Run migration

Author: Sg312

apps/sim/app/workspace/[workspaceId]/settings/[section]/settings.tsx

@@ -31,7 +31,6 @@ import {
 import { AuditLogsSkeleton } from '@/ee/audit-logs/components/audit-logs-skeleton'
 import { DataDrainsSkeleton } from '@/ee/data-drains/components/data-drains-skeleton'
 import { DataRetentionSkeleton } from '@/ee/data-retention/components/data-retention-skeleton'
-import { useGeneralSettings } from '@/hooks/queries/general-settings'
 
 /**
  * Generic skeleton fallback for sections without a dedicated skeleton.
@@ -147,13 +146,6 @@ const Admin = dynamic(
     import('@/app/workspace/[workspaceId]/settings/components/admin/admin').then((m) => m.Admin),
   { loading: () => <AdminSkeleton /> }
 )
-const Mothership = dynamic(
-  () =>
-    import('@/app/workspace/[workspaceId]/settings/components/mothership/mothership').then(
-      (m) => m.Mothership
-    ),
-  { loading: () => <SettingsSectionSkeleton /> }
-)
 const RecentlyDeleted = dynamic(
   () =>
     import(
@@ -200,24 +192,21 @@ export function SettingsPage({ section }: SettingsPageProps) {
   const searchParams = useSearchParams()
   const mcpServerId = searchParams.get('mcpServerId')
   const { data: session, isPending: sessionLoading } = useSession()
-  const { data: generalSettings, isLoading: generalSettingsLoading } = useGeneralSettings()
   const posthog = usePostHog()
 
   const isAdminRole = session?.user?.role === 'admin'
-  const isEffectiveSuperUser = isAdminRole && (generalSettings?.superUserModeEnabled ?? false)
   const effectiveSection =
     !isBillingEnabled && (section === 'subscription' || section === 'organization')
       ? 'general'
       : section === 'credential-sets' && !isCredentialSetsEnabled
         ? 'general'
         : section === 'admin' && !sessionLoading && !isAdminRole
           ? 'general'
-          : section === 'mothership' &&
-              !sessionLoading &&
-              !generalSettingsLoading &&
-              !isEffectiveSuperUser
-            ? 'general'
-            : section
+          : section === 'mothership' && !sessionLoading && isAdminRole
+            ? 'admin'
+            : section === 'mothership' && !sessionLoading && !isAdminRole
+              ? 'general'
+              : section
 
   const label =
     allNavigationItems.find((item) => item.id === effectiveSection)?.label ?? effectiveSection
@@ -258,7 +247,6 @@ export function SettingsPage({ section }: SettingsPageProps) {
       {effectiveSection === 'inbox' && <Inbox />}
       {effectiveSection === 'recently-deleted' && <RecentlyDeleted />}
       {effectiveSection === 'admin' && <Admin />}
-      {effectiveSection === 'mothership' && <Mothership />}
     </div>
   )
 }

apps/sim/app/workspace/[workspaceId]/settings/components/admin/admin.tsx

@@ -1,8 +1,9 @@
 'use client'
 
-import { useMemo, useState } from 'react'
+import { useCallback, useMemo, useState } from 'react'
 import { useParams } from 'next/navigation'
 import { Badge, Button, Input as EmcnInput, Label, Skeleton, Switch } from '@/components/emcn'
+import type { MothershipEnvironment } from '@/lib/api/contracts'
 import { useSession } from '@/lib/auth/auth-client'
 import { cn } from '@/lib/core/utils/cn'
 import {
@@ -17,6 +18,12 @@ import { useImportWorkflow } from '@/hooks/queries/workflows'
 
 const PAGE_SIZE = 20 as const
 
+const ENV_OPTIONS: { id: MothershipEnvironment; label: string }[] = [
+  { id: 'dev', label: 'Dev' },
+  { id: 'staging', label: 'Staging' },
+  { id: 'prod', label: 'Prod' },
+]
+
 export function Admin() {
   const params = useParams()
   const workspaceId = params?.workspaceId as string
@@ -63,6 +70,18 @@ export function Admin() {
     }
   }
 
+  const handleMothershipEnvironmentChange = useCallback(
+    async (nextEnvironment: MothershipEnvironment) => {
+      if (nextEnvironment !== settings?.mothershipEnvironment && !updateSetting.isPending) {
+        await updateSetting.mutateAsync({
+          key: 'mothershipEnvironment',
+          value: nextEnvironment,
+        })
+      }
+    },
+    [settings?.mothershipEnvironment, updateSetting]
+  )
+
   const handleImport = () => {
     if (!workflowId.trim()) return
     importWorkflow.mutate(
@@ -119,13 +138,45 @@ export function Admin() {
   ])
   return (
     <div className='flex h-full flex-col gap-6'>
-      <div className='flex items-center justify-between'>
-        <Label htmlFor='super-user-mode'>Super admin mode</Label>
-        <Switch
-          id='super-user-mode'
-          checked={settings?.superUserModeEnabled ?? false}
-          onCheckedChange={handleSuperUserModeToggle}
-        />
+      <div className='flex flex-col gap-4'>
+        <div className='flex items-center justify-between'>
+          <Label htmlFor='super-user-mode'>Super admin mode</Label>
+          <Switch
+            id='super-user-mode'
+            checked={settings?.superUserModeEnabled ?? false}
+            disabled={updateSetting.isPending}
+            onCheckedChange={handleSuperUserModeToggle}
+          />
+        </div>
+
+        {settings?.superUserModeEnabled && (
+          <div className='flex items-center justify-between gap-3'>
+            <div className='flex flex-col gap-1'>
+              <Label className='text-[var(--text-primary)] text-sm'>Mothership Environment</Label>
+              <p className='text-[var(--text-secondary)] text-xs'>
+                Controls which Copilot backend this admin session talks to.
+              </p>
+            </div>
+            <div className='flex gap-1'>
+              {ENV_OPTIONS.map((opt) => (
+                <button
+                  key={opt.id}
+                  type='button'
+                  onClick={() => handleMothershipEnvironmentChange(opt.id)}
+                  disabled={updateSetting.isPending}
+                  className={cn(
+                    'rounded-md px-3 py-1 font-medium text-sm transition-colors',
+                    (settings?.mothershipEnvironment ?? 'prod') === opt.id
+                      ? 'bg-[var(--surface-hover)] text-[var(--text-primary)]'
+                      : 'text-[var(--text-tertiary)] hover-hover:hover:text-[var(--text-secondary)]'
+                  )}
+                >
+                  {opt.label}
+                </button>
+              ))}
+            </div>
+          </div>
+        )}
       </div>
 
       <div className='h-px bg-[var(--border-secondary)]' />

apps/sim/app/workspace/[workspaceId]/settings/navigation.ts

@@ -218,12 +218,4 @@ export const allNavigationItems: NavigationItem[] = [
     section: 'superuser',
     requiresAdminRole: true,
   },
-  {
-    id: 'mothership',
-    label: 'Mothership',
-    icon: Server,
-    section: 'superuser',
-    requiresAdminRole: true,
-    requiresSuperUser: true,
-  },
 ]

packages/db/migrations/0205_funny_sleepwalker.sql

@@ -0,0 +1 @@
+ALTER TABLE "settings" ADD COLUMN "mothership_environment" text DEFAULT 'prod' NOT NULL;