forked from reiver/go-telnet
/
help.go
120 lines (85 loc) · 2.21 KB
/
help.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
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
112
113
114
115
116
117
118
119
120
package telsh
import (
"github.com/reiver/go-oi"
"github.com/reiver/go-telnet"
"io"
"sort"
)
type internalHelpProducer struct {
shellHandler *ShellHandler
}
func Help(shellHandler *ShellHandler) Producer {
producer := internalHelpProducer{
shellHandler:shellHandler,
}
return &producer
}
func (producer *internalHelpProducer) Produce(telnet.Context, string, ...string) Handler {
return newHelpHandler(producer)
}
type internalHelpHandler struct {
helpProducer *internalHelpProducer
err error
stdin io.ReadCloser
stdout io.WriteCloser
stderr io.WriteCloser
stdinPipe io.WriteCloser
stdoutPipe io.ReadCloser
stderrPipe io.ReadCloser
}
func newHelpHandler(helpProducer *internalHelpProducer) *internalHelpHandler {
stdin, stdinPipe := io.Pipe()
stdoutPipe, stdout := io.Pipe()
stderrPipe, stderr := io.Pipe()
handler := internalHelpHandler{
helpProducer:helpProducer,
err:nil,
stdin:stdin,
stdout:stdout,
stderr:stderr,
stdinPipe:stdinPipe,
stdoutPipe:stdoutPipe,
stderrPipe:stderrPipe,
}
return &handler
}
func (handler *internalHelpHandler) Run() error {
if nil != handler.err {
return handler.err
}
//@TODO: Should this be reaching inside of ShellHandler? Maybe there should be ShellHandler public methods instead.
keys := make([]string, 1+len(handler.helpProducer.shellHandler.producers))
i:=0
for key,_ := range handler.helpProducer.shellHandler.producers {
keys[i] = key
i++
}
keys[i] = handler.helpProducer.shellHandler.ExitCommandName
sort.Strings(keys)
for _, key := range keys {
oi.LongWriteString(handler.stdout, key)
oi.LongWriteString(handler.stdout, "\r\n")
}
handler.stdin.Close()
handler.stdout.Close()
handler.stderr.Close()
return handler.err
}
func (handler *internalHelpHandler) StdinPipe() (io.WriteCloser, error) {
if nil != handler.err {
return nil, handler.err
}
return handler.stdinPipe, nil
}
func (handler *internalHelpHandler) StdoutPipe() (io.ReadCloser, error) {
if nil != handler.err {
return nil, handler.err
}
return handler.stdoutPipe, nil
}
func (handler *internalHelpHandler) StderrPipe() (io.ReadCloser, error) {
if nil != handler.err {
return nil, handler.err
}
return handler.stderrPipe, nil
}