feat: add setVelocity method

scripts/analyze.sh

@@ -1,4 +1,4 @@
-curl -o scripts/roblox.d.lua https://raw.githubusercontent.com/JohnnyMorganz/luau-lsp/main/scripts/globalTypes.d.luau
+curl -o scripts/roblox.d.luau https://raw.githubusercontent.com/JohnnyMorganz/luau-lsp/main/scripts/globalTypes.d.luau
 
 rojo sourcemap test.project.json -o sourcemap.json
 

src/createMotion.luau

@@ -107,7 +107,7 @@ local function createMotion<T>(initialValue: T, options: types.MotionOptions?):
 				end
 
 				state.complete = false
-				state.velocity = state.velocity :: number + impulses
+				state.velocity += impulses
 			end
 
 			return
@@ -123,10 +123,23 @@ local function createMotion<T>(initialValue: T, options: types.MotionOptions?):
 			end
 
 			state.complete = false
-			state.velocity = state.velocity :: number + amount
+			state.velocity += amount
 		end
+	end
+
+	local function setVelocity(_self, velocity)
+		local intermediates = intermediate.to(velocity)
+
+		for key, amount in intermediates do
+			local state = motionState[key]
+
+			if not state or not state.velocity then
+				continue
+			end
 
-		return
+			state.complete = false
+			state.velocity = amount
+		end
 	end
 
 	local function to(_self, solvers)
@@ -268,10 +281,11 @@ local function createMotion<T>(initialValue: T, options: types.MotionOptions?):
 		start = start,
 		stop = stop,
 		get = get,
-		getVelocity = getVelocity,
 		set = set,
-		patch = patch,
+		getVelocity = getVelocity,
+		setVelocity = setVelocity,
 		impulse = impulse,
+		patch = patch,
 		to = to,
 		immediate = immediate,
 		linear = linear,

src/createMotion.spec.luau

@@ -12,6 +12,8 @@ return function()
 		expect(motion.stop).to.be.a("function")
 		expect(motion.get).to.be.a("function")
 		expect(motion.set).to.be.a("function")
+		expect(motion.getVelocity).to.be.a("function")
+		expect(motion.setVelocity).to.be.a("function")
 		expect(motion.patch).to.be.a("function")
 		expect(motion.impulse).to.be.a("function")
 		expect(motion.to).to.be.a("function")

src/index.d.ts

@@ -81,8 +81,9 @@ declare namespace Ripple {
 		start(): Cleanup;
 		stop(): void;
 		get(): T;
-		getVelocity(): T;
 		set(value: PartialMotionGoal<T>): void;
+		getVelocity(): T;
+		setVelocity(velocity: PartialMotionGoal<T>): void;
 		impulse(velocity: PartialMotionGoal<T>): void;
 		to(solver: MotionSolver<T> | MapSolvers<PartialMotionGoal<T>>): void;
 		immediate(goal: PartialMotionGoal<T>): void;

src/types.luau

@@ -41,10 +41,11 @@ export type Motion<T = number> = {
 	start: (self: Motion<T>) -> Cleanup,
 	stop: (self: Motion<T>) -> (),
 	get: (self: Motion<T>) -> T,
-	getVelocity: (self: Motion<T>) -> T,
 	set: (self: Motion<T>, value: T | { [any]: number }) -> (),
-	patch: (self: Motion<T>, patch: { [any]: Partial<MotionState> }) -> (),
+	getVelocity: (self: Motion<T>) -> T,
+	setVelocity: (self: Motion<T>, value: T | { [any]: number }) -> (),
 	impulse: (self: Motion<T>, impulse: T | { [any]: number }) -> (),
+	patch: (self: Motion<T>, patch: { [any]: Partial<MotionState> }) -> (),
 	to: (self: Motion<T>, goal: MotionSolver | { [any]: MotionSolver }) -> (),
 	immediate: (self: Motion<T>, goal: T) -> (),
 	spring: (self: Motion<T>, goal: T, options: SpringOptions?) -> (),