windows-exe-decompiler-mcp-server 0.1.0

package/dist/database.d.ts

@@ -0,0 +1,227 @@
+/**
+ * Database module for Windows EXE Decompiler MCP Server
+ * Manages SQLite database schema and operations
+ */
+import Database from 'better-sqlite3';
+/**
+ * Database interface types
+ */
+export interface Sample {
+    id: string;
+    sha256: string;
+    md5: string | null;
+    size: number;
+    file_type: string | null;
+    created_at: string;
+    source: string | null;
+}
+export interface Analysis {
+    id: string;
+    sample_id: string;
+    stage: string;
+    backend: string;
+    status: string;
+    started_at: string | null;
+    finished_at: string | null;
+    output_json: string | null;
+    metrics_json: string | null;
+}
+export interface Function {
+    sample_id: string;
+    address: string;
+    name: string | null;
+    size: number | null;
+    score: number | null;
+    tags: string | null;
+    summary: string | null;
+    caller_count: number | null;
+    callee_count: number | null;
+    is_entry_point: number | null;
+    is_exported: number | null;
+    callees: string | null;
+}
+export interface Artifact {
+    id: string;
+    sample_id: string;
+    type: string;
+    path: string;
+    sha256: string;
+    mime: string | null;
+    created_at: string;
+}
+export interface CachedResult {
+    key: string;
+    data: unknown;
+    created_at: string;
+    expires_at: string | null;
+}
+/**
+ * Database manager class
+ */
+export declare class DatabaseManager {
+    private db;
+    constructor(dbPath: string);
+    /**
+     * Initialize database schema
+     */
+    private initializeSchema;
+    /**
+     * Get the underlying database instance
+     */
+    getDatabase(): Database.Database;
+    /**
+     * Close the database connection
+     */
+    close(): void;
+    /**
+     * Execute a transaction
+     */
+    transaction<T>(fn: () => T): T;
+    /**
+     * Insert a new sample
+     */
+    insertSample(sample: Sample): void;
+    /**
+     * Find a sample by ID
+     */
+    findSample(sampleId: string): Sample | undefined;
+    /**
+     * Find a sample by SHA256
+     */
+    findSampleBySha256(sha256: string): Sample | undefined;
+    /**
+     * Insert a new analysis
+     */
+    insertAnalysis(analysis: Analysis): void;
+    /**
+     * Update an analysis
+     */
+    updateAnalysis(analysisId: string, updates: Partial<Omit<Analysis, 'id' | 'sample_id'>>): void;
+    /**
+     * Find an analysis by ID
+     */
+    findAnalysis(analysisId: string): Analysis | undefined;
+    /**
+     * Find all analyses for a sample
+     */
+    findAnalysesBySample(sampleId: string): Analysis[];
+    /**
+     * Find recent samples ordered by creation time.
+     */
+    findRecentSamples(limit?: number): Sample[];
+    /**
+     * Mark stale running analyses as failed so persisted status does not remain misleading.
+     */
+    reapStaleAnalyses(maxRuntimeMs: number, sampleId?: string): Analysis[];
+    /**
+     * Insert a new function
+     */
+    insertFunction(func: Function): void;
+    /**
+     * Find all functions for a sample
+     */
+    findFunctions(sampleId: string): Function[];
+    /**
+     * Find functions by sample with score ordering
+     */
+    findFunctionsByScore(sampleId: string, limit?: number): Function[];
+    /**
+     * Update a function
+     */
+    updateFunction(sampleId: string, address: string, updates: Partial<Omit<Function, 'sample_id' | 'address'>>): void;
+    /**
+     * Insert a new artifact
+     */
+    insertArtifact(artifact: Artifact): void;
+    /**
+     * Find all artifacts for a sample
+     */
+    findArtifacts(sampleId: string): Artifact[];
+    /**
+     * Find artifacts by sample and type
+     */
+    findArtifactsByType(sampleId: string, type: string): Artifact[];
+    /**
+     * Get cached result from database
+     * Requirements: 20.5
+     */
+    getCachedResult(key: string): Promise<{
+        data: unknown;
+        createdAt?: string;
+        expiresAt?: string;
+        sampleSha256?: string;
+    } | null>;
+    /**
+     * Set cached result in database
+     * Requirements: 20.5
+     */
+    setCachedResult(key: string, data: unknown, expiresAt?: string, sampleSha256?: string): Promise<void>;
+    /**
+     * Delete expired cache entries
+     */
+    cleanExpiredCache(): number;
+    /**
+     * Get recent cache entries for prewarming
+     * Requirements: 26.1 (cache prewarming), 26.2 (query optimization)
+     *
+     * @param limit - Maximum number of entries to return
+     * @returns Array of cache entries ordered by creation time (most recent first)
+     */
+    getRecentCacheEntries(limit: number): Promise<Array<{
+        key: string;
+        data: string;
+        expires_at: string | null;
+    }>>;
+    /**
+     * Get cache entries for a specific sample
+     * Requirements: 26.1 (cache prewarming), 26.2 (query optimization)
+     *
+     * @param sampleSha256 - SHA256 hash of the sample
+     * @returns Array of cache entries for the sample
+     */
+    getCacheEntriesBySample(sampleSha256: string): Promise<Array<{
+        key: string;
+        data: string;
+        expires_at: string | null;
+    }>>;
+    /**
+     * Batch insert functions for better performance
+     * Requirements: 26.2 (database query optimization)
+     *
+     * @param functions - Array of functions to insert
+     */
+    insertFunctionsBatch(functions: Function[]): void;
+    /**
+     * Batch insert artifacts for better performance
+     * Requirements: 26.2 (database query optimization)
+     *
+     * @param artifacts - Array of artifacts to insert
+     */
+    insertArtifactsBatch(artifacts: Artifact[]): void;
+    /**
+     * Optimize database by running VACUUM and ANALYZE
+     * Requirements: 26.2 (database query optimization)
+     *
+     * Should be run periodically to maintain performance
+     */
+    optimizeDatabase(): void;
+    /**
+     * Get database statistics for monitoring
+     * Requirements: 26.2 (database query optimization)
+     *
+     * @returns Object with database statistics
+     */
+    getDatabaseStats(): {
+        sampleCount: number;
+        analysisCount: number;
+        functionCount: number;
+        artifactCount: number;
+        cacheCount: number;
+        dbSizeBytes: number;
+    };
+}
+/**
+ * Create and initialize a database instance
+ */
+export declare function createDatabase(dbPath: string): DatabaseManager;
+//# sourceMappingURL=database.d.ts.map