gigadrive 2.11.0 → 2.13.0

package/dist/index.mjs

@@ -9,7 +9,7 @@ import { Effect as Effect29, Layer, LogLevel, Logger } from "effect";
 // package.json
 var package_default = {
   name: "gigadrive",
-  version: "2.11.0",
+  version: "2.13.0",
   repository: {
     type: "git",
     url: "https://github.com/Gigadrive/sdk.git",
@@ -3323,29 +3323,32 @@ var withClient = (run) => Effect22.gen(function* () {
   return yield* Effect22.tryPromise({ try: () => run(client), catch: toApiRequestError });
 });
 var signalOnInterrupt = (signal) => (command, exit) => Exit.isInterrupted(exit) ? Effect22.tryPromise(() => command.kill(signal)).pipe(Effect22.timeout("2 seconds"), Effect22.ignore) : Effect22.void;
-var createInput = (flags) => Effect22.gen(function* () {
-  const env = yield* keyValues("--env", flags.env);
-  const metadata = yield* keyValues("--metadata", flags.metadata);
-  const timeoutMs = yield* Option14.match(flags.timeout, {
-    onNone: () => Effect22.succeed(void 0),
-    onSome: (text) => parseLifetime("--timeout", text)
-  });
-  if (flags.denyEgress && flags.allowEgress.length > 0) {
+var checkEgressFlags = (denyEgress, allowEgress) => Effect22.gen(function* () {
+  if (denyEgress && allowEgress.length > 0) {
     return yield* new InvalidMicroVmOptionsError({
       message: "--allow-egress cannot be combined with --deny-egress."
     });
   }
-  if (flags.allowEgress.length > 32) {
+  if (allowEgress.length > 32) {
     return yield* new InvalidMicroVmOptionsError({ message: "--allow-egress takes at most 32 entries." });
   }
   const network = /^\d{1,3}(\.\d{1,3}){3}\/\d{1,2}$/;
   const domain = /^(\*\.)?([a-z0-9-]+\.)+[a-z0-9-]*[a-z][a-z0-9-]*\.?$/i;
-  const badEntry = flags.allowEgress.find((entry) => !network.test(entry.trim()) && !domain.test(entry.trim()));
+  const badEntry = allowEgress.find((entry) => !network.test(entry.trim()) && !domain.test(entry.trim()));
   if (badEntry !== void 0) {
     return yield* new InvalidMicroVmOptionsError({
       message: `--allow-egress must be a domain name such as api.openai.com or *.github.com, or an IPv4 network such as 140.82.112.0/20 (got ${badEntry}).`
     });
   }
+});
+var createInput = (flags) => Effect22.gen(function* () {
+  const env = yield* keyValues("--env", flags.env);
+  const metadata = yield* keyValues("--metadata", flags.metadata);
+  const timeoutMs = yield* Option14.match(flags.timeout, {
+    onNone: () => Effect22.succeed(void 0),
+    onSome: (text) => parseLifetime("--timeout", text)
+  });
+  yield* checkEgressFlags(flags.denyEgress, flags.allowEgress);
   const badPort = flags.ports.find((port) => !Number.isInteger(port) || port < 1 || port > 65535);
   if (badPort !== void 0) {
     return yield* new InvalidMicroVmOptionsError({ message: `--port must be between 1 and 65535 (got ${badPort}).` });
@@ -3361,6 +3364,8 @@ var createInput = (flags) => Effect22.gen(function* () {
     ...Option14.match(flags.memory, { onNone: () => ({}), onSome: (memoryMb) => ({ memoryMb }) }),
     ...Option14.match(flags.disk, { onNone: () => ({}), onSome: (diskMb) => ({ diskMb }) }),
     ...timeoutMs !== void 0 ? { timeoutMs } : {},
+    ...flags.pauseOnTimeout ? { onTimeout: "pause" } : {},
+    ...flags.autoResume ? { autoResume: true } : {},
     ...flags.ports.length > 0 ? { ports: [...flags.ports] } : {},
     ...env ? { env } : {},
     ...metadata ? { metadata } : {},
@@ -3368,7 +3373,7 @@ var createInput = (flags) => Effect22.gen(function* () {
       network: {
         egress: flags.denyEgress ? "deny" : "allow",
         ...flags.privatePreviews ? { previews: "private" } : {},
-        ...flags.allowEgress.length > 0 ? { allow: [...flags.allowEgress] } : {}
+        ...flags.allowEgress.length > 0 ? { allow: flags.allowEgress.map((entry) => entry.trim()) } : {}
       }
     } : {},
     ...Option14.match(flags.region, { onNone: () => ({}), onSome: (region) => ({ region }) })
@@ -3440,6 +3445,14 @@ var createCommand = Command12.make(
       ),
       Options7.repeated
     ),
+    pauseOnTimeout: Options7.boolean("pause-on-timeout").pipe(
+      Options7.withDescription("Pause the microVM when its timeout ends instead of stopping it")
+    ),
+    autoResume: Options7.boolean("auto-resume").pipe(
+      Options7.withDescription(
+        "Resume the paused microVM when a command, file call or public preview request reaches it"
+      )
+    ),
     region: Options7.text("region").pipe(Options7.withDescription("Region to place the microVM in"), Options7.optional),
     json: jsonOption2
   },
@@ -3495,6 +3508,36 @@ var getCommand = Command12.make(
     for (const line of describeMicroVm(vm)) yield* Console12.log(line);
   }).pipe(reportErrors2("get the microVM"))
 ).pipe(Command12.withDescription("Show a microVM"));
+var formatGigabytes = (bytes) => `${(bytes / 1e9).toFixed(1)} GB`;
+var describeMicroVmMetrics = (metrics) => {
+  const latest = metrics.cpu.at(-1);
+  const peak = metrics.cpu.reduce((max, point) => Math.max(max, point.cpuPercent), 0);
+  return [
+    `CPU (last ${metrics.windowMinutes} min, ${metrics.vcpus} vCPU): ${latest ? `${latest.cpuPercent}% at ${latest.time}, peak ${peak}%` : "no samples"}`,
+    metrics.current ? `Memory: ${formatGigabytes(metrics.current.memoryUsedBytes)} of ${formatGigabytes(metrics.current.memoryTotalBytes)}` : "Memory: unavailable (not running, or no answer)",
+    metrics.current ? `Disk:   ${formatGigabytes(metrics.current.diskUsedBytes)} of ${formatGigabytes(metrics.current.diskTotalBytes)}` : "Disk:   unavailable (not running, or no answer)"
+  ];
+};
+var metricsCommand = Command12.make(
+  "metrics",
+  {
+    microvmId: microvmIdArg,
+    window: Options7.integer("window").pipe(
+      Options7.withDescription("Minutes of CPU history, 1 to 1440 (default: 15)"),
+      Options7.optional
+    ),
+    json: jsonOption2
+  },
+  ({ microvmId, window, json }) => Effect22.gen(function* () {
+    const apiClient = yield* ApiClientService;
+    const windowMinutes = Option14.getOrUndefined(window);
+    const metrics = yield* apiClient.request(
+      (client) => client.microvms.getMetrics(microvmId, windowMinutes === void 0 ? {} : { windowMinutes })
+    );
+    if (json) return yield* Console12.log(JSON.stringify(metrics, null, 2));
+    for (const line of describeMicroVmMetrics(metrics)) yield* Console12.log(line);
+  }).pipe(reportErrors2("read the microVM metrics"))
+).pipe(Command12.withDescription("Show a microVM's CPU, memory and disk use"));
 var execInMicroVm = (flags, io = nodeSessionIo()) => Effect22.gen(function* () {
   const env = yield* keyValues("--env", flags.env);
   const timeout = yield* optionalDuration("--timeout", flags.timeout);
@@ -3756,6 +3799,46 @@ var extendCommand = Command12.make(
   },
   ({ microvmId, duration }) => extendMicroVm(microvmId, duration).pipe(Effect22.asVoid, reportErrors2("extend the microVM"))
 ).pipe(Command12.withDescription("Set how long a microVM keeps running, counted from now"));
+var updateMicroVmNetwork = (microvmId, flags) => Effect22.gen(function* () {
+  const chosen = [flags.denyEgress, flags.allowEgress.length > 0, flags.allowAllEgress].filter(Boolean).length;
+  if (chosen !== 1) {
+    return yield* new InvalidMicroVmOptionsError({
+      message: "Pass exactly one of --deny-egress, --allow-egress or --allow-all-egress."
+    });
+  }
+  yield* checkEgressFlags(flags.denyEgress, flags.allowEgress);
+  const apiClient = yield* ApiClientService;
+  const vm = yield* apiClient.request(
+    (client) => client.microvms.updateNetwork(
+      microvmId,
+      flags.denyEgress ? { egress: "deny" } : flags.allowEgress.length > 0 ? { egress: "allow", allow: flags.allowEgress.map((entry) => entry.trim()) } : { egress: "allow" }
+    )
+  );
+  const reach = vm.network.egress === "deny" ? "no outbound traffic" : vm.network.allow ? vm.network.allow.join(", ") : "the whole internet";
+  yield* Console12.log(`${vm.id} now reaches ${reach}.`);
+  return vm;
+});
+var networkCommand = Command12.make(
+  "network",
+  {
+    microvmId: microvmIdArg,
+    denyEgress: Options7.boolean("deny-egress").pipe(Options7.withDescription("Block all outbound network traffic")),
+    allowEgress: Options7.text("allow-egress").pipe(
+      Options7.withDescription(
+        "Only reach this domain name (api.openai.com, *.github.com for every subdomain) or public IPv4 network (140.82.112.0/20) (repeatable, up to 32)"
+      ),
+      Options7.repeated
+    ),
+    allowAllEgress: Options7.boolean("allow-all-egress").pipe(
+      Options7.withDescription("Reach the whole internet again (private networks stay blocked)")
+    )
+  },
+  ({ microvmId, ...flags }) => updateMicroVmNetwork(microvmId, flags).pipe(Effect22.asVoid, reportErrors2("update the microVM network"))
+).pipe(
+  Command12.withDescription(
+    "Change what a running or paused microVM may reach; connections the new rules no longer allow are closed"
+  )
+);
 var imagesListCommand = Command12.make(
   "ls",
   { app: appOption6, json: jsonOption2 },
@@ -3856,12 +3939,14 @@ var SUBCOMMANDS = [
   createCommand,
   listCommand2,
   getCommand,
+  metricsCommand,
   execCommand,
   shellCommand,
   copyCommand,
   stopCommand,
   removeCommand2,
   extendCommand,
+  networkCommand,
   imagesCommand
 ];
 var microvmsCommand = Command12.make("microvms", {}, () => Effect22.void).pipe(