From a02781a544fdf9ebf76beee315c66aff1b70ace1 Mon Sep 17 00:00:00 2001
From: Guillaume Le Floch <glfloch@gmail.com>
Date: Sat, 23 Jan 2021 13:41:44 +0100
Subject: [PATCH] Enhance dev console for hibernate-search extension

---
 .../devconsole/DevConsoleProcessor.java       | 25 +++++++++++
 .../resources/dev-templates/embedded.html     |  4 ++
 .../resources/dev-templates/entities.html     | 41 +++++++++++++++++++
 .../runtime/pom.xml                           | 10 +++++
 .../HibernateSearchDevConsoleRecorder.java    | 37 +++++++++++++++++
 .../devconsole/HibernateSearchSupplier.java   | 23 +++++++++++
 .../main/resources/dev-templates/main.html    |  4 +-
 7 files changed, 142 insertions(+), 2 deletions(-)
 create mode 100644 extensions/hibernate-search-orm-elasticsearch/deployment/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/devconsole/DevConsoleProcessor.java
 create mode 100644 extensions/hibernate-search-orm-elasticsearch/deployment/src/main/resources/dev-templates/embedded.html
 create mode 100644 extensions/hibernate-search-orm-elasticsearch/deployment/src/main/resources/dev-templates/entities.html
 create mode 100644 extensions/hibernate-search-orm-elasticsearch/runtime/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/runtime/devconsole/HibernateSearchDevConsoleRecorder.java
 create mode 100644 extensions/hibernate-search-orm-elasticsearch/runtime/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/runtime/devconsole/HibernateSearchSupplier.java

diff --git a/extensions/hibernate-search-orm-elasticsearch/deployment/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/devconsole/DevConsoleProcessor.java b/extensions/hibernate-search-orm-elasticsearch/deployment/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/devconsole/DevConsoleProcessor.java
new file mode 100644
index 0000000000000..94d1c46300e91
--- /dev/null
+++ b/extensions/hibernate-search-orm-elasticsearch/deployment/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/devconsole/DevConsoleProcessor.java
@@ -0,0 +1,25 @@
+package io.quarkus.hibernate.search.orm.elasticsearch.devconsole;
+
+import static io.quarkus.deployment.annotations.ExecutionTime.STATIC_INIT;
+
+import io.quarkus.deployment.IsDevelopment;
+import io.quarkus.deployment.annotations.BuildStep;
+import io.quarkus.deployment.annotations.Record;
+import io.quarkus.devconsole.spi.DevConsoleRouteBuildItem;
+import io.quarkus.devconsole.spi.DevConsoleRuntimeTemplateInfoBuildItem;
+import io.quarkus.hibernate.search.orm.elasticsearch.runtime.devconsole.HibernateSearchDevConsoleRecorder;
+import io.quarkus.hibernate.search.orm.elasticsearch.runtime.devconsole.HibernateSearchSupplier;
+
+public class DevConsoleProcessor {
+
+    @BuildStep(onlyIf = IsDevelopment.class)
+    public DevConsoleRuntimeTemplateInfoBuildItem collectBeanInfo() {
+        return new DevConsoleRuntimeTemplateInfoBuildItem("entities", new HibernateSearchSupplier());
+    }
+
+    @BuildStep
+    @Record(value = STATIC_INIT, optional = true)
+    DevConsoleRouteBuildItem invokeEndpoint(HibernateSearchDevConsoleRecorder recorder) {
+        return new DevConsoleRouteBuildItem("entities", "POST", recorder.indexEntity());
+    }
+}
diff --git a/extensions/hibernate-search-orm-elasticsearch/deployment/src/main/resources/dev-templates/embedded.html b/extensions/hibernate-search-orm-elasticsearch/deployment/src/main/resources/dev-templates/embedded.html
new file mode 100644
index 0000000000000..192a385b27b1a
--- /dev/null
+++ b/extensions/hibernate-search-orm-elasticsearch/deployment/src/main/resources/dev-templates/embedded.html
@@ -0,0 +1,4 @@
+<a href="{urlbase}/entities" class="badge badge-light">
+    <i class="fa fa-search fa-fw"></i>
+    Indexed entities <span class="badge badge-light">{info:entities.size()}</span></a>
+<br>
diff --git a/extensions/hibernate-search-orm-elasticsearch/deployment/src/main/resources/dev-templates/entities.html b/extensions/hibernate-search-orm-elasticsearch/deployment/src/main/resources/dev-templates/entities.html
new file mode 100644
index 0000000000000..3f81b7ba4ba15
--- /dev/null
+++ b/extensions/hibernate-search-orm-elasticsearch/deployment/src/main/resources/dev-templates/entities.html
@@ -0,0 +1,41 @@
+{#include main}
+{#title}Index Entities{/title}
+{#body}
+<form method="post" enctype="application/x-www-form-urlencoded">
+    <input id="index" type="submit" class="btn btn-primary mb-2" value="Reindex Entities" >
+    <table id="table" class="table table-striped">
+        <thead class="thead-dark">
+        <tr>
+            <th scope="col">
+                <div class="custom-control">
+                    <input type="checkbox" class="form-check-input" id="check-all">
+                </div>
+            </th>
+            <th scope="col">Entity</th>
+        </tr>
+        </thead>
+        <tbody>
+        {#for entity in info:entities}
+        <tr>
+            <td>
+                <div class="custom-control">
+                    <input type="checkbox" class="form-check-input checkbox" name="{entity}" id="{entity}">
+                </div>
+            </td>
+            <td>{entity}</td>
+        </tr>
+        {/for}
+        </tbody>
+    </table>
+</form>
+<script type="text/javascript">
+    jQuery('#check-all').change(function() {
+        if (this.checked) {
+            jQuery('.checkbox').prop('checked', true);
+        } else {
+            jQuery('.checkbox').prop('checked', false);
+        }
+    });
+</script>
+{/body}
+{/include}
diff --git a/extensions/hibernate-search-orm-elasticsearch/runtime/pom.xml b/extensions/hibernate-search-orm-elasticsearch/runtime/pom.xml
index 0cef80b35aaa5..55696ce4d6f35 100644
--- a/extensions/hibernate-search-orm-elasticsearch/runtime/pom.xml
+++ b/extensions/hibernate-search-orm-elasticsearch/runtime/pom.xml
@@ -37,6 +37,16 @@
             <groupId>org.graalvm.nativeimage</groupId>
             <artifactId>svm</artifactId>
         </dependency>
+        <dependency>
+            <groupId>io.vertx</groupId>
+            <artifactId>vertx-web</artifactId>
+            <optional>true</optional>
+        </dependency>
+        <dependency>
+            <groupId>io.quarkus</groupId>
+            <artifactId>quarkus-vertx-http</artifactId>
+            <optional>true</optional>
+        </dependency>
     </dependencies>
 
     <build>
diff --git a/extensions/hibernate-search-orm-elasticsearch/runtime/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/runtime/devconsole/HibernateSearchDevConsoleRecorder.java b/extensions/hibernate-search-orm-elasticsearch/runtime/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/runtime/devconsole/HibernateSearchDevConsoleRecorder.java
new file mode 100644
index 0000000000000..cd49ef9553ef3
--- /dev/null
+++ b/extensions/hibernate-search-orm-elasticsearch/runtime/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/runtime/devconsole/HibernateSearchDevConsoleRecorder.java
@@ -0,0 +1,37 @@
+package io.quarkus.hibernate.search.orm.elasticsearch.runtime.devconsole;
+
+import java.time.Duration;
+import java.util.stream.Collectors;
+
+import org.hibernate.search.mapper.orm.entity.SearchIndexedEntity;
+import org.hibernate.search.mapper.orm.mapping.SearchMapping;
+
+import io.quarkus.devconsole.runtime.spi.DevConsolePostHandler;
+import io.quarkus.runtime.annotations.Recorder;
+import io.vertx.core.Handler;
+import io.vertx.core.MultiMap;
+import io.vertx.ext.web.RoutingContext;
+
+@Recorder
+public class HibernateSearchDevConsoleRecorder {
+
+    public Handler<RoutingContext> indexEntity() {
+        return new DevConsolePostHandler() {
+            @Override
+            protected void handlePostAsync(RoutingContext event, MultiMap form) throws Exception {
+                if (form.isEmpty()) {
+                    return;
+                }
+                SearchMapping searchMapping = HibernateSearchSupplier.searchMapping();
+                searchMapping.scope(Object.class,
+                        searchMapping.allIndexedEntities().stream()
+                                .map(SearchIndexedEntity::jpaName)
+                                .filter(form::contains)
+                                .collect(Collectors.toList()))
+                        .massIndexer()
+                        .startAndWait();
+                flashMessage(event, "Entities successfully reindexed", Duration.ofSeconds(10));
+            }
+        };
+    }
+}
diff --git a/extensions/hibernate-search-orm-elasticsearch/runtime/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/runtime/devconsole/HibernateSearchSupplier.java b/extensions/hibernate-search-orm-elasticsearch/runtime/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/runtime/devconsole/HibernateSearchSupplier.java
new file mode 100644
index 0000000000000..4666a20a0f97f
--- /dev/null
+++ b/extensions/hibernate-search-orm-elasticsearch/runtime/src/main/java/io/quarkus/hibernate/search/orm/elasticsearch/runtime/devconsole/HibernateSearchSupplier.java
@@ -0,0 +1,23 @@
+package io.quarkus.hibernate.search.orm.elasticsearch.runtime.devconsole;
+
+import java.util.List;
+import java.util.function.Supplier;
+import java.util.stream.Collectors;
+
+import org.hibernate.search.mapper.orm.entity.SearchIndexedEntity;
+import org.hibernate.search.mapper.orm.mapping.SearchMapping;
+
+import io.quarkus.arc.Arc;
+
+public class HibernateSearchSupplier implements Supplier<List<String>> {
+    @Override
+    public List<String> get() {
+        return searchMapping().allIndexedEntities().stream().map(SearchIndexedEntity::jpaName).sorted()
+                .collect(Collectors.toList());
+
+    }
+
+    public static SearchMapping searchMapping() {
+        return Arc.container().instance(SearchMapping.class).get();
+    }
+}
diff --git a/extensions/vertx-http/deployment/src/main/resources/dev-templates/main.html b/extensions/vertx-http/deployment/src/main/resources/dev-templates/main.html
index a38578c0c4c41..92402c5f62c8d 100644
--- a/extensions/vertx-http/deployment/src/main/resources/dev-templates/main.html
+++ b/extensions/vertx-http/deployment/src/main/resources/dev-templates/main.html
@@ -25,6 +25,8 @@
   </title>
  </head>
  <body>
+  <script src="{frameworkRootPath}/dev/resources/js/jquery-3.5.1.min.js"></script>
+  <script src="{frameworkRootPath}/dev/resources/js/bootstrap.bundle.min.js"></script>
   <nav class="navbar sticky-top navbar-dark bg-dark">
     <a class="navbar-brand" href="{frameworkRootPath}/dev/">
       <img src="{frameworkRootPath}/dev/resources/images/quarkus_icon_rgb_reverse.svg" width="40" height="30" class="d-inline-block align-middle" alt="Quarkus"/>
@@ -59,8 +61,6 @@
    {/if}
    {#insert body/}
   </div>
-  <script src="{frameworkRootPath}/dev/resources/js/jquery-3.5.1.min.js"></script>
-  <script src="{frameworkRootPath}/dev/resources/js/bootstrap.bundle.min.js"></script>
   {|
   <script>
     if (jQuery(".alert").length) {