fix(chaoxing): JSON parsing error in content field (AlistGo#5877)

* fix(chaoxing):fix JSON parsing error in `content` field * fix(chaoxing): optimizing `UnmarshalJSON` implementation * fix(chaoxing): use `objectID` when is empty
EtherealAO · Apr 12, 2024 · 8562405 · 8562405
1 parent e8f9f94
commit 8562405
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 38 deletions.
diff --git a/drivers/chaoxing/types.go b/drivers/chaoxing/types.go
@@ -1,7 +1,9 @@
 package chaoxing
 
 import (
+	"bytes"
 	"fmt"
+	"strconv"
 	"time"
 
 	"github.com/alist-org/alist/v3/internal/model"
@@ -88,44 +90,59 @@ type UserAuth struct {
 	} `json:"operationAuth"`
 }
 
+// 手机端学习通上传的文件的json内容(content字段)与网页端上传的有所不同
+// 网页端json `"puid": 54321, "size": 12345`
+// 手机端json `"puid": "54321". "size": "12345"`
+type int_str int
+
+// json 字符串数字和纯数字解析
+func (ios *int_str) UnmarshalJSON(data []byte) error {
+	intValue, err := strconv.Atoi(string(bytes.Trim(data, "\"")))
+	if err != nil {
+		return err
+	}
+	*ios = int_str(intValue)
+	return nil
+}
+
 type File struct {
 	Cataid  int `json:"cataid"`
 	Cfid    int `json:"cfid"`
 	Content struct {
-		Cfid             int    `json:"cfid"`
-		Pid              int    `json:"pid"`
-		FolderName       string `json:"folderName"`
-		ShareType        int    `json:"shareType"`
-		Preview          string `json:"preview"`
-		Filetype         string `json:"filetype"`
-		PreviewURL       string `json:"previewUrl"`
-		IsImg            bool   `json:"isImg"`
-		ParentPath       string `json:"parentPath"`
-		Icon             string `json:"icon"`
-		Suffix           string `json:"suffix"`
-		Duration         int    `json:"duration"`
-		Pantype          string `json:"pantype"`
-		Puid             int    `json:"puid"`
-		Filepath         string `json:"filepath"`
-		Crc              string `json:"crc"`
-		Isfile           bool   `json:"isfile"`
-		Residstr         string `json:"residstr"`
-		ObjectID         string `json:"objectId"`
-		Extinfo          string `json:"extinfo"`
-		Thumbnail        string `json:"thumbnail"`
-		Creator          int    `json:"creator"`
-		ResTypeValue     int    `json:"resTypeValue"`
-		UploadDateFormat string `json:"uploadDateFormat"`
-		DisableOpt       bool   `json:"disableOpt"`
-		DownPath         string `json:"downPath"`
-		Sort             int    `json:"sort"`
-		Topsort          int    `json:"topsort"`
-		Restype          string `json:"restype"`
-		Size             int    `json:"size"`
-		UploadDate       string `json:"uploadDate"`
-		FileSize         string `json:"fileSize"`
-		Name             string `json:"name"`
-		FileID           string `json:"fileId"`
+		Cfid             int     `json:"cfid"`
+		Pid              int     `json:"pid"`
+		FolderName       string  `json:"folderName"`
+		ShareType        int     `json:"shareType"`
+		Preview          string  `json:"preview"`
+		Filetype         string  `json:"filetype"`
+		PreviewURL       string  `json:"previewUrl"`
+		IsImg            bool    `json:"isImg"`
+		ParentPath       string  `json:"parentPath"`
+		Icon             string  `json:"icon"`
+		Suffix           string  `json:"suffix"`
+		Duration         int     `json:"duration"`
+		Pantype          string  `json:"pantype"`
+		Puid             int_str `json:"puid"`
+		Filepath         string  `json:"filepath"`
+		Crc              string  `json:"crc"`
+		Isfile           bool    `json:"isfile"`
+		Residstr         string  `json:"residstr"`
+		ObjectID         string  `json:"objectId"`
+		Extinfo          string  `json:"extinfo"`
+		Thumbnail        string  `json:"thumbnail"`
+		Creator          int     `json:"creator"`
+		ResTypeValue     int     `json:"resTypeValue"`
+		UploadDateFormat string  `json:"uploadDateFormat"`
+		DisableOpt       bool    `json:"disableOpt"`
+		DownPath         string  `json:"downPath"`
+		Sort             int     `json:"sort"`
+		Topsort          int     `json:"topsort"`
+		Restype          string  `json:"restype"`
+		Size             int_str `json:"size"`
+		UploadDate       string  `json:"uploadDate"`
+		FileSize         string  `json:"fileSize"`
+		Name             string  `json:"name"`
+		FileID           string  `json:"fileId"`
 	} `json:"content"`
 	CreatorID  int    `json:"creatorId"`
 	DesID      string `json:"des_id"`
@@ -204,7 +221,6 @@ type UploadFileDataRsp struct {
 	} `json:"data"`
 }
 
-
 type UploadDoneParam struct {
 	Cataid string `json:"cataid"`
 	Key    string `json:"key"`

diff --git a/drivers/chaoxing/util.go b/drivers/chaoxing/util.go
@@ -79,7 +79,7 @@ func (d *ChaoXing) GetFiles(parent string) ([]File, error) {
 		return nil, err
 	}
 	if resp.Result != 1 {
-		msg:=fmt.Sprintf("error code is:%d", resp.Result)
+		msg := fmt.Sprintf("error code is:%d", resp.Result)
 		return nil, errors.New(msg)
 	}
 	if len(resp.List) > 0 {
@@ -97,8 +97,12 @@ func (d *ChaoXing) GetFiles(parent string) ([]File, error) {
 	if err != nil {
 		return nil, err
 	}
-	if len(resps.List) > 0 {
-		files = append(files, resps.List...)
+	for _, file := range resps.List {
+		// 手机端超星上传的文件没有fileID字段，但ObjectID与fileID相同，可代替
+		if file.Content.FileID == "" {
+			file.Content.FileID = file.Content.ObjectID
+		}
+		files = append(files, file)
 	}
 	return files, nil
 }