feat: add AI workflow editor

internal/ai/application/runtime/tool_catalog_test.go

@@ -1,10 +1,19 @@
 package runtime
 
 import (
+	"encoding/json"
+	"strings"
 	"testing"
 
+	"agent-desk/internal/ai/workflow/dsl"
+	workflowregistry "agent-desk/internal/ai/workflow/registry"
 	"agent-desk/internal/models"
+	"agent-desk/internal/pkg/enums"
 	"agent-desk/internal/pkg/toolx"
+
+	"github.com/glebarez/sqlite"
+	"github.com/mlogclub/simple/sqls"
+	"gorm.io/gorm"
 )
 
 func TestNormalizeAllowedToolCodes(t *testing.T) {
@@ -60,3 +69,91 @@ func TestBuildRuntimeStaticTools(t *testing.T) {
 		t.Fatalf("expected %d runtime static tools, got %d", len(toolx.ListRuntimeStaticToolSpecs()), len(ret))
 	}
 }
+
+func TestToolCatalogIncludesPublishedWorkflowGraphTools(t *testing.T) {
+	setupWorkflowRuntimeTestDB(t)
+	version := createWorkflowRuntimeTestVersion(t, dsl.Definition{
+		SchemaVersion: 1,
+		EntryNodeID:   "start",
+		Nodes: []dsl.Node{
+			{ID: "start", Type: workflowregistry.NodeTypeStart, Name: "Start"},
+			{ID: "draft", Type: workflowregistry.NodeTypePrepareTicketDraft, Name: "Draft Ticket"},
+			{ID: "create", Type: workflowregistry.NodeTypeCreateTicket, Name: "Create Ticket"},
+			{ID: "handoff", Type: workflowregistry.NodeTypeHandoffToHuman, Name: "Handoff"},
+		},
+	})
+
+	catalog := newToolCatalog()
+	ret := catalog.parseAgentAllowedToolCodes(models.AIAgent{
+		RuntimeMode:       enums.AIAgentRuntimeModeWorkflow,
+		WorkflowVersionID: version.ID,
+	})
+
+	assertContainsToolCode(t, ret, toolx.GraphPrepareTicketDraft.Code)
+	assertContainsToolCode(t, ret, toolx.GraphCreateTicketConfirm.Code)
+	assertContainsToolCode(t, ret, toolx.GraphHandoffConversation.Code)
+}
+
+func TestApplyWorkflowInstructionAppendsPublishedWorkflow(t *testing.T) {
+	setupWorkflowRuntimeTestDB(t)
+	version := createWorkflowRuntimeTestVersion(t, dsl.Definition{
+		SchemaVersion: 1,
+		EntryNodeID:   "start",
+		Nodes: []dsl.Node{
+			{ID: "start", Type: workflowregistry.NodeTypeStart, Name: "Start"},
+			{ID: "handoff", Type: workflowregistry.NodeTypeHandoffToHuman, Name: "Handoff"},
+		},
+	})
+
+	agent := applyWorkflowInstruction(models.AIAgent{
+		SystemPrompt:      "Base prompt.",
+		RuntimeMode:       enums.AIAgentRuntimeModeWorkflow,
+		WorkflowVersionID: version.ID,
+	})
+	if agent.SystemPrompt == "Base prompt." {
+		t.Fatalf("expected workflow appendix to be appended")
+	}
+	if !strings.Contains(agent.SystemPrompt, "Published customer-service workflow") {
+		t.Fatalf("missing workflow appendix: %s", agent.SystemPrompt)
+	}
+}
+
+func setupWorkflowRuntimeTestDB(t *testing.T) {
+	t.Helper()
+	db, err := gorm.Open(sqlite.Open("file:"+t.Name()+"?mode=memory&cache=shared"), &gorm.Config{})
+	if err != nil {
+		t.Fatalf("open sqlite db: %v", err)
+	}
+	if err := db.AutoMigrate(&models.AIWorkflowVersion{}); err != nil {
+		t.Fatalf("auto migrate: %v", err)
+	}
+	sqls.SetDB(db)
+}
+
+func createWorkflowRuntimeTestVersion(t *testing.T, def dsl.Definition) *models.AIWorkflowVersion {
+	t.Helper()
+	definition, err := json.Marshal(def)
+	if err != nil {
+		t.Fatalf("marshal definition: %v", err)
+	}
+	version := &models.AIWorkflowVersion{
+		WorkflowID: 1,
+		Version:    1,
+		Status:     enums.StatusOk,
+		Definition: string(definition),
+	}
+	if err := sqls.DB().Create(version).Error; err != nil {
+		t.Fatalf("create workflow version: %v", err)
+	}
+	return version
+}
+
+func assertContainsToolCode(t *testing.T, items []string, want string) {
+	t.Helper()
+	for _, item := range items {
+		if item == want {
+			return
+		}
+	}
+	t.Fatalf("expected tool code %s in %#v", want, items)
+}