path: root/rust/src/core.rs

#![allow(dead_code)]

use std::collections::HashMap;
use std::fs::File;
use std::io::prelude::*;
use std::num::ToPrimitive;
use time;

use types::{MalVal,MalRet,err_val,err_str,err_string,
            _nil,_true,_false,_int,string,
            list,vector,listm,vectorm,hash_mapm,func,funcm,malfuncd};
use types::MalType::{Nil, Int, Strn, List, Vector, Hash_Map, Func, MalFunc, Atom};
use types;
use readline;
use reader;
use printer;

// General functions
fn equal_q(a: Vec<MalVal>) -> MalRet {
    if a.len() != 2 {
        return err_str("Wrong arity to equal? call");
    }
    if a[0] == a[1] {Ok(_true())} else {Ok(_false())}
}

// Errors/Exceptions
fn throw(a: Vec<MalVal>) -> MalRet {
    if a.len() != 1 {
        return err_str("Wrong arity to throw call");
    }
    err_val(a[0].clone())
}

// String routines
fn pr_str(a: Vec<MalVal>) -> MalRet {
    Ok(string(printer::pr_list(&a, true, "", "", " ")))
}

fn str(a: Vec<MalVal>) -> MalRet {
    Ok(string(printer::pr_list(&a, false, "", "", "")))
}

fn prn(a: Vec<MalVal>) -> MalRet {
    println!("{}", printer::pr_list(&a, true, "", "", " "));
    Ok(_nil())
}

fn println(a: Vec<MalVal>) -> MalRet {
    println!("{}", printer::pr_list(&a, false, "", "", " "));
    Ok(_nil())
}

fn readline(a: Vec<MalVal>) -> MalRet {
    match *a[0] {
        Strn(ref a0) => match readline::mal_readline(&a0) {
            Some(line) => Ok(string(line)),
            None       => err_val(_nil()),
        },
        _ => err_str("read_string called with non-string"),
    }
}

fn read_string(a: Vec<MalVal>) -> MalRet {
    match *a[0] {
        Strn(ref a0) => reader::read_str(a0.to_string()),
        _ => err_str("read_string called with non-string"),
    }
}

fn slurp(a: Vec<MalVal>) -> MalRet {
    match *a[0] {
        Strn(ref a0) => {
            let mut s = String::new();
            match File::open(a0).and_then(|mut f| f.read_to_string(&mut s)) {
                Ok(_) => Ok(string(s)),
                Err(e) => err_string(e.to_string()),
            }
        },
        _ => err_str("slurp called with non-string"),
    }
}


// Numeric functions
fn int_op<F>(f: F, a: Vec<MalVal>) -> MalRet
    where F: FnOnce(isize, isize) -> isize
{
    match *a[0] {
        Int(a0) => match *a[1] {
            Int(a1) => Ok(_int(f(a0,a1))),
            _ => err_str("second arg must be an int"),
        },
        _ => err_str("first arg must be an int"),
    }
}

fn bool_op<F>(f: F, a: Vec<MalVal>) -> MalRet
    where F: FnOnce(isize, isize) -> bool
{
    match *a[0] {
        Int(a0) => match *a[1] {
            Int(a1) => {
                match f(a0,a1) {
                    true => Ok(_true()),
                    false => Ok(_false()),
                }
            },
            _ => err_str("second arg must be an int"),
        },
        _ => err_str("first arg must be an int"),
    }
}

pub fn add(a: Vec<MalVal>) -> MalRet { int_op(|i,j| { i+j }, a) }
pub fn sub(a: Vec<MalVal>) -> MalRet { int_op(|i,j| { i-j }, a) }
pub fn mul(a: Vec<MalVal>) -> MalRet { int_op(|i,j| { i*j }, a) }
pub fn div(a: Vec<MalVal>) -> MalRet { int_op(|i,j| { i/j }, a) }

pub fn lt (a: Vec<MalVal>) -> MalRet { bool_op(|i,j| { i<j }, a) }
pub fn lte(a: Vec<MalVal>) -> MalRet { bool_op(|i,j| { i<=j }, a) }
pub fn gt (a: Vec<MalVal>) -> MalRet { bool_op(|i,j| { i>j }, a) }
pub fn gte(a: Vec<MalVal>) -> MalRet { bool_op(|i,j| { i>=j }, a) }

pub fn time_ms(_a: Vec<MalVal>) -> MalRet {
    //let x = time::now();
    let now = time::get_time();
    let now_ms = (now.sec * 1000).to_isize().unwrap() +
                 (now.nsec.to_isize().unwrap() / 1000000);
    Ok(_int(now_ms))
}


// Hash Map functions
pub fn assoc(a: Vec<MalVal>) -> MalRet {
    if a.len() < 3 {
        return err_str("Wrong arity to assoc call");
    }
    match *a[0] {
        Hash_Map(ref hm,_) => types::_assoc(hm, a[1..].to_vec()),
        Nil => types::hash_mapv(a[1..].to_vec()),
        _ => err_str("assoc onto non-hash map"),
    }
}

pub fn dissoc(a: Vec<MalVal>) -> MalRet {
    if a.len() < 2 {
        return err_str("Wrong arity to dissoc call");
    }
    match *a[0] {
        Hash_Map(ref hm,_) => types::_dissoc(hm, a[1..].to_vec()),
        Nil => Ok(_nil()),
        _ => err_str("dissoc onto non-hash map"),
    }
}

pub fn get(a: Vec<MalVal>) -> MalRet {
    if a.len() != 2 {
        return err_str("Wrong arity to get call");
    }
    let hm = match *a[0] {
        Hash_Map(ref hm,_) => hm,
        Nil => return Ok(_nil()),
        _ => return err_str("get on non-hash map"),
    };
    match *a[1] {
        Strn(ref key) => {
            match hm.get(key) {
                Some(v) => Ok(v.clone()),
                None    => Ok(_nil()),
            }
        },
        _ => err_str("get with non-string key"),
    }
}

pub fn contains_q(a: Vec<MalVal>) -> MalRet {
    if a.len() != 2 {
        return err_str("Wrong arity to contains? call");
    }
    let hm = match *a[0] {
        Hash_Map(ref hm,_) => hm,
        Nil => return Ok(_false()),
        _ => return err_str("contains? on non-hash map"),
    };
    match *a[1] {
        Strn(ref key) => {
            match hm.contains_key(key) {
                true  => Ok(_true()),
                false => Ok(_false()),
            }
        }
        _ => err_str("contains? with non-string key"),
    }
}

pub fn keys(a: Vec<MalVal>) -> MalRet {
    if a.len() != 1 {
        return err_str("Wrong arity to keys call");
    }
    let hm = match *a[0] {
        Hash_Map(ref hm,_) => hm,
        Nil => return Ok(_nil()),
        _ => return err_str("contains? on non-hash map"),
    };
    Ok(list(hm.keys().map(|s| string(s.to_string())).collect()))
}

pub fn vals(a: Vec<MalVal>) -> MalRet {
    if a.len() != 1 {
        return err_str("Wrong arity to values call");
    }
    let hm = match *a[0] {
        Hash_Map(ref hm,_) => hm,
        Nil => return Ok(_nil()),
        _ => return err_str("contains? on non-hash map"),
    };
    Ok(list(hm.values().map(|s| s.clone()).collect()))
}

// Sequence functions
pub fn cons(a: Vec<MalVal>) -> MalRet {
    match *a[1] {
        List(ref v,_) | Vector(ref v,_) => {
            let mut new_v = v.clone();
            new_v.insert(0, a[0].clone());
            Ok(list(new_v))
        },
        _ => err_str("Second arg to cons not a sequence"),
    }
}

pub fn concat(a: Vec<MalVal>) -> MalRet {
    let mut new_v: Vec<MalVal> = vec![];
    for lst in a.iter() {
        match **lst {
            List(ref l,_) | Vector(ref l,_) => new_v.push_all(l),
            _ => return err_str("concat called with non-sequence"),
        }
    }
    Ok(list(new_v))
}

pub fn nth(a: Vec<MalVal>) -> MalRet {
    if a.len() != 2 {
        return err_str("Wrong arity to nth call");
    }
    let seq = match *a[0] {
        List(ref v,_) | Vector(ref v,_) => v,
        _ => return err_str("nth called with non-sequence"),
    };
    let idx = match *a[1] {
        Int(i) => {
            match i.to_usize() {
                Some(ui) => ui,
                None => return Ok(_nil()),
            }
        },
        _ => return err_str("nth called with non-integer index"),
    };
    if idx >= seq.len() {
        err_str("nth: index out of range")
    } else {
        Ok(seq[idx].clone())
    }
}

pub fn first(a: Vec<MalVal>) -> MalRet {
    if a.len() != 1 {
        return err_str("Wrong arity to first call");
    }
    let seq = match *a[0] {
        List(ref v,_) | Vector(ref v,_) => v,
        _ => return err_str("first called with non-sequence"),
    };
    if seq.len() == 0 {
        Ok(_nil())
    } else {
        Ok(seq[0].clone())
    }
}

pub fn rest(a: Vec<MalVal>) -> MalRet {
    if a.len() != 1 {
        return err_str("Wrong arity to rest call");
    }
    let seq = match *a[0] {
        List(ref v,_) | Vector(ref v,_) => v,
        _ => return err_str("rest called with non-sequence"),
    };
    if seq.len() == 0 {
        Ok(list(vec![]))
    } else {
        Ok(list(seq[1..].to_vec()))
    }
}

pub fn empty_q(a: Vec<MalVal>) -> MalRet {
    if a.len() != 1 {
        return err_str("Wrong arity to empty? call");
    }
    match *a[0] {
        List(ref v,_) | Vector(ref v,_) => {
            match v.len() {
                0 => Ok(_true()),
                _ => Ok(_false()),
            }
        },
        _ => err_str("empty? called on non-sequence"),
    }
}

pub fn count(a: Vec<MalVal>) -> MalRet {
    if a.len() != 1 {
        return err_str("Wrong arity to count call");
    }
    match *a[0] {
        List(ref v,_) | Vector(ref v,_) => Ok(_int(v.len().to_isize().unwrap())),
        Nil => Ok(_int(0)),
        _ => err_str("count called on non-sequence"),
    }
}

pub fn apply(a: Vec<MalVal>) -> MalRet {
    if a.len() < 2 {
        return err_str("apply call needs 2 or more arguments");
    }
    let ref f = a[0];
    let mut args = a[1..a.len()-1].to_vec();
    match *a[a.len()-1] {
        List(ref v, _) | Vector(ref v, _) => {
            args.push_all(v);
            f.apply(args)
        },
        _ => err_str("apply call with non-sequence"),
    }
}

pub fn map(a: Vec<MalVal>) -> MalRet {
    if a.len() != 2 {
        return err_str("Wrong arity to map call");
    }
    let mut results:Vec<MalVal> = vec![];
    match *a[1] {
        List(ref v,_) | Vector(ref v,_) => {
            for mv in v.iter() {
                let res = try!(a[0].apply(vec![mv.clone()]));
                results.push(res);
            }
        },
        _ => return err_str("map call with non-sequence"),
    }
    Ok(list(results))
}

pub fn conj(a: Vec<MalVal>) -> MalRet {
    if a.len() < 2 {
        return err_str("Wrong arity to conj call");
    }
    let mut new_v: Vec<MalVal> = vec![];
    match *a[0] {
        List(ref l,_) => {
            new_v.push_all(l);
            for mv in a.iter().skip(1) {
                new_v.insert(0, mv.clone());
            }
            Ok(list(new_v))
        }
        Vector(ref l,_) => {
            new_v.push_all(l);
            for mv in a.iter().skip(1) {
                new_v.push(mv.clone());
            }
            Ok(vector(new_v))
        }
        _ => err_str("conj called with non-sequence"),
    }
}


// Metadata functions
fn with_meta(a: Vec<MalVal>) -> MalRet {
    if a.len() != 2 {
        return err_str("Wrong arity to with-meta call");
    }
    let meta = a[1].clone();
    match *a[0] {
        List(ref v,_) => Ok(listm(v.clone(), meta)),
        Vector(ref v,_) => Ok(vectorm(v.clone(), meta)),
        Hash_Map(ref hm,_) => Ok(hash_mapm(hm.clone(), meta)),
        MalFunc(ref mfd,_) => Ok(malfuncd(mfd.clone(), meta)),
        Func(f,_) => Ok(funcm(f, meta)),
        _ => err_str("type does not support metadata"),
    }
}

fn meta(a: Vec<MalVal>) -> MalRet {
    if a.len() != 1 {
        return err_str("Wrong arity to meta call");
    }
    match *a[0] {
        List(_,ref meta) |
        Vector(_,ref meta) |
        Hash_Map(_,ref meta) |
        MalFunc(_,ref meta) |
        Func(_,ref meta) => Ok(meta.clone()),
        _ => err_str("type does not support metadata"),
    }
}

// Atom functions
fn deref(a: Vec<MalVal>) -> MalRet {
    if a.len() != 1 {
        return err_str("Wrong arity to deref call");
    }
    match *a[0] {
        Atom(ref val) => Ok(val.borrow().clone()),
        _ => err_str("deref called on non-atom"),
    }
}

fn reset_bang(a: Vec<MalVal>) -> MalRet {
    if a.len() != 2 {
        return err_str("Wrong arity to map call");
    }
    match *a[0] {
        Atom(ref val) => {
            let mut val_cell = val.borrow_mut();
            *val_cell = a[1].clone();
            Ok(a[1].clone())
        },
        _ => err_str("reset! called on non-atom"),
    }
}

fn swap_bang(a: Vec<MalVal>) -> MalRet {
    if a.len() < 2 {
        return err_str("Wrong arity to swap_q call");
    }
    let f = a[1].clone();
    match *a[0] {
        Atom(ref val) => {
            let mut val_cell = val.borrow_mut();
            let mut args = a[2..].to_vec();
            args.insert(0, val_cell.clone());
            *val_cell = try!(f.apply(args));
            Ok(val_cell.clone())
        },
        _ => err_str("swap! called on non-atom"),
    }
}


pub fn ns() -> HashMap<String,MalVal> {
    let mut ns = HashMap::new();;

    ns.insert("=".to_string(), func(equal_q));
    ns.insert("throw".to_string(), func(throw));
    ns.insert("nil?".to_string(), func(types::nil_q));
    ns.insert("true?".to_string(), func(types::true_q));
    ns.insert("false?".to_string(), func(types::false_q));
    ns.insert("symbol".to_string(), func(types::_symbol));
    ns.insert("symbol?".to_string(), func(types::symbol_q));
    ns.insert("keyword".to_string(), func(types::_keyword));
    ns.insert("keyword?".to_string(), func(types::keyword_q));

    ns.insert("pr-str".to_string(), func(pr_str));
    ns.insert("str".to_string(), func(str));
    ns.insert("prn".to_string(), func(prn));
    ns.insert("println".to_string(), func(println));
    ns.insert("readline".to_string(), func(readline));
    ns.insert("read-string".to_string(), func(read_string));
    ns.insert("slurp".to_string(), func(slurp));

    ns.insert("<".to_string(),  func(lt));
    ns.insert("<=".to_string(), func(lte));
    ns.insert(">".to_string(),  func(gt));
    ns.insert(">=".to_string(), func(gte));
    ns.insert("+".to_string(), func(add));
    ns.insert("-".to_string(), func(sub));
    ns.insert("*".to_string(), func(mul));
    ns.insert("/".to_string(), func(div));
    ns.insert("time-ms".to_string(), func(time_ms));

    ns.insert("list".to_string(), func(types::listv));
    ns.insert("list?".to_string(), func(types::list_q));
    ns.insert("vector".to_string(), func(types::vectorv));
    ns.insert("vector?".to_string(), func(types::vector_q));
    ns.insert("hash-map".to_string(), func(types::hash_mapv));
    ns.insert("map?".to_string(), func(types::hash_map_q));
    ns.insert("assoc".to_string(), func(assoc));
    ns.insert("dissoc".to_string(), func(dissoc));
    ns.insert("get".to_string(), func(get));
    ns.insert("contains?".to_string(), func(contains_q));
    ns.insert("keys".to_string(), func(keys));
    ns.insert("vals".to_string(), func(vals));

    ns.insert("sequential?".to_string(), func(types::sequential_q));
    ns.insert("cons".to_string(), func(cons));
    ns.insert("concat".to_string(), func(concat));
    ns.insert("empty?".to_string(), func(empty_q));
    ns.insert("nth".to_string(), func(nth));
    ns.insert("first".to_string(), func(first));
    ns.insert("rest".to_string(), func(rest));
    ns.insert("count".to_string(), func(count));
    ns.insert("apply".to_string(), func(apply));
    ns.insert("map".to_string(), func(map));
    ns.insert("conj".to_string(), func(conj));

    ns.insert("with-meta".to_string(), func(with_meta));
    ns.insert("meta".to_string(), func(meta));
    ns.insert("atom".to_string(), func(types::atom));
    ns.insert("atom?".to_string(), func(types::atom_q));
    ns.insert("deref".to_string(), func(deref));
    ns.insert("reset!".to_string(), func(reset_bang));
    ns.insert("swap!".to_string(), func(swap_bang));

    return ns;
}