调整调试的细节

Signed-off-by: 716 <[email protected]>

common/common.go

@@ -4,4 +4,6 @@ import "time"
 
 const HTMLTimeOut = 20 * time.Second	// HttpClient 超时时间
 
-const DownloadSubsPerSite = 1 // 默认，每个网站下载一个字幕，允许额外传参调整
+const DownloadSubsPerSite = 1 // 默认，每个网站下载一个字幕，允许额外传参调整
+
+const DebugFolder = "debugThings"

common/util.go

@@ -4,6 +4,8 @@ import (
 	"fmt"
 	"github.com/go-resty/resty/v2"
 	"net/http"
+	"os"
+	"path"
 	"regexp"
 	"strings"
 )
@@ -86,6 +88,16 @@ func AddBaseUrl(baseUrl, url string) string {
 	return fmt.Sprintf("%s%s", baseUrl, url)
 }
 
+func GetDebugFolder() (string, error) {
+	nowProcessRoot, _ := os.Getwd()
+	nowProcessRoot = path.Join(nowProcessRoot, DebugFolder)
+	err := os.MkdirAll(nowProcessRoot, os.ModePerm)
+	if err != nil {
+		return "", err
+	}
+	return nowProcessRoot, err
+}
+
 // ReqParam 可选择传入的参数
 type ReqParam struct {
 	UserExtList []string	// 用户确认的视频后缀名支持列表

sub_supplier/subhd/subhd.go

@@ -14,6 +14,7 @@ import (
 	"math"
 	"net/url"
 	"os"
+	"path"
 	"path/filepath"
 	"regexp"
 	"strings"
@@ -431,7 +432,12 @@ search:
 
 	if s.reqParam.DebugMode == true {
 		//截圖保存
-		page.MustScreenshot("result.png")
+		nowProcessRoot, err := common.GetDebugFolder()
+		if err == nil {
+			page.MustScreenshot(path.Join(nowProcessRoot, "result.png"))
+		} else {
+			println(err.Error())
+		}
 	}
 }