mirror of
https://shylinux.com/x/icebergs
synced 2025-04-27 01:48:27 +08:00
80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
package ssh
|
|
|
|
import (
|
|
"io"
|
|
"strings"
|
|
|
|
ice "shylinux.com/x/icebergs"
|
|
"shylinux.com/x/icebergs/base/ctx"
|
|
"shylinux.com/x/icebergs/base/mdb"
|
|
psh "shylinux.com/x/icebergs/base/ssh"
|
|
"shylinux.com/x/icebergs/base/tcp"
|
|
kit "shylinux.com/x/toolkits"
|
|
)
|
|
|
|
func _ssh_watch(m *ice.Message, h string, output io.Writer, input io.Reader) io.Closer {
|
|
r, w := io.Pipe()
|
|
bio := io.TeeReader(input, w)
|
|
m.Go(func() { io.Copy(output, r) })
|
|
|
|
i, buf := 0, make([]byte, ice.MOD_BUFS)
|
|
m.Go(func() {
|
|
for {
|
|
n, e := bio.Read(buf[i:])
|
|
if e != nil {
|
|
break
|
|
}
|
|
|
|
switch buf[i] {
|
|
case '\r', '\n':
|
|
cmd := strings.TrimSpace(string(buf[:i]))
|
|
m.Cmdy(mdb.INSERT, m.Prefix(CHANNEL), kit.Keys(mdb.HASH, h), mdb.LIST, mdb.TYPE, CMD, mdb.TEXT, cmd)
|
|
i = 0
|
|
default:
|
|
if i += n; i >= ice.MOD_BUFS {
|
|
i = 0
|
|
}
|
|
}
|
|
}
|
|
})
|
|
return r
|
|
}
|
|
|
|
const CHANNEL = "channel"
|
|
|
|
func init() {
|
|
psh.Index.MergeCommands(ice.Commands{
|
|
CHANNEL: {Name: "channel hash id auto", Help: "通道", Actions: ice.MergeActions(ice.Actions{
|
|
ice.CTX_INIT: {Hand: func(m *ice.Message, arg ...string) {
|
|
mdb.HashSelectUpdate(m, mdb.FOREACH, func(value ice.Map) {
|
|
kit.Value(value, mdb.STATUS, tcp.CLOSE)
|
|
})
|
|
}},
|
|
ctx.COMMAND: {Name: "command cmd=pwd", Help: "命令", Hand: func(m *ice.Message, arg ...string) {
|
|
mdb.ZoneInsert(m, m.OptionSimple(mdb.HASH), mdb.TYPE, CMD, mdb.TEXT, m.Option(CMD))
|
|
if w, ok := mdb.HashTarget(m, m.Option(mdb.HASH), nil).(io.Writer); ok {
|
|
w.Write([]byte(m.Option(CMD) + ice.NL))
|
|
m.Sleep300ms()
|
|
}
|
|
}},
|
|
mdb.REPEAT: {Name: "repeat", Help: "执行", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmdy("", ctx.COMMAND, CMD, m.Option(mdb.TEXT))
|
|
}},
|
|
}, mdb.HashAction(mdb.FIELD, "time,hash,status,tty,count,username,hostport")), Hand: func(m *ice.Message, arg ...string) {
|
|
if len(arg) == 0 { // 通道列表
|
|
m.Action(mdb.PRUNES)
|
|
mdb.HashSelect(m, arg...).Tables(func(value ice.Maps) {
|
|
m.PushButton(kit.Select("", ctx.COMMAND, value[mdb.STATUS] == tcp.OPEN), mdb.REMOVE)
|
|
})
|
|
return
|
|
}
|
|
|
|
// 通道命令
|
|
m.Action(ctx.COMMAND)
|
|
m.Fields(len(arg[1:]), "time,id,type,text")
|
|
mdb.ZoneSelect(m, arg...)
|
|
m.PushAction(mdb.REPEAT)
|
|
}},
|
|
})
|
|
}
|