@myvillage/cli 1.2.2 → 1.5.0

package/src/utils/local-agent.js

@@ -0,0 +1,168 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync, readdirSync, unlinkSync } from 'fs';
+import { join } from 'path';
+import { getConfigDir } from './config.js';
+import { parse as parseYaml, stringify as stringifyYaml } from 'yaml';
+
+// ── Directory Helpers ───────────────────────────────────
+
+export function getAgentsDir() {
+  const dir = join(getConfigDir(), 'agents');
+  if (!existsSync(dir)) {
+    mkdirSync(dir, { recursive: true });
+  }
+  return dir;
+}
+
+export function getAgentDir(name) {
+  return join(getAgentsDir(), name);
+}
+
+export function agentExists(name) {
+  const dir = getAgentDir(name);
+  return existsSync(dir) && existsSync(join(dir, 'agent.config.yaml'));
+}
+
+// ── Config Read/Write ───────────────────────────────────
+
+export function readAgentConfig(name) {
+  const configPath = join(getAgentDir(name), 'agent.config.yaml');
+  if (!existsSync(configPath)) return null;
+  try {
+    const raw = readFileSync(configPath, 'utf-8');
+    return parseYaml(raw);
+  } catch {
+    return null;
+  }
+}
+
+export function writeAgentConfig(name, config) {
+  const configPath = join(getAgentDir(name), 'agent.config.yaml');
+  writeFileSync(configPath, stringifyYaml(config, { lineWidth: 0 }));
+}
+
+// ── Tools YAML Read/Write ───────────────────────────────
+
+export function readToolsYaml(name) {
+  const toolsPath = join(getAgentDir(name), 'tools.yaml');
+  if (!existsSync(toolsPath)) return { servers: {} };
+  try {
+    const raw = readFileSync(toolsPath, 'utf-8');
+    return parseYaml(raw) || { servers: {} };
+  } catch {
+    return { servers: {} };
+  }
+}
+
+export function writeToolsYaml(name, toolsConfig) {
+  const toolsPath = join(getAgentDir(name), 'tools.yaml');
+  writeFileSync(toolsPath, stringifyYaml(toolsConfig, { lineWidth: 0 }));
+}
+
+// ── Daemon Status ───────────────────────────────────────
+
+export function getDaemonInfo(name) {
+  const pidFile = join(getAgentDir(name), 'daemon.pid');
+  if (!existsSync(pidFile)) return null;
+  try {
+    return JSON.parse(readFileSync(pidFile, 'utf-8'));
+  } catch {
+    return null;
+  }
+}
+
+export function isDaemonRunning(name) {
+  const info = getDaemonInfo(name);
+  if (!info) return false;
+  try {
+    process.kill(info.pid, 0);
+    return true;
+  } catch {
+    // Process dead, clean up stale PID file
+    cleanupPidFile(name);
+    return false;
+  }
+}
+
+export function cleanupPidFile(name) {
+  const pidFile = join(getAgentDir(name), 'daemon.pid');
+  try {
+    if (existsSync(pidFile)) unlinkSync(pidFile);
+  } catch {
+    // ignore
+  }
+}
+
+// ── List Local Agents ───────────────────────────────────
+
+export function listLocalAgents() {
+  const agentsDir = getAgentsDir();
+  if (!existsSync(agentsDir)) return [];
+
+  const entries = readdirSync(agentsDir, { withFileTypes: true });
+  const agents = [];
+
+  for (const entry of entries) {
+    if (!entry.isDirectory()) continue;
+    const configPath = join(agentsDir, entry.name, 'agent.config.yaml');
+    if (!existsSync(configPath)) continue;
+
+    try {
+      const config = parseYaml(readFileSync(configPath, 'utf-8'));
+      const running = isDaemonRunning(entry.name);
+      const daemonInfo = getDaemonInfo(entry.name);
+
+      agents.push({
+        name: entry.name,
+        config,
+        isRunning: running,
+        lastHeartbeat: daemonInfo?.lastHeartbeat || null,
+      });
+    } catch {
+      // Skip malformed agent directories
+    }
+  }
+
+  return agents;
+}
+
+// ── Log Reading ─────────────────────────────────────────
+
+export function readAgentLogs(name, options = {}) {
+  const logsDir = join(getAgentDir(name), 'logs');
+  if (!existsSync(logsDir)) return [];
+
+  const { since, limit = 50 } = options;
+
+  // Find log files, sorted newest first
+  const files = readdirSync(logsDir)
+    .filter(f => f.endsWith('.jsonl'))
+    .sort()
+    .reverse();
+
+  const entries = [];
+
+  for (const file of files) {
+    const lines = readFileSync(join(logsDir, file), 'utf-8')
+      .split('\n')
+      .filter(Boolean);
+
+    for (const line of lines.reverse()) {
+      try {
+        const entry = JSON.parse(line);
+        if (since && new Date(entry.ts) < new Date(since)) continue;
+        entries.push(entry);
+        if (entries.length >= limit) return entries.reverse();
+      } catch {
+        // skip malformed lines
+      }
+    }
+  }
+
+  return entries.reverse();
+}
+
+export function getLogFilePath(name) {
+  const logsDir = join(getAgentDir(name), 'logs');
+  const today = new Date().toISOString().slice(0, 10);
+  return join(logsDir, `${today}.jsonl`);
+}