dirac-lang 0.1.69 → 0.1.71

package/dist/{agent-RH5SYOLV.js → agent-WJUVRA6D.js}

@@ -2,8 +2,8 @@ import {
   SessionServer,
   getSocketPath,
   isSessionRunning
-} from "./chunk-65TBONV6.js";
-import "./chunk-QUFYZS2S.js";
+} from "./chunk-YVI7T3DS.js";
+import "./chunk-QPO3WKH2.js";
 import "./chunk-HRHAMPOB.js";
 import "./chunk-VC23AJJJ.js";
 import "./chunk-M57VI7KL.js";

package/dist/{chunk-6SKZC4N7.js → chunk-PLH7YN7I.js}

@@ -1,6 +1,6 @@
 import {
   integrate
-} from "./chunk-QUFYZS2S.js";
+} from "./chunk-QPO3WKH2.js";
 import {
   DiracParser
 } from "./chunk-HRHAMPOB.js";

package/dist/{chunk-QUFYZS2S.js → chunk-QPO3WKH2.js}

@@ -181,6 +181,7 @@ async function executeAssign(session, element) {
   const name = element.attributes.name;
   const valueAttr = element.attributes.value;
   const trimAttr = element.attributes.trim;
+  const typeAttr = element.attributes.type;
   if (!name) {
     throw new Error("<assign> requires name attribute");
   }
@@ -203,6 +204,12 @@ async function executeAssign(session, element) {
   if (trimAttr === "true" && typeof value === "string") {
     value = value.trim();
   }
+  if (typeAttr === "cat") {
+    const existingValue = getVariable(session, name);
+    if (existingValue !== void 0) {
+      value = String(existingValue) + String(value);
+    }
+  }
   for (let i = session.variables.length - 1; i >= 0; i--) {
     if (session.variables[i].name === name) {
       session.variables[i].value = value;
@@ -474,12 +481,12 @@ async function executeIf(session, element) {
   const condition = await evaluatePredicate(session, conditionElement);
   if (condition) {
     if (thenElement) {
-      const { integrateChildren: integrateChildren2 } = await import("./interpreter-N4ERLMWW.js");
+      const { integrateChildren: integrateChildren2 } = await import("./interpreter-Y2AU6RG6.js");
       await integrateChildren2(session, thenElement);
     }
   } else {
     if (elseElement) {
-      const { integrateChildren: integrateChildren2 } = await import("./interpreter-N4ERLMWW.js");
+      const { integrateChildren: integrateChildren2 } = await import("./interpreter-Y2AU6RG6.js");
       await integrateChildren2(session, elseElement);
     }
   }
@@ -492,7 +499,7 @@ async function evaluatePredicate(session, predicateElement) {
     return await evaluateCondition(session, predicateElement);
   }
   const outputLengthBefore = session.output.length;
-  const { integrate: integrate2 } = await import("./interpreter-N4ERLMWW.js");
+  const { integrate: integrate2 } = await import("./interpreter-Y2AU6RG6.js");
   await integrate2(session, predicateElement);
   const newOutputChunks = session.output.slice(outputLengthBefore);
   const result = newOutputChunks.join("").trim();
@@ -515,11 +522,11 @@ async function evaluateCondition(session, condElement) {
   }
   const outputLengthBefore = session.output.length;
   const args = [];
-  const { integrate: integrate2 } = await import("./interpreter-N4ERLMWW.js");
+  const { integrate: integrate2 } = await import("./interpreter-Y2AU6RG6.js");
   for (const child of condElement.children) {
     if (child.tag.toLowerCase() === "arg") {
       const argOutputStart = session.output.length;
-      const { integrateChildren: integrateChildren2 } = await import("./interpreter-N4ERLMWW.js");
+      const { integrateChildren: integrateChildren2 } = await import("./interpreter-Y2AU6RG6.js");
       await integrateChildren2(session, child);
       const newChunks = session.output.slice(argOutputStart);
       const argValue = newChunks.join("");
@@ -1740,7 +1747,7 @@ async function executeTagCheck(session, element) {
       const executeTag = correctedTag || tagName;
       console.error(`[tag-check] Executing <${executeTag}/> as all checks passed and execute=true.`);
       const elementToExecute = correctedTag ? { ...child, tag: correctedTag } : child;
-      const { integrate: integrate2 } = await import("./interpreter-N4ERLMWW.js");
+      const { integrate: integrate2 } = await import("./interpreter-Y2AU6RG6.js");
       await integrate2(session, elementToExecute);
     }
   }
@@ -1749,7 +1756,7 @@ async function executeTagCheck(session, element) {
 // src/tags/throw.ts
 async function executeThrow(session, element) {
   const exceptionName = element.attributes?.name || "exception";
-  const { integrateChildren: integrateChildren2 } = await import("./interpreter-N4ERLMWW.js");
+  const { integrateChildren: integrateChildren2 } = await import("./interpreter-Y2AU6RG6.js");
   const exceptionDom = {
     tag: "exception-content",
     attributes: { name: exceptionName },
@@ -1762,7 +1769,7 @@ async function executeThrow(session, element) {
 // src/tags/try.ts
 async function executeTry(session, element) {
   setExceptionBoundary(session);
-  const { integrateChildren: integrateChildren2 } = await import("./interpreter-N4ERLMWW.js");
+  const { integrateChildren: integrateChildren2 } = await import("./interpreter-Y2AU6RG6.js");
   await integrateChildren2(session, element);
   unsetExceptionBoundary(session);
 }
@@ -1772,7 +1779,7 @@ async function executeCatch(session, element) {
   const exceptionName = element.attributes?.name || "exception";
   const caughtCount = lookupException(session, exceptionName);
   if (caughtCount > 0) {
-    const { integrateChildren: integrateChildren2 } = await import("./interpreter-N4ERLMWW.js");
+    const { integrateChildren: integrateChildren2 } = await import("./interpreter-Y2AU6RG6.js");
     await integrateChildren2(session, element);
   }
   flushCurrentException(session);
@@ -1781,7 +1788,7 @@ async function executeCatch(session, element) {
 // src/tags/exception.ts
 async function executeException(session, element) {
   const exceptions = getCurrentExceptions(session);
-  const { integrateChildren: integrateChildren2 } = await import("./interpreter-N4ERLMWW.js");
+  const { integrateChildren: integrateChildren2 } = await import("./interpreter-Y2AU6RG6.js");
   for (const exceptionDom of exceptions) {
     await integrateChildren2(session, exceptionDom);
   }
@@ -2046,6 +2053,7 @@ var SubroutineRegistry = class {
       this.index.subroutines = this.index.subroutines.filter((s) => s.filePath !== filePath);
       const subroutines = this.extractSubroutines(ast, filePath);
       this.index.subroutines.push(...subroutines);
+      this.saveIndex();
       return subroutines.length;
     } catch (err) {
       if (process.env.DEBUG_REGISTRY === "1") {
@@ -2242,14 +2250,13 @@ async function executeRegistryStats(session, element) {
 }
 
 // src/tags/load-context.ts
-var registry2 = new SubroutineRegistry();
 async function executeLoadContext(session, element) {
   let query = element.attributes.query;
   if (!query && element.text) {
     query = element.text.trim();
   }
   if (!query && element.children.length > 0) {
-    const { integrate: integrate2 } = await import("./interpreter-N4ERLMWW.js");
+    const { integrate: integrate2 } = await import("./interpreter-Y2AU6RG6.js");
     const beforeOutput = session.output.length;
     for (const child of element.children) {
       await integrate2(session, child);
@@ -2264,14 +2271,14 @@ async function executeLoadContext(session, element) {
   const limitAttr = element.attributes.limit;
   const shouldImport = element.attributes.import !== "false";
   const outputVar = element.attributes.output;
-  const stats = registry2.getStats();
+  const stats = registry.getStats();
   if (stats.totalSubroutines === 0) {
     emit(session, `[load-context] Registry is empty. Use :index <path> or <index-subroutines path="..."> first.
 `);
     return;
   }
   const limit = limitAttr ? parseInt(limitAttr, 10) : 5;
-  const results = registry2.search(query, limit);
+  const results = registry.search(query, limit);
   if (results.length === 0) {
     emit(session, `[load-context] No subroutines found for query: "${query}". Try indexing more libraries.
 `);
@@ -2324,7 +2331,7 @@ async function executeLoadContext(session, element) {
 }
 
 // src/tags/save-subroutine.ts
-import { writeFileSync as writeFileSync4, mkdirSync as mkdirSync4, existsSync as existsSync5, readFileSync as readFileSync3 } from "fs";
+import { writeFileSync as writeFileSync4, mkdirSync as mkdirSync4, existsSync as existsSync5 } from "fs";
 import { resolve as resolve3, dirname as dirname4, join as join4 } from "path";
 import { homedir as homedir4 } from "os";
 async function executeSaveSubroutine(session, element) {
@@ -2352,40 +2359,13 @@ async function executeSaveSubroutine(session, element) {
     content = generateXMLNotation(subroutine);
   }
   let filePath;
-  let shouldMerge = false;
   if (file) {
     filePath = resolve3(process.cwd(), file);
   } else if (pathAttr) {
     const targetDir = join4(homedir4(), ".dirac", "lib", pathAttr);
     filePath = join4(targetDir, `${name}.di`);
-  } else if (subroutine.sourcePath && existsSync5(subroutine.sourcePath)) {
-    const sourceFile = subroutine.sourcePath;
-    const sourceContent = readFileSync3(sourceFile, "utf-8");
-    const parser = new DiracParser();
-    try {
-      const ast = parser.parse(sourceContent);
-      const subroutineCount = countSubroutines(ast);
-      if (subroutineCount === 1) {
-        filePath = sourceFile;
-        emit(session, `Updating single-subroutine file: ${filePath}
-`);
-      } else {
-        emit(session, `Source file contains ${subroutineCount} subroutines - creating separate file
-`);
-        const timestamp = (/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-").slice(0, -5);
-        const defaultDir = join4(homedir4(), ".dirac", "lib", timestamp);
-        filePath = join4(defaultDir, `${name}.di`);
-      }
-    } catch (error) {
-      emit(session, `Warning: Could not parse source file - creating new file
-`);
-      const timestamp = (/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-").slice(0, -5);
-      const defaultDir = join4(homedir4(), ".dirac", "lib", timestamp);
-      filePath = join4(defaultDir, `${name}.di`);
-    }
   } else {
-    const timestamp = (/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-").slice(0, -5);
-    const defaultDir = join4(homedir4(), ".dirac", "lib", timestamp);
+    const defaultDir = join4(homedir4(), ".dirac", "lib", "user");
     filePath = join4(defaultDir, `${name}.di`);
   }
   const dir = dirname4(filePath);
@@ -2395,12 +2375,16 @@ async function executeSaveSubroutine(session, element) {
   writeFileSync4(filePath, content, "utf-8");
   emit(session, `Subroutine '${name}' saved to: ${filePath}
 `);
+  const savedSub = session.subroutines.find((s) => s.name === name);
+  if (savedSub) {
+    savedSub.sourcePath = filePath;
+    savedSub.modified = false;
+  }
   if (session.debug) {
     console.error(`[save-subroutine] Saved '${name}' to: ${filePath}`);
   }
   try {
-    const registry3 = new SubroutineRegistry();
-    const count = registry3.indexFile(filePath);
+    const count = registry.indexFile(filePath);
     if (session.debug) {
       console.error(`[save-subroutine] Re-indexed ${filePath}: ${count} subroutine(s)`);
     }
@@ -2518,21 +2502,9 @@ function serializeChildrenBraKet(children, indent) {
 function escapeXml3(text) {
   return text.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;").replace(/"/g, "&quot;").replace(/'/g, "&apos;");
 }
-function countSubroutines(element) {
-  let count = 0;
-  if (element.tag === "subroutine") {
-    count = 1;
-  }
-  if (element.children) {
-    for (const child of element.children) {
-      count += countSubroutines(child);
-    }
-  }
-  return count;
-}
 
 // src/tags/edit-subroutine.ts
-import { writeFileSync as writeFileSync5, readFileSync as readFileSync4, unlinkSync } from "fs";
+import { writeFileSync as writeFileSync5, readFileSync as readFileSync3, unlinkSync, existsSync as existsSync6 } from "fs";
 import { tmpdir } from "os";
 import { join as join5 } from "path";
 import { spawnSync } from "child_process";
@@ -2552,7 +2524,36 @@ async function executeEditSubroutine(session, element) {
   if (!subroutine) {
     throw new Error(`Subroutine '${name}' not found in session`);
   }
-  const xml = serializeSubroutineToXML(subroutine);
+  let xml;
+  if (subroutine.sourcePath && existsSync6(subroutine.sourcePath)) {
+    try {
+      const sourceContent = readFileSync3(subroutine.sourcePath, "utf-8");
+      const match = sourceContent.match(
+        new RegExp(`<subroutine\\s+name="${name}"[\\s\\S]*?<\\/subroutine>`, "i")
+      );
+      if (match) {
+        xml = match[0];
+        if (session.debug) {
+          console.error(`[edit-subroutine] Loaded from source: ${subroutine.sourcePath}`);
+        }
+      } else {
+        xml = serializeSubroutineToXML(subroutine);
+        if (session.debug) {
+          console.error(`[edit-subroutine] Could not extract from source, using serialization`);
+        }
+      }
+    } catch (err) {
+      xml = serializeSubroutineToXML(subroutine);
+      if (session.debug) {
+        console.error(`[edit-subroutine] Error reading source file, using serialization`);
+      }
+    }
+  } else {
+    xml = serializeSubroutineToXML(subroutine);
+    if (session.debug) {
+      console.error(`[edit-subroutine] No source file, serializing from AST`);
+    }
+  }
   const tempFile = join5(tmpdir(), `dirac-edit-${name}-${Date.now()}.di`);
   writeFileSync5(tempFile, xml, "utf-8");
   if (session.debug) {
@@ -2572,7 +2573,7 @@ async function executeEditSubroutine(session, element) {
     unlinkSync(tempFile);
     throw new Error(`Editor exited with code ${result.status}`);
   }
-  const editedContent = readFileSync4(tempFile, "utf-8");
+  const editedContent = readFileSync3(tempFile, "utf-8");
   unlinkSync(tempFile);
   if (session.debug) {
     console.error(`[edit-subroutine] Editor closed, re-importing subroutine`);
@@ -2580,6 +2581,10 @@ async function executeEditSubroutine(session, element) {
   const parser = new DiracParser();
   const ast = parser.parse(editedContent);
   await integrate(session, ast);
+  const editedSub = session.subroutines.find((s) => s.name === name);
+  if (editedSub) {
+    editedSub.modified = true;
+  }
   emit(session, `Subroutine '${name}' updated in session (use save-subroutine to persist)
 `);
 }
@@ -2591,9 +2596,14 @@ function serializeSubroutineToXML(sub) {
   return lines.join("\n");
 }
 function serializeElement2(el, lines, indent) {
-  if (!el || !el.tag) {
-    if (el && typeof el === "string") {
-      lines.push(indent + el);
+  if (!el.tag || el.tag === "") {
+    if (el.text) {
+      let lastIdx = lines.length - 1;
+      if (lastIdx >= 0 && !lines[lastIdx].endsWith(">")) {
+        lines[lastIdx] += el.text;
+      } else {
+        lines.push(indent + el.text);
+      }
     }
     return;
   }
@@ -2606,22 +2616,46 @@ function serializeElement2(el, lines, indent) {
     }
   }
   const hasChildren = el.children && el.children.length > 0;
-  const hasText = el.text && el.text.trim();
-  if (!hasChildren && !hasText) {
-    lines.push(tag + " />");
-  } else if (hasText && !hasChildren) {
-    lines.push(tag + ">" + el.text + `</${el.tag}>`);
+  if (!hasChildren) {
+    let lastIdx = lines.length - 1;
+    if (lastIdx >= 0 && !lines[lastIdx].endsWith(">") && !lines[lastIdx].trim().startsWith("<")) {
+      lines[lastIdx] += tag.slice(indent.length) + " />";
+    } else {
+      lines.push(tag + " />");
+    }
   } else {
-    lines.push(tag + ">");
-    if (hasText) {
-      lines.push(indent + "  " + el.text);
+    let lastIdx = lines.length - 1;
+    const shouldInline = lastIdx >= 0 && !lines[lastIdx].endsWith(">");
+    if (shouldInline) {
+      lines[lastIdx] += tag.slice(indent.length) + ">";
+    } else {
+      lines.push(tag + ">");
     }
-    if (hasChildren) {
-      for (const child of el.children) {
-        serializeElement2(child, lines, indent + "  ");
+    let allInline = true;
+    for (let i = 0; i < el.children.length; i++) {
+      const child = el.children[i];
+      if (!child.tag || child.tag === "") {
+        lastIdx = lines.length - 1;
+        if (child.text) {
+          lines[lastIdx] += child.text;
+        }
+      } else {
+        const isSimpleVar = child.tag === "variable" && child.attributes && child.attributes.name;
+        if (isSimpleVar) {
+          lastIdx = lines.length - 1;
+          lines[lastIdx] += `<variable name="${child.attributes.name}" />`;
+        } else {
+          allInline = false;
+          serializeElement2(child, lines, indent + "  ");
+        }
       }
     }
-    lines.push(`${indent}</${el.tag}>`);
+    lastIdx = lines.length - 1;
+    if (allInline || lines[lastIdx].indexOf(`<${el.tag}`) !== -1) {
+      lines[lastIdx] += `</${el.tag}>`;
+    } else {
+      lines.push(`${indent}</${el.tag}>`);
+    }
   }
 }
 
@@ -2645,7 +2679,7 @@ async function executeForeach(session, element) {
     const parser2 = new DiracParser2();
     try {
       const fromElement = parser2.parse(fromAttr);
-      const { integrate: integrate2 } = await import("./interpreter-N4ERLMWW.js");
+      const { integrate: integrate2 } = await import("./interpreter-Y2AU6RG6.js");
       await integrate2(session, fromElement);
     } catch (e) {
       session.output = savedOutput;

package/dist/{chunk-65TBONV6.js → chunk-YVI7T3DS.js}

@@ -1,6 +1,6 @@
 import {
   integrate
-} from "./chunk-QUFYZS2S.js";
+} from "./chunk-QPO3WKH2.js";
 import {
   DiracParser
 } from "./chunk-HRHAMPOB.js";

package/dist/cli.js

@@ -4,8 +4,8 @@ import {
 } from "./chunk-AJSYOXXZ.js";
 import {
   execute
-} from "./chunk-6SKZC4N7.js";
-import "./chunk-QUFYZS2S.js";
+} from "./chunk-PLH7YN7I.js";
+import "./chunk-QPO3WKH2.js";
 import "./chunk-HRHAMPOB.js";
 import "./chunk-VC23AJJJ.js";
 import "./chunk-M57VI7KL.js";
@@ -16,7 +16,7 @@ import "dotenv/config";
 // package.json
 var package_default = {
   name: "dirac-lang",
-  version: "0.1.68",
+  version: "0.1.70",
   description: "LLM-Augmented Declarative Execution",
   type: "module",
   main: "dist/index.js",
@@ -109,7 +109,7 @@ async function main() {
   const args = process.argv.slice(2);
   const calledAs = process.argv[1];
   if (calledAs && calledAs.endsWith("/dish")) {
-    const { DiracShell } = await import("./shell-BYGHH5YH.js");
+    const { DiracShell } = await import("./shell-BTRWJIV3.js");
     const shellConfig = loadShellConfig(args);
     const shell = new DiracShell(shellConfig);
     await shell.start();
@@ -158,11 +158,11 @@ async function main() {
   if (args[0] === "agent") {
     const subcommand = args[1];
     if (subcommand === "daemon") {
-      const { runAgentDaemon } = await import("./agent-RH5SYOLV.js");
+      const { runAgentDaemon } = await import("./agent-WJUVRA6D.js");
       await runAgentDaemon();
       return;
     }
-    const { AgentCLI } = await import("./agent-RH5SYOLV.js");
+    const { AgentCLI } = await import("./agent-WJUVRA6D.js");
     const agent = new AgentCLI();
     switch (subcommand) {
       case "start":
@@ -189,8 +189,8 @@ async function main() {
     return;
   }
   if (args[0] === "shell") {
-    const { DiracShell } = await import("./shell-BYGHH5YH.js");
-    const { SessionServer, isSessionRunning, getSocketPath } = await import("./session-server-XV4RB5IW.js");
+    const { DiracShell } = await import("./shell-BTRWJIV3.js");
+    const { SessionServer, isSessionRunning, getSocketPath } = await import("./session-server-L5HBRT5P.js");
     const { SessionClient } = await import("./session-client-3VTC5MLO.js");
     const daemonMode = args.includes("--daemon") || args.includes("-d");
     const agentMode = args.includes("--agent") || args.includes("-a");
@@ -236,7 +236,7 @@ async function main() {
     return;
   }
   if (args[0] === "daemon") {
-    const { SessionServer } = await import("./session-server-XV4RB5IW.js");
+    const { SessionServer } = await import("./session-server-L5HBRT5P.js");
     const server = new SessionServer();
     await server.start();
     console.log("Session daemon started");

package/dist/{cron-VEWKZ2ZU.js → cron-HFJ3HK5G.js}

@@ -3,7 +3,7 @@ import {
   listCronJobs,
   stopAllCronJobs,
   stopCronJob
-} from "./chunk-QUFYZS2S.js";
+} from "./chunk-QPO3WKH2.js";
 import "./chunk-HRHAMPOB.js";
 import "./chunk-VC23AJJJ.js";
 import "./chunk-M57VI7KL.js";

package/dist/index.d.ts

@@ -58,6 +58,7 @@ interface Subroutine {
     parameters?: ParameterMetadata[];
     meta?: Record<string, string>;
     sourcePath?: string;
+    modified?: boolean;
 }
 /**
  * Execution context - maps to MaskSession in MASK

package/dist/index.js

@@ -2,10 +2,10 @@ import {
   createLLMAdapter,
   execute,
   executeUserCommand
-} from "./chunk-6SKZC4N7.js";
+} from "./chunk-PLH7YN7I.js";
 import {
   integrate
-} from "./chunk-QUFYZS2S.js";
+} from "./chunk-QPO3WKH2.js";
 import {
   DiracParser
 } from "./chunk-HRHAMPOB.js";

package/dist/{interpreter-N4ERLMWW.js → interpreter-Y2AU6RG6.js}

@@ -1,7 +1,7 @@
 import {
   integrate,
   integrateChildren
-} from "./chunk-QUFYZS2S.js";
+} from "./chunk-QPO3WKH2.js";
 import "./chunk-HRHAMPOB.js";
 import "./chunk-VC23AJJJ.js";
 import "./chunk-M57VI7KL.js";

package/dist/{run-at-OPGYI26L.js → run-at-JFHLEZNU.js}

@@ -3,7 +3,7 @@ import {
   cancelScheduledRun,
   executeRunAt,
   listScheduledRuns
-} from "./chunk-QUFYZS2S.js";
+} from "./chunk-QPO3WKH2.js";
 import "./chunk-HRHAMPOB.js";
 import "./chunk-VC23AJJJ.js";
 import "./chunk-M57VI7KL.js";

package/dist/{schedule-ZGPBTVXG.js → schedule-R3ZBENYA.js}

@@ -3,7 +3,7 @@ import {
   listScheduledTasks,
   stopAllScheduledTasks,
   stopScheduledTask
-} from "./chunk-QUFYZS2S.js";
+} from "./chunk-QPO3WKH2.js";
 import "./chunk-HRHAMPOB.js";
 import "./chunk-VC23AJJJ.js";
 import "./chunk-M57VI7KL.js";

package/dist/{session-server-XV4RB5IW.js → session-server-L5HBRT5P.js}

@@ -2,8 +2,8 @@ import {
   SessionServer,
   getSocketPath,
   isSessionRunning
-} from "./chunk-65TBONV6.js";
-import "./chunk-QUFYZS2S.js";
+} from "./chunk-YVI7T3DS.js";
+import "./chunk-QPO3WKH2.js";
 import "./chunk-HRHAMPOB.js";
 import "./chunk-VC23AJJJ.js";
 import "./chunk-M57VI7KL.js";

package/dist/{shell-BYGHH5YH.js → shell-BTRWJIV3.js}

@@ -4,7 +4,7 @@ import {
 } from "./chunk-AJSYOXXZ.js";
 import {
   integrate
-} from "./chunk-QUFYZS2S.js";
+} from "./chunk-QPO3WKH2.js";
 import {
   DiracParser
 } from "./chunk-HRHAMPOB.js";
@@ -271,7 +271,7 @@ var DiracShell = class {
     } catch (err) {
     }
   }
-  checkUnsavedSubroutines() {
+  getUnsavedSubroutines() {
     const unsaved = [];
     const excludePaths = [
       path.join(os.homedir(), ".dirac", "lib"),
@@ -283,8 +283,12 @@ var DiracShell = class {
       if (sub.meta && sub.meta["hide-from-llm"] === "true") {
         continue;
       }
+      if (sub.modified) {
+        unsaved.push(sub.name);
+        continue;
+      }
       if (!sub.sourcePath) {
-        unsaved.push({ name: sub.name, reason: "created in session" });
+        unsaved.push(sub.name);
       } else {
         const isExcluded = excludePaths.some(
           (excludePath) => sub.sourcePath.startsWith(excludePath)
@@ -292,28 +296,51 @@ var DiracShell = class {
         if (isExcluded) {
           continue;
         }
-        if (fs.existsSync(sub.sourcePath)) {
-          try {
-            const content = fs.readFileSync(sub.sourcePath, "utf-8");
-            const parser = new DiracParser();
-            const ast = parser.parse(content);
-            const count = this.countSubroutinesInAST(ast);
-            if (count > 1) {
-              const shortPath = sub.sourcePath.replace(os.homedir(), "~");
-              unsaved.push({ name: sub.name, reason: `from multi-sub file (${shortPath})` });
-            }
-          } catch (err) {
-          }
-        }
       }
     }
-    if (unsaved.length > 0) {
-      console.log("\n\u26A0\uFE0F  Warning: You have unsaved subroutines:");
-      for (const { name, reason } of unsaved) {
-        console.log(`   - ${name} (${reason})`);
-      }
-      console.log('\nUse :save <name> or |save-subroutine name="..."> to persist them.\n');
+    return unsaved;
+  }
+  async promptSaveUnsaved(unsaved) {
+    console.log("\n\u26A0\uFE0F  Warning: You have unsaved subroutines created in this session:");
+    for (const name of unsaved) {
+      console.log(`   - ${name}`);
     }
+    console.log("\nOptions:");
+    console.log("  a - Save all and exit");
+    console.log("  n - Exit without saving");
+    console.log("  c - Cancel (return to shell)");
+    return new Promise((resolve2) => {
+      const confirmRl = readline.createInterface({
+        input: process.stdin,
+        output: process.stdout
+      });
+      confirmRl.question("\nYour choice [a/n/c]: ", async (answer) => {
+        confirmRl.close();
+        const choice = answer.trim().toLowerCase();
+        if (choice === "a") {
+          console.log("\nSaving all unsaved subroutines...\n");
+          for (const name of unsaved) {
+            try {
+              const xml = `<save-subroutine name="${name}" format="xml" />`;
+              const ast = this.xmlParser.parse(xml);
+              await integrate(this.session, ast);
+              if (this.session.output.length > 0) {
+                console.log(this.session.output.join(""));
+                this.session.output = [];
+              }
+            } catch (error) {
+              console.error(`Error saving ${name}:`, error instanceof Error ? error.message : String(error));
+            }
+          }
+          resolve2(true);
+        } else if (choice === "n") {
+          resolve2(true);
+        } else {
+          console.log("\n(Exit canceled - returning to shell)\n");
+          resolve2(false);
+        }
+      });
+    });
   }
   countSubroutinesInAST(element) {
     let count = 0;
@@ -327,24 +354,44 @@ var DiracShell = class {
     }
     return count;
   }
+  finalizeExit() {
+    this.saveHistory();
+    if (this.client) {
+      this.client.disconnect();
+    }
+    import("./schedule-R3ZBENYA.js").then(({ stopAllScheduledTasks }) => {
+      stopAllScheduledTasks();
+      console.log("\nGoodbye!");
+      process.exit(0);
+    }).catch(() => {
+      console.log("\nGoodbye!");
+      process.exit(0);
+    });
+  }
   setupHandlers() {
     this.rl.on("line", async (input) => {
       await this.handleInput(input);
     });
-    this.rl.on("close", () => {
-      this.checkUnsavedSubroutines();
-      this.saveHistory();
-      if (this.client) {
-        this.client.disconnect();
+    this.rl.on("close", async () => {
+      const unsaved = this.getUnsavedSubroutines();
+      if (unsaved.length > 0) {
+        const shouldExit = await this.promptSaveUnsaved(unsaved);
+        if (shouldExit) {
+          this.finalizeExit();
+        } else {
+          this.rl = readline.createInterface({
+            input: process.stdin,
+            output: process.stdout,
+            prompt: "> ",
+            historySize: MAX_HISTORY,
+            completer: this.completer.bind(this)
+          });
+          this.setupHandlers();
+          this.rl.prompt();
+        }
+      } else {
+        this.finalizeExit();
       }
-      import("./schedule-ZGPBTVXG.js").then(({ stopAllScheduledTasks }) => {
-        stopAllScheduledTasks();
-        console.log("\nGoodbye!");
-        process.exit(0);
-      }).catch(() => {
-        console.log("\nGoodbye!");
-        process.exit(0);
-      });
     });
     this.rl.on("SIGINT", () => {
       if (this.inputBuffer.length > 0) {
@@ -703,7 +750,7 @@ Examples:
         break;
       case "tasks":
         try {
-          const { listScheduledTasks } = await import("./schedule-ZGPBTVXG.js");
+          const { listScheduledTasks } = await import("./schedule-R3ZBENYA.js");
           const tasks = listScheduledTasks();
           if (tasks.length === 0) {
             console.log("No scheduled tasks running.");
@@ -722,7 +769,7 @@ Examples:
           console.log("Usage: :stop <task-name>");
         } else {
           try {
-            const { stopScheduledTask } = await import("./schedule-ZGPBTVXG.js");
+            const { stopScheduledTask } = await import("./schedule-R3ZBENYA.js");
             const taskName = args[0];
             const stopped = stopScheduledTask(taskName);
             if (stopped) {
@@ -737,7 +784,7 @@ Examples:
         break;
       case "stopall":
         try {
-          const { stopAllScheduledTasks } = await import("./schedule-ZGPBTVXG.js");
+          const { stopAllScheduledTasks } = await import("./schedule-R3ZBENYA.js");
           stopAllScheduledTasks();
           console.log("All scheduled tasks stopped.");
         } catch (error) {
@@ -746,7 +793,7 @@ Examples:
         break;
       case "crons":
         try {
-          const { listCronJobs } = await import("./cron-VEWKZ2ZU.js");
+          const { listCronJobs } = await import("./cron-HFJ3HK5G.js");
           const jobs = listCronJobs();
           if (jobs.length === 0) {
             console.log("No cron jobs running.");
@@ -766,7 +813,7 @@ Examples:
           console.log("Usage: :stopcron <job-name>");
         } else {
           try {
-            const { stopCronJob } = await import("./cron-VEWKZ2ZU.js");
+            const { stopCronJob } = await import("./cron-HFJ3HK5G.js");
             const jobName = args[0];
             const stopped = stopCronJob(jobName);
             if (stopped) {
@@ -781,7 +828,7 @@ Examples:
         break;
       case "stopallcrons":
         try {
-          const { stopAllCronJobs } = await import("./cron-VEWKZ2ZU.js");
+          const { stopAllCronJobs } = await import("./cron-HFJ3HK5G.js");
           stopAllCronJobs();
           console.log("All cron jobs stopped.");
         } catch (error) {
@@ -790,7 +837,7 @@ Examples:
         break;
       case "scheduled":
         try {
-          const { listScheduledRuns } = await import("./run-at-OPGYI26L.js");
+          const { listScheduledRuns } = await import("./run-at-JFHLEZNU.js");
           const runs = listScheduledRuns();
           if (runs.length === 0) {
             console.log("No scheduled runs pending.");
@@ -810,7 +857,7 @@ Examples:
           console.log("Usage: :cancel <run-name>");
         } else {
           try {
-            const { cancelScheduledRun } = await import("./run-at-OPGYI26L.js");
+            const { cancelScheduledRun } = await import("./run-at-JFHLEZNU.js");
             const runName = args[0];
             const cancelled = cancelScheduledRun(runName);
             if (cancelled) {
@@ -825,7 +872,7 @@ Examples:
         break;
       case "cancelall":
         try {
-          const { cancelAllScheduledRuns } = await import("./run-at-OPGYI26L.js");
+          const { cancelAllScheduledRuns } = await import("./run-at-JFHLEZNU.js");
           cancelAllScheduledRuns();
           console.log("All scheduled runs cancelled.");
         } catch (error) {

package/dist/test-runner.js

@@ -1,6 +1,6 @@
 import {
   integrate
-} from "./chunk-QUFYZS2S.js";
+} from "./chunk-QPO3WKH2.js";
 import {
   DiracParser
 } from "./chunk-HRHAMPOB.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dirac-lang",
-  "version": "0.1.69",
+  "version": "0.1.71",
   "description": "LLM-Augmented Declarative Execution",
   "type": "module",
   "main": "dist/index.js",