src/main/java/org/zwobble/couscous/interpreter/StackFrameBuilder.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

package org.zwobble.couscous.interpreter;

import java.util.Map;

import org.zwobble.couscous.ast.VariableDeclaration;
import org.zwobble.couscous.ast.VariableNode;
import org.zwobble.couscous.interpreter.values.InterpreterValue;

import com.google.common.collect.ImmutableMap;

import java.util.Optional;

public class StackFrameBuilder {
    private final ImmutableMap.Builder<VariableDeclaration, Optional<InterpreterValue>> variables
        = ImmutableMap.builder();
    
    public StackFrameBuilder declare(VariableNode variable, InterpreterValue value) {
        variables.put(variable.getDeclaration(), Optional.of(value));
        return this;
    }
    
    public StackFrameBuilder declare(VariableNode variable) {
        variables.put(variable.getDeclaration(), Optional.empty());
        return this;
    }
    
    public Map<VariableDeclaration, Optional<InterpreterValue>> build() {
        return variables.build();
    }
}