parse.rs

//! トークンの並びを構文木にする

use crate::error::Error;
use crate::node::{self, Align, Class, Level, Limits, Node, Spacing};
use crate::symbol;
use crate::token::{Located, Token, tokenize};


/// 式を囲んでいた書き方が指す組み方
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Wrapper {
    /// 囲みがなかった
    None,
    /// 文中に置く囲み。$…$ と \(…\) と \begin{math}
    Inline,
    /// 独立した行に置く囲み。$$…$$ と \[…\] と \begin{equation}、それに \begin{align} と \begin{gather}
    Display,
}


/// LaTeX の文字列を構文木にする。囲みがあれば外す
pub fn parse(source: &str) -> Result<Node, Error> {
    Ok(parse_wrapped(source)?.0)
}


/// 構文木と、囲みが指していた組み方を返す
///
/// 囲みは付いていても付いていなくてもよい。付いていればそれが display か inline かを決め、
/// 付いていなければ利用側の指定がそのまま使われる
pub fn parse_wrapped(source: &str) -> Result<(Node, Wrapper), Error> {
    let (tokens, wrapper) = unwrap(tokenize(source)?);

    if let Some(stray) = tokens.iter().find(|found| found.token == Token::Character('$')) {
        let message = "$ は式を囲む記号です。文字として出すには \\$ と書きます";

        return Err(Error::new(message, stray.position));
    }

    let mut parser = Parser {
        tokens,
        position: 0,
        end: source.chars().count(),
        stop_at_bracket: false,
        last_limits: Limits::Never,
    };
    let row = parser.parse_row()?;

    if parser.position < parser.tokens.len() {
        return Err(parser.unexpected());
    }

    Ok((row, wrapper))
}


/// 囲みが組み方を指しているならそれを、指していなければ渡された値を返す
pub fn display_of(wrapper: Wrapper, fallback: bool) -> bool {
    match wrapper {
        Wrapper::Display => true,
        Wrapper::Inline => false,
        Wrapper::None => fallback,
    }
}


fn is_character(found: &Located, character: char) -> bool {
    found.token == Token::Character(character)
}


fn is_command(found: &Located, name: &str) -> bool {
    matches!(&found.token, Token::Command(found) if found == name)
}


/// 前から head 個、後ろから tail 個を落とす
fn trim(tokens: Vec<Located>, head: usize, tail: usize) -> Vec<Located> {
    let keep = tokens.len().saturating_sub(head + tail);

    tokens.into_iter().skip(head).take(keep).collect()
}


/// 式を囲んでいる書き方を外す
fn unwrap(tokens: Vec<Located>) -> (Vec<Located>, Wrapper) {
    let count = tokens.len();

    if count < 2 {
        return (tokens, Wrapper::None);
    }

    if is_character(&tokens[0], '$') && is_character(&tokens[count - 1], '$') {
        if count >= 4 && is_character(&tokens[1], '$') && is_character(&tokens[count - 2], '$') {
            return (trim(tokens, 2, 2), Wrapper::Display);
        }

        return (trim(tokens, 1, 1), Wrapper::Inline);
    }

    if is_command(&tokens[0], "[") && is_command(&tokens[count - 1], "]") {
        return (trim(tokens, 1, 1), Wrapper::Display);
    }

    if is_command(&tokens[0], "(") && is_command(&tokens[count - 1], ")") {
        return (trim(tokens, 1, 1), Wrapper::Inline);
    }

    unwrap_environment(tokens)
}


/// 数式に入るための環境を外す。行列のように数式の中で使う環境はここでは外さない
///
/// align と gather は、LaTeX では数式の外に置いて display で組む環境である。
/// 中身は行を揃えて組むので、外さずに残して display を指す
fn unwrap_environment(tokens: Vec<Located>) -> (Vec<Located>, Wrapper) {
    let Some(name) = environment_name(&tokens, 0, "begin") else {
        return (tokens, Wrapper::None);
    };
    let (wrapper, keep) = match name.as_str() {
        "equation" | "equation*" | "displaymath" => (Wrapper::Display, false),
        "math" => (Wrapper::Inline, false),
        "align" | "align*" | "alignat" | "alignat*" | "gather" | "gather*" => (Wrapper::Display, true),
        _ => return (tokens, Wrapper::None),
    };
    // \begin{名前} も \end{名前} も、コマンドと波括弧 2 つと名前の字数ぶんある
    let edge = name.chars().count() + 3;

    if tokens.len() < edge * 2 {
        return (tokens, Wrapper::None);
    }

    match environment_name(&tokens, tokens.len() - edge, "end") {
        Some(closing) if closing == name && keep => (tokens, wrapper),
        Some(closing) if closing == name => (trim(tokens, edge, edge), wrapper),
        _ => (tokens, Wrapper::None),
    }
}


/// at から始まる \begin{名前} や \end{名前} の名前を読む
fn environment_name(tokens: &[Located], at: usize, keyword: &str) -> Option<String> {
    if !is_command(tokens.get(at)?, keyword) || tokens.get(at + 1)?.token != Token::LeftBrace {
        return None;
    }

    let mut name = String::new();
    let mut index = at + 2;

    while let Some(found) = tokens.get(index) {
        match &found.token {
            Token::Character(character)
                if character.is_ascii_alphanumeric() || *character == '*' =>
            {
                name.push(*character);
                index += 1;
            }
            Token::RightBrace => return Some(name),
            _ => return None,
        }
    }

    None
}


struct Parser {
    tokens: Vec<Located>,
    position: usize,
    /// 元の文字列の長さ。終端でのエラー位置に使う
    end: usize,
    /// \sqrt[...] の中を読んでいるあいだだけ true
    stop_at_bracket: bool,
    /// 直前に読んだアトムが、添字を上下に取るかどうか
    last_limits: Limits,
}


impl Parser {
    fn peek(&self) -> Option<Token> {
        self.tokens.get(self.position).map(|located| located.token.clone())
    }

    /// いま見ているトークンの、元の文字列での位置
    fn spot(&self) -> usize {
        match self.tokens.get(self.position) {
            Some(located) => located.position,
            None => self.end.saturating_sub(1),
        }
    }

    fn unexpected(&self) -> Error {
        let message = match self.tokens.get(self.position).map(|located| &located.token) {
            None => "式が途中で終わっています".to_string(),
            Some(Token::RightBrace) => "対応する開き波括弧がない閉じ波括弧があります".to_string(),
            Some(Token::Ampersand) => "& は行列の中でだけ使えます".to_string(),
            Some(Token::RowBreak) => "\\\\ は行列の中でだけ使えます".to_string(),
            Some(Token::Caret) => "^ の位置が正しくありません".to_string(),
            Some(Token::Underscore) => "_ の位置が正しくありません".to_string(),
            Some(Token::Text(_)) => "\\text の位置が正しくありません".to_string(),
            Some(Token::LeftBrace) => "開き波括弧の位置が正しくありません".to_string(),
            Some(Token::Command(name)) if name == "end" => {
                "対応する \\begin がない \\end があります".to_string()
            }
            Some(Token::Command(name)) if name == "right" => {
                "対応する \\left がない \\right があります".to_string()
            }
            Some(Token::Command(name)) => format!("予期しない \\{name} があります"),
            Some(Token::Character(character)) => format!("予期しない文字 {character} があります"),
        };
        Error::new(message, self.spot())
    }

    /// 要素の列がここで終わるか
    fn at_boundary(&self) -> bool {
        match self.tokens.get(self.position).map(|located| &located.token) {
            None | Some(Token::RightBrace) | Some(Token::Ampersand) | Some(Token::RowBreak) => true,
            Some(Token::Command(name)) => name == "end" || name == "right",
            Some(Token::Character(']')) => self.stop_at_bracket,
            _ => false,
        }
    }

    /// 区切りまで要素を読んで Row にする。区切りのトークンは消費しない
    fn parse_row(&mut self) -> Result<Node, Error> {
        self.parse_row_from(Vec::new())
    }

    fn parse_row_from(&mut self, mut items: Vec<Node>) -> Result<Node, Error> {
        while !self.at_boundary() {
            // 組み方の切り替えは、そこから先の全部にかかる
            if let Some(level) = self.style_switch() {
                let rest = self.parse_row()?;
                items.push(Node::Sized {level, body: Box::new(rest)});
                break;
            }
            items.push(self.parse_item()?);
        }
        fix_unary(&mut items);
        Ok(Node::Row(items))
    }

    /// 要素 1 つ。上付きと下付きが付いていればまとめて読む
    fn parse_item(&mut self) -> Result<Node, Error> {
        let base = if matches!(self.peek(), Some(Token::Caret) | Some(Token::Underscore)) {
            node::empty()
        } else {
            self.parse_atom()?
        };
        let mut limits = self.last_limits;
        let mut superscript = None;
        let mut subscript = None;
        loop {
            match self.peek() {
                Some(Token::Caret) => self.parse_script(&mut superscript, '^')?,
                Some(Token::Underscore) => self.parse_script(&mut subscript, '_')?,
                Some(Token::Command(name)) if name == "limits" => {
                    self.position += 1;
                    limits = Limits::Always;
                }
                Some(Token::Command(name)) if name == "nolimits" => {
                    self.position += 1;
                    limits = Limits::Never;
                }
                _ => break,
            }
        }
        if superscript.is_none() && subscript.is_none() {
            return Ok(base);
        }
        Ok(Node::Scripts {base: Box::new(base), superscript, subscript, limits})
    }

    /// ^ か _ の位置から、その引数を読んで slot に入れる
    fn parse_script(&mut self, slot: &mut Option<Box<Node>>, mark: char) -> Result<(), Error> {
        let start = self.spot();
        self.position += 1;
        if slot.is_some() {
            return Err(Error::new(format!("{mark} が二重になっています"), start));
        }
        let argument = self
            .parse_atom()
            .map_err(|error| prefix(error, &format!("{mark} の後: ")))?;
        *slot = Some(Box::new(argument));
        Ok(())
    }

    /// アトム 1 つ。波括弧のまとまり、コマンド、1 文字のどれか
    fn parse_atom(&mut self) -> Result<Node, Error> {
        self.last_limits = Limits::Never;
        let start = self.spot();
        match self.peek() {
            Some(Token::LeftBrace) => {
                self.position += 1;
                self.finish_group(start)
            }
            Some(Token::Character(character)) => {
                self.position += 1;
                Ok(character_node(character))
            }
            Some(Token::Text(body)) => {
                self.position += 1;
                Ok(Node::Text(body))
            }
            Some(Token::Command(name)) if name != "end" && name != "right" => {
                self.position += 1;
                self.parse_command(&name, start)
            }
            _ => Err(self.unexpected()),
        }
    }

    /// 開き波括弧を読んだ直後から、閉じ波括弧までを読む
    fn finish_group(&mut self, start: usize) -> Result<Node, Error> {
        let inside = self.parse_row()?;
        match self.peek() {
            Some(Token::RightBrace) => {
                self.position += 1;
                Ok(inside)
            }
            None => Err(Error::new("開き波括弧に対応する閉じ波括弧がありません", start)),
            _ => Err(self.unexpected()),
        }
    }

    fn parse_command(&mut self, name: &str, start: usize) -> Result<Node, Error> {
        match name {
            "frac" | "dfrac" | "tfrac" => {
                let numerator = self.argument(&format!("\\{name} の分子"))?;
                let denominator = self.argument(&format!("\\{name} の分母"))?;
                let fraction = Node::Fraction {
                    numerator: Box::new(numerator),
                    denominator: Box::new(denominator),
                    bar: true,
                };
                Ok(fixed_level(name, fraction))
            }
            "binom" | "dbinom" | "tbinom" => {
                let upper = self.argument(&format!("\\{name} の上"))?;
                let lower = self.argument(&format!("\\{name} の下"))?;
                let stack = Node::Fraction {
                    numerator: Box::new(upper),
                    denominator: Box::new(lower),
                    bar: false,
                };
                let fenced = Node::Fenced {left: "(", right: ")", body: Box::new(stack)};
                Ok(fixed_level(name, fenced))
            }
            "sqrt" => {
                let index = self.parse_optional()?;
                let body = self.argument("\\sqrt の中身")?;
                Ok(Node::Root {index: index.map(Box::new), body: Box::new(body)})
            }
            "overline" | "bar" => {
                let body = self.argument("\\overline の中身")?;
                Ok(Node::Line {body: Box::new(body), above: true})
            }
            "cfrac" => {
                let numerator = self.argument("\\cfrac の分子")?;
                let denominator = self.argument("\\cfrac の分母")?;
                let fraction = Node::Fraction {
                    numerator: Box::new(numerator),
                    denominator: Box::new(denominator),
                    bar: true,
                };
                Ok(Node::Sized {level: Level::Display, body: Box::new(fraction)})
            }
            "overset" | "stackrel" => {
                let above = self.argument(&format!("\\{name} の上"))?;
                let base = self.argument(&format!("\\{name} の土台"))?;
                Ok(Node::Scripts {
                    base: Box::new(base),
                    superscript: Some(Box::new(above)),
                    subscript: None,
                    limits: Limits::Always,
                })
            }
            "underset" => {
                let below = self.argument("\\underset の下")?;
                let base = self.argument("\\underset の土台")?;
                Ok(Node::Scripts {
                    base: Box::new(base),
                    superscript: None,
                    subscript: Some(Box::new(below)),
                    limits: Limits::Always,
                })
            }
            "overbrace" | "underbrace" => {
                let above = name == "overbrace";
                let body = self.argument(&format!("\\{name} の中身"))?;
                let mark = if above {"\u{23de}"} else {"\u{23df}"};
                self.last_limits = Limits::Always;
                Ok(Node::Accent {mark, body: Box::new(body), above})
            }
            "overrightarrow" | "overleftarrow" => {
                let body = self.argument(&format!("\\{name} の中身"))?;
                let mark = if name == "overrightarrow" {"\u{20d7}"} else {"\u{20d6}"};
                Ok(Node::Accent {mark, body: Box::new(body), above: true})
            }
            "xrightarrow" | "xleftarrow" | "xhookrightarrow" | "xrightleftharpoons" => {
                let below = self.parse_optional()?;
                let above = self.argument(&format!("\\{name} の上"))?;
                let mark = match name {
                    "xleftarrow" => "\u{27f5}",
                    "xhookrightarrow" => "\u{21aa}",
                    "xrightleftharpoons" => "\u{21cc}",
                    _ => "\u{27f6}",
                };
                Ok(Node::Extend {
                    mark,
                    above: Some(Box::new(above)),
                    below: below.map(Box::new),
                })
            }
            "substack" => self.parse_substack(start),
            "pmod" | "mod" => {
                let value = self.argument(&format!("\\{name} の中身"))?;
                Ok(modulo(value, name == "pmod"))
            }
            "big" | "Big" | "bigg" | "Bigg" | "bigl" | "Bigl" | "biggl" | "Biggl"
            | "bigr" | "Bigr" | "biggr" | "Biggr" | "bigm" | "Bigm" | "biggm" | "Biggm" => {
                self.parse_sized_delimiter(name)
            }
            "displaystyle" | "textstyle" | "scriptstyle" | "scriptscriptstyle" => {
                Err(Error::new(format!("\\{name} はまとまりの先頭に置きます"), start))
            }
            "underline" => {
                let body = self.argument("\\underline の中身")?;
                Ok(Node::Line {body: Box::new(body), above: false})
            }
            "phantom" | "hphantom" | "vphantom" => {
                let body = self.argument(&format!("\\{name} の中身"))?;

                Ok(Node::Phantom {
                    body: Box::new(body),
                    width: name != "vphantom",
                    height: name != "hphantom",
                })
            }
            "operatorname" => {
                // \operatorname* は \lim と同じく、display のとき添字を上下に置く
                let starred = matches!(self.peek(), Some(Token::Character('*')));

                if starred {
                    self.position += 1;
                }

                let body = self.argument("\\operatorname の中身")?;
                self.last_limits = if starred {Limits::Auto} else {Limits::Never};

                Ok(Node::Operator {body: Box::new(body)})
            }
            "left" => self.parse_fence(start),
            "begin" => self.parse_environment(start),
            "limits" | "nolimits" => Err(Error::new(
                format!("\\{name} は演算子の直後にだけ置けます"),
                start,
            )),
            "hline" => Err(Error::new("\\hline は行列の行頭にだけ置けます", start)),
            "[" | "]" | "(" | ")" => Err(Error::new(
                format!("\\{name} は式全体を囲むときにだけ使えます"),
                start,
            )),
            _ => self.parse_named(name, start),
        }
    }

    /// 表を引いて決まるコマンド
    fn parse_named(&mut self, name: &str, start: usize) -> Result<Node, Error> {
        if let Some(style) = symbol::style(name) {
            let body = self.argument(&format!("\\{name} の中身"))?;
            return Ok(Node::Styled {style, body: Box::new(body)});
        }
        if let Some(mark) = symbol::accent(name) {
            let body = self.argument(&format!("\\{name} の中身"))?;
            return Ok(Node::Accent {mark, body: Box::new(body), above: true});
        }
        if let Some(width) = symbol::space(name) {
            return Ok(Node::Symbol {text: String::new(), class: Class::Space(width)});
        }
        match symbol::symbol(name) {
            Some((text, class)) => {
                self.last_limits = if symbol::takes_limits(name) {
                    Limits::Auto
                } else {
                    Limits::Never
                };
                Ok(Node::Symbol {text: text.to_string(), class})
            }
            None => Err(Error::new(format!("対応していないコマンドです: \\{name}"), start)),
        }
    }

    /// コマンドの引数を 1 つ読む。失敗したらどのコマンドの引数かを添える
    fn argument(&mut self, owner: &str) -> Result<Node, Error> {
        self.parse_atom().map_err(|error| prefix(error, &format!("{owner}: ")))
    }

    /// 組み方を切り替えるコマンドなら、その段階を返して読み進める
    fn style_switch(&mut self) -> Option<Level> {
        let Some(Token::Command(name)) = self.peek() else {
            return None;
        };
        let level = match name.as_str() {
            "displaystyle" => Level::Display,
            "textstyle" => Level::Text,
            "scriptstyle" => Level::Script,
            "scriptscriptstyle" => Level::ScriptScript,
            _ => return None,
        };
        self.position += 1;
        Some(level)
    }


    /// \big などの、大きさを決めた括弧を読む
    fn parse_sized_delimiter(&mut self, name: &str) -> Result<Node, Error> {
        let text = self.read_delimiter(&format!("\\{name}"))?;
        let height = if name.starts_with("Bigg") {
            3.0
        } else if name.starts_with("bigg") {
            2.4
        } else if name.starts_with("Big") {
            1.8
        } else {
            1.2
        };
        let class = match name.chars().last() {
            Some('l') => Class::Open,
            Some('r') => Class::Close,
            Some('m') => Class::Relation,
            _ => Class::Ordinary,
        };
        Ok(Node::Delimiter {text, height, class})
    }


    /// \substack{a \\ b} を読む
    fn parse_substack(&mut self, start: usize) -> Result<Node, Error> {
        if self.peek() != Some(Token::LeftBrace) {
            return Err(Error::new("\\substack の後には {…} が必要です", start));
        }

        self.position += 1;
        let mut rows = Vec::new();

        loop {
            rows.push(vec![self.parse_row()?]);

            match self.peek() {
                Some(Token::RowBreak) => self.position += 1,
                Some(Token::RightBrace) => {
                    self.position += 1;

                    return Ok(Node::Matrix {
                        rows,
                        left: "",
                        right: "",
                        align: vec![Align::Center],
                        lines: Vec::new(),
                        spacing: Spacing::Matrix,
                    });
                }
                _ => return Err(self.unexpected()),
            }
        }
    }


    /// \sqrt[3] の [3] を読む
    fn parse_optional(&mut self) -> Result<Option<Node>, Error> {
        if self.peek() != Some(Token::Character('[')) {
            return Ok(None);
        }
        let start = self.spot();
        self.position += 1;
        let outer = self.stop_at_bracket;
        self.stop_at_bracket = true;
        let inside = self.parse_row();
        self.stop_at_bracket = outer;
        let inside = inside?;
        match self.peek() {
            Some(Token::Character(']')) => {
                self.position += 1;
                Ok(Some(inside))
            }
            _ => Err(Error::new("[ に対応する ] がありません", start)),
        }
    }

    /// \left から \right までを読む
    fn parse_fence(&mut self, start: usize) -> Result<Node, Error> {
        let left = self.read_delimiter("\\left")?;
        let body = self.parse_row()?;
        match self.peek() {
            Some(Token::Command(name)) if name == "right" => {
                self.position += 1;
                let right = self.read_delimiter("\\right")?;
                Ok(Node::Fenced {left, right, body: Box::new(body)})
            }
            _ => Err(Error::new("\\left に対応する \\right がありません", start)),
        }
    }

    /// \left と \right の後の括弧を読む
    fn read_delimiter(&mut self, owner: &str) -> Result<&'static str, Error> {
        let spot = self.spot();
        let key = match self.peek() {
            Some(Token::Character(character)) => character.to_string(),
            Some(Token::Command(name)) => format!("\\{name}"),
            _ => return Err(Error::new(format!("{owner} の後には括弧が必要です"), spot)),
        };
        match symbol::delimiter(&key) {
            Some(found) => {
                self.position += 1;
                Ok(found)
            }
            None => Err(Error::new(format!("{owner} の括弧には使えません: {key}"), spot)),
        }
    }

    fn parse_environment(&mut self, start: usize) -> Result<Node, Error> {
        let name = self.parse_environment_name(start)?;

        if let Some(spacing) = aligned_spacing(&name) {
            return self.parse_aligned(&name, spacing, start);
        }

        let (left, right, default_align) = match name.as_str() {
            "matrix" => ("", "", Align::Center),
            "pmatrix" => ("(", ")", Align::Center),
            "bmatrix" => ("[", "]", Align::Center),
            "Bmatrix" => ("{", "}", Align::Center),
            "vmatrix" => ("|", "|", Align::Center),
            "Vmatrix" => ("‖", "‖", Align::Center),
            "array" => ("", "", Align::Center),
            "cases" => ("{", "", Align::Left),
            _ => return Err(Error::new(format!("対応していない環境です: {name}"), start)),
        };
        let mut align = if name == "array" {
            self.parse_column_spec(start)?
        } else {
            Vec::new()
        };
        let (rows, lines) = self.parse_cells(&name, start, false)?;
        self.close_environment(&name, start)?;
        let columns = rows.iter().map(Vec::len).max().unwrap_or(0);
        while align.len() < columns {
            align.push(default_align);
        }
        Ok(Node::Matrix {rows, left, right, align, lines, spacing: Spacing::Matrix})
    }

    fn parse_aligned(&mut self, name: &str, spacing: Spacing, start: usize) -> Result<Node, Error> {
        let gather = matches!(name, "gather" | "gather*" | "gathered");
        let pair_count = if spacing == Spacing::Alignat {Some(self.parse_pair_count(start)?)} else {None};
        let (rows, lines) = self.parse_cells(name, start, !gather)?;
        self.close_environment(name, start)?;
        let columns = rows.iter().map(Vec::len).max().unwrap_or(0);

        if gather && columns > 1 {
            return Err(Error::new(format!("{name} では & を使えません"), start));
        }

        if let Some(count) = pair_count.filter(|count| columns > count * 2) {
            let message = format!("{name} の列が {{{count}}} で指定した組の数より多くあります");
            return Err(Error::new(message, start));
        }

        let align = (0..columns)
            .map(|index| match (gather, index % 2) {
                (true, _) => Align::Center,
                (false, 0) => Align::Right,
                _ => Align::Left,
            })
            .collect();
        let rows = rows
            .into_iter()
            .map(|row| {
                row.into_iter()
                    .map(|cell| Node::Sized {level: Level::Display, body: Box::new(cell)})
                    .collect()
            })
            .collect();

        Ok(Node::Matrix {rows, left: "", right: "", align, lines, spacing})
    }

    fn parse_pair_count(&mut self, start: usize) -> Result<usize, Error> {
        if self.peek() != Some(Token::LeftBrace) {
            return Err(Error::new("alignat の後には {組の数} が必要です", start));
        }

        self.position += 1;
        let mut digits = String::new();

        while let Some(Token::Character(character)) = self.peek() {
            if !character.is_ascii_digit() {
                break;
            }

            digits.push(character);
            self.position += 1;
        }

        if self.peek() != Some(Token::RightBrace) {
            return Err(Error::new("組の数は数字で書きます", self.spot()));
        }

        self.position += 1;

        match digits.parse::<usize>() {
            Ok(count) if count > 0 => Ok(count),
            _ => Err(Error::new("組の数は 1 以上の数字で書きます", start)),
        }
    }

    fn close_environment(&mut self, name: &str, start: usize) -> Result<(), Error> {
        let closing = self.parse_environment_name(start)?;

        if closing != name {
            let message = format!("\\begin{{{name}}} と \\end{{{closing}}} が食い違っています");
            return Err(Error::new(message, start));
        }

        Ok(())
    }

    /// \begin と \end の後の {環境名}
    fn parse_environment_name(&mut self, start: usize) -> Result<String, Error> {
        if self.peek() != Some(Token::LeftBrace) {
            return Err(Error::new("\\begin と \\end の後には {環境名} が必要です", start));
        }
        self.position += 1;
        let mut name = String::new();
        loop {
            match self.peek() {
                Some(Token::Character(character))
                    if character.is_ascii_alphanumeric() || character == '*' =>
                {
                    name.push(character);
                    self.position += 1;
                }
                Some(Token::RightBrace) => {
                    self.position += 1;
                    return Ok(name);
                }
                _ => return Err(Error::new("環境名は英数字で書きます", self.spot())),
            }
        }
    }

    /// array の {ccc} を読む
    fn parse_column_spec(&mut self, start: usize) -> Result<Vec<Align>, Error> {
        if self.peek() != Some(Token::LeftBrace) {
            return Err(Error::new("array の後には {列の指定} が必要です", start));
        }
        self.position += 1;
        let mut align = Vec::new();
        loop {
            let found = match self.peek() {
                Some(Token::Character('l')) => Some(Align::Left),
                Some(Token::Character('c')) => Some(Align::Center),
                Some(Token::Character('r')) => Some(Align::Right),
                Some(Token::Character('|')) => None,
                Some(Token::RightBrace) => {
                    self.position += 1;
                    return Ok(align);
                }
                _ => return Err(Error::new("列の指定に使えるのは l c r です", self.spot())),
            };
            self.position += 1;
            if let Some(found) = found {
                align.push(found);
            }
        }
    }

    /// 行列の中身。セルは & で、行は \\ で区切る
    ///
    /// seed が true なら、2 列目・4 列目…の頭に空のまとまりを置く。LaTeX の align と同じく、
    /// &= の = が、左に何もなくても関係記号の空きを取る
    fn parse_cells(
        &mut self,
        name: &str,
        start: usize,
        seed: bool,
    ) -> Result<(Vec<Vec<Node>>, Vec<usize>), Error> {
        let mut rows: Vec<Vec<Node>> = Vec::new();
        let mut lines: Vec<usize> = Vec::new();
        let mut row: Vec<Node> = Vec::new();
        loop {
            if row.is_empty() {
                while self.peek() == Some(Token::Command("hline".to_string())) {
                    self.position += 1;
                    lines.push(rows.len());
                }
            }
            let lead = if seed && row.len() % 2 == 1 {vec![node::empty()]} else {Vec::new()};
            row.push(self.parse_row_from(lead)?);
            match self.peek() {
                Some(Token::Ampersand) => self.position += 1,
                Some(Token::RowBreak) => {
                    self.position += 1;
                    rows.push(std::mem::take(&mut row));
                }
                Some(Token::Command(command)) if command == "end" => {
                    self.position += 1;
                    let trailing = row.len() == 1 && node::is_empty(&row[0]) && !rows.is_empty();
                    if !trailing {
                        rows.push(row);
                    }
                    return Ok((rows, lines));
                }
                None => {
                    let message = format!("\\begin{{{name}}} に対応する \\end がありません");
                    return Err(Error::new(message, start));
                }
                _ => return Err(self.unexpected()),
            }
        }
    }
}


fn aligned_spacing(name: &str) -> Option<Spacing> {
    match name {
        "align" | "align*" | "aligned" | "split" | "gather" | "gather*" | "gathered" => Some(Spacing::Align),
        "alignat" | "alignat*" | "alignedat" => Some(Spacing::Alignat),
        _ => None,
    }
}


/// \pmod{n} と \mod{n} の組み立て
fn modulo(value: Node, parenthesis: bool) -> Node {
    let space = |width: f64| Node::Symbol {text: String::new(), class: Class::Space(width)};
    let word = |text: &str, class: Class| Node::Symbol {text: text.to_string(), class};
    let mut items = vec![space(1.0)];

    if parenthesis {
        items.push(word("(", Class::Open));
    }

    items.push(word("mod", Class::Ordinary));
    items.push(space(4.0 / 18.0));
    items.push(value);

    if parenthesis {
        items.push(word(")", Class::Close));
    }

    Node::Row(items)
}


/// \dfrac や \tbinom のように、組み方の段階を名前で決めているものを包む
///
/// d で始まるものは display、t で始まるものは text の組み方で組む。\cfrac と同じ作りである
fn fixed_level(name: &str, body: Node) -> Node {
    let level = match name.chars().next() {
        Some('d') => Level::Display,
        Some('t') => Level::Text,
        _ => return body,
    };

    Node::Sized {level, body: Box::new(body)}
}


/// エラーの文言の前に、どこで起きたかを足す
fn prefix(error: Error, text: &str) -> Error {
    Error {message: format!("{text}{}", error.message), position: error.position}
}


/// 1 文字を記号のノードにする
fn character_node(character: char) -> Node {
    let (text, class) = match character {
        'a'..='z' | 'A'..='Z' => (character.to_string(), Class::Variable),
        '+' => ("+".to_string(), Class::Binary),
        '-' => ("−".to_string(), Class::Binary),
        '*' => ("∗".to_string(), Class::Binary),
        '=' | '<' | '>' => (character.to_string(), Class::Relation),
        ':' => (":".to_string(), Class::Relation),
        '(' | '[' => (character.to_string(), Class::Open),
        ')' | ']' => (character.to_string(), Class::Close),
        ',' | ';' => (character.to_string(), Class::Punctuation),
        '\'' => ("′".to_string(), Class::Ordinary),
        _ => (character.to_string(), Class::Ordinary),
    };
    Node::Symbol {text, class}
}


/// 行頭や記号の直後の + − は二項演算子ではないので、ふつうの記号に直す
fn fix_unary(items: &mut [Node]) {
    let mut wants_operand = true;
    for item in items.iter_mut() {
        match item {
            Node::Symbol {class, ..} => {
                if matches!(class, Class::Space(_)) {
                    continue;
                }
                if *class == Class::Binary && wants_operand {
                    *class = Class::Ordinary;
                }
                wants_operand = matches!(
                    *class,
                    Class::Binary
                        | Class::Relation
                        | Class::Open
                        | Class::Punctuation
                        | Class::Large
                        | Class::Function
                );
            }
            // \operatorname の後も、\sin の後と同じく + − は符号になる
            Node::Operator {..} => wants_operand = true,
            _ => wants_operand = false,
        }
    }
}


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

    fn symbol_at(node: &Node, index: usize) -> (String, Class) {
        match node {
            Node::Row(items) => match &items[index] {
                Node::Symbol {text, class} => (text.clone(), *class),
                other => panic!("記号ではありません: {other:?}"),
            },
            other => panic!("Row ではありません: {other:?}"),
        }
    }

    #[test]
    fn reads_a_fraction() {
        let tree = parse(r"\frac{a}{b}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Fraction {bar: true, ..}));
    }

    #[test]
    fn fixed_fractions_carry_their_level() {
        for (source, wanted) in [
            (r"\dfrac{a}{b}", Level::Display),
            (r"\tfrac{a}{b}", Level::Text),
            (r"\dbinom{n}{k}", Level::Display),
            (r"\tbinom{n}{k}", Level::Text),
        ] {
            let tree = parse(source).unwrap();
            let Node::Row(items) = &tree else {panic!("Row ではありません")};
            assert!(
                matches!(&items[0], Node::Sized {level, ..} if *level == wanted),
                "段階が付いていません: {source}"
            );
        }

        let tree = parse(r"\frac{a}{b}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Fraction {..}), "\\frac に段階が付いています");
    }

    #[test]
    fn unary_minus_becomes_ordinary() {
        assert_eq!(symbol_at(&parse("-x").unwrap(), 0).1, Class::Ordinary);
        assert_eq!(symbol_at(&parse("a-b").unwrap(), 1).1, Class::Binary);
        assert_eq!(symbol_at(&parse("=-1").unwrap(), 1).1, Class::Ordinary);
    }

    #[test]
    fn spaces_do_not_break_the_unary_rule() {
        assert_eq!(symbol_at(&parse(r"a \quad - b").unwrap(), 2).1, Class::Binary);
    }

    #[test]
    fn reads_scripts() {
        let tree = parse("x_i^2").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        let Node::Scripts {superscript, subscript, ..} = &items[0] else {
            panic!("Scripts ではありません")
        };
        assert!(superscript.is_some());
        assert!(subscript.is_some());
    }

    #[test]
    fn large_operators_take_limits_by_default() {
        let tree = parse(r"\sum_{k}^{n}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Scripts {limits: Limits::Auto, ..}));

        let tree = parse(r"\int_0^1").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Scripts {limits: Limits::Never, ..}));

        let tree = parse(r"\sum\nolimits_k").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Scripts {limits: Limits::Never, ..}));
    }

    #[test]
    fn reads_root_with_an_index() {
        let tree = parse(r"\sqrt[3]{x}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        let Node::Root {index, ..} = &items[0] else {panic!("Root ではありません")};
        assert!(index.is_some());
        assert!(matches!(parse(r"\sqrt{x}").unwrap(), Node::Row(ref items)
            if matches!(items[0], Node::Root {index: None, ..})));
    }

    #[test]
    fn reads_fences() {
        let tree = parse(r"\left( x \right)").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(&items[0], Node::Fenced {left: "(", right: ")", ..}));

        let tree = parse(r"\left. x \right|").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(&items[0], Node::Fenced {left: "", right: "|", ..}));
    }

    #[test]
    fn matrix_shape_and_lines() {
        fn shape(source: &str) -> (Vec<usize>, Vec<usize>) {
            match parse(source).unwrap() {
                Node::Row(items) => match &items[0] {
                    Node::Matrix {rows, lines, ..} => {
                        (rows.iter().map(Vec::len).collect(), lines.clone())
                    }
                    other => panic!("行列ではありません: {other:?}"),
                },
                other => panic!("Row ではありません: {other:?}"),
            }
        }
        assert_eq!(shape(r"\begin{bmatrix} 1 & 2 & 3 \\ 4 & 5 & 6 \end{bmatrix}").0, vec![3, 3]);
        assert_eq!(shape(r"\begin{matrix} a \\ b \\ \end{matrix}").0, vec![1, 1]);
        assert_eq!(shape(r"\begin{array}{cc} \hline a & b \\ \hline \end{array}").1, vec![0, 1]);
    }

    #[test]
    fn array_reads_its_column_spec() {
        let tree = parse(r"\begin{array}{l|rc} a & b & c \end{array}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        let Node::Matrix {align, ..} = &items[0] else {panic!("行列ではありません")};
        assert_eq!(align, &vec![Align::Left, Align::Right, Align::Center]);
    }

    #[test]
    fn cases_align_to_the_left() {
        let tree = parse(r"\begin{cases} a & b \\ c & d \end{cases}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        let Node::Matrix {align, left, ..} = &items[0] else {panic!("行列ではありません")};
        assert_eq!(*left, "{");
        assert_eq!(align, &vec![Align::Left, Align::Left]);
    }

    #[test]
    fn reads_text_and_styles() {
        assert_eq!(parse(r"\text{a b}").unwrap(), Node::Row(vec![Node::Text("a b".to_string())]));
        let tree = parse(r"\mathbb{R}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Styled {style: crate::node::Style::Blackboard, ..}));
    }

    #[test]
    fn operator_names_read_as_functions() {
        let tree = parse(r"\operatorname{ht} x").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Operator {..}));
        assert_eq!(crate::node::class_of(&items[0]), Class::Function);

        let tree = parse(r"\operatorname{gr}_i").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Scripts {limits: Limits::Never, ..}));

        let tree = parse(r"\operatorname*{argmax}_x").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Scripts {limits: Limits::Auto, ..}));

        assert_eq!(symbol_at(&parse(r"\operatorname{sgn} -x").unwrap(), 1).1, Class::Ordinary);
    }

    #[test]
    fn wrappers_choose_the_shape_and_come_off() {
        let plain = parse("x^2").unwrap();

        for (source, wanted) in [
            ("$x^2$", Wrapper::Inline),
            ("$$x^2$$", Wrapper::Display),
            (r"\[x^2\]", Wrapper::Display),
            (r"\(x^2\)", Wrapper::Inline),
            (r"\begin{equation} x^2 \end{equation}", Wrapper::Display),
            (r"\begin{equation*} x^2 \end{equation*}", Wrapper::Display),
            (r"\begin{displaymath} x^2 \end{displaymath}", Wrapper::Display),
            (r"\begin{math} x^2 \end{math}", Wrapper::Inline),
            ("x^2", Wrapper::None),
        ] {
            let (node, wrapper) = parse_wrapped(source).unwrap();
            assert_eq!(wrapper, wanted, "囲みの読み違い: {source}");
            assert_eq!(node, plain, "中身が変わっています: {source}");
        }
    }

    #[test]
    fn environments_used_inside_math_are_not_wrappers() {
        let (node, wrapper) = parse_wrapped(r"\begin{pmatrix} a & b \end{pmatrix}").unwrap();
        assert_eq!(wrapper, Wrapper::None);
        let Node::Row(items) = &node else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Matrix {..}));
    }

    fn matrix_of(source: &str) -> (Vec<Vec<Node>>, Vec<Align>, Spacing) {
        match parse(source).unwrap() {
            Node::Row(items) => match &items[0] {
                Node::Matrix {rows, align, spacing, ..} => (rows.clone(), align.clone(), *spacing),
                other => panic!("行列ではありません: {other:?}"),
            },
            other => panic!("Row ではありません: {other:?}"),
        }
    }

    #[test]
    fn align_alternates_right_and_left() {
        for source in [
            r"\begin{align} a &= b & c &= d \end{align}",
            r"\begin{align*} a &= b & c &= d \end{align*}",
            r"\begin{aligned} a &= b & c &= d \end{aligned}",
        ] {
            let (rows, align, spacing) = matrix_of(source);
            assert_eq!(align, vec![Align::Right, Align::Left, Align::Right, Align::Left], "{source}");
            assert_eq!(spacing, Spacing::Align, "{source}");
            assert_eq!(rows[0].len(), 4, "{source}");
        }

        let (rows, _, _) = matrix_of(r"\begin{align} a &= b \\ c &= d \\ \end{align}");
        assert_eq!(rows.len(), 2, "最後の \\\\ で空の行ができています");
    }

    #[test]
    fn aligned_cells_are_display_and_the_left_columns_start_with_an_empty_group() {
        let (rows, _, _) = matrix_of(r"\begin{aligned} a &= b \end{aligned}");
        let Node::Sized {level: Level::Display, body} = &rows[0][1] else {panic!("display で包まれていません")};
        let Node::Row(items) = body.as_ref() else {panic!("Row ではありません")};
        assert!(node::is_empty(&items[0]), "空のまとまりがありません");
        assert!(matches!(&items[1], Node::Symbol {class: Class::Relation, ..}));

        let Node::Sized {body, ..} = &rows[0][0] else {panic!("display で包まれていません")};
        let Node::Row(items) = body.as_ref() else {panic!("Row ではありません")};
        assert!(matches!(&items[0], Node::Symbol {class: Class::Variable, ..}), "右寄せの列に空のまとまりがあります");
    }

    #[test]
    fn gather_centers_a_single_column() {
        for source in [
            r"\begin{gather} a \\ b + c \end{gather}",
            r"\begin{gather*} a \\ b + c \end{gather*}",
            r"\begin{gathered} a \\ b + c \end{gathered}",
        ] {
            let (rows, align, spacing) = matrix_of(source);
            assert_eq!(rows.len(), 2, "{source}");
            assert_eq!(align, vec![Align::Center], "{source}");
            assert_eq!(spacing, Spacing::Align, "{source}");
        }

        assert!(parse(r"\begin{gather} a & b \end{gather}").is_err());
    }

    #[test]
    fn alignat_reads_its_pair_count() {
        for source in [
            r"\begin{alignat}{2} a &= b & c &= d \end{alignat}",
            r"\begin{alignat*}{2} a &= b & c &= d \end{alignat*}",
            r"\begin{alignedat}{2} a &= b & c &= d \end{alignedat}",
        ] {
            let (_, align, spacing) = matrix_of(source);
            assert_eq!(align.len(), 4, "{source}");
            assert_eq!(spacing, Spacing::Alignat, "{source}");
        }

        assert!(parse(r"\begin{alignat}{1} a &= b & c &= d \end{alignat}").is_err());
        assert!(parse(r"\begin{alignat} a &= b \end{alignat}").is_err());
        assert!(parse(r"\begin{alignat}{0} a &= b \end{alignat}").is_err());
        assert!(parse(r"\begin{alignat}{x} a &= b \end{alignat}").is_err());
    }

    #[test]
    fn align_and_gather_are_display_and_keep_their_rows() {
        for source in [
            r"\begin{align} a &= b \\ c &= d \end{align}",
            r"\begin{align*} a &= b \end{align*}",
            r"\begin{gather} a \\ b \end{gather}",
            r"\begin{gather*} a \\ b \end{gather*}",
            r"\begin{alignat}{1} a &= b \end{alignat}",
            r"\begin{alignat*}{1} a &= b \end{alignat*}",
        ] {
            let (node, wrapper) = parse_wrapped(source).unwrap();
            assert_eq!(wrapper, Wrapper::Display, "{source}");
            let Node::Row(items) = &node else {panic!("Row ではありません")};
            assert!(matches!(items[0], Node::Matrix {..}), "行列になっていません: {source}");
        }

        for source in [
            r"\begin{aligned} a &= b \end{aligned}",
            r"\begin{gathered} a \\ b \end{gathered}",
            r"\begin{split} a &= b \end{split}",
        ] {
            let (_, wrapper) = parse_wrapped(source).unwrap();
            assert_eq!(wrapper, Wrapper::None, "{source}");
        }
    }

    #[test]
    fn split_sits_inside_equation() {
        let source = r"\begin{equation} \begin{split} a &= b \\ &= c \end{split} \end{equation}";
        let (node, wrapper) = parse_wrapped(source).unwrap();
        assert_eq!(wrapper, Wrapper::Display);
        let Node::Row(items) = &node else {panic!("Row ではありません")};
        let Node::Matrix {rows, spacing, ..} = &items[0] else {panic!("行列ではありません")};
        assert_eq!(rows.len(), 2);
        assert_eq!(*spacing, Spacing::Align);
    }

    #[test]
    fn a_lone_dollar_is_an_error() {
        for source in ["$x", "x$", "a $ b", "$$x$"] {
            let error = parse(source).unwrap_err();
            assert!(error.message.contains("囲む記号"), "{source}: {}", error.message);
        }

        // 文字として出したいときはエスケープする
        assert!(parse(r"\$100").is_ok());
        // \text の中はそのまま通す
        assert!(parse(r"\text{100$}").is_ok());
    }

    #[test]
    fn wrapper_commands_are_refused_in_the_middle() {
        let error = parse(r"x + \[ y \]").unwrap_err();
        assert!(error.message.contains("式全体を囲む"), "{}", error.message);
    }

    #[test]
    fn stacking_commands_force_the_limits() {
        let tree = parse(r"\overset{f}{\to}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert!(matches!(items[0], Node::Scripts {limits: Limits::Always, ..}));

        let tree = parse(r"\underset{g}{\to}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        let Node::Scripts {subscript, superscript, ..} = &items[0] else {
            panic!("Scripts ではありません")
        };
        assert!(subscript.is_some() && superscript.is_none());
    }

    #[test]
    fn a_style_switch_covers_the_rest_of_the_row() {
        let tree = parse(r"a \displaystyle b c").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        assert_eq!(items.len(), 2, "切り替えから先がまとまっていません");
        assert!(matches!(items[1], Node::Sized {level: Level::Display, ..}));
    }

    #[test]
    fn sized_delimiters_carry_their_height() {
        let tree = parse(r"\bigl( \Bigg]").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        let Node::Delimiter {height, class, ..} = &items[0] else {panic!("Delimiter ではありません")};
        assert_eq!(*height, 1.2);
        assert_eq!(*class, Class::Open);
        let Node::Delimiter {height, ..} = &items[1] else {panic!("Delimiter ではありません")};
        assert_eq!(*height, 3.0);
    }

    #[test]
    fn an_extending_arrow_takes_a_label_above_and_below() {
        let tree = parse(r"\xrightarrow[g]{f}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        let Node::Extend {above, below, ..} = &items[0] else {panic!("Extend ではありません")};
        assert!(above.is_some() && below.is_some());

        let tree = parse(r"\xrightarrow{f}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        let Node::Extend {below, ..} = &items[0] else {panic!("Extend ではありません")};
        assert!(below.is_none());
    }

    #[test]
    fn substack_stacks_its_rows() {
        let tree = parse(r"\substack{i = 1 \\ i \neq j}").unwrap();
        let Node::Row(items) = &tree else {panic!("Row ではありません")};
        let Node::Matrix {rows, left, ..} = &items[0] else {panic!("行列ではありません")};
        assert_eq!(rows.len(), 2);
        assert_eq!(*left, "");
    }

    #[test]
    fn reports_where_it_failed() {
        let error = parse(r"x + \foo").unwrap_err();
        assert_eq!(error.position, 4);
        assert!(error.message.contains("\\foo"));

        let error = parse(r"\frac{a}{b").unwrap_err();
        assert!(error.message.starts_with("\\frac の分母: "));
    }

    #[test]
    fn errors() {
        assert!(parse("{x").is_err());
        assert!(parse("x}").is_err());
        assert!(parse(r"\foo").is_err());
        assert!(parse(r"\frac{a}").is_err());
        assert!(parse("x^").is_err());
        assert!(parse("x^1^2").is_err());
        assert!(parse("a & b").is_err());
        assert!(parse(r"a \\ b").is_err());
        assert!(parse(r"\begin{pmatrix} a & b").is_err());
        assert!(parse(r"\begin{pmatrix} a \end{bmatrix}").is_err());
        assert!(parse(r"\begin{foo} a \end{foo}").is_err());
        assert!(parse(r"\end{matrix}").is_err());
        assert!(parse(r"\left( x").is_err());
        assert!(parse(r"x \right)").is_err());
        assert!(parse(r"\left x \right x").is_err());
        assert!(parse(r"\sqrt[3{x}").is_err());
    }
}