angular-three 4.0.0-next.13 → 4.0.0-next.14

package/fesm2022/angular-three.mjs

@@ -3008,32 +3008,34 @@ i0.ɵɵngDeclareClassMetadata({ minVersion: "12.0.0", version: "19.1.4", ngImpor
  * `injectBeforeRender` invokes its callback on every frame. Hence, the notion of tracking
  * changes (i.e: signals) does not really matter since we're getting latest values of the things we need on every frame anyway.
  *
- * If `priority` is dynamic, consumers should set up `injectBeforeRender` in
- * an `effect` and track `priority` changes. Make use of `onCleanup` to clean up
- * previous before render subscription
+ * If `priority` is a Signal, `injectBeforeRender` will set up an Effect internally and returns the `EffectRef#destroy` instead.
  *
  * @example
  * ```ts
- * const injector = inject(Injector);
- *
- * effect((onCleanup) => {
- *   const priority = this.priority(); // track priority
- *
- *   const sub = injectBeforeRender(
- *    ({ gl, camera }) => {
- *      // before render logic
- *    },
- *    {
- *      priority,
- *      injector, // injector is needed since injectBeforeRender is invoked in effect body
- *    }
- *   });
- *
- *   onCleanup(() => sub());
- * });
+ * const destroy = injectBeforeRender(
+ *  ({ gl, camera }) => {
+ *    // before render logic
+ *  },
+ *  {
+ *    priority: this.priority, // this.priority is a Signal<number>
+ *  }
+ * )
  * ```
  */
 function injectBeforeRender(cb, { priority = 0, injector } = {}) {
+    if (typeof priority === 'function') {
+        const effectRef = assertInjector(injectBeforeRender, injector, () => {
+            const store = injectStore();
+            const ref = effect((onCleanup) => {
+                const p = priority();
+                const sub = store.snapshot.internal.subscribe(cb, p, store);
+                onCleanup(() => sub());
+            });
+            inject(DestroyRef).onDestroy(() => void ref.destroy());
+            return ref;
+        });
+        return effectRef.destroy.bind(effectRef);
+    }
     return assertInjector(injectBeforeRender, injector, () => {
         const store = injectStore();
         const sub = store.snapshot.internal.subscribe(cb, priority, store);