cartog 0.8.1

use anyhow::Result;
use tree_sitter::{Language, Parser};

use super::{js_shared, ExtractionResult, Extractor};

pub struct JavaScriptExtractor {
    parser: Parser,
    queries: js_shared::JsQueries,
}

impl JavaScriptExtractor {
    pub fn new() -> Self {
        let lang = Language::new(tree_sitter_javascript::LANGUAGE);
        let mut parser = Parser::new();
        parser
            .set_language(&lang)
            .expect("JavaScript grammar should always load");
        let queries = js_shared::JsQueries::new(&lang);
        Self { parser, queries }
    }
}

impl Default for JavaScriptExtractor {
    fn default() -> Self {
        Self::new()
    }
}

impl Extractor for JavaScriptExtractor {
    fn extract(&mut self, source: &str, file_path: &str) -> Result<ExtractionResult> {
        js_shared::extract(&mut self.parser, &self.queries, source, file_path)
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::types::{EdgeKind, SymbolKind, Visibility};

    fn extract_js(source: &str) -> ExtractionResult {
        let mut ext = JavaScriptExtractor::new();
        ext.extract(source, "test.js").unwrap()
    }

    #[test]
    fn test_function_declaration() {
        let result = extract_js(
            r#"
function greet(name) {
    return `Hello, ${name}!`;
}
"#,
        );
        assert_eq!(result.symbols.len(), 1);
        assert_eq!(result.symbols[0].name, "greet");
        assert_eq!(result.symbols[0].kind, SymbolKind::Function);
    }

    #[test]
    fn test_arrow_function() {
        let result = extract_js(
            r#"
const add = (a, b) => a + b;
"#,
        );
        let func = result.symbols.iter().find(|s| s.name == "add");
        assert!(func.is_some());
        assert_eq!(func.unwrap().kind, SymbolKind::Function);
    }

    #[test]
    fn test_class_with_inheritance() {
        let result = extract_js(
            r#"
class Animal {
    constructor(name) {
        this.name = name;
    }

    speak() {
        return `${this.name} makes a sound`;
    }
}

class Dog extends Animal {
    speak() {
        return `${this.name} barks`;
    }
}
"#,
        );

        let classes: Vec<_> = result
            .symbols
            .iter()
            .filter(|s| s.kind == SymbolKind::Class)
            .collect();
        assert_eq!(classes.len(), 2);

        let inherits: Vec<_> = result
            .edges
            .iter()
            .filter(|e| e.kind == EdgeKind::Inherits)
            .collect();
        assert_eq!(inherits.len(), 1);
        assert_eq!(inherits[0].target_name, "Animal");
    }

    #[test]
    fn test_imports_require() {
        let result = extract_js(
            r#"
import { readFile, writeFile } from 'fs';
import path from 'path';
"#,
        );

        let imports: Vec<_> = result
            .symbols
            .iter()
            .filter(|s| s.kind == SymbolKind::Import)
            .collect();
        assert_eq!(imports.len(), 2);
    }

    #[test]
    fn test_function_calls() {
        let result = extract_js(
            r#"
function process() {
    const data = fetchData();
    transform(data);
}
"#,
        );

        let calls: Vec<_> = result
            .edges
            .iter()
            .filter(|e| e.kind == EdgeKind::Calls)
            .collect();
        assert_eq!(calls.len(), 2);
    }

    #[test]
    fn test_exported_class() {
        let result = extract_js(
            r#"
export class Router {
    constructor(routes) {
        this.routes = routes;
    }

    match(path) {
        return this.routes[path];
    }
}

export function createRouter() {
    return new Router({});
}
"#,
        );

        let router = result.symbols.iter().find(|s| s.name == "Router").unwrap();
        assert_eq!(router.kind, SymbolKind::Class);

        let create = result
            .symbols
            .iter()
            .find(|s| s.name == "createRouter")
            .unwrap();
        assert_eq!(create.kind, SymbolKind::Function);

        let method_names: Vec<&str> = result
            .symbols
            .iter()
            .filter(|s| s.kind == SymbolKind::Method)
            .map(|s| s.name.as_str())
            .collect();
        assert!(method_names.contains(&"constructor"));
        assert!(method_names.contains(&"match"));
    }

    #[test]
    fn test_top_level_expression_no_crash() {
        // Top-level calls have no parent context — should not crash or produce
        // spurious edges (calls require a source symbol)
        let result = extract_js(
            r#"
console.log("booting");
setupDatabase();

function setupDatabase() {}
"#,
        );

        let func = result.symbols.iter().find(|s| s.name == "setupDatabase");
        assert!(func.is_some());
        // No calls edges from top-level (no context symbol to attach to)
        let orphan_calls: Vec<_> = result
            .edges
            .iter()
            .filter(|e| e.kind == EdgeKind::Calls && e.source_id.is_empty())
            .collect();
        assert!(orphan_calls.is_empty());
    }

    #[test]
    fn test_jsdoc_comment() {
        let result = extract_js(
            r#"
/** Adds two numbers. */
function add(a, b) {
    return a + b;
}
"#,
        );

        let add = result.symbols.iter().find(|s| s.name == "add").unwrap();
        assert_eq!(add.docstring.as_deref(), Some("Adds two numbers."));
    }

    #[test]
    fn test_throw_expressions() {
        let result = extract_js(
            r#"
function validate(x) {
    if (!x) throw new TypeError("missing");
    if (x < 0) throw new RangeError("negative");
}
"#,
        );

        let raises: Vec<_> = result
            .edges
            .iter()
            .filter(|e| e.kind == EdgeKind::Raises)
            .map(|e| e.target_name.as_str())
            .collect();
        assert!(raises.contains(&"TypeError"));
        assert!(raises.contains(&"RangeError"));
    }

    #[test]
    fn test_private_field_convention() {
        let result = extract_js(
            r#"
class Cache {
    #store;

    constructor() {
        this.#store = new Map();
    }

    _internalMethod() {
        return this.#store;
    }
}
"#,
        );

        let store = result.symbols.iter().find(|s| s.name == "#store");
        assert!(store.is_some());
        assert_eq!(store.unwrap().visibility, Visibility::Private);

        let internal = result.symbols.iter().find(|s| s.name == "_internalMethod");
        assert!(internal.is_some());
        assert_eq!(internal.unwrap().visibility, Visibility::Protected);
    }
}