feat: add s3 driver

drivers/s3/driver.go

@@ -0,0 +1,160 @@
+package s3
+
+import (
+	"bytes"
+	"context"
+	"fmt"
+	"io"
+	"net/url"
+	stdpath "path"
+	"time"
+
+	"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/session"
+	"github.com/aws/aws-sdk-go/service/s3"
+	"github.com/aws/aws-sdk-go/service/s3/s3manager"
+	log "github.com/sirupsen/logrus"
+)
+
+type S3 struct {
+	model.Storage
+	Addition
+	Session    *session.Session
+	client     *s3.S3
+	linkClient *s3.S3
+}
+
+func (d *S3) Config() driver.Config {
+	return config
+}
+
+func (d *S3) GetAddition() driver.Additional {
+	return d.Addition
+}
+
+func (d *S3) Init(ctx context.Context, storage model.Storage) error {
+	d.Storage = storage
+	err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
+	if err != nil {
+		return err
+	}
+	if d.Region == "" {
+		d.Region = "alist"
+	}
+	err = d.initSession()
+	if err != nil {
+		return err
+	}
+	d.client = d.getClient(false)
+	d.linkClient = d.getClient(true)
+	return nil
+}
+
+func (d *S3) Drop(ctx context.Context) error {
+	return nil
+}
+
+func (d *S3) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
+	if d.ListObjectVersion == "v2" {
+		return d.listV2(dir.GetPath())
+	}
+	return d.listV1(dir.GetPath())
+}
+
+//func (d *S3) Get(ctx context.Context, path string) (model.Obj, error) {
+//	// this is optional
+//	return nil, errs.NotImplement
+//}
+
+func (d *S3) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
+	path := getKey(file.GetPath(), false)
+	disposition := fmt.Sprintf(`attachment;filename="%s"`, url.QueryEscape(stdpath.Base(path)))
+	input := &s3.GetObjectInput{
+		Bucket: &d.Bucket,
+		Key:    &path,
+		//ResponseContentDisposition: &disposition,
+	}
+	if d.CustomHost == "" {
+		input.ResponseContentDisposition = &disposition
+	}
+	req, _ := d.linkClient.GetObjectRequest(input)
+	var link string
+	var err error
+	if d.CustomHost != "" {
+		err = req.Build()
+		link = req.HTTPRequest.URL.String()
+	} else {
+		link, err = req.Presign(time.Hour * time.Duration(d.SignURLExpire))
+	}
+	if err != nil {
+		return nil, err
+	}
+	return &model.Link{
+		URL: link,
+	}, nil
+}
+
+func (d *S3) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
+	return d.Put(ctx, &model.Object{
+		Path: stdpath.Join(parentDir.GetPath(), dirName),
+	}, &model.FileStream{
+		Obj: &model.Object{
+			Name:     getPlaceholderName(d.Placeholder),
+			Modified: time.Now(),
+		},
+		ReadCloser: io.NopCloser(bytes.NewReader([]byte{})),
+		Mimetype:   "application/octet-stream",
+	}, func(int) {})
+}
+
+func (d *S3) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
+	err := d.Copy(ctx, srcObj, dstDir)
+	if err != nil {
+		return err
+	}
+	return d.Remove(ctx, srcObj)
+}
+
+func (d *S3) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
+	err := d.copy(srcObj.GetPath(), stdpath.Join(stdpath.Dir(srcObj.GetPath()), newName), srcObj.IsDir())
+	if err != nil {
+		return err
+	}
+	return d.Remove(ctx, srcObj)
+}
+
+func (d *S3) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
+	return d.copy(srcObj.GetPath(), stdpath.Join(dstDir.GetPath(), stdpath.Base(srcObj.GetPath())), srcObj.IsDir())
+}
+
+func (d *S3) Remove(ctx context.Context, obj model.Obj) error {
+	key := getKey(obj.GetPath(), obj.IsDir())
+	input := &s3.DeleteObjectInput{
+		Bucket: &d.Bucket,
+		Key:    &key,
+	}
+	_, err := d.client.DeleteObject(input)
+	return err
+}
+
+func (d *S3) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
+	uploader := s3manager.NewUploader(d.Session)
+	key := getKey(stdpath.Join(dstDir.GetPath(), stream.GetName()), false)
+	log.Debugln("key:", key)
+	input := &s3manager.UploadInput{
+		Bucket: &d.Bucket,
+		Key:    &key,
+		Body:   stream,
+	}
+	_, err := uploader.Upload(input)
+	return err
+}
+
+func (d *S3) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
+	return nil, errs.NotSupport
+}
+
+var _ driver.Driver = (*S3)(nil)