rink-core 0.9.0

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.

use crate::ast::*;
use crate::output::Digits;
use crate::types::{BigInt, BigRat, Numeric, TimeZone};
use std::iter::Peekable;
use std::str::Chars;

#[derive(Debug, Clone)]
pub enum Token {
    Newline,
    Comment(usize),
    Ident(String),
    Decimal(String, Option<String>, Option<String>),
    Exponent(String),
    Hex(String),
    Oct(String),
    Bin(String),
    Quote(String),
    Slash,
    Pipe,
    Semicolon,
    Equals,
    Caret,
    Eof,
    LPar,
    RPar,
    Plus,
    Minus,
    Asterisk,
    DashArrow,
    Colon,
    DoubleLAngle,
    DoubleRAngle,
    KeywordMod,
    KeywordXor,
    KeywordOr,
    KeywordAnd,
    Date(Vec<DateToken>),
    Comma,
    Degree(Degree),
    Percent,
    Error(String),
}

fn describe(token: &Token) -> String {
    match *token {
        Token::Newline | Token::Comment(_) => "\\n".to_owned(),
        Token::Ident(_) => "ident".to_owned(),
        Token::Decimal(_, _, _) => "number".to_owned(),
        Token::Exponent(_) => "exponent".to_owned(),
        Token::Hex(_) => "hex".to_owned(),
        Token::Oct(_) => "octal".to_owned(),
        Token::Bin(_) => "binary".to_owned(),
        Token::Quote(_) => "quote".to_owned(),
        Token::Slash => "`/`".to_owned(),
        Token::Pipe => "`|`".to_owned(),
        Token::Semicolon => "`;`".to_owned(),
        Token::Equals => "`=`".to_owned(),
        Token::Caret => "`^`".to_owned(),
        Token::Eof => "eof".to_owned(),
        Token::LPar => "`(`".to_owned(),
        Token::RPar => "`)`".to_owned(),
        Token::Plus => "`+`".to_owned(),
        Token::Minus => "`-`".to_owned(),
        Token::Asterisk => "`*`".to_owned(),
        Token::DashArrow => "`->`".to_owned(),
        Token::Colon => "`:`".to_owned(),
        Token::DoubleLAngle => "`<<`".to_owned(),
        Token::DoubleRAngle => "`>>`".to_owned(),
        Token::KeywordMod => "`mod`".to_owned(),
        Token::KeywordXor => "`xor`".to_owned(),
        Token::KeywordOr => "`or`".to_owned(),
        Token::KeywordAnd => "`and`".to_owned(),
        Token::Date(_) => "date literal".to_owned(),
        Token::Comma => "`,`".to_owned(),
        Token::Percent => "%".to_owned(),
        Token::Degree(ref deg) => format!("`{}`", deg),
        Token::Error(ref e) => format!("<{}>", e),
    }
}

#[derive(Clone)]
pub struct TokenIterator<'a>(Peekable<Chars<'a>>);

impl<'a> TokenIterator<'a> {
    pub fn new(input: &'a str) -> TokenIterator<'a> {
        TokenIterator(input.chars().peekable())
    }
}

/// List of currency signs like `$` etc.
/// Based on the unicode "Currrency Symbol" (Sc) general category.
fn is_currency(ch: char) -> bool {
    match ch {
        '$' | '¢' | '£' | '¤' | '¥' | '֏' | '؋' | '߾' | '߿' | '৲' | '৳' | '৻' | '૱' | '௹' | '฿'
        | '៛' | '₠' | '₡' | '₢' | '₣' | '₤' | '₥' | '₦' | '₧' | '₨' | '₩' | '₪' | '₫' | '€'
        | '₭' | '₮' | '₯' | '₰' | '₱' | '₲' | '₳' | '₴' | '₵' | '₶' | '₷' | '₸' | '₹' | '₺'
        | '₻' | '₼' | '₽' | '₾' | '₿' | '⃀' | '꠸' | '﷼' | '﹩' | '＄' | '￠' | '￡' | '￥'
        | '￦' | '𑿝' | '𑿞' | '𑿟' | '𑿠' | '𞋿' | '𞲰' => true,
        _ => false,
    }
}

fn digit_from_superscript(sup: char) -> Option<char> {
    // From the Unicode "Superscripts and Subscripts" block, U+2070 to U+209F
    match sup {
        '⁰' => Some('0'),
        '¹' => Some('1'),
        '²' => Some('2'),
        '³' => Some('3'),
        '⁴' => Some('4'),
        '⁵' => Some('5'),
        '⁶' => Some('6'),
        '⁷' => Some('7'),
        '⁸' => Some('8'),
        '⁹' => Some('9'),
        _ => None,
    }
}

impl<'a> Iterator for TokenIterator<'a> {
    type Item = Token;

    fn next(&mut self) -> Option<Token> {
        if self.0.peek().is_none() {
            return Some(Token::Eof);
        }
        let res = match self.0.next().unwrap() {
            ' ' | '\t' => return self.next(),
            '\n' => Token::Newline,
            '(' => Token::LPar,
            ')' => Token::RPar,
            '+' => Token::Plus,
            ';' => Token::Semicolon,
            '%' => Token::Percent,
            '=' => Token::Equals,
            '^' => Token::Caret,
            ',' => Token::Comma,
            // U+2044 fraction slash '⁄'
            // U+2215 division slash '∕'
            '|' | '\u{2044}' | '\u{2215}' => Token::Pipe,
            ':' => Token::Colon,
            '→' => Token::DashArrow,
            '<' if self.0.peek().cloned() == Some('<') => {
                self.0.next();
                Token::DoubleLAngle
            }
            '>' if self.0.peek().cloned() == Some('>') => {
                self.0.next();
                Token::DoubleRAngle
            }
            '*' => {
                if self.0.peek().cloned() == Some('*') {
                    self.0.next();
                    Token::Caret
                } else {
                    Token::Asterisk
                }
            }
            // U+22C5 dot operator '⋅'
            // U+00D7 multiplication sign '×'
            '⋅' | '×' => Token::Asterisk,
            '-' => match self.0.peek().cloned() {
                Some('>') => {
                    self.0.next();
                    Token::DashArrow
                }
                _ => Token::Minus,
            },
            // U+2212 minus sign '−'
            '\u{2212}' => Token::Minus,
            // U+00F7 division sign '÷'
            '÷' => Token::Slash,
            '/' => match self.0.peek() {
                Some(&'/') => loop {
                    match self.0.next() {
                        None | Some('\n') => return Some(Token::Comment(1)),
                        _ => (),
                    }
                },
                Some(&'*') => {
                    let mut lines = 0;
                    loop {
                        if let Some(&'\n') = self.0.peek() {
                            lines += 1;
                        }
                        if let Some('*') = self.0.next() {
                            if let Some(&'/') = self.0.peek() {
                                self.0.next();
                                return Some(Token::Comment(lines));
                            }
                        }
                        if self.0.peek() == None {
                            return Some(Token::Error("Expected `*/`, got EOF".to_string()));
                        }
                    }
                }
                _ => Token::Slash,
            },
            x @ '0'..='9' | x @ '.' => {
                if x == '0' && self.0.peek() == Some(&'x') {
                    self.0.next();
                    let mut hex = String::new();

                    while let Some(c) = self.0.peek().cloned() {
                        match c {
                            '0'..='9' | 'a'..='f' | 'A'..='F' => hex.push(self.0.next().unwrap()),
                            '\u{2009}' | '_' => {
                                self.0.next();
                            }
                            _ => break,
                        }
                    }
                    if hex.is_empty() {
                        return Some(Token::Error(
                            "Malformed hexadecimal literal: No digits after 0x".to_owned(),
                        ));
                    }
                    return Some(Token::Hex(hex));
                }

                if x == '0' && self.0.peek() == Some(&'o') {
                    self.0.next();
                    let mut oct = String::new();

                    while let Some(c) = self.0.peek().cloned() {
                        match c {
                            '0'..='7' => oct.push(self.0.next().unwrap()),
                            '\u{2009}' | '_' => {
                                self.0.next();
                            }
                            _ => break,
                        }
                    }
                    if oct.is_empty() {
                        return Some(Token::Error(
                            "Malformed octal literal: No digits after 0o".to_owned(),
                        ));
                    }
                    return Some(Token::Oct(oct));
                }

                if x == '0' && self.0.peek() == Some(&'b') {
                    self.0.next();
                    let mut bin = String::new();

                    while let Some(c) = self.0.peek().cloned() {
                        match c {
                            '0' | '1' => bin.push(self.0.next().unwrap()),
                            '\u{2009}' | '_' => {
                                self.0.next();
                            }
                            _ => break,
                        }
                    }
                    if bin.is_empty() {
                        return Some(Token::Error(
                            "Malformed binary literal: No digits after 0b".to_owned(),
                        ));
                    }
                    return Some(Token::Bin(bin));
                }

                let mut integer = String::new();
                let mut frac = None;
                let mut exp = None;

                // integer component
                if x != '.' {
                    integer.push(x);
                    while let Some(c) = self.0.peek().cloned() {
                        match c {
                            '0'..='9' => integer.push(self.0.next().unwrap()),
                            '\u{2009}' | '_' => {
                                self.0.next();
                            }
                            _ => break,
                        }
                    }
                } else {
                    integer.push('0');
                }
                // fractional component
                if x == '.' || Some('.') == self.0.peek().cloned() {
                    let mut buf = String::new();
                    if x != '.' {
                        self.0.next();
                    }
                    while let Some(c) = self.0.peek().cloned() {
                        match c {
                            '0'..='9' => buf.push(self.0.next().unwrap()),
                            '\u{2009}' | '_' => {
                                self.0.next();
                            }
                            _ => break,
                        }
                    }
                    if buf.is_empty() {
                        return Some(Token::Error(
                            "Malformed number literal: No digits after decimal point".to_owned(),
                        ));
                    }
                    frac = Some(buf)
                }
                // exponent
                if let Some('e') = self.0.peek().cloned().map(|x| x.to_ascii_lowercase()) {
                    let mut buf = String::new();
                    self.0.next();
                    if let Some('e') = self.0.peek().cloned().map(|x| x.to_ascii_lowercase()) {
                        self.0.next();
                    }
                    if let Some(c) = self.0.peek().cloned() {
                        match c {
                            '-' => {
                                buf.push(self.0.next().unwrap());
                            }
                            '+' => {
                                self.0.next();
                            }
                            _ => (),
                        }
                    }
                    while let Some(c) = self.0.peek().cloned() {
                        match c {
                            '0'..='9' => buf.push(self.0.next().unwrap()),
                            // U+2009 thin space ' '
                            '\u{2009}' | '_' => {
                                self.0.next();
                            }
                            _ => break,
                        }
                    }
                    if buf.is_empty() {
                        return Some(Token::Error(
                            "Malformed number literal: No digits after exponent".to_owned(),
                        ));
                    }
                    exp = Some(buf)
                }
                Token::Decimal(integer, frac, exp)
            }
            x if digit_from_superscript(x).is_some() => {
                let mut integer = String::new();
                integer.push(digit_from_superscript(x).unwrap());
                while let Some(c) = self.0.peek().cloned() {
                    if let Some(digit) = digit_from_superscript(c) {
                        self.0.next();
                        integer.push(digit);
                    } else {
                        break;
                    }
                }
                Token::Exponent(integer)
            }
            '\\' => match self.0.next() {
                Some('u') => {
                    let mut buf = String::new();
                    while let Some(c) = self.0.peek().cloned() {
                        if c.is_digit(16) {
                            buf.push(self.0.next().unwrap());
                        } else {
                            break;
                        }
                    }
                    let v = u32::from_str_radix(&*buf, 16).unwrap();
                    if let Some(c) = ::std::char::from_u32(v) {
                        let mut buf = String::new();
                        buf.push(c);
                        Token::Ident(buf)
                    } else {
                        Token::Error(format!("Invalid unicode scalar: {:x}", v))
                    }
                }
                _ => Token::Error("Unexpected \\".to_string()),
            },
            '\'' => {
                let mut buf = String::new();
                loop {
                    match self.0.next() {
                        None | Some('\n') => {
                            return Some(Token::Error("Unexpected newline or EOF".to_string()))
                        }
                        Some('\\') => match self.0.next() {
                            Some('\'') => buf.push('\''),
                            Some('n') => buf.push('\n'),
                            Some('t') => buf.push('\t'),
                            Some(c) => {
                                return Some(Token::Error(format!(
                                    "Invalid escape sequence \\{}",
                                    c
                                )))
                            }
                            None => return Some(Token::Error("Unexpected EOF".to_string())),
                        },
                        Some('\'') => break,
                        Some(c) => buf.push(c),
                    }
                }
                Token::Quote(buf)
            }
            '#' => {
                let mut toks = vec![];
                while self.0.peek().is_some() {
                    let res = match self.0.next().unwrap() {
                        '#' => break,
                        ':' => DateToken::Colon,
                        '-' => DateToken::Dash,
                        '+' => DateToken::Plus,
                        x if x.is_whitespace() => {
                            while self.0.peek().map(|c| c.is_whitespace()).unwrap_or(false) {
                                self.0.next();
                            }
                            DateToken::Space
                        }
                        x if x.is_digit(10) => {
                            let mut integer = String::new();
                            integer.push(x);
                            while let Some(c) = self.0.peek().cloned() {
                                if c.is_digit(10) {
                                    self.0.next();
                                    integer.push(c);
                                } else {
                                    break;
                                }
                            }
                            let frac = if let Some('.') = self.0.peek().cloned() {
                                let mut frac = String::new();
                                self.0.next();
                                while let Some(c) = self.0.peek().cloned() {
                                    if c.is_digit(10) {
                                        self.0.next();
                                        frac.push(c);
                                    } else {
                                        break;
                                    }
                                }
                                Some(frac)
                            } else {
                                None
                            };
                            DateToken::Number(integer, frac)
                        }
                        x => {
                            let mut buf = String::new();
                            buf.push(x);
                            while let Some(c) = self.0.peek().cloned() {
                                if !"#:-+ ".contains(c) && !c.is_digit(10) {
                                    self.0.next();
                                    buf.push(c);
                                } else {
                                    break;
                                }
                            }
                            DateToken::Literal(buf)
                        } //x => DateToken::Error(format!("Unexpected character '{}'", x))
                    };
                    toks.push(res);
                }
                if let Some(&DateToken::Space) = toks.first() {
                    toks.remove(0);
                }
                if let Some(&DateToken::Space) = toks.last() {
                    toks.pop();
                }
                Token::Date(toks)
            }
            '"' => {
                let mut buf = String::new();
                while let Some(c) = self.0.next() {
                    if c == '\\' {
                        if let Some(c) = self.0.next() {
                            buf.push(c);
                        }
                    } else if c == '"' {
                        break;
                    } else {
                        buf.push(c);
                    }
                }
                Token::Ident(buf)
            }
            x => {
                let mut buf = String::new();
                let mut prev = x;
                buf.push(x);
                while let Some(c) = self.0.peek().cloned() {
                    if digit_from_superscript(c).is_some() {
                        // split x² into Ident(x) + Exponent(2)
                        break;
                    } else if c.is_digit(10) && is_currency(prev) {
                        // split $10 into Ident($) + Decimal(10)
                        break;
                    } else if c.is_alphanumeric() || c == '_' || c == '$' {
                        prev = self.0.next().unwrap();
                        buf.push(prev);
                    } else {
                        break;
                    }
                }
                match &*buf {
                    "degC" | "°C" | "celsius" | "℃" => Token::Degree(Degree::Celsius),
                    "degF" | "°F" | "fahrenheit" | "℉" => Token::Degree(Degree::Fahrenheit),
                    "degRé" | "°Ré" | "degRe" | "°Re" | "réaumur" | "reaumur" => {
                        Token::Degree(Degree::Reaumur)
                    }
                    "degRø" | "°Rø" | "degRo" | "°Ro" | "rømer" | "romer" => {
                        Token::Degree(Degree::Romer)
                    }
                    "degDe" | "°De" | "delisle" => Token::Degree(Degree::Delisle),
                    "degN" | "°N" | "degnewton" => Token::Degree(Degree::Newton),
                    "per" => Token::Slash,
                    "to" | "in" => Token::DashArrow,
                    "mod" => Token::KeywordMod,
                    "and" => Token::KeywordAnd,
                    "or" => Token::KeywordOr,
                    "xor" => Token::KeywordXor,
                    _ => Token::Ident(buf),
                }
            }
        };
        Some(res)
    }
}

pub type Iter<'a> = Peekable<TokenIterator<'a>>;

fn attr_from_name(name: &str) -> Option<&'static str> {
    match name {
        "int" | "international" => Some("int"),
        "UKSJJ" => Some("UKSJJ"),
        "UKB" => Some("UKB"),
        "UKC" => Some("UKC"),
        "UKK" => Some("UKK"),
        "imperial" | "british" | "UK" => Some("br"),
        "survey" | "geodetic" => Some("survey"),
        "irish" => Some("irish"),
        "aust" | "australian" => Some("aust"),
        "roman" => Some("roman"),
        "egyptian" => Some("egyptian"),
        "greek" => Some("greek"),
        "olympic" => Some("olympic"),
        _ => None,
    }
}

fn parse_function(iter: &mut Iter<'_>, func: Function) -> Expr {
    let args = match iter.peek().cloned().unwrap() {
        Token::LPar => {
            iter.next();
            let mut args = vec![];
            loop {
                if let Some(&Token::RPar) = iter.peek() {
                    iter.next();
                    break;
                }
                args.push(parse_expr(iter));
                match iter.peek().cloned().unwrap() {
                    Token::Comma => {
                        iter.next();
                    }
                    Token::RPar => (),
                    x => {
                        return Expr::new_error(format!(
                            "Expected `,` or `)`, got {}",
                            describe(&x)
                        ))
                    }
                }
            }
            args
        }
        _ => vec![parse_unary(iter)],
    };
    Expr::new_call(func, args)
}

fn parse_radix(num: &str, base: u32, description: &str) -> Expr {
    BigInt::from_str_radix(num, base)
        .map(|x| BigRat::ratio(&x, &BigInt::one()))
        .map(Numeric::Rational)
        .map(Expr::new_const)
        .unwrap_or_else(|_| Expr::new_error(format!("Failed to parse {}", description)))
}

fn parse_term(iter: &mut Iter<'_>) -> Expr {
    match iter.next().unwrap() {
        Token::Ident(ref id) => {
            if let Some(func) = Function::from_name(id) {
                parse_function(iter, func)
            } else if let Some(attr) = attr_from_name(id) {
                match iter.peek().cloned().unwrap() {
                    Token::Ident(ref name) => {
                        iter.next();
                        Expr::new_unit(format!("{}{}", attr, name))
                    }
                    x => Expr::new_error(format!(
                        "Attribute must be followed by ident, got {}",
                        describe(&x)
                    )),
                }
            } else {
                match iter.peek().cloned().unwrap() {
                    Token::Ident(ref s) if s == "of" => {
                        iter.next();
                        Expr::new_of(id, parse_juxt(iter))
                    }
                    _ => Expr::new_unit(id.to_string()),
                }
            }
        }
        Token::Quote(string) => Expr::Quote { string },
        Token::Decimal(num, frac, exp) => Numeric::from_parts(
            &*num,
            frac.as_ref().map(|x| &**x),
            exp.as_ref().map(|x| &**x),
        )
        .map(Expr::new_const)
        .unwrap_or_else(Expr::new_error),
        Token::Hex(num) => parse_radix(&*num, 16, "hex"),
        Token::Oct(num) => parse_radix(&*num, 8, "octal"),
        Token::Bin(num) => parse_radix(&*num, 2, "binary"),
        Token::LPar => {
            let res = parse_expr(iter);
            match iter.next().unwrap() {
                Token::RPar => res,
                x => Expr::new_error(format!("Expected `)`, got {}", describe(&x))),
            }
        }
        Token::Percent => Expr::new_unit("percent".to_owned()),
        Token::Date(tokens) => Expr::Date { tokens },
        Token::Comment(_) => parse_term(iter),
        x => Expr::new_error(format!("Expected term, got {}", describe(&x))),
    }
}

fn parse_suffix(iter: &mut Iter<'_>) -> Expr {
    let left = parse_term(iter);
    match *iter.peek().unwrap() {
        Token::Percent => {
            let mut left = left;
            while let Some(&Token::Percent) = iter.peek() {
                iter.next();
                left = Expr::new_mul(vec![left, Expr::new_unit("percent".to_owned())]);
            }
            left
        }
        _ => left,
    }
}

fn parse_pow(iter: &mut Iter<'_>) -> Expr {
    let left = parse_suffix(iter);
    match *iter.peek().unwrap() {
        Token::Caret => {
            iter.next();
            let right = parse_pow(iter);
            Expr::new_pow(left, right)
        }
        Token::Exponent(ref exp_str) => {
            let res = Numeric::from_parts(&exp_str, None, None);
            let exp = res.map(Expr::new_const).unwrap_or_else(Expr::new_error);
            iter.next();
            Expr::new_pow(left, exp)
        }
        _ => left,
    }
}

fn parse_unary(iter: &mut Iter<'_>) -> Expr {
    match *iter.peek().unwrap() {
        Token::Plus => {
            iter.next();
            Expr::new_plus(parse_unary(iter))
        }
        Token::Minus => {
            iter.next();
            Expr::new_negate(parse_unary(iter))
        }
        _ => parse_pow(iter),
    }
}

fn parse_frac(iter: &mut Iter<'_>) -> Expr {
    let left = parse_unary(iter);
    match *iter.peek().unwrap() {
        Token::Pipe => {
            iter.next();
            let right = parse_unary(iter);
            Expr::new_frac(left, right)
        }
        _ => left,
    }
}

fn parse_juxt(iter: &mut Iter<'_>) -> Expr {
    let mut terms = vec![parse_frac(iter)];
    loop {
        match iter.peek().cloned().unwrap() {
            Token::Asterisk
            | Token::Slash
            | Token::Comma
            | Token::Equals
            | Token::Plus
            | Token::Minus
            | Token::DashArrow
            | Token::RPar
            | Token::Newline
            | Token::DoubleLAngle
            | Token::DoubleRAngle
            | Token::KeywordMod
            | Token::KeywordAnd
            | Token::KeywordOr
            | Token::KeywordXor
            | Token::Comment(_)
            | Token::Eof => break,
            Token::Degree(deg) => {
                iter.next();
                terms = vec![Expr::new_suffix(deg, Expr::new_mul(terms))]
            }
            _ => terms.push(parse_frac(iter)),
        }
    }
    if terms.len() == 1 {
        terms.pop().unwrap()
    } else {
        Expr::new_mul(terms)
    }
}

fn parse_div(iter: &mut Iter<'_>) -> Expr {
    let mut terms = vec![parse_juxt(iter)];
    loop {
        match iter.peek().cloned().unwrap() {
            Token::Slash => {
                iter.next();
                let left = Expr::new_mul(terms.drain(..).collect());
                terms = vec![Expr::new_frac(left, parse_juxt(iter))];
            }
            Token::Asterisk => {
                iter.next();
                terms.push(parse_juxt(iter));
            }
            Token::DoubleLAngle => {
                iter.next();
                let left = Expr::new_mul(terms.drain(..).collect());
                terms = vec![Expr::new_bin(BinOpType::ShiftL, left, parse_juxt(iter))];
            }
            Token::DoubleRAngle => {
                iter.next();
                let left = Expr::new_mul(terms.drain(..).collect());
                terms = vec![Expr::new_bin(BinOpType::ShiftR, left, parse_juxt(iter))];
            }
            Token::KeywordMod => {
                iter.next();
                let left = Expr::new_mul(terms.drain(..).collect());
                terms = vec![Expr::new_bin(BinOpType::Mod, left, parse_juxt(iter))];
            }
            Token::KeywordAnd => {
                iter.next();
                let left = Expr::new_mul(terms.drain(..).collect());
                terms = vec![Expr::new_bin(BinOpType::And, left, parse_juxt(iter))];
            }
            Token::KeywordOr => {
                iter.next();
                let left = Expr::new_mul(terms.drain(..).collect());
                terms = vec![Expr::new_bin(BinOpType::Or, left, parse_juxt(iter))];
            }
            Token::KeywordXor => {
                iter.next();
                let left = Expr::new_mul(terms.drain(..).collect());
                terms = vec![Expr::new_bin(BinOpType::Xor, left, parse_juxt(iter))];
            }
            _ => break,
        }
    }
    if terms.len() == 1 {
        terms.pop().unwrap()
    } else {
        Expr::new_mul(terms)
    }
}

fn parse_add(iter: &mut Iter<'_>) -> Expr {
    let mut left = parse_div(iter);
    loop {
        match *iter.peek().unwrap() {
            Token::Plus => {
                iter.next();
                let right = parse_div(iter);
                left = Expr::new_add(left, right)
            }
            Token::Minus => {
                iter.next();
                let right = parse_div(iter);
                left = Expr::new_sub(left, right)
            }
            _ => return left,
        }
    }
}

fn parse_eq(iter: &mut Iter<'_>) -> Expr {
    let left = parse_add(iter);
    match iter.peek().cloned().unwrap() {
        Token::Equals => {
            iter.next();
            let right = parse_add(iter);
            Expr::new_equals(left, right)
        }
        _ => left,
    }
}

pub fn parse_expr(iter: &mut Iter<'_>) -> Expr {
    parse_eq(iter)
}

pub fn parse_unitlist(iter: &mut Iter<'_>) -> Option<Vec<String>> {
    let mut expecting_term = true;
    let mut res = vec![];
    loop {
        match iter.next().unwrap() {
            Token::Ident(ref ident) if expecting_term => {
                res.push(ident.clone());
                expecting_term = false;
            }
            Token::Comma | Token::Semicolon if !expecting_term => {
                expecting_term = true;
            }
            Token::Eof | Token::Newline | Token::Comment(_) if !expecting_term => break,
            _ => return None,
        }
    }
    if res.len() > 1 {
        Some(res)
    } else {
        None
    }
}

pub fn parse_offset(iter: &mut Iter<'_>) -> Option<i64> {
    use std::str::FromStr;

    let sign = match iter.next().unwrap() {
        Token::Plus => 1,
        Token::Minus => -1,
        _ => return None,
    };
    let hour = match iter.next().unwrap() {
        Token::Decimal(ref i, None, None) if i.len() == 2 => i.clone(),
        _ => return None,
    };
    match iter.next().unwrap() {
        Token::Colon => (),
        _ => return None,
    }
    let min = match iter.next().unwrap() {
        Token::Decimal(ref i, None, None) if i.len() == 2 => i.clone(),
        _ => return None,
    };
    Some(sign * (i64::from_str(&*hour).unwrap() * 3600 + i64::from_str(&*min).unwrap() * 60))
}

pub fn parse_query(iter: &mut Iter<'_>) -> Query {
    match iter.peek().cloned() {
        Some(Token::Ident(ref s)) if s == "factorize" => {
            iter.next();
            return Query::Factorize(parse_eq(iter));
        }
        Some(Token::Ident(ref s)) if s == "units" => {
            iter.next();
            if let Some(Token::Ident(ref s)) = iter.peek().cloned() {
                if s == "for" || s == "of" {
                    iter.next();
                }
            }
            return Query::UnitsFor(parse_eq(iter));
        }
        Some(Token::Ident(ref s)) if s == "search" => {
            iter.next();
            if let Some(Token::Ident(ref s)) = iter.peek().cloned() {
                return Query::Search(s.clone());
            }
        }
        _ => (),
    }
    let left = parse_eq(iter);
    match iter.peek().cloned().unwrap() {
        Token::DashArrow => {
            iter.next();
            let mut copy = iter.clone();
            if let Some(res) = parse_unitlist(&mut copy) {
                *iter = copy;
                return Query::Convert(left, Conversion::List(res), None, Digits::Default);
            }
            let digits = match iter.peek().cloned().unwrap() {
                Token::Ident(ref s) if s == "digits" => {
                    iter.next();
                    match iter.peek().cloned() {
                        Some(Token::Decimal(int, None, None)) => {
                            iter.next();
                            match u64::from_str_radix(&*int, 10) {
                                Ok(v) => Digits::Digits(v),
                                Err(e) => {
                                    return Query::Error(format!("Failed to parse digits: {}", e))
                                }
                            }
                        }
                        _ => Digits::FullInt,
                    }
                }
                Token::Ident(ref s) if s == "frac" || s == "fraction" || s == "ratio" => {
                    iter.next();
                    Digits::Fraction
                }
                Token::Ident(ref s) if s == "sci" || s == "scientific" => {
                    iter.next();
                    Digits::Scientific
                }
                Token::Ident(ref s) if s == "eng" || s == "engineering" => {
                    iter.next();
                    Digits::Engineering
                }
                _ => Digits::Default,
            };
            let base = match iter.peek().cloned().unwrap() {
                Token::Ident(ref s) if s == "base" => {
                    iter.next();
                    match iter.next() {
                        Some(Token::Decimal(int, None, None)) => {
                            match u64::from_str_radix(&*int, 10) {
                                Ok(v @ 2..=36) => Some(v as u8),
                                Ok(v) => {
                                    return Query::Error(format!(
                                        "Unsupported base {}, must be from 2 to 36",
                                        v
                                    ))
                                }
                                Err(e) => {
                                    return Query::Error(format!("Failed to parse base: {}", e))
                                }
                            }
                        }
                        Some(x) => {
                            return Query::Error(format!(
                                "Expected decimal base, got {}",
                                describe(&x)
                            ))
                        }
                        None => return Query::Error("Expected decimal base, got eof".to_string()),
                    }
                }
                Token::Ident(ref s) if s == "hex" || s == "hexadecimal" || s == "base16" => {
                    iter.next();
                    Some(16)
                }
                Token::Ident(ref s) if s == "oct" || s == "octal" || s == "base8" => {
                    iter.next();
                    Some(8)
                }
                Token::Ident(ref s) if s == "bin" || s == "binary" || s == "base2" => {
                    iter.next();
                    Some(2)
                }
                _ => None,
            };
            let right = match iter.peek().cloned().unwrap() {
                Token::Eof => Conversion::None,
                Token::Degree(deg) => Conversion::Degree(deg),
                Token::Plus | Token::Minus => {
                    let mut old = iter.clone();
                    if let Some(off) = parse_offset(iter) {
                        Conversion::Offset(off)
                    } else {
                        Conversion::Expr(parse_eq(&mut old))
                    }
                }
                Token::Ident(ref s) if is_valid_timezone(s) => Conversion::Timezone(
                    TimeZone::get(s).expect("Running TimeZone::lookup a second time failed"),
                ),
                _ => Conversion::Expr(parse_eq(iter)),
            };
            Query::Convert(left, right, base, digits)
        }
        _ => Query::Expr(left),
    }
}

fn is_valid_timezone(s: &String) -> bool {
    s != "GB" && TimeZone::get(s).is_ok()
}

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

    fn parse(input: &str) -> String {
        parse_expr(&mut TokenIterator::new(input).peekable()).to_string()
    }

    #[test]
    fn add_assoc() {
        assert_eq!(parse("a + b - c + d - e"), "((a + b) - c + d) - e");
    }

    #[test]
    fn sub_crash_regression() {
        assert_eq!(parse("-"), "-<error: Expected term, got eof>");
    }

    #[test]
    fn multiplication() {
        assert_eq!(parse("a⋅b"), parse("a*b"));
        assert_eq!(parse("a×b"), parse("a*b"));
    }

    #[test]
    fn division() {
        assert_eq!(parse("2|3"), parse("2/3"));
        assert_eq!(parse("2∕3"), parse("2/3"));
        assert_eq!(parse("2÷3"), parse("2/3"));
        assert_eq!(parse("2⁄3"), parse("2/3"));
    }

    #[test]
    fn exponents() {
        assert_eq!(parse("2¹³⁶²⁷⁹⁸⁴¹−1"), parse("2^136279841−1"));
        assert_eq!(parse("e³"), parse("e^3"));
        assert_eq!(parse("1m/s²"), parse("1m/s^2"));
        assert_eq!(parse("1kg*m²/s²"), parse("1kg*m^2/s^2"));
        assert_eq!(parse("1V/m²"), parse("1V/m^2"));
        assert_eq!(parse("x¹²³⁴⁵⁶⁷⁸⁹⁰"), parse("x^1234567890"));
        assert_eq!(parse("¹"), "<error: Expected term, got exponent>");
    }

    #[test]
    fn mul_assoc() {
        assert_eq!(
            parse("a b * c / d / e f * g h"),
            "((((a b) * c) / d) / e f) * (g h)"
        );
        assert_eq!(parse("a|b c / g e|f"), "((a / b) * c) / (g * (e / f))");
        assert_eq!(parse("a / b / c"), "(a / b) / c");
    }

    #[test]
    fn parse_extra_ops() {
        assert_eq!(parse("a b mod c d"), "a b mod c d");
        assert_eq!(parse("a b << c d"), "a b << c d");
        assert_eq!(parse("a b >> c d"), "a b >> c d");
        assert_eq!(parse("a b and c d"), "a b and c d");
        assert_eq!(parse("a b or c d"), "a b or c d");
        assert_eq!(parse("a b xor c d"), "a b xor c d");
        assert_eq!(parse("a / b c mod d e / f"), "((a / b c) mod d e) / f");
    }

    #[test]
    fn suffix_prec() {
        assert_eq!(parse("a b °C + x y °F"), "a b °C + x y °F");
        assert_eq!(parse("a b °C c"), "(a b °C) * c");
        assert_eq!(parse("a °C / x"), "a °C / x");
        assert_eq!(parse("a °C * x"), "(a °C) * x");
    }

    #[test]
    fn number_lex() {
        assert_eq!(
            parse("1e"),
            "<error: Expected term, got <Malformed number literal: No digits after exponent>>"
        );
        assert_eq!(
            parse("1."),
            "<error: Expected term, got <Malformed number literal: No digits after decimal point>>"
        );
    }

    #[test]
    fn mono_unit_list() {
        use crate::ast::*;
        match parse_query(&mut TokenIterator::new("foo -> bar").peekable()) {
            Query::Convert(_, Conversion::Expr(_), _, _) => (),
            x => panic!("Expected Convert(_, Expr(_), _), got {:?}", x),
        }
    }

    #[test]
    fn test_of() {
        assert_eq!(parse("foo of 1 abc def / 12"), "(foo of 1 abc def) / 12");
    }

    #[test]
    fn test_prefixed_currency() {
        assert_eq!(parse("$2.5"), "$ * 2.5");
        assert_eq!(parse("£3"), "£ * 3");
        assert_eq!(parse("$.01"), "$ * 0.01");
        assert_eq!(parse("$asdf"), "$asdf");
        assert_eq!(parse("C$"), "C$");
    }

    #[test]
    fn test_pow_prec() {
        assert_eq!(parse("-2^5"), "-(2^5)");
    }
}