增加录屏回放demo

pkg/api/recording.go

@@ -0,0 +1,93 @@
+package api
+
+import (
+	"encoding/json"
+	"next-terminal/pkg/utils"
+	"os"
+	"time"
+)
+
+type Env struct {
+	Shell string `json:"SHELL"`
+	Term  string `json:"TERM"`
+}
+
+type Header struct {
+	Title     string `json:"title"`
+	Version   int    `json:"version"`
+	Height    int    `json:"height"`
+	Width     int    `json:"width"`
+	Env       Env    `json:"env"`
+	Timestamp int    `json:"timestamp"`
+}
+
+type Recorder struct {
+	file      *os.File
+	timestamp int
+}
+
+func NewRecorder(filename string) (recorder *Recorder, err error) {
+	recorder = &Recorder{}
+
+	if utils.FileExists(filename) {
+		if err := os.RemoveAll(filename); err != nil {
+			return nil, err
+		}
+	}
+	var file *os.File
+	file, err = os.Create(filename)
+	if err != nil {
+		return nil, err
+	}
+
+	recorder.file = file
+	return recorder, nil
+}
+
+func (recorder *Recorder) Close() {
+	if recorder.file != nil {
+		recorder.file.Close()
+	}
+}
+
+func (recorder *Recorder) WriteHeader(header *Header) (err error) {
+	var p []byte
+
+	if p, err = json.Marshal(header); err != nil {
+		return
+	}
+
+	if _, err := recorder.file.Write(p); err != nil {
+		return err
+	}
+	if _, err := recorder.file.Write([]byte("\n")); err != nil {
+		return err
+	}
+
+	recorder.timestamp = header.Timestamp
+
+	return
+}
+
+func (recorder *Recorder) WriteData(data string) (err error) {
+	now := int(time.Now().UnixNano())
+
+	delta := float64(now-recorder.timestamp*1000*1000*1000) / 1000 / 1000 / 1000
+
+	row := make([]interface{}, 0)
+	row = append(row, delta)
+	row = append(row, "o")
+	row = append(row, data)
+
+	var s []byte
+	if s, err = json.Marshal(row); err != nil {
+		return
+	}
+	if _, err := recorder.file.Write(s); err != nil {
+		return err
+	}
+	if _, err := recorder.file.Write([]byte("\n")); err != nil {
+		return err
+	}
+	return
+}

pkg/api/ssh.go

@@ -142,6 +142,24 @@ func SSHEndpoint(c echo.Context) error {
 	}
 	_ = WriteMessage(ws, msg)
 
+	recorder, err := NewRecorder("./" + assetId + ".cast")
+	if err != nil {
+		return err
+	}
+
+	header := &Header{
+		Title:     "test",
+		Version:   2,
+		Height:    height,
+		Width:     width,
+		Env:       Env{Shell: "/bin/bash", Term: "xterm-256color"},
+		Timestamp: int(time.Now().Unix()),
+	}
+
+	if err := recorder.WriteHeader(header); err != nil {
+		return err
+	}
+
 	var mut sync.Mutex
 	var active = true
 
@@ -150,6 +168,7 @@ func SSHEndpoint(c echo.Context) error {
 			mut.Lock()
 			if !active {
 				logrus.Debugf("会话: %v -> %v 关闭", sshClient.LocalAddr().String(), sshClient.RemoteAddr().String())
+				recorder.Close()
 				break
 			}
 			mut.Unlock()
@@ -159,10 +178,14 @@ func SSHEndpoint(c echo.Context) error {
 				continue
 			}
 			if n > 0 {
+				s := string(p)
 				msg := Message{
 					Type:    Data,
-					Content: string(p),
+					Content: s,
 				}
+				// 录屏
+				_ = recorder.WriteData(s)
+
 				message, err := json.Marshal(msg)
 				if err != nil {
 					logrus.Warnf("生成Json失败 %v", err)