aboutsummaryrefslogtreecommitdiff
path: root/interpreter.py
blob: c77aa1c8327268da4dc00a476318a63c5762a22d (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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#!/usr/bin/python3 -tt

# Copyright 2012 Jussi Pakkanen

# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at

#     http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import parser
import nodes
import environment

class InterpreterException(Exception):
    pass

class InvalidCode(InterpreterException):
    pass

class Interpreter():
    
    def __init__(self, code):
        self.ast = parser.build_ast(code)
        self.sanity_check_ast()
        self.project = None
        self.compilers = []
        
    def sanity_check_ast(self):
        if not isinstance(self.ast, nodes.CodeBlock):
            raise InvalidCode('AST is of invalid type. Possibly a bug in the parser.')
        if len(self.ast.get_statements()) == 0:
            raise InvalidCode('No statements in code.')
        first = self.ast.get_statements()[0]
        if not isinstance(first, nodes.FunctionCall) or first.get_function_name() != 'project':
            raise InvalidCode('First statement must be a call to project')
    
    def run(self):
        i = 0
        statements = self.ast.get_statements()
        while i < len(statements):
            cur = statements[i]
            if isinstance(cur, nodes.FunctionCall):
                self.function_call(cur)
            else:
                print("Unknown statement in line %d." % cur.lineno())
            i += 1

    def function_call(self, node):
        func_name = node.get_function_name()
        args = node.arguments.arguments
        if func_name == 'project':
            if len(args) != 1:
                raise InvalidCode('Project() must have one and only one argument.')
            if not isinstance(args[0], nodes.StringStatement):
                raise InvalidCode('Project() argument must be a string.')
            if self.project is not None:
                raise InvalidCode('Second call to project() on line %d.' % node.lineno())
            self.project = args[0].get_string()
            print("Project name is %s." % self.project)
        elif func_name == 'message':
            if len(args) != 1:
                raise InvalidCode('Function message() must have only one argument')
            if not isinstance(args[0], nodes.StringStatement):
                raise InvalidCode('Argument to message() must be a string')
            print('Message: %s' % args[0].get_string())
        elif func_name == 'language':
            if len(args) != 1:
                raise InvalidCode('Function language() must have only one argument')
            if not isinstance(args[0], nodes.StringStatement):
                raise InvalidCode('Argument to language() must be a string')
            if len(self.compilers) > 0:
                raise InvalidCode('Function language() can only be called once (line %d).' % node.lineno())
            lang = args[0].get_string()
            if lang.lower() == 'c':
                self.compilers.append(environment.detect_c_compiler('gcc'))
            else:
                raise InvalidCode('Tried to use unknown language "%s".' % lang)
        else:
            raise InvalidCode('Unknown function "%s".' % func_name)

if __name__ == '__main__':
    code = """project('myawesomeproject')
    message('I can haz text printed out?')
    message('It workses!')
    language('c')
    """
    i = Interpreter(code)
    i.run()