standalone compiler

2011-10-17 17:02:47 +02:00
parent fa3c8b4cfe
commit 5a1bca0615
3 changed files with 130 additions and 0 deletions
@@ -4,6 +4,7 @@
    <modules>
      <module fileurl="file://$PROJECT_DIR$/Jet.iml" filepath="$PROJECT_DIR$/Jet.iml" />
      <module fileurl="file://$PROJECT_DIR$/compiler/backend/backend.iml" filepath="$PROJECT_DIR$/compiler/backend/backend.iml" />
+      <module fileurl="file://$PROJECT_DIR$/compiler/cli/cli.iml" filepath="$PROJECT_DIR$/compiler/cli/cli.iml" />
      <module fileurl="file://$PROJECT_DIR$/examples/examples.iml" filepath="$PROJECT_DIR$/examples/examples.iml" />
      <module fileurl="file://$PROJECT_DIR$/compiler/frontend/frontend.iml" filepath="$PROJECT_DIR$/compiler/frontend/frontend.iml" />
      <module fileurl="file://$PROJECT_DIR$/compiler/frontend.java/frontend.java.iml" filepath="$PROJECT_DIR$/compiler/frontend.java/frontend.java.iml" />
@@ -0,0 +1,15 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="JAVA_MODULE" version="4">
+  <component name="NewModuleRootManager" inherit-compiler-output="true">
+    <exclude-output />
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
+    </content>
+    <orderEntry type="jdk" jdkName="IDEA 10.x" jdkType="IDEA JDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+    <orderEntry type="module" module-name="backend" />
+    <orderEntry type="module" module-name="frontend" />
+    <orderEntry type="module" module-name="frontend.java" />
+  </component>
+</module>
+
@@ -0,0 +1,114 @@
+package org.jetbrains.jet.cli;
+
+import com.google.common.collect.Lists;
+import com.intellij.core.JavaCoreEnvironment;
+import com.intellij.openapi.Disposable;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.util.io.FileUtil;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.psi.PsiFile;
+import com.intellij.psi.PsiManager;
+import org.jetbrains.jet.codegen.ClassFileFactory;
+import org.jetbrains.jet.codegen.GenerationState;
+import org.jetbrains.jet.lang.cfg.pseudocode.JetControlFlowDataTraceFactory;
+import org.jetbrains.jet.lang.diagnostics.Diagnostic;
+import org.jetbrains.jet.lang.parsing.JetParserDefinition;
+import org.jetbrains.jet.lang.psi.JetFile;
+import org.jetbrains.jet.lang.psi.JetNamespace;
+import org.jetbrains.jet.lang.resolve.AnalyzingUtils;
+import org.jetbrains.jet.lang.resolve.BindingContext;
+import org.jetbrains.jet.lang.resolve.java.JavaDefaultImports;
+import org.jetbrains.jet.plugin.JetFileType;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.List;
+
+/**
+ * @author yole
+ */
+public class KotlinCompiler {
+    public static void main(String[] args) {
+        System.setProperty("java.awt.headless", "true");
+        if (args.length < 1) {
+            System.out.println("Usage: KotlinCompiler <filename>");
+            return;
+        }
+        String javaHome = System.getenv("JAVA_HOME");
+        if (javaHome == null) {
+            System.out.println("JAVA_HOME environment variable needs to be defined");
+            return;
+        }
+        File rtJar = findRtJar(javaHome);
+        if (rtJar == null || !rtJar.exists()) {
+            System.out.print("No rt.jar found under JAVA_HOME");
+            return;
+        }
+
+        Disposable root = new Disposable() {
+            @Override
+            public void dispose() {
+            }
+        };
+        JavaCoreEnvironment environment = new JavaCoreEnvironment(root);
+        environment.addToClasspath(rtJar);
+
+        environment.registerFileType(JetFileType.INSTANCE, "kt");
+        environment.registerParserDefinition(new JetParserDefinition());
+        VirtualFile vFile = environment.getLocalFileSystem().findFileByPath(args [0]);
+
+        Project project = environment.getProject();
+        GenerationState generationState = new GenerationState(project, false);
+        List<JetNamespace> namespaces = Lists.newArrayList();
+        PsiFile psiFile = PsiManager.getInstance(project).findFile(vFile);
+        if (psiFile instanceof JetFile) {
+            namespaces.add(((JetFile) psiFile).getRootNamespace());
+        }
+
+        BindingContext bindingContext = AnalyzingUtils.getInstance(JavaDefaultImports.JAVA_DEFAULT_IMPORTS).analyzeNamespaces(project, namespaces, JetControlFlowDataTraceFactory.EMPTY);
+
+        boolean errors = false;
+        for (Diagnostic diagnostic : bindingContext.getDiagnostics()) {
+            switch (diagnostic.getSeverity()) {
+                case ERROR:
+                    errors = true;
+                    report(diagnostic);
+                    break;
+                case INFO:
+                    report(diagnostic);                            
+                    break;
+                case WARNING:
+                    report(diagnostic);                            
+                    break;
+            }
+        }
+
+        if (!errors) {
+            generationState.compileCorrectNamespaces(bindingContext, namespaces);
+
+            final ClassFileFactory factory = generationState.getFactory();
+            List<String> files = factory.files();
+            for (String file : files) {
+                File target = new File(vFile.getParent().getPath(), file);
+                try {
+                    FileUtil.writeToFile(target, factory.asBytes(file));
+                } catch (IOException e) {
+                    System.out.println(e.getMessage());
+                }
+            }
+        }
+
+    }
+
+    private static void report(Diagnostic diagnostic) {
+        System.out.println(diagnostic.getMessage());
+    }
+
+    private static File findRtJar(String javaHome) {
+        File rtJar = new File(javaHome, "jre/lib/rt.jar");
+        if (rtJar.exists()) {
+            return rtJar;
+        }
+        return null;
+    }
+}