import type { MemoryEntryStatus, MemoryEntryType, MemorySearchStatus } from '../../contracts/memory';
import type { DreamNeighborPageResult, DreamNeighborQuery, VectorSnapshot, MemoryVectorIndexManifest, MemoryVectorIndexPresetStatus, MemoryVectorIndexState } from '../../contracts/vector_index';
import type { LivingMemoryVectorIndexErrorCode } from './errors';
/**
 * 持久账本只接受终态；进行中与瞬态条件只存在于主进程 overlay。
 * 展示用的 MemoryVectorIndexPresetStatus 合并 overlay 后可为四态，两者不可混用。
 */
export interface VectorIndexPresetStateMark {
    presetId: string;
    state: 'ready' | 'dirty';
    expectedCount: number;
    indexedCount: number;
    lastError: string | null;
    updatedAt: number;
}
export interface VectorIndexInspection {
    vectorExtensionVersion: string;
    manifest: MemoryVectorIndexManifest | null;
    indexedCount: number;
    inventory: {
        presetId: string;
        indexedCount: number;
    }[];
    presets: MemoryVectorIndexPresetStatus[];
}
export interface VectorIndexInventoryItem {
    memoryId: string;
    presetId: string;
    sourceConversationId: string | null;
    status: MemoryEntryStatus;
    type: MemoryEntryType;
    contentHash: string;
    keywordsHash: string;
    updatedAt: number;
}
export interface VectorIndexDocument extends VectorIndexInventoryItem {
    keywords: string[];
}
export interface VectorIndexReplaceUpsert {
    vectorAction: 'replace';
    document: VectorIndexDocument;
    vector: Float32Array<ArrayBuffer>;
}
export interface VectorIndexPreserveUpsert {
    vectorAction: 'preserve';
    document: VectorIndexDocument;
}
export type VectorIndexUpsert = VectorIndexReplaceUpsert | VectorIndexPreserveUpsert;
export interface VectorIndexMutation {
    presetId: string;
    upserts: VectorIndexUpsert[];
    deletes: string[];
}
export interface VectorIndexFilter {
    presetId: string;
    conversationId?: string;
    excludedMemoryIds?: string[];
    types: MemoryEntryType[] | null;
    memoryStatus: MemorySearchStatus;
}
export interface VectorIndexKnnQuery extends VectorIndexFilter {
    vector: Float32Array<ArrayBuffer>;
    limit: number;
}
export interface VectorIndexHybridQuery extends VectorIndexKnnQuery {
    keywords: string[];
    minSimilarity: number;
}
export interface VectorIndexKnnHit {
    memoryId: string;
    cosineScore: number;
}
export interface VectorIndexHybridHit extends VectorIndexKnnHit {
    keywordMatchCount: number;
    boostedScore: number;
}
export interface VectorIndexInventoryPage {
    items: VectorIndexInventoryItem[];
    nextCursor: string | null;
}
export interface VectorIndexWorkerCommandMap {
    open: {
        input: {
            databaseDirectory: string;
            previousDatabaseDirectory: string;
        };
        result: VectorIndexInspection;
    };
    openCandidate: {
        input: {
            databaseDirectory: string;
        };
        result: VectorIndexInspection;
    };
    inspect: {
        input: Record<never, never>;
        result: VectorIndexInspection;
    };
    upgradeSchema: {
        input: Record<never, never>;
        result: {
            schemaVersion: number;
        };
    };
    queryHybrid: {
        input: VectorIndexHybridQuery;
        result: VectorIndexHybridHit[];
    };
    readVectorSnapshot: {
        input: {
            presetId: string;
            memoryIds: string[];
        };
        result: VectorSnapshot;
    };
    buildDreamAnnIndex: {
        input: Record<never, never>;
        result: {
            built: boolean;
        };
    };
    queryDreamNeighbors: {
        input: {
            queries: DreamNeighborQuery[];
            cancellation: Int32Array<SharedArrayBuffer>;
        };
        result: DreamNeighborPageResult;
    };
    applyMutation: {
        input: VectorIndexMutation;
        result: {
            indexedCount: number;
        };
    };
    clearPreset: {
        input: {
            presetId: string;
        };
        result: {
            deletedCount: number;
        };
    };
    readInventoryPage: {
        input: {
            presetId: string | null;
            afterMemoryId: string | null;
            limit: number;
        };
        result: VectorIndexInventoryPage;
    };
    markPresetState: {
        input: VectorIndexPresetStateMark;
        result: MemoryVectorIndexPresetStatus;
    };
    openRebuildDatabase: {
        input: {
            databaseDirectory: string;
            manifest: MemoryVectorIndexManifest;
        };
        result: VectorIndexInspection;
    };
    appendRebuildBatch: {
        input: {
            presetId: string;
            upserts: VectorIndexReplaceUpsert[];
        };
        result: {
            indexedCount: number;
        };
    };
    validateAndCloseRebuildDatabase: {
        input: {
            expectedCount: number;
        };
        result: {
            indexedCount: number;
        };
    };
    closeDatabase: {
        input: Record<never, never>;
        result: {
            closed: true;
        };
    };
    abortRebuild: {
        input: Record<never, never>;
        result: VectorIndexInspection;
    };
    dispose: {
        input: Record<never, never>;
        result: {
            disposed: true;
        };
    };
}
export type VectorIndexWorkerCommandName = keyof VectorIndexWorkerCommandMap;
export type VectorIndexWorkerCommand<Name extends VectorIndexWorkerCommandName = VectorIndexWorkerCommandName> = Name extends VectorIndexWorkerCommandName ? {
    type: Name;
} & VectorIndexWorkerCommandMap[Name]['input'] : never;
export type VectorIndexWorkerResult<Name extends VectorIndexWorkerCommandName> = VectorIndexWorkerCommandMap[Name]['result'];
export interface VectorIndexWorkerRequest<Name extends VectorIndexWorkerCommandName = VectorIndexWorkerCommandName> {
    id: number;
    command: VectorIndexWorkerCommand<Name>;
}
export interface VectorIndexWorkerError {
    name: string;
    message: string;
    stack: string | null;
    code: LivingMemoryVectorIndexErrorCode | null;
    state: MemoryVectorIndexState | null;
}
export interface VectorIndexWorkerWarning {
    type: 'warning';
    error: VectorIndexWorkerError;
}
export type VectorIndexWorkerResponse = {
    [Name in VectorIndexWorkerCommandName]: {
        id: number;
        type: Name;
        ok: true;
        result: VectorIndexWorkerResult<Name>;
    } | {
        id: number;
        type: Name;
        ok: false;
        error: VectorIndexWorkerError;
    };
}[VectorIndexWorkerCommandName];