Add hidden CLI option to specify a partition to provision on (#31128)

GitOrigin-RevId: 0c563b01c2a8995ef9c04a90493fb596cda067f7

Author: emmaling27

src/cli/configure.ts

@@ -75,6 +75,7 @@ export async function deploymentCredentialsOrConfigure(
     url?: string | undefined;
     adminKey?: string | undefined;
   },
+  partitionId?: number | undefined,
 ): Promise<
   DeploymentCredentials & {
     deploymentName?: DeploymentName;
@@ -90,7 +91,11 @@ export async function deploymentCredentialsOrConfigure(
   const { projectSlug, teamSlug } = await selectProject(
     ctx,
     chosenConfiguration,
-    { team: cmdOptions.team, project: cmdOptions.project },
+    {
+      team: cmdOptions.team,
+      project: cmdOptions.project,
+      partitionId,
+    },
   );
   const deploymentOptions: DeploymentOptions = cmdOptions.prod
     ? { kind: "prod" }
@@ -149,6 +154,7 @@ async function selectProject(
   cmdOptions: {
     team?: string | undefined;
     project?: string | undefined;
+    partitionId?: number;
   },
 ): Promise<{ teamSlug: string; projectSlug: string }> {
   let result:
@@ -280,6 +286,7 @@ async function selectNewProject(
   config: {
     team?: string | undefined;
     project?: string | undefined;
+    partitionId?: number | undefined;
   },
 ) {
   const { teamSlug: selectedTeam, chosen: didChooseBetweenTeams } =
@@ -299,6 +306,7 @@ async function selectNewProject(
     ({ projectSlug, teamSlug, projectsRemaining } = await createProject(ctx, {
       teamSlug: selectedTeam,
       projectName,
+      partitionId: config.partitionId,
     }));
   } catch (err) {
     logFailure(ctx, "Unable to create project.");

src/cli/dev.ts

@@ -125,6 +125,7 @@ export const dev = new Command("dev")
   .addOption(new Option("--local-backend-version <version>").hideHelp())
   .addOption(new Option("--local-force-upgrade").default(false).hideHelp())
   .addOption(new Option("--live-component-sources").hideHelp())
+  .addOption(new Option("--partition-id <id>").hideHelp())
   .showHelpAfterError()
   .action(async (cmdOptions) => {
     const ctx = oneoffContext();
@@ -194,12 +195,20 @@ export const dev = new Command("dev")
       }
     }
 
+    const partitionId = cmdOptions.partitionId
+      ? parseInt(cmdOptions.partitionId)
+      : undefined;
     const configure =
       cmdOptions.configure === true ? "ask" : cmdOptions.configure ?? null;
-    const credentials = await deploymentCredentialsOrConfigure(ctx, configure, {
-      ...cmdOptions,
-      localOptions,
-    });
+    const credentials = await deploymentCredentialsOrConfigure(
+      ctx,
+      configure,
+      {
+        ...cmdOptions,
+        localOptions,
+      },
+      partitionId,
+    );
 
     await usageStateWarning(ctx);
 

src/cli/lib/api.ts

@@ -36,7 +36,8 @@ export async function createProject(
   {
     teamSlug: selectedTeamSlug,
     projectName,
-  }: { teamSlug: string; projectName: string },
+    partitionId,
+  }: { teamSlug: string; projectName: string; partitionId?: number },
 ): Promise<{
   projectSlug: string;
   teamSlug: string;
@@ -49,6 +50,7 @@ export async function createProject(
     // to provisioning prod on creation.
     deploymentType: "dev",
     backendVersionOverride: process.env.CONVEX_BACKEND_VERSION_OVERRIDE,
+    partitionId,
   };
   const data = await bigBrainAPI({
     ctx,