geekai/api/service/oss/localstorage.go

package oss

import (
	"chatplus/core/types"
	"chatplus/utils"
	"fmt"
	"github.com/gin-gonic/gin"
	"net/url"
	"os"
	"path/filepath"
	"strings"
)

type LocalStorage struct {
	config   *types.LocalStorageConfig
	proxyURL string
}

func NewLocalStorage(config *types.AppConfig) LocalStorage {
	return LocalStorage{
		config:   &config.OSS.Local,
		proxyURL: config.ProxyURL,
	}
}

func (s LocalStorage) PutFile(ctx *gin.Context, name string) (File, error) {
	file, err := ctx.FormFile(name)
	if err != nil {
		return File{}, fmt.Errorf("error with get form: %v", err)
	}

	path, err := utils.GenUploadPath(s.config.BasePath, file.Filename)
	if err != nil {
		return File{}, fmt.Errorf("error with generate filename: %s", err.Error())
	}
	// 将文件保存到指定路径
	err = ctx.SaveUploadedFile(file, path)
	if err != nil {
		return File{}, fmt.Errorf("error with save upload file: %s", err.Error())
	}

	ext := filepath.Ext(file.Filename)
	return File{
		Name: file.Filename,
		URL:  utils.GenUploadUrl(s.config.BasePath, s.config.BaseURL, path),
		Ext:  ext,
		Size: file.Size,
	}, nil
}

func (s LocalStorage) PutImg(imageURL string, useProxy bool) (string, error) {
	parse, err := url.Parse(imageURL)
	if err != nil {
		return "", fmt.Errorf("error with parse image URL: %v", err)
	}
	filename := filepath.Base(parse.Path)
	filePath, err := utils.GenUploadPath(s.config.BasePath, filename)
	if err != nil {
		return "", fmt.Errorf("error with generate image dir: %v", err)
	}

	if useProxy {
		err = utils.DownloadFile(imageURL, filePath, s.proxyURL)
	} else {
		err = utils.DownloadFile(imageURL, filePath, "")
	}
	if err != nil {
		return "", fmt.Errorf("error with download image: %v", err)
	}

	return utils.GenUploadUrl(s.config.BasePath, s.config.BaseURL, filePath), nil
}

func (s LocalStorage) Delete(fileURL string) error {
	filePath := strings.Replace(fileURL, s.config.BaseURL, s.config.BasePath, 1)
	return os.Remove(filePath)
}

var _ Uploader = LocalStorage{}