update io and time (#769)

docs/misc/sign/wbi.md

@@ -282,7 +282,7 @@ import (
         "crypto/md5"
         "encoding/hex"
         "fmt"
-	"io/ioutil"
+        "io"
         "net/http"
         "net/url"
         "sort"
@@ -294,11 +294,93 @@ import (
         "github.com/tidwall/gjson"
 )
 
-var mixinKeyEncTab = []int{
+var (
+        mixinKeyEncTab = []int{
                 46, 47, 18, 2, 53, 8, 23, 32, 15, 50, 10, 31, 58, 3, 45, 35, 27, 43, 5, 49,
                 33, 9, 42, 19, 29, 28, 14, 39, 12, 38, 41, 13, 37, 48, 7, 16, 24, 55, 40,
                 61, 26, 17, 0, 1, 60, 51, 30, 4, 22, 25, 54, 21, 56, 59, 6, 63, 57, 62, 11,
                 36, 20, 34, 44, 52,
+        }
+        cache          sync.Map
+        lastUpdateTime time.Time
+)
+
+func main() {
+        urlStr := "https://api.bilibili.com/x/space/wbi/acc/info?mid=1850091"
+        newUrlStr, err := signAndGenerateURL(urlStr)
+        if err != nil {
+                fmt.Printf("Error: %s", err)
+                return
+        }
+        req, err := http.NewRequest("GET", newUrlStr, nil)
+        if err != nil {
+                fmt.Printf("Error: %s", err)
+                return
+        }
+        req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
+        response, err := http.DefaultClient.Do(req)
+        if err != nil {
+                fmt.Printf("Request failed: %s", err)
+                return
+        }
+        defer response.Body.Close()
+        body, err := io.ReadAll(response.Body)
+        if err != nil {
+                fmt.Printf("Failed to read response: %s", err)
+                return
+        }
+        fmt.Println(string(body))
+}
+
+func signAndGenerateURL(urlStr string) (string, error) {
+        urlObj, err := url.Parse(urlStr)
+        if err != nil {
+                return "", err
+        }
+        imgKey, subKey := getWbiKeysCached()
+        query := urlObj.Query()
+        params := map[string]string{}
+        for k, v := range query {
+                params[k] = v[0]
+        }
+        newParams := encWbi(params, imgKey, subKey)
+        for k, v := range newParams {
+                query.Set(k, v)
+        }
+        urlObj.RawQuery = query.Encode()
+        newUrlStr := urlObj.String()
+        return newUrlStr, nil
+}
+
+func encWbi(params map[string]string, imgKey, subKey string) map[string]string {
+        mixinKey := getMixinKey(imgKey + subKey)
+        currTime := strconv.FormatInt(time.Now().Unix(), 10)
+        params["wts"] = currTime
+
+        // Sort keys
+        keys := make([]string, 0, len(params))
+        for k := range params {
+                keys = append(keys, k)
+        }
+        sort.Strings(keys)
+
+        // Remove unwanted characters
+        for k, v := range params {
+                v = sanitizeString(v)
+                params[k] = v
+        }
+
+        // Build URL parameters
+        query := url.Values{}
+        for _, k := range keys {
+                query.Set(k, params[k])
+        }
+        queryStr := query.Encode()
+
+        // Calculate w_rid
+        hash := md5.Sum([]byte(queryStr + mixinKey))
+        params["w_rid"] = hex.EncodeToString(hash[:])
+        return params
 }
 
 func getMixinKey(orig string) string {
@@ -310,67 +392,42 @@ func getMixinKey(orig string) string {
         }
         return str.String()[:32]
 }
-func EncWbi(params map[string]string, imgKey string, subKey string) map[string]string {
+
-	mixinKey := getMixinKey(imgKey + subKey)
+func sanitizeString(s string) string {
-	currTime := strconv.FormatInt(time.Now().Unix(), 10)
+        unwantedChars := []string{"!", "'", "(", ")", "*"}
-	params["wts"] = currTime
+        for _, char := range unwantedChars {
-	// Sort keys
+                s = strings.ReplaceAll(s, char, "")
-	keys := make([]string, 0, len(params))
-	for k := range params {
-		keys = append(keys, k)
         }
-	sort.Strings(keys)
+        return s
-	// Remove unwanted characters
-	for k, v := range params {
-		v = strings.ReplaceAll(v, "!", "")
-		v = strings.ReplaceAll(v, "'", "")
-		v = strings.ReplaceAll(v, "(", "")
-		v = strings.ReplaceAll(v, ")", "")
-		v = strings.ReplaceAll(v, "*", "")
-		params[k] = v
-	}
-	// Build URL parameters
-	var str strings.Builder
-	for _, k := range keys {
-		str.WriteString(fmt.Sprintf("%s=%s&", k, params[k]))
-	}
-	query := strings.TrimSuffix(str.String(), "&")
-	// Calculate w_rid
-	hash := md5.Sum([]byte(query + mixinKey))
-	params["w_rid"] = hex.EncodeToString(hash[:])
-	return params
 }
 
-var cache sync.Map
-var lastUpdateTime time.Time
-
 func updateCache() {
-	if time.Now().Sub(lastUpdateTime).Minutes() < 10 {
+        if time.Since(lastUpdateTime).Minutes() < 10 {
                 return
         }
-	imgKey, subKey := GetWbiKeys()
+        imgKey, subKey := getWbiKeys()
         cache.Store("imgKey", imgKey)
         cache.Store("subKey", subKey)
         lastUpdateTime = time.Now()
 }
 
-func GetWbiKeysCached() (string, string) {
+func getWbiKeysCached() (string, string) {
         updateCache()
         imgKeyI, _ := cache.Load("imgKey")
         subKeyI, _ := cache.Load("subKey")
         return imgKeyI.(string), subKeyI.(string)
 }
 
-func GetWbiKeys() (string, string) {
+func getWbiKeys() (string, string) {
         resp, err := http.Get("https://api.bilibili.com/x/web-interface/nav")
         if err != nil {
-		fmt.Println("Error:", err)
+                fmt.Printf("Error: %s", err)
                 return "", ""
         }
         defer resp.Body.Close()
-	body, err := ioutil.ReadAll(resp.Body)
+        body, err := io.ReadAll(resp.Body)
         if err != nil {
-		fmt.Println("Error:", err)
+                fmt.Printf("Error: %s", err)
                 return "", ""
         }
         json := string(body)
@@ -380,47 +437,6 @@ func GetWbiKeys() (string, string) {
         subKey := strings.Split(strings.Split(subURL, "/")[len(strings.Split(subURL, "/"))-1], ".")[0]
         return imgKey, subKey
 }
-
-func SignURL(urlStr string) string {
-	urlObj, _ := url.Parse(urlStr)
-	imgKey, subKey := GetWbiKeysCached()
-	fmt.Println(imgKey, subKey)
-	query := urlObj.Query()
-	params := map[string]string{}
-	for k, v := range query {
-		params[k] = v[0]
-	}
-	newParams := EncWbi(params, imgKey, subKey)
-	for k, v := range newParams {
-		query.Set(k, v)
-	}
-	urlObj.RawQuery = query.Encode()
-	newUrlStr := urlObj.String()
-	return newUrlStr
-}
-func main() {
-	urlStr := "https://api.bilibili.com/x/space/wbi/acc/info?mid=1850091"
-	newUrlStr := utils.SignURL(urlStr) // 签名
-	req, err := http.NewRequest("GET", newUrlStr, nil)
-	if err != nil {
-		return nil, err
-	}
-	// 设置请求头，模拟浏览器
-	req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64)     AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
-	response, err := http.DefaultClient.Do(req)
-	fmt.Println(newUrlStr)
-	if err != nil {
-		fmt.Printf("请求发送失败：%s", err)
-		return nil, err
-	}
-	defer response.Body.Close()
-	body, err := ioutil.ReadAll(response.Body)
-	if err != nil {
-		fmt.Printf("读取响应失败：%s", err)
-		return nil, err
-	}
-	fmt.Println(string(body))
-}
 ```
 
 ### CSharp