mirror of
https://github.com/wavetermdev/waveterm
synced 2026-05-06 15:08:34 +00:00
01b5d71709
lots of changes. new wshrpc implementation. unify websocket, web, blockcontroller, domain sockets, and terminal inputs to all use the new rpc system. lots of moving files around to deal with circular dependencies use new wshrpc as a client in wsh cmd
98 lines
2.5 KiB
Go
98 lines
2.5 KiB
Go
// Copyright 2024, Command Line Inc.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package webcmd
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"github.com/wavetermdev/thenextwave/pkg/shellexec"
|
|
"github.com/wavetermdev/thenextwave/pkg/tsgen/tsgenmeta"
|
|
"github.com/wavetermdev/thenextwave/pkg/util/utilfn"
|
|
"github.com/wavetermdev/thenextwave/pkg/wshutil"
|
|
)
|
|
|
|
const (
|
|
WSCommand_SetBlockTermSize = "setblocktermsize"
|
|
WSCommand_BlockInput = "blockinput"
|
|
WSCommand_Rpc = "rpc"
|
|
)
|
|
|
|
type WSCommandType interface {
|
|
GetWSCommand() string
|
|
}
|
|
|
|
func WSCommandTypeUnionMeta() tsgenmeta.TypeUnionMeta {
|
|
return tsgenmeta.TypeUnionMeta{
|
|
BaseType: reflect.TypeOf((*WSCommandType)(nil)).Elem(),
|
|
TypeFieldName: "wscommand",
|
|
Types: []reflect.Type{
|
|
reflect.TypeOf(SetBlockTermSizeWSCommand{}),
|
|
reflect.TypeOf(BlockInputWSCommand{}),
|
|
reflect.TypeOf(WSRpcCommand{}),
|
|
},
|
|
}
|
|
}
|
|
|
|
type WSRpcCommand struct {
|
|
WSCommand string `json:"wscommand" tstype:"\"rpc\""`
|
|
Message *wshutil.RpcMessage `json:"message"`
|
|
}
|
|
|
|
func (cmd *WSRpcCommand) GetWSCommand() string {
|
|
return cmd.WSCommand
|
|
}
|
|
|
|
type SetBlockTermSizeWSCommand struct {
|
|
WSCommand string `json:"wscommand" tstype:"\"setblocktermsize\""`
|
|
BlockId string `json:"blockid"`
|
|
TermSize shellexec.TermSize `json:"termsize"`
|
|
}
|
|
|
|
func (cmd *SetBlockTermSizeWSCommand) GetWSCommand() string {
|
|
return cmd.WSCommand
|
|
}
|
|
|
|
type BlockInputWSCommand struct {
|
|
WSCommand string `json:"wscommand" tstype:"\"blockinput\""`
|
|
BlockId string `json:"blockid"`
|
|
InputData64 string `json:"inputdata64"`
|
|
}
|
|
|
|
func (cmd *BlockInputWSCommand) GetWSCommand() string {
|
|
return cmd.WSCommand
|
|
}
|
|
|
|
func ParseWSCommandMap(cmdMap map[string]any) (WSCommandType, error) {
|
|
cmdType, ok := cmdMap["wscommand"].(string)
|
|
if !ok {
|
|
return nil, fmt.Errorf("no wscommand field in command map")
|
|
}
|
|
switch cmdType {
|
|
case WSCommand_SetBlockTermSize:
|
|
var cmd SetBlockTermSizeWSCommand
|
|
err := utilfn.DoMapStucture(&cmd, cmdMap)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error decoding SetBlockTermSizeWSCommand: %w", err)
|
|
}
|
|
return &cmd, nil
|
|
case WSCommand_BlockInput:
|
|
var cmd BlockInputWSCommand
|
|
err := utilfn.DoMapStucture(&cmd, cmdMap)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error decoding BlockInputWSCommand: %w", err)
|
|
}
|
|
return &cmd, nil
|
|
case WSCommand_Rpc:
|
|
var cmd WSRpcCommand
|
|
err := utilfn.DoMapStucture(&cmd, cmdMap)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error decoding WSRpcCommand: %w", err)
|
|
}
|
|
return &cmd, nil
|
|
default:
|
|
return nil, fmt.Errorf("unknown wscommand type %q", cmdType)
|
|
}
|
|
|
|
}
|