forked from x/icebergs
78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
package ctx
|
|
|
|
import (
|
|
ice "github.com/shylinux/icebergs"
|
|
"github.com/shylinux/icebergs/base/mdb"
|
|
kit "github.com/shylinux/toolkits"
|
|
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
func _command_search(m *ice.Message, kind, name, text string) {
|
|
if !(kind == COMMAND || kind == kit.MDB_FOREACH && name != "") {
|
|
return
|
|
}
|
|
|
|
ice.Pulse.Travel(func(p *ice.Context, s *ice.Context, key string, cmd *ice.Command) {
|
|
if strings.HasPrefix(key, "_") || strings.HasPrefix(key, "/") {
|
|
return
|
|
}
|
|
if name != "" && name != key && name != s.Name {
|
|
return
|
|
}
|
|
|
|
m.PushSearch(kit.SSH_CMD, COMMAND,
|
|
"context", s.Cap(ice.CTX_FOLLOW), "command", key,
|
|
kit.MDB_TYPE, kind, kit.MDB_NAME, key, kit.MDB_TEXT, s.Cap(ice.CTX_FOLLOW),
|
|
)
|
|
})
|
|
}
|
|
func _command_list(m *ice.Message, name string) {
|
|
p := m.Spawn(m.Source())
|
|
if name != "" {
|
|
// 命令详情
|
|
p.Search(name, func(p *ice.Context, s *ice.Context, key string, cmd *ice.Command) {
|
|
m.Push(kit.MDB_KEY, s.Cap(ice.CTX_FOLLOW))
|
|
m.Push(kit.MDB_NAME, kit.Format(cmd.Name))
|
|
m.Push(kit.MDB_HELP, kit.Simple(cmd.Help)[0])
|
|
m.Push(kit.MDB_META, kit.Formats(cmd.Meta))
|
|
m.Push(kit.MDB_LIST, kit.Formats(cmd.List))
|
|
})
|
|
return
|
|
}
|
|
|
|
list := []string{}
|
|
for k := range p.Target().Commands {
|
|
if k[0] == '/' || k[0] == '_' {
|
|
continue // 内部命令
|
|
}
|
|
list = append(list, k)
|
|
}
|
|
sort.Strings(list)
|
|
|
|
// 命令列表
|
|
for _, k := range list {
|
|
v := p.Target().Commands[k]
|
|
m.Push(kit.MDB_KEY, k)
|
|
m.Push(kit.MDB_NAME, kit.Format(v.Name))
|
|
m.Push(kit.MDB_HELP, kit.Simple(v.Help)[0])
|
|
}
|
|
}
|
|
|
|
const COMMAND = "command"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{
|
|
Commands: map[string]*ice.Command{
|
|
COMMAND: {Name: "command key auto", Help: "命令", Action: map[string]*ice.Action{
|
|
mdb.SEARCH: {Name: "search type name text", Help: "搜索", Hand: func(m *ice.Message, arg ...string) {
|
|
_command_search(m, arg[0], arg[1], arg[2])
|
|
}},
|
|
}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
_command_list(m, strings.Join(arg, "."))
|
|
}},
|
|
},
|
|
})
|
|
}
|