syntax = "proto3";

package exa.context_module_pb;

import "buf/validate/validate.proto";
import "exa/chat_pb/chat.proto";
import "exa/codeium_common_pb/codeium_common.proto";

option go_package = "github.com/Exafunction/Exafunction/exa/context_module_pb";

message ContextChangeEvent {
    ContextRefreshReason context_refresh_reason = 6;
    
    oneof context_change_event {
        ContextChangeActiveDocument context_change_active_document = 1;
        ContextChangeCursorPosition context_change_cursor_position = 2;
        ContextChangeChatMessageReceived context_change_chat_message_received = 3;
        ContextChangeOpenDocuments context_change_open_documents = 4;
        ContextChangeOracleItems context_change_oracle_items = 5;
        ContextChangePinnedContext context_change_pinned_context = 7;
        ContextChangePinnedGuideline context_change_pinned_guideline = 8;
        ContextChangeActiveNode context_change_active_node = 10;
    }
    
    reserved 9;
}

message ContextChangeActiveDocument {
    codeium_common_pb.Document document = 1;
}

message ContextChangeCursorPosition {
    codeium_common_pb.Document document = 2;
    
    reserved 1, 3;
    
    reserved "position", "refresh_reason";
}

message ContextChangeChatMessageReceived {
    repeated chat_pb.ChatMessage chat_messages = 1;
}

message ContextChangeOpenDocuments {
    repeated codeium_common_pb.Document other_open_documents = 1;
}

message ContextChangeOracleItems {
    repeated codeium_common_pb.CodeContextItem oracle_items = 1;
}

message ContextChangePinnedContext {
    oneof scope {
        codeium_common_pb.ContextScope pinned_scope = 1;
        codeium_common_pb.ContextScope mentioned_scope = 2;
    }
}

message ContextChangePinnedGuideline {
    codeium_common_pb.Guideline guideline = 1;
}

message ContextChangeActiveNode {
    codeium_common_pb.CodeContextItem active_node = 1;
    codeium_common_pb.Document document = 2;
    bool actual_node_change = 3;
}

message RetrievedCodeContextItemMetadata {
    repeated codeium_common_pb.CodeContextSource context_sources = 1;
    codeium_common_pb.CodeContextType context_type = 2;
    string scorer = 3;
    float score = 4;
    map<string, CodeContextProviderMetadata> provider_metadata = 5;
    bool is_in_pinned_scope = 6;
}

message CciWithSubrangeWithRetrievalMetadata {
    codeium_common_pb.CciWithSubrange cci_with_subrange = 1;
    RetrievedCodeContextItemMetadata metadata = 2;
}

message CodeContextItemWithRetrievalMetadata {
    codeium_common_pb.CodeContextItem code_context_item = 1;
    RetrievedCodeContextItemMetadata metadata = 2;
}

message FileNameWithRetrievalMetadata {
    string absolute_uri = 1;
    RetrievedCodeContextItemMetadata metadata = 2;
}

message CodeContextProviderMetadata {
    float relative_weight = 1;
}

message ContextModuleStats {
    ContextModuleStateStats context_module_state_stats = 1;
    CodeContextItemIndexStats code_context_item_index_stats = 2;
    int64 get_stats_latency_ms = 3;
    int64 context_module_age_s = 4;
}

message ContextModuleStateStats {
    int64 cci_per_source_bytes = 1;
    int64 active_document_bytes = 2;
    int64 other_open_documents_bytes = 3;
}

message CodeContextItemIndexStats {
    int64 all_ccis_bytes = 1;
    int64 num_ccis_tracked = 2;
    int64 term_frequency_map_bytes = 3;
    int64 num_terms_tracked = 4;
    int64 file_to_cci_map_bytes = 5;
    int64 num_files_tracked = 6;
    int64 last_modified_bytes = 7;
    int64 hash_map_bytes = 8;
}

message PersistentContextModuleState {
    codeium_common_pb.Guideline pinned_guideline = 1;
    codeium_common_pb.ContextScope pinned_context_scope = 2;
}

message ContextModuleResult {
    repeated CciWithSubrangeWithRetrievalMetadata retrieved_cci_with_subranges = 1;
    codeium_common_pb.Document active_document = 2;
    codeium_common_pb.DocumentOutline active_document_outline = 5;
    LocalNodeState local_node_state = 3;
    codeium_common_pb.Guideline guideline = 4;
    repeated codeium_common_pb.Document open_documents = 6;
}

message LocalNodeState {
    codeium_common_pb.CodeContextItem current_node = 1;
    codeium_common_pb.CodeContextItem closest_above_node = 2;
    codeium_common_pb.CodeContextItem closest_below_node = 3;
}

enum ContextChangeType {
    CONTEXT_CHANGE_TYPE_UNSPECIFIED = 0;
    CONTEXT_CHANGE_TYPE_ACTIVE_DOCUMENT = 1;
    CONTEXT_CHANGE_TYPE_CURSOR_POSITION = 2;
    CONTEXT_CHANGE_TYPE_CHAT_MESSAGE_RECEIVED = 3;
    CONTEXT_CHANGE_TYPE_OPEN_DOCUMENTS = 4;
    CONTEXT_CHANGE_TYPE_ORACLE_ITEMS = 5;
    CONTEXT_CHANGE_TYPE_PINNED_CONTEXT = 6;
    CONTEXT_CHANGE_TYPE_PINNED_GUIDELINE = 7;
    CONTEXT_CHANGE_TYPE_ACTIVE_NODE = 9;
}

enum ContextUseCase {
    CONTEXT_USE_CASE_UNSPECIFIED = 0;
    CONTEXT_USE_CASE_AUTOCOMPLETE = 1;
    CONTEXT_USE_CASE_CHAT = 2;
    CONTEXT_USE_CASE_CHAT_COMPLETION = 3;
    CONTEXT_USE_CASE_CORTEX_RESEARCH = 4;
    CONTEXT_USE_CASE_EVAL = 5;
    CONTEXT_USE_CASE_CHAT_COMPLETION_GENERATE = 6;
    CONTEXT_USE_CASE_SUPERCOMPLETE = 7;
    CONTEXT_USE_CASE_FAST_APPLY = 8;
    CONTEXT_USE_CASE_COMMAND_TERMINAL = 9;
}

enum ContextRefreshReason {
    CONTEXT_REFRESH_REASON_UNSPECIFIED = 0;
    CONTEXT_REFRESH_REASON_AUTOCOMPLETE = 1;
    CONTEXT_REFRESH_REASON_CHAT = 2;
    CONTEXT_REFRESH_REASON_IDE_ACTION = 4;
    CONTEXT_REFRESH_REASON_CHAT_COMPLETION = 5;
}