chore: import clean Agent Desk snapshot (3/7)

2026-10-16 11:58:36 +00:00
parent 89c5231c4c
commit fe4a99cf4b
127 changed files with 29641 additions and 0 deletions
--- a/internal/services/im_message_asset.go
+++ b/internal/services/im_message_asset.go
@@ -0,0 +1,116 @@
+package services
+
+import (
+	"agent-desk/internal/models"
+	"agent-desk/internal/pkg/enums"
+	"agent-desk/internal/pkg/errorsx"
+	"agent-desk/internal/services/storage"
+	"encoding/json"
+	"strings"
+)
+
+type imMessageAssetPayload struct {
+	AssetID    string              `json:"assetId"`
+	Provider   enums.AssetProvider `json:"provider,omitempty"`
+	StorageKey string              `json:"storageKey,omitempty"`
+	Filename   string              `json:"filename,omitempty"`
+	FileSize   int64               `json:"fileSize,omitempty"`
+	MimeType   string              `json:"mimeType,omitempty"`
+	URL        string              `json:"url,omitempty"`
+}
+
+func parseIMMessageAssetPayload(payload string) (*imMessageAssetPayload, error) {
+	payload = strings.TrimSpace(payload)
+	if payload == "" {
+		return nil, errorsx.InvalidParamI18n("error.e0346")
+	}
+	ret := &imMessageAssetPayload{}
+	if err := json.Unmarshal([]byte(payload), ret); err != nil {
+		return nil, errorsx.InvalidParamI18n("error.e0344")
+	}
+	ret.AssetID = strings.TrimSpace(ret.AssetID)
+	ret.Provider = enums.AssetProvider(strings.TrimSpace(string(ret.Provider)))
+	ret.StorageKey = strings.TrimSpace(ret.StorageKey)
+	if ret.AssetID == "" {
+		return nil, errorsx.InvalidParamI18n("error.e0345")
+	}
+	return ret, nil
+}
+
+func buildIMMessageAssetPayload(asset *models.Asset) (string, error) {
+	if asset == nil {
+		return "", errorsx.InvalidParamI18n("error.e0342")
+	}
+	payload, err := json.Marshal(imMessageAssetPayload{
+		AssetID:    asset.AssetID,
+		Provider:   asset.Provider,
+		StorageKey: asset.StorageKey,
+		Filename:   asset.Filename,
+		FileSize:   asset.FileSize,
+		MimeType:   asset.MimeType,
+	})
+	if err != nil {
+		return "", err
+	}
+	return string(payload), nil
+}
+
+func buildIMMessageAssetPayloadForResponse(payload string) string {
+	assetPayload, err := parseIMMessageAssetPayload(payload)
+	if err != nil {
+		return strings.TrimSpace(payload)
+	}
+	assetPayload = hydrateIMMessageAssetPayload(assetPayload)
+	if assetPayload.Provider != "" && assetPayload.StorageKey != "" {
+		if provider, err := storage.NewProvider(assetPayload.Provider); err == nil {
+			assetPayload.URL = provider.GetSignedURL(assetPayload.StorageKey)
+		}
+	}
+	data, err := json.Marshal(assetPayload)
+	if err != nil {
+		return strings.TrimSpace(payload)
+	}
+	return string(data)
+}
+
+func hydrateIMMessageAssetPayload(payload *imMessageAssetPayload) *imMessageAssetPayload {
+	if payload == nil {
+		return nil
+	}
+	if payload.Provider != "" && payload.StorageKey != "" {
+		return payload
+	}
+	if payload.AssetID == "" {
+		return payload
+	}
+	asset := AssetService.GetByAssetID(payload.AssetID)
+	if asset == nil {
+		return payload
+	}
+	if payload.Provider == "" {
+		payload.Provider = asset.Provider
+	}
+	if payload.StorageKey == "" {
+		payload.StorageKey = strings.TrimSpace(asset.StorageKey)
+	}
+	if payload.Filename == "" {
+		payload.Filename = strings.TrimSpace(asset.Filename)
+	}
+	if payload.FileSize <= 0 {
+		payload.FileSize = asset.FileSize
+	}
+	if payload.MimeType == "" {
+		payload.MimeType = strings.TrimSpace(asset.MimeType)
+	}
+	return payload
+}
+
+func validateConversationAsset(asset *models.Asset, conversationID int64, messageType enums.IMMessageType) error {
+	if asset == nil {
+		return errorsx.InvalidParamI18n("error.e0342")
+	}
+	if asset.Status != enums.AssetStatusSuccess {
+		return errorsx.InvalidParamI18n("error.e0343")
+	}
+	return nil
+}