@cnstra/core 1.0.0 → 1.1.0

package/README.md

@@ -44,7 +44,7 @@ const userService = neuron('user-service', {
 })
 .dendrite({
   collateral: userCreated,
-  reaction: async (payload, axon) => {
+  response: async (payload, axon) => {
     const userData = payload as { id: string; name: string };
     
     // Process the user creation
@@ -59,10 +59,10 @@ const userService = neuron('user-service', {
 });
 
 // Create the CNS system
-const cns = new CNS({ userCreated }, [userService]);
+const cns = new CNS([userService]);
 
 // Stimulate the system
-await cns.stimulate('userCreated', {
+await cns.stimulate(userCreated, {
   id: '123',
   name: 'John Doe'
 });
@@ -97,8 +97,9 @@ Adds a dendrite (input receptor) to a neuron. Returns the neuron for chaining.
 myNeuron
   .dendrite({
     collateral: inputCollateral,
-    reaction: async (payload, axon) => {
+    response: async (payload, axon, ctx) => {
       // Process input and return output signal
+      // ctx parameter provides local context storage for this neuron
       return axon.output.createSignal(result);
     }
   });
@@ -110,21 +111,20 @@ The main orchestrator that manages signal flow between neurons.
 
 #### Constructor
 ```typescript
-new CNS(afferentAxon, neurons)
+new CNS(neurons)
 ```
 
 **Parameters:**
-- `afferentAxon`: Object containing collaterals that can be used as input triggers
 - `neurons`: Array of neurons that process signals
 
 #### `stimulate()` Method
 ```typescript
-await cns.stimulate(axonKey, payload, options?)
+await cns.stimulate(collateral, payload, options?)
 ```
 
 **Parameters:**
-- `axonKey`: Key from the afferent axon to trigger
-- `payload`: Signal payload with type and data
+- `collateral`: The collateral instance to trigger
+- `payload`: Signal payload data
 - `options`: Optional configuration object
 
 **Options:**
@@ -143,7 +143,7 @@ const processor = neuron('processor', {
 })
 .dendrite({
   collateral: inputCollateral,
-  reaction: async (payload, axon) => {
+  response: async (payload, axon) => {
     // This signal is created but NOT processed
     axon.output.createSignal({ message: 'Hello' });
     
@@ -160,7 +160,7 @@ Multiple neurons can listen to the same collateral:
 const emailService = neuron('email-service', { emailSent })
   .dendrite({
     collateral: userCreated,
-    reaction: async (payload, axon) => {
+    response: async (payload, axon) => {
       return axon.emailSent.createSignal({ to: 'user@example.com' });
     }
   });
@@ -168,13 +168,13 @@ const emailService = neuron('email-service', { emailSent })
 const notificationService = neuron('notification-service', { notificationSent })
   .dendrite({
     collateral: userCreated,
-    reaction: async (payload, axon) => {
+    response: async (payload, axon) => {
       return axon.notificationSent.createSignal({ message: 'User created' });
     }
   });
 
 // Both neurons will process the userCreated signal
-const cns = new CNS({ userCreated }, [emailService, notificationService]);
+const cns = new CNS([emailService, notificationService]);
 ```
 
 ### Conditional Logic
@@ -185,7 +185,7 @@ const router = neuron('router', {
 })
 .dendrite({
   collateral: requestCollateral,
-  reaction: async (payload, axon) => {
+  response: async (payload, axon) => {
     try {
       const result = await processRequest(payload);
       return axon.success.createSignal(result);
@@ -196,6 +196,26 @@ const router = neuron('router', {
 });
 ```
 
+### Context Management
+
+Neurons can maintain local state using the context parameter:
+
+```typescript
+const statefulNeuron = neuron('stateful', { output: outputCollateral })
+  .dendrite({
+    collateral: inputCollateral,
+    response: async (payload, axon, ctx) => {
+      // Get current state
+      const currentState = ctx.get();
+      
+      // Update state
+      ctx.set({ count: (currentState?.count || 0) + 1 });
+      
+      return axon.output.createSignal({ count: ctx.get()?.count });
+    }
+  });
+```
+
 ## 🧪 Testing
 
 ```bash

package/dist/index.cjs

@@ -1,37 +1,22 @@
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+'use strict';
+
+// src/CNSStimulationContextStore.ts
+var CNSStimulationContextStore = class {
+  constructor(ctx = /* @__PURE__ */ new Map()) {
+    this.ctx = ctx;
+  }
+  get(key) {
+    return this.ctx.get(key);
+  }
+  set(key, value) {
+    this.ctx.set(key, value);
   }
-  return to;
 };
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-
-// src/index.ts
-var index_exports = {};
-__export(index_exports, {
-  CNS: () => CNS,
-  CNSCollateral: () => CNSCollateral,
-  collateral: () => collateral,
-  neuron: () => neuron
-});
-module.exports = __toCommonJS(index_exports);
 
 // src/CNS.ts
 var asArray = (x) => Array.isArray(x) ? x : [x];
 var CNS = class {
-  constructor(afferentLayerAsAxon, neurons) {
-    this.afferentLayerAsAxon = afferentLayerAsAxon;
+  constructor(neurons) {
     this.neurons = neurons;
     // index by collateral *type* => list of subscribers
     this.subIndex = /* @__PURE__ */ new Map();
@@ -51,20 +36,23 @@ var CNS = class {
       }
     }
   }
-  async fanOut(incoming, allowType, seen, queue) {
-    const subscribers = this.subIndex.get(incoming.edgeId);
+  async fanOut(incoming, allowType, seen, queue, ctx) {
+    const subscribers = this.subIndex.get(incoming.collateralId);
     if (!subscribers || subscribers.length === 0) return;
     for (const { neuron: neuron2, dendrite } of subscribers) {
-      const k = `${neuron2.id}::${incoming.edgeId}::${incoming.spikeId}`;
+      const k = `${neuron2.id}::${incoming.collateralId}::${incoming.spikeId}`;
       if (seen.has(k)) continue;
       seen.add(k);
-      const out = await dendrite.reaction(incoming.payload, neuron2.axon);
+      const out = await dendrite.response(incoming.payload, neuron2.axon, {
+        get: () => ctx.get(neuron2.id),
+        set: (value) => ctx.set(neuron2.id, value)
+      });
       if (out === void 0) continue;
       for (const spike of asArray(out)) {
         if (allowType && !allowType(spike.type))
           continue;
         queue.push({
-          edgeId: spike.type,
+          collateralId: spike.type,
           payload: spike.payload,
           hops: incoming.hops + 1,
           spikeId: incoming.spikeId
@@ -73,13 +61,13 @@ var CNS = class {
       }
     }
   }
-  async stimulate(axonKey, payload, opts) {
-    const afferent = this.afferentLayerAsAxon[axonKey];
+  async stimulate(collateral2, payload, opts) {
+    const ctx = opts?.ctx ?? new CNSStimulationContextStore();
     const spikeId = opts?.spikeId || Math.random().toString(36).slice(2);
     const maxHops = opts?.maxHops ?? 1e3;
     const queue = [
       {
-        edgeId: afferent.id,
+        collateralId: collateral2.id,
         payload,
         hops: 0,
         spikeId
@@ -90,7 +78,7 @@ var CNS = class {
       if (opts?.abortSignal?.aborted) break;
       const item = queue.shift();
       opts?.onTrace?.({
-        edgeId: item.edgeId,
+        collateralId: item.collateralId,
         hops: item.hops,
         payload: item.payload
       });
@@ -99,7 +87,8 @@ var CNS = class {
         item,
         opts?.allowType,
         seen,
-        queue
+        queue,
+        ctx
       );
     }
   }
@@ -125,16 +114,72 @@ var neuron = (id, axon) => {
     id,
     dendrites,
     dendrite(s) {
-      dendrites.push(s);
+      dendrites.push(
+        s
+      );
       return api;
     }
   };
   return api;
 };
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  CNS,
-  CNSCollateral,
-  collateral,
-  neuron
+var withCtx = () => ({
+  neuron: (id, axon) => {
+    return neuron(id, axon);
+  }
+});
+var request = collateral("request");
+var success = collateral("success");
+var error = collateral("error");
+neuron("router", { success, error }).dendrite({
+  collateral: request,
+  response: async (payload, axon) => {
+    const value = payload.value;
+    if (value > 0) {
+      return axon.success.createSignal({
+        result: `Success: ${value}`
+      });
+    } else {
+      return axon.error.createSignal({
+        error: `Error: ${value} is not positive`
+      });
+    }
+  }
+});
+var a1 = {
+  test: collateral("test")
+};
+var a2 = {
+  test2: collateral("test2")
+};
+var a3 = {
+  test3: collateral("test3")
+};
+var n1 = withCtx().neuron("n1", a1).dendrite({
+  collateral: a2.test2,
+  response: async (payload, axon, ctx) => {
+    ctx.set({
+      testCtx: payload.test2
+    });
+    ctx.get()?.testCtx;
+    return axon.test.createSignal({
+      test: payload.test2
+    });
+  }
 });
+var n2 = neuron("n2", a2).dendrite({
+  collateral: a3.test3,
+  response: async (payload, axon) => {
+    return axon.test2.createSignal({
+      test2: payload.test3
+    });
+  }
+});
+new CNS([n1, n2]);
+
+exports.CNS = CNS;
+exports.CNSCollateral = CNSCollateral;
+exports.collateral = collateral;
+exports.neuron = neuron;
+exports.withCtx = withCtx;
+//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/CNSStimulationContextStore.ts","../src/CNS.ts","../src/CNSCollateral.ts","../src/factory/index.ts"],"names":["neuron","collateral"],"mappings":";;;AAEO,IAAM,6BAAN,MAAwE;AAAA,EAC3E,WAAA,CAA6B,GAAA,mBAA4B,IAAI,GAAA,EAAI,EAAG;AAAvC,IAAA,IAAA,CAAA,GAAA,GAAA,GAAA;AAAA,EAAwC;AAAA,EAErE,IAAI,GAAA,EAAsB;AACtB,IAAA,OAAO,IAAA,CAAK,GAAA,CAAI,GAAA,CAAI,GAAG,CAAA;AAAA,EAC3B;AAAA,EAEA,GAAA,CAAI,KAAa,KAAA,EAAsB;AACnC,IAAA,IAAA,CAAK,GAAA,CAAI,GAAA,CAAI,GAAA,EAAK,KAAK,CAAA;AAAA,EAC3B;AACJ,CAAA;;;ACNA,IAAM,OAAA,GAAU,CAAI,CAAA,KAAgB,KAAA,CAAM,QAAQ,CAAC,CAAA,GAAI,CAAA,GAAI,CAAC,CAAC,CAAA;AAiBtD,IAAM,MAAN,MAIL;AAAA,EAOE,YAA+B,OAAA,EAAoB;AAApB,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA;AAL/B;AAAA,IAAA,IAAA,CAAQ,QAAA,uBAAe,GAAA,EAGrB;AAGE,IAAA,IAAA,CAAK,UAAA,EAAW;AAAA,EACpB;AAAA,EAEQ,UAAA,GAAa;AACjB,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AACpB,IAAA,KAAA,MAAWA,OAAAA,IAAU,KAAK,OAAA,EAAS;AAC/B,MAAA,KAAA,MAAW,QAAA,IAAYA,QAAO,SAAA,EAAW;AACrC,QAAA,MAAM,GAAA,GAAM,SAAS,UAAA,CAAW,EAAA;AAChC,QAAA,MAAM,MAAM,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,GAAG,KAAK,EAAC;AACvC,QAAA,GAAA,CAAI,IAAA,CAAK,EAAE,MAAA,EAAAA,OAAAA,EAAQ,UAAiC,CAAA;AACpD,QAAA,IAAA,CAAK,QAAA,CAAS,GAAA;AAAA,UACV,GAAA;AAAA,UACA;AAAA,SACJ;AAAA,MACJ;AAAA,IACJ;AAAA,EACJ;AAAA,EAEA,MAAc,MAAA,CACV,QAAA,EACA,SAAA,EACA,IAAA,EACA,OACA,GAAA,EACF;AACE,IAAA,MAAM,WAAA,GAAc,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,SAAS,YAAY,CAAA;AAC3D,IAAA,IAAI,CAAC,WAAA,IAAe,WAAA,CAAY,MAAA,KAAW,CAAA,EAAG;AAE9C,IAAA,KAAA,MAAW,EAAE,MAAA,EAAAA,OAAAA,EAAQ,QAAA,MAAc,WAAA,EAAa;AAC5C,MAAA,MAAM,CAAA,GAAI,GAAGA,OAAAA,CAAO,EAAE,KAAK,QAAA,CAAS,YAAY,CAAA,EAAA,EAAK,QAAA,CAAS,OAAO,CAAA,CAAA;AACrE,MAAA,IAAI,IAAA,CAAK,GAAA,CAAI,CAAC,CAAA,EAAG;AACjB,MAAA,IAAA,CAAK,IAAI,CAAC,CAAA;AAEV,MAAA,MAAM,MAAM,MAAM,QAAA,CAAS,SAAS,QAAA,CAAS,OAAA,EAASA,QAAO,IAAA,EAAM;AAAA,QAC/D,GAAA,EAAK,MAAM,GAAA,CAAI,GAAA,CAAIA,QAAO,EAAE,CAAA;AAAA,QAC5B,KAAK,CAAA,KAAA,KAAS,GAAA,CAAI,GAAA,CAAIA,OAAAA,CAAO,IAAI,KAAK;AAAA,OACzC,CAAA;AACD,MAAA,IAAI,QAAQ,MAAA,EAAW;AAEvB,MAAA,KAAA,MAAW,KAAA,IAAS,OAAA,CAAQ,GAAG,CAAA,EAAG;AAC9B,QAAA,IAAI,SAAA,IAAa,CAAC,SAAA,CAAU,KAAA,CAAM,IAAqB,CAAA;AACnD,UAAA;AAEJ,QAAA,KAAA,CAAM,IAAA,CAAK;AAAA,UACP,cAAc,KAAA,CAAM,IAAA;AAAA,UACpB,SAAS,KAAA,CAAM,OAAA;AAAA,UACf,IAAA,EAAM,SAAS,IAAA,GAAO,CAAA;AAAA,UACtB,SAAS,QAAA,CAAS;AAAA;AAAA,SACrB,CAAA;AAAA,MACL;AAAA,IACJ;AAAA,EACJ;AAAA,EAEA,MAAM,SAAA,CAIFC,WAAAA,EAIA,OAAA,EACA,IAAA,EAYa;AACb,IAAA,MAAM,GAAA,GAAM,IAAA,EAAM,GAAA,IAAO,IAAI,0BAAA,EAA2B;AACxD,IAAA,MAAM,OAAA,GAAU,IAAA,EAAM,OAAA,IAAW,IAAA,CAAK,MAAA,GAAS,QAAA,CAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAC,CAAA;AACnE,IAAA,MAAM,OAAA,GAAU,MAAM,OAAA,IAAW,GAAA;AAEjC,IAAA,MAAM,KAAA,GAAqC;AAAA,MACvC;AAAA,QACI,cAAcA,WAAAA,CAAW,EAAA;AAAA,QACzB,OAAA;AAAA,QACA,IAAA,EAAM,CAAA;AAAA,QACN;AAAA;AACJ,KACJ;AACA,IAAA,MAAM,IAAA,uBAAW,GAAA,EAAY;AAE7B,IAAA,OAAO,MAAM,MAAA,EAAQ;AACjB,MAAA,IAAI,IAAA,EAAM,aAAa,OAAA,EAAS;AAEhC,MAAA,MAAM,IAAA,GAAO,MAAM,KAAA,EAAM;AACzB,MAAA,IAAA,EAAM,OAAA,GAAU;AAAA,QACZ,cAAc,IAAA,CAAK,YAAA;AAAA,QACnB,MAAM,IAAA,CAAK,IAAA;AAAA,QACX,SAAS,IAAA,CAAK;AAAA,OACjB,CAAA;AACD,MAAA,IAAI,IAAA,CAAK,QAAQ,OAAA,EAAS;AAE1B,MAAA,MAAM,IAAA,CAAK,MAAA;AAAA,QACP,IAAA;AAAA,QACA,IAAA,EAAM,SAAA;AAAA,QACN,IAAA;AAAA,QACA,KAAA;AAAA,QACA;AAAA,OACJ;AAAA,IACJ;AAAA,EACJ;AACJ;;;AC9IO,IAAM,gBAAN,MAAkD;AAAA,EACrD,YAA4B,EAAA,EAAS;AAAT,IAAA,IAAA,CAAA,EAAA,GAAA,EAAA;AAAA,EAAU;AAAA,EAEtC,gBACO,IAAA,EACsB;AACzB,IAAA,MAAM,OAAA,GAAW,KAAa,CAAC,CAAA;AAC/B,IAAA,OAAO,EAAE,IAAA,EAAM,IAAA,CAAK,EAAA,EAAI,OAAA,EAAQ;AAAA,EACpC;AACJ;;;ACNO,IAAM,UAAA,GAAa,CACtB,EAAA,KACC,IAAI,cAA6B,EAAE;AA4CjC,IAAM,MAAA,GAAS,CAUlB,EAAA,EACA,IAAA,KAOC;AACD,EAAA,MAAM,YAOA,EAAC;AAEP,EAAA,MAAM,GAAA,GAMF;AAAA,IACA,IAAA;AAAA,IACA,EAAA;AAAA,IACA,SAAA;AAAA,IACA,SAII,CAAA,EAQF;AACE,MAAA,SAAA,CAAU,IAAA;AAAA,QACN;AAAA,OAQJ;AACA,MAAA,OAAO,GAAA;AAAA,IACX;AAAA,GACJ;AAEA,EAAA,OAAO,GAAA;AACX;AAEO,IAAM,UAAU,OAAsB;AAAA,EACzC,MAAA,EAAQ,CASJ,EAAA,EACA,IAAA,KACC;AACD,IAAA,OAAO,MAAA,CAML,IAAI,IAAI,CAAA;AAAA,EACd;AACJ,CAAA;AAEA,IAAM,OAAA,GAAU,WAA8B,SAAS,CAAA;AACvD,IAAM,OAAA,GAAU,WAA+B,SAAS,CAAA;AACxD,IAAM,KAAA,GAAQ,WAA8B,OAAO,CAAA;AAEpC,OAAO,QAAA,EAAU,EAAE,SAAS,KAAA,EAAO,EAAE,QAAA,CAAS;AAAA,EACzD,UAAA,EAAY,OAAA;AAAA,EACZ,QAAA,EAAU,OAAO,OAAA,EAAS,IAAA,KAAS;AAC/B,IAAA,MAAM,QAAS,OAAA,CAA8B,KAAA;AAC7C,IAAA,IAAI,QAAQ,CAAA,EAAG;AACX,MAAA,OAAO,IAAA,CAAK,QAAQ,YAAA,CAAa;AAAA,QAC7B,MAAA,EAAQ,YAAY,KAAK,CAAA;AAAA,OAC5B,CAAA;AAAA,IACL,CAAA,MAAO;AACH,MAAA,OAAO,IAAA,CAAK,MAAM,YAAA,CAAa;AAAA,QAC3B,KAAA,EAAO,UAAU,KAAK,CAAA,gBAAA;AAAA,OACzB,CAAA;AAAA,IACL;AAAA,EACJ;AACJ,CAAC;AAED,IAAM,EAAA,GAAK;AAAA,EACP,IAAA,EAAM,WAA6B,MAAM;AAC7C,CAAA;AAEA,IAAM,EAAA,GAAK;AAAA,EACP,KAAA,EAAO,WAA8B,OAAO;AAChD,CAAA;AAEA,IAAM,EAAA,GAAK;AAAA,EACP,KAAA,EAAO,WAA8B,OAAO;AAChD,CAAA;AAEA,IAAM,KAAK,OAAA,EAA6B,CACnC,OAAO,IAAA,EAAM,EAAE,EACf,QAAA,CAAS;AAAA,EACN,YAAY,EAAA,CAAG,KAAA;AAAA,EACf,QAAA,EAAU,OAAO,OAAA,EAAS,IAAA,EAAM,GAAA,KAAQ;AACpC,IAAA,GAAA,CAAI,GAAA,CAAI;AAAA,MACJ,SAAS,OAAA,CAAQ;AAAA,KACpB,CAAA;AACD,IAAA,GAAA,CAAI,KAAI,EAAG,OAAA;AACX,IAAA,OAAO,IAAA,CAAK,KAAK,YAAA,CAAa;AAAA,MAC1B,MAAM,OAAA,CAAQ;AAAA,KACjB,CAAA;AAAA,EACL;AACJ,CAAC,CAAA;AAEL,IAAM,EAAA,GAAK,MAAA,CAAO,IAAA,EAAM,EAAE,EAAE,QAAA,CAAS;AAAA,EACjC,YAAY,EAAA,CAAG,KAAA;AAAA,EACf,QAAA,EAAU,OAAO,OAAA,EAAS,IAAA,KAAS;AAC/B,IAAA,OAAO,IAAA,CAAK,MAAM,YAAA,CAAa;AAAA,MAC3B,OAAO,OAAA,CAAQ;AAAA,KAClB,CAAA;AAAA,EACL;AACJ,CAAC,CAAA;AAEW,IAAI,GAAA,CAAI,CAAC,EAAA,EAAI,EAAE,CAAC","file":"index.cjs","sourcesContent":["import { ICNSStimulationContextStore } from './interfaces/ICNSStimulationContextStore';\n\nexport class CNSStimulationContextStore implements ICNSStimulationContextStore {\n    constructor(private readonly ctx: Map<string, unknown> = new Map()) {}\n\n    get(key: string): unknown {\n        return this.ctx.get(key);\n    }\n\n    set(key: string, value: unknown): void {\n        this.ctx.set(key, value);\n    }\n}\n","import { TCNSNeuron } from './types/TCNSNeuron';\nimport { TCNSDendrite } from './types/TCNSDendrite';\nimport { CNSCollateral } from './CNSCollateral';\nimport { ICNSStimulationContextStore } from './interfaces/ICNSStimulationContextStore';\nimport { CNSStimulationContextStore } from './CNSStimulationContextStore';\n\nconst asArray = <T>(x: T | T[]) => (Array.isArray(x) ? x : [x]);\n\ntype TSubscriber<\n    TNeuron extends TCNSNeuron<any, any, any, any, any, any, any>,\n    TDendrite extends TCNSDendrite<any, any, any, any, any, any>\n> = {\n    neuron: TNeuron;\n    dendrite: TDendrite;\n};\n\ntype TQueueItem<TCollateralId extends string> = {\n    collateralId: TCollateralId;\n    payload: unknown;\n    hops: number;\n    spikeId: string;\n};\n\nexport class CNS<\n    TCollateralId extends string,\n    TNeuron extends TCNSNeuron<any, any, any, any, any, any, any>,\n    TDendrite extends TCNSDendrite<any, any, any, any, any, any>\n> {\n    // index by collateral *type* => list of subscribers\n    private subIndex = new Map<\n        TCollateralId,\n        TSubscriber<TNeuron, TDendrite>[]\n    >();\n\n    constructor(protected readonly neurons: TNeuron[]) {\n        this.buildIndex();\n    }\n\n    private buildIndex() {\n        this.subIndex.clear();\n        for (const neuron of this.neurons) {\n            for (const dendrite of neuron.dendrites) {\n                const key = dendrite.collateral.id;\n                const arr = this.subIndex.get(key) ?? [];\n                arr.push({ neuron, dendrite: dendrite as TDendrite });\n                this.subIndex.set(\n                    key,\n                    arr as TSubscriber<TNeuron, TDendrite>[]\n                );\n            }\n        }\n    }\n\n    private async fanOut(\n        incoming: TQueueItem<TCollateralId>,\n        allowType: ((t: TCollateralId) => boolean) | undefined,\n        seen: Set<string>,\n        queue: TQueueItem<TCollateralId>[],\n        ctx: ICNSStimulationContextStore\n    ) {\n        const subscribers = this.subIndex.get(incoming.collateralId);\n        if (!subscribers || subscribers.length === 0) return;\n\n        for (const { neuron, dendrite } of subscribers) {\n            const k = `${neuron.id}::${incoming.collateralId}::${incoming.spikeId}`;\n            if (seen.has(k)) continue;\n            seen.add(k);\n\n            const out = await dendrite.response(incoming.payload, neuron.axon, {\n                get: () => ctx.get(neuron.id),\n                set: value => ctx.set(neuron.id, value),\n            });\n            if (out === undefined) continue;\n\n            for (const spike of asArray(out)) {\n                if (allowType && !allowType(spike.type as TCollateralId))\n                    continue; // TODO: add allowType\n\n                queue.push({\n                    collateralId: spike.type as TCollateralId,\n                    payload: spike.payload,\n                    hops: incoming.hops + 1,\n                    spikeId: incoming.spikeId, // keep the same spike id across the cascade\n                });\n            }\n        }\n    }\n\n    async stimulate<\n        TAfferentCollateralId extends TCollateralId,\n        TAfferentCollateralPayload\n    >(\n        collateral: CNSCollateral<\n            TAfferentCollateralId,\n            TAfferentCollateralPayload\n        >,\n        payload: TAfferentCollateralPayload,\n        opts?: {\n            maxHops?: number;\n            allowType?: (t: TCollateralId) => boolean;\n            onTrace?: (e: {\n                collateralId: TCollateralId;\n                hops: number;\n                payload: unknown;\n            }) => void;\n            abortSignal?: AbortSignal;\n            spikeId?: string;\n            ctx?: ICNSStimulationContextStore;\n        }\n    ): Promise<void> {\n        const ctx = opts?.ctx ?? new CNSStimulationContextStore();\n        const spikeId = opts?.spikeId || Math.random().toString(36).slice(2);\n        const maxHops = opts?.maxHops ?? 1000;\n\n        const queue: TQueueItem<TCollateralId>[] = [\n            {\n                collateralId: collateral.id,\n                payload,\n                hops: 0,\n                spikeId,\n            },\n        ];\n        const seen = new Set<string>();\n\n        while (queue.length) {\n            if (opts?.abortSignal?.aborted) break;\n\n            const item = queue.shift()!;\n            opts?.onTrace?.({\n                collateralId: item.collateralId,\n                hops: item.hops,\n                payload: item.payload,\n            });\n            if (item.hops >= maxHops) continue;\n\n            await this.fanOut(\n                item as TQueueItem<TCollateralId>,\n                opts?.allowType,\n                seen,\n                queue as TQueueItem<TCollateralId>[],\n                ctx\n            );\n        }\n    }\n}\n","import { TCNSSignal } from './types/TCNSSignal';\n\nexport class CNSCollateral<TId extends string, TPayload> {\n    constructor(public readonly id: TId) {}\n\n    createSignal(\n        ...args: [TPayload] extends [undefined] ? [] : [payload: TPayload]\n    ): TCNSSignal<TId, TPayload> {\n        const payload = (args as any)[0] as TPayload;\n        return { type: this.id, payload };\n    }\n}\n","import { CNS } from '../CNS';\nimport { CNSCollateral } from '../CNSCollateral';\nimport { TCNSAxon } from '../types/TCNSAxon';\nimport { TCNSDendrite } from '../types/TCNSDendrite';\n\nexport const collateral = <TPayload = undefined, TId extends string = string>(\n    id: TId\n) => new CNSCollateral<TId, TPayload>(id);\n\ntype InterNeuronAPI<\n    TContextValue,\n    TIdType extends string,\n    TReceiverCollateralIdType extends string,\n    TReceiverAxonCollateralPayload,\n    TAxonType extends TCNSAxon<\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload\n    >\n> = {\n    id: TIdType;\n    axon: TAxonType;\n    dendrites: TCNSDendrite<\n        TContextValue,\n        string,\n        unknown,\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload,\n        TAxonType\n    >[];\n    dendrite: <\n        TSenderExactCollateralIdType extends string,\n        TSenderExactAxonCollateralPayload\n    >(\n        s: TCNSDendrite<\n            TContextValue,\n            TSenderExactCollateralIdType,\n            TSenderExactAxonCollateralPayload,\n            TReceiverCollateralIdType,\n            TReceiverAxonCollateralPayload,\n            TAxonType\n        >\n    ) => InterNeuronAPI<\n        TContextValue,\n        TIdType,\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload,\n        TAxonType\n    >;\n};\n\n// Concrete builder\nexport const neuron = <\n    TContextValue,\n    TIdType extends string,\n    TReceiverCollateralIdType extends string,\n    TReceiverAxonCollateralPayload,\n    TAxonType extends TCNSAxon<\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload\n    >\n>(\n    id: TIdType,\n    axon: TAxonType\n): InterNeuronAPI<\n    TContextValue,\n    TIdType,\n    TReceiverCollateralIdType,\n    TReceiverAxonCollateralPayload,\n    TAxonType\n> => {\n    const dendrites: TCNSDendrite<\n        TContextValue,\n        string,\n        unknown,\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload,\n        TAxonType\n    >[] = [];\n\n    const api: InterNeuronAPI<\n        TContextValue,\n        TIdType,\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload,\n        TAxonType\n    > = {\n        axon,\n        id,\n        dendrites,\n        dendrite<\n            TSenderExactCollateralIdType extends string,\n            TSenderExactAxonCollateralPayload\n        >(\n            s: TCNSDendrite<\n                TContextValue,\n                TSenderExactCollateralIdType,\n                TSenderExactAxonCollateralPayload,\n                TReceiverCollateralIdType,\n                TReceiverAxonCollateralPayload,\n                TAxonType\n            >\n        ) {\n            dendrites.push(\n                s as TCNSDendrite<\n                    TContextValue,\n                    string,\n                    unknown,\n                    TReceiverCollateralIdType,\n                    TReceiverAxonCollateralPayload,\n                    TAxonType\n                >\n            );\n            return api; // keep full API for chaining\n        },\n    };\n\n    return api;\n};\n\nexport const withCtx = <TContextValue>() => ({\n    neuron: <\n        TIdType extends string,\n        TReceiverCollateralIdType extends string,\n        TReceiverAxonCollateralPayload,\n        TAxonType extends TCNSAxon<\n            TReceiverCollateralIdType,\n            TReceiverAxonCollateralPayload\n        >\n    >(\n        id: TIdType,\n        axon: TAxonType\n    ) => {\n        return neuron<\n            TContextValue,\n            TIdType,\n            TReceiverCollateralIdType,\n            TReceiverAxonCollateralPayload,\n            TAxonType\n        >(id, axon);\n    },\n});\n\nconst request = collateral<{ value: number }>('request');\nconst success = collateral<{ result: string }>('success');\nconst error = collateral<{ error: string }>('error');\n\nconst router = neuron('router', { success, error }).dendrite({\n    collateral: request,\n    response: async (payload, axon) => {\n        const value = (payload as { value: number }).value;\n        if (value > 0) {\n            return axon.success.createSignal({\n                result: `Success: ${value}`,\n            });\n        } else {\n            return axon.error.createSignal({\n                error: `Error: ${value} is not positive`,\n            });\n        }\n    },\n});\n\nconst a1 = {\n    test: collateral<{ test: string }>('test'),\n};\n\nconst a2 = {\n    test2: collateral<{ test2: string }>('test2'),\n};\n\nconst a3 = {\n    test3: collateral<{ test3: string }>('test3'),\n};\n\nconst n1 = withCtx<{ testCtx: string }>()\n    .neuron('n1', a1)\n    .dendrite({\n        collateral: a2.test2,\n        response: async (payload, axon, ctx) => {\n            ctx.set({\n                testCtx: payload.test2,\n            });\n            ctx.get()?.testCtx;\n            return axon.test.createSignal({\n                test: payload.test2,\n            });\n        },\n    });\n\nconst n2 = neuron('n2', a2).dendrite({\n    collateral: a3.test3,\n    response: async (payload, axon) => {\n        return axon.test2.createSignal({\n            test2: payload.test3,\n        });\n    },\n});\n\nconst cns = new CNS([n1, n2]);\n"]}

package/dist/index.js

@@ -1,8 +1,20 @@
+// src/CNSStimulationContextStore.ts
+var CNSStimulationContextStore = class {
+  constructor(ctx = /* @__PURE__ */ new Map()) {
+    this.ctx = ctx;
+  }
+  get(key) {
+    return this.ctx.get(key);
+  }
+  set(key, value) {
+    this.ctx.set(key, value);
+  }
+};
+
 // src/CNS.ts
 var asArray = (x) => Array.isArray(x) ? x : [x];
 var CNS = class {
-  constructor(afferentLayerAsAxon, neurons) {
-    this.afferentLayerAsAxon = afferentLayerAsAxon;
+  constructor(neurons) {
     this.neurons = neurons;
     // index by collateral *type* => list of subscribers
     this.subIndex = /* @__PURE__ */ new Map();
@@ -22,20 +34,23 @@ var CNS = class {
       }
     }
   }
-  async fanOut(incoming, allowType, seen, queue) {
-    const subscribers = this.subIndex.get(incoming.edgeId);
+  async fanOut(incoming, allowType, seen, queue, ctx) {
+    const subscribers = this.subIndex.get(incoming.collateralId);
     if (!subscribers || subscribers.length === 0) return;
     for (const { neuron: neuron2, dendrite } of subscribers) {
-      const k = `${neuron2.id}::${incoming.edgeId}::${incoming.spikeId}`;
+      const k = `${neuron2.id}::${incoming.collateralId}::${incoming.spikeId}`;
       if (seen.has(k)) continue;
       seen.add(k);
-      const out = await dendrite.reaction(incoming.payload, neuron2.axon);
+      const out = await dendrite.response(incoming.payload, neuron2.axon, {
+        get: () => ctx.get(neuron2.id),
+        set: (value) => ctx.set(neuron2.id, value)
+      });
       if (out === void 0) continue;
       for (const spike of asArray(out)) {
         if (allowType && !allowType(spike.type))
           continue;
         queue.push({
-          edgeId: spike.type,
+          collateralId: spike.type,
           payload: spike.payload,
           hops: incoming.hops + 1,
           spikeId: incoming.spikeId
@@ -44,13 +59,13 @@ var CNS = class {
       }
     }
   }
-  async stimulate(axonKey, payload, opts) {
-    const afferent = this.afferentLayerAsAxon[axonKey];
+  async stimulate(collateral2, payload, opts) {
+    const ctx = opts?.ctx ?? new CNSStimulationContextStore();
     const spikeId = opts?.spikeId || Math.random().toString(36).slice(2);
     const maxHops = opts?.maxHops ?? 1e3;
     const queue = [
       {
-        edgeId: afferent.id,
+        collateralId: collateral2.id,
         payload,
         hops: 0,
         spikeId
@@ -61,7 +76,7 @@ var CNS = class {
       if (opts?.abortSignal?.aborted) break;
       const item = queue.shift();
       opts?.onTrace?.({
-        edgeId: item.edgeId,
+        collateralId: item.collateralId,
         hops: item.hops,
         payload: item.payload
       });
@@ -70,7 +85,8 @@ var CNS = class {
         item,
         opts?.allowType,
         seen,
-        queue
+        queue,
+        ctx
       );
     }
   }
@@ -96,15 +112,68 @@ var neuron = (id, axon) => {
     id,
     dendrites,
     dendrite(s) {
-      dendrites.push(s);
+      dendrites.push(
+        s
+      );
       return api;
     }
   };
   return api;
 };
-export {
-  CNS,
-  CNSCollateral,
-  collateral,
-  neuron
+var withCtx = () => ({
+  neuron: (id, axon) => {
+    return neuron(id, axon);
+  }
+});
+var request = collateral("request");
+var success = collateral("success");
+var error = collateral("error");
+neuron("router", { success, error }).dendrite({
+  collateral: request,
+  response: async (payload, axon) => {
+    const value = payload.value;
+    if (value > 0) {
+      return axon.success.createSignal({
+        result: `Success: ${value}`
+      });
+    } else {
+      return axon.error.createSignal({
+        error: `Error: ${value} is not positive`
+      });
+    }
+  }
+});
+var a1 = {
+  test: collateral("test")
+};
+var a2 = {
+  test2: collateral("test2")
 };
+var a3 = {
+  test3: collateral("test3")
+};
+var n1 = withCtx().neuron("n1", a1).dendrite({
+  collateral: a2.test2,
+  response: async (payload, axon, ctx) => {
+    ctx.set({
+      testCtx: payload.test2
+    });
+    ctx.get()?.testCtx;
+    return axon.test.createSignal({
+      test: payload.test2
+    });
+  }
+});
+var n2 = neuron("n2", a2).dendrite({
+  collateral: a3.test3,
+  response: async (payload, axon) => {
+    return axon.test2.createSignal({
+      test2: payload.test3
+    });
+  }
+});
+new CNS([n1, n2]);
+
+export { CNS, CNSCollateral, collateral, neuron, withCtx };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/CNSStimulationContextStore.ts","../src/CNS.ts","../src/CNSCollateral.ts","../src/factory/index.ts"],"names":["neuron","collateral"],"mappings":";AAEO,IAAM,6BAAN,MAAwE;AAAA,EAC3E,WAAA,CAA6B,GAAA,mBAA4B,IAAI,GAAA,EAAI,EAAG;AAAvC,IAAA,IAAA,CAAA,GAAA,GAAA,GAAA;AAAA,EAAwC;AAAA,EAErE,IAAI,GAAA,EAAsB;AACtB,IAAA,OAAO,IAAA,CAAK,GAAA,CAAI,GAAA,CAAI,GAAG,CAAA;AAAA,EAC3B;AAAA,EAEA,GAAA,CAAI,KAAa,KAAA,EAAsB;AACnC,IAAA,IAAA,CAAK,GAAA,CAAI,GAAA,CAAI,GAAA,EAAK,KAAK,CAAA;AAAA,EAC3B;AACJ,CAAA;;;ACNA,IAAM,OAAA,GAAU,CAAI,CAAA,KAAgB,KAAA,CAAM,QAAQ,CAAC,CAAA,GAAI,CAAA,GAAI,CAAC,CAAC,CAAA;AAiBtD,IAAM,MAAN,MAIL;AAAA,EAOE,YAA+B,OAAA,EAAoB;AAApB,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA;AAL/B;AAAA,IAAA,IAAA,CAAQ,QAAA,uBAAe,GAAA,EAGrB;AAGE,IAAA,IAAA,CAAK,UAAA,EAAW;AAAA,EACpB;AAAA,EAEQ,UAAA,GAAa;AACjB,IAAA,IAAA,CAAK,SAAS,KAAA,EAAM;AACpB,IAAA,KAAA,MAAWA,OAAAA,IAAU,KAAK,OAAA,EAAS;AAC/B,MAAA,KAAA,MAAW,QAAA,IAAYA,QAAO,SAAA,EAAW;AACrC,QAAA,MAAM,GAAA,GAAM,SAAS,UAAA,CAAW,EAAA;AAChC,QAAA,MAAM,MAAM,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,GAAG,KAAK,EAAC;AACvC,QAAA,GAAA,CAAI,IAAA,CAAK,EAAE,MAAA,EAAAA,OAAAA,EAAQ,UAAiC,CAAA;AACpD,QAAA,IAAA,CAAK,QAAA,CAAS,GAAA;AAAA,UACV,GAAA;AAAA,UACA;AAAA,SACJ;AAAA,MACJ;AAAA,IACJ;AAAA,EACJ;AAAA,EAEA,MAAc,MAAA,CACV,QAAA,EACA,SAAA,EACA,IAAA,EACA,OACA,GAAA,EACF;AACE,IAAA,MAAM,WAAA,GAAc,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,SAAS,YAAY,CAAA;AAC3D,IAAA,IAAI,CAAC,WAAA,IAAe,WAAA,CAAY,MAAA,KAAW,CAAA,EAAG;AAE9C,IAAA,KAAA,MAAW,EAAE,MAAA,EAAAA,OAAAA,EAAQ,QAAA,MAAc,WAAA,EAAa;AAC5C,MAAA,MAAM,CAAA,GAAI,GAAGA,OAAAA,CAAO,EAAE,KAAK,QAAA,CAAS,YAAY,CAAA,EAAA,EAAK,QAAA,CAAS,OAAO,CAAA,CAAA;AACrE,MAAA,IAAI,IAAA,CAAK,GAAA,CAAI,CAAC,CAAA,EAAG;AACjB,MAAA,IAAA,CAAK,IAAI,CAAC,CAAA;AAEV,MAAA,MAAM,MAAM,MAAM,QAAA,CAAS,SAAS,QAAA,CAAS,OAAA,EAASA,QAAO,IAAA,EAAM;AAAA,QAC/D,GAAA,EAAK,MAAM,GAAA,CAAI,GAAA,CAAIA,QAAO,EAAE,CAAA;AAAA,QAC5B,KAAK,CAAA,KAAA,KAAS,GAAA,CAAI,GAAA,CAAIA,OAAAA,CAAO,IAAI,KAAK;AAAA,OACzC,CAAA;AACD,MAAA,IAAI,QAAQ,MAAA,EAAW;AAEvB,MAAA,KAAA,MAAW,KAAA,IAAS,OAAA,CAAQ,GAAG,CAAA,EAAG;AAC9B,QAAA,IAAI,SAAA,IAAa,CAAC,SAAA,CAAU,KAAA,CAAM,IAAqB,CAAA;AACnD,UAAA;AAEJ,QAAA,KAAA,CAAM,IAAA,CAAK;AAAA,UACP,cAAc,KAAA,CAAM,IAAA;AAAA,UACpB,SAAS,KAAA,CAAM,OAAA;AAAA,UACf,IAAA,EAAM,SAAS,IAAA,GAAO,CAAA;AAAA,UACtB,SAAS,QAAA,CAAS;AAAA;AAAA,SACrB,CAAA;AAAA,MACL;AAAA,IACJ;AAAA,EACJ;AAAA,EAEA,MAAM,SAAA,CAIFC,WAAAA,EAIA,OAAA,EACA,IAAA,EAYa;AACb,IAAA,MAAM,GAAA,GAAM,IAAA,EAAM,GAAA,IAAO,IAAI,0BAAA,EAA2B;AACxD,IAAA,MAAM,OAAA,GAAU,IAAA,EAAM,OAAA,IAAW,IAAA,CAAK,MAAA,GAAS,QAAA,CAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAC,CAAA;AACnE,IAAA,MAAM,OAAA,GAAU,MAAM,OAAA,IAAW,GAAA;AAEjC,IAAA,MAAM,KAAA,GAAqC;AAAA,MACvC;AAAA,QACI,cAAcA,WAAAA,CAAW,EAAA;AAAA,QACzB,OAAA;AAAA,QACA,IAAA,EAAM,CAAA;AAAA,QACN;AAAA;AACJ,KACJ;AACA,IAAA,MAAM,IAAA,uBAAW,GAAA,EAAY;AAE7B,IAAA,OAAO,MAAM,MAAA,EAAQ;AACjB,MAAA,IAAI,IAAA,EAAM,aAAa,OAAA,EAAS;AAEhC,MAAA,MAAM,IAAA,GAAO,MAAM,KAAA,EAAM;AACzB,MAAA,IAAA,EAAM,OAAA,GAAU;AAAA,QACZ,cAAc,IAAA,CAAK,YAAA;AAAA,QACnB,MAAM,IAAA,CAAK,IAAA;AAAA,QACX,SAAS,IAAA,CAAK;AAAA,OACjB,CAAA;AACD,MAAA,IAAI,IAAA,CAAK,QAAQ,OAAA,EAAS;AAE1B,MAAA,MAAM,IAAA,CAAK,MAAA;AAAA,QACP,IAAA;AAAA,QACA,IAAA,EAAM,SAAA;AAAA,QACN,IAAA;AAAA,QACA,KAAA;AAAA,QACA;AAAA,OACJ;AAAA,IACJ;AAAA,EACJ;AACJ;;;AC9IO,IAAM,gBAAN,MAAkD;AAAA,EACrD,YAA4B,EAAA,EAAS;AAAT,IAAA,IAAA,CAAA,EAAA,GAAA,EAAA;AAAA,EAAU;AAAA,EAEtC,gBACO,IAAA,EACsB;AACzB,IAAA,MAAM,OAAA,GAAW,KAAa,CAAC,CAAA;AAC/B,IAAA,OAAO,EAAE,IAAA,EAAM,IAAA,CAAK,EAAA,EAAI,OAAA,EAAQ;AAAA,EACpC;AACJ;;;ACNO,IAAM,UAAA,GAAa,CACtB,EAAA,KACC,IAAI,cAA6B,EAAE;AA4CjC,IAAM,MAAA,GAAS,CAUlB,EAAA,EACA,IAAA,KAOC;AACD,EAAA,MAAM,YAOA,EAAC;AAEP,EAAA,MAAM,GAAA,GAMF;AAAA,IACA,IAAA;AAAA,IACA,EAAA;AAAA,IACA,SAAA;AAAA,IACA,SAII,CAAA,EAQF;AACE,MAAA,SAAA,CAAU,IAAA;AAAA,QACN;AAAA,OAQJ;AACA,MAAA,OAAO,GAAA;AAAA,IACX;AAAA,GACJ;AAEA,EAAA,OAAO,GAAA;AACX;AAEO,IAAM,UAAU,OAAsB;AAAA,EACzC,MAAA,EAAQ,CASJ,EAAA,EACA,IAAA,KACC;AACD,IAAA,OAAO,MAAA,CAML,IAAI,IAAI,CAAA;AAAA,EACd;AACJ,CAAA;AAEA,IAAM,OAAA,GAAU,WAA8B,SAAS,CAAA;AACvD,IAAM,OAAA,GAAU,WAA+B,SAAS,CAAA;AACxD,IAAM,KAAA,GAAQ,WAA8B,OAAO,CAAA;AAEpC,OAAO,QAAA,EAAU,EAAE,SAAS,KAAA,EAAO,EAAE,QAAA,CAAS;AAAA,EACzD,UAAA,EAAY,OAAA;AAAA,EACZ,QAAA,EAAU,OAAO,OAAA,EAAS,IAAA,KAAS;AAC/B,IAAA,MAAM,QAAS,OAAA,CAA8B,KAAA;AAC7C,IAAA,IAAI,QAAQ,CAAA,EAAG;AACX,MAAA,OAAO,IAAA,CAAK,QAAQ,YAAA,CAAa;AAAA,QAC7B,MAAA,EAAQ,YAAY,KAAK,CAAA;AAAA,OAC5B,CAAA;AAAA,IACL,CAAA,MAAO;AACH,MAAA,OAAO,IAAA,CAAK,MAAM,YAAA,CAAa;AAAA,QAC3B,KAAA,EAAO,UAAU,KAAK,CAAA,gBAAA;AAAA,OACzB,CAAA;AAAA,IACL;AAAA,EACJ;AACJ,CAAC;AAED,IAAM,EAAA,GAAK;AAAA,EACP,IAAA,EAAM,WAA6B,MAAM;AAC7C,CAAA;AAEA,IAAM,EAAA,GAAK;AAAA,EACP,KAAA,EAAO,WAA8B,OAAO;AAChD,CAAA;AAEA,IAAM,EAAA,GAAK;AAAA,EACP,KAAA,EAAO,WAA8B,OAAO;AAChD,CAAA;AAEA,IAAM,KAAK,OAAA,EAA6B,CACnC,OAAO,IAAA,EAAM,EAAE,EACf,QAAA,CAAS;AAAA,EACN,YAAY,EAAA,CAAG,KAAA;AAAA,EACf,QAAA,EAAU,OAAO,OAAA,EAAS,IAAA,EAAM,GAAA,KAAQ;AACpC,IAAA,GAAA,CAAI,GAAA,CAAI;AAAA,MACJ,SAAS,OAAA,CAAQ;AAAA,KACpB,CAAA;AACD,IAAA,GAAA,CAAI,KAAI,EAAG,OAAA;AACX,IAAA,OAAO,IAAA,CAAK,KAAK,YAAA,CAAa;AAAA,MAC1B,MAAM,OAAA,CAAQ;AAAA,KACjB,CAAA;AAAA,EACL;AACJ,CAAC,CAAA;AAEL,IAAM,EAAA,GAAK,MAAA,CAAO,IAAA,EAAM,EAAE,EAAE,QAAA,CAAS;AAAA,EACjC,YAAY,EAAA,CAAG,KAAA;AAAA,EACf,QAAA,EAAU,OAAO,OAAA,EAAS,IAAA,KAAS;AAC/B,IAAA,OAAO,IAAA,CAAK,MAAM,YAAA,CAAa;AAAA,MAC3B,OAAO,OAAA,CAAQ;AAAA,KAClB,CAAA;AAAA,EACL;AACJ,CAAC,CAAA;AAEW,IAAI,GAAA,CAAI,CAAC,EAAA,EAAI,EAAE,CAAC","file":"index.js","sourcesContent":["import { ICNSStimulationContextStore } from './interfaces/ICNSStimulationContextStore';\n\nexport class CNSStimulationContextStore implements ICNSStimulationContextStore {\n    constructor(private readonly ctx: Map<string, unknown> = new Map()) {}\n\n    get(key: string): unknown {\n        return this.ctx.get(key);\n    }\n\n    set(key: string, value: unknown): void {\n        this.ctx.set(key, value);\n    }\n}\n","import { TCNSNeuron } from './types/TCNSNeuron';\nimport { TCNSDendrite } from './types/TCNSDendrite';\nimport { CNSCollateral } from './CNSCollateral';\nimport { ICNSStimulationContextStore } from './interfaces/ICNSStimulationContextStore';\nimport { CNSStimulationContextStore } from './CNSStimulationContextStore';\n\nconst asArray = <T>(x: T | T[]) => (Array.isArray(x) ? x : [x]);\n\ntype TSubscriber<\n    TNeuron extends TCNSNeuron<any, any, any, any, any, any, any>,\n    TDendrite extends TCNSDendrite<any, any, any, any, any, any>\n> = {\n    neuron: TNeuron;\n    dendrite: TDendrite;\n};\n\ntype TQueueItem<TCollateralId extends string> = {\n    collateralId: TCollateralId;\n    payload: unknown;\n    hops: number;\n    spikeId: string;\n};\n\nexport class CNS<\n    TCollateralId extends string,\n    TNeuron extends TCNSNeuron<any, any, any, any, any, any, any>,\n    TDendrite extends TCNSDendrite<any, any, any, any, any, any>\n> {\n    // index by collateral *type* => list of subscribers\n    private subIndex = new Map<\n        TCollateralId,\n        TSubscriber<TNeuron, TDendrite>[]\n    >();\n\n    constructor(protected readonly neurons: TNeuron[]) {\n        this.buildIndex();\n    }\n\n    private buildIndex() {\n        this.subIndex.clear();\n        for (const neuron of this.neurons) {\n            for (const dendrite of neuron.dendrites) {\n                const key = dendrite.collateral.id;\n                const arr = this.subIndex.get(key) ?? [];\n                arr.push({ neuron, dendrite: dendrite as TDendrite });\n                this.subIndex.set(\n                    key,\n                    arr as TSubscriber<TNeuron, TDendrite>[]\n                );\n            }\n        }\n    }\n\n    private async fanOut(\n        incoming: TQueueItem<TCollateralId>,\n        allowType: ((t: TCollateralId) => boolean) | undefined,\n        seen: Set<string>,\n        queue: TQueueItem<TCollateralId>[],\n        ctx: ICNSStimulationContextStore\n    ) {\n        const subscribers = this.subIndex.get(incoming.collateralId);\n        if (!subscribers || subscribers.length === 0) return;\n\n        for (const { neuron, dendrite } of subscribers) {\n            const k = `${neuron.id}::${incoming.collateralId}::${incoming.spikeId}`;\n            if (seen.has(k)) continue;\n            seen.add(k);\n\n            const out = await dendrite.response(incoming.payload, neuron.axon, {\n                get: () => ctx.get(neuron.id),\n                set: value => ctx.set(neuron.id, value),\n            });\n            if (out === undefined) continue;\n\n            for (const spike of asArray(out)) {\n                if (allowType && !allowType(spike.type as TCollateralId))\n                    continue; // TODO: add allowType\n\n                queue.push({\n                    collateralId: spike.type as TCollateralId,\n                    payload: spike.payload,\n                    hops: incoming.hops + 1,\n                    spikeId: incoming.spikeId, // keep the same spike id across the cascade\n                });\n            }\n        }\n    }\n\n    async stimulate<\n        TAfferentCollateralId extends TCollateralId,\n        TAfferentCollateralPayload\n    >(\n        collateral: CNSCollateral<\n            TAfferentCollateralId,\n            TAfferentCollateralPayload\n        >,\n        payload: TAfferentCollateralPayload,\n        opts?: {\n            maxHops?: number;\n            allowType?: (t: TCollateralId) => boolean;\n            onTrace?: (e: {\n                collateralId: TCollateralId;\n                hops: number;\n                payload: unknown;\n            }) => void;\n            abortSignal?: AbortSignal;\n            spikeId?: string;\n            ctx?: ICNSStimulationContextStore;\n        }\n    ): Promise<void> {\n        const ctx = opts?.ctx ?? new CNSStimulationContextStore();\n        const spikeId = opts?.spikeId || Math.random().toString(36).slice(2);\n        const maxHops = opts?.maxHops ?? 1000;\n\n        const queue: TQueueItem<TCollateralId>[] = [\n            {\n                collateralId: collateral.id,\n                payload,\n                hops: 0,\n                spikeId,\n            },\n        ];\n        const seen = new Set<string>();\n\n        while (queue.length) {\n            if (opts?.abortSignal?.aborted) break;\n\n            const item = queue.shift()!;\n            opts?.onTrace?.({\n                collateralId: item.collateralId,\n                hops: item.hops,\n                payload: item.payload,\n            });\n            if (item.hops >= maxHops) continue;\n\n            await this.fanOut(\n                item as TQueueItem<TCollateralId>,\n                opts?.allowType,\n                seen,\n                queue as TQueueItem<TCollateralId>[],\n                ctx\n            );\n        }\n    }\n}\n","import { TCNSSignal } from './types/TCNSSignal';\n\nexport class CNSCollateral<TId extends string, TPayload> {\n    constructor(public readonly id: TId) {}\n\n    createSignal(\n        ...args: [TPayload] extends [undefined] ? [] : [payload: TPayload]\n    ): TCNSSignal<TId, TPayload> {\n        const payload = (args as any)[0] as TPayload;\n        return { type: this.id, payload };\n    }\n}\n","import { CNS } from '../CNS';\nimport { CNSCollateral } from '../CNSCollateral';\nimport { TCNSAxon } from '../types/TCNSAxon';\nimport { TCNSDendrite } from '../types/TCNSDendrite';\n\nexport const collateral = <TPayload = undefined, TId extends string = string>(\n    id: TId\n) => new CNSCollateral<TId, TPayload>(id);\n\ntype InterNeuronAPI<\n    TContextValue,\n    TIdType extends string,\n    TReceiverCollateralIdType extends string,\n    TReceiverAxonCollateralPayload,\n    TAxonType extends TCNSAxon<\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload\n    >\n> = {\n    id: TIdType;\n    axon: TAxonType;\n    dendrites: TCNSDendrite<\n        TContextValue,\n        string,\n        unknown,\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload,\n        TAxonType\n    >[];\n    dendrite: <\n        TSenderExactCollateralIdType extends string,\n        TSenderExactAxonCollateralPayload\n    >(\n        s: TCNSDendrite<\n            TContextValue,\n            TSenderExactCollateralIdType,\n            TSenderExactAxonCollateralPayload,\n            TReceiverCollateralIdType,\n            TReceiverAxonCollateralPayload,\n            TAxonType\n        >\n    ) => InterNeuronAPI<\n        TContextValue,\n        TIdType,\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload,\n        TAxonType\n    >;\n};\n\n// Concrete builder\nexport const neuron = <\n    TContextValue,\n    TIdType extends string,\n    TReceiverCollateralIdType extends string,\n    TReceiverAxonCollateralPayload,\n    TAxonType extends TCNSAxon<\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload\n    >\n>(\n    id: TIdType,\n    axon: TAxonType\n): InterNeuronAPI<\n    TContextValue,\n    TIdType,\n    TReceiverCollateralIdType,\n    TReceiverAxonCollateralPayload,\n    TAxonType\n> => {\n    const dendrites: TCNSDendrite<\n        TContextValue,\n        string,\n        unknown,\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload,\n        TAxonType\n    >[] = [];\n\n    const api: InterNeuronAPI<\n        TContextValue,\n        TIdType,\n        TReceiverCollateralIdType,\n        TReceiverAxonCollateralPayload,\n        TAxonType\n    > = {\n        axon,\n        id,\n        dendrites,\n        dendrite<\n            TSenderExactCollateralIdType extends string,\n            TSenderExactAxonCollateralPayload\n        >(\n            s: TCNSDendrite<\n                TContextValue,\n                TSenderExactCollateralIdType,\n                TSenderExactAxonCollateralPayload,\n                TReceiverCollateralIdType,\n                TReceiverAxonCollateralPayload,\n                TAxonType\n            >\n        ) {\n            dendrites.push(\n                s as TCNSDendrite<\n                    TContextValue,\n                    string,\n                    unknown,\n                    TReceiverCollateralIdType,\n                    TReceiverAxonCollateralPayload,\n                    TAxonType\n                >\n            );\n            return api; // keep full API for chaining\n        },\n    };\n\n    return api;\n};\n\nexport const withCtx = <TContextValue>() => ({\n    neuron: <\n        TIdType extends string,\n        TReceiverCollateralIdType extends string,\n        TReceiverAxonCollateralPayload,\n        TAxonType extends TCNSAxon<\n            TReceiverCollateralIdType,\n            TReceiverAxonCollateralPayload\n        >\n    >(\n        id: TIdType,\n        axon: TAxonType\n    ) => {\n        return neuron<\n            TContextValue,\n            TIdType,\n            TReceiverCollateralIdType,\n            TReceiverAxonCollateralPayload,\n            TAxonType\n        >(id, axon);\n    },\n});\n\nconst request = collateral<{ value: number }>('request');\nconst success = collateral<{ result: string }>('success');\nconst error = collateral<{ error: string }>('error');\n\nconst router = neuron('router', { success, error }).dendrite({\n    collateral: request,\n    response: async (payload, axon) => {\n        const value = (payload as { value: number }).value;\n        if (value > 0) {\n            return axon.success.createSignal({\n                result: `Success: ${value}`,\n            });\n        } else {\n            return axon.error.createSignal({\n                error: `Error: ${value} is not positive`,\n            });\n        }\n    },\n});\n\nconst a1 = {\n    test: collateral<{ test: string }>('test'),\n};\n\nconst a2 = {\n    test2: collateral<{ test2: string }>('test2'),\n};\n\nconst a3 = {\n    test3: collateral<{ test3: string }>('test3'),\n};\n\nconst n1 = withCtx<{ testCtx: string }>()\n    .neuron('n1', a1)\n    .dendrite({\n        collateral: a2.test2,\n        response: async (payload, axon, ctx) => {\n            ctx.set({\n                testCtx: payload.test2,\n            });\n            ctx.get()?.testCtx;\n            return axon.test.createSignal({\n                test: payload.test2,\n            });\n        },\n    });\n\nconst n2 = neuron('n2', a2).dendrite({\n    collateral: a3.test3,\n    response: async (payload, axon) => {\n        return axon.test2.createSignal({\n            test2: payload.test3,\n        });\n    },\n});\n\nconst cns = new CNS([n1, n2]);\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cnstra/core",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "Neural network-inspired event flow system for building reactive applications with dependency-injected neurons and synapses",
   "keywords": [
     "neural-network",
@@ -36,7 +36,8 @@
     }
   },
   "scripts": {
-    "build": "tsup src/index.ts --dts --format esm,cjs --tsconfig tsconfig.build.json",
+    "build": "tsup && tsc --emitDeclarationOnly",
+    "build:types": "tsc --emitDeclarationOnly",
     "prepublishOnly": "npm run build",
     "test": "jest",
     "test:types": "tsd",

package/dist/index.d.cts

@@ -1,56 +0,0 @@
-type TCNSSignal<TType extends string, TPayload> = {
-    type: TType;
-    payload?: TPayload;
-};
-
-declare class CNSCollateral<TId extends string, TPayload> {
-    readonly id: TId;
-    constructor(id: TId);
-    createSignal(...args: [TPayload] extends [undefined] ? [] : [payload: TPayload]): TCNSSignal<TId, TPayload>;
-}
-
-type TCNSAxon<TCId extends string, TInputPayload = unknown> = Record<TCId, CNSCollateral<TCId, TInputPayload>>;
-
-type TCNDendrite<TSenderCollateralIdType extends string, TReceiverAxonCollateralPayload, TReceiverCollateralIdType extends string, TReceiverAxon extends TCNSAxon<TReceiverCollateralIdType, TReceiverAxonCollateralPayload>> = {
-    collateral: CNSCollateral<TSenderCollateralIdType, TReceiverAxonCollateralPayload>;
-    reaction: (payload: TReceiverAxonCollateralPayload, axon: TReceiverAxon) => Promise<ReturnType<CNSCollateral<TReceiverCollateralIdType, TReceiverAxonCollateralPayload>['createSignal']>> | ReturnType<CNSCollateral<TReceiverCollateralIdType, TReceiverAxonCollateralPayload>['createSignal']> | Promise<void> | void;
-};
-
-type TCNSNeuron<TId extends string, TCollateralId extends string, TAxon extends TCNSAxon<TCollateralId, unknown>, TSenderCollateralId extends string, TDendrites extends TCNDendrite<TSenderCollateralId, unknown, TCollateralId, TAxon>[]> = {
-    id: TId;
-    axon: TAxon;
-    dendrites: TDendrites;
-};
-
-declare class CNS<TAfferentCollateralId extends string, TAfferentAxon extends TCNSAxon<TAfferentCollateralId, unknown>, TNeuronId extends string, TCollateralId extends string, TAxon extends TCNSAxon<TCollateralId, any>> {
-    protected readonly afferentLayerAsAxon: TAfferentAxon;
-    protected readonly neurons: TCNSNeuron<TNeuronId, TCollateralId, TAxon, TCollateralId, TCNDendrite<TCollateralId, any, TCollateralId, TAxon>[]>[];
-    private subIndex;
-    constructor(afferentLayerAsAxon: TAfferentAxon, neurons: TCNSNeuron<TNeuronId, TCollateralId, TAxon, TCollateralId, TCNDendrite<TCollateralId, any, TCollateralId, TAxon>[]>[]);
-    private buildIndex;
-    private fanOut;
-    stimulate<K extends keyof TAfferentAxon, TACollateralId extends string>(axonKey: K, payload: ReturnType<TAfferentAxon[K]['createSignal']> & {
-        type: TACollateralId;
-    }, opts?: {
-        maxHops?: number;
-        allowType?: (t: TCollateralId | TAfferentCollateralId) => boolean;
-        onTrace?: (e: {
-            edgeId: TCollateralId | TAfferentCollateralId;
-            hops: number;
-            payload: unknown;
-        }) => void;
-        abortSignal?: AbortSignal;
-        spikeId?: string;
-    }): Promise<void>;
-}
-
-declare const collateral: <TPayload = undefined, TId extends string = string>(id: TId) => CNSCollateral<TId, TPayload>;
-type InterNeuronAPI<TIdType extends string, TSenderCollateralIdType extends string, TReceiverAxonCollateralPayload, TReceiverCollateralIdType extends string, TReceiverAxon extends TCNSAxon<TReceiverCollateralIdType, TReceiverAxonCollateralPayload>> = {
-    id: TIdType;
-    axon: TReceiverAxon;
-    dendrites: TCNDendrite<TSenderCollateralIdType, TReceiverAxonCollateralPayload, TReceiverCollateralIdType, TReceiverAxon>[];
-    dendrite: (s: TCNDendrite<TSenderCollateralIdType, TReceiverAxonCollateralPayload, TReceiverCollateralIdType, TReceiverAxon>) => InterNeuronAPI<TIdType, TSenderCollateralIdType, TReceiverAxonCollateralPayload, TReceiverCollateralIdType, TReceiverAxon>;
-};
-declare const neuron: <TIdType extends string, TSenderCollateralIdType extends string, TReceiverAxonCollateralPayload, TReceiverCollateralIdType extends string, TReceiverAxon extends TCNSAxon<TReceiverCollateralIdType, TReceiverAxonCollateralPayload>>(id: TIdType, axon: TReceiverAxon) => InterNeuronAPI<TIdType, TSenderCollateralIdType, TReceiverAxonCollateralPayload, TReceiverCollateralIdType, TReceiverAxon>;
-
-export { CNS, CNSCollateral, type TCNDendrite, type TCNSAxon, type TCNSNeuron, type TCNSSignal, collateral, neuron };

package/dist/index.d.ts

@@ -1,56 +0,0 @@
-type TCNSSignal<TType extends string, TPayload> = {
-    type: TType;
-    payload?: TPayload;
-};
-
-declare class CNSCollateral<TId extends string, TPayload> {
-    readonly id: TId;
-    constructor(id: TId);
-    createSignal(...args: [TPayload] extends [undefined] ? [] : [payload: TPayload]): TCNSSignal<TId, TPayload>;
-}
-
-type TCNSAxon<TCId extends string, TInputPayload = unknown> = Record<TCId, CNSCollateral<TCId, TInputPayload>>;
-
-type TCNDendrite<TSenderCollateralIdType extends string, TReceiverAxonCollateralPayload, TReceiverCollateralIdType extends string, TReceiverAxon extends TCNSAxon<TReceiverCollateralIdType, TReceiverAxonCollateralPayload>> = {
-    collateral: CNSCollateral<TSenderCollateralIdType, TReceiverAxonCollateralPayload>;
-    reaction: (payload: TReceiverAxonCollateralPayload, axon: TReceiverAxon) => Promise<ReturnType<CNSCollateral<TReceiverCollateralIdType, TReceiverAxonCollateralPayload>['createSignal']>> | ReturnType<CNSCollateral<TReceiverCollateralIdType, TReceiverAxonCollateralPayload>['createSignal']> | Promise<void> | void;
-};
-
-type TCNSNeuron<TId extends string, TCollateralId extends string, TAxon extends TCNSAxon<TCollateralId, unknown>, TSenderCollateralId extends string, TDendrites extends TCNDendrite<TSenderCollateralId, unknown, TCollateralId, TAxon>[]> = {
-    id: TId;
-    axon: TAxon;
-    dendrites: TDendrites;
-};
-
-declare class CNS<TAfferentCollateralId extends string, TAfferentAxon extends TCNSAxon<TAfferentCollateralId, unknown>, TNeuronId extends string, TCollateralId extends string, TAxon extends TCNSAxon<TCollateralId, any>> {
-    protected readonly afferentLayerAsAxon: TAfferentAxon;
-    protected readonly neurons: TCNSNeuron<TNeuronId, TCollateralId, TAxon, TCollateralId, TCNDendrite<TCollateralId, any, TCollateralId, TAxon>[]>[];
-    private subIndex;
-    constructor(afferentLayerAsAxon: TAfferentAxon, neurons: TCNSNeuron<TNeuronId, TCollateralId, TAxon, TCollateralId, TCNDendrite<TCollateralId, any, TCollateralId, TAxon>[]>[]);
-    private buildIndex;
-    private fanOut;
-    stimulate<K extends keyof TAfferentAxon, TACollateralId extends string>(axonKey: K, payload: ReturnType<TAfferentAxon[K]['createSignal']> & {
-        type: TACollateralId;
-    }, opts?: {
-        maxHops?: number;
-        allowType?: (t: TCollateralId | TAfferentCollateralId) => boolean;
-        onTrace?: (e: {
-            edgeId: TCollateralId | TAfferentCollateralId;
-            hops: number;
-            payload: unknown;
-        }) => void;
-        abortSignal?: AbortSignal;
-        spikeId?: string;
-    }): Promise<void>;
-}
-
-declare const collateral: <TPayload = undefined, TId extends string = string>(id: TId) => CNSCollateral<TId, TPayload>;
-type InterNeuronAPI<TIdType extends string, TSenderCollateralIdType extends string, TReceiverAxonCollateralPayload, TReceiverCollateralIdType extends string, TReceiverAxon extends TCNSAxon<TReceiverCollateralIdType, TReceiverAxonCollateralPayload>> = {
-    id: TIdType;
-    axon: TReceiverAxon;
-    dendrites: TCNDendrite<TSenderCollateralIdType, TReceiverAxonCollateralPayload, TReceiverCollateralIdType, TReceiverAxon>[];
-    dendrite: (s: TCNDendrite<TSenderCollateralIdType, TReceiverAxonCollateralPayload, TReceiverCollateralIdType, TReceiverAxon>) => InterNeuronAPI<TIdType, TSenderCollateralIdType, TReceiverAxonCollateralPayload, TReceiverCollateralIdType, TReceiverAxon>;
-};
-declare const neuron: <TIdType extends string, TSenderCollateralIdType extends string, TReceiverAxonCollateralPayload, TReceiverCollateralIdType extends string, TReceiverAxon extends TCNSAxon<TReceiverCollateralIdType, TReceiverAxonCollateralPayload>>(id: TIdType, axon: TReceiverAxon) => InterNeuronAPI<TIdType, TSenderCollateralIdType, TReceiverAxonCollateralPayload, TReceiverCollateralIdType, TReceiverAxon>;
-
-export { CNS, CNSCollateral, type TCNDendrite, type TCNSAxon, type TCNSNeuron, type TCNSSignal, collateral, neuron };