From c6d3fc820da4cbd71e1f84806dcd0a9168875d82 Mon Sep 17 00:00:00 2001
From: Rob Percival <robpercival@google.com>
Date: Fri, 19 May 2017 17:33:19 +0100
Subject: [PATCH] Example of loading flags from a file

---
 cmd/createtree/main.go | 18 ++++++++++++++++++
 cmd/flags.go           | 24 ++++++++++++++++++++++++
 2 files changed, 42 insertions(+)
 create mode 100644 cmd/flags.go

diff --git a/cmd/createtree/main.go b/cmd/createtree/main.go
index ce72149683..9cc0a0bac3 100644
--- a/cmd/createtree/main.go
+++ b/cmd/createtree/main.go
@@ -41,12 +41,14 @@ import (
 	"github.com/golang/protobuf/ptypes"
 	"github.com/golang/protobuf/ptypes/any"
 	"github.com/google/trillian"
+	"github.com/google/trillian/cmd"
 	"github.com/google/trillian/crypto/keyspb"
 	"github.com/google/trillian/crypto/sigpb"
 	"google.golang.org/grpc"
 )
 
 var (
+	configFile      = flag.String("config", "", "Config file containing flags")
 	adminServerAddr = flag.String("admin_server", "", "Address of the gRPC Trillian Admin Server (host:port)")
 
 	treeState          = flag.String("tree_state", trillian.TreeState_ACTIVE.String(), "State of the new tree")
@@ -175,6 +177,22 @@ func newOptsFromFlags() *createOpts {
 func main() {
 	flag.Parse()
 
+	if *configFile != "" {
+		if flag.NFlag() != 1 {
+			fmt.Printf("No other flags can be provided when --config is set")
+			os.Exit(1)
+		}
+
+		if err := cmd.ParseFlagFile(*configFile); err != nil {
+			fmt.Fprintf(os.Stderr, "Failed to parse %v: %v\n", *configFile, err)
+			os.Exit(1)
+		}
+
+		// Alternative to printing error if more than just "--config" flag is provided:
+		// let command-line flags take precedent by re-parsing from the command-line.
+		// flag.Parse()
+	}
+
 	ctx := context.Background()
 	tree, err := createTree(ctx, newOptsFromFlags())
 	if err != nil {
diff --git a/cmd/flags.go b/cmd/flags.go
new file mode 100644
index 0000000000..25232ae211
--- /dev/null
+++ b/cmd/flags.go
@@ -0,0 +1,24 @@
+package cmd
+
+import (
+	"flag"
+	"io/ioutil"
+
+	"github.com/mattn/go-shellwords"
+)
+
+func ParseFlagFile(path string) error {
+	file, err := ioutil.ReadFile(path)
+	if err != nil {
+		return err
+	}
+
+	p := shellwords.NewParser()
+	p.ParseEnv = true
+	args, err := p.Parse(string(file))
+	if err != nil {
+		return err
+	}
+
+	return flag.CommandLine.Parse(args)
+}