use std::ops::Range;
use std::sync::Arc;

use crate::syntax::{Green, GreenNode, NodeKind};

use super::{
    is_newline, parse, parse_block, parse_markup_elements, parse_template, TokenMode,
};

type ReparseFunc =
    fn(&str, &str, usize, isize, &[Green], bool) -> Option<(Vec<Green>, bool, usize)>;

/// Allows partial refreshs of the [`Green`] node tree.
///
/// This struct holds a description of a change. Its methods can be used to try
/// and apply the change to a green tree.
pub struct Reparser<'a> {
    /// The new source code, with the change applied.
    src: &'a str,
    /// Which range in the old source file was changed.
    replace_range: Range<usize>,
    /// How many characters replaced the text in `replace_range`.
    replace_len: usize,
}

impl<'a> Reparser<'a> {
    /// Create a new reparser.
    pub fn new(src: &'a str, replace_range: Range<usize>, replace_len: usize) -> Self {
        Self { src, replace_range, replace_len }
    }
}

impl Reparser<'_> {
    /// Find the innermost child that is incremental safe.
    pub fn reparse(&self, green: &mut Arc<GreenNode>) -> Range<usize> {
        self.reparse_step(Arc::make_mut(green), 0, true).unwrap_or_else(|| {
            *green = parse(self.src);
            0 .. self.src.len()
        })
    }

    fn reparse_step(
        &self,
        green: &mut GreenNode,
        mut offset: usize,
        outermost: bool,
    ) -> Option<Range<usize>> {
        let child_mode = green.kind().mode().unwrap_or(TokenMode::Code);
        let original_count = green.children().len();

        let mut search = SearchState::default();
        let mut ahead_nontrivia = None;
        // Whether the first node that should be replaced is at start.
        let mut at_start = true;
        let mut end_outermost = false;

        // Find the the first child in the range of children to reparse.
        for (i, child) in green.children().iter().enumerate() {
            let pos = GreenPos { idx: i, offset };
            let child_span = offset .. offset + child.len();

            match search {
                SearchState::NoneFound => {
                    // The edit is contained within the span of the current element.
                    if child_span.contains(&self.replace_range.start)
                        && child_span.end >= self.replace_range.end
                    {
                        // In Markup mode, we want to consider a non-whitespace
                        // neighbor if the edit is on the node boundary.
                        search = if child_span.end == self.replace_range.end
                            && child_mode == TokenMode::Markup
                        {
                            SearchState::RequireNonWS(pos)
                        } else {
                            // println!("found containing block {:?}", green.kind());
                            SearchState::Contained(pos)
                        };
                    } else if child_span.contains(&self.replace_range.start) {
                        search = SearchState::Inside(pos);
                    } else {
                        if !child.kind().is_space()
                            && child.kind() != &NodeKind::Semicolon
                        {
                            ahead_nontrivia = Some((pos, at_start));
                        }
                        at_start = child.kind().is_at_start(at_start);
                    }
                }
                SearchState::Inside(start) => {
                    if child_span.end == self.replace_range.end {
                        search = SearchState::RequireNonWS(start);
                    } else if child_span.end > self.replace_range.end {
                        search = SearchState::SpanFound(start, pos);
                    }
                }
                SearchState::RequireNonWS(start) => {
                    if !child.kind().is_trivia() {
                        search = SearchState::SpanFound(start, pos);
                    }
                }
                _ => unreachable!(),
            }

            offset += child.len();
            end_outermost = outermost && i + 1 == original_count;
            if search.end().is_some() {
                break;
            }
        }

        if let SearchState::Contained(pos) = search {
            let child = &mut green.children_mut()[pos.idx];
            let prev_len = child.len();

            if let Some(range) = match child {
                Green::Node(node) => {
                    self.reparse_step(Arc::make_mut(node), pos.offset, end_outermost)
                }
                Green::Token(_) => None,
            } {
                let new_len = child.len();
                green.update_parent(new_len, prev_len);
                return Some(range);
            }

            let superseded_span = pos.offset .. pos.offset + prev_len;
            let func: Option<ReparseFunc> = match child.kind() {
                NodeKind::Template => Some(parse_template),
                NodeKind::Block => Some(parse_block),
                _ => None,
            };

            if let Some(func) = func {
                if let Some(result) = self.replace(
                    green,
                    func,
                    pos.idx .. pos.idx + 1,
                    superseded_span,
                    at_start,
                    end_outermost,
                    outermost,
                ) {
                    return Some(result);
                }
            }
        }

        if !matches!(green.kind(), NodeKind::Markup(_)) {
            return None;
        }

        let (mut start, end) = search.end()?;
        if let Some((ahead, ahead_at_start)) = ahead_nontrivia {
            let ahead_kind = green.children()[ahead.idx].kind();

            if start.offset == self.replace_range.start
                || ahead_kind.only_at_start()
                || ahead_kind == &NodeKind::LineComment
            {
                start = ahead;
                at_start = ahead_at_start;
            }
        }

        let superseded_span =
            start.offset .. end.offset + green.children()[end.idx].len();
        self.replace(
            green,
            parse_markup_elements,
            start.idx .. end.idx + 1,
            superseded_span,
            at_start,
            end_outermost,
            outermost,
        )
    }

    fn replace(
        &self,
        green: &mut GreenNode,
        func: ReparseFunc,
        superseded_idx: Range<usize>,
        superseded_span: Range<usize>,
        at_start: bool,
        outermost: bool,
        parent_outermost: bool,
    ) -> Option<Range<usize>> {
        let differential: isize =
            self.replace_len as isize - self.replace_range.len() as isize;
        let newborn_span = superseded_span.start
            ..
            (superseded_span.end as isize + differential) as usize;
        let superseded_start = superseded_idx.start;

        let mut prefix = "";
        for (i, c) in self.src[.. newborn_span.start].char_indices().rev() {
            if is_newline(c) {
                break;
            }
            prefix = &self.src[i .. newborn_span.start];
        }

        // // println!("reparsing...");

        let (newborns, terminated, amount) = func(
            &prefix,
            &self.src[newborn_span.start ..],
            newborn_span.len(),
            differential,
            &green.children()[superseded_start ..],
            at_start,
        )?;

        // // println!("Reparse success");

        // Do not accept unclosed nodes if the old node wasn't at the right edge
        // of the tree.
        if !outermost && !terminated {
            return None;
        }

        if !parent_outermost && green.children()[superseded_start ..].len() == amount {
            return None;
        }

        green.replace_children(superseded_start .. superseded_start + amount, newborns);
        Some(newborn_span)
    }
}

#[derive(Clone, Copy, Debug, PartialEq)]
struct GreenPos {
    idx: usize,
    offset: usize,
}

/// Encodes the state machine of the search for the node which is pending for
/// replacement.
#[derive(Clone, Copy, Debug, PartialEq)]
enum SearchState {
    /// Neither an end nor a start have been found as of now.
    /// The last non-whitespace child is continually saved.
    NoneFound,
    /// The search has concluded by finding a node that fully contains the
    /// modifications.
    Contained(GreenPos),
    /// The search has found the start of the modified nodes.
    Inside(GreenPos),
    /// The search has found the end of the modified nodes but the change
    /// touched its boundries so another non-trivia node is needed.
    RequireNonWS(GreenPos),
    /// The search has concluded by finding a start and an end index for nodes
    /// with a pending reparse.
    SpanFound(GreenPos, GreenPos),
}

impl Default for SearchState {
    fn default() -> Self {
        Self::NoneFound
    }
}

impl SearchState {
    fn end(&self) -> Option<(GreenPos, GreenPos)> {
        match self {
            Self::NoneFound => None,
            Self::Contained(s) => Some((*s, *s)),
            Self::Inside(_) => None,
            Self::RequireNonWS(_) => None,
            Self::SpanFound(s, e) => Some((*s, *e)),
        }
    }
}

impl NodeKind {
    /// Whether this node has to appear at the start of a line.
    pub fn only_at_start(&self) -> bool {
        match self {
            Self::Heading | Self::Enum | Self::List => true,
            _ => false,
        }
    }
}

#[cfg(test)]
#[rustfmt::skip]
mod tests {
    use super::*;
    use crate::parse::parse;
    use crate::parse::tests::check;
    use crate::source::SourceFile;

    #[track_caller]
    fn test(prev: &str, range: Range<usize>, with: &str, goal: Range<usize>) {
        let mut source = SourceFile::detached(prev);
        let range = source.edit(range, with);
        check(source.src(), source.root(), &parse(source.src()));
        assert_eq!(range, goal);
    }

    #[test]
    fn test_parse_incremental_simple_replacements() {
        test("hello  world", 7 .. 12, "walkers", 0 .. 14);
        test("some content", 0..12, "", 0..0);
        test("", 0..0, "do it", 0..5);
        test("a d e", 1 .. 3, " b c d", 0 .. 9);
        test("a #f() e", 1 .. 6, " b c d", 0 .. 9);
        test("a\nb\nc\nd\ne\n", 5 .. 5, "c", 4 .. 7);
        test("a\n\nb\n\nc\n\nd\n\ne\n", 7 .. 7, "c", 6 .. 10);
        test("a\nb\nc *hel a b lo* d\nd\ne", 13..13, "c ", 6..20);
        test("~~ {a} ~~", 4 .. 5, "b", 3 .. 6);
        test("{(0, 1, 2)}", 5 .. 6, "11pt", 0..14);
        test("\n= A heading", 3 .. 3, "n evocative", 3 .. 23);
        test("for~your~thing", 9 .. 9, "a", 8 .. 15);
        test("a your thing a", 6 .. 7, "a", 0 .. 14);
        test("{call(); abc}", 7 .. 7, "[]", 0 .. 15);
        test("#call() abc", 7 .. 7, "[]", 0 .. 10);
        // Investigate
        test("hi[\n- item\n- item 2\n    - item 3]", 11 .. 11, "  ", 2 .. 35);
        test("hi\n- item\nno item\n    - item 3", 10 .. 10, "- ", 3..19);
        test("#grid(columns: (auto, 1fr, 40%), [*plonk*], rect(width: 100%, height: 1pt, fill: conifer), [thing])", 16 .. 20, "none", 0..99);
        test("#grid(columns: (auto, 1fr, 40%), [*plonk*], rect(width: 100%, height: 1pt, fill: conifer), [thing])", 33 .. 42, "[_gronk_]", 33..42);
        test("#grid(columns: (auto, 1fr, 40%), [*plonk*], rect(width: 100%, height: 1pt, fill: conifer), [thing])", 34 .. 41, "_bar_", 33 .. 40);
        test("{let i=1; for x in range(5) {i}}", 6 .. 6, " ", 0 .. 33);
        test("{let i=1; for x in range(5) {i}}", 13 .. 14, "  ", 0 .. 33);
        // Investigate
        test("hello~~{x}", 7 .. 10, "#f()", 0 .. 11);
        test("this~is -- in my opinion -- spectacular", 8 .. 10, "---", 5 .. 25);
        test("understanding `code` is complicated", 15 .. 15, "C ", 14 .. 22);
        test("{ let x = g() }", 10 .. 12, "f(54", 0 .. 17);
        test("a #let rect with (fill: eastern)\nb", 16 .. 31, " (stroke: conifer", 2 .. 34);
        test(r#"a ```typst hello``` b"#, 16 .. 17, "", 0 .. 20);
        test(r#"a ```typst hello```"#, 16 .. 17, "", 2 .. 18);
        test("#for", 4 .. 4, "//", 0 .. 6);
    }

    #[test]
    fn test_parse_incremental_whitespace_invariants() {
        test("hello \\ world", 7 .. 8, "a ", 6 .. 14);
        test("hello \\ world", 7 .. 8, " a", 6 .. 14);
        test("x = y", 1 .. 1, " + y", 0 .. 6);
        test("x = y", 1 .. 1, " + y\n", 0 .. 7);
        test("abc\n= a heading\njoke", 3 .. 4, "\nmore\n\n", 0 .. 21);
        test("abc\n= a heading\njoke", 3 .. 4, "\nnot ", 0 .. 19);
        test("#let x = (1, 2 + ;~ Five\r\n\r", 20 .. 23, "2.", 18 .. 23);
        test("hey #myfriend", 4 .. 4, "\\", 0 .. 14);
        test("hey  #myfriend", 4 .. 4, "\\", 3 .. 6);
        test("= foo\nbar\n - a\n - b", 6 .. 9, "", 0..11)
    }

    #[test]
    fn test_parse_incremental_type_invariants() {
        test("a #for x in array {x}", 18 .. 21, "[#x]", 0 .. 22);
        test("a #let x = 1 {5}", 3 .. 6, "if", 2 .. 11);
        test("a {let x = 1 {5}} b", 3 .. 6, "if", 2 .. 16);
        test("#let x = 1 {5}", 4 .. 4, " if", 0 .. 13);
        test("{let x = 1 {5}}", 4 .. 4, " if", 0 .. 18);
        test("a // b c #f()", 3 .. 4, "", 2 .. 12);
        test("{\nf()\n//g(a)\n}", 6 .. 8, "", 0 .. 12);
        test("a{\nf()\n//g(a)\n}b", 7 .. 9, "", 1 .. 13);
        test("a #while x {\n g(x) \n}  b", 11 .. 11, "//", 0 .. 26);
        test("{(1, 2)}", 1 .. 1, "while ", 0 .. 14);
        test("a b c", 1 .. 1, "{[}", 0 .. 8);
    }

    #[test]
    fn test_parse_incremental_wrongly_or_unclosed_things() {
        test(r#"{"hi"}"#, 4 .. 5, "c", 0 .. 6);
        test(r"this \u{abcd}", 8 .. 9, "", 5 .. 12);
        test(r"this \u{abcd} that", 12 .. 13, "", 5 .. 17);
        test(r"{{let x = z}; a = 1} b", 6 .. 6, "//", 0 .. 24);
        test("a b c", 1 .. 1, " /* letters */", 0 .. 19);
        test("a b c", 1 .. 1, " /* letters", 0 .. 16);
        test("{if i==1 {a} else [b]; b()}", 12 .. 12, " /* letters */", 0 .. 41);
        test("{if i==1 {a} else [b]; b()}", 12 .. 12, " /* letters", 0 .. 38);
        test("~~~~", 2 .. 2, "[]", 1 .. 5);
        test("a[]b", 2 .. 2, "{", 1 .. 4);
        test("[hello]", 2 .. 3, "]", 0 .. 7);
        test("{a}", 1 .. 2, "b", 0 .. 3);
        test("{ a; b; c }", 5 .. 6, "[}]", 0 .. 13);
        test("#a()\n~", 3..4, "{}", 0..7);
        test("[]\n~", 1..2, "#if i==0 {true}", 0..18);
    }
}