feat(message): enhance message normalization and asset handling in HTML content

internal/pkg/utils/message_test.go

@@ -6,6 +6,11 @@ import (
 	"cs-agent/internal/pkg/enums"
 	"strings"
 	"testing"
+	"time"
+
+	"github.com/glebarez/sqlite"
+	"github.com/mlogclub/simple/sqls"
+	"gorm.io/gorm"
 )
 
 func TestBuildIMMessageAssetPayloadForResponseAddsSignedURL(t *testing.T) {
@@ -94,3 +99,86 @@ func TestBuildRenderableMessageTransformsPayloadAndHTML(t *testing.T) {
 		t.Fatalf("expected html content signed src, got: %s", htmlContent)
 	}
 }
+
+func TestNormalizeMessageHTMLAssetsEnrichesImageDataAttrs(t *testing.T) {
+	setupMessageTestDB(t)
+	config.SetCurrent(&config.Config{
+		Storage: config.StorageConfig{
+			Default: enums.AssetProviderLocal,
+			Local: config.LocalStorageConfig{
+				BaseURL: "https://files.example.com",
+			},
+		},
+	})
+	createTestAsset(t, &models.Asset{
+		AssetID:    "asset_local_1",
+		Provider:   enums.AssetProviderLocal,
+		StorageKey: "images/demo.png",
+		Filename:   "demo.png",
+		FileSize:   123,
+		MimeType:   "image/png",
+		Status:     enums.AssetStatusSuccess,
+	})
+
+	got := NormalizeMessageHTMLAssets(`<p><img src="https://files.example.com/images/demo.png" alt="demo"></p>`)
+
+	if !strings.Contains(got, `data-asset-id="asset_local_1"`) {
+		t.Fatalf("expected data-asset-id added, got: %s", got)
+	}
+	if !strings.Contains(got, `data-provider="local"`) {
+		t.Fatalf("expected data-provider added, got: %s", got)
+	}
+	if !strings.Contains(got, `data-storage-key="images/demo.png"`) {
+		t.Fatalf("expected data-storage-key added, got: %s", got)
+	}
+	if strings.Contains(got, `src=`) {
+		t.Fatalf("expected src removed after asset binding, got: %s", got)
+	}
+}
+
+func TestNormalizeMessageHTMLAssetsKeepsUnknownImageSrc(t *testing.T) {
+	setupMessageTestDB(t)
+	config.SetCurrent(&config.Config{
+		Storage: config.StorageConfig{
+			Default: enums.AssetProviderLocal,
+			Local: config.LocalStorageConfig{
+				BaseURL: "https://files.example.com",
+			},
+		},
+	})
+
+	got := NormalizeMessageHTMLAssets(`<p><img src="https://unknown.example.com/demo.png" alt="demo"></p>`)
+
+	if !strings.Contains(got, `src="https://unknown.example.com/demo.png"`) {
+		t.Fatalf("expected unknown image src kept, got: %s", got)
+	}
+	if strings.Contains(got, `data-asset-id=`) || strings.Contains(got, `data-provider=`) || strings.Contains(got, `data-storage-key=`) {
+		t.Fatalf("expected no asset attrs added for unknown image, got: %s", got)
+	}
+}
+
+func setupMessageTestDB(t *testing.T) {
+	t.Helper()
+	db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
+	if err != nil {
+		t.Fatalf("open sqlite failed: %v", err)
+	}
+	if err := db.AutoMigrate(&models.Asset{}); err != nil {
+		t.Fatalf("auto migrate asset failed: %v", err)
+	}
+	sqls.SetDB(db)
+}
+
+func createTestAsset(t *testing.T, item *models.Asset) {
+	t.Helper()
+	now := time.Now()
+	if item.CreatedAt.IsZero() {
+		item.CreatedAt = now
+	}
+	if item.UpdatedAt.IsZero() {
+		item.UpdatedAt = now
+	}
+	if err := sqls.DB().Create(item).Error; err != nil {
+		t.Fatalf("create asset failed: %v", err)
+	}
+}