llvy.ltd/casdoor
1
0
Fork 0
mirror of https://github.com/casdoor/casdoor.git synced 2025-07-02 11:20:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add local file system storage provider (#224)

Browse Source 
Signed-off-by: sh1luo <690898835@qq.com>
This commit is contained in:
sh1luo
2021-08-08 14:18:44 +08:00
committed by Yang Luo
parent 44150a6781
commit c55fa4f452
9 changed files with 178 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
object/storage.go
View File

@ -17,14 +17,16 @@ package object
import (
"bytes"
"fmt"
"strings"
"github.com/casbin/casdoor/storage"
"github.com/casbin/casdoor/util"
)
func UploadAvatar(provider *Provider, username string, avatar []byte) error {
func UploadAvatar(provider *Provider, username string, avatar []byte) (string, error) {
storageProvider := storage.GetStorageProvider(provider.Type, provider.ClientId, provider.ClientSecret, provider.RegionId, provider.Bucket, provider.Endpoint)
if storageProvider == nil {
return fmt.Errorf("the provider type: %s is not supported", provider.Type)
return "", fmt.Errorf("the provider type: %s is not supported", provider.Type)
}
if provider.Domain == "" {
@ -34,5 +36,22 @@ func UploadAvatar(provider *Provider, username string, avatar []byte) error {
path := fmt.Sprintf("%s/%s.png", util.UrlJoin(util.GetUrlPath(provider.Domain), "/avatar"), username)
_, err := storageProvider.Put(path, bytes.NewReader(avatar))
return err
if err != nil {
return "", err
}
host := ""
if provider.Type != "Local File System" {
// provider.Domain = "https://cdn.casbin.com/casdoor/"
host = util.GetUrlHost(provider.Domain)
if !strings.HasPrefix(host, "http://") && !strings.HasPrefix(host, "https://") {
host = fmt.Sprintf("https://%s", host)
}
} else {
// provider.Domain = "http://localhost:8000" or "https://door.casbin.com"
host = util.UrlJoin(provider.Domain, "/files")
}
fileUrl := fmt.Sprintf("%s?time=%s", util.UrlJoin(host, path), util.GetCurrentUnixTime())
return fileUrl, nil
}