magellan 3.3.3

//! SCIP export functionality
//!
//! Implements SCIP (Source Code Intelligence Protocol) export for Magellan.
//! SCIP is a language-agnostic protocol for code indexing defined by Sourcegraph.
//!
//! # FQN Collision Limitation
//!
//! SCIP export uses the display_fqn (human-readable) for symbol encoding,
//! which may result in collisions when multiple symbols share the same name
//! in different files. The SCIP format does not natively support Magellan's
//! symbol_id field for disambiguation.
//!
//! For complete ambiguity resolution, use Magellan's native JSON/JSONL exports
//! which include both symbol_id (stable 32-char BLAKE3 hash) and canonical_fqn
//! (full identity with file path).

use anyhow::Result;
use protobuf::{EnumOrUnknown, Message};
use scip::types::{
    symbol_information::Kind, Document, Index, Metadata, Occurrence, PositionEncoding,
    SymbolInformation, SymbolRole,
};
use std::collections::HashMap;

use crate::graph::schema::SymbolNode;
use crate::ingest::detect::detect_language;

use super::CodeGraph;

// Import the GraphBackend trait for backend methods
use sqlitegraph::{BackendDirection, GraphBackend, NeighborQuery, SnapshotId};

/// SCIP export configuration
#[derive(Debug, Clone)]
pub struct ScipExportConfig {
    /// Project root for SCIP metadata
    pub project_root: String,
    /// Optional project name
    pub project_name: Option<String>,
    /// Optional version string
    pub version: Option<String>,
}

impl Default for ScipExportConfig {
    fn default() -> Self {
        Self {
            project_root: ".".to_string(),
            project_name: None,
            version: None,
        }
    }
}

/// Convert Magellan FQN to SCIP symbol format
///
/// SCIP symbol format: `scheme package/descriptor1/descriptor2/symbol.`
/// For Magellan, we use "magellan" as scheme and language as package.
///
/// # Arguments
/// * `fqn` - Fully qualified name from Magellan
/// * `language` - Programming language string
///
/// # Returns
/// SCIP-encoded symbol string
fn magellan_symbol_to_scip(fqn: &str, language: &str) -> String {
    // Split FQN based on language-specific separator
    let parts: Vec<&str> = match language {
        "rust" | "cpp" => fqn.split("::").collect(),
        "python" | "java" | "javascript" | "typescript" => fqn.split('.').collect(),
        _ => vec![fqn],
    };

    // Build SCIP symbol string
    // Format: magellan lang/descriptor1/descriptor2/symbol.
    let symbol = parts.last().unwrap_or(&"");

    // Build descriptor path (all parts except the last symbol name)
    let descriptors: Vec<String> = parts
        .iter()
        .take(parts.len().saturating_sub(1))
        .map(|s| s.to_string())
        .collect();

    // Construct full symbol string
    if descriptors.is_empty() {
        format!("magellan {}/{}.", language, symbol)
    } else {
        format!(
            "magellan {}/{}/{}.",
            language,
            descriptors.join("/"),
            symbol
        )
    }
}

/// Map Magellan symbol kind to SCIP symbol kind
fn map_symbol_kind(kind: &str) -> Kind {
    match kind {
        "Function" => Kind::Function,
        "Method" => Kind::Method,
        "Struct" => Kind::Class,
        "Enum" => Kind::Enum,
        "Module" => Kind::Namespace,
        "Class" => Kind::Class,
        "Interface" => Kind::Interface,
        "TypeAlias" => Kind::TypeAlias,
        "Union" => Kind::Union,
        "Namespace" => Kind::Namespace,
        _ => Kind::UnspecifiedKind,
    }
}

/// Export graph to SCIP format
///
/// Builds a SCIP index containing:
/// - Metadata with tool info and project root
/// - Documents per file with occurrences (definitions and references)
/// - Proper SCIP symbol encoding
///
/// # FQN Collision Limitation
///
/// **Note:** SCIP symbol format uses display_fqn (human-readable) for encoding,
/// which may have collisions when multiple symbols share the same name in different files.
/// Use Magellan's native export formats (JSON/JSONL) for complete ambiguity resolution
/// via symbol_id and canonical_fqn.
///
/// # Arguments
/// * `graph` - CodeGraph to export
/// * `config` - SCIP export configuration
///
/// # Returns
/// SCIP protobuf bytes
pub fn export_scip(graph: &CodeGraph, config: &ScipExportConfig) -> Result<Vec<u8>> {
    let mut index = Index::new();

    // Build metadata
    let mut metadata = Metadata::new();

    // Document FQN collision limitation in SCIP export
    // SCIP symbol format uses display_fqn (human-readable) which may have collisions
    // when multiple symbols share the same name in different files. Use Magellan's
    // native export formats for complete ambiguity resolution via symbol_id.
    // Note: scip 0.6.1 ToolInfo doesn't have public name/version fields
    // The metadata.project_root is the primary metadata we can set

    // Set metadata fields
    // metadata.tool_info = MessageField::some(tool_info); // This field may not exist or be different
    metadata.project_root = config.project_root.clone();

    // Set text encoding to UTF-8 (all sources are UTF-8)
    use scip::types::TextEncoding;
    metadata.text_document_encoding = EnumOrUnknown::new(TextEncoding::UTF8);

    // Set metadata on index
    index.metadata = protobuf::MessageField::some(metadata);

    // Collect all entities by file
    let mut file_to_symbols: HashMap<String, Vec<(i64, SymbolNode)>> = HashMap::new();
    let mut file_to_references: HashMap<String, Vec<super::ReferenceNode>> = HashMap::new();

    // Global symbol map for cross-file reference resolution
    // Maps (FQN or name) -> SCIP symbol string
    let mut global_symbol_map: HashMap<String, String> = HashMap::new();

    // Get all entity IDs
    let entity_ids = graph.files.backend.entity_ids()?;
    let snapshot = SnapshotId::current();

    // First pass: collect symbols and references by file
    for entity_id in entity_ids {
        let entity = graph.files.backend.get_node(snapshot, entity_id)?;

        match entity.kind.as_str() {
            "Symbol" => {
                if let Ok(symbol_node) = serde_json::from_value::<SymbolNode>(entity.data.clone()) {
                    // Find the file path from DEFINES edge
                    let file_path = if let Some(file_id) = graph
                        .files
                        .backend
                        .neighbors(
                            snapshot,
                            entity_id,
                            NeighborQuery {
                                direction: BackendDirection::Incoming,
                                edge_type: Some("DEFINES".to_string()),
                            },
                        )?
                        .first()
                    {
                        let file_entity = graph.files.backend.get_node(snapshot, *file_id)?;
                        if let Ok(file_node) =
                            serde_json::from_value::<super::FileNode>(file_entity.data)
                        {
                            file_node.path
                        } else {
                            continue;
                        }
                    } else {
                        continue;
                    };

                    // Detect language for this file to encode the symbol
                    let language = detect_language(std::path::Path::new(&file_path))
                        .map(|l| l.as_str().to_string())
                        .unwrap_or_else(|| "unknown".to_string());

                    // Build SCIP symbol for global map
                    let fqn = symbol_node.fqn.as_deref().unwrap_or("");
                    let scip_symbol = if !fqn.is_empty() {
                        magellan_symbol_to_scip(fqn, &language)
                    } else {
                        let name = symbol_node.name.as_deref().unwrap_or("");
                        format!("magellan {}/{}.", language, name)
                    };

                    // Add to global symbol map (keyed by both FQN and name)
                    if !fqn.is_empty() {
                        global_symbol_map.insert(fqn.to_string(), scip_symbol.clone());
                    }
                    if let Some(ref name) = symbol_node.name {
                        global_symbol_map.insert(name.clone(), scip_symbol.clone());
                    }

                    file_to_symbols
                        .entry(file_path)
                        .or_default()
                        .push((entity_id, symbol_node));
                }
            }
            "Reference" => {
                // Extract file from reference node
                if let Ok(ref_node) =
                    serde_json::from_value::<super::ReferenceNode>(entity.data.clone())
                {
                    file_to_references
                        .entry(ref_node.file.clone())
                        .or_default()
                        .push(ref_node);
                }
            }
            _ => {
                // Ignore other entity types
            }
        }
    }

    // Build SCIP documents
    for (file_path, symbols) in file_to_symbols {
        // Detect language from file path
        let language = if let Some(lang) = detect_language(std::path::Path::new(&file_path)) {
            lang.as_str().to_string()
        } else {
            "unknown".to_string()
        };

        // Create document
        let mut document = Document::new();
        document.relative_path = file_path.clone();
        document.language = language.clone();

        // Set position encoding to UTF-8 code units (line/col based)
        document.position_encoding =
            EnumOrUnknown::new(PositionEncoding::UTF8CodeUnitOffsetFromLineStart);

        // Add symbol occurrences (definitions)
        for (_node_id, symbol) in &symbols {
            let mut occurrence = Occurrence::new();

            // Set range [line_start, col_start, line_end, col_end]
            occurrence.range = vec![
                symbol.start_line as i32,
                symbol.start_col as i32,
                symbol.end_line as i32,
                symbol.end_col as i32,
            ];

            // Build SCIP symbol from FQN
            let fqn = symbol.fqn.as_deref().unwrap_or("");
            let scip_symbol = if !fqn.is_empty() {
                magellan_symbol_to_scip(fqn, &language)
            } else {
                // Fallback to simple name
                let name = symbol.name.as_deref().unwrap_or("");
                format!("magellan {}/{}.", language, name)
            };

            occurrence.symbol = scip_symbol.clone();

            // Set symbol roles: Definition = 1
            occurrence.symbol_roles = SymbolRole::Definition as i32;

            document.occurrences.push(occurrence);

            // Add symbol information to document.symbols
            let mut sym_info = SymbolInformation::new();
            sym_info.kind = EnumOrUnknown::new(map_symbol_kind(&symbol.kind));

            if let Some(ref name) = symbol.name {
                sym_info.display_name = name.clone();
            }

            sym_info.symbol = scip_symbol;

            document.symbols.push(sym_info);
        }

        // Add reference occurrences
        if let Some(refs) = file_to_references.get(&file_path) {
            for ref_node in refs {
                let mut occurrence = Occurrence::new();

                occurrence.range = vec![
                    ref_node.start_line as i32,
                    ref_node.start_col as i32,
                    ref_node.end_line as i32,
                    ref_node.end_col as i32,
                ];

                // Set symbol roles: ReadAccess = 8 (used for references)
                occurrence.symbol_roles = SymbolRole::ReadAccess as i32;

                // For references, we don't have direct access to the target FQN.
                // Use a placeholder indicating the reference couldn't be resolved.
                // In a full implementation, we'd store the target symbol's FQN in the reference.
                occurrence.symbol = "magellan unknown/.".to_string();

                document.occurrences.push(occurrence);
            }
        }

        index.documents.push(document);
    }

    // Serialize to protobuf bytes
    let bytes = index.write_to_bytes()?;
    Ok(bytes)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_magellan_symbol_to_scip_rust() {
        // Test Rust symbol conversion (:: separator)
        let result = magellan_symbol_to_scip("crate::module::function", "rust");
        assert_eq!(result, "magellan rust/crate/module/function.");
    }

    #[test]
    fn test_magellan_symbol_to_scip_rust_simple() {
        // Test simple Rust symbol (no namespace)
        let result = magellan_symbol_to_scip("myfunction", "rust");
        assert_eq!(result, "magellan rust/myfunction.");
    }

    #[test]
    fn test_magellan_symbol_to_scip_python() {
        // Test Python symbol conversion (. separator)
        let result = magellan_symbol_to_scip("package.module.function", "python");
        assert_eq!(result, "magellan python/package/module/function.");
    }

    #[test]
    fn test_magellan_symbol_to_scip_java() {
        // Test Java symbol conversion
        let result = magellan_symbol_to_scip("com.example.Class.method", "java");
        assert_eq!(result, "magellan java/com/example/Class/method.");
    }

    #[test]
    fn test_magellan_symbol_to_scip_cpp() {
        // Test C++ symbol conversion (:: separator)
        let result = magellan_symbol_to_scip("std::vector::push_back", "cpp");
        assert_eq!(result, "magellan cpp/std/vector/push_back.");
    }

    #[test]
    fn test_magellan_symbol_to_scip_javascript() {
        // Test JavaScript symbol conversion
        let result = magellan_symbol_to_scip("module.function", "javascript");
        assert_eq!(result, "magellan javascript/module/function.");
    }

    #[test]
    fn test_magellan_symbol_to_scip_typescript() {
        // Test TypeScript symbol conversion
        let result = magellan_symbol_to_scip("namespace.Class.method", "typescript");
        assert_eq!(result, "magellan typescript/namespace/Class/method.");
    }

    #[test]
    fn test_magellan_symbol_to_scip_unknown_language() {
        // Test unknown language (no separator)
        let result = magellan_symbol_to_scip("some_symbol", "unknown");
        assert_eq!(result, "magellan unknown/some_symbol.");
    }

    #[test]
    fn test_map_symbol_kind() {
        // Test symbol kind mapping
        assert_eq!(map_symbol_kind("Function"), Kind::Function);
        assert_eq!(map_symbol_kind("Method"), Kind::Method);
        assert_eq!(map_symbol_kind("Struct"), Kind::Class);
        assert_eq!(map_symbol_kind("Enum"), Kind::Enum);
        assert_eq!(map_symbol_kind("Module"), Kind::Namespace);
        assert_eq!(map_symbol_kind("Class"), Kind::Class);
        assert_eq!(map_symbol_kind("Interface"), Kind::Interface);
        assert_eq!(map_symbol_kind("TypeAlias"), Kind::TypeAlias);
        assert_eq!(map_symbol_kind("Union"), Kind::Union);
        assert_eq!(map_symbol_kind("Namespace"), Kind::Namespace);
        assert_eq!(map_symbol_kind("Unknown"), Kind::UnspecifiedKind);
    }

    #[test]
    fn test_scip_export_config_default() {
        let config = ScipExportConfig::default();
        assert_eq!(config.project_root, ".");
        assert!(config.project_name.is_none());
        assert!(config.version.is_none());
    }
}