io.vertx.codetrans.lang.kotlin.KotlinLang Maven / Gradle / Ivy

Go to download
package io.vertx.codetrans.lang.kotlin;

import io.vertx.codetrans.CodeBuilder;
import io.vertx.codetrans.Lang;
import io.vertx.codetrans.Script;

import java.io.File;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.net.URL;
import java.net.URLClassLoader;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @author Sergey Mashkov
 */
public class KotlinLang implements Lang {

  @Override
  public String id() {
    return "kotlin";
  }

  @Override
  public File createSourceFile(File root, List className, String methodName) {
    File folder = new File(root, className.stream().collect(Collectors.joining(File.separator)));
    if (methodName != null) {
      folder = new File(folder, methodName);
    }
    return new File(folder.getParent(), folder.getName() + ".kt");
  }

  @Override
  public Script loadScript(ClassLoader loader, String path, String method) throws Exception {
    loader = new URLClassLoader(new URL[]{ new File(new File("target"), "kotlin-classes").toURI().toURL() }, loader);
    String fqn = path.replace('/', '.') + "." + method;
    Class c = loader.loadClass(fqn);
    Field accessor = c.getDeclaredField("INSTANCE");
    Method m = c.getDeclaredMethod(method);
    File f = new File(("src/test/generated/kotlin/" + path + "/" + method + ".kt").replace('/', File.separatorChar));
    String source = new String(Files.readAllBytes(f.toPath()));
    return new Script() {
      @Override
      public String getSource() {
        return source;
      }
      @Override
      public void run(Map globals) throws Exception {
        Object instance = accessor.get(null);
        m.invoke(instance);
      }
    };
  }

  @Override
  public String getExtension() {
    return "kt";
  }

  @Override
  public CodeBuilder codeBuilder() {
    return new KotlinCodeBuilder();
  }
}