mirror of
https://shylinux.com/x/icebergs
synced 2025-04-25 17:18:05 +08:00
88 lines
3.3 KiB
Go
88 lines
3.3 KiB
Go
package web
|
|
|
|
import (
|
|
"net/http"
|
|
"path"
|
|
|
|
ice "shylinux.com/x/icebergs"
|
|
"shylinux.com/x/icebergs/base/cli"
|
|
"shylinux.com/x/icebergs/base/ctx"
|
|
"shylinux.com/x/icebergs/base/mdb"
|
|
"shylinux.com/x/icebergs/base/nfs"
|
|
"shylinux.com/x/icebergs/base/web/html"
|
|
kit "shylinux.com/x/toolkits"
|
|
)
|
|
|
|
const TOKEN = "token"
|
|
|
|
func init() {
|
|
Index.MergeCommands(ice.Commands{
|
|
TOKEN: {Help: "令牌桶", Icon: "Keychain.png", Actions: ice.MergeActions(ice.Actions{
|
|
mdb.PRUNES: {Name: "prunes", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmds("").Table(func(value ice.Maps) {
|
|
if value[mdb.STATUS] != "valid" {
|
|
m.Cmd("", mdb.REMOVE, value)
|
|
}
|
|
})
|
|
}},
|
|
}, mdb.HashAction(mdb.SHORT, mdb.UNIQ, mdb.EXPIRE, mdb.MONTH, html.CHECKBOX, ice.TRUE)), Hand: func(m *ice.Message, arg ...string) {
|
|
if mdb.HashSelect(m, arg...); len(arg) > 0 {
|
|
m.EchoScript(kit.MergeURL2(m.Option(ice.MSG_USERWEB), nfs.PS, TOKEN, arg[0]))
|
|
} else {
|
|
now := m.Time()
|
|
list := map[string]bool{}
|
|
m.Table(func(value ice.Maps) {
|
|
key := kit.Keys(value[mdb.TYPE], value[mdb.TEXT])
|
|
if value[mdb.TIME] < now {
|
|
m.Push(mdb.STATUS, mdb.EXPIRED)
|
|
} else if list[key] {
|
|
m.Push(mdb.STATUS, mdb.INVALID)
|
|
} else {
|
|
m.Push(mdb.STATUS, mdb.VALID)
|
|
}
|
|
list[key] = true
|
|
})
|
|
m.Action(mdb.CREATE, mdb.PRUNES)
|
|
}
|
|
}},
|
|
})
|
|
}
|
|
|
|
const (
|
|
DEV_REQUEST_TEXT = "devRequestText"
|
|
DEV_CREATE_TOKEN = "devCreateToken"
|
|
)
|
|
|
|
func DevTokenAction(name, origin string) ice.Actions {
|
|
return ice.Actions{
|
|
DEV_REQUEST_TEXT: {Hand: func(m *ice.Message, arg ...string) { m.Echo(UserHost(m)) }},
|
|
DEV_CREATE_TOKEN: {Hand: func(m *ice.Message, arg ...string) {}},
|
|
mdb.DEV_REQUEST: {Help: "请求", Icon: "bi bi-person-down", Hand: func(m *ice.Message, arg ...string) {
|
|
back := m.Options(ice.MSG_USERWEB, m.Option(ice.MSG_USERHOST)).MergePod("")
|
|
m.ProcessOpen(m.Options(ice.MSG_USERWEB, m.Option(origin), ice.MSG_USERPOD, "").MergePodCmd("", m.ShortKey(),
|
|
ctx.ACTION, mdb.DEV_CHOOSE, cli.BACK, back, cli.DAEMON, m.Option(ice.MSG_DAEMON),
|
|
m.OptionSimple(name), mdb.TEXT, m.Cmdx("", DEV_REQUEST_TEXT),
|
|
))
|
|
}},
|
|
mdb.DEV_CHOOSE: {Hand: func(m *ice.Message, arg ...string) {
|
|
m.EchoInfoButton(kit.JoinWord(m.ShortKey(), m.Cmdx(nfs.CAT, path.Join(nfs.SRC_TEMPLATE, "web.token", "saveto.html")), m.Option(cli.BACK)), mdb.DEV_RESPONSE)
|
|
}},
|
|
mdb.DEV_RESPONSE: {Help: "确认", Hand: func(m *ice.Message, arg ...string) {
|
|
if !m.WarnNotAllow(m.Option(ice.MSG_METHOD) != http.MethodPost) {
|
|
m.ProcessReplace(m.ParseLink(m.Option(cli.BACK)).MergePodCmd("", m.ShortKey(), ctx.ACTION, mdb.DEV_CONFIRM, m.OptionSimple(cli.DAEMON),
|
|
m.OptionSimple(name), TOKEN, m.Cmdx(TOKEN, mdb.CREATE, mdb.TYPE, m.CommandKey(), mdb.NAME, m.Option(ice.MSG_USERNAME), m.OptionSimple(mdb.TEXT))))
|
|
}
|
|
}},
|
|
mdb.DEV_CONFIRM: {Hand: func(m *ice.Message, arg ...string) {
|
|
m.EchoInfoButton(kit.JoinWord(m.ShortKey(), m.Cmdx(nfs.CAT, path.Join(nfs.SRC_TEMPLATE, "web.token", "savefrom.html")), m.Option(name)), mdb.DEV_CREATE)
|
|
}},
|
|
mdb.DEV_CREATE: {Help: "创建", Hand: func(m *ice.Message, arg ...string) {
|
|
if !m.WarnNotAllow(m.Option(ice.MSG_METHOD) != http.MethodPost) {
|
|
defer kit.If(m.Option(cli.DAEMON), func(p string) { m.Cmd(SPACE, p, html.REFRESH) })
|
|
mdb.HashModify(m, m.OptionSimple(name, TOKEN))
|
|
m.Cmdy("", DEV_CREATE_TOKEN).ProcessClose()
|
|
}
|
|
}},
|
|
}
|
|
}
|