mayfly-go/server/internal/machine/application/machine_file.go

package application

import (
	"fmt"
	"io"
	"io/fs"
	"mayfly-go/internal/machine/domain/entity"
	"mayfly-go/internal/machine/domain/repository"
	"mayfly-go/internal/machine/infrastructure/machine"
	"mayfly-go/pkg/biz"
	"mayfly-go/pkg/gormx"
	"mayfly-go/pkg/model"
	"os"
	"strings"

	"github.com/pkg/sftp"
)

type MachineFile interface {
	// 分页获取机器文件信息列表
	GetPageList(condition *entity.MachineFile, pageParam *model.PageParam, toEntity any, orderBy ...string) *model.PageResult[any]

	// 根据条件获取
	GetMachineFile(condition *entity.MachineFile, cols ...string) error

	// 根据id获取
	GetById(id uint64, cols ...string) *entity.MachineFile

	Save(entity *entity.MachineFile)

	Delete(id uint64)

	// 获取文件关联的机器信息，主要用于记录日志使用
	GetMachine(fileId uint64) *machine.Info

	/**  sftp 相关操作 **/

	// 创建目录
	MkDir(fid uint64, path string)

	// 创建文件
	CreateFile(fid uint64, path string)

	// 读取目录
	ReadDir(fid uint64, path string) []fs.FileInfo

	// 获取指定目录内容大小
	GetDirSize(fid uint64, path string) string

	// 获取文件stat
	FileStat(fid uint64, path string) string

	// 读取文件内容
	ReadFile(fileId uint64, path string) *sftp.File

	// 写文件
	WriteFileContent(fileId uint64, path string, content []byte)

	// 文件上传
	UploadFile(fileId uint64, path, filename string, reader io.Reader)

	// 移除文件
	RemoveFile(fileId uint64, path string)
}

func newMachineFileApp(machineFileRepo repository.MachineFile, machineRepo repository.Machine) MachineFile {
	return &machineFileAppImpl{machineRepo: machineRepo, machineFileRepo: machineFileRepo}

}

type machineFileAppImpl struct {
	machineFileRepo repository.MachineFile
	machineRepo     repository.Machine
}

// 分页获取机器脚本信息列表
func (m *machineFileAppImpl) GetPageList(condition *entity.MachineFile, pageParam *model.PageParam, toEntity any, orderBy ...string) *model.PageResult[any] {
	return m.machineFileRepo.GetPageList(condition, pageParam, toEntity, orderBy...)
}

// 根据条件获取
func (m *machineFileAppImpl) GetMachineFile(condition *entity.MachineFile, cols ...string) error {
	return m.machineFileRepo.GetMachineFile(condition, cols...)
}

// 根据id获取
func (m *machineFileAppImpl) GetById(id uint64, cols ...string) *entity.MachineFile {
	return m.machineFileRepo.GetById(id, cols...)
}

// 保存机器文件配置
func (m *machineFileAppImpl) Save(entity *entity.MachineFile) {
	biz.NotNil(m.machineRepo.GetById(entity.MachineId, "Name"), "该机器不存在")

	if entity.Id != 0 {
		gormx.UpdateById(entity)
	} else {
		gormx.Insert(entity)
	}
}

// 根据id删除
func (m *machineFileAppImpl) Delete(id uint64) {
	m.machineFileRepo.Delete(id)
}

func (m *machineFileAppImpl) ReadDir(fid uint64, path string) []fs.FileInfo {
	path, machineId := m.checkAndReturnPathMid(fid, path)
	if !strings.HasSuffix(path, "/") {
		path = path + "/"
	}

	sftpCli := m.getSftpCli(machineId)
	fis, err := sftpCli.ReadDir(path)
	biz.ErrIsNilAppendErr(err, "读取目录失败: %s")
	return fis
}

func (m *machineFileAppImpl) GetDirSize(fid uint64, path string) string {
	_, machineId := m.checkAndReturnPathMid(fid, path)
	res, err := GetMachineApp().GetCli(machineId).Run(fmt.Sprintf("du -sh %s", path))
	if err != nil {
		// 若存在目录为空，则可能会返回如下内容。最后一行即为真正目录内容所占磁盘空间大小
		//du: cannot access ‘/proc/19087/fd/3’: No such file or directory\n
		//du: cannot access ‘/proc/19087/fdinfo/3’: No such file or directory\n
		//18G     /\n
		if res == "" {
			panic(biz.NewBizErr(fmt.Sprintf("获取目录大小失败: %s", err.Error())))
		}
		strs := strings.Split(res, "\n")
		res = strs[len(strs)-2]

		if !strings.Contains(res, "\t") {
			panic(biz.NewBizErr(res))
		}
	}
	// 返回 32K\t/tmp\n
	return strings.Split(res, "\t")[0]
}

func (m *machineFileAppImpl) FileStat(fid uint64, path string) string {
	_, machineId := m.checkAndReturnPathMid(fid, path)
	res, err := GetMachineApp().GetCli(machineId).Run(fmt.Sprintf("stat -L %s", path))
	biz.ErrIsNil(err, res)
	return res
}

func (m *machineFileAppImpl) MkDir(fid uint64, path string) {
	path, machineId := m.checkAndReturnPathMid(fid, path)
	if !strings.HasSuffix(path, "/") {
		path = path + "/"
	}

	sftpCli := m.getSftpCli(machineId)
	err := sftpCli.Mkdir(path)
	biz.ErrIsNilAppendErr(err, "创建目录失败: %s")
}

func (m *machineFileAppImpl) CreateFile(fid uint64, path string) {
	path, machineId := m.checkAndReturnPathMid(fid, path)
	sftpCli := m.getSftpCli(machineId)
	file, err := sftpCli.Create(path)
	biz.ErrIsNilAppendErr(err, "创建文件失败: %s")
	defer file.Close()
}

func (m *machineFileAppImpl) ReadFile(fileId uint64, path string) *sftp.File {
	path, machineId := m.checkAndReturnPathMid(fileId, path)
	sftpCli := m.getSftpCli(machineId)
	// 读取文件内容
	fc, err := sftpCli.Open(path)
	biz.ErrIsNilAppendErr(err, "打开文件失败: %s")
	return fc
}

// 写文件内容
func (m *machineFileAppImpl) WriteFileContent(fileId uint64, path string, content []byte) {
	_, machineId := m.checkAndReturnPathMid(fileId, path)

	sftpCli := m.getSftpCli(machineId)
	f, err := sftpCli.OpenFile(path, os.O_WRONLY|os.O_TRUNC|os.O_CREATE|os.O_RDWR)
	biz.ErrIsNilAppendErr(err, "打开文件失败: %s")
	defer f.Close()

	fi, _ := f.Stat()
	biz.IsTrue(!fi.IsDir(), "该路径不是文件")
	f.Write(content)
}

// 上传文件
func (m *machineFileAppImpl) UploadFile(fileId uint64, path, filename string, reader io.Reader) {
	path, machineId := m.checkAndReturnPathMid(fileId, path)
	if !strings.HasSuffix(path, "/") {
		path = path + "/"
	}

	sftpCli := m.getSftpCli(machineId)
	createfile, err := sftpCli.Create(path + filename)
	biz.ErrIsNilAppendErr(err, "创建文件失败: %s")
	defer createfile.Close()

	io.Copy(createfile, reader)
}

// 删除文件
func (m *machineFileAppImpl) RemoveFile(fileId uint64, path string) {
	path, machineId := m.checkAndReturnPathMid(fileId, path)

	sftpCli := m.getSftpCli(machineId)
	file, err := sftpCli.Open(path)
	biz.ErrIsNilAppendErr(err, "打开文件失败: %s")
	fi, _ := file.Stat()
	if fi.IsDir() {
		err = sftpCli.RemoveDirectory(path)
		// 如果文件夹有内容会删除失败，则使用rm -rf命令删除
		if err != nil {
			GetMachineApp().GetCli(machineId).Run(fmt.Sprintf("rm -rf %s", path))
			err = nil
		}
	} else {
		err = sftpCli.Remove(path)
	}
	biz.ErrIsNilAppendErr(err, "删除文件失败: %s")
}

// 获取sftp client
func (m *machineFileAppImpl) getSftpCli(machineId uint64) *sftp.Client {
	return GetMachineApp().GetCli(machineId).GetSftpCli()
}

func (m *machineFileAppImpl) GetMachine(fileId uint64) *machine.Info {
	return GetMachineApp().GetCli(m.GetById(fileId).MachineId).GetMachine()
}

// 校验并返回实际可访问的文件path
func (m *machineFileAppImpl) checkAndReturnPathMid(fid uint64, inputPath string) (string, uint64) {
	biz.IsTrue(fid != 0, "文件id不能为空")
	mf := m.GetById(fid)
	biz.NotNil(mf, "文件不存在")
	if inputPath != "" {
		// 接口传入的地址需为配置路径的子路径
		biz.IsTrue(strings.HasPrefix(inputPath, mf.Path), "无权访问该目录或文件")
		return inputPath, mf.MachineId
	} else {
		return mf.Path, mf.MachineId
	}
}