package slugser

import (
	"context"
	"strconv"
	"strings"
	"videoweb/pub/common/srvdef"
	"videoweb/pub/models/actress"
	"videoweb/pub/models/contenttagmod"
	"videoweb/pub/models/playletcategorymod"
	"videoweb/pub/models/playletmod"
	"videoweb/pub/models/videocategorymod"
	"videoweb/pub/models/videoinformationmod"

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

// Resolve 解析实体ID：slug 非空时先按 slug 查 ID，否则使用传入 ID。
func Resolve(id int64, slug string, required bool, emptyMsg string, lookup func(string) (int64, *srvdef.SrvErr)) (int64, *srvdef.SrvErr) {
	slug = strings.TrimSpace(slug)
	if slug != "" {
		if lookup == nil {
			return 0, srvdef.New(srvdef.ErrParamError, emptyMsg, emptyMsg)
		}
		return lookup(slug)
	}
	if required && id <= 0 {
		if emptyMsg == "" {
			emptyMsg = "参数错误"
		}
		return 0, srvdef.New(srvdef.ErrParamError, emptyMsg, emptyMsg)
	}
	return id, nil
}

// ParseIDOrSlug 解析路径参数：纯正整数当作ID，否则当作slug。
func ParseIDOrSlug(raw string) (int64, string) {
	raw = strings.TrimSpace(raw)
	if raw == "" {
		return 0, ""
	}
	if id, err := strconv.ParseInt(raw, 10, 64); err == nil && id > 0 {
		return id, ""
	}
	return 0, raw
}

func VideoID(ctx context.Context, id int64, slug string, required bool) (int64, *srvdef.SrvErr) {
	return Resolve(id, slug, required, "视频ID不能为空", func(s string) (int64, *srvdef.SrvErr) {
		item, err := videoinformationmod.FindOneVideoInformation(ctx, bson.M{
			videoinformationmod.FN.VideoInformation.Slug: s,
		}, options.FindOne().SetProjection(bson.M{videoinformationmod.FN.VideoInformation.Id: 1}))
		if err != nil {
			return 0, err
		}
		return item.Id, nil
	})
}

func VideoCategoryID(ctx context.Context, id int64, slug string, required bool) (int64, *srvdef.SrvErr) {
	return Resolve(id, slug, required, "视频分类ID不能为空", func(s string) (int64, *srvdef.SrvErr) {
		item, err := videocategorymod.FindOneVideoCategory(ctx, bson.M{
			videocategorymod.FN.VideoCategory.Slug: s,
		})
		if err != nil {
			return 0, err
		}
		return item.Id, nil
	})
}

func PlayletID(ctx context.Context, id int64, slug string, required bool) (int64, *srvdef.SrvErr) {
	return Resolve(id, slug, required, "短剧ID不能为空", func(s string) (int64, *srvdef.SrvErr) {
		item, err := playletmod.FindOnePlaylet(ctx, bson.M{
			playletmod.FN.Playlet.Slug: s,
		}, options.FindOne().SetProjection(bson.M{playletmod.FN.Playlet.Id: 1}))
		if err != nil {
			return 0, err
		}
		return item.Id, nil
	})
}

func PlayletCategoryID(ctx context.Context, id int64, slug string, required bool) (int64, *srvdef.SrvErr) {
	return Resolve(id, slug, required, "短剧分类ID不能为空", func(s string) (int64, *srvdef.SrvErr) {
		item, err := playletcategorymod.FindOnePlayletCategory(ctx, bson.M{
			playletcategorymod.FN.PlayletCategory.Slug: s,
		}, options.FindOne().SetProjection(bson.M{playletcategorymod.FN.PlayletCategory.Id: 1}))
		if err != nil {
			return 0, err
		}
		return item.Id, nil
	})
}

func ActressID(ctx context.Context, id int64, slug string, required bool) (int64, *srvdef.SrvErr) {
	return Resolve(id, slug, required, "女优ID不能为空", func(s string) (int64, *srvdef.SrvErr) {
		item, err := actress.FindOneActress(ctx, bson.M{
			actress.FN.Actress.Slug: s,
		})
		if err != nil {
			return 0, err
		}
		return item.Id, nil
	})
}

func ContentTagID(ctx context.Context, id int64, slug string, required bool) (int64, *srvdef.SrvErr) {
	return Resolve(id, slug, required, "标签ID不能为空", func(s string) (int64, *srvdef.SrvErr) {
		item, err := contenttagmod.FindOneContentTag(ctx, bson.M{
			contenttagmod.FN.ContentTag.Slug: s,
		}, options.FindOne().SetProjection(bson.M{contenttagmod.FN.ContentTag.Id: 1}))
		if err != nil {
			return 0, err
		}
		return item.Id, nil
	})
}