将命令行字符串解析为Golang中的标志和参数

时间:2015-12-06 14:53:24

标签: go arguments package command-line-arguments flags

我正在寻找一个包含-v --format "some example" -i test等字符串的包,然后将其解析成一串字符串,正确处理引号,空格等:

-v
--format
some example
-i
test

我已经检查了内置的flag包以及Github上的其他标志处理包,但它们似乎都没有处理将原始字符串解析为令牌的特殊情况。在尝试自己做之前,我宁愿寻找一个包裹,因为我确定有很多特殊情况需要处理。

有什么建议吗?

4 个答案:

答案 0 :(得分:4)

有关信息,这是我最终创建的功能。

它将命令拆分为其参数。例如,cat -v "some file.txt"将返回["cat", "-v", "some file.txt"]

它还可以正确处理转义字符,特别是空格。因此cat -v some\ file.txt也会正确地分为["cat", "-v", "some file.txt"]

func parseCommandLine(command string) ([]string, error) {
    var args []string
    state := "start"
    current := ""
    quote := "\""
    escapeNext := true
    for i := 0; i < len(command); i++ {
        c := command[i]

        if state == "quotes" {
            if string(c) != quote {
                current += string(c)
            } else {
                args = append(args, current)
                current = ""
                state = "start"
            }
            continue
        }

        if (escapeNext) {
            current += string(c)
            escapeNext = false
            continue
        }

        if (c == '\\') {
            escapeNext = true
            continue
        }

        if c == '"' || c == '\'' {
            state = "quotes"
            quote = string(c)
            continue
        }

        if state == "arg" {
            if c == ' ' || c == '\t' {
                args = append(args, current)
                current = ""
                state = "start"
            } else {
                current += string(c)
            }
            continue
        }

        if c != ' ' && c != '\t' {
            state = "arg"
            current += string(c)
        }
    }

    if state == "quotes" {
        return []string{}, errors.New(fmt.Sprintf("Unclosed quote in command line: %s", command))
    }

    if current != "" {
        args = append(args, current)
    }

    return args, nil
}

答案 1 :(得分:2)

如果args在命令行上传递给你的程序,那么shell应该处理这个并且os.Args将被正确填充。例如,在您的情况下,os.Args[1:]将等于

[]string{"-v", "--format", "some example", "-i", "test"}

如果您只是因为某些原因而拥有该字符串,并且您想模仿shell将对其执行的操作,那么我建议使用https://github.com/kballard/go-shellquote

这样的包

答案 2 :(得分:2)

外观类似于shlex

import "github.com/google/shlex"
shlex.Split("one \"two three\" four") -> []string{"one", "two three", "four"}

答案 3 :(得分:0)

hedzr/cmdr可能很好。它是类似getopt的命令行解析器,重量轻,流利的api或经典样式。