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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
module LeapCli; module Commands
desc 'Run a shell command remotely'
long_desc "Runs the specified command COMMAND on each node in the FILTER set. " +
"For example, `leap run 'uname -a' webapp`"
command :run do |c|
c.switch 'stream', :default => false, :desc => 'If set, stream the output as it arrives. (default: --stream for a single node, --no-stream for multiple nodes)'
c.flag 'port', :arg_name => 'SSH_PORT', :desc => 'Override default SSH port used when trying to connect to the server.'
c.desc 'Run an arbitrary shell command.'
c.arg_name 'FILTER', optional: true
c.command :command do |command|
command.action do |global, options, args|
run_shell_command(global, options, args)
end
end
c.desc 'Generate one or more new invite codes.'
c.arg_name '[COUNT] [ENVIRONMENT]'
c.command :invite do |invite|
invite.action do |global_options,options,args|
run_new_invites(global_options, options, args)
end
end
c.default_command :command
end
private
def run_shell_command(global, options, args)
require 'leap_cli/ssh'
cmd = args[0]
filter = args[1..-1]
cmd = global[:force] ? cmd : LeapCli::SSH::Options.sanitize_command(cmd)
nodes = manager.filter!(filter)
if nodes.size == 1 || options[:stream]
stream_command(nodes, cmd, options)
else
capture_command(nodes, cmd, options)
end
end
CMD_NEW_INVITES="cd /srv/leap/webapp; RAILS_ENV=production bundle exec rake \"generate_invites[NUM,USES]\""
def run_new_invites(global, options, args)
require 'leap_cli/ssh'
count = 1
uses = 1
env = nil
arg1 = args.shift
arg2 = args.shift
if arg1 && arg2
env = manager.env(arg2)
count = arg1
elsif arg1
env = manager.env(arg1)
else
env = manager.env(nil)
end
unless env
bail! "Environment name you specified does not match one that is available. See `leap env ls` for the available names"
end
env_name = env.name == 'default' ? nil : env.name
webapp_nodes = env.nodes[:environment => env_name][:services => 'webapp'].first
if webapp_nodes.empty?
bail! "Could not find a webapp node for the specified environment"
end
stream_command(
webapp_nodes,
CMD_NEW_INVITES.sub('NUM', count.to_s).sub('USES', uses.to_s),
options
)
end
def capture_command(nodes, cmd, options)
SSH.remote_command(nodes, options) do |ssh, host|
output = ssh.capture(cmd, :log_output => false)
if output
logger = LeapCli.new_logger
logger.log(:ran, "`" + cmd + "`", host: host.hostname, color: :green) do
logger.log(output, wrap: true)
end
end
end
end
def stream_command(nodes, cmd, options)
SSH.remote_command(nodes, options) do |ssh, host|
ssh.stream(cmd, :log_cmd => true, :log_finish => true, :fail_msg => 'oops')
end
end
end; end
|
