mortgram-cli 1.0.1 → 1.0.2

package/dist/index.js

@@ -9,6 +9,8 @@ const spawner_1 = require("./lib/spawner");
 const config_1 = require("./lib/config");
 const dotenv_1 = __importDefault(require("dotenv"));
 const path_1 = __importDefault(require("path"));
+const axios_1 = __importDefault(require("axios"));
+const chalk_1 = __importDefault(require("chalk"));
 // Load environment variables from the parent directory if possible, or current
 dotenv_1.default.config({ path: path_1.default.join(process.cwd(), "..", ".env.local"), quiet: true });
 dotenv_1.default.config({ quiet: true }); // Fallback to current dir
@@ -16,20 +18,53 @@ const program = new commander_1.Command();
 program
     .name("mortgram")
     .description("MORTGRAM Observer CLI - Zero-code observability wrapper")
-    .version("1.0.0");
+    .version("1.0.1");
 program
     .command("login <key>")
     .description("Authenticate your terminal with your MORTGRAM API Key")
-    .action((key) => {
-    (0, config_1.saveConfig)(key);
-    console.log("MORTGRAM: Authentication successful. Key saved to ~/.mortgramrc");
+    .action(async (key) => {
+    try {
+        const apiUrl = process.env.MORTGRAM_API_URL || "http://localhost:3000/api";
+        console.log(chalk_1.default.yellow("Authenticating with MORTGRAM HQ..."));
+        const res = await axios_1.default.get(`${apiUrl}/profile`, {
+            headers: { "Authorization": `Bearer ${key}` }
+        });
+        if (res.data.profile) {
+            const ablyKey = res.data.profile.ably_key;
+            (0, config_1.saveConfig)(key, ablyKey);
+            console.log(chalk_1.default.green("✔ Authentication successful."));
+            if (ablyKey) {
+                console.log(chalk_1.default.green("✔ Neural Link established (Ably Key retrieved)."));
+            }
+            else {
+                console.log(chalk_1.default.yellow("⚠ Neural Link warning: No Ably Key returned from server."));
+            }
+            console.log(chalk_1.default.gray("Configuration saved to ~/.mortgramrc"));
+        }
+        else {
+            console.error(chalk_1.default.red("Authentication failed: Invalid response from server."));
+        }
+    }
+    catch (error) {
+        console.error(chalk_1.default.red(`Authentication failed: ${error.message}`));
+        if (error.response?.status === 401) {
+            console.error(chalk_1.default.red("Error: Invalid API Key."));
+        }
+    }
 });
 program
     .command("run <command...>")
     .description("Run an agent command with MORTGRAM observability")
     .action(async (commandParts) => {
     const fullCommand = commandParts.join(" ");
-    console.log(`MORTGRAM: Wrapping command: "${fullCommand}"`);
-    await (0, spawner_1.spawnAgent)(fullCommand);
+    console.log(chalk_1.default.blue(`MORTGRAM: Wrapping command: "${fullCommand}"`));
+    console.log(chalk_1.default.blue(`MORTGRAM: Listening for uplink...`));
+    try {
+        await (0, spawner_1.spawnAgent)(fullCommand);
+    }
+    catch (e) {
+        console.error(chalk_1.default.red(`MORTGRAM: Failed to launch agent: ${e.message}`));
+        process.exit(1);
+    }
 });
 program.parse();

package/dist/lib/config.js

@@ -9,8 +9,8 @@ const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const os_1 = __importDefault(require("os"));
 const CONFIG_FILE = path_1.default.join(os_1.default.homedir(), ".mortgramrc");
-function saveConfig(key) {
-    const config = { api_key: key };
+function saveConfig(apiKey, ablyKey) {
+    const config = { api_key: apiKey, ably_key: ablyKey };
     fs_1.default.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
 }
 function loadConfig() {
@@ -18,11 +18,14 @@ function loadConfig() {
         try {
             const content = fs_1.default.readFileSync(CONFIG_FILE, "utf-8");
             const config = JSON.parse(content);
-            return config.api_key || null;
+            return {
+                api_key: config.api_key || null,
+                ably_key: config.ably_key || null
+            };
         }
         catch (e) {
-            return null;
+            return { api_key: null, ably_key: null };
         }
     }
-    return null;
+    return { api_key: null, ably_key: null };
 }

package/dist/lib/logger.js

@@ -9,7 +9,8 @@ const dotenv_1 = __importDefault(require("dotenv"));
 const config_1 = require("./config");
 // ensure env is loaded
 dotenv_1.default.config({ quiet: true });
-const API_KEY = process.env.MG_LIVE_KEY || (0, config_1.loadConfig)();
+const config = (0, config_1.loadConfig)();
+const API_KEY = process.env.MG_LIVE_KEY || config.api_key;
 const API_URL = process.env.MORTGRAM_API_URL || "http://localhost:3000/api/ingest";
 const AGENT_ID = process.env.MG_AGENT_ID || "cli-agent";
 async function ingestLog(type, content, metadata = {}) {

package/dist/lib/neural.js

@@ -5,29 +5,39 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.startNeuralLink = startNeuralLink;
 const ably_1 = __importDefault(require("ably"));
+const logger_1 = require("./logger");
 const chalk_1 = __importDefault(require("chalk"));
 const dotenv_1 = __importDefault(require("dotenv"));
-dotenv_1.default.config();
-const ABLY_KEY = process.env.ABLY_API_KEY;
-const AGENT_ID = process.env.MG_AGENT_ID || "cli-agent";
-let client = null;
+const config_1 = require("./config");
+dotenv_1.default.config({ quiet: true });
 async function startNeuralLink(childProcess) {
-    if (!ABLY_KEY) {
-        console.warn(chalk_1.default.yellow("MORTGRAM: No ABLY_API_KEY found. Neural link disabled."));
+    const config = (0, config_1.loadConfig)();
+    const API_KEY = process.env.ABLY_API_KEY || config.ably_key;
+    const AGENT_ID = process.env.MG_AGENT_ID || "cli-agent"; // In real usage, this might be dynamic or passed as arg
+    if (!API_KEY) {
+        console.log(chalk_1.default.yellow("MORTGRAM: No ABLY_API_KEY found. Neural link disabled."));
         return { close: () => { } };
     }
-    client = new ably_1.default.Realtime(ABLY_KEY);
-    const channel = client.channels.get(`control:${AGENT_ID}`);
-    await channel.subscribe("KILL", (message) => {
-        console.log(chalk_1.default.bgRed.white.bold("\n!!! MORTGRAM NEURAL LINK: KILL SIGNAL RECEIVED !!!\n"));
-        childProcess.kill('SIGKILL');
+    const realtime = new ably_1.default.Realtime(API_KEY);
+    // Connect to specific agent control channel
+    const channel = realtime.channels.get(`control:${AGENT_ID}`);
+    await channel.subscribe("kill", async (message) => {
+        console.log(chalk_1.default.red.bold("\n⚡ MORTGRAM: KILL SIGNAL RECEIVED ⚡"));
+        console.log(chalk_1.default.red(`Reason: ${message.data.reason || "Manual Termination"}`));
+        // Log the event before dying
+        await (0, logger_1.ingestLog)("action", "TERMINATED via Neural Link", {
+            status: "killed",
+            reason: message.data.reason
+        });
+        // Kill the child process
+        childProcess.kill("SIGKILL");
+        // Close Ably connection
+        realtime.close();
         process.exit(0);
     });
-    console.log(chalk_1.default.blue(`MORTGRAM: Neural Link active on channel control:${AGENT_ID}`));
+    // Also subscribe to PAUSE signals if we implement them later
+    // await channel.subscribe("pause", ...);
     return {
-        close: () => {
-            if (client)
-                client.close();
-        }
+        close: () => realtime.close()
     };
 }

package/dist/lib/spawner.js

@@ -13,8 +13,9 @@ const chalk_1 = __importDefault(require("chalk"));
 let childProcess = null;
 async function spawnAgent(command) {
     console.log(chalk_1.default.green("MORTGRAM: Initializing Observer..."));
+    // Immediate handshake
+    await (0, logger_1.ingestLog)("log", "Observer Linked. Awaiting agent output.", { status: "online" });
     // Parse command string into cmd and args
-    // crude parsing, careful with quotes
     const parts = command.match(/(?:[^\s"]+|"[^"]*")+/g) || [];
     if (parts.length === 0)
         return;
@@ -44,8 +45,10 @@ async function spawnAgent(command) {
         process.stderr.write(line); // Pass through
         analyzeLog(line, true);
     });
-    childProcess.on('close', (code) => {
+    childProcess.on('close', async (code) => {
         console.log(chalk_1.default.gray(`MORTGRAM: Child process exited with code ${code}`));
+        // Final log
+        await (0, logger_1.ingestLog)("log", `Agent process terminated (code ${code})`, { status: "offline" });
         clearInterval(vitalsInterval);
         neural.close();
         process.exit(code);

package/dist/lib/vitals.js

@@ -10,7 +10,8 @@ const guard_1 = require("./guard");
 const dotenv_1 = __importDefault(require("dotenv"));
 const config_1 = require("./config");
 dotenv_1.default.config({ quiet: true });
-const API_KEY = process.env.MG_LIVE_KEY || (0, config_1.loadConfig)();
+const config = (0, config_1.loadConfig)();
+const API_KEY = process.env.MG_LIVE_KEY || config.api_key;
 const API_URL = process.env.MORTGRAM_API_URL ? `${process.env.MORTGRAM_API_URL}/heartbeat` : "http://localhost:3000/api/cron/heartbeat";
 // Note: reusing cron heartbeat or creating a specific agent heartbeat endpoint?
 // For CLI, we likely want a direct ingestion endpoint or use the log ingestion with type "heartbeat"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mortgram-cli",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "Zero-code observability wrapper for AI agents",
   "main": "dist/index.js",
   "scripts": {