🎇 s3 support

drivers/s3/s3.go

@@ -0,0 +1,118 @@
+package s3
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/drivers/base"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/utils"
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/credentials"
+	"github.com/aws/aws-sdk-go/aws/request"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/s3"
+	log "github.com/sirupsen/logrus"
+	"net/http"
+	"net/url"
+	"path"
+	"strings"
+)
+
+var sessionsMap map[string]*session.Session
+
+func (driver S3) NewSession(account *model.Account) (*session.Session, error) {
+	cfg := &aws.Config{
+		Credentials: credentials.NewStaticCredentials(account.AccessKey, account.AccessSecret, ""),
+		Region:      &account.Region,
+		Endpoint:    &account.Endpoint,
+	}
+	return session.NewSession(cfg)
+}
+
+func (driver S3) GetClient(account *model.Account) (*s3.S3, error) {
+	s, ok := sessionsMap[account.Name]
+	if !ok {
+		return nil, fmt.Errorf("can't find [%s] session", account.Name)
+	}
+	client := s3.New(s)
+	if account.CustomHost != "" {
+		cURL, err := url.Parse(account.CustomHost)
+		if err != nil {
+			return nil, err
+		}
+		client.Handlers.Build.PushBack(func(r *request.Request) {
+			if r.HTTPRequest.Method != http.MethodGet {
+				return
+			}
+			r.HTTPRequest.URL.Scheme = cURL.Scheme
+			r.HTTPRequest.URL.Host = cURL.Host
+		})
+	}
+	return client, nil
+}
+
+func (driver S3) List(prefix string, account *model.Account) ([]model.File, error) {
+	prefix = driver.GetKey(prefix, account, true)
+	log.Debugf("list: %s", prefix)
+	client, err := driver.GetClient(account)
+	if err != nil {
+		return nil, err
+	}
+	files := make([]model.File, 0)
+	marker := ""
+	for {
+		input := &s3.ListObjectsInput{
+			Bucket:    &account.Bucket,
+			Marker:    &marker,
+			Prefix:    &prefix,
+			Delimiter: aws.String("/"),
+		}
+		listObjectsResult, err := client.ListObjects(input)
+
+		if err != nil {
+			return nil, err
+		}
+		for _, object := range listObjectsResult.CommonPrefixes {
+			file := model.File{
+				//Id:        *object.Key,
+				Name:      utils.Base(strings.Trim(*object.Prefix, "/")),
+				Driver:    driver.Config().Name,
+				UpdatedAt: account.UpdatedAt,
+				TimeStr:   "-",
+				Type:      conf.FOLDER,
+			}
+			files = append(files, file)
+		}
+		for _, object := range listObjectsResult.Contents {
+			file := model.File{
+				//Id:        *object.Key,
+				Name:      utils.Base(*object.Key),
+				Size:      *object.Size,
+				Driver:    driver.Config().Name,
+				UpdatedAt: object.LastModified,
+				Type:      utils.GetFileType(path.Ext(*object.Key)),
+			}
+			files = append(files, file)
+		}
+		if *listObjectsResult.IsTruncated {
+			marker = *listObjectsResult.NextMarker
+		} else {
+			break
+		}
+	}
+	return files, nil
+}
+
+func (driver S3) GetKey(path string, account *model.Account, dir bool) string {
+	path = utils.Join(account.RootFolder, path)
+	path = strings.TrimPrefix(path, "/")
+	if path != "" && dir {
+		path += "/"
+	}
+	return path
+}
+
+func init() {
+	sessionsMap = make(map[string]*session.Session, 0)
+	base.RegisterDriver(&S3{})
+}