refactor(workspace): defer per-row balances to keep switcher parity with cloud

Author: comfydesigner

src/locales/en/main.json

@@ -3337,7 +3337,6 @@
     "nameValidationError": "Name must be 1–50 characters using letters, numbers, spaces, or common punctuation."
   },
   "workspaceSwitcher": {
-    "noCredits": "No credits",
     "scopeCaption": "Workspaces only affect which credits you use.",
     "scopeTooltip": "Runs that use partner nodes spend credits from this workspace. Unlike on Cloud, every workspace saves to your usual output folder.",
     "switchWorkspace": "Switch workspace",

src/platform/workspace/api/workspaceApi.ts

@@ -446,24 +446,6 @@ export const workspaceApi = {
     }
   },
 
-  /**
-   * Get credit balance for a specific workspace via an explicitly minted
-   * token, without switching the active workspace context.
-   */
-  async getBillingBalanceWithToken(
-    accessToken: string
-  ): Promise<BillingBalanceResponse> {
-    try {
-      const response = await workspaceApiClient.get<BillingBalanceResponse>(
-        workspaceApiUrl('/billing/balance'),
-        { headers: { Authorization: `Bearer ${accessToken}` } }
-      )
-      return response.data
-    } catch (err) {
-      handleAxiosError(err)
-    }
-  },
-
   /**
    * Get available subscription plans
    * GET /api/billing/plans

src/platform/workspace/components/WorkspaceSwitcherPopover.test.ts

@@ -23,20 +23,6 @@ const distributionMocks = vi.hoisted(() => ({ isCloud: true }))
 
 vi.mock('@/platform/distribution/types', () => distributionMocks)
 
-const balanceMocks = vi.hoisted(() => ({
-  balanceByWorkspaceId: {} as Record<
-    string,
-    | { status: 'loading' }
-    | { status: 'ready'; cents: number }
-    | { status: 'error' }
-  >,
-  loadBalances: vi.fn()
-}))
-
-vi.mock('@/platform/workspace/composables/useWorkspaceBalances', () => ({
-  useWorkspaceBalances: () => balanceMocks
-}))
-
 const LONG_WORKSPACE_NAME =
   'Quantum Renaissance Collective for Hyperdimensional Latent Diffusion Research and Experimental Workflow Engineering'
 
@@ -49,7 +35,6 @@ const i18n = createI18n({
         personal: 'Personal',
         roleOwner: 'Owner',
         roleMember: 'Member',
-        noCredits: 'No credits',
         scopeCaption: 'Workspaces only affect which credits you use.',
         scopeTooltip:
           'Runs that use partner nodes spend credits from this workspace. Unlike on Cloud, every workspace saves to your usual output folder.',
@@ -121,30 +106,6 @@ function renderComponent(
 }
 
 describe('WorkspaceSwitcherPopover', () => {
-  it('shows per-workspace balances on non-active rows and asks for them on mount', () => {
-    balanceMocks.balanceByWorkspaceId['ws-team-long'] = {
-      status: 'ready',
-      cents: 1000
-    }
-    renderComponent()
-
-    // 1000 cents at the 2.11 credits-per-cent rate
-    expect(screen.getByText('2,110')).toBeInTheDocument()
-    expect(balanceMocks.loadBalances).toHaveBeenCalledWith(
-      expect.arrayContaining(['ws-team-long'])
-    )
-  })
-
-  it('labels an empty workspace "No credits" instead of a number', () => {
-    balanceMocks.balanceByWorkspaceId['ws-team-long'] = {
-      status: 'ready',
-      cents: 0
-    }
-    renderComponent()
-
-    expect(screen.getByText('No credits')).toBeInTheDocument()
-  })
-
   it('shows the credits-scope caption off cloud', () => {
     distributionMocks.isCloud = false
     renderComponent()

src/platform/workspace/components/WorkspaceSwitcherPopover.vue

@@ -64,12 +64,6 @@
                   v-if="isCurrentWorkspace(workspace)"
                   class="pi pi-check shrink-0 text-sm text-base-foreground"
                 />
-                <span
-                  v-else-if="rowBalanceLabel(workspace)"
-                  class="shrink-0 text-sm text-muted-foreground tabular-nums"
-                >
-                  {{ rowBalanceLabel(workspace) }}
-                </span>
               </button>
             </div>
           </div>
@@ -129,14 +123,12 @@
 
 <script setup lang="ts">
 import { storeToRefs } from 'pinia'
-import { computed, onMounted, watch } from 'vue'
+import { computed } from 'vue'
 import { useI18n } from 'vue-i18n'
 
 import WorkspaceProfilePic from '@/platform/workspace/components/WorkspaceProfilePic.vue'
 import { useBillingContext } from '@/composables/billing/useBillingContext'
 import { isCloud } from '@/platform/distribution/types'
-import { formatCreditsFromCents } from '@/base/credits/comfyCredits'
-import { useWorkspaceBalances } from '@/platform/workspace/composables/useWorkspaceBalances'
 import { useWorkspaceSwitch } from '@/platform/workspace/composables/useWorkspaceSwitch'
 import { useWorkspaceTierLabel } from '@/platform/workspace/composables/useWorkspaceTierLabel'
 import type {
@@ -225,29 +217,4 @@ async function handleSelectWorkspace(workspace: AvailableWorkspace) {
 function handleCreateWorkspace() {
   emit('create')
 }
-
-const { locale } = useI18n()
-const { balanceByWorkspaceId, loadBalances } = useWorkspaceBalances()
-
-const peekableWorkspaceIds = computed(() =>
-  availableWorkspaces.value
-    .filter((w) => !isCurrentWorkspace(w))
-    .map((w) => w.id)
-)
-
-onMounted(() => loadBalances(peekableWorkspaceIds.value))
-watch(peekableWorkspaceIds, (ids) => loadBalances(ids))
-
-function rowBalanceLabel(workspace: AvailableWorkspace): string {
-  const state = balanceByWorkspaceId[workspace.id]
-  if (!state || state.status === 'loading' || state.status === 'error') {
-    return ''
-  }
-  if (state.cents <= 0) return t('workspaceSwitcher.noCredits')
-  return formatCreditsFromCents({
-    cents: state.cents,
-    locale: locale.value,
-    numberOptions: { minimumFractionDigits: 0, maximumFractionDigits: 0 }
-  })
-}
 </script>

src/platform/workspace/composables/useWorkspaceBalances.test.ts

@@ -555,22 +555,6 @@ export const useWorkspaceAuthStore = defineStore('workspaceAuth', () => {
    * callers onto a single in-flight mint, backs off after failure, and returns
    * null so callers fail closed rather than downgrade to the personal identity.
    */
-  /**
-   * Read-only token mint for a workspace the user is not switched into, used
-   * to peek at per-workspace data (e.g. switcher balances). Never touches the
-   * active workspace context or the stored token state.
-   */
-  async function peekWorkspaceToken(
-    workspaceId: string
-  ): Promise<string | null> {
-    try {
-      const minted = await requestToken(workspaceId)
-      return minted.token
-    } catch {
-      return null
-    }
-  }
-
   async function ensureWorkspaceToken(
     preferredWorkspaceId?: string
   ): Promise<string | null> {
@@ -1030,7 +1014,6 @@ export const useWorkspaceAuthStore = defineStore('workspaceAuth', () => {
     getWorkspaceAuthHeader,
     ensureWorkspaceAuthHeader,
     ensureWorkspaceToken,
-    peekWorkspaceToken,
     getWorkspaceToken,
     getUnifiedToken,
     clearWorkspaceContext