antcolony

5.3 KB

1. // quota.hl — the Claude usage of this host's Claude account (mission 020; concept §3 "Claude quota per account:
2. // reserve threshold, e.g. no new starts above 70 % of the 5 h window").
3. //
4. // HOW (tested 2026-09-24, Claude Code 2.1.280 on Loreana, .scratch/m020/usage-probe/): `claude -p /usage` is a LOCAL
5. // command — no model call: result `total_cost_usd: 0`, `num_turns: 0`, no tokens, ~1–4 s. With
6. // `--output-format json --verbose` the answer is a message list whose assistant message carries
7. // usage_report.rate_limits.limits = [{ kind: 'session', percent: 5, resets_at: '2026-09-24T18:19:59.77+00:00' },
8. // { kind: 'weekly_all', percent: 41, … }, { kind: 'weekly_scoped', … }]
9. // and the text "Current session: 5% used · resets …" (the fallback when the structured part is missing).
10. // The probe runs with `--permission-mode plan --permission-prompts none --strict-mcp-config --model haiku
11. // --max-turns 1` so that, if `/usage` ever stopped being local, it would still be one cheap refused turn;
12. // a probe that cost anything is reported (`cost`).
13. import { readJson, isList, isObject, isString } from './util.hl'
14. import { runClaude } from './claude.hl'
16. static digit = (c) => { return c >= 48 && c <= 57 }
18. // integer division for non-negative numbers (no Math)
19. static idiv = (a, b) => { return (a - a % b) / b }
21. // days since 1970-01-01 of a civil date (H. Hinnant's days_from_civil)
22. static daysFromCivil = (y0, m, d) => {
23. let y = m <= 2 ? y0 - 1 : y0
24. let era = idiv(y, 400)
25. let yoe = y - era * 400
26. let mp = m > 2 ? m - 3 : m + 9
27. let doy = idiv(153 * mp + 2, 5) + d - 1
28. let doe = yoe * 365 + idiv(yoe, 4) - idiv(yoe, 100) + doy
29. return era * 146097 + doe - 719468
30. }
32. // 'YYYY-MM-DDTHH:MM:SS[.frac][Z|±HH:MM]' → epoch ms, or null
33. static isoToMs = (s) => {
34. if (s == null || s.length < 19) { return null }
35. let y = toNumber(s.slice(0, 4))
36. let mo = toNumber(s.slice(5, 7))
37. let d = toNumber(s.slice(8, 10))
38. let h = toNumber(s.slice(11, 13))
39. let mi = toNumber(s.slice(14, 16))
40. let se = toNumber(s.slice(17, 19))
41. if (y == null || mo == null || d == null || h == null || mi == null || se == null) { return null }
42. let ms = ((daysFromCivil(y, mo, d) * 24 + h) * 60 + mi) * 60000 + se * 1000
43. let i = 19
44. if (i < s.length && s.charCodeAt(i) == 46) {
45. i = i + 1
46. let frac = ''
47. while (i < s.length && digit(s.charCodeAt(i))) {
48. frac = frac + s.slice(i, i + 1)
49. i = i + 1
50. }
51. if (frac.length >= 3) { ms = ms + toNumber(frac.slice(0, 3)) }
52. }
53. if (i < s.length) {
54. let sign = s.slice(i, i + 1)
55. if ((sign == '+' || sign == '-') && s.length >= i + 6) {
56. let oh = toNumber(s.slice(i + 1, i + 3))
57. let om = toNumber(s.slice(i + 4, i + 6))
58. if (oh != null && om != null) {
59. let off = (oh * 60 + om) * 60000
60. ms = sign == '+' ? ms - off : ms + off
61. }
62. }
63. }
64. return ms
65. }
67. // the number in 'Current session: 5% used' → 5 or null
68. static percentAfter = (text, label) => {
69. let i = text.indexOf(label)
70. if (i < 0) { return null }
71. let j = i + label.length
72. let k = j
73. while (k < text.length && digit(text.charCodeAt(k))) { k = k + 1 }
74. if (k == j) { return null }
75. return toNumber(text.slice(j, k))
76. }
78. // the probe's output (parsed JSON) → { percent, resetsMs, resetsAt, weekly, cost, how } or { error }
79. static parseUsage = (value) => {
80. let limits = null
81. let text = null
82. let cost = null
83. let msgs = isList(value) ? value : [value]
84. for (m of msgs) {
85. if (isObject(m)) {
86. if (m.type == 'assistant' && m.usage_report != null && m.usage_report.rate_limits != null && isList(m.usage_report.rate_limits.limits)) { limits = m.usage_report.rate_limits.limits }
87. if (m.type == 'result') {
88. if (isString(m.result)) { text = m.result }
89. cost = m.total_cost_usd
90. }
91. }
92. }
93. if (limits != null) {
94. let out = { percent = null resetsMs = null resetsAt = null weekly = null weeklyResetsMs = null cost = cost how = 'usage_report' }
95. for (l of limits) {
96. if (isObject(l) && l.kind == 'session') {
97. out.percent = l.percent
98. out.resetsAt = l.resets_at
99. out.resetsMs = isoToMs(l.resets_at)
100. }
101. if (isObject(l) && l.kind == 'weekly_all') {
102. out.weekly = l.percent
103. out.weeklyResetsMs = isoToMs(l.resets_at)
104. }
105. }
106. if (out.percent != null) { return out }
107. }
108. if (text != null) {
109. let p = percentAfter(text, 'Current session: ')
110. if (p != null) { return { percent = p resetsMs = null resetsAt = null weekly = percentAfter(text, 'Current week (all models): ') cost = cost how = 'text' } }
111. }
112. return { error = 'no "session" limit in the /usage answer' }
113. }
115. // runs the probe (bin = the Claude binary, dir = where its files go); calls done(reading) with
116. // { percent, resetsMs, weekly, cost, … } or { error }. → the process
117. static probeUsage = (bin, dir, done) => {
118. let outFile = dir + '/.quota-probe.json'
119. let errFile = dir + '/.quota-probe.err'
120. let args = ['-p' '/usage' '--output-format' 'json' '--verbose' '--model' 'haiku' '--max-turns' '1' '--permission-mode' 'plan' '--permission-prompts' 'none' '--strict-mcp-config']
121. return runClaude({ label = 'quota' folder = dir inFile = '/dev/null' outFile = outFile errFile = errFile bin = bin args = args timeout = '60' }, (x) => {
122. if (x.timedOut) {
123. done({ error = 'the /usage probe timed out (60 s)' })
124. return null
125. }
126. let rj = readJson(outFile)
127. if (rj.error != null) {
128. done({ error = 'the /usage probe gave no JSON (exit ' + x.ev.exit + '): ' + rj.error })
129. return null
130. }
131. done(parseUsage(rj.value))
132. return null
133. })
134. }