Introduce new assistant module

Signed-off-by: Phillip Kruger <[email protected]>

Author: phillip-kruger

bom/application/pom.xml

@@ -594,7 +594,16 @@
             </dependency>
 
             <!-- Quarkus libraries -->
-
+            <dependency>
+                <groupId>io.quarkus</groupId>
+                <artifactId>quarkus-assistant</artifactId>
+                <version>${project.version}</version>
+            </dependency>
+            <dependency>
+                <groupId>io.quarkus</groupId>
+                <artifactId>quarkus-assistant-deployment</artifactId>
+                <version>${project.version}</version>
+            </dependency>
             <dependency>
                 <groupId>io.quarkus</groupId>
                 <artifactId>quarkus-caffeine</artifactId>

bom/dev-ui/pom.xml

@@ -33,6 +33,7 @@
         <codeblock.version>1.1.1</codeblock.version>
         <qomponent.version>1.0.4</qomponent.version>
         <directory-tree.version>1.0.3</directory-tree.version>
+        <dot.version>0.0.1</dot.version>
         <hpcc-js-wasm.version>2.15.3</hpcc-js-wasm.version>
         <yargs.version>17.7.2</yargs.version>
         <cliui.version>8.0.1</cliui.version>
@@ -50,6 +51,7 @@
         <ansi-styles.version>4.3.0</ansi-styles.version>
         <color-convert.version>2.0.1</color-convert.version>
         <color-name.version>1.1.4</color-name.version>
+        <viz-js.version>3.12.0</viz-js.version>
     </properties>
 
     <dependencyManagement>
@@ -277,6 +279,14 @@
                 <scope>runtime</scope>
             </dependency>
 
+            <!-- Dot files render -->
+            <dependency>
+                <groupId>org.mvnpm.at.qomponent</groupId>
+                <artifactId>qui-dot</artifactId>
+                <version>${dot.version}</version>
+                <scope>runtime</scope>
+            </dependency>
+            
             <!-- Markdown render -->
             <dependency>
                 <groupId>org.mvnpm</groupId>
@@ -335,6 +345,14 @@
                 <version>${es-module-shims.version}</version>
                 <scope>runtime</scope>
             </dependency>
+            
+            <!-- Viz.js to render dot file -->
+            <dependency>
+                <groupId>org.mvnpm.at.viz-js</groupId>
+                <artifactId>viz</artifactId>
+                <version>${viz-js.version}</version>
+                <scope>runtime</scope>
+            </dependency>
         </dependencies>
     </dependencyManagement>
 </project>

devtools/bom-descriptor-json/pom.xml

@@ -226,6 +226,19 @@
                         </exclusion>
                     </exclusions>
                 </dependency>
+                <dependency>
+                    <groupId>io.quarkus</groupId>
+                    <artifactId>quarkus-assistant</artifactId>
+                    <version>${project.version}</version>
+                    <type>pom</type>
+                    <scope>test</scope>
+                    <exclusions>
+                        <exclusion>
+                            <groupId>*</groupId>
+                            <artifactId>*</artifactId>
+                        </exclusion>
+                    </exclusions>
+                </dependency>
                 <dependency>
                     <groupId>io.quarkus</groupId>
                     <artifactId>quarkus-avro</artifactId>

docs/pom.xml

@@ -237,6 +237,19 @@
                 </exclusion>
             </exclusions>
         </dependency>
+        <dependency>
+            <groupId>io.quarkus</groupId>
+            <artifactId>quarkus-assistant-deployment</artifactId>
+            <version>${project.version}</version>
+            <type>pom</type>
+            <scope>test</scope>
+            <exclusions>
+                <exclusion>
+                    <groupId>*</groupId>
+                    <artifactId>*</artifactId>
+                </exclusion>
+            </exclusions>
+        </dependency>
         <dependency>
             <groupId>io.quarkus</groupId>
             <artifactId>quarkus-avro-deployment</artifactId>

extensions/agroal/deployment/pom.xml

@@ -25,6 +25,10 @@
             <groupId>io.quarkus</groupId>
             <artifactId>quarkus-datasource-deployment</artifactId>
         </dependency>
+        <dependency>
+            <groupId>io.quarkus</groupId>
+            <artifactId>quarkus-assistant-deployment</artifactId>
+        </dependency>
         <dependency>
             <groupId>io.quarkus</groupId>
             <artifactId>quarkus-agroal</artifactId>

extensions/agroal/deployment/src/main/java/io/quarkus/agroal/deployment/AgroalProcessor.java

@@ -73,7 +73,6 @@ class AgroalProcessor {
 
     private static final Logger log = Logger.getLogger(AgroalProcessor.class);
 
-    private static final String OPEN_TELEMETRY_DRIVER = "io.opentelemetry.instrumentation.jdbc.OpenTelemetryDriver";
     private static final DotName DATA_SOURCE = DotName.createSimple(javax.sql.DataSource.class.getName());
     private static final DotName AGROAL_DATA_SOURCE = DotName.createSimple(AgroalDataSource.class.getName());
 

extensions/agroal/deployment/src/main/java/io/quarkus/agroal/deployment/devui/AgroalDevUIProcessor.java

@@ -1,14 +1,19 @@
 package io.quarkus.agroal.deployment.devui;
 
+import java.util.Optional;
+
 import io.quarkus.agroal.runtime.DataSourcesJdbcBuildTimeConfig;
 import io.quarkus.agroal.runtime.dev.ui.DatabaseInspector;
+import io.quarkus.assistant.deployment.Assistant;
+import io.quarkus.assistant.deployment.AssistantBuildItem;
 import io.quarkus.deployment.IsLocalDevelopment;
 import io.quarkus.deployment.annotations.BuildProducer;
 import io.quarkus.deployment.annotations.BuildStep;
 import io.quarkus.deployment.annotations.BuildSteps;
 import io.quarkus.deployment.builditem.LaunchModeBuildItem;
 import io.quarkus.dev.spi.DevModeType;
 import io.quarkus.devui.spi.JsonRPCProvidersBuildItem;
+import io.quarkus.devui.spi.buildtime.BuildTimeActionBuildItem;
 import io.quarkus.devui.spi.page.CardPageBuildItem;
 import io.quarkus.devui.spi.page.Page;
 
@@ -37,8 +42,35 @@ void devUI(DataSourcesJdbcBuildTimeConfig config,
         }
     }
 
+    @BuildStep
+    void createBuildTimeActions(Optional<AssistantBuildItem> assistantBuildItem,
+            BuildProducer<BuildTimeActionBuildItem> buildTimeActionProducer) {
+
+        if (assistantBuildItem.isPresent()) {
+            BuildTimeActionBuildItem bta = new BuildTimeActionBuildItem();
+
+            Assistant assistant = assistantBuildItem.get().getAssistant();
+
+            // TODO: If currentInsertScript is empty, maybe send tables schema
+
+            bta.addAction("generateMoreData", params -> {
+                return assistant.assist(USER_MESSAGE, params);
+            });
+
+            buildTimeActionProducer.produce(bta);
+        }
+    }
+
     @BuildStep
     JsonRPCProvidersBuildItem createJsonRPCService() {
         return new JsonRPCProvidersBuildItem(DatabaseInspector.class);
     }
+
+    private static final String USER_MESSAGE = """
+            Given the provided sql script:
+            {{currentInsertScript}}
+            Can you add 10 more inserts into the script and return the result
+            (including the provided entries, so update the script)
+            Return the result in a field called `script`.
+            """;
 }