Upstream/waveterm
1
0
Fork 0
mirror of https://github.com/wavetermdev/waveterm.git synced 2025-01-02 18:39:05 +01:00
Code Issues Packages Projects Releases Wiki Activity

write wsh conn disconnectall

Browse Source
This commit is contained in:
sawka 2024-09-06 15:58:42 -07:00
parent bd43a4b61d
commit b85cfc9d2d
1 changed files with 25 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
cmd/wsh/cmd/wshcmd-conn.go
View File

@ -46,6 +46,25 @@ func connStatus() error {
return nil
}
func connDisconnectAll() error {
resp, err := wshclient.ConnStatusCommand(RpcClient, nil)
if err != nil {
return fmt.Errorf("getting connection status: %w", err)
}
if len(resp) == 0 {
return nil
}
for _, conn := range resp {
if conn.Status == "connected" {
err := connDisconnect(conn.Connection)
if err != nil {
WriteStdout("error disconnecting %q: %v\n", conn.Connection, err)
}
}
}
return nil
}
func connEnsure(connName string) error {
err := wshclient.ConnEnsureCommand(RpcClient, connName, &wshrpc.RpcOpts{Timeout: 60000})
if err != nil {
@ -67,9 +86,9 @@ func connReinstall(connName string) error {
func connDisconnect(connName string) error {
err := wshclient.ConnDisconnectCommand(RpcClient, connName, &wshrpc.RpcOpts{Timeout: 10000})
if err != nil {
return fmt.Errorf("disconnecting connection: %w", err)
return fmt.Errorf("disconnecting %q error: %w", connName, err)
}
WriteStdout("disconnected connection %q\n", connName)
WriteStdout("disconnected %q\n", connName)
return nil
}
@ -85,9 +104,9 @@ func connConnect(connName string) error {
func connRun(cmd *cobra.Command, args []string) error {
connCmd := args[0]
var connName string
if connCmd != "status" {
if connCmd != "status" && connCmd != "disconnectall" {
if len(args) < 2 {
return fmt.Errorf("connection name is required")
return fmt.Errorf("connection name is required %q", connCmd)
}
connName = args[1]
_, err := remote.ParseOpts(connName)
@ -103,6 +122,8 @@ func connRun(cmd *cobra.Command, args []string) error {
return connReinstall(connName)
} else if connCmd == "disconnect" {
return connDisconnect(connName)
} else if connCmd == "disconnectall" {
return connDisconnectAll()
} else if connCmd == "connect" {
return connConnect(connName)
} else {