131 lines
3.5 KiB
Go
131 lines
3.5 KiB
Go
package xfs
|
||
|
||
import (
|
||
"crypto/md5"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/aliyun/alibaba-cloud-sdk-go/services/sts"
|
||
"github.com/yaoapp/gou"
|
||
"github.com/yaoapp/kun/any"
|
||
"github.com/yaoapp/kun/exception"
|
||
"github.com/yaoapp/kun/maps"
|
||
"github.com/yaoapp/xiang/share"
|
||
"github.com/yaoapp/xun"
|
||
)
|
||
|
||
func init() {
|
||
// 注册处理器
|
||
gou.RegisterProcessHandler("xiang.fs.Upload", processUpload)
|
||
gou.RegisterProcessHandler("xiang.fs.GetToken", processGetToken)
|
||
gou.RegisterProcessHandler("xiang.fs.GetURL", processGetURL)
|
||
gou.RegisterProcessHandler("xiang.fs.ReadFile", processReadFile)
|
||
}
|
||
|
||
// processUpload 上传文件到本地服务器
|
||
func processUpload(process *gou.Process) interface{} {
|
||
process.ValidateArgNums(1)
|
||
tmpfile, ok := process.Args[0].(xun.UploadFile)
|
||
if !ok {
|
||
exception.New("上传文件参数错误", 400, process.Args[0]).Throw()
|
||
}
|
||
|
||
hash := md5.Sum([]byte(time.Now().Format("20060102-15:04:05")))
|
||
fingerprint := string(hex.EncodeToString(hash[:]))
|
||
fingerprint = strings.ToUpper(fingerprint)
|
||
|
||
dir := time.Now().Format("20060102")
|
||
ext := filepath.Ext(tmpfile.Name)
|
||
filename := filepath.Join(dir, fmt.Sprintf("%s%s", fingerprint, ext))
|
||
Stor.MustMkdirAll(dir, os.ModePerm)
|
||
|
||
content, err := New("/").ReadFile(tmpfile.TempFile)
|
||
if err != nil {
|
||
exception.New("不能读取上传文件 %s", 500, err.Error()).Throw()
|
||
}
|
||
|
||
Stor.MustWriteFile(filename, content, os.ModePerm)
|
||
return filename
|
||
}
|
||
|
||
// processGetContent 返回文件正文
|
||
func processReadFile(process *gou.Process) interface{} {
|
||
process.ValidateArgNums(1)
|
||
filename := process.ArgsString(0)
|
||
encode := process.ArgsBool(1, true)
|
||
|
||
stats, err := Stor.Stat(filename)
|
||
if err != nil {
|
||
exception.New("读取文件信息失败 %s", 500, err.Error()).Throw()
|
||
}
|
||
|
||
var content string
|
||
body := Stor.MustReadFile(filename)
|
||
if encode {
|
||
content = Encode(body)
|
||
} else {
|
||
content = string(body)
|
||
}
|
||
|
||
return maps.Map{
|
||
"size": stats.Size(),
|
||
"content": content,
|
||
}
|
||
}
|
||
|
||
// processGetToken 上传文件到腾讯云对象存储 COS
|
||
func processGetToken(process *gou.Process) interface{} {
|
||
process.ValidateArgNums(1)
|
||
name := process.ArgsString(0, "oss")
|
||
// bucket := process.ArgsString(1)
|
||
if name != "oss" {
|
||
exception.New("暂时支持 oss 存储", 400).Throw()
|
||
}
|
||
|
||
if share.App.Storage.OSS == nil {
|
||
exception.New("未配置 OSS 存储", 400).Throw()
|
||
}
|
||
|
||
app := share.App.Storage.OSS
|
||
id := strings.Split(app.Endpoint, ".")[0]
|
||
client, err := sts.NewClientWithAccessKey(id, app.ID, app.Secret)
|
||
if err != nil {
|
||
exception.New("配置错误 %s", 400, err.Error()).Throw()
|
||
}
|
||
|
||
//构建请求对象。
|
||
request := sts.CreateAssumeRoleRequest()
|
||
request.Scheme = "https"
|
||
request.Domain = "sts.aliyuncs.com"
|
||
|
||
//设置参数。关于参数含义和设置方法，请参见API参考。
|
||
request.RoleArn = app.RoleArn
|
||
request.RoleSessionName = app.SessionName
|
||
|
||
//发起请求，并得到响应。
|
||
response, err := client.AssumeRole(request)
|
||
if err != nil {
|
||
exception.New("配置错误 %s", 400, err.Error()).Throw()
|
||
}
|
||
|
||
res := any.Of(response.Credentials).Map()
|
||
res.Set("Endpoint", app.Endpoint)
|
||
return res.MapStrAny
|
||
}
|
||
|
||
// processGetURL 返回文件CDN地址
|
||
func processGetURL(process *gou.Process) interface{} {
|
||
process.ValidateArgNums(1)
|
||
filename := process.ArgsString(0)
|
||
// typ := process.ArgsString(1)
|
||
// stats, err := Stor.Stat(filename)
|
||
// if err != nil {
|
||
// exception.New("读取文件信息失败 %s", 500, err.Error()).Throw()
|
||
// }
|
||
body := Stor.MustReadFile(filename)
|
||
return string(body)
|
||
}
|