With.java

package org.yinwang.pysonar.ast;

import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.yinwang.pysonar.Scope;
import org.yinwang.pysonar.types.Type;

import java.util.List;

public class With extends Node {

    @NotNull
    public List<Withitem> items;
    public Block body;


    public With(@NotNull List<Withitem> items, Block body, int start, int end) {
        super(start, end);
        this.items = items;
        this.body = body;
        addChildren(items);
        addChildren(body);
    }

    @NotNull
    @Override
    public Type resolve(@NotNull Scope s, int tag) {
        for (Withitem item : items) {
            Type val = resolveExpr(item.context_expr, s, tag);
            if (item.optional_vars != null) {
                NameBinder.bind(s, item.optional_vars, val, tag);
            }
        }
        return resolveExpr(body, s, tag);
    }

    @NotNull
    @Override
    public String toString() {
        return "<With:" + items + ":" + body + ">";
    }

    @Override
    public void visit(@NotNull NodeVisitor v) {
        if (v.visit(this)) {
            for (Withitem item : items) {
                visitNode(item, v);
            }

            visitNode(body, v);
        }
    }
}