//! Generate a minimized version of a given input, by removing parts of it.
//! This is useful for debugging, when you have a large input that causes a failure,
//! and you want to find the smallest input that still causes the failure.
//!
//! Most users will want to use the binary instead of this module directly.
//! e.g. `cargo run --release --bin minimize -- full <file_that_triggers_parsing_bug>`

use crate::test_helpers::{Input, InputKind};
use bumpalo::Bump;
use roc_parse::{ast::Malformed, remove_spaces::RemoveSpaces};

pub fn print_minimizations(text: &str, kind: InputKind) {
    let Some(original_error) = round_trip_once_and_extract_error(text, kind) else {
        eprintln!("No error found");
        return;
    };

    eprintln!("Error found: {}", original_error);
    eprintln!("Proceeding with minimization");

    let mut s = text.to_string();

    loop {
        let mut found = false;
        for update in candidate_minimizations(s.clone()) {
            let mut new_s = String::with_capacity(s.len());
            let mut offset = 0;
            for (start, end, replacement) in update.replacements.clone() {
                new_s.push_str(&s[offset..start]);
                new_s.push_str(&replacement);
                offset = end;
            }
            new_s.push_str(&s[offset..]);

            assert!(
                new_s.len() < s.len(),
                "replacements: {:?}",
                update.replacements
            );

            if let Some(result) = round_trip_once_and_extract_error(&new_s, kind) {
                if result == original_error {
                    eprintln!("Successfully minimized, new length: {}", new_s.len());
                    s = new_s;
                    found = true;
                    break;
                }
            }
        }

        if !found {
            eprintln!("No more minimizations found");
            break;
        }
    }

    eprintln!("Final result:");
    println!("{}", s);
}

fn round_trip_once_and_extract_error(text: &str, kind: InputKind) -> Option<String> {
    let input = kind.with_text(text);
    let res = std::panic::catch_unwind(|| round_trip_once(input));

    match res {
        Ok(res) => res,
        Err(e) => {
            if let Some(s) = e.downcast_ref::<&'static str>() {
                return Some(s.to_string());
            }
            if let Some(s) = e.downcast_ref::<String>() {
                return Some(s.clone());
            }
            Some("Panic during parsing".to_string())
        }
    }
}

fn round_trip_once(input: Input<'_>) -> Option<String> {
    let arena = Bump::new();

    let actual = match input.parse_in(&arena) {
        Ok(a) => a,
        Err(e) => {
            return Some(format!(
                "Initial parse failed: {:?}",
                e.remove_spaces(&arena)
            ))
        }
    };

    if actual.is_malformed() {
        return Some("Initial parse is malformed".to_string());
    }

    let output = actual.format();

    let reparsed_ast = match output.as_ref().parse_in(&arena) {
        Ok(r) => r,
        Err(e) => return Some(format!("Reparse failed: {:?}", e.remove_spaces(&arena))),
    };

    let ast_normalized = actual.remove_spaces(&arena);
    let reparsed_ast_normalized = reparsed_ast.remove_spaces(&arena);

    if format!("{ast_normalized:?}") != format!("{reparsed_ast_normalized:?}") {
        return Some("Different ast".to_string());
    }

    None
}

struct Update {
    replacements: Vec<(usize, usize, String)>,
}

fn candidate_minimizations(s: String) -> Box<dyn Iterator<Item = Update>> {
    let mut line_offsets = vec![0];
    line_offsets.extend(s.match_indices('\n').map(|(i, _)| i + 1));
    let line_count = line_offsets.len();
    let s_len = s.len();

    let line_indents = line_offsets
        .iter()
        .map(|&offset| s[offset..].chars().take_while(|&c| c == ' ').count())
        .collect::<Vec<_>>();

    let line_offsets_clone = line_offsets.clone();

    // first, try to remove every group of 1, 2, 3, ... lines - in reverse order (so, trying removing n lines first, then n-1, etc)
    let line_removals = (1..=line_count).rev().flat_map(move |n| {
        let line_offsets_clone = line_offsets.clone();
        (0..line_count - n).map(move |start| {
            let end = start + n;
            let start_offset = line_offsets_clone[start];
            let end_offset = line_offsets_clone[end];
            let replacement = String::new();
            let replacements = vec![(start_offset, end_offset, replacement)];
            Update { replacements }
        })
    });

    let line_offsets = line_offsets_clone;
    let line_offsets_clone = line_offsets.clone();

    // then, try to dedent every group of 1, 2, 3, ... lines - in reverse order (so, trying dedenting n lines first, then n-1, etc)
    // just remove one space at a time, for now
    let line_dedents = (1..=line_count).rev().flat_map(move |n| {
        let line_offsets_clone = line_offsets.clone();
        let line_indents_clone = line_indents.clone();
        (0..line_count - n).filter_map(move |start| {
            // first check if all lines are either zero-width or have greater than zero indent
            let end = start + n;
            for i in start..end {
                if line_indents_clone[i] == 0
                    && line_offsets_clone[i] + 1
                        < line_offsets_clone.get(i + 1).cloned().unwrap_or(s_len)
                {
                    return None;
                }
            }

            let mut replacements = vec![];
            for i in start..end {
                let offset = line_offsets_clone[i];
                let indent = line_indents_clone[i];
                if indent > 0 {
                    replacements.push((offset, offset + 1, String::new()));
                }
            }
            Some(Update { replacements })
        })
    });

    // then, try to select every range of 1, 2, 3, ... lines - in normal order this time!
    // we remove the lines before and after the range
    let line_selects = (1..line_count - 1).flat_map(move |n| {
        assert!(n > 0);
        let line_offsets_clone = line_offsets_clone.clone();
        (0..line_count - n).map(move |start| {
            let end = start + n;
            let start_offset = line_offsets_clone[start];
            let end_offset = line_offsets_clone[end];
            assert!(end_offset > start_offset);
            assert!(start_offset > 0 || end_offset < s_len);
            let replacements = vec![
                (0, start_offset, String::new()),
                (end_offset, s_len, String::new()),
            ];
            Update { replacements }
        })
    });

    // then, try to remove every range of 1, 2, 3, ... characters - in reverse order (so, trying removing n characters first, then n-1, etc)
    let charseq_removals = (1..s.len()).rev().flat_map(move |n| {
        (0..s.len() - n).map(move |start| {
            let end = start + n;
            let replacement = String::new();
            let replacements = vec![(start, end, replacement)];
            Update { replacements }
        })
    });

    Box::new(
        line_removals
            .chain(line_dedents)
            .chain(line_selects)
            .chain(charseq_removals)
            .filter(|u| !u.replacements.is_empty()),
    )
}