summaryrefslogtreecommitdiff
path: root/bin/actinide-repl
blob: 99c605a5df88afc00d1665c165d78646b95c4220 (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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#!/usr/bin/env python3

import sys

import readline

import actinide as a
import actinide.tokenizer as ak
import actinide.reader as ar
import actinide.ports as ap
import actinide.types as at

class ConsolePort(ap.Port):
    def __init__(self):
        self.line = None
        self.prompt = ">>> "

    def next_prompt(self):
        prompt, self.prompt = self.prompt, "... "
        return prompt

    def reset_prompt(self):
        self.prompt = ">>> "

    def ensure_line(self):
        if self.line == None:
            try:
                self.line = input(self.next_prompt()) + '\n'
            except EOFError:
                self.line = ''

    def peek(self, n):
        self.ensure_line()
        return self.line[:n]

    def read(self, n):
        self.ensure_line()
        result, self.line = self.line[:n], self.line[n:]
        if self.line == '':
            self.line = None
        return result

    def read_fully(self, n):
        def read_fully_():
            result = self.read(1024)
            yield result
            while result != '':
                result = self.read(1024)
                yield result
        return ''.join(read_fully_())

    def forget_input(self):
        self.line = None

def repl(session, port):
    while True:
        try:
            port.reset_prompt()
            form = session.read(port)
            if form is ar.EOF:
                print()
                return 0
            results = session.eval(form)
            print(*(session.display(result) for result in results))
        except (ak.TokenError, ar.SyntaxError) as e:
            print(e)
            port.forget_input()
        except Exception as e:
            print(e)
        except KeyboardInterrupt:
            print()
            print("(Interrupted)")

def main():
    port = ConsolePort()
    session = a.Session()

    return repl(session, port)

if __name__ == '__main__':
    sys.exit(main())