feat: add minio service implementation, download midjourney image to local storage

api/utils/upload.go

@@ -0,0 +1,68 @@
+package utils
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+	"os"
+	"path/filepath"
+	"strings"
+	"time"
+)
+
+// GenUploadPath 生成上传文件路径
+func GenUploadPath(basePath, filename string) (string, error) {
+	now := time.Now()
+	dir := fmt.Sprintf("%s/upload/%d/%d", basePath, now.Year(), now.Month())
+	_, err := os.Stat(dir)
+	if err != nil {
+		err = os.MkdirAll(dir, 0755)
+		if err != nil {
+			return "", fmt.Errorf("创建上传目录失败：%s", err)
+		}
+	}
+	fileExt := filepath.Ext(filename)
+	return fmt.Sprintf("%s/%d%s", dir, now.UnixMilli(), fileExt), nil
+}
+
+// GenUploadUrl 生成上传文件 URL
+func GenUploadUrl(basePath, baseUrl string, filePath string) string {
+	return strings.Replace(filePath, basePath, baseUrl, 1)
+}
+
+func DownloadFile(fileURL string, filepath string, proxy string) error {
+	var client *http.Client
+	if proxy == "" {
+		client = http.DefaultClient
+	} else {
+		proxyURL, _ := url.Parse(proxy)
+		client = &http.Client{
+			Transport: &http.Transport{
+				Proxy: http.ProxyURL(proxyURL),
+			},
+		}
+	}
+	req, err := http.NewRequest("GET", fileURL, nil)
+	if err != nil {
+		return err
+	}
+
+	resp, err := client.Do(req)
+	if err != nil {
+		return err
+	}
+
+	file, err := os.Create(filepath)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	_, err = io.Copy(file, resp.Body)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}