triflux 10.3.2 → 10.3.4

package/scripts/lib/mcp-server-catalog.mjs

@@ -1,118 +1,118 @@
-export const SEARCH_SERVER_ORDER = Object.freeze(['brave-search', 'tavily', 'exa']);
-
-export const MCP_SERVER_TOOL_CATALOG = Object.freeze({
-  context7: Object.freeze(['resolve-library-id', 'query-docs']),
-  'brave-search': Object.freeze(['brave_web_search', 'brave_news_search']),
-  exa: Object.freeze(['web_search_exa', 'get_code_context_exa']),
-  tavily: Object.freeze(['tavily_search', 'tavily_extract']),
-  playwright: Object.freeze([
-    'browser_navigate',
-    'browser_navigate_back',
-    'browser_snapshot',
-    'browser_take_screenshot',
-    'browser_wait_for',
-  ]),
-  'sequential-thinking': Object.freeze(['sequentialthinking']),
-});
-
-export const MCP_SERVER_DOMAIN_TAGS = Object.freeze({
-  context7: Object.freeze(['docs', 'reference', 'api', 'sdk', 'library']),
-  'brave-search': Object.freeze(['web', 'search', 'news', 'current']),
-  exa: Object.freeze(['code', 'repository', 'examples', 'search']),
-  tavily: Object.freeze(['research', 'search', 'news', 'verification', 'current']),
-  playwright: Object.freeze(['browser', 'ui', 'visual', 'e2e']),
-  'sequential-thinking': Object.freeze(['analysis', 'planning', 'reasoning', 'security', 'review']),
-});
-
-export const DOMAIN_TAG_KEYWORDS = Object.freeze({
-  docs: Object.freeze(['docs', 'documentation', 'manual', 'guide', '문서', '가이드', '매뉴얼']),
-  reference: Object.freeze(['reference', 'spec', 'schema', 'official', '레퍼런스', '공식', '스펙', '스키마']),
-  api: Object.freeze(['api', 'endpoint', 'interface', 'sdk', '호출', '엔드포인트']),
-  sdk: Object.freeze(['sdk', 'library', 'package', 'framework', '라이브러리', '패키지', '프레임워크']),
-  library: Object.freeze(['library', 'package', 'framework', 'module', '라이브러리', '패키지', '모듈']),
-  web: Object.freeze(['web', 'site', 'article', 'forum', 'blog', 'reddit', '웹', '사이트', '기사', '포럼', '블로그']),
-  search: Object.freeze(['search', 'browse', 'lookup', 'find', '검색', '조회', '탐색', '찾아']),
-  news: Object.freeze(['latest', 'recent', 'news', 'today', 'release', 'announcement', '최신', '최근', '뉴스', '오늘', '릴리즈', '공지']),
-  current: Object.freeze(['current', 'status', 'pricing', 'changelog', 'up-to-date', '현재', '상태', '가격', '변경사항']),
-  research: Object.freeze(['research', 'verify', 'fact-check', 'investigate', '리서치', '검증', '조사']),
-  verification: Object.freeze(['verify', 'validation', 'fact-check', 'audit', '검증', '확인', '감사']),
-  code: Object.freeze(['code', 'repo', 'repository', 'source', 'implementation', 'bug', 'fix', 'test', 'snippet', 'cli', '코드', '리포', '저장소', '구현', '버그', '테스트', '예제', '스크립트']),
-  repository: Object.freeze(['repo', 'repository', 'source', 'git', 'github', '리포', '저장소', '소스']),
-  examples: Object.freeze(['example', 'examples', 'snippet', 'sample', '예제', '샘플']),
-  browser: Object.freeze(['browser', 'page', 'dom', 'screenshot', 'render', '브라우저', '페이지', '스크린샷', '렌더']),
-  ui: Object.freeze(['ui', 'ux', 'layout', 'responsive', 'css', 'html', '디자인', '레이아웃', '반응형']),
-  visual: Object.freeze(['visual', 'screenshot', 'layout', 'render', 'screen', '화면', '시각', '스크린샷']),
-  e2e: Object.freeze(['playwright', 'e2e', 'click', 'navigate', 'automation', 'playwright', '클릭', '이동', '자동화']),
-  analysis: Object.freeze(['analysis', 'analyze', 'audit', 'compare', 'root cause', '분석', '검토', '비교', '원인']),
-  planning: Object.freeze(['plan', 'planning', 'strategy', 'design', '계획', '전략', '설계']),
-  reasoning: Object.freeze(['reason', 'reasoning', 'think', 'critique', '추론', '사고', '비평']),
-  security: Object.freeze(['security', 'risk', 'threat', 'vulnerability', '보안', '위험', '취약점']),
-  review: Object.freeze(['review', 'reviewer', 'inspect', '리뷰', '검수']),
-});
-
-export const SERVER_EXPLICIT_KEYWORDS = Object.freeze({
-  context7: Object.freeze(['context7']),
-  'brave-search': Object.freeze(['brave', 'brave-search']),
-  exa: Object.freeze(['exa']),
-  tavily: Object.freeze(['tavily']),
-  playwright: Object.freeze(['playwright']),
-  'sequential-thinking': Object.freeze(['sequential-thinking', 'sequential thinking']),
-});
-
-export function uniqueStrings(values = []) {
-  return [...new Set(
-    values
-      .filter((value) => typeof value === 'string' && value.trim())
-      .map((value) => value.trim()),
-  )];
-}
-
-export function inferDomainTagsFromText(text = '') {
-  if (typeof text !== 'string' || !text.trim()) return [];
-  const normalized = text.toLocaleLowerCase();
-  const matched = [];
-
-  for (const [tag, keywords] of Object.entries(DOMAIN_TAG_KEYWORDS)) {
-    if (keywords.some((keyword) => normalized.includes(String(keyword).toLocaleLowerCase()))) {
-      matched.push(tag);
-    }
-  }
-
-  return uniqueStrings(matched);
-}
-
-export function getDefaultServerMetadata(serverName = '') {
-  const toolCount = MCP_SERVER_TOOL_CATALOG[serverName]?.length || 0;
-  const domainTags = uniqueStrings([
-    ...(MCP_SERVER_DOMAIN_TAGS[serverName] || []),
-    ...inferDomainTagsFromText(serverName),
-  ]);
-
-  return {
-    tool_count: toolCount,
-    domain_tags: domainTags,
-  };
-}
-
-export function normalizeServerMetadata(serverName = '', metadata = {}) {
-  const fallback = getDefaultServerMetadata(serverName);
-  const toolCount = Number.isFinite(metadata.tool_count)
-    ? Math.max(0, Math.trunc(metadata.tool_count))
-    : fallback.tool_count;
-  const domainTags = uniqueStrings([
-    ...fallback.domain_tags,
-    ...(Array.isArray(metadata.domain_tags) ? metadata.domain_tags : []),
-    ...inferDomainTagsFromText([
-      serverName,
-      typeof metadata.command === 'string' ? metadata.command : '',
-      typeof metadata.url === 'string' ? metadata.url : '',
-      ...(Array.isArray(metadata.args) ? metadata.args : []),
-      ...(metadata.env && typeof metadata.env === 'object' ? Object.keys(metadata.env) : []),
-    ].join(' ')),
-  ]);
-
-  return {
-    tool_count: toolCount,
-    domain_tags: domainTags,
-  };
-}
+export const SEARCH_SERVER_ORDER = Object.freeze(['brave-search', 'tavily', 'exa']);
+
+export const MCP_SERVER_TOOL_CATALOG = Object.freeze({
+  context7: Object.freeze(['resolve-library-id', 'query-docs']),
+  'brave-search': Object.freeze(['brave_web_search', 'brave_news_search']),
+  exa: Object.freeze(['web_search_exa', 'get_code_context_exa']),
+  tavily: Object.freeze(['tavily_search', 'tavily_extract']),
+  playwright: Object.freeze([
+    'browser_navigate',
+    'browser_navigate_back',
+    'browser_snapshot',
+    'browser_take_screenshot',
+    'browser_wait_for',
+  ]),
+  'sequential-thinking': Object.freeze(['sequentialthinking']),
+});
+
+export const MCP_SERVER_DOMAIN_TAGS = Object.freeze({
+  context7: Object.freeze(['docs', 'reference', 'api', 'sdk', 'library']),
+  'brave-search': Object.freeze(['web', 'search', 'news', 'current']),
+  exa: Object.freeze(['code', 'repository', 'examples', 'search']),
+  tavily: Object.freeze(['research', 'search', 'news', 'verification', 'current']),
+  playwright: Object.freeze(['browser', 'ui', 'visual', 'e2e']),
+  'sequential-thinking': Object.freeze(['analysis', 'planning', 'reasoning', 'security', 'review']),
+});
+
+export const DOMAIN_TAG_KEYWORDS = Object.freeze({
+  docs: Object.freeze(['docs', 'documentation', 'manual', 'guide', '문서', '가이드', '매뉴얼']),
+  reference: Object.freeze(['reference', 'spec', 'schema', 'official', '레퍼런스', '공식', '스펙', '스키마']),
+  api: Object.freeze(['api', 'endpoint', 'interface', 'sdk', '호출', '엔드포인트']),
+  sdk: Object.freeze(['sdk', 'library', 'package', 'framework', '라이브러리', '패키지', '프레임워크']),
+  library: Object.freeze(['library', 'package', 'framework', 'module', '라이브러리', '패키지', '모듈']),
+  web: Object.freeze(['web', 'site', 'article', 'forum', 'blog', 'reddit', '웹', '사이트', '기사', '포럼', '블로그']),
+  search: Object.freeze(['search', 'browse', 'lookup', 'find', '검색', '조회', '탐색', '찾아']),
+  news: Object.freeze(['latest', 'recent', 'news', 'today', 'release', 'announcement', '최신', '최근', '뉴스', '오늘', '릴리즈', '공지']),
+  current: Object.freeze(['current', 'status', 'pricing', 'changelog', 'up-to-date', '현재', '상태', '가격', '변경사항']),
+  research: Object.freeze(['research', 'verify', 'fact-check', 'investigate', '리서치', '검증', '조사']),
+  verification: Object.freeze(['verify', 'validation', 'fact-check', 'audit', '검증', '확인', '감사']),
+  code: Object.freeze(['code', 'repo', 'repository', 'source', 'implementation', 'bug', 'fix', 'test', 'snippet', 'cli', '코드', '리포', '저장소', '구현', '버그', '테스트', '예제', '스크립트']),
+  repository: Object.freeze(['repo', 'repository', 'source', 'git', 'github', '리포', '저장소', '소스']),
+  examples: Object.freeze(['example', 'examples', 'snippet', 'sample', '예제', '샘플']),
+  browser: Object.freeze(['browser', 'page', 'dom', 'screenshot', 'render', '브라우저', '페이지', '스크린샷', '렌더']),
+  ui: Object.freeze(['ui', 'ux', 'layout', 'responsive', 'css', 'html', '디자인', '레이아웃', '반응형']),
+  visual: Object.freeze(['visual', 'screenshot', 'layout', 'render', 'screen', '화면', '시각', '스크린샷']),
+  e2e: Object.freeze(['playwright', 'e2e', 'click', 'navigate', 'automation', 'playwright', '클릭', '이동', '자동화']),
+  analysis: Object.freeze(['analysis', 'analyze', 'audit', 'compare', 'root cause', '분석', '검토', '비교', '원인']),
+  planning: Object.freeze(['plan', 'planning', 'strategy', 'design', '계획', '전략', '설계']),
+  reasoning: Object.freeze(['reason', 'reasoning', 'think', 'critique', '추론', '사고', '비평']),
+  security: Object.freeze(['security', 'risk', 'threat', 'vulnerability', '보안', '위험', '취약점']),
+  review: Object.freeze(['review', 'reviewer', 'inspect', '리뷰', '검수']),
+});
+
+export const SERVER_EXPLICIT_KEYWORDS = Object.freeze({
+  context7: Object.freeze(['context7']),
+  'brave-search': Object.freeze(['brave', 'brave-search']),
+  exa: Object.freeze(['exa']),
+  tavily: Object.freeze(['tavily']),
+  playwright: Object.freeze(['playwright']),
+  'sequential-thinking': Object.freeze(['sequential-thinking', 'sequential thinking']),
+});
+
+export function uniqueStrings(values = []) {
+  return [...new Set(
+    values
+      .filter((value) => typeof value === 'string' && value.trim())
+      .map((value) => value.trim()),
+  )];
+}
+
+export function inferDomainTagsFromText(text = '') {
+  if (typeof text !== 'string' || !text.trim()) return [];
+  const normalized = text.toLocaleLowerCase();
+  const matched = [];
+
+  for (const [tag, keywords] of Object.entries(DOMAIN_TAG_KEYWORDS)) {
+    if (keywords.some((keyword) => normalized.includes(String(keyword).toLocaleLowerCase()))) {
+      matched.push(tag);
+    }
+  }
+
+  return uniqueStrings(matched);
+}
+
+export function getDefaultServerMetadata(serverName = '') {
+  const toolCount = MCP_SERVER_TOOL_CATALOG[serverName]?.length || 0;
+  const domainTags = uniqueStrings([
+    ...(MCP_SERVER_DOMAIN_TAGS[serverName] || []),
+    ...inferDomainTagsFromText(serverName),
+  ]);
+
+  return {
+    tool_count: toolCount,
+    domain_tags: domainTags,
+  };
+}
+
+export function normalizeServerMetadata(serverName = '', metadata = {}) {
+  const fallback = getDefaultServerMetadata(serverName);
+  const toolCount = Number.isFinite(metadata.tool_count)
+    ? Math.max(0, Math.trunc(metadata.tool_count))
+    : fallback.tool_count;
+  const domainTags = uniqueStrings([
+    ...fallback.domain_tags,
+    ...(Array.isArray(metadata.domain_tags) ? metadata.domain_tags : []),
+    ...inferDomainTagsFromText([
+      serverName,
+      typeof metadata.command === 'string' ? metadata.command : '',
+      typeof metadata.url === 'string' ? metadata.url : '',
+      ...(Array.isArray(metadata.args) ? metadata.args : []),
+      ...(metadata.env && typeof metadata.env === 'object' ? Object.keys(metadata.env) : []),
+    ].join(' ')),
+  ]);
+
+  return {
+    tool_count: toolCount,
+    domain_tags: domainTags,
+  };
+}

package/scripts/lib/skill-state.mjs

@@ -0,0 +1,220 @@
+import {
+  access,
+  mkdir,
+  open,
+  readdir,
+  readFile,
+  rename,
+  rm,
+  writeFile,
+} from "node:fs/promises";
+import { basename, join } from "node:path";
+
+const DEFAULT_STATE_DIR = join(process.cwd(), ".tfx", "state");
+const STOP_HOOKS = new Map();
+
+function stateFilePath(stateDir, skillName) {
+  return join(stateDir, `${skillName}-active.json`);
+}
+
+function assertValidSkillName(skillName) {
+  if (basename(skillName) !== skillName) {
+    throw new Error(`Invalid skill name: ${skillName}`);
+  }
+}
+
+function assertValidOnStop(onStop) {
+  if (onStop !== undefined && typeof onStop !== "function") {
+    throw new TypeError("onStop must be a function");
+  }
+}
+
+async function readStateFile(filePath) {
+  try {
+    const raw = await readFile(filePath, "utf8");
+    return JSON.parse(raw);
+  } catch {
+    return null;
+  }
+}
+
+function rememberStopHook(filePath, onStop) {
+  if (typeof onStop === "function") {
+    STOP_HOOKS.set(filePath, onStop);
+    return;
+  }
+  STOP_HOOKS.delete(filePath);
+}
+
+function logStopHookWarning(message, error) {
+  if (error) {
+    console.warn(message, error);
+    return;
+  }
+  console.warn(message);
+}
+
+async function runStopHook(filePath, state, onStop) {
+  if (!state?.hasStopHook) {
+    STOP_HOOKS.delete(filePath);
+    return;
+  }
+
+  const hook = onStop ?? STOP_HOOKS.get(filePath);
+  STOP_HOOKS.delete(filePath);
+
+  if (typeof hook !== "function") {
+    return;
+  }
+
+  try {
+    await hook({
+      skillName: state.skillName,
+      filePath,
+      state,
+    });
+  } catch (error) {
+    logStopHookWarning(
+      `Failed to run stop-hook for skill: ${state.skillName}`,
+      error,
+    );
+  }
+}
+
+/**
+ * Activate a skill by writing its state file.
+ * Throws if the skill is already active.
+ *
+ * @param {string} skillName
+ * @param {{ stateDir?: string, onStop?: (() => Promise<void> | void) }} options
+ */
+export async function activateSkill(
+  skillName,
+  { stateDir = DEFAULT_STATE_DIR, onStop } = {},
+) {
+  assertValidSkillName(skillName);
+  assertValidOnStop(onStop);
+
+  await mkdir(stateDir, { recursive: true });
+
+  const filePath = stateFilePath(stateDir, skillName);
+  const lockPath = `${filePath}.lock`;
+  const tmpPath = `${filePath}.${process.pid}.${Date.now()}.tmp`;
+  let lockHandle;
+
+  try {
+    lockHandle = await open(lockPath, "wx");
+    try {
+      await access(filePath);
+      throw new Error(`Skill already active: ${skillName}`);
+    } catch (error) {
+      if (error?.message === `Skill already active: ${skillName}`) {
+        throw error;
+      }
+      if (error?.code !== "ENOENT") {
+        throw error;
+      }
+    }
+
+    const state = {
+      skillName,
+      pid: process.pid,
+      activatedAt: Date.now(),
+      hasStopHook: typeof onStop === "function",
+    };
+    await writeFile(tmpPath, JSON.stringify(state), "utf8");
+    await rename(tmpPath, filePath);
+    rememberStopHook(filePath, onStop);
+  } finally {
+    await rm(tmpPath, { force: true }).catch(() => {});
+    if (lockHandle) {
+      await lockHandle.close().catch(() => {});
+    }
+    await rm(lockPath, { force: true }).catch(() => {});
+  }
+}
+
+/**
+ * Deactivate a skill by removing its state file.
+ * Does not throw if the file does not exist.
+ *
+ * @param {string} skillName
+ * @param {{ stateDir?: string, onStop?: (() => Promise<void> | void) }} options
+ */
+export async function deactivateSkill(
+  skillName,
+  { stateDir = DEFAULT_STATE_DIR, onStop } = {},
+) {
+  assertValidOnStop(onStop);
+
+  const filePath = stateFilePath(stateDir, skillName);
+  const state = await readStateFile(filePath);
+
+  try {
+    await runStopHook(filePath, state, onStop);
+  } finally {
+    STOP_HOOKS.delete(filePath);
+    await rm(filePath, { force: true }).catch(() => {});
+  }
+}
+
+/**
+ * Return all currently active skills by scanning *-active.json files.
+ *
+ * @param {{ stateDir?: string }} options
+ * @returns {Promise<Array<{ skillName: string, pid: number, activatedAt: number, hasStopHook?: boolean }>>}
+ */
+export async function getActiveSkills({ stateDir = DEFAULT_STATE_DIR } = {}) {
+  let entries;
+  try {
+    entries = await readdir(stateDir);
+  } catch {
+    return [];
+  }
+
+  const results = [];
+  for (const entry of entries) {
+    if (!entry.endsWith("-active.json")) continue;
+    const state = await readStateFile(join(stateDir, entry));
+    if (state) {
+      results.push(state);
+    }
+  }
+  return results;
+}
+
+/**
+ * Remove state files for skills whose processes are no longer alive.
+ *
+ * @param {{ stateDir?: string }} options
+ * @returns {Promise<string[]>} list of pruned skill names
+ */
+export async function pruneOrphanSkillStates({
+  stateDir = DEFAULT_STATE_DIR,
+} = {}) {
+  const active = await getActiveSkills({ stateDir });
+  const pruned = [];
+
+  for (const state of active) {
+    let alive = true;
+    try {
+      process.kill(state.pid, 0);
+    } catch {
+      alive = false;
+    }
+
+    if (!alive) {
+      const filePath = stateFilePath(stateDir, state.skillName);
+      STOP_HOOKS.delete(filePath);
+      if (state.hasStopHook) {
+        logStopHookWarning(
+          `Skipping stop-hook for orphaned skill state: ${state.skillName}`,
+        );
+      }
+      await rm(filePath, { force: true }).catch(() => {});
+      pruned.push(state.skillName);
+    }
+  }
+
+  return pruned;
+}