dirac-lang 0.1.60 → 0.1.61

package/dist/agent-OXQZBLNQ.js

@@ -0,0 +1,226 @@
+import {
+  SessionServer,
+  getSocketPath,
+  isSessionRunning
+} from "./chunk-AU3YU63U.js";
+import "./chunk-ZAA6R2TC.js";
+import "./chunk-HRHAMPOB.js";
+import "./chunk-NKA6ZJDV.js";
+import "./chunk-3UW6GWYQ.js";
+
+// src/agent.ts
+import fs from "fs";
+import path from "path";
+import os from "os";
+import { spawn } from "child_process";
+var AGENT_DIR = path.join(os.homedir(), ".dirac");
+var PID_FILE = path.join(AGENT_DIR, "session.pid");
+var LOG_FILE = path.join(AGENT_DIR, "agent.log");
+var AgentCLI = class {
+  async start() {
+    if (isSessionRunning()) {
+      console.log("Agent is already running");
+      const pid = this.getRunningPid();
+      if (pid) {
+        console.log(`PID: ${pid}`);
+        console.log(`Socket: ${getSocketPath()}`);
+        console.log(`Logs: ${LOG_FILE}`);
+      }
+      return;
+    }
+    if (!fs.existsSync(AGENT_DIR)) {
+      fs.mkdirSync(AGENT_DIR, { recursive: true });
+    }
+    console.log("Starting DIRAC agent...");
+    const logFd = fs.openSync(LOG_FILE, "a");
+    const child = spawn(
+      process.argv[0],
+      // node executable
+      [process.argv[1], "agent", "daemon"],
+      // dirac agent daemon
+      {
+        detached: true,
+        stdio: ["ignore", logFd, logFd],
+        // Redirect stdout/stderr to log file
+        cwd: process.cwd()
+      }
+    );
+    fs.closeSync(logFd);
+    child.unref();
+    let attempts = 0;
+    const maxAttempts = 10;
+    while (attempts < maxAttempts) {
+      await new Promise((resolve) => setTimeout(resolve, 500));
+      if (isSessionRunning()) {
+        console.log("\u2713 Agent started successfully");
+        console.log(`  PID: ${child.pid}`);
+        console.log(`  Socket: ${getSocketPath()}`);
+        console.log(`  Logs: ${LOG_FILE}`);
+        console.log("");
+        console.log("Connect with: dirac shell --agent");
+        return;
+      }
+      attempts++;
+    }
+    console.error("\u2717 Failed to start agent");
+    console.error("Check logs:", LOG_FILE);
+    process.exit(1);
+  }
+  async stop() {
+    const pid = this.getRunningPid();
+    if (!pid) {
+      console.log("Agent is not running");
+      return;
+    }
+    console.log(`Stopping agent (PID: ${pid})...`);
+    try {
+      process.kill(pid, "SIGTERM");
+      let attempts = 0;
+      while (attempts < 10) {
+        await new Promise((resolve) => setTimeout(resolve, 500));
+        try {
+          process.kill(pid, 0);
+          attempts++;
+        } catch {
+          console.log("\u2713 Agent stopped");
+          this.cleanup();
+          return;
+        }
+      }
+      console.log("Agent did not stop gracefully, forcing...");
+      process.kill(pid, "SIGKILL");
+      this.cleanup();
+      console.log("\u2713 Agent stopped (forced)");
+    } catch (error) {
+      if (error.code === "ESRCH") {
+        console.log("Agent was not running");
+        this.cleanup();
+      } else {
+        console.error("Error stopping agent:", error.message);
+        process.exit(1);
+      }
+    }
+  }
+  async status() {
+    const pid = this.getRunningPid();
+    if (!pid) {
+      console.log("Status: NOT RUNNING");
+      return;
+    }
+    try {
+      process.kill(pid, 0);
+      console.log("Status: RUNNING");
+      console.log(`PID: ${pid}`);
+      console.log(`Socket: ${getSocketPath()}`);
+      console.log(`Logs: ${LOG_FILE}`);
+      if (fs.existsSync(PID_FILE)) {
+        const stats = fs.statSync(PID_FILE);
+        const uptime = Date.now() - stats.mtimeMs;
+        console.log(`Uptime: ${this.formatUptime(uptime)}`);
+      }
+    } catch (error) {
+      if (error.code === "ESRCH") {
+        console.log("Status: NOT RUNNING (stale PID file)");
+        this.cleanup();
+      } else {
+        throw error;
+      }
+    }
+  }
+  async restart() {
+    console.log("Restarting agent...");
+    await this.stop();
+    await new Promise((resolve) => setTimeout(resolve, 1e3));
+    await this.start();
+  }
+  async logs(follow = false) {
+    if (!fs.existsSync(LOG_FILE)) {
+      console.log("No logs found");
+      return;
+    }
+    if (follow) {
+      const { spawn: spawn2 } = await import("child_process");
+      const tail = spawn2("tail", ["-f", LOG_FILE], {
+        stdio: "inherit"
+      });
+      process.on("SIGINT", () => {
+        tail.kill();
+        process.exit(0);
+      });
+    } else {
+      const content = fs.readFileSync(LOG_FILE, "utf-8");
+      const lines = content.split("\n");
+      const lastLines = lines.slice(-50);
+      console.log(lastLines.join("\n"));
+    }
+  }
+  getRunningPid() {
+    if (!fs.existsSync(PID_FILE)) {
+      return null;
+    }
+    try {
+      const pidStr = fs.readFileSync(PID_FILE, "utf-8").trim();
+      return parseInt(pidStr, 10);
+    } catch {
+      return null;
+    }
+  }
+  cleanup() {
+    if (fs.existsSync(PID_FILE)) {
+      fs.unlinkSync(PID_FILE);
+    }
+  }
+  formatUptime(ms) {
+    const seconds = Math.floor(ms / 1e3);
+    const minutes = Math.floor(seconds / 60);
+    const hours = Math.floor(minutes / 60);
+    const days = Math.floor(hours / 24);
+    if (days > 0) return `${days}d ${hours % 24}h`;
+    if (hours > 0) return `${hours}h ${minutes % 60}m`;
+    if (minutes > 0) return `${minutes}m`;
+    return `${seconds}s`;
+  }
+};
+async function runAgentDaemon() {
+  console.log(`[${(/* @__PURE__ */ new Date()).toISOString()}] Starting DIRAC agent daemon`);
+  let config = {};
+  try {
+    const fs2 = await import("fs");
+    const path2 = await import("path");
+    const yaml = await import("js-yaml");
+    const configPath = path2.default.resolve(process.cwd(), "config.yml");
+    if (fs2.default.existsSync(configPath)) {
+      const configData = yaml.default.load(fs2.default.readFileSync(configPath, "utf-8"));
+      config = {
+        llmProvider: configData.llmProvider,
+        llmModel: configData.llmModel,
+        customLLMUrl: configData.customLLMUrl,
+        initScript: configData.initScript
+      };
+      console.log(`[${(/* @__PURE__ */ new Date()).toISOString()}] Loaded config: LLM=${config.llmProvider}/${config.llmModel}`);
+    }
+  } catch (err) {
+    console.log(`[${(/* @__PURE__ */ new Date()).toISOString()}] No config.yml found, using defaults`);
+  }
+  const server = new SessionServer(config);
+  try {
+    await server.start();
+    console.log(`[${(/* @__PURE__ */ new Date()).toISOString()}] Agent daemon started successfully`);
+    const shutdown = async () => {
+      console.log(`[${(/* @__PURE__ */ new Date()).toISOString()}] Shutting down agent daemon`);
+      await server.shutdown();
+      process.exit(0);
+    };
+    process.on("SIGTERM", shutdown);
+    process.on("SIGINT", shutdown);
+    await new Promise(() => {
+    });
+  } catch (error) {
+    console.error(`[${(/* @__PURE__ */ new Date()).toISOString()}] Failed to start agent:`, error);
+    process.exit(1);
+  }
+}
+export {
+  AgentCLI,
+  runAgentDaemon
+};

package/dist/chunk-AU3YU63U.js

@@ -0,0 +1,208 @@
+import {
+  integrate
+} from "./chunk-ZAA6R2TC.js";
+import {
+  DiracParser
+} from "./chunk-HRHAMPOB.js";
+import {
+  createSession
+} from "./chunk-3UW6GWYQ.js";
+
+// src/session-server.ts
+import net from "net";
+import fs from "fs";
+import path from "path";
+import os from "os";
+var SOCKET_DIR = path.join(os.homedir(), ".dirac");
+var SOCKET_PATH = path.join(SOCKET_DIR, "session.sock");
+var PID_PATH = path.join(SOCKET_DIR, "session.pid");
+var SessionServer = class {
+  server;
+  session;
+  clients = /* @__PURE__ */ new Map();
+  clientIdCounter = 0;
+  constructor(config) {
+    this.session = createSession(config || {});
+    this.server = net.createServer(this.handleConnection.bind(this));
+  }
+  async start() {
+    if (!fs.existsSync(SOCKET_DIR)) {
+      fs.mkdirSync(SOCKET_DIR, { recursive: true });
+    }
+    if (fs.existsSync(SOCKET_PATH)) {
+      fs.unlinkSync(SOCKET_PATH);
+    }
+    return new Promise((resolve, reject) => {
+      this.server.listen(SOCKET_PATH, () => {
+        console.log(`[session-server] Listening on ${SOCKET_PATH}`);
+        fs.writeFileSync(PID_PATH, process.pid.toString());
+        fs.chmodSync(SOCKET_PATH, 384);
+        resolve();
+      });
+      this.server.on("error", (error) => {
+        console.error("[session-server] Error:", error);
+        reject(error);
+      });
+    });
+  }
+  handleConnection(socket) {
+    const clientId = `client-${++this.clientIdCounter}`;
+    const client = { id: clientId, socket };
+    this.clients.set(clientId, client);
+    console.log(`[session-server] Client connected: ${clientId}`);
+    let buffer = "";
+    socket.on("data", async (data) => {
+      buffer += data.toString();
+      const lines = buffer.split("\n");
+      buffer = lines.pop() || "";
+      for (const line of lines) {
+        if (!line.trim()) continue;
+        try {
+          const message = JSON.parse(line);
+          await this.handleMessage(client, message);
+        } catch (error) {
+          this.sendError(client, `Invalid message: ${error}`);
+        }
+      }
+    });
+    socket.on("end", () => {
+      console.log(`[session-server] Client disconnected: ${clientId}`);
+      this.clients.delete(clientId);
+    });
+    socket.on("error", (error) => {
+      console.error(`[session-server] Client error (${clientId}):`, error);
+      this.clients.delete(clientId);
+    });
+    this.sendResponse(client, {
+      type: "welcome",
+      sessionId: clientId,
+      message: "Connected to DIRAC session"
+    });
+  }
+  async handleMessage(client, message) {
+    const { type, data } = message;
+    switch (type) {
+      case "execute":
+        await this.executeCommand(client, data.command);
+        break;
+      case "getVars":
+        this.sendResponse(client, {
+          type: "vars",
+          data: Object.keys(this.session.variables)
+        });
+        break;
+      case "getState":
+        this.sendResponse(client, {
+          type: "state",
+          data: {
+            variables: this.session.variables,
+            subroutines: this.session.subroutines,
+            varBoundary: this.session.varBoundary,
+            subBoundary: this.session.subBoundary
+          }
+        });
+        break;
+      case "shutdown":
+        this.sendResponse(client, {
+          type: "shutdown",
+          message: "Server shutting down"
+        });
+        await this.shutdown();
+        break;
+      default:
+        this.sendError(client, `Unknown message type: ${type}`);
+    }
+  }
+  async executeCommand(client, command) {
+    try {
+      this.session.output = [];
+      const parser = new DiracParser();
+      const ast = parser.parse(command);
+      await integrate(this.session, ast);
+      const output = this.session.output.join("");
+      this.sendResponse(client, {
+        type: "output",
+        data: output
+      });
+    } catch (error) {
+      this.sendError(client, error.message);
+    }
+  }
+  sendResponse(client, response) {
+    try {
+      client.socket.write(JSON.stringify(response) + "\n");
+    } catch (error) {
+      console.error(`[session-server] Failed to send response:`, error);
+    }
+  }
+  sendError(client, message) {
+    this.sendResponse(client, {
+      type: "error",
+      message
+    });
+  }
+  async shutdown() {
+    console.log("[session-server] Shutting down...");
+    for (const client of this.clients.values()) {
+      client.socket.end();
+    }
+    this.clients.clear();
+    return new Promise((resolve) => {
+      this.server.close(() => {
+        if (fs.existsSync(SOCKET_PATH)) {
+          fs.unlinkSync(SOCKET_PATH);
+        }
+        if (fs.existsSync(PID_PATH)) {
+          fs.unlinkSync(PID_PATH);
+        }
+        console.log("[session-server] Shutdown complete");
+        resolve();
+      });
+    });
+  }
+};
+function isSessionRunning() {
+  if (!fs.existsSync(PID_PATH)) {
+    return false;
+  }
+  try {
+    const pid = parseInt(fs.readFileSync(PID_PATH, "utf-8").trim(), 10);
+    process.kill(pid, 0);
+    return true;
+  } catch (error) {
+    if (fs.existsSync(PID_PATH)) fs.unlinkSync(PID_PATH);
+    try {
+      if (fs.existsSync(SOCKET_PATH)) fs.unlinkSync(SOCKET_PATH);
+    } catch (e) {
+    }
+    return false;
+  }
+}
+function getSocketPath() {
+  return SOCKET_PATH;
+}
+if (import.meta.url === `file://${process.argv[1]}`) {
+  const server = new SessionServer();
+  server.start().then(() => {
+    console.log("[session-server] Server started successfully");
+    process.on("SIGINT", async () => {
+      console.log("\n[session-server] Received SIGINT, shutting down...");
+      await server.shutdown();
+      process.exit(0);
+    });
+    process.on("SIGTERM", async () => {
+      console.log("\n[session-server] Received SIGTERM, shutting down...");
+      await server.shutdown();
+      process.exit(0);
+    });
+  }).catch((error) => {
+    console.error("[session-server] Failed to start:", error);
+    process.exit(1);
+  });
+}
+
+export {
+  SessionServer,
+  isSessionRunning,
+  getSocketPath
+};

package/dist/cli.js

@@ -75,9 +75,22 @@ async function main() {
   if (args.includes("--help") || args.includes("-h")) {
     console.log("Usage: dirac <file.di|file.bk>");
     console.log("       dirac shell [options]");
+    console.log("       dirac agent <command>");
     console.log("");
     console.log("Commands:");
     console.log("  shell             Start interactive shell (REPL)");
+    console.log("  shell --agent     Connect shell to running agent daemon");
+    console.log("  agent start       Start persistent agent daemon");
+    console.log("  agent stop        Stop agent daemon");
+    console.log("  agent status      Check agent status");
+    console.log("  agent restart     Restart agent daemon");
+    console.log("  agent logs        Show agent logs");
+    console.log("  shell --daemon    Start shell with persistent daemon (experimental)");
+    console.log("");
+    console.log("Shell tips:");
+    console.log("  Press Ctrl-Z to suspend shell, then `bg` to run in background");
+    console.log("  Use `fg` to bring it back to foreground");
+    console.log("  Cron jobs and run-at tasks continue running in background");
     console.log("");
     console.log("File formats:");
     console.log("  .di               XML notation (verbose)");
@@ -98,13 +111,54 @@ async function main() {
     console.log(package_default.version);
     process.exit(0);
   }
+  if (args[0] === "agent") {
+    const subcommand = args[1];
+    if (subcommand === "daemon") {
+      const { runAgentDaemon } = await import("./agent-OXQZBLNQ.js");
+      await runAgentDaemon();
+      return;
+    }
+    const { AgentCLI } = await import("./agent-OXQZBLNQ.js");
+    const agent = new AgentCLI();
+    switch (subcommand) {
+      case "start":
+        await agent.start();
+        break;
+      case "stop":
+        await agent.stop();
+        break;
+      case "status":
+        await agent.status();
+        break;
+      case "restart":
+        await agent.restart();
+        break;
+      case "logs":
+        const follow = args.includes("-f") || args.includes("--follow");
+        await agent.logs(follow);
+        break;
+      default:
+        console.error("Unknown agent command:", subcommand);
+        console.error("Available commands: start, stop, status, restart, logs");
+        process.exit(1);
+    }
+    return;
+  }
   if (args[0] === "shell") {
-    const { DiracShell } = await import("./shell-WFLXZW3I.js");
+    const { DiracShell } = await import("./shell-OFBL57OA.js");
+    const { SessionServer, isSessionRunning, getSocketPath } = await import("./session-server-QZN2RMYD.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");
     const shellConfig = { debug: false };
     for (let i = 1; i < args.length; i++) {
       const arg = args[i];
       if (arg === "--debug") {
         shellConfig.debug = true;
+      } else if (arg === "--daemon" || arg === "-d") {
+        continue;
+      } else if (arg === "--agent" || arg === "-a") {
+        continue;
       } else if ((arg === "-f" || arg === "--config") && i + 1 < args.length) {
         const configPath = resolve(args[++i]);
         if (fs.existsSync(configPath)) {
@@ -131,10 +185,59 @@ async function main() {
         }
       }
     }
+    if (agentMode) {
+      if (!isSessionRunning()) {
+        console.error("Error: No agent is running");
+        console.error("Start the agent first with: dirac agent start");
+        process.exit(1);
+      }
+      console.log("Connecting to agent...");
+      const client = new SessionClient({ socketPath: getSocketPath() });
+      client.on("error", (error) => {
+        console.error("Agent error:", error);
+      });
+      try {
+        await client.connect();
+        console.log("Connected to agent at", getSocketPath());
+        console.log("Session state is persistent across disconnects");
+        console.log("");
+        const shell2 = new DiracShell(shellConfig);
+        shell2.setClient(client);
+        process.on("SIGINT", async () => {
+          await client.disconnect();
+          process.exit(0);
+        });
+        await shell2.start();
+      } catch (error) {
+        console.error("Failed to connect to agent:", error);
+        process.exit(1);
+      }
+      return;
+    }
+    if (daemonMode) {
+      console.log("Note: Daemon mode is under development.");
+      console.log("For now, use Ctrl-Z + bg to background the shell.");
+      console.log("");
+    }
     const shell = new DiracShell(shellConfig);
     await shell.start();
     return;
   }
+  if (args[0] === "daemon") {
+    const { SessionServer } = await import("./session-server-QZN2RMYD.js");
+    const server = new SessionServer();
+    await server.start();
+    console.log("Session daemon started");
+    process.on("SIGINT", async () => {
+      await server.shutdown();
+      process.exit(0);
+    });
+    process.on("SIGTERM", async () => {
+      await server.shutdown();
+      process.exit(0);
+    });
+    return;
+  }
   if (args.length === 0) {
     console.error("Usage: dirac <file.di|file.bk>");
     console.error("       dirac shell [options]");

package/dist/session-client-3VTC5MLO.js

@@ -0,0 +1,177 @@
+// src/session-client.ts
+import net from "net";
+import { EventEmitter } from "events";
+var SessionClient = class extends EventEmitter {
+  socket = null;
+  buffer = "";
+  connected = false;
+  options;
+  constructor(options) {
+    super();
+    this.options = {
+      autoReconnect: false,
+      reconnectDelay: 1e3,
+      ...options
+    };
+  }
+  async connect() {
+    return new Promise((resolve, reject) => {
+      this.socket = net.connect(this.options.socketPath);
+      this.socket.on("connect", () => {
+        this.emit("socket-connected");
+      });
+      this.socket.on("data", (data) => {
+        this.buffer += data.toString();
+        const lines = this.buffer.split("\n");
+        this.buffer = lines.pop() || "";
+        for (const line of lines) {
+          if (!line.trim()) continue;
+          try {
+            const message = JSON.parse(line);
+            this.handleMessage(message);
+          } catch (error) {
+            this.emit("error", new Error(`Invalid message: ${error}`));
+          }
+        }
+      });
+      this.socket.on("end", () => {
+        this.connected = false;
+        this.emit("disconnected");
+        if (this.options.autoReconnect) {
+          setTimeout(() => this.connect(), this.options.reconnectDelay);
+        }
+      });
+      this.socket.on("error", (error) => {
+        this.connected = false;
+        reject(error);
+        this.emit("error", error);
+      });
+      this.once("welcome", () => {
+        this.connected = true;
+        resolve();
+      });
+    });
+  }
+  handleMessage(message) {
+    const { type, data, message: msg } = message;
+    switch (type) {
+      case "welcome":
+        this.emit("welcome", msg);
+        break;
+      case "output":
+        this.emit("output", data);
+        break;
+      case "error":
+        this.emit("error", new Error(msg || data));
+        break;
+      case "vars":
+        this.emit("vars", data);
+        break;
+      case "state":
+        this.emit("state", data);
+        break;
+      case "shutdown":
+        this.emit("shutdown");
+        break;
+      default:
+        this.emit("message", message);
+    }
+  }
+  async execute(command) {
+    return new Promise((resolve, reject) => {
+      if (!this.socket) {
+        reject(new Error("Not connected to session: socket is null"));
+        return;
+      }
+      if (!this.connected) {
+        reject(new Error("Not connected to session: connected flag is false"));
+        return;
+      }
+      this.send({ type: "execute", data: { command } });
+      const timeout = setTimeout(() => {
+        reject(new Error("Execution timeout"));
+      }, 3e4);
+      const onOutput = (output) => {
+        clearTimeout(timeout);
+        this.off("error", onError);
+        resolve(output);
+      };
+      const onError = (error) => {
+        clearTimeout(timeout);
+        this.off("output", onOutput);
+        reject(error);
+      };
+      this.once("output", onOutput);
+      this.once("error", onError);
+    });
+  }
+  async getVars() {
+    return new Promise((resolve, reject) => {
+      if (!this.connected || !this.socket) {
+        reject(new Error("Not connected to session"));
+        return;
+      }
+      this.send({ type: "getVars" });
+      const timeout = setTimeout(() => {
+        reject(new Error("Timeout getting variables"));
+      }, 5e3);
+      const onVars = (vars) => {
+        clearTimeout(timeout);
+        resolve(vars);
+      };
+      this.once("vars", onVars);
+    });
+  }
+  async getState() {
+    return new Promise((resolve, reject) => {
+      if (!this.socket) {
+        reject(new Error("Not connected to session: socket is null"));
+        return;
+      }
+      if (!this.connected) {
+        reject(new Error("Not connected to session: connected flag is false"));
+        return;
+      }
+      this.send({ type: "getState" });
+      const timeout = setTimeout(() => {
+        reject(new Error("Timeout getting state"));
+      }, 5e3);
+      const onState = (state) => {
+        clearTimeout(timeout);
+        resolve(state);
+      };
+      this.once("state", onState);
+    });
+  }
+  async shutdown() {
+    return new Promise((resolve) => {
+      if (!this.connected || !this.socket) {
+        resolve();
+        return;
+      }
+      this.send({ type: "shutdown" });
+      this.once("shutdown", () => {
+        resolve();
+      });
+    });
+  }
+  send(message) {
+    if (!this.socket || !this.connected) {
+      throw new Error("Not connected to session");
+    }
+    this.socket.write(JSON.stringify(message) + "\n");
+  }
+  disconnect() {
+    if (this.socket) {
+      this.socket.end();
+      this.socket = null;
+    }
+    this.connected = false;
+  }
+  isConnected() {
+    return this.connected;
+  }
+};
+export {
+  SessionClient
+};

package/dist/session-server-QZN2RMYD.js

@@ -0,0 +1,14 @@
+import {
+  SessionServer,
+  getSocketPath,
+  isSessionRunning
+} from "./chunk-AU3YU63U.js";
+import "./chunk-ZAA6R2TC.js";
+import "./chunk-HRHAMPOB.js";
+import "./chunk-NKA6ZJDV.js";
+import "./chunk-3UW6GWYQ.js";
+export {
+  SessionServer,
+  getSocketPath,
+  isSessionRunning
+};

package/dist/{shell-WFLXZW3I.js → shell-OFBL57OA.js}

@@ -23,6 +23,7 @@ var HISTORY_FILE = path.join(os.homedir(), ".dirac_history");
 var MAX_HISTORY = 1e3;
 var DiracShell = class {
   session;
+  client = null;
   braketParser;
   xmlParser;
   rl;
@@ -45,6 +46,12 @@ var DiracShell = class {
     this.loadHistory();
     this.setupHandlers();
   }
+  /**
+   * Set client for daemon mode
+   */
+  setClient(client) {
+    this.client = client;
+  }
   completer(line) {
     const attrMatch = line.match(/\|([a-z0-9_-]+)\s+.*?([a-z0-9_-]*)$/i);
     if (attrMatch) {
@@ -150,6 +157,9 @@ var DiracShell = class {
     });
     this.rl.on("close", () => {
       this.saveHistory();
+      if (this.client) {
+        this.client.disconnect();
+      }
       import("./schedule-QZ5U3YBS.js").then(({ stopAllScheduledTasks }) => {
         stopAllScheduledTasks();
         console.log("\nGoodbye!");
@@ -249,6 +259,17 @@ var DiracShell = class {
     this.inputBuffer = [];
     this.baseIndent = null;
     try {
+      if (this.client) {
+        const xml2 = this.braketParser.parse(input);
+        if (this.config.debug) {
+          console.log("[Debug] Sending to agent:\n", xml2);
+        }
+        const output = await this.client.execute(xml2);
+        if (output) {
+          console.log(output);
+        }
+        return;
+      }
       this.session.output = [];
       const xml = this.braketParser.parse(input);
       if (this.config.debug) {
@@ -321,29 +342,56 @@ Examples:
 `);
         break;
       case "vars":
-        if (this.session.variables.length === 0) {
-          console.log("No variables defined");
-        } else {
-          console.log("Variables:");
-          for (const v of this.session.variables) {
-            if (v.visible) {
-              console.log(`  ${v.name} = ${JSON.stringify(v.value)}`);
+        try {
+          let variables;
+          if (this.client) {
+            const state = await this.client.getState();
+            variables = state.variables || [];
+          } else {
+            variables = this.session.variables;
+          }
+          if (variables.length === 0) {
+            console.log("No variables defined");
+          } else {
+            console.log("Variables:");
+            for (const v of variables) {
+              if (v.visible) {
+                console.log(`  ${v.name} = ${JSON.stringify(v.value)}`);
+              }
             }
           }
+        } catch (error) {
+          console.error("Error getting variables:", error instanceof Error ? error.message : String(error));
         }
         break;
       case "subs":
-        if (this.session.subroutines.length === 0) {
-          console.log("No subroutines defined");
-        } else {
-          console.log("Subroutines:");
-          for (const s of this.session.subroutines) {
-            const params = s.parameters?.map((p) => p.name).join(", ") || "";
-            console.log(`  ${s.name}(${params})`);
-            if (s.description) {
-              console.log(`    ${s.description}`);
+        try {
+          let subroutines;
+          if (this.client) {
+            const state = await this.client.getState();
+            console.log("State received:", JSON.stringify(state, null, 2).substring(0, 500));
+            subroutines = state.subroutines || [];
+          } else {
+            subroutines = this.session.subroutines;
+          }
+          if (!Array.isArray(subroutines)) {
+            console.error("Error: subroutines is not an array, it is:", typeof subroutines);
+            break;
+          }
+          if (subroutines.length === 0) {
+            console.log("No subroutines defined");
+          } else {
+            console.log("Subroutines:");
+            for (const s of subroutines) {
+              const params = s.parameters?.map((p) => p.name).join(", ") || "";
+              console.log(`  ${s.name}(${params})`);
+              if (s.description) {
+                console.log(`    ${s.description}`);
+              }
             }
           }
+        } catch (error) {
+          console.error("Error getting subroutines:", error instanceof Error ? error.message : String(error));
         }
         break;
       case "clear":
@@ -709,9 +757,14 @@ Examples:
       }
       console.log(`Loading init script: ${scriptPath}`);
       const scriptContent = fs.readFileSync(resolvedPath, "utf-8");
-      const xml = this.braketParser.parse(scriptContent);
-      const ast = this.xmlParser.parse(xml);
-      await integrate(this.session, ast);
+      if (this.client) {
+        const xml = this.braketParser.parse(scriptContent);
+        await this.client.execute(xml);
+      } else {
+        const xml = this.braketParser.parse(scriptContent);
+        const ast = this.xmlParser.parse(xml);
+        await integrate(this.session, ast);
+      }
       console.log(`Init script loaded.
 `);
     } catch (err) {

package/package.json

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