triflux 10.4.0 → 10.7.0

package/tui/gemini-profile.mjs

@@ -0,0 +1,299 @@
+#!/usr/bin/env node
+// tui/gemini-profile.mjs — Interactive Gemini Profile Manager
+// Codex config.toml 대칭 구조 — JSON 기반 프로필 CRUD
+import {
+  copyFileSync,
+  existsSync,
+  mkdirSync,
+  readFileSync,
+  writeFileSync,
+} from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import {
+  BOLD,
+  box,
+  CYAN,
+  clear,
+  confirm,
+  DIM,
+  divider,
+  fail,
+  GREEN,
+  info,
+  input,
+  label,
+  ok,
+  onExit,
+  RED,
+  RESET,
+  select,
+  showCursor,
+  table,
+  WHITE,
+  warn,
+  YELLOW,
+} from "./core.mjs";
+
+const GEMINI_DIR = join(homedir(), ".gemini");
+const CONFIG_PATH = join(GEMINI_DIR, "triflux-profiles.json");
+
+const KNOWN_MODELS = [
+  { label: "gemini-3.1-pro-preview", hint: "3.1 Pro — 플래그십" },
+  { label: "gemini-3-flash-preview", hint: "3.0 Flash — 빠른 응답" },
+  { label: "gemini-2.5-pro", hint: "2.5 Pro — 안정" },
+  { label: "gemini-2.5-flash", hint: "2.5 Flash — 경량" },
+  { label: "gemini-2.5-flash-lite", hint: "2.5 Flash Lite — 최경량" },
+  { label: "직접 입력", hint: "" },
+];
+
+const DEFAULT_CONFIG = {
+  model: "gemini-3.1-pro-preview",
+  profiles: {
+    pro31: {
+      model: "gemini-3.1-pro-preview",
+      hint: "3.1 Pro — 플래그십 (1M ctx, 멀티모달)",
+    },
+    flash3: {
+      model: "gemini-3-flash-preview",
+      hint: "3.0 Flash — 빠른 응답, 비용 효율",
+    },
+    pro25: { model: "gemini-2.5-pro", hint: "2.5 Pro — 안정 (추론 강화)" },
+    flash25: { model: "gemini-2.5-flash", hint: "2.5 Flash — 경량 범용" },
+    lite25: { model: "gemini-2.5-flash-lite", hint: "2.5 Flash Lite — 최경량" },
+  },
+};
+
+// ── JSON Config ──
+
+function readConfig() {
+  if (!existsSync(CONFIG_PATH)) return structuredClone(DEFAULT_CONFIG);
+  try {
+    return JSON.parse(readFileSync(CONFIG_PATH, "utf8"));
+  } catch {
+    warn("triflux-profiles.json 파싱 실패, 기본값 사용");
+    return structuredClone(DEFAULT_CONFIG);
+  }
+}
+
+function getProfiles(config) {
+  return Object.entries(config.profiles || {}).map(([name, p]) => ({
+    name,
+    model: p.model || config.model,
+    hint: p.hint || "",
+  }));
+}
+
+// ── UI Flows ──
+
+function showStatus(config) {
+  const profiles = getProfiles(config);
+
+  console.log();
+  label("기본 모델", `${WHITE}${config.model || "미설정"}${RESET}`);
+  console.log();
+
+  if (profiles.length === 0) {
+    warn("등록된 프로필이 없습니다.");
+    return;
+  }
+
+  const headers = ["프로필", "모델", "설명"];
+  const rows = profiles.map((p) => [
+    `${CYAN}${p.name}${RESET}`,
+    modelColor(p.model),
+    p.hint ? `${DIM}${p.hint}${RESET}` : "",
+  ]);
+
+  table(headers, rows);
+}
+
+function modelColor(model) {
+  if (!model) return `${DIM}inherit${RESET}`;
+  if (model.includes("pro")) return `${YELLOW}${model}${RESET}`;
+  if (model.includes("flash-lite")) return `${GREEN}${model}${RESET}`;
+  if (model.includes("flash")) return `${CYAN}${model}${RESET}`;
+  return `${WHITE}${model}${RESET}`;
+}
+
+async function pickModel(current) {
+  const idx = KNOWN_MODELS.findIndex((m) => m.label === current);
+  const choice = await select("모델 선택", KNOWN_MODELS, {
+    initial: Math.max(0, idx),
+  });
+  if (!choice) return null;
+  if (choice.value.label === "직접 입력") {
+    return await input("모델 ID", current || "");
+  }
+  return choice.value.label;
+}
+
+async function editProfile(config) {
+  const profiles = getProfiles(config);
+  if (profiles.length === 0) {
+    warn("편집할 프로필이 없습니다.");
+    return config;
+  }
+
+  const options = profiles.map((p) => ({
+    label: p.name,
+    hint: `${DIM}${p.model}${RESET}`,
+  }));
+
+  const picked = await select("편집할 프로필", options);
+  if (!picked) return config;
+
+  const profile = profiles[picked.index];
+  console.log();
+  info(`현재: ${BOLD}${profile.name}${RESET} → ${profile.model}`);
+
+  const newModel = await pickModel(profile.model);
+  if (newModel === null) return config;
+
+  const hint = await input("설명 (Enter로 유지)", profile.hint);
+
+  console.log();
+  info(`변경: ${profile.model} → ${BOLD}${newModel}${RESET}`);
+
+  if (!(await confirm("저장하시겠습니까?"))) return config;
+
+  config.profiles[profile.name] = {
+    model: newModel,
+    hint: hint || profile.hint,
+  };
+  save(config);
+  ok(`${profile.name} 프로필 저장 완료`);
+  return readConfig();
+}
+
+async function editDefault(config) {
+  info(`현재 기본 모델: ${BOLD}${config.model || "미설정"}${RESET}`);
+
+  const newModel = await pickModel(config.model);
+  if (newModel === null) return config;
+
+  console.log();
+  info(`변경: ${config.model} → ${BOLD}${newModel}${RESET}`);
+
+  if (!(await confirm("저장하시겠습니까?"))) return config;
+
+  config.model = newModel;
+  save(config);
+  ok("기본 모델 저장 완료");
+  return readConfig();
+}
+
+async function addProfile(config) {
+  const name = await input("새 프로필 이름");
+  if (!name) return config;
+
+  if (config.profiles[name]) {
+    fail(`'${name}' 프로필이 이미 존재합니다.`);
+    return config;
+  }
+
+  const model = await pickModel("");
+  if (!model) return config;
+
+  const hint = await input("설명 (선택)", "");
+
+  console.log();
+  info(`추가: ${BOLD}${name}${RESET} → ${model}`);
+  if (!(await confirm("저장하시겠습니까?"))) return config;
+
+  config.profiles[name] = { model, hint };
+  save(config);
+  ok(`${name} 프로필 추가 완료`);
+  return readConfig();
+}
+
+async function removeProfile(config) {
+  const profiles = getProfiles(config);
+  if (profiles.length === 0) {
+    warn("삭제할 프로필이 없습니다.");
+    return config;
+  }
+
+  const options = profiles.map((p) => ({ label: p.name, hint: p.model }));
+  const picked = await select("삭제할 프로필", options);
+  if (!picked) return config;
+
+  const name = profiles[picked.index].name;
+  if (
+    !(await confirm(`${RED}${name}${RESET} 프로필을 삭제하시겠습니까?`, false))
+  ) {
+    return config;
+  }
+
+  delete config.profiles[name];
+  save(config);
+  ok(`${name} 프로필 삭제 완료`);
+  return readConfig();
+}
+
+function save(config) {
+  if (!existsSync(GEMINI_DIR)) mkdirSync(GEMINI_DIR, { recursive: true });
+
+  // Backup before write
+  if (existsSync(CONFIG_PATH)) {
+    copyFileSync(CONFIG_PATH, CONFIG_PATH + ".bak");
+  }
+
+  writeFileSync(CONFIG_PATH, JSON.stringify(config, null, 2) + "\n", "utf8");
+}
+
+// ── Main Loop ──
+
+const MENU = [
+  { label: "프로필 모델 변경", hint: "모델 수정" },
+  { label: "기본 모델 변경", hint: "top-level default" },
+  { label: "프로필 추가", hint: "새 프로필 생성" },
+  { label: "프로필 삭제", hint: "기존 프로필 제거" },
+  { label: "종료", hint: "Ctrl+C" },
+];
+
+async function main() {
+  onExit(() => {});
+  clear();
+
+  let config = readConfig();
+
+  while (true) {
+    box("Gemini Profile Manager", 46);
+    showStatus(config);
+    console.log();
+
+    const choice = await select("작업 선택", MENU);
+    if (!choice || choice.index === 4) {
+      console.log();
+      info("종료합니다.");
+      showCursor();
+      break;
+    }
+
+    console.log();
+    switch (choice.index) {
+      case 0:
+        config = await editProfile(config);
+        break;
+      case 1:
+        config = await editDefault(config);
+        break;
+      case 2:
+        config = await addProfile(config);
+        break;
+      case 3:
+        config = await removeProfile(config);
+        break;
+    }
+
+    console.log();
+    divider(46);
+  }
+}
+
+main().catch((e) => {
+  showCursor();
+  console.error(e);
+  process.exit(1);
+});

package/tui/monitor-data.mjs

@@ -0,0 +1,152 @@
+import { readdirSync, readFileSync, unlinkSync } from "node:fs";
+import { get as httpGet } from "node:http";
+import { join } from "node:path";
+
+const HUB_URL = "http://127.0.0.1:27888";
+const AGENT_FILE_PREFIX = "tfx-agent-";
+const AGENT_FILE_SUFFIX = ".json";
+
+// 임시 디렉터리 경로를 플랫폼별 우선순위로 고른다.
+function getTmpDir(env = process.env) {
+  return env.TMPDIR || env.TEMP || "/tmp";
+}
+
+// 모니터가 읽을 에이전트 메타 파일만 고른다.
+function isAgentFile(name) {
+  return name.startsWith(AGENT_FILE_PREFIX) && name.endsWith(AGENT_FILE_SUFFIX);
+}
+
+// JSON 파싱 실패는 삼키고 null로 처리한다.
+function readAgentRecord(filePath, deps) {
+  try {
+    return JSON.parse(deps.readFileSync(filePath, "utf8"));
+  } catch {
+    return null;
+  }
+}
+
+// PID 생존 여부를 확인한다.
+// Windows에서 process.kill(pid, 0)이 불안정할 수 있으므로
+// 24시간 이상 경과한 에이전트는 좀비로 간주한다.
+const MAX_AGENT_AGE_S = 86400;
+
+function isAlive(pid, deps, startedAt = 0) {
+  const age = Math.floor(deps.now() / 1000) - (Number(startedAt) || 0);
+  if (age > MAX_AGENT_AGE_S) return false;
+  try {
+    deps.kill(pid, 0);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+// 좀비 파일 삭제 실패는 무시한다.
+function removeZombie(filePath, deps) {
+  try {
+    deps.unlinkSync(filePath);
+  } catch {}
+}
+
+// 표시용 경과 시간을 계산한다.
+function toAgentView(record, now) {
+  const started = Number(record.started) || 0;
+  return {
+    pid: Number(record.pid),
+    cli: String(record.cli || ""),
+    agent: String(record.agent || ""),
+    started,
+    elapsed: Math.max(0, now - started),
+    alive: true,
+  };
+}
+
+// /tmp/tfx-agent-*.json 파일에서 살아있는 에이전트 목록을 읽는다.
+function pollAgents(deps = {}) {
+  const resolved = {
+    readdirSync,
+    readFileSync,
+    unlinkSync,
+    kill: process.kill.bind(process),
+    env: process.env,
+    now: Date.now,
+    ...deps,
+  };
+  const dir = getTmpDir(resolved.env);
+  let names = [];
+
+  try {
+    names = resolved.readdirSync(dir).filter(isAgentFile);
+  } catch {
+    return [];
+  }
+
+  const now = Math.floor(resolved.now() / 1000);
+  const agents = [];
+  for (const name of names) {
+    const filePath = join(dir, name);
+    const record = readAgentRecord(filePath, resolved);
+    if (!record) continue;
+    if (!isAlive(Number(record.pid), resolved, record.started)) {
+      removeZombie(filePath, resolved);
+      continue;
+    }
+    agents.push(toAgentView(record, now));
+  }
+  return agents;
+}
+
+// hub /status 경로 URL을 안전하게 만든다.
+function buildStatusUrl(hubUrl = HUB_URL) {
+  return new URL("/status", hubUrl).toString();
+}
+
+// 응답 바디를 JSON으로 해석하고 핵심 필드만 추린다.
+function parseStatus(body) {
+  const parsed = JSON.parse(body);
+  return {
+    online: true,
+    uptime: parsed.uptime,
+    queueDepth: parsed.queueDepth,
+    agents: parsed.agents,
+  };
+}
+
+// node:http.get으로 허브 상태를 2초 안에 조회한다.
+function fetchHubStatus(hubUrl = HUB_URL, deps = {}) {
+  const get = deps.get || httpGet;
+  return new Promise((resolve) => {
+    let settled = false;
+    const finish = (value) => {
+      if (settled) return;
+      settled = true;
+      resolve(value);
+    };
+    try {
+      const req = get(buildStatusUrl(hubUrl), (res) => {
+        let body = "";
+        res.setEncoding?.("utf8");
+        res.on("data", (chunk) => {
+          body += chunk;
+        });
+        res.on("end", () => {
+          try {
+            finish(
+              res.statusCode === 200 ? parseStatus(body) : { online: false },
+            );
+          } catch {
+            finish({ online: false });
+          }
+        });
+        res.on("error", () => finish({ online: false }));
+      });
+
+      req.setTimeout?.(2000, () => req.destroy(new Error("timeout")));
+      req.on?.("error", () => finish({ online: false }));
+    } catch {
+      finish({ online: false });
+    }
+  });
+}
+
+export { fetchHubStatus, pollAgents };