mirror of
https://gitee.com/johng/gf.git
synced 2024-12-05 05:37:55 +08:00
147 lines
3.5 KiB
Go
147 lines
3.5 KiB
Go
// Copyright 2017 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
|
||
//
|
||
// This Source Code Form is subject to the terms of the MIT License.
|
||
// If a copy of the MIT was not distributed with this file,
|
||
// You can obtain one at https://gitee.com/johng/gf.
|
||
//
|
||
|
||
// 命令行管理
|
||
package gcmd
|
||
|
||
import (
|
||
"os"
|
||
"regexp"
|
||
"errors"
|
||
"strconv"
|
||
"strings"
|
||
)
|
||
|
||
// 命令行参数列表
|
||
type gCmdValue struct {
|
||
values []string
|
||
}
|
||
|
||
// 命令行选项列表
|
||
type gCmdOption struct {
|
||
options map[string]string
|
||
}
|
||
|
||
// 终端管理对象(全局)
|
||
var Value = &gCmdValue{} // 终端参数-命令参数列表
|
||
var Option = &gCmdOption{} // 终端参数-选项参数列表
|
||
var cmdFuncMap = make(map[string]func()) // 终端命令及函数地址对应表
|
||
|
||
// 检查并初始化console参数，在包加载的时候触发
|
||
// 初始化时执行，不影响运行时性能
|
||
func init() {
|
||
Option.options = make(map[string]string)
|
||
reg := regexp.MustCompile(`\-\-{0,1}(\w+?)=(.+)`)
|
||
for i := 0; i < len(os.Args); i++ {
|
||
result := reg.FindStringSubmatch(os.Args[i])
|
||
if len(result) > 1 {
|
||
Option.options[result[1]] = result[2]
|
||
} else {
|
||
Value.values = append(Value.values, os.Args[i])
|
||
}
|
||
}
|
||
}
|
||
|
||
// 返回所有的命令行参数values
|
||
func (c *gCmdValue) GetAll() []string {
|
||
return c.values
|
||
}
|
||
|
||
// 返回所有的命令行参数options
|
||
func (c *gCmdOption) GetAll() map[string]string {
|
||
return c.options
|
||
}
|
||
|
||
// 获得一条指定索引位置的value参数
|
||
func (c *gCmdValue) Get(index uint8) string {
|
||
if index < uint8(len(c.values)) {
|
||
return c.values[index]
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// 类型转换
|
||
func (c *gCmdValue) GetInt(key uint8) int {
|
||
if v := c.Get(key); v != "" {
|
||
i, _ := strconv.Atoi(v)
|
||
return i
|
||
}
|
||
return 0
|
||
}
|
||
|
||
// 类型转换bool
|
||
func (c *gCmdValue) GetBool(key uint8) bool {
|
||
v := c.Get(key)
|
||
v = strings.ToLower(v)
|
||
if v != "" && v != "0" && v != "false" {
|
||
return true
|
||
}
|
||
return false
|
||
}
|
||
|
||
// 获得一条指定索引位置的option参数;
|
||
func (c *gCmdOption) Get(key string) string {
|
||
if option, ok := c.options[key]; ok {
|
||
return option
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// 类型转换int
|
||
func (c *gCmdOption) GetInt(key string) int {
|
||
if v := c.Get(key); v != "" {
|
||
i, _ := strconv.Atoi(v)
|
||
return i
|
||
}
|
||
return 0
|
||
}
|
||
|
||
// 类型转换bool
|
||
func (c *gCmdOption) GetBool(key string) bool {
|
||
v := c.Get(key)
|
||
v = strings.ToLower(v)
|
||
if v != "" && v != "0" && v != "false" {
|
||
return true
|
||
}
|
||
return false
|
||
}
|
||
|
||
// 绑定命令行参数及对应的命令函数，注意参数是函数的内存地址
|
||
// 如果操作失败返回错误信息
|
||
func BindHandle (cmd string, f func()) error {
|
||
if _, ok := cmdFuncMap[cmd]; ok {
|
||
return errors.New("duplicated handle for command:" + cmd)
|
||
} else {
|
||
cmdFuncMap[cmd] = f
|
||
return nil
|
||
}
|
||
}
|
||
|
||
// 执行命令对应的函数
|
||
func RunHandle (cmd string) error {
|
||
if handle, ok := cmdFuncMap[cmd]; ok {
|
||
handle()
|
||
return nil
|
||
} else {
|
||
return errors.New("no handle found for command:" + cmd)
|
||
}
|
||
}
|
||
|
||
// 自动识别命令参数并执行命令参数对应的函数
|
||
func AutoRun () error {
|
||
if cmd := Value.Get(1); cmd != "" {
|
||
if handle, ok := cmdFuncMap[cmd]; ok {
|
||
handle()
|
||
return nil
|
||
} else {
|
||
return errors.New("no handle found for command:" + cmd)
|
||
}
|
||
} else {
|
||
return errors.New("no command found")
|
||
}
|
||
}
|