main/core/parser/ast/FunctionDefineNode.java


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

package com.jantuomi.interpreter.main.core.parser.ast;

import com.jantuomi.interpreter.main.core.parser.datatype.DataContainer;
import com.jantuomi.interpreter.main.core.parser.datatype.StringDataContainer;
import com.jantuomi.interpreter.main.core.runtime.Function;
import com.jantuomi.interpreter.main.core.runtime.State;
import com.jantuomi.interpreter.main.core.tokenizer.token.Token;

import java.util.ArrayList;
import java.util.List;

/**
 * Created by jan on 17.6.2016.
 */
public class FunctionDefineNode extends VarargOperatorNode {

    private String name;

    public void setBody(ASTNode body) {
        this.body = body;
    }

    public void setName(String name) {
        this.name = name;
    }

    private ASTNode body;

    public FunctionDefineNode(Token token) {
        super(token);
    }

    @Override
    public DataContainer evaluate() {
        List<String> argumentNames = new ArrayList<>();
        for (ASTNode arg : args) {
            SymbolNode argSym = (SymbolNode) arg;
            argumentNames.add(argSym.getName());
        }
        Function function = new Function(argumentNames, (BlockBodyNode) body);
        State.getInstance().addFunctionToScope(name, function);

        return new StringDataContainer("function " + name);
    }

    @Override
    public void print(int indent) {

        for (int i = 0; i < indent; i++) {
            System.out.print("\t");
        }

        System.out.println(String.format("Function %s", name));

        for (int i = 0; i < indent + 1; i++) {
            System.out.print("\t");
        }

        System.out.println(String.format("Arguments: "));

        for (ASTNode node : args) {
            node.print(indent + 2);
        }

        for (int i = 0; i < indent + 1; i++) {
            System.out.print("\t");
        }

        System.out.println(String.format("Body statements: "));
        body.print(indent + 2);
    }
}