Enables AI extension integration

Adds support for AI-powered features via a new extension mechanism.
This includes dynamically loading the AI extension's CSS and JS,
and retrying the plugin mounting process with exponential backoff
to ensure proper initialization.

Also, provides a basic GRPC querying functionality which could call AI
methods, and converts the AI response to standard data format.
The AI plugin can be built from source or downloaded from a binary URL.

Author: KariHall619

cmd/testdata/stores.yaml

@@ -10,7 +10,7 @@ stores:
       kind:
         name: atest-ext-ai
         enabled: true
-      url: ""
+      url: "unix:///tmp/atest-ext-ai.sock"
       readonly: false
       disabled: false
       properties:

console/atest-ui/src/views/Extension.vue

@@ -9,33 +9,55 @@ const props = defineProps<Props>()
 const loading = ref(true)
 const loadPlugin = async (): Promise<void> => {
     try {
+        // First load CSS
         API.GetPageOfCSS(props.name, (d) => {
             const style = document.createElement('style');
             style.textContent = d.message;
             document.head.appendChild(style);
         });
 
+        // Then load JS and mount plugin
         API.GetPageOfJS(props.name, (d) => {
             const script = document.createElement('script');
             script.type = 'text/javascript';
             script.textContent = d.message;
             document.head.appendChild(script);
 
-            const plugin = window.ATestPlugin;
-            
-            if (plugin && plugin.mount) {
-                console.log('extension load success');
-                const container = document.getElementById("plugin-container");
-                if (container) {
-                    container.innerHTML = ''; // Clear previous content
-                    plugin.mount(container);
+            // Implement retry mechanism with exponential backoff
+            const checkPluginLoad = (retries = 0, maxRetries = 10) => {
+                const plugin = (window as any).ATestPlugin;
+
+                console.log(`Plugin load attempt ${retries + 1}/${maxRetries + 1}`);
+
+                if (plugin && plugin.mount) {
+                    console.log('extension load success');
+                    const container = document.getElementById("plugin-container");
+                    if (container) {
+                        container.innerHTML = ''; // Clear previous content
+                        plugin.mount(container);
+                        loading.value = false;
+                    } else {
+                        console.error('Plugin container not found');
+                        loading.value = false;
+                    }
+                } else if (retries < maxRetries) {
+                    // Incremental retry mechanism: 50ms, 100ms, 150ms...
+                    const delay = 50 + retries * 50;
+                    console.log(`ATestPlugin not ready, retrying in ${delay}ms (attempt ${retries + 1}/${maxRetries + 1})`);
+                    setTimeout(() => checkPluginLoad(retries + 1, maxRetries), delay);
+                } else {
+                    console.error('ATestPlugin not found or missing mount method after max retries');
+                    console.error('Window.ATestPlugin value:', (window as any).ATestPlugin);
+                    loading.value = false;
                 }
-            }
+            };
+
+            // Start the retry mechanism
+            checkPluginLoad();
         });
     } catch (error) {
-        console.log(`extension load error: ${(error as Error).message}`)
-    } finally {
-        console.log('extension load finally');
+        console.log(`extension load error: ${(error as Error).message}`);
+        loading.value = false; // Set loading to false on error
     }
 };
 try {

pkg/testing/remote/grpc_store.go

@@ -18,9 +18,11 @@ package remote
 
 import (
 	"context"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/linuxsuren/api-testing/pkg/logging"
@@ -316,6 +318,12 @@ func (g *gRPCLoader) PProf(name string) []byte {
 }
 
 func (g *gRPCLoader) Query(query map[string]string) (result testing.DataResult, err error) {
+	// Detect AI method calls
+	if method := query["method"]; strings.HasPrefix(method, "ai.") {
+		return g.handleAIQuery(query)
+	}
+
+	// Original standard query logic
 	var dataResult *server.DataQueryResult
 	offset, _ := strconv.ParseInt(query["offset"], 10, 64)
 	limit, _ := strconv.ParseInt(query["limit"], 10, 64)
@@ -444,3 +452,65 @@ func (g *gRPCLoader) Close() {
 		g.conn.Close()
 	}
 }
+
+// handleAIQuery handles AI-specific queries
+func (g *gRPCLoader) handleAIQuery(query map[string]string) (testing.DataResult, error) {
+	method := query["method"]
+
+	var dataQuery *server.DataQuery
+	switch method {
+	case "ai.generate":
+		dataQuery = &server.DataQuery{
+			Type: "ai",
+			Key:  "generate",
+			Sql:  g.encodeAIGenerateParams(query),
+		}
+	case "ai.capabilities":
+		dataQuery = &server.DataQuery{
+			Type: "ai",
+			Key:  "capabilities",
+			Sql:  "", // No additional parameters needed
+		}
+	default:
+		return testing.DataResult{}, fmt.Errorf("unsupported AI method: %s", method)
+	}
+
+	// Call existing gRPC Query
+	dataResult, err := g.client.Query(g.ctx, dataQuery)
+	if err != nil {
+		return testing.DataResult{}, err
+	}
+
+	// Convert response to testing.DataResult format
+	return g.convertAIResponse(dataResult), nil
+}
+
+// encodeAIGenerateParams encodes AI generation parameters into SQL field
+func (g *gRPCLoader) encodeAIGenerateParams(query map[string]string) string {
+	params := map[string]string{
+		"model":  query["model"],
+		"prompt": query["prompt"],
+		"config": query["config"],
+	}
+	data, _ := json.Marshal(params)
+	return string(data)
+}
+
+// convertAIResponse converts AI response to standard format
+func (g *gRPCLoader) convertAIResponse(dataResult *server.DataQueryResult) testing.DataResult {
+	result := testing.DataResult{
+		Pairs: pairToMap(dataResult.Data),
+	}
+
+	// Map AI-specific response fields
+	if content := result.Pairs["generated_sql"]; content != "" {
+		result.Pairs["content"] = content // Follow AI interface standard
+	}
+	if result.Pairs["error"] != "" {
+		result.Pairs["success"] = "false"
+	} else {
+		result.Pairs["success"] = "true"
+	}
+
+	return result
+}

tools/make/run.mk

@@ -5,12 +5,28 @@
 include tools/make/env.mk
 
 ATEST_UI = console/atest-ui
+AI_PLUGIN_DIR := $(or $(AI_PLUGIN_SOURCE),../atest-ext-ai)
 
 ##@ Local runs & init env
 
+.PHONY: build-ai-plugin
+build-ai-plugin:
+	@if [ -n "$(AI_PLUGIN_BINARY_URL)" ]; then \
+		echo "📥 Downloading AI plugin binary from $(AI_PLUGIN_BINARY_URL)..."; \
+		mkdir -p bin; \
+		curl -L "$(AI_PLUGIN_BINARY_URL)" | tar xz -C bin/ --strip-components=1; \
+		echo "✅ AI plugin binary downloaded"; \
+	elif [ -d "$(AI_PLUGIN_DIR)" ]; then \
+		echo "🔨 Building AI plugin from source..."; \
+		cd $(AI_PLUGIN_DIR) && make build; \
+		echo "✅ AI plugin built from source"; \
+	else \
+		echo "⚠️  AI plugin directory not found, skipping"; \
+	fi
+
 .PHONY: run-server
 run-server: ## Run the API Testing server
-run-server: build-ui run-backend
+run-server: build-ui build-ai-plugin run-backend
 run-backend:
 	go run . server --local-storage 'bin/*.yaml' --console-path ${ATEST_UI}/dist \
 		--extension-registry ghcr.io --download-timeout 10m