fix contracts ls

Author: Flaque

src/lib/contracts.ts

@@ -2,15 +2,70 @@ import { Command } from "commander";
 import { loadConfig } from "../helpers/config";
 import { logLoginMessageAndQuit } from "../helpers/errors";
 import { getApiUrl } from "../helpers/urls";
+import Table from "cli-table3";
+
+interface Contract {
+  object: string;
+  status: string;
+  id: string;
+  created_at: string;
+  instance_type: string;
+  shape: {
+    // These are date strings
+    intervals: string[];
+    quantities: number[];
+  };
+  colocate_with: string[];
+  cluster_id: string;
+}
+
+function printTable(data: Contract[]) {
+  const table = new Table({
+    head: [
+      "ID",
+      "Status",
+      "Instance Type",
+      // Found by looking at the first interval
+      "Starts At",
+      // Found by looking at the last interval
+      "Ends At",
+    ],
+  });
+
+  for (const contract of data) {
+    const startsAt = contract.shape.intervals[0];
+    const endsAt =
+      contract.shape.intervals[contract.shape.intervals.length - 1];
+    table.push([
+      contract.id,
+      contract.status,
+      contract.instance_type,
+      new Date(startsAt).toLocaleString(),
+      new Date(endsAt).toLocaleString(),
+    ]);
+  }
+
+  console.log(table.toString());
+}
 
 export function registerContracts(program: Command) {
   program
     .command("contracts")
     .description("Manage contracts")
     .addCommand(
-      new Command("list").description("List all contracts").action(async () => {
-        await listContracts();
-      }),
+      new Command("list")
+        .alias("ls")
+        .option("--json", "Output in JSON format")
+        .description("List all contracts")
+        .action(async (options) => {
+          if (options.json) {
+            console.log(await listContracts());
+          } else {
+            const data = await listContracts();
+            printTable(data.data);
+          }
+          process.exit(0);
+        }),
     );
 }
 
@@ -29,5 +84,5 @@ async function listContracts() {
   });
 
   const data = await response.json();
-  console.log(data);
+  return data;
 }

src/lib/upgrade.ts

@@ -21,5 +21,7 @@ export function registerUpgrade(program: Command) {
       } else {
         await Bun.$`bash -c "$(curl -fsSL https://www.sfcompute.com/cli/install)"`;
       }
+
+      process.exit(0);
     });
 }