aboutsummaryrefslogtreecommitdiff
path: root/go/src/printer/printer.go
blob: fa1c0fbf6a325706d378d0f1cf902976a74d2bd4 (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
package printer

import (
    "fmt"
    "strings"
)

import (
    "types"
)

func _pr_list(lst []types.MalType, pr bool, start string, end string) string {
    str_list := make([]string, 0, len(lst))
    for _, e := range lst {
        str_list = append(str_list, Pr_str(e, pr))
    }
    return start + strings.Join(str_list, " ") + end
}

func Pr_str(obj types.MalType, print_readably bool) string {
    switch tobj := obj.(type) {
    case types.List:
        return _pr_list(tobj.Val, print_readably, "(", ")")
    case types.Vector:
        return _pr_list(tobj.Val, print_readably, "[", "]")
    case map[string]types.MalType:
        str_list := make([]string, 0, len(tobj)*2)
        for k, v := range tobj {
            str_list = append(str_list, Pr_str(k, print_readably))
            str_list = append(str_list, Pr_str(v, print_readably))
        }
        return "{" + strings.Join(str_list, " ") + "}"
    case string:
        if print_readably {
            // TODO: quote backslash, quote, and newline
            return `"` + fmt.Sprintf("%v", obj) + `"`
        } else {
            return fmt.Sprintf("%v", obj)
        }
    case types.Symbol:
        return tobj.Val
    case nil:
        return "nil"
    default:
        return fmt.Sprintf("%v", obj)
    }
}