cassian-cli 0.3.6 → 0.3.8

package/dist/commands/init.js

@@ -1,18 +1,17 @@
-import { createInterface } from "readline";
-import { writeFileSync, existsSync } from "fs";
-import { green, dim, bold } from "../lib/output.js";
-import { fatal } from "../lib/errors.js";
+import { existsSync } from "fs";
+import { writeFileSync } from "fs";
+import * as p from "@clack/prompts";
 import { AGENT_URL } from "../lib/constants.js";
 const ALL_GPUS = [
-    { label: "H100 SXM", value: "h100-sxm", memory: "80GB", hint: "best for large training" },
-    { label: "H100 PCIe", value: "h100-pcie", memory: "80GB" },
-    { label: "A100 SXM", value: "a100-sxm", memory: "80GB", hint: "reliable workhorse" },
-    { label: "A100 PCIe", value: "a100-pcie", memory: "80GB" },
-    { label: "L40S", value: "l40s", memory: "48GB", hint: "great for inference" },
-    { label: "L4", value: "l4", memory: "24GB", hint: "efficient inference" },
-    { label: "A10G", value: "a10g", memory: "24GB", hint: "good all-rounder" },
-    { label: "RTX 4090", value: "rtx4090", memory: "24GB", hint: "fast, affordable" },
-    { label: "RTX 3090", value: "rtx3090", memory: "24GB" },
+    { value: "h100-sxm", label: "H100 SXM", memory: "80GB", hint: "best for large training" },
+    { value: "h100-pcie", label: "H100 PCIe", memory: "80GB" },
+    { value: "a100-sxm", label: "A100 SXM", memory: "80GB", hint: "reliable workhorse" },
+    { value: "a100-pcie", label: "A100 PCIe", memory: "80GB" },
+    { value: "l40s", label: "L40S", memory: "48GB", hint: "great for inference" },
+    { value: "l4", label: "L4", memory: "24GB", hint: "efficient inference" },
+    { value: "a10g", label: "A10G", memory: "24GB", hint: "good all-rounder" },
+    { value: "rtx4090", label: "RTX 4090", memory: "24GB", hint: "fast, affordable" },
+    { value: "rtx3090", label: "RTX 3090", memory: "24GB" },
 ];
 async function fetchAvailability() {
     try {
@@ -29,40 +28,6 @@ async function fetchAvailability() {
         return {};
     }
 }
-function buildGpuOptions(availability) {
-    return ALL_GPUS.map(gpu => {
-        const count = availability[gpu.value] ?? 0;
-        const parts = [gpu.memory];
-        if (gpu.hint)
-            parts.push(gpu.hint);
-        if (count > 0) {
-            parts.push(`${count} available`);
-        }
-        else {
-            parts.push("unavailable");
-        }
-        return { label: gpu.label, value: gpu.value, hint: parts.join(" — ") };
-    });
-}
-function prompt(rl, question, fallback) {
-    return new Promise((resolve) => {
-        rl.question(question, (answer) => resolve(answer.trim() || fallback));
-    });
-}
-function promptChoice(rl, question, choices, fallback) {
-    return new Promise((resolve) => {
-        console.log(question);
-        choices.forEach((c, i) => {
-            const marker = c.value === fallback ? green("›") : " ";
-            const hint = c.hint ? dim(` — ${c.hint}`) : "";
-            console.log(`  ${marker} ${i + 1}) ${c.label}${hint}`);
-        });
-        rl.question(dim(`  [1-${choices.length}, default ${choices.findIndex(c => c.value === fallback) + 1}]: `), (answer) => {
-            const idx = parseInt(answer.trim()) - 1;
-            resolve(idx >= 0 && idx < choices.length ? choices[idx].value : fallback);
-        });
-    });
-}
 function buildYaml(opts) {
     const storageLine = opts.storage ? "\nstorage: true\n" : "";
     const excludeLines = opts.exclude.length
@@ -78,11 +43,11 @@ ${storageLine}${excludeLines}`;
 }
 export async function init(options = {}) {
     if (existsSync("cassian.yaml")) {
-        fatal("cassian.yaml already exists in this directory.");
+        p.log.error("cassian.yaml already exists in this directory.");
+        process.exit(1);
     }
     const dirName = process.cwd().split("/").pop().toLowerCase().replace(/[^a-z0-9-]/g, "-");
-    const skipPrompts = options.yes ?? false;
-    if (skipPrompts) {
+    if (options.yes) {
         const yaml = buildYaml({
             name: options.name || dirName,
             gpuType: options.gpu || "rtx3090",
@@ -92,56 +57,98 @@ export async function init(options = {}) {
             exclude: ["node_modules/", ".venv/", "__pycache__/", "*.pyc", "wandb/"],
         });
         writeFileSync("cassian.yaml", yaml);
-        console.log();
-        console.log(`  ${green("✓")} Created cassian.yaml`);
-        console.log();
-        console.log(`  Run ${green("cassian up")} to start your instance.`);
-        console.log();
+        p.log.success("Created cassian.yaml");
+        p.log.info("Run \x1b[32mcassian up\x1b[0m to start your instance.");
         return;
     }
-    // Interactive wizard
-    const rl = createInterface({ input: process.stdin, output: process.stdout });
-    console.log();
-    console.log(bold("  Let's set up your Cassian project."));
-    console.log(dim("  Press enter to accept defaults.\n"));
-    const name = options.name || await prompt(rl, `  Instance name ${dim(`[${dirName}]`)}: `, dirName);
     console.log();
+    const name = options.name || await p.text({
+        message: "Instance name",
+        placeholder: dirName,
+        defaultValue: dirName,
+        validate: (v) => {
+            if (!v || !/^[a-z0-9][a-z0-9-]*$/.test(v))
+                return "Lowercase letters, numbers, and hyphens only";
+        },
+    });
+    if (p.isCancel(name)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
     const availability = await fetchAvailability();
-    const gpuOptions = buildGpuOptions(availability);
-    const availableTypes = Object.keys(availability).filter(k => availability[k] > 0);
-    const defaultGpu = availableTypes[0] || "rtx3090";
-    const gpuType = options.gpu || await promptChoice(rl, "  GPU type:", gpuOptions, defaultGpu);
-    console.log();
-    const gpuCountStr = options.gpuCount || await promptChoice(rl, "  Number of GPUs:", [
-        { label: "1", value: "1" }, { label: "2", value: "2" },
-        { label: "4", value: "4" }, { label: "8", value: "8" },
-    ], "1");
+    const gpuType = options.gpu || await p.select({
+        message: "GPU type",
+        options: ALL_GPUS.map(gpu => {
+            const count = availability[gpu.value] ?? 0;
+            const parts = [gpu.memory];
+            if (gpu.hint)
+                parts.push(gpu.hint);
+            if (count > 0) {
+                parts.push(`\x1b[32m${count} available\x1b[0m`);
+            }
+            else {
+                parts.push(`\x1b[90munavailable\x1b[0m`);
+            }
+            return { value: gpu.value, label: gpu.label, hint: parts.join(" · ") };
+        }),
+        initialValue: Object.keys(availability).find(k => availability[k] > 0) || "rtx3090",
+    });
+    if (p.isCancel(gpuType)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
+    const gpuCountStr = options.gpuCount || await p.select({
+        message: "Number of GPUs",
+        options: [
+            { value: "1", label: "1" },
+            { value: "2", label: "2" },
+            { value: "4", label: "4" },
+            { value: "8", label: "8" },
+        ],
+        initialValue: "1",
+    });
+    if (p.isCancel(gpuCountStr)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
     const gpuCount = parseInt(gpuCountStr);
-    console.log();
-    const diskSize = options.disk || await promptChoice(rl, "  Disk size (fast storage for code + training):", [
-        { label: "20 GB", value: "20G" },
-        { label: "50 GB", value: "50G", hint: "recommended" },
-        { label: "100 GB", value: "100G" },
-        { label: "200 GB", value: "200G" },
-        { label: "500 GB", value: "500G" },
-    ], "50G");
-    console.log();
-    console.log(dim("  Cloud storage gives you unlimited persistent space at /workspace/storage"));
-    console.log(dim("  Great for datasets, model weights, and large files"));
-    const storageAnswer = await prompt(rl, `  Enable cloud storage? ${dim("[Y/n]")}: `, "Y");
-    const storage = storageAnswer.toLowerCase() !== "n";
-    console.log();
-    console.log(dim("  Folders to exclude (build artifacts, can be recreated)"));
-    const excludeRaw = await prompt(rl, `  Exclude ${dim("[node_modules/, .venv/, __pycache__/, *.pyc, wandb/]")}: `, "");
-    const exclude = excludeRaw
-        ? excludeRaw.split(",").map(s => s.trim()).filter(Boolean)
-        : ["node_modules/", ".venv/", "__pycache__/", "*.pyc", "wandb/"];
-    rl.close();
-    const yaml = buildYaml({ name, gpuType, gpuCount, diskSize, storage, exclude });
+    const diskSize = options.disk || await p.select({
+        message: "Disk size (fast NVMe storage)",
+        options: [
+            { value: "20G", label: "20 GB" },
+            { value: "50G", label: "50 GB", hint: "recommended" },
+            { value: "100G", label: "100 GB" },
+            { value: "200G", label: "200 GB" },
+            { value: "500G", label: "500 GB" },
+        ],
+        initialValue: "50G",
+    });
+    if (p.isCancel(diskSize)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
+    const storageEnabled = await p.confirm({
+        message: "Enable cloud storage? \x1b[90m(unlimited, persists at /workspace/storage)\x1b[0m",
+        initialValue: true,
+    });
+    if (p.isCancel(storageEnabled)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
+    const excludeRaw = await p.text({
+        message: "Exclude patterns \x1b[90m(comma-separated, not persisted)\x1b[0m",
+        placeholder: "node_modules/, .venv/, __pycache__/, *.pyc, wandb/",
+        defaultValue: "node_modules/, .venv/, __pycache__/, *.pyc, wandb/",
+    });
+    if (p.isCancel(excludeRaw)) {
+        p.cancel("Setup cancelled.");
+        process.exit(0);
+    }
+    const exclude = excludeRaw.split(",").map(s => s.trim()).filter(Boolean);
+    const yaml = buildYaml({ name, gpuType, gpuCount, diskSize, storage: storageEnabled, exclude });
     writeFileSync("cassian.yaml", yaml);
     console.log();
-    console.log(`  ${green("✓")} Created cassian.yaml`);
-    console.log();
-    console.log(`  Run ${green("cassian up")} to start your instance.`);
+    console.log(`  \x1b[32m✓\x1b[0m Created \x1b[1mcassian.yaml\x1b[0m`);
+    console.log(`  Run \x1b[32mcassian up\x1b[0m to start your instance.`);
     console.log();
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cassian-cli",
-  "version": "0.3.6",
+  "version": "0.3.8",
   "description": "The Cassian GPU cloud CLI — provision GPUs, sync files, and run workloads from your terminal.",
   "type": "module",
   "bin": {
@@ -28,6 +28,7 @@
     "node": ">=18"
   },
   "dependencies": {
+    "@clack/prompts": "^1.4.0",
     "@types/tar": "^6.1.13",
     "chalk": "^5.3.0",
     "chokidar": "^5.0.0",