Northern_Lights/alist
1
0
Fork 0
mirror of https://github.com/AlistGo/alist.git synced 2025-12-19 11:00:06 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

WPS Final Update. (Without Uploading)

Browse Source
This commit is contained in:
x-spy 2025-12-12 14:18:58 +08:00
parent e17baab2d5
commit afd5d09f2f
3 changed files with 270 additions and 247 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

162
drivers/wps/driver.go
View File

@ -3,7 +3,6 @@ package wps
import ( import (
"context" "context"
"fmt" "fmt"
"net/http"
"github.com/alist-org/alist/v3/internal/driver" "github.com/alist-org/alist/v3/internal/driver"
"github.com/alist-org/alist/v3/internal/errs" "github.com/alist-org/alist/v3/internal/errs"
@ -35,183 +34,44 @@ func (d *Wps) Drop(ctx context.Context) error {
return nil return nil
} }
func (d *Wps) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) { func (d *Wps) List(ctx context.Context, dir model.Obj, _ model.ListArgs) ([]model.Obj, error) {
basePath := "/" basePath := "/"
if dir != nil { if dir != nil {
if p := dir.GetPath(); p != "" { if p := dir.GetPath(); p != "" {
basePath = p basePath = p
} }
} }
node, err := d.resolvePath(ctx, basePath) return d.list(ctx, basePath)
if err != nil {
return nil, err
}
if node.kind == "root" {
groups, err := d.getGroups(ctx)
if err != nil {
return nil, err
}
res := make([]model.Obj, 0, len(groups))
for _, g := range groups {
path := joinPath(basePath, g.Name)
obj := &Obj{
id: path,
name: g.Name,
ctime: parseTime(0),
mtime: parseTime(0),
isDir: true,
path: path,
}
res = append(res, obj)
}
return res, nil
}
if node.kind != "group" && node.kind != "folder" {
return nil, nil
}
parentID := int64(0)
if node.file != nil && node.kind == "folder" {
parentID = node.file.ID
}
files, err := d.getFiles(ctx, node.group.GroupID, parentID)
if err != nil {
return nil, err
}
res := make([]model.Obj, 0, len(files))
for _, f := range files {
res = append(res, fileToObj(basePath, f))
}
return res, nil
} }
func (d *Wps) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) { func (d *Wps) Link(ctx context.Context, file model.Obj, _ model.LinkArgs) (*model.Link, error) {
path := file.GetPath() if file == nil {
node, err := d.resolvePath(ctx, path)
if err != nil {
return nil, err
}
if node.kind != "file" || node.file == nil {
return nil, errs.NotSupport return nil, errs.NotSupport
} }
if node.file.FilePerms.Download == 0 { return d.link(ctx, file.GetPath())
return nil, fmt.Errorf("no download permission")
}
url := fmt.Sprintf("%s/3rd/drive/api/v5/groups/%d/files/%d/download?support_checksums=sha1", endpoint, node.group.GroupID, node.file.ID)
var resp downloadResp
_, err = d.request(ctx).SetResult(&resp).Get(url)
if err != nil {
return nil, err
}
if resp.URL == "" {
return nil, fmt.Errorf("empty download url")
}
link := &model.Link{
URL: resp.URL,
Header: http.Header{},
}
return link, nil
} }
func (d *Wps) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error { func (d *Wps) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
basePath := "/" return d.makeDir(ctx, parentDir, dirName)
if parentDir != nil {
if p := parentDir.GetPath(); p != "" {
basePath = p
}
}
node, err := d.resolvePath(ctx, basePath)
if err != nil {
return err
}
if node.kind != "group" && node.kind != "folder" {
return errs.NotSupport
}
parentID := int64(0)
if node.file != nil && node.kind == "folder" {
parentID = node.file.ID
}
return d.createFolder(ctx, node.group.GroupID, parentID, dirName)
} }
func (d *Wps) Move(ctx context.Context, srcObj, dstDir model.Obj) error { func (d *Wps) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
if srcObj == nil || dstDir == nil { return d.move(ctx, srcObj, dstDir)
return errs.NotSupport
}
srcNode, err := d.resolvePath(ctx, srcObj.GetPath())
if err != nil {
return err
}
if (srcNode.kind != "file" && srcNode.kind != "folder") || srcNode.file == nil {
return errs.NotSupport
}
dstNode, err := d.resolvePath(ctx, dstDir.GetPath())
if err != nil {
return err
}
if dstNode.kind != "group" && dstNode.kind != "folder" {
return errs.NotSupport
}
targetParentID := int64(0)
if dstNode.file != nil && dstNode.kind == "folder" {
targetParentID = dstNode.file.ID
}
return d.moveFile(ctx, srcNode.group.GroupID, srcNode.file.ID, dstNode.group.GroupID, targetParentID)
} }
func (d *Wps) Rename(ctx context.Context, srcObj model.Obj, newName string) error { func (d *Wps) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
if srcObj == nil { return d.rename(ctx, srcObj, newName)
return errs.NotSupport
}
node, err := d.resolvePath(ctx, srcObj.GetPath())
if err != nil {
return err
}
if (node.kind != "file" && node.kind != "folder") || node.file == nil {
return errs.NotSupport
}
return d.renameFile(ctx, node.group.GroupID, node.file.ID, newName)
} }
func (d *Wps) Copy(ctx context.Context, srcObj, dstDir model.Obj) error { func (d *Wps) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
if srcObj == nil || dstDir == nil { return d.copy(ctx, srcObj, dstDir)
return errs.NotSupport
}
srcNode, err := d.resolvePath(ctx, srcObj.GetPath())
if err != nil {
return err
}
if (srcNode.kind != "file" && srcNode.kind != "folder") || srcNode.file == nil {
return errs.NotSupport
}
dstNode, err := d.resolvePath(ctx, dstDir.GetPath())
if err != nil {
return err
}
if dstNode.kind != "group" && dstNode.kind != "folder" {
return errs.NotSupport
}
targetParentID := int64(0)
if dstNode.file != nil && dstNode.kind == "folder" {
targetParentID = dstNode.file.ID
}
return d.copyFile(ctx, srcNode.group.GroupID, srcNode.file.ID, dstNode.group.GroupID, targetParentID)
} }
func (d *Wps) Remove(ctx context.Context, obj model.Obj) error { func (d *Wps) Remove(ctx context.Context, obj model.Obj) error {
if obj == nil { return d.remove(ctx, obj)
return errs.NotSupport
}
node, err := d.resolvePath(ctx, obj.GetPath())
if err != nil {
return err
}
if (node.kind != "file" && node.kind != "folder") || node.file == nil {
return errs.NotSupport
}
return d.deleteFile(ctx, node.group.GroupID, node.file.ID)
} }
func (d *Wps) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error { func (d *Wps) Put(ctx context.Context, _ model.Obj, _ model.FileStreamer, _ driver.UpdateProgress) error {
return errs.NotSupport return errs.NotSupport
} }

33
drivers/wps/types.go
View File

@ -49,39 +49,6 @@ type downloadResp struct {
Result string `json:"result"` Result string `json:"result"`
} }
type opResp struct {
Result string `json:"result"`
Msg string `json:"msg"`
}
type renameReq struct {
Fname string `json:"fname"`
}
type mkdirReq struct {
GroupID int64 `json:"groupid"`
Name string `json:"name"`
ParentID int64 `json:"parentid"`
}
type moveReq struct {
FileIDs []int64 `json:"fileids"`
TargetGroupID int64 `json:"target_groupid"`
TargetParentID int64 `json:"target_parentid"`
}
type deleteReq struct {
FileIDs []int64 `json:"fileids"`
}
type copyReq struct {
FileIDs []int64 `json:"fileids"`
GroupID int64 `json:"groupid"`
TargetGroupID int64 `json:"target_groupid"`
TargetParentID int64 `json:"target_parentid"`
DuplicatedNameModel int `json:"duplicated_name_model"`
}
type Obj struct { type Obj struct {
id string id string
name string name string

322
drivers/wps/util.go
View File

@ -3,11 +3,14 @@ package wps
import ( import (
"context" "context"
"fmt" "fmt"
"net/http"
"strconv" "strconv"
"strings" "strings"
"time" "time"
"github.com/alist-org/alist/v3/drivers/base" "github.com/alist-org/alist/v3/drivers/base"
"github.com/alist-org/alist/v3/internal/errs"
"github.com/alist-org/alist/v3/internal/model"
"github.com/go-resty/resty/v2" "github.com/go-resty/resty/v2"
) )
@ -19,6 +22,11 @@ type resolvedNode struct {
file *FileInfo file *FileInfo
} }
type apiResult struct {
Result string `json:"result"`
Msg string `json:"msg"`
}
func (d *Wps) request(ctx context.Context) *resty.Request { func (d *Wps) request(ctx context.Context) *resty.Request {
return base.RestyClient.R(). return base.RestyClient.R().
SetHeader("Cookie", d.Cookie). SetHeader("Cookie", d.Cookie).
@ -26,15 +34,24 @@ func (d *Wps) request(ctx context.Context) *resty.Request {
SetContext(ctx) SetContext(ctx)
} }
func (d *Wps) jsonRequest(ctx context.Context) *resty.Request {
return d.request(ctx).
SetHeader("Content-Type", "application/json").
SetHeader("Origin", "https://365.kdocs.cn/")
}
func (d *Wps) ensureCompanyID(ctx context.Context) error { func (d *Wps) ensureCompanyID(ctx context.Context) error {
if d.companyID != "" { if d.companyID != "" {
return nil return nil
} }
var resp workspaceResp var resp workspaceResp
_, err := d.request(ctx).SetResult(&resp).Get(endpoint + "/3rd/plussvr/compose/v1/users/self/workspaces?fields=name&comp_status=active") r, err := d.request(ctx).SetResult(&resp).SetError(&resp).Get(endpoint + "/3rd/plussvr/compose/v1/users/self/workspaces?fields=name&comp_status=active")
if err != nil { if err != nil {
return err return err
} }
if r != nil && r.IsError() {
return fmt.Errorf("http error: %d", r.StatusCode())
}
if len(resp.Companies) == 0 { if len(resp.Companies) == 0 {
return fmt.Errorf("no company id") return fmt.Errorf("no company id")
} }
@ -48,83 +65,33 @@ func (d *Wps) getGroups(ctx context.Context) ([]Group, error) {
} }
var resp groupsResp var resp groupsResp
url := fmt.Sprintf("%s/3rd/plus/groups/v1/companies/%s/users/self/groups/private", endpoint, d.companyID) url := fmt.Sprintf("%s/3rd/plus/groups/v1/companies/%s/users/self/groups/private", endpoint, d.companyID)
_, err := d.request(ctx).SetResult(&resp).Get(url) r, err := d.request(ctx).SetResult(&resp).SetError(&resp).Get(url)
if err != nil { if err != nil {
return nil, err return nil, err
} }
if r != nil && r.IsError() {
return nil, fmt.Errorf("http error: %d", r.StatusCode())
}
return resp.Groups, nil return resp.Groups, nil
} }
func (d *Wps) getFiles(ctx context.Context, groupID, parentID int64) ([]FileInfo, error) { func (d *Wps) getFiles(ctx context.Context, groupID, parentID int64) ([]FileInfo, error) {
var resp filesResp var resp filesResp
url := fmt.Sprintf("%s/3rd/drive/api/v5/groups/%d/files", endpoint, groupID) url := fmt.Sprintf("%s/3rd/drive/api/v5/groups/%d/files", endpoint, groupID)
_, err := d.request(ctx).SetQueryParam("parentid", strconv.FormatInt(parentID, 10)).SetResult(&resp).Get(url) r, err := d.request(ctx).
SetQueryParam("parentid", strconv.FormatInt(parentID, 10)).
SetResult(&resp).
SetError(&resp).
Get(url)
if err != nil { if err != nil {
return nil, err return nil, err
} }
if r != nil && r.IsError() {
return nil, fmt.Errorf("http error: %d", r.StatusCode())
}
return resp.Files, nil return resp.Files, nil
} }
func checkResult(result, msg string) error {
if result == "" || result == "ok" {
return nil
}
if msg != "" {
return fmt.Errorf("%s: %s", result, msg)
}
return fmt.Errorf("%s", result)
}
func (d *Wps) renameFile(ctx context.Context, groupID, fileID int64, name string) error {
var resp opResp
url := fmt.Sprintf("%s/3rd/drive/api/v3/groups/%d/files/%d", endpoint, groupID, fileID)
_, err := d.request(ctx).SetBody(renameReq{Fname: name}).SetResult(&resp).Put(url)
if err != nil {
return err
}
return checkResult(resp.Result, resp.Msg)
}
func (d *Wps) createFolder(ctx context.Context, groupID, parentID int64, name string) error {
var resp opResp
url := endpoint + "/3rd/drive/api/v5/files/folder"
_, err := d.request(ctx).SetBody(mkdirReq{GroupID: groupID, Name: name, ParentID: parentID}).SetResult(&resp).Post(url)
if err != nil {
return err
}
return checkResult(resp.Result, resp.Msg)
}
func (d *Wps) moveFile(ctx context.Context, groupID, fileID, targetGroupID, targetParentID int64) error {
var resp opResp
url := fmt.Sprintf("%s/3rd/drive/api/v3/groups/%d/files/batch/move", endpoint, groupID)
_, err := d.request(ctx).SetBody(moveReq{FileIDs: []int64{fileID}, TargetGroupID: targetGroupID, TargetParentID: targetParentID}).SetResult(&resp).Post(url)
if err != nil {
return err
}
return checkResult(resp.Result, resp.Msg)
}
func (d *Wps) deleteFile(ctx context.Context, groupID, fileID int64) error {
var resp opResp
url := fmt.Sprintf("%s/3rd/drive/api/v3/groups/%d/files/batch/delete", endpoint, groupID)
_, err := d.request(ctx).SetBody(deleteReq{FileIDs: []int64{fileID}}).SetResult(&resp).Post(url)
if err != nil {
return err
}
return checkResult(resp.Result, resp.Msg)
}
func (d *Wps) copyFile(ctx context.Context, groupID, fileID, targetGroupID, targetParentID int64) error {
var resp opResp
url := fmt.Sprintf("%s/3rd/drive/api/v3/groups/%d/files/batch/copy", endpoint, groupID)
_, err := d.request(ctx).SetBody(copyReq{FileIDs: []int64{fileID}, GroupID: groupID, TargetGroupID: targetGroupID, TargetParentID: targetParentID, DuplicatedNameModel: 1}).SetResult(&resp).Post(url)
if err != nil {
return err
}
return checkResult(resp.Result, resp.Msg)
}
func parseTime(v int64) time.Time { func parseTime(v int64) time.Time {
if v <= 0 { if v <= 0 {
return time.Time{} return time.Time{}
@ -183,6 +150,9 @@ func (d *Wps) resolvePath(ctx context.Context, path string) (*resolvedNode, erro
if found == nil { if found == nil {
return nil, fmt.Errorf("path not found") return nil, fmt.Errorf("path not found")
} }
if i < len(segs)-1 && found.Type != "folder" {
return nil, fmt.Errorf("path not found")
}
last = *found last = *found
parentID = found.ID parentID = found.ID
} }
@ -207,3 +177,229 @@ func fileToObj(basePath string, f FileInfo) *Obj {
canDownload: f.FilePerms.Download != 0, canDownload: f.FilePerms.Download != 0,
} }
} }
func (d *Wps) doJSON(ctx context.Context, method, url string, body interface{}) error {
var result apiResult
req := d.jsonRequest(ctx).SetBody(body).SetResult(&result).SetError(&result)
var (
resp *resty.Response
err error
)
switch method {
case http.MethodPost:
resp, err = req.Post(url)
case http.MethodPut:
resp, err = req.Put(url)
default:
return errs.NotSupport
}
if err != nil {
return err
}
if result.Result != "" && result.Result != "ok" {
if result.Msg == "" {
result.Msg = "unknown error"
}
return fmt.Errorf("%s: %s", result.Result, result.Msg)
}
if resp != nil && resp.IsError() {
if result.Msg != "" {
return fmt.Errorf("%s", result.Msg)
}
return fmt.Errorf("http error: %d", resp.StatusCode())
}
return nil
}
func (d *Wps) list(ctx context.Context, basePath string) ([]model.Obj, error) {
if strings.TrimSpace(basePath) == "" {
basePath = "/"
}
node, err := d.resolvePath(ctx, basePath)
if err != nil {
return nil, err
}
if node.kind == "root" {
groups, err := d.getGroups(ctx)
if err != nil {
return nil, err
}
res := make([]model.Obj, 0, len(groups))
for _, g := range groups {
path := joinPath(basePath, g.Name)
obj := &Obj{
id: path,
name: g.Name,
ctime: parseTime(0),
mtime: parseTime(0),
isDir: true,
path: path,
}
res = append(res, obj)
}
return res, nil
}
if node.kind != "group" && node.kind != "folder" {
return nil, nil
}
parentID := int64(0)
if node.file != nil && node.kind == "folder" {
parentID = node.file.ID
}
files, err := d.getFiles(ctx, node.group.GroupID, parentID)
if err != nil {
return nil, err
}
res := make([]model.Obj, 0, len(files))
for _, f := range files {
res = append(res, fileToObj(basePath, f))
}
return res, nil
}
func (d *Wps) link(ctx context.Context, path string) (*model.Link, error) {
node, err := d.resolvePath(ctx, path)
if err != nil {
return nil, err
}
if node.kind != "file" || node.file == nil {
return nil, errs.NotSupport
}
if node.file.FilePerms.Download == 0 {
return nil, fmt.Errorf("no download permission")
}
url := fmt.Sprintf("%s/3rd/drive/api/v5/groups/%d/files/%d/download?support_checksums=sha1", endpoint, node.group.GroupID, node.file.ID)
var resp downloadResp
r, err := d.request(ctx).SetResult(&resp).SetError(&resp).Get(url)
if err != nil {
return nil, err
}
if r != nil && r.IsError() {
return nil, fmt.Errorf("http error: %d", r.StatusCode())
}
if resp.URL == "" {
return nil, fmt.Errorf("empty download url")
}
return &model.Link{URL: resp.URL, Header: http.Header{}}, nil
}
func (d *Wps) makeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
if parentDir == nil {
return errs.NotSupport
}
node, err := d.resolvePath(ctx, parentDir.GetPath())
if err != nil {
return err
}
if node.kind != "group" && node.kind != "folder" {
return errs.NotSupport
}
parentID := int64(0)
if node.file != nil && node.kind == "folder" {
parentID = node.file.ID
}
body := map[string]interface{}{
"groupid": node.group.GroupID,
"name": dirName,
"parentid": parentID,
}
return d.doJSON(ctx, http.MethodPost, endpoint+"/3rd/drive/api/v5/files/folder", body)
}
func (d *Wps) move(ctx context.Context, srcObj, dstDir model.Obj) error {
if srcObj == nil || dstDir == nil {
return errs.NotSupport
}
nodeSrc, err := d.resolvePath(ctx, srcObj.GetPath())
if err != nil {
return err
}
nodeDst, err := d.resolvePath(ctx, dstDir.GetPath())
if err != nil {
return err
}
if nodeSrc.kind != "file" && nodeSrc.kind != "folder" {
return errs.NotSupport
}
if nodeDst.kind != "group" && nodeDst.kind != "folder" {
return errs.NotSupport
}
targetParentID := int64(0)
if nodeDst.file != nil && nodeDst.kind == "folder" {
targetParentID = nodeDst.file.ID
}
body := map[string]interface{}{
"fileids": []int64{nodeSrc.file.ID},
"target_groupid": nodeDst.group.GroupID,
"target_parentid": targetParentID,
}
url := fmt.Sprintf("%s/3rd/drive/api/v3/groups/%d/files/batch/move", endpoint, nodeSrc.group.GroupID)
return d.doJSON(ctx, http.MethodPost, url, body)
}
func (d *Wps) rename(ctx context.Context, srcObj model.Obj, newName string) error {
if srcObj == nil {
return errs.NotSupport
}
node, err := d.resolvePath(ctx, srcObj.GetPath())
if err != nil {
return err
}
if node.kind != "file" && node.kind != "folder" {
return errs.NotSupport
}
url := fmt.Sprintf("%s/3rd/drive/api/v3/groups/%d/files/%d", endpoint, node.group.GroupID, node.file.ID)
body := map[string]string{"fname": newName}
return d.doJSON(ctx, http.MethodPut, url, body)
}
func (d *Wps) copy(ctx context.Context, srcObj, dstDir model.Obj) error {
if srcObj == nil || dstDir == nil {
return errs.NotSupport
}
nodeSrc, err := d.resolvePath(ctx, srcObj.GetPath())
if err != nil {
return err
}
nodeDst, err := d.resolvePath(ctx, dstDir.GetPath())
if err != nil {
return err
}
if nodeSrc.kind != "file" && nodeSrc.kind != "folder" {
return errs.NotSupport
}
if nodeDst.kind != "group" && nodeDst.kind != "folder" {
return errs.NotSupport
}
targetParentID := int64(0)
if nodeDst.file != nil && nodeDst.kind == "folder" {
targetParentID = nodeDst.file.ID
}
body := map[string]interface{}{
"fileids": []int64{nodeSrc.file.ID},
"groupid": nodeSrc.group.GroupID,
"target_groupid": nodeDst.group.GroupID,
"target_parentid": targetParentID,
"duplicated_name_model": 1,
}
url := fmt.Sprintf("%s/3rd/drive/api/v3/groups/%d/files/batch/copy", endpoint, nodeSrc.group.GroupID)
return d.doJSON(ctx, http.MethodPost, url, body)
}
func (d *Wps) remove(ctx context.Context, obj model.Obj) error {
if obj == nil {
return errs.NotSupport
}
node, err := d.resolvePath(ctx, obj.GetPath())
if err != nil {
return err
}
if node.kind != "file" && node.kind != "folder" {
return errs.NotSupport
}
body := map[string]interface{}{
"fileids": []int64{node.file.ID},
}
url := fmt.Sprintf("%s/3rd/drive/api/v3/groups/%d/files/batch/delete", endpoint, node.group.GroupID)
return d.doJSON(ctx, http.MethodPost, url, body)
}