diScope

index.ts

@@ -96,6 +96,21 @@ export const diExists = () => (als.getStore() == null) === false
 
 export const diContext = (): AlsContext => ({ deps: new Map(), once: new Map(), state: new Map() })
 
+export const diScope = <T extends { [key: string]: any }>(scope: T, init?: () => void): T => {
+  const ctx = diContext()
+  if (init) {
+    diInit(init, ctx)
+  }
+  return Object.fromEntries(
+    Object.entries(scope).map(([key, fn]) => [
+      key,
+      function (this: unknown, ...args: unknown[]) {
+        return diInit(() => fn.apply(this, args), ctx)
+      },
+    ])
+  ) as T
+}
+
 const storeOrError = () => {
   const store = als.getStore()
 

test.ts

@@ -16,6 +16,7 @@ import {
   clearGlobalState,
   diHas,
   diContext,
+  diScope,
 } from './index.js'
 import EventEmitter from 'events'
 
@@ -344,3 +345,22 @@ test('diContext', () => {
     state: new Map(),
   })
 })
+
+test('diScope', () => {
+  const inc = dis((resp: number, n: number) => resp + n, 0)
+  const checkScope = di((scope: string) => diDep<boolean>(scope))
+
+  const scope1 = diScope({ inc, checkScope }, () => diSet('scope1', true))
+  const scope2 = diScope({ inc, checkScope }, () => diSet('scope2', true))
+
+  assert.strictEqual(scope1.checkScope('scope1'), true)
+  assert.rejects(async () => scope1.checkScope('scope2'), new Error('Dependency with key scope2 not registered!'))
+  assert.strictEqual(scope2.checkScope('scope2'), true)
+  assert.rejects(async () => scope2.checkScope('scope1'), new Error('Dependency with key scope1 not registered!'))
+
+  scope1.inc(2)
+  scope2.inc(5)
+
+  assert.strictEqual(scope1.inc(), 2)
+  assert.strictEqual(scope2.inc(), 5)
+})