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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
# # Helpers: Exec helpers
# Provides `pretty_system` which Mina uses to parse SSH output, and delegate to
# the appropriate Output helper.
module Mina
module ExecHelpers
# ### pretty_system
# __Internal:__ A pretty version of the default `#system` commands, but
# indents and puts color.
#
# Returns the exit code in integer form.
#
def pretty_system(code)
require 'shellwords'
cmds = Shellwords.shellsplit(code)
coathooks = 0
status =
Tools.popen4(*cmds) do |pid, i, o, e|
# Handle `^C`.
trap("INT") { Sys.handle_sigint(coathooks += 1, pid, self) }
# __In the background,__ make stdin passthru, and stream stderr.
th_err = Sys.stream_stderr!(e) { |str| print_stderr str }
th_in = Sys.stream_stdin! { |chr| i.putc chr }
# __In the foreground,__ stream stdout to the output helper.
Sys.stream_stdout(o) { |ch| print_char ch }
th_err.join
th_in.terminate
end
status.exitstatus
end
# ## Private methods
# Delegate functions, mostly.
module Sys
extend self
# ### Sys.handle_sigint!
# Called when a `^C` is pressed. The param `count` is how many times it's
# been pressed since. Returns nothing.
def handle_sigint(count, pid, this)
puts ""
if count > 1
this.print_status "Mina: SIGINT received again. Force quitting..."
Process.kill "KILL", pid
else
this.print_status "Mina: SIGINT received."
Process.kill "TERM", pid
end
end
# ### Sys.stream_stderr!
# __Internal:__ Read from stderr stream `err` *[0]*, supress expected
# errors *[1]*, and yield. Returns the thread.
def stream_stderr!(err, &blk)
Thread.new do
begin
while str = err.gets #[0]
next if str.include? "bash: no job control in this shell" #[1]
next if str.include? "stdin is not a terminal"
yield str.strip #[2]
end
rescue Interrupt
end
end
end
# ### Sys.stream_stdin!
# __Internal:__ Read from the real stdin stream and pass it onto the given
# stdin stream `i`. Returns the thread.
def stream_stdin!(&blk)
Thread.new do
begin
while (char = STDIN.getbyte rescue nil)
yield char if char
end
rescue Interrupt
# rubinius
rescue SignalException
end
end
end
# ### Sys.stream_stdout
# __Internal:__ Read from given stdout stream `o` and delegate it to the
# output helper.
def stream_stdout(o, &blk)
while str = o.getc
# Ruby 1.8.7 fix
str = str.chr if str.is_a? Fixnum
yield str
end
end
end
end
end
|
