forked from x/icebergs
86 lines
2.7 KiB
Go
86 lines
2.7 KiB
Go
package vim
|
|
|
|
import (
|
|
ice "github.com/shylinux/icebergs"
|
|
"github.com/shylinux/icebergs/base/aaa"
|
|
"github.com/shylinux/icebergs/base/mdb"
|
|
"github.com/shylinux/icebergs/base/tcp"
|
|
"github.com/shylinux/icebergs/base/web"
|
|
kit "github.com/shylinux/toolkits"
|
|
|
|
"io/ioutil"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
SID = "sid"
|
|
ARG = "arg"
|
|
SUB = "sub"
|
|
PWD = "pwd"
|
|
PID = "pid"
|
|
BUF = "buf"
|
|
ROW = "row"
|
|
COL = "col"
|
|
)
|
|
const (
|
|
LOGOUT = "logout"
|
|
LOGIN = "login"
|
|
)
|
|
const SESS = "sess"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{
|
|
Configs: map[string]*ice.Config{
|
|
SESS: {Name: SESS, Help: "会话流", Value: kit.Data(
|
|
kit.MDB_FIELD, "time,hash,status,username,hostname,pid,pwd",
|
|
)},
|
|
},
|
|
Commands: map[string]*ice.Command{
|
|
SESS: {Name: "sess hash auto prunes", Help: "会话流", Action: map[string]*ice.Action{
|
|
mdb.PRUNES: {Name: "prunes", Help: "清理", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Option(mdb.FIELDS, m.Conf(SESS, kit.META_FIELD))
|
|
m.Cmdy(mdb.PRUNES, m.Prefix(SESS), "", mdb.HASH, kit.MDB_STATUS, LOGOUT)
|
|
}},
|
|
}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
m.Option(mdb.FIELDS, kit.Select(m.Conf(SESS, kit.META_FIELD), mdb.DETAIL, len(arg) > 0))
|
|
m.Cmdy(mdb.SELECT, m.Prefix(SESS), "", mdb.HASH, kit.MDB_HASH, arg)
|
|
}},
|
|
|
|
web.LOGIN: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if f, _, e := m.R.FormFile(SUB); e == nil {
|
|
defer f.Close()
|
|
// 文件参数
|
|
if b, e := ioutil.ReadAll(f); e == nil {
|
|
m.Option(SUB, string(b))
|
|
}
|
|
}
|
|
|
|
m.Debug("what %v", m.Option(SID))
|
|
if sid := strings.TrimSpace(m.Option(SID)); m.Option(SID, sid) != "" {
|
|
if msg := m.Cmd(SESS, sid); msg.Append(kit.MDB_STATUS) != "" {
|
|
m.Option(aaa.USERNAME, msg.Append(aaa.USERNAME))
|
|
m.Option(tcp.HOSTNAME, msg.Append(tcp.HOSTNAME))
|
|
} else {
|
|
// 登录失败
|
|
}
|
|
}
|
|
m.Debug("what %v", m.Option(SID))
|
|
m.Render(ice.RENDER_RESULT)
|
|
}},
|
|
"/sess": {Name: "/sess", Help: "会话", Action: map[string]*ice.Action{
|
|
LOGOUT: {Name: "logout", Help: "退出", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmdy(mdb.MODIFY, m.Prefix(SESS), "", mdb.HASH, kit.MDB_HASH, m.Option(SID), kit.MDB_STATUS, LOGOUT)
|
|
}},
|
|
}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if m.Option(SID) == "" { // 新建会话
|
|
m.Option(SID, m.Cmdx(mdb.INSERT, m.Prefix(SESS), "", mdb.HASH, kit.MDB_STATUS, LOGIN,
|
|
aaa.USERNAME, m.Option(aaa.USERNAME), tcp.HOSTNAME, m.Option(tcp.HOSTNAME), PID, m.Option(PID), PWD, m.Option(PWD)))
|
|
} else { // 复用会话
|
|
m.Cmdy(mdb.MODIFY, m.Prefix(SESS), "", mdb.HASH, kit.MDB_HASH, m.Option(SID), kit.MDB_STATUS, LOGIN)
|
|
}
|
|
m.Echo(m.Option(SID))
|
|
}},
|
|
},
|
|
})
|
|
}
|