mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2024-09-21 08:06:01 +08:00
277 lines
7.1 KiB
Go
277 lines
7.1 KiB
Go
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/1Panel-dev/1Panel/backend/app/dto/request"
|
|
"github.com/1Panel-dev/1Panel/backend/app/dto/response"
|
|
"github.com/1Panel-dev/1Panel/backend/buserr"
|
|
"github.com/1Panel-dev/1Panel/backend/constant"
|
|
|
|
"github.com/1Panel-dev/1Panel/backend/global"
|
|
"github.com/1Panel-dev/1Panel/backend/utils/common"
|
|
"github.com/1Panel-dev/1Panel/backend/utils/files"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type FileService struct {
|
|
}
|
|
|
|
type IFileService interface {
|
|
GetFileList(op request.FileOption) (response.FileInfo, error)
|
|
SearchUploadWithPage(req request.SearchUploadWithPage) (int64, interface{}, error)
|
|
GetFileTree(op request.FileOption) ([]response.FileTree, error)
|
|
Create(op request.FileCreate) error
|
|
Delete(op request.FileDelete) error
|
|
BatchDelete(op request.FileBatchDelete) error
|
|
ChangeMode(op request.FileCreate) error
|
|
Compress(c request.FileCompress) error
|
|
DeCompress(c request.FileDeCompress) error
|
|
GetContent(op request.FileOption) (response.FileInfo, error)
|
|
SaveContent(edit request.FileEdit) error
|
|
FileDownload(d request.FileDownload) (string, error)
|
|
DirSize(req request.DirSizeReq) (response.DirSizeRes, error)
|
|
ChangeName(req request.FileRename) error
|
|
Wget(w request.FileWget) (string, error)
|
|
MvFile(m request.FileMove) error
|
|
ChangeOwner(req request.FileRoleUpdate) error
|
|
}
|
|
|
|
func NewIFileService() IFileService {
|
|
return &FileService{}
|
|
}
|
|
|
|
func (f *FileService) GetFileList(op request.FileOption) (response.FileInfo, error) {
|
|
var fileInfo response.FileInfo
|
|
if _, err := os.Stat(op.Path); err != nil && os.IsNotExist(err) {
|
|
return fileInfo, nil
|
|
}
|
|
info, err := files.NewFileInfo(op.FileOption)
|
|
if err != nil {
|
|
return fileInfo, err
|
|
}
|
|
fileInfo.FileInfo = *info
|
|
return fileInfo, nil
|
|
}
|
|
|
|
func (f *FileService) SearchUploadWithPage(req request.SearchUploadWithPage) (int64, interface{}, error) {
|
|
var (
|
|
files []response.UploadInfo
|
|
backData []response.UploadInfo
|
|
)
|
|
_ = filepath.Walk(req.Path, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
if !info.IsDir() {
|
|
files = append(files, response.UploadInfo{
|
|
CreatedAt: info.ModTime().Format("2006-01-02 15:04:05"),
|
|
Size: int(info.Size()),
|
|
Name: info.Name(),
|
|
})
|
|
}
|
|
return nil
|
|
})
|
|
total, start, end := len(files), (req.Page-1)*req.PageSize, req.Page*req.PageSize
|
|
if start > total {
|
|
backData = make([]response.UploadInfo, 0)
|
|
} else {
|
|
if end >= total {
|
|
end = total
|
|
}
|
|
backData = files[start:end]
|
|
}
|
|
return int64(total), backData, nil
|
|
}
|
|
|
|
func (f *FileService) GetFileTree(op request.FileOption) ([]response.FileTree, error) {
|
|
var treeArray []response.FileTree
|
|
info, err := files.NewFileInfo(op.FileOption)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
node := response.FileTree{
|
|
ID: common.GetUuid(),
|
|
Name: info.Name,
|
|
Path: info.Path,
|
|
}
|
|
for _, v := range info.Items {
|
|
if v.IsDir {
|
|
node.Children = append(node.Children, response.FileTree{
|
|
ID: common.GetUuid(),
|
|
Name: v.Name,
|
|
Path: v.Path,
|
|
})
|
|
}
|
|
}
|
|
return append(treeArray, node), nil
|
|
}
|
|
|
|
func (f *FileService) Create(op request.FileCreate) error {
|
|
fo := files.NewFileOp()
|
|
if fo.Stat(op.Path) {
|
|
return buserr.New(constant.ErrFileIsExit)
|
|
}
|
|
if op.IsDir {
|
|
return fo.CreateDir(op.Path, fs.FileMode(op.Mode))
|
|
} else {
|
|
if op.IsLink {
|
|
if !fo.Stat(op.LinkPath) {
|
|
return buserr.New(constant.ErrLinkPathNotFound)
|
|
}
|
|
return fo.LinkFile(op.LinkPath, op.Path, op.IsSymlink)
|
|
} else {
|
|
return fo.CreateFile(op.Path)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (f *FileService) Delete(op request.FileDelete) error {
|
|
fo := files.NewFileOp()
|
|
if op.IsDir {
|
|
return fo.DeleteDir(op.Path)
|
|
} else {
|
|
return fo.DeleteFile(op.Path)
|
|
}
|
|
}
|
|
|
|
func (f *FileService) BatchDelete(op request.FileBatchDelete) error {
|
|
fo := files.NewFileOp()
|
|
if op.IsDir {
|
|
for _, file := range op.Paths {
|
|
if err := fo.DeleteDir(file); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
} else {
|
|
for _, file := range op.Paths {
|
|
if err := fo.DeleteFile(file); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (f *FileService) ChangeMode(op request.FileCreate) error {
|
|
fo := files.NewFileOp()
|
|
if op.Sub {
|
|
return fo.ChmodR(op.Path, op.Mode)
|
|
} else {
|
|
return fo.Chmod(op.Path, fs.FileMode(op.Mode))
|
|
}
|
|
}
|
|
|
|
func (f *FileService) ChangeOwner(req request.FileRoleUpdate) error {
|
|
fo := files.NewFileOp()
|
|
return fo.ChownR(req.Path, req.User, req.Group, req.Sub)
|
|
}
|
|
|
|
func (f *FileService) Compress(c request.FileCompress) error {
|
|
fo := files.NewFileOp()
|
|
if !c.Replace && fo.Stat(filepath.Join(c.Dst, c.Name)) {
|
|
return buserr.New(constant.ErrFileIsExit)
|
|
}
|
|
return fo.Compress(c.Files, c.Dst, c.Name, files.CompressType(c.Type))
|
|
}
|
|
|
|
func (f *FileService) DeCompress(c request.FileDeCompress) error {
|
|
fo := files.NewFileOp()
|
|
return fo.Decompress(c.Path, c.Dst, files.CompressType(c.Type))
|
|
}
|
|
|
|
func (f *FileService) GetContent(op request.FileOption) (response.FileInfo, error) {
|
|
info, err := files.NewFileInfo(op.FileOption)
|
|
if err != nil {
|
|
return response.FileInfo{}, err
|
|
}
|
|
return response.FileInfo{FileInfo: *info}, nil
|
|
}
|
|
|
|
func (f *FileService) SaveContent(edit request.FileEdit) error {
|
|
info, err := files.NewFileInfo(files.FileOption{
|
|
Path: edit.Path,
|
|
Expand: false,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fo := files.NewFileOp()
|
|
return fo.WriteFile(edit.Path, strings.NewReader(edit.Content), info.FileMode)
|
|
}
|
|
|
|
func (f *FileService) ChangeName(req request.FileRename) error {
|
|
fo := files.NewFileOp()
|
|
return fo.Rename(req.OldName, req.NewName)
|
|
}
|
|
|
|
func (f *FileService) Wget(w request.FileWget) (string, error) {
|
|
fo := files.NewFileOp()
|
|
key := "file-wget-" + common.GetUuid()
|
|
return key, fo.DownloadFileWithProcess(w.Url, filepath.Join(w.Path, w.Name), key)
|
|
}
|
|
|
|
func (f *FileService) MvFile(m request.FileMove) error {
|
|
fo := files.NewFileOp()
|
|
if !fo.Stat(m.NewPath) {
|
|
return buserr.New(constant.ErrPathNotFound)
|
|
}
|
|
for _, path := range m.OldPaths {
|
|
if path == m.NewPath || strings.Contains(m.NewPath, path) {
|
|
return buserr.New(constant.ErrMovePathFailed)
|
|
}
|
|
}
|
|
if m.Type == "cut" {
|
|
return fo.Cut(m.OldPaths, m.NewPath)
|
|
}
|
|
var errs []error
|
|
if m.Type == "copy" {
|
|
for _, src := range m.OldPaths {
|
|
if err := fo.Copy(src, m.NewPath); err != nil {
|
|
errs = append(errs, err)
|
|
global.LOG.Errorf("copy file [%s] to [%s] failed, err: %s", src, m.NewPath, err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
var errString string
|
|
for _, err := range errs {
|
|
errString += err.Error() + "\n"
|
|
}
|
|
if errString != "" {
|
|
return errors.New(errString)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (f *FileService) FileDownload(d request.FileDownload) (string, error) {
|
|
filePath := d.Paths[0]
|
|
if d.Compress {
|
|
tempPath := filepath.Join(os.TempDir(), fmt.Sprintf("%d", time.Now().UnixNano()))
|
|
if err := os.MkdirAll(tempPath, os.ModePerm); err != nil {
|
|
return "", err
|
|
}
|
|
fo := files.NewFileOp()
|
|
if err := fo.Compress(d.Paths, tempPath, d.Name, files.CompressType(d.Type)); err != nil {
|
|
return "", err
|
|
}
|
|
filePath = filepath.Join(tempPath, d.Name)
|
|
}
|
|
return filePath, nil
|
|
}
|
|
|
|
func (f *FileService) DirSize(req request.DirSizeReq) (response.DirSizeRes, error) {
|
|
fo := files.NewFileOp()
|
|
size, err := fo.GetDirSize(req.Path)
|
|
if err != nil {
|
|
return response.DirSizeRes{}, err
|
|
}
|
|
return response.DirSizeRes{Size: size}, nil
|
|
}
|