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

Go to download
package io.vertx.codetrans;

import io.vertx.codegen.ClassKind;
import io.vertx.codegen.TypeInfo;

import java.util.Collections;
import java.util.List;

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

  private final List entries;

  public JsonObjectLiteralModel() {
    this(Collections.emptyList());
  }

  private JsonObjectLiteralModel(List entries) {
    this.entries = entries;
  }

  public Iterable getMembers() {
    return entries;
  }

  @Override
  public ExpressionModel as(TypeInfo type) {
    if (type.getKind() != ClassKind.JSON_OBJECT) {
      throw new UnsupportedOperationException();
    }
    return this;
  }

  @Override
  public ExpressionModel onMethodInvocation(String methodName, List parameterTypes, List argumentModels, List argumenTypes) {
    switch (methodName) {
      case "put":
        return new JsonObjectLiteralModel(Helper.append(entries, new Member.Single(argumentModels.get(0)).append(argumentModels.get(1))));
      default:
        throw new UnsupportedOperationException("Method " + methodName + " not yet implemented");
    }
  }

  @Override
  public void render(CodeWriter writer) {
    writer.getLang().renderJsonObject(this, writer);
  }
}