feat: 支持关联多标签、计划任务立即执行、标签相关操作优化

server/internal/machine/application/machine_term_op.go

@@ -0,0 +1,93 @@
+package application
+
+import (
+	"context"
+	"fmt"
+	"mayfly-go/internal/machine/config"
+	"mayfly-go/internal/machine/domain/entity"
+	"mayfly-go/internal/machine/domain/repository"
+	"mayfly-go/internal/machine/mcm"
+	"mayfly-go/pkg/base"
+	"mayfly-go/pkg/contextx"
+	"mayfly-go/pkg/errorx"
+	"mayfly-go/pkg/model"
+	"mayfly-go/pkg/utils/stringx"
+	"os"
+	"path"
+	"time"
+
+	"github.com/gorilla/websocket"
+)
+
+type MachineTermOp interface {
+	base.App[*entity.MachineTermOp]
+
+	// 终端连接操作
+	TermConn(ctx context.Context, cli *mcm.Cli, wsConn *websocket.Conn, rows, cols int) error
+
+	GetPageList(condition *entity.MachineTermOp, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error)
+}
+
+func newMachineTermOpApp(machineTermOpRepo repository.MachineTermOp) MachineTermOp {
+	return &machineTermOpAppImpl{
+		base.AppImpl[*entity.MachineTermOp, repository.MachineTermOp]{Repo: machineTermOpRepo},
+	}
+}
+
+type machineTermOpAppImpl struct {
+	base.AppImpl[*entity.MachineTermOp, repository.MachineTermOp]
+}
+
+func (a *machineTermOpAppImpl) TermConn(ctx context.Context, cli *mcm.Cli, wsConn *websocket.Conn, rows, cols int) error {
+	var recorder *mcm.Recorder
+	var termOpRecord *entity.MachineTermOp
+
+	// 开启终端操作记录
+	if cli.Info.EnableRecorder == 1 {
+		now := time.Now()
+		la := contextx.GetLoginAccount(ctx)
+
+		termOpRecord = new(entity.MachineTermOp)
+
+		termOpRecord.CreateTime = &now
+		termOpRecord.Creator = la.Username
+		termOpRecord.CreatorId = la.Id
+
+		termOpRecord.MachineId = cli.Info.Id
+		termOpRecord.Username = cli.Info.Username
+
+		// 回放文件路径为: 基础配置路径/操作日期(202301)/day/hour/randstr.cast
+		recRelPath := path.Join(now.Format("200601"), fmt.Sprintf("%d", now.Day()), fmt.Sprintf("%d", now.Hour()))
+		// 文件绝对路径
+		recAbsPath := path.Join(config.GetMachine().TerminalRecPath, recRelPath)
+		os.MkdirAll(recAbsPath, 0766)
+		filename := fmt.Sprintf("%s.cast", stringx.RandByChars(18, stringx.LowerChars))
+		f, err := os.OpenFile(path.Join(recAbsPath, filename), os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0766)
+		if err != nil {
+			return errorx.NewBiz("创建终端回放记录文件失败: %s", err.Error())
+		}
+		defer f.Close()
+
+		termOpRecord.RecordFilePath = path.Join(recRelPath, filename)
+		recorder = mcm.NewRecorder(f)
+	}
+
+	mts, err := mcm.NewTerminalSession(stringx.Rand(16), wsConn, cli, rows, cols, recorder)
+	if err != nil {
+		return err
+	}
+
+	mts.Start()
+	defer mts.Stop()
+
+	if termOpRecord != nil {
+		now := time.Now()
+		termOpRecord.EndTime = &now
+		return a.Insert(ctx, termOpRecord)
+	}
+	return nil
+}
+
+func (a *machineTermOpAppImpl) GetPageList(condition *entity.MachineTermOp, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
+	return a.GetRepo().GetPageList(condition, pageParam, toEntity)
+}