feat(strm): keep local download file (#1707)

Author: sevxn007

drivers/strm/hook.go

@@ -27,7 +27,7 @@ func UpdateLocalStrm(ctx context.Context, path string, objs []model.Obj) {
 			localPath := stdpath.Join(localParentPath, obj.GetName())
 			generateStrm(ctx, driver, obj, localPath)
 		}
-		deleteExtraFiles(localParentPath, objs)
+		deleteExtraFiles(driver, localParentPath, objs)
 	}
 
 	_ = strmTrie.VisitPrefixes(patricia.Prefix(path), func(needPathPrefix patricia.Prefix, item patricia.Item) error {
@@ -129,23 +129,37 @@ func generateStrm(ctx context.Context, driver *Strm, obj model.Obj, localPath st
 	}
 }
 
-func deleteExtraFiles(localPath string, objs []model.Obj) {
+func deleteExtraFiles(driver *Strm, localPath string, objs []model.Obj) {
 	localFiles, err := getLocalFiles(localPath)
 	if err != nil {
 		log.Errorf("Failed to read local files from %s: %v", localPath, err)
 		return
 	}
 
 	objsSet := make(map[string]struct{})
+	objsBaseNameSet := make(map[string]struct{})
 	for _, obj := range objs {
 		if obj.IsDir() {
 			continue
 		}
-		objsSet[stdpath.Join(localPath, obj.GetName())] = struct{}{}
+		objName := obj.GetName()
+		objsSet[stdpath.Join(localPath, objName)] = struct{}{}
+
+		objBaseName := strings.TrimSuffix(objName, utils.SourceExt(objName))
+		objsBaseNameSet[stdpath.Join(localPath, objBaseName[:len(objBaseName)-1])] = struct{}{}
 	}
 
 	for _, localFile := range localFiles {
 		if _, exists := objsSet[localFile]; !exists {
+			ext := utils.Ext(localFile)
+			localFileName := stdpath.Base(localFile)
+			localFileBaseName := strings.TrimSuffix(localFile, utils.SourceExt(localFileName))
+			_, nameExists := objsBaseNameSet[localFileBaseName[:len(localFileBaseName)-1]]
+			_, downloadFile := driver.downloadSuffix[ext]
+			if driver.KeepLocalDownloadFile && nameExists && downloadFile {
+				continue
+			}
+
 			err := os.Remove(localFile)
 			if err != nil {
 				log.Errorf("Failed to delete file: %s, error: %v\n", localFile, err)

drivers/strm/meta.go

@@ -6,16 +6,17 @@ import (
 )
 
 type Addition struct {
-	Paths             string `json:"paths" required:"true" type:"text"`
-	SiteUrl           string `json:"siteUrl" type:"text" required:"false" help:"The prefix URL of the strm file"`
-	PathPrefix        string `json:"PathPrefix" type:"text" required:"false" default:"/d"  help:"Path prefix"`
-	DownloadFileTypes string `json:"downloadFileTypes" type:"text" default:"ass,srt,vtt,sub,strm" required:"false" help:"Files need to download with strm (usally subtitles)"`
-	FilterFileTypes   string `json:"filterFileTypes" type:"text" default:"mp4,mkv,flv,avi,wmv,ts,rmvb,webm,mp3,flac,aac,wav,ogg,m4a,wma,alac" required:"false" help:"Supports suffix name of strm file"`
-	EncodePath        bool   `json:"encodePath" default:"true" required:"true" help:"encode the path in the strm file"`
-	WithoutUrl        bool   `json:"withoutUrl" default:"false" help:"strm file content without URL prefix"`
-	SaveStrmToLocal   bool   `json:"SaveStrmToLocal" default:"false" help:"save strm file locally"`
-	SaveStrmLocalPath string `json:"SaveStrmLocalPath" type:"text" help:"save strm file local path"`
-	Version           int
+	Paths                 string `json:"paths" required:"true" type:"text"`
+	SiteUrl               string `json:"siteUrl" type:"text" required:"false" help:"The prefix URL of the strm file"`
+	PathPrefix            string `json:"PathPrefix" type:"text" required:"false" default:"/d"  help:"Path prefix"`
+	DownloadFileTypes     string `json:"downloadFileTypes" type:"text" default:"ass,srt,vtt,sub,strm" required:"false" help:"Files need to download with strm (usally subtitles)"`
+	FilterFileTypes       string `json:"filterFileTypes" type:"text" default:"mp4,mkv,flv,avi,wmv,ts,rmvb,webm,mp3,flac,aac,wav,ogg,m4a,wma,alac" required:"false" help:"Supports suffix name of strm file"`
+	EncodePath            bool   `json:"encodePath" default:"true" required:"true" help:"encode the path in the strm file"`
+	WithoutUrl            bool   `json:"withoutUrl" default:"false" help:"strm file content without URL prefix"`
+	SaveStrmToLocal       bool   `json:"SaveStrmToLocal" default:"false" help:"save strm file locally"`
+	SaveStrmLocalPath     string `json:"SaveStrmLocalPath" type:"text" help:"save strm file local path"`
+	KeepLocalDownloadFile bool   `json:"KeepLocalDownloadFile" default:"false" help:"keep local download files"`
+	Version               int
 }
 
 var config = driver.Config{