io.vertx.codetrans.MapModel Maven / Gradle / Ivy

Go to download
package io.vertx.codetrans;

import io.vertx.codegen.TypeInfo;

import java.util.List;

/**
 * @author Julien Viet
 */
public class MapModel extends ExpressionModel {

  private final ExpressionModel expression;

  public MapModel(ExpressionModel expression) {
    this.expression = expression;
  }

  @Override
  public ExpressionModel onMethodInvocation(String methodName, List parameterTypes, List argumentModels, List argumenTypes) {
    switch (methodName) {
      case "get":
        return ExpressionModel.render(writer -> {
          writer.getLang().renderMapGet(expression, argumentModels.get(0), writer);
        });
      case "forEach":
        LambdaExpressionModel lambda = (LambdaExpressionModel) argumentModels.get(0);
        return ExpressionModel.render(writer -> {
          writer.getLang().renderMapForEach(
              expression,
              lambda.getParameterNames().get(0),
              lambda.getParameterTypes().get(0),
              lambda.getParameterNames().get(1),
              lambda.getParameterTypes().get(1),
              lambda.getBodyKind(),
              lambda.getBody(),
              writer);
        });
      default:
        throw new UnsupportedOperationException("Map " + methodName + " method not supported");
    }
  }

  @Override
  public void render(CodeWriter writer) {
    expression.render(writer);
  }
}