@cnstra/core 2.0.1 → 2.2.0

package/dist/index.js

@@ -948,6 +948,72 @@ var CNS = class {
   }
 };
 
+// src/CNSPersistOptionsRegistry.ts
+var CNSPersistOptionsRegistry = class {
+  constructor() {
+    this.neurons = /* @__PURE__ */ new Map();
+    this.neuronNames = /* @__PURE__ */ new WeakMap();
+    this.collaterals = /* @__PURE__ */ new Map();
+    this.stimulations = /* @__PURE__ */ new Map();
+  }
+  addNeuron(neuron2, options) {
+    this.neurons.set(options.name, neuron2);
+    this.neuronNames.set(neuron2, options.name);
+  }
+  getNeuron(name) {
+    return this.neurons.get(name);
+  }
+  getNeuronName(neuron2) {
+    return this.neuronNames.get(neuron2);
+  }
+  getNamedNeurons() {
+    return this.neurons;
+  }
+  getCollateralName(collateral2) {
+    for (const [name, col] of this.collaterals) {
+      if (col === collateral2) return name;
+    }
+    return void 0;
+  }
+  removeNeuron(name) {
+    const neuron2 = this.neurons.get(name);
+    if (neuron2) this.neuronNames.delete(neuron2);
+    this.neurons.delete(name);
+  }
+  addCollateral(collateral2, options) {
+    this.collaterals.set(options.name, collateral2);
+  }
+  getCollateral(name) {
+    return this.collaterals.get(name);
+  }
+  removeCollateral(name) {
+    this.collaterals.delete(name);
+  }
+  addStimulation(stimulation, options) {
+    this.stimulations.set(options.stimulationId, stimulation);
+  }
+  getStimulation(stimulationId) {
+    return this.stimulations.get(stimulationId);
+  }
+  removeStimulation(stimulationId) {
+    this.stimulations.delete(stimulationId);
+  }
+  register(name, neuron2, collateralNames) {
+    this.addNeuron(neuron2, { name, neuron: neuron2 });
+    for (const [key, col] of Object.entries(neuron2.axon)) {
+      this.addCollateral(col, {
+        name: collateralNames?.[key] ?? key,
+        collateral: col
+      });
+    }
+    return this;
+  }
+  registerCollateral(name, collateral2) {
+    this.addCollateral(collateral2, { name, collateral: collateral2 });
+    return this;
+  }
+};
+
 // src/CNSCollateral.ts
 var CNSCollateral = class {
   createSignal(payload) {
@@ -955,8 +1021,8 @@ var CNSCollateral = class {
   }
 };
 
-// src/utils/CNSDrainGuard.ts
-var CNSDrainGuard = class {
+// src/utils/CNSStimulationGate.ts
+var CNSStimulationGate = class {
   constructor(guardOptions) {
     this.guardOptions = guardOptions;
   }
@@ -1146,7 +1212,23 @@ var modality = (afferentPaths) => {
     afferentPaths
   };
 };
+var createPersistRegistry = (namedNeurons) => {
+  const registry = new CNSPersistOptionsRegistry();
+  for (const [name, entry] of Object.entries(namedNeurons)) {
+    const isEntry = "neuron" in entry && !("axon" in entry);
+    const neuron2 = isEntry ? entry.neuron : entry;
+    const colNames = isEntry ? entry.collaterals ?? {} : {};
+    registry.addNeuron(neuron2, { name, neuron: neuron2 });
+    for (const [key, col] of Object.entries(neuron2.axon)) {
+      registry.addCollateral(col, {
+        name: colNames[key] ?? key,
+        collateral: col
+      });
+    }
+  }
+  return registry;
+};
 
-export { CNS, CNSCollateral, CNSDrainGuard, CNSStimulationContextStore, afferentPath, collateral, modality, neuron, withCtx };
+export { CNS, CNSCollateral, CNSPersistOptionsRegistry, CNSStimulationContextStore, CNSStimulationGate, afferentPath, collateral, createPersistRegistry, modality, neuron, withCtx };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map