@mostok/codexes 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +15 -0
- package/README.md +88 -0
- package/dist/accounts/account-registry.d.ts +30 -0
- package/dist/accounts/account-registry.js +263 -0
- package/dist/accounts/account-registry.js.map +1 -0
- package/dist/accounts/account-resolution.d.ts +16 -0
- package/dist/accounts/account-resolution.js +71 -0
- package/dist/accounts/account-resolution.js.map +1 -0
- package/dist/accounts/resolve-active-account.d.ts +6 -0
- package/dist/accounts/resolve-active-account.js +32 -0
- package/dist/accounts/resolve-active-account.js.map +1 -0
- package/dist/cli.d.ts +1 -0
- package/dist/cli.js +3426 -0
- package/dist/cli.js.map +7 -0
- package/dist/commands/account-add/run-account-add-command.d.ts +2 -0
- package/dist/commands/account-add/run-account-add-command.js +296 -0
- package/dist/commands/account-add/run-account-add-command.js.map +1 -0
- package/dist/commands/account-list/run-account-list-command.d.ts +2 -0
- package/dist/commands/account-list/run-account-list-command.js +48 -0
- package/dist/commands/account-list/run-account-list-command.js.map +1 -0
- package/dist/commands/account-remove/run-account-remove-command.d.ts +2 -0
- package/dist/commands/account-remove/run-account-remove-command.js +52 -0
- package/dist/commands/account-remove/run-account-remove-command.js.map +1 -0
- package/dist/commands/account-use/run-account-use-command.d.ts +2 -0
- package/dist/commands/account-use/run-account-use-command.js +78 -0
- package/dist/commands/account-use/run-account-use-command.js.map +1 -0
- package/dist/commands/root/run-root-command.d.ts +2 -0
- package/dist/commands/root/run-root-command.js +175 -0
- package/dist/commands/root/run-root-command.js.map +1 -0
- package/dist/config/wrapper-config.d.ts +24 -0
- package/dist/config/wrapper-config.js +145 -0
- package/dist/config/wrapper-config.js.map +1 -0
- package/dist/core/bootstrap.d.ts +8 -0
- package/dist/core/bootstrap.js +32 -0
- package/dist/core/bootstrap.js.map +1 -0
- package/dist/core/context.d.ts +33 -0
- package/dist/core/context.js +72 -0
- package/dist/core/context.js.map +1 -0
- package/dist/core/paths.d.ts +12 -0
- package/dist/core/paths.js +30 -0
- package/dist/core/paths.js.map +1 -0
- package/dist/logging/logger.d.ts +18 -0
- package/dist/logging/logger.js +56 -0
- package/dist/logging/logger.js.map +1 -0
- package/dist/process/codex-launch-spec.d.ts +5 -0
- package/dist/process/codex-launch-spec.js +80 -0
- package/dist/process/codex-launch-spec.js.map +1 -0
- package/dist/process/find-codex-binary.d.ts +14 -0
- package/dist/process/find-codex-binary.js +73 -0
- package/dist/process/find-codex-binary.js.map +1 -0
- package/dist/process/run-codex-login.d.ts +14 -0
- package/dist/process/run-codex-login.js +97 -0
- package/dist/process/run-codex-login.js.map +1 -0
- package/dist/process/spawn-codex-command.d.ts +7 -0
- package/dist/process/spawn-codex-command.js +69 -0
- package/dist/process/spawn-codex-command.js.map +1 -0
- package/dist/runtime/activate-account/activate-account.d.ts +27 -0
- package/dist/runtime/activate-account/activate-account.js +298 -0
- package/dist/runtime/activate-account/activate-account.js.map +1 -0
- package/dist/runtime/auth-state-probe.d.ts +57 -0
- package/dist/runtime/auth-state-probe.js +394 -0
- package/dist/runtime/auth-state-probe.js.map +1 -0
- package/dist/runtime/init/initialize-runtime.d.ts +19 -0
- package/dist/runtime/init/initialize-runtime.js +275 -0
- package/dist/runtime/init/initialize-runtime.js.map +1 -0
- package/dist/runtime/lock/runtime-lock.d.ts +11 -0
- package/dist/runtime/lock/runtime-lock.js +99 -0
- package/dist/runtime/lock/runtime-lock.js.map +1 -0
- package/dist/runtime/login-workspace.d.ts +18 -0
- package/dist/runtime/login-workspace.js +171 -0
- package/dist/runtime/login-workspace.js.map +1 -0
- package/dist/runtime/runtime-contract.d.ts +44 -0
- package/dist/runtime/runtime-contract.js +79 -0
- package/dist/runtime/runtime-contract.js.map +1 -0
- package/dist/selection/account-auth-state.d.ts +23 -0
- package/dist/selection/account-auth-state.js +132 -0
- package/dist/selection/account-auth-state.js.map +1 -0
- package/dist/selection/select-account.d.ts +11 -0
- package/dist/selection/select-account.js +168 -0
- package/dist/selection/select-account.js.map +1 -0
- package/dist/selection/usage-cache.d.ts +24 -0
- package/dist/selection/usage-cache.js +106 -0
- package/dist/selection/usage-cache.js.map +1 -0
- package/dist/selection/usage-client.d.ts +23 -0
- package/dist/selection/usage-client.js +143 -0
- package/dist/selection/usage-client.js.map +1 -0
- package/dist/selection/usage-normalize.d.ts +7 -0
- package/dist/selection/usage-normalize.js +209 -0
- package/dist/selection/usage-normalize.js.map +1 -0
- package/dist/selection/usage-probe-coordinator.d.ts +18 -0
- package/dist/selection/usage-probe-coordinator.js +69 -0
- package/dist/selection/usage-probe-coordinator.js.map +1 -0
- package/dist/selection/usage-types.d.ts +59 -0
- package/dist/selection/usage-types.js +2 -0
- package/dist/selection/usage-types.js.map +1 -0
- package/package.json +59 -0
|
@@ -0,0 +1,143 @@
|
|
|
1
|
+
import { readAccountAuthState } from "./account-auth-state.js";
|
|
2
|
+
import { normalizeWhamUsageResponse } from "./usage-normalize.js";
|
|
3
|
+
const WHAM_USAGE_URL = "https://chatgpt.com/backend-api/wham/usage";
|
|
4
|
+
export async function probeAccountUsage(input) {
|
|
5
|
+
const fetchImpl = input.fetchImpl ?? fetch;
|
|
6
|
+
input.logger.info("selection.usage_probe.start", {
|
|
7
|
+
accountId: input.account.id,
|
|
8
|
+
label: input.account.label,
|
|
9
|
+
timeoutMs: input.probeConfig.probeTimeoutMs,
|
|
10
|
+
useAccountIdHeader: input.probeConfig.useAccountIdHeader,
|
|
11
|
+
});
|
|
12
|
+
const authState = await readAccountAuthState({
|
|
13
|
+
account: input.account,
|
|
14
|
+
logger: input.logger,
|
|
15
|
+
});
|
|
16
|
+
if (!authState.ok) {
|
|
17
|
+
input.logger.warn("selection.usage_probe.auth_missing", {
|
|
18
|
+
accountId: input.account.id,
|
|
19
|
+
label: input.account.label,
|
|
20
|
+
category: authState.category,
|
|
21
|
+
filePath: authState.filePath,
|
|
22
|
+
});
|
|
23
|
+
return {
|
|
24
|
+
ok: false,
|
|
25
|
+
account: input.account,
|
|
26
|
+
category: "auth-missing",
|
|
27
|
+
message: authState.message,
|
|
28
|
+
source: "fresh",
|
|
29
|
+
};
|
|
30
|
+
}
|
|
31
|
+
try {
|
|
32
|
+
const response = await fetchImpl(WHAM_USAGE_URL, {
|
|
33
|
+
method: "GET",
|
|
34
|
+
headers: buildUsageHeaders({
|
|
35
|
+
accessToken: authState.state.accessToken,
|
|
36
|
+
accountId: authState.state.accountId,
|
|
37
|
+
useAccountIdHeader: input.probeConfig.useAccountIdHeader,
|
|
38
|
+
}),
|
|
39
|
+
signal: AbortSignal.timeout(input.probeConfig.probeTimeoutMs),
|
|
40
|
+
});
|
|
41
|
+
input.logger.debug("selection.usage_probe.http_complete", {
|
|
42
|
+
accountId: input.account.id,
|
|
43
|
+
label: input.account.label,
|
|
44
|
+
status: response.status,
|
|
45
|
+
ok: response.ok,
|
|
46
|
+
});
|
|
47
|
+
if (!response.ok) {
|
|
48
|
+
input.logger.warn("selection.usage_probe.http_error", {
|
|
49
|
+
accountId: input.account.id,
|
|
50
|
+
label: input.account.label,
|
|
51
|
+
status: response.status,
|
|
52
|
+
});
|
|
53
|
+
return {
|
|
54
|
+
ok: false,
|
|
55
|
+
account: input.account,
|
|
56
|
+
category: "http-error",
|
|
57
|
+
message: `Usage probe returned HTTP ${response.status}.`,
|
|
58
|
+
source: "fresh",
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
const body = (await response.json());
|
|
62
|
+
if (!isRecord(body)) {
|
|
63
|
+
input.logger.warn("selection.usage_probe.invalid_response", {
|
|
64
|
+
accountId: input.account.id,
|
|
65
|
+
label: input.account.label,
|
|
66
|
+
bodyType: typeof body,
|
|
67
|
+
});
|
|
68
|
+
return {
|
|
69
|
+
ok: false,
|
|
70
|
+
account: input.account,
|
|
71
|
+
category: "invalid-response",
|
|
72
|
+
message: "Usage probe returned a non-object JSON payload.",
|
|
73
|
+
source: "fresh",
|
|
74
|
+
};
|
|
75
|
+
}
|
|
76
|
+
const snapshot = normalizeWhamUsageResponse({
|
|
77
|
+
accountIdHint: authState.state.accountId ?? input.account.id,
|
|
78
|
+
logger: input.logger,
|
|
79
|
+
raw: body,
|
|
80
|
+
});
|
|
81
|
+
input.logger.info("selection.usage_probe.success", {
|
|
82
|
+
accountId: input.account.id,
|
|
83
|
+
label: input.account.label,
|
|
84
|
+
snapshotStatus: snapshot.status,
|
|
85
|
+
dailyRemaining: snapshot.dailyRemaining,
|
|
86
|
+
weeklyRemaining: snapshot.weeklyRemaining,
|
|
87
|
+
limitReached: snapshot.limitReached,
|
|
88
|
+
});
|
|
89
|
+
return {
|
|
90
|
+
ok: true,
|
|
91
|
+
account: input.account,
|
|
92
|
+
snapshot,
|
|
93
|
+
source: "fresh",
|
|
94
|
+
};
|
|
95
|
+
}
|
|
96
|
+
catch (error) {
|
|
97
|
+
if (isAbortError(error)) {
|
|
98
|
+
input.logger.warn("selection.usage_probe.timeout", {
|
|
99
|
+
accountId: input.account.id,
|
|
100
|
+
label: input.account.label,
|
|
101
|
+
timeoutMs: input.probeConfig.probeTimeoutMs,
|
|
102
|
+
});
|
|
103
|
+
return {
|
|
104
|
+
ok: false,
|
|
105
|
+
account: input.account,
|
|
106
|
+
category: "timeout",
|
|
107
|
+
message: `Usage probe timed out after ${input.probeConfig.probeTimeoutMs}ms.`,
|
|
108
|
+
source: "fresh",
|
|
109
|
+
};
|
|
110
|
+
}
|
|
111
|
+
input.logger.error("selection.usage_probe.request_failed", {
|
|
112
|
+
accountId: input.account.id,
|
|
113
|
+
label: input.account.label,
|
|
114
|
+
message: error instanceof Error ? error.message : String(error),
|
|
115
|
+
});
|
|
116
|
+
return {
|
|
117
|
+
ok: false,
|
|
118
|
+
account: input.account,
|
|
119
|
+
category: "invalid-response",
|
|
120
|
+
message: error instanceof Error ? error.message : String(error),
|
|
121
|
+
source: "fresh",
|
|
122
|
+
};
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
function buildUsageHeaders(input) {
|
|
126
|
+
const headers = new Headers({
|
|
127
|
+
accept: "application/json",
|
|
128
|
+
authorization: `Bearer ${input.accessToken}`,
|
|
129
|
+
"user-agent": "codexes/0.1 experimental-usage-probe",
|
|
130
|
+
});
|
|
131
|
+
if (input.useAccountIdHeader && input.accountId) {
|
|
132
|
+
headers.set("OpenAI-Account-ID", input.accountId);
|
|
133
|
+
}
|
|
134
|
+
return headers;
|
|
135
|
+
}
|
|
136
|
+
function isAbortError(error) {
|
|
137
|
+
return (error instanceof Error &&
|
|
138
|
+
(error.name === "AbortError" || error.name === "TimeoutError"));
|
|
139
|
+
}
|
|
140
|
+
function isRecord(value) {
|
|
141
|
+
return typeof value === "object" && value !== null;
|
|
142
|
+
}
|
|
143
|
+
//# sourceMappingURL=usage-client.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usage-client.js","sourceRoot":"","sources":["../../src/selection/usage-client.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAC;AAGlE,MAAM,cAAc,GAAG,4CAA4C,CAAC;AAuBpE,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,KAKvC;IACC,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC;IAE3C,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,6BAA6B,EAAE;QAC/C,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE;QAC3B,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK;QAC1B,SAAS,EAAE,KAAK,CAAC,WAAW,CAAC,cAAc;QAC3C,kBAAkB,EAAE,KAAK,CAAC,WAAW,CAAC,kBAAkB;KACzD,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,MAAM,oBAAoB,CAAC;QAC3C,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,MAAM,EAAE,KAAK,CAAC,MAAM;KACrB,CAAC,CAAC;IACH,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC;QAClB,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,oCAAoC,EAAE;YACtD,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE;YAC3B,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK;YAC1B,QAAQ,EAAE,SAAS,CAAC,QAAQ;YAC5B,QAAQ,EAAE,SAAS,CAAC,QAAQ;SAC7B,CAAC,CAAC;QAEH,OAAO;YACL,EAAE,EAAE,KAAK;YACT,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,QAAQ,EAAE,cAAc;YACxB,OAAO,EAAE,SAAS,CAAC,OAAO;YAC1B,MAAM,EAAE,OAAO;SAChB,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,cAAc,EAAE;YAC/C,MAAM,EAAE,KAAK;YACb,OAAO,EAAE,iBAAiB,CAAC;gBACzB,WAAW,EAAE,SAAS,CAAC,KAAK,CAAC,WAAW;gBACxC,SAAS,EAAE,SAAS,CAAC,KAAK,CAAC,SAAS;gBACpC,kBAAkB,EAAE,KAAK,CAAC,WAAW,CAAC,kBAAkB;aACzD,CAAC;YACF,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,cAAc,CAAC;SAC9D,CAAC,CAAC;QAEH,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE;YACxD,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE;YAC3B,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK;YAC1B,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,EAAE,EAAE,QAAQ,CAAC,EAAE;SAChB,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE;gBACpD,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE;gBAC3B,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK;gBAC1B,MAAM,EAAE,QAAQ,CAAC,MAAM;aACxB,CAAC,CAAC;YACH,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ,EAAE,YAAY;gBACtB,OAAO,EAAE,6BAA6B,QAAQ,CAAC,MAAM,GAAG;gBACxD,MAAM,EAAE,OAAO;aAChB,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAY,CAAC;QAChD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACpB,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE;gBAC1D,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE;gBAC3B,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK;gBAC1B,QAAQ,EAAE,OAAO,IAAI;aACtB,CAAC,CAAC;YACH,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ,EAAE,kBAAkB;gBAC5B,OAAO,EAAE,iDAAiD;gBAC1D,MAAM,EAAE,OAAO;aAChB,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,0BAA0B,CAAC;YAC1C,aAAa,EAAE,SAAS,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,EAAE;YAC5D,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,GAAG,EAAE,IAA4B;SAClC,CAAC,CAAC;QAEH,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE;YACjD,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE;YAC3B,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK;YAC1B,cAAc,EAAE,QAAQ,CAAC,MAAM;YAC/B,cAAc,EAAE,QAAQ,CAAC,cAAc;YACvC,eAAe,EAAE,QAAQ,CAAC,eAAe;YACzC,YAAY,EAAE,QAAQ,CAAC,YAAY;SACpC,CAAC,CAAC;QAEH,OAAO;YACL,EAAE,EAAE,IAAI;YACR,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,QAAQ;YACR,MAAM,EAAE,OAAO;SAChB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;YACxB,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE;gBACjD,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE;gBAC3B,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK;gBAC1B,SAAS,EAAE,KAAK,CAAC,WAAW,CAAC,cAAc;aAC5C,CAAC,CAAC;YACH,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ,EAAE,SAAS;gBACnB,OAAO,EAAE,+BAA+B,KAAK,CAAC,WAAW,CAAC,cAAc,KAAK;gBAC7E,MAAM,EAAE,OAAO;aAChB,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,sCAAsC,EAAE;YACzD,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE;YAC3B,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK;YAC1B,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAChE,CAAC,CAAC;QACH,OAAO;YACL,EAAE,EAAE,KAAK;YACT,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,QAAQ,EAAE,kBAAkB;YAC5B,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YAC/D,MAAM,EAAE,OAAO;SAChB,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,KAI1B;IACC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC;QAC1B,MAAM,EAAE,kBAAkB;QAC1B,aAAa,EAAE,UAAU,KAAK,CAAC,WAAW,EAAE;QAC5C,YAAY,EAAE,sCAAsC;KACrD,CAAC,CAAC;IAEH,IAAI,KAAK,CAAC,kBAAkB,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;QAChD,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;IACpD,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,CACL,KAAK,YAAY,KAAK;QACtB,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,CAAC,CAC/D,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACrD,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import type { Logger } from "../logging/logger.js";
|
|
2
|
+
import type { NormalizedUsageSnapshot, WhamUsageResponseRaw } from "./usage-types.js";
|
|
3
|
+
export declare function normalizeWhamUsageResponse(input: {
|
|
4
|
+
accountIdHint?: string | null;
|
|
5
|
+
logger: Logger;
|
|
6
|
+
raw: WhamUsageResponseRaw;
|
|
7
|
+
}): NormalizedUsageSnapshot;
|
|
@@ -0,0 +1,209 @@
|
|
|
1
|
+
export function normalizeWhamUsageResponse(input) {
|
|
2
|
+
input.logger.debug("selection.usage_normalize.start", {
|
|
3
|
+
accountIdHint: input.accountIdHint ?? null,
|
|
4
|
+
topLevelKeys: Object.keys(input.raw).sort(),
|
|
5
|
+
});
|
|
6
|
+
const daily = normalizeUsageWindow({
|
|
7
|
+
accountIdHint: input.accountIdHint,
|
|
8
|
+
logger: input.logger,
|
|
9
|
+
raw: resolveUsageWindow(input.raw, "daily"),
|
|
10
|
+
window: "daily",
|
|
11
|
+
});
|
|
12
|
+
const weekly = normalizeUsageWindow({
|
|
13
|
+
accountIdHint: input.accountIdHint,
|
|
14
|
+
logger: input.logger,
|
|
15
|
+
raw: resolveUsageWindow(input.raw, "weekly"),
|
|
16
|
+
window: "weekly",
|
|
17
|
+
});
|
|
18
|
+
const accountId = pickString(input.raw.account_id, input.raw.accountId, input.accountIdHint);
|
|
19
|
+
const allowed = typeof input.raw.allowed === "boolean" ? input.raw.allowed : true;
|
|
20
|
+
const limitReached = typeof input.raw.limit_reached === "boolean"
|
|
21
|
+
? input.raw.limit_reached
|
|
22
|
+
: daily.limitReached || weekly.limitReached;
|
|
23
|
+
const status = classifyUsageStatus({
|
|
24
|
+
allowed,
|
|
25
|
+
dailyRemaining: daily.remaining,
|
|
26
|
+
limitReached,
|
|
27
|
+
weeklyRemaining: weekly.remaining,
|
|
28
|
+
});
|
|
29
|
+
const snapshot = {
|
|
30
|
+
accountId,
|
|
31
|
+
allowed,
|
|
32
|
+
limitReached,
|
|
33
|
+
dailyRemaining: daily.remaining,
|
|
34
|
+
weeklyRemaining: weekly.remaining,
|
|
35
|
+
dailyResetsAt: daily.resetsAt,
|
|
36
|
+
weeklyResetsAt: weekly.resetsAt,
|
|
37
|
+
dailyPercentUsed: daily.percentUsed,
|
|
38
|
+
weeklyPercentUsed: weekly.percentUsed,
|
|
39
|
+
observedAt: new Date().toISOString(),
|
|
40
|
+
status,
|
|
41
|
+
statusReason: describeUsageStatus(status),
|
|
42
|
+
windows: {
|
|
43
|
+
daily,
|
|
44
|
+
weekly,
|
|
45
|
+
},
|
|
46
|
+
};
|
|
47
|
+
input.logger.debug("selection.usage_normalize.complete", {
|
|
48
|
+
accountId: snapshot.accountId,
|
|
49
|
+
allowed: snapshot.allowed,
|
|
50
|
+
limitReached: snapshot.limitReached,
|
|
51
|
+
dailyRemaining: snapshot.dailyRemaining,
|
|
52
|
+
weeklyRemaining: snapshot.weeklyRemaining,
|
|
53
|
+
dailyResetsAt: snapshot.dailyResetsAt,
|
|
54
|
+
weeklyResetsAt: snapshot.weeklyResetsAt,
|
|
55
|
+
status: snapshot.status,
|
|
56
|
+
statusReason: snapshot.statusReason,
|
|
57
|
+
});
|
|
58
|
+
return snapshot;
|
|
59
|
+
}
|
|
60
|
+
function normalizeUsageWindow(input) {
|
|
61
|
+
if (!input.raw) {
|
|
62
|
+
input.logger.debug("selection.usage_normalize.window_missing", {
|
|
63
|
+
accountIdHint: input.accountIdHint ?? null,
|
|
64
|
+
window: input.window,
|
|
65
|
+
});
|
|
66
|
+
return {
|
|
67
|
+
limit: null,
|
|
68
|
+
used: null,
|
|
69
|
+
remaining: null,
|
|
70
|
+
limitReached: false,
|
|
71
|
+
resetsAt: null,
|
|
72
|
+
percentUsed: null,
|
|
73
|
+
source: null,
|
|
74
|
+
};
|
|
75
|
+
}
|
|
76
|
+
const limit = pickNumber(input.raw.limit);
|
|
77
|
+
const used = pickNumber(input.raw.used, calculateUsed(limit, pickNumber(input.raw.remaining)));
|
|
78
|
+
const remaining = pickNumber(input.raw.remaining, calculateRemaining(limit, used));
|
|
79
|
+
const limitReached = typeof input.raw.limit_reached === "boolean"
|
|
80
|
+
? input.raw.limit_reached
|
|
81
|
+
: remaining !== null
|
|
82
|
+
? remaining <= 0
|
|
83
|
+
: false;
|
|
84
|
+
const percentUsed = pickNumber(input.raw.percent_used, input.raw.percentage_used, calculatePercentUsed(limit, used, remaining));
|
|
85
|
+
const resetsAt = normalizeTimestamp(input.raw.reset_at, input.raw.resets_at, input.raw.next_reset_at);
|
|
86
|
+
const source = resolveWindowSource(input.raw);
|
|
87
|
+
input.logger.debug("selection.usage_normalize.window_complete", {
|
|
88
|
+
accountIdHint: input.accountIdHint ?? null,
|
|
89
|
+
window: input.window,
|
|
90
|
+
limit,
|
|
91
|
+
used,
|
|
92
|
+
remaining,
|
|
93
|
+
limitReached,
|
|
94
|
+
percentUsed,
|
|
95
|
+
resetsAt,
|
|
96
|
+
source,
|
|
97
|
+
});
|
|
98
|
+
return {
|
|
99
|
+
limit,
|
|
100
|
+
used,
|
|
101
|
+
remaining,
|
|
102
|
+
limitReached,
|
|
103
|
+
resetsAt,
|
|
104
|
+
percentUsed,
|
|
105
|
+
source,
|
|
106
|
+
};
|
|
107
|
+
}
|
|
108
|
+
function resolveUsageWindow(raw, window) {
|
|
109
|
+
const candidates = [raw[window], raw.usage?.[window], raw.quotas?.[window]];
|
|
110
|
+
for (const candidate of candidates) {
|
|
111
|
+
if (isRecord(candidate)) {
|
|
112
|
+
return candidate;
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
return null;
|
|
116
|
+
}
|
|
117
|
+
function resolveWindowSource(raw) {
|
|
118
|
+
if (typeof raw.source === "string") {
|
|
119
|
+
return raw.source;
|
|
120
|
+
}
|
|
121
|
+
if (typeof raw.kind === "string") {
|
|
122
|
+
return raw.kind;
|
|
123
|
+
}
|
|
124
|
+
return null;
|
|
125
|
+
}
|
|
126
|
+
function classifyUsageStatus(input) {
|
|
127
|
+
if (!input.allowed) {
|
|
128
|
+
return "not-allowed";
|
|
129
|
+
}
|
|
130
|
+
if (input.limitReached) {
|
|
131
|
+
return "limit-reached";
|
|
132
|
+
}
|
|
133
|
+
if (input.dailyRemaining === null && input.weeklyRemaining === null) {
|
|
134
|
+
return "missing-usage-data";
|
|
135
|
+
}
|
|
136
|
+
return "usable";
|
|
137
|
+
}
|
|
138
|
+
function describeUsageStatus(status) {
|
|
139
|
+
switch (status) {
|
|
140
|
+
case "not-allowed":
|
|
141
|
+
return "usage endpoint reported that the account is not allowed to launch";
|
|
142
|
+
case "limit-reached":
|
|
143
|
+
return "usage endpoint reported an exhausted limit window";
|
|
144
|
+
case "missing-usage-data":
|
|
145
|
+
return "usage endpoint did not expose enough quota fields to rank this account";
|
|
146
|
+
case "usable":
|
|
147
|
+
return "usage endpoint exposed enough quota fields to rank this account";
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
function normalizeTimestamp(...values) {
|
|
151
|
+
for (const value of values) {
|
|
152
|
+
if (typeof value === "string") {
|
|
153
|
+
const parsed = new Date(value);
|
|
154
|
+
if (!Number.isNaN(parsed.valueOf())) {
|
|
155
|
+
return parsed.toISOString();
|
|
156
|
+
}
|
|
157
|
+
}
|
|
158
|
+
if (typeof value === "number" && Number.isFinite(value)) {
|
|
159
|
+
const normalizedValue = value > 10_000_000_000 ? value : value * 1_000;
|
|
160
|
+
const parsed = new Date(normalizedValue);
|
|
161
|
+
if (!Number.isNaN(parsed.valueOf())) {
|
|
162
|
+
return parsed.toISOString();
|
|
163
|
+
}
|
|
164
|
+
}
|
|
165
|
+
}
|
|
166
|
+
return null;
|
|
167
|
+
}
|
|
168
|
+
function calculateRemaining(limit, used) {
|
|
169
|
+
if (limit === null || used === null) {
|
|
170
|
+
return null;
|
|
171
|
+
}
|
|
172
|
+
return limit - used;
|
|
173
|
+
}
|
|
174
|
+
function calculateUsed(limit, remaining) {
|
|
175
|
+
if (limit === null || remaining === null) {
|
|
176
|
+
return null;
|
|
177
|
+
}
|
|
178
|
+
return limit - remaining;
|
|
179
|
+
}
|
|
180
|
+
function calculatePercentUsed(limit, used, remaining) {
|
|
181
|
+
if (limit === null || limit <= 0) {
|
|
182
|
+
return null;
|
|
183
|
+
}
|
|
184
|
+
const numerator = used ?? calculateUsed(limit, remaining);
|
|
185
|
+
if (numerator === null) {
|
|
186
|
+
return null;
|
|
187
|
+
}
|
|
188
|
+
return Number(((numerator / limit) * 100).toFixed(2));
|
|
189
|
+
}
|
|
190
|
+
function pickString(...values) {
|
|
191
|
+
for (const value of values) {
|
|
192
|
+
if (typeof value === "string" && value.trim().length > 0) {
|
|
193
|
+
return value;
|
|
194
|
+
}
|
|
195
|
+
}
|
|
196
|
+
return null;
|
|
197
|
+
}
|
|
198
|
+
function pickNumber(...values) {
|
|
199
|
+
for (const value of values) {
|
|
200
|
+
if (typeof value === "number" && Number.isFinite(value)) {
|
|
201
|
+
return value;
|
|
202
|
+
}
|
|
203
|
+
}
|
|
204
|
+
return null;
|
|
205
|
+
}
|
|
206
|
+
function isRecord(value) {
|
|
207
|
+
return typeof value === "object" && value !== null;
|
|
208
|
+
}
|
|
209
|
+
//# sourceMappingURL=usage-normalize.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usage-normalize.js","sourceRoot":"","sources":["../../src/selection/usage-normalize.ts"],"names":[],"mappings":"AASA,MAAM,UAAU,0BAA0B,CAAC,KAI1C;IACC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE;QACpD,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,IAAI;QAC1C,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE;KAC5C,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,oBAAoB,CAAC;QACjC,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,GAAG,EAAE,kBAAkB,CAAC,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC;QAC3C,MAAM,EAAE,OAAO;KAChB,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,oBAAoB,CAAC;QAClC,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,GAAG,EAAE,kBAAkB,CAAC,KAAK,CAAC,GAAG,EAAE,QAAQ,CAAC;QAC5C,MAAM,EAAE,QAAQ;KACjB,CAAC,CAAC;IACH,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;IAC7F,MAAM,OAAO,GAAG,OAAO,KAAK,CAAC,GAAG,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;IAClF,MAAM,YAAY,GAChB,OAAO,KAAK,CAAC,GAAG,CAAC,aAAa,KAAK,SAAS;QAC1C,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa;QACzB,CAAC,CAAC,KAAK,CAAC,YAAY,IAAI,MAAM,CAAC,YAAY,CAAC;IAChD,MAAM,MAAM,GAAG,mBAAmB,CAAC;QACjC,OAAO;QACP,cAAc,EAAE,KAAK,CAAC,SAAS;QAC/B,YAAY;QACZ,eAAe,EAAE,MAAM,CAAC,SAAS;KAClC,CAAC,CAAC;IACH,MAAM,QAAQ,GAA4B;QACxC,SAAS;QACT,OAAO;QACP,YAAY;QACZ,cAAc,EAAE,KAAK,CAAC,SAAS;QAC/B,eAAe,EAAE,MAAM,CAAC,SAAS;QACjC,aAAa,EAAE,KAAK,CAAC,QAAQ;QAC7B,cAAc,EAAE,MAAM,CAAC,QAAQ;QAC/B,gBAAgB,EAAE,KAAK,CAAC,WAAW;QACnC,iBAAiB,EAAE,MAAM,CAAC,WAAW;QACrC,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACpC,MAAM;QACN,YAAY,EAAE,mBAAmB,CAAC,MAAM,CAAC;QACzC,OAAO,EAAE;YACP,KAAK;YACL,MAAM;SACP;KACF,CAAC;IAEF,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE;QACvD,SAAS,EAAE,QAAQ,CAAC,SAAS;QAC7B,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,YAAY,EAAE,QAAQ,CAAC,YAAY;QACnC,cAAc,EAAE,QAAQ,CAAC,cAAc;QACvC,eAAe,EAAE,QAAQ,CAAC,eAAe;QACzC,aAAa,EAAE,QAAQ,CAAC,aAAa;QACrC,cAAc,EAAE,QAAQ,CAAC,cAAc;QACvC,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,YAAY,EAAE,QAAQ,CAAC,YAAY;KACpC,CAAC,CAAC;IAEH,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,oBAAoB,CAAC,KAK7B;IACC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QACf,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE;YAC7D,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,IAAI;YAC1C,MAAM,EAAE,KAAK,CAAC,MAAM;SACrB,CAAC,CAAC;QAEH,OAAO;YACL,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,IAAI;YACV,SAAS,EAAE,IAAI;YACf,YAAY,EAAE,KAAK;YACnB,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,IAAI;YACjB,MAAM,EAAE,IAAI;SACb,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC1C,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,aAAa,CAAC,KAAK,EAAE,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC/F,MAAM,SAAS,GAAG,UAAU,CAC1B,KAAK,CAAC,GAAG,CAAC,SAAS,EACnB,kBAAkB,CAAC,KAAK,EAAE,IAAI,CAAC,CAChC,CAAC;IACF,MAAM,YAAY,GAChB,OAAO,KAAK,CAAC,GAAG,CAAC,aAAa,KAAK,SAAS;QAC1C,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa;QACzB,CAAC,CAAC,SAAS,KAAK,IAAI;YAClB,CAAC,CAAC,SAAS,IAAI,CAAC;YAChB,CAAC,CAAC,KAAK,CAAC;IACd,MAAM,WAAW,GAAG,UAAU,CAC5B,KAAK,CAAC,GAAG,CAAC,YAAY,EACtB,KAAK,CAAC,GAAG,CAAC,eAAe,EACzB,oBAAoB,CAAC,KAAK,EAAE,IAAI,EAAE,SAAS,CAAC,CAC7C,CAAC;IACF,MAAM,QAAQ,GAAG,kBAAkB,CACjC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAClB,KAAK,CAAC,GAAG,CAAC,SAAS,EACnB,KAAK,CAAC,GAAG,CAAC,aAAa,CACxB,CAAC;IACF,MAAM,MAAM,GAAG,mBAAmB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAE9C,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE;QAC9D,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,IAAI;QAC1C,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,KAAK;QACL,IAAI;QACJ,SAAS;QACT,YAAY;QACZ,WAAW;QACX,QAAQ;QACR,MAAM;KACP,CAAC,CAAC;IAEH,OAAO;QACL,KAAK;QACL,IAAI;QACJ,SAAS;QACT,YAAY;QACZ,QAAQ;QACR,WAAW;QACX,MAAM;KACP,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CACzB,GAAyB,EACzB,MAA0B;IAE1B,MAAM,UAAU,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAE5E,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,IAAI,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YACxB,OAAO,SAAgC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,mBAAmB,CAAC,GAAwB;IACnD,IAAI,OAAO,GAAG,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QACnC,OAAO,GAAG,CAAC,MAAM,CAAC;IACpB,CAAC;IAED,IAAI,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QACjC,OAAO,GAAG,CAAC,IAAI,CAAC;IAClB,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,mBAAmB,CAAC,KAK5B;IACC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;QACvB,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,KAAK,CAAC,cAAc,KAAK,IAAI,IAAI,KAAK,CAAC,eAAe,KAAK,IAAI,EAAE,CAAC;QACpE,OAAO,oBAAoB,CAAC;IAC9B,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,mBAAmB,CAAC,MAA2B;IACtD,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,aAAa;YAChB,OAAO,mEAAmE,CAAC;QAC7E,KAAK,eAAe;YAClB,OAAO,mDAAmD,CAAC;QAC7D,KAAK,oBAAoB;YACvB,OAAO,wEAAwE,CAAC;QAClF,KAAK,QAAQ;YACX,OAAO,iEAAiE,CAAC;IAC7E,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,GAAG,MAAiD;IAC9E,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;gBACpC,OAAO,MAAM,CAAC,WAAW,EAAE,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACxD,MAAM,eAAe,GAAG,KAAK,GAAG,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC;YACvE,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,eAAe,CAAC,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;gBACpC,OAAO,MAAM,CAAC,WAAW,EAAE,CAAC;YAC9B,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,kBAAkB,CACzB,KAAoB,EACpB,IAAmB;IAEnB,IAAI,KAAK,KAAK,IAAI,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QACpC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,GAAG,IAAI,CAAC;AACtB,CAAC;AAED,SAAS,aAAa,CACpB,KAAoB,EACpB,SAAwB;IAExB,IAAI,KAAK,KAAK,IAAI,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;QACzC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,GAAG,SAAS,CAAC;AAC3B,CAAC;AAED,SAAS,oBAAoB,CAC3B,KAAoB,EACpB,IAAmB,EACnB,SAAwB;IAExB,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACjC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,IAAI,aAAa,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAC1D,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,MAAM,CAAC,CAAC,CAAC,SAAS,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;AACxD,CAAC;AAED,SAAS,UAAU,CACjB,GAAG,MAAwC;IAE3C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzD,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,UAAU,CACjB,GAAG,MAAwC;IAE3C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACxD,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACrD,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { AccountRecord } from "../accounts/account-registry.js";
|
|
2
|
+
import type { ExperimentalSelectionConfig } from "../config/wrapper-config.js";
|
|
3
|
+
import type { Logger } from "../logging/logger.js";
|
|
4
|
+
import { type AccountUsageProbeResult } from "./usage-client.js";
|
|
5
|
+
import type { NormalizedUsageSnapshot } from "./usage-types.js";
|
|
6
|
+
export type AccountUsageResolution = {
|
|
7
|
+
ok: true;
|
|
8
|
+
account: AccountRecord;
|
|
9
|
+
snapshot: NormalizedUsageSnapshot;
|
|
10
|
+
source: "cache" | "fresh";
|
|
11
|
+
} | AccountUsageProbeResult;
|
|
12
|
+
export declare function resolveAccountUsageSnapshots(input: {
|
|
13
|
+
accounts: AccountRecord[];
|
|
14
|
+
cacheFilePath: string;
|
|
15
|
+
fetchImpl?: typeof fetch;
|
|
16
|
+
logger: Logger;
|
|
17
|
+
probeConfig: ExperimentalSelectionConfig;
|
|
18
|
+
}): Promise<AccountUsageResolution[]>;
|
|
@@ -0,0 +1,69 @@
|
|
|
1
|
+
import { probeAccountUsage } from "./usage-client.js";
|
|
2
|
+
import { loadUsageCache, persistUsageCache, resolveFreshUsageCacheEntry, } from "./usage-cache.js";
|
|
3
|
+
export async function resolveAccountUsageSnapshots(input) {
|
|
4
|
+
const now = Date.now();
|
|
5
|
+
input.logger.info("selection.usage_probe_coordinator.start", {
|
|
6
|
+
accountCount: input.accounts.length,
|
|
7
|
+
cacheFilePath: input.cacheFilePath,
|
|
8
|
+
cacheTtlMs: input.probeConfig.cacheTtlMs,
|
|
9
|
+
timeoutMs: input.probeConfig.probeTimeoutMs,
|
|
10
|
+
});
|
|
11
|
+
const cacheEntries = await loadUsageCache({
|
|
12
|
+
cacheFilePath: input.cacheFilePath,
|
|
13
|
+
logger: input.logger,
|
|
14
|
+
});
|
|
15
|
+
const freshCacheEntries = [...cacheEntries];
|
|
16
|
+
const resolutions = await Promise.all(input.accounts.map(async (account) => {
|
|
17
|
+
const cached = resolveFreshUsageCacheEntry({
|
|
18
|
+
accountId: account.id,
|
|
19
|
+
entries: freshCacheEntries,
|
|
20
|
+
logger: input.logger,
|
|
21
|
+
now,
|
|
22
|
+
ttlMs: input.probeConfig.cacheTtlMs,
|
|
23
|
+
});
|
|
24
|
+
if (cached) {
|
|
25
|
+
return {
|
|
26
|
+
ok: true,
|
|
27
|
+
account,
|
|
28
|
+
snapshot: cached.snapshot,
|
|
29
|
+
source: "cache",
|
|
30
|
+
};
|
|
31
|
+
}
|
|
32
|
+
const fresh = await probeAccountUsage({
|
|
33
|
+
account,
|
|
34
|
+
fetchImpl: input.fetchImpl,
|
|
35
|
+
logger: input.logger,
|
|
36
|
+
probeConfig: input.probeConfig,
|
|
37
|
+
});
|
|
38
|
+
if (fresh.ok) {
|
|
39
|
+
upsertCacheEntry(freshCacheEntries, {
|
|
40
|
+
accountId: account.id,
|
|
41
|
+
accountLabel: account.label,
|
|
42
|
+
cachedAt: new Date(now).toISOString(),
|
|
43
|
+
snapshot: fresh.snapshot,
|
|
44
|
+
});
|
|
45
|
+
}
|
|
46
|
+
return fresh;
|
|
47
|
+
}));
|
|
48
|
+
await persistUsageCache({
|
|
49
|
+
cacheFilePath: input.cacheFilePath,
|
|
50
|
+
entries: freshCacheEntries,
|
|
51
|
+
logger: input.logger,
|
|
52
|
+
});
|
|
53
|
+
input.logger.info("selection.usage_probe_coordinator.complete", {
|
|
54
|
+
accountCount: input.accounts.length,
|
|
55
|
+
cacheHitCount: resolutions.filter((entry) => entry.ok && entry.source === "cache").length,
|
|
56
|
+
freshSuccessCount: resolutions.filter((entry) => entry.ok && entry.source === "fresh").length,
|
|
57
|
+
failureCount: resolutions.filter((entry) => !entry.ok).length,
|
|
58
|
+
});
|
|
59
|
+
return resolutions;
|
|
60
|
+
}
|
|
61
|
+
function upsertCacheEntry(entries, nextEntry) {
|
|
62
|
+
const existingIndex = entries.findIndex((entry) => entry.accountId === nextEntry.accountId);
|
|
63
|
+
if (existingIndex >= 0) {
|
|
64
|
+
entries.splice(existingIndex, 1, nextEntry);
|
|
65
|
+
return;
|
|
66
|
+
}
|
|
67
|
+
entries.push(nextEntry);
|
|
68
|
+
}
|
|
69
|
+
//# sourceMappingURL=usage-probe-coordinator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usage-probe-coordinator.js","sourceRoot":"","sources":["../../src/selection/usage-probe-coordinator.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,iBAAiB,EAAgC,MAAM,mBAAmB,CAAC;AACpF,OAAO,EACL,cAAc,EACd,iBAAiB,EACjB,2BAA2B,GAE5B,MAAM,kBAAkB,CAAC;AAY1B,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,KAMlD;IACC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAEvB,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,yCAAyC,EAAE;QAC3D,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,MAAM;QACnC,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,UAAU,EAAE,KAAK,CAAC,WAAW,CAAC,UAAU;QACxC,SAAS,EAAE,KAAK,CAAC,WAAW,CAAC,cAAc;KAC5C,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,MAAM,cAAc,CAAC;QACxC,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,MAAM,EAAE,KAAK,CAAC,MAAM;KACrB,CAAC,CAAC;IACH,MAAM,iBAAiB,GAAsB,CAAC,GAAG,YAAY,CAAC,CAAC;IAC/D,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CACnC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;QACnC,MAAM,MAAM,GAAG,2BAA2B,CAAC;YACzC,SAAS,EAAE,OAAO,CAAC,EAAE;YACrB,OAAO,EAAE,iBAAiB;YAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,GAAG;YACH,KAAK,EAAE,KAAK,CAAC,WAAW,CAAC,UAAU;SACpC,CAAC,CAAC;QACH,IAAI,MAAM,EAAE,CAAC;YACX,OAAO;gBACL,EAAE,EAAE,IAAa;gBACjB,OAAO;gBACP,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,MAAM,EAAE,OAAgB;aACzB,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,iBAAiB,CAAC;YACpC,OAAO;YACP,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,WAAW,EAAE,KAAK,CAAC,WAAW;SAC/B,CAAC,CAAC;QACH,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC;YACb,gBAAgB,CAAC,iBAAiB,EAAE;gBAClC,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,YAAY,EAAE,OAAO,CAAC,KAAK;gBAC3B,QAAQ,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE;gBACrC,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACzB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC,CAAC,CACH,CAAC;IAEF,MAAM,iBAAiB,CAAC;QACtB,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,OAAO,EAAE,iBAAiB;QAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;KACrB,CAAC,CAAC;IAEH,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,4CAA4C,EAAE;QAC9D,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,MAAM;QACnC,aAAa,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC,MAAM;QACzF,iBAAiB,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC,MAAM;QAC7F,YAAY,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,MAAM;KAC9D,CAAC,CAAC;IAEH,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,SAAS,gBAAgB,CAAC,OAA0B,EAAE,SAA0B;IAC9E,MAAM,aAAa,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,SAAS,CAAC,CAAC;IAC5F,IAAI,aAAa,IAAI,CAAC,EAAE,CAAC;QACvB,OAAO,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;QAC5C,OAAO;IACT,CAAC;IAED,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AAC1B,CAAC"}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
export interface UsageQuotaWindowRaw {
|
|
2
|
+
limit?: number | null;
|
|
3
|
+
used?: number | null;
|
|
4
|
+
remaining?: number | null;
|
|
5
|
+
limit_reached?: boolean | null;
|
|
6
|
+
reset_at?: number | string | null;
|
|
7
|
+
resets_at?: number | string | null;
|
|
8
|
+
next_reset_at?: number | string | null;
|
|
9
|
+
percent_used?: number | null;
|
|
10
|
+
percentage_used?: number | null;
|
|
11
|
+
[key: string]: unknown;
|
|
12
|
+
}
|
|
13
|
+
export interface WhamUsageResponseRaw {
|
|
14
|
+
account_id?: string | null;
|
|
15
|
+
accountId?: string | null;
|
|
16
|
+
allowed?: boolean | null;
|
|
17
|
+
limit_reached?: boolean | null;
|
|
18
|
+
daily?: UsageQuotaWindowRaw | null;
|
|
19
|
+
weekly?: UsageQuotaWindowRaw | null;
|
|
20
|
+
usage?: {
|
|
21
|
+
daily?: UsageQuotaWindowRaw | null;
|
|
22
|
+
weekly?: UsageQuotaWindowRaw | null;
|
|
23
|
+
[key: string]: unknown;
|
|
24
|
+
} | null;
|
|
25
|
+
quotas?: {
|
|
26
|
+
daily?: UsageQuotaWindowRaw | null;
|
|
27
|
+
weekly?: UsageQuotaWindowRaw | null;
|
|
28
|
+
[key: string]: unknown;
|
|
29
|
+
} | null;
|
|
30
|
+
[key: string]: unknown;
|
|
31
|
+
}
|
|
32
|
+
export type UsageSnapshotStatus = "usable" | "not-allowed" | "limit-reached" | "missing-usage-data";
|
|
33
|
+
export interface NormalizedUsageWindow {
|
|
34
|
+
limit: number | null;
|
|
35
|
+
used: number | null;
|
|
36
|
+
remaining: number | null;
|
|
37
|
+
limitReached: boolean;
|
|
38
|
+
resetsAt: string | null;
|
|
39
|
+
percentUsed: number | null;
|
|
40
|
+
source: string | null;
|
|
41
|
+
}
|
|
42
|
+
export interface NormalizedUsageSnapshot {
|
|
43
|
+
accountId: string | null;
|
|
44
|
+
allowed: boolean;
|
|
45
|
+
limitReached: boolean;
|
|
46
|
+
dailyRemaining: number | null;
|
|
47
|
+
weeklyRemaining: number | null;
|
|
48
|
+
dailyResetsAt: string | null;
|
|
49
|
+
weeklyResetsAt: string | null;
|
|
50
|
+
dailyPercentUsed: number | null;
|
|
51
|
+
weeklyPercentUsed: number | null;
|
|
52
|
+
observedAt: string;
|
|
53
|
+
status: UsageSnapshotStatus;
|
|
54
|
+
statusReason: string;
|
|
55
|
+
windows: {
|
|
56
|
+
daily: NormalizedUsageWindow;
|
|
57
|
+
weekly: NormalizedUsageWindow;
|
|
58
|
+
};
|
|
59
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usage-types.js","sourceRoot":"","sources":["../../src/selection/usage-types.ts"],"names":[],"mappings":""}
|
package/package.json
ADDED
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
{
|
|
2
|
+
"name": "@mostok/codexes",
|
|
3
|
+
"version": "0.1.0",
|
|
4
|
+
"description": "Transparent multi-account wrapper around the Codex CLI.",
|
|
5
|
+
"type": "module",
|
|
6
|
+
"bin": {
|
|
7
|
+
"codexes": "dist/cli.js"
|
|
8
|
+
},
|
|
9
|
+
"repository": {
|
|
10
|
+
"type": "git",
|
|
11
|
+
"url": "git+https://github.com/Mostok/codexes.git"
|
|
12
|
+
},
|
|
13
|
+
"homepage": "https://github.com/Mostok/codexes#readme",
|
|
14
|
+
"bugs": {
|
|
15
|
+
"url": "https://github.com/Mostok/codexes/issues"
|
|
16
|
+
},
|
|
17
|
+
"files": [
|
|
18
|
+
"dist",
|
|
19
|
+
"README.md",
|
|
20
|
+
"LICENSE"
|
|
21
|
+
],
|
|
22
|
+
"publishConfig": {
|
|
23
|
+
"access": "public",
|
|
24
|
+
"registry": "https://registry.npmjs.org/"
|
|
25
|
+
},
|
|
26
|
+
"engines": {
|
|
27
|
+
"node": ">=20.11.0"
|
|
28
|
+
},
|
|
29
|
+
"scripts": {
|
|
30
|
+
"build": "node ./scripts/build.mjs",
|
|
31
|
+
"clean": "powershell -NoProfile -Command \"if (Test-Path dist) { Remove-Item -Recurse -Force dist }; if (Test-Path artifacts) { Remove-Item -Recurse -Force artifacts }; if (Test-Path .tmp) { Remove-Item -Recurse -Force .tmp }\"",
|
|
32
|
+
"dev": "tsx src/cli.ts",
|
|
33
|
+
"probe:auth-state": "tsx src/runtime/auth-state-probe.ts",
|
|
34
|
+
"pack:tarball": "node ./scripts/package-release.mjs",
|
|
35
|
+
"prepare": "npm run build",
|
|
36
|
+
"prepublishOnly": "npm test",
|
|
37
|
+
"smoke:packaged": "node ./scripts/smoke-packaged.mjs",
|
|
38
|
+
"start": "node dist/cli.js",
|
|
39
|
+
"test": "node --import tsx --test ./test/*.test.ts",
|
|
40
|
+
"typecheck": "tsc --noEmit -p tsconfig.json"
|
|
41
|
+
},
|
|
42
|
+
"keywords": [
|
|
43
|
+
"codex",
|
|
44
|
+
"cli",
|
|
45
|
+
"wrapper",
|
|
46
|
+
"accounts"
|
|
47
|
+
],
|
|
48
|
+
"author": "Mostok",
|
|
49
|
+
"license": "ISC",
|
|
50
|
+
"devDependencies": {
|
|
51
|
+
"@types/node": "^24.5.2",
|
|
52
|
+
"esbuild": "^0.25.9",
|
|
53
|
+
"tsx": "^4.20.5",
|
|
54
|
+
"typescript": "^5.9.2"
|
|
55
|
+
},
|
|
56
|
+
"dependencies": {
|
|
57
|
+
"@openai/codex": "^0.120.0"
|
|
58
|
+
}
|
|
59
|
+
}
|