org.vertx.mods.VertxCommand Maven / Gradle / Ivy

Go to download
package org.vertx.mods;

import org.crsh.command.ScriptException;
import org.crsh.groovy.GroovyCommand;
import org.vertx.java.core.Vertx;
import org.vertx.java.core.json.DecodeException;
import org.vertx.java.core.json.JsonObject;
import org.vertx.java.platform.Container;
import org.vertx.java.platform.impl.DefaultContainer;
import org.vertx.java.platform.impl.DefaultPlatformManager;
import org.vertx.java.platform.impl.Deployment;
import org.vertx.java.platform.impl.PlatformManagerInternal;

import java.lang.reflect.Field;
import java.util.Collections;
import java.util.List;
import java.util.Map;

/** @author Julien Viet */
public class VertxCommand extends GroovyCommand {

  protected final Vertx getVertx() {
    return (Vertx)context.getAttributes().get("vertx");
  }

  protected final Container getContainer() {
    return (Container)context.getAttributes().get("container");
  }

  protected final PlatformManagerInternal getManager() {
    try {
      Container container = getContainer();
      Field f = DefaultContainer.class.getDeclaredField("mgr");
      f.setAccessible(true);
      return (PlatformManagerInternal)f.get(container);
    }
    catch (Exception e) {
      e.printStackTrace();
      throw new ScriptException("Could not access verticle manager");
    }
  }

  protected final Map getDeployments() {
    try {
      PlatformManagerInternal mgr = getManager();
      Field d = DefaultPlatformManager.class.getDeclaredField("deployments");
      d.setAccessible(true);
      return Collections.unmodifiableMap((Map)d.get(mgr));
    }
    catch (Exception e) {
      throw new ScriptException("Could not access deployments");
    }
  }

  public static String join(List parts) {
    StringBuilder sb = new StringBuilder();
    for (String part : parts) {
      sb.append(part);
    }
    return sb.toString();
  }

  public static JsonObject parseJson(List parts) throws ScriptException {
    return parseJson(join(parts));
  }

  public static JsonObject parseJson(String s) throws ScriptException {
    try {
      return new JsonObject(s);
    }
    catch (DecodeException ignore) {
      throw new ScriptException("Invalid JSON:" + s);
    }
  }
}