mirror of
https://github.com/wavetermdev/waveterm.git
synced 2025-01-08 19:38:51 +01:00
6a1b2c8bd4
* working on an in-memory store for screen information * nostrpos sentinel * textareainput now tracks selection (to update backend) * make websocket connections much safer. add a defer/panic handler for each ws message handled on backend. don't allow client to reconnect to backend ws handler more than once per second (handles issue with lots of fast fails) * use onSelect to have frontend textarea sync state to backend ScreenMem store * restore cmdline when switching screens * prettier
142 lines
4.2 KiB
Go
142 lines
4.2 KiB
Go
// Copyright 2023, Command Line Inc.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package scpacket
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"strings"
|
|
|
|
"github.com/alessio/shellescape"
|
|
"github.com/wavetermdev/waveterm/waveshell/pkg/base"
|
|
"github.com/wavetermdev/waveterm/waveshell/pkg/packet"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/sstore"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/utilfn"
|
|
)
|
|
|
|
const FeCommandPacketStr = "fecmd"
|
|
const WatchScreenPacketStr = "watchscreen"
|
|
const FeInputPacketStr = "feinput"
|
|
const RemoteInputPacketStr = "remoteinput"
|
|
const CmdInputTextPacketStr = "cmdinputtext"
|
|
|
|
type FeCommandPacketType struct {
|
|
Type string `json:"type"`
|
|
MetaCmd string `json:"metacmd"`
|
|
MetaSubCmd string `json:"metasubcmd,omitempty"`
|
|
Args []string `json:"args,omitempty"`
|
|
Kwargs map[string]string `json:"kwargs,omitempty"`
|
|
RawStr string `json:"rawstr,omitempty"`
|
|
UIContext *UIContextType `json:"uicontext,omitempty"`
|
|
Interactive bool `json:"interactive"`
|
|
}
|
|
|
|
func (pk *FeCommandPacketType) GetRawStr() string {
|
|
if pk.RawStr != "" {
|
|
return pk.RawStr
|
|
}
|
|
cmd := "/" + pk.MetaCmd
|
|
if pk.MetaSubCmd != "" {
|
|
cmd = cmd + ":" + pk.MetaSubCmd
|
|
}
|
|
var args []string
|
|
for k, v := range pk.Kwargs {
|
|
argStr := fmt.Sprintf("%s=%s", shellescape.Quote(k), shellescape.Quote(v))
|
|
args = append(args, argStr)
|
|
}
|
|
for _, arg := range pk.Args {
|
|
args = append(args, shellescape.Quote(arg))
|
|
}
|
|
if len(args) == 0 {
|
|
return cmd
|
|
}
|
|
return cmd + " " + strings.Join(args, " ")
|
|
}
|
|
|
|
type UIContextType struct {
|
|
SessionId string `json:"sessionid"`
|
|
ScreenId string `json:"screenid"`
|
|
Remote *sstore.RemotePtrType `json:"remote,omitempty"`
|
|
WinSize *packet.WinSize `json:"winsize,omitempty"`
|
|
Build string `json:"build,omitempty"`
|
|
}
|
|
|
|
type FeInputPacketType struct {
|
|
Type string `json:"type"`
|
|
CK base.CommandKey `json:"ck"`
|
|
Remote sstore.RemotePtrType `json:"remote"`
|
|
InputData64 string `json:"inputdata64"`
|
|
SigName string `json:"signame,omitempty"`
|
|
WinSize *packet.WinSize `json:"winsize,omitempty"`
|
|
}
|
|
|
|
type RemoteInputPacketType struct {
|
|
Type string `json:"type"`
|
|
RemoteId string `json:"remoteid"`
|
|
InputData64 string `json:"inputdata64"`
|
|
}
|
|
|
|
type WatchScreenPacketType struct {
|
|
Type string `json:"type"`
|
|
SessionId string `json:"sessionid"`
|
|
ScreenId string `json:"screenid"`
|
|
Connect bool `json:"connect"`
|
|
AuthKey string `json:"authkey"`
|
|
}
|
|
|
|
type CmdInputTextPacketType struct {
|
|
Type string `json:"type"`
|
|
SeqNum int `json:"seqnum"`
|
|
ScreenId string `json:"screenid"`
|
|
Text utilfn.StrWithPos `json:"text"`
|
|
}
|
|
|
|
func init() {
|
|
packet.RegisterPacketType(FeCommandPacketStr, reflect.TypeOf(FeCommandPacketType{}))
|
|
packet.RegisterPacketType(WatchScreenPacketStr, reflect.TypeOf(WatchScreenPacketType{}))
|
|
packet.RegisterPacketType(FeInputPacketStr, reflect.TypeOf(FeInputPacketType{}))
|
|
packet.RegisterPacketType(RemoteInputPacketStr, reflect.TypeOf(RemoteInputPacketType{}))
|
|
packet.RegisterPacketType(CmdInputTextPacketStr, reflect.TypeOf(CmdInputTextPacketType{}))
|
|
}
|
|
|
|
func (*CmdInputTextPacketType) GetType() string {
|
|
return CmdInputTextPacketStr
|
|
}
|
|
|
|
func MakeCmdInputTextPacket(screenId string) *CmdInputTextPacketType {
|
|
return &CmdInputTextPacketType{Type: CmdInputTextPacketStr, ScreenId: screenId}
|
|
}
|
|
|
|
func (*FeCommandPacketType) GetType() string {
|
|
return FeCommandPacketStr
|
|
}
|
|
|
|
func MakeFeCommandPacket() *FeCommandPacketType {
|
|
return &FeCommandPacketType{Type: FeCommandPacketStr}
|
|
}
|
|
|
|
func (*FeInputPacketType) GetType() string {
|
|
return FeInputPacketStr
|
|
}
|
|
|
|
func MakeFeInputPacket() *FeInputPacketType {
|
|
return &FeInputPacketType{Type: FeInputPacketStr}
|
|
}
|
|
|
|
func (*WatchScreenPacketType) GetType() string {
|
|
return WatchScreenPacketStr
|
|
}
|
|
|
|
func MakeWatchScreenPacket() *WatchScreenPacketType {
|
|
return &WatchScreenPacketType{Type: WatchScreenPacketStr}
|
|
}
|
|
|
|
func MakeRemoteInputPacket() *RemoteInputPacketType {
|
|
return &RemoteInputPacketType{Type: RemoteInputPacketStr}
|
|
}
|
|
|
|
func (*RemoteInputPacketType) GetType() string {
|
|
return RemoteInputPacketStr
|
|
}
|