package io.vertx.codetrans;
import com.sun.tools.javac.code.Symbol;
import io.vertx.codetrans.expression.ExpressionModel;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;
public class VisitContext {
final CodeBuilder builder;
private final Map<Symbol, ExpressionModel> aliases;
private final Set<String> referencedMethods;
private final Set<String> referencedFields;
public VisitContext(CodeBuilder builder) {
this.builder = builder;
this.aliases = Collections.emptyMap();
this.referencedMethods = new LinkedHashSet<>();
this.referencedFields = new LinkedHashSet<>();
}
private VisitContext(CodeBuilder builder, Map<Symbol, ExpressionModel> aliases, Set<String> referencedMethods, Set<String> referencedFields) {
this.builder = builder;
this.aliases = aliases;
this.referencedMethods = referencedMethods;
this.referencedFields = referencedFields;
}
public VisitContext putAlias(Symbol symbol, ExpressionModel builder) {
HashMap<Symbol, ExpressionModel> clone = new HashMap<>(aliases);
clone.put(symbol, builder);
return new VisitContext(this.builder, clone, referencedMethods, referencedFields);
}
public ExpressionModel getAlias(Symbol symbol) {
return aliases.get(symbol);
}
public Set<String> getReferencedMethods() {
return referencedMethods;
}
public Set<String> getReferencedFields() {
return referencedFields;
}
}