feat: add pikpak driver

drivers/all.go

@@ -3,6 +3,7 @@ package drivers
 import (
 	_ "github.com/alist-org/alist/v3/drivers/local"
 	_ "github.com/alist-org/alist/v3/drivers/onedrive"
+	_ "github.com/alist-org/alist/v3/drivers/pikpak"
 	_ "github.com/alist-org/alist/v3/drivers/virtual"
 )
 

drivers/pikpak/driver.go

@@ -0,0 +1,220 @@
+package local
+
+import (
+	"context"
+	"crypto/sha1"
+	"encoding/hex"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"strings"
+
+	"github.com/alist-org/alist/v3/drivers/base"
+	"github.com/alist-org/alist/v3/internal/conf"
+	"github.com/alist-org/alist/v3/internal/driver"
+	"github.com/alist-org/alist/v3/internal/errs"
+	"github.com/alist-org/alist/v3/internal/model"
+	"github.com/alist-org/alist/v3/pkg/utils"
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/credentials"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/s3/s3manager"
+	"github.com/go-resty/resty/v2"
+	jsoniter "github.com/json-iterator/go"
+	"github.com/pkg/errors"
+	log "github.com/sirupsen/logrus"
+)
+
+type PikPak struct {
+	model.Storage
+	Addition
+	RefreshToken string
+	AccessToken  string
+}
+
+func (d *PikPak) Config() driver.Config {
+	return config
+}
+
+func (d *PikPak) GetAddition() driver.Additional {
+	return d.Addition
+}
+
+func (d *PikPak) Init(ctx context.Context, storage model.Storage) error {
+	d.Storage = storage
+	err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
+	if err != nil {
+		return errors.Wrap(err, "error while unmarshal addition")
+	}
+	return d.login()
+}
+
+func (d *PikPak) Drop(ctx context.Context) error {
+	return nil
+}
+
+func (d *PikPak) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
+	files, err := d.getFiles(dir.GetID())
+	if err != nil {
+		return nil, err
+	}
+	objs := make([]model.Obj, len(files))
+	for i := 0; i < len(files); i++ {
+		objs[i] = fileToObj(files[i])
+	}
+	return objs, nil
+}
+
+func (d *PikPak) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
+	var resp File
+	_, err := d.request(fmt.Sprintf("https://api-drive.mypikpak.com/drive/v1/files/%s?_magic=2021&thumbnail_size=SIZE_LARGE", file.GetID()),
+		http.MethodGet, nil, &resp)
+	if err != nil {
+		return nil, err
+	}
+	link := model.Link{
+		URL: resp.WebContentLink,
+	}
+	if len(resp.Medias) > 0 && resp.Medias[0].Link.Url != "" {
+		log.Debugln("use media link")
+		link.URL = resp.Medias[0].Link.Url
+	}
+	return &link, nil
+}
+
+func (d *PikPak) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
+	_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files", http.MethodPost, func(req *resty.Request) {
+		req.SetBody(base.Json{
+			"kind":      "drive#folder",
+			"parent_id": parentDir.GetID(),
+			"name":      dirName,
+		})
+	}, nil)
+	return err
+}
+
+func (d *PikPak) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
+	_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files:batchMove", http.MethodPost, func(req *resty.Request) {
+		req.SetBody(base.Json{
+			"ids": []string{srcObj.GetID()},
+			"to": base.Json{
+				"parent_id": dstDir.GetID(),
+			},
+		})
+	}, nil)
+	return err
+}
+
+func (d *PikPak) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
+	_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files/"+srcObj.GetID(), http.MethodPatch, func(req *resty.Request) {
+		req.SetBody(base.Json{
+			"name": newName,
+		})
+	}, nil)
+	return err
+}
+
+func (d *PikPak) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
+	_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files:batchCopy", http.MethodPost, func(req *resty.Request) {
+		req.SetBody(base.Json{
+			"ids": []string{srcObj.GetID()},
+			"to": base.Json{
+				"parent_id": dstDir.GetID(),
+			},
+		})
+	}, nil)
+	return err
+}
+
+func (d *PikPak) Remove(ctx context.Context, obj model.Obj) error {
+	_, err := d.request("https://api-drive.mypikpak.com/drive/v1/files:batchTrash", http.MethodPost, func(req *resty.Request) {
+		req.SetBody(base.Json{
+			"ids": []string{obj.GetID()},
+		})
+	}, nil)
+	return err
+}
+
+func (d *PikPak) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
+	var tempFile *os.File
+	var err error
+	if f, ok := stream.GetReadCloser().(*os.File); ok {
+		tempFile = f
+	} else {
+		tempFile, err = os.CreateTemp(conf.Conf.TempDir, "file-*")
+		if err != nil {
+			return err
+		}
+		defer func() {
+			_ = tempFile.Close()
+			_ = os.Remove(tempFile.Name())
+		}()
+		_, err = io.Copy(tempFile, stream)
+		if err != nil {
+			return err
+		}
+		_, err = tempFile.Seek(0, io.SeekStart)
+		if err != nil {
+			return err
+		}
+	}
+	// cal sha1
+	s := sha1.New()
+	_, err = io.Copy(s, tempFile)
+	if err != nil {
+		return err
+	}
+	_, err = tempFile.Seek(0, io.SeekStart)
+	if err != nil {
+		return err
+	}
+	sha1Str := hex.EncodeToString(s.Sum(nil))
+	data := base.Json{
+		"kind":        "drive#file",
+		"name":        stream.GetName(),
+		"size":        stream.GetSize(),
+		"hash":        strings.ToUpper(sha1Str),
+		"upload_type": "UPLOAD_TYPE_RESUMABLE",
+		"objProvider": base.Json{"provider": "UPLOAD_TYPE_UNKNOWN"},
+		"parent_id":   dstDir.GetID(),
+	}
+	res, err := d.request("https://api-drive.mypikpak.com/drive/v1/files", http.MethodPost, func(req *resty.Request) {
+		req.SetBody(data)
+	}, nil)
+	if err != nil {
+		return err
+	}
+	params := jsoniter.Get(res, "resumable").Get("params")
+	endpoint := params.Get("endpoint").ToString()
+	endpointS := strings.Split(endpoint, ".")
+	endpoint = strings.Join(endpointS[1:], ".")
+	accessKeyId := params.Get("access_key_id").ToString()
+	accessKeySecret := params.Get("access_key_secret").ToString()
+	securityToken := params.Get("security_token").ToString()
+	key := params.Get("key").ToString()
+	bucket := params.Get("bucket").ToString()
+	cfg := &aws.Config{
+		Credentials: credentials.NewStaticCredentials(accessKeyId, accessKeySecret, securityToken),
+		Region:      aws.String("pikpak"),
+		Endpoint:    &endpoint,
+	}
+	ss, err := session.NewSession(cfg)
+	if err != nil {
+		return err
+	}
+	uploader := s3manager.NewUploader(ss)
+	input := &s3manager.UploadInput{
+		Bucket: &bucket,
+		Key:    &key,
+		Body:   tempFile,
+	}
+	_, err = uploader.Upload(input)
+	return err
+}
+
+func (d *PikPak) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
+	return nil, errs.NotSupport
+}
+
+var _ driver.Driver = (*PikPak)(nil)

drivers/pikpak/meta.go

@@ -0,0 +1,26 @@
+package local
+
+import (
+	"github.com/alist-org/alist/v3/internal/driver"
+	"github.com/alist-org/alist/v3/internal/operations"
+)
+
+type Addition struct {
+	driver.RootFolderId
+	Username string `json:"username" required:"true"`
+	Password string `json:"password" required:"true"`
+}
+
+var config = driver.Config{
+	Name:        "PikPak",
+	LocalSort:   true,
+	DefaultRoot: "",
+}
+
+func New() driver.Driver {
+	return &PikPak{}
+}
+
+func init() {
+	operations.RegisterDriver(config, New)
+}

drivers/pikpak/types.go

@@ -0,0 +1,75 @@
+package local
+
+import (
+	"strconv"
+	"time"
+
+	"github.com/alist-org/alist/v3/internal/model"
+)
+
+type RespErr struct {
+	ErrorCode int    `json:"error_code"`
+	Error     string `json:"error"`
+}
+
+type Files struct {
+	Files         []File `json:"files"`
+	NextPageToken string `json:"next_page_token"`
+}
+
+type File struct {
+	Id             string    `json:"id"`
+	Kind           string    `json:"kind"`
+	Name           string    `json:"name"`
+	ModifiedTime   time.Time `json:"modified_time"`
+	Size           string    `json:"size"`
+	ThumbnailLink  string    `json:"thumbnail_link"`
+	WebContentLink string    `json:"web_content_link"`
+	Medias         []Media   `json:"medias"`
+}
+
+func fileToObj(f File) model.ObjectThumbnail {
+	size, _ := strconv.ParseInt(f.Size, 10, 64)
+	return model.ObjectThumbnail{
+		Object: model.Object{
+			ID:       f.Id,
+			Name:     f.Name,
+			Size:     size,
+			Modified: f.ModifiedTime,
+			IsFolder: f.Kind == "drive#folder",
+		},
+		Thumbnail: model.Thumbnail{
+			Thumbnail: f.ThumbnailLink,
+		},
+	}
+}
+
+type Media struct {
+	MediaId   string `json:"media_id"`
+	MediaName string `json:"media_name"`
+	Video     struct {
+		Height     int    `json:"height"`
+		Width      int    `json:"width"`
+		Duration   int    `json:"duration"`
+		BitRate    int    `json:"bit_rate"`
+		FrameRate  int    `json:"frame_rate"`
+		VideoCodec string `json:"video_codec"`
+		AudioCodec string `json:"audio_codec"`
+		VideoType  string `json:"video_type"`
+	} `json:"video"`
+	Link struct {
+		Url    string    `json:"url"`
+		Token  string    `json:"token"`
+		Expire time.Time `json:"expire"`
+	} `json:"link"`
+	NeedMoreQuota  bool          `json:"need_more_quota"`
+	VipTypes       []interface{} `json:"vip_types"`
+	RedirectLink   string        `json:"redirect_link"`
+	IconLink       string        `json:"icon_link"`
+	IsDefault      bool          `json:"is_default"`
+	Priority       int           `json:"priority"`
+	IsOrigin       bool          `json:"is_origin"`
+	ResolutionName string        `json:"resolution_name"`
+	IsVisible      bool          `json:"is_visible"`
+	Category       string        `json:"category"`
+}