App import: override service registry

Author: jknack

jooby/src/main/java/io/jooby/Jooby.java

@@ -258,6 +258,15 @@ public Jooby use(@Nonnull Router router) {
     return this;
   }
 
+  /**
+   * The underlying router.
+   *
+   * @return The underlying router.
+   */
+  public @Nonnull Router getRouter() {
+    return router;
+  }
+
   @Nonnull @Override
   public Jooby use(@Nonnull Predicate<Context> predicate, @Nonnull Router router) {
     this.router.use(predicate, router);

jooby/src/main/java/io/jooby/internal/RouterImpl.java

@@ -223,6 +223,7 @@ public RouterImpl(ClassLoader loader) {
 
   @Nonnull @Override
   public Router use(@Nonnull Predicate<Context> predicate, @Nonnull Router router) {
+    syncState(router);
     Chi tree = new Chi();
     if (predicateMap == null) {
       predicateMap = new LinkedHashMap<>();
@@ -236,7 +237,20 @@ public Router use(@Nonnull Predicate<Context> predicate, @Nonnull Router router)
     return this;
   }
 
+  private void syncState(Router router) {
+    if (router instanceof Jooby) {
+      Jooby app = (Jooby) router;
+      syncState(app.getRouter());
+    } else if (router instanceof RouterImpl) {
+      RouterImpl that = (RouterImpl) router;
+      // Inherited the services from router owner
+      // TODO: what to do with existing services? Is there anything we can do?
+      that.services = this.services;
+    }
+  }
+
   @Nonnull @Override public Router use(@Nonnull String path, @Nonnull Router router) {
+    syncState(router);
     String prefix = Router.leadingSlash(path);
     for (Route route : router.getRoutes()) {
       String routePattern = new PathBuilder(prefix, route.getPattern()).toString();

starters/quartz-starter/src/main/java/starter/App.java

@@ -20,7 +20,7 @@ public class App extends Jooby {
 
     install(new QuartzModule(SampleJob.class, BeanJob.class));
 
-    use("/quartz", new QuartzApp(require(Scheduler.class)));
+    use("/quartz", new QuartzApp());
   }
 
   public static void main(String[] args) {