editor.util.transform.java.JavaToGosu Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2014 Guidewire Software, Inc.
 */

package editor.util.transform.java;


import com.sun.source.tree.CompilationUnitTree;
import com.sun.tools.javac.api.JavacTaskImpl;
import com.sun.tools.javac.api.JavacTool;
import editor.util.transform.java.visitor.GosuVisitor;


import javax.tools.JavaCompiler;
import javax.tools.StandardJavaFileManager;
import java.io.IOException;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List;

public class JavaToGosu {

  public static String convertString(String javaSource) {
    Wrap[] wraps = new Wrap[]{
            new Wrap("", "", "", ""),
            new Wrap("class _JAVA_TO_GOSU_INTERNAL_ {",
                    "\n}",
                    "class _JAVA_TO_GOSU_INTERNAL_  {\n\n",
                    "\n}\n"),
            new Wrap("class _JAVA_TO_GOSU_INTERNAL_ { void _JAVA_TO_GOSU_INTERNAL_METHOD_() {",
                    "\n}}",
                    "class _JAVA_TO_GOSU_INTERNAL_  {\n\n  function _JAVA_TO_GOSU_INTERNAL_METHOD_() : void {\n",
                    "\n  }\n\n}\n"),
            new Wrap("class _JAVA_TO_GOSU_INTERNAL_ { void _JAVA_TO_GOSU_INTERNAL_METHOD_() {",
                    ";\n}}",
                    "class _JAVA_TO_GOSU_INTERNAL_  {\n\n  function _JAVA_TO_GOSU_INTERNAL_METHOD_() : void {\n",
                    "\n  }\n\n}\n"),
            new Wrap("class _JAVA_TO_GOSU_INTERNAL_ { void _JAVA_TO_GOSU_INTERNAL_METHOD_() { if(",
                    ");\n}}",
                    "class _JAVA_TO_GOSU_INTERNAL_  {\n\n  function _JAVA_TO_GOSU_INTERNAL_METHOD_() : void {\n    if (",
                    ") {\n      \n    }\n  }\n\n}\n")
    };
    String javaWrapStart = "";
    String javaWrapEnd = "";
    String gosuWrapStart = "";
    String gosuWrapEnd = "";
    List trees = null;
    boolean parsed = false;
    String output = null;
    int i = wraps.length - 1;

    while (i >= 0 && !parsed) {
      javaWrapStart = wraps[i].JAVA_WRAP_START;
      javaWrapEnd = wraps[i].JAVA_WRAP_END;
      gosuWrapStart = wraps[i].GOSU_WRAP_START;
      gosuWrapEnd = wraps[i].GOSU_WRAP_END;
      trees = new ArrayList();
      String src = javaWrapStart + javaSource + javaWrapEnd;
      parsed = parseJava(trees, src);
      i--;
    }
    if (!parsed) {
      return "";
    }
    GosuVisitor visitor = new GosuVisitor(2);
    for (CompilationUnitTree tree : trees) {
      tree.accept(visitor, null);
    }
    output = visitor.getOutput().toString();

    if (!javaWrapStart.isEmpty()) {
      int b = output.indexOf(gosuWrapStart) + gosuWrapStart.length();
      int e = output.lastIndexOf(gosuWrapEnd);
      output = unIndent(output.substring(b, e));
    }
    return output;
  }

  private static String unIndent(String output) {
    StringBuilder src = new StringBuilder();
    String[] lines = output.split("\n");
    int tab = 0;
    char[] chars = lines[0].toCharArray();
    while (tab < chars.length && chars[tab] == ' ') {
      tab++;
    }
    for (String line : lines) {
      if (!line.trim().isEmpty()) {
        if (line.length() > tab && line.substring(0, tab).trim().isEmpty()) {
          src.append(line.substring(tab));
        } else {
          src.append(line);
        }
        src.append("\n");
      }
    }
    return src.toString();
  }

  private static boolean parseJava(List trees, String src) {
    JavaCompiler compiler = JavacTool.create();
    StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
    boolean success = true;

    ArrayList javaStringObjects = new ArrayList();
    javaStringObjects.add(new JavaStringObject(src));
    StringWriter errors = new StringWriter();
    JavaCompiler.CompilationTask task = compiler.getTask(errors, fileManager, null, null, null, javaStringObjects);
    JavacTaskImpl javacTask = (JavacTaskImpl) task;
    try {
      Iterable iterable = javacTask.parse();
      for (CompilationUnitTree x : iterable) {
        trees.add(x);
      }
      fileManager.close();
    } catch (IOException e) {
      success = false;
    }
    if (!errors.toString().isEmpty()) {
      success = false;
    }
    return success;
  }
}