forked from x/icebergs
266 lines
8.0 KiB
Go
266 lines
8.0 KiB
Go
package cli
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"runtime"
|
|
"strings"
|
|
|
|
ice "shylinux.com/x/icebergs"
|
|
"shylinux.com/x/icebergs/base/aaa"
|
|
"shylinux.com/x/icebergs/base/ctx"
|
|
"shylinux.com/x/icebergs/base/gdb"
|
|
"shylinux.com/x/icebergs/base/mdb"
|
|
"shylinux.com/x/icebergs/base/nfs"
|
|
kit "shylinux.com/x/toolkits"
|
|
)
|
|
|
|
func _runtime_init(m *ice.Message) {
|
|
// 版本信息 make
|
|
kit.Fetch(kit.UnMarshal(kit.Format(ice.Info.Make)), func(key string, value ice.Any) {
|
|
m.Conf(RUNTIME, kit.Keys(MAKE, strings.ToLower(key)), value)
|
|
})
|
|
|
|
// 主机信息 host
|
|
m.Conf(RUNTIME, kit.Keys(HOST, GOARCH), runtime.GOARCH)
|
|
m.Conf(RUNTIME, kit.Keys(HOST, GOOS), runtime.GOOS)
|
|
m.Conf(RUNTIME, kit.Keys(HOST, PID), os.Getpid())
|
|
m.Conf(RUNTIME, kit.Keys(HOST, HOME), kit.Env(HOME))
|
|
m.Conf(RUNTIME, kit.Keys(HOST, MAXPROCS), runtime.GOMAXPROCS(0))
|
|
|
|
// 启动目录 boot
|
|
m.Conf(RUNTIME, kit.Keys(BOOT, HOSTNAME), kit.Env("HOSTNAME"))
|
|
if name, e := os.Hostname(); e == nil && name != "" {
|
|
m.Conf(RUNTIME, kit.Keys(BOOT, HOSTNAME), name)
|
|
}
|
|
m.Conf(RUNTIME, kit.Keys(BOOT, PATHNAME), path.Base(kit.Env("PWD")))
|
|
if name, e := os.Getwd(); e == nil && name != "" {
|
|
m.Conf(RUNTIME, kit.Keys(BOOT, PATHNAME), path.Base(name))
|
|
}
|
|
m.Conf(RUNTIME, kit.Keys(BOOT, USERNAME), kit.Select(kit.UserName(), kit.Select(kit.Env("WORKSPACE_GIT_USERNAME"), kit.Env(CTX_USER))))
|
|
ice.Info.HostName = m.Conf(RUNTIME, kit.Keys(BOOT, HOSTNAME))
|
|
ice.Info.PathName = m.Conf(RUNTIME, kit.Keys(BOOT, PATHNAME))
|
|
ice.Info.UserName = m.Conf(RUNTIME, kit.Keys(BOOT, USERNAME))
|
|
aaa.UserRoot(ice.Pulse)
|
|
|
|
// 启动程序 boot
|
|
m.Conf(RUNTIME, kit.Keys(BOOT, mdb.COUNT), kit.Int(m.Conf(RUNTIME, kit.Keys(BOOT, mdb.COUNT)))+1)
|
|
bin := _system_find(m, os.Args[0])
|
|
m.Conf(RUNTIME, kit.Keys(BOOT, ice.BIN), bin)
|
|
if s, e := os.Stat(bin); e == nil {
|
|
m.Conf(RUNTIME, kit.Keys(BOOT, nfs.SIZE), kit.FmtSize(s.Size()))
|
|
if f, e := os.Open(bin); e == nil {
|
|
defer f.Close()
|
|
m.Conf(RUNTIME, kit.Keys(BOOT, mdb.HASH), kit.Hashs(f))
|
|
}
|
|
}
|
|
|
|
// 环境变量 conf
|
|
for _, k := range ENV_LIST {
|
|
switch m.Conf(RUNTIME, kit.Keys(CONF, k), kit.Env(k)); k {
|
|
case CTX_PID:
|
|
ice.Info.PidPath = kit.Select("var/run/ice.pid", kit.Env(k))
|
|
case CTX_SHARE:
|
|
ice.Info.CtxShare = kit.Env(k)
|
|
case CTX_RIVER:
|
|
ice.Info.CtxRiver = kit.Env(k)
|
|
}
|
|
}
|
|
|
|
// 系统版本 osid
|
|
osid := runtime.GOOS
|
|
m.Cmd(nfs.CAT, "/etc/os-release", func(text string) {
|
|
if ls := kit.Split(text, "="); len(ls) > 1 {
|
|
switch ls[0] {
|
|
case "ID", "ID_LIKE":
|
|
osid = strings.TrimSpace(ls[1] + ice.SP + osid)
|
|
}
|
|
}
|
|
})
|
|
m.Conf(RUNTIME, kit.Keys(HOST, OSID), osid)
|
|
|
|
switch strings.Split(os.Getenv(TERM), "-")[0] {
|
|
case "xterm", "screen":
|
|
ice.Info.Colors = true
|
|
default:
|
|
ice.Info.Colors = false
|
|
}
|
|
|
|
}
|
|
func _runtime_hostinfo(m *ice.Message) {
|
|
m.Push("nCPU", strings.Count(m.Cmdx(nfs.CAT, "/proc/cpuinfo"), "processor"))
|
|
for i, ls := range strings.Split(m.Cmdx(nfs.CAT, "/proc/meminfo"), ice.NL) {
|
|
vs := kit.Split(ls, ": ")
|
|
if m.Push(strings.TrimSpace(vs[0]), kit.FmtSize(kit.Int64(strings.TrimSpace(vs[1]))*1024)); i > 1 {
|
|
break
|
|
}
|
|
}
|
|
m.Push("uptime", kit.Split(m.Cmdx(SYSTEM, "uptime"), ice.FS)[0])
|
|
}
|
|
func _runtime_diskinfo(m *ice.Message) {
|
|
m.Spawn().Split(kit.Replace(m.Cmdx(SYSTEM, "df", "-h"), "Mounted on", "Mountedon"), "", ice.SP, ice.NL).Table(func(index int, value ice.Maps, head []string) {
|
|
if strings.HasPrefix(value["Filesystem"], "/dev") {
|
|
m.Push("", value, head)
|
|
}
|
|
})
|
|
m.RenameAppend("%iused", "piused", "Use%", "Usep")
|
|
ctx.DisplayStory(m, "pie.js?field=Size")
|
|
}
|
|
|
|
func NodeInfo(m *ice.Message, kind, name string) {
|
|
m.Conf(RUNTIME, kit.Keys(NODE, mdb.TIME), m.Time())
|
|
ice.Info.NodeType = m.Conf(RUNTIME, kit.Keys(NODE, mdb.TYPE), kind)
|
|
ice.Info.NodeName = m.Conf(RUNTIME, kit.Keys(NODE, mdb.NAME), strings.ReplaceAll(name, ice.PT, "_"))
|
|
}
|
|
|
|
const (
|
|
MAKE = "make"
|
|
TEST = "test"
|
|
HOST = "host"
|
|
BOOT = "boot"
|
|
CONF = "conf"
|
|
NODE = "node"
|
|
)
|
|
const (
|
|
GOARCH = "GOARCH"
|
|
AMD64 = "amd64"
|
|
X86 = "386"
|
|
ARM = "arm"
|
|
ARM64 = "arm64"
|
|
MIPSLE = "mipsle"
|
|
|
|
GOOS = "GOOS"
|
|
LINUX = "linux"
|
|
DARWIN = "darwin"
|
|
WINDOWS = "windows"
|
|
)
|
|
const (
|
|
PATH = "PATH"
|
|
HOME = "HOME"
|
|
USER = "USER"
|
|
TERM = "TERM"
|
|
SHELL = "SHELL"
|
|
)
|
|
const (
|
|
CTX_SHY = "ctx_shy"
|
|
CTX_DEV = "ctx_dev"
|
|
CTX_OPS = "ctx_ops"
|
|
CTX_POD = "ctx_pod"
|
|
CTX_ARG = "ctx_arg"
|
|
CTX_PID = "ctx_pid"
|
|
CTX_LOG = "ctx_log"
|
|
|
|
CTX_USER = "ctx_user"
|
|
CTX_SHARE = "ctx_share"
|
|
CTX_RIVER = "ctx_river"
|
|
CTX_DAEMON = "ctx_daemon"
|
|
|
|
MAKE_DOMAIN = "make.domain"
|
|
)
|
|
|
|
var ENV_LIST = []string{
|
|
TERM, SHELL, CTX_SHY, CTX_DEV, CTX_OPS, CTX_ARG, CTX_PID, CTX_USER, CTX_SHARE, CTX_RIVER, CTX_DAEMON,
|
|
}
|
|
|
|
const (
|
|
HOSTNAME = "hostname"
|
|
PATHNAME = "pathname"
|
|
USERNAME = "username"
|
|
)
|
|
const (
|
|
IFCONFIG = "ifconfig"
|
|
HOSTINFO = "hostinfo"
|
|
USERINFO = "userinfo"
|
|
PROCINFO = "procinfo"
|
|
PROCKILL = "prockill"
|
|
DISKINFO = "diskinfo"
|
|
BOOTINFO = "bootinfo"
|
|
MAXPROCS = "maxprocs"
|
|
)
|
|
const RUNTIME = "runtime"
|
|
|
|
func init() {
|
|
Index.MergeCommands(ice.Commands{
|
|
RUNTIME: {Name: "runtime info=ifconfig,hostinfo,hostname,userinfo,procinfo,diskinfo,bootinfo,api,cli,cmd,env auto", Help: "运行环境", Actions: ice.Actions{
|
|
ice.CTX_INIT: {Hand: func(m *ice.Message, arg ...string) {
|
|
cs := m.Target().Configs
|
|
if _, ok := cs[RUNTIME]; !ok {
|
|
cs[RUNTIME] = &ice.Config{Value: kit.Dict()}
|
|
}
|
|
_runtime_init(m)
|
|
}},
|
|
IFCONFIG: {Name: "ifconfig", Help: "网卡配置", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmdy("tcp.host")
|
|
}},
|
|
HOSTINFO: {Name: "hostinfo", Help: "主机信息", Hand: func(m *ice.Message, arg ...string) {
|
|
_runtime_hostinfo(m)
|
|
}},
|
|
HOSTNAME: {Name: "hostname", Help: "主机域名", Hand: func(m *ice.Message, arg ...string) {
|
|
if len(arg) > 0 {
|
|
ice.Info.HostName = m.Conf(RUNTIME, kit.Keys(BOOT, HOSTNAME), m.Conf(RUNTIME, kit.Keys(NODE, mdb.NAME), arg[0]))
|
|
}
|
|
m.Echo(ice.Info.HostName)
|
|
}},
|
|
USERINFO: {Name: "userinfo", Help: "用户信息", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Split(m.Cmdx(SYSTEM, "who"), "user term time")
|
|
}},
|
|
PROCINFO: {Name: "procinfo", Help: "进程信息", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Split(m.Cmdx(SYSTEM, "ps", "u")).PushAction(PROCKILL)
|
|
m.StatusTimeCount()
|
|
}},
|
|
PROCKILL: {Name: "prockill", Help: "结束进程", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmdy(gdb.SIGNAL, gdb.STOP, m.Option("PID"))
|
|
m.ProcessRefresh30ms()
|
|
}},
|
|
DISKINFO: {Name: "diskinfo", Help: "磁盘信息", Hand: func(m *ice.Message, arg ...string) {
|
|
_runtime_diskinfo(m)
|
|
}},
|
|
MAXPROCS: {Name: "maxprocs", Help: "最大并发", Hand: func(m *ice.Message, arg ...string) {
|
|
if len(arg) > 0 {
|
|
runtime.GOMAXPROCS(kit.Int(m.Conf(RUNTIME, kit.Keys(HOST, MAXPROCS), kit.Select("1", arg, 0))))
|
|
}
|
|
m.Echo("%d", runtime.GOMAXPROCS(0))
|
|
}},
|
|
API: {Name: "api", Help: "接口命令", Hand: func(m *ice.Message, arg ...string) {
|
|
for k, v := range ice.Info.Route {
|
|
m.Push(nfs.PATH, k)
|
|
m.Push(nfs.FILE, v)
|
|
}
|
|
m.Sort(nfs.PATH).StatusTimeCount()
|
|
}},
|
|
CLI: {Name: "cli", Help: "文件模块", Hand: func(m *ice.Message, arg ...string) {
|
|
for k, v := range ice.Info.File {
|
|
m.Push(nfs.FILE, k)
|
|
m.Push(mdb.NAME, v)
|
|
}
|
|
m.Sort(nfs.FILE).StatusTimeCount()
|
|
}},
|
|
CMD: {Name: "cmd", Help: "模块命令", Hand: func(m *ice.Message, arg ...string) {
|
|
m.OptionFields(ctx.INDEX, mdb.NAME, mdb.HELP)
|
|
m.Cmdy(ctx.COMMAND, mdb.SEARCH, ctx.COMMAND).StatusTimeCount()
|
|
}},
|
|
ENV: {Name: "env", Help: "环境变量", Hand: func(m *ice.Message, arg ...string) {
|
|
for _, v := range os.Environ() {
|
|
ls := strings.SplitN(v, "=", 2)
|
|
m.Push(mdb.NAME, ls[0])
|
|
m.Push(mdb.VALUE, ls[1])
|
|
}
|
|
m.StatusTimeCount()
|
|
}},
|
|
MAKE_DOMAIN: {Name: "make.domain", Help: "编译主机", Hand: func(m *ice.Message, arg ...string) {
|
|
if os.Getenv(CTX_DEV) == "" || os.Getenv(CTX_POD) == "" {
|
|
m.Echo(m.Conf(RUNTIME, MAKE_DOMAIN))
|
|
} else {
|
|
m.Echo(kit.MergePOD(os.Getenv(CTX_DEV), os.Getenv(CTX_POD)))
|
|
}
|
|
}},
|
|
}, Hand: func(m *ice.Message, arg ...string) {
|
|
if len(arg) > 0 && arg[0] == BOOTINFO {
|
|
arg = arg[1:]
|
|
}
|
|
m.Cmdy(ctx.CONFIG, RUNTIME, arg)
|
|
ctx.DisplayStoryJSON(m)
|
|
}},
|
|
})
|
|
}
|