/* Flesh: Flexible Shell
 * Copyright (C) 2021 Ava Hahn
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

use crate::eval::eval;
use crate::error::{Traceback, start_trace};
use crate::segment::{Ctr, Seg, Type};
use crate::stdlib::{CONSOLE_XDIM_VNAME, FLESH_DEFAULT_CONS_WIDTH};
use crate::sym::{SymTable, Symbol, UserFn, ValueType, Args};
use std::env;
use std::rc::Rc;

const QUOTE_DOCSTRING: &str = "takes a single unevaluated tree and returns it as it is: unevaluated.";
fn quote_callback(ast: &Seg, _syms: &mut SymTable) -> Result<Ctr, Traceback> {
    if ast.len() > 1 {
        Err(start_trace(("quote", "do not quote more than one thing at a time").into()))
    } else {
        Ok(*ast.car.clone())
    }
}

const EVAL_DOCSTRING: &str = "takes an unevaluated argument and evaluates it.
Specifically, does one pass of the tree simplification algorithm.
If you have a variable referencing another variable you will get the
referenced variable.";
fn eval_callback(ast: &Seg, syms: &mut SymTable) -> Result<Ctr, Traceback> {
    if ast.len() > 1 {
        Err(start_trace(
            ("eval", "do not eval more than one thing at a time")
                .into()))
    } else {
        match *ast.car {
            Ctr::Seg(ref s) => {
                match eval(s, syms) {
                    Err(e) => Err(e.with_trace(
                        ("eval", "evaluation failure")
                            .into())),
                    Ok(s) => if let Ctr::Seg(ref inner) = *s {
                        match eval(inner, syms) {
                            Err(e) => Err(e.with_trace(
                                ("eval", "evaluation failure")
                                    .into())),
                            Ok(s) => Ok(*s),
                        }
                    } else {
                        Ok(*s)
                    },
                }
            }

            Ctr::Symbol(ref sym) => {
                let intermediate = syms.call_symbol(sym, &Seg::new(), true);
                if let Err(e) = intermediate {
                    return Err(e.with_trace(
                        ("eval", "evaluation failure")
                            .into()))
                }
                let res = *intermediate?;
                if let Ctr::Seg(ref s) = res {
                    match eval(s, syms) {
                        Err(e) => Err(e.with_trace(
                            ("eval", "evaluation failure")
                                .into())),
                        Ok(s) => Ok(*s),
                    }
                } else {
                    Ok(res)
                }
            },
            _ => Ok(*ast.car.clone())
        }
    }
}

const HELP_DOCSTRING: &str = "prints help text for a given symbol. Expects only one argument.";
fn help_callback(ast: &Seg, syms: &mut SymTable) -> Result<Ctr, Traceback> {
    if ast.len() != 1 {
        return Err(start_trace(("help", "expected one input").into()));
    }
    if let Ctr::Symbol(ref symbol) = *ast.car {
        if let Some(sym) = syms.get(symbol) {
            let args_str: String;
            if let ValueType::VarForm(_) = sym.value {
                args_str = "(its a variable)".to_string();
            } else {
                args_str = sym.args.to_string();
            }
            println!(
                "NAME: {0}\n
ARGS: {1}\n
DOCUMENTATION:\n
{2}\n
CURRENT VALUE AND/OR BODY:
{3}",
                sym.name, args_str, sym.docs, sym.value
            );
        } else {
            return Err(start_trace(("help", format!("{symbol} is undefined")).into()));
        }
    } else {
        return Err(start_trace(("help", "expected input to be a symbol").into()));
    }

    Ok(Ctr::None)
}

const ISSET_DOCSTRING: &str = "accepts a single argument: a symbol.
returns true or false according to whether or not the symbol is found in the symbol table.";
fn isset_callback(ast: &Seg, syms: &mut SymTable) -> Result<Ctr, Traceback> {
    if ast.len() != 1 {
        Err(start_trace(("set?", "expcted one input").into()))
    } else if let Ctr::Symbol(ref symbol) = *ast.car {
        Ok(Ctr::Bool(syms.get(symbol).is_some()))
    } else {
        Err(start_trace(("set?", "expected argument to be a input").into()))
    }
}

const ENV_DOCSTRING: &str = "takes no arguments
prints out all available symbols and their associated values";
fn env_callback(_ast: &Seg, syms: &mut SymTable) -> Result<Ctr, Traceback> {
    // get width of current output
    let xdim: i128;
    if let Ctr::Integer(dim) = *syms
        .call_symbol(&CONSOLE_XDIM_VNAME.to_string(), &Seg::new(), true)
        .unwrap_or_else(|_: Traceback| Box::new(Ctr::None)) {
            xdim = dim;
        } else {
            println!("{} contains non integer value, defaulting to {}",
                     CONSOLE_XDIM_VNAME, FLESH_DEFAULT_CONS_WIDTH);
            xdim = FLESH_DEFAULT_CONS_WIDTH as i128;
        }

    let mut v_col_len = 0;
    let mut f_col_len = 0;
    let mut functions = vec![];
    let mut variables = vec![];
    for (name, val) in syms.iter() {
        if let ValueType::VarForm(l) = &val.value {
            let token: String = match l.to_type() {
                Type::Lambda => format!("{}: <lambda>", name),
                Type::Seg => format!("{}: <form>", name),
                _ => format!("{}: {}", name, val.value),
            };

            if token.len() > v_col_len && token.len() < xdim as usize {
                v_col_len = token.len();
            }

            variables.push(token);
        } else {
            if f_col_len < name.len() && name.len() < xdim as usize {
                f_col_len = name.len();
            }
            functions.push(name.clone());
        }
    }

    let mut n_v_cols = xdim / v_col_len as i128;
    // now decrement to make sure theres room for two spaces of padding
    while n_v_cols > 1 && xdim % (v_col_len as i128) < (2 * n_v_cols) {
        n_v_cols -= 1;
    }
    // again for functions
    let mut n_f_cols = xdim / f_col_len as i128;
    while n_f_cols > 1 && xdim & (f_col_len as i128) < (2 * n_f_cols) {
        n_f_cols -= 1;
    }

    let mut col_iter = 0;
    println!("VARIABLES:");
    for var in variables {
        print!("{:v_col_len$}", var);
        col_iter += 1;
        if col_iter % n_v_cols == 0 {
            println!();
        } else {
            print!("  ");
        }
    }
    println!("\nFUNCTIONS:");
    col_iter = 0;
    for func in functions {
        print!("{:f_col_len$}", func);
        col_iter += 1;
        if col_iter % n_f_cols == 0 {
            println!();
        } else {
            print!("  ");
        }
    }
    Ok(Ctr::None)
}

const LAMBDA_DOCSTRING: &str = "Takes two arguments of any type.
No args are evaluated when lambda is called.
Lambda makes sure the first argument is a list of symbols (or 'arguments') to the lambda function.
The next arg is stored in a tree to evaluate on demand.

Example: (lambda (x y) (add x y))
This can then be evaluated like so:
((lambda (x y) (add x y)) 1 2)
which is functionally equivalent to:
(add 1 2)";
fn lambda_callback(
    ast: &Seg,
    _syms: &mut SymTable
) -> Result<Ctr, Traceback> {
    let mut args = vec![];
    if let Ctr::Seg(ref arg_head) = *ast.car {
        if !arg_head.circuit(&mut |arg: &Ctr| -> bool {
            if let Ctr::Symbol(ref s) = *arg {
                args.push(s.clone());
                true
            } else if let Ctr::None = *arg {
                // no args case
                true
            } else {
                false
            }
        }) {
            Err(start_trace(("lambda", "lambda inputs should all be symbols").into()))
        } else if let Ctr::Seg(ref eval_head) = *ast.cdr {
            if let Ctr::Seg(_) = *eval_head.car {
                Ok(Ctr::Lambda(UserFn{
                    ast: Box::new(eval_head.clone()),
                    arg_syms: args,
                }))
            } else {
                Err(start_trace(("lambda", "expected list of forms for lambda body").into()))
            }
        } else {
            Err(start_trace(("lambda", "not enough args").into()))
        }
    } else {
        Err(start_trace(("lambda", "expected list of lambda inputs").into()))
    }
}

const GETDOC_DOCSTRING: &str = "accepts an unevaluated symbol, returns the doc string.
Returns an error if symbol is undefined.

Note: make sure to quote the input like this:
    (get-doc (quote symbol-name))";
fn getdoc_callback(ast: &Seg, syms: &mut SymTable) -> Result<Ctr, Traceback> {
    if let Ctr::Symbol(ref symbol) = *ast.car {
        if let Some(sym) = syms.get(symbol) {
            Ok(Ctr::String(sym.docs.clone()))
        } else {
            Err(start_trace(("get-doc", "input is undefined").into()))
        }
    } else {
        Err(start_trace(("get-doc", "expected input to be a symbol").into()))
    }
}

const SETDOC_DOCSTRING: &str = "accepts a symbol and a doc string.
Returns an error if symbol is undefined, otherwise sets the symbols docstring to the argument.

Note: make sure to quote the input like this:
    (set-doc (quote symbol-name) my-new-docs)";
fn setdoc_callback(ast: &Seg, syms: &mut SymTable) -> Result<Ctr, Traceback> {
    if ast.len() != 2 {
        Err(start_trace(
            ("set-doc", "expected two inputs")
                .into()))
    } else if let Ctr::Symbol(ref symbol) = *ast.car {
        if let Some(mut sym) = syms.remove(symbol) {
            if let Ctr::Seg(ref doc_node) = *ast.cdr {
                if let Ctr::String(ref doc) = *doc_node.car {
                    sym.docs = doc.clone();
                    syms.insert(sym.name.clone(), sym);
                    Ok(Ctr::None)
                } else {
                    syms.insert(sym.name.clone(), sym);
                    Err(start_trace(
                        ("set-doc", "expected second input to be a string")
                            .into()))
                }
            } else {
                Err(start_trace(
                    ("set-doc", "missing second input somehow")
                        .into()))
            }
        } else {
            Err(start_trace(
                ("set-doc", format!("{symbol} is undefined"))
                    .into()))
        }
    } else {
        Err(start_trace(
            ("set-doc", "first input must be a symbol")
                .into()))
    }
}

const STORE_DOCSTRING: &str = "allows user to define functions and variables.
  A call may take one of three forms:
  1. variable declaration:
     Takes a name, doc string, and a value.
     (def myvar 'my special variable' 'my var value')
  2. function declaration:
     Takes a name, doc string, list of arguments, and one or more bodies to evaluate.
     Result of evaluating the final body is returned.
     (def myfunc 'does a thing' (myarg1 myarg2) (dothing myarg1 myarg2) (add myarg1 myarg2))
  3. symbol un-definition:
     Takes just a name. Removes variable from table.
     (def useless-var)

Additionally, passing a tree as a name will trigger def to evaluate the tree and try to derive
a value from it. If it does not return a ";
fn store_callback(ast: &Seg, syms: &mut SymTable, env_cfg: bool) -> Result<Ctr, Traceback> {
    let is_var = ast.len() == 3;
    let name: String;
    let docs: String;

    match *ast.car {
        Ctr::String(ref s) => name = s.clone(),
        Ctr::Symbol(ref s) => name = s.clone(),
        Ctr::Seg(ref s) => match eval(s, syms) {
            Err(e) => return Err(e.with_trace(("def", "failed to evaluate symbol name").into())),
            Ok(s) => match *s {
                Ctr::String(ref s) => name = s.clone(),
                Ctr::Symbol(ref s) => name = s.clone(),
                _ => {
                    return Err(start_trace(
                        ("def", "expected symbol name input to evaluate to a symbol or a string")
                            .into()));
                },
            }
        },
        _ => return Err(start_trace(
            ("def", "expected a string or a symbol as input for symbol name")
                .into()))
    }

    // remove var case
    if ast.len() == 1 {
        syms.remove(&name);
        if env_cfg {
            env::remove_var(name);
        }

        return Ok(Ctr::None)
    } else if ast.len() < 3 || ast.len() > 4 {
        return Err(start_trace(("def", "expected 3 or 4 inputs").into()))
    }

    let mut iter: &Seg;
    if let Ctr::Seg(ref s) = *ast.cdr {
        iter = s;
    } else {
        return Err(start_trace(("def", "not enough inputs").into()))
    }

    match *iter.car {
        Ctr::String(ref s) => docs = s.clone(),
        Ctr::Symbol(ref s) => match syms.call_symbol(s, &Seg::new(), true) {
            Ok(d) => if let Ctr::String(doc) = *d {
                docs = doc;
            } else {
                return Err(start_trace(("def", "expected docs input to evaluate to a string").into()))
            },

            Err(e) => return Err(e.with_trace(
                ("def", "couldnt evaluate docs form")
                    .into()))
        },
        _ => return Err(start_trace(
            ("def", "expected docs input to at least evaluate to a string if not be one")
                .into()))
    }

    if let Ctr::Seg(ref s) = *iter.cdr {
        iter = s;
    } else {
        return Err(start_trace(("def", "not enough inputs").into()))
    }

    let mut outer_scope_val: Seg = Seg::new();
    let noseg = Seg::new(); // similarly, rust shouldnt need this either
    let mut args = &noseg;
    let mut var_val_form: &Seg = &outer_scope_val;
    let mut expand = false;
    match *iter.car {
        Ctr::Seg(ref s) if !is_var => args = s,
        Ctr::Seg(ref s) if is_var => var_val_form = s,
        _ if is_var => {
            expand = true;
            outer_scope_val = Seg::from_mono(Box::new(*iter.car.clone()));
            var_val_form = &outer_scope_val;
        },
        _ if !is_var => return Err(start_trace(("def", "expected a list of inputs").into())),
        _ => unimplemented!(), // rustc is haunted and cursed
    }

    if is_var {
        let var_eval_result = eval(var_val_form, syms);
        if let Err(e) = var_eval_result {
            return Err(e.with_trace(
                ("def", format!("couldnt evaluate {var_val_form}"))
                    .into()))
        }
        let var_eval_final = *var_eval_result?;
        let var_val: Ctr = match var_eval_final {
            Ctr::Seg(ref s) if expand => *s.car.clone(),
            Ctr::Seg(ref s) if !expand => Ctr::Seg(s.clone()),
            _ => var_eval_final,
        };

        let outer_seg = Seg::from_mono(Box::new(var_val.clone()));
        syms.insert(
            name.clone(),
            Symbol::from_ast(&name, &docs, &outer_seg, None),
        );
        if env_cfg {
            match var_val.to_type() {
                Type::Lambda => {},
                Type::Seg => {},
                _ => {
                    let mut s = var_val.to_string();
                    if let Ctr::String(tok) = var_val {
                        s = tok;
                    }
                    env::set_var(name.clone(), s);
                }
            }
        }
        return Ok(Ctr::None)
    }

    let mut arg_list = vec![];
    if !args.circuit(&mut |c: &Ctr| -> bool {
        if let Ctr::Symbol(s) = c {
            arg_list.push(s.clone());
            true
        } else if let Ctr::None = c {
            // no args case
            true
        } else {
            false
        }
    }) {
        return Err(start_trace(
            ("def", "all inputs to function must be of type symbol")
                .into()))
    }

    if let Ctr::Seg(ref eval_bodies) = *iter.cdr {
        syms.insert(
            name.clone(),
            Symbol::from_ast(
                &name, &docs,
                eval_bodies,
                Some(arg_list),
            ),
        );
        Ok(Ctr::None)
    } else {
        Err(start_trace(
            ("def", "expected one or more forms to evaluate in function body")
                .into()))
    }
}

pub fn add_decl_lib_static(syms: &mut SymTable) {
    syms.insert(
        "help".to_string(),
        Symbol {
            name: String::from("help"),
            args: Args::Strict(vec![Type::Symbol]),
            conditional_branches: true,
            docs: HELP_DOCSTRING.to_string(),
            value: ValueType::Internal(Rc::new(help_callback)),
            ..Default::default()
        },
    );

    syms.insert(
        "set?".to_string(),
        Symbol {
            name: String::from("set?"),
            args: Args::Strict(vec![Type::Symbol]),
            conditional_branches: true,
            docs: ISSET_DOCSTRING.to_string(),
            value: ValueType::Internal(Rc::new(isset_callback)),
            ..Default::default()
        },
    );

    syms.insert(
        "env".to_string(),
        Symbol {
            name: String::from("env"),
            args: Args::None,
            conditional_branches: false,
            docs: ENV_DOCSTRING.to_string(),
            value: ValueType::Internal(Rc::new(env_callback)),
            ..Default::default()
        },
    );

        syms.insert(
        "quote".to_string(),
        Symbol {
            name: String::from("quote"),
            args: Args::Lazy(1),
            conditional_branches: true,
            docs: QUOTE_DOCSTRING.to_string(),
            value: ValueType::Internal(Rc::new(quote_callback)),
            ..Default::default()
        },
    );

    syms.insert(
        "q".to_string(),
        Symbol {
            name: String::from("quote"),
            args: Args::Lazy(1),
            conditional_branches: true,
            docs: QUOTE_DOCSTRING.to_string(),
            value: ValueType::Internal(Rc::new(quote_callback)),
            ..Default::default()
        },
    );

    syms.insert(
        "eval".to_string(),
        Symbol {
            name: String::from("eval"),
            args: Args::Lazy(1),
            conditional_branches: true,
            docs: EVAL_DOCSTRING.to_string(),
            value: ValueType::Internal(Rc::new(eval_callback)),
            ..Default::default()
        },
    );

    syms.insert(
        "lambda".to_string(),
        Symbol {
            name: String::from("lambda"),
            args: Args::Lazy(2),
            conditional_branches: true,
            docs: LAMBDA_DOCSTRING.to_string(),
            value: ValueType::Internal(Rc::new(lambda_callback)),
            ..Default::default()
        }
    );

    syms.insert(
        "get-doc".to_string(),
        Symbol {
            name: String::from("get-doc"),
            args: Args::Strict(vec![Type::Symbol]),
            conditional_branches: false,
            docs: GETDOC_DOCSTRING.to_string(),
            value: ValueType::Internal(Rc::new(getdoc_callback)),
            ..Default::default()
        }
    );

    syms.insert(
        "set-doc".to_string(),
        Symbol {
            name: String::from("get-doc"),
            args: Args::Strict(vec![Type::Symbol, Type::String]),
            conditional_branches: false,
            docs: SETDOC_DOCSTRING.to_string(),
            value: ValueType::Internal(Rc::new(setdoc_callback)),
            ..Default::default()
        }
    );
}

pub fn add_decl_lib_dynamic(syms: &mut SymTable, env: bool) {
    syms.insert(
        "def".to_string(),
        Symbol {
            name: String::from("define"),
            args: Args::Infinite,
            conditional_branches: true,
            docs: STORE_DOCSTRING.to_string(),
            value: ValueType::Internal(Rc::new(
                move |ast: &Seg, syms: &mut SymTable| -> Result<Ctr, Traceback> {
                    store_callback(ast, syms, env)
                },
            )),
            ..Default::default()
        },
    );
}