atabey 0.0.7 → 0.0.8

package/framework-mcp/dist/src/shared/logger.js

@@ -0,0 +1,116 @@
+import chalk from "chalk";
+import path from "path";
+import { ensureDir, appendFile } from "./fs.js";
+export var LogLevel;
+(function (LogLevel) {
+    LogLevel[LogLevel["DEBUG"] = 0] = "DEBUG";
+    LogLevel[LogLevel["INFO"] = 1] = "INFO";
+    LogLevel[LogLevel["WARN"] = 2] = "WARN";
+    LogLevel[LogLevel["ERROR"] = 3] = "ERROR";
+    LogLevel[LogLevel["FATAL"] = 4] = "FATAL";
+})(LogLevel || (LogLevel = {}));
+export class EnterpriseLogger {
+    config;
+    static instance;
+    constructor() {
+        const isProd = process.env.NODE_ENV === "production";
+        this.config = {
+            minLevel: isProd ? LogLevel.INFO : LogLevel.DEBUG,
+            enableColors: !isProd,
+            jsonFormat: isProd,
+        };
+    }
+    static getInstance() {
+        if (!EnterpriseLogger.instance) {
+            EnterpriseLogger.instance = new EnterpriseLogger();
+        }
+        return EnterpriseLogger.instance;
+    }
+    configure(config) {
+        this.config = { ...this.config, ...config };
+        if (this.config.logFile) {
+            try {
+                ensureDir(path.dirname(this.config.logFile));
+            }
+            catch (err) {
+                // Directly write to stderr — using console here would be circular
+                process.stderr.write(`[Logger] Failed to create log directory: ${err}\n`);
+            }
+        }
+    }
+    formatMessage(level, message, meta) {
+        const timestamp = new Date().toISOString();
+        const pid = process.pid;
+        if (this.config.jsonFormat) {
+            return JSON.stringify({
+                timestamp,
+                level: LogLevel[level],
+                pid,
+                message,
+                meta,
+            });
+        }
+        const levelName = LogLevel[level].padEnd(5);
+        let coloredLevel = levelName;
+        if (this.config.enableColors) {
+            switch (level) {
+                case LogLevel.DEBUG:
+                    coloredLevel = chalk.blue(levelName);
+                    break;
+                case LogLevel.INFO:
+                    coloredLevel = chalk.green(levelName);
+                    break;
+                case LogLevel.WARN:
+                    coloredLevel = chalk.yellow(levelName);
+                    break;
+                case LogLevel.ERROR:
+                    coloredLevel = chalk.red(levelName);
+                    break;
+                case LogLevel.FATAL:
+                    coloredLevel = chalk.bgRed.white.bold(levelName);
+                    break;
+            }
+        }
+        const metaStr = meta ? ` | Meta: ${JSON.stringify(meta)}` : "";
+        return `[${timestamp}] [PID:${pid}] [${coloredLevel}]: ${message}${metaStr}`;
+    }
+    log(level, message, meta) {
+        if (level < this.config.minLevel)
+            return;
+        const formatted = this.formatMessage(level, message, meta);
+        // Route to stderr for ERROR/FATAL, stdout otherwise.
+        // Using process.write directly here is intentional — this IS the logger.
+        if (level >= LogLevel.ERROR) {
+            process.stderr.write(formatted + "\n");
+        }
+        else {
+            process.stdout.write(formatted + "\n");
+        }
+        // Output to file if configured
+        if (this.config.logFile) {
+            try {
+                appendFile(this.config.logFile, formatted + "\n");
+            }
+            catch (err) {
+                // Directly write to stderr — using console here would be circular
+                process.stderr.write(`[Logger] Failed to write to log file: ${err}\n`);
+            }
+        }
+    }
+    debug(message, meta) {
+        this.log(LogLevel.DEBUG, message, meta);
+    }
+    info(message, meta) {
+        this.log(LogLevel.INFO, message, meta);
+    }
+    warn(message, meta) {
+        this.log(LogLevel.WARN, message, meta);
+    }
+    error(message, meta) {
+        this.log(LogLevel.ERROR, message, meta);
+    }
+    fatal(message, meta) {
+        this.log(LogLevel.FATAL, message, meta);
+    }
+}
+export const logger = EnterpriseLogger.getInstance();

package/framework-mcp/dist/src/shared/storage.js

@@ -0,0 +1,207 @@
+import Database from "better-sqlite3";
+import path from "path";
+import fs from "fs";
+import { getFrameworkDir } from "../cli/utils/memory.js";
+/**
+ * [DB] Atabey Storage Engine
+ * Central SQLite database for enterprise-scale agent state management.
+ */
+export class Storage {
+    static db = null;
+    static getDB() {
+        if (!this.db) {
+            const frameworkDir = getFrameworkDir();
+            if (!fs.existsSync(frameworkDir)) {
+                fs.mkdirSync(frameworkDir, { recursive: true });
+            }
+            const dbPath = path.join(frameworkDir, "atabey.db");
+            this.db = new Database(dbPath);
+            this.initializeSchema();
+        }
+        return this.db;
+    }
+    static initializeSchema() {
+        const db = this.db;
+        // Agents Table
+        db.exec(`
+            CREATE TABLE IF NOT EXISTS agents (
+                name TEXT PRIMARY KEY,
+                state TEXT DEFAULT 'READY',
+                task TEXT DEFAULT 'Idle',
+                last_updated DATETIME DEFAULT CURRENT_TIMESTAMP
+            )
+        `);
+        // Messages (Hermes) Table
+        db.exec(`
+            CREATE TABLE IF NOT EXISTS messages (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
+                sender TEXT,
+                receiver TEXT,
+                category TEXT,
+                content TEXT,
+                trace_id TEXT,
+                parent_id TEXT,
+                status TEXT DEFAULT 'PENDING',
+                priority TEXT DEFAULT 'NORMAL',
+                requires_approval BOOLEAN DEFAULT 0
+            )
+        `);
+        // Tasks Table
+        db.exec(`
+            CREATE TABLE IF NOT EXISTS tasks (
+                id TEXT PRIMARY KEY,
+                trace_id TEXT,
+                description TEXT,
+                agent TEXT,
+                status TEXT DEFAULT 'PENDING',
+                priority TEXT DEFAULT 'NORMAL',
+                dependencies TEXT, -- JSON Array
+                created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+            )
+        `);
+        // Logs Table
+        db.exec(`
+            CREATE TABLE IF NOT EXISTS logs (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
+                agent TEXT,
+                action TEXT,
+                trace_id TEXT,
+                status TEXT,
+                summary TEXT,
+                findings TEXT
+            )
+        `);
+        // Metadata (State) Table
+        db.exec(`
+            CREATE TABLE IF NOT EXISTS metadata (
+                key TEXT PRIMARY KEY,
+                value TEXT
+            )
+        `);
+    }
+    // --- Agent Operations ---
+    static updateAgentStatus(name, state, task, lastUpdated) {
+        const db = this.getDB();
+        const cleanName = name.replace("@", "");
+        const timestamp = lastUpdated || new Date().toISOString();
+        db.prepare(`
+            INSERT INTO agents (name, state, task, last_updated) 
+            VALUES (?, ?, ?, ?)
+            ON CONFLICT(name) DO UPDATE SET 
+                state = excluded.state,
+                task = excluded.task,
+                last_updated = excluded.last_updated
+        `).run(cleanName, state, task, timestamp);
+    }
+    static getAllAgents() {
+        return this.getDB().prepare("SELECT * FROM agents").all();
+    }
+    // --- Message Operations ---
+    static saveMessage(msg) {
+        const db = this.getDB();
+        db.prepare(`
+            INSERT INTO messages (sender, receiver, category, content, trace_id, parent_id, status, priority, requires_approval)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+        `).run(msg.from, msg.to, msg.category, msg.content, msg.traceId, msg.parentId || null, msg.status || "PENDING", msg.priority || "NORMAL", msg.requiresApproval ? 1 : 0);
+    }
+    static getPendingMessages() {
+        const rows = this.getDB().prepare("SELECT * FROM messages WHERE status IN ('PENDING', 'APPROVED') ORDER BY priority DESC, timestamp ASC").all();
+        return rows.map(r => ({
+            id: r.id,
+            timestamp: r.timestamp,
+            from: r.sender,
+            to: r.receiver,
+            category: r.category,
+            content: r.content,
+            traceId: r.trace_id,
+            parentId: r.parent_id || undefined,
+            status: r.status,
+            priority: r.priority,
+            requiresApproval: r.requires_approval === 1
+        }));
+    }
+    static updateMessageStatus(id, status) {
+        this.getDB().prepare("UPDATE messages SET status = ? WHERE id = ?").run(status, id);
+    }
+    // --- Task Operations ---
+    static saveTask(task) {
+        const db = this.getDB();
+        db.prepare(`
+            INSERT INTO tasks (id, trace_id, description, agent, status, priority, dependencies)
+            VALUES (?, ?, ?, ?, ?, ?, ?)
+            ON CONFLICT(id) DO UPDATE SET
+                status = excluded.status,
+                description = excluded.description
+        `).run(task.id, task.traceId, task.description, task.agent, task.status, task.priority, JSON.stringify(task.dependencies || []));
+    }
+    static getTasks(traceId) {
+        const db = this.getDB();
+        let rows;
+        if (traceId) {
+            rows = db.prepare("SELECT * FROM tasks WHERE trace_id = ?").all(traceId);
+        }
+        else {
+            rows = db.prepare("SELECT * FROM tasks").all();
+        }
+        return rows.map(r => {
+            let deps = [];
+            try {
+                const parsed = JSON.parse(r.dependencies || "[]");
+                // Handle cases where dependencies might be double-stringified or are raw JSON strings
+                deps = Array.isArray(parsed) ? parsed : JSON.parse(parsed);
+            }
+            catch {
+                // Keep empty array
+            }
+            return {
+                id: r.id,
+                traceId: r.trace_id,
+                description: r.description,
+                agent: r.agent,
+                status: r.status,
+                priority: r.priority,
+                dependencies: deps
+            };
+        });
+    }
+    static getLogs(agentName) {
+        const db = this.getDB();
+        let rows;
+        const cleanName = agentName ? agentName.replace("@", "") : undefined;
+        if (cleanName) {
+            rows = db.prepare("SELECT * FROM logs WHERE agent = ? ORDER BY timestamp DESC").all(cleanName);
+        }
+        else {
+            rows = db.prepare("SELECT * FROM logs ORDER BY timestamp DESC").all();
+        }
+        return rows.map(r => ({
+            id: r.id,
+            timestamp: r.timestamp,
+            agent: r.agent,
+            action: r.action,
+            trace_id: r.trace_id || undefined,
+            status: r.status,
+            summary: r.summary,
+            findings: r.findings || undefined
+        }));
+    }
+    // --- Metadata Operations ---
+    static setMetadata(key, value) {
+        this.getDB().prepare(`
+            INSERT INTO metadata (key, value) VALUES (?, ?)
+            ON CONFLICT(key) DO UPDATE SET value = excluded.value
+        `).run(key, value);
+    }
+    static getMetadata(key) {
+        const row = this.getDB().prepare("SELECT value FROM metadata WHERE key = ?").get(key);
+        return row ? row.value : null;
+    }
+    static reset() {
+        if (this.db) {
+            this.db.close();
+            this.db = null;
+        }
+    }
+}

package/framework-mcp/dist/src/shared/types.js

@@ -0,0 +1,12 @@
+/**
+ * Agent Atabey Framework — Internal Branded Types
+ * Used to enforce absolute type safety within the core orchestration logic.
+ */
+/**
+ * Casts a raw string to a Branded Type.
+ * Use this only at the boundaries of the system.
+ */
+export function asTraceID(val) { return val; }
+export function asAgentID(val) { return val; }
+export function asPhaseID(val) { return val; }
+export function asProjectPath(val) { return val; }

package/framework-mcp/dist/tools/control_plane/locking.js

@@ -0,0 +1,82 @@
+import fs from "fs";
+import path from "path";
+import { resolveFrameworkDir } from "../../utils/security.js";
+/**
+ * Handles acquiring a stateful lock on a resource.
+ */
+export async function handleAcquireLock(projectRoot, args) {
+    const { resource, agent, ttl = 60 } = args;
+    const frameworkDir = resolveFrameworkDir(projectRoot);
+    const lockDir = path.join(projectRoot, frameworkDir, "locks");
+    const lockPath = path.join(lockDir, `${resource}.lock`);
+    try {
+        if (!fs.existsSync(lockDir))
+            fs.mkdirSync(lockDir, { recursive: true });
+        // Check for stale lock first
+        if (fs.existsSync(lockPath)) {
+            const stat = fs.statSync(lockPath);
+            const now = new Date().getTime();
+            const age = (now - stat.mtimeMs) / 1000;
+            if (age < ttl) {
+                return {
+                    isError: true,
+                    content: [{ type: "text", text: `Resource '${resource}' is currently locked by another agent.` }]
+                };
+            }
+            // Lock expired, safe to override by renaming first (POSIX atomic operation)
+            const tempLockPath = `${lockPath}.${Math.random().toString(36).substring(2)}.old`;
+            try {
+                fs.renameSync(lockPath, tempLockPath);
+                fs.unlinkSync(tempLockPath);
+            }
+            catch {
+                // If rename failed, it means another agent already evicting/deleting it.
+                // Do not delete anything else; proceed and let writeFileSync (wx flag) fail if a new lock exists.
+            }
+        }
+        // Use 'wx' flag for atomic file creation
+        const lockData = JSON.stringify({ agent, timestamp: new Date().toISOString() });
+        fs.writeFileSync(lockPath, lockData, { flag: "wx" });
+        return {
+            content: [{ type: "text", text: `[OK] Lock acquired for resource '${resource}' by ${agent}.` }]
+        };
+    }
+    catch (e) {
+        const error = e;
+        if (error.code === "EEXIST") {
+            return {
+                isError: true,
+                content: [{ type: "text", text: `Resource '${resource}' was just acquired by another agent.` }]
+            };
+        }
+        return {
+            isError: true,
+            content: [{ type: "text", text: `Failed to acquire lock: ${String(e)}` }]
+        };
+    }
+}
+/**
+ * Handles releasing a lock.
+ */
+export async function handleReleaseLock(projectRoot, args) {
+    const { resource, agent } = args;
+    const frameworkDir = resolveFrameworkDir(projectRoot);
+    const lockPath = path.join(projectRoot, frameworkDir, "locks", `${resource}.lock`);
+    try {
+        if (!fs.existsSync(lockPath)) {
+            return { content: [{ type: "text", text: `ℹ️ No lock found for resource '${resource}'.` }] };
+        }
+        const lockData = JSON.parse(fs.readFileSync(lockPath, "utf8"));
+        if (lockData.agent !== agent) {
+            return {
+                isError: true,
+                content: [{ type: "text", text: `[ERROR] Denied: You do not own the lock for '${resource}'. Owned by ${lockData.agent}.` }]
+            };
+        }
+        fs.unlinkSync(lockPath);
+        return { content: [{ type: "text", text: `[OK] Lock released for resource '${resource}' by ${agent}.` }] };
+    }
+    catch (e) {
+        return { isError: true, content: [{ type: "text", text: `Failed to release lock: ${String(e)}` }] };
+    }
+}

package/framework-mcp/dist/tools/control_plane/registry.js

@@ -0,0 +1,35 @@
+import fs from "fs";
+import path from "path";
+import { resolveFrameworkDir } from "../../utils/security.js";
+/**
+ * Handles agent registration with the Control Plane.
+ * This can be used to validate permissions and active status.
+ */
+export async function handleRegisterAgent(projectRoot, args) {
+    const { agent, role, capability = 5, specialties } = args;
+    const frameworkDir = resolveFrameworkDir(projectRoot);
+    const registryDir = path.join(projectRoot, frameworkDir, "registry");
+    const agentFile = path.join(registryDir, `${agent.replace("@", "")}_active.json`);
+    try {
+        if (!fs.existsSync(registryDir))
+            fs.mkdirSync(registryDir, { recursive: true });
+        const agentData = {
+            agent,
+            role,
+            capability,
+            specialties,
+            last_seen: new Date().toISOString(),
+            status: "ACTIVE"
+        };
+        fs.writeFileSync(agentFile, JSON.stringify(agentData, null, 2));
+        return {
+            content: [{ type: "text", text: `[ATABEY] Agent ${agent} (${role}) registered successfully in the Atabey Control Plane.` }]
+        };
+    }
+    catch (e) {
+        return {
+            isError: true,
+            content: [{ type: "text", text: `Failed to register agent: ${String(e)}` }]
+        };
+    }
+}