feat(backup): add restore flag

pkg/internal/qiniu.go

@@ -5,7 +5,9 @@ import (
 	"context"
 	"errors"
 	"io"
+	"net/http"
 	"path/filepath"
+	"time"
 
 	"github.com/eiblog/eiblog/pkg/config"
 
@@ -15,9 +17,10 @@ import (
 
 // UploadParams upload params
 type UploadParams struct {
-	Name string
-	Size int64
-	Data io.Reader
+	Name           string
+	Size           int64
+	Data           io.Reader
+	NoCompletePath bool
 
 	Conf config.Qiniu
 }
@@ -28,7 +31,10 @@ func QiniuUpload(params UploadParams) (string, error) {
 		params.Conf.SecretKey == "" {
 		return "", errors.New("qiniu config error")
 	}
-	key := completeQiniuKey(params.Name)
+	key := params.Name
+	if !params.NoCompletePath {
+		key = filepath.Base(params.Name)
+	}
 
 	mac := qbox.NewMac(params.Conf.AccessKey,
 		params.Conf.SecretKey)
@@ -65,15 +71,19 @@ func QiniuUpload(params UploadParams) (string, error) {
 
 // DeleteParams delete params
 type DeleteParams struct {
-	Name string
-	Days int
+	Name           string
+	Days           int
+	NoCompletePath bool
 
 	Conf config.Qiniu
 }
 
 // QiniuDelete 删除文件
 func QiniuDelete(params DeleteParams) error {
-	key := completeQiniuKey(params.Name)
+	key := params.Name
+	if !params.NoCompletePath {
+		key = completeQiniuKey(params.Name)
+	}
 
 	mac := qbox.NewMac(params.Conf.AccessKey,
 		params.Conf.SecretKey)
@@ -95,6 +105,47 @@ func QiniuDelete(params DeleteParams) error {
 	return bucketManager.Delete(params.Conf.Bucket, key)
 }
 
+// ContentParams list params
+type ContentParams struct {
+	Prefix string
+
+	Conf config.Qiniu
+}
+
+// QiniuContent 获取文件列表
+func QiniuContent(params ContentParams) ([]byte, error) {
+	mac := qbox.NewMac(params.Conf.AccessKey,
+		params.Conf.SecretKey)
+	// region
+	region, err := storage.GetRegion(params.Conf.AccessKey, params.Conf.Bucket)
+	if err != nil {
+		return nil, err
+	}
+	cfg := &storage.Config{
+		UseHTTPS: true,
+		Region:   region,
+	}
+	// manager
+	bucketManager := storage.NewBucketManager(mac, cfg)
+	// list file
+	files, _, _, _, err := bucketManager.ListFiles(params.Conf.Bucket, params.Prefix, "", "", 2)
+	if err != nil {
+		return nil, err
+	}
+	if len(files) == 0 {
+		return nil, errors.New("no file")
+	}
+	deadline := time.Now().Add(time.Second * 60).Unix()
+	url := storage.MakePrivateURLv2(mac, "https://"+params.Conf.Domain, files[0].Key, deadline)
+	resp, err := http.Get(url)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	return io.ReadAll(resp.Body)
+}
+
 // completeQiniuKey 修复路径
 func completeQiniuKey(name string) string {
 	ext := filepath.Ext(name)

pkg/internal/qiniu_test.go

@@ -46,3 +46,18 @@ func TestQiniuUpload(t *testing.T) {
 		})
 	}
 }
+
+func TestQiniuContent(t *testing.T) {
+	params := ContentParams{
+		Conf: config.Qiniu{
+			AccessKey: os.Getenv("QINIU_ACCESSKEY"),
+			SecretKey: os.Getenv("QINIU_SECRETKEY"),
+			Bucket:    os.Getenv("QINIU_BUCKET"),
+			Domain:    "bu.st.deepzz.com",
+		},
+	}
+	_, err := QiniuContent(params)
+	if err != nil {
+		t.Errorf("QiniuList error = %v", err)
+	}
+}