forked from sanbuphy/learn-coding-agent
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbilling.ts
More file actions
78 lines (62 loc) · 2.36 KB
/
billing.ts
File metadata and controls
78 lines (62 loc) · 2.36 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
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)
)
}