io.vertx.codetrans.expression.DataObjectModel Maven / Gradle / Ivy

Go to download
package io.vertx.codetrans.expression;

import io.vertx.codegen.type.TypeInfo;
import io.vertx.codetrans.CodeBuilder;
import io.vertx.codetrans.CodeWriter;
import io.vertx.codetrans.MethodSignature;
import io.vertx.codetrans.TypeArg;

import java.util.List;

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

  final ExpressionModel expression;

  public DataObjectModel(CodeBuilder builder, ExpressionModel expression) {
    super(builder);
    this.expression = expression;
  }

  @Override
  public ExpressionModel onMethodInvocation(TypeInfo receiverType, MethodSignature method, TypeInfo returnType, List typeArguments, List argumentModels, List argumentTypes) {
    String methodName = method.getName();
    if (DataObjectLiteralModel.isSet(methodName)) {
      return builder.render(writer -> {
        writer.renderDataObjectAssign(expression,
            DataObjectLiteralModel.unwrapSet(methodName),
            argumentModels.get(0).toDataObjectValue());
      });
    }
    if (DataObjectLiteralModel.isGet(methodName)) {
      return builder.render(writer -> {
        writer.renderDataObjectMemberSelect(expression,
            DataObjectLiteralModel.unwrapSet(methodName));
      });
    }
    if (isToJson(method)) {
      return builder.render(writer -> {
        writer.renderDataObjectToJson((IdentifierModel) expression);
      });
    }
    throw new UnsupportedOperationException("Unsupported method " + method + " on object model");
  }

  private boolean isToJson(MethodSignature method) {
    return "toJson".equals(method.getName())
      && method.getReturnType().getKind().json
      && method.getParameterTypes().isEmpty();
  }

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