feat(platform): 插件协议系统 - 支持声明式渠道、工作流与参考媒体映射

2026-08-26 13:11:10 +08:00
parent 054e3bbbff
commit 2f3a813ccb
58 changed files with 2074 additions and 706 deletions
--- a/backend/internal/service/protocol_runtime_test.go
+++ b/backend/internal/service/protocol_runtime_test.go
@@ -1,9 +1,14 @@
 package service

 import (
+	"archive/zip"
+	"bytes"
 	"context"
+	"encoding/json"
 	"net/http"
 	"net/http/httptest"
+	"os"
+	"path/filepath"
 	"strings"
 	"testing"

@@ -23,7 +28,7 @@ func TestPluginViewIncludesDocumentationForEveryBundledProtocol(t *testing.T) {
 		if plugin.Source != "bundled" {
 			continue
 		}
-		document := strings.TrimSpace(plugin.Manifest.Protocol.Documentation)
+		document := strings.TrimSpace(plugin.Manifest.Documentation)
 		if document == "" || !strings.Contains(document, "## 影策运行时合同") {
 			t.Errorf("bundled plugin %q has no complete documentation in PluginView", plugin.Manifest.ID)
 		}
@@ -37,18 +42,23 @@ func TestPluginRuntimeIsTheProtocolSourceOfTruth(t *testing.T) {
 		t.Fatal(err)
 	}
 	if got := center.registrySnapshot().List("", "", false); len(got) == 0 {
-		t.Fatal("bundled protocols were not reconciled into a new plugin directory")
+		t.Fatal("bundled providers were not reconciled into the unified plugin registry")
 	}
-	manifest := []byte(`{"apiVersion":"v1","metadata":{"id":"uploaded-runtime","version":"1.0.0","name":"Uploaded Runtime","vendor":"Test","categories":["video"],"scopes":["canvas"],"documentation":"# Uploaded Runtime"},"create":{"method":"POST","path":"/tasks","fields":{"prompt":"request.prompt"}},"response":{"statusPaths":["status"]}}`)
-	plugin, err := center.install(manifest)
+	if _, err := center.install([]byte(`{"apiVersion":"yingce.plugin/v1"}`), "legacy.json"); err == nil {
+		t.Fatal("bare JSON manifest was accepted by the upload runtime")
+	}
+	manifest := []byte(`{"apiVersion":"yingce.plugin/v1","id":"uploaded-runtime","version":"1.0.0","name":"Uploaded Runtime","author":"Test","documentation":"# Uploaded Runtime","contributes":{"providers":[{"id":"uploaded-runtime","label":"Uploaded Runtime","capabilities":["video"],"scopes":["canvas"],"create":{"method":"POST","path":"/tasks","fields":{"prompt":"request.prompt"}},"response":{"statusPaths":["status"]}}]}}`)
+	plugin, err := center.install(testPluginPackage(t, manifest), "uploaded-runtime.yingce-plugin")
 	if err != nil {
 		t.Fatal(err)
 	}
 	if plugin.Status != "enabled" || !center.registrySnapshot().IsCapability("uploaded-runtime", protocol.CapabilityVideo) {
 		t.Fatalf("installed plugin was not activated: %#v", plugin)
 	}
-	if _, err := center.install(manifest); err == nil {
-		t.Fatal("duplicate plugin id overwrote an installed plugin")
+	updatedManifest := []byte(`{"apiVersion":"yingce.plugin/v1","id":"uploaded-runtime","version":"2.0.0","name":"Uploaded Runtime v2","author":"Test","documentation":"# Uploaded Runtime v2","contributes":{"providers":[{"id":"uploaded-runtime","label":"Uploaded Runtime v2","capabilities":["video"],"scopes":["canvas"],"create":{"method":"POST","path":"/tasks","fields":{"prompt":"request.prompt"}},"response":{"statusPaths":["status"]}}]}}`)
+	updated, err := center.install(testPluginPackage(t, updatedManifest), "uploaded-runtime-v2.yingce-plugin")
+	if err != nil || updated.Manifest.Version != "2.0.0" || updated.Manifest.Name != "Uploaded Runtime v2" {
+		t.Fatalf("plugin update = %#v, err = %v", updated, err)
 	}
 	if _, err := center.setEnabled("uploaded-runtime", false); err != nil {
 		t.Fatal(err)
@@ -67,19 +77,77 @@ func TestPluginRuntimeIsTheProtocolSourceOfTruth(t *testing.T) {
 	}
 }

+func TestPluginRuntimeDropsRemovedBundledProtocol(t *testing.T) {
+	data, err := os.ReadFile(filepath.Join("..", "..", "..", "plugin-packages", "autodl-comfyui.yingce-plugin"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	pkg, err := protocol.ParsePluginPackage(data)
+	if err != nil {
+		t.Fatal(err)
+	}
+	registryData, err := json.Marshal([]pluginRegistryRecord{{ID: "autodl-comfyui", Raw: pkg.ManifestRaw, Source: "bundled"}})
+	if err != nil {
+		t.Fatal(err)
+	}
+	dataDir := t.TempDir()
+	if err := os.WriteFile(filepath.Join(dataDir, "plugin_registry.json"), registryData, 0o600); err != nil {
+		t.Fatal(err)
+	}
+	center, err := newPluginRuntime(dataDir)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if _, ok := center.registrySnapshot().Resolve("autodl-comfyui"); ok {
+		t.Fatal("removed bundled AutoDL protocol survived bootstrap")
+	}
+}
+
+func TestAutoDLPluginPackageInstallsThroughUploadRuntime(t *testing.T) {
+	data, err := os.ReadFile(filepath.Join("..", "..", "..", "plugin-packages", "autodl-comfyui.yingce-plugin"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	center, err := newPluginRuntime(t.TempDir())
+	if err != nil {
+		t.Fatal(err)
+	}
+	plugin, err := center.install(data, "autodl-comfyui.yingce-plugin")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if plugin.Status != "enabled" || plugin.Manifest.ID != "autodl-comfyui" || plugin.Package != protocol.PluginPackageFormat {
+		t.Fatalf("installed AutoDL plugin = %#v", plugin)
+	}
+	if !center.registrySnapshot().IsCapability("autodl-comfyui", protocol.CapabilityVideo) {
+		t.Fatal("AutoDL package provider was not registered")
+	}
+	catalog := (&Service{pluginRuntime: center}).PluginProviderCatalog(string(protocol.SurfaceAdminSystemChannel), string(protocol.CapabilityVideo), false)
+	var autoDL *PluginProviderCatalogItem
+	for index := range catalog {
+		if catalog[index].ID == "autodl-comfyui" {
+			autoDL = &catalog[index]
+		}
+		if catalog[index].ID == "autodl-comfyui-plugin" {
+			t.Fatal("legacy AutoDL provider ID leaked into administrator catalog")
+		}
+	}
+	if autoDL == nil || len(autoDL.Workflows) != 1 {
+		t.Fatalf("AutoDL administrator catalog = %#v", catalog)
+	}
+}
+
 func TestDeclarativeProtocolRuntimeExecutesCreatePollAndDownload(t *testing.T) {
 	manifest := []byte(`{
-		"apiVersion":"v1",
-			"metadata":{"id":"test-declarative-video-runtime","version":"1.0.0","name":"Test Declarative Video","vendor":"Test","categories":["video"],"scopes":["canvas"],"documentation":"# Test Declarative Video"},
-		"create":{"method":"POST","path":"/tasks","fields":{"model":"request.model","prompt":"request.prompt","seconds":"request.duration"}},
-		"poll":{"method":"GET","path":"/tasks/{{taskId}}"},
-		"response":{"taskIdPaths":["id"],"statusPaths":["status"],"resultUrlPaths":["video_url"],"resultKind":"video"}
+		"apiVersion":"yingce.plugin/v1",
+			"id":"test-declarative-video-runtime","version":"1.0.0","name":"Test Declarative Video","author":"Test","documentation":"# Test Declarative Video",
+		"contributes":{"providers":[{"id":"test-declarative-video-runtime","label":"Test Declarative Video","capabilities":["video"],"scopes":["canvas"],"create":{"method":"POST","path":"/tasks","fields":{"model":"request.model","prompt":"request.prompt","seconds":"request.duration"}},"poll":{"method":"GET","path":"/tasks/{{taskId}}"},"response":{"taskIdPaths":["id"],"statusPaths":["status"],"resultPaths":["video_url"],"resultKind":"video"}}]}
 	}`)
 	center, err := newPluginRuntime(t.TempDir())
 	if err != nil {
 		t.Fatal(err)
 	}
-	if _, err := center.install(manifest); err != nil {
+	if _, err := center.install(testPluginPackage(t, manifest), "test-declarative-video-runtime.yingce-plugin"); err != nil {
 		t.Fatal(err)
 	}

@@ -112,3 +180,63 @@ func TestDeclarativeProtocolRuntimeExecutesCreatePollAndDownload(t *testing.T) {
 		t.Fatalf("result = %#v", result)
 	}
 }
+
+func TestDeclarativeProtocolRuntimeMapsReferenceImageURL(t *testing.T) {
+	manifest := []byte(`{
+		"apiVersion":"yingce.plugin/v1",
+		"id":"test-declarative-reference-image-runtime","version":"1.0.0","name":"Test Declarative Reference Image","author":"Test","documentation":"# Test Declarative Reference Image",
+		"contributes":{"providers":[{"id":"test-declarative-reference-image-runtime","label":"Test Declarative Reference Image","capabilities":["video"],"scopes":["canvas"],"create":{"method":"POST","path":"/tasks","fields":{"prompt":"request.prompt","ref_image_0":"request.images.0.url"}},"response":{"statusPaths":["status"],"messagePaths":["msg"]}}]}
+	}`)
+	center, err := newPluginRuntime(t.TempDir())
+	if err != nil {
+		t.Fatal(err)
+	}
+	if _, err := center.install(testPluginPackage(t, manifest), "test-declarative-reference-image-runtime.yingce-plugin"); err != nil {
+		t.Fatal(err)
+	}
+
+	var createBody map[string]any
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path != "/v1/tasks" {
+			http.NotFound(w, r)
+			return
+		}
+		if err := json.NewDecoder(r.Body).Decode(&createBody); err != nil {
+			t.Errorf("decode create body: %v", err)
+		}
+		w.Header().Set("Content-Type", "application/json")
+		_, _ = w.Write([]byte(`{"status":"failed","msg":"stop after request capture"}`))
+	}))
+	defer server.Close()
+
+	config := providerConfig{BaseURL: server.URL + "/v1", APIKey: "key", Model: "test-model", APIFormat: "openai", InterfaceType: "test-declarative-reference-image-runtime", AllowLocalChannel: true}
+	ctx := withProviderOutboundPolicy(context.Background(), config)
+	ctx = withProtocolRegistry(ctx, center.registrySnapshot())
+	_, err = runDeclarativeProtocolTask(ctx, canvasGenerationInput{
+		Mode: "video", Prompt: "a clip", Config: config,
+		ReferenceImages: []providerMedia{{URL: "https://cdn.example/reference.png"}},
+	})
+	if err == nil || !strings.Contains(err.Error(), "stop after request capture") {
+		t.Fatalf("runDeclarativeProtocolTask() error = %v", err)
+	}
+	if createBody["prompt"] != "a clip" || createBody["ref_image_0"] != "https://cdn.example/reference.png" {
+		t.Fatalf("create body = %#v", createBody)
+	}
+}
+
+func testPluginPackage(t *testing.T, manifest []byte) []byte {
+	t.Helper()
+	var buffer bytes.Buffer
+	writer := zip.NewWriter(&buffer)
+	file, err := writer.Create("manifest.json")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if _, err := file.Write(manifest); err != nil {
+		t.Fatal(err)
+	}
+	if err := writer.Close(); err != nil {
+		t.Fatal(err)
+	}
+	return buffer.Bytes()
+}