fix: reset workspaces when re-enabling account

Author: ndycode

lib/accounts.ts

@@ -765,7 +765,11 @@ export class AccountManager {
 		if (index < 0 || index >= this.accounts.length) return null;
 		const account = this.accounts[index];
 		if (!account) return null;
+		const wasEnabled = account.enabled !== false;
 		account.enabled = enabled;
+		if (enabled && !wasEnabled) {
+			this.resetWorkspaces(account);
+		}
 		return account;
 	}
 
@@ -850,6 +854,22 @@ export class AccountManager {
 		}
 	}
 
+	private resetWorkspaces(account: ManagedAccount): void {
+		if (!account.workspaces || account.workspaces.length === 0) {
+			return;
+		}
+
+		for (const workspace of account.workspaces) {
+			workspace.enabled = true;
+			delete workspace.disabledAt;
+		}
+
+		const currentIdx = account.currentWorkspaceIndex ?? 0;
+		if (currentIdx < 0 || currentIdx >= account.workspaces.length) {
+			account.currentWorkspaceIndex = 0;
+		}
+	}
+
 	getCurrentWorkspace(account: ManagedAccount): Workspace | null {
 		if (!account.workspaces || account.workspaces.length === 0) {
 			return null;

test/accounts.test.ts

@@ -256,6 +256,43 @@ describe("AccountManager", () => {
     expect(account.workspaces?.map((workspace) => workspace.enabled)).toEqual([false, true]);
   });
 
+  it("re-enabling an exhausted account restores its workspaces", () => {
+    const now = Date.now();
+    const stored = {
+      version: 3 as const,
+      activeIndex: 0,
+      accounts: [
+        {
+          refreshToken: "token-1",
+          addedAt: now,
+          lastUsed: now,
+          enabled: false,
+          workspaces: [
+            { id: "workspace-1", name: "Workspace 1", enabled: false, disabledAt: now - 2_000 },
+            { id: "workspace-2", name: "Workspace 2", enabled: false, disabledAt: now - 1_000 },
+          ],
+          currentWorkspaceIndex: 1,
+        },
+      ],
+    };
+
+    const manager = new AccountManager(undefined, stored);
+    const account = manager.setAccountEnabled(0, true);
+    expect(account).not.toBeNull();
+    if (!account) {
+      throw new Error("account should exist");
+    }
+
+    expect(account.enabled).toBe(true);
+    expect(manager.hasEnabledWorkspaces(account)).toBe(true);
+    expect(manager.getEnabledWorkspaceCount(account)).toBe(2);
+    expect(manager.getCurrentWorkspace(account)?.id).toBe("workspace-2");
+    expect(account.workspaces).toEqual([
+      { id: "workspace-1", name: "Workspace 1", enabled: true },
+      { id: "workspace-2", name: "Workspace 2", enabled: true },
+    ]);
+  });
+
   it("checks account availability by enabled/rate-limit/cooldown state", () => {
     const now = Date.now();
     const stored = {