EnvironmentClass.java

package stanhebben.zenscript.compiler;

import org.objectweb.asm.ClassVisitor;
import stanhebben.zenscript.*;
import stanhebben.zenscript.expression.partial.IPartialExpression;
import stanhebben.zenscript.symbols.IZenSymbol;
import stanhebben.zenscript.type.ZenType;
import stanhebben.zenscript.util.ZenPosition;

import java.lang.reflect.Type;
import java.util.*;

/**
 * @author Stan
 */
public class EnvironmentClass implements IEnvironmentClass {
    
    private final ClassVisitor output;
    private final IEnvironmentGlobal global;
    private final Map<String, IZenSymbol> local;
    
    public EnvironmentClass(ClassVisitor output, IEnvironmentGlobal global) {
        this.output = output;
        this.global = global;
        this.local = new HashMap<>();
    }
    
    @Override
    public ClassVisitor getClassOutput() {
        return output;
    }
    
    @Override
    public ZenType getType(Type type) {
        return global.getType(type);
    }
    
    @Override
    public IZenCompileEnvironment getEnvironment() {
        return global.getEnvironment();
    }
    
    @Override
    public TypeExpansion getExpansion(String name) {
        return global.getExpansion(name);
    }
    
    @Override
    public String makeClassName() {
        return global.makeClassName();
    }
    
    @Override
    public String makeClassNameWithMiddleName(String middleName) {
        return global.makeClassNameWithMiddleName(middleName);
    }
    
    @Override
    public boolean containsClass(String name) {
        return global.containsClass(name);
    }
    
    @Override
    public void putClass(String name, byte[] data) {
        global.putClass(name, data);
    }
    
    @Override
    public IPartialExpression getValue(String name, ZenPosition position) {
        if(local.containsKey(name)) {
            return local.get(name).instance(position);
        } else {
            return global.getValue(name, position);
        }
    }
    
    @Override
    public void putValue(String name, IZenSymbol value, ZenPosition position) {
        if(local.containsKey(name)) {
            error(position, "Value already defined in this scope: " + name);
        } else {
            local.put(name, value);
        }
    }
    
    @Override
    public void error(ZenPosition position, String message) {
        global.error(position, message);
    }
    
    @Override
    public void warning(ZenPosition position, String message) {
        global.warning(position, message);
    }
    
    @Override
    public void info(ZenPosition position, String message) {
        global.info(position, message);
    }
    
    @Override
    public Set<String> getClassNames() {
        return global.getClassNames();
    }
    
    @Override
    public byte[] getClass(String name) {
        return global.getClass(name);
    }
    
    @Override
    public void error(String message) {
        global.error(message);
    }
    
    @Override
    public void error(String message, Throwable e) {
        global.error(message, e);
    }
    
    @Override
    public void warning(String message) {
        global.warning(message);
    }
    
    @Override
    public void info(String message) {
        global.info(message);
    }
}