in storage-tencentyuncos/tencentyuncos.go [79:142]
func (s *Storage) UploadFile(ctx *plugin.GinContext, condition plugin.UploadFileCondition) (resp plugin.UploadFileResponse) {
resp = plugin.UploadFileResponse{}
BucketURL, _ := url.Parse(fmt.Sprintf("https://%s.cos.%s.myqcloud.com", s.Config.BucketName, s.Config.Region))
BaseURL := &cos.BaseURL{BucketURL: BucketURL}
client := cos.NewClient(BaseURL, &http.Client{
Transport: &cos.AuthorizationTransport{
SecretID: s.Config.SecretID,
SecretKey: s.Config.SecretKey,
},
})
_, err := client.Bucket.IsExist(ctx)
if err != nil {
resp.OriginalError = fmt.Errorf("head bucket failed: %v", err)
resp.DisplayErrorMsg = plugin.MakeTranslator(i18n.ErrMisStorageConfig)
return resp
}
file, err := ctx.FormFile("file")
if err != nil {
resp.OriginalError = fmt.Errorf("get upload file failed: %v", err)
resp.DisplayErrorMsg = plugin.MakeTranslator(i18n.ErrFileNotFound)
return resp
}
if s.IsUnsupportedFileType(file.Filename, condition) {
resp.OriginalError = fmt.Errorf("file type not allowed")
resp.DisplayErrorMsg = plugin.MakeTranslator(i18n.ErrUnsupportedFileType)
return resp
}
if s.ExceedFileSizeLimit(file.Size, condition) {
resp.OriginalError = fmt.Errorf("file size too large")
resp.DisplayErrorMsg = plugin.MakeTranslator(i18n.ErrOverFileSizeLimit)
return resp
}
openFile, err := file.Open()
if err != nil {
resp.OriginalError = fmt.Errorf("get file failed: %v", err)
resp.DisplayErrorMsg = plugin.MakeTranslator(i18n.ErrFileNotFound)
return resp
}
defer openFile.Close()
objectKey := s.createObjectKey(file.Filename, condition.Source)
var options *cos.ObjectPutOptions
if s.Config.ACL == "public-read" {
options = &cos.ObjectPutOptions{
ACLHeaderOptions: &cos.ACLHeaderOptions{
XCosACL: "public-read",
},
}
}
_, err = client.Object.Put(ctx, objectKey, openFile, options)
if err != nil {
resp.OriginalError = fmt.Errorf("upload file failed: %v", err)
resp.DisplayErrorMsg = plugin.MakeTranslator(i18n.ErrUploadFileFailed)
return resp
}
resp.FullURL = s.Config.VisitUrlPrefix + objectKey
return resp
}