feat: support LEAK watermark

capture/capture.go

@@ -72,6 +72,9 @@ func (c *Capture) resolveFileInfo(fc *model.FileContext, file string) error {
 	if fc.Number.IsChineseSubtitle() {
 		fc.SaveFileBase += "-C"
 	}
+	if fc.Number.IsLeak() {
+		fc.SaveFileBase += "-LEAK"
+	}
 	if fc.Number.IsMultiCD() {
 		fc.SaveFileBase += "-CD" + strconv.FormatInt(int64(fc.Number.MultiCDIndex()), 10)
 	}

constant/constant.go

@@ -4,4 +4,5 @@ const (
 	TagUncensored      = "无码"
 	TagChineseSubtitle = "中文字幕"
 	Tag4K              = "4K"
+	TagLeak            = "无码流出"
 )

image/resource/resource.go

@@ -12,3 +12,6 @@ var ResIMGUncensored []byte
 
 //go:embed 4k.png
 var ResIMG4K []byte
+
+//go:embed leak.png
+var ResIMGLeak []byte

image/watermark.go

@@ -14,8 +14,22 @@ const (
 	WMChineseSubtitle Watermark = 1
 	WMUncensored      Watermark = 2
 	WM4K              Watermark = 3
+	WMLeak            Watermark = 4
 )
 
+var resMap = make(map[Watermark][]byte)
+
+func registerResource() {
+	resMap[WMChineseSubtitle] = resource.ResIMGSubtitle
+	resMap[WM4K] = resource.ResIMG4K
+	resMap[WMUncensored] = resource.ResIMGUncensored
+	resMap[WMLeak] = resource.ResIMGLeak
+}
+
+func init() {
+	registerResource()
+}
+
 const (
 	defaultMaxWaterMarkCount               = 4                    //最大的水印个数
 	defaultWaterMarkWidthToImageWidthRatio = float64(31.58) / 100 //水印与整张图片的宽度比, W(watermark)/W(image) = 0.3158
@@ -57,18 +71,8 @@ func addWatermarkToImage(img image.Image, wms []image.Image) (image.Image, error
 }
 
 func selectWatermarkResource(w Watermark) ([]byte, bool) {
-	var out []byte
-	switch w {
-	case WMChineseSubtitle:
-		out = resource.ResIMGSubtitle
-	case WMUncensored:
-		out = resource.ResIMGUncensored
-	case WM4K:
-		out = resource.ResIMG4K
-	default:
-		break
-	}
-	if len(out) == 0 {
+	out, ok := resMap[w]
+	if !ok {
 		return nil, false
 	}
 	rs := make([]byte, len(out))

number/model.go

@@ -7,6 +7,7 @@ type Number struct {
 	multiCDIndex      int
 	isUncensorMovie   bool
 	is4k              bool
+	isLeak            bool
 }
 
 func (n *Number) Number() string {
@@ -32,3 +33,7 @@ func (n *Number) IsUncensorMovie() bool {
 func (n *Number) Is4K() bool {
 	return n.is4k
 }
+
+func (n *Number) IsLeak() bool {
+	return n.isLeak
+}

number/number.go

@@ -14,6 +14,7 @@ var defaultSuffixResolverList = []suffixInfoResolveFunc{
 	resolveIsChineseSubTitle,
 	resolveCDInfo,
 	resolve4K,
+	resolveLeak,
 }
 
 var defaultNumberInfoResolverList = []numberInfoResolveFunc{
@@ -66,6 +67,14 @@ func resolveCDInfo(info *Number, str string) bool {
 	return true
 }
 
+func resolveLeak(info *Number, str string) bool {
+	if str != "LEAK" {
+		return false
+	}
+	info.isLeak = true
+	return true
+}
+
 func resolve4K(info *Number, str string) bool {
 	if str != "4K" {
 		return false

number/number_test.go

@@ -66,6 +66,11 @@ func TestNumber(t *testing.T) {
 			isChineseSubtitle: true,
 			is4k:              true,
 		},
+		"abc-leak-c.mp4": {
+			number:            "ABC",
+			isLeak:            true,
+			isChineseSubtitle: true,
+		},
 	}
 	for file, info := range checkList {
 		rs, err := ParseWithFileName(file)

processor/handler/tag_padder.go

@@ -19,6 +19,9 @@ func (h *tagPadder) Handle(ctx context.Context, fc *model.FileContext) error {
 	if fc.Number.Is4K() {
 		fc.Meta.Genres = append(fc.Meta.Genres, constant.Tag4K)
 	}
+	if fc.Number.IsLeak() {
+		fc.Meta.Genres = append(fc.Meta.Genres, constant.TagLeak)
+	}
 	fc.Meta.Genres = utils.DedupStringList(fc.Meta.Genres)
 	return nil
 }

processor/handler/watermark_handler.go

@@ -31,6 +31,9 @@ func (h *watermark) Handle(ctx context.Context, fc *model.FileContext) error {
 	if fc.Number.IsChineseSubtitle() {
 		tags = append(tags, image.WMChineseSubtitle)
 	}
+	if fc.Number.IsLeak() {
+		tags = append(tags, image.WMLeak)
+	}
 	if len(tags) == 0 {
 		logutil.GetLogger(ctx).Debug("no watermark tag found, skip watermark proc")
 		return nil