forked from scaleway/scaleway-cli
/
cmd_exec.go
64 lines (57 loc) · 2.13 KB
/
cmd_exec.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
// Copyright (C) 2015 Scaleway. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE.md file.
package cli
import "github.com/scaleway/scaleway-cli/pkg/commands"
var cmdExec = &Command{
Exec: runExec,
UsageLine: "exec [OPTIONS] SERVER [COMMAND] [ARGS...]",
Description: "Run a command on a running server",
Help: "Run a command on a running server.",
Examples: `
$ scw exec myserver
$ scw exec myserver bash
$ scw exec --gateway=myotherserver myserver bash
$ scw exec myserver 'tmux a -t joe || tmux new -s joe || bash'
$ SCW_SECURE_EXEC=1 scw exec myserver bash
$ scw exec -w $(scw start $(scw create ubuntu-trusty)) bash
$ scw exec $(scw start -w $(scw create ubuntu-trusty)) bash
$ scw exec myserver tmux new -d sleep 10
$ scw exec myserver ls -la | grep password
$ cat local-file | scw exec myserver 'cat > remote/path'
`,
}
func init() {
cmdExec.Flag.BoolVar(&execHelp, []string{"h", "-help"}, false, "Print usage")
cmdExec.Flag.Float64Var(&execTimeout, []string{"T", "-timeout"}, 0, "Set timeout values to seconds")
cmdExec.Flag.BoolVar(&execW, []string{"w", "-wait"}, false, "Wait for SSH to be ready")
cmdExec.Flag.StringVar(&execGateway, []string{"g", "-gateway"}, "", "Use a SSH gateway")
cmdExec.Flag.StringVar(&execSSHUser, []string{"-user"}, "root", "Specify SSH user")
cmdExec.Flag.IntVar(&execSSHPort, []string{"-p", "-port"}, 22, "Specify SSH port")
}
// Flags
var execW bool // -w, --wait flag
var execTimeout float64 // -T flag
var execHelp bool // -h, --help flag
var execGateway string // -g, --gateway flag
var execSSHUser string // --user flag
var execSSHPort int // -p, --port flag
func runExec(cmd *Command, rawArgs []string) error {
if execHelp {
return cmd.PrintUsage()
}
if len(rawArgs) < 1 {
return cmd.PrintShortUsage()
}
args := commands.ExecArgs{
Timeout: execTimeout,
Wait: execW,
Gateway: execGateway,
Server: rawArgs[0],
Command: rawArgs[1:],
SSHUser: execSSHUser,
SSHPort: execSSHPort,
}
ctx := cmd.GetContext(rawArgs)
return commands.RunExec(ctx, args)
}