| 1 | import { AlertTriangle, CheckCircle2, ChevronRight, ReceiptText, X } from "lucide-react"; |
| 2 | import { Form, data } from "react-router"; |
| 3 | |
| 4 | import { COST_PROVIDERS, type CostProvider, type ProviderConnection, type ProviderCosts, isCostProvider } from "@g1t/contracts"; |
| 5 | |
| 6 | import type { Route } from "./+types/provider-costs"; |
| 7 | import { AiProviderMark } from "../../components/connectors"; |
| 8 | import { Unavailable } from "../../components/boundary"; |
| 9 | import { ErrorText, Field, Input, SubmitButton, TimeAgo } from "../../components/ui"; |
| 10 | import { Card } from "../../components/ui/card"; |
| 11 | import { page } from "../../lib/meta"; |
| 12 | import { money } from "../../lib/money"; |
| 13 | import { type Failed, isFailed, part } from "../../lib/partial"; |
| 14 | import { billing } from "../../lib/services.server"; |
| 15 | import { assertSameOrigin, getViewer, requireUser, roleIn, unwrap } from "../../lib/session.server"; |
| 16 | |
| 17 | export function meta({ params, ...args }: Route.MetaArgs) { |
| 18 | return page(args, { title: `Provider costs · ${params.owner} · g1t` }); |
| 19 | } |
| 20 | |
| 21 | /** How many days the page totals. */ |
| 22 | const RECENT_DAYS = 30; |
| 23 | |
| 24 | /** Which key each provider's report needs, and where it is made. */ |
| 25 | const KEYS: Record<CostProvider, { key: string; where: string; placeholder: string }> = { |
| 26 | anthropic: { |
| 27 | key: "an Admin API key", |
| 28 | where: "Make one in the Claude Console under Settings, Admin keys. Only an organization admin can.", |
| 29 | placeholder: "sk-ant-admin…", |
| 30 | }, |
| 31 | openai: { |
| 32 | key: "an admin key", |
| 33 | where: "Make one in the OpenAI platform under Organization settings, Admin keys. Only an organization owner can.", |
| 34 | placeholder: "sk-admin-…", |
| 35 | }, |
| 36 | }; |
| 37 | |
| 38 | /** |
| 39 | * Settings, Provider costs: the AI providers whose own cost reports g1t |
| 40 | * reads for the workspace (services/billing/src/provider_costs.rs). Every |
| 41 | * member can see which are connected, when they were last read and what |
| 42 | * went wrong; only owners connect and disconnect them. Keys are never |
| 43 | * shown again, only their last four characters. |
| 44 | */ |
| 45 | export async function loader({ params, context }: Route.LoaderArgs) { |
| 46 | const viewer = getViewer(context); |
| 47 | const role = roleIn(viewer, params.owner); |
| 48 | if (!role) throw data(null, { status: 404 }); |
| 49 | const slug = params.owner.toLowerCase(); |
| 50 | const since = new Date(Date.now() - (RECENT_DAYS - 1) * 86_400_000).toISOString().slice(0, 10); |
| 51 | // The connections are what the page is about; the totals beside them are a part. |
| 52 | const [connections, costs] = await Promise.all([ |
| 53 | billing.providerConnections(slug, viewer), |
| 54 | part(billing.providerCosts(slug, viewer, since), "What the providers reported"), |
| 55 | ]); |
| 56 | return { |
| 57 | slug, |
| 58 | owner: role === "owner", |
| 59 | connections: unwrap(connections), |
| 60 | totals: totalsOf(costs), |
| 61 | }; |
| 62 | } |
| 63 | |
| 64 | /** Each provider's total over the recent days, or the part's failure. */ |
| 65 | function totalsOf(costs: Awaited<ReturnType<typeof billing.providerCosts>> | Failed): Partial<Record<CostProvider, number>> | Failed { |
| 66 | if (isFailed(costs)) return costs; |
| 67 | if (!costs.ok) return { failed: true, what: "What the providers reported" }; |
| 68 | const totals: Partial<Record<CostProvider, number>> = {}; |
| 69 | for (const day of (costs.value as ProviderCosts).days) totals[day.provider] = (totals[day.provider] ?? 0) + day.micros; |
| 70 | return totals; |
| 71 | } |
| 72 | |
| 73 | export async function action({ params, context, request }: Route.ActionArgs) { |
| 74 | assertSameOrigin(request); |
| 75 | const user = requireUser(context, request); |
| 76 | const slug = params.owner.toLowerCase(); |
| 77 | const form = await request.formData(); |
| 78 | const provider = String(form.get("provider") ?? ""); |
| 79 | if (!isCostProvider(provider)) return { ok: false as const, provider: null, error: "Choose a provider." }; |
| 80 | try { |
| 81 | if (form.get("intent") === "disconnect") { |
| 82 | const removed = await billing.disconnectProvider(user, slug, provider); |
| 83 | return removed.ok ? { ok: true as const, provider, disconnected: true as const } : { ok: false as const, provider, error: removed.error.message }; |
| 84 | } |
| 85 | const credential = String(form.get("credential") ?? "").trim(); |
| 86 | if (!credential) return { ok: false as const, provider, error: `Paste ${COST_PROVIDERS[provider].label}'s admin key.` }; |
| 87 | const connected = await billing.connectProvider(user, slug, provider, credential); |
| 88 | return connected.ok ? { ok: true as const, provider, connected: connected.value } : { ok: false as const, provider, error: connected.error.message }; |
| 89 | } catch (error) { |
| 90 | console.error("provider costs: billing did not answer", error); |
| 91 | return { ok: false as const, provider, error: "Billing didn't answer. Try again in a moment." }; |
| 92 | } |
| 93 | } |
| 94 | |
| 95 | const SECTION = "border-t border-line pt-8 first:border-t-0 first:pt-0"; |
| 96 | |
| 97 | export default function ProviderCostsPage({ loaderData, actionData }: Route.ComponentProps) { |
| 98 | const { slug, owner, connections, totals } = loaderData; |
| 99 | const connected = actionData?.ok && "connected" in actionData ? actionData.connected : null; |
| 100 | const failure = actionData && !actionData.ok ? actionData : null; |
| 101 | const byProvider = new Map(connections.map((c) => [c.provider, c])); |
| 102 | const unconnected = (Object.keys(COST_PROVIDERS) as CostProvider[]).filter((p) => !byProvider.has(p)); |
| 103 | return ( |
| 104 | <div className="max-w-2xl"> |
| 105 | <header className="mb-8 border-b border-line pb-6"> |
| 106 | <h1 className="flex items-center gap-2 text-2xl font-semibold tracking-tight"> |
| 107 | <span className="text-muted"> |
| 108 | <ReceiptText size={18} /> |
| 109 | </span> |
| 110 | Provider costs |
| 111 | </h1> |
| 112 | <p className="mt-1.5 text-sm text-muted"> |
| 113 | What your AI providers say {slug}'s organization spent, day by day, read from each provider's own cost report. It counts every use of that |
| 114 | organization, in g1t or not: CI scripts, other tools, people's own sessions. g1t reads it once a day. |
| 115 | </p> |
| 116 | </header> |
| 117 | |
| 118 | {!owner && ( |
| 119 | <Card asChild radius="lg" className="mb-8 px-4 py-3 text-sm text-muted"> |
| 120 | <p>Only workspace owners can connect or disconnect a provider. They are shown so you know where these figures come from.</p> |
| 121 | </Card> |
| 122 | )} |
| 123 | |
| 124 | {connected && <Connected connection={connected} />} |
| 125 | {failure && !failure.provider && <ErrorText>{failure.error}</ErrorText>} |
| 126 | |
| 127 | <div className="space-y-8"> |
| 128 | <section className={SECTION} aria-labelledby="connected-heading"> |
| 129 | <h2 id="connected-heading" className="font-medium"> |
| 130 | Connected providers |
| 131 | </h2> |
| 132 | {connections.length === 0 ? ( |
| 133 | <p className="mt-3 text-sm text-muted">{owner ? "None yet. Connect one below." : `None yet. An owner of ${slug} can connect one.`}</p> |
| 134 | ) : ( |
| 135 | <Card asChild divided className="mt-4 overflow-hidden"> |
| 136 | <ul> |
| 137 | {connections.map((connection) => ( |
| 138 | <li key={connection.provider} className="p-4"> |
| 139 | <ConnectionRow |
| 140 | connection={connection} |
| 141 | owner={owner} |
| 142 | total={isFailed(totals) ? totals : (totals[connection.provider] ?? 0)} |
| 143 | error={failure?.provider === connection.provider ? failure.error : null} |
| 144 | /> |
| 145 | </li> |
| 146 | ))} |
| 147 | </ul> |
| 148 | </Card> |
| 149 | )} |
| 150 | <p className="mt-3 text-xs text-faint"> |
| 151 | Keys are sealed when saved and never shown again. Disconnecting deletes the key and every cost read with it. |
| 152 | </p> |
| 153 | </section> |
| 154 | |
| 155 | {owner && unconnected.length > 0 && ( |
| 156 | <section className={SECTION} aria-labelledby="connect-heading"> |
| 157 | <h2 id="connect-heading" className="font-medium"> |
| 158 | Connect a provider |
| 159 | </h2> |
| 160 | <p className="mt-1.5 text-xs text-faint"> |
| 161 | g1t checks the key with one small report request, then reads the last 90 days. It reads costs only: the key is never used to run models. |
| 162 | </p> |
| 163 | <div className="mt-4 space-y-4"> |
| 164 | {unconnected.map((provider) => ( |
| 165 | <Card key={provider} asChild className="p-4"> |
| 166 | <div> |
| 167 | <div className="flex items-center gap-3"> |
| 168 | <AiProviderMark provider={provider} size={28} /> |
| 169 | <div className="min-w-0"> |
| 170 | <h3 className="text-sm font-medium">{COST_PROVIDERS[provider].label}</h3> |
| 171 | <p className="text-xs text-muted">{KEYS[provider].where}</p> |
| 172 | </div> |
| 173 | </div> |
| 174 | <KeyForm provider={provider} error={failure?.provider === provider ? failure.error : null} label="Connect" /> |
| 175 | </div> |
| 176 | </Card> |
| 177 | ))} |
| 178 | </div> |
| 179 | </section> |
| 180 | )} |
| 181 | </div> |
| 182 | </div> |
| 183 | ); |
| 184 | } |
| 185 | |
| 186 | function ConnectionRow({ connection, owner, total, error }: { connection: ProviderConnection; owner: boolean; total: number | Failed; error: string | null }) { |
| 187 | const { provider } = connection; |
| 188 | return ( |
| 189 | <div> |
| 190 | <div className="flex items-center gap-3"> |
| 191 | <AiProviderMark provider={provider} /> |
| 192 | <div className="min-w-0 grow"> |
| 193 | <p className="truncate text-sm font-medium">{COST_PROVIDERS[provider].label}</p> |
| 194 | <p className="truncate text-xs text-muted"> |
| 195 | Key {connection.hint} · connected by {connection.connectedBy} <TimeAgo at={connection.connectedAt} /> |
| 196 | </p> |
| 197 | </div> |
| 198 | {owner && ( |
| 199 | <Form method="post" className="shrink-0"> |
| 200 | <input type="hidden" name="provider" value={provider} /> |
| 201 | <SubmitButton |
| 202 | variant="outline" |
| 203 | name="intent" |
| 204 | value="disconnect" |
| 205 | match={{ provider }} |
| 206 | pending="Disconnecting…" |
| 207 | aria-label={`Disconnect ${COST_PROVIDERS[provider].label}`} |
| 208 | > |
| 209 | <X size={14} /> |
| 210 | </SubmitButton> |
| 211 | </Form> |
| 212 | )} |
| 213 | </div> |
| 214 | <div className="mt-3 flex flex-wrap items-center gap-x-4 gap-y-1 text-xs text-muted"> |
| 215 | <span> |
| 216 | {connection.lastSyncedAt ? ( |
| 217 | <> |
| 218 | Last read <TimeAgo at={connection.lastSyncedAt} /> |
| 219 | </> |
| 220 | ) : ( |
| 221 | "Not read yet" |
| 222 | )} |
| 223 | {connection.syncedFrom && <>, back to {connection.syncedFrom}</>} |
| 224 | </span> |
| 225 | {isFailed(total) ? ( |
| 226 | <Unavailable what={total.what} compact /> |
| 227 | ) : ( |
| 228 | <span> |
| 229 | {money(total)} in the last {RECENT_DAYS} days |
| 230 | </span> |
| 231 | )} |
| 232 | </div> |
| 233 | {connection.lastError && ( |
| 234 | <p className="mt-3 flex items-start gap-2 text-xs text-warn"> |
| 235 | <AlertTriangle size={13} className="mt-0.5 shrink-0" /> |
| 236 | {connection.lastError} |
| 237 | </p> |
| 238 | )} |
| 239 | <ErrorText>{error}</ErrorText> |
| 240 | {owner && ( |
| 241 | <details className="group mt-3"> |
| 242 | <summary className="inline-flex cursor-pointer list-none items-center gap-1 text-xs text-muted hover:text-fg"> |
| 243 | <ChevronRight size={12} className="transition-transform group-open:rotate-90" /> |
| 244 | Replace the key |
| 245 | </summary> |
| 246 | <p className="mt-2 text-xs text-faint">The costs read with the old key are deleted and the last 90 days are read again with the new one.</p> |
| 247 | <KeyForm provider={provider} error={null} label="Replace" /> |
| 248 | </details> |
| 249 | )} |
| 250 | </div> |
| 251 | ); |
| 252 | } |
| 253 | |
| 254 | function KeyForm({ provider, error, label }: { provider: CostProvider; error: string | null; label: string }) { |
| 255 | return ( |
| 256 | <Form method="post" className="mt-4 space-y-3"> |
| 257 | <input type="hidden" name="provider" value={provider} /> |
| 258 | <Field label={`${COST_PROVIDERS[provider].label} admin key`} hint={`Write-only: g1t never shows it again. Use ${KEYS[provider].key}.`}> |
| 259 | <Input |
| 260 | name="credential" |
| 261 | type="password" |
| 262 | required |
| 263 | placeholder={KEYS[provider].placeholder} |
| 264 | autoComplete="new-password" |
| 265 | spellCheck={false} |
| 266 | className="max-w-sm font-mono text-[0.8125rem]" |
| 267 | /> |
| 268 | </Field> |
| 269 | <ErrorText>{error}</ErrorText> |
| 270 | <SubmitButton variant="outline" name="intent" value="connect" match={{ provider }} pending="Checking the key…"> |
| 271 | {label} |
| 272 | </SubmitButton> |
| 273 | </Form> |
| 274 | ); |
| 275 | } |
| 276 | |
| 277 | /** What happened, right after connecting: shown once. */ |
| 278 | function Connected({ connection }: { connection: ProviderConnection }) { |
| 279 | const label = COST_PROVIDERS[connection.provider].label; |
| 280 | return ( |
| 281 | <div className="mb-8 rounded-xl border border-success-dim/60 bg-success/5 p-5"> |
| 282 | <p className="flex items-center gap-2 font-medium"> |
| 283 | <CheckCircle2 size={16} className="text-success" /> |
| 284 | {label} is connected. |
| 285 | </p> |
| 286 | <p className="mt-2 text-sm text-muted"> |
| 287 | {connection.lastError |
| 288 | ? `The key works and is saved, but the first read failed: ${connection.lastError}` |
| 289 | : connection.syncedFrom |
| 290 | ? `Its report is read back to ${connection.syncedFrom}, and again every day.` |
| 291 | : "The key works and is saved. Its report is read once a day."} |
| 292 | </p> |
| 293 | </div> |
| 294 | ); |
| 295 | } |