修复 「1.2.2 用户管理-用户列表勾选单一用户会全选 」 close #216

server/api/term_handler.go

@@ -0,0 +1,104 @@
+package api
+
+import (
+	"context"
+	"time"
+	"unicode/utf8"
+
+	"next-terminal/server/dto"
+	"next-terminal/server/global/session"
+	"next-terminal/server/term"
+
+	"github.com/gorilla/websocket"
+)
+
+type TermHandler struct {
+	sessionId    string
+	isRecording  bool
+	ws           *websocket.Conn
+	nextTerminal *term.NextTerminal
+	ctx          context.Context
+	cancel       context.CancelFunc
+	dataChan     chan rune
+	tick         *time.Ticker
+}
+
+func NewTermHandler(sessionId string, isRecording bool, ws *websocket.Conn, nextTerminal *term.NextTerminal) *TermHandler {
+	ctx, cancel := context.WithCancel(context.Background())
+	tick := time.NewTicker(time.Millisecond * time.Duration(60))
+	return &TermHandler{
+		sessionId:    sessionId,
+		isRecording:  isRecording,
+		ws:           ws,
+		nextTerminal: nextTerminal,
+		ctx:          ctx,
+		cancel:       cancel,
+		dataChan:     make(chan rune),
+		tick:         tick,
+	}
+}
+
+func (r TermHandler) Start() {
+	go r.readFormTunnel()
+	go r.writeToWebsocket()
+}
+
+func (r TermHandler) Stop() {
+	r.tick.Stop()
+	r.cancel()
+}
+
+func (r TermHandler) readFormTunnel() {
+	for {
+		select {
+		case <-r.ctx.Done():
+			return
+		default:
+			rn, size, err := r.nextTerminal.StdoutReader.ReadRune()
+			if err != nil {
+				return
+			}
+			if size > 0 {
+				r.dataChan <- rn
+			}
+		}
+	}
+}
+
+func (r TermHandler) writeToWebsocket() {
+	var buf []byte
+	for {
+		select {
+		case <-r.ctx.Done():
+			return
+		case <-r.tick.C:
+			if len(buf) > 0 {
+				s := string(buf)
+				if err := WriteMessage(r.ws, dto.NewMessage(Data, s)); err != nil {
+					return
+				}
+				// 录屏
+				if r.isRecording {
+					_ = r.nextTerminal.Recorder.WriteData(s)
+				}
+				nextSession := session.GlobalSessionManager.GetById(r.sessionId)
+				// 监控
+				if nextSession != nil && len(nextSession.Observer.All()) > 0 {
+					obs := nextSession.Observer.All()
+					for _, ob := range obs {
+						_ = WriteMessage(ob.WebSocket, dto.NewMessage(Data, s))
+					}
+				}
+				buf = []byte{}
+			}
+		case data := <-r.dataChan:
+			if data != utf8.RuneError {
+				p := make([]byte, utf8.RuneLen(data))
+				utf8.EncodeRune(p, data)
+				buf = append(buf, p...)
+			} else {
+				buf = append(buf, []byte("@")...)
+			}
+		}
+	}
+}