From 96153f29bb2a598097c980ed398015c3a104f5d7 Mon Sep 17 00:00:00 2001
From: Will Chen <willchen90@gmail.com>
Date: Tue, 20 May 2025 21:51:34 -0700
Subject: [PATCH] Deprecate pro budget in app since it's not accurate (#210)

---
 src/lib/schemas.ts |  5 ++++-
 src/main.ts        | 11 +----------
 src/main/pro.ts    | 17 +----------------
 3 files changed, 6 insertions(+), 27 deletions(-)

diff --git a/src/lib/schemas.ts b/src/lib/schemas.ts
index 4fd0b10..f87368f 100644
--- a/src/lib/schemas.ts
+++ b/src/lib/schemas.ts
@@ -114,14 +114,17 @@ export const UserSettingsSchema = z.object({
   telemetryUserId: z.string().optional(),
   hasRunBefore: z.boolean().optional(),
   enableDyadPro: z.boolean().optional(),
-  dyadProBudget: DyadProBudgetSchema.optional(),
   experiments: ExperimentsSchema.optional(),
   lastShownReleaseNotesVersion: z.string().optional(),
   maxChatTurnsInContext: z.number().optional(),
   enableProSaverMode: z.boolean().optional(),
   enableProLazyEditsMode: z.boolean().optional(),
   enableProSmartFilesContextMode: z.boolean().optional(),
+
+  ////////////////////////////////
   // DEPRECATED.
+  ////////////////////////////////
+  dyadProBudget: DyadProBudgetSchema.optional(),
   runtimeMode: RuntimeModeSchema.optional(),
 });
 
diff --git a/src/main.ts b/src/main.ts
index e5da056..1e4f4c4 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -189,21 +189,12 @@ function handleDeepLinkReturn(url: string) {
   // dyad://dyad-pro-return?key=123&budget_reset_at=2025-05-26T16:31:13.492000Z&max_budget=100
   if (parsed.hostname === "dyad-pro-return") {
     const apiKey = parsed.searchParams.get("key");
-    // UTC time
-    // budget_reset_at: '2025-05-26T16:31:13.492000Z'
-    const budgetResetAt = parsed.searchParams.get("budget_reset_at");
-    const maxBudget = Number(parsed.searchParams.get("max_budget"));
     if (!apiKey) {
-      dialog.showErrorBox(
-        "Invalid URL",
-        "Expected key, budget_reset_at, and max_budget",
-      );
+      dialog.showErrorBox("Invalid URL", "Expected key");
       return;
     }
     handleDyadProReturn({
       apiKey,
-      budgetResetAt,
-      maxBudget,
     });
     // Send message to renderer to trigger re-render
     mainWindow?.webContents.send("deep-link-received", {
diff --git a/src/main/pro.ts b/src/main/pro.ts
index 1dcaca1..d67fef0 100644
--- a/src/main/pro.ts
+++ b/src/main/pro.ts
@@ -1,14 +1,6 @@
 import { readSettings, writeSettings } from "./settings";
 
-export function handleDyadProReturn({
-  apiKey,
-  budgetResetAt,
-  maxBudget,
-}: {
-  apiKey: string;
-  budgetResetAt: string | null | undefined;
-  maxBudget: number | null | undefined;
-}) {
+export function handleDyadProReturn({ apiKey }: { apiKey: string }) {
   const settings = readSettings();
   writeSettings({
     providerSettings: {
@@ -20,13 +12,6 @@ export function handleDyadProReturn({
         },
       },
     },
-    dyadProBudget:
-      budgetResetAt && maxBudget
-        ? {
-            budgetResetAt,
-            maxBudget,
-          }
-        : undefined,
     enableDyadPro: true,
   });
 }