gigadrive 2.15.0 → 2.16.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.15.0",
+  version: "2.16.0",
   repository: {
     type: "git",
     url: "https://github.com/Gigadrive/sdk.git",
@@ -3063,6 +3063,7 @@ var describeMicroVm = (vm, now = Date.now()) => {
     `Size:        ${vm.vcpus} vCPU, ${formatMemory(vm.memoryMb)} memory, ${formatMemory(vm.diskMb)} disk`,
     `Region:      ${vm.region}`,
     `Network:     ${vm.network.egress === "deny" ? "outbound traffic blocked" : vm.network.allow && vm.network.allow.length > 0 ? `only ${vm.network.allow.join(", ")}` : "internet access"}`,
+    ...(vm.network.inject ?? []).map((rule) => `Injects:     ${rule.headers.join(", ")} for ${rule.domain}`),
     `Previews:    ${vm.network.previews === "private" ? "private (preview token required)" : "public"}`,
     `Application: ${vm.applicationId}`,
     `Created:     ${vm.createdAt}`
@@ -3373,6 +3374,26 @@ var parseVolumeFlags = (values) => Effect22.gen(function* () {
   }
   return volumes;
 });
+var parseInjectFlags = (values) => Effect22.gen(function* () {
+  const rules = /* @__PURE__ */ new Map();
+  for (const value of values) {
+    const colon = value.indexOf(":");
+    const equals = value.indexOf("=", colon + 1);
+    const domain = colon > 0 ? value.slice(0, colon).trim().toLowerCase() : "";
+    const name = colon > 0 && equals > colon ? value.slice(colon + 1, equals).trim() : "";
+    if (domain === "" || name === "" || equals === value.length - 1) {
+      return yield* new InvalidMicroVmOptionsError({
+        // The entry is not echoed: it may hold the credential.
+        message: '--inject must be DOMAIN:HEADER=VALUE, such as "api.openai.com:Authorization=Bearer $OPENAI_API_KEY".'
+      });
+    }
+    rules.set(domain, { ...rules.get(domain), [name]: value.slice(equals + 1) });
+  }
+  if (rules.size > 8) {
+    return yield* new InvalidMicroVmOptionsError({ message: "--inject takes at most 8 domains." });
+  }
+  return [...rules].map(([domain, headers]) => ({ domain, headers }));
+});
 var checkEgressFlags = (denyEgress, allowEgress) => Effect22.gen(function* () {
   if (denyEgress && allowEgress.length > 0) {
     return yield* new InvalidMicroVmOptionsError({
@@ -3403,6 +3424,10 @@ var createInput = (flags) => Effect22.gen(function* () {
   if (badPort !== void 0) {
     return yield* new InvalidMicroVmOptionsError({ message: `--port must be between 1 and 65535 (got ${badPort}).` });
   }
+  const inject = yield* parseInjectFlags(flags.inject);
+  if (inject.length > 0 && flags.denyEgress) {
+    return yield* new InvalidMicroVmOptionsError({ message: "--inject cannot be combined with --deny-egress." });
+  }
   const volumes = yield* parseVolumeFlags(flags.volumes);
   if (volumes.length > 0 && flags.pauseOnTimeout) {
     return yield* new InvalidMicroVmOptionsError({ message: "--volume cannot be combined with --pause-on-timeout." });
@@ -3424,11 +3449,12 @@ var createInput = (flags) => Effect22.gen(function* () {
     ...env ? { env } : {},
     ...metadata ? { metadata } : {},
     ...volumes.length > 0 ? { volumes } : {},
-    ...flags.denyEgress || flags.privatePreviews || flags.allowEgress.length > 0 ? {
+    ...flags.denyEgress || flags.privatePreviews || flags.allowEgress.length > 0 || inject.length > 0 ? {
       network: {
         egress: flags.denyEgress ? "deny" : "allow",
         ...flags.privatePreviews ? { previews: "private" } : {},
-        ...flags.allowEgress.length > 0 ? { allow: flags.allowEgress.map((entry) => entry.trim()) } : {}
+        ...flags.allowEgress.length > 0 ? { allow: flags.allowEgress.map((entry) => entry.trim()) } : {},
+        ...inject.length > 0 ? { inject } : {}
       }
     } : {},
     ...Option14.match(flags.region, { onNone: () => ({}), onSome: (region) => ({ region }) })
@@ -3519,6 +3545,12 @@ var createCommand = Command12.make(
       ),
       Options7.repeated
     ),
+    inject: Options7.text("inject").pipe(
+      Options7.withDescription(
+        "Add a header to HTTPS requests to a domain, as DOMAIN:HEADER=VALUE (repeatable, up to 8 domains); the value never enters the microVM"
+      ),
+      Options7.repeated
+    ),
     pauseOnTimeout: Options7.boolean("pause-on-timeout").pipe(
       Options7.withDescription("Pause the microVM when its timeout ends instead of stopping it")
     ),
@@ -3888,15 +3920,31 @@ var updateMicroVmNetwork = (microvmId, flags) => Effect22.gen(function* () {
     });
   }
   yield* checkEgressFlags(flags.denyEgress, flags.allowEgress);
+  const inject = yield* parseInjectFlags(flags.inject);
+  if (inject.length > 0 && flags.denyEgress) {
+    return yield* new InvalidMicroVmOptionsError({ message: "--inject cannot be combined with --deny-egress." });
+  }
+  if (flags.clearInject && (inject.length > 0 || flags.denyEgress)) {
+    return yield* new InvalidMicroVmOptionsError({
+      message: "--clear-inject cannot be combined with --inject or --deny-egress."
+    });
+  }
   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" }
+      flags.denyEgress ? { egress: "deny" } : {
+        egress: "allow",
+        ...flags.allowEgress.length > 0 ? { allow: flags.allowEgress.map((entry) => entry.trim()) } : {},
+        ...inject.length > 0 || flags.clearInject ? { inject } : {}
+      }
     )
   );
   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}.`);
+  for (const rule of vm.network.inject ?? []) {
+    yield* Console12.log(`Adds ${rule.headers.join(", ")} to HTTPS requests to ${rule.domain}.`);
+  }
   return vm;
 });
 var networkCommand = Command12.make(
@@ -3912,6 +3960,15 @@ var networkCommand = Command12.make(
     ),
     allowAllEgress: Options7.boolean("allow-all-egress").pipe(
       Options7.withDescription("Reach the whole internet again (private networks stay blocked)")
+    ),
+    inject: Options7.text("inject").pipe(
+      Options7.withDescription(
+        "Add a header to HTTPS requests to a domain, as DOMAIN:HEADER=VALUE (repeatable, up to 8 domains); the value never enters the microVM"
+      ),
+      Options7.repeated
+    ),
+    clearInject: Options7.boolean("clear-inject").pipe(
+      Options7.withDescription("Remove every header rule added with --inject")
     )
   },
   ({ microvmId, ...flags }) => updateMicroVmNetwork(microvmId, flags).pipe(Effect22.asVoid, reportErrors2("update the microVM network"))