feat: integrate Storage config into providers (#198)

Signed-off-by: Kininaru <shiftregister233@outlook.com>

object/storage.go

@@ -0,0 +1,136 @@
+// Copyright 2021 The casbin Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package object
+
+import (
+	"bytes"
+	"fmt"
+
+	awss3 "github.com/aws/aws-sdk-go/service/s3"
+	"github.com/casbin/casdoor/util"
+	"github.com/qor/oss"
+	"github.com/qor/oss/aliyun"
+	//"github.com/qor/oss/qiniu"
+	"github.com/qor/oss/s3"
+)
+
+func getAliyunClient(provider *Provider) oss.StorageInterface {
+	if util.IsStrsEmpty(
+		provider.Endpoint,
+		provider.ClientId,
+		provider.ClientSecret,
+		provider.Bucket) {
+		return nil
+	}
+
+	ret := aliyun.New(&aliyun.Config{
+		AccessID:  provider.ClientId,
+		AccessKey: provider.ClientSecret,
+		Bucket:    provider.Bucket,
+		Endpoint:  provider.Endpoint,
+	})
+
+	if len(provider.Domain) == 0 {
+		provider.Domain = ret.GetEndpoint()
+		UpdateProvider(provider.GetId(), provider)
+	}
+	return ret
+}
+
+func getQiniuClient(provider *Provider) oss.StorageInterface {
+	fmt.Println("Casdoor does not support Qiniu now.")
+	return nil
+//	endpoint := section.Key("endpoint").String()
+//	accessId := section.Key("accessId").String()
+//	accessKey := section.Key("accessKey").String()
+//  domain = section.Key("domain").String()
+//	bucket := section.Key("bucket").String()
+//	region := section.Key("region").String()
+//	if accessId == "" || accessKey == "" || bucket == "" || endpoint == "" || region == "" {
+//		return "Config oss.conf wrong"
+//	}
+//	storage = qiniu.New(&qiniu.Config{
+//		AccessID: accessId,
+//		AccessKey: accessKey,
+//		Bucket: bucket,
+//		Region: region,
+//		Endpoint: endpoint,
+//	})
+//	return ""
+}
+
+func getAwss3Client(provider *Provider) oss.StorageInterface {
+	if util.IsStrsEmpty(
+		provider.Endpoint,
+		provider.ClientId,
+		provider.ClientSecret,
+		provider.Bucket,
+		provider.RegionId) {
+		return nil
+	}
+
+	ret := s3.New(&s3.Config{
+		AccessID:  provider.ClientId,
+		AccessKey: provider.ClientSecret,
+		Region:    provider.RegionId,
+		Bucket:    provider.Bucket,
+		Endpoint:  provider.Endpoint,
+		ACL:       awss3.BucketCannedACLPublicRead,
+	})
+
+	if len(provider.Domain) == 0 {
+		provider.Domain = ret.GetEndpoint()
+		UpdateProvider(provider.GetId(), provider)
+	}
+	return ret
+}
+
+func getStorageClient(provider *Provider) oss.StorageInterface {
+	if provider == nil || provider.Category != "Storage" {
+		return nil
+	}
+
+	switch provider.Type {
+	case "Aliyun":
+		return getAliyunClient(provider)
+	case "Qiniu":
+		return getQiniuClient(provider)
+	case "AWSS3":
+		return getAwss3Client(provider)
+	}
+
+	return nil
+}
+
+func UploadAvatar(provider *Provider, username string, avatar []byte) string {
+	if provider == nil {
+		return "invalid Storage provider"
+	}
+	storage := getStorageClient(provider)
+	if storage == nil {
+		return "oss provider not exists"
+	}
+
+	path := fmt.Sprintf("/casdoor/avatar/%s.png", username)
+	_, err := storage.Put(path, bytes.NewReader(avatar))
+	if err != nil {
+		panic(err)
+	}
+	return ""
+}
+
+func GetAvatarPath(provider *Provider) string {
+	return fmt.Sprintf("https://%s/casdoor/avatar/", provider.Domain)
+}