atabey-shared 0.1.1-beta → 0.1.2

package/dist/modules/storage/storage.d.ts

@@ -0,0 +1,341 @@
+import Database from "better-sqlite3";
+import type { ApiEndpoint } from "../agent/api-catalog.js";
+import { type BackupResult, type IntegrityCheckResult } from "./migrations.js";
+import { AgentID, LogID, MessageID, TaskID, TraceID } from "../primitives/types.js";
+export interface AgentRow {
+    name: AgentID;
+    state: string;
+    task: string;
+    last_updated: string;
+}
+export interface MessageRow {
+    id?: MessageID;
+    timestamp: string;
+    from: AgentID;
+    to: AgentID;
+    category: string;
+    content: string;
+    traceId: TraceID;
+    parentId?: TaskID;
+    status: string;
+    priority: string;
+    requiresApproval: boolean;
+}
+export interface TaskRow {
+    id: TaskID;
+    traceId: TraceID;
+    description: string;
+    agent: AgentID;
+    status: string;
+    priority: string;
+    dependencies: string[];
+    createdAt?: string;
+}
+export interface LogRow {
+    id?: LogID;
+    timestamp: string;
+    agent: AgentID;
+    action: string;
+    trace_id?: TraceID;
+    status: string;
+    summary: string;
+    findings?: string;
+    prev_hash?: string;
+    hash?: string;
+}
+export interface CostDbRow {
+    id: number;
+    provider: string;
+    model: string;
+    input_tokens: number;
+    output_tokens: number;
+    cost: number;
+    timestamp: string;
+    trace_id: string | null;
+}
+export interface MessageDbRow {
+    id: number;
+    timestamp: string;
+    sender: string;
+    receiver: string;
+    category: string;
+    content: string;
+    trace_id: string;
+    parent_id: string | null;
+    status: string;
+    priority: string;
+    requires_approval: number;
+}
+export interface TaskDbRow {
+    id: string;
+    trace_id: string;
+    description: string;
+    agent: string;
+    status: string;
+    priority: string;
+    dependencies: string;
+    created_at: string;
+}
+export interface LogDbRow {
+    id: number;
+    timestamp: string;
+    agent: string;
+    action: string;
+    trace_id: string | null;
+    status: string;
+    summary: string;
+    findings: string | null;
+    prev_hash: string | null;
+    hash: string | null;
+}
+/**
+ * Resolves the framework directory (.atabey) intelligently by ascending the directory tree,
+ * preventing subpackages and test runners from spraying databases in random subfolders.
+ */
+export declare function resolveFrameworkDir(): string;
+/**
+ * [DB] Canonical Atabey Storage Engine (Single Source of Truth)
+ * Central SQLite database for enterprise-scale multi-agent state management.
+ */
+export declare class AtabeyStorage {
+    private static db;
+    static getDB(): Database.Database;
+    private static initializeSchema;
+    static getKnowledgeList(): Array<{
+        filename: string;
+        updated_at: string;
+    }>;
+    static getKnowledgeFile(filename: string): {
+        filename: string;
+        content: string;
+        updated_at: string;
+    } | null;
+    static saveKnowledgeFile(filename: string, content: string): void;
+    static deleteKnowledgeFile(filename: string): boolean;
+    static hasUsers(): boolean;
+    static getUsers(): Array<{
+        name: string;
+        token: string;
+        role: string;
+    }>;
+    static getUserByToken(token: string): {
+        name: string;
+        role: string;
+    } | null;
+    static createUser(name: string, token: string, role: string): void;
+    static deleteUser(name: string): boolean;
+    static createAgent(name: string, description: string, skills: string): void;
+    static deleteAgent(name: string): boolean;
+    static updateAgentDetails(name: string, description: string, state: string, task: string, skills: string): void;
+    static onMessageSaved: ((msg: {
+        traceId: string;
+        category: string;
+    }) => void) | null;
+    static onMessageStatusUpdated: ((id: number, status: string) => void) | null;
+    static saveLog(log: {
+        agent: string;
+        action: string;
+        trace_id?: string;
+        status: string;
+        summary: string;
+        findings?: string;
+    }): number;
+    static verifyLogIntegrity(): {
+        valid: boolean;
+        checked?: number;
+        failedLogId?: number;
+        reason?: string;
+        tip?: string;
+    };
+    static saveCost(cost: {
+        provider: string;
+        model: string;
+        inputTokens: number;
+        outputTokens: number;
+        cost: number;
+        timestamp: string;
+        traceId?: string;
+    }): void;
+    static getAllCosts(): Array<{
+        provider: string;
+        model: string;
+        inputTokens: number;
+        outputTokens: number;
+        cost: number;
+        timestamp: string;
+        traceId?: string;
+    }>;
+    static updateAgentStatus(name: string, state: string, task: string, lastUpdated?: string): void;
+    static getAllAgents(): AgentRow[];
+    static saveMessage(msg: MessageRow): void;
+    static getPendingMessages(): MessageRow[];
+    static updateMessageStatus(id: number, status: string): void;
+    static saveTask(task: TaskRow): void;
+    static getTasks(traceId?: TraceID): TaskRow[];
+    static getLogs(agentName?: string): LogRow[];
+    static setMetadata(key: string, value: string): void;
+    static getMetadata(key: string): string | null;
+    static saveLoopCooldown(agent: string, cooldownUntil: number, detail: string | null, cooldownCount?: number): void;
+    static getLoopCooldown(agent: string): {
+        agent: string;
+        cooldownUntil: number;
+        detail: string | null;
+        cooldownCount: number;
+    } | null;
+    static clearLoopCooldown(agent: string): boolean;
+    static clearAllLoopCooldowns(): void;
+    static listActiveLoopCooldowns(): Array<{
+        agent: string;
+        cooldownUntil: number;
+        detail: string | null;
+        cooldownCount: number;
+    }>;
+    static purgeExpiredLoopCooldowns(): number;
+    static saveApproval(app: {
+        traceId: string;
+        toolName: string;
+        agent: string;
+        riskScore: number;
+        riskLevel: string;
+        reason: string;
+        args: Record<string, unknown>;
+        createdAt: number;
+        expiresAt: number;
+        status?: "PENDING" | "APPROVED" | "REJECTED" | "EXPIRED";
+        passcode?: string;
+    }): void;
+    static saveApprovalRequest(request: {
+        traceId: string;
+        toolName: string;
+        agent: string;
+        riskScore: number;
+        riskLevel: string;
+        reason: string;
+        args: Record<string, unknown>;
+        createdAt: number;
+        expiresAt: number;
+        status?: "PENDING" | "APPROVED" | "REJECTED" | "EXPIRED";
+        passcode?: string;
+    }): void;
+    static getApproval(traceId: string): {
+        traceId: string;
+        toolName: string;
+        agent: string;
+        riskScore: number;
+        riskLevel: string;
+        reason: string;
+        args: Record<string, unknown>;
+        createdAt: number;
+        expiresAt: number;
+        status: "PENDING" | "APPROVED" | "REJECTED" | "EXPIRED";
+        passcode?: string;
+    } | null;
+    static getPendingApprovals(): Array<{
+        traceId: string;
+        toolName: string;
+        agent: string;
+        riskScore: number;
+        riskLevel: string;
+        reason: string;
+        args: Record<string, unknown>;
+        createdAt: number;
+        expiresAt: number;
+        status: "PENDING" | "APPROVED" | "REJECTED" | "EXPIRED";
+        passcode?: string;
+    }>;
+    static updateApprovalStatus(traceId: string, status: string): void;
+    static clearAllApprovals(): void;
+    static saveDiscipline(agent: string, stats: {
+        totalCalls: number;
+        violations: number;
+        lastViolation: string | null;
+        cooldownUntil: number;
+    }): void;
+    static getDiscipline(agent: string): {
+        agent: string;
+        totalCalls: number;
+        violations: number;
+        lastViolation: string | null;
+        cooldownUntil: number;
+    } | null;
+    static getAllDiscipline(): Array<{
+        agent: string;
+        totalCalls: number;
+        violations: number;
+        lastViolation: string | null;
+        cooldownUntil: number;
+    }>;
+    static clearAllDiscipline(): void;
+    static recordDisciplineCall(agent: string, tool: string, timestamp: number, argsSize: number): void;
+    static getRecentDisciplineCalls(agent: string, sinceTs: number): Array<{
+        tool: string;
+        timestamp: number;
+        argsSize: number;
+    }>;
+    static pruneDisciplineCalls(beforeTs: number): void;
+    static clearDisciplineCalls(agent?: string): void;
+    static getCodeFileHash(filePath: string): string | null;
+    static upsertCodeFile(filePath: string, hash: string): void;
+    static replaceCodeSymbols(filePath: string, symbols: Array<{
+        name: string;
+        kind: string;
+        startLine: number;
+        endLine: number;
+        signature: string;
+        exported: boolean;
+    }>): void;
+    static replaceCodeImports(filePath: string, specifiers: string[]): void;
+    static deleteCodeFile(filePath: string): void;
+    static getAllCodeFilePaths(): string[];
+    static searchCodeSymbols(query: string, limit?: number): Array<{
+        file: string;
+        name: string;
+        kind: string;
+        startLine: number;
+        endLine: number;
+        signature: string;
+        exported: boolean;
+    }>;
+    static getCodeGraphStats(): {
+        files: number;
+        symbols: number;
+        imports: number;
+    };
+    static clearCodeGraph(): void;
+    static checkIntegrity(): {
+        ok: boolean;
+        message: string;
+    };
+    static verifyDatabaseIntegrity(mode?: "quick" | "full"): IntegrityCheckResult;
+    static getMigrationStatus(): {
+        currentVersion: number;
+        applied: import("./migrations.js").AppliedMigration[];
+        pending: Array<{
+            version: number;
+            name: string;
+        }>;
+    };
+    static backup(destinationPath?: string): BackupResult;
+    static restore(backupPath: string, targetDbPath?: string): {
+        restoredPath: string;
+        sha256: string;
+        schemaVersion: number;
+    };
+    static saveApiEndpoint(endpoint: ApiEndpoint): void;
+    static getApiEndpoints(filter?: {
+        tag?: string;
+        method?: string;
+        search?: string;
+    }): ApiEndpoint[];
+    static findApiEndpoint(pathQuery: string, method?: string): ApiEndpoint[];
+    static deleteApiEndpoint(method: string, path: string): boolean;
+    /** Replace the whole catalog with a fresh scan; returns upserted/deleted counts. */
+    static replaceApiCatalog(endpoints: ApiEndpoint[]): {
+        upserted: number;
+        deleted: number;
+    };
+    private static rowToApiEndpoint;
+    static reset(): void;
+}
+/** Canonical alias */
+export declare const Storage: typeof AtabeyStorage;