Pro Trial models (#2387)

src/app/TitleBar.tsx

@@ -216,7 +216,11 @@ export function DyadProButton({
       )}
       size="sm"
     >
-      {isDyadProEnabled ? "Pro" : "Pro (off)"}
+      {isDyadProEnabled
+        ? userBudget?.isTrial
+          ? "Pro Trial"
+          : "Pro"
+        : "Pro (off)"}
       {userBudget && isDyadProEnabled && (
         <AICreditStatus userBudget={userBudget} />
       )}

src/hooks/useTrialModelRestriction.ts

+import { useEffect } from "react";
+import { useUserBudgetInfo } from "./useUserBudgetInfo";
+import { useSettings } from "./useSettings";
+import { isDyadProEnabled } from "../lib/schemas";
+
+const AUTO_MODEL = { name: "auto", provider: "auto" };
+
+export function useTrialModelRestriction() {
+  const { userBudget, isLoadingUserBudget } = useUserBudgetInfo();
+  const { settings, updateSettings } = useSettings();
+
+  const isTrial =
+    (userBudget?.isTrial && settings && isDyadProEnabled(settings)) ?? false;
+  const isOnAutoModel =
+    settings?.selectedModel?.provider === "auto" &&
+    settings?.selectedModel?.name === "auto";
+
+  // Auto-switch to auto model if user is on trial and not already on auto
+  useEffect(() => {
+    if (isTrial && settings && !isOnAutoModel && !isLoadingUserBudget) {
+      updateSettings({ selectedModel: AUTO_MODEL });
+    }
+  }, [isTrial, isOnAutoModel, isLoadingUserBudget, settings, updateSettings]);
+
+  return {
+    isTrial,
+    isLoadingTrialStatus: isLoadingUserBudget,
+  };
+}

src/ipc/handlers/pro_handlers.ts

@@ -11,6 +11,7 @@ export const UserInfoResponseSchema = z.object({
   totalCredits: z.number(),
   budgetResetDate: z.string(), // ISO date string from API
   userId: z.string(),
+  isTrial: z.boolean().optional().default(false),
 });
 export type UserInfoResponse = z.infer<typeof UserInfoResponseSchema>;
 
@@ -30,6 +31,7 @@ export function registerProHandlers() {
         totalCredits: 1000,
         budgetResetDate: resetDate,
         redactedUserId: "<redacted-user-id-testing>",
+        isTrial: false,
       };
     }
     logger.info("Attempting to fetch user budget information.");
@@ -83,6 +85,7 @@ export function registerProHandlers() {
         totalCredits: data.totalCredits,
         budgetResetDate: new Date(data.budgetResetDate),
         redactedUserId: redactedUserId,
+        isTrial: data.isTrial,
       });
 
       return userBudgetInfo;

src/ipc/types/system.ts

@@ -73,6 +73,7 @@ export const UserBudgetInfoSchema = z
     totalCredits: z.number(),
     budgetResetDate: z.date(),
     redactedUserId: z.string(),
+    isTrial: z.boolean(),
   })
   .nullable();