Merge branch 'master' into issue_166_string_ops

[jackhill/mal.git] / es6 / core.js

import { _equal_Q, _clone, _list_Q, _sequential_Q,
         _keyword, _keyword_Q, _vector, _vector_Q,
         _hash_map, _hash_map_Q, _assoc_BANG, _dissoc_BANG,
         _symbol, _symbol_Q, Atom } from './types'
import { pr_str } from './printer'
import { readline } from './node_readline'
import { read_str } from './reader'

// Errors/Exceptions
function mal_throw(exc) { throw exc; }

// String functions
function slurp(f) {
    if (typeof require !== 'undefined') {
        return require('fs').readFileSync(f, 'utf-8')
    } else {
        var req = new XMLHttpRequest()
        req.open('GET', f, false)
        req.send()
        if (req.status == 200) {
            return req.responseText
        } else {
            throw new Error(`Failed to slurp file: ${f}`)
        }
    }
}

// Sequence functions
function nth(lst, idx) {
    if (idx < lst.length) { return lst[idx]; }
    else                  { throw new Error('nth: index out of range'); }
}

function conj(lst, ...args) {
    if (_list_Q(lst)) {
        return args.reverse().concat(lst)
    } else {
        return _vector(...lst.concat(args))
    }
}

function seq(obj) {
    if (_list_Q(obj)) {
        return obj.length > 0 ? obj : null
    } else if (_vector_Q(obj)) {
        return obj.length > 0 ? obj.slice(0) : null
    } else if (typeof obj === "string" && obj[0] !== '\u029e') {
        return obj.length > 0 ? obj.split('') : null
    } else if (obj === null) {
        return null
    } else {
        throw new Error('seq: called on non-sequence')
    }
}

// hash-map functions

function keys(hm) {
    // TODO: Array.from(hm.keys()) when supported
    let ks = []
    for (let k of hm.keys()) { ks.push(k) }
    return ks
}

function vals(hm) {
    // TODO: Array.from(hm.keys()) when supported
    let vs = []
    for (let v of hm.values()) { vs.push(v) }
    return vs
}

// Metadata functions
function with_meta(obj, m) {
    let new_obj = _clone(obj)
    new_obj.meta = m
    return new_obj
}

// core_ns is namespace of type functions
export const core_ns = new Map([
        ['=', _equal_Q],
        ['throw', mal_throw],

        ['nil?', a => a === null],
        ['true?', a => a === true],
        ['false?', a => a === false],
        ['string?', a => typeof a === "string" && a[0] !== '\u029e'],
        ['symbol', a => _symbol(a)],
        ['symbol?', a => _symbol_Q(a)],
        ['keyword', a => _keyword(a)],
        ['keyword?', a => _keyword_Q(a)],

        ['pr-str', (...a) => a.map(e => pr_str(e,1)).join(' ')],
        ['str', (...a) => a.map(e => pr_str(e,0)).join('')],
        ['prn', (...a) => console.log(...a.map(e => pr_str(e,1))) || null],
        ['println', (...a) => console.log(...a.map(e => pr_str(e,0))) || null],
        ['read-string', read_str],
        ['readline', readline],
        ['slurp', slurp],

        ['<' , (a,b) => a<b],
        ['<=', (a,b) => a<=b],
        ['>' , (a,b) => a>b],
        ['>=', (a,b) => a>=b],
        ['+' , (a,b) => a+b],
        ['-' , (a,b) => a-b],
        ['*' , (a,b) => a*b],
        ['/' , (a,b) => a/b],
        ["time-ms", () => new Date().getTime()],

        ['list', (...a) => a],
        ['list?', _list_Q],
        ['vector', _vector],
        ['vector?', _vector_Q],
        ['hash-map', _hash_map],
        ['map?', _hash_map_Q],
        ['assoc', (m,...a) => _assoc_BANG(_clone(m), ...a)],
        ['dissoc', (m,...a) => _dissoc_BANG(_clone(m), ...a)],
        ['get', (m,a) => m === null ? null : m.has(a) ? m.get(a) : null],
        ['contains?', (m,a) => m.has(a)],
        ['keys', keys],
        ['vals', vals],

        ['sequential?', _sequential_Q],
        ['cons', (a,b) => [a].concat(b)],
        ['concat', (...a) => a.reduce((x,y) => x.concat(y), [])],
        ['nth', nth],
        ['first', a => a !== null && a.length > 0 ? a[0] : null],
        ['rest', a => a === null ? [] : a.slice(1)],
        ['empty?', a => a.length === 0],
        ['count', a => a === null ? 0 : a.length],
        ['apply', (f,...a) => f(...a.slice(0, -1).concat(a[a.length-1]))],
        ['map', (f,a) => a.map(x => f(x))],

        ['conj', conj],
        ['seq', seq],

        ['meta', a => 'meta' in a ? a['meta'] : null],
        ['with-meta', with_meta],
        ['atom', a => new Atom(a)],
        ['atom?', a => a instanceof Atom],
        ['deref', atm => atm.val],
        ['reset!', (atm,a) => atm.val = a],
        ['swap!', (atm,f,...args) => atm.val = f(...[atm.val].concat(args))]
        ])