mirror of
https://github.com/wavetermdev/waveterm.git
synced 2024-12-31 18:18:02 +01:00
8acda3525b
* Break update code out of sstore * add license disclaimers * missed one * add another * fix regression in openai updates, remove unnecessary functions * another copyright * update casts * fix issue with variadic updates * remove logs * remove log * remove unnecessary log * save work * moved a bunch of stuff to scbus * make modelupdate an object * fix new screen not updating active screen * add comment * make updates into packet types * different cast * update comments, remove unused methods * add one more comment * add an IsEmpty() on model updates to prevent sending empty updates to client
5537 lines
181 KiB
Go
5537 lines
181 KiB
Go
// Copyright 2023, Command Line Inc.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package cmdrunner
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"log"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
"unicode"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/gorilla/websocket"
|
|
"github.com/kevinburke/ssh_config"
|
|
"github.com/wavetermdev/waveterm/waveshell/pkg/base"
|
|
"github.com/wavetermdev/waveterm/waveshell/pkg/packet"
|
|
"github.com/wavetermdev/waveterm/waveshell/pkg/server"
|
|
"github.com/wavetermdev/waveterm/waveshell/pkg/shellenv"
|
|
"github.com/wavetermdev/waveterm/waveshell/pkg/shellutil"
|
|
"github.com/wavetermdev/waveterm/waveshell/pkg/shexec"
|
|
"github.com/wavetermdev/waveterm/waveshell/pkg/utilfn"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/comp"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/dbutil"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/pcloud"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/releasechecker"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/remote"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/remote/openai"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/scbase"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/scbus"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/scpacket"
|
|
"github.com/wavetermdev/waveterm/wavesrv/pkg/sstore"
|
|
"golang.org/x/mod/semver"
|
|
)
|
|
|
|
const (
|
|
HistoryTypeScreen = "screen"
|
|
HistoryTypeSession = "session"
|
|
HistoryTypeGlobal = "global"
|
|
)
|
|
|
|
func init() {
|
|
comp.RegisterSimpleCompFn(comp.CGTypeMeta, simpleCompMeta)
|
|
comp.RegisterSimpleCompFn(comp.CGTypeCommandMeta, simpleCompCommandMeta)
|
|
}
|
|
|
|
const DefaultUserId = "user"
|
|
const MaxNameLen = 50
|
|
const MaxShareNameLen = 150
|
|
const MaxRendererLen = 50
|
|
const MaxRemoteAliasLen = 50
|
|
const PasswordUnchangedSentinel = "--unchanged--"
|
|
const DefaultPTERM = "MxM"
|
|
const MaxCommandLen = 4096
|
|
const MaxSignalLen = 12
|
|
const MaxSignalNum = 64
|
|
const MaxEvalDepth = 5
|
|
const MaxOpenAIAPITokenLen = 100
|
|
const MaxOpenAIModelLen = 100
|
|
const MaxSidebarSections = 5
|
|
|
|
const TermFontSizeMin = 8
|
|
const TermFontSizeMax = 24
|
|
|
|
const TsFormatStr = "2006-01-02 15:04:05"
|
|
|
|
const OpenAIPacketTimeout = 10 * time.Second
|
|
const OpenAIStreamTimeout = 5 * time.Minute
|
|
|
|
const OpenAICloudCompletionTelemetryOffErrorMsg = "In order to protect against abuse, you must have telemetry turned on in order to use Wave's free AI features. If you do not want to turn telemetry on, you can still use Wave's AI features by adding your own OpenAI key in Settings. Note that when you use your own key, requests are not proxied through Wave's servers and will be sent directly to the OpenAI API."
|
|
|
|
const (
|
|
KwArgRenderer = "renderer"
|
|
KwArgView = "view"
|
|
KwArgState = "state"
|
|
KwArgTemplate = "template"
|
|
KwArgLang = "lang"
|
|
)
|
|
|
|
var ColorNames = []string{"yellow", "blue", "pink", "mint", "cyan", "violet", "orange", "green", "red", "white"}
|
|
var TabIcons = []string{"square", "sparkle", "fire", "ghost", "cloud", "compass", "crown", "droplet", "graduation-cap", "heart", "file"}
|
|
var RemoteColorNames = []string{"red", "green", "yellow", "blue", "magenta", "cyan", "white", "orange"}
|
|
var RemoteSetArgs = []string{"alias", "connectmode", "key", "password", "autoinstall", "color"}
|
|
var ConfirmFlags = []string{"hideshellprompt"}
|
|
var SidebarNames = []string{"main"}
|
|
|
|
var ScreenCmds = []string{"run", "comment", "cd", "cr", "clear", "sw", "reset", "signal", "chat"}
|
|
var NoHistCmds = []string{"_compgen", "line", "history", "_killserver"}
|
|
var GlobalCmds = []string{"session", "screen", "remote", "set", "client", "telemetry", "bookmark", "bookmarks"}
|
|
|
|
var SetVarNameMap map[string]string = map[string]string{
|
|
"tabcolor": "screen.tabcolor",
|
|
"tabicon": "screen.tabicon",
|
|
"pterm": "screen.pterm",
|
|
"anchor": "screen.anchor",
|
|
"focus": "screen.focus",
|
|
"line": "screen.line",
|
|
"index": "screen.index",
|
|
}
|
|
|
|
var SetVarScopes = []SetVarScope{
|
|
{ScopeName: "global", VarNames: []string{}},
|
|
{ScopeName: "client", VarNames: []string{"telemetry"}},
|
|
{ScopeName: "session", VarNames: []string{"name", "pos"}},
|
|
{ScopeName: "screen", VarNames: []string{"name", "tabcolor", "tabicon", "pos", "pterm", "anchor", "focus", "line", "index"}},
|
|
{ScopeName: "line", VarNames: []string{}},
|
|
// connection = remote, remote = remoteinstance
|
|
{ScopeName: "connection", VarNames: []string{"alias", "connectmode", "key", "password", "autoinstall", "color"}},
|
|
{ScopeName: "remote", VarNames: []string{}},
|
|
}
|
|
|
|
var userHostRe = regexp.MustCompile(`^(sudo@)?([a-z][a-z0-9._@\\-]*)@([a-z0-9][a-z0-9.-]*)(?::([0-9]+))?$`)
|
|
var remoteAliasRe = regexp.MustCompile("^[a-zA-Z0-9][a-zA-Z0-9._-]*$")
|
|
var genericNameRe = regexp.MustCompile("^[a-zA-Z][a-zA-Z0-9_ .()<>,/\"'\\[\\]{}=+$@!*-]*$")
|
|
var rendererRe = regexp.MustCompile("^[a-zA-Z][a-zA-Z0-9_.:-]*$")
|
|
var positionRe = regexp.MustCompile("^((S?\\+|E?-)?[0-9]+|(\\+|-|S|E))$")
|
|
var wsRe = regexp.MustCompile("\\s+")
|
|
var sigNameRe = regexp.MustCompile("^((SIG[A-Z0-9]+)|(\\d+))$")
|
|
|
|
type contextType string
|
|
|
|
var historyContextKey = contextType("history")
|
|
var depthContextKey = contextType("depth")
|
|
|
|
type SetVarScope struct {
|
|
ScopeName string
|
|
VarNames []string
|
|
}
|
|
|
|
type historyContextType struct {
|
|
LineId string
|
|
LineNum int64
|
|
RemotePtr *sstore.RemotePtrType
|
|
FeState sstore.FeStateType
|
|
InitialStatus string
|
|
}
|
|
|
|
type MetaCmdFnType = func(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error)
|
|
type MetaCmdEntryType struct {
|
|
IsAlias bool
|
|
Fn MetaCmdFnType
|
|
}
|
|
|
|
var MetaCmdFnMap = make(map[string]MetaCmdEntryType)
|
|
|
|
func init() {
|
|
registerCmdFn("run", RunCommand)
|
|
registerCmdFn("eval", EvalCommand)
|
|
registerCmdFn("comment", CommentCommand)
|
|
registerCmdFn("cr", CrCommand)
|
|
registerCmdFn("connect", CrCommand)
|
|
registerCmdFn("_compgen", CompGenCommand)
|
|
registerCmdFn("clear", ClearCommand)
|
|
registerCmdFn("reset", RemoteResetCommand)
|
|
registerCmdFn("signal", SignalCommand)
|
|
registerCmdFn("sync", SyncCommand)
|
|
|
|
registerCmdFn("session", SessionCommand)
|
|
registerCmdFn("session:open", SessionOpenCommand)
|
|
registerCmdAlias("session:new", SessionOpenCommand)
|
|
registerCmdFn("session:set", SessionSetCommand)
|
|
registerCmdFn("session:delete", SessionDeleteCommand)
|
|
registerCmdFn("session:archive", SessionArchiveCommand)
|
|
registerCmdFn("session:showall", SessionShowAllCommand)
|
|
registerCmdFn("session:show", SessionShowCommand)
|
|
registerCmdFn("session:openshared", SessionOpenSharedCommand)
|
|
|
|
registerCmdFn("screen", ScreenCommand)
|
|
registerCmdFn("screen:archive", ScreenArchiveCommand)
|
|
registerCmdFn("screen:delete", ScreenDeleteCommand)
|
|
registerCmdFn("screen:open", ScreenOpenCommand)
|
|
registerCmdAlias("screen:new", ScreenOpenCommand)
|
|
registerCmdFn("screen:set", ScreenSetCommand)
|
|
registerCmdFn("screen:showall", ScreenShowAllCommand)
|
|
registerCmdFn("screen:reset", ScreenResetCommand)
|
|
registerCmdFn("screen:webshare", ScreenWebShareCommand)
|
|
registerCmdFn("screen:reorder", ScreenReorderCommand)
|
|
|
|
registerCmdAlias("remote", RemoteCommand)
|
|
registerCmdFn("remote:show", RemoteShowCommand)
|
|
registerCmdFn("remote:showall", RemoteShowAllCommand)
|
|
registerCmdFn("remote:new", RemoteNewCommand)
|
|
registerCmdFn("remote:archive", RemoteArchiveCommand)
|
|
registerCmdFn("remote:set", RemoteSetCommand)
|
|
registerCmdFn("remote:disconnect", RemoteDisconnectCommand)
|
|
registerCmdFn("remote:connect", RemoteConnectCommand)
|
|
registerCmdFn("remote:install", RemoteInstallCommand)
|
|
registerCmdFn("remote:installcancel", RemoteInstallCancelCommand)
|
|
registerCmdFn("remote:reset", RemoteResetCommand)
|
|
registerCmdFn("remote:parse", RemoteConfigParseCommand)
|
|
|
|
registerCmdFn("copyfile", CopyFileCommand)
|
|
|
|
registerCmdFn("screen:resize", ScreenResizeCommand)
|
|
|
|
registerCmdFn("line", LineCommand)
|
|
registerCmdFn("line:show", LineShowCommand)
|
|
registerCmdFn("line:star", LineStarCommand)
|
|
registerCmdFn("line:bookmark", LineBookmarkCommand)
|
|
registerCmdFn("line:pin", LinePinCommand)
|
|
registerCmdFn("line:archive", LineArchiveCommand)
|
|
registerCmdFn("line:delete", LineDeleteCommand)
|
|
registerCmdFn("line:setheight", LineSetHeightCommand)
|
|
registerCmdFn("line:view", LineViewCommand)
|
|
registerCmdFn("line:set", LineSetCommand)
|
|
registerCmdFn("line:restart", LineRestartCommand)
|
|
|
|
registerCmdFn("client", ClientCommand)
|
|
registerCmdFn("client:show", ClientShowCommand)
|
|
registerCmdFn("client:set", ClientSetCommand)
|
|
registerCmdFn("client:notifyupdatewriter", ClientNotifyUpdateWriterCommand)
|
|
registerCmdFn("client:accepttos", ClientAcceptTosCommand)
|
|
registerCmdFn("client:setconfirmflag", ClientConfirmFlagCommand)
|
|
registerCmdFn("client:setsidebar", ClientSetSidebarCommand)
|
|
registerCmdFn("client:setglobalshortcut", ClientSetGlobalShortcut)
|
|
|
|
registerCmdFn("sidebar:open", SidebarOpenCommand)
|
|
registerCmdFn("sidebar:close", SidebarCloseCommand)
|
|
registerCmdFn("sidebar:add", SidebarAddCommand)
|
|
registerCmdFn("sidebar:remove", SidebarRemoveCommand)
|
|
|
|
registerCmdFn("telemetry", TelemetryCommand)
|
|
registerCmdFn("telemetry:on", TelemetryOnCommand)
|
|
registerCmdFn("telemetry:off", TelemetryOffCommand)
|
|
registerCmdFn("telemetry:send", TelemetrySendCommand)
|
|
registerCmdFn("telemetry:show", TelemetryShowCommand)
|
|
|
|
registerCmdFn("releasecheck", ReleaseCheckCommand)
|
|
registerCmdFn("releasecheck:autoon", ReleaseCheckOnCommand)
|
|
registerCmdFn("releasecheck:autooff", ReleaseCheckOffCommand)
|
|
|
|
registerCmdFn("history", HistoryCommand)
|
|
registerCmdFn("history:viewall", HistoryViewAllCommand)
|
|
registerCmdFn("history:purge", HistoryPurgeCommand)
|
|
|
|
registerCmdFn("bookmarks:show", BookmarksShowCommand)
|
|
|
|
registerCmdFn("bookmark:set", BookmarkSetCommand)
|
|
registerCmdFn("bookmark:delete", BookmarkDeleteCommand)
|
|
|
|
registerCmdFn("chat", OpenAICommand)
|
|
|
|
registerCmdFn("_killserver", KillServerCommand)
|
|
registerCmdFn("_dumpstate", DumpStateCommand)
|
|
|
|
registerCmdFn("set", SetCommand)
|
|
|
|
registerCmdFn("view:stat", ViewStatCommand)
|
|
registerCmdFn("view:test", ViewTestCommand)
|
|
|
|
registerCmdFn("edit:test", EditTestCommand)
|
|
|
|
// CodeEditCommand is overloaded to do codeedit and codeview
|
|
registerCmdFn("codeedit", CodeEditCommand)
|
|
registerCmdFn("codeview", CodeEditCommand)
|
|
|
|
registerCmdFn("imageview", ImageViewCommand)
|
|
registerCmdFn("mdview", MarkdownViewCommand)
|
|
registerCmdFn("markdownview", MarkdownViewCommand)
|
|
|
|
registerCmdFn("csvview", CSVViewCommand)
|
|
}
|
|
|
|
func getValidCommands() []string {
|
|
var rtn []string
|
|
for key, val := range MetaCmdFnMap {
|
|
if val.IsAlias {
|
|
continue
|
|
}
|
|
rtn = append(rtn, "/"+key)
|
|
}
|
|
return rtn
|
|
}
|
|
|
|
func registerCmdFn(cmdName string, fn MetaCmdFnType) {
|
|
MetaCmdFnMap[cmdName] = MetaCmdEntryType{Fn: fn}
|
|
}
|
|
|
|
func registerCmdAlias(cmdName string, fn MetaCmdFnType) {
|
|
MetaCmdFnMap[cmdName] = MetaCmdEntryType{IsAlias: true, Fn: fn}
|
|
}
|
|
|
|
func GetCmdStr(pk *scpacket.FeCommandPacketType) string {
|
|
if pk.MetaSubCmd == "" {
|
|
return pk.MetaCmd
|
|
}
|
|
return pk.MetaCmd + ":" + pk.MetaSubCmd
|
|
}
|
|
|
|
func HandleCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
metaCmd := SubMetaCmd(pk.MetaCmd)
|
|
var cmdName string
|
|
if pk.MetaSubCmd == "" {
|
|
cmdName = metaCmd
|
|
} else {
|
|
cmdName = fmt.Sprintf("%s:%s", pk.MetaCmd, pk.MetaSubCmd)
|
|
}
|
|
entry := MetaCmdFnMap[cmdName]
|
|
if entry.Fn == nil {
|
|
if MetaCmdFnMap[metaCmd].Fn != nil {
|
|
return nil, fmt.Errorf("invalid /%s subcommand '%s'", metaCmd, pk.MetaSubCmd)
|
|
}
|
|
return nil, fmt.Errorf("invalid command '/%s', no handler", cmdName)
|
|
}
|
|
return entry.Fn(ctx, pk)
|
|
}
|
|
|
|
func firstArg(pk *scpacket.FeCommandPacketType) string {
|
|
if len(pk.Args) == 0 {
|
|
return ""
|
|
}
|
|
return pk.Args[0]
|
|
}
|
|
|
|
func argN(pk *scpacket.FeCommandPacketType, n int) string {
|
|
if len(pk.Args) <= n {
|
|
return ""
|
|
}
|
|
return pk.Args[n]
|
|
}
|
|
|
|
// will trim strings for whitespace
|
|
func resolveCommaSepListToMap(arg string) map[string]bool {
|
|
if arg == "" {
|
|
return nil
|
|
}
|
|
rtn := make(map[string]bool)
|
|
fields := strings.Split(arg, ",")
|
|
for _, field := range fields {
|
|
field = strings.TrimSpace(field)
|
|
rtn[field] = true
|
|
}
|
|
return rtn
|
|
}
|
|
|
|
func resolveBool(arg string, def bool) bool {
|
|
if arg == "" {
|
|
return def
|
|
}
|
|
if arg == "0" || arg == "false" {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func defaultStr(arg string, def string) string {
|
|
if arg == "" {
|
|
return def
|
|
}
|
|
return arg
|
|
}
|
|
|
|
func resolveFile(arg string) (string, error) {
|
|
if arg == "" {
|
|
return "", nil
|
|
}
|
|
fileName := base.ExpandHomeDir(arg)
|
|
if !strings.HasPrefix(fileName, "/") {
|
|
return "", fmt.Errorf("must be absolute, cannot be a relative path")
|
|
}
|
|
fd, err := os.Open(fileName)
|
|
if fd != nil {
|
|
fd.Close()
|
|
}
|
|
if err != nil {
|
|
return "", fmt.Errorf("cannot open file: %v", err)
|
|
}
|
|
return fileName, nil
|
|
}
|
|
|
|
func resolvePosInt(arg string, def int) (int, error) {
|
|
if arg == "" {
|
|
return def, nil
|
|
}
|
|
ival, err := strconv.Atoi(arg)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
if ival <= 0 {
|
|
return 0, fmt.Errorf("must be greater than 0")
|
|
}
|
|
return ival, nil
|
|
}
|
|
|
|
func isAllDigits(arg string) bool {
|
|
if len(arg) == 0 {
|
|
return false
|
|
}
|
|
for i := 0; i < len(arg); i++ {
|
|
if arg[i] >= '0' && arg[i] <= '9' {
|
|
continue
|
|
}
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func resolveNonNegInt(arg string, def int) (int, error) {
|
|
if arg == "" {
|
|
return def, nil
|
|
}
|
|
ival, err := strconv.Atoi(arg)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
if ival < 0 {
|
|
return 0, fmt.Errorf("cannot be negative")
|
|
}
|
|
return ival, nil
|
|
}
|
|
|
|
var histExpansionRe = regexp.MustCompile(`^!(\d+)$`)
|
|
|
|
func doCmdHistoryExpansion(ctx context.Context, ids resolvedIds, cmdStr string) (string, error) {
|
|
if !strings.HasPrefix(cmdStr, "!") {
|
|
return "", nil
|
|
}
|
|
if strings.HasPrefix(cmdStr, "! ") {
|
|
return "", nil
|
|
}
|
|
if cmdStr == "!!" {
|
|
return doHistoryExpansion(ctx, ids, -1)
|
|
}
|
|
if strings.HasPrefix(cmdStr, "!-") {
|
|
return "", fmt.Errorf("wave does not support negative history offsets, use a stable positive history offset instead: '![linenum]'")
|
|
}
|
|
m := histExpansionRe.FindStringSubmatch(cmdStr)
|
|
if m == nil {
|
|
return "", fmt.Errorf("unsupported history substitution, can use '!!' or '![linenum]'")
|
|
}
|
|
ival, err := strconv.Atoi(m[1])
|
|
if err != nil {
|
|
return "", fmt.Errorf("invalid history expansion")
|
|
}
|
|
return doHistoryExpansion(ctx, ids, ival)
|
|
}
|
|
|
|
func doHistoryExpansion(ctx context.Context, ids resolvedIds, hnum int) (string, error) {
|
|
if hnum == 0 {
|
|
return "", fmt.Errorf("invalid history expansion, cannot expand line number '0'")
|
|
}
|
|
if hnum < -1 {
|
|
return "", fmt.Errorf("invalid history expansion, cannot expand negative history offsets")
|
|
}
|
|
foundHistoryNum := hnum
|
|
if hnum == -1 {
|
|
var err error
|
|
foundHistoryNum, err = sstore.GetLastHistoryLineNum(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return "", fmt.Errorf("cannot expand history, error finding last history item: %v", err)
|
|
}
|
|
if foundHistoryNum == 0 {
|
|
return "", fmt.Errorf("cannot expand history, no last history item")
|
|
}
|
|
}
|
|
hitem, err := sstore.GetHistoryItemByLineNum(ctx, ids.ScreenId, foundHistoryNum)
|
|
if err != nil {
|
|
return "", fmt.Errorf("cannot get history item '%d': %v", foundHistoryNum, err)
|
|
}
|
|
if hitem == nil {
|
|
return "", fmt.Errorf("cannot expand history, history item '%d' not found", foundHistoryNum)
|
|
}
|
|
return hitem.CmdStr, nil
|
|
}
|
|
|
|
func getEvalDepth(ctx context.Context) int {
|
|
depthVal := ctx.Value(depthContextKey)
|
|
if depthVal == nil {
|
|
return 0
|
|
}
|
|
return depthVal.(int)
|
|
}
|
|
|
|
func SyncCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/run error: %w", err)
|
|
}
|
|
runPacket := packet.MakeRunPacket()
|
|
runPacket.ReqId = uuid.New().String()
|
|
runPacket.CK = base.MakeCommandKey(ids.ScreenId, scbase.GenWaveUUID())
|
|
runPacket.UsePty = true
|
|
ptermVal := defaultStr(pk.Kwargs["wterm"], DefaultPTERM)
|
|
runPacket.TermOpts, err = GetUITermOpts(pk.UIContext.WinSize, ptermVal)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/sync error, invalid 'wterm' value %q: %v", ptermVal, err)
|
|
}
|
|
runPacket.Command = ":"
|
|
runPacket.ReturnState = true
|
|
rcOpts := remote.RunCommandOpts{
|
|
SessionId: ids.SessionId,
|
|
ScreenId: ids.ScreenId,
|
|
RemotePtr: ids.Remote.RemotePtr,
|
|
}
|
|
cmd, callback, err := remote.RunCommand(ctx, rcOpts, runPacket)
|
|
if callback != nil {
|
|
defer callback()
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cmd.RawCmdStr = pk.GetRawStr()
|
|
update, err := addLineForCmd(ctx, "/sync", true, ids, cmd, "terminal", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(sstore.InteractiveUpdate(pk.Interactive))
|
|
scbus.MainUpdateBus.DoScreenUpdate(ids.ScreenId, update)
|
|
return nil, nil
|
|
}
|
|
|
|
func getRendererArg(pk *scpacket.FeCommandPacketType) (string, error) {
|
|
rval := pk.Kwargs[KwArgView]
|
|
if rval == "" {
|
|
rval = pk.Kwargs[KwArgRenderer]
|
|
}
|
|
if rval == "" {
|
|
return "", nil
|
|
}
|
|
err := validateRenderer(rval)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return rval, nil
|
|
}
|
|
|
|
func getTemplateArg(pk *scpacket.FeCommandPacketType) (string, error) {
|
|
rval := pk.Kwargs[KwArgTemplate]
|
|
if rval == "" {
|
|
return "", nil
|
|
}
|
|
// TODO validate
|
|
return rval, nil
|
|
}
|
|
|
|
func getLangArg(pk *scpacket.FeCommandPacketType) (string, error) {
|
|
// TODO better error checking
|
|
if len(pk.Kwargs[KwArgLang]) > 50 {
|
|
return "", nil // TODO return error, don't fail silently
|
|
}
|
|
return pk.Kwargs[KwArgLang], nil
|
|
}
|
|
|
|
func RunCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/run error: %w", err)
|
|
}
|
|
renderer, err := getRendererArg(pk)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/run error, invalid view/renderer: %w", err)
|
|
}
|
|
templateArg, err := getTemplateArg(pk)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/run error, invalid template: %w", err)
|
|
}
|
|
langArg, err := getLangArg(pk)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/run error, invalid lang: %w", err)
|
|
}
|
|
cmdStr := firstArg(pk)
|
|
expandedCmdStr, err := doCmdHistoryExpansion(ctx, ids, cmdStr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if expandedCmdStr != "" {
|
|
newPk := scpacket.MakeFeCommandPacket()
|
|
newPk.MetaCmd = "eval"
|
|
newPk.Args = []string{expandedCmdStr}
|
|
newPk.Kwargs = pk.Kwargs
|
|
newPk.RawStr = pk.RawStr
|
|
newPk.UIContext = pk.UIContext
|
|
newPk.Interactive = pk.Interactive
|
|
evalDepth := getEvalDepth(ctx)
|
|
ctxWithDepth := context.WithValue(ctx, depthContextKey, evalDepth+1)
|
|
return EvalCommand(ctxWithDepth, newPk)
|
|
}
|
|
isRtnStateCmd := IsReturnStateCommand(cmdStr)
|
|
// runPacket.State is set in remote.RunCommand()
|
|
runPacket := packet.MakeRunPacket()
|
|
runPacket.ReqId = uuid.New().String()
|
|
runPacket.CK = base.MakeCommandKey(ids.ScreenId, scbase.GenWaveUUID())
|
|
runPacket.UsePty = true
|
|
ptermVal := defaultStr(pk.Kwargs["wterm"], DefaultPTERM)
|
|
runPacket.TermOpts, err = GetUITermOpts(pk.UIContext.WinSize, ptermVal)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/run error, invalid 'pterm' value %q: %v", ptermVal, err)
|
|
}
|
|
runPacket.Command = strings.TrimSpace(cmdStr)
|
|
runPacket.ReturnState = resolveBool(pk.Kwargs["rtnstate"], isRtnStateCmd)
|
|
rcOpts := remote.RunCommandOpts{
|
|
SessionId: ids.SessionId,
|
|
ScreenId: ids.ScreenId,
|
|
RemotePtr: ids.Remote.RemotePtr,
|
|
}
|
|
cmd, callback, err := remote.RunCommand(ctx, rcOpts, runPacket)
|
|
if callback != nil {
|
|
defer callback()
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cmd.RawCmdStr = pk.GetRawStr()
|
|
lineState := make(map[string]any)
|
|
if templateArg != "" {
|
|
lineState[sstore.LineState_Template] = templateArg
|
|
}
|
|
if langArg != "" {
|
|
lineState[sstore.LineState_Lang] = langArg
|
|
}
|
|
update, err := addLineForCmd(ctx, "/run", true, ids, cmd, renderer, lineState)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(sstore.InteractiveUpdate(pk.Interactive))
|
|
// this update is sent asynchronously for timing issues. the cmd update comes async as well
|
|
// so if we return this directly it sometimes gets evaluated first. by pushing it on the MainBus
|
|
// it ensures it happens after the command creation event.
|
|
scbus.MainUpdateBus.DoScreenUpdate(ids.ScreenId, update)
|
|
return nil, nil
|
|
}
|
|
|
|
func implementRunInSidebar(ctx context.Context, screenId string, lineId string) (*sstore.ScreenType, error) {
|
|
screen, err := sidebarSetOpen(ctx, "run", screenId, true, "")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
screen.ScreenViewOpts.Sidebar.SidebarLineId = lineId
|
|
err = sstore.ScreenUpdateViewOpts(ctx, screenId, screen.ScreenViewOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/run error updating screenviewopts: %v", err)
|
|
}
|
|
return screen, nil
|
|
}
|
|
|
|
func addToHistory(ctx context.Context, pk *scpacket.FeCommandPacketType, historyContext historyContextType, isMetaCmd bool, hadError bool) error {
|
|
cmdStr := firstArg(pk)
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
hitem := &sstore.HistoryItemType{
|
|
HistoryId: scbase.GenWaveUUID(),
|
|
Ts: time.Now().UnixMilli(),
|
|
UserId: DefaultUserId,
|
|
SessionId: ids.SessionId,
|
|
ScreenId: ids.ScreenId,
|
|
LineId: historyContext.LineId,
|
|
LineNum: historyContext.LineNum,
|
|
HadError: hadError,
|
|
CmdStr: cmdStr,
|
|
IsMetaCmd: isMetaCmd,
|
|
FeState: historyContext.FeState,
|
|
Status: historyContext.InitialStatus,
|
|
}
|
|
if hitem.Status == "" {
|
|
if hadError {
|
|
hitem.Status = sstore.CmdStatusError
|
|
} else {
|
|
hitem.Status = "done"
|
|
}
|
|
}
|
|
if !isMetaCmd && historyContext.RemotePtr != nil {
|
|
hitem.Remote = *historyContext.RemotePtr
|
|
}
|
|
err = sstore.InsertHistoryItem(ctx, hitem)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func EvalCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("usage: /eval [command], no command passed to eval")
|
|
}
|
|
if len(pk.Args[0]) > MaxCommandLen {
|
|
return nil, fmt.Errorf("command length too long len:%d, max:%d", len(pk.Args[0]), MaxCommandLen)
|
|
}
|
|
evalDepth := getEvalDepth(ctx)
|
|
if pk.Interactive && evalDepth == 0 {
|
|
sstore.UpdateActivityWrap(ctx, sstore.ActivityUpdate{NumCommands: 1}, "numcommands")
|
|
}
|
|
if evalDepth > MaxEvalDepth {
|
|
return nil, fmt.Errorf("alias/history expansion max-depth exceeded")
|
|
}
|
|
var historyContext historyContextType
|
|
ctxWithHistory := context.WithValue(ctx, historyContextKey, &historyContext)
|
|
var update scbus.UpdatePacket
|
|
newPk, rtnErr := EvalMetaCommand(ctxWithHistory, pk)
|
|
if rtnErr == nil {
|
|
update, rtnErr = HandleCommand(ctxWithHistory, newPk)
|
|
} else {
|
|
return nil, fmt.Errorf("error in Eval Meta Command: %v", rtnErr)
|
|
}
|
|
if !resolveBool(pk.Kwargs["nohist"], false) {
|
|
// TODO should this be "pk" or "newPk" (2nd arg)
|
|
err := addToHistory(ctx, pk, historyContext, (newPk.MetaCmd != "run"), (rtnErr != nil))
|
|
if err != nil {
|
|
log.Printf("[error] adding to history: %v\n", err)
|
|
// fall through (non-fatal error)
|
|
}
|
|
}
|
|
var hasModelUpdate bool
|
|
var modelUpdate *scbus.ModelUpdatePacketType
|
|
if update == nil {
|
|
hasModelUpdate = true
|
|
modelUpdate = scbus.MakeUpdatePacket()
|
|
update = modelUpdate
|
|
} else if mu, ok := update.(*scbus.ModelUpdatePacketType); ok {
|
|
hasModelUpdate = true
|
|
modelUpdate = mu
|
|
}
|
|
if resolveBool(newPk.Kwargs["sidebar"], false) && historyContext.LineId != "" && hasModelUpdate {
|
|
ids, resolveErr := resolveUiIds(ctx, newPk, R_Session|R_Screen)
|
|
// we are ignoring resolveErr (if not nil). obviously can't add to sidebar and
|
|
// either another error already happened, or this command was never about the sidebar
|
|
if resolveErr == nil {
|
|
screen, sidebarErr := implementRunInSidebar(ctx, ids.ScreenId, historyContext.LineId)
|
|
if sidebarErr == nil {
|
|
sstore.AddScreenUpdate(modelUpdate, screen)
|
|
} else {
|
|
sstore.AddInfoMsgUpdateError(modelUpdate, fmt.Sprintf("cannot move command to sidebar: %v", sidebarErr))
|
|
}
|
|
}
|
|
}
|
|
return update, rtnErr
|
|
}
|
|
|
|
func ScreenArchiveCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session) // don't force R_Screen
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:archive cannot archive screen: %w", err)
|
|
}
|
|
screenId := ids.ScreenId
|
|
if len(pk.Args) > 0 {
|
|
ri, err := resolveSessionScreen(ctx, ids.SessionId, pk.Args[0], ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:archive cannot resolve screen arg: %v", err)
|
|
}
|
|
screenId = ri.Id
|
|
}
|
|
if screenId == "" {
|
|
return nil, fmt.Errorf("/screen:archive no active screen or screen arg passed")
|
|
}
|
|
archiveVal := true
|
|
if len(pk.Args) > 1 {
|
|
archiveVal = resolveBool(pk.Args[1], true)
|
|
}
|
|
var update scbus.UpdatePacket
|
|
if archiveVal {
|
|
update, err = sstore.ArchiveScreen(ctx, ids.SessionId, screenId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return update, nil
|
|
} else {
|
|
log.Printf("unarchive screen %s\n", screenId)
|
|
err = sstore.UnArchiveScreen(ctx, ids.SessionId, screenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:archive cannot un-archive screen: %v", err)
|
|
}
|
|
screen, err := sstore.GetScreenById(ctx, screenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:archive cannot get updated screen obj: %v", err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*screen)
|
|
return update, nil
|
|
}
|
|
}
|
|
|
|
func ScreenDeleteCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session) // don't force R_Screen
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:delete cannot delete screen: %w", err)
|
|
}
|
|
screenId := ids.ScreenId
|
|
if len(pk.Args) > 0 {
|
|
ri, err := resolveSessionScreen(ctx, ids.SessionId, pk.Args[0], ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:delete cannot resolve screen arg: %v", err)
|
|
}
|
|
screenId = ri.Id
|
|
}
|
|
if screenId == "" {
|
|
return nil, fmt.Errorf("/screen:delete no active screen or screen arg passed")
|
|
}
|
|
update, err := sstore.DeleteScreen(ctx, screenId, false, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return update, nil
|
|
}
|
|
|
|
func ScreenOpenCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:open cannot open screen: %w", err)
|
|
}
|
|
activate := resolveBool(pk.Kwargs["activate"], true)
|
|
newName := pk.Kwargs["name"]
|
|
if newName != "" {
|
|
err := validateName(newName, "screen")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
update, err := sstore.InsertScreen(ctx, ids.SessionId, newName, sstore.ScreenCreateOpts{}, activate)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return update, nil
|
|
}
|
|
|
|
func ScreenReorderCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
// Resolve the UI IDs for the session and screen
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Extract the screen ID and the new index from the packet
|
|
screenId := ids.ScreenId
|
|
newScreenIdxStr := pk.Kwargs["index"]
|
|
newScreenIdx, err := resolvePosInt(newScreenIdxStr, 1)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid new screen index: %v", err)
|
|
}
|
|
|
|
// Call SetScreenIdx to update the screen's index in the database
|
|
err = sstore.SetScreenIdx(ctx, ids.SessionId, screenId, newScreenIdx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating screen index: %v", err)
|
|
}
|
|
|
|
// Retrieve all session screens
|
|
screens, err := sstore.GetSessionScreens(ctx, ids.SessionId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error retrieving updated screen: %v", err)
|
|
}
|
|
|
|
// Prepare the update packet to send back to the client
|
|
update := scbus.MakeUpdatePacket()
|
|
for _, screen := range screens {
|
|
update.AddUpdate(*screen)
|
|
}
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoMsg: "screen indices updated successfully",
|
|
TimeoutMs: 2000,
|
|
})
|
|
|
|
return update, nil
|
|
}
|
|
|
|
var screenAnchorRe = regexp.MustCompile("^(\\d+)(?::(-?\\d+))?$")
|
|
|
|
func ScreenSetCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var varsUpdated []string
|
|
var setNonAnchor bool // anchor does not receive an update
|
|
updateMap := make(map[string]interface{})
|
|
if pk.Kwargs["name"] != "" {
|
|
newName := pk.Kwargs["name"]
|
|
err = validateName(newName, "screen")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
updateMap[sstore.ScreenField_Name] = newName
|
|
varsUpdated = append(varsUpdated, "name")
|
|
setNonAnchor = true
|
|
}
|
|
if pk.Kwargs["sharename"] != "" {
|
|
shareName := pk.Kwargs["sharename"]
|
|
err = validateShareName(shareName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
updateMap[sstore.ScreenField_ShareName] = shareName
|
|
varsUpdated = append(varsUpdated, "sharename")
|
|
setNonAnchor = true
|
|
}
|
|
if pk.Kwargs["tabcolor"] != "" {
|
|
color := pk.Kwargs["tabcolor"]
|
|
err = validateColor(color, "screen tabcolor")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
updateMap[sstore.ScreenField_TabColor] = color
|
|
varsUpdated = append(varsUpdated, "tabcolor")
|
|
setNonAnchor = true
|
|
}
|
|
if pk.Kwargs["tabicon"] != "" {
|
|
icon := pk.Kwargs["tabicon"]
|
|
updateMap[sstore.ScreenField_TabIcon] = icon
|
|
varsUpdated = append(varsUpdated, "tabicon")
|
|
setNonAnchor = true
|
|
}
|
|
if pk.Kwargs["pos"] != "" {
|
|
varsUpdated = append(varsUpdated, "pos")
|
|
setNonAnchor = true
|
|
}
|
|
if pk.Kwargs["focus"] != "" {
|
|
focusVal := pk.Kwargs["focus"]
|
|
if focusVal != sstore.ScreenFocusInput && focusVal != sstore.ScreenFocusCmd {
|
|
return nil, fmt.Errorf("/screen:set invalid focus argument %q, must be %s", focusVal, formatStrs([]string{sstore.ScreenFocusInput, sstore.ScreenFocusCmd}, "or", false))
|
|
}
|
|
varsUpdated = append(varsUpdated, "focus")
|
|
updateMap[sstore.ScreenField_Focus] = focusVal
|
|
setNonAnchor = true
|
|
}
|
|
if pk.Kwargs["line"] != "" {
|
|
screen, err := sstore.GetScreenById(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:set cannot get screen: %v", err)
|
|
}
|
|
var selectedLineStr string
|
|
if screen.SelectedLine > 0 {
|
|
selectedLineStr = strconv.Itoa(int(screen.SelectedLine))
|
|
}
|
|
ritem, err := resolveLine(ctx, screen.SessionId, screen.ScreenId, pk.Kwargs["line"], selectedLineStr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:set error resolving line: %v", err)
|
|
}
|
|
if ritem == nil {
|
|
return nil, fmt.Errorf("/screen:set could not resolve line %q", pk.Kwargs["line"])
|
|
}
|
|
varsUpdated = append(varsUpdated, "line")
|
|
setNonAnchor = true
|
|
updateMap[sstore.ScreenField_SelectedLine] = ritem.Num
|
|
}
|
|
if pk.Kwargs["anchor"] != "" {
|
|
m := screenAnchorRe.FindStringSubmatch(pk.Kwargs["anchor"])
|
|
if m == nil {
|
|
return nil, fmt.Errorf("/screen:set invalid anchor argument (must be [line] or [line]:[offset])")
|
|
}
|
|
anchorLine, _ := strconv.Atoi(m[1])
|
|
varsUpdated = append(varsUpdated, "anchor")
|
|
updateMap[sstore.ScreenField_AnchorLine] = anchorLine
|
|
if m[2] != "" {
|
|
anchorOffset, _ := strconv.Atoi(m[2])
|
|
updateMap[sstore.ScreenField_AnchorOffset] = anchorOffset
|
|
} else {
|
|
updateMap[sstore.ScreenField_AnchorOffset] = 0
|
|
}
|
|
}
|
|
if len(varsUpdated) == 0 {
|
|
return nil, fmt.Errorf("/screen:set no updates, can set %s", formatStrs([]string{"name", "pos", "tabcolor", "tabicon", "focus", "anchor", "line", "sharename"}, "or", false))
|
|
}
|
|
screen, err := sstore.UpdateScreen(ctx, ids.ScreenId, updateMap)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating screen: %v", err)
|
|
}
|
|
if !setNonAnchor {
|
|
return nil, nil
|
|
}
|
|
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*screen, sstore.InfoMsgType{
|
|
InfoMsg: fmt.Sprintf("screen updated %s", formatStrs(varsUpdated, "and", false)),
|
|
TimeoutMs: 2000,
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func ScreenCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen cannot switch to screen: %w", err)
|
|
}
|
|
firstArg := firstArg(pk)
|
|
if firstArg == "" {
|
|
return nil, fmt.Errorf("usage /screen [screen-name|screen-index|screen-id], no param specified")
|
|
}
|
|
ritem, err := resolveSessionScreen(ctx, ids.SessionId, firstArg, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
update, err := sstore.SwitchScreenById(ctx, ids.SessionId, ritem.Id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return update, nil
|
|
}
|
|
|
|
var sidebarWidthRe = regexp.MustCompile("^\\d+(px|%)$")
|
|
|
|
func sidebarSetOpen(ctx context.Context, cmdStr string, screenId string, open bool, width string) (*sstore.ScreenType, error) {
|
|
if width != "" && !sidebarWidthRe.MatchString(width) {
|
|
return nil, fmt.Errorf("/%s invalid width specified, must be either a px value or a percent (e.g. '300px' or '50%%')", cmdStr)
|
|
}
|
|
if strings.HasSuffix(width, "%") {
|
|
percentNum, _ := strconv.Atoi(width[:len(width)-1])
|
|
if percentNum < 10 || percentNum > 90 {
|
|
return nil, fmt.Errorf("/%s invalid width specified, percentage must be between 10%% and 90%%", cmdStr)
|
|
}
|
|
}
|
|
if strings.HasSuffix(width, "px") {
|
|
pxNum, _ := strconv.Atoi(width[:len(width)-2])
|
|
if pxNum < 200 {
|
|
return nil, fmt.Errorf("/%s invalid width specified, minimum sizebar width is 200px", cmdStr)
|
|
}
|
|
}
|
|
screen, err := sstore.GetScreenById(ctx, screenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/%s cannot get screen: %v", cmdStr, err)
|
|
}
|
|
if screen.ScreenViewOpts.Sidebar == nil {
|
|
screen.ScreenViewOpts.Sidebar = &sstore.ScreenSidebarOptsType{}
|
|
}
|
|
screen.ScreenViewOpts.Sidebar.Open = open
|
|
if width != "" {
|
|
screen.ScreenViewOpts.Sidebar.Width = width
|
|
}
|
|
err = sstore.ScreenUpdateViewOpts(ctx, screenId, screen.ScreenViewOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/%s error updating screenviewopts: %v", cmdStr, err)
|
|
}
|
|
return screen, nil
|
|
}
|
|
|
|
func SidebarOpenCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
screen, err := sidebarSetOpen(ctx, GetCmdStr(pk), ids.ScreenId, true, pk.Kwargs["width"])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*screen)
|
|
return update, nil
|
|
}
|
|
|
|
func SidebarCloseCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
screen, err := sidebarSetOpen(ctx, GetCmdStr(pk), ids.ScreenId, false, "")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*screen)
|
|
return update, nil
|
|
}
|
|
|
|
func SidebarAddCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var addLineId string
|
|
if lineArg, ok := pk.Kwargs["line"]; ok {
|
|
lineId, err := sstore.FindLineIdByArg(ctx, ids.ScreenId, lineArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error looking up lineid: %v", err)
|
|
}
|
|
addLineId = lineId
|
|
}
|
|
if addLineId == "" {
|
|
return nil, fmt.Errorf("/%s must specify line=[lineid] to add to the sidebar", GetCmdStr(pk))
|
|
}
|
|
screen, err := sidebarSetOpen(ctx, GetCmdStr(pk), ids.ScreenId, true, pk.Kwargs["width"])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
screen.ScreenViewOpts.Sidebar.SidebarLineId = addLineId
|
|
err = sstore.ScreenUpdateViewOpts(ctx, ids.ScreenId, screen.ScreenViewOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/%s error updating screenviewopts: %v", GetCmdStr(pk), err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*screen)
|
|
return update, nil
|
|
}
|
|
|
|
func SidebarRemoveCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
screen, err := sstore.GetScreenById(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/%s cannot get screeen: %v", GetCmdStr(pk), err)
|
|
}
|
|
sidebar := screen.ScreenViewOpts.Sidebar
|
|
if sidebar == nil {
|
|
return nil, nil
|
|
}
|
|
sidebar.SidebarLineId = ""
|
|
sidebar.Open = false
|
|
err = sstore.ScreenUpdateViewOpts(ctx, ids.ScreenId, screen.ScreenViewOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/%s error updating screenviewopts: %v", GetCmdStr(pk), err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*screen)
|
|
return update, nil
|
|
}
|
|
|
|
func createRemoteViewRemoteIdUpdate(remoteId string) scbus.UpdatePacket {
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.RemoteViewType{
|
|
PtyRemoteId: remoteId,
|
|
})
|
|
return update
|
|
}
|
|
|
|
func createRemoteViewRemoteEditUpdate(redit *sstore.RemoteEditType) scbus.UpdatePacket {
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.RemoteViewType{
|
|
RemoteEdit: redit,
|
|
})
|
|
return update
|
|
}
|
|
|
|
func prettyPrintByteSize(size int64) string {
|
|
gbSize := float64(size) / float64(1073741824)
|
|
if gbSize > 1 {
|
|
return fmt.Sprintf("%.2f Gigabytes", gbSize)
|
|
}
|
|
mbSize := float64(size) / float64(1048576)
|
|
if mbSize > 1 {
|
|
return fmt.Sprintf("%.2f Megabytes", mbSize)
|
|
}
|
|
kbSize := float64(size) / float64(1024)
|
|
if kbSize > 1 {
|
|
return fmt.Sprintf("%.2f Kilobytes", kbSize)
|
|
}
|
|
return fmt.Sprintf("%v Bytes", size)
|
|
}
|
|
|
|
// this can only be called in a defer func, because recover() only works inside of a defe
|
|
func deferWriteCmdStatus(ctx context.Context, cmd *sstore.CmdType, startTime time.Time, exitSuccess bool, outputPos int64) {
|
|
r := recover()
|
|
if r != nil {
|
|
panicMsg := fmt.Sprintf("panic: %v", r)
|
|
log.Printf("panic: %v\n", panicMsg)
|
|
writeStringToPty(ctx, cmd, panicMsg, &outputPos)
|
|
}
|
|
duration := time.Since(startTime)
|
|
cmdStatus := sstore.CmdStatusDone
|
|
var exitCode int
|
|
if !exitSuccess {
|
|
cmdStatus = sstore.CmdStatusError
|
|
exitCode = 1
|
|
}
|
|
ck := base.MakeCommandKey(cmd.ScreenId, cmd.LineId)
|
|
donePk := packet.MakeCmdDonePacket(ck)
|
|
donePk.Ts = time.Now().UnixMilli()
|
|
donePk.ExitCode = exitCode
|
|
donePk.DurationMs = duration.Milliseconds()
|
|
update, err := sstore.UpdateCmdDoneInfo(context.Background(), ck, donePk, cmdStatus)
|
|
if err != nil {
|
|
// nothing to do
|
|
log.Printf("error updating cmddoneinfo (in openai): %v\n", err)
|
|
return
|
|
}
|
|
scbus.MainUpdateBus.DoScreenUpdate(cmd.ScreenId, update)
|
|
}
|
|
|
|
func checkForWriteReady(ctx context.Context, iter *packet.RpcResponseIter) (string, error) {
|
|
readyIf, err := iter.Next(ctx)
|
|
if err != nil {
|
|
return "", fmt.Errorf("error getting write ready response: %v\r\n", err)
|
|
}
|
|
readyPk, ok := readyIf.(*packet.WriteFileReadyPacketType)
|
|
if !ok {
|
|
return "", fmt.Errorf("bad write ready packet received %v", readyIf)
|
|
}
|
|
if readyPk.Error != "" {
|
|
return "", fmt.Errorf("ready error: %v", readyPk.Error)
|
|
}
|
|
return readyPk.RespId, nil
|
|
}
|
|
|
|
func checkForWriteFinished(ctx context.Context, iter *packet.RpcResponseIter) error {
|
|
doneIf, err := iter.Next(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("error while getting done response: %v", err)
|
|
}
|
|
writeDonePk, ok := doneIf.(*packet.WriteFileDonePacketType)
|
|
if !ok {
|
|
return fmt.Errorf("bad done packet received: %T", doneIf)
|
|
}
|
|
if writeDonePk.Error != "" {
|
|
return fmt.Errorf("done error: %v", writeDonePk.Error)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func doCopyLocalFileToRemote(ctx context.Context, cmd *sstore.CmdType, remote_msh *remote.MShellProc, localPath string, destPath string, outputPos int64) {
|
|
var exitSuccess bool
|
|
startTime := time.Now()
|
|
defer func() {
|
|
deferWriteCmdStatus(ctx, cmd, startTime, exitSuccess, outputPos)
|
|
}()
|
|
localFile, err := os.Open(localPath)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Error, unable to open file %v: %v\r\n", localFile, localPath), &outputPos)
|
|
return
|
|
}
|
|
defer localFile.Close()
|
|
writePk := packet.MakeWriteFilePacket()
|
|
writePk.ReqId = uuid.New().String()
|
|
writePk.Path = destPath
|
|
iter, err := remote_msh.WriteFile(ctx, writePk)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Error starting file write: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
defer iter.Close()
|
|
_, err = checkForWriteReady(ctx, iter)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Write ready packet error: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
fileStat, err := localFile.Stat()
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("error: could not get file stat: %v", err), &outputPos)
|
|
return
|
|
}
|
|
fileSizeBytes := fileStat.Size()
|
|
bytesWritten := int64(0)
|
|
lastFileTransferPercentage := float64(0)
|
|
fileTransferPercentage := float64(0)
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Source File Size: %s\r\n", prettyPrintByteSize(fileSizeBytes)), &outputPos)
|
|
writeStringToPty(ctx, cmd, "[", &outputPos)
|
|
var buffer [server.MaxFileDataPacketSize]byte
|
|
bufSlice := buffer[:]
|
|
for {
|
|
dataPk := packet.MakeFileDataPacket(writePk.ReqId)
|
|
bytesRead, err := io.ReadFull(localFile, bufSlice)
|
|
if err == io.ErrUnexpectedEOF || err == io.EOF {
|
|
dataPk.Eof = true
|
|
} else if err != nil {
|
|
dataErr := fmt.Sprintf("error reading file data: %v", err)
|
|
dataPk.Error = dataErr
|
|
remote_msh.SendFileData(dataPk)
|
|
writeStringToPty(ctx, cmd, dataErr, &outputPos)
|
|
return
|
|
}
|
|
if bytesRead > 0 {
|
|
dataPk.Data = make([]byte, bytesRead)
|
|
copy(dataPk.Data, bufSlice[0:bytesRead])
|
|
bytesWritten += int64(len(dataPk.Data))
|
|
fileTransferPercentage = float64(bytesWritten) / float64(fileSizeBytes)
|
|
|
|
if fileTransferPercentage-lastFileTransferPercentage > float64(0.05) {
|
|
writeStringToPty(ctx, cmd, "-", &outputPos)
|
|
lastFileTransferPercentage = fileTransferPercentage
|
|
}
|
|
}
|
|
remote_msh.SendFileData(dataPk)
|
|
if dataPk.Eof {
|
|
break
|
|
}
|
|
}
|
|
err = checkForWriteFinished(ctx, iter)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Write finished packet error %v", err), &outputPos)
|
|
return
|
|
}
|
|
writeStringToPty(ctx, cmd, "] done. \r\n", &outputPos)
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Finished transferring. Transferred %v bytes\r\n", fileSizeBytes), &outputPos)
|
|
exitSuccess = true
|
|
}
|
|
|
|
func getStatusBarString(filePercentageInt int) string {
|
|
statusBarString := "\x1b[2k\r["
|
|
for count := 0; count < 20; count++ {
|
|
if (filePercentageInt - count*5) > 0 {
|
|
statusBarString += "-"
|
|
} else {
|
|
statusBarString += " "
|
|
}
|
|
}
|
|
if filePercentageInt < 100 {
|
|
statusBarString += fmt.Sprintf("] %v%%", filePercentageInt)
|
|
} else {
|
|
statusBarString += "]"
|
|
}
|
|
return statusBarString
|
|
}
|
|
|
|
func doCopyRemoteFileToRemote(ctx context.Context, cmd *sstore.CmdType, sourceMsh *remote.MShellProc, destMsh *remote.MShellProc, sourcePath string, destPath string, outputPos int64) {
|
|
var exitSuccess bool
|
|
startTime := time.Now()
|
|
defer func() {
|
|
deferWriteCmdStatus(ctx, cmd, startTime, exitSuccess, outputPos)
|
|
}()
|
|
streamPk := packet.MakeStreamFilePacket()
|
|
streamPk.ReqId = uuid.New().String()
|
|
streamPk.Path = sourcePath
|
|
sourceStreamIter, err := sourceMsh.StreamFile(ctx, streamPk)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Error getting file data packet: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
defer sourceStreamIter.Close()
|
|
respIf, err := sourceStreamIter.Next(ctx)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Error getting next packet: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
resp, ok := respIf.(*packet.StreamFileResponseType)
|
|
if !ok {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Error in getting packet response: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
if resp == nil || resp.Error != "" {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Response packet has error: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
fileSizeBytes := resp.Info.Size
|
|
if fileSizeBytes == 0 {
|
|
writeStringToPty(ctx, cmd, "Source file does not exist or is empty - exiting\r\n", &outputPos)
|
|
return
|
|
}
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Source File Size: %v\r\n", prettyPrintByteSize(fileSizeBytes)), &outputPos)
|
|
writePk := packet.MakeWriteFilePacket()
|
|
writePk.ReqId = uuid.New().String()
|
|
writePk.Path = destPath
|
|
destWriteIter, err := destMsh.WriteFile(ctx, writePk)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Error starting file write: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
defer destWriteIter.Close()
|
|
_, err = checkForWriteReady(ctx, destWriteIter)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Write ready packet error: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
bytesWritten := int64(0)
|
|
lastFilePercentageInt := int(0)
|
|
fileTransferPercentage := float64(0)
|
|
writeStringToPty(ctx, cmd, "[", &outputPos)
|
|
for {
|
|
dataPkIf, err := sourceStreamIter.Next(ctx)
|
|
if err != nil {
|
|
log.Printf("error in read-file while getting data: %v\n", err)
|
|
return
|
|
}
|
|
if dataPkIf == nil {
|
|
break
|
|
}
|
|
dataPk, ok := dataPkIf.(*packet.FileDataPacketType)
|
|
if !ok {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("error in read-file, invalid data packet type: %T\r\n", dataPkIf), &outputPos)
|
|
return
|
|
}
|
|
if dataPk.Error != "" {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("in read-file, data packet error: %s\r\n", dataPk.Error), &outputPos)
|
|
return
|
|
}
|
|
writeDataPk := packet.MakeFileDataPacket(writePk.ReqId)
|
|
writeDataPk.Eof = dataPk.Eof
|
|
writeDataPk.Error = dataPk.Error
|
|
writeDataPk.Type = dataPk.Type
|
|
writeDataPk.Data = make([]byte, int64(len(dataPk.Data)))
|
|
copy(writeDataPk.Data, dataPk.Data)
|
|
err = destMsh.SendFileData(writeDataPk)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("error sending file to dest: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
bytesWritten += int64(len(dataPk.Data))
|
|
fileTransferPercentage = float64(bytesWritten) / float64(fileSizeBytes)
|
|
filePercentageInt := int(fileTransferPercentage * 100)
|
|
if filePercentageInt-lastFilePercentageInt > 5 {
|
|
statusBarString := getStatusBarString(filePercentageInt)
|
|
writeStringToPty(ctx, cmd, statusBarString, &outputPos)
|
|
lastFilePercentageInt = filePercentageInt
|
|
}
|
|
}
|
|
err = checkForWriteFinished(ctx, destWriteIter)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("\r\nWrite finished packet error %v", err), &outputPos)
|
|
return
|
|
}
|
|
writeStringToPty(ctx, cmd, getStatusBarString(100), &outputPos)
|
|
writeStringToPty(ctx, cmd, " done. \r\n", &outputPos)
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Finished transferring. Transferred %v bytes\r\n", bytesWritten), &outputPos)
|
|
exitSuccess = true
|
|
}
|
|
|
|
func doCopyLocalFileToLocal(ctx context.Context, cmd *sstore.CmdType, sourcePath string, destPath string, outputPos int64) {
|
|
var exitSuccess bool
|
|
var bytesWritten int64
|
|
startTime := time.Now()
|
|
defer func() {
|
|
deferWriteCmdStatus(ctx, cmd, startTime, exitSuccess, outputPos)
|
|
}()
|
|
sourceFile, err := os.Open(sourcePath)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("error opening source file %v", err), &outputPos)
|
|
return
|
|
}
|
|
defer sourceFile.Close()
|
|
sourceFileStat, err := sourceFile.Stat()
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("error getting filestat %v", err), &outputPos)
|
|
return
|
|
}
|
|
fileSizeBytes := sourceFileStat.Size()
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Source File Size: %v\r\n", prettyPrintByteSize(fileSizeBytes)), &outputPos)
|
|
destFile, err := os.Create(destPath)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("error creating dest file %v", err), &outputPos)
|
|
return
|
|
}
|
|
defer destFile.Close()
|
|
bytesWritten, err = io.Copy(destFile, sourceFile)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("error copying files %v", err), &outputPos)
|
|
return
|
|
}
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Finished transferring. Transferred %v bytes\r\n", bytesWritten), &outputPos)
|
|
exitSuccess = true
|
|
}
|
|
|
|
func doCopyRemoteFileToLocal(ctx context.Context, cmd *sstore.CmdType, remote_msh *remote.MShellProc, sourcePath string, localPath string, outputPos int64) {
|
|
var exitSuccess bool
|
|
startTime := time.Now()
|
|
defer func() {
|
|
deferWriteCmdStatus(ctx, cmd, startTime, exitSuccess, outputPos)
|
|
}()
|
|
streamPk := packet.MakeStreamFilePacket()
|
|
streamPk.ReqId = uuid.New().String()
|
|
streamPk.Path = sourcePath
|
|
iter, err := remote_msh.StreamFile(ctx, streamPk)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Error getting file data packet: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
defer iter.Close()
|
|
respIf, err := iter.Next(ctx)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Error getting next packet: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
resp, ok := respIf.(*packet.StreamFileResponseType)
|
|
if !ok {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Error in getting packet response: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
if resp == nil || resp.Error != "" {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Response packet has error: %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
fileSizeBytes := resp.Info.Size
|
|
if fileSizeBytes == 0 {
|
|
writeStringToPty(ctx, cmd, "Source file doesn't exist or file is empty - exiting\r\n", &outputPos)
|
|
return
|
|
}
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Source File Size: %s\r\n", prettyPrintByteSize(fileSizeBytes)), &outputPos)
|
|
localFile, err := os.Create(localPath)
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Error creating file on local %v\r\n", err), &outputPos)
|
|
return
|
|
}
|
|
defer localFile.Close()
|
|
bytesWritten := int64(0)
|
|
lastFileTransferPercentage := float64(0)
|
|
fileTransferPercentage := float64(0)
|
|
writeStringToPty(ctx, cmd, "[", &outputPos)
|
|
for {
|
|
dataPkIf, err := iter.Next(ctx)
|
|
if err != nil {
|
|
log.Printf("error in read-file while getting data: %v\n", err)
|
|
return
|
|
}
|
|
if dataPkIf == nil {
|
|
break
|
|
}
|
|
dataPk, ok := dataPkIf.(*packet.FileDataPacketType)
|
|
if !ok {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("error in read-file, invalid data packet type: %T\r\n", dataPkIf), &outputPos)
|
|
return
|
|
}
|
|
if dataPk.Error != "" {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("in read-file, data packet error: %s", dataPk.Error), &outputPos)
|
|
return
|
|
}
|
|
localFile.Write(dataPk.Data)
|
|
bytesWritten += int64(len(dataPk.Data))
|
|
fileTransferPercentage = float64(bytesWritten) / float64(fileSizeBytes)
|
|
|
|
if fileTransferPercentage-lastFileTransferPercentage > float64(0.05) {
|
|
writeStringToPty(ctx, cmd, "-", &outputPos)
|
|
lastFileTransferPercentage = fileTransferPercentage
|
|
}
|
|
}
|
|
writeStringToPty(ctx, cmd, "] done. \r\n", &outputPos)
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Finished transferring. Transferred %v bytes\n", fileSizeBytes), &outputPos)
|
|
exitSuccess = true
|
|
}
|
|
|
|
func writeStringToPty(ctx context.Context, cmd *sstore.CmdType, outputString string, outputPos *int64) {
|
|
outBytes := []byte(outputString)
|
|
update, err := sstore.AppendToCmdPtyBlob(ctx, cmd.ScreenId, cmd.LineId, outBytes, *outputPos)
|
|
*outputPos += int64(len(outBytes))
|
|
if err != nil {
|
|
log.Printf("error writing to pty: %v", err)
|
|
}
|
|
scbus.MainUpdateBus.DoScreenUpdate(cmd.ScreenId, update)
|
|
err = sstore.SetStatusIndicatorLevel(ctx, cmd.ScreenId, sstore.StatusIndicatorLevel_Output, false)
|
|
if err != nil {
|
|
// This is not a fatal error, so just log it
|
|
log.Printf("error setting status indicator level to output in writeStringToPty: %v\n", err)
|
|
}
|
|
}
|
|
|
|
func parseCopyFileParam(info string) (remote string, path string, err error) {
|
|
stringsList := strings.Split(info, ":")
|
|
if len(stringsList) == 1 {
|
|
// use cur remote
|
|
return "", stringsList[0], nil
|
|
} else if len(stringsList) == 2 {
|
|
remote := strings.Trim(stringsList[0], "[] ")
|
|
return remote, stringsList[1], nil
|
|
} else {
|
|
return "error", "error", fmt.Errorf("malformed arguments")
|
|
}
|
|
}
|
|
|
|
func CopyFileCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("usage: /copyfile [file to copy] local=[path to copy to on local]")
|
|
}
|
|
ids, err := resolveUiIds(ctx, pk, R_Screen|R_Session|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to resolve connected remote id: %v", err)
|
|
}
|
|
sourceInfo := pk.Args[0]
|
|
sourceRemote, sourcePath, err := parseCopyFileParam(sourceInfo)
|
|
var sourceRemoteId *ResolvedRemote
|
|
var destRemoteId *ResolvedRemote
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error: malformed arguments - usage: [remote]:path ")
|
|
} else if sourceRemote == "" {
|
|
// use cur remote
|
|
sourceRemote = ConnectedRemote
|
|
sourceRemoteId = ids.Remote
|
|
if ids.Remote.RemoteCopy.IsLocal() {
|
|
sourceRemote = LocalRemote
|
|
}
|
|
} else {
|
|
pk.Kwargs["remote"] = sourceRemote
|
|
sourceIds, err := resolveUiIds(ctx, pk, R_Remote)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error resolving remote id %v", err)
|
|
}
|
|
sourceRemoteId = sourceIds.Remote
|
|
}
|
|
destInfo := pk.Args[1]
|
|
destRemote, destPath, err := parseCopyFileParam(destInfo)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error: malformed arguments - usage: [remote]:path ")
|
|
} else if destRemote == "" {
|
|
destRemote = ConnectedRemote
|
|
destRemoteId = ids.Remote
|
|
if ids.Remote.RemoteCopy.IsLocal() {
|
|
destRemote = LocalRemote
|
|
}
|
|
} else {
|
|
pk.Kwargs["remote"] = destRemote
|
|
destIds, err := resolveUiIds(ctx, pk, R_Remote)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error resolving remote id %v", err)
|
|
}
|
|
destRemoteId = destIds.Remote
|
|
}
|
|
if destPath == "" {
|
|
return nil, fmt.Errorf("error: malformed arguments - usage: [remote]:path ")
|
|
}
|
|
|
|
var sourceFullPath string
|
|
var destFullPath string
|
|
sourceMsh := sourceRemoteId.MShell
|
|
if sourceMsh == nil {
|
|
return nil, fmt.Errorf("failure getting source remote mshell")
|
|
}
|
|
sourceRRState := sourceMsh.GetRemoteRuntimeState()
|
|
sourcePathWithHome, err := sourceRRState.ExpandHomeDir(sourcePath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("expand home dir err: %v", err)
|
|
}
|
|
sourceFullPath = sourcePathWithHome
|
|
if (sourceRemote == ConnectedRemote || sourceRemote == LocalRemote) && !filepath.IsAbs(sourcePathWithHome) && sourceRemoteId.FeState != nil {
|
|
sourceCwd := sourceRemoteId.FeState["cwd"]
|
|
if sourceCwd != "" {
|
|
sourceFullPath = filepath.Join(sourceCwd, sourcePathWithHome)
|
|
}
|
|
}
|
|
if destPath[len(destPath)-1:] == "/" {
|
|
sourceFileName := filepath.Base(sourceFullPath)
|
|
destPath = filepath.Join(destPath, sourceFileName)
|
|
}
|
|
destMsh := destRemoteId.MShell
|
|
if destMsh == nil {
|
|
return nil, fmt.Errorf("failure getting dest remote mshell")
|
|
}
|
|
destRRState := destMsh.GetRemoteRuntimeState()
|
|
destPathWithHome, err := destRRState.ExpandHomeDir(destPath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("expand home dir err: %v", err)
|
|
}
|
|
destFullPath = destPathWithHome
|
|
if (destRemote == ConnectedRemote || destRemote == LocalRemote) && !filepath.IsAbs(destPathWithHome) && destRemoteId.FeState != nil {
|
|
destCwd := destRemoteId.FeState["cwd"]
|
|
if destCwd != "" {
|
|
destFullPath = filepath.Join(destCwd, destPathWithHome)
|
|
}
|
|
}
|
|
var outputPos int64
|
|
outputStr := fmt.Sprintf("Copying [%v]:%v to [%v]:%v\r\n", sourceRemoteId.DisplayName, sourceFullPath, destRemoteId.DisplayName, destFullPath)
|
|
termopts := sstore.TermOpts{Rows: shellutil.DefaultTermRows, Cols: shellutil.DefaultTermCols, FlexRows: true, MaxPtySize: remote.DefaultMaxPtySize}
|
|
cmd, err := makeDynCmd(ctx, "copy file", ids, pk.GetRawStr(), termopts)
|
|
writeStringToPty(ctx, cmd, outputStr, &outputPos)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update, err := addLineForCmd(ctx, "/copy file", false, ids, cmd, "", nil)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(sstore.InteractiveUpdate(pk.Interactive))
|
|
if destRemote != ConnectedRemote && destRemoteId != nil && !destRemoteId.RState.IsConnected() {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Attempting to autoconnect to remote %v\r\n", destRemote), &outputPos)
|
|
err = destRemoteId.MShell.TryAutoConnect()
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Couldn't connect to remote %v\r\n", sourceRemote), &outputPos)
|
|
} else {
|
|
writeStringToPty(ctx, cmd, "Auto connect successful\r\n", &outputPos)
|
|
}
|
|
}
|
|
if sourceRemote != LocalRemote && sourceRemoteId != nil && !sourceRemoteId.RState.IsConnected() {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Attempting to autoconnect to remote %v\r\n", sourceRemote), &outputPos)
|
|
err = sourceRemoteId.MShell.TryAutoConnect()
|
|
if err != nil {
|
|
writeStringToPty(ctx, cmd, fmt.Sprintf("Couldn't connect to remote %v\r\n", sourceRemote), &outputPos)
|
|
} else {
|
|
writeStringToPty(ctx, cmd, "Auto connect successful\r\n", &outputPos)
|
|
}
|
|
}
|
|
scbus.MainUpdateBus.DoScreenUpdate(cmd.ScreenId, update)
|
|
update = scbus.MakeUpdatePacket()
|
|
if destRemote == LocalRemote && sourceRemote == LocalRemote {
|
|
go doCopyLocalFileToLocal(context.Background(), cmd, sourceFullPath, destFullPath, outputPos)
|
|
} else if destRemote == LocalRemote && sourceRemote != LocalRemote {
|
|
go doCopyRemoteFileToLocal(context.Background(), cmd, sourceMsh, sourceFullPath, destFullPath, outputPos)
|
|
} else if destRemote != LocalRemote && sourceRemote == LocalRemote {
|
|
go doCopyLocalFileToRemote(context.Background(), cmd, destMsh, sourceFullPath, destFullPath, outputPos)
|
|
} else if destRemote != LocalRemote && sourceRemote != LocalRemote {
|
|
go doCopyRemoteFileToRemote(context.Background(), cmd, sourceMsh, destMsh, sourceFullPath, destFullPath, outputPos)
|
|
}
|
|
return update, nil
|
|
}
|
|
|
|
func RemoteInstallCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_Remote)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
mshell := ids.Remote.MShell
|
|
go mshell.RunInstall()
|
|
return createRemoteViewRemoteIdUpdate(ids.Remote.RemotePtr.RemoteId), nil
|
|
}
|
|
|
|
func RemoteInstallCancelCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_Remote)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
mshell := ids.Remote.MShell
|
|
go mshell.CancelInstall()
|
|
return createRemoteViewRemoteIdUpdate(ids.Remote.RemotePtr.RemoteId), nil
|
|
}
|
|
|
|
func RemoteConnectCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_Remote)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
go ids.Remote.MShell.Launch(true)
|
|
return createRemoteViewRemoteIdUpdate(ids.Remote.RemotePtr.RemoteId), nil
|
|
}
|
|
|
|
func RemoteDisconnectCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_Remote)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
force := resolveBool(pk.Kwargs["force"], false)
|
|
go ids.Remote.MShell.Disconnect(force)
|
|
return createRemoteViewRemoteIdUpdate(ids.Remote.RemotePtr.RemoteId), nil
|
|
}
|
|
|
|
func makeRemoteEditUpdate_new(err error) scbus.UpdatePacket {
|
|
redit := &sstore.RemoteEditType{
|
|
RemoteEdit: true,
|
|
}
|
|
if err != nil {
|
|
redit.ErrorStr = err.Error()
|
|
}
|
|
return createRemoteViewRemoteEditUpdate(redit)
|
|
}
|
|
|
|
func makeRemoteEditErrorReturn_new(visual bool, err error) (scbus.UpdatePacket, error) {
|
|
if visual {
|
|
return makeRemoteEditUpdate_new(err), nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
func makeRemoteEditUpdate_edit(ids resolvedIds, err error) scbus.UpdatePacket {
|
|
redit := &sstore.RemoteEditType{
|
|
RemoteEdit: true,
|
|
}
|
|
redit.RemoteId = ids.Remote.RemotePtr.RemoteId
|
|
if ids.Remote.RemoteCopy.SSHOpts != nil {
|
|
redit.KeyStr = ids.Remote.RemoteCopy.SSHOpts.SSHIdentity
|
|
redit.HasPassword = (ids.Remote.RemoteCopy.SSHOpts.SSHPassword != "")
|
|
}
|
|
if err != nil {
|
|
redit.ErrorStr = err.Error()
|
|
}
|
|
return createRemoteViewRemoteEditUpdate(redit)
|
|
}
|
|
|
|
func makeRemoteEditErrorReturn_edit(ids resolvedIds, visual bool, err error) (scbus.UpdatePacket, error) {
|
|
if visual {
|
|
return makeRemoteEditUpdate_edit(ids, err), nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
type RemoteEditArgs struct {
|
|
CanonicalName string
|
|
SSHOpts *sstore.SSHOpts
|
|
ConnectMode string
|
|
Alias string
|
|
AutoInstall bool
|
|
Color string
|
|
ShellPref string
|
|
EditMap map[string]interface{}
|
|
}
|
|
|
|
func parseRemoteEditArgs(isNew bool, pk *scpacket.FeCommandPacketType, isLocal bool) (*RemoteEditArgs, error) {
|
|
var canonicalName string
|
|
var sshOpts *sstore.SSHOpts
|
|
var isSudo bool
|
|
|
|
if isNew {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/remote:new must specify user@host argument (set visual=1 to edit in UI)")
|
|
}
|
|
userHost := pk.Args[0]
|
|
m := userHostRe.FindStringSubmatch(userHost)
|
|
if m == nil {
|
|
return nil, fmt.Errorf("invalid format of user@host argument")
|
|
}
|
|
sudoStr, remoteUser, remoteHost, remotePortStr := m[1], m[2], m[3], m[4]
|
|
var uhPort int
|
|
if remotePortStr != "" {
|
|
var err error
|
|
uhPort, err = strconv.Atoi(remotePortStr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid port specified on user@host argument")
|
|
}
|
|
}
|
|
if sudoStr != "" {
|
|
isSudo = true
|
|
}
|
|
if pk.Kwargs["sudo"] != "" {
|
|
sudoArg := resolveBool(pk.Kwargs["sudo"], false)
|
|
if isSudo && !sudoArg {
|
|
return nil, fmt.Errorf("invalid 'sudo' argument, with sudo kw arg set to false")
|
|
}
|
|
if !isSudo && sudoArg {
|
|
isSudo = true
|
|
}
|
|
}
|
|
sshOpts = &sstore.SSHOpts{
|
|
Local: false,
|
|
SSHHost: remoteHost,
|
|
SSHUser: remoteUser,
|
|
IsSudo: isSudo,
|
|
}
|
|
portVal, err := resolvePosInt(pk.Kwargs["port"], 0)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid port %q: %v", pk.Kwargs["port"], err)
|
|
}
|
|
if portVal != 0 && uhPort != 0 && portVal != uhPort {
|
|
return nil, fmt.Errorf("invalid port argument, does not match port specified in 'user@host:port' argument")
|
|
}
|
|
if portVal == 0 && uhPort != 0 {
|
|
portVal = uhPort
|
|
}
|
|
if portVal < 0 || portVal > 65535 {
|
|
// 0 is used as a sentinel value for the default in this case
|
|
return nil, fmt.Errorf("invalid port argument, \"%d\" is not in the range of 1 to 65535", portVal)
|
|
}
|
|
sshOpts.SSHPort = portVal
|
|
canonicalName = remoteUser + "@" + remoteHost
|
|
if portVal != 0 && portVal != 22 {
|
|
canonicalName = canonicalName + ":" + strconv.Itoa(portVal)
|
|
}
|
|
if isSudo {
|
|
canonicalName = "sudo@" + canonicalName
|
|
}
|
|
} else {
|
|
if pk.Kwargs["sudo"] != "" {
|
|
return nil, fmt.Errorf("cannot update 'sudo' value")
|
|
}
|
|
if pk.Kwargs["port"] != "" {
|
|
return nil, fmt.Errorf("cannot update 'port' value")
|
|
}
|
|
}
|
|
alias := pk.Kwargs["alias"]
|
|
if alias != "" {
|
|
if len(alias) > MaxRemoteAliasLen {
|
|
return nil, fmt.Errorf("alias too long, max length = %d", MaxRemoteAliasLen)
|
|
}
|
|
if !remoteAliasRe.MatchString(alias) {
|
|
return nil, fmt.Errorf("invalid alias format")
|
|
}
|
|
}
|
|
var shellPref string
|
|
if isNew {
|
|
shellPref = sstore.ShellTypePref_Detect
|
|
}
|
|
if pk.Kwargs["shellpref"] != "" {
|
|
shellPref = pk.Kwargs["shellpref"]
|
|
}
|
|
if shellPref != "" && shellPref != packet.ShellType_bash && shellPref != packet.ShellType_zsh && shellPref != sstore.ShellTypePref_Detect {
|
|
return nil, fmt.Errorf("invalid shellpref %q, must be %s", shellPref, formatStrs([]string{packet.ShellType_bash, packet.ShellType_zsh, sstore.ShellTypePref_Detect}, "or", false))
|
|
}
|
|
var connectMode string
|
|
if isNew {
|
|
connectMode = sstore.ConnectModeAuto
|
|
}
|
|
if pk.Kwargs["connectmode"] != "" {
|
|
connectMode = pk.Kwargs["connectmode"]
|
|
}
|
|
if connectMode != "" && !sstore.IsValidConnectMode(connectMode) {
|
|
err := fmt.Errorf("invalid connectmode %q: valid modes are %s", connectMode, formatStrs([]string{sstore.ConnectModeStartup, sstore.ConnectModeAuto, sstore.ConnectModeManual}, "or", false))
|
|
return nil, err
|
|
}
|
|
keyFile, err := resolveFile(pk.Kwargs["key"])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid ssh keyfile %q: %v", pk.Kwargs["key"], err)
|
|
}
|
|
color := pk.Kwargs["color"]
|
|
if color != "" {
|
|
err := validateRemoteColor(color, "remote color")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
sshPassword := pk.Kwargs["password"]
|
|
if sshOpts != nil {
|
|
sshOpts.SSHIdentity = keyFile
|
|
sshOpts.SSHPassword = sshPassword
|
|
}
|
|
|
|
// set up editmap
|
|
editMap := make(map[string]interface{})
|
|
if _, found := pk.Kwargs[sstore.RemoteField_Alias]; found {
|
|
editMap[sstore.RemoteField_Alias] = alias
|
|
}
|
|
if connectMode != "" {
|
|
if isLocal {
|
|
return nil, fmt.Errorf("Cannot edit connect mode for 'local' remote")
|
|
}
|
|
editMap[sstore.RemoteField_ConnectMode] = connectMode
|
|
}
|
|
if _, found := pk.Kwargs["key"]; found {
|
|
if isLocal {
|
|
return nil, fmt.Errorf("Cannot edit ssh key file for 'local' remote")
|
|
}
|
|
editMap[sstore.RemoteField_SSHKey] = keyFile
|
|
}
|
|
if _, found := pk.Kwargs[sstore.RemoteField_Color]; found {
|
|
editMap[sstore.RemoteField_Color] = color
|
|
}
|
|
if _, found := pk.Kwargs["password"]; found && pk.Kwargs["password"] != PasswordUnchangedSentinel {
|
|
if isLocal {
|
|
return nil, fmt.Errorf("Cannot edit ssh password for 'local' remote")
|
|
}
|
|
editMap[sstore.RemoteField_SSHPassword] = sshPassword
|
|
}
|
|
if _, found := pk.Kwargs["shellpref"]; found {
|
|
editMap[sstore.RemoteField_ShellPref] = shellPref
|
|
}
|
|
|
|
return &RemoteEditArgs{
|
|
SSHOpts: sshOpts,
|
|
ConnectMode: connectMode,
|
|
Alias: alias,
|
|
AutoInstall: true,
|
|
CanonicalName: canonicalName,
|
|
Color: color,
|
|
EditMap: editMap,
|
|
ShellPref: shellPref,
|
|
}, nil
|
|
}
|
|
|
|
func RemoteNewCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
visualEdit := resolveBool(pk.Kwargs["visual"], false)
|
|
isSubmitted := resolveBool(pk.Kwargs["submit"], false)
|
|
if visualEdit && !isSubmitted && len(pk.Args) == 0 {
|
|
return makeRemoteEditUpdate_new(nil), nil
|
|
}
|
|
editArgs, err := parseRemoteEditArgs(true, pk, false)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/remote:new %v", err)
|
|
}
|
|
r := &sstore.RemoteType{
|
|
RemoteId: scbase.GenWaveUUID(),
|
|
RemoteType: sstore.RemoteTypeSsh,
|
|
RemoteAlias: editArgs.Alias,
|
|
RemoteCanonicalName: editArgs.CanonicalName,
|
|
RemoteUser: editArgs.SSHOpts.SSHUser,
|
|
RemoteHost: editArgs.SSHOpts.SSHHost,
|
|
ConnectMode: editArgs.ConnectMode,
|
|
AutoInstall: editArgs.AutoInstall,
|
|
SSHOpts: editArgs.SSHOpts,
|
|
SSHConfigSrc: sstore.SSHConfigSrcTypeManual,
|
|
ShellPref: editArgs.ShellPref,
|
|
}
|
|
if editArgs.Color != "" {
|
|
r.RemoteOpts = &sstore.RemoteOptsType{Color: editArgs.Color}
|
|
}
|
|
err = remote.AddRemote(ctx, r, true)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot create remote %q: %v", r.RemoteCanonicalName, err)
|
|
}
|
|
// SUCCESS
|
|
return createRemoteViewRemoteIdUpdate(r.RemoteId), nil
|
|
}
|
|
|
|
func RemoteSetCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_Remote)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
visualEdit := resolveBool(pk.Kwargs["visual"], false)
|
|
isSubmitted := resolveBool(pk.Kwargs["submit"], false)
|
|
editArgs, err := parseRemoteEditArgs(false, pk, ids.Remote.MShell.IsLocal())
|
|
if err != nil {
|
|
return makeRemoteEditErrorReturn_edit(ids, visualEdit, fmt.Errorf("/remote:new %v", err))
|
|
}
|
|
if visualEdit && !isSubmitted && len(editArgs.EditMap) == 0 {
|
|
return makeRemoteEditUpdate_edit(ids, nil), nil
|
|
}
|
|
if !visualEdit && len(editArgs.EditMap) == 0 {
|
|
return nil, fmt.Errorf("/remote:set no updates, can set %s. (set visual=1 to edit in UI)", formatStrs(RemoteSetArgs, "or", false))
|
|
}
|
|
err = ids.Remote.MShell.UpdateRemote(ctx, editArgs.EditMap)
|
|
if err != nil {
|
|
return makeRemoteEditErrorReturn_edit(ids, visualEdit, fmt.Errorf("/remote:new error updating remote: %v", err))
|
|
}
|
|
if visualEdit {
|
|
return createRemoteViewRemoteIdUpdate(ids.Remote.RemoteCopy.RemoteId), nil
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoMsg: fmt.Sprintf("remote %q updated", ids.Remote.DisplayName),
|
|
TimeoutMs: 2000,
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func RemoteShowCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_Remote)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
state := ids.Remote.RState
|
|
return createRemoteViewRemoteIdUpdate(state.RemoteId), nil
|
|
}
|
|
|
|
func RemoteShowAllCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
stateArr := remote.GetAllRemoteRuntimeState()
|
|
var buf bytes.Buffer
|
|
for _, rstate := range stateArr {
|
|
var name string
|
|
if rstate.RemoteAlias == "" {
|
|
name = rstate.RemoteCanonicalName
|
|
} else {
|
|
name = fmt.Sprintf("%s (%s)", rstate.RemoteCanonicalName, rstate.RemoteAlias)
|
|
}
|
|
buf.WriteString(fmt.Sprintf("%-12s %-5s %8s %s\n", rstate.Status, rstate.RemoteType, rstate.RemoteId[0:8], name))
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.RemoteViewType{
|
|
RemoteShowAll: true,
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func resolveSshConfigPatterns(configFiles []string) ([]string, error) {
|
|
// using two separate containers to track order and have O(1) lookups
|
|
// since go does not have an ordered map primitive
|
|
var discoveredPatterns []string
|
|
alreadyUsed := make(map[string]bool)
|
|
alreadyUsed[""] = true // this excludes the empty string from potential alias
|
|
var openedFiles []fs.File
|
|
|
|
defer func() {
|
|
for _, openedFile := range openedFiles {
|
|
openedFile.Close()
|
|
}
|
|
}()
|
|
|
|
var errs []error
|
|
for _, configFile := range configFiles {
|
|
fd, openErr := os.Open(configFile)
|
|
openedFiles = append(openedFiles, fd)
|
|
if fd == nil {
|
|
errs = append(errs, openErr)
|
|
continue
|
|
}
|
|
|
|
cfg, _ := ssh_config.Decode(fd)
|
|
for _, host := range cfg.Hosts {
|
|
// for each host, find the first good alias
|
|
for _, hostPattern := range host.Patterns {
|
|
hostPatternStr := hostPattern.String()
|
|
if strings.Index(hostPatternStr, "*") == -1 || alreadyUsed[hostPatternStr] == true {
|
|
discoveredPatterns = append(discoveredPatterns, hostPatternStr)
|
|
alreadyUsed[hostPatternStr] = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if len(errs) == len(configFiles) {
|
|
errs = append([]error{fmt.Errorf("no ssh config files could be opened:\n")}, errs...)
|
|
return nil, errors.Join(errs...)
|
|
}
|
|
if len(discoveredPatterns) == 0 {
|
|
return nil, fmt.Errorf("no compatible hostnames found in ssh config files")
|
|
}
|
|
|
|
return discoveredPatterns, nil
|
|
}
|
|
|
|
type HostInfoType struct {
|
|
Host string
|
|
User string
|
|
CanonicalName string
|
|
Port int
|
|
SshKeyFile string
|
|
ConnectMode string
|
|
Ignore bool
|
|
ShellPref string
|
|
}
|
|
|
|
func createSshImportSummary(changeList map[string][]string) string {
|
|
totalNumChanges := len(changeList["create"]) + len(changeList["delete"]) + len(changeList["update"]) + len(changeList["createErr"]) + len(changeList["deleteErr"]) + len(changeList["updateErr"])
|
|
if totalNumChanges == 0 {
|
|
return "No changes made from ssh config import"
|
|
}
|
|
remoteStatusMsgs := map[string]string{
|
|
"delete": "Deleted %d connection%s: %s",
|
|
"create": "Created %d connection%s: %s",
|
|
"update": "Edited %d connection%s: %s",
|
|
"deleteErr": "Error deleting %d connection%s: %s",
|
|
"createErr": "Error creating %d connection%s: %s",
|
|
"updateErr": "Error editing %d connection%s: %s",
|
|
}
|
|
|
|
changeTypeKeys := []string{"delete", "create", "update", "deleteErr", "createErr", "updateErr"}
|
|
|
|
var outMsgs []string
|
|
for _, changeTypeKey := range changeTypeKeys {
|
|
changes := changeList[changeTypeKey]
|
|
if len(changes) > 0 {
|
|
rawStatusMsg := remoteStatusMsgs[changeTypeKey]
|
|
var pluralize string
|
|
if len(changes) == 1 {
|
|
pluralize = ""
|
|
} else {
|
|
pluralize = "s"
|
|
}
|
|
newMsg := fmt.Sprintf(rawStatusMsg, len(changes), pluralize, strings.Join(changes, ", "))
|
|
outMsgs = append(outMsgs, newMsg)
|
|
}
|
|
}
|
|
|
|
var pluralize string
|
|
if totalNumChanges == 1 {
|
|
pluralize = ""
|
|
} else {
|
|
pluralize = "s"
|
|
}
|
|
return fmt.Sprintf("%d connection%s changed:\n\n%s", totalNumChanges, pluralize, strings.Join(outMsgs, "\n\n"))
|
|
}
|
|
|
|
func NewHostInfo(hostName string) (*HostInfoType, error) {
|
|
userName, _ := ssh_config.GetStrict(hostName, "User")
|
|
if userName == "" {
|
|
// we cannot store a remote with a missing user
|
|
// in the current setup
|
|
return nil, fmt.Errorf("could not parse \"%s\" - no User in config\n", hostName)
|
|
}
|
|
canonicalName := userName + "@" + hostName
|
|
|
|
// check if user and host are okay
|
|
m := userHostRe.FindStringSubmatch(canonicalName)
|
|
if m == nil || m[2] == "" || m[3] == "" {
|
|
return nil, fmt.Errorf("could not parse \"%s\" - %s did not fit user@host requirement\n", hostName, canonicalName)
|
|
}
|
|
|
|
portStr, _ := ssh_config.GetStrict(hostName, "Port")
|
|
var portVal int
|
|
if portStr != "" && portStr != "22" {
|
|
canonicalName = canonicalName + ":" + portStr
|
|
var err error
|
|
portVal, err = strconv.Atoi(portStr)
|
|
if err != nil {
|
|
// do not make assumptions about port if incorrectly configured
|
|
return nil, fmt.Errorf("could not parse \"%s\" (%s) - %s could not be converted to a valid port\n", hostName, canonicalName, portStr)
|
|
}
|
|
if portVal <= 0 || portVal > 65535 {
|
|
return nil, fmt.Errorf("could not parse port \"%d\": number is not valid for a port\n", portVal)
|
|
}
|
|
}
|
|
identityFile, _ := ssh_config.GetStrict(hostName, "IdentityFile")
|
|
passwordAuth, _ := ssh_config.GetStrict(hostName, "PasswordAuthentication")
|
|
|
|
cfgWaveOptionsStr, _ := ssh_config.GetStrict(hostName, "WaveOptions")
|
|
cfgWaveOptionsStr = strings.ToLower(cfgWaveOptionsStr)
|
|
cfgWaveOptions := make(map[string]string)
|
|
setBracketArgs(cfgWaveOptions, cfgWaveOptionsStr)
|
|
|
|
shouldIgnore := false
|
|
if result, _ := strconv.ParseBool(cfgWaveOptions["ignore"]); result {
|
|
shouldIgnore = true
|
|
}
|
|
|
|
var sshKeyFile string
|
|
connectMode := sstore.ConnectModeAuto
|
|
if cfgWaveOptions["connectmode"] == "manual" {
|
|
connectMode = sstore.ConnectModeManual
|
|
} else if _, err := os.Stat(base.ExpandHomeDir(identityFile)); err == nil {
|
|
sshKeyFile = identityFile
|
|
} else if passwordAuth == "yes" {
|
|
connectMode = sstore.ConnectModeManual
|
|
}
|
|
|
|
shellPref := sstore.ShellTypePref_Detect
|
|
if cfgWaveOptions["shellpref"] == "bash" {
|
|
shellPref = "bash"
|
|
} else if cfgWaveOptions["shellpref"] == "zsh" {
|
|
shellPref = "zsh"
|
|
}
|
|
|
|
outHostInfo := new(HostInfoType)
|
|
outHostInfo.Host = hostName
|
|
outHostInfo.User = userName
|
|
outHostInfo.CanonicalName = canonicalName
|
|
outHostInfo.Port = portVal
|
|
outHostInfo.SshKeyFile = sshKeyFile
|
|
outHostInfo.ConnectMode = connectMode
|
|
outHostInfo.Ignore = shouldIgnore
|
|
outHostInfo.ShellPref = shellPref
|
|
return outHostInfo, nil
|
|
}
|
|
|
|
func RemoteConfigParseCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
home := base.GetHomeDir()
|
|
localConfig := filepath.Join(home, ".ssh", "config")
|
|
systemConfig := filepath.Join("/", "ssh", "config")
|
|
sshConfigFiles := []string{localConfig, systemConfig}
|
|
ssh_config.ReloadConfigs()
|
|
hostPatterns, hostPatternsErr := resolveSshConfigPatterns(sshConfigFiles)
|
|
if hostPatternsErr != nil {
|
|
return nil, hostPatternsErr
|
|
}
|
|
previouslyImportedRemotes, dbQueryErr := sstore.GetAllImportedRemotes(ctx)
|
|
if dbQueryErr != nil {
|
|
return nil, dbQueryErr
|
|
}
|
|
|
|
var parsedHostData []*HostInfoType
|
|
hostInfoInConfig := make(map[string]*HostInfoType)
|
|
for _, hostPattern := range hostPatterns {
|
|
hostInfo, hostInfoErr := NewHostInfo(hostPattern)
|
|
if hostInfoErr != nil {
|
|
log.Printf("sshconfig-import: %s", hostInfoErr)
|
|
continue
|
|
}
|
|
parsedHostData = append(parsedHostData, hostInfo)
|
|
hostInfoInConfig[hostInfo.CanonicalName] = hostInfo
|
|
}
|
|
|
|
remoteChangeList := make(map[string][]string)
|
|
|
|
// remove all previously imported remotes that
|
|
// no longer have a canonical pattern in the config files
|
|
for importedRemoteCanonicalName, importedRemote := range previouslyImportedRemotes {
|
|
var err error
|
|
hostInfo := hostInfoInConfig[importedRemoteCanonicalName]
|
|
if !importedRemote.Archived && (hostInfo == nil || hostInfo.Ignore) {
|
|
err = remote.ArchiveRemote(ctx, importedRemote.RemoteId)
|
|
if err != nil {
|
|
remoteChangeList["deleteErr"] = append(remoteChangeList["deleteErr"], importedRemote.RemoteCanonicalName)
|
|
log.Printf("sshconfig-import: failed to remove remote \"%s\" (%s)\n", importedRemote.RemoteAlias, importedRemote.RemoteCanonicalName)
|
|
} else {
|
|
remoteChangeList["delete"] = append(remoteChangeList["delete"], importedRemote.RemoteCanonicalName)
|
|
log.Printf("sshconfig-import: archived remote \"%s\" (%s)\n", importedRemote.RemoteAlias, importedRemote.RemoteCanonicalName)
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, hostInfo := range parsedHostData {
|
|
previouslyImportedRemote := previouslyImportedRemotes[hostInfo.CanonicalName]
|
|
if hostInfo.Ignore {
|
|
log.Printf("sshconfig-import: ignore remote[%s] as specified in config file\n", hostInfo.CanonicalName)
|
|
continue
|
|
}
|
|
if previouslyImportedRemote != nil && !previouslyImportedRemote.Archived {
|
|
// this already existed and was created via import
|
|
// it needs to be updated instead of created
|
|
editMap := make(map[string]interface{})
|
|
editMap[sstore.RemoteField_Alias] = hostInfo.Host
|
|
editMap[sstore.RemoteField_ConnectMode] = hostInfo.ConnectMode
|
|
if hostInfo.SshKeyFile != "" {
|
|
editMap[sstore.RemoteField_SSHKey] = hostInfo.SshKeyFile
|
|
}
|
|
editMap[sstore.RemoteField_ShellPref] = hostInfo.ShellPref
|
|
msh := remote.GetRemoteById(previouslyImportedRemote.RemoteId)
|
|
if msh == nil {
|
|
remoteChangeList["updateErr"] = append(remoteChangeList["updateErr"], hostInfo.CanonicalName)
|
|
log.Printf("strange, msh for remote %s [%s] not found\n", hostInfo.CanonicalName, previouslyImportedRemote.RemoteId)
|
|
continue
|
|
}
|
|
|
|
if msh.Remote.ConnectMode == hostInfo.ConnectMode && msh.Remote.SSHOpts.SSHIdentity == hostInfo.SshKeyFile && msh.Remote.RemoteAlias == hostInfo.Host && msh.Remote.ShellPref == hostInfo.ShellPref {
|
|
// silently skip this one. it didn't fail, but no changes were needed
|
|
continue
|
|
}
|
|
|
|
err := msh.UpdateRemote(ctx, editMap)
|
|
if err != nil {
|
|
remoteChangeList["updateErr"] = append(remoteChangeList["updateErr"], hostInfo.CanonicalName)
|
|
log.Printf("error updating remote[%s]: %v\n", hostInfo.CanonicalName, err)
|
|
continue
|
|
}
|
|
remoteChangeList["update"] = append(remoteChangeList["update"], hostInfo.CanonicalName)
|
|
log.Printf("sshconfig-import: found previously imported remote with canonical name \"%s\": it has been updated\n", hostInfo.CanonicalName)
|
|
} else {
|
|
sshOpts := &sstore.SSHOpts{
|
|
Local: false,
|
|
SSHHost: hostInfo.Host,
|
|
SSHUser: hostInfo.User,
|
|
IsSudo: false,
|
|
SSHPort: hostInfo.Port,
|
|
}
|
|
if hostInfo.SshKeyFile != "" {
|
|
sshOpts.SSHIdentity = hostInfo.SshKeyFile
|
|
}
|
|
|
|
// this is new and must be created for the first time
|
|
r := &sstore.RemoteType{
|
|
RemoteId: scbase.GenWaveUUID(),
|
|
RemoteType: sstore.RemoteTypeSsh,
|
|
RemoteAlias: hostInfo.Host,
|
|
RemoteCanonicalName: hostInfo.CanonicalName,
|
|
RemoteUser: hostInfo.User,
|
|
RemoteHost: hostInfo.Host,
|
|
ConnectMode: hostInfo.ConnectMode,
|
|
AutoInstall: true,
|
|
SSHOpts: sshOpts,
|
|
SSHConfigSrc: sstore.SSHConfigSrcTypeImport,
|
|
ShellPref: sstore.ShellTypePref_Detect,
|
|
}
|
|
err := remote.AddRemote(ctx, r, false)
|
|
if err != nil {
|
|
remoteChangeList["createErr"] = append(remoteChangeList["createErr"], hostInfo.CanonicalName)
|
|
log.Printf("sshconfig-import: failed to add remote \"%s\" (%s): it is being skipped\n", hostInfo.Host, hostInfo.CanonicalName)
|
|
continue
|
|
}
|
|
remoteChangeList["create"] = append(remoteChangeList["create"], hostInfo.CanonicalName)
|
|
log.Printf("sshconfig-import: created remote \"%s\" (%s)\n", hostInfo.Host, hostInfo.CanonicalName)
|
|
}
|
|
}
|
|
|
|
outMsg := createSshImportSummary(remoteChangeList)
|
|
visualEdit := resolveBool(pk.Kwargs["visual"], false)
|
|
if visualEdit {
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.AlertMessageType{
|
|
Title: "SSH Config Import",
|
|
Message: outMsg,
|
|
Markdown: true,
|
|
})
|
|
return update, nil
|
|
} else {
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoMsg: outMsg,
|
|
})
|
|
return update, nil
|
|
}
|
|
}
|
|
|
|
func ScreenShowAllCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session)
|
|
screenArr, err := sstore.GetSessionScreens(ctx, ids.SessionId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:showall error getting screen list: %v", err)
|
|
}
|
|
var buf bytes.Buffer
|
|
for _, screen := range screenArr {
|
|
var archivedStr string
|
|
if screen.Archived {
|
|
archivedStr = " (archived)"
|
|
}
|
|
screenIdxStr := "-"
|
|
if screen.ScreenIdx != 0 {
|
|
screenIdxStr = strconv.Itoa(int(screen.ScreenIdx))
|
|
}
|
|
outStr := fmt.Sprintf("%-30s %s %s\n", screen.Name+archivedStr, screen.ScreenId, screenIdxStr)
|
|
buf.WriteString(outStr)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoTitle: fmt.Sprintf("all screens for session"),
|
|
InfoLines: splitLinesForInfo(buf.String()),
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func ScreenResetCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
localRemote := remote.GetLocalRemote()
|
|
if localRemote == nil {
|
|
return nil, fmt.Errorf("error getting local remote (not found)")
|
|
}
|
|
rptr := sstore.RemotePtrType{RemoteId: localRemote.RemoteId}
|
|
sessionUpdate := &sstore.SessionType{SessionId: ids.SessionId}
|
|
ris, err := sstore.ScreenReset(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error resetting screen: %v", err)
|
|
}
|
|
sessionUpdate.Remotes = append(sessionUpdate.Remotes, ris...)
|
|
err = sstore.UpdateCurRemote(ctx, ids.ScreenId, rptr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot reset screen remote back to local: %w", err)
|
|
}
|
|
outputStr := "reset screen state (all remote state reset)"
|
|
cmd, err := makeStaticCmd(ctx, "screen:reset", ids, pk.GetRawStr(), []byte(outputStr))
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update, err := addLineForCmd(ctx, "/screen:reset", false, ids, cmd, "", nil)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(sstore.InteractiveUpdate(pk.Interactive), sessionUpdate)
|
|
return update, nil
|
|
}
|
|
|
|
func RemoteArchiveCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_Remote)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = remote.ArchiveRemote(ctx, ids.Remote.RemotePtr.RemoteId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("archiving remote: %v", err)
|
|
}
|
|
update := sstore.InfoMsgUpdate("remote [%s] archived", ids.Remote.DisplayName)
|
|
localRemote := remote.GetLocalRemote()
|
|
rptr := sstore.RemotePtrType{RemoteId: localRemote.GetRemoteId()}
|
|
err = sstore.UpdateCurRemote(ctx, ids.ScreenId, rptr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot switch remote back to local: %w", err)
|
|
}
|
|
screen, err := sstore.GetScreenById(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot get updated screen: %w", err)
|
|
}
|
|
update.AddUpdate(*screen)
|
|
return update, nil
|
|
}
|
|
|
|
func RemoteCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
return nil, fmt.Errorf("/remote requires a subcommand: %s", formatStrs([]string{"show"}, "or", false))
|
|
}
|
|
|
|
func crShowCommand(ctx context.Context, pk *scpacket.FeCommandPacketType, ids resolvedIds) (scbus.UpdatePacket, error) {
|
|
var buf bytes.Buffer
|
|
riArr, err := sstore.GetRIsForScreen(ctx, ids.SessionId, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot get remote instances: %w", err)
|
|
}
|
|
rmap := remote.GetRemoteMap()
|
|
for _, ri := range riArr {
|
|
rptr := sstore.RemotePtrType{RemoteId: ri.RemoteId, Name: ri.Name}
|
|
msh := rmap[ri.RemoteId]
|
|
if msh == nil {
|
|
continue
|
|
}
|
|
baseDisplayName := msh.GetDisplayName()
|
|
displayName := rptr.GetDisplayName(baseDisplayName)
|
|
cwdStr := "-"
|
|
if ri.FeState["cwd"] != "" {
|
|
cwdStr = ri.FeState["cwd"]
|
|
}
|
|
buf.WriteString(fmt.Sprintf("%-30s %-50s\n", displayName, cwdStr))
|
|
}
|
|
riBaseMap := make(map[string]bool)
|
|
for _, ri := range riArr {
|
|
if ri.Name == "" {
|
|
riBaseMap[ri.RemoteId] = true
|
|
}
|
|
}
|
|
for remoteId, msh := range rmap {
|
|
if riBaseMap[remoteId] {
|
|
continue
|
|
}
|
|
feState := msh.GetDefaultFeState(msh.GetShellPref())
|
|
if feState == nil {
|
|
continue
|
|
}
|
|
cwdStr := "-"
|
|
if feState["cwd"] != "" {
|
|
cwdStr = feState["cwd"]
|
|
}
|
|
buf.WriteString(fmt.Sprintf("%-30s %-50s (default)\n", msh.GetDisplayName(), cwdStr))
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoLines: splitLinesForInfo(buf.String()),
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func GetFullRemoteDisplayName(rptr *sstore.RemotePtrType, rstate *remote.RemoteRuntimeState) string {
|
|
if rptr == nil {
|
|
return "(invalid)"
|
|
}
|
|
if rstate.RemoteAlias != "" {
|
|
fullName := rstate.RemoteAlias
|
|
if rptr.Name != "" {
|
|
fullName = fullName + ":" + rptr.Name
|
|
}
|
|
return fmt.Sprintf("[%s] (%s)", fullName, rstate.RemoteCanonicalName)
|
|
} else {
|
|
if rptr.Name != "" {
|
|
return fmt.Sprintf("[%s:%s]", rstate.RemoteCanonicalName, rptr.Name)
|
|
}
|
|
return fmt.Sprintf("[%s]", rstate.RemoteCanonicalName)
|
|
}
|
|
}
|
|
|
|
func writeErrorToPty(cmd *sstore.CmdType, errStr string, outputPos int64) {
|
|
errPk := openai.CreateErrorPacket(errStr)
|
|
errBytes, err := packet.MarshalPacket(errPk)
|
|
if err != nil {
|
|
log.Printf("error writing error packet to openai response: %v\n", err)
|
|
return
|
|
}
|
|
errCtx, cancelFn := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancelFn()
|
|
update, err := sstore.AppendToCmdPtyBlob(errCtx, cmd.ScreenId, cmd.LineId, errBytes, outputPos)
|
|
if err != nil {
|
|
log.Printf("error writing ptyupdate for openai response: %v\n", err)
|
|
return
|
|
}
|
|
scbus.MainUpdateBus.DoScreenUpdate(cmd.ScreenId, update)
|
|
return
|
|
}
|
|
|
|
func writePacketToPty(ctx context.Context, cmd *sstore.CmdType, pk packet.PacketType, outputPos *int64) error {
|
|
outBytes, err := packet.MarshalPacket(pk)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
update, err := sstore.AppendToCmdPtyBlob(ctx, cmd.ScreenId, cmd.LineId, outBytes, *outputPos)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*outputPos += int64(len(outBytes))
|
|
scbus.MainUpdateBus.DoScreenUpdate(cmd.ScreenId, update)
|
|
return nil
|
|
}
|
|
|
|
func doOpenAICompletion(cmd *sstore.CmdType, opts *sstore.OpenAIOptsType, prompt []packet.OpenAIPromptMessageType) {
|
|
var outputPos int64
|
|
var hadError bool
|
|
startTime := time.Now()
|
|
ctx, cancelFn := context.WithTimeout(context.Background(), 30*time.Second)
|
|
defer cancelFn()
|
|
defer func() {
|
|
r := recover()
|
|
if r != nil {
|
|
panicMsg := fmt.Sprintf("panic: %v", r)
|
|
log.Printf("panic in doOpenAICompletion: %s\n", panicMsg)
|
|
writeErrorToPty(cmd, panicMsg, outputPos)
|
|
hadError = true
|
|
}
|
|
duration := time.Since(startTime)
|
|
cmdStatus := sstore.CmdStatusDone
|
|
var exitCode int
|
|
if hadError {
|
|
cmdStatus = sstore.CmdStatusError
|
|
exitCode = 1
|
|
}
|
|
ck := base.MakeCommandKey(cmd.ScreenId, cmd.LineId)
|
|
donePk := packet.MakeCmdDonePacket(ck)
|
|
donePk.Ts = time.Now().UnixMilli()
|
|
donePk.ExitCode = exitCode
|
|
donePk.DurationMs = duration.Milliseconds()
|
|
update, err := sstore.UpdateCmdDoneInfo(context.Background(), ck, donePk, cmdStatus)
|
|
if err != nil {
|
|
// nothing to do
|
|
log.Printf("error updating cmddoneinfo (in openai): %v\n", err)
|
|
return
|
|
}
|
|
scbus.MainUpdateBus.DoScreenUpdate(cmd.ScreenId, update)
|
|
}()
|
|
var respPks []*packet.OpenAIPacketType
|
|
var err error
|
|
// run open ai completion locally
|
|
respPks, err = openai.RunCompletion(ctx, opts, prompt)
|
|
if err != nil {
|
|
writeErrorToPty(cmd, fmt.Sprintf("error calling OpenAI API: %v", err), outputPos)
|
|
return
|
|
}
|
|
for _, pk := range respPks {
|
|
err = writePacketToPty(ctx, cmd, pk, &outputPos)
|
|
if err != nil {
|
|
writeErrorToPty(cmd, fmt.Sprintf("error writing response to ptybuffer: %v", err), outputPos)
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func writePacketToUpdateBus(ctx context.Context, cmd *sstore.CmdType, pk *packet.OpenAICmdInfoChatMessage) {
|
|
update := sstore.UpdateWithAddNewOpenAICmdInfoPacket(ctx, cmd.ScreenId, pk)
|
|
scbus.MainUpdateBus.DoScreenUpdate(cmd.ScreenId, update)
|
|
}
|
|
|
|
func updateAsstResponseAndWriteToUpdateBus(ctx context.Context, cmd *sstore.CmdType, pk *packet.OpenAICmdInfoChatMessage, messageID int) {
|
|
update, err := sstore.UpdateWithUpdateOpenAICmdInfoPacket(ctx, cmd.ScreenId, messageID, pk)
|
|
if err != nil {
|
|
log.Printf("Open AI Update packet err: %v\n", err)
|
|
}
|
|
scbus.MainUpdateBus.DoScreenUpdate(cmd.ScreenId, update)
|
|
}
|
|
|
|
func getCmdInfoEngineeredPrompt(userQuery string, curLineStr string) string {
|
|
rtn := "You are an expert on the command line terminal. Your task is to help me write a command."
|
|
if curLineStr != "" {
|
|
rtn += "My current command is: " + curLineStr
|
|
}
|
|
rtn += ". My question is: " + userQuery + "."
|
|
return rtn
|
|
}
|
|
|
|
func doOpenAICmdInfoCompletion(cmd *sstore.CmdType, clientId string, opts *sstore.OpenAIOptsType, prompt []packet.OpenAIPromptMessageType, curLineStr string) {
|
|
var hadError bool
|
|
log.Println("had error: ", hadError)
|
|
ctx, cancelFn := context.WithTimeout(context.Background(), OpenAIStreamTimeout)
|
|
defer cancelFn()
|
|
defer func() {
|
|
r := recover()
|
|
if r != nil {
|
|
panicMsg := fmt.Sprintf("panic: %v", r)
|
|
log.Printf("panic in doOpenAICompletion: %s\n", panicMsg)
|
|
hadError = true
|
|
}
|
|
}()
|
|
var ch chan *packet.OpenAIPacketType
|
|
var err error
|
|
if opts.APIToken == "" {
|
|
var conn *websocket.Conn
|
|
ch, conn, err = openai.RunCloudCompletionStream(ctx, clientId, opts, prompt)
|
|
if conn != nil {
|
|
defer conn.Close()
|
|
}
|
|
} else {
|
|
ch, err = openai.RunCompletionStream(ctx, opts, prompt)
|
|
}
|
|
asstOutputPk := &packet.OpenAICmdInfoPacketOutputType{
|
|
Model: "",
|
|
Created: 0,
|
|
FinishReason: "",
|
|
Message: "",
|
|
}
|
|
asstOutputMessageID := sstore.ScreenMemGetCmdInfoMessageCount(cmd.ScreenId)
|
|
asstMessagePk := &packet.OpenAICmdInfoChatMessage{IsAssistantResponse: true, AssistantResponse: asstOutputPk, MessageID: asstOutputMessageID}
|
|
if err != nil {
|
|
asstOutputPk.Error = fmt.Sprintf("Error calling OpenAI API: %v", err)
|
|
writePacketToUpdateBus(ctx, cmd, asstMessagePk)
|
|
return
|
|
}
|
|
writePacketToUpdateBus(ctx, cmd, asstMessagePk)
|
|
doneWaitingForPackets := false
|
|
for !doneWaitingForPackets {
|
|
select {
|
|
case <-time.After(OpenAIPacketTimeout):
|
|
// timeout reading from channel
|
|
hadError = true
|
|
doneWaitingForPackets = true
|
|
asstOutputPk.Error = "timeout waiting for server response"
|
|
updateAsstResponseAndWriteToUpdateBus(ctx, cmd, asstMessagePk, asstOutputMessageID)
|
|
break
|
|
case pk, ok := <-ch:
|
|
if ok {
|
|
// got a packet
|
|
if pk.Error != "" {
|
|
hadError = true
|
|
asstOutputPk.Error = pk.Error
|
|
}
|
|
if pk.Model != "" && pk.Index == 0 {
|
|
asstOutputPk.Model = pk.Model
|
|
asstOutputPk.Created = pk.Created
|
|
asstOutputPk.FinishReason = pk.FinishReason
|
|
if pk.Text != "" {
|
|
asstOutputPk.Message += pk.Text
|
|
}
|
|
}
|
|
if pk.Index == 0 {
|
|
if pk.FinishReason != "" {
|
|
asstOutputPk.FinishReason = pk.FinishReason
|
|
}
|
|
if pk.Text != "" {
|
|
asstOutputPk.Message += pk.Text
|
|
}
|
|
}
|
|
asstMessagePk.AssistantResponse = asstOutputPk
|
|
updateAsstResponseAndWriteToUpdateBus(ctx, cmd, asstMessagePk, asstOutputMessageID)
|
|
|
|
} else {
|
|
// channel closed
|
|
doneWaitingForPackets = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func doOpenAIStreamCompletion(cmd *sstore.CmdType, clientId string, opts *sstore.OpenAIOptsType, prompt []packet.OpenAIPromptMessageType) {
|
|
var outputPos int64
|
|
var hadError bool
|
|
startTime := time.Now()
|
|
ctx, cancelFn := context.WithTimeout(context.Background(), OpenAIStreamTimeout)
|
|
defer cancelFn()
|
|
defer func() {
|
|
r := recover()
|
|
if r != nil {
|
|
panicMsg := fmt.Sprintf("panic: %v", r)
|
|
log.Printf("panic in doOpenAICompletion: %s\n", panicMsg)
|
|
writeErrorToPty(cmd, panicMsg, outputPos)
|
|
hadError = true
|
|
}
|
|
duration := time.Since(startTime)
|
|
cmdStatus := sstore.CmdStatusDone
|
|
var exitCode int
|
|
if hadError {
|
|
cmdStatus = sstore.CmdStatusError
|
|
exitCode = 1
|
|
}
|
|
ck := base.MakeCommandKey(cmd.ScreenId, cmd.LineId)
|
|
donePk := packet.MakeCmdDonePacket(ck)
|
|
donePk.Ts = time.Now().UnixMilli()
|
|
donePk.ExitCode = exitCode
|
|
donePk.DurationMs = duration.Milliseconds()
|
|
update, err := sstore.UpdateCmdDoneInfo(context.Background(), ck, donePk, cmdStatus)
|
|
if err != nil {
|
|
// nothing to do
|
|
log.Printf("error updating cmddoneinfo (in openai): %v\n", err)
|
|
return
|
|
}
|
|
scbus.MainUpdateBus.DoScreenUpdate(cmd.ScreenId, update)
|
|
}()
|
|
var ch chan *packet.OpenAIPacketType
|
|
var err error
|
|
if opts.APIToken == "" {
|
|
var conn *websocket.Conn
|
|
ch, conn, err = openai.RunCloudCompletionStream(ctx, clientId, opts, prompt)
|
|
if conn != nil {
|
|
defer conn.Close()
|
|
}
|
|
} else {
|
|
ch, err = openai.RunCompletionStream(ctx, opts, prompt)
|
|
}
|
|
if err != nil {
|
|
writeErrorToPty(cmd, fmt.Sprintf("error calling OpenAI API: %v", err), outputPos)
|
|
return
|
|
}
|
|
doneWaitingForPackets := false
|
|
for !doneWaitingForPackets {
|
|
select {
|
|
case <-time.After(OpenAIPacketTimeout):
|
|
// timeout reading from channel
|
|
hadError = true
|
|
pk := openai.CreateErrorPacket(fmt.Sprintf("timeout waiting for server response"))
|
|
err = writePacketToPty(ctx, cmd, pk, &outputPos)
|
|
if err != nil {
|
|
log.Printf("error writing response to ptybuffer: %v", err)
|
|
return
|
|
}
|
|
doneWaitingForPackets = true
|
|
break
|
|
case pk, ok := <-ch:
|
|
if ok {
|
|
// got a packet
|
|
if pk.Error != "" {
|
|
hadError = true
|
|
}
|
|
err = writePacketToPty(ctx, cmd, pk, &outputPos)
|
|
if err != nil {
|
|
hadError = true
|
|
log.Printf("error writing response to ptybuffer: %v", err)
|
|
return
|
|
}
|
|
} else {
|
|
// channel closed
|
|
doneWaitingForPackets = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func BuildOpenAIPromptArrayWithContext(messages []*packet.OpenAICmdInfoChatMessage) []packet.OpenAIPromptMessageType {
|
|
rtn := make([]packet.OpenAIPromptMessageType, 0)
|
|
for _, msg := range messages {
|
|
content := msg.UserEngineeredQuery
|
|
if msg.UserEngineeredQuery == "" {
|
|
content = msg.UserQuery
|
|
}
|
|
msgRole := sstore.OpenAIRoleUser
|
|
if msg.IsAssistantResponse {
|
|
msgRole = sstore.OpenAIRoleAssistant
|
|
content = msg.AssistantResponse.Message
|
|
}
|
|
rtn = append(rtn, packet.OpenAIPromptMessageType{Role: msgRole, Content: content})
|
|
}
|
|
return rtn
|
|
}
|
|
|
|
func OpenAICommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/%s error: %w", GetCmdStr(pk), err)
|
|
}
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
if clientData.OpenAIOpts == nil {
|
|
return nil, fmt.Errorf("error retrieving client open ai options")
|
|
}
|
|
opts := clientData.OpenAIOpts
|
|
if opts.APIToken == "" {
|
|
if clientData.ClientOpts.NoTelemetry {
|
|
return nil, fmt.Errorf(OpenAICloudCompletionTelemetryOffErrorMsg)
|
|
}
|
|
}
|
|
if opts.Model == "" {
|
|
opts.Model = openai.DefaultModel
|
|
}
|
|
if opts.MaxTokens == 0 {
|
|
opts.MaxTokens = openai.DefaultMaxTokens
|
|
}
|
|
promptStr := firstArg(pk)
|
|
ptermVal := defaultStr(pk.Kwargs["wterm"], DefaultPTERM)
|
|
pkTermOpts, err := GetUITermOpts(pk.UIContext.WinSize, ptermVal)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("openai error, invalid 'pterm' value %q: %v", ptermVal, err)
|
|
}
|
|
termOpts := convertTermOpts(pkTermOpts)
|
|
cmd, err := makeDynCmd(ctx, GetCmdStr(pk), ids, pk.GetRawStr(), *termOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("openai error, cannot make dyn cmd")
|
|
}
|
|
if resolveBool(pk.Kwargs["cmdinfo"], false) {
|
|
if promptStr == "" {
|
|
// this is requesting an update without wanting an openai query
|
|
update := sstore.UpdateWithCurrentOpenAICmdInfoChat(cmd.ScreenId, nil)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting update for CmdInfoChat %v", err)
|
|
}
|
|
return update, nil
|
|
}
|
|
curLineStr := defaultStr(pk.Kwargs["curline"], "")
|
|
userQueryPk := &packet.OpenAICmdInfoChatMessage{UserQuery: promptStr, MessageID: sstore.ScreenMemGetCmdInfoMessageCount(cmd.ScreenId)}
|
|
engineeredQuery := getCmdInfoEngineeredPrompt(promptStr, curLineStr)
|
|
userQueryPk.UserEngineeredQuery = engineeredQuery
|
|
writePacketToUpdateBus(ctx, cmd, userQueryPk)
|
|
prompt := BuildOpenAIPromptArrayWithContext(sstore.ScreenMemGetCmdInfoChat(cmd.ScreenId).Messages)
|
|
go doOpenAICmdInfoCompletion(cmd, clientData.ClientId, opts, prompt, curLineStr)
|
|
update := scbus.MakeUpdatePacket()
|
|
return update, nil
|
|
}
|
|
prompt := []packet.OpenAIPromptMessageType{{Role: sstore.OpenAIRoleUser, Content: promptStr}}
|
|
if resolveBool(pk.Kwargs["cmdinfoclear"], false) {
|
|
update := sstore.UpdateWithClearOpenAICmdInfo(cmd.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error clearing CmdInfoChat: %v", err)
|
|
}
|
|
return update, nil
|
|
}
|
|
if promptStr == "" {
|
|
return nil, fmt.Errorf("openai error, prompt string is blank")
|
|
}
|
|
line, err := sstore.AddOpenAILine(ctx, ids.ScreenId, DefaultUserId, cmd)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot add new line: %v", err)
|
|
}
|
|
if resolveBool(pk.Kwargs["stream"], true) {
|
|
go doOpenAIStreamCompletion(cmd, clientData.ClientId, opts, prompt)
|
|
} else {
|
|
go doOpenAICompletion(cmd, opts, prompt)
|
|
}
|
|
updateHistoryContext(ctx, line, cmd, nil)
|
|
updateMap := make(map[string]interface{})
|
|
updateMap[sstore.ScreenField_SelectedLine] = line.LineNum
|
|
updateMap[sstore.ScreenField_Focus] = sstore.ScreenFocusInput
|
|
screen, err := sstore.UpdateScreen(ctx, ids.ScreenId, updateMap)
|
|
if err != nil {
|
|
// ignore error again (nothing to do)
|
|
log.Printf("openai error updating screen selected line: %v\n", err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
sstore.AddLineUpdate(update, line, cmd)
|
|
update.AddUpdate(*screen)
|
|
return update, nil
|
|
}
|
|
|
|
func CrCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/%s error: %w", GetCmdStr(pk), err)
|
|
}
|
|
newRemote := firstArg(pk)
|
|
if newRemote == "" {
|
|
return crShowCommand(ctx, pk, ids)
|
|
}
|
|
_, rptr, rstate, err := resolveRemote(ctx, newRemote, ids.SessionId, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if rptr == nil {
|
|
return nil, fmt.Errorf("/%s error: remote %q not found", GetCmdStr(pk), newRemote)
|
|
}
|
|
if rstate.Archived {
|
|
return nil, fmt.Errorf("/%s error: remote %q cannot switch to archived remote", GetCmdStr(pk), newRemote)
|
|
}
|
|
err = sstore.UpdateCurRemote(ctx, ids.ScreenId, *rptr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/%s error: cannot update curremote: %w", GetCmdStr(pk), err)
|
|
}
|
|
noHist := resolveBool(pk.Kwargs["nohist"], false)
|
|
if noHist {
|
|
screen, err := sstore.GetScreenById(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/%s error: cannot resolve screen for update: %w", GetCmdStr(pk), err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*screen, sstore.InteractiveUpdate(pk.Interactive))
|
|
return update, nil
|
|
}
|
|
outputStr := fmt.Sprintf("connected to %s", GetFullRemoteDisplayName(rptr, rstate))
|
|
cmd, err := makeStaticCmd(ctx, GetCmdStr(pk), ids, pk.GetRawStr(), []byte(outputStr))
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update, err := addLineForCmd(ctx, "/"+GetCmdStr(pk), false, ids, cmd, "", nil)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(sstore.InteractiveUpdate(pk.Interactive))
|
|
return update, nil
|
|
}
|
|
|
|
func makeDynCmd(ctx context.Context, metaCmd string, ids resolvedIds, cmdStr string, termOpts sstore.TermOpts) (*sstore.CmdType, error) {
|
|
cmd := &sstore.CmdType{
|
|
ScreenId: ids.ScreenId,
|
|
LineId: scbase.GenWaveUUID(),
|
|
CmdStr: cmdStr,
|
|
RawCmdStr: cmdStr,
|
|
Remote: ids.Remote.RemotePtr,
|
|
TermOpts: termOpts,
|
|
Status: sstore.CmdStatusRunning,
|
|
RunOut: nil,
|
|
}
|
|
if ids.Remote.StatePtr != nil {
|
|
cmd.StatePtr = *ids.Remote.StatePtr
|
|
}
|
|
if ids.Remote.FeState != nil {
|
|
cmd.FeState = ids.Remote.FeState
|
|
}
|
|
err := sstore.CreateCmdPtyFile(ctx, cmd.ScreenId, cmd.LineId, cmd.TermOpts.MaxPtySize)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, fmt.Errorf("cannot create local ptyout file for %s command: %w", metaCmd, err)
|
|
}
|
|
return cmd, nil
|
|
}
|
|
|
|
func makeStaticCmd(ctx context.Context, metaCmd string, ids resolvedIds, cmdStr string, cmdOutput []byte) (*sstore.CmdType, error) {
|
|
cmd := &sstore.CmdType{
|
|
ScreenId: ids.ScreenId,
|
|
LineId: scbase.GenWaveUUID(),
|
|
CmdStr: cmdStr,
|
|
RawCmdStr: cmdStr,
|
|
Remote: ids.Remote.RemotePtr,
|
|
TermOpts: sstore.TermOpts{Rows: shellutil.DefaultTermRows, Cols: shellutil.DefaultTermCols, FlexRows: true, MaxPtySize: remote.DefaultMaxPtySize},
|
|
Status: sstore.CmdStatusDone,
|
|
RunOut: nil,
|
|
}
|
|
if ids.Remote.StatePtr != nil {
|
|
cmd.StatePtr = *ids.Remote.StatePtr
|
|
}
|
|
if ids.Remote.FeState != nil {
|
|
cmd.FeState = ids.Remote.FeState
|
|
}
|
|
err := sstore.CreateCmdPtyFile(ctx, cmd.ScreenId, cmd.LineId, cmd.TermOpts.MaxPtySize)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, fmt.Errorf("cannot create local ptyout file for %s command: %w", metaCmd, err)
|
|
}
|
|
// can ignore ptyupdate
|
|
_, err = sstore.AppendToCmdPtyBlob(ctx, ids.ScreenId, cmd.LineId, cmdOutput, 0)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, fmt.Errorf("cannot append to local ptyout file for %s command: %v", metaCmd, err)
|
|
}
|
|
return cmd, nil
|
|
}
|
|
|
|
func addLineForCmd(ctx context.Context, metaCmd string, shouldFocus bool, ids resolvedIds, cmd *sstore.CmdType, renderer string, lineState map[string]any) (*scbus.ModelUpdatePacketType, error) {
|
|
rtnLine, err := sstore.AddCmdLine(ctx, ids.ScreenId, DefaultUserId, cmd, renderer, lineState)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
screen, err := sstore.GetScreenById(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
// ignore error here, because the command has already run (nothing to do)
|
|
log.Printf("%s error getting screen: %v\n", metaCmd, err)
|
|
}
|
|
if screen != nil {
|
|
updateMap := make(map[string]interface{})
|
|
updateMap[sstore.ScreenField_SelectedLine] = rtnLine.LineNum
|
|
if shouldFocus {
|
|
updateMap[sstore.ScreenField_Focus] = sstore.ScreenFocusCmd
|
|
}
|
|
screen, err = sstore.UpdateScreen(ctx, ids.ScreenId, updateMap)
|
|
if err != nil {
|
|
// ignore error again (nothing to do)
|
|
log.Printf("%s error updating screen selected line: %v\n", metaCmd, err)
|
|
}
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
sstore.AddLineUpdate(update, rtnLine, cmd)
|
|
update.AddUpdate(*screen)
|
|
sstore.IncrementNumRunningCmds_Update(update, cmd.ScreenId, 1)
|
|
updateHistoryContext(ctx, rtnLine, cmd, cmd.FeState)
|
|
return update, nil
|
|
}
|
|
|
|
func updateHistoryContext(ctx context.Context, line *sstore.LineType, cmd *sstore.CmdType, feState sstore.FeStateType) {
|
|
ctxVal := ctx.Value(historyContextKey)
|
|
if ctxVal == nil {
|
|
return
|
|
}
|
|
hctx := ctxVal.(*historyContextType)
|
|
if line != nil {
|
|
hctx.LineId = line.LineId
|
|
hctx.LineNum = line.LineNum
|
|
}
|
|
if cmd != nil {
|
|
hctx.RemotePtr = &cmd.Remote
|
|
hctx.InitialStatus = cmd.Status
|
|
} else {
|
|
hctx.InitialStatus = sstore.CmdStatusDone
|
|
}
|
|
hctx.FeState = feState
|
|
}
|
|
|
|
func makeInfoFromComps(compType string, comps []string, hasMore bool) scbus.UpdatePacket {
|
|
sort.Slice(comps, func(i int, j int) bool {
|
|
c1 := comps[i]
|
|
c2 := comps[j]
|
|
c1mc := strings.HasPrefix(c1, "^")
|
|
c2mc := strings.HasPrefix(c2, "^")
|
|
if c1mc && !c2mc {
|
|
return true
|
|
}
|
|
if !c1mc && c2mc {
|
|
return false
|
|
}
|
|
return c1 < c2
|
|
})
|
|
if len(comps) == 0 {
|
|
comps = []string{"(no completions)"}
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoTitle: fmt.Sprintf("%s completions", compType),
|
|
InfoComps: comps,
|
|
InfoCompsMore: hasMore,
|
|
})
|
|
return update
|
|
}
|
|
|
|
func simpleCompCommandMeta(ctx context.Context, prefix string, compCtx comp.CompContext, args []interface{}) (*comp.CompReturn, error) {
|
|
if strings.HasPrefix(prefix, "/") {
|
|
compsCmd, _ := comp.DoSimpleComp(ctx, comp.CGTypeCommand, prefix, compCtx, nil)
|
|
compsMeta, _ := simpleCompMeta(ctx, prefix, compCtx, nil)
|
|
return comp.CombineCompReturn(comp.CGTypeCommandMeta, compsCmd, compsMeta), nil
|
|
} else {
|
|
compsCmd, _ := comp.DoSimpleComp(ctx, comp.CGTypeCommand, prefix, compCtx, nil)
|
|
compsBareCmd, _ := simpleCompBareCmds(ctx, prefix, compCtx, nil)
|
|
return comp.CombineCompReturn(comp.CGTypeCommand, compsCmd, compsBareCmd), nil
|
|
}
|
|
}
|
|
|
|
func simpleCompBareCmds(ctx context.Context, prefix string, compCtx comp.CompContext, args []interface{}) (*comp.CompReturn, error) {
|
|
rtn := comp.CompReturn{}
|
|
for _, bmc := range BareMetaCmds {
|
|
if strings.HasPrefix(bmc.CmdStr, prefix) {
|
|
rtn.Entries = append(rtn.Entries, comp.CompEntry{Word: bmc.CmdStr, IsMetaCmd: true})
|
|
}
|
|
}
|
|
return &rtn, nil
|
|
}
|
|
|
|
func simpleCompMeta(ctx context.Context, prefix string, compCtx comp.CompContext, args []interface{}) (*comp.CompReturn, error) {
|
|
rtn := comp.CompReturn{}
|
|
validCommands := getValidCommands()
|
|
for _, cmd := range validCommands {
|
|
if strings.HasPrefix(cmd, "/_") && !strings.HasPrefix(prefix, "/_") {
|
|
continue
|
|
}
|
|
if strings.HasPrefix(cmd, prefix) {
|
|
rtn.Entries = append(rtn.Entries, comp.CompEntry{Word: cmd, IsMetaCmd: true})
|
|
}
|
|
}
|
|
return &rtn, nil
|
|
}
|
|
|
|
func doMetaCompGen(ctx context.Context, pk *scpacket.FeCommandPacketType, prefix string, forDisplay bool) ([]string, bool, error) {
|
|
ids, err := resolveUiIds(ctx, pk, 0) // best effort
|
|
var comps []string
|
|
var hasMore bool
|
|
if ids.Remote != nil && ids.Remote.RState.IsConnected() {
|
|
comps, hasMore, err = doCompGen(ctx, pk, prefix, "file", forDisplay)
|
|
if err != nil {
|
|
return nil, false, err
|
|
}
|
|
}
|
|
validCommands := getValidCommands()
|
|
for _, cmd := range validCommands {
|
|
if strings.HasPrefix(cmd, prefix) {
|
|
if forDisplay {
|
|
comps = append(comps, "^"+cmd)
|
|
} else {
|
|
comps = append(comps, cmd)
|
|
}
|
|
}
|
|
}
|
|
return comps, hasMore, nil
|
|
}
|
|
|
|
func doCompGen(ctx context.Context, pk *scpacket.FeCommandPacketType, prefix string, compType string, forDisplay bool) ([]string, bool, error) {
|
|
if compType == "metacommand" {
|
|
return doMetaCompGen(ctx, pk, prefix, forDisplay)
|
|
}
|
|
if !packet.IsValidCompGenType(compType) {
|
|
return nil, false, fmt.Errorf("/_compgen invalid type '%s'", compType)
|
|
}
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, false, fmt.Errorf("/_compgen error: %w", err)
|
|
}
|
|
cgPacket := packet.MakeCompGenPacket()
|
|
cgPacket.ReqId = uuid.New().String()
|
|
cgPacket.CompType = compType
|
|
cgPacket.Prefix = prefix
|
|
cgPacket.Cwd = ids.Remote.FeState["cwd"]
|
|
resp, err := ids.Remote.MShell.PacketRpc(ctx, cgPacket)
|
|
if err != nil {
|
|
return nil, false, err
|
|
}
|
|
if err = resp.Err(); err != nil {
|
|
return nil, false, err
|
|
}
|
|
comps := utilfn.GetStrArr(resp.Data, "comps")
|
|
hasMore := utilfn.GetBool(resp.Data, "hasmore")
|
|
return comps, hasMore, nil
|
|
}
|
|
|
|
func CompGenCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, 0) // best-effort
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/_compgen error: %w", err)
|
|
}
|
|
cmdLine := firstArg(pk)
|
|
pos := len(cmdLine)
|
|
if pk.Kwargs["comppos"] != "" {
|
|
posArg, err := strconv.Atoi(pk.Kwargs["comppos"])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/_compgen invalid comppos '%s': %w", pk.Kwargs["comppos"], err)
|
|
}
|
|
pos = posArg
|
|
}
|
|
if pos < 0 {
|
|
pos = 0
|
|
}
|
|
if pos > len(cmdLine) {
|
|
pos = len(cmdLine)
|
|
}
|
|
showComps := resolveBool(pk.Kwargs["compshow"], false)
|
|
cmdSP := utilfn.StrWithPos{Str: cmdLine, Pos: pos}
|
|
compCtx := comp.CompContext{}
|
|
if ids.Remote != nil {
|
|
rptr := ids.Remote.RemotePtr
|
|
compCtx.RemotePtr = &rptr
|
|
if ids.Remote.FeState != nil {
|
|
compCtx.Cwd = ids.Remote.FeState["cwd"]
|
|
}
|
|
}
|
|
compCtx.ForDisplay = showComps
|
|
crtn, newSP, err := comp.DoCompGen(ctx, cmdSP, compCtx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if crtn == nil {
|
|
return nil, nil
|
|
}
|
|
if showComps {
|
|
compStrs := crtn.GetCompDisplayStrs()
|
|
return makeInfoFromComps(crtn.CompType, compStrs, crtn.HasMore), nil
|
|
}
|
|
if newSP == nil || cmdSP == *newSP {
|
|
return nil, nil
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.CmdLineUpdate(utilfn.StrWithPos{Str: newSP.Str, Pos: newSP.Pos}))
|
|
return update, nil
|
|
}
|
|
|
|
func CommentCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/comment error: %w", err)
|
|
}
|
|
text := firstArg(pk)
|
|
if strings.TrimSpace(text) == "" {
|
|
return nil, fmt.Errorf("cannot post empty comment")
|
|
}
|
|
rtnLine, err := sstore.AddCommentLine(ctx, ids.ScreenId, DefaultUserId, text)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
updateHistoryContext(ctx, rtnLine, nil, nil)
|
|
updateMap := make(map[string]interface{})
|
|
updateMap[sstore.ScreenField_SelectedLine] = rtnLine.LineNum
|
|
updateMap[sstore.ScreenField_Focus] = sstore.ScreenFocusInput
|
|
screen, err := sstore.UpdateScreen(ctx, ids.ScreenId, updateMap)
|
|
if err != nil {
|
|
// ignore error again (nothing to do)
|
|
log.Printf("/comment error updating screen selected line: %v\n", err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
sstore.AddLineUpdate(update, rtnLine, nil)
|
|
update.AddUpdate(*screen)
|
|
return update, nil
|
|
}
|
|
|
|
func maybeQuote(s string, quote bool) string {
|
|
if quote {
|
|
return fmt.Sprintf("%q", s)
|
|
}
|
|
return s
|
|
}
|
|
|
|
func mapToStrs(m map[string]bool) []string {
|
|
var rtn []string
|
|
for key, val := range m {
|
|
if val {
|
|
rtn = append(rtn, key)
|
|
}
|
|
}
|
|
return rtn
|
|
}
|
|
|
|
func formatStrs(strs []string, conj string, quote bool) string {
|
|
if len(strs) == 0 {
|
|
return "(none)"
|
|
}
|
|
if len(strs) == 1 {
|
|
return maybeQuote(strs[0], quote)
|
|
}
|
|
if len(strs) == 2 {
|
|
return fmt.Sprintf("%s %s %s", maybeQuote(strs[0], quote), conj, maybeQuote(strs[1], quote))
|
|
}
|
|
var buf bytes.Buffer
|
|
for idx := 0; idx < len(strs)-1; idx++ {
|
|
buf.WriteString(maybeQuote(strs[idx], quote))
|
|
buf.WriteString(", ")
|
|
}
|
|
buf.WriteString(conj)
|
|
buf.WriteString(" ")
|
|
buf.WriteString(maybeQuote(strs[len(strs)-1], quote))
|
|
return buf.String()
|
|
}
|
|
|
|
func validateName(name string, typeStr string) error {
|
|
if len(name) > MaxNameLen {
|
|
return fmt.Errorf("%s name too long, max length is %d", typeStr, MaxNameLen)
|
|
}
|
|
if !genericNameRe.MatchString(name) {
|
|
return fmt.Errorf("invalid %s name", typeStr)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func validateShareName(name string) error {
|
|
if len(name) > MaxShareNameLen {
|
|
return fmt.Errorf("share name too long, max length is %d", MaxShareNameLen)
|
|
}
|
|
for _, ch := range name {
|
|
if !unicode.IsPrint(ch) {
|
|
return fmt.Errorf("invalid character %q in share name", string(ch))
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func validateRenderer(renderer string) error {
|
|
if renderer == "" {
|
|
return nil
|
|
}
|
|
if len(renderer) > MaxRendererLen {
|
|
return fmt.Errorf("renderer name too long, max length is %d", MaxRendererLen)
|
|
}
|
|
if !rendererRe.MatchString(renderer) {
|
|
return fmt.Errorf("invalid renderer format")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func validateColor(color string, typeStr string) error {
|
|
for _, c := range ColorNames {
|
|
if color == c {
|
|
return nil
|
|
}
|
|
}
|
|
return fmt.Errorf("invalid %s, valid colors are: %s", typeStr, formatStrs(ColorNames, "or", false))
|
|
}
|
|
|
|
func validateRemoteColor(color string, typeStr string) error {
|
|
for _, c := range RemoteColorNames {
|
|
if color == c {
|
|
return nil
|
|
}
|
|
}
|
|
return fmt.Errorf("invalid %s, valid colors are: %s", typeStr, formatStrs(RemoteColorNames, "or", false))
|
|
}
|
|
|
|
func SessionOpenSharedCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
activity := sstore.ActivityUpdate{ClickShared: 1}
|
|
sstore.UpdateActivityWrap(ctx, activity, "click-shared")
|
|
return nil, fmt.Errorf("shared sessions are not available in this version of prompt (stay tuned)")
|
|
}
|
|
|
|
func SessionOpenCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
activate := resolveBool(pk.Kwargs["activate"], true)
|
|
newName := pk.Kwargs["name"]
|
|
if newName != "" {
|
|
err := validateName(newName, "session")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
update, err := sstore.InsertSessionWithName(ctx, newName, activate)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return update, nil
|
|
}
|
|
|
|
func makeExternLink(urlStr string) string {
|
|
return fmt.Sprintf(`https://extern?%s`, url.QueryEscape(urlStr))
|
|
}
|
|
|
|
func ScreenWebShareCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
return nil, fmt.Errorf("websharing is no longer available")
|
|
}
|
|
|
|
func SessionDeleteCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, 0) // don't force R_Session
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
sessionId := ""
|
|
if len(pk.Args) >= 1 {
|
|
ritem, err := resolveSession(ctx, pk.Args[0], ids.SessionId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/session:delete error resolving session %q: %w", pk.Args[0], err)
|
|
}
|
|
if ritem == nil {
|
|
return nil, fmt.Errorf("/session:delete session %q not found", pk.Args[0])
|
|
}
|
|
sessionId = ritem.Id
|
|
} else {
|
|
sessionId = ids.SessionId
|
|
}
|
|
if sessionId == "" {
|
|
return nil, fmt.Errorf("/session:delete no sessionid found")
|
|
}
|
|
update, err := sstore.DeleteSession(ctx, sessionId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot delete session: %v", err)
|
|
}
|
|
return update, nil
|
|
}
|
|
|
|
func SessionArchiveCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, 0) // don't force R_Session
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
sessionId := ""
|
|
if len(pk.Args) >= 1 {
|
|
ritem, err := resolveSession(ctx, pk.Args[0], ids.SessionId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/session:archive error resolving session %q: %w", pk.Args[0], err)
|
|
}
|
|
if ritem == nil {
|
|
return nil, fmt.Errorf("/session:archive session %q not found", pk.Args[0])
|
|
}
|
|
sessionId = ritem.Id
|
|
} else {
|
|
sessionId = ids.SessionId
|
|
}
|
|
if sessionId == "" {
|
|
return nil, fmt.Errorf("/session:archive no sessionid found")
|
|
}
|
|
archiveVal := true
|
|
if len(pk.Args) >= 2 {
|
|
archiveVal = resolveBool(pk.Args[1], true)
|
|
}
|
|
if archiveVal {
|
|
update, err := sstore.ArchiveSession(ctx, sessionId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot archive session: %v", err)
|
|
}
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoMsg: "session archived",
|
|
})
|
|
return update, nil
|
|
} else {
|
|
activate := resolveBool(pk.Kwargs["activate"], false)
|
|
update, err := sstore.UnArchiveSession(ctx, sessionId, activate)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot un-archive session: %v", err)
|
|
}
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoMsg: "session un-archived",
|
|
})
|
|
return update, nil
|
|
}
|
|
}
|
|
|
|
func SessionShowCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
session, err := sstore.GetSessionById(ctx, ids.SessionId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot get session: %w", err)
|
|
}
|
|
if session == nil {
|
|
return nil, fmt.Errorf("session not found")
|
|
}
|
|
var buf bytes.Buffer
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "sessionid", session.SessionId))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "name", session.Name))
|
|
if session.SessionIdx != 0 {
|
|
buf.WriteString(fmt.Sprintf(" %-15s %d\n", "index", session.SessionIdx))
|
|
}
|
|
if session.Archived {
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "archived", "true"))
|
|
ts := time.UnixMilli(session.ArchivedTs)
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "archivedts", ts.Format(TsFormatStr)))
|
|
}
|
|
stats, err := sstore.GetSessionStats(ctx, ids.SessionId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting session stats: %w", err)
|
|
}
|
|
var screenArchiveStr string
|
|
if stats.NumArchivedScreens > 0 {
|
|
screenArchiveStr = fmt.Sprintf(" (%d archived)", stats.NumArchivedScreens)
|
|
}
|
|
buf.WriteString(fmt.Sprintf(" %-15s %d%s\n", "screens", stats.NumScreens, screenArchiveStr))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %d\n", "lines", stats.NumLines))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %d\n", "cmds", stats.NumCmds))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %0.2fM\n", "disksize", float64(stats.DiskStats.TotalSize)/1000000))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "disk-location", stats.DiskStats.Location))
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoTitle: "session info",
|
|
InfoLines: splitLinesForInfo(buf.String()),
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func SessionShowAllCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
sessions, err := sstore.GetBareSessions(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error retrieving sessions: %v", err)
|
|
}
|
|
var buf bytes.Buffer
|
|
for _, session := range sessions {
|
|
var archivedStr string
|
|
if session.Archived {
|
|
archivedStr = " (archived)"
|
|
}
|
|
sessionIdxStr := "-"
|
|
if session.SessionIdx != 0 {
|
|
sessionIdxStr = strconv.Itoa(int(session.SessionIdx))
|
|
}
|
|
outStr := fmt.Sprintf("%-30s %s %s\n", session.Name+archivedStr, session.SessionId, sessionIdxStr)
|
|
buf.WriteString(outStr)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoTitle: "all sessions",
|
|
InfoLines: splitLinesForInfo(buf.String()),
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func SessionSetCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var varsUpdated []string
|
|
if pk.Kwargs["name"] != "" {
|
|
newName := pk.Kwargs["name"]
|
|
err = validateName(newName, "session")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = sstore.SetSessionName(ctx, ids.SessionId, newName)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("setting session name: %v", err)
|
|
}
|
|
varsUpdated = append(varsUpdated, "name")
|
|
}
|
|
if len(varsUpdated) == 0 {
|
|
return nil, fmt.Errorf("/session:set no updates, can set %s", formatStrs([]string{"name", "pos"}, "or", false))
|
|
}
|
|
bareSession, err := sstore.GetBareSessionById(ctx, ids.SessionId)
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*bareSession, sstore.InfoMsgType{
|
|
InfoMsg: fmt.Sprintf("session updated %s", formatStrs(varsUpdated, "and", false)),
|
|
TimeoutMs: 2000,
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func SessionCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
firstArg := firstArg(pk)
|
|
if firstArg == "" {
|
|
return nil, fmt.Errorf("usage /session [name|id|pos], no param specified")
|
|
}
|
|
ritem, err := resolveSession(ctx, firstArg, ids.SessionId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = sstore.SetActiveSessionId(ctx, ritem.Id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.ActiveSessionIdUpdate(ritem.Id))
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoMsg: fmt.Sprintf("switched to session %q", ritem.Name),
|
|
TimeoutMs: 2000,
|
|
})
|
|
|
|
// Reset the status indicator for the new active screen
|
|
session, err := sstore.GetSessionById(ctx, ritem.Id)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot get session: %w", err)
|
|
}
|
|
if session == nil {
|
|
return nil, fmt.Errorf("session not found")
|
|
}
|
|
err = sstore.ResetStatusIndicator_Update(update, session.ActiveScreenId)
|
|
if err != nil {
|
|
// this is not a fatal error, just log it
|
|
log.Printf("error resetting status indicator after session command: %v\n", err)
|
|
}
|
|
|
|
return update, nil
|
|
}
|
|
|
|
func RemoteResetCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_Remote)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
shellType := ids.Remote.ShellType
|
|
if pk.Kwargs["shell"] != "" {
|
|
shellArg := pk.Kwargs["shell"]
|
|
if shellArg != packet.ShellType_bash && shellArg != packet.ShellType_zsh {
|
|
return nil, fmt.Errorf("/reset invalid shell type %q", shellArg)
|
|
}
|
|
shellType = shellArg
|
|
}
|
|
ssPk, err := ids.Remote.MShell.ReInit(ctx, shellType)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if ssPk == nil || ssPk.State == nil {
|
|
return nil, fmt.Errorf("invalid initpk received from remote (no remote state)")
|
|
}
|
|
feState := sstore.FeStateFromShellState(ssPk.State)
|
|
remoteInst, err := sstore.UpdateRemoteState(ctx, ids.SessionId, ids.ScreenId, ids.Remote.RemotePtr, feState, ssPk.State, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
outputStr := fmt.Sprintf("reset remote state (shell:%s)", ssPk.State.GetShellType())
|
|
cmd, err := makeStaticCmd(ctx, "reset", ids, pk.GetRawStr(), []byte(outputStr))
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update, err := addLineForCmd(ctx, "/reset", false, ids, cmd, "", nil)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(sstore.MakeSessionUpdateForRemote(ids.SessionId, remoteInst), sstore.InteractiveUpdate(pk.Interactive))
|
|
return update, nil
|
|
}
|
|
|
|
func ClearCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resolveBool(pk.Kwargs["archive"], false) {
|
|
update, err := sstore.ArchiveScreenLines(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("clearing screen (archiving): %v", err)
|
|
}
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoMsg: fmt.Sprintf("screen cleared (all lines archived)"),
|
|
TimeoutMs: 2000,
|
|
})
|
|
return update, nil
|
|
} else {
|
|
update, err := sstore.DeleteScreenLines(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("clearing screen: %v", err)
|
|
}
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoMsg: fmt.Sprintf("screen cleared"),
|
|
TimeoutMs: 2000,
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
}
|
|
|
|
func HistoryPurgeCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/history:purge requires at least one argument (history id)")
|
|
}
|
|
var historyIds []string
|
|
for _, historyArg := range pk.Args {
|
|
_, err := uuid.Parse(historyArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid historyid (must be uuid)")
|
|
}
|
|
historyIds = append(historyIds, historyArg)
|
|
}
|
|
err := sstore.PurgeHistoryByIds(ctx, historyIds)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/history:purge error purging items: %v", err)
|
|
}
|
|
return sstore.InfoMsgUpdate("removed history items"), nil
|
|
}
|
|
|
|
const HistoryViewPageSize = 50
|
|
|
|
var cmdFilterLs = regexp.MustCompile(`^ls(\s|$)`)
|
|
var cmdFilterCd = regexp.MustCompile(`^cd(\s|$)`)
|
|
|
|
func historyCmdFilter(hitem *sstore.HistoryItemType) bool {
|
|
cmdStr := hitem.CmdStr
|
|
if cmdStr == "" || strings.Index(cmdStr, ";") != -1 || strings.Index(cmdStr, "\n") != -1 {
|
|
return true
|
|
}
|
|
if cmdFilterLs.MatchString(cmdStr) {
|
|
return false
|
|
}
|
|
if cmdFilterCd.MatchString(cmdStr) {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func HistoryViewAllCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
_, err := resolveUiIds(ctx, pk, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
offset, err := resolveNonNegInt(pk.Kwargs["offset"], 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rawOffset, err := resolveNonNegInt(pk.Kwargs["rawoffset"], 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
opts := sstore.HistoryQueryOpts{MaxItems: HistoryViewPageSize, Offset: offset, RawOffset: rawOffset}
|
|
if pk.Kwargs["text"] != "" {
|
|
opts.SearchText = pk.Kwargs["text"]
|
|
}
|
|
if pk.Kwargs["searchsession"] != "" {
|
|
sessionId, err := resolveSessionArg(pk.Kwargs["searchsession"])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid searchsession: %v", err)
|
|
}
|
|
opts.SessionId = sessionId
|
|
}
|
|
if pk.Kwargs["searchremote"] != "" {
|
|
rptr, err := resolveRemoteArg(pk.Kwargs["searchremote"])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid searchremote: %v", err)
|
|
}
|
|
if rptr != nil {
|
|
opts.RemoteId = rptr.RemoteId
|
|
}
|
|
}
|
|
if pk.Kwargs["fromts"] != "" {
|
|
fromTs, err := resolvePosInt(pk.Kwargs["fromts"], 0)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid fromts (must be unixtime (milliseconds): %v", err)
|
|
}
|
|
if fromTs > 0 {
|
|
opts.FromTs = int64(fromTs)
|
|
}
|
|
}
|
|
if pk.Kwargs["meta"] != "" {
|
|
opts.NoMeta = !resolveBool(pk.Kwargs["meta"], true)
|
|
}
|
|
if resolveBool(pk.Kwargs["filter"], false) {
|
|
opts.FilterFn = historyCmdFilter
|
|
}
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid meta arg (must be boolean): %v", err)
|
|
}
|
|
hresult, err := sstore.GetHistoryItems(ctx, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
hvdata := &sstore.HistoryViewData{
|
|
Items: hresult.Items,
|
|
Offset: hresult.Offset,
|
|
RawOffset: hresult.RawOffset,
|
|
NextRawOffset: hresult.NextRawOffset,
|
|
HasMore: hresult.HasMore,
|
|
}
|
|
lines, cmds, err := sstore.GetLineCmdsFromHistoryItems(ctx, hvdata.Items)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
hvdata.Lines = lines
|
|
hvdata.Cmds = cmds
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(&sstore.MainViewUpdate{MainView: sstore.MainViewHistory, HistoryView: hvdata})
|
|
return update, nil
|
|
}
|
|
|
|
const DefaultMaxHistoryItems = 10000
|
|
|
|
func HistoryCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_Remote)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
maxItems, err := resolvePosInt(pk.Kwargs["maxitems"], DefaultMaxHistoryItems)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid maxitems value '%s' (must be a number): %v", pk.Kwargs["maxitems"], err)
|
|
}
|
|
if maxItems < 0 {
|
|
return nil, fmt.Errorf("invalid maxitems value '%d' (cannot be negative)", maxItems)
|
|
}
|
|
if maxItems == 0 {
|
|
maxItems = DefaultMaxHistoryItems
|
|
}
|
|
htype := HistoryTypeScreen
|
|
hSessionId := ids.SessionId
|
|
hScreenId := ids.ScreenId
|
|
if pk.Kwargs["type"] != "" {
|
|
htype = pk.Kwargs["type"]
|
|
if htype != HistoryTypeScreen && htype != HistoryTypeSession && htype != HistoryTypeGlobal {
|
|
return nil, fmt.Errorf("invalid history type '%s', valid types: %s", htype, formatStrs([]string{HistoryTypeScreen, HistoryTypeSession, HistoryTypeGlobal}, "or", false))
|
|
}
|
|
}
|
|
if htype == HistoryTypeGlobal {
|
|
hSessionId = ""
|
|
hScreenId = ""
|
|
} else if htype == HistoryTypeSession {
|
|
hScreenId = ""
|
|
}
|
|
hopts := sstore.HistoryQueryOpts{MaxItems: maxItems, SessionId: hSessionId, ScreenId: hScreenId}
|
|
hresult, err := sstore.GetHistoryItems(ctx, hopts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
show := !resolveBool(pk.Kwargs["noshow"], false)
|
|
if show {
|
|
sstore.UpdateActivityWrap(ctx, sstore.ActivityUpdate{HistoryView: 1}, "history")
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.HistoryInfoType{
|
|
HistoryType: htype,
|
|
SessionId: ids.SessionId,
|
|
ScreenId: ids.ScreenId,
|
|
Items: hresult.Items,
|
|
Show: show,
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func splitLinesForInfo(str string) []string {
|
|
rtn := strings.Split(str, "\n")
|
|
if rtn[len(rtn)-1] == "" {
|
|
return rtn[:len(rtn)-1]
|
|
}
|
|
return rtn
|
|
}
|
|
|
|
func resizeRunningCommand(ctx context.Context, cmd *sstore.CmdType, newCols int) error {
|
|
siPk := packet.MakeSpecialInputPacket()
|
|
siPk.CK = base.MakeCommandKey(cmd.ScreenId, cmd.LineId)
|
|
siPk.WinSize = &packet.WinSize{Rows: int(cmd.TermOpts.Rows), Cols: newCols}
|
|
msh := remote.GetRemoteById(cmd.Remote.RemoteId)
|
|
if msh == nil {
|
|
return fmt.Errorf("cannot resize, cmd remote not found")
|
|
}
|
|
err := msh.SendSpecialInput(siPk)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
newTermOpts := cmd.TermOpts
|
|
newTermOpts.Cols = int64(newCols)
|
|
err = sstore.UpdateCmdTermOpts(ctx, cmd.ScreenId, cmd.LineId, newTermOpts)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ScreenResizeCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
colsStr := pk.Kwargs["cols"]
|
|
if colsStr == "" {
|
|
return nil, fmt.Errorf("/screen:resize requires a numeric 'cols' argument")
|
|
}
|
|
cols, err := strconv.Atoi(colsStr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:resize requires a numeric 'cols' argument: %v", err)
|
|
}
|
|
if cols <= 0 {
|
|
return nil, fmt.Errorf("/screen:resize invalid zero/negative 'cols' argument")
|
|
}
|
|
cols = base.BoundInt(cols, shexec.MinTermCols, shexec.MaxTermCols)
|
|
runningCmds, err := sstore.GetRunningScreenCmds(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/screen:resize cannot get running commands: %v", err)
|
|
}
|
|
if len(runningCmds) == 0 {
|
|
return nil, nil
|
|
}
|
|
includeMap := resolveCommaSepListToMap(pk.Kwargs["include"])
|
|
excludeMap := resolveCommaSepListToMap(pk.Kwargs["exclude"])
|
|
for _, cmd := range runningCmds {
|
|
if excludeMap[cmd.LineId] {
|
|
continue
|
|
}
|
|
if len(includeMap) > 0 && !includeMap[cmd.LineId] {
|
|
continue
|
|
}
|
|
if int(cmd.TermOpts.Cols) != cols {
|
|
resizeRunningCommand(ctx, cmd, cols)
|
|
}
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func LineCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
return nil, fmt.Errorf("/line requires a subcommand: %s", formatStrs([]string{"show", "star", "hide", "delete", "setheight", "set"}, "or", false))
|
|
}
|
|
|
|
func LineSetHeightCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(pk.Args) != 2 {
|
|
return nil, fmt.Errorf("/line:setheight requires 2 arguments (linearg and height)")
|
|
}
|
|
lineArg := pk.Args[0]
|
|
lineId, err := sstore.FindLineIdByArg(ctx, ids.ScreenId, lineArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error looking up lineid: %v", err)
|
|
}
|
|
heightVal, err := resolveNonNegInt(pk.Args[1], 0)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:setheight invalid height val: %v", err)
|
|
}
|
|
if heightVal > 10000 {
|
|
return nil, fmt.Errorf("/line:setheight invalid height val (too large): %d", heightVal)
|
|
}
|
|
err = sstore.UpdateLineHeight(ctx, ids.ScreenId, lineId, heightVal)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:setheight error updating height: %v", err)
|
|
}
|
|
// we don't need to pass the updated line height (it is "write only")
|
|
return nil, nil
|
|
}
|
|
|
|
func LineRestartCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var lineId string
|
|
if len(pk.Args) >= 1 {
|
|
lineArg := pk.Args[0]
|
|
resolvedLineId, err := sstore.FindLineIdByArg(ctx, ids.ScreenId, lineArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error looking up lineid: %v", err)
|
|
}
|
|
lineId = resolvedLineId
|
|
} else {
|
|
selectedLineId, err := sstore.GetScreenSelectedLineId(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting selected lineid: %v", err)
|
|
}
|
|
lineId = selectedLineId
|
|
}
|
|
if lineId == "" {
|
|
return nil, fmt.Errorf("%s requires a lineid to operate on", GetCmdStr(pk))
|
|
}
|
|
line, cmd, err := sstore.GetLineCmdByLineId(ctx, ids.ScreenId, lineId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting line: %v", err)
|
|
}
|
|
if line == nil {
|
|
return nil, fmt.Errorf("line not found")
|
|
}
|
|
if cmd == nil {
|
|
return nil, fmt.Errorf("cannot restart line (no cmd found)")
|
|
}
|
|
if cmd.Status == sstore.CmdStatusRunning || cmd.Status == sstore.CmdStatusDetached {
|
|
killCtx, cancel := context.WithTimeout(ctx, 2*time.Second)
|
|
defer cancel()
|
|
err = ids.Remote.MShell.KillRunningCommandAndWait(killCtx, base.MakeCommandKey(ids.ScreenId, lineId))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
ids.Remote.MShell.ResetDataPos(base.MakeCommandKey(ids.ScreenId, lineId))
|
|
err = sstore.ClearCmdPtyFile(ctx, ids.ScreenId, lineId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error clearing existing pty file: %v", err)
|
|
}
|
|
runPacket := packet.MakeRunPacket()
|
|
runPacket.ReqId = uuid.New().String()
|
|
runPacket.CK = base.MakeCommandKey(ids.ScreenId, lineId)
|
|
runPacket.UsePty = true
|
|
// TODO how can we preseve the original termopts?
|
|
runPacket.TermOpts, err = GetUITermOpts(pk.UIContext.WinSize, DefaultPTERM)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting creating termopts for command: %w", err)
|
|
}
|
|
runPacket.Command = cmd.CmdStr
|
|
runPacket.ReturnState = false
|
|
rcOpts := remote.RunCommandOpts{
|
|
SessionId: ids.SessionId,
|
|
ScreenId: ids.ScreenId,
|
|
RemotePtr: ids.Remote.RemotePtr,
|
|
StatePtr: &cmd.StatePtr,
|
|
NoCreateCmdPtyFile: true,
|
|
}
|
|
cmd, callback, err := remote.RunCommand(ctx, rcOpts, runPacket)
|
|
sstore.IncrementNumRunningCmds(cmd.ScreenId, 1)
|
|
if callback != nil {
|
|
defer callback()
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
newTs := time.Now().UnixMilli()
|
|
err = sstore.UpdateCmdForRestart(ctx, runPacket.CK, newTs, cmd.CmdPid, cmd.RemotePid, convertTermOpts(runPacket.TermOpts))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating cmd for restart: %w", err)
|
|
}
|
|
line, cmd, err = sstore.GetLineCmdByLineId(ctx, ids.ScreenId, lineId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting updated line/cmd: %w", err)
|
|
}
|
|
cmd.Restarted = true
|
|
update := scbus.MakeUpdatePacket()
|
|
sstore.AddLineUpdate(update, line, cmd)
|
|
update.AddUpdate(sstore.InteractiveUpdate(pk.Interactive))
|
|
screen, focusErr := focusScreenLine(ctx, ids.ScreenId, line.LineNum)
|
|
if focusErr != nil {
|
|
// not a fatal error, so just log
|
|
log.Printf("error focusing screen line: %v\n", focusErr)
|
|
}
|
|
if screen != nil {
|
|
update.AddUpdate(*screen)
|
|
}
|
|
return update, nil
|
|
}
|
|
|
|
func focusScreenLine(ctx context.Context, screenId string, lineNum int64) (*sstore.ScreenType, error) {
|
|
screen, err := sstore.GetScreenById(ctx, screenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting screen: %v", err)
|
|
}
|
|
if screen == nil {
|
|
return nil, fmt.Errorf("screen not found")
|
|
}
|
|
updateMap := make(map[string]interface{})
|
|
updateMap[sstore.ScreenField_SelectedLine] = lineNum
|
|
updateMap[sstore.ScreenField_Focus] = sstore.ScreenFocusCmd
|
|
screen, err = sstore.UpdateScreen(ctx, screenId, updateMap)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating screen: %v", err)
|
|
}
|
|
return screen, nil
|
|
}
|
|
|
|
func LineSetCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(pk.Args) != 1 {
|
|
return nil, fmt.Errorf("/line:set requires 1 argument (linearg)")
|
|
}
|
|
lineArg := pk.Args[0]
|
|
lineId, err := sstore.FindLineIdByArg(ctx, ids.ScreenId, lineArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error looking up lineid: %v", err)
|
|
}
|
|
var varsUpdated []string
|
|
if renderer, found := pk.Kwargs[KwArgRenderer]; found {
|
|
if err = validateRenderer(renderer); err != nil {
|
|
return nil, fmt.Errorf("invalid renderer value: %w", err)
|
|
}
|
|
err = sstore.UpdateLineRenderer(ctx, ids.ScreenId, lineId, renderer)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error changing line renderer: %v", err)
|
|
}
|
|
varsUpdated = append(varsUpdated, KwArgRenderer)
|
|
}
|
|
if view, found := pk.Kwargs[KwArgView]; found {
|
|
if err = validateRenderer(view); err != nil {
|
|
return nil, fmt.Errorf("invalid view value: %w", err)
|
|
}
|
|
err = sstore.UpdateLineRenderer(ctx, ids.ScreenId, lineId, view)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error changing line view: %v", err)
|
|
}
|
|
varsUpdated = append(varsUpdated, KwArgView)
|
|
}
|
|
if stateJson, found := pk.Kwargs[KwArgState]; found {
|
|
if len(stateJson) > sstore.MaxLineStateSize {
|
|
return nil, fmt.Errorf("invalid state value (too large), size[%d], max[%d]", len(stateJson), sstore.MaxLineStateSize)
|
|
}
|
|
var stateMap map[string]any
|
|
err = json.Unmarshal([]byte(stateJson), &stateMap)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid state value, cannot parse json: %v", err)
|
|
}
|
|
err = sstore.UpdateLineState(ctx, ids.ScreenId, lineId, stateMap)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot update linestate: %v", err)
|
|
}
|
|
varsUpdated = append(varsUpdated, KwArgState)
|
|
}
|
|
if len(varsUpdated) == 0 {
|
|
return nil, fmt.Errorf("/line:set requires a value to set: %s", formatStrs([]string{KwArgView, KwArgState}, "or", false))
|
|
}
|
|
updatedLine, err := sstore.GetLineById(ctx, ids.ScreenId, lineId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:set cannot retrieve updated line: %v", err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
sstore.AddLineUpdate(update, updatedLine, nil)
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoMsg: fmt.Sprintf("line updated %s", formatStrs(varsUpdated, "and", false)),
|
|
TimeoutMs: 2000,
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func LineViewCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) != 3 {
|
|
return nil, fmt.Errorf("usage /line:view [session] [screen] [line]")
|
|
}
|
|
sessionArg := pk.Args[0]
|
|
screenArg := pk.Args[1]
|
|
lineArg := pk.Args[2]
|
|
sessionId, err := resolveSessionArg(sessionArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:view invalid session arg: %v", err)
|
|
}
|
|
if sessionId == "" {
|
|
return nil, fmt.Errorf("/line:view no session found")
|
|
}
|
|
screenRItem, err := resolveSessionScreen(ctx, sessionId, screenArg, "")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:view invalid screen arg: %v", err)
|
|
}
|
|
if screenRItem == nil {
|
|
return nil, fmt.Errorf("/line:view no screen found")
|
|
}
|
|
screen, err := sstore.GetScreenById(ctx, screenRItem.Id)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:view could not get screen: %v", err)
|
|
}
|
|
lineRItem, err := resolveLine(ctx, sessionId, screen.ScreenId, lineArg, "")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:view invalid line arg: %v", err)
|
|
}
|
|
update, err := sstore.SwitchScreenById(ctx, sessionId, screenRItem.Id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if lineRItem != nil {
|
|
updateMap := make(map[string]interface{})
|
|
updateMap[sstore.ScreenField_SelectedLine] = lineRItem.Num
|
|
updateMap[sstore.ScreenField_AnchorLine] = lineRItem.Num
|
|
updateMap[sstore.ScreenField_AnchorOffset] = 0
|
|
screen, err = sstore.UpdateScreen(ctx, screenRItem.Id, updateMap)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(*screen)
|
|
}
|
|
return update, nil
|
|
}
|
|
|
|
func BookmarksShowCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
// no resolve ui ids!
|
|
var tagName string // defaults to ''
|
|
if len(pk.Args) > 0 {
|
|
tagName = pk.Args[0]
|
|
}
|
|
bms, err := sstore.GetBookmarks(ctx, tagName)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve bookmarks: %v", err)
|
|
}
|
|
sstore.UpdateActivityWrap(ctx, sstore.ActivityUpdate{BookmarksView: 1}, "bookmarks")
|
|
update := scbus.MakeUpdatePacket()
|
|
|
|
update.AddUpdate(&sstore.MainViewUpdate{
|
|
MainView: sstore.MainViewBookmarks,
|
|
BookmarksView: &sstore.BookmarksUpdate{Bookmarks: bms},
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func BookmarkSetCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/bookmark:set requires one argument (bookmark id)")
|
|
}
|
|
bookmarkArg := pk.Args[0]
|
|
bookmarkId, err := sstore.GetBookmarkIdByArg(ctx, bookmarkArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error trying to resolve bookmark: %v", err)
|
|
}
|
|
if bookmarkId == "" {
|
|
return nil, fmt.Errorf("bookmark not found")
|
|
}
|
|
editMap := make(map[string]interface{})
|
|
if descStr, found := pk.Kwargs["desc"]; found {
|
|
editMap[sstore.BookmarkField_Desc] = descStr
|
|
}
|
|
if cmdStr, found := pk.Kwargs["cmdstr"]; found {
|
|
editMap[sstore.BookmarkField_CmdStr] = cmdStr
|
|
}
|
|
if len(editMap) == 0 {
|
|
return nil, fmt.Errorf("no fields set, can set %s", formatStrs([]string{"desc", "cmdstr"}, "or", false))
|
|
}
|
|
err = sstore.EditBookmark(ctx, bookmarkId, editMap)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error trying to edit bookmark: %v", err)
|
|
}
|
|
bm, err := sstore.GetBookmarkById(ctx, bookmarkId, "")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error retrieving edited bookmark: %v", err)
|
|
}
|
|
bms := []*sstore.BookmarkType{bm}
|
|
update := scbus.MakeUpdatePacket()
|
|
sstore.AddBookmarksUpdate(update, bms, nil)
|
|
update.AddUpdate(sstore.InfoMsgUpdate("bookmark edited"))
|
|
return update, nil
|
|
}
|
|
|
|
func BookmarkDeleteCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/bookmark:delete requires one argument (bookmark id)")
|
|
}
|
|
bookmarkArg := pk.Args[0]
|
|
bookmarkId, err := sstore.GetBookmarkIdByArg(ctx, bookmarkArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error trying to resolve bookmark: %v", err)
|
|
}
|
|
if bookmarkId == "" {
|
|
return nil, fmt.Errorf("bookmark not found")
|
|
}
|
|
err = sstore.DeleteBookmark(ctx, bookmarkId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error deleting bookmark: %v", err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
bms := []*sstore.BookmarkType{{BookmarkId: bookmarkId, Remove: true}}
|
|
sstore.AddBookmarksUpdate(update, bms, nil)
|
|
update.AddUpdate(sstore.InfoMsgUpdate("bookmark deleted"))
|
|
return update, nil
|
|
}
|
|
|
|
func LineBookmarkCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/line:bookmark requires an argument (line number or id)")
|
|
}
|
|
lineArg := pk.Args[0]
|
|
lineId, err := sstore.FindLineIdByArg(ctx, ids.ScreenId, lineArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error looking up lineid: %v", err)
|
|
}
|
|
if lineId == "" {
|
|
return nil, fmt.Errorf("line %q not found", lineArg)
|
|
}
|
|
_, cmdObj, err := sstore.GetLineCmdByLineId(ctx, ids.ScreenId, lineId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:bookmark error getting line: %v", err)
|
|
}
|
|
if cmdObj == nil {
|
|
return nil, fmt.Errorf("cannot bookmark non-cmd line")
|
|
}
|
|
existingBmIds, err := sstore.GetBookmarkIdsByCmdStr(ctx, cmdObj.CmdStr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error trying to retrieve current boookmarks: %v", err)
|
|
}
|
|
var newBmId string
|
|
if len(existingBmIds) > 0 {
|
|
newBmId = existingBmIds[0]
|
|
} else {
|
|
newBm := &sstore.BookmarkType{
|
|
BookmarkId: uuid.New().String(),
|
|
CreatedTs: time.Now().UnixMilli(),
|
|
CmdStr: cmdObj.CmdStr,
|
|
Alias: "",
|
|
Tags: nil,
|
|
Description: "",
|
|
}
|
|
err = sstore.InsertBookmark(ctx, newBm)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot insert bookmark: %v", err)
|
|
}
|
|
newBmId = newBm.BookmarkId
|
|
}
|
|
bms, err := sstore.GetBookmarks(ctx, "")
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(&sstore.MainViewUpdate{
|
|
MainView: sstore.MainViewBookmarks,
|
|
BookmarksView: &sstore.BookmarksUpdate{Bookmarks: bms, SelectedBookmark: newBmId},
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func LinePinCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func LineStarCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/line:star requires an argument (line number or id)")
|
|
}
|
|
if len(pk.Args) > 2 {
|
|
return nil, fmt.Errorf("/line:star only takes up to 2 arguments (line-number and star-value)")
|
|
}
|
|
lineArg := pk.Args[0]
|
|
lineId, err := sstore.FindLineIdByArg(ctx, ids.ScreenId, lineArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error looking up lineid: %v", err)
|
|
}
|
|
if lineId == "" {
|
|
return nil, fmt.Errorf("line %q not found", lineArg)
|
|
}
|
|
starVal, err := resolveNonNegInt(pk.Args[1], 1)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:star invalid star-value (not integer): %v", err)
|
|
}
|
|
if starVal > 5 {
|
|
return nil, fmt.Errorf("/line:star invalid star-value must be in the range of 0-5")
|
|
}
|
|
err = sstore.UpdateLineStar(ctx, ids.ScreenId, lineId, starVal)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:star error updating star value: %v", err)
|
|
}
|
|
lineObj, err := sstore.GetLineById(ctx, ids.ScreenId, lineId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:star error getting line: %v", err)
|
|
}
|
|
if lineObj == nil {
|
|
// no line (which is strange given we checked for it above). just return a nop.
|
|
return nil, nil
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
sstore.AddLineUpdate(update, lineObj, nil)
|
|
return update, nil
|
|
}
|
|
|
|
func LineArchiveCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/line:archive requires an argument (line number or id)")
|
|
}
|
|
lineArg := pk.Args[0]
|
|
lineId, err := sstore.FindLineIdByArg(ctx, ids.ScreenId, lineArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error looking up lineid: %v", err)
|
|
}
|
|
if lineId == "" {
|
|
return nil, fmt.Errorf("line %q not found", lineArg)
|
|
}
|
|
shouldArchive := true
|
|
if len(pk.Args) >= 2 {
|
|
shouldArchive = resolveBool(pk.Args[1], true)
|
|
}
|
|
err = sstore.SetLineArchivedById(ctx, ids.ScreenId, lineId, shouldArchive)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:archive error updating hidden status: %v", err)
|
|
}
|
|
lineObj, err := sstore.GetLineById(ctx, ids.ScreenId, lineId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:archive error getting line: %v", err)
|
|
}
|
|
if lineObj == nil {
|
|
// no line (which is strange given we checked for it above). just return a nop.
|
|
return nil, nil
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
sstore.AddLineUpdate(update, lineObj, nil)
|
|
return update, nil
|
|
}
|
|
|
|
func LineDeleteCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/line:delete requires at least one argument (line number or id)")
|
|
}
|
|
var lineIds []string
|
|
for _, lineArg := range pk.Args {
|
|
lineId, err := sstore.FindLineIdByArg(ctx, ids.ScreenId, lineArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error looking up lineid: %v", err)
|
|
}
|
|
if lineId == "" {
|
|
return nil, fmt.Errorf("line %q not found", lineArg)
|
|
}
|
|
lineIds = append(lineIds, lineId)
|
|
}
|
|
err = sstore.DeleteLinesByIds(ctx, ids.ScreenId, lineIds)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:delete error deleting lines: %v", err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
for _, lineId := range lineIds {
|
|
line := &sstore.LineType{ScreenId: ids.ScreenId, LineId: lineId, Remove: true}
|
|
sstore.AddLineUpdate(update, line, nil)
|
|
}
|
|
screen, err := sstore.FixupScreenSelectedLine(ctx, ids.ScreenId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/line:delete error fixing up screen: %v", err)
|
|
}
|
|
if screen != nil {
|
|
update.AddUpdate(*screen)
|
|
}
|
|
return update, nil
|
|
}
|
|
|
|
func LineShowCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/line:show requires an argument (line number or id)")
|
|
}
|
|
lineArg := pk.Args[0]
|
|
lineId, err := sstore.FindLineIdByArg(ctx, ids.ScreenId, lineArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error looking up lineid: %v", err)
|
|
}
|
|
if lineId == "" {
|
|
return nil, fmt.Errorf("line %q not found", lineArg)
|
|
}
|
|
line, cmd, err := sstore.GetLineCmdByLineId(ctx, ids.ScreenId, lineId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting line: %v", err)
|
|
}
|
|
if line == nil {
|
|
return nil, fmt.Errorf("line %q not found", lineArg)
|
|
}
|
|
var buf bytes.Buffer
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "screenid", line.ScreenId))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "lineid", line.LineId))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "type", line.LineType))
|
|
lineNumStr := strconv.FormatInt(line.LineNum, 10)
|
|
if line.LineNumTemp {
|
|
lineNumStr = "~" + lineNumStr
|
|
}
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "linenum", lineNumStr))
|
|
ts := time.UnixMilli(line.Ts)
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "ts", ts.Format(TsFormatStr)))
|
|
if line.Ephemeral {
|
|
buf.WriteString(fmt.Sprintf(" %-15s %v\n", "ephemeral", true))
|
|
}
|
|
if line.Renderer != "" {
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "renderer", line.Renderer))
|
|
} else {
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "renderer", "terminal"))
|
|
}
|
|
if cmd != nil {
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "remote", cmd.Remote.MakeFullRemoteRef()))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "status", cmd.Status))
|
|
if cmd.FeState["cwd"] != "" {
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "cwd", cmd.FeState["cwd"]))
|
|
}
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "termopts", formatTermOpts(cmd.TermOpts)))
|
|
if cmd.TermOpts != cmd.OrigTermOpts {
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "orig-termopts", formatTermOpts(cmd.OrigTermOpts)))
|
|
}
|
|
if cmd.RtnState {
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "rtnstate", "true"))
|
|
}
|
|
stat, _ := sstore.StatCmdPtyFile(ctx, cmd.ScreenId, cmd.LineId)
|
|
if stat == nil {
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "file", "-"))
|
|
} else {
|
|
fileDataStr := fmt.Sprintf("v%d data=%d offset=%d max=%s", stat.Version, stat.DataSize, stat.FileOffset, scbase.NumFormatB2(stat.MaxSize))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "file", stat.Location))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "file-data", fileDataStr))
|
|
}
|
|
if cmd.RestartTs > 0 {
|
|
restartTs := time.UnixMilli(cmd.RestartTs)
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "restartts", restartTs.Format(TsFormatStr)))
|
|
}
|
|
if cmd.DoneTs != 0 {
|
|
doneTs := time.UnixMilli(cmd.DoneTs)
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "donets", doneTs.Format(TsFormatStr)))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %d\n", "exitcode", cmd.ExitCode))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %dms\n", "duration", cmd.DurationMs))
|
|
}
|
|
}
|
|
stateStr := dbutil.QuickJson(line.LineState)
|
|
if len(stateStr) > 80 {
|
|
stateStr = stateStr[0:77] + "..."
|
|
}
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "state", stateStr))
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoTitle: fmt.Sprintf("line %d info", line.LineNum),
|
|
InfoLines: splitLinesForInfo(buf.String()),
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func SetCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
var setMap map[string]map[string]string
|
|
setMap = make(map[string]map[string]string)
|
|
_, err := resolveUiIds(ctx, pk, 0) // best effort
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for argIdx, rawArgVal := range pk.Args {
|
|
eqIdx := strings.Index(rawArgVal, "=")
|
|
if eqIdx == -1 {
|
|
return nil, fmt.Errorf("/set invalid argument %d, does not contain an '='", argIdx)
|
|
}
|
|
argName := rawArgVal[:eqIdx]
|
|
argVal := rawArgVal[eqIdx+1:]
|
|
ok, scopeName, varName := resolveSetArg(argName)
|
|
if !ok {
|
|
return nil, fmt.Errorf("/set invalid setvar %q", argName)
|
|
}
|
|
if _, ok := setMap[scopeName]; !ok {
|
|
setMap[scopeName] = make(map[string]string)
|
|
}
|
|
setMap[scopeName][varName] = argVal
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func makeStreamFilePk(ids resolvedIds, pk *scpacket.FeCommandPacketType) (*packet.StreamFilePacketType, error) {
|
|
cwd := ids.Remote.FeState["cwd"]
|
|
fileArg := pk.Args[0]
|
|
if fileArg == "" {
|
|
return nil, fmt.Errorf("/view:stat file argument must be set (cannot be empty)")
|
|
}
|
|
streamPk := packet.MakeStreamFilePacket()
|
|
streamPk.ReqId = uuid.New().String()
|
|
if filepath.IsAbs(fileArg) {
|
|
streamPk.Path = fileArg
|
|
} else {
|
|
streamPk.Path = filepath.Join(cwd, fileArg)
|
|
}
|
|
return streamPk, nil
|
|
}
|
|
|
|
func ViewStatCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/view:stat requires an argument (file name)")
|
|
}
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
streamPk, err := makeStreamFilePk(ids, pk)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
streamPk.StatOnly = true
|
|
msh := ids.Remote.MShell
|
|
iter, err := msh.StreamFile(ctx, streamPk)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/view:stat error: %v", err)
|
|
}
|
|
defer iter.Close()
|
|
respIf, err := iter.Next(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/view:stat error getting response: %v", err)
|
|
}
|
|
resp, ok := respIf.(*packet.StreamFileResponseType)
|
|
if !ok {
|
|
return nil, fmt.Errorf("/view:stat error, bad response packet type: %T", respIf)
|
|
}
|
|
if resp.Error != "" {
|
|
return nil, fmt.Errorf("/view:stat error: %s", resp.Error)
|
|
}
|
|
if resp.Info == nil {
|
|
return nil, fmt.Errorf("/view:stat error, no file info")
|
|
}
|
|
var buf bytes.Buffer
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "path", resp.Info.Name))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %d\n", "size", resp.Info.Size))
|
|
modTs := time.UnixMilli(resp.Info.ModTs)
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "modts", modTs.Format(TsFormatStr)))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %v\n", "isdir", resp.Info.IsDir))
|
|
modeStr := fs.FileMode(resp.Info.Perm).String()
|
|
if len(modeStr) > 9 {
|
|
modeStr = modeStr[len(modeStr)-9:]
|
|
}
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "perms", modeStr))
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoTitle: fmt.Sprintf("view stat %q", streamPk.Path),
|
|
InfoLines: splitLinesForInfo(buf.String()),
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func ViewTestCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/view:test requires an argument (file name)")
|
|
}
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
streamPk, err := makeStreamFilePk(ids, pk)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
msh := ids.Remote.MShell
|
|
iter, err := msh.StreamFile(ctx, streamPk)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/view:test error: %v", err)
|
|
}
|
|
defer iter.Close()
|
|
respIf, err := iter.Next(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/view:test error getting response: %v", err)
|
|
}
|
|
resp, ok := respIf.(*packet.StreamFileResponseType)
|
|
if !ok {
|
|
return nil, fmt.Errorf("/view:test error, bad response packet type: %T", respIf)
|
|
}
|
|
if resp.Error != "" {
|
|
return nil, fmt.Errorf("/view:test error: %s", resp.Error)
|
|
}
|
|
if resp.Info == nil {
|
|
return nil, fmt.Errorf("/view:test error, no file info")
|
|
}
|
|
var buf bytes.Buffer
|
|
var numPackets int
|
|
for {
|
|
dataPkIf, err := iter.Next(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/view:test error while getting data: %w", err)
|
|
}
|
|
if dataPkIf == nil {
|
|
break
|
|
}
|
|
dataPk, ok := dataPkIf.(*packet.FileDataPacketType)
|
|
if !ok {
|
|
return nil, fmt.Errorf("/view:test invalid data packet type: %T", dataPkIf)
|
|
}
|
|
if dataPk.Error != "" {
|
|
return nil, fmt.Errorf("/view:test error returned while getting data: %s", dataPk.Error)
|
|
}
|
|
numPackets++
|
|
buf.Write(dataPk.Data)
|
|
}
|
|
buf.WriteString(fmt.Sprintf("\n\ntotal packets: %d\n", numPackets))
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoTitle: fmt.Sprintf("view file %q", streamPk.Path),
|
|
InfoLines: splitLinesForInfo(buf.String()),
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func CodeEditCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("%s requires an argument (file name)", GetCmdStr(pk))
|
|
}
|
|
// TODO more error checking on filename format?
|
|
if pk.Args[0] == "" {
|
|
return nil, fmt.Errorf("%s argument cannot be empty", GetCmdStr(pk))
|
|
}
|
|
langArg, err := getLangArg(pk)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("%s invalid 'lang': %v", GetCmdStr(pk), err)
|
|
}
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
outputStr := fmt.Sprintf("%s %q", GetCmdStr(pk), pk.Args[0])
|
|
cmd, err := makeStaticCmd(ctx, GetCmdStr(pk), ids, pk.GetRawStr(), []byte(outputStr))
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
// set the line state
|
|
lineState := make(map[string]any)
|
|
lineState[sstore.LineState_Source] = "file"
|
|
lineState[sstore.LineState_File] = pk.Args[0]
|
|
if GetCmdStr(pk) == "codeview" {
|
|
lineState[sstore.LineState_Mode] = "view"
|
|
} else {
|
|
lineState[sstore.LineState_Mode] = "edit"
|
|
}
|
|
if langArg != "" {
|
|
lineState[sstore.LineState_Lang] = langArg
|
|
}
|
|
update, err := addLineForCmd(ctx, "/"+GetCmdStr(pk), true, ids, cmd, "code", lineState)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(sstore.InteractiveUpdate(pk.Interactive))
|
|
return update, nil
|
|
}
|
|
|
|
func CSVViewCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("%s requires an argument (file name)", GetCmdStr(pk))
|
|
}
|
|
// TODO more error checking on filename format?
|
|
if pk.Args[0] == "" {
|
|
return nil, fmt.Errorf("%s argument cannot be empty", GetCmdStr(pk))
|
|
}
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
outputStr := fmt.Sprintf("%s %q", GetCmdStr(pk), pk.Args[0])
|
|
cmd, err := makeStaticCmd(ctx, GetCmdStr(pk), ids, pk.GetRawStr(), []byte(outputStr))
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
// set the line state
|
|
lineState := make(map[string]any)
|
|
lineState[sstore.LineState_Source] = "file"
|
|
lineState[sstore.LineState_File] = pk.Args[0]
|
|
update, err := addLineForCmd(ctx, "/"+GetCmdStr(pk), true, ids, cmd, "csv", lineState)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(sstore.InteractiveUpdate(pk.Interactive))
|
|
return update, nil
|
|
}
|
|
|
|
func ImageViewCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("%s requires an argument (file name)", GetCmdStr(pk))
|
|
}
|
|
// TODO more error checking on filename format?
|
|
if pk.Args[0] == "" {
|
|
return nil, fmt.Errorf("%s argument cannot be empty", GetCmdStr(pk))
|
|
}
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
outputStr := fmt.Sprintf("%s %q", GetCmdStr(pk), pk.Args[0])
|
|
cmd, err := makeStaticCmd(ctx, GetCmdStr(pk), ids, pk.GetRawStr(), []byte(outputStr))
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
// set the line state
|
|
lineState := make(map[string]any)
|
|
lineState[sstore.LineState_Source] = "file"
|
|
lineState[sstore.LineState_File] = pk.Args[0]
|
|
update, err := addLineForCmd(ctx, "/"+GetCmdStr(pk), false, ids, cmd, "image", lineState)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(sstore.InteractiveUpdate(pk.Interactive))
|
|
return update, nil
|
|
}
|
|
|
|
func MarkdownViewCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("%s requires an argument (file name)", GetCmdStr(pk))
|
|
}
|
|
// TODO more error checking on filename format?
|
|
if pk.Args[0] == "" {
|
|
return nil, fmt.Errorf("%s argument cannot be empty", GetCmdStr(pk))
|
|
}
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
outputStr := fmt.Sprintf("%s %q", GetCmdStr(pk), pk.Args[0])
|
|
cmd, err := makeStaticCmd(ctx, GetCmdStr(pk), ids, pk.GetRawStr(), []byte(outputStr))
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
// set the line state
|
|
lineState := make(map[string]any)
|
|
lineState[sstore.LineState_Source] = "file"
|
|
lineState[sstore.LineState_File] = pk.Args[0]
|
|
update, err := addLineForCmd(ctx, "/"+GetCmdStr(pk), false, ids, cmd, "markdown", lineState)
|
|
if err != nil {
|
|
// TODO tricky error since the command was a success, but we can't show the output
|
|
return nil, err
|
|
}
|
|
update.AddUpdate(sstore.InteractiveUpdate(pk.Interactive))
|
|
return update, nil
|
|
}
|
|
|
|
func EditTestCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/edit:test requires an argument (file name)")
|
|
}
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_RemoteConnected)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
content, ok := pk.Kwargs["content"]
|
|
if !ok {
|
|
return nil, fmt.Errorf("/edit:test no content for file specified")
|
|
}
|
|
fileArg := pk.Args[0]
|
|
if fileArg == "" {
|
|
return nil, fmt.Errorf("/view:stat file argument must be set (cannot be empty)")
|
|
}
|
|
writePk := packet.MakeWriteFilePacket()
|
|
writePk.ReqId = uuid.New().String()
|
|
writePk.UseTemp = true
|
|
cwd := ids.Remote.FeState["cwd"]
|
|
if filepath.IsAbs(fileArg) {
|
|
writePk.Path = fileArg
|
|
} else {
|
|
writePk.Path = filepath.Join(cwd, fileArg)
|
|
}
|
|
msh := ids.Remote.MShell
|
|
iter, err := msh.PacketRpcIter(ctx, writePk)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/edit:test error: %v", err)
|
|
}
|
|
// first packet should be WriteFileReady
|
|
readyIf, err := iter.Next(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/edit:test error while getting ready response: %w", err)
|
|
}
|
|
readyPk, ok := readyIf.(*packet.WriteFileReadyPacketType)
|
|
if !ok {
|
|
return nil, fmt.Errorf("/edit:test bad ready packet received: %T", readyIf)
|
|
}
|
|
if readyPk.Error != "" {
|
|
return nil, fmt.Errorf("/edit:test %s", readyPk.Error)
|
|
}
|
|
dataPk := packet.MakeFileDataPacket(writePk.ReqId)
|
|
dataPk.Data = []byte(content)
|
|
dataPk.Eof = true
|
|
err = msh.SendFileData(dataPk)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/edit:test error sending data packet: %v", err)
|
|
}
|
|
doneIf, err := iter.Next(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("/edit:test error while getting done response: %w", err)
|
|
}
|
|
donePk, ok := doneIf.(*packet.WriteFileDonePacketType)
|
|
if !ok {
|
|
return nil, fmt.Errorf("/edit:test bad done packet received: %T", doneIf)
|
|
}
|
|
if donePk.Error != "" {
|
|
return nil, fmt.Errorf("/edit:test %s", donePk.Error)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoTitle: fmt.Sprintf("edit test, wrote %q", writePk.Path),
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func SignalCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(pk.Args) == 0 {
|
|
return nil, fmt.Errorf("/signal requires a first argument (line number or id)")
|
|
}
|
|
if len(pk.Args) == 1 {
|
|
return nil, fmt.Errorf("/signal requires a second argument (signal name)")
|
|
}
|
|
lineArg := pk.Args[0]
|
|
lineId, err := sstore.FindLineIdByArg(ctx, ids.ScreenId, lineArg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error looking up lineid: %v", err)
|
|
}
|
|
line, cmd, err := sstore.GetLineCmdByLineId(ctx, ids.ScreenId, lineId)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting line: %v", err)
|
|
}
|
|
if line == nil {
|
|
return nil, fmt.Errorf("line %q not found", lineArg)
|
|
}
|
|
if cmd == nil {
|
|
return nil, fmt.Errorf("line %q does not have a command", lineArg)
|
|
}
|
|
if cmd.Status != sstore.CmdStatusRunning {
|
|
return nil, fmt.Errorf("line %q command is not running, cannot send signal", lineArg)
|
|
}
|
|
sigArg := pk.Args[1]
|
|
if isAllDigits(sigArg) {
|
|
val, _ := strconv.Atoi(sigArg)
|
|
if val <= 0 || val > MaxSignalNum {
|
|
return nil, fmt.Errorf("signal number is out of bounds: %q", sigArg)
|
|
}
|
|
} else if !strings.HasPrefix(sigArg, "SIG") {
|
|
sigArg = "SIG" + sigArg
|
|
}
|
|
sigArg = strings.ToUpper(sigArg)
|
|
if len(sigArg) > 12 {
|
|
return nil, fmt.Errorf("invalid signal (too long): %q", sigArg)
|
|
}
|
|
if !sigNameRe.MatchString(sigArg) {
|
|
return nil, fmt.Errorf("invalid signal name/number: %q", sigArg)
|
|
}
|
|
msh := remote.GetRemoteById(cmd.Remote.RemoteId)
|
|
if msh == nil {
|
|
return nil, fmt.Errorf("cannot send signal, no remote found for command")
|
|
}
|
|
if !msh.IsConnected() {
|
|
return nil, fmt.Errorf("cannot send signal, remote is not connected")
|
|
}
|
|
siPk := packet.MakeSpecialInputPacket()
|
|
siPk.CK = base.MakeCommandKey(cmd.ScreenId, cmd.LineId)
|
|
siPk.SigName = sigArg
|
|
err = msh.SendSpecialInput(siPk)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot send signal: %v", err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgUpdate("sent line %s signal %s", lineArg, sigArg))
|
|
return update, nil
|
|
}
|
|
|
|
func KillServerCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
go func() {
|
|
log.Printf("received /killserver, shutting down\n")
|
|
time.Sleep(1 * time.Second)
|
|
syscall.Kill(syscall.Getpid(), syscall.SIGINT)
|
|
}()
|
|
return nil, nil
|
|
}
|
|
|
|
func DumpStateCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
ids, err := resolveUiIds(ctx, pk, R_Session|R_Screen|R_Remote)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
currentState, err := sstore.GetFullState(ctx, *ids.Remote.StatePtr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting state: %v", err)
|
|
}
|
|
feState := sstore.FeStateFromShellState(currentState)
|
|
shellenv.DumpVarMapFromState(currentState)
|
|
return sstore.InfoMsgUpdate("current connection state sent to log. festate: %s", dbutil.QuickJson(feState)), nil
|
|
}
|
|
|
|
func ClientCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
return nil, fmt.Errorf("/client requires a subcommand: %s", formatStrs([]string{"show", "set"}, "or", false))
|
|
}
|
|
|
|
func ClientNotifyUpdateWriterCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
pcloud.ResetUpdateWriterNumFailures()
|
|
sstore.NotifyUpdateWriter()
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgUpdate("notified update writer"))
|
|
return update, nil
|
|
}
|
|
|
|
func boolToStr(v bool, trueStr string, falseStr string) string {
|
|
if v {
|
|
return trueStr
|
|
}
|
|
return falseStr
|
|
}
|
|
|
|
func ClientAcceptTosCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
clientOpts := clientData.ClientOpts
|
|
clientOpts.AcceptedTos = time.Now().UnixMilli()
|
|
err = sstore.SetClientOpts(ctx, clientOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client data: %v", err)
|
|
}
|
|
clientData, err = sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve updated client data: %v", err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*clientData)
|
|
return update, nil
|
|
}
|
|
|
|
var confirmKeyRe = regexp.MustCompile(`^[a-z][a-z0-9_]*$`)
|
|
|
|
// confirm flags must be all lowercase and only contain letters, numbers, and underscores (and start with letter)
|
|
func ClientConfirmFlagCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
// Check for valid arguments length
|
|
if len(pk.Args) < 2 {
|
|
return nil, fmt.Errorf("invalid arguments: expected at least 2, got %d", len(pk.Args))
|
|
}
|
|
|
|
// Extract confirmKey and value from pk.Args
|
|
confirmKey := pk.Args[0]
|
|
if !confirmKeyRe.MatchString(confirmKey) {
|
|
return nil, fmt.Errorf("invalid confirm flag key: %s", confirmKey)
|
|
}
|
|
value := resolveBool(pk.Args[1], true)
|
|
validKey := utilfn.ContainsStr(ConfirmFlags, confirmKey)
|
|
if !validKey {
|
|
return nil, fmt.Errorf("invalid confirm flag key: %s", confirmKey)
|
|
}
|
|
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
|
|
// Initialize ConfirmFlags if it's nil
|
|
if clientData.ClientOpts.ConfirmFlags == nil {
|
|
clientData.ClientOpts.ConfirmFlags = make(map[string]bool)
|
|
}
|
|
|
|
// Set the confirm flag
|
|
clientData.ClientOpts.ConfirmFlags[confirmKey] = value
|
|
|
|
err = sstore.SetClientOpts(ctx, clientData.ClientOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client data: %v", err)
|
|
}
|
|
|
|
// Retrieve updated client data
|
|
clientData, err = sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve updated client data: %v", err)
|
|
}
|
|
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*clientData)
|
|
|
|
return update, nil
|
|
}
|
|
|
|
func ClientSetGlobalShortcut(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
newShortcut := firstArg(pk)
|
|
if len(newShortcut) > 50 {
|
|
return nil, fmt.Errorf("invalid shortcut (maxlen = 50)")
|
|
}
|
|
clientOpts := clientData.ClientOpts
|
|
clientOpts.GlobalShortcut = newShortcut
|
|
clientOpts.GlobalShortcutEnabled = (newShortcut != "")
|
|
err = sstore.SetClientOpts(ctx, clientOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client data: %v", err)
|
|
}
|
|
clientData.ClientOpts = clientOpts
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*clientData)
|
|
return update, nil
|
|
}
|
|
|
|
func ClientSetSidebarCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
|
|
// Handle collapsed
|
|
collapsed, ok := pk.Kwargs["collapsed"]
|
|
if !ok {
|
|
return nil, fmt.Errorf("collapsed key not provided")
|
|
}
|
|
collapsedValue := resolveBool(collapsed, false)
|
|
|
|
// Handle width
|
|
var width int
|
|
if w, exists := pk.Kwargs["width"]; exists {
|
|
width, err = resolveNonNegInt(w, 0)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error resolving width: %v", err)
|
|
}
|
|
} else if clientData.ClientOpts.MainSidebar != nil {
|
|
width = clientData.ClientOpts.MainSidebar.Width
|
|
}
|
|
|
|
// Initialize SidebarCollapsed if it's nil
|
|
if clientData.ClientOpts.MainSidebar == nil {
|
|
clientData.ClientOpts.MainSidebar = new(sstore.SidebarValueType)
|
|
}
|
|
|
|
// Set the sidebar values
|
|
var sv sstore.SidebarValueType
|
|
sv.Collapsed = collapsedValue
|
|
if width != 0 {
|
|
sv.Width = width
|
|
}
|
|
clientData.ClientOpts.MainSidebar = &sv
|
|
|
|
// Update client data
|
|
err = sstore.SetClientOpts(ctx, clientData.ClientOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client data: %v", err)
|
|
}
|
|
|
|
// Retrieve updated client data
|
|
clientData, err = sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve updated client data: %v", err)
|
|
}
|
|
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*clientData)
|
|
|
|
return update, nil
|
|
}
|
|
|
|
func validateOpenAIAPIToken(key string) error {
|
|
if len(key) > MaxOpenAIAPITokenLen {
|
|
return fmt.Errorf("invalid openai token, too long")
|
|
}
|
|
for idx, ch := range key {
|
|
if !unicode.IsPrint(ch) {
|
|
return fmt.Errorf("invalid openai token, char at idx:%d is invalid %q", idx, string(ch))
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func validateOpenAIModel(model string) error {
|
|
if len(model) == 0 {
|
|
return nil
|
|
}
|
|
if len(model) > MaxOpenAIModelLen {
|
|
return fmt.Errorf("invalid openai model, too long")
|
|
}
|
|
for idx, ch := range model {
|
|
if !unicode.IsPrint(ch) {
|
|
return fmt.Errorf("invalid openai model, char at idx:%d is invalid %q", idx, string(ch))
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ClientSetCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
var varsUpdated []string
|
|
if fontSizeStr, found := pk.Kwargs["termfontsize"]; found {
|
|
newFontSize, err := resolveNonNegInt(fontSizeStr, 0)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid termfontsize, must be a number between 8-15: %v", err)
|
|
}
|
|
if newFontSize < TermFontSizeMin || newFontSize > TermFontSizeMax {
|
|
return nil, fmt.Errorf("invalid termfontsize, must be a number between %d-%d", TermFontSizeMin, TermFontSizeMax)
|
|
}
|
|
feOpts := clientData.FeOpts
|
|
feOpts.TermFontSize = newFontSize
|
|
err = sstore.UpdateClientFeOpts(ctx, feOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client feopts: %v", err)
|
|
}
|
|
varsUpdated = append(varsUpdated, "termfontsize")
|
|
}
|
|
if apiToken, found := pk.Kwargs["openaiapitoken"]; found {
|
|
err = validateOpenAIAPIToken(apiToken)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
varsUpdated = append(varsUpdated, "openaiapitoken")
|
|
aiOpts := clientData.OpenAIOpts
|
|
if aiOpts == nil {
|
|
aiOpts = &sstore.OpenAIOptsType{}
|
|
clientData.OpenAIOpts = aiOpts
|
|
}
|
|
aiOpts.APIToken = apiToken
|
|
err = sstore.UpdateClientOpenAIOpts(ctx, *aiOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client openai api token: %v", err)
|
|
}
|
|
}
|
|
if aiModel, found := pk.Kwargs["openaimodel"]; found {
|
|
err = validateOpenAIModel(aiModel)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
varsUpdated = append(varsUpdated, "openaimodel")
|
|
aiOpts := clientData.OpenAIOpts
|
|
if aiOpts == nil {
|
|
aiOpts = &sstore.OpenAIOptsType{}
|
|
clientData.OpenAIOpts = aiOpts
|
|
}
|
|
aiOpts.Model = aiModel
|
|
err = sstore.UpdateClientOpenAIOpts(ctx, *aiOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client openai model: %v", err)
|
|
}
|
|
}
|
|
if maxTokensStr, found := pk.Kwargs["openaimaxtokens"]; found {
|
|
maxTokens, err := strconv.Atoi(maxTokensStr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client openai maxtokens, invalid number: %v", err)
|
|
}
|
|
if maxTokens < 0 || maxTokens > 1000000 {
|
|
return nil, fmt.Errorf("error updating client openai maxtokens, out of range: %d", maxTokens)
|
|
}
|
|
varsUpdated = append(varsUpdated, "openaimaxtokens")
|
|
aiOpts := clientData.OpenAIOpts
|
|
if aiOpts == nil {
|
|
aiOpts = &sstore.OpenAIOptsType{}
|
|
clientData.OpenAIOpts = aiOpts
|
|
}
|
|
aiOpts.MaxTokens = maxTokens
|
|
err = sstore.UpdateClientOpenAIOpts(ctx, *aiOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client openai maxtokens: %v", err)
|
|
}
|
|
}
|
|
if maxChoicesStr, found := pk.Kwargs["openaimaxchoices"]; found {
|
|
maxChoices, err := strconv.Atoi(maxChoicesStr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client openai maxchoices, invalid number: %v", err)
|
|
}
|
|
if maxChoices < 0 || maxChoices > 10 {
|
|
return nil, fmt.Errorf("error updating client openai maxchoices, out of range: %d", maxChoices)
|
|
}
|
|
varsUpdated = append(varsUpdated, "openaimaxchoices")
|
|
aiOpts := clientData.OpenAIOpts
|
|
if aiOpts == nil {
|
|
aiOpts = &sstore.OpenAIOptsType{}
|
|
clientData.OpenAIOpts = aiOpts
|
|
}
|
|
aiOpts.MaxChoices = maxChoices
|
|
err = sstore.UpdateClientOpenAIOpts(ctx, *aiOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client openai maxchoices: %v", err)
|
|
}
|
|
}
|
|
if aiBaseURL, found := pk.Kwargs["openaibaseurl"]; found {
|
|
aiOpts := clientData.OpenAIOpts
|
|
if aiOpts == nil {
|
|
aiOpts = &sstore.OpenAIOptsType{}
|
|
clientData.OpenAIOpts = aiOpts
|
|
}
|
|
aiOpts.BaseURL = aiBaseURL
|
|
varsUpdated = append(varsUpdated, "openaibaseurl")
|
|
err = sstore.UpdateClientOpenAIOpts(ctx, *aiOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error updating client openai base url: %v", err)
|
|
}
|
|
}
|
|
if len(varsUpdated) == 0 {
|
|
return nil, fmt.Errorf("/client:set requires a value to set: %s", formatStrs([]string{"termfontsize", "openaiapitoken", "openaimodel", "openaibaseurl", "openaimaxtokens", "openaimaxchoices"}, "or", false))
|
|
}
|
|
clientData, err = sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve updated client data: %v", err)
|
|
}
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(*clientData)
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoMsg: fmt.Sprintf("client updated %s", formatStrs(varsUpdated, "and", false)),
|
|
TimeoutMs: 2000,
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func ClientShowCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
dbVersion, err := sstore.GetDBVersion(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve db version: %v\n", err)
|
|
}
|
|
clientVersion := "-"
|
|
if pk.UIContext != nil && pk.UIContext.Build != "" {
|
|
clientVersion = pk.UIContext.Build
|
|
}
|
|
var buf bytes.Buffer
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "userid", clientData.UserId))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "clientid", clientData.ClientId))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "telemetry", boolToStr(clientData.ClientOpts.NoTelemetry, "off", "on")))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "release-check", boolToStr(clientData.ClientOpts.NoReleaseCheck, "off", "on")))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %d\n", "db-version", dbVersion))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "client-version", clientVersion))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s %s\n", "server-version", scbase.WaveVersion, scbase.BuildTime))
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s (%s)\n", "arch", scbase.ClientArch(), scbase.UnameKernelRelease()))
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoTitle: fmt.Sprintf("client info"),
|
|
InfoLines: splitLinesForInfo(buf.String()),
|
|
})
|
|
|
|
return update, nil
|
|
}
|
|
|
|
func TelemetryCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
return nil, fmt.Errorf("/telemetry requires a subcommand: %s", formatStrs([]string{"show", "on", "off", "send"}, "or", false))
|
|
}
|
|
|
|
func setNoTelemetry(ctx context.Context, clientData *sstore.ClientData, noTelemetryVal bool) error {
|
|
clientOpts := clientData.ClientOpts
|
|
clientOpts.NoTelemetry = noTelemetryVal
|
|
err := sstore.SetClientOpts(ctx, clientOpts)
|
|
if err != nil {
|
|
return fmt.Errorf("error trying to update client telemetry: %v", err)
|
|
}
|
|
log.Printf("client no-telemetry setting updated to %v\n", noTelemetryVal)
|
|
go func() {
|
|
cloudCtx, cancelFn := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancelFn()
|
|
err := pcloud.SendNoTelemetryUpdate(cloudCtx, clientOpts.NoTelemetry)
|
|
if err != nil {
|
|
log.Printf("[error] sending no-telemetry update: %v\n", err)
|
|
log.Printf("note that telemetry update has still taken effect locally, and will be respected by the client\n")
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
func TelemetryOnCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
if !clientData.ClientOpts.NoTelemetry {
|
|
return sstore.InfoMsgUpdate("telemetry is already on"), nil
|
|
}
|
|
err = setNoTelemetry(ctx, clientData, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
go func() {
|
|
cloudCtx, cancelFn := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancelFn()
|
|
err := pcloud.SendTelemetry(cloudCtx, false)
|
|
if err != nil {
|
|
// ignore error, but log
|
|
log.Printf("[error] sending telemetry update (in /telemetry:on): %v\n", err)
|
|
}
|
|
}()
|
|
clientData, err = sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve updated client data: %v", err)
|
|
}
|
|
update := sstore.InfoMsgUpdate("telemetry is now on")
|
|
update.AddUpdate(*clientData)
|
|
return update, nil
|
|
}
|
|
|
|
func TelemetryOffCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
if clientData.ClientOpts.NoTelemetry {
|
|
return sstore.InfoMsgUpdate("telemetry is already off"), nil
|
|
}
|
|
err = setNoTelemetry(ctx, clientData, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
clientData, err = sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve updated client data: %v", err)
|
|
}
|
|
update := sstore.InfoMsgUpdate("telemetry is now off")
|
|
update.AddUpdate(*clientData)
|
|
return update, nil
|
|
}
|
|
|
|
func TelemetryShowCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
var buf bytes.Buffer
|
|
buf.WriteString(fmt.Sprintf(" %-15s %s\n", "telemetry", boolToStr(clientData.ClientOpts.NoTelemetry, "off", "on")))
|
|
update := scbus.MakeUpdatePacket()
|
|
update.AddUpdate(sstore.InfoMsgType{
|
|
InfoTitle: fmt.Sprintf("telemetry info"),
|
|
InfoLines: splitLinesForInfo(buf.String()),
|
|
})
|
|
return update, nil
|
|
}
|
|
|
|
func TelemetrySendCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
force := resolveBool(pk.Kwargs["force"], false)
|
|
if clientData.ClientOpts.NoTelemetry && !force {
|
|
return nil, fmt.Errorf("cannot send telemetry, telemetry is off. pass force=1 to force the send, or turn on telemetry with /telemetry:on")
|
|
}
|
|
err = pcloud.SendTelemetry(ctx, force)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to send telemetry: %v", err)
|
|
}
|
|
return sstore.InfoMsgUpdate("telemetry sent"), nil
|
|
}
|
|
|
|
func runReleaseCheck(ctx context.Context, force bool) error {
|
|
rslt, err := releasechecker.CheckNewRelease(ctx, force)
|
|
|
|
if err != nil {
|
|
return fmt.Errorf("error checking for new release: %v", err)
|
|
}
|
|
|
|
if rslt == releasechecker.Failure {
|
|
return fmt.Errorf("error checking for new release, see log for details")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func setNoReleaseCheck(ctx context.Context, clientData *sstore.ClientData, noReleaseCheckValue bool) error {
|
|
clientOpts := clientData.ClientOpts
|
|
clientOpts.NoReleaseCheck = noReleaseCheckValue
|
|
err := sstore.SetClientOpts(ctx, clientOpts)
|
|
if err != nil {
|
|
return fmt.Errorf("error trying to update client releaseCheck setting: %v", err)
|
|
}
|
|
log.Printf("client no-release-check setting updated to %v\n", noReleaseCheckValue)
|
|
return nil
|
|
}
|
|
|
|
func ReleaseCheckOnCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
if !clientData.ClientOpts.NoReleaseCheck {
|
|
return sstore.InfoMsgUpdate("release check is already on"), nil
|
|
}
|
|
err = setNoReleaseCheck(ctx, clientData, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
go func() {
|
|
releaseCheckCtx, cancelFn := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancelFn()
|
|
releaseCheckErr := runReleaseCheck(releaseCheckCtx, true)
|
|
if releaseCheckErr != nil {
|
|
log.Printf("error checking for new release after enabling auto release check: %v\n", releaseCheckErr)
|
|
}
|
|
}()
|
|
|
|
clientData, err = sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve updated client data: %v", err)
|
|
}
|
|
update := sstore.InfoMsgUpdate("automatic release checking is now on")
|
|
update.AddUpdate(*clientData)
|
|
return update, nil
|
|
}
|
|
|
|
func ReleaseCheckOffCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve client data: %v", err)
|
|
}
|
|
if clientData.ClientOpts.NoReleaseCheck {
|
|
return sstore.InfoMsgUpdate("release check is already off"), nil
|
|
}
|
|
err = setNoReleaseCheck(ctx, clientData, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
clientData, err = sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve updated client data: %v", err)
|
|
}
|
|
update := sstore.InfoMsgUpdate("automatic release checking is now off")
|
|
update.AddUpdate(*clientData)
|
|
return update, nil
|
|
}
|
|
|
|
func ReleaseCheckCommand(ctx context.Context, pk *scpacket.FeCommandPacketType) (scbus.UpdatePacket, error) {
|
|
err := runReleaseCheck(ctx, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
clientData, err := sstore.EnsureClientData(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot retrieve updated client data: %v", err)
|
|
}
|
|
|
|
var rsp string
|
|
if semver.Compare(scbase.WaveVersion, clientData.ReleaseInfo.LatestVersion) < 0 {
|
|
rsp = "new release available to download: https://www.waveterm.dev/download"
|
|
} else {
|
|
rsp = "no new release available"
|
|
}
|
|
|
|
update := sstore.InfoMsgUpdate(rsp)
|
|
update.AddUpdate(*clientData)
|
|
return update, nil
|
|
}
|
|
|
|
func formatTermOpts(termOpts sstore.TermOpts) string {
|
|
if termOpts.Cols == 0 {
|
|
return "???"
|
|
}
|
|
rtnStr := fmt.Sprintf("%dx%d", termOpts.Rows, termOpts.Cols)
|
|
if termOpts.FlexRows {
|
|
rtnStr += " flexrows"
|
|
}
|
|
if termOpts.MaxPtySize > 0 {
|
|
rtnStr += " maxbuf=" + scbase.NumFormatB2(termOpts.MaxPtySize)
|
|
}
|
|
return rtnStr
|
|
}
|
|
|
|
type ColMeta struct {
|
|
Title string
|
|
MinCols int
|
|
MaxCols int
|
|
}
|
|
|
|
func toInterfaceArr(sarr []string) []interface{} {
|
|
rtn := make([]interface{}, len(sarr))
|
|
for idx, s := range sarr {
|
|
rtn[idx] = s
|
|
}
|
|
return rtn
|
|
}
|
|
|
|
func formatTextTable(totalCols int, data [][]string, colMeta []ColMeta) []string {
|
|
numCols := len(colMeta)
|
|
maxColLen := make([]int, len(colMeta))
|
|
for i, cm := range colMeta {
|
|
maxColLen[i] = cm.MinCols
|
|
}
|
|
for _, row := range data {
|
|
for i := 0; i < numCols && i < len(row); i++ {
|
|
dlen := len(row[i])
|
|
if dlen > maxColLen[i] {
|
|
maxColLen[i] = dlen
|
|
}
|
|
}
|
|
}
|
|
fmtStr := ""
|
|
for idx, clen := range maxColLen {
|
|
if idx != 0 {
|
|
fmtStr += " "
|
|
}
|
|
fmtStr += fmt.Sprintf("%%%ds", clen)
|
|
}
|
|
var rtn []string
|
|
for _, row := range data {
|
|
sval := fmt.Sprintf(fmtStr, toInterfaceArr(row)...)
|
|
rtn = append(rtn, sval)
|
|
}
|
|
return rtn
|
|
}
|
|
|
|
func isValidInScope(scopeName string, varName string) bool {
|
|
for _, varScope := range SetVarScopes {
|
|
if varScope.ScopeName == scopeName {
|
|
return utilfn.ContainsStr(varScope.VarNames, varName)
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// returns (is-valid, scope, name)
|
|
// TODO write a full resolver to allow for indexed arguments. e.g. session[1].screen[1].screen.pterm="25x80"
|
|
func resolveSetArg(argName string) (bool, string, string) {
|
|
dotIdx := strings.Index(argName, ".")
|
|
if dotIdx == -1 {
|
|
argName = SetVarNameMap[argName]
|
|
dotIdx = strings.Index(argName, ".")
|
|
}
|
|
if argName == "" {
|
|
return false, "", ""
|
|
}
|
|
scopeName := argName[0:dotIdx]
|
|
varName := argName[dotIdx+1:]
|
|
if !isValidInScope(scopeName, varName) {
|
|
return false, "", ""
|
|
}
|
|
return true, scopeName, varName
|
|
}
|