@locusai/telegram 0.13.3 → 0.14.0

package/bin/telegram.js

@@ -22719,7 +22719,7 @@ var init_constants = __esm(() => {
 });
 
 // ../shared/src/enums.ts
-var UserRole, MembershipRole, TaskStatus, TaskPriority, AssigneeRole, SprintStatus, EventType;
+var UserRole, MembershipRole, TaskStatus, TaskPriority, AssigneeRole, SprintStatus, InstanceStatus, AwsRegion, EventType;
 var init_enums = __esm(() => {
   ((UserRole2) => {
     UserRole2["USER"] = "USER";
@@ -22756,6 +22756,16 @@ var init_enums = __esm(() => {
     SprintStatus2["ACTIVE"] = "ACTIVE";
     SprintStatus2["COMPLETED"] = "COMPLETED";
   })(SprintStatus ||= {});
+  ((InstanceStatus2) => {
+    InstanceStatus2["PROVISIONING"] = "PROVISIONING";
+    InstanceStatus2["RUNNING"] = "RUNNING";
+    InstanceStatus2["STOPPED"] = "STOPPED";
+    InstanceStatus2["TERMINATED"] = "TERMINATED";
+    InstanceStatus2["ERROR"] = "ERROR";
+  })(InstanceStatus ||= {});
+  ((AwsRegion2) => {
+    AwsRegion2["US_EAST_1"] = "us-east-1";
+  })(AwsRegion ||= {});
   ((EventType2) => {
     EventType2["TASK_CREATED"] = "TASK_CREATED";
     EventType2["TASK_UPDATED"] = "TASK_UPDATED";
@@ -23103,6 +23113,76 @@ var init_auth = __esm(() => {
   });
 });
 
+// ../shared/src/models/aws-instance.ts
+var InstanceAction, AwsCredentialsSchema, IntegrationSchema, AwsInstanceSchema, CreateAwsInstanceSchema, UpdateAwsInstanceSchema, SaveAwsCredentialsSchema, ProvisionAwsInstanceSchema, InstanceActionBodySchema, InstanceIdParamSchema, CIDR_REGEX, UpdateSecurityRulesSchema;
+var init_aws_instance = __esm(() => {
+  init_zod();
+  init_common();
+  init_enums();
+  ((InstanceAction2) => {
+    InstanceAction2["START"] = "START";
+    InstanceAction2["STOP"] = "STOP";
+    InstanceAction2["TERMINATE"] = "TERMINATE";
+  })(InstanceAction ||= {});
+  AwsCredentialsSchema = exports_external.object({
+    accessKeyId: exports_external.string().min(1),
+    secretAccessKey: exports_external.string().min(1),
+    region: exports_external.enum(AwsRegion).default("us-east-1" /* US_EAST_1 */)
+  });
+  IntegrationSchema = exports_external.object({
+    name: exports_external.string(),
+    config: exports_external.record(exports_external.string(), exports_external.string())
+  });
+  AwsInstanceSchema = BaseEntitySchema.extend({
+    workspaceId: exports_external.uuid(),
+    instanceId: exports_external.string(),
+    status: exports_external.enum(InstanceStatus),
+    instanceType: exports_external.enum(["t3.micro", "t3.small", "t3.medium"]),
+    region: exports_external.enum(AwsRegion).default("us-east-1" /* US_EAST_1 */),
+    publicIp: exports_external.string().nullable().optional(),
+    launchTime: exports_external.union([exports_external.date(), exports_external.number()]).nullable().optional(),
+    repoUrl: exports_external.string().nullable().optional(),
+    integrations: exports_external.array(IntegrationSchema).default([])
+  });
+  CreateAwsInstanceSchema = exports_external.object({
+    workspaceId: exports_external.uuid(),
+    instanceType: exports_external.enum(["t3.micro", "t3.small", "t3.medium"]).default("t3.micro"),
+    region: exports_external.enum(AwsRegion).default("us-east-1" /* US_EAST_1 */),
+    repoUrl: exports_external.string().optional(),
+    integrations: exports_external.array(IntegrationSchema).optional().default([])
+  });
+  UpdateAwsInstanceSchema = exports_external.object({
+    status: exports_external.enum(InstanceStatus).optional(),
+    instanceType: exports_external.enum(["t3.micro", "t3.small", "t3.medium"]).optional(),
+    publicIp: exports_external.string().nullable().optional(),
+    launchTime: exports_external.union([exports_external.date(), exports_external.number()]).nullable().optional(),
+    repoUrl: exports_external.string().nullable().optional(),
+    integrations: exports_external.array(IntegrationSchema).optional()
+  });
+  SaveAwsCredentialsSchema = exports_external.object({
+    accessKeyId: exports_external.string().min(16),
+    secretAccessKey: exports_external.string().min(1),
+    region: exports_external.string().default("us-east-1")
+  });
+  ProvisionAwsInstanceSchema = exports_external.object({
+    repoUrl: exports_external.string().min(1),
+    githubToken: exports_external.string().min(1),
+    instanceType: exports_external.enum(["t3.micro", "t3.small", "t3.medium"]).default("t3.small"),
+    integrations: exports_external.array(IntegrationSchema).optional().default([])
+  });
+  InstanceActionBodySchema = exports_external.object({
+    action: exports_external.nativeEnum(InstanceAction)
+  });
+  InstanceIdParamSchema = exports_external.object({
+    workspaceId: exports_external.string().uuid("Invalid Workspace ID"),
+    instanceId: exports_external.string().uuid("Invalid Instance ID")
+  });
+  CIDR_REGEX = /^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\/\d{1,2}$/;
+  UpdateSecurityRulesSchema = exports_external.object({
+    allowedIps: exports_external.array(exports_external.string().regex(CIDR_REGEX, "Invalid CIDR format (e.g. 1.2.3.4/32)"))
+  });
+});
+
 // ../shared/src/models/ci.ts
 var RecordCiSchema;
 var init_ci = __esm(() => {
@@ -23490,6 +23570,7 @@ var init_models = __esm(() => {
   init_activity();
   init_agent();
   init_auth();
+  init_aws_instance();
   init_ci();
   init_doc();
   init_doc_group();
@@ -38388,6 +38469,49 @@ var init_docs = __esm(() => {
   };
 });
 
+// ../sdk/src/modules/instances.ts
+var InstancesModule;
+var init_instances = __esm(() => {
+  InstancesModule = class InstancesModule extends BaseModule {
+    async list(workspaceId) {
+      const { data } = await this.api.get(`/workspaces/${workspaceId}/aws-instances`);
+      return data.instances;
+    }
+    async get(workspaceId, instanceId) {
+      const { data } = await this.api.get(`/workspaces/${workspaceId}/aws-instances/${instanceId}`);
+      return data.instance;
+    }
+    async provision(workspaceId, body) {
+      const { data } = await this.api.post(`/workspaces/${workspaceId}/aws-instances`, body);
+      return data.instance;
+    }
+    async performAction(workspaceId, instanceId, action) {
+      const { data } = await this.api.post(`/workspaces/${workspaceId}/aws-instances/${instanceId}/actions`, { action });
+      return data.instance;
+    }
+    async sync(workspaceId, instanceId) {
+      const { data } = await this.api.post(`/workspaces/${workspaceId}/aws-instances/${instanceId}/sync`);
+      return data.instance;
+    }
+    async checkUpdates(workspaceId, instanceId) {
+      const { data } = await this.api.get(`/workspaces/${workspaceId}/aws-instances/${instanceId}/updates`);
+      return data.update;
+    }
+    async applyUpdate(workspaceId, instanceId) {
+      const { data } = await this.api.post(`/workspaces/${workspaceId}/aws-instances/${instanceId}/updates`);
+      return data.update;
+    }
+    async getSecurity(workspaceId, instanceId) {
+      const { data } = await this.api.get(`/workspaces/${workspaceId}/aws-instances/${instanceId}/security`);
+      return data.rules;
+    }
+    async updateSecurity(workspaceId, instanceId, body) {
+      const { data } = await this.api.put(`/workspaces/${workspaceId}/aws-instances/${instanceId}/security`, body);
+      return data.rules;
+    }
+  };
+});
+
 // ../sdk/src/modules/invitations.ts
 var InvitationsModule;
 var init_invitations = __esm(() => {
@@ -38620,6 +38744,17 @@ var init_workspaces = __esm(() => {
     async deleteApiKey(workspaceId, keyId) {
       await this.api.delete(`/workspaces/${workspaceId}/api-keys/${keyId}`);
     }
+    async getAwsCredentials(workspaceId) {
+      const { data } = await this.api.get(`/workspaces/${workspaceId}/aws-credentials`);
+      return data.credential;
+    }
+    async saveAwsCredentials(workspaceId, body) {
+      const { data } = await this.api.put(`/workspaces/${workspaceId}/aws-credentials`, body);
+      return data.credential;
+    }
+    async deleteAwsCredentials(workspaceId) {
+      await this.api.delete(`/workspaces/${workspaceId}/aws-credentials`);
+    }
   };
 });
 
@@ -38668,6 +38803,7 @@ class LocusClient {
   invitations;
   docs;
   ci;
+  instances;
   constructor(config2) {
     this.emitter = new LocusEmitter;
     this.api = axios_default.create({
@@ -38687,6 +38823,7 @@ class LocusClient {
     this.invitations = new InvitationsModule(this.api, this.emitter);
     this.docs = new DocsModule(this.api, this.emitter);
     this.ci = new CiModule(this.api, this.emitter);
+    this.instances = new InstancesModule(this.api, this.emitter);
     if (config2.retryOptions) {
       this.setupRetryInterceptor(config2.retryOptions);
     }
@@ -38752,6 +38889,7 @@ var init_src2 = __esm(() => {
   init_auth2();
   init_ci2();
   init_docs();
+  init_instances();
   init_invitations();
   init_organizations();
   init_sprints();
@@ -38762,6 +38900,7 @@ var init_src2 = __esm(() => {
   init_auth2();
   init_ci2();
   init_docs();
+  init_instances();
   init_invitations();
   init_organizations();
   init_sprints();
@@ -44423,6 +44562,7 @@ var PLAN_TIMEOUT = 3600000;
 var STREAMING_DEFAULT_TIMEOUT = 3600000;
 var GIT_TIMEOUT = 60000;
 var DEV_TIMEOUT = 600000;
+var UPGRADE_TIMEOUT = 300000;
 
 // src/commands/dev.ts
 var USAGE2 = `<b>Usage:</b> /dev &lt;command&gt;
@@ -44601,6 +44741,7 @@ var HELP_TEXT = `<b>Locus Bot — Command Center</b>
 /config set &lt;key&gt; &lt;value&gt; — Update a setting
 /config unset &lt;key&gt; — Remove a setting
 /model [name] — View or switch AI model
+/upgrade — Upgrade Locus CLI &amp; restart bot
 /help — Show this message`;
 async function startCommand(ctx) {
   console.log("[start] User started bot");
@@ -45266,6 +45407,121 @@ async function approveTaskCommand(ctx, config2) {
     await ctx.reply(formatError2(`Failed to approve task: ${err instanceof Error ? err.message : String(err)}`), { parse_mode: "HTML" });
   }
 }
+// src/commands/upgrade.ts
+import { spawn as spawn5 } from "node:child_process";
+import { homedir as homedir3 } from "node:os";
+import { join as join10 } from "node:path";
+function runCommand2(cmd, args, timeout) {
+  return new Promise((resolve2) => {
+    const proc = spawn5(cmd, args, {
+      stdio: ["pipe", "pipe", "pipe"]
+    });
+    let stdout = "";
+    let stderr = "";
+    let killed = false;
+    proc.stdout?.on("data", (data) => {
+      stdout += data.toString();
+    });
+    proc.stderr?.on("data", (data) => {
+      stderr += data.toString();
+    });
+    const timer = setTimeout(() => {
+      killed = true;
+      proc.kill("SIGTERM");
+    }, timeout);
+    proc.on("close", (exitCode) => {
+      clearTimeout(timer);
+      resolve2({ stdout, stderr, exitCode, killed });
+    });
+    proc.on("error", (err) => {
+      clearTimeout(timer);
+      resolve2({
+        stdout,
+        stderr: stderr || err.message,
+        exitCode: 1,
+        killed: false
+      });
+    });
+  });
+}
+function getRestartInfo() {
+  const platform = process.platform;
+  if (platform === "linux") {
+    return {
+      label: "systemd (Linux)",
+      commands: [
+        { cmd: "sudo", args: ["systemctl", "restart", "locus-telegram"] }
+      ]
+    };
+  }
+  if (platform === "darwin") {
+    const plistPath = join10(homedir3(), "Library/LaunchAgents/com.locus.telegram.plist");
+    return {
+      label: "launchctl (macOS)",
+      commands: [
+        { cmd: "launchctl", args: ["unload", plistPath] },
+        { cmd: "launchctl", args: ["load", plistPath] }
+      ]
+    };
+  }
+  return null;
+}
+async function upgradeCommand(ctx) {
+  console.log("[upgrade] Starting upgrade");
+  const restartInfo = getRestartInfo();
+  if (!restartInfo) {
+    await ctx.reply(formatError2(`Unsupported platform: ${process.platform}. Upgrade is only supported on Linux (systemd) and macOS (launchctl).`), { parse_mode: "HTML" });
+    return;
+  }
+  await ctx.reply(formatInfo("Running <code>sudo locus upgrade</code>..."), {
+    parse_mode: "HTML"
+  });
+  const upgradeResult = await runCommand2("sudo", ["locus", "upgrade"], UPGRADE_TIMEOUT);
+  const output = stripAnsi((upgradeResult.stdout + upgradeResult.stderr).trim());
+  if (upgradeResult.killed) {
+    await ctx.reply(formatError2("Upgrade command timed out and was killed."), {
+      parse_mode: "HTML"
+    });
+    return;
+  }
+  if (upgradeResult.exitCode !== 0) {
+    const truncated = truncateOutput(output, 3000);
+    await ctx.reply(`${formatError2("Upgrade failed:")}
+
+<pre>${escapeHtml(truncated)}</pre>`, { parse_mode: "HTML" });
+    return;
+  }
+  if (output) {
+    const truncated = truncateOutput(output, 3000);
+    await ctx.reply(`${formatSuccess("Upgrade completed:")}
+
+<pre>${escapeHtml(truncated)}</pre>`, { parse_mode: "HTML" });
+  } else {
+    await ctx.reply(formatSuccess("Upgrade completed."), {
+      parse_mode: "HTML"
+    });
+  }
+  await ctx.reply(formatInfo(`Restarting locus-telegram via ${restartInfo.label}... The bot will be back shortly.`), { parse_mode: "HTML" });
+  for (const { cmd, args } of restartInfo.commands) {
+    console.log(`[upgrade] Running: ${cmd} ${args.join(" ")}`);
+    const result = await runCommand2(cmd, args, 30000);
+    if (result.exitCode !== 0) {
+      const errOutput = stripAnsi((result.stdout + result.stderr).trim());
+      console.error(`[upgrade] Restart command failed: ${cmd} ${args.join(" ")} — ${errOutput}`);
+      try {
+        await ctx.reply(formatError2(`Restart failed: <code>${escapeHtml(`${cmd} ${args.join(" ")}`)}</code>
+
+<pre>${escapeHtml(truncateOutput(errOutput, 2000))}</pre>`), { parse_mode: "HTML" });
+      } catch {}
+      return;
+    }
+  }
+  try {
+    await ctx.reply(formatSuccess("Bot restarted successfully."), {
+      parse_mode: "HTML"
+    });
+  } catch {}
+}
 // src/commands/workspace.ts
 async function workspaceCommand(ctx, config2) {
   console.log("[workspace] Fetching workspace info");
@@ -45309,8 +45565,8 @@ async function workspaceCommand(ctx, config2) {
   }
 }
 // src/executor.ts
-import { spawn as spawn5 } from "node:child_process";
-import { join as join10 } from "node:path";
+import { spawn as spawn6 } from "node:child_process";
+import { join as join11 } from "node:path";
 function timestamp2() {
   return new Date().toLocaleTimeString("en-GB", { hour12: false });
 }
@@ -45326,7 +45582,7 @@ class CliExecutor {
   }
   resolveCommand(args) {
     if (this.config.testMode) {
-      const cliPath = join10(this.config.projectPath, "packages/cli/src/cli.ts");
+      const cliPath = join11(this.config.projectPath, "packages/cli/src/cli.ts");
       return { cmd: "bun", cmdArgs: ["run", cliPath, ...args] };
     }
     return { cmd: "locus", cmdArgs: args };
@@ -45339,7 +45595,7 @@ class CliExecutor {
     const startTime = Date.now();
     log2(id, `Process started: ${fullCommand}`);
     return new Promise((resolve2) => {
-      const proc = spawn5(cmd, cmdArgs, {
+      const proc = spawn6(cmd, cmdArgs, {
         cwd: this.config.projectPath,
         env: buildSpawnEnv(),
         stdio: ["pipe", "pipe", "pipe"]
@@ -45389,7 +45645,7 @@ class CliExecutor {
     const fullCommand = `${cmd} ${cmdArgs.join(" ")}`;
     const startTime = Date.now();
     log2(id, `Process started (streaming): ${fullCommand}`);
-    const proc = spawn5(cmd, cmdArgs, {
+    const proc = spawn6(cmd, cmdArgs, {
       cwd: this.config.projectPath,
       env: buildSpawnEnv(),
       stdio: ["pipe", "pipe", "pipe"]
@@ -45535,6 +45791,7 @@ function createBot(config2) {
   bot.command("activity", (ctx) => activityCommand(ctx, config2));
   bot.command("workspace", (ctx) => workspaceCommand(ctx, config2));
   bot.command("status", (ctx) => statusCommand(ctx, executor));
+  bot.command("upgrade", (ctx) => upgradeCommand(ctx));
   bot.on("text", (ctx) => {
     const text = ctx.message.text;
     const chatId = ctx.chat.id;
@@ -45581,6 +45838,7 @@ function createBot(config2) {
     { command: "config", description: "Show/update settings" },
     { command: "model", description: "View or switch AI model" },
     { command: "status", description: "Show running processes" },
+    { command: "upgrade", description: "Upgrade Locus CLI & restart bot" },
     { command: "help", description: "Show all commands" }
   ]).catch((err) => console.error("Failed to set bot commands:", err));
   return bot;
@@ -45589,12 +45847,12 @@ function createBot(config2) {
 // src/config.ts
 var import_dotenv = __toESM(require_main(), 1);
 import { existsSync as existsSync9, readFileSync as readFileSync8 } from "node:fs";
-import { join as join11 } from "node:path";
+import { join as join12 } from "node:path";
 import_dotenv.default.config();
 var SETTINGS_FILE2 = "settings.json";
 var CONFIG_DIR2 = ".locus";
 function loadSettings2(projectPath) {
-  const settingsPath = join11(projectPath, CONFIG_DIR2, SETTINGS_FILE2);
+  const settingsPath = join12(projectPath, CONFIG_DIR2, SETTINGS_FILE2);
   if (!existsSync9(settingsPath)) {
     return null;
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@locusai/telegram",
-  "version": "0.13.3",
+  "version": "0.14.0",
   "description": "Telegram bot for Locus - remote control your AI agents from Telegram",
   "type": "module",
   "bin": {
@@ -31,8 +31,8 @@
   "author": "",
   "license": "MIT",
   "dependencies": {
-    "@locusai/sdk": "^0.13.3",
-    "@locusai/shared": "^0.13.3",
+    "@locusai/sdk": "^0.14.0",
+    "@locusai/shared": "^0.14.0",
     "dotenv": "^16.4.7",
     "telegraf": "^4.16.3"
   },