chore: initialize recovered claude workspace
This commit is contained in:
78
src/utils/billing.ts
Normal file
78
src/utils/billing.ts
Normal file
@@ -0,0 +1,78 @@
|
||||
import {
|
||||
getAnthropicApiKey,
|
||||
getAuthTokenSource,
|
||||
getSubscriptionType,
|
||||
isClaudeAISubscriber,
|
||||
} from './auth.js'
|
||||
import { getGlobalConfig } from './config.js'
|
||||
import { isEnvTruthy } from './envUtils.js'
|
||||
|
||||
export function hasConsoleBillingAccess(): boolean {
|
||||
// Check if cost reporting is disabled via environment variable
|
||||
if (isEnvTruthy(process.env.DISABLE_COST_WARNINGS)) {
|
||||
return false
|
||||
}
|
||||
|
||||
const isSubscriber = isClaudeAISubscriber()
|
||||
|
||||
// This might be wrong if user is signed into Max but also using an API key, but
|
||||
// we already show a warning on launch in that case
|
||||
if (isSubscriber) return false
|
||||
|
||||
// Check if user has any form of authentication
|
||||
const authSource = getAuthTokenSource()
|
||||
const hasApiKey = getAnthropicApiKey() !== null
|
||||
|
||||
// If user has no authentication at all (logged out), don't show costs
|
||||
if (!authSource.hasToken && !hasApiKey) {
|
||||
return false
|
||||
}
|
||||
|
||||
const config = getGlobalConfig()
|
||||
const orgRole = config.oauthAccount?.organizationRole
|
||||
const workspaceRole = config.oauthAccount?.workspaceRole
|
||||
|
||||
if (!orgRole || !workspaceRole) {
|
||||
return false // hide cost for grandfathered users who have not re-authed since we've added roles
|
||||
}
|
||||
|
||||
// Users have billing access if they are admins or billing roles at either workspace or organization level
|
||||
return (
|
||||
['admin', 'billing'].includes(orgRole) ||
|
||||
['workspace_admin', 'workspace_billing'].includes(workspaceRole)
|
||||
)
|
||||
}
|
||||
|
||||
// Mock billing access for /mock-limits testing (set by mockRateLimits.ts)
|
||||
let mockBillingAccessOverride: boolean | null = null
|
||||
|
||||
export function setMockBillingAccessOverride(value: boolean | null): void {
|
||||
mockBillingAccessOverride = value
|
||||
}
|
||||
|
||||
export function hasClaudeAiBillingAccess(): boolean {
|
||||
// Check for mock billing access first (for /mock-limits testing)
|
||||
if (mockBillingAccessOverride !== null) {
|
||||
return mockBillingAccessOverride
|
||||
}
|
||||
|
||||
if (!isClaudeAISubscriber()) {
|
||||
return false
|
||||
}
|
||||
|
||||
const subscriptionType = getSubscriptionType()
|
||||
|
||||
// Consumer plans (Max/Pro) - individual users always have billing access
|
||||
if (subscriptionType === 'max' || subscriptionType === 'pro') {
|
||||
return true
|
||||
}
|
||||
|
||||
// Team/Enterprise - check for admin or billing roles
|
||||
const config = getGlobalConfig()
|
||||
const orgRole = config.oauthAccount?.organizationRole
|
||||
|
||||
return (
|
||||
!!orgRole &&
|
||||
['admin', 'billing', 'owner', 'primary_owner'].includes(orgRole)
|
||||
)
|
||||
}
|
||||
Reference in New Issue
Block a user