Add toggle for disabling Dyad Pro (#24)

src/app/TitleBar.tsx

@@ -7,7 +7,7 @@ import { Button } from "@/components/ui/button";
 // @ts-ignore
 import logo from "../../assets/logo_transparent.png";
 import { providerSettingsRoute } from "@/routes/settings/providers/$provider";
-
+import { cn } from "@/lib/utils";
 export const TitleBar = () => {
   const [selectedAppId] = useAtom(selectedAppIdAtom);
   const { apps } = useLoadApps();
@@ -27,6 +27,7 @@ export const TitleBar = () => {
   };
 
   const isDyadPro = !!settings?.providerSettings?.auto?.apiKey?.value;
+  const isDyadProEnabled = settings?.enableDyadPro;
 
   return (
     <div className="@container z-11 w-full h-11 bg-(--sidebar) absolute top-0 left-0 app-region-drag flex items-center">
@@ -51,10 +52,13 @@ export const TitleBar = () => {
             });
           }}
           variant="outline"
-          className="ml-4 no-app-region-drag h-7 bg-indigo-600 text-white dark:bg-indigo-600 dark:text-white"
+          className={cn(
+            "ml-4 no-app-region-drag h-7 bg-indigo-600 text-white dark:bg-indigo-600 dark:text-white",
+            !isDyadProEnabled && "bg-zinc-600 dark:bg-zinc-600"
+          )}
           size="sm"
         >
-          Dyad Pro
+          {isDyadProEnabled ? "Dyad Pro" : "Dyad Pro (disabled)"}
         </Button>
       )}
     </div>

src/components/settings/ProviderSettingsPage.tsx

@@ -23,6 +23,9 @@ import {
 import { Input } from "@/components/ui/input";
 import { Button } from "@/components/ui/button";
 import { IpcClient } from "@/ipc/ipc_client";
+import { Switch } from "@/components/ui/switch";
+import { showError } from "@/lib/toast";
+import { UserSettings } from "@/lib/schemas";
 
 interface ProviderSettingsPageProps {
   provider: string;
@@ -44,6 +47,8 @@ export function ProviderSettingsPage({ provider }: ProviderSettingsPageProps) {
     updateSettings,
   } = useSettings();
 
+  const isDyad = provider === "auto";
+
   const [apiKeyInput, setApiKeyInput] = useState("");
   const [isSaving, setIsSaving] = useState(false);
   const [saveError, setSaveError] = useState<string | null>(null);
@@ -95,7 +100,7 @@ export function ProviderSettingsPage({ provider }: ProviderSettingsPageProps) {
     setIsSaving(true);
     setSaveError(null);
     try {
-      await updateSettings({
+      const settingsUpdate: Partial<UserSettings> = {
         providerSettings: {
           ...settings?.providerSettings,
           [provider]: {
@@ -105,7 +110,11 @@ export function ProviderSettingsPage({ provider }: ProviderSettingsPageProps) {
             },
           },
         },
-      });
+      };
+      if (isDyad) {
+        settingsUpdate.enableDyadPro = true;
+      }
+      await updateSettings(settingsUpdate);
       setApiKeyInput(""); // Clear input on success
       // Optionally show a success message
     } catch (error: any) {
@@ -139,6 +148,20 @@ export function ProviderSettingsPage({ provider }: ProviderSettingsPageProps) {
     }
   };
 
+  // --- Toggle Dyad Pro Handler ---
+  const handleToggleDyadPro = async (enabled: boolean) => {
+    setIsSaving(true);
+    try {
+      await updateSettings({
+        enableDyadPro: enabled,
+      });
+    } catch (error: any) {
+      showError(`Error toggling Dyad Pro: ${error}`);
+    } finally {
+      setIsSaving(false);
+    }
+  };
+
   // Effect to clear input error when input changes
   useEffect(() => {
     if (saveError) {
@@ -205,7 +228,7 @@ export function ProviderSettingsPage({ provider }: ProviderSettingsPageProps) {
             ) : (
               <KeyRound className="mr-2 h-4 w-4" />
             )}
-            {isConfigured ? "Manage API Keys" : "Setup API Key"}
+            {getKeyButtonText({ isConfigured, isDyad })}
             <ExternalLink className="ml-2 h-4 w-4" />
           </Button>
         )}
@@ -299,6 +322,7 @@ export function ProviderSettingsPage({ provider }: ProviderSettingsPageProps) {
               </AccordionContent>
             </AccordionItem>
 
+            {!isDyad && (
               <AccordionItem
                 value="env-key"
                 className="border rounded-lg px-4 bg-(--background-lightest)"
@@ -324,8 +348,8 @@ export function ProviderSettingsPage({ provider }: ProviderSettingsPageProps) {
                         )}
                         {activeKeySource === "settings" && (
                           <p className="text-xs text-yellow-600 dark:text-yellow-400 mt-1">
-                          This key is currently being overridden by the key set
-                          in Settings.
+                            This key is currently being overridden by the key
+                            set in Settings.
                           </p>
                         )}
                       </AlertDescription>
@@ -350,9 +374,41 @@ export function ProviderSettingsPage({ provider }: ProviderSettingsPageProps) {
                   </p>
                 </AccordionContent>
               </AccordionItem>
+            )}
           </Accordion>
         )}
+
+        {isDyad && !settingsLoading && (
+          <div className="mt-6 flex items-center justify-between p-4 bg-(--background-lightest) rounded-lg border">
+            <div>
+              <h3 className="font-medium">Enable Dyad Pro</h3>
+              <p className="text-sm text-gray-600 dark:text-gray-400">
+                Toggle to enable Dyad Pro
+              </p>
+            </div>
+            <Switch
+              checked={settings?.enableDyadPro}
+              onCheckedChange={handleToggleDyadPro}
+              disabled={isSaving}
+            />
+          </div>
+        )}
       </div>
     </div>
   );
 }
+
+function getKeyButtonText({
+  isConfigured,
+  isDyad,
+}: {
+  isConfigured: boolean;
+  isDyad: boolean;
+}) {
+  if (isDyad) {
+    return isConfigured
+      ? "Manage Dyad Pro Subscription"
+      : "Setup Dyad Pro Subscription";
+  }
+  return isConfigured ? "Manage API Keys" : "Setup API Key";
+}

src/ipc/utils/get_model_client.ts

@@ -46,7 +46,7 @@ export function getModelClient(
   }
 
   const dyadApiKey = settings.providerSettings?.auto?.apiKey?.value;
-  if (dyadApiKey) {
+  if (dyadApiKey && settings.enableDyadPro) {
     const provider = createOpenAI({
       apiKey: dyadApiKey,
       baseURL: "https://llm-gateway.dyad.sh/v1",

src/lib/schemas.ts

@@ -107,6 +107,7 @@ export const UserSettingsSchema = z.object({
   telemetryConsent: z.enum(["opted_in", "opted_out", "unset"]).optional(),
   telemetryUserId: z.string().optional(),
   hasRunBefore: z.boolean().optional(),
+  enableDyadPro: z.boolean().optional(),
 
   experiments: ExperimentsSchema.optional(),
   // DEPRECATED.