@cnstra/core 2.0.0 → 2.1.0

package/dist/index.js

@@ -948,6 +948,58 @@ 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);
+  }
+};
+
 // src/CNSCollateral.ts
 var CNSCollateral = class {
   createSignal(payload) {
@@ -955,6 +1007,55 @@ var CNSCollateral = class {
   }
 };
 
+// src/utils/CNSStimulationGate.ts
+var CNSStimulationGate = class {
+  constructor(guardOptions) {
+    this.guardOptions = guardOptions;
+  }
+  isDraining() {
+    return this.currentDrain !== void 0;
+  }
+  getCurrentStimulation() {
+    return this.currentStimulation;
+  }
+  drain() {
+    if (this.currentDrain) return this.currentDrain;
+    const stimulation = this.guardOptions.cns.stimulate(
+      this.guardOptions.signal,
+      this.createStimulationOptions()
+    );
+    const drain = stimulation.waitUntilComplete().finally(() => {
+      if (this.currentDrain !== drain) return;
+      this.currentDrain = void 0;
+      this.currentStimulation = void 0;
+      this.currentAbortController = void 0;
+    });
+    this.currentStimulation = stimulation;
+    this.currentDrain = drain;
+    return drain;
+  }
+  abort(reason) {
+    if (!this.currentAbortController) return false;
+    if (this.currentAbortController.signal.aborted) return false;
+    this.currentAbortController.abort(reason);
+    return true;
+  }
+  createStimulationOptions() {
+    const options = this.guardOptions.options;
+    if (options?.abortSignal) {
+      return options;
+    }
+    if (typeof AbortController === "undefined") {
+      return options ?? {};
+    }
+    this.currentAbortController = new AbortController();
+    return {
+      ...options,
+      abortSignal: this.currentAbortController.signal
+    };
+  }
+};
+
 // src/factory/index.ts
 var collateral = () => new CNSCollateral();
 var neuron = (axon) => {
@@ -1097,7 +1198,20 @@ var modality = (afferentPaths) => {
     afferentPaths
   };
 };
+var createPersistRegistry = (namedNeurons) => {
+  const registry = new CNSPersistOptionsRegistry();
+  for (const [name, neuron2] of Object.entries(namedNeurons)) {
+    registry.addNeuron(neuron2, { name, neuron: neuron2 });
+    for (const [key, col] of Object.entries(neuron2.axon)) {
+      registry.addCollateral(col, {
+        name: key,
+        collateral: col
+      });
+    }
+  }
+  return registry;
+};
 
-export { CNS, CNSCollateral, 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