use anyhow::Result;
use searchi_core::{SearchIndex, SearchOptions, SourceLocation};
use std::fs;

#[test]
fn code_scope_filters_before_limit_and_extracts_build_files() -> Result<()> {
    let root = tempfile::tempdir()?;
    for number in 0..100 {
        fs::write(
            root.path().join(format!("early{number}.txt")),
            "refreshAccessToken",
        )?;
    }
    fs::write(
        root.path().join("last.go"),
        "func refreshAccessToken() {}\n",
    )?;
    fs::write(
        root.path().join("Makefile"),
        "refreshAccessToken:\n\techo ready\n",
    )?;
    fs::write(root.path().join("large.rs"), "\n".repeat(1_000_000))?;
    let mut index = SearchIndex::open(":memory:")?;
    index.index_folder(root.path())?;
    assert_eq!(index.search("large.rs", 1)?[0].filename, "large.rs");
    let options = SearchOptions {
        code_only: true,
        ..Default::default()
    };
    let results = index.search_with_options("refreshAccessToken", 2, &options)?;
    assert_eq!(results.len(), 2);
    assert!(results
        .iter()
        .all(|r| r.filename == "last.go" || r.filename == "Makefile"));
    assert!(results
        .iter()
        .all(|r| r.source.is_some() && r.snippet.contains("refreshAccessToken")));
    Ok(())
}

#[test]
fn overlapping_source_passages_preserve_routine_and_actual_lines() -> Result<()> {
    let root = tempfile::tempdir()?;
    let file = root.path().join("module.swift");
    let mut lines = vec!["// filler\n"; 166];
    lines[116] = "func importantRoutine() {\n";
    lines[136] = "  return finalValue\n";
    lines[137] = "}\n";
    fs::write(&file, lines.concat())?;
    let mut index = SearchIndex::open(":memory:")?;
    index.index_folder(root.path())?;
    let path = fs::canonicalize(file)?.to_string_lossy().into_owned();
    let passage = index
        .source_passage(
            &path,
            &SourceLocation {
                kind: "line".into(),
                number: 97,
            },
            8192,
        )?
        .unwrap();
    assert_eq!(passage.text, lines[96..].concat());
    assert!(passage.text.contains("importantRoutine") && passage.text.contains("finalValue"));
    Ok(())
}

#[test]
fn old_code_index_reextracts_unchanged_source_on_upgrade() -> Result<()> {
    let root = tempfile::tempdir()?;
    let files = root.path().join("files");
    fs::create_dir(&files)?;
    fs::write(files.join("module.java"), "class Evidence {}")?;
    fs::write(files.join("notes.txt"), "ordinary evidence")?;
    let db = root.path().join("index.db");
    let mut index = SearchIndex::open(&db)?;
    index.index_folder(&files)?;
    drop(index);
    let connection = rusqlite::Connection::open(&db)?;
    connection.execute_batch("PRAGMA user_version=10;")?;
    drop(connection);
    let mut index = SearchIndex::open(&db)?;
    assert_eq!(index.search("module.java", 1)?[0].filename, "module.java");
    let report = index.index_folder(&files)?;
    assert_eq!(report.indexed, 1);
    assert_eq!(report.unchanged, 1);
    assert!(index
        .search("Evidence", 10)?
        .iter()
        .any(|r| r.filename == "module.java" && r.source.is_some()));
    Ok(())
}