forked from x/icebergs
99 lines
3.3 KiB
Go
99 lines
3.3 KiB
Go
package aaa
|
|
|
|
import (
|
|
ice "shylinux.com/x/icebergs"
|
|
"shylinux.com/x/icebergs/base/gdb"
|
|
"shylinux.com/x/icebergs/base/mdb"
|
|
kit "shylinux.com/x/toolkits"
|
|
"shylinux.com/x/toolkits/logs"
|
|
)
|
|
|
|
func _sess_check(m *ice.Message, sessid string) {
|
|
m.Option(ice.MSG_USERROLE, VOID)
|
|
m.Option(ice.MSG_USERNAME, "")
|
|
m.Option(ice.MSG_USERNICK, "")
|
|
if sessid == "" {
|
|
return
|
|
}
|
|
|
|
_source := logs.FileLineMeta(logs.FileLine(-1, 3))
|
|
mdb.HashSelectDetail(m, sessid, func(value ice.Map) {
|
|
if m.Warn(kit.Time(kit.Format(value[mdb.TIME])) < kit.Time(m.Time()), ice.ErrNotValid, sessid) {
|
|
return // 会话超时
|
|
}
|
|
m.Auth(
|
|
USERROLE, m.Option(ice.MSG_USERROLE, value[USERROLE]),
|
|
USERNAME, m.Option(ice.MSG_USERNAME, value[USERNAME]),
|
|
USERNICK, m.Option(ice.MSG_USERNICK, value[USERNICK]),
|
|
_source,
|
|
)
|
|
})
|
|
}
|
|
func _sess_create(m *ice.Message, username string) (h string) {
|
|
if m.Warn(username == "", ice.ErrNotValid, username) {
|
|
return
|
|
}
|
|
if msg := m.Cmd(USER, username); msg.Length() > 0 {
|
|
h = mdb.HashCreate(m, msg.AppendSimple(USERROLE, USERNAME, USERNICK), IP, m.Option(ice.MSG_USERIP), UA, m.Option(ice.MSG_USERUA))
|
|
} else {
|
|
h = mdb.HashCreate(m, m.OptionSimple(USERROLE, USERNAME, USERNICK), IP, m.Option(ice.MSG_USERIP), UA, m.Option(ice.MSG_USERUA))
|
|
}
|
|
gdb.Event(m, SESS_CREATE, SESS, h, USERNAME, username)
|
|
return h
|
|
}
|
|
|
|
const (
|
|
IP = "ip"
|
|
UA = "ua"
|
|
)
|
|
const (
|
|
CHECK = "check"
|
|
GRANT = "grant"
|
|
LOGIN = "login"
|
|
LOGOUT = "logout"
|
|
SESSID = "sessid"
|
|
)
|
|
const (
|
|
SESS_CREATE = "sess.create"
|
|
)
|
|
const SESS = "sess"
|
|
|
|
func init() {
|
|
Index.MergeCommands(ice.Commands{
|
|
SESS: {Name: "sess hash auto prunes", Help: "会话", Actions: ice.MergeActions(ice.Actions{
|
|
mdb.CREATE: {Name: "create username", Help: "创建", Hand: func(m *ice.Message, arg ...string) {
|
|
_sess_create(m, m.Option(USERNAME))
|
|
}},
|
|
SESS_CREATE: {Name: "sess.create", Help: "事件", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmd(mdb.EXPORT, m.Prefix(SESS), "", mdb.HASH)
|
|
m.Cmd(mdb.IMPORT, m.Prefix(SESS), "", mdb.HASH)
|
|
}},
|
|
USER_CREATE: {Name: "user.create", Help: "检查", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmd(mdb.EXPORT, m.Prefix(USER), "", mdb.HASH)
|
|
m.Cmd(mdb.IMPORT, m.Prefix(USER), "", mdb.HASH)
|
|
}},
|
|
CHECK: {Name: "check sessid", Help: "检查", Hand: func(m *ice.Message, arg ...string) {
|
|
_sess_check(m, m.Option(SESSID))
|
|
}},
|
|
}, mdb.HashAction(mdb.SHORT, mdb.UNIQ, mdb.FIELD, "time,hash,userrole,username,usernick,ip,ua", mdb.EXPIRE, "720h"))},
|
|
})
|
|
}
|
|
|
|
func SessCreate(m *ice.Message, username string) string {
|
|
return m.Option(ice.MSG_SESSID, m.Cmdx(SESS, mdb.CREATE, username))
|
|
}
|
|
func SessCheck(m *ice.Message, sessid string) bool {
|
|
m.Option(ice.MSG_USERROLE, VOID)
|
|
m.Option(ice.MSG_USERNAME, "")
|
|
m.Option(ice.MSG_USERNICK, "")
|
|
return sessid != "" && m.Cmdy(SESS, CHECK, sessid).Option(ice.MSG_USERNAME) != ""
|
|
}
|
|
func UserLogout(m *ice.Message) {
|
|
m.Cmd(SESS, mdb.REMOVE, kit.Dict(mdb.HASH, m.Option(ice.MSG_SESSID)))
|
|
}
|
|
func SessAuth(m *ice.Message, value ice.Maps, arg ...string) {
|
|
m.Option(ice.MSG_USERROLE, value[aaa.USERROLE])
|
|
m.Option(ice.MSG_USERNAME, value[aaa.USERNAME])
|
|
m.Option(ice.MSG_USERNICK, value[aaa.USERNICK])
|
|
m.Auth(aaa.USERROLE, value[aaa.USERROLE], aaa.USERNAME, value[aaa.USERNAME], aaa.USERNICK, value[aaa.USERNICK], arg, logs.FileLineMeta(logs.FileLine(2, 3)))
|
|
} |