mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 00:10:25 +08:00
94 lines
2.8 KiB
Go
94 lines
2.8 KiB
Go
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)
|
|
}
|