chore: import upstream snapshot with attribution

apps/sim/lib/copilot/server/agent-url.test.ts

@@ -0,0 +1,145 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+import {
+  getMothershipBaseURL,
+  getMothershipSourceEnvHeaders,
+  MOTHERSHIP_SOURCE_ENV_HEADER,
+} from './agent-url'
+
+const { dbMock, envMock, mockRows } = vi.hoisted(() => {
+  const mockRows: any[] = []
+  const dbMock = {
+    select: vi.fn(() => ({
+      from: vi.fn(() => ({
+        leftJoin: vi.fn(() => ({
+          where: vi.fn(() => ({
+            limit: vi.fn(async () => mockRows),
+          })),
+        })),
+      })),
+    })),
+  }
+  const envMock = {
+    COPILOT_DEV_URL: 'https://dev.mothership.test',
+    COPILOT_STAGING_URL: 'https://staging.mothership.test',
+    COPILOT_PROD_URL: 'https://prod.mothership.test',
+    COPILOT_SOURCE_ENV: undefined as string | undefined,
+  }
+  return { dbMock, envMock, mockRows }
+})
+
+vi.mock('@sim/db', () => ({ db: dbMock }))
+vi.mock('@sim/db/schema', () => ({
+  settings: {
+    userId: 'settings.userId',
+    superUserModeEnabled: 'settings.superUserModeEnabled',
+    mothershipEnvironment: 'settings.mothershipEnvironment',
+  },
+  user: {
+    id: 'user.id',
+    role: 'user.role',
+  },
+}))
+vi.mock('drizzle-orm', () => ({
+  eq: vi.fn(() => ({})),
+}))
+vi.mock('@/lib/api/contracts', () => ({
+  mothershipEnvironmentSchema: {
+    safeParse: (value: unknown) =>
+      ['default', 'dev', 'staging', 'prod'].includes(String(value))
+        ? { success: true, data: value }
+        : { success: false },
+  },
+}))
+vi.mock('@/lib/copilot/constants', () => ({
+  SIM_AGENT_API_URL: 'https://default.mothership.test',
+  SIM_AGENT_API_URL_DEFAULT: 'https://fallback.mothership.test',
+}))
+vi.mock('@/lib/core/config/env', () => ({
+  env: envMock,
+}))
+
+describe('getMothershipBaseURL', () => {
+  beforeEach(() => {
+    mockRows.length = 0
+    dbMock.select.mockClear()
+    envMock.COPILOT_SOURCE_ENV = undefined
+  })
+
+  it('uses the default URL when there is no user context', async () => {
+    await expect(getMothershipBaseURL()).resolves.toBe('https://default.mothership.test')
+    await expect(getMothershipBaseURL({ environment: 'dev' })).resolves.toBe(
+      'https://default.mothership.test'
+    )
+  })
+
+  it('ignores stored and explicit environments for non-admin users', async () => {
+    mockRows.push({
+      role: 'user',
+      superUserModeEnabled: true,
+      mothershipEnvironment: 'dev',
+    })
+
+    await expect(getMothershipBaseURL({ userId: 'user-1', environment: 'staging' })).resolves.toBe(
+      'https://default.mothership.test'
+    )
+  })
+
+  it('ignores stored and explicit environments when super user mode is off', async () => {
+    mockRows.push({
+      role: 'admin',
+      superUserModeEnabled: false,
+      mothershipEnvironment: 'dev',
+    })
+
+    await expect(getMothershipBaseURL({ userId: 'admin-1', environment: 'prod' })).resolves.toBe(
+      'https://default.mothership.test'
+    )
+  })
+
+  it('uses default for super admins until they select a concrete environment', async () => {
+    mockRows.push({
+      role: 'admin',
+      superUserModeEnabled: true,
+      mothershipEnvironment: 'default',
+    })
+
+    await expect(getMothershipBaseURL({ userId: 'admin-1' })).resolves.toBe(
+      'https://default.mothership.test'
+    )
+  })
+
+  it('allows effective super admins to use a selected environment', async () => {
+    mockRows.push({
+      role: 'admin',
+      superUserModeEnabled: true,
+      mothershipEnvironment: 'dev',
+    })
+
+    await expect(getMothershipBaseURL({ userId: 'admin-1' })).resolves.toBe(
+      'https://dev.mothership.test'
+    )
+    await expect(getMothershipBaseURL({ userId: 'admin-1', environment: 'staging' })).resolves.toBe(
+      'https://staging.mothership.test'
+    )
+  })
+})
+
+describe('getMothershipSourceEnvHeaders', () => {
+  beforeEach(() => {
+    envMock.COPILOT_SOURCE_ENV = undefined
+  })
+
+  it('emits the source environment header for known hosted environments', () => {
+    envMock.COPILOT_SOURCE_ENV = 'dev'
+
+    expect(getMothershipSourceEnvHeaders()).toEqual({
+      [MOTHERSHIP_SOURCE_ENV_HEADER]: 'dev',
+    })
+  })
+
+  it('omits the source environment header for unknown values', () => {
+    envMock.COPILOT_SOURCE_ENV = 'local'
+
+    expect(getMothershipSourceEnvHeaders()).toEqual({})
+  })
+})

apps/sim/lib/copilot/server/agent-url.ts

@@ -0,0 +1,79 @@
+import { db } from '@sim/db'
+import { settings, user } from '@sim/db/schema'
+import { eq } from 'drizzle-orm'
+import { type MothershipEnvironment, mothershipEnvironmentSchema } from '@/lib/api/contracts'
+import { SIM_AGENT_API_URL, SIM_AGENT_API_URL_DEFAULT } from '@/lib/copilot/constants'
+import { env } from '@/lib/core/config/env'
+
+export interface GetMothershipBaseURLOptions {
+  userId?: string | null
+  environment?: MothershipEnvironment
+  fallbackUrl?: string | null
+}
+
+type ConcreteMothershipEnvironment = Exclude<MothershipEnvironment, 'default'>
+type MothershipSourceEnvironment = 'dev' | 'staging' | 'prod'
+
+export const MOTHERSHIP_SOURCE_ENV_HEADER = 'X-Sim-Source-Env'
+
+const ENVIRONMENT_URLS: Record<ConcreteMothershipEnvironment, string | undefined> = {
+  // env vars
+  dev: env.COPILOT_DEV_URL,
+  staging: env.COPILOT_STAGING_URL,
+  prod: env.COPILOT_PROD_URL,
+}
+
+const SOURCE_ENVIRONMENTS = new Set<MothershipSourceEnvironment>(['dev', 'staging', 'prod'])
+
+function normalizeUrl(url: string | undefined): string | null {
+  if (!url) return null
+  return url.startsWith('http://') || url.startsWith('https://') ? url : null
+}
+
+function getConfiguredEnvironmentUrl(environment: MothershipEnvironment): string | null {
+  if (environment === 'default') return null
+  return normalizeUrl(ENVIRONMENT_URLS[environment])
+}
+
+function getDefaultMothershipBaseURL(fallbackUrl?: string | null): string {
+  const fallback = typeof fallbackUrl === 'string' ? fallbackUrl : undefined
+  return normalizeUrl(fallback) ?? normalizeUrl(SIM_AGENT_API_URL) ?? SIM_AGENT_API_URL_DEFAULT
+}
+
+export async function getMothershipBaseURL(
+  options: GetMothershipBaseURLOptions = {}
+): Promise<string> {
+  const defaultUrl = getDefaultMothershipBaseURL(options.fallbackUrl)
+
+  const { userId } = options
+  if (!userId) return defaultUrl
+
+  const [row] = await db
+    .select({
+      role: user.role,
+      superUserModeEnabled: settings.superUserModeEnabled,
+      mothershipEnvironment: settings.mothershipEnvironment,
+    })
+    .from(user)
+    .leftJoin(settings, eq(settings.userId, user.id))
+    .where(eq(user.id, userId))
+    .limit(1)
+
+  const effectiveSuperUser = row?.role === 'admin' && (row.superUserModeEnabled ?? false)
+  if (!effectiveSuperUser) return defaultUrl
+
+  const selectedEnvironment = options.environment ?? row.mothershipEnvironment
+  const parsedEnvironment = mothershipEnvironmentSchema.safeParse(selectedEnvironment)
+  const environment = parsedEnvironment.success ? parsedEnvironment.data : 'default'
+
+  return getConfiguredEnvironmentUrl(environment) ?? defaultUrl
+}
+
+export function getMothershipSourceEnvHeaders(): Record<string, string> {
+  const sourceEnv = env.COPILOT_SOURCE_ENV?.trim().toLowerCase()
+  if (!sourceEnv || !SOURCE_ENVIRONMENTS.has(sourceEnv as MothershipSourceEnvironment)) {
+    return {}
+  }
+
+  return { [MOTHERSHIP_SOURCE_ENV_HEADER]: sourceEnv }
+}