fix(rapier): clean up framestepper; remove afterNextRender and autoEffect

Author: nartc

libs/rapier/src/lib/frame-stepper.ts

@@ -1,6 +1,5 @@
-import { afterNextRender, Directive, input } from '@angular/core';
-import { injectBeforeRender } from 'angular-three';
-import { injectAutoEffect } from 'ngxtension/auto-effect';
+import { Directive, effect, input } from '@angular/core';
+import { injectStore } from 'angular-three';
 import { NgtrPhysicsOptions } from './types';
 
 @Directive({ standalone: true, selector: 'ngtr-frame-stepper' })
@@ -11,39 +10,37 @@ export class NgtrFrameStepper {
 	type = input.required<NgtrPhysicsOptions['updateLoop']>();
 
 	constructor() {
-		const autoEffect = injectAutoEffect();
+		const store = injectStore();
 
-		afterNextRender(() => {
-			autoEffect((injector) => {
-				const ready = this.ready();
-				if (!ready) return;
+		effect((onCleanup) => {
+			const ready = this.ready();
+			if (!ready) return;
 
-				const [type, updatePriority, stepFn] = [this.type(), this.updatePriority(), this.stepFn()];
-				if (type === 'follow') {
-					return injectBeforeRender(
-						({ delta }) => {
-							stepFn(delta);
-						},
-						{ priority: updatePriority, injector },
-					);
-				}
-
-				let lastFrame = 0;
-				let raf: ReturnType<typeof requestAnimationFrame> = 0;
-				const loop = () => {
-					const now = performance.now();
-					const delta = now - lastFrame;
-					raf = requestAnimationFrame(loop);
-					stepFn(delta);
-					lastFrame = now;
-				};
+			const [type, updatePriority, stepFn] = [this.type(), this.updatePriority(), this.stepFn()];
+			if (type === 'follow') {
+				const cleanup = store.snapshot.internal.subscribe(
+					({ delta }) => {
+						stepFn(delta);
+					},
+					updatePriority,
+					store,
+				);
+				onCleanup(() => cleanup());
+				return;
+			}
 
+			let lastFrame = 0;
+			let raf: ReturnType<typeof requestAnimationFrame> = 0;
+			const loop = () => {
+				const now = performance.now();
+				const delta = now - lastFrame;
 				raf = requestAnimationFrame(loop);
+				stepFn(delta);
+				lastFrame = now;
+			};
 
-				return () => {
-					cancelAnimationFrame(raf);
-				};
-			});
+			raf = requestAnimationFrame(loop);
+			onCleanup(() => cancelAnimationFrame(raf));
 		});
 	}
 }