pleumrouter 0.5.26 → 0.5.27

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/bin/pleum.mjs CHANGED
@@ -124,9 +124,11 @@ try {
124
124
  case "models":
125
125
  await listModels(rest);
126
126
  break;
127
- case "balance":
128
- await balance();
127
+ case "balance": {
128
+ const isSwitch = rest.includes("--switch");
129
+ await balance({ interactive: isSwitch, switchOnly: isSwitch });
129
130
  break;
131
+ }
130
132
  case "usage":
131
133
  await usage(rest);
132
134
  break;
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "pleumrouter",
3
- "version": "0.5.26",
3
+ "version": "0.5.27",
4
4
  "description": "Official PleumRouter terminal CLI — launch any coding agent against PleumRouter.",
5
5
  "type": "module",
6
6
  "bin": {
package/src/balance.mjs CHANGED
@@ -1,16 +1,177 @@
1
- import { authFetch } from "./config.mjs";
1
+ import { authFetch, getActiveContext, setActiveContext } from "./config.mjs";
2
+ import { style } from "./ansi.mjs";
3
+ import { select } from "./interactive.mjs";
4
+ import { t } from "./i18n.mjs";
2
5
 
3
- export async function balance() {
6
+ export function formatKrwCredit(amount) {
7
+ return Number(amount || 0).toLocaleString("ko-KR", {
8
+ minimumFractionDigits: 1,
9
+ maximumFractionDigits: 1,
10
+ });
11
+ }
12
+
13
+ export async function fetchCreditSummary() {
4
14
  const res = await authFetch("/credits");
5
15
  if (!res.ok) {
6
16
  const body = await res.text().catch(() => "");
7
- console.error(`잔액 조회 실패 (HTTP ${res.status})${body ? `: ${body}` : ""}`);
17
+ throw new Error(`잔액 조회 실패 (HTTP ${res.status})${body ? `: ${body}` : ""}`);
18
+ }
19
+ const personal = await res.json();
20
+
21
+ let orgs = [];
22
+ try {
23
+ const orgRes = await authFetch("/orgs");
24
+ if (orgRes.ok) {
25
+ orgs = await orgRes.json();
26
+ }
27
+ } catch {}
28
+
29
+ const teamWallets = [];
30
+ if (Array.isArray(orgs) && orgs.length > 0) {
31
+ await Promise.all(
32
+ orgs.map(async (org) => {
33
+ try {
34
+ const wRes = await authFetch(`/orgs/${org.id}/wallet`);
35
+ if (wRes.ok) {
36
+ const wallet = await wRes.json();
37
+ teamWallets.push({
38
+ id: String(org.id),
39
+ name: org.name,
40
+ role: org.role,
41
+ balance_krw: wallet.available_krw ?? wallet.balance_krw ?? 0,
42
+ });
43
+ }
44
+ } catch {}
45
+ })
46
+ );
47
+ }
48
+
49
+ return {
50
+ personal: {
51
+ balance_krw: personal.balance_krw ?? 0,
52
+ balance_usd: personal.balance_usd,
53
+ },
54
+ teams: teamWallets,
55
+ };
56
+ }
57
+
58
+ export function buildCreditOptions({ personal, teams = [], activeContext = { type: "personal" } }) {
59
+ const options = [];
60
+ const isPersonalActive = activeContext.type === "personal" || !activeContext.org_id;
61
+ const personalBalanceStr = `${formatKrwCredit(personal.balance_krw)} ${t("credits")}${
62
+ personal.balance_usd != null ? ` / $${personal.balance_usd.toFixed(4)}` : ""
63
+ }`;
64
+
65
+ options.push({
66
+ id: "personal",
67
+ type: "personal",
68
+ name: t("Personal"),
69
+ title: t("Personal Credit"),
70
+ balance: personalBalanceStr,
71
+ isActive: isPersonalActive,
72
+ label: `${isPersonalActive ? "✓ " : " "}${t("Personal Credit")}: ${personalBalanceStr}`,
73
+ });
74
+
75
+ for (const team of teams) {
76
+ const isTeamActive = activeContext.type === "team" && String(activeContext.org_id) === String(team.id);
77
+ const teamBalanceStr = `${formatKrwCredit(team.balance_krw)} ${t("credits")}`;
78
+ const roleStr = team.role ? ` (${team.role})` : "";
79
+ const teamTitle = `${t("Team")}: ${team.name}${roleStr}`;
80
+ options.push({
81
+ id: String(team.id),
82
+ type: "team",
83
+ name: team.name,
84
+ title: teamTitle,
85
+ balance: teamBalanceStr,
86
+ isActive: isTeamActive,
87
+ label: `${isTeamActive ? "✓ " : " "}${teamTitle}: ${teamBalanceStr}`,
88
+ });
89
+ }
90
+
91
+ return options;
92
+ }
93
+
94
+ export async function switchCreditWorkspace({ summary, selectFn = select } = {}) {
95
+ const data = summary || (await fetchCreditSummary());
96
+ const active = getActiveContext();
97
+ const options = buildCreditOptions({
98
+ personal: data.personal,
99
+ teams: data.teams,
100
+ activeContext: active,
101
+ });
102
+
103
+ if (options.length <= 1) {
104
+ console.log(t("You are not a member of any team. Using personal credit."));
105
+ return active;
106
+ }
107
+
108
+ const picked = await selectFn(options, {
109
+ message: t("Select credit workspace"),
110
+ formatItem: (x) => x.label,
111
+ limit: 10,
112
+ });
113
+
114
+ if (!picked) return active;
115
+
116
+ const targetOrgId = picked.type === "team" ? picked.id : null;
117
+
118
+ try {
119
+ const res = await authFetch("/auth/cli-context", {
120
+ method: "POST",
121
+ headers: { "Content-Type": "application/json" },
122
+ body: JSON.stringify({ org_id: targetOrgId }),
123
+ });
124
+ if (!res.ok) {
125
+ const err = await res.text().catch(() => "");
126
+ console.error(`✗ ${t("Failed to update credit context")}: ${err}`);
127
+ return active;
128
+ }
129
+ } catch (e) {
130
+ console.error(`✗ ${t("Failed to update credit context")}: ${e.message}`);
131
+ return active;
132
+ }
133
+
134
+ const newContext = setActiveContext({
135
+ type: picked.type,
136
+ org_id: targetOrgId,
137
+ org_name: picked.type === "team" ? picked.name : null,
138
+ });
139
+
140
+ console.log(`✓ ${t("Active credit set to: {name}", { name: picked.title })}`);
141
+ return newContext;
142
+ }
143
+
144
+ export async function balance({ interactive = false, switchOnly = false } = {}) {
145
+ let summary;
146
+ try {
147
+ summary = await fetchCreditSummary();
148
+ } catch (e) {
149
+ console.error(e.message);
8
150
  process.exit(1);
9
151
  }
10
- const data = await res.json();
11
- const plm = data.balance_krw;
12
- const usd = data.balance_usd != null ? ` / $${data.balance_usd.toFixed(4)}` : "";
13
- // 원장 눈금이 0.1이라 잔액은 소수 1자리 고정 표기(딱 떨어져도 ".0"을 남긴다).
14
- const shown = plm.toLocaleString("ko-KR", { minimumFractionDigits: 1, maximumFractionDigits: 1 });
15
- console.log(`잔액: ${shown} 크레딧${usd}`);
152
+
153
+ if (switchOnly) {
154
+ return await switchCreditWorkspace({ summary });
155
+ }
156
+
157
+ const active = getActiveContext();
158
+ const options = buildCreditOptions({
159
+ personal: summary.personal,
160
+ teams: summary.teams,
161
+ activeContext: active,
162
+ });
163
+
164
+ if (!interactive || options.length <= 1) {
165
+ console.log(t("Credit Balance:"));
166
+ for (const opt of options) {
167
+ const mark = opt.isActive ? style.bold(style.plum("▸ [Active] ")) : " [ ] ";
168
+ console.log(`${mark}${opt.title}: ${opt.balance}`);
169
+ }
170
+ if (options.length > 1 && !interactive) {
171
+ console.log(`\n${style.dim(t("Run `pleum balance --switch` to switch credit workspace."))}`);
172
+ }
173
+ return;
174
+ }
175
+
176
+ return await switchCreditWorkspace({ summary });
16
177
  }
package/src/config.mjs CHANGED
@@ -194,11 +194,41 @@ export async function refreshAccessToken(cfg = readConfig()) {
194
194
  }
195
195
  }
196
196
 
197
- /** Bearer 요청 — 401이면 1회 refresh 후 재시도. */
197
+ /** 활성 크레딧/워크스페이스 컨텍스트 조회 (기본값 personal) */
198
+ export function getActiveContext(cfg = readConfig()) {
199
+ const ctx = cfg?.active_context;
200
+ if (ctx && ctx.type === "team" && ctx.org_id) {
201
+ return { type: "team", org_id: ctx.org_id, org_name: ctx.org_name || ctx.org_id };
202
+ }
203
+ return { type: "personal", org_id: null, org_name: null };
204
+ }
205
+
206
+ /** 활성 크레딧/워크스페이스 컨텍스트 저장 */
207
+ export function setActiveContext(context, cfg = readConfig()) {
208
+ const updated = { ...(cfg ?? {}) };
209
+ if (!context || context.type === "personal" || !context.org_id) {
210
+ updated.active_context = { type: "personal" };
211
+ } else {
212
+ updated.active_context = {
213
+ type: "team",
214
+ org_id: context.org_id,
215
+ org_name: context.org_name || null,
216
+ };
217
+ }
218
+ writeConfig(updated);
219
+ return updated.active_context;
220
+ }
221
+
222
+ /** Bearer 요청 — 401이면 1회 refresh 후 재시도. 활성 팀 컨텍스트가 있으면 X-Pleum-Org 첨부. */
198
223
  export async function authFetch(path, init = {}) {
199
224
  const token = await requireAuth();
200
225
  const url = path.startsWith("http") ? path : `${V1}${path.startsWith("/") ? path : `/${path}`}`;
201
- const headers = { ...(init.headers || {}), Authorization: `Bearer ${token}` };
226
+ const ctx = getActiveContext();
227
+ const customHeaders = {};
228
+ if (ctx.type === "team" && ctx.org_id) {
229
+ customHeaders["X-Pleum-Org"] = ctx.org_id;
230
+ }
231
+ const headers = { ...customHeaders, ...(init.headers || {}), Authorization: `Bearer ${token}` };
202
232
  let res = await fetch(url, { ...init, headers });
203
233
  if (res.status !== 401) return res;
204
234
 
@@ -207,6 +237,7 @@ export async function authFetch(path, init = {}) {
207
237
  if (!refreshed) return res;
208
238
  return fetch(url, {
209
239
  ...init,
210
- headers: { ...(init.headers || {}), Authorization: `Bearer ${refreshed}` },
240
+ headers: { ...customHeaders, ...(init.headers || {}), Authorization: `Bearer ${refreshed}` },
211
241
  });
212
242
  }
243
+
package/src/launch.mjs CHANGED
@@ -1,8 +1,8 @@
1
1
  import { spawn } from "node:child_process";
2
- import { mkdirSync, writeFileSync, existsSync, readFileSync } from "node:fs";
2
+ import { mkdirSync, writeFileSync, existsSync, readFileSync, rmSync } from "node:fs";
3
3
  import { homedir } from "node:os";
4
4
  import { join, delimiter, dirname, resolve } from "node:path";
5
- import { requireAuth } from "./config.mjs";
5
+ import { requireAuth, getActiveContext } from "./config.mjs";
6
6
  import {
7
7
  RECIPES,
8
8
  MANUAL,
@@ -95,6 +95,8 @@ export function launchHelp() {
95
95
 
96
96
  // exit:false면 process.exit 대신 자식 종료 코드를 반환 — 런처 메뉴의 실행 후 복귀용.
97
97
  export async function launch(tool, model, passthrough, { exit = true } = {}) {
98
+ recoverAgySettings();
99
+
98
100
  const key = await requireAuth();
99
101
  const resolved = resolveLaunchTool(tool);
100
102
  const r = RECIPES[resolved];
@@ -126,7 +128,41 @@ export async function launch(tool, model, passthrough, { exit = true } = {}) {
126
128
  if (r.type === "crush-home") {
127
129
  env.CRUSH_GLOBAL_CONFIG = writeIsolatedHome("crush-home", "crush.json", crushJson(model));
128
130
  }
129
- if (r.type === "agy-settings") ensureAgySettings(model);
131
+
132
+ let restoreAgy = null;
133
+ if (r.type === "agy-settings") {
134
+ restoreAgy = setupAgySettings(model);
135
+ }
136
+
137
+ const cleanup = () => {
138
+ if (restoreAgy) {
139
+ restoreAgy();
140
+ restoreAgy = null;
141
+ }
142
+ };
143
+
144
+ const onSignal = () => {
145
+ cleanup();
146
+ if (exit) process.exit(130);
147
+ };
148
+
149
+ process.once("SIGINT", onSignal);
150
+ process.once("SIGTERM", onSignal);
151
+ process.once("SIGHUP", onSignal);
152
+ process.once("exit", cleanup);
153
+
154
+ const removeListeners = () => {
155
+ process.removeListener("SIGINT", onSignal);
156
+ process.removeListener("SIGTERM", onSignal);
157
+ process.removeListener("SIGHUP", onSignal);
158
+ process.removeListener("exit", cleanup);
159
+ };
160
+
161
+ const activeCtx = getActiveContext();
162
+ const creditMsg = activeCtx.type === "team"
163
+ ? `${t("Team")}: ${activeCtx.org_name}`
164
+ : t("Personal");
165
+ console.log(`ℹ ${t("Credit:")} ${creditMsg}`);
130
166
 
131
167
  const args = [...r.args(model), ...passthrough];
132
168
  if (r.note) console.log(`ℹ ${t(r.note)}`);
@@ -136,6 +172,8 @@ export async function launch(tool, model, passthrough, { exit = true } = {}) {
136
172
  try {
137
173
  command = resolveLaunchCommand(r.bin, args);
138
174
  } catch (e) {
175
+ cleanup();
176
+ removeListeners();
139
177
  console.error(`✗ ${sanitizeTerminalText(e.message)}`);
140
178
  if (exit) process.exit(1);
141
179
  return 1;
@@ -148,6 +186,8 @@ export async function launch(tool, model, passthrough, { exit = true } = {}) {
148
186
  shell: false,
149
187
  });
150
188
  child.on("error", (e) => {
189
+ cleanup();
190
+ removeListeners();
151
191
  if (e.code === "ENOENT") {
152
192
  console.error(`✗ ${t("'{bin}' not found. Install:\n {install}", { bin: r.bin, install: r.install })}`);
153
193
  if (exit) process.exit(127);
@@ -158,6 +198,8 @@ export async function launch(tool, model, passthrough, { exit = true } = {}) {
158
198
  resolve(1);
159
199
  });
160
200
  child.on("exit", (code) => {
201
+ cleanup();
202
+ removeListeners();
161
203
  if (exit) process.exit(code ?? 0);
162
204
  resolve(code ?? 0);
163
205
  });
@@ -171,12 +213,142 @@ function writeIsolatedHome(name, filename, contents) {
171
213
  return home;
172
214
  }
173
215
 
174
- function ensureAgySettings(model) {
175
- const dir = join(homedir(), ".gemini", "antigravity-cli");
176
- const path = join(dir, "settings.json");
177
- mkdirSync(dir, { recursive: true, mode: 0o700 });
178
- const existing = existsSync(path) ? readFileSync(path, "utf8") : "";
179
- writeFileSync(path, `${JSON.stringify(mergeAgySettings(existing, model), null, 2)}\n`, { mode: 0o600 });
216
+ export function getAgyPaths() {
217
+ const settingsPath = join(homedir(), ".gemini", "antigravity-cli", "settings.json");
218
+ const backupPath = join(homedir(), ".pleum", "agy-settings-backup.json");
219
+ return { settingsPath, backupPath };
220
+ }
221
+
222
+ export function setupAgySettings(model, {
223
+ paths = getAgyPaths(),
224
+ read = readFileSync,
225
+ write = writeFileSync,
226
+ exists = existsSync,
227
+ mkdir = mkdirSync,
228
+ rm = rmSync,
229
+ } = {}) {
230
+ const { settingsPath, backupPath } = paths;
231
+ const originalExists = exists(settingsPath);
232
+ let originalContent = null;
233
+ let parsed = null;
234
+
235
+ if (originalExists) {
236
+ try {
237
+ originalContent = read(settingsPath, "utf8");
238
+ parsed = JSON.parse(originalContent);
239
+ } catch {
240
+ parsed = null;
241
+ }
242
+ }
243
+
244
+ // 1. 비정상 종료 대비 백업 메타데이터 저장
245
+ try {
246
+ mkdir(dirname(backupPath), { recursive: true, mode: 0o700 });
247
+ write(
248
+ backupPath,
249
+ JSON.stringify(
250
+ {
251
+ settingsPath,
252
+ originalExists,
253
+ originalHadModelProvider: Boolean(parsed && "modelProvider" in parsed),
254
+ originalModelProvider: parsed?.modelProvider ?? null,
255
+ originalHadModel: Boolean(parsed && "model" in parsed),
256
+ originalModel: parsed?.model ?? null,
257
+ originalContent,
258
+ },
259
+ null,
260
+ 2,
261
+ ),
262
+ { mode: 0o600 },
263
+ );
264
+ } catch {}
265
+
266
+ // 2. 임시 설정 병합
267
+ mkdir(dirname(settingsPath), { recursive: true, mode: 0o700 });
268
+ write(settingsPath, `${JSON.stringify(mergeAgySettings(originalContent || "", model), null, 2)}\n`, { mode: 0o600 });
269
+
270
+ // 3. 종료 시 스마트 복구 클로저 (세션 중 사용자가 추가한 권한/워크스페이스 보존)
271
+ let restored = false;
272
+ return function restore() {
273
+ if (restored) return;
274
+ restored = true;
275
+ try {
276
+ if (!originalExists) {
277
+ if (exists(settingsPath)) rm(settingsPath, { force: true });
278
+ } else {
279
+ const currentContent = exists(settingsPath) ? read(settingsPath, "utf8") : null;
280
+ if (currentContent) {
281
+ try {
282
+ const currentObj = JSON.parse(currentContent);
283
+ if (parsed && "modelProvider" in parsed) {
284
+ currentObj.modelProvider = parsed.modelProvider;
285
+ } else {
286
+ delete currentObj.modelProvider;
287
+ }
288
+ if (parsed && "model" in parsed) {
289
+ currentObj.model = parsed.model;
290
+ } else {
291
+ delete currentObj.model;
292
+ }
293
+ write(settingsPath, `${JSON.stringify(currentObj, null, 2)}\n`, { mode: 0o600 });
294
+ } catch {
295
+ if (originalContent != null) write(settingsPath, originalContent, { mode: 0o600 });
296
+ }
297
+ } else if (originalContent != null) {
298
+ write(settingsPath, originalContent, { mode: 0o600 });
299
+ }
300
+ }
301
+ } catch {
302
+ // 복구 실패 시 최선 처리
303
+ } finally {
304
+ try {
305
+ if (exists(backupPath)) rm(backupPath, { force: true });
306
+ } catch {}
307
+ }
308
+ };
309
+ }
310
+
311
+ export function recoverAgySettings({
312
+ paths = getAgyPaths(),
313
+ read = readFileSync,
314
+ write = writeFileSync,
315
+ exists = existsSync,
316
+ rm = rmSync,
317
+ } = {}) {
318
+ const { backupPath } = paths;
319
+ if (!exists(backupPath)) return;
320
+ try {
321
+ const raw = read(backupPath, "utf8");
322
+ const backup = JSON.parse(raw);
323
+ const settingsPath = backup?.settingsPath;
324
+ if (settingsPath && exists(settingsPath)) {
325
+ if (!backup.originalExists) {
326
+ rm(settingsPath, { force: true });
327
+ } else {
328
+ const curContent = read(settingsPath, "utf8");
329
+ try {
330
+ const curObj = JSON.parse(curContent);
331
+ if (backup.originalHadModelProvider) {
332
+ curObj.modelProvider = backup.originalModelProvider;
333
+ } else {
334
+ delete curObj.modelProvider;
335
+ }
336
+ if (backup.originalHadModel) {
337
+ curObj.model = backup.originalModel;
338
+ } else {
339
+ delete curObj.model;
340
+ }
341
+ write(settingsPath, `${JSON.stringify(curObj, null, 2)}\n`, { mode: 0o600 });
342
+ } catch {
343
+ if (backup.originalContent != null) write(settingsPath, backup.originalContent, { mode: 0o600 });
344
+ }
345
+ }
346
+ }
347
+ } catch {} finally {
348
+ try {
349
+ if (exists(backupPath)) rm(backupPath, { force: true });
350
+ } catch {}
351
+ }
180
352
  }
181
353
 
182
354
  function printManual(tool, key) {
@@ -186,3 +358,4 @@ function printManual(tool, key) {
186
358
  console.log(m.snippet());
187
359
  console.log(`\n${t("2) Export the key and run:")}\n export PLEUM_API_KEY=${key}\n`);
188
360
  }
361
+
@@ -262,5 +262,17 @@ export const ES = {
262
262
  "✓ installed": "✓ instalado",
263
263
  "✗ not installed —": "✗ no instalado —",
264
264
  "Log out": "Cerrar sesión",
265
+ "Credit:": "Crédito:",
266
+ "Team": "Equipo",
267
+ "Personal": "Personal",
268
+ "Personal Credit": "Crédito personal",
269
+ "Credit workspace": "Espacio de crédito",
270
+ "Credit Balance:": "Saldo de crédito:",
271
+ "credits": "créditos",
272
+ "Select credit workspace": "Seleccionar espacio de crédito",
273
+ "Active credit set to: {name}": "Crédito activo configurado en: {name}",
274
+ "Failed to update credit context": "Error al actualizar el contexto de crédito",
275
+ "You are not a member of any team. Using personal credit.": "No eres miembro de ningún equipo. Usando crédito personal.",
276
+ "Run `pleum balance --switch` to switch credit workspace.": "Ejecuta `pleum balance --switch` para cambiar de espacio de crédito.",
265
277
  };
266
278
 
@@ -262,5 +262,17 @@ export const JA = {
262
262
  "✓ installed": "✓ インストール済み",
263
263
  "✗ not installed —": "✗ 未インストール —",
264
264
  "Log out": "ログアウト",
265
+ "Credit:": "クレジット:",
266
+ "Team": "チーム",
267
+ "Personal": "個人",
268
+ "Personal Credit": "個人クレジット",
269
+ "Credit workspace": "クレジットワークスペース",
270
+ "Credit Balance:": "クレジット残高:",
271
+ "credits": "クレジット",
272
+ "Select credit workspace": "クレジットワークスペースを選択",
273
+ "Active credit set to: {name}": "有効なクレジットを {name} に設定しました。",
274
+ "Failed to update credit context": "クレジットコンテキストの更新に失敗しました",
275
+ "You are not a member of any team. Using personal credit.": "所属しているチームがありません。個人クレジットを使用します。",
276
+ "Run `pleum balance --switch` to switch credit workspace.": "`pleum balance --switch` でクレジットワークスペースを変更できます。",
265
277
  };
266
278
 
@@ -325,5 +325,17 @@ export const KO = {
325
325
  "↑/↓ move • enter select • esc back": "↑/↓ 이동 • enter 선택 • esc 뒤로",
326
326
  "↑/↓ move • enter set model • esc back": "↑/↓ 이동 • enter 모델 설정 • esc 뒤로",
327
327
  "Log out": "로그아웃",
328
+ "Credit:": "크레딧:",
329
+ "Team": "팀",
330
+ "Personal": "개인",
331
+ "Personal Credit": "개인 크레딧",
332
+ "Credit workspace": "크레딧 워크스페이스",
333
+ "Credit Balance:": "크레딧 잔액:",
334
+ "credits": "크레딧",
335
+ "Select credit workspace": "크레딧 워크스페이스 선택",
336
+ "Active credit set to: {name}": "활성 크레딧이 {name}으로 설정되었습니다.",
337
+ "Failed to update credit context": "크레딧 컨텍스트 변경 실패",
338
+ "You are not a member of any team. Using personal credit.": "소속된 팀이 없습니다. 개인 크레딧을 사용합니다.",
339
+ "Run `pleum balance --switch` to switch credit workspace.": "`pleum balance --switch` 명령으로 크레딧 워크스페이스를 변경할 수 있습니다.",
328
340
  };
329
341
 
@@ -262,5 +262,17 @@ export const ZH = {
262
262
  "✓ installed": "✓ 已安装",
263
263
  "✗ not installed —": "✗ 未安装 —",
264
264
  "Log out": "登出",
265
+ "Credit:": "额度:",
266
+ "Team": "团队",
267
+ "Personal": "个人",
268
+ "Personal Credit": "个人额度",
269
+ "Credit workspace": "额度工作区",
270
+ "Credit Balance:": "额度余额:",
271
+ "credits": "额度",
272
+ "Select credit workspace": "选择额度工作区",
273
+ "Active credit set to: {name}": "已将有效额度设置为: {name}",
274
+ "Failed to update credit context": "更新额度上下文失败",
275
+ "You are not a member of any team. Using personal credit.": "您不属于任何团队。使用个人额度。",
276
+ "Run `pleum balance --switch` to switch credit workspace.": "运行 `pleum balance --switch` 可切换额度工作区。",
265
277
  };
266
278
 
package/src/menu.mjs CHANGED
@@ -13,7 +13,7 @@ import { spawn } from "node:child_process";
13
13
  import { createInterface } from "node:readline";
14
14
  import { style, displayWidth, terminalWidth, hyperlink, sanitizeTerminalText } from "./ansi.mjs";
15
15
  import { keyFeeder, select } from "./interactive.mjs";
16
- import { readConfig, writeConfig, isLoggedIn, touchSession } from "./config.mjs";
16
+ import { readConfig, writeConfig, isLoggedIn, touchSession, getActiveContext } from "./config.mjs";
17
17
  import { RECIPES, MANUAL, DEFAULT_MODEL } from "./recipes.mjs";
18
18
  import { onPath, launch } from "./launch.mjs";
19
19
  import { fetchModels, matchModels, listModels, formatModelLabel } from "./models.mjs";
@@ -56,7 +56,7 @@ const autoInstallable = (r) => !/^https?:/.test(r.install); // URL이면 수동
56
56
  // 최상위 메뉴 항목(순수 함수 — 테스트 대상). 에이전트는 없음 — "agents" 한 항목으로 서브메뉴 진입.
57
57
  // 로그인 항목도 없다 — CLI는 로그인 사용자용이라 mainMenu()가 진입 시점에 게이트로 처리하고,
58
58
  // 로그인/로그아웃 전환은 설정>계정이 소유한다.
59
- export function buildItems({ savedModels = {} }) {
59
+ export function buildItems({ savedModels = {}, activeContext = { type: "personal" } }) {
60
60
  const items = [];
61
61
  items.push({
62
62
  id: "chat",
@@ -65,11 +65,14 @@ export function buildItems({ savedModels = {} }) {
65
65
  desc: `${t("Start an interactive chat")}${savedModels.chat ? ` · ${t("model")}: ${savedModels.chat}` : ""}`,
66
66
  });
67
67
  items.push({ id: "agents", title: t("Run an agent"), desc: t("Pick a coding agent to run through PleumRouter") });
68
+ const contextDesc = activeContext?.type === "team" && activeContext.org_name
69
+ ? `${t("Team")}: ${activeContext.org_name}`
70
+ : t("Personal");
68
71
  items.push(
69
72
  { id: "models", title: t("Model list"), desc: t("All models — price and context included") },
70
73
  { id: "top", title: t("Leaderboard"), desc: t("Model rankings · provider share") },
71
74
  { id: "usage", title: t("Usage report"), desc: t("Spend over the last 30 days") },
72
- { id: "balance", title: t("Balance"), desc: t("Check your credit balance") },
75
+ { id: "balance", title: t("Balance"), desc: `${t("Check your credit balance")} · ${contextDesc}` },
73
76
  { id: "health", title: t("Health check"), desc: t("Gateway connectivity · login status") },
74
77
  { id: "settings", title: t("Settings"), desc: t("Language, default models, account, updates") },
75
78
  );
@@ -275,7 +278,10 @@ export async function runList(items, { header, footer, initialCursor = 0 } = {})
275
278
  async function menuSelect() {
276
279
  const version = JSON.parse(readFileSync(new URL("../package.json", import.meta.url), "utf8")).version;
277
280
  const cfg = readConfig();
278
- const items = buildItems({ savedModels: cfg?.launch_models ?? {} });
281
+ const items = buildItems({
282
+ savedModels: cfg?.launch_models ?? {},
283
+ activeContext: getActiveContext(cfg),
284
+ });
279
285
  const last = cfg?.last_selection;
280
286
  const initialCursor = last ? Math.max(0, items.findIndex((it) => it.id === last)) : 0;
281
287
  return runList(items, {
@@ -377,7 +383,7 @@ export async function mainMenu() {
377
383
  await usage([]);
378
384
  continue;
379
385
  case "balance":
380
- await balance();
386
+ await balance({ interactive: true });
381
387
  continue;
382
388
  case "health":
383
389
  await health();
package/src/settings.mjs CHANGED
@@ -8,7 +8,7 @@
8
8
  import { readFileSync } from "node:fs";
9
9
  import { style, sanitizeTerminalText } from "./ansi.mjs";
10
10
  import { select, confirm } from "./interactive.mjs";
11
- import { readConfig, writeConfig, isLoggedIn } from "./config.mjs";
11
+ import { readConfig, writeConfig, isLoggedIn, getActiveContext } from "./config.mjs";
12
12
  import { getLocale, LANGS, t } from "./i18n.mjs";
13
13
  import { RECIPES } from "./recipes.mjs";
14
14
  import { fetchModels, matchModels, formatModelLabel } from "./models.mjs";
@@ -16,13 +16,18 @@ import { login, logout } from "./login.mjs";
16
16
  import { upgrade } from "./upgrade.mjs";
17
17
  import { completion } from "./completion.mjs";
18
18
  import { lang } from "./lang.mjs";
19
+ import { switchCreditWorkspace } from "./balance.mjs";
19
20
  import { runList, saveModel } from "./menu.mjs";
20
21
 
21
22
  // Settings 서브메뉴 항목(순수 함수 — 테스트 대상).
22
- export function buildSettingsItems({ lang, loggedIn, keyPrefix, version }) {
23
+ export function buildSettingsItems({ lang, loggedIn, keyPrefix, version, activeContext = { type: "personal" } }) {
24
+ const contextDesc = activeContext?.type === "team" && activeContext.org_name
25
+ ? `${t("Team")}: ${activeContext.org_name}`
26
+ : t("Personal Credit");
23
27
  return [
24
28
  { id: "settings:lang", title: t("Language"), desc: `${t("current")}: ${LANGS[lang] ?? lang}` },
25
29
  { id: "settings:models", title: t("Default models"), desc: t("Set the default model used per agent") },
30
+ { id: "settings:context", title: t("Credit workspace"), desc: contextDesc },
26
31
  {
27
32
  id: "settings:account",
28
33
  // 제목 자체가 다음 동작을 말한다 — 로그인 상태에서 Enter는 로그아웃이다.
@@ -73,6 +78,7 @@ async function settingsSelect() {
73
78
  loggedIn: isLoggedIn(cfg),
74
79
  keyPrefix: cfg?.access_token ? "oauth" : cfg?.key ? cfg.key.slice(0, 8) : null,
75
80
  version: readVersion(),
81
+ activeContext: getActiveContext(cfg),
76
82
  });
77
83
  return runList(items, {
78
84
  header: style.bold(style.plum(t("Settings"))),
@@ -152,6 +158,9 @@ export async function runSettings() {
152
158
  if (r === "quit") return "quit";
153
159
  continue;
154
160
  }
161
+ case "settings:context":
162
+ await switchCreditWorkspace();
163
+ continue;
155
164
  case "settings:account":
156
165
  await runAccount();
157
166
  continue;