feat: add createRunInContext

Author: nartc

libs/angular-three/src/index.ts

@@ -3,6 +3,7 @@ export * from './lib/di/before-render';
 export * from './lib/di/catalogue';
 export * from './lib/di/destroy';
 export * from './lib/di/ref';
+export * from './lib/di/run-in-context';
 export * from './lib/directives/args';
 export * from './lib/directives/repeat';
 export * from './lib/loader';

libs/angular-three/src/lib/di/run-in-context.ts

@@ -0,0 +1,22 @@
+import { EnvironmentInjector, inject, Injector } from '@angular/core';
+
+export function createRunInContext() {
+    const nodeInjector = inject(Injector);
+    const envInjector = inject(EnvironmentInjector);
+
+    const originalGet = envInjector.get;
+
+    return <TReturn>(cb: () => TReturn): TReturn => {
+        envInjector.get = (...args: Parameters<EnvironmentInjector['get']>) => {
+            try {
+                const fromNodeInjector = nodeInjector.get(...(args as Parameters<Injector['get']>));
+                if (fromNodeInjector) return fromNodeInjector;
+                return originalGet(...args);
+            } catch (e) {
+                return originalGet(...args);
+            }
+        };
+
+        return envInjector.runInContext(cb);
+    };
+}