invoke repl from command line

2012-06-09 00:28:17 +04:00
parent bc82284d83
commit 48b1ab417b
2 changed files with 108 additions and 0 deletions
@@ -29,6 +29,7 @@ import org.jetbrains.jet.cli.jvm.compiler.K2JVMCompileEnvironmentConfiguration;
 import org.jetbrains.jet.cli.jvm.compiler.CompileEnvironmentUtil;
 import org.jetbrains.jet.cli.jvm.compiler.JetCoreEnvironment;
 import org.jetbrains.jet.cli.jvm.compiler.KotlinToJVMBytecodeCompiler;
+import org.jetbrains.jet.cli.jvm.repl.ReplFromTerminal;
 import org.jetbrains.jet.codegen.CompilationException;
 import org.jetbrains.jet.lang.resolve.java.CompilerDependencies;
 import org.jetbrains.jet.lang.resolve.java.CompilerSpecialMode;
@@ -84,6 +85,12 @@ public class K2JVMCompiler extends CLICompiler<K2JVMCompilerArguments, K2JVMComp
        }

        CompilerDependencies dependencies = new CompilerDependencies(mode, CompilerDependencies.findRtJar(), jdkHeadersJar, runtimeJar);
+
+        if (!arguments.script && arguments.module == null && arguments.src == null && arguments.freeArgs.isEmpty()) {
+            ReplFromTerminal.run(rootDisposable, dependencies);
+            return ExitCode.OK;
+        }
+
        JetCoreEnvironment environment = JetCoreEnvironment.getCoreEnvironmentForJVM(rootDisposable, dependencies);
        List<String> scriptArgs = arguments.script ? arguments.freeArgs.subList(1, arguments.freeArgs.size()) : Collections.<String>emptyList();
        K2JVMCompileEnvironmentConfiguration configuration = new K2JVMCompileEnvironmentConfiguration(environment, messageCollector, arguments.script, scriptArgs);
@@ -0,0 +1,101 @@
+/*
+ * Copyright 2010-2012 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.jet.cli.jvm.repl;
+
+import com.intellij.openapi.Disposable;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.jet.lang.resolve.java.CompilerDependencies;
+import org.jetbrains.jet.lang.resolve.java.CompilerSpecialMode;
+import org.jetbrains.jet.utils.ExceptionUtils;
+
+import javax.sound.midi.SysexMessage;
+import java.io.Console;
+import java.io.File;
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * @author Stepan Koltsov
+ */
+public class ReplFromTerminal {
+
+    private ReplInterpreter replInterpreter;
+    private Throwable replInitializationFailed;
+    private final Object waitRepl = new Object();
+
+    public ReplFromTerminal(@NotNull final Disposable disposable, @NotNull final CompilerDependencies compilerDependencies) {
+        final List<File> extraClasspath = Collections.<File>emptyList();
+        new Thread("initialize-repl") {
+            @Override
+            public void run() {
+                try {
+                    replInterpreter = new ReplInterpreter(disposable, compilerDependencies, extraClasspath);
+                } catch (Throwable e) {
+                    replInitializationFailed = e;
+                }
+                synchronized (waitRepl) {
+                    waitRepl.notifyAll();
+                }
+            }
+        }.start();
+    }
+
+    private ReplInterpreter getReplInterpreter() {
+        if (replInterpreter != null) {
+            return replInterpreter;
+        }
+        synchronized (waitRepl) {
+            while (replInterpreter == null && replInitializationFailed == null) {
+                try {
+                    waitRepl.wait();
+                } catch (Throwable e) {
+                    throw ExceptionUtils.rethrow(e);
+                }
+            }
+            if (replInterpreter != null) {
+                return replInterpreter;
+            }
+            throw ExceptionUtils.rethrow(replInitializationFailed);
+        }
+    }
+
+    private void doRun() {
+        System.out.println("Kotlin");
+        while (true) {
+            boolean next = one();
+            if (!next) {
+                break;
+            }
+        }
+    }
+
+    private boolean one() {
+        System.out.print(">>> ");
+        String line = System.console().readLine();
+        if (line == null) {
+            return false;
+        }
+        Object value = getReplInterpreter().eval(line);
+        System.out.println(value);
+        return true;
+    }
+
+    public static void run(@NotNull Disposable disposable, @NotNull CompilerDependencies compilerDependencies) {
+        new ReplFromTerminal(disposable, compilerDependencies).doRun();
+    }
+
+}