237 lines
8.1 KiB
Go
237 lines
8.1 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"math"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func adaptLibTVDetail(detail *libTVDetail) (*LibTVImportResult, error) {
|
|
if detail == nil {
|
|
return nil, errors.New("LibTV 画布数据为空")
|
|
}
|
|
now := time.Now()
|
|
batchID := strconv.FormatInt(now.UnixMilli(), 36) + "-" + strings.ReplaceAll(uuid.NewString(), "-", "")[:8]
|
|
result := &LibTVImportResult{
|
|
BatchID: batchID,
|
|
BatchCreatedAt: now,
|
|
ProjectUUID: detail.ProjectMeta.UUID,
|
|
ProjectName: detail.ProjectMeta.Name,
|
|
Nodes: make([]LibTVCanvasNode, 0),
|
|
Connections: make([]LibTVCanvasConnection, 0),
|
|
SkippedNodes: make([]LibTVImportIssue, 0),
|
|
SkippedConnections: make([]LibTVImportIssue, 0),
|
|
Warnings: make([]LibTVImportWarning, 0),
|
|
}
|
|
mapping := make(map[string]string, len(detail.NodeList))
|
|
seenNodeKeys := make(map[string]struct{}, len(detail.NodeList))
|
|
for _, raw := range detail.NodeList {
|
|
nodeKey := strings.TrimSpace(raw.NodeKey)
|
|
if nodeKey == "" {
|
|
result.SkippedNodes = append(result.SkippedNodes, LibTVImportIssue{Name: raw.Name, Reason: "节点缺少 nodeKey"})
|
|
continue
|
|
}
|
|
if _, exists := seenNodeKeys[nodeKey]; exists {
|
|
result.SkippedNodes = append(result.SkippedNodes, LibTVImportIssue{ID: nodeKey, Name: raw.Name, Reason: "节点 nodeKey 重复"})
|
|
continue
|
|
}
|
|
seenNodeKeys[nodeKey] = struct{}{}
|
|
|
|
var data libTVNodeData
|
|
if err := json.Unmarshal([]byte(raw.Data), &data); err != nil {
|
|
result.SkippedNodes = append(result.SkippedNodes, LibTVImportIssue{ID: nodeKey, Name: raw.Name, Reason: "节点数据格式无效"})
|
|
continue
|
|
}
|
|
taskStatus := data.TaskInfo.Status
|
|
if taskStatus == 0 && raw.TaskInfo.Status != 0 {
|
|
taskStatus = raw.TaskInfo.Status
|
|
}
|
|
sourceType := strings.ToLower(strings.TrimSpace(data.Type))
|
|
kind := sourceType
|
|
if sourceType == "material-style" {
|
|
// 风格素材没有普通生成 URL，但有可复用封面时可以降级为影策图片参考节点。
|
|
kind = "image"
|
|
} else if kind != "image" && kind != "video" {
|
|
result.SkippedNodes = append(result.SkippedNodes, LibTVImportIssue{ID: nodeKey, Name: raw.Name, Reason: "暂不支持的节点类型"})
|
|
continue
|
|
}
|
|
mediaURL, mediaIndex := firstLibTVMediaURL(data.URL)
|
|
if mediaURL == "" && sourceType == "material-style" {
|
|
mediaURL, mediaIndex = firstLibTVMediaURL([]string{data.CoverURL})
|
|
}
|
|
|
|
x, okX := finiteFloat(raw.Position.X)
|
|
y, okY := finiteFloat(raw.Position.Y)
|
|
if !okX || !okY {
|
|
result.SkippedNodes = append(result.SkippedNodes, LibTVImportIssue{ID: nodeKey, Name: raw.Name, Reason: "节点坐标无效"})
|
|
continue
|
|
}
|
|
width, okWidth := positiveFiniteFloat(raw.Measured.Width)
|
|
height, okHeight := positiveFiniteFloat(raw.Measured.Height)
|
|
if !okWidth {
|
|
width = 480
|
|
}
|
|
if !okHeight {
|
|
height = 300
|
|
}
|
|
item := LibTVCanvasNode{
|
|
ID: "libtv-" + batchID + "-" + nodeKey,
|
|
Type: kind,
|
|
Title: strings.TrimSpace(raw.Name),
|
|
X: x,
|
|
Y: y,
|
|
Width: width,
|
|
Height: height,
|
|
Content: mediaURL,
|
|
Prompt: mapString(data.Params, "prompt"),
|
|
Model: mapString(data.Params, "model"),
|
|
Metadata: LibTVImportMetadata{
|
|
Provider: "libtv",
|
|
ProjectUUID: result.ProjectUUID,
|
|
NodeKey: nodeKey,
|
|
BatchID: batchID,
|
|
SourceType: sourceType,
|
|
StyleAssetUUID: strings.TrimSpace(data.StyleAssetUUID),
|
|
StyleVersionUUID: strings.TrimSpace(data.StyleVersionUUID),
|
|
StyleName: strings.TrimSpace(data.StyleName),
|
|
},
|
|
}
|
|
if item.Title == "" {
|
|
item.Title = kind + " 节点"
|
|
}
|
|
if mediaIndex >= 0 && mediaIndex < len(data.ResourceMeta.Items) {
|
|
resource := data.ResourceMeta.Items[mediaIndex]
|
|
item.NaturalWidth = resource.Width
|
|
item.NaturalHeight = resource.Height
|
|
if resource.DurationSec > 0 && !math.IsNaN(resource.DurationSec) && !math.IsInf(resource.DurationSec, 0) {
|
|
item.DurationMs = int64(math.Round(resource.DurationSec * 1000))
|
|
}
|
|
}
|
|
item.MimeType = inferMediaMime(kind, mediaURL)
|
|
if mediaURL == "" {
|
|
result.PlaceholderNodeCount++
|
|
if taskStatus == 3 {
|
|
// 没有历史结果的失败节点仍保留拓扑，并以失败占位状态呈现。
|
|
item.Status = "error"
|
|
item.ErrorDetails = firstNonEmptyString(data.TaskInfo.FailedReason, raw.TaskInfo.FailedReason, "LibTV 生成任务失败")
|
|
} else {
|
|
item.Status = "idle"
|
|
}
|
|
} else {
|
|
item.Status = "success"
|
|
if taskStatus == 3 {
|
|
// 最近一次任务失败不应覆盖节点中仍可用的历史资源。
|
|
result.ReusedFailedNodeCount++
|
|
}
|
|
}
|
|
mapping[nodeKey] = item.ID
|
|
result.Nodes = append(result.Nodes, item)
|
|
if sourceType == "material-style" {
|
|
result.ConvertedSpecialCount++
|
|
}
|
|
if !okWidth || !okHeight {
|
|
result.Warnings = append(result.Warnings, LibTVImportWarning{ID: nodeKey, Message: fmt.Sprintf("节点“%s”尺寸无效，已使用默认尺寸。", item.Title)})
|
|
}
|
|
if len(data.URL) > 1 && mediaURL != "" {
|
|
result.MultiResultNodeCount++
|
|
result.Warnings = append(result.Warnings, LibTVImportWarning{ID: nodeKey, Message: fmt.Sprintf("节点“%s”包含 %d 个结果，已导入第 %d 个。", item.Title, len(data.URL), mediaIndex+1)})
|
|
}
|
|
if data.IsStale || raw.IsStale {
|
|
result.StaleNodeCount++
|
|
result.Warnings = append(result.Warnings, LibTVImportWarning{ID: nodeKey, Message: fmt.Sprintf("节点“%s”已标记为过期，但现有资源仍已导入。", item.Title)})
|
|
}
|
|
}
|
|
|
|
seenConnectionIDs := make(map[string]struct{}, len(detail.ConnectionList))
|
|
for _, raw := range detail.ConnectionList {
|
|
connectionID := strings.TrimSpace(raw.ConnectionID)
|
|
if connectionID == "" {
|
|
result.SkippedConnections = append(result.SkippedConnections, LibTVImportIssue{Reason: "连接缺少 connectionId"})
|
|
continue
|
|
}
|
|
if _, exists := seenConnectionIDs[connectionID]; exists {
|
|
result.SkippedConnections = append(result.SkippedConnections, LibTVImportIssue{ID: connectionID, Reason: "连接 connectionId 重复"})
|
|
continue
|
|
}
|
|
seenConnectionIDs[connectionID] = struct{}{}
|
|
from, okFrom := mapping[strings.TrimSpace(raw.Source)]
|
|
to, okTo := mapping[strings.TrimSpace(raw.Target)]
|
|
if !okFrom || !okTo {
|
|
result.SkippedConnections = append(result.SkippedConnections, LibTVImportIssue{ID: connectionID, Reason: "连接端点节点未导入"})
|
|
continue
|
|
}
|
|
result.Connections = append(result.Connections, LibTVCanvasConnection{ID: "libtv-" + batchID + "-" + connectionID, FromNodeID: from, ToNodeID: to})
|
|
}
|
|
result.ImportedNodeCount = len(result.Nodes)
|
|
result.ImportedConnectionCount = len(result.Connections)
|
|
if result.ImportedNodeCount == 0 {
|
|
return nil, errors.New("LibTV 画布没有可导入的有效节点")
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func mapString(values map[string]any, key string) string {
|
|
value, _ := values[key].(string)
|
|
return strings.TrimSpace(value)
|
|
}
|
|
|
|
func isLibTVProjectUUID(value string) bool {
|
|
if len(value) != 32 {
|
|
return false
|
|
}
|
|
for _, char := range value {
|
|
if !((char >= '0' && char <= '9') || (char >= 'a' && char <= 'f') || (char >= 'A' && char <= 'F')) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func firstLibTVMediaURL(values []string) (string, int) {
|
|
for index, value := range values {
|
|
parsed, err := url.Parse(strings.TrimSpace(value))
|
|
if err != nil || parsed.Scheme != "https" || parsed.Hostname() == "" || parsed.User != nil {
|
|
continue
|
|
}
|
|
return parsed.String(), index
|
|
}
|
|
return "", -1
|
|
}
|
|
|
|
func finiteFloat(value string) (float64, bool) {
|
|
parsed, err := strconv.ParseFloat(strings.TrimSpace(value), 64)
|
|
if err != nil || math.IsNaN(parsed) || math.IsInf(parsed, 0) {
|
|
return 0, false
|
|
}
|
|
return parsed, true
|
|
}
|
|
|
|
func positiveFiniteFloat(value string) (float64, bool) {
|
|
parsed, ok := finiteFloat(value)
|
|
return parsed, ok && parsed > 0
|
|
}
|
|
|
|
func inferMediaMime(kind, rawURL string) string {
|
|
lower := strings.ToLower(rawURL)
|
|
if strings.Contains(lower, ".png") {
|
|
return "image/png"
|
|
}
|
|
if strings.Contains(lower, ".jpg") || strings.Contains(lower, ".jpeg") {
|
|
return "image/jpeg"
|
|
}
|
|
if strings.Contains(lower, ".webp") {
|
|
return "image/webp"
|
|
}
|
|
if kind == "video" {
|
|
return "video/mp4"
|
|
}
|
|
return "image/*"
|
|
}
|