workstation.md 0.1.1 → 0.3.0

package/bin/workstation.d.ts

@@ -1,14 +1,2 @@
 #!/usr/bin/env node
-declare const API_URL: string;
-interface WorkstationInfo {
-    id: string;
-    host: string;
-    port: number;
-    web: string;
-}
-declare function apiRequest(path: string, options?: RequestInit): Promise<any>;
-declare function create(args: string[]): Promise<void>;
-declare function destroy(wsId: string): Promise<void>;
-declare function list(): Promise<void>;
-declare function usage(): void;
-declare function main(): Promise<void>;
+export {};

package/bin/workstation.js

@@ -1,5 +1,8 @@
 #!/usr/bin/env node
-"use strict";
+import { execSync } from "child_process";
+import { writeFileSync, mkdtempSync, readFileSync } from "fs";
+import { tmpdir } from "os";
+import { join } from "path";
 const API_URL = process.env.WORKSTATION_API_URL || "https://api.workstation.md";
 async function apiRequest(path, options = {}) {
     const res = await fetch(`${API_URL}${path}`, {
@@ -16,21 +19,65 @@ async function apiRequest(path, options = {}) {
     }
     return res.json();
 }
+function findKeyPath(args) {
+    for (let i = 0; i < args.length; i++) {
+        if (args[i] === "--key" && args[i + 1]) {
+            return args[i + 1];
+        }
+    }
+    // Default: try common key paths
+    const home = process.env.HOME || "~";
+    const defaults = [
+        join(home, ".ssh", "id_ed25519"),
+        join(home, ".ssh", "id_rsa"),
+    ];
+    for (const p of defaults) {
+        try {
+            readFileSync(p);
+            return p;
+        }
+        catch { }
+    }
+    console.error("No SSH private key found. Use --key <path> to specify.");
+    process.exit(1);
+}
+function signWithSSHKey(keyPath, data) {
+    const tmp = mkdtempSync(join(tmpdir(), "ws-"));
+    const dataPath = join(tmp, "data");
+    writeFileSync(dataPath, data);
+    try {
+        execSync(`ssh-keygen -Y sign -f "${keyPath}" -n workstation.md "${dataPath}"`, { stdio: ["pipe", "pipe", "pipe"] });
+        const sigPath = dataPath + ".sig";
+        return readFileSync(sigPath, "utf-8");
+    }
+    catch (e) {
+        console.error("Failed to sign with SSH key:", e.stderr?.toString() || e.message);
+        process.exit(1);
+    }
+}
 async function create(args) {
     let pubkey;
+    let name;
     for (let i = 0; i < args.length; i++) {
         if (args[i] === "--pubkey" && args[i + 1]) {
             pubkey = args[i + 1];
             i++;
         }
+        else if (args[i] === "--name" && args[i + 1]) {
+            name = args[i + 1];
+            i++;
+        }
     }
     if (!pubkey) {
-        console.error("Usage: workstation create --pubkey <public_key>");
+        console.error("Usage: workstation create --pubkey <public_key> [--name <name>]");
         process.exit(1);
     }
+    const body = { pubkey };
+    if (name)
+        body.name = name;
     const info = await apiRequest("/create", {
         method: "POST",
-        body: JSON.stringify({ pubkey }),
+        body: JSON.stringify(body),
     });
     console.log(JSON.stringify(info, null, 2));
 }
@@ -38,6 +85,15 @@ async function destroy(wsId) {
     const result = await apiRequest(`/${wsId}`, { method: "DELETE" });
     console.log(JSON.stringify(result, null, 2));
 }
+async function extend(wsId, args) {
+    const keyPath = findKeyPath(args);
+    const signature = signWithSSHKey(keyPath, wsId);
+    const result = await apiRequest(`/${wsId}/extend`, {
+        method: "POST",
+        body: JSON.stringify({ signature }),
+    });
+    console.log(JSON.stringify(result, null, 2));
+}
 async function list() {
     const items = await apiRequest("/list");
     if (items.length === 0) {
@@ -45,13 +101,17 @@ async function list() {
         return;
     }
     for (const ws of items) {
-        console.log(`${ws.id}\tssh -p ${ws.port} root@${ws.host}\t${ws.web}`);
+        const ttl = ws.expires - Math.floor(Date.now() / 1000);
+        const hours = Math.max(0, Math.floor(ttl / 3600));
+        const mins = Math.max(0, Math.floor((ttl % 3600) / 60));
+        console.log(`${ws.id}\tssh -p ${ws.port} root@${ws.host}\t${ws.web}\t${hours}h${mins}m remaining`);
     }
 }
 function usage() {
     console.log(`Usage:
-  workstation create --pubkey <public_key>    Create a new workstation
+  workstation create --pubkey <key> [--name <name>]  Create a workstation (24h TTL)
   workstation <id> destroy                    Destroy a workstation
+  workstation <id> extend [--key <path>]      Extend TTL by 24h (proves key ownership)
   workstation list                            List active workstations`);
 }
 async function main() {
@@ -70,6 +130,9 @@ async function main() {
     else if (args.length >= 2 && args[1] === "destroy") {
         await destroy(command);
     }
+    else if (args.length >= 2 && args[1] === "extend") {
+        await extend(command, args.slice(2));
+    }
     else {
         usage();
         process.exit(1);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "workstation.md",
-  "version": "0.1.1",
+  "version": "0.3.0",
   "description": "Cloud Linux workstations for AI agents",
   "type": "module",
   "bin": {