mirror of
https://github.com/wgh136/nysoure.git
synced 2025-12-16 15:51:14 +00:00
feat: add FTP storage functionality with API integration
This commit is contained in:
164
server/storage/ftp.go
Normal file
164
server/storage/ftp.go
Normal file
@@ -0,0 +1,164 @@
|
||||
package storage
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"os"
|
||||
"path"
|
||||
"time"
|
||||
|
||||
"github.com/gofiber/fiber/v3/log"
|
||||
"github.com/google/uuid"
|
||||
"github.com/jlaffaye/ftp"
|
||||
)
|
||||
|
||||
type FTPStorage struct {
|
||||
Host string // FTP服务器地址,例如: "ftp.example.com:21"
|
||||
Username string // FTP用户名
|
||||
Password string // FTP密码
|
||||
BasePath string // FTP服务器上的基础路径,例如: "/uploads"
|
||||
Domain string // 文件服务器域名,用于生成下载链接,例如: "files.example.com"
|
||||
}
|
||||
|
||||
func (f *FTPStorage) Upload(filePath string, fileName string) (string, error) {
|
||||
// 连接到FTP服务器
|
||||
conn, err := ftp.Dial(f.Host, ftp.DialWithTimeout(10*time.Second))
|
||||
if err != nil {
|
||||
log.Error("Failed to connect to FTP server: ", err)
|
||||
return "", errors.New("failed to connect to FTP server")
|
||||
}
|
||||
defer conn.Quit()
|
||||
|
||||
// 登录
|
||||
err = conn.Login(f.Username, f.Password)
|
||||
if err != nil {
|
||||
log.Error("Failed to login to FTP server: ", err)
|
||||
return "", errors.New("failed to login to FTP server")
|
||||
}
|
||||
|
||||
// 生成唯一的存储键
|
||||
storageKey := uuid.NewString() + "/" + fileName
|
||||
remotePath := path.Join(f.BasePath, storageKey)
|
||||
|
||||
// 创建远程目录
|
||||
remoteDir := path.Dir(remotePath)
|
||||
err = f.createRemoteDir(conn, remoteDir)
|
||||
if err != nil {
|
||||
log.Error("Failed to create remote directory: ", err)
|
||||
return "", errors.New("failed to create remote directory")
|
||||
}
|
||||
|
||||
// 打开本地文件
|
||||
file, err := os.Open(filePath)
|
||||
if err != nil {
|
||||
log.Error("Failed to open local file: ", err)
|
||||
return "", errors.New("failed to open local file")
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// 上传文件
|
||||
err = conn.Stor(remotePath, file)
|
||||
if err != nil {
|
||||
log.Error("Failed to upload file to FTP server: ", err)
|
||||
return "", errors.New("failed to upload file to FTP server")
|
||||
}
|
||||
|
||||
return storageKey, nil
|
||||
}
|
||||
|
||||
func (f *FTPStorage) Download(storageKey string, fileName string) (string, error) {
|
||||
// 返回文件下载链接:域名 + 存储键
|
||||
if f.Domain == "" {
|
||||
return "", errors.New("domain is not configured")
|
||||
}
|
||||
return "https://" + f.Domain + "/" + storageKey, nil
|
||||
}
|
||||
|
||||
func (f *FTPStorage) Delete(storageKey string) error {
|
||||
// 连接到FTP服务器
|
||||
conn, err := ftp.Dial(f.Host, ftp.DialWithTimeout(10*time.Second))
|
||||
if err != nil {
|
||||
log.Error("Failed to connect to FTP server: ", err)
|
||||
return errors.New("failed to connect to FTP server")
|
||||
}
|
||||
defer conn.Quit()
|
||||
|
||||
// 登录
|
||||
err = conn.Login(f.Username, f.Password)
|
||||
if err != nil {
|
||||
log.Error("Failed to login to FTP server: ", err)
|
||||
return errors.New("failed to login to FTP server")
|
||||
}
|
||||
|
||||
// 删除文件
|
||||
remotePath := path.Join(f.BasePath, storageKey)
|
||||
err = conn.Delete(remotePath)
|
||||
if err != nil {
|
||||
log.Error("Failed to delete file from FTP server: ", err)
|
||||
return errors.New("failed to delete file from FTP server")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *FTPStorage) ToString() string {
|
||||
data, _ := json.Marshal(f)
|
||||
return string(data)
|
||||
}
|
||||
|
||||
func (f *FTPStorage) FromString(config string) error {
|
||||
var ftpConfig FTPStorage
|
||||
if err := json.Unmarshal([]byte(config), &ftpConfig); err != nil {
|
||||
return err
|
||||
}
|
||||
f.Host = ftpConfig.Host
|
||||
f.Username = ftpConfig.Username
|
||||
f.Password = ftpConfig.Password
|
||||
f.BasePath = ftpConfig.BasePath
|
||||
f.Domain = ftpConfig.Domain
|
||||
|
||||
if f.Host == "" || f.Username == "" || f.Password == "" || f.Domain == "" {
|
||||
return errors.New("invalid FTP configuration")
|
||||
}
|
||||
if f.BasePath == "" {
|
||||
f.BasePath = "/"
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *FTPStorage) Type() string {
|
||||
return "ftp"
|
||||
}
|
||||
|
||||
// createRemoteDir 递归创建远程目录
|
||||
func (f *FTPStorage) createRemoteDir(conn *ftp.ServerConn, dirPath string) error {
|
||||
if dirPath == "" || dirPath == "/" || dirPath == "." {
|
||||
return nil
|
||||
}
|
||||
|
||||
// 尝试进入目录,如果失败则创建
|
||||
err := conn.ChangeDir(dirPath)
|
||||
if err == nil {
|
||||
// 目录存在,返回根目录
|
||||
conn.ChangeDir("/")
|
||||
return nil
|
||||
}
|
||||
|
||||
// 递归创建父目录
|
||||
parentDir := path.Dir(dirPath)
|
||||
if parentDir != dirPath {
|
||||
err = f.createRemoteDir(conn, parentDir)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
// 创建当前目录
|
||||
err = conn.MakeDir(dirPath)
|
||||
if err != nil {
|
||||
// 忽略目录已存在的错误
|
||||
return nil
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
@@ -43,6 +43,14 @@ func NewStorage(s model.Storage) IStorage {
|
||||
return nil
|
||||
}
|
||||
return &r
|
||||
|
||||
case "ftp":
|
||||
r := FTPStorage{}
|
||||
err := r.FromString(s.Config)
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
return &r
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user