forked from x/icebergs
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package code
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"os"
|
|
"os/exec"
|
|
"sync"
|
|
|
|
"github.com/creack/pty"
|
|
ice "shylinux.com/x/icebergs"
|
|
"shylinux.com/x/icebergs/base/mdb"
|
|
"shylinux.com/x/icebergs/base/web"
|
|
kit "shylinux.com/x/toolkits"
|
|
)
|
|
|
|
const XTERM = "xterm"
|
|
|
|
func init() {
|
|
cache := sync.Map{}
|
|
add := func(m *ice.Message, key string) string {
|
|
cmd := exec.Command("/bin/sh")
|
|
cmd.Env = append(os.Environ(), "TERM=xterm")
|
|
tty, err := pty.Start(cmd)
|
|
m.Assert(err)
|
|
m.Go(func() {
|
|
buf := make([]byte, 1024)
|
|
for {
|
|
if n, e := tty.Read(buf); m.Assert(e) {
|
|
m.PushNotice("grow", base64.StdEncoding.EncodeToString(buf[:n]))
|
|
}
|
|
}
|
|
})
|
|
|
|
cache.Store(key, tty)
|
|
return key
|
|
}
|
|
Index.MergeCommands(ice.Commands{
|
|
XTERM: {Name: "xterm auto", Help: "终端", Actions: ice.MergeAction(ice.Actions{
|
|
ice.CTX_INIT: {Hand: func(m *ice.Message, arg ...string) {
|
|
m.Watch(web.SPACE_STOP, m.PrefixKey())
|
|
}},
|
|
web.SPACE_STOP: {Name: "space.stop", Help: "断开连接", Hand: func(m *ice.Message, arg ...string) {
|
|
|
|
}},
|
|
"input": {Name: "input", Help: "输入", Hand: func(m *ice.Message, arg ...string) {
|
|
if w, ok := cache.Load(m.Option("channel")); ok {
|
|
if w, ok := w.(*os.File); ok {
|
|
w.Write([]byte(arg[0]))
|
|
return
|
|
}
|
|
}
|
|
}},
|
|
}, mdb.HashAction()), Hand: func(m *ice.Message, arg ...string) {
|
|
m.DisplayLocal("", "channel", add(m, kit.Hashs(mdb.UNIQ)))
|
|
}},
|
|
})
|
|
}
|