aboutsummaryrefslogtreecommitdiff
path: root/lua/printer.lua
blob: 8c1cdad5d95d2bf7320fc955b94f0d12576e6fa5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
local string = require('string')
local table = require('table')
local types = require('types')
local utils = require('utils')

local M = {}

function M._pr_str(obj, print_readably)
    local _r = print_readably
    if utils.instanceOf(obj, types.Symbol) then
        return obj.val
    elseif types._list_Q(obj) then
        return "(" .. table.concat(utils.map(function(e)
            return M._pr_str(e,_r) end, obj), " ") .. ")"
    elseif types._vector_Q(obj) then
        return "[" .. table.concat(utils.map(function(e)
            return M._pr_str(e,_r) end, obj), " ") .. "]"
    elseif types._hash_map_Q(obj) then
        local res = {}
        for k,v in pairs(obj) do
            res[#res+1] = M._pr_str(k, _r)
            res[#res+1] = M._pr_str(v, _r)
        end
        return "{".. table.concat(res, " ").."}"
    elseif type(obj) == 'string' then
        if string.sub(obj,1,1) == "\177" then
            return ':' .. string.sub(obj,2)
        else
            if _r then
                local sval = obj:gsub('\\', '\\\\')
                sval = sval:gsub('"', '\\"')
                sval = sval:gsub('\n', '\\n')
                return '"' .. sval .. '"'
            else
                return obj
            end
        end
    elseif obj == types.Nil then
        return "nil"
    elseif obj == true then
        return "true"
    elseif obj == false then
        return "false"
    elseif types._malfunc_Q(obj) then
        return "(fn* "..M._pr_str(obj.params).." "..M._pr_str(obj.ast)..")"
    elseif types._atom_Q(obj) then
        return "(atom "..M._pr_str(obj.val)..")"
    elseif type(obj) == 'function' then
        return "#<function>"
    else
        return string.format("%s", obj)
    end
end

return M