//! Single-file processing pipeline.
//!
//! Handles reading, transforming, caching, and outputting a single file or
//! stdin stream. Multi-file orchestration lives in [`crate::multi`].

use std::fs;
use std::io::{self, BufWriter, Read, Write};
use std::path::Path;

use rskim_core::{
    Language, Mode, TransformConfig, detect_language_from_path, transform_auto_with_config,
    transform_with_config, transform_with_line_map,
};

use crate::{cache, cascade, cascade::TruncationOptions, tokens};

/// Maximum input size to prevent memory exhaustion (50MB)
const MAX_INPUT_SIZE: usize = 50 * 1024 * 1024;

/// Options for processing a single file
#[derive(Debug, Clone, Copy)]
pub(crate) struct ProcessOptions {
    /// Transformation mode
    pub(crate) mode: Mode,
    /// Explicit language override (None for auto-detection)
    pub(crate) explicit_lang: Option<Language>,
    /// Whether to use cache
    pub(crate) use_cache: bool,
    /// Whether to compute token statistics (for --show-stats)
    pub(crate) show_stats: bool,
    /// Truncation options (max_lines, last_lines, token_budget)
    pub(crate) trunc: TruncationOptions,
    /// Whether to annotate output with source line numbers (`--line-numbers` / `-n`)
    pub(crate) line_numbers: bool,
}

/// Result of processing a file
#[derive(Debug)]
#[must_use]
pub(crate) struct ProcessResult {
    /// Transformed output
    pub(crate) output: String,
    /// Original token count (if computed)
    pub(crate) original_tokens: Option<usize>,
    /// Transformed token count (if computed)
    pub(crate) transformed_tokens: Option<usize>,
    /// Whether the output guardrail was triggered (compressed > raw)
    pub(crate) guardrail_triggered: bool,
    /// Parse quality tier: "full", "degraded", or "passthrough".
    ///
    /// - "passthrough" — Mode::Full, no transformation applied
    /// - "degraded"    — tree-sitter reported syntax errors
    /// - "full"        — clean parse, no errors
    ///
    /// `None` for cache hits (tier was not recorded at write time).
    pub(crate) parse_tier: Option<&'static str>,
}

/// Determine the parse quality tier from the mode and whether the parser reported errors.
///
/// - "passthrough" — Mode::Full; no transformation was applied
/// - "degraded"    — tree-sitter reported syntax errors
/// - "full"        — clean parse, no syntax errors
pub(crate) fn parse_tier_from(mode: Mode, has_errors: bool) -> &'static str {
    if mode == Mode::Full {
        "passthrough"
    } else if has_errors {
        "degraded"
    } else {
        "full"
    }
}

/// Apply line number annotations to `output` after the guardrail decision.
///
/// When `guardrail_triggered` is true the output is raw source, so an identity
/// map is used.  When `computed_map` is `Some`, it is used directly.
///
/// When `computed_map` is `None` (serde non-full modes, or language detection
/// failure), line numbers are **skipped** because the output is restructured
/// and has no meaningful line-for-line correspondence with the original source.
///
/// AC-11: Identity map is applied when guardrail emits raw source.
/// AC-15: Serde non-full modes skip line numbers (computed_map is None).
pub(crate) fn apply_line_numbers(
    output: String,
    line_numbers: bool,
    guardrail_triggered: bool,
    computed_map: Option<Vec<usize>>,
) -> String {
    if !line_numbers {
        return output;
    }
    if guardrail_triggered {
        // Guardrail emitted raw source — use identity map
        let map = crate::format::identity_line_map(&output);
        return crate::format::format_with_line_numbers(&output, &map);
    }
    match computed_map {
        Some(map) => crate::format::format_with_line_numbers(&output, &map),
        // No line map available (serde non-full, language detection failure):
        // skip annotation — restructured output has no source correspondence.
        None => output,
    }
}

/// Count tokens for both original and transformed text, returning `(None, None)` on failure.
///
/// Centralises the paired token-counting pattern used across the processing pipeline.
pub(crate) fn count_token_pair(
    original: &str,
    transformed: &str,
) -> (Option<usize>, Option<usize>) {
    match (
        tokens::count_tokens(original),
        tokens::count_tokens(transformed),
    ) {
        (Ok(orig), Ok(trans)) => (Some(orig), Some(trans)),
        _ => (None, None),
    }
}

/// Report token statistics to stderr if token counts are available
pub(crate) fn report_token_stats(
    original_tokens: Option<usize>,
    transformed_tokens: Option<usize>,
    suffix: &str,
) {
    if let (Some(orig), Some(trans)) = (original_tokens, transformed_tokens) {
        let stats = tokens::TokenStats::new(orig, trans);
        eprintln!("\n[skim] {}{}", stats.format(), suffix);
    }
}

/// Write a single-input result to stdout and optionally report token stats to stderr.
///
/// Used by both `process_stdin` and the single-file path in `main()`.
/// Multi-file paths use their own output logic in `process_files()`.
pub(crate) fn write_result_and_stats(
    result: &ProcessResult,
    show_stats: bool,
) -> anyhow::Result<()> {
    let stdout = io::stdout();
    let mut writer = BufWriter::new(stdout.lock());
    write!(writer, "{}", result.output)?;
    writer.flush()?;

    if show_stats {
        report_token_stats(result.original_tokens, result.transformed_tokens, "");
    }

    Ok(())
}

/// Try to return a result from cache, handling token recount when needed.
///
/// Returns `Some(ProcessResult)` on cache hit, `None` on cache miss.
/// When stats are requested but the cached entry lacks token counts,
/// the original file is read to compute them on the fly.
fn try_cached_result(
    path: &Path,
    options: &ProcessOptions,
) -> anyhow::Result<Option<ProcessResult>> {
    if !options.use_cache {
        return Ok(None);
    }

    let Some(hit) = cache::read_cache(path, options.mode, &options.trunc, options.line_numbers)
    else {
        return Ok(None);
    };

    // If the cache entry was written without token counts, read the original
    // file and count tokens for both source and output -- but only when
    // --show-stats is active. Analytics background threads handle their own
    // token counting, so we don't erode cache speedup for analytics alone.
    let needs_recount = hit.original_tokens.is_none() && options.show_stats;
    let (orig_tokens, trans_tokens) = if needs_recount {
        let contents = read_and_validate(path)?;
        count_token_pair(&contents, &hit.content)
    } else {
        (hit.original_tokens, hit.transformed_tokens)
    };

    Ok(Some(ProcessResult {
        output: hit.content,
        original_tokens: orig_tokens,
        transformed_tokens: trans_tokens,
        guardrail_triggered: false,
        parse_tier: None, // tier was not recorded at cache-write time
    }))
}

/// Read a file and validate it doesn't exceed the maximum input size.
fn read_and_validate(path: &Path) -> anyhow::Result<String> {
    let contents = fs::read_to_string(path)?;
    if contents.len() > MAX_INPUT_SIZE {
        anyhow::bail!(
            "File too large: {} bytes exceeds maximum of {} bytes ({}MB)",
            contents.len(),
            MAX_INPUT_SIZE,
            MAX_INPUT_SIZE / 1024 / 1024
        );
    }
    Ok(contents)
}

/// Transform file contents, trying auto-detection first and falling back to
/// `explicit_lang` when provided.
///
/// Returns `(transformed_output, mode_used, has_errors, source_line_map)` where:
/// - `has_errors` reflects whether the parser encountered syntax errors
/// - `source_line_map` is `Some(map)` when `options.line_numbers` is true and the
///   transform produced a meaningful source line map; `None` otherwise
///
/// For cascade paths (token_budget is set) `has_errors` is always `false` and
/// line numbers are applied after mode selection.
fn run_transform(
    contents: &str,
    path: &Path,
    options: &ProcessOptions,
) -> anyhow::Result<(String, Mode, bool, Option<Vec<usize>>)> {
    let explicit_lang = options.explicit_lang;
    // Non-line-number transform closure (used for cascade mode selection)
    let transform_file = |config: &TransformConfig| -> anyhow::Result<Option<String>> {
        // Try auto-detection first; fall back to explicit language if provided.
        let auto_result = transform_auto_with_config(contents, path, config);
        if let Ok(output) = auto_result {
            return Ok(Some(output));
        }
        let Some(language) = explicit_lang else {
            return Err(auto_result.unwrap_err().into());
        };
        Ok(Some(transform_with_config(contents, language, config)?))
    };

    match options.trunc.token_budget {
        Some(budget) => {
            let language = explicit_lang
                .or_else(|| detect_language_from_path(path))
                .unwrap_or_else(|| {
                    eprintln!(
                        "[skim] warning: language detection failed for '{}', defaulting to TypeScript",
                        path.display(),
                    );
                    Language::TypeScript
                });

            // AC-10: Token counting for mode selection does NOT include line number annotations.
            // Run cascade WITHOUT line_numbers to select the best mode.
            let (output, mode) = cascade::cascade_for_token_budget(
                options.mode,
                &options.trunc,
                budget,
                language,
                transform_file,
            )?;

            // If line numbers requested, re-run the selected mode WITH line_numbers.
            // Use the re-run output directly as the final output (avoids double transform).
            let (final_output, line_map) = if options.line_numbers {
                let config = cascade::build_config_with_opts(mode, &options.trunc, true);
                let (rerun_output, _has_errors, map) =
                    transform_with_line_map(contents, language, &config)?;
                (rerun_output, map)
            } else {
                (output, None)
            };

            Ok((final_output, mode, false, line_map))
        }
        None => {
            let language = explicit_lang.or_else(|| detect_language_from_path(path));

            // Use transform_with_line_map when we can identify the language
            if let Some(lang) = language {
                let config = cascade::build_config_with_opts(
                    options.mode,
                    &options.trunc,
                    options.line_numbers,
                );
                let (output, has_errors, line_map) =
                    transform_with_line_map(contents, lang, &config)?;
                Ok((output, options.mode, has_errors, line_map))
            } else {
                // Language detection failed — try auto-detect via path extension.
                // Can't get line map without a known language.
                let config = cascade::build_config(options.mode, &options.trunc);
                let output = transform_file(&config)?.ok_or_else(|| {
                    anyhow::anyhow!("Language detection failed and no --language specified")
                })?;
                Ok((output, options.mode, false, None))
            }
        }
    }
}

/// Process stdin input and return transformed content with optional token statistics.
///
/// Reads from stdin with a size limit, resolves the language from `--language` or
/// `--filename`, transforms the source (with optional token-budget cascade), and
/// computes token stats when `show_stats` is enabled.
pub(crate) fn process_stdin(
    options: ProcessOptions,
    filename_hint: Option<&str>,
) -> anyhow::Result<ProcessResult> {
    let mut buffer = String::with_capacity(64 * 1024);
    let bytes_read = io::stdin()
        .take(MAX_INPUT_SIZE as u64 + 1)
        .read_to_string(&mut buffer)?;

    if bytes_read > MAX_INPUT_SIZE {
        anyhow::bail!(
            "Input too large: {} bytes exceeds maximum of {} bytes ({}MB)",
            bytes_read,
            MAX_INPUT_SIZE,
            MAX_INPUT_SIZE / 1024 / 1024
        );
    }

    let filename_lang = filename_hint.and_then(|f| Language::from_path(Path::new(f)));

    let language = options.explicit_lang.or(filename_lang).ok_or_else(|| {
        if let Some(fname) = filename_hint {
            anyhow::anyhow!(
                "Language detection failed: unrecognized filename '{}'\n\
                 Supported extensions: .ts, .tsx, .js, .jsx, .cjs, .mjs, .py, .rs, .go, .java, .c, .h, .cpp, .hpp, .cxx, .cc, .md, .json, .yaml, .yml, .toml\n\
                 Hint: use --language to specify the language explicitly\n\
                 Example: cat file | skim - --language=typescript",
                fname
            )
        } else {
            anyhow::anyhow!(
                "Language detection failed: reading from stdin requires --language or --filename\n\
                 Example: cat file.ts | skim - --language=typescript\n\
                 Example: git show HEAD:main.rs | skim - --filename=main.rs"
            )
        }
    })?;

    let (transformed, stdin_has_errors, stdin_line_map) = match options.trunc.token_budget {
        Some(budget) => {
            // AC-10: Cascade mode selection without line numbers, then re-run with line numbers
            let (output, mode) = cascade::cascade_for_token_budget(
                options.mode,
                &options.trunc,
                budget,
                language,
                |config| Ok(Some(transform_with_config(&buffer, language, config)?)),
            )?;
            // Use the re-run output directly as the final output (avoids double transform).
            let (cascade_output, line_map) = if options.line_numbers {
                let config = cascade::build_config_with_opts(mode, &options.trunc, true);
                let (rerun, _errs, map) = transform_with_line_map(&buffer, language, &config)?;
                (rerun, map)
            } else {
                (output, None)
            };
            (cascade_output, false, line_map)
        }
        None => {
            let config =
                cascade::build_config_with_opts(options.mode, &options.trunc, options.line_numbers);
            let (output, has_errors, line_map) =
                transform_with_line_map(&buffer, language, &config)?;
            (output, has_errors, line_map)
        }
    };

    // Determine parse quality tier before guardrail.
    let parse_tier = Some(parse_tier_from(options.mode, stdin_has_errors));

    // Apply output guardrail: if compressed output is larger than raw, emit raw instead.
    // Same protection as process_file; token counting happens after so stats reflect
    // the final output. Guardrail comparison uses UN-annotated output.
    let (final_output, guardrail_triggered) =
        if options.mode != Mode::Full && options.trunc.token_budget.is_none() {
            let outcome = crate::output::guardrail::apply_to_stderr(buffer.clone(), transformed)?;
            let triggered = outcome.was_triggered();
            (outcome.into_output(), triggered)
        } else {
            (transformed, false)
        };

    // Apply line number formatting AFTER guardrail, BEFORE token stats.
    let final_output = apply_line_numbers(
        final_output,
        options.line_numbers,
        guardrail_triggered,
        stdin_line_map,
    );

    // Only pay the tiktoken BPE cost on the main thread when --show-stats
    // is set. Analytics background threads compute their own token counts.
    let (orig_tokens, trans_tokens) = if options.show_stats {
        count_token_pair(&buffer, &final_output)
    } else {
        (None, None)
    };

    Ok(ProcessResult {
        output: final_output,
        original_tokens: orig_tokens,
        transformed_tokens: trans_tokens,
        guardrail_triggered,
        parse_tier,
    })
}

/// Process a single file and return transformed content with optional token statistics.
pub(crate) fn process_file(path: &Path, options: ProcessOptions) -> anyhow::Result<ProcessResult> {
    if let Some(result) = try_cached_result(path, &options)? {
        return Ok(result);
    }

    let contents = read_and_validate(path)?;
    let (result, mode_used, has_errors, line_map) = run_transform(&contents, path, &options)?;

    // Determine parse quality tier before guardrail (guardrail may swap output,
    // but the parse tier reflects the transformation, not the final selection).
    let parse_tier = Some(parse_tier_from(options.mode, has_errors));

    // Apply output guardrail: if compressed output is larger than raw, emit raw instead.
    // Token counting happens AFTER this decision so stats reflect the final output.
    // Guardrail comparison uses UN-annotated output (before line number formatting).
    let (final_output, guardrail_triggered) =
        if options.mode != Mode::Full && options.trunc.token_budget.is_none() {
            let outcome = crate::output::guardrail::apply_to_stderr(contents.clone(), result)?;
            let triggered = outcome.was_triggered();
            (outcome.into_output(), triggered)
        } else {
            (result, false)
        };

    // Apply line number formatting AFTER guardrail, BEFORE cache write and token stats.
    // AC-12: Cache key includes line_numbers (handled in cache::read_cache/write_cache).
    let final_output = apply_line_numbers(
        final_output,
        options.line_numbers,
        guardrail_triggered,
        line_map,
    );

    // Only pay the tiktoken BPE cost on the main thread when --show-stats
    // is set. Analytics background threads compute their own token counts.
    let (orig_tokens, trans_tokens) = if options.show_stats {
        count_token_pair(&contents, &final_output)
    } else {
        (None, None)
    };

    // Cache the transform result (post-guardrail, post-line-number-formatting).
    // Cache write failures are non-fatal; don't fail the transformation.
    if options.use_cache {
        let effective_mode = (mode_used != options.mode).then_some(mode_used);
        let _ = cache::write_cache(&cache::CacheWriteParams {
            path,
            mode: options.mode,
            content: &final_output,
            original_tokens: orig_tokens,
            transformed_tokens: trans_tokens,
            trunc: options.trunc,
            effective_mode,
            parse_tier: parse_tier.map(str::to_string),
            line_numbers: options.line_numbers,
        });
    }

    Ok(ProcessResult {
        output: final_output,
        original_tokens: orig_tokens,
        transformed_tokens: trans_tokens,
        guardrail_triggered,
        parse_tier,
    })
}

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

    // ========================================================================
    // count_token_pair tests
    // ========================================================================

    #[test]
    fn count_token_pair_returns_some_for_valid_input() {
        let (orig, trans) = count_token_pair("hello world", "hello");
        assert!(orig.is_some(), "original tokens should be Some");
        assert!(trans.is_some(), "transformed tokens should be Some");
        assert!(
            orig.unwrap() > trans.unwrap(),
            "original should have more tokens than transformed"
        );
    }

    #[test]
    fn count_token_pair_returns_some_for_empty_strings() {
        let (orig, trans) = count_token_pair("", "");
        assert_eq!(orig, Some(0));
        assert_eq!(trans, Some(0));
    }

    #[test]
    fn count_token_pair_original_equals_transformed_for_identical_input() {
        let text = "fn main() { println!(\"hello\"); }";
        let (orig, trans) = count_token_pair(text, text);
        assert_eq!(orig, trans);
    }

    // ========================================================================
    // report_token_stats tests
    // ========================================================================

    #[test]
    fn report_token_stats_does_not_panic_with_none_values() {
        // Should be a no-op when tokens are None
        report_token_stats(None, None, "");
        report_token_stats(Some(100), None, "");
        report_token_stats(None, Some(50), "");
    }

    #[test]
    fn report_token_stats_does_not_panic_with_valid_values() {
        // Should write to stderr without panicking
        report_token_stats(Some(1000), Some(200), " (test)");
    }

    // ========================================================================
    // read_and_validate tests
    // ========================================================================

    #[test]
    fn read_and_validate_rejects_nonexistent_file() {
        let result = read_and_validate(Path::new("/nonexistent/file.txt"));
        assert!(result.is_err());
    }

    // ========================================================================
    // parse_tier_from tests (B4-B5)
    // ========================================================================

    #[test]
    fn test_parse_tier_passthrough() {
        assert_eq!(parse_tier_from(Mode::Full, false), "passthrough");
        assert_eq!(parse_tier_from(Mode::Full, true), "passthrough");
    }

    #[test]
    fn test_parse_tier_degraded() {
        assert_eq!(parse_tier_from(Mode::Structure, true), "degraded");
        assert_eq!(parse_tier_from(Mode::Signatures, true), "degraded");
        assert_eq!(parse_tier_from(Mode::Minimal, true), "degraded");
    }

    #[test]
    fn test_parse_tier_full() {
        assert_eq!(parse_tier_from(Mode::Structure, false), "full");
        assert_eq!(parse_tier_from(Mode::Signatures, false), "full");
        assert_eq!(parse_tier_from(Mode::Types, false), "full");
    }

    // ========================================================================
    // apply_line_numbers tests
    // ========================================================================

    /// Branch: line_numbers disabled — output is returned unchanged regardless of
    /// guardrail_triggered or computed_map.
    #[test]
    fn apply_line_numbers_disabled_returns_output_unchanged() {
        let output = "fn foo() {}\nfn bar() {}\n".to_string();
        let result = apply_line_numbers(output.clone(), false, false, Some(vec![1, 2]));
        assert_eq!(
            result, output,
            "disabled line numbers must not modify output"
        );

        let result2 = apply_line_numbers(output.clone(), false, true, None);
        assert_eq!(
            result2, output,
            "disabled line numbers must not modify output even when guardrail triggered"
        );
    }

    /// Branch: guardrail_triggered — identity map is applied regardless of computed_map.
    #[test]
    fn apply_line_numbers_guardrail_uses_identity_map() {
        let output = "line one\nline two\n".to_string();
        // computed_map is None here; the guardrail path must build its own identity map.
        let result = apply_line_numbers(output.clone(), true, true, None);
        // Identity map annotates each line with its 1-based output line number.
        // format_with_line_numbers uses "<n>\t<content>" format.
        assert!(
            result.contains("1\t") && result.contains("2\t"),
            "guardrail path must annotate both output lines; got: {result:?}"
        );
        // The raw content must still be present.
        assert!(
            result.contains("line one"),
            "output text must survive annotation"
        );
    }

    /// Branch: computed_map is None (serde non-full modes or language detection failure) —
    /// output is returned unannotated even when line_numbers is true.
    #[test]
    fn apply_line_numbers_none_map_skips_annotation() {
        let output = "key: value\n".to_string();
        let result = apply_line_numbers(output.clone(), true, false, None);
        assert_eq!(
            result, output,
            "None computed_map must skip line-number annotation"
        );
    }

    /// Branch: computed_map is Some — the provided map is forwarded to
    /// format_with_line_numbers to produce annotated output.
    #[test]
    fn apply_line_numbers_some_map_annotates_output() {
        // Simulate a 2-line transform output that maps to source lines 1 and 5.
        let output = "fn foo() {}\nfn bar() {}\n".to_string();
        let map = vec![1usize, 5];
        let result = apply_line_numbers(output.clone(), true, false, Some(map));
        // The annotation must include the source line numbers from the provided map.
        // format_with_line_numbers uses "<n>\t<content>" format.
        assert!(
            result.contains("1\t") && result.contains("5\t"),
            "provided map line numbers must appear in output; got: {result:?}"
        );
        assert!(
            result.contains("fn foo()"),
            "content must survive annotation"
        );
    }
}
