refactor(*): 平台架构 - 拆分后端领域模块并同步插件协议与模型能力
This commit is contained in:
ddCat committed
1 parent
916043fb5a
commit
6e5cbc8b05
644 files changed
+62731
-2320
No files matched your search
@@ -0,0 +1,318 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"infinite-canvas/backend/internal/kernel"
|
||||
"infinite-canvas/backend/internal/model"
|
||||
)
|
||||
|
||||
// TaskSummary 是任务列表/会话详情的读模型，不直接复用数据库 Task，避免把
|
||||
// 渠道模型、供应线路和受保护输入泄露到普通用户接口。
|
||||
type TaskSummary struct {
|
||||
ID string `json:"id"`
|
||||
SessionID string `json:"sessionId,omitempty"`
|
||||
ProjectID string `json:"projectId,omitempty"`
|
||||
Type string `json:"type"`
|
||||
Status model.TaskStatus `json:"status"`
|
||||
Stage string `json:"stage"`
|
||||
Progress int `json:"progress"`
|
||||
Prompt string `json:"prompt"`
|
||||
Operation string `json:"operation,omitempty"`
|
||||
Provider string `json:"provider,omitempty"`
|
||||
Model string `json:"model,omitempty"`
|
||||
ProviderRequestID string `json:"providerRequestId,omitempty"`
|
||||
ProviderCancelStatus model.ProviderCancelStatus `json:"providerCancelStatus,omitempty"`
|
||||
ProviderCancelError string `json:"providerCancelError,omitempty"`
|
||||
ProviderCancelAttempts int `json:"providerCancelAttempts,omitempty"`
|
||||
ProviderCancelRequestedAt *time.Time `json:"providerCancelRequestedAt,omitempty"`
|
||||
ProviderCancelledAt *time.Time `json:"providerCancelledAt,omitempty"`
|
||||
ErrorCode string `json:"errorCode,omitempty"`
|
||||
PreviewURL string `json:"previewUrl,omitempty"`
|
||||
PreviewKind string `json:"previewKind,omitempty"`
|
||||
PreviewPosterURL string `json:"previewPosterUrl,omitempty"`
|
||||
Attempts int `json:"attempts"`
|
||||
StartedAt *time.Time `json:"startedAt"`
|
||||
CompletedAt *time.Time `json:"completedAt"`
|
||||
CreatedAt time.Time `json:"createdAt"`
|
||||
UpdatedAt time.Time `json:"updatedAt"`
|
||||
Billing *TaskBillingSummary `json:"billing,omitempty"`
|
||||
ClientContext *TaskClientContext `json:"clientContext,omitempty"`
|
||||
}
|
||||
|
||||
type TaskClientContext struct {
|
||||
NodeID string `json:"nodeId,omitempty"`
|
||||
ConversationID string `json:"conversationId,omitempty"`
|
||||
MessageID string `json:"messageId,omitempty"`
|
||||
BatchIndex int `json:"batchIndex,omitempty"`
|
||||
BatchCount int `json:"batchCount,omitempty"`
|
||||
DomainProjectID string `json:"domainProjectId,omitempty"`
|
||||
ChapterID string `json:"chapterId,omitempty"`
|
||||
ChapterOperation string `json:"chapterOperation,omitempty"`
|
||||
ShotID string `json:"shotId,omitempty"`
|
||||
WorkflowStepID string `json:"workflowStepId,omitempty"`
|
||||
ArtifactType string `json:"artifactType,omitempty"`
|
||||
}
|
||||
|
||||
type TaskBillingSummary struct {
|
||||
AmountMicrocredits int64 `json:"amountMicrocredits"`
|
||||
Status model.BillingStatus `json:"status"`
|
||||
}
|
||||
|
||||
func taskSummariesForOutput(tasks []model.Task) []TaskSummary {
|
||||
return taskSummariesForOutputWithBilling(tasks, nil)
|
||||
}
|
||||
|
||||
func taskSummariesForOutputWithBilling(tasks []model.Task, orders map[string]model.BillingOrder) []TaskSummary {
|
||||
result := make([]TaskSummary, 0, len(tasks))
|
||||
for _, task := range tasks {
|
||||
summary := taskSummaryForOutput(task)
|
||||
if order, ok := orders[task.ID]; ok {
|
||||
summary.Billing = &TaskBillingSummary{AmountMicrocredits: order.AmountMicrocredits, Status: order.Status}
|
||||
if summary.ProviderRequestID == "" {
|
||||
summary.ProviderRequestID = order.ProviderRequestID
|
||||
}
|
||||
}
|
||||
result = append(result, summary)
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func taskBillingTaskIDs(tasks []model.Task) []string {
|
||||
ids := make([]string, 0, len(tasks))
|
||||
seen := map[string]struct{}{}
|
||||
for _, task := range tasks {
|
||||
if task.BillingOrderID == "" {
|
||||
continue
|
||||
}
|
||||
if _, ok := seen[task.ID]; ok {
|
||||
continue
|
||||
}
|
||||
seen[task.ID] = struct{}{}
|
||||
ids = append(ids, task.ID)
|
||||
}
|
||||
return ids
|
||||
}
|
||||
|
||||
func taskSummaryForOutput(task model.Task) TaskSummary {
|
||||
errorCode := ""
|
||||
if isContentModerationFailure(task.Error) {
|
||||
errorCode = contentModerationErrorCode
|
||||
}
|
||||
previewURL, previewKind, previewPosterURL := taskMediaPreviewWithPoster(task.ResultJSON, task.Type)
|
||||
return TaskSummary{
|
||||
ID: task.ID,
|
||||
SessionID: task.SessionID,
|
||||
ProjectID: task.ProjectID,
|
||||
Type: task.Type,
|
||||
Status: task.Status,
|
||||
Stage: task.Stage,
|
||||
Progress: task.Progress,
|
||||
Prompt: truncateRunes(task.Prompt, 500),
|
||||
Operation: task.Operation,
|
||||
Provider: task.Provider,
|
||||
Model: task.Model,
|
||||
ProviderRequestID: task.ProviderRequestID,
|
||||
ProviderCancelStatus: task.ProviderCancelStatus,
|
||||
ProviderCancelError: task.ProviderCancelError,
|
||||
ProviderCancelAttempts: task.ProviderCancelAttempts,
|
||||
ProviderCancelRequestedAt: task.ProviderCancelRequestedAt,
|
||||
ProviderCancelledAt: task.ProviderCancelledAt,
|
||||
ErrorCode: errorCode,
|
||||
PreviewURL: previewURL,
|
||||
PreviewKind: previewKind,
|
||||
PreviewPosterURL: previewPosterURL,
|
||||
Attempts: task.Attempts,
|
||||
StartedAt: task.StartedAt,
|
||||
CompletedAt: task.CompletedAt,
|
||||
CreatedAt: task.CreatedAt,
|
||||
UpdatedAt: task.UpdatedAt,
|
||||
ClientContext: taskClientContext(task.InputJSON),
|
||||
}
|
||||
}
|
||||
|
||||
// 列表只暴露页面恢复所需的非敏感关联 ID，不下发完整任务输入或其他 metadata。
|
||||
func taskClientContext(raw string) *TaskClientContext {
|
||||
var input struct {
|
||||
Metadata struct {
|
||||
Source string `json:"source"`
|
||||
NodeID string `json:"nodeId"`
|
||||
ConversationID string `json:"conversationId"`
|
||||
MessageID string `json:"messageId"`
|
||||
BatchIndex int `json:"batchIndex"`
|
||||
BatchCount int `json:"batchCount"`
|
||||
DomainProjectID string `json:"domainProjectId"`
|
||||
ChapterID string `json:"chapterId"`
|
||||
Operation string `json:"operation"`
|
||||
ShotID string `json:"shotId"`
|
||||
WorkflowStepID string `json:"workflowStepId"`
|
||||
ArtifactType string `json:"artifactType"`
|
||||
} `json:"metadata"`
|
||||
}
|
||||
if json.Unmarshal([]byte(raw), &input) != nil {
|
||||
return nil
|
||||
}
|
||||
metadata := input.Metadata
|
||||
context := &TaskClientContext{NodeID: metadata.NodeID}
|
||||
if metadata.Source == "create-page" && metadata.ConversationID != "" && metadata.MessageID != "" {
|
||||
context.ConversationID = metadata.ConversationID
|
||||
context.MessageID = metadata.MessageID
|
||||
context.BatchIndex = metadata.BatchIndex
|
||||
context.BatchCount = metadata.BatchCount
|
||||
return context
|
||||
}
|
||||
if metadata.ShotID != "" && metadata.WorkflowStepID != "" {
|
||||
context.DomainProjectID = metadata.DomainProjectID
|
||||
context.ShotID = metadata.ShotID
|
||||
context.WorkflowStepID = metadata.WorkflowStepID
|
||||
context.ArtifactType = metadata.ArtifactType
|
||||
return context
|
||||
}
|
||||
chapterOperation := ""
|
||||
if metadata.Operation == "chapter_character_breakdown" {
|
||||
chapterOperation = "characters"
|
||||
} else if metadata.Source == "short-drama-chapter-storyboard" {
|
||||
chapterOperation = "storyboard"
|
||||
}
|
||||
if chapterOperation == "" || metadata.DomainProjectID == "" || metadata.ChapterID == "" {
|
||||
if context.NodeID == "" {
|
||||
return nil
|
||||
}
|
||||
return context
|
||||
}
|
||||
context.DomainProjectID = metadata.DomainProjectID
|
||||
context.ChapterID = metadata.ChapterID
|
||||
context.ChapterOperation = chapterOperation
|
||||
return context
|
||||
}
|
||||
|
||||
// 列表只暴露首个可访问媒体地址，避免把完整生成结果和内嵌数据带回前端。
|
||||
func taskMediaPreview(raw string, taskType string) (string, string) {
|
||||
previewURL, previewKind, _ := taskMediaPreviewWithPoster(raw, taskType)
|
||||
return previewURL, previewKind
|
||||
}
|
||||
|
||||
func taskMediaPreviewWithPoster(raw string, taskType string) (string, string, string) {
|
||||
if strings.TrimSpace(raw) == "" {
|
||||
return "", "", ""
|
||||
}
|
||||
var payload any
|
||||
if json.Unmarshal([]byte(raw), &payload) != nil {
|
||||
return "", "", ""
|
||||
}
|
||||
defaultKind := "image"
|
||||
if strings.Contains(strings.ToLower(taskType), "video") {
|
||||
defaultKind = "video"
|
||||
}
|
||||
previewURL, previewKind := findTaskMediaPreview(payload, defaultKind)
|
||||
posterURL := findTaskMediaPoster(payload)
|
||||
if previewKind == "image" && posterURL == "" {
|
||||
posterURL = previewURL
|
||||
}
|
||||
return previewURL, previewKind, posterURL
|
||||
}
|
||||
|
||||
func findTaskMediaPoster(value any) string {
|
||||
switch item := value.(type) {
|
||||
case []any:
|
||||
for _, child := range item {
|
||||
if posterURL := findTaskMediaPoster(child); posterURL != "" {
|
||||
return posterURL
|
||||
}
|
||||
}
|
||||
case map[string]any:
|
||||
for _, key := range []string{"posterUrl", "posterURL", "thumbnailUrl", "thumbnailURL", "coverUrl", "coverURL", "poster", "thumbnail", "cover"} {
|
||||
child, exists := item[key]
|
||||
if !exists {
|
||||
continue
|
||||
}
|
||||
if previewURL, previewKind := findTaskMediaPreview(child, "image"); previewURL != "" && previewKind == "image" {
|
||||
return previewURL
|
||||
}
|
||||
}
|
||||
for _, child := range item {
|
||||
if posterURL := findTaskMediaPoster(child); posterURL != "" {
|
||||
return posterURL
|
||||
}
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func findTaskMediaPreview(value any, hint string) (string, string) {
|
||||
switch item := value.(type) {
|
||||
case string:
|
||||
text := strings.TrimSpace(item)
|
||||
if !isResourceFileURL(text) && !strings.HasPrefix(text, "http://") && !strings.HasPrefix(text, "https://") {
|
||||
return "", ""
|
||||
}
|
||||
kind := hint
|
||||
lower := strings.ToLower(text)
|
||||
if strings.Contains(lower, ".mp4") || strings.Contains(lower, ".webm") || strings.Contains(lower, ".mov") {
|
||||
kind = "video"
|
||||
} else if kind != "video" {
|
||||
kind = "image"
|
||||
}
|
||||
return text, kind
|
||||
case []any:
|
||||
for _, child := range item {
|
||||
if previewURL, previewKind := findTaskMediaPreview(child, hint); previewURL != "" {
|
||||
return previewURL, previewKind
|
||||
}
|
||||
}
|
||||
case map[string]any:
|
||||
for _, key := range []string{"images", "image", "video", "dataUrl", "url", "resultUrl", "outputUrl"} {
|
||||
child, exists := item[key]
|
||||
if !exists {
|
||||
continue
|
||||
}
|
||||
childHint := hint
|
||||
if key == "video" {
|
||||
childHint = "video"
|
||||
} else if key == "images" || key == "image" {
|
||||
childHint = "image"
|
||||
}
|
||||
if previewURL, previewKind := findTaskMediaPreview(child, childHint); previewURL != "" {
|
||||
return previewURL, previewKind
|
||||
}
|
||||
}
|
||||
}
|
||||
return "", ""
|
||||
}
|
||||
|
||||
func truncateRunes(value string, limit int) string {
|
||||
return kernel.TruncateRunes(value, limit)
|
||||
}
|
||||
|
||||
func taskForOutput(task model.Task) *model.Task {
|
||||
task.InputJSON = publicTaskInputJSON(task.InputJSON)
|
||||
// 普通任务接口只暴露前台模型身份；渠道模型和供应线路属于管理员内部信息。
|
||||
task.LogicalModelRevisionID = ""
|
||||
task.RouteID = ""
|
||||
task.ChannelModelID = ""
|
||||
return &task
|
||||
}
|
||||
|
||||
func publicTaskInputJSON(raw string) string {
|
||||
if strings.TrimSpace(raw) == "" {
|
||||
return ""
|
||||
}
|
||||
var input map[string]any
|
||||
if err := json.Unmarshal([]byte(raw), &input); err != nil {
|
||||
return ""
|
||||
}
|
||||
public := map[string]any{}
|
||||
// 任务完成后仍需依靠这些非敏感 ID 恢复项目产物归属；密钥等配置继续被过滤。
|
||||
for _, key := range []string{"mode", "metadata", "workflowStepId", "domainProjectId", "assetVersionId", "resourceId", "mediaType", "role"} {
|
||||
if value, ok := input[key]; ok {
|
||||
public[key] = value
|
||||
}
|
||||
}
|
||||
if len(public) == 0 {
|
||||
return ""
|
||||
}
|
||||
data, _ := json.Marshal(public)
|
||||
return string(data)
|
||||
}
|
||||
Reference in new issue
Block a user