/* Copyright (C) 2019  Akshay Oppiliappan <nerdypepper@tuta.io>
 * Refer to LICENCE for more information.
 * */

use std::collections::HashMap;

use crate::CONFIGURATION;

use crate::error::{
    CalcError,
    Math
};

#[derive(Debug, Copy, Clone, PartialEq)]
pub struct Operator {
    token: char,
    pub operation: fn(f64, f64) -> f64,
    pub precedence: u8,
    pub is_left_associative: bool,
}

impl Operator {
    fn token_from_op(token: char,
                     operation: fn(f64, f64) -> f64,
                     precedence: u8,
                     is_left_associative: bool) -> Token {
        Token::Operator(
            Operator {
                token,
                operation,
                precedence,
                is_left_associative
            }
        )
    }
    pub fn operate(self, x: f64, y: f64) -> Result<f64, CalcError> {
        if self.token == '/' && y == 0. {
            return Err(CalcError::Math(Math::DivideByZero))
        } else {
            Ok((self.operation)(x, y))
        }
    }
}

#[derive(Debug, Clone, PartialEq)]
pub struct Function {
    token: String,
    relation: fn(f64) -> f64,
}

impl Function {
    fn token_from_fn(token: String, relation: fn(f64) -> f64) -> Token {
        Token::Function(
            Function {
                token,
                relation
            }
        )
    }
    pub fn apply(self, arg: f64) -> Result<f64, CalcError> {
        let result = (self.relation)(arg);
        if !result.is_finite() {
            return Err(CalcError::Math(Math::OutOfBounds));
        } else {
            Ok(result)
        }
    }
}


#[derive(Debug, Clone, PartialEq)]
pub enum Token {
    Operator(Operator),
    Num(f64),
    Function(Function),
    LParen,
    RParen
}

fn get_functions() -> HashMap<&'static str, Token> {
    return [
        ("sin",   Function::token_from_fn("sin".into(),   |x| is_radian_mode(x, CONFIGURATION.radian_mode).sin())),
        ("cos",   Function::token_from_fn("cos".into(),   |x| is_radian_mode(x, CONFIGURATION.radian_mode).cos())),
        ("tan",   Function::token_from_fn("tan".into(),   |x| is_radian_mode(x, CONFIGURATION.radian_mode).tan())),
        ("csc",   Function::token_from_fn("csc".into(),   |x| is_radian_mode(x, CONFIGURATION.radian_mode).sin().recip())),
        ("sec",   Function::token_from_fn("sec".into(),   |x| is_radian_mode(x, CONFIGURATION.radian_mode).cos().recip())),
        ("cot",   Function::token_from_fn("cot".into(),   |x| is_radian_mode(x, CONFIGURATION.radian_mode).tan().recip())),
        ("sinh",  Function::token_from_fn("sinh".into(),  |x| x.sinh())),
        ("cosh",  Function::token_from_fn("cosh".into(),  |x| x.cosh())),
        ("tanh",  Function::token_from_fn("tanh".into(),  |x| x.tanh())),
        ("ln",    Function::token_from_fn("ln".into(),    |x| x.ln())),
        ("log",   Function::token_from_fn("log".into(),   |x| x.log10())),
        ("sqrt",  Function::token_from_fn("sqrt".into(),  |x| x.sqrt())),
        ("ceil",  Function::token_from_fn("ceil".into(),  |x| x.ceil())),
        ("floor", Function::token_from_fn("floor".into(), |x| x.floor())),
        ("rad",   Function::token_from_fn("rad".into(),   |x| x.to_radians())),
        ("deg",   Function::token_from_fn("deg".into(),   |x| x.to_degrees())),
        ("abs",   Function::token_from_fn("abs".into(),   |x| x.abs())),
        ("asin",  Function::token_from_fn("asin".into(),  |x| x.asin())),
        ("acos",  Function::token_from_fn("acos".into(),  |x| x.acos())),
        ("atan",  Function::token_from_fn("atan".into(),  |x| x.atan())),
        ("acsc",  Function::token_from_fn("acsc".into(),  |x| (1./x).asin())),
        ("asec",  Function::token_from_fn("asec".into(),  |x| (1./x).acos())),
        ("acot",  Function::token_from_fn("acot".into(),  |x| (1./x).atan())),
        // single arg function s can be added here
    ].iter().cloned().collect();
}

pub fn get_operators() -> HashMap<char, Token> {
    return [
        ('+', Operator::token_from_op('+', |x, y| x + y, 2, true)),
        ('-', Operator::token_from_op('-', |x, y| x - y, 2, true)),
        ('*', Operator::token_from_op('*', |x, y| x * y, 3, true)),
        ('/', Operator::token_from_op('/', |x, y| x / y, 3, true)),
        ('%', Operator::token_from_op('%', |x, y| x % y, 3, true)),
        ('^', Operator::token_from_op('^', |x, y| x.powf(y) , 4, false)),
        ('!', Operator::token_from_op('!', |x, _| factorial(x) , 4, true)),
    ].iter().cloned().collect();
}

fn factorial (n: f64) -> f64 {
        n.signum() * (1.. n.abs() as u64 +1).fold(1, |p, n| p*n) as f64
}

pub fn lexer(input: &str, prev_ans: &mut f64) -> Result<Vec<Token>, CalcError> {
    let functions: HashMap<&str, Token> = get_functions();
    let operators: HashMap<char, Token> = get_operators();

    let mut num_vec: String = String::new();
    let mut char_vec: String = String::new();
    let mut result: Vec<Token> = vec![];
    let mut last_char_is_op = true;

    for letter in input.chars() {
        match letter {
            '0'...'9' | '.' => {
                if char_vec.len() > 0 {
                    if let Some(_) = functions.get(&char_vec[..]) {
                        return Err(CalcError::Syntax(format!("Function '{}' expected parentheses", char_vec)))
                    } else {
                        return Err(CalcError::Syntax(format!("Unexpected character '{}'", char_vec)))
                    }
                }
                num_vec.push(letter);
                last_char_is_op = false; 
            },
            '_' => {
                if char_vec.len() > 0 {
                    if let Some(_) = functions.get(&char_vec[..]) {
                        return Err(CalcError::Syntax(format!("Function '{}' expected parentheses", char_vec)))
                    } else {
                        return Err(CalcError::Syntax(format!("Unexpected character '{}'", char_vec)))
                    }
                }
                let parse_num = num_vec.parse::<f64>().ok();
                if let Some(x) = parse_num {
                    result.push(Token::Num(x));
                    result.push(operators.get(&'*').unwrap().clone());
                    num_vec.clear();
                }
                last_char_is_op = false; 
                result.push(Token::Num(*prev_ans));
            }
            'a'...'z' | 'A'...'Z' => {
                let parse_num = num_vec.parse::<f64>().ok();
                if let Some(x) = parse_num {
                    result.push(Token::Num(x));
                    result.push(operators.get(&'*').unwrap().clone());
                    num_vec.clear();
                }
                char_vec.push(letter);
                last_char_is_op = false;
            },
            '+' | '-' => {
                let op_token = operators.get(&letter).unwrap().clone();
                let parse_num = num_vec.parse::<f64>().ok();
                if !last_char_is_op {
                    if let Some(x) = parse_num {
                        result.push(Token::Num(x));
                        num_vec.clear();
                        last_char_is_op = true;
                    }
                    result.push(op_token);
                } else if last_char_is_op {
                    result.push(Token::LParen);
                    result.push(Token::Num((letter.to_string() + "1").parse::<f64>().unwrap()));
                    result.push(Token::RParen);
                    result.push(Operator::token_from_op('*', |x, y| x * y, 10, true));
                }
            },
            '/' | '*' | '%' | '^' | '!' => {
                drain_num_stack(&mut num_vec, &mut result);
                let operator_token: Token = operators.get(&letter).unwrap().clone();
                result.push(operator_token);
                last_char_is_op = true; 
                if letter == '!' {
                    result.push(Token::Num(1.));
                    last_char_is_op = false; 
                }
            },
            '('  => {
                if char_vec.len() > 0 {
                    if let Some(res) = functions.get(&char_vec[..]) {
                        result.push(res.clone());
                    } else {
                        return Err(CalcError::Syntax(format!("Unknown function '{}'", char_vec)))
                    }
                    char_vec.clear();
                } else {
                    let parse_num = num_vec.parse::<f64>().ok();
                    if let Some(x) = parse_num {
                        result.push(Token::Num(x));
                        result.push(operators.get(&'*').unwrap().clone());
                        num_vec.clear();
                    }
                }

                if let Some(x) = result.last() {
                    match x {
                        Token::RParen => {
                            result.push(operators.get(&'*').unwrap().clone());
                        },
                        _ => {}
                    };
                }
                result.push(Token::LParen);
                last_char_is_op = true; 
            },
            ')' => {
                drain_num_stack(&mut num_vec, &mut result);
                result.push(Token::RParen);
                last_char_is_op = false;  
            },
            ' ' => {},
            _ => {
                return Err(CalcError::Syntax(format!("Unexpected token: '{}'", letter)))
            }
        }
    }
    // println!("{:?}", result);
    drain_num_stack(&mut num_vec, &mut result);
    Ok(result)
}

fn drain_num_stack(num_vec: &mut String, result: &mut Vec<Token>) {
    let parse_num = num_vec.parse::<f64>().ok();
    if let Some(x) = parse_num {
        result.push(Token::Num(x));
        num_vec.clear();
    }
}

fn is_radian_mode(x: f64, is_radian: bool) -> f64 {
    if is_radian {
        return x
    } else {
        x.to_radians()
    }
}