Initial commit

Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>

web/skd/playletwatermark/expire.go

@@ -0,0 +1,38 @@
+package playletwatermark
+
+import (
+	"context"
+	"time"
+	"videoweb/pub/common/log"
+	"videoweb/pub/common/srvdef"
+	"videoweb/pub/models/playletchaptermod"
+
+	"go.mongodb.org/mongo-driver/bson"
+)
+
+const watermarkStuckTimeout = 48 * time.Hour
+
+// MarkStuckChaptersFailed 兜底：已登记/处理中超过 2 天仍未完成的剧集标记为处理失败。
+// media 原片已 CompletedFail 时由 submit 立即回写，不必等这 2 天。
+func MarkStuckChaptersFailed() {
+	ctx := context.Background()
+	now := time.Now()
+	cutoff := now.Add(-watermarkStuckTimeout)
+	err := playletchaptermod.UpdateManyPlayletChapter(ctx, bson.M{
+		playletchaptermod.FN.PlayletChapter.WatermarkStatus: bson.M{
+			"$in": []playletchaptermod.WatermarkStatus{
+				playletchaptermod.WatermarkStatusRegistered,
+				playletchaptermod.WatermarkStatusProcessing,
+			},
+		},
+		playletchaptermod.FN.PlayletChapter.UpdatedAt: bson.M{"$lt": cutoff},
+	}, bson.M{"$set": bson.M{
+		playletchaptermod.FN.PlayletChapter.WatermarkStatus: playletchaptermod.WatermarkStatusFailed,
+		playletchaptermod.FN.PlayletChapter.UpdatedAt:       now,
+	}})
+	if err != nil && err.Code != srvdef.ErrNoData {
+		log.Error("playletwatermark:MarkStuckChaptersFailed 更新失败", log.E(err))
+		return
+	}
+	log.Info("playletwatermark:MarkStuckChaptersFailed end", log.Any("cutoff", cutoff))
+}

web/skd/playletwatermark/submit.go

@@ -0,0 +1,367 @@
+package playletwatermark
+
+import (
+	"context"
+	"errors"
+	"net/url"
+	"path"
+	"strings"
+	"sync"
+	"time"
+	"videoweb/pub/common/log"
+	"videoweb/pub/models/cfg"
+	"videoweb/pub/models/playletchaptermod"
+	"videoweb/pub/models/playletmod"
+	"videoweb/pub/pubdef"
+	"videoweb/pub/storage/proxy"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+const (
+	submitBatchSize    = 10
+	taskAddRetryBatch  = 20
+	mediaFileVersionV1 = "v1"
+	mediaFileVersionV2 = "v2"
+)
+
+var (
+	submitRunning bool
+	submitLock    sync.Mutex
+)
+
+// SubmitPendingChapters 定时提交未加水印剧集到 media（每10分钟最多10条）
+func SubmitPendingChapters() {
+	submitLock.Lock()
+	if submitRunning {
+		submitLock.Unlock()
+		return
+	}
+	submitRunning = true
+	submitLock.Unlock()
+	defer func() {
+		submitLock.Lock()
+		submitRunning = false
+		submitLock.Unlock()
+	}()
+
+	ctx := context.Background()
+	log.Info("playletwatermark:SubmitPendingChapters begin")
+	MarkStuckChaptersFailed()
+
+	sysCfg, err := cfg.FindOneSysCfg(ctx, bson.M{cfg.FN.SysCfg.Id: 1})
+	if err != nil || sysCfg == nil {
+		log.Error("playletwatermark:SubmitPendingChapters 读取系统配置失败", log.E(err))
+		return
+	}
+	if !hasVideoWatermark(sysCfg) {
+		log.Warn("playletwatermark:SubmitPendingChapters 水印素材未配置，跳过")
+		return
+	}
+
+	registerPendingSources(ctx)
+	submitWatermarkTasks(ctx, sysCfg)
+	log.Info("playletwatermark:SubmitPendingChapters end")
+}
+
+// registerPendingSources 将未处理剧集的相对 m3u8 登记到 media
+func registerPendingSources(ctx context.Context) {
+	list, err := playletchaptermod.FindManyPlayletChapter(ctx, bson.M{
+		playletchaptermod.FN.PlayletChapter.WatermarkStatus: bson.M{
+			"$in": []playletchaptermod.WatermarkStatus{
+				playletchaptermod.WatermarkStatusPending,
+				playletchaptermod.WatermarkStatusRegistered,
+			},
+		},
+		playletchaptermod.FN.PlayletChapter.VideoUrl: bson.M{"$gt": ""},
+		"$or": []bson.M{
+			{playletchaptermod.FN.PlayletChapter.MediaSourceId: bson.M{"$exists": false}},
+			{playletchaptermod.FN.PlayletChapter.MediaSourceId: ""},
+		},
+	}, options.Find().
+		SetSort(bson.D{{Key: playletchaptermod.FN.PlayletChapter.Id, Value: 1}}).
+		SetLimit(submitBatchSize))
+	if err != nil {
+		log.Error("playletwatermark:registerPendingSources 查询失败", log.E(err))
+		return
+	}
+	for _, item := range list {
+		if item == nil {
+			continue
+		}
+		sourceID, syncErr := RegisterChapterSource(item)
+		if syncErr != nil {
+			log.Error("playletwatermark:registerPendingSources SyncMediaFile 失败",
+				log.Any("chapterId", item.Id), log.Any("videoUrl", item.VideoUrl), log.E(syncErr))
+			continue
+		}
+		now := time.Now()
+		if updateErr := playletchaptermod.UpdateOnePlayletChapter(ctx,
+			bson.M{playletchaptermod.FN.PlayletChapter.Id: item.Id},
+			bson.M{"$set": bson.M{
+				playletchaptermod.FN.PlayletChapter.MediaSourceId:   sourceID,
+				playletchaptermod.FN.PlayletChapter.WatermarkStatus: playletchaptermod.WatermarkStatusRegistered,
+				playletchaptermod.FN.PlayletChapter.UpdatedAt:       now,
+			}}); updateErr != nil {
+			log.Error("playletwatermark:registerPendingSources 更新章节失败",
+				log.Any("chapterId", item.Id), log.E(updateErr))
+		}
+	}
+}
+
+// RegisterChapterSource 将剧集原片登记到 media 切片服，返回 mediaSourceId。
+func RegisterChapterSource(item *playletchaptermod.PlayletChapter) (string, error) {
+	if item == nil {
+		return "", errors.New("chapter is nil")
+	}
+	fieldPath, version := normalizeMediaFilePath(item.VideoUrl)
+	if fieldPath == "" {
+		return "", errors.New("invalid video path")
+	}
+	sourceID := primitive.NewObjectID()
+	req := &proxy.MediaFileSyncReq{
+		AppId: pubdef.AppId,
+		VideoList: []proxy.MediaFileSyncItem{
+			{
+				ID:               sourceID.Hex(),
+				Title:            mediaWatermarkTitle(playletForChapter(item), item),
+				TaskID:           sourceID.Hex(),
+				FieldNameFs:      fieldPath,
+				Width:            item.Width,
+				Height:           item.Height,
+				VideoTime:        item.MediaTime,
+				Size:             item.MediaSize,
+				MediaFileVersion: version,
+				IsPlaylet:        true,
+			},
+		},
+	}
+	result, syncErr := proxy.SyncMediaFile(req, "")
+	if syncErr != nil {
+		return "", syncErr
+	}
+	if result == nil || result.Code != proxy.Success {
+		return "", errors.New("SyncMediaFile failed")
+	}
+	return sourceID.Hex(), nil
+}
+
+// submitWatermarkTasks 对已登记原片尝试 task/add，成功后进入处理中
+func submitWatermarkTasks(ctx context.Context, sysCfg *cfg.SysCfg) {
+	list, err := playletchaptermod.FindManyPlayletChapter(ctx, bson.M{
+		playletchaptermod.FN.PlayletChapter.WatermarkStatus: playletchaptermod.WatermarkStatusRegistered,
+		playletchaptermod.FN.PlayletChapter.MediaSourceId:   bson.M{"$gt": ""},
+	}, options.Find().
+		SetSort(bson.D{{Key: playletchaptermod.FN.PlayletChapter.Id, Value: 1}}).
+		SetLimit(taskAddRetryBatch))
+	if err != nil {
+		log.Error("playletwatermark:submitWatermarkTasks 查询失败", log.E(err))
+		return
+	}
+	for _, item := range list {
+		if item == nil || item.MediaSourceId == "" {
+			continue
+		}
+		if strings.TrimSpace(item.MediaTaskId) != "" {
+			if !restartWatermarkTask(ctx, item, sysCfg) {
+				continue
+			}
+		} else if !addWatermarkTask(ctx, item, sysCfg) {
+			continue
+		}
+		_ = playletchaptermod.UpdateOnePlayletChapter(ctx,
+			bson.M{
+				playletchaptermod.FN.PlayletChapter.Id:              item.Id,
+				playletchaptermod.FN.PlayletChapter.WatermarkStatus: playletchaptermod.WatermarkStatusRegistered,
+			},
+			bson.M{"$set": bson.M{
+				playletchaptermod.FN.PlayletChapter.WatermarkStatus: playletchaptermod.WatermarkStatusProcessing,
+				playletchaptermod.FN.PlayletChapter.UpdatedAt:       time.Now(),
+			}})
+	}
+}
+
+func addWatermarkTask(ctx context.Context, item *playletchaptermod.PlayletChapter, sysCfg *cfg.SysCfg) bool {
+	req := &proxy.MediaTaskAddReq{
+		MediaSourceId:            item.MediaSourceId,
+		AppId:                    pubdef.AppId,
+		Title:                    mediaWatermarkTitle(playletForChapter(item), item),
+		WatermarkImage:           sysCfg.WatermarkImage,
+		WatermarkMedia:           sysCfg.WatermarkMedia,
+		WatermarkMediaTop:        sysCfg.WatermarkMediaTop,
+		WatermarkMediaTopRight:   sysCfg.WatermarkMediaTopRight,
+		WatermarkMediaBottomLeft: sysCfg.WatermarkMediaBottomLeft,
+	}
+	result, addErr := proxy.AddMediaTask(req, "")
+	if addErr != nil || result == nil {
+		log.Info("playletwatermark:submitWatermarkTasks task/add 请求失败",
+			log.Any("chapterId", item.Id),
+			log.Any("mediaSourceId", item.MediaSourceId),
+			log.E(addErr))
+		return false
+	}
+	// 200=新建成功；6007=任务已存在。原片未处理完时保持已登记，下轮再试。
+	if result.Code != proxy.Success && result.Code != 6007 {
+		log.Info("playletwatermark:submitWatermarkTasks task/add 暂未成功",
+			log.Any("chapterId", item.Id),
+			log.Any("mediaSourceId", item.MediaSourceId),
+			log.Any("result", result))
+		markFailedIfMediaSourceFailed(ctx, item)
+		return false
+	}
+	return true
+}
+
+func restartWatermarkTask(ctx context.Context, item *playletchaptermod.PlayletChapter, sysCfg *cfg.SysCfg) bool {
+	watermarkImage := sysCfg.WatermarkImage
+	watermarkMedia := sysCfg.WatermarkMedia
+	watermarkMediaTop := sysCfg.WatermarkMediaTop
+	watermarkMediaTopRight := sysCfg.WatermarkMediaTopRight
+	watermarkMediaBottomLeft := sysCfg.WatermarkMediaBottomLeft
+	req := &proxy.MediaTaskUpdateWatermarkReq{
+		Id:                       item.MediaTaskId,
+		AppId:                    pubdef.AppId,
+		WatermarkImage:           &watermarkImage,
+		WatermarkMedia:           &watermarkMedia,
+		WatermarkMediaTop:        &watermarkMediaTop,
+		WatermarkMediaTopRight:   &watermarkMediaTopRight,
+		WatermarkMediaBottomLeft: &watermarkMediaBottomLeft,
+	}
+	result, updateErr := proxy.UpdateMediaWatermark(req, "")
+	if updateErr != nil || result == nil {
+		log.Info("playletwatermark:submitWatermarkTasks update/watermark 请求失败",
+			log.Any("chapterId", item.Id),
+			log.Any("mediaTaskId", item.MediaTaskId),
+			log.E(updateErr))
+		return false
+	}
+	if result.Code != proxy.Success {
+		log.Info("playletwatermark:submitWatermarkTasks update/watermark 暂未成功",
+			log.Any("chapterId", item.Id),
+			log.Any("mediaTaskId", item.MediaTaskId),
+			log.Any("result", result))
+		markFailedIfMediaSourceFailed(ctx, item)
+		return false
+	}
+	return true
+}
+
+func markFailedIfMediaSourceFailed(ctx context.Context, item *playletchaptermod.PlayletChapter) {
+	if item == nil || item.MediaSourceId == "" {
+		return
+	}
+	source, err := proxy.QueryMediaSource(item.MediaSourceId, pubdef.AppId, "")
+	if err != nil {
+		log.Info("playletwatermark:markFailedIfMediaSourceFailed 查询原片失败",
+			log.Any("chapterId", item.Id),
+			log.Any("mediaSourceId", item.MediaSourceId),
+			log.E(err))
+		return
+	}
+	if source == nil || source.Status != proxy.MediaSourceCompletedFail {
+		return
+	}
+	now := time.Now()
+	if updateErr := playletchaptermod.UpdateOnePlayletChapter(ctx,
+		bson.M{
+			playletchaptermod.FN.PlayletChapter.Id: item.Id,
+			playletchaptermod.FN.PlayletChapter.WatermarkStatus: bson.M{
+				"$in": []playletchaptermod.WatermarkStatus{
+					playletchaptermod.WatermarkStatusRegistered,
+					playletchaptermod.WatermarkStatusProcessing,
+				},
+			},
+		},
+		bson.M{"$set": bson.M{
+			playletchaptermod.FN.PlayletChapter.WatermarkStatus: playletchaptermod.WatermarkStatusFailed,
+			playletchaptermod.FN.PlayletChapter.UpdatedAt:       now,
+		}}); updateErr != nil {
+		log.Error("playletwatermark:markFailedIfMediaSourceFailed 更新章节失败",
+			log.Any("chapterId", item.Id),
+			log.E(updateErr))
+		return
+	}
+	log.Info("playletwatermark:markFailedIfMediaSourceFailed 原片已失败，章节标为处理失败",
+		log.Any("chapterId", item.Id),
+		log.Any("mediaSourceId", item.MediaSourceId),
+		log.Any("failReason", source.FailReason))
+}
+
+// normalizeMediaFilePath 将相对/绝对 m3u8 地址规范化为 media 可识别的相对路径（不含 .m3u8）
+func normalizeMediaFilePath(raw string) (fieldPath string, version string) {
+	raw = strings.TrimSpace(raw)
+	if raw == "" {
+		return "", ""
+	}
+	version = mediaFileVersionV2
+	if strings.HasPrefix(raw, "http://") || strings.HasPrefix(raw, "https://") {
+		u, err := url.Parse(raw)
+		if err != nil {
+			return "", ""
+		}
+		raw = u.Path
+	}
+	raw = strings.TrimPrefix(raw, "/")
+	if strings.HasPrefix(raw, "v1/") {
+		version = mediaFileVersionV1
+	}
+	if strings.HasSuffix(strings.ToLower(raw), ".m3u8") {
+		raw = strings.TrimSuffix(raw, ".m3u8")
+		raw = strings.TrimSuffix(raw, ".M3U8")
+	}
+	raw = path.Clean(raw)
+	if raw == "." || raw == "" {
+		return "", ""
+	}
+	return raw, version
+}
+
+// mediaWatermarkTitle 传给 media 的标题：短剧剧名（没有则用标题）+ "-" + 剧集标题
+func mediaWatermarkTitle(playlet *playletmod.Playlet, chapter *playletchaptermod.PlayletChapter) string {
+	playletName := ""
+	if playlet != nil {
+		playletName = strings.TrimSpace(playlet.PlayName)
+		if playletName == "" {
+			playletName = strings.TrimSpace(playlet.Title)
+		}
+	}
+	episodeTitle := ""
+	if chapter != nil {
+		episodeTitle = strings.TrimSpace(chapter.Title)
+	}
+	if playletName != "" && episodeTitle != "" {
+		return playletName + "-" + episodeTitle
+	}
+	if playletName != "" {
+		return playletName
+	}
+	return episodeTitle
+}
+
+func playletForChapter(item *playletchaptermod.PlayletChapter) *playletmod.Playlet {
+	if item == nil || item.PlayletId <= 0 {
+		return nil
+	}
+	playlet, err := playletmod.FindOnePlaylet(context.Background(), bson.M{
+		playletmod.FN.Playlet.Id: item.PlayletId,
+	}, options.FindOne().SetProjection(bson.M{
+		playletmod.FN.Playlet.PlayName: 1,
+		playletmod.FN.Playlet.Title:    1,
+	}))
+	if err != nil {
+		return nil
+	}
+	return playlet
+}
+
+func hasVideoWatermark(sysCfg *cfg.SysCfg) bool {
+	if sysCfg == nil {
+		return false
+	}
+	return strings.TrimSpace(sysCfg.WatermarkMedia) != "" ||
+		strings.TrimSpace(sysCfg.WatermarkMediaTop) != "" ||
+		strings.TrimSpace(sysCfg.WatermarkMediaTopRight) != "" ||
+		strings.TrimSpace(sysCfg.WatermarkMediaBottomLeft) != ""
+}

web/skd/playletwatermark/sync.go

@@ -0,0 +1,284 @@
+package playletwatermark
+
+import (
+	"context"
+	"net/http"
+	"strings"
+	"sync"
+	"time"
+	"videoweb/pub/common/log"
+	"videoweb/pub/common/srvdef"
+	"videoweb/pub/models/playletchaptermod"
+	"videoweb/pub/models/playletmod"
+	"videoweb/pub/models/syncfilemod"
+	"videoweb/pub/pubdef"
+	"videoweb/pub/storage/proxy"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+const (
+	syncPageSize       int64 = 1000
+	syncOverlapWindow        = 10 * time.Minute
+	reconcileBatchSize int64 = 100
+)
+
+var (
+	syncRunning bool
+	syncLock    sync.Mutex
+)
+
+// SyncCompletedChapters 定时从 media 拉取已完成水印任务，回写剧集地址与状态
+func SyncCompletedChapters() {
+	syncLock.Lock()
+	if syncRunning {
+		syncLock.Unlock()
+		return
+	}
+	syncRunning = true
+	syncLock.Unlock()
+	defer func() {
+		syncLock.Lock()
+		syncRunning = false
+		syncLock.Unlock()
+	}()
+
+	ctx := context.Background()
+	log.Info("playletwatermark:SyncCompletedChapters begin")
+
+	record, err := ensureSyncRecord(ctx)
+	if err != nil {
+		log.Error("playletwatermark:SyncCompletedChapters 获取同步水印失败", log.E(err))
+		return
+	}
+
+	recordTime, syncOK := syncIncrementalCompletedChapters(ctx, record.RecordTime)
+	if syncOK {
+		if updateErr := syncfilemod.UpdateOneSyncRecord(ctx, bson.M{
+			syncfilemod.FN.SyncRecord.Job: syncfilemod.PlayletWatermarkJob,
+		}, bson.M{"$set": bson.M{
+			syncfilemod.FN.SyncRecord.RecordTime: recordTime,
+			syncfilemod.FN.SyncRecord.SuccessAt:  time.Now(),
+			syncfilemod.FN.SyncRecord.UpdatedAt:  time.Now(),
+		}}); updateErr != nil {
+			log.Error("playletwatermark:SyncCompletedChapters 更新同步水印失败", log.E(updateErr))
+		}
+	}
+
+	reconcileProcessingChapters(ctx)
+	log.Info("playletwatermark:SyncCompletedChapters end",
+		log.Any("recordTime", recordTime),
+		log.Any("syncOK", syncOK))
+}
+
+// syncIncrementalCompletedChapters 按更新时间水印增量拉取。
+// 每次回退一段时间重复扫描，避免临界时间及同一时间戳分页造成漏数据；章节回写是幂等的。
+func syncIncrementalCompletedChapters(ctx context.Context, recordTime time.Time) (time.Time, bool) {
+	cursor := recordTime
+	if !cursor.IsZero() {
+		cursor = cursor.Add(-syncOverlapWindow)
+	}
+	maxRecordTime := recordTime
+	for {
+		req := proxy.MediaTaskSyncReq{
+			AppId:          pubdef.AppId,
+			LastUpdateTime: cursor,
+			PageSize:       syncPageSize,
+		}
+		result, syncErr := proxy.SyncMedia(&req, "")
+		if syncErr != nil || result == nil || result.Code != http.StatusOK {
+			log.Error("playletwatermark:SyncCompletedChapters SyncMedia 失败", log.Any("req", req), log.E(syncErr))
+			return recordTime, false
+		}
+		pageMaxTime := cursor
+		for _, media := range result.Data {
+			if media.UpdatedAt.After(pageMaxTime) {
+				pageMaxTime = media.UpdatedAt
+			}
+			if media.UpdatedAt.After(maxRecordTime) {
+				maxRecordTime = media.UpdatedAt
+			}
+			applyCompletedMedia(ctx, media, "incremental")
+		}
+		if int64(len(result.Data)) < syncPageSize {
+			break
+		}
+		if !pageMaxTime.After(cursor) {
+			log.Error("playletwatermark:SyncCompletedChapters 增量分页水印未推进",
+				log.Any("cursor", cursor),
+				log.Any("count", len(result.Data)))
+			return recordTime, false
+		}
+		cursor = pageMaxTime
+	}
+	return maxRecordTime, true
+}
+
+// reconcileProcessingChapters 反向对账所有处理中章节。
+// 它不依赖增量水印，可补偿本地更新失败、时间戳临界和历史水印越过造成的漏回写。
+func reconcileProcessingChapters(ctx context.Context) {
+	var lastID int64
+	for {
+		list, err := playletchaptermod.FindManyPlayletChapter(ctx, bson.M{
+			playletchaptermod.FN.PlayletChapter.Id:              bson.M{"$gt": lastID},
+			playletchaptermod.FN.PlayletChapter.WatermarkStatus: playletchaptermod.WatermarkStatusProcessing,
+			playletchaptermod.FN.PlayletChapter.MediaSourceId:   bson.M{"$gt": ""},
+		}, options.Find().
+			SetSort(bson.D{{Key: playletchaptermod.FN.PlayletChapter.Id, Value: 1}}).
+			SetLimit(reconcileBatchSize))
+		if err != nil {
+			log.Error("playletwatermark:reconcileProcessingChapters 查询章节失败", log.E(err))
+			return
+		}
+		if len(list) == 0 {
+			return
+		}
+
+		sourceIDs := make([]string, 0, len(list))
+		for _, item := range list {
+			if item == nil {
+				continue
+			}
+			sourceIDs = append(sourceIDs, item.MediaSourceId)
+			if item.Id > lastID {
+				lastID = item.Id
+			}
+		}
+		if len(sourceIDs) > 0 {
+			result, queryErr := proxy.QueryMedias(&proxy.MediaQueryesReq{
+				AppId:          pubdef.AppId,
+				MediaSourceIds: sourceIDs,
+			}, "")
+			if queryErr != nil || result == nil || result.Code != http.StatusOK {
+				log.Error("playletwatermark:reconcileProcessingChapters 查询 media 任务失败",
+					log.Any("sourceCount", len(sourceIDs)),
+					log.Any("result", result),
+					log.E(queryErr))
+				return
+			}
+			for _, media := range result.Data {
+				applyCompletedMedia(ctx, media, "reconcile")
+			}
+		}
+		if int64(len(list)) < reconcileBatchSize {
+			return
+		}
+	}
+}
+
+func applyCompletedMedia(ctx context.Context, media proxy.MediaResp, source string) bool {
+	sourceID := media.MediaSourceId.Hex()
+	if sourceID == "" || sourceID == "000000000000000000000000" {
+		log.Warn("playletwatermark:applyCompletedMedia mediaSourceId 无效",
+			log.Any("mediaTaskId", media.Id.Hex()),
+			log.Any("source", source))
+		return false
+	}
+	fieldPath := strings.TrimSpace(media.FieldNameFs)
+	if fieldPath == "" {
+		log.Warn("playletwatermark:applyCompletedMedia 视频路径为空",
+			log.Any("mediaSourceId", sourceID),
+			log.Any("mediaTaskId", media.Id.Hex()),
+			log.Any("source", source))
+		return false
+	}
+	if !strings.HasSuffix(strings.ToLower(fieldPath), ".m3u8") {
+		fieldPath += ".m3u8"
+	}
+	taskID := media.Id.Hex()
+	setDoc := bson.M{
+		playletchaptermod.FN.PlayletChapter.WatermarkVideoUrl: fieldPath,
+		playletchaptermod.FN.PlayletChapter.WatermarkStatus:   playletchaptermod.WatermarkStatusDone,
+		playletchaptermod.FN.PlayletChapter.MediaTaskId:       taskID,
+		playletchaptermod.FN.PlayletChapter.UpdatedAt:         time.Now(),
+	}
+	if sourceName := strings.TrimSpace(media.SourceNameFs); sourceName != "" &&
+		isOriginalPlayletBySourceID(ctx, sourceID) {
+		// 原创才回填原片地址，供换水印/重置复用；非原创保留媒资原始 VideoUrl。
+		setDoc[playletchaptermod.FN.PlayletChapter.VideoUrl] = sourceName
+	}
+	updateErr := playletchaptermod.UpdateOnePlayletChapter(ctx, bson.M{
+		playletchaptermod.FN.PlayletChapter.MediaSourceId:   sourceID,
+		playletchaptermod.FN.PlayletChapter.WatermarkStatus: playletchaptermod.WatermarkStatusProcessing,
+	}, bson.M{"$set": setDoc})
+	if updateErr == nil {
+		log.Info("playletwatermark:applyCompletedMedia 回写成功",
+			log.Any("mediaSourceId", sourceID),
+			log.Any("mediaTaskId", taskID),
+			log.Any("watermarkVideoUrl", fieldPath),
+			log.Any("source", source))
+		return true
+	}
+	if updateErr.Code != srvdef.ErrNoData {
+		log.Error("playletwatermark:applyCompletedMedia 更新章节失败",
+			log.Any("mediaSourceId", sourceID),
+			log.Any("source", source),
+			log.E(updateErr))
+		return false
+	}
+
+	chapter, findErr := playletchaptermod.FindOnePlayletChapter(ctx, bson.M{
+		playletchaptermod.FN.PlayletChapter.MediaSourceId: sourceID,
+	})
+	if findErr != nil {
+		if findErr.Code != srvdef.ErrNoData {
+			log.Error("playletwatermark:applyCompletedMedia 查询未匹配章节失败",
+				log.Any("mediaSourceId", sourceID),
+				log.E(findErr))
+		} else {
+			log.Warn("playletwatermark:applyCompletedMedia 未找到对应章节",
+				log.Any("mediaSourceId", sourceID),
+				log.Any("mediaTaskId", taskID),
+				log.Any("source", source))
+		}
+		return false
+	}
+	if chapter.WatermarkStatus != playletchaptermod.WatermarkStatusDone {
+		log.Warn("playletwatermark:applyCompletedMedia 章节状态不允许回写",
+			log.Any("chapterId", chapter.Id),
+			log.Any("mediaSourceId", sourceID),
+			log.Any("watermarkStatus", chapter.WatermarkStatus),
+			log.Any("source", source))
+	}
+	return false
+}
+
+func isOriginalPlayletBySourceID(ctx context.Context, sourceID string) bool {
+	chapter, err := playletchaptermod.FindOnePlayletChapter(ctx, bson.M{
+		playletchaptermod.FN.PlayletChapter.MediaSourceId: sourceID,
+	}, options.FindOne().SetProjection(bson.M{
+		playletchaptermod.FN.PlayletChapter.PlayletId: 1,
+	}))
+	if err != nil || chapter == nil || chapter.PlayletId <= 0 {
+		return false
+	}
+	playlet, err := playletmod.FindOnePlaylet(ctx, bson.M{
+		playletmod.FN.Playlet.Id: chapter.PlayletId,
+	}, options.FindOne().SetProjection(bson.M{
+		playletmod.FN.Playlet.IsOriginal: 1,
+	}))
+	return err == nil && playlet != nil && playlet.IsOriginal
+}
+
+func ensureSyncRecord(ctx context.Context) (*syncfilemod.SyncRecord, *srvdef.SrvErr) {
+	filter := bson.M{syncfilemod.FN.SyncRecord.Job: syncfilemod.PlayletWatermarkJob}
+	record, err := syncfilemod.FindOneSyncRecord(ctx, filter)
+	if err == nil {
+		return record, nil
+	}
+	if err.Code != srvdef.ObjectDayStatDoesNotExist {
+		return nil, err
+	}
+	now := time.Now()
+	record = &syncfilemod.SyncRecord{
+		Job:        syncfilemod.PlayletWatermarkJob,
+		RecordTime: time.Time{},
+		CreatedAt:  now,
+		UpdatedAt:  now,
+	}
+	if insertErr := syncfilemod.InsertOneSyncRecord(ctx, record); insertErr != nil {
+		return nil, insertErr
+	}
+	return record, nil
+}