随机修改文件名

转存秒传随机修改文件名增加成功率

baidupcs/transfer.go

@@ -20,6 +20,7 @@ type (
 	TransferOption struct {
 		Download bool // 是否直接开始下载
 		Collect  bool // 多文件整合
+		Rname    bool // 随机改文件名
 	}
 )
 

internal/pcscommand/transfer.go

@@ -18,7 +18,7 @@ func RunShareTransfer(params []string, opt *baidupcs.TransferOption) {
 	if len(params) == 1 {
 		link = params[0]
 		if strings.Contains(link, "bdlink=") || !strings.Contains(link, "pan.baidu.com/") {
-			RunRapidTransfer(link)
+			RunRapidTransfer(link, opt.Rname)
 			//fmt.Printf("%s失败: %s\n", baidupcs.OperationShareFileSavetoLocal, "秒传已不再被支持")
 			return
 		}
@@ -114,7 +114,7 @@ func RunShareTransfer(params []string, opt *baidupcs.TransferOption) {
 }
 
 // RunRapidTransfer 执行秒传链接解析及保存
-func RunRapidTransfer(link string) {
+func RunRapidTransfer(link string, rnameOpt ...bool) {
 	if strings.Contains(link, "bdlink=") || strings.Contains(link, "bdpan://") {
 		r, _ := regexp.Compile(`(bdlink=|bdpan://)([^\s]+)`)
 		link1 := r.FindStringSubmatch(link)[2]
@@ -125,17 +125,21 @@ func RunRapidTransfer(link string) {
 		}
 		link = string(decodeBytes)
 	}
+	rname := false
+	if len(rnameOpt) > 0 {
+		rname = rnameOpt[0]
+	}
 	link = strings.TrimSpace(link)
 	substrs := strings.SplitN(link, "#", 4)
 	if len(substrs) == 4 {
 		md5, slicemd5 := substrs[0], substrs[1]
 		size, _ := strconv.ParseInt(substrs[2], 10, 64)
-		filename := path.Join(GetActiveUser().Workdir, substrs[3])
+		filename := path.Join(GetActiveUser().Workdir, randReplaceStr(substrs[3], rname))
 		RunRapidUpload(filename, md5, slicemd5, size)
 	} else if len(substrs) == 3 {
 		md5 := substrs[0]
 		size, _ := strconv.ParseInt(substrs[1], 10, 64)
-		filename := path.Join(GetActiveUser().Workdir, substrs[2])
+		filename := path.Join(GetActiveUser().Workdir, randReplaceStr(substrs[2], rname))
 		RunRapidUpload(filename, md5, "", size)
 	} else {
 		fmt.Printf("%s失败: %s\n", baidupcs.OperationRapidLinkSavetoLocal, "秒传链接格式错误")

internal/pcscommand/util.go

@@ -3,8 +3,12 @@ package pcscommand
 import (
 	"errors"
 	"fmt"
+	"math/rand"
+	"path"
 )
 
+const letters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
+
 var (
 	// ErrShellPatternMultiRes 多条通配符匹配结果
 	ErrShellPatternMultiRes = errors.New("多条通配符匹配结果")
@@ -62,3 +66,23 @@ func matchPathByShellPattern(patterns ...string) (pcspaths []string, err error)
 	}
 	return pcspaths, nil
 }
+
+
+
+func randReplaceStr(s string, rname bool) string {
+	if !rname {
+		return s
+	}
+	filenameAll := path.Base(s)
+	fileSuffix := path.Ext(s)
+	filePrefix := filenameAll[0:len(filenameAll) - len(fileSuffix)]
+	runes := []rune(filePrefix)
+
+	for i := 0; i< len(filePrefix); i++ {
+		runes[i] = rune(letters[rand.Int63()%int64(len(letters))])
+		if i == 3 {
+			break
+		}
+	}
+	return path.Join(path.Dir(s), string(runes) + fileSuffix)
+}

main.go

@@ -1477,6 +1477,7 @@ func main() {
 				opt := &baidupcs.TransferOption{
 					Download: c.Bool("download"),
 					Collect:  c.Bool("collect"),
+					Rname:    c.Bool("rname"),
 				}
 				pcscommand.RunShareTransfer(c.Args(), opt)
 				return nil
@@ -1490,6 +1491,10 @@ func main() {
 					Name:  "collect",
 					Usage: "多文件整合到一个文件夹中转存",
 				},
+				cli.BoolFlag{
+					Name:  "rname",
+					Usage: "秒传随机替换4位文件名提高成功率",
+				},
 			},
 		},
 		{