feat(nodes): add feature-flag-gated --enable-infiniband to sf nodes create

The option is only mounted when the user's account has the
`infiniband-preview` flag enabled (checked via the existing posthog/
feature-flag helper) and forwarded to the create factory. When enabled,
it shows up in `sf nodes create --help` and forwards
`_preview_enable_infiniband: true` to the nodes SDK, exposing the
InfiniBand preview surface in the sfcompute/sfcompute API.

Generated with [Indent](https://indent.com)
Co-Authored-By: Indent <noreply@indent.com>

Author: sigmachirality

src/lib/nodes/create.ts

@@ -68,7 +68,11 @@ function validateCount(val: string): number {
   return parsed;
 }
 
-export function createCreateCommand() {
+export function createCreateCommand({
+  enableInfiniband,
+}: {
+  enableInfiniband: boolean;
+}) {
   const create = new Command("create")
     .description("Create reserved or auto reserved nodes")
     .showHelpAfterError()
@@ -240,12 +244,22 @@ Examples:\n
     )
     .action(createNodesAction);
 
+  if (enableInfiniband) {
+    create.addOption(
+      new Option(
+        "--enable-infiniband",
+        "Enable InfiniBand passthrough on the node VMs.",
+      ),
+    );
+  }
+
   return create;
 
   async function createNodesAction(
     names: typeof create.args,
     options: ReturnType<typeof create.opts> & {
       image?: string;
+      enableInfiniband?: boolean;
     },
   ) {
     try {
@@ -282,6 +296,9 @@ Examples:\n
         cloud_init_user_data: encodedUserData,
         image_id: options.image,
         node_type: isReserved ? "reserved" : "autoreserved",
+        ...(options.enableInfiniband
+          ? { _preview_enable_infiniband: true }
+          : {}),
       };
 
       if (isReserved) {

src/lib/nodes/index.ts

@@ -1,6 +1,7 @@
 import console from "node:console";
 import type { Command } from "@commander-js/extra-typings";
 
+import { isFeatureEnabled } from "../posthog.ts";
 import { createCreateCommand } from "./create.ts";
 import deleteCommand from "./delete.ts";
 import extend from "./extend.ts";
@@ -14,14 +15,16 @@ import set from "./set.ts";
 import ssh from "./ssh.ts";
 
 export async function registerNodes(program: Command) {
+  const enableInfiniband = await isFeatureEnabled("infiniband-preview");
+
   const nodes = program
     .command("nodes")
     .alias("node")
     .showHelpAfterError()
     .description("Manage compute nodes")
     .addCommand(list)
     .addCommand(get)
-    .addCommand(createCreateCommand())
+    .addCommand(createCreateCommand({ enableInfiniband }))
     .addCommand(extend)
     .addCommand(release)
     .addCommand(deleteCommand)

src/lib/posthog.ts

@@ -62,7 +62,7 @@ const trackEvent = ({
   }
 };
 
-type FeatureFlags = "procurements" | "zones";
+type FeatureFlags = "procurements" | "zones" | "infiniband-preview";
 
 /**
  * Checks if a feature is enabled for the current user.