mirror of
https://github.com/eiblog/eiblog.git
synced 2026-02-04 13:52:26 +08:00
102 lines
2.3 KiB
Go
102 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"path/filepath"
|
|
|
|
"github.com/eiblog/eiblog/setting"
|
|
"github.com/qiniu/api.v7/auth/qbox"
|
|
"github.com/qiniu/api.v7/storage"
|
|
)
|
|
|
|
// 进度条
|
|
func onProgress(fsize, uploaded int64) {
|
|
d := int(float64(uploaded) / float64(fsize) * 100)
|
|
if fsize == uploaded {
|
|
fmt.Printf("\rUpload completed! \n")
|
|
} else {
|
|
fmt.Printf("\r%02d%% uploaded ", int(d))
|
|
}
|
|
}
|
|
|
|
// 上传文件
|
|
func FileUpload(name string, size int64, data io.Reader) (string, error) {
|
|
if setting.Conf.Qiniu.AccessKey == "" || setting.Conf.Qiniu.SecretKey == "" {
|
|
return "", errors.New("qiniu config error")
|
|
}
|
|
|
|
key := getKey(name)
|
|
mac := qbox.NewMac(setting.Conf.Qiniu.AccessKey, setting.Conf.Qiniu.SecretKey)
|
|
// 设置上传的策略
|
|
putPolicy := &storage.PutPolicy{
|
|
Scope: setting.Conf.Qiniu.Bucket,
|
|
Expires: 3600,
|
|
InsertOnly: 1,
|
|
}
|
|
// 上传token
|
|
upToken := putPolicy.UploadToken(mac)
|
|
|
|
// 上传配置
|
|
cfg := &storage.Config{
|
|
Zone: &storage.ZoneHuadong,
|
|
UseHTTPS: true,
|
|
}
|
|
// uploader
|
|
uploader := storage.NewFormUploader(cfg)
|
|
ret := new(storage.PutRet)
|
|
putExtra := &storage.PutExtra{}
|
|
|
|
err := uploader.Put(context.Background(), ret, upToken, key, data, size, putExtra)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
url := "https://" + setting.Conf.Qiniu.Domain + "/" + key
|
|
return url, nil
|
|
}
|
|
|
|
// 删除文件
|
|
func FileDelete(name string) error {
|
|
key := getKey(name)
|
|
|
|
mac := qbox.NewMac(setting.Conf.Qiniu.AccessKey, setting.Conf.Qiniu.SecretKey)
|
|
// 上传配置
|
|
cfg := &storage.Config{
|
|
Zone: &storage.ZoneHuadong,
|
|
UseHTTPS: true,
|
|
}
|
|
// manager
|
|
bucketManager := storage.NewBucketManager(mac, cfg)
|
|
// Delete
|
|
err := bucketManager.Delete(setting.Conf.Qiniu.Bucket, key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// 修复路径
|
|
func getKey(name string) string {
|
|
ext := filepath.Ext(name)
|
|
var key string
|
|
switch ext {
|
|
case ".bmp", ".png", ".jpg", ".gif", ".ico":
|
|
key = "blog/img/" + name
|
|
case ".mov", ".mp4":
|
|
key = "blog/video/" + name
|
|
case ".go", ".js", ".css", ".cpp", ".php", ".rb",
|
|
".java", ".py", ".sql", ".lua", ".html",
|
|
".sh", ".xml", ".cs":
|
|
key = "blog/code/" + name
|
|
case ".txt", ".md", ".ini", ".yaml", ".yml",
|
|
".doc", ".ppt", ".pdf":
|
|
key = "blog/document/" + name
|
|
case ".zip", ".rar", ".tar", ".gz":
|
|
key = "blog/archive/" + name
|
|
}
|
|
return key
|
|
}
|