forked from x/icebergs
91 lines
2.8 KiB
Go
91 lines
2.8 KiB
Go
package tcp
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
|
|
ice "shylinux.com/x/icebergs"
|
|
"shylinux.com/x/icebergs/base/aaa"
|
|
"shylinux.com/x/icebergs/base/mdb"
|
|
kit "shylinux.com/x/toolkits"
|
|
)
|
|
|
|
func _host_list(m *ice.Message, name string) {
|
|
if ifs, e := net.Interfaces(); m.Assert(e) {
|
|
for _, v := range ifs {
|
|
if name != "" && !strings.Contains(v.Name, name) {
|
|
continue
|
|
}
|
|
if len(v.HardwareAddr.String()) == 0 {
|
|
continue
|
|
}
|
|
if ips, e := v.Addrs(); m.Assert(e) {
|
|
for _, x := range ips {
|
|
ip := strings.Split(x.String(), ice.PS)
|
|
if strings.Contains(ip[0], ice.DF) || len(ip) == 0 {
|
|
continue
|
|
}
|
|
m.Push(mdb.INDEX, v.Index)
|
|
m.Push(mdb.NAME, v.Name)
|
|
m.Push(aaa.IP, ip[0])
|
|
m.Push("mask", ip[1])
|
|
m.Push("hard", v.HardwareAddr.String())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if len(m.Appendv(aaa.IP)) == 0 {
|
|
m.Push(mdb.INDEX, -1)
|
|
m.Push(mdb.NAME, LOCALHOST)
|
|
m.Push(aaa.IP, "127.0.0.1")
|
|
m.Push("mask", "255.0.0.0")
|
|
m.Push("hard", "")
|
|
}
|
|
m.SortInt(mdb.INDEX)
|
|
m.StatusTimeCount()
|
|
}
|
|
|
|
const (
|
|
LOCALHOST = "localhost"
|
|
|
|
ISLOCAL = "islocal"
|
|
PUBLISH = "publish"
|
|
)
|
|
const HOST = "host"
|
|
|
|
func init() {
|
|
Index.MergeCommands(ice.Commands{
|
|
HOST: {Name: "host name auto", Help: "主机", Actions: ice.MergeActions(ice.Actions{
|
|
ice.CTX_INIT: {Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmd("", func(value ice.Maps) { m.Cmd("", aaa.WHITE, LOCALHOST, value[aaa.IP]) })
|
|
}},
|
|
aaa.WHITE: {Name: "white name text", Help: "白名单", Hand: func(m *ice.Message, arg ...string) {
|
|
mdb.HashCreate(m, mdb.TYPE, m.ActionKey(), m.OptionSimple(mdb.NAME, mdb.TEXT))
|
|
}},
|
|
aaa.BLACK: {Name: "black name text", Help: "黑名单", Hand: func(m *ice.Message, arg ...string) {
|
|
mdb.HashCreate(m, mdb.TYPE, m.ActionKey(), m.OptionSimple(mdb.NAME, mdb.TEXT))
|
|
}},
|
|
ISLOCAL: {Hand: func(m *ice.Message, arg ...string) {
|
|
if arg[0] = strings.Split(strings.TrimPrefix(arg[0], "["), "]")[0]; arg[0] == "::1" || strings.HasPrefix(arg[0], "127.") {
|
|
m.Echo(ice.OK)
|
|
} else if mdb.HashSelectField(m, arg[0], mdb.TYPE) == aaa.WHITE {
|
|
m.Echo(ice.OK)
|
|
}
|
|
}},
|
|
PUBLISH: {Hand: func(m *ice.Message, arg ...string) {
|
|
if strings.Contains(arg[0], LOCALHOST) {
|
|
arg[0] = strings.Replace(arg[0], LOCALHOST, m.Cmd("").Append(aaa.IP), 1)
|
|
} else if strings.Contains(arg[0], "127.0.0.1") {
|
|
arg[0] = strings.Replace(arg[0], "127.0.0.1", m.Cmd("").Append(aaa.IP), 1)
|
|
}
|
|
m.Echo(arg[0])
|
|
}},
|
|
}, mdb.HashAction(mdb.SHORT, mdb.TEXT), mdb.ClearHashOnExitAction()), Hand: func(m *ice.Message, arg ...string) {
|
|
_host_list(m, kit.Select("", arg, 0))
|
|
}},
|
|
})
|
|
}
|
|
|
|
func IsLocalHost(m *ice.Message, ip string) bool { return m.Cmdx(HOST, ISLOCAL, ip) == ice.OK }
|
|
func PublishLocalhost(m *ice.Message, url string) string { return m.Cmdx(HOST, PUBLISH, url) }
|