Command mode support

bom/deployment/pom.xml

@@ -674,12 +674,6 @@
                 <version>${project.version}</version>
             </dependency>
 
-            <dependency>
-                <groupId>io.quarkus</groupId>
-                <artifactId>quarkus-development-mode</artifactId>
-                <version>${project.version}</version>
-            </dependency>
-
             <dependency>
                 <groupId>io.quarkus</groupId>
                 <artifactId>quarkus-scala-deployment</artifactId>

bom/runtime/pom.xml

@@ -272,6 +272,18 @@
                 <artifactId>quarkus-development-mode-spi</artifactId>
                 <version>${project.version}</version>
             </dependency>
+            <dependency>
+                <groupId>io.quarkus</groupId>
+                <artifactId>quarkus-ide-launcher</artifactId>
+                <version>${project.version}</version>
+                <exclusions>
+                    <exclusion>
+                        <!-- This has shaded dependencies -->
+                        <groupId>*</groupId>
+                        <artifactId>*</artifactId>
+                    </exclusion>
+                </exclusions>
+            </dependency>
             <dependency>
                 <groupId>io.quarkus</groupId>
                 <artifactId>quarkus-arc</artifactId>

core/deployment/src/main/java/io/quarkus/deployment/QuarkusAugmentor.java

@@ -33,6 +33,7 @@
 import io.quarkus.deployment.builditem.GeneratedResourceBuildItem;
 import io.quarkus.deployment.builditem.LaunchModeBuildItem;
 import io.quarkus.deployment.builditem.LiveReloadBuildItem;
+import io.quarkus.deployment.builditem.RawCommandLineArgumentsBuildItem;
 import io.quarkus.deployment.builditem.ShutdownContextBuildItem;
 import io.quarkus.deployment.pkg.builditem.BuildSystemTargetBuildItem;
 import io.quarkus.deployment.pkg.builditem.CurateOutcomeBuildItem;
@@ -107,6 +108,7 @@ public BuildResult run() throws Exception {
                     .addInitial(DeploymentClassLoaderBuildItem.class)
                     .addInitial(ArchiveRootBuildItem.class)
                     .addInitial(ShutdownContextBuildItem.class)
+                    .addInitial(RawCommandLineArgumentsBuildItem.class)
                     .addInitial(LaunchModeBuildItem.class)
                     .addInitial(LiveReloadBuildItem.class)
                     .addInitial(AdditionalApplicationArchiveBuildItem.class)
@@ -132,6 +134,7 @@ public BuildResult run() throws Exception {
                     .produce(liveReloadBuildItem)
                     .produce(new ArchiveRootBuildItem(root, rootFs == null ? root : rootFs.getPath("/"), excludedFromIndexing))
                     .produce(new ShutdownContextBuildItem())
+                    .produce(new RawCommandLineArgumentsBuildItem())
                     .produce(new LaunchModeBuildItem(launchMode))
                     .produce(new BuildSystemTargetBuildItem(targetDir, baseName))
                     .produce(new DeploymentClassLoaderBuildItem(deploymentClassLoader))

core/deployment/src/main/java/io/quarkus/deployment/builditem/RawCommandLineArgumentsBuildItem.java

@@ -0,0 +1,38 @@
+package io.quarkus.deployment.builditem;
+
+import java.util.function.Supplier;
+
+import io.quarkus.builder.item.SimpleBuildItem;
+import io.quarkus.deployment.recording.BytecodeRecorderImpl;
+import io.quarkus.runtime.StartupContext;
+
+/**
+ * A build item that represents the raw command line arguments as they were passed to the application.
+ * 
+ * This can be passed directly to bytecode recorders that take a {@code Supplier<String[]>}.
+ *
+ * No filtering is done on these parameters.
+ */
+public final class RawCommandLineArgumentsBuildItem extends SimpleBuildItem
+        implements BytecodeRecorderImpl.ReturnedProxy, Supplier<String[]> {
+
+    /**
+     * As this object directly implements ReturnedProxy it can be
+     * passed into bytecode recorders. The runtime value of the command line parameters
+     * in places directly into the startup context under this key.
+     */
+    @Override
+    public String __returned$proxy$key() {
+        return StartupContext.RAW_COMMAND_LINE_ARGS;
+    }
+
+    @Override
+    public boolean __static$$init() {
+        return true;
+    }
+
+    @Override
+    public String[] get() {
+        throw new IllegalStateException("Can only be called at runtime");
+    }
+}

core/devmode/src/main/java/io/quarkus/dev/ClassLoaderCompiler.java → core/deployment/src/main/java/io/quarkus/deployment/dev/ClassLoaderCompiler.java

@@ -1,4 +1,4 @@
-package io.quarkus.dev;
+package io.quarkus.deployment.dev;
 
 import java.io.File;
 import java.io.IOException;

core/devmode/src/main/java/io/quarkus/dev/CompilationProvider.java → core/deployment/src/main/java/io/quarkus/deployment/dev/CompilationProvider.java

@@ -1,4 +1,4 @@
-package io.quarkus.dev;
+package io.quarkus.deployment.dev;
 
 import java.io.File;
 import java.nio.charset.Charset;

core/devmode/src/main/java/io/quarkus/dev/CompilerFlags.java → core/deployment/src/main/java/io/quarkus/deployment/dev/CompilerFlags.java

@@ -1,4 +1,4 @@
-package io.quarkus.dev;
+package io.quarkus.deployment.dev;
 
 import java.util.ArrayList;
 import java.util.Collection;

core/devmode/src/main/java/io/quarkus/dev/DevModeContext.java → core/deployment/src/main/java/io/quarkus/deployment/dev/DevModeContext.java

@@ -1,4 +1,4 @@
-package io.quarkus.dev;
+package io.quarkus.deployment.dev;
 
 import java.io.File;
 import java.io.Serializable;
@@ -29,6 +29,7 @@ public class DevModeContext implements Serializable {
     private String sourceEncoding;
 
     private final List<File> classesRoots = new ArrayList<>();
+    private final List<URL> additionalClassPathElements = new ArrayList<>();
     private File frameworkClassesDir;
     private File cacheDir;
     private File projectDir;
@@ -82,6 +83,10 @@ public List<File> getClassesRoots() {
         return classesRoots;
     }
 
+    public List<URL> getAdditionalClassPathElements() {
+        return additionalClassPathElements;
+    }
+
     public File getFrameworkClassesDir() {
         return frameworkClassesDir;
     }

core/devmode/src/main/java/io/quarkus/dev/DevModeMain.java → core/deployment/src/main/java/io/quarkus/deployment/dev/DevModeMain.java

@@ -1,4 +1,5 @@
-package io.quarkus.dev;
+
+package io.quarkus.deployment.dev;
 
 import java.io.Closeable;
 import java.io.DataInputStream;
@@ -21,6 +22,7 @@
 import io.quarkus.bootstrap.app.AdditionalDependency;
 import io.quarkus.bootstrap.app.CuratedApplication;
 import io.quarkus.bootstrap.app.QuarkusBootstrap;
+import io.quarkus.dev.appstate.ApplicationStateNotification;
 
 /**
  * The main entry point for the dev mojo execution
@@ -154,5 +156,6 @@ public void close() throws IOException {
         if (realCloseable != null) {
             realCloseable.close();
         }
+        ApplicationStateNotification.waitForApplicationStop();
     }
 }

core/devmode/src/main/java/io/quarkus/dev/HotDeploymentConfigFileBuildStep.java → core/deployment/src/main/java/io/quarkus/deployment/dev/HotDeploymentConfigFileBuildStep.java

@@ -1,4 +1,4 @@
-package io.quarkus.dev;
+package io.quarkus.deployment.dev;
 
 import java.util.List;
 import java.util.Map;

core/devmode/src/main/java/io/quarkus/dev/IsolatedDevModeMain.java → core/deployment/src/main/java/io/quarkus/deployment/dev/IsolatedDevModeMain.java

@@ -1,4 +1,4 @@
-package io.quarkus.dev;
+package io.quarkus.deployment.dev;
 
 import java.io.ByteArrayInputStream;
 import java.io.ByteArrayOutputStream;
@@ -31,6 +31,7 @@
 import io.quarkus.deployment.builditem.ApplicationClassPredicateBuildItem;
 import io.quarkus.dev.spi.HotReplacementSetup;
 import io.quarkus.runner.bootstrap.AugmentActionImpl;
+import io.quarkus.runtime.ApplicationLifecycleManager;
 import io.quarkus.runtime.Timing;
 import io.quarkus.runtime.configuration.QuarkusConfigFactory;
 import io.quarkus.runtime.logging.InitialConfigurator;
@@ -49,6 +50,7 @@ public class IsolatedDevModeMain implements BiConsumer<CuratedApplication, Map<S
     static volatile RuntimeUpdatesProcessor runtimeUpdatesProcessor;
     private static volatile CuratedApplication curatedApplication;
     private static volatile AugmentAction augmentAction;
+    private static volatile boolean restarting;
 
     private synchronized void firstStart() {
         ClassLoader old = Thread.currentThread().getContextClassLoader();
@@ -57,7 +59,31 @@ private synchronized void firstStart() {
             //ok, we have resolved all the deps
             try {
                 StartupAction start = augmentAction.createInitialRuntimeApplication();
-                runner = start.run();
+                //this is a bit yuck, but we need replace the default
+                //exit handler in the runtime class loader
+                //TODO: look at implementing a common core classloader, that removes the need for this sort of crappy hack
+                curatedApplication.getBaseRuntimeClassLoader().loadClass(ApplicationLifecycleManager.class.getName())
+                        .getMethod("setDefaultExitCodeHandler", Consumer.class)
+                        .invoke(null, new Consumer<Integer>() {
+                            @Override
+                            public void accept(Integer integer) {
+                                if (restarting || ApplicationLifecycleManager.isVmShuttingDown()
+                                        || context.isAbortOnFailedStart()) {
+                                    return;
+                                }
+                                System.out.println("Quarkus application exited with code " + integer);
+                                System.out.println("Press Enter to restart");
+                                try {
+                                    while (System.in.read() != '\n') {
+                                    }
+                                    runtimeUpdatesProcessor.checkForChangedClasses();
+                                    restartApp(runtimeUpdatesProcessor.checkForFileChange());
+                                } catch (Exception e) {
+                                    log.error("Failed to restart", e);
+                                }
+                            }
+                        });
+                runner = start.runMainClass();
             } catch (Throwable t) {
                 deploymentProblem = t;
                 if (context.isAbortOnFailedStart()) {
@@ -91,7 +117,9 @@ private synchronized void firstStart() {
     }
 
     public synchronized void restartApp(Set<String> changedResources) {
+        restarting = true;
         stop();
+        restarting = false;
         Timing.restart(curatedApplication.getAugmentClassLoader());
         deploymentProblem = null;
         ClassLoader old = Thread.currentThread().getContextClassLoader();
@@ -100,11 +128,10 @@ public synchronized void restartApp(Set<String> changedResources) {
             //ok, we have resolved all the deps
             try {
                 StartupAction start = augmentAction.reloadExistingApplication(changedResources);
-                runner = start.run();
+                runner = start.runMainClass();
             } catch (Throwable t) {
                 deploymentProblem = t;
                 log.error("Failed to start quarkus", t);
-
             }
         } finally {
             Thread.currentThread().setContextClassLoader(old);
@@ -169,6 +196,8 @@ public void stop() {
     }
 
     public void close() {
+        //don't attempt to restart in the exit code handler
+        restarting = true;
         try {
             stop();
         } finally {

core/devmode/src/main/java/io/quarkus/dev/JavaCompilationProvider.java → core/deployment/src/main/java/io/quarkus/deployment/dev/JavaCompilationProvider.java

@@ -1,4 +1,4 @@
-package io.quarkus.dev;
+package io.quarkus.deployment.dev;
 
 import java.io.File;
 import java.io.IOException;

core/deployment/src/main/java/io/quarkus/deployment/dev/LauncherMain.java

@@ -0,0 +1,38 @@
+package io.quarkus.deployment.dev;
+
+import java.io.File;
+import java.net.URL;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.Collections;
+
+import org.eclipse.microprofile.config.spi.ConfigProviderResolver;
+
+import io.smallrye.config.SmallRyeConfigProviderResolver;
+
+/**
+ * The entry point when launched from an IDE
+ */
+public class LauncherMain {
+
+    public static void main(Path appClasses, Path wiring, URL[] classPath, String... args) throws Exception {
+        DevModeContext context = new DevModeContext();
+        context.setAbortOnFailedStart(true);
+        context.setTest(false);
+        context.setCacheDir(Files.createTempDirectory("quarkus-cache").toFile());
+        context.setSourceEncoding("UTF-8");
+        File appClassesFile = appClasses.toFile();
+        context.getClassesRoots().add(appClassesFile);
+
+        //TODO: huge hacks
+        File src = new File(appClassesFile, "../../src/main/java");
+        File res = new File(appClassesFile, "../../src/main/resources");
+
+        context.getModules().add(new DevModeContext.ModuleInfo("main", new File("").getAbsolutePath(),
+                Collections.singleton(src.getAbsolutePath()), appClassesFile.getAbsolutePath(), res.getAbsolutePath()));
+        //the loading of this is super weird, and does its own class loader delegation for some reason
+        ConfigProviderResolver.setInstance(new SmallRyeConfigProviderResolver());
+        DevModeMain main = new DevModeMain(context);
+        main.start();
+    }
+}

core/devmode/src/main/java/io/quarkus/dev/RuntimeUpdatesProcessor.java → core/deployment/src/main/java/io/quarkus/deployment/dev/RuntimeUpdatesProcessor.java

@@ -1,4 +1,4 @@
-package io.quarkus.dev;
+package io.quarkus.deployment.dev;
 
 import static java.util.stream.Collectors.groupingBy;
 import static java.util.stream.Collectors.toList;

core/deployment/src/main/java/io/quarkus/deployment/pkg/PackageConfig.java

@@ -33,10 +33,15 @@ public class PackageConfig {
     public ManifestConfig manifest;
 
     /**
-     * The entry point of the application. In most cases this should not be modified.
+     * The entry point of the application. This can either be a a fully qualified name of a standard Java
+     * class with a main method, or {@link io.quarkus.runtime.QuarkusApplication}.
+     * 
+     * If your application has main classes annotated with {@link io.quarkus.runtime.annotations.QuarkusMain}
+     * then this can also reference the name given in the annotation, to avoid the need to specify fully qualified
+     * names in the config.
      */
-    @ConfigItem(defaultValue = "io.quarkus.runner.GeneratedMain")
-    public String mainClass;
+    @ConfigItem
+    public Optional<String> mainClass;
 
     /**
      * Files that should not be copied to the output artifact