1import fs from 'fs'
2import { getPlainNoMappedSec, type secretNameType } from '../../serverF/secretsF/getPlainSecAI.mts'
3import { assertDefined } from '../../sharedF/assertsF/assertDefinedF.mts'
4import { assertTruthy } from '../../sharedF/assertsF/assertTruthyAI.mts'
5import { throwIf } from '../../sharedF/throwErrorF/throwIfF.mts'
6import { pathDownJoin } from '../../serverF/pathF.mts'
7import { secretsDir } from '../../serverF/secretsDirF.mts'
8import { betLog } from '../../sharedF/betterConsLogF.mts'
9import { getReqAppCfg } from '../k8sF/ctxF/appCfgCtxF.mts'
10
11const opnRtrProvisionerKey = () => getPlainNoMappedSec('openrouter-provisioner-apikey' as secretNameType)
12
13export const opnRtrProvisionerFetch = async (urlPath: string, {method = 'GET', bodyH, apiKey}: {method?: string, bodyH?: Record<string, unknown>, apiKey?: string} = {}) => {
14  const token = apiKey || opnRtrProvisionerKey()
15  const resp = await fetch('https://openrouter.ai' + urlPath, {
16    method,
17    headers: {'Authorization': `Bearer ${token}`, 'Content-Type': 'application/json'},
18    body: bodyH ? JSON.stringify(bodyH) : undefined,
19  })
20  return resp
21}
22
23const listOpnRtrKeys = async () => {
24  const resp = await opnRtrProvisionerFetch('/api/v1/keys')
25  assertTruthy(resp.ok, {status: resp.status})
26  const {data} = await resp.json()
27  return data
28}
29
30export const opnRtrSecretName = (name: string) => `openrouter-${name}`
31
32export const ensureOpnRtrKey = async ({name, limit, limitReset}: {name: string, limit?: number, limitReset?: string}) => {
33  assertDefined(name, {name})
34  const secName = opnRtrSecretName(name)
35  const secPath = pathDownJoin(secretsDir, secName)
36  if (fs.existsSync(secPath)) return secName
37
38  betLog({provisioningOpnRtrKey: name, limit, limitReset})
39  const existingKeys = await listOpnRtrKeys()
40  const existingKeyOnServer = existingKeys.find((k: {name: string}) => k.name === name)
41  throwIf(() => existingKeyOnServer, {name, secName, existingKeyOnServer})
42
43  const resp = await opnRtrProvisionerFetch('/api/v1/keys', {
44    method: 'POST',
45    bodyH: {name, limit, limit_reset: limitReset},
46  })
47  const respStatus = resp.status, respBody = !resp.ok ? await resp.text().catch(() => '') : undefined
48  assertTruthy(resp.ok, {respStatus, respBody})
49  const {key} = await resp.json()
50  assertDefined(key, {name})
51  fs.writeFileSync(secPath, key)
52  betLog({provisionedOpnRtrKey: name, secretFile: secName})
53  return secName
54}
55
56export const opnRtrKeyUsageFromSecret = async ({secretName}: {secretName: string}) => {
57  const secPath = pathDownJoin(secretsDir, secretName)
58  const apiKey = fs.readFileSync(secPath, 'utf-8').trim()
59  const resp = await opnRtrProvisionerFetch('/api/v1/key', {apiKey})
60  const usageRespStatus = resp.status
61  assertTruthy(resp.ok, {usageRespStatus})
62  const {data} = await resp.json()
63  return data
64}
65
66export const openrouterusage = async () => {
67  const {name} = getReqAppCfg()
68  assertDefined(name, {name})
69  const secretName = opnRtrSecretName(name)
70  const usage = await opnRtrKeyUsageFromSecret({secretName})
71  console.log(`OpenRouter key: ${usage.name || name}`)
72  console.log(`  limit:     $${usage.limit ?? 'none'}  (resets: ${usage.limit_reset || 'never'})`)
73  console.log(`  remaining: $${usage.limit_remaining ?? 'n/a'}`)
74  console.log(`  daily:     $${usage.usage_daily}`)
75  console.log(`  weekly:    $${usage.usage_weekly}`)
76  console.log(`  monthly:   $${usage.usage_monthly}`)
77  console.log(`  total:     $${usage.usage}`)
78}
79openrouterusage.cliDescript = 'show OpenRouter API key usage and budget'
80