@cortex-context/cli 0.0.13 → 0.0.15

package/dist/index.js

@@ -18309,12 +18309,26 @@ async function deployLocalStack(workspacePath, opts = {}) {
   if (!(0, import_fs5.existsSync)(templatePath)) {
     return { started: false, error: `Template not found: ${templatePath}` };
   }
-  if (!(0, import_fs5.existsSync)(destPath)) {
-    (0, import_fs5.cpSync)(templatePath, destPath);
-  }
+  (0, import_fs5.cpSync)(templatePath, destPath);
+  const specsSample = (0, import_path5.join)(workspacePath, "specs-sample");
+  const reposSample = (0, import_path5.join)(workspacePath, "repos-sample");
+  (0, import_fs5.mkdirSync)(specsSample, { recursive: true });
+  (0, import_fs5.mkdirSync)(reposSample, { recursive: true });
   const envPath = (0, import_path5.join)(workspacePath, ".env");
-  const envContent = `CORTEX_API_TOKEN=${opts.token ?? ""}
-`;
+  let existingEnv = "";
+  if ((0, import_fs5.existsSync)(envPath)) {
+    existingEnv = (0, import_fs5.readFileSync)(envPath, "utf-8");
+  }
+  const setEnvVar = (content, key, value) => {
+    const re = new RegExp(`^${key}=.*$`, "m");
+    const line = `${key}=${value}`;
+    return re.test(content) ? content.replace(re, line) : content + (content.endsWith("\n") || !content ? "" : "\n") + line + "\n";
+  };
+  let envContent = existingEnv;
+  envContent = setEnvVar(envContent, "CORTEX_API_TOKEN", opts.token ?? "");
+  envContent = setEnvVar(envContent, "SPECS_DIR_HOST", opts.specsDir ?? "./specs-sample");
+  envContent = setEnvVar(envContent, "REPOS_DIR_HOST", opts.reposDir ?? "./repos-sample");
+  if (!envContent.endsWith("\n")) envContent += "\n";
   (0, import_fs5.writeFileSync)(envPath, envContent, { encoding: "utf-8" });
   const imageTag = opts.embeddings ? "latest-embeddings" : "latest";
   const cortexImage = `ghcr.io/rodrigoroldan/cortex-context:${imageTag}`;
@@ -18342,6 +18356,66 @@ ${stderr.trim()}` : ""}`
     };
   }
 }
+async function upgradeLocalStack(workspacePath, opts = {}) {
+  const composePath = (0, import_path5.join)(workspacePath, "docker-compose.local.yml");
+  if (!(0, import_fs5.existsSync)(composePath)) {
+    return {
+      upgraded: false,
+      error: `docker-compose.local.yml not found in ${workspacePath}. Run: cortex-context server`
+    };
+  }
+  const imageTag = opts.embeddings ? "latest-embeddings" : "latest";
+  const cortexImage = `ghcr.io/rodrigoroldan/cortex-context:${imageTag}`;
+  const composeCmd = `docker compose --project-name cortex-local -f "${composePath}"`;
+  try {
+    (0, import_child_process.execSync)(`${composeCmd} pull cortex-api`, {
+      cwd: workspacePath,
+      stdio: ["inherit", "inherit", "pipe"],
+      env: { ...process.env, CORTEX_IMAGE: cortexImage }
+    });
+  } catch (err) {
+    const stderr = err instanceof Error && "stderr" in err ? err.stderr?.toString() ?? "" : "";
+    const msg = err instanceof Error ? err.message : String(err);
+    return {
+      upgraded: false,
+      error: `docker compose pull failed: ${msg}${stderr ? `
+${stderr.trim()}` : ""}`
+    };
+  }
+  try {
+    (0, import_child_process.execSync)(`${composeCmd} up -d --no-deps --force-recreate cortex-api`, {
+      cwd: workspacePath,
+      stdio: ["inherit", "inherit", "pipe"],
+      env: { ...process.env, CORTEX_IMAGE: cortexImage }
+    });
+  } catch (err) {
+    const stderr = err instanceof Error && "stderr" in err ? err.stderr?.toString() ?? "" : "";
+    const msg = err instanceof Error ? err.message : String(err);
+    return {
+      upgraded: false,
+      error: `docker compose up failed: ${msg}${stderr ? `
+${stderr.trim()}` : ""}`
+    };
+  }
+  return { upgraded: true };
+}
+function restartCortexApi(workspacePath) {
+  const composePath = (0, import_path5.join)(workspacePath, "docker-compose.local.yml");
+  if (!(0, import_fs5.existsSync)(composePath))
+    return { restarted: false, error: "docker-compose.local.yml not found" };
+  try {
+    (0, import_child_process.execSync)(
+      `docker compose --project-name cortex-local -f "${composePath}" restart cortex-api`,
+      { cwd: workspacePath, stdio: "ignore" }
+    );
+    return { restarted: true };
+  } catch (err) {
+    return {
+      restarted: false,
+      error: err instanceof Error ? err.message : String(err)
+    };
+  }
+}
 function isDockerAvailable() {
   try {
     (0, import_child_process.execSync)("docker --version", { stdio: "ignore" });
@@ -18619,7 +18693,7 @@ async function initCommand(options) {
     const spinner2 = ora(
       "  Starting Cortex stack with Docker Compose..."
     ).start();
-    const dockerResult = await deployLocalStack(workspacePath);
+    const dockerResult = await deployLocalStack(workspacePath, { token: cortexToken });
     if (!dockerResult.started) {
       spinner2.fail(
         source_default.red(`  \u2717 Docker deploy failed: ${dockerResult.error}`)
@@ -28720,6 +28794,39 @@ var import_os5 = require("os");
 var import_prompts6 = __toESM(require_prompts3());
 var CORTEX_DIR = (0, import_path15.join)((0, import_os5.homedir)(), ".cortex-context");
 var ENV_FILE = (0, import_path15.join)(CORTEX_DIR, ".env");
+async function applyTokenRestart() {
+  if (!isDockerAvailable()) return;
+  console.log(
+    source_default.dim("  Restarting cortex-api container to apply new token...")
+  );
+  const result = restartCortexApi(CORTEX_DIR);
+  if (!result.restarted) {
+    console.log(
+      source_default.yellow(
+        `  \u26A0  Could not restart container: ${result.error ?? "unknown error"}`
+      )
+    );
+    console.log(
+      source_default.dim(
+        "    docker compose -f ~/.cortex-context/docker-compose.local.yml restart cortex-api"
+      )
+    );
+    return;
+  }
+  const config2 = readConfig();
+  const url = config2.url ?? "http://localhost:8082";
+  console.log(source_default.dim(`  Waiting for Cortex API at ${url}...`));
+  const { healthy } = await waitForCortexHealth(url, 6e4);
+  if (healthy) {
+    console.log(source_default.green("  \u2713 Container restarted and healthy"));
+  } else {
+    console.log(
+      source_default.yellow(
+        "  \u26A0  Container restarted but did not become healthy within 60s"
+      )
+    );
+  }
+}
 function readEnvToken() {
   if (!(0, import_fs14.existsSync)(ENV_FILE)) return "";
   const content = (0, import_fs14.readFileSync)(ENV_FILE, "utf-8");
@@ -28756,15 +28863,8 @@ async function tokenCommand(options) {
     console.log(
       source_default.green(newToken ? "  \u2713 Token updated" : "  \u2713 Token cleared")
     );
-    if (newToken) {
-      console.log("");
-      console.log(source_default.dim("  Restart the Cortex container to apply:"));
-      console.log(
-        source_default.cyan(
-          "    docker compose -f ~/.cortex-context/docker-compose.local.yml restart cortex-api"
-        )
-      );
-    }
+    console.log("");
+    await applyTokenRestart();
     console.log("");
     return;
   }
@@ -28805,12 +28905,8 @@ async function tokenCommand(options) {
         `    cortex-context init --url ${config3.url ?? "http://localhost:8082"} --token ${newToken}`
       )
     );
-    console.log(source_default.dim("  Restart the Cortex container:"));
-    console.log(
-      source_default.cyan(
-        "    docker compose -f ~/.cortex-context/docker-compose.local.yml restart cortex-api"
-      )
-    );
+    console.log("");
+    await applyTokenRestart();
     console.log("");
     return;
   }
@@ -28834,11 +28930,114 @@ async function tokenCommand(options) {
   console.log("");
 }
 
-// src/cli.ts
-var import_fs15 = require("fs");
+// src/commands/upgrade.ts
 var import_path16 = require("path");
+var import_os6 = require("os");
+var import_fs15 = require("fs");
+async function upgradeCommand(options) {
+  console.log("");
+  console.log(source_default.bold.cyan("  \u25C6 Cortex Context \u2014 Upgrade"));
+  console.log(
+    source_default.dim(
+      "  Pulls the latest backend image and recreates the cortex-api container."
+    )
+  );
+  console.log("");
+  if (!isDockerAvailable()) {
+    console.error(source_default.red("  \u2717 Docker is not available on $PATH."));
+    console.log(
+      source_default.dim("    Install Docker or ensure it is running, then retry.")
+    );
+    process.exit(1);
+  }
+  const workDir = options.dir ?? (0, import_path16.join)((0, import_os6.homedir)(), ".cortex-context");
+  const composePath = (0, import_path16.join)(workDir, "docker-compose.local.yml");
+  if (!(0, import_fs15.existsSync)(composePath)) {
+    console.error(
+      source_default.red(`  \u2717 docker-compose.local.yml not found in ${workDir}`)
+    );
+    console.log(
+      source_default.dim("    The local Cortex stack has not been set up yet. Run:")
+    );
+    console.log(source_default.cyan("      cortex-context server"));
+    console.log("");
+    process.exit(1);
+  }
+  const imageTag = options.embeddings ? "latest-embeddings" : "latest";
+  const cortexImage = `ghcr.io/rodrigoroldan/cortex-context:${imageTag}`;
+  console.log(source_default.dim(`  Target image: ${cortexImage}`));
+  console.log(source_default.dim(`  Compose dir:  ${workDir}`));
+  console.log("");
+  console.log(source_default.bold("  Step 1 \u2014 Pulling latest image..."));
+  if (options.pullOnly) {
+    const { execSync: execSync3 } = await import("child_process");
+    try {
+      execSync3(
+        `docker compose --project-name cortex-local -f "${composePath}" pull cortex-api`,
+        {
+          cwd: workDir,
+          stdio: ["inherit", "inherit", "pipe"],
+          env: { ...process.env, CORTEX_IMAGE: cortexImage }
+        }
+      );
+      console.log("");
+      console.log(
+        source_default.green("  \u2713 Image pulled successfully (--pull-only mode)")
+      );
+      console.log(
+        source_default.dim(
+          "    Run without --pull-only to recreate the container with the new image."
+        )
+      );
+      console.log("");
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      console.error(source_default.red(`  \u2717 Pull failed: ${msg}`));
+      process.exit(1);
+    }
+    return;
+  }
+  const result = await upgradeLocalStack(workDir, {
+    embeddings: options.embeddings
+  });
+  if (!result.upgraded) {
+    console.error(source_default.red(`  \u2717 Upgrade failed: ${result.error}`));
+    process.exit(1);
+  }
+  console.log("");
+  console.log(source_default.bold("  Step 2 \u2014 Waiting for Cortex API to be healthy..."));
+  const config2 = readConfig();
+  const cortexUrl = config2.url ?? "http://localhost:8082";
+  const { healthy } = await waitForCortexHealth(cortexUrl, 12e4);
+  console.log("");
+  if (healthy) {
+    console.log(source_default.green("  \u2713 Cortex API is healthy"));
+    console.log("");
+    console.log(source_default.bold("  \u25C6 Upgrade complete!"));
+    console.log(source_default.dim(`    API running at ${cortexUrl}`));
+    console.log("");
+  } else {
+    console.log(
+      source_default.yellow(
+        "  \u26A0  Container is up but did not become healthy within 120s"
+      )
+    );
+    console.log(source_default.dim("    Check container logs:"));
+    console.log(
+      source_default.cyan(
+        `    docker compose -f ${workDir}/docker-compose.local.yml logs cortex-api`
+      )
+    );
+    console.log("");
+    process.exit(1);
+  }
+}
+
+// src/cli.ts
+var import_fs16 = require("fs");
+var import_path17 = require("path");
 var pkg = JSON.parse(
-  (0, import_fs15.readFileSync)((0, import_path16.join)(__dirname, "..", "package.json"), "utf-8")
+  (0, import_fs16.readFileSync)((0, import_path17.join)(__dirname, "..", "package.json"), "utf-8")
 );
 function createCli() {
   const program3 = new Command();
@@ -28892,6 +29091,15 @@ function createCli() {
     "--set <value>",
     "Set a specific token value (empty string clears auth)"
   ).action(tokenCommand);
+  program3.command("upgrade").description(
+    "Pull the latest Cortex backend image and restart the local stack"
+  ).option(
+    "--dir <path>",
+    "Directory with docker-compose.local.yml (defaults to ~/.cortex-context)"
+  ).option("--embeddings", "Use the embeddings-enabled image variant").option(
+    "--pull-only",
+    "Pull the latest image without recreating the container"
+  ).action(upgradeCommand);
   return program3;
 }