Go言語でのflagパッケージでコマンドラインオプションをコマンドライン引数の前後に置きたい

Question

現在Go言語で簡易なcurlコマンドを実装しています。
コマンドラインオプションは-v, -x, -o, -dのみを想定していて以下のように設定しています。

// オプションの設定
	var showHeader bool
	flag.BoolVar(&showHeader, "v", false, "-v, --verbose  Make the operation more talkative")
	var outputFile string
	flag.StringVar(&outputFile, "o", "", "-o, --output <file>  Write to file instead of stdout")
	var postValues string
	flag.StringVar(&postValues, "d", "", "-d, --data <data>  HTTP POST data")
	var requestType string
	flag.StringVar(&requestType, "X", "GET", "-X, --request <command>  Specify request command to use")
	flag.Parse()

そのあとHTTPクライアントの設定などを行った後、以下のように分岐してリクエストを飛ばしています。

addr := flag.Arg(0)
// GETかPOSTで分岐
	if requestType == "GET"{
		get(client, addr, showHeader, outputFile)
	}else if requestType == "POST"{
		post(client, addr, showHeader, values)
	}else{
		fmt.Printf("%s: requestType is not correct!\n", os.Args[0])
		fmt.Printf("%s: try 'kcurl --help' or 'kcurl --manual' for more information\n", os.Args[0])
		os.Exit(1)
	}

しかし、作成したプログラムでは、
go run main.go -v -o test.txt 'http://google.com'のようにコマンドライン引数（URL）の前にコマンドラインオプションを置く形式しか動かず、go run main.go 'http://google.com' -o test.txt -vのように後ろに置く形式ではflag.Argsに-v -o test.txtが文字列として認識されて保存されてしまい、-vオプションの値や-oオプションの値がデフォルト値のままとなってしまっています。

fmt.Println(flag.Args())	
fmt.Println(showHeader, outputFile, postValues, requestType)

// [http://google.com -v -o test.txt]
// false   GET

このような場合、どういうふうにflagを設定すればいいのでしょうか。
教えていただけると幸いです。

Answer 1

• flagは必ず[options] args...という並びを期待します。
• flagだけで質問のようなパターンを処理するには、flagに渡す前に加工する必要があります。
• デフォルトflagインスタンスを使う場合、コマンドライン引数は自動で渡されちゃうのでNewFlagSetを使いましょう

flags := flag.NewFlagSet(...)
...
args := os.Args[1:]
... // <- argsを加工
flags.Parse(args)

別解としてはもっと柔軟性の高いコマンドラインパーサーを使います。
例えば： https://github.com/alecthomas/kingpin

Comments

[tedmosby]

ありがとうございます！