feat: Forward endpoint URL to migrations package

src/bin.ts

@@ -2409,9 +2409,11 @@ yargs(rawArgs)
  async (argv) => {
  await applyInsecureStorage(argv.insecureStorage);
  const { resolveOptionalApiKey } = await import('./lib/api-key.js');
+ const { getActiveEnvironment } = await import('./lib/config-store.js');
  const { getMigrationsPassthroughArgs, runMigrations } = await import('./commands/migrations.js');
  const passthrough = getMigrationsPassthroughArgs(rawArgs);
- await runMigrations(passthrough, resolveOptionalApiKey({ apiKey: argv.apiKey }));
+ const endpoint = getActiveEnvironment()?.endpoint;
+ await runMigrations(passthrough, resolveOptionalApiKey({ apiKey: argv.apiKey }), endpoint);
  },
  )
  .command(

src/commands/migrations.spec.ts

@@ -13,6 +13,7 @@ describe('runMigrations', () => {
  beforeEach(() => {
  vi.clearAllMocks();
  delete process.env.WORKOS_SECRET_KEY;
+ delete process.env.WORKOS_API_URL;
  });
 
  it('sets WORKOS_SECRET_KEY from the provided API key', async () => {
@@ -42,6 +43,16 @@ describe('runMigrations', () => {
  expect(mockParseAsync).toHaveBeenCalledWith(args, { from: 'user' });
  });
 
+ it('sets WORKOS_API_URL when apiBaseUrl is provided', async () => {
+ await runMigrations(['import', '--csv', 'users.csv'], 'sk_test_123', 'https://api.staging.workos.com');
+ expect(process.env.WORKOS_API_URL).toBe('https://api.staging.workos.com');
+ });
+
+ it('does not set WORKOS_API_URL when apiBaseUrl is undefined', async () => {
+ await runMigrations(['import', '--csv', 'users.csv'], 'sk_test_123');
+ expect(process.env.WORKOS_API_URL).toBeUndefined();
+ });
+
  it('removes WorkOS global flags from migrations passthrough args', () => {
  expect(
  getMigrationsPassthroughArgs([

src/commands/migrations.ts

@@ -43,10 +43,13 @@ export function getMigrationsPassthroughArgs(rawArgs: string[]): string[] {
  return passthrough;
 }
 
-export async function runMigrations(args: string[], apiKey?: string): Promise<void> {
+export async function runMigrations(args: string[], apiKey?: string, apiBaseUrl?: string): Promise<void> {
  if (apiKey) {
  process.env.WORKOS_SECRET_KEY = apiKey;
  }
+ if (apiBaseUrl) {
+ process.env.WORKOS_API_URL = apiBaseUrl;
+ }
 
  const { program } = (await import('@workos/migrations/dist/cli/index.js')) as {
  program: {