web_huangguo_server/web/service/srv_playlet/theme.go

package srv_playlet

import (
	"context"
	"regexp"
	"strings"
	"time"
	"videoweb/pub/common/response"
	"videoweb/pub/common/srvdef"
	"videoweb/pub/models/playletmod"
	"videoweb/pub/models/playletthememod"
	"videoweb/web/api/proto"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo/options"
)

// ListPlayletTheme 分页查询题材列表
func ListPlayletTheme(ctx context.Context, req *proto.PlayletThemeListReq) (*proto.PlayletThemeListPageResp, *srvdef.SrvErr) {
	if req == nil {
		return nil, srvdef.New(srvdef.ErrParamError, "", "")
	}
	filter := bson.M{}
	if req.Id != nil && *req.Id > 0 {
		filter[playletthememod.FN.PlayletTheme.Id] = *req.Id
	}
	if req.Status != nil {
		filter[playletthememod.FN.PlayletTheme.Status] = *req.Status
	}
	if req.IsRecommend != nil {
		filter[playletthememod.FN.PlayletTheme.IsRecommend] = *req.IsRecommend
	}
	if name := strings.TrimSpace(req.Name); name != "" {
		filter[playletthememod.FN.PlayletTheme.Name] = bson.M{"$regex": regexp.QuoteMeta(name)}
	}
	if slug := strings.TrimSpace(req.Slug); slug != "" {
		filter[playletthememod.FN.PlayletTheme.Slug] = slug
	}
	pageNum, pageSize := normalizePage(req.PageNum, req.PageSize)
	totalCount, err := playletthememod.CountPlayletTheme(ctx, filter)
	if err != nil {
		return nil, err
	}
	resp := &proto.PlayletThemeListPageResp{
		BasePageResp: response.BasePageResp{
			TotalCount: totalCount,
			PageSize:   int(pageSize),
			PageNum:    int(pageNum),
			List:       make([]*proto.PlayletThemeListResp, 0),
		},
	}
	if totalCount == 0 {
		return resp, nil
	}
	list, err := playletthememod.FindManyPlayletTheme(ctx, filter, options.Find().
		SetSort(bson.D{
			{Key: playletthememod.FN.PlayletTheme.Rank, Value: -1},
			{Key: playletthememod.FN.PlayletTheme.Id, Value: -1},
		}).
		SetSkip((pageNum-1)*pageSize).
		SetLimit(pageSize))
	if err != nil {
		return nil, err
	}
	items := make([]*proto.PlayletThemeListResp, 0, len(list))
	for _, item := range list {
		if item == nil {
			continue
		}
		items = append(items, &proto.PlayletThemeListResp{
			Id:          item.Id,
			Name:        item.Name,
			Slug:        item.Slug,
			Icon:        item.Icon,
			Status:      item.Status,
			IsRecommend: item.IsRecommend,
			Rank:        item.Rank,
			SeoInfo:     normalizeSeoValue(item.SeoInfo),
			CreatedAt:   item.CreatedAt,
			UpdatedAt:   item.UpdatedAt,
		})
	}
	resp.List = items
	return resp, nil
}

// AddPlayletTheme 新增题材
func AddPlayletTheme(ctx context.Context, req *proto.PlayletThemeAddReq) (*proto.PlayletThemeAddResp, *srvdef.SrvErr) {
	if req == nil || strings.TrimSpace(req.Name) == "" {
		return nil, srvdef.New(srvdef.ErrParamError, "题材名称不能为空", "")
	}
	doc := &playletthememod.PlayletTheme{
		Name:        strings.TrimSpace(req.Name),
		Slug:        strings.TrimSpace(req.Slug),
		Icon:        req.Icon,
		Status:      req.Status,
		IsRecommend: req.IsRecommend,
		Rank:        req.Rank,
		SeoInfo:     normalizeSeoValue(req.SeoInfo),
	}
	if err := ensureThemeSlug(ctx, doc, 0); err != nil {
		return nil, err
	}
	if err := playletthememod.InsertOnePlayletTheme(ctx, doc); err != nil {
		return nil, err
	}
	clearAppPlayletHomeCache()
	return &proto.PlayletThemeAddResp{Id: doc.Id}, nil
}

// UpdatePlayletTheme 更新题材
func UpdatePlayletTheme(ctx context.Context, req *proto.PlayletThemeUpdateReq) *srvdef.SrvErr {
	if req == nil || req.Id <= 0 {
		return srvdef.New(srvdef.ErrParamError, "id无效", "")
	}
	info, err := playletthememod.FindOnePlayletTheme(ctx, bson.M{playletthememod.FN.PlayletTheme.Id: req.Id})
	if err != nil {
		return err
	}
	setDoc := bson.M{playletthememod.FN.PlayletTheme.UpdatedAt: time.Now()}
	if req.Name != nil {
		name := strings.TrimSpace(*req.Name)
		if name == "" {
			return srvdef.New(srvdef.ErrParamError, "题材名称不能为空", "")
		}
		setDoc[playletthememod.FN.PlayletTheme.Name] = name
	}
	if req.Icon != nil {
		setDoc[playletthememod.FN.PlayletTheme.Icon] = *req.Icon
	}
	if req.Status != nil {
		setDoc[playletthememod.FN.PlayletTheme.Status] = *req.Status
	}
	if req.IsRecommend != nil {
		setDoc[playletthememod.FN.PlayletTheme.IsRecommend] = *req.IsRecommend
	}
	if req.Rank != nil {
		setDoc[playletthememod.FN.PlayletTheme.Rank] = *req.Rank
	}
	if req.SeoInfo != nil {
		setDoc["seoInfo"] = normalizeSeoValue(*req.SeoInfo)
	}
	if err := prepareThemeSlugOnUpdate(ctx, req.Id, info, req, setDoc); err != nil {
		return err
	}
	if len(setDoc) <= 1 {
		return srvdef.New(srvdef.ErrParamError, "没有可更新字段", "")
	}
	if err := playletthememod.UpdateOnePlayletTheme(ctx,
		bson.M{playletthememod.FN.PlayletTheme.Id: req.Id},
		bson.M{"$set": setDoc},
	); err != nil {
		return err
	}
	clearAppPlayletHomeCache()
	return nil
}

// SetPlayletThemeRecommend 单/批量设置题材是否推荐
func SetPlayletThemeRecommend(ctx context.Context, req *proto.PlayletThemeRecommendSetReq) *srvdef.SrvErr {
	if req == nil {
		return srvdef.New(srvdef.ErrParamError, "参数不能为空", "")
	}
	ids := dedupePositiveInt64Ids(req.Ids)
	if len(ids) == 0 {
		return srvdef.New(srvdef.ErrParamError, "题材ID不能为空", "")
	}
	if err := playletthememod.UpdateManyPlayletTheme(ctx,
		bson.M{playletthememod.FN.PlayletTheme.Id: bson.M{"$in": ids}},
		bson.M{"$set": bson.M{
			playletthememod.FN.PlayletTheme.IsRecommend: req.IsRecommend,
			playletthememod.FN.PlayletTheme.UpdatedAt:   time.Now(),
		}},
	); err != nil {
		return err
	}
	clearAppPlayletHomeCache()
	return nil
}

// DelPlayletTheme 删除题材（被短剧引用时禁止删除）
func DelPlayletTheme(ctx context.Context, id int64) *srvdef.SrvErr {
	if id <= 0 {
		return srvdef.New(srvdef.ErrParamError, "id无效", "")
	}
	usedCount, err := playletmod.CountPlaylet(ctx, bson.M{
		playletmod.FN.Playlet.ThemeIds: bson.M{"$in": []int64{id}},
	})
	if err != nil {
		return err
	}
	if usedCount > 0 {
		return srvdef.New(srvdef.ErrParamError, "题材已被短剧引用，无法删除", "")
	}
	count, err := playletthememod.DelOnePlayletTheme(ctx, bson.M{playletthememod.FN.PlayletTheme.Id: id})
	if err != nil {
		return err
	}
	if count == 0 {
		return srvdef.New(srvdef.ErrNoData, "", "")
	}
	clearAppPlayletHomeCache()
	return nil
}