Add beginnings of new cr command

go/src/infra/tools/cr/main.go

+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+/*
+Core of the Chrome Infrastructure CLI SDK. Provides autoupdate, module
+discovery, module installation, and help.
+*/
+
+package main
+
+import (
+        "flag"
+        "fmt"
+        "os"
+        "text/tabwriter"
+
+        "infra/tools/cr/lib/subcommand"
+        "infra/tools/cr/lib/terminal"
+
+        "infra/tools/cr/cmd/firstrun"
+)
+
+var (
+        // The subcommand which will be executed.
+        cmd *subcommand.Subcommand
+
+        // Accessible variables for global flags.
+        help    bool
+        verbose bool
+)
+
+var subcommands = map[string]*subcommand.Subcommand{
+        "firstrun": firstrun.FirstrunCmd,
+}
+
+func printSubcommands() {
+        fmt.Println("Available subcommands are:")
+        colwriter := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
+        for key, value := range subcommands {
+                fmt.Fprintf(colwriter, "\t%v\t%v", key, value.ShortHelp)
+        }
+        colwriter.Flush()
+        fmt.Println("")
+}
+
+func printCrHelp() {
+        fmt.Println("The cr tool intelligently manages the Chrome Infra command-line SDK.")
+        fmt.Println("")
+        fmt.Println("You must provide a subcommand. Run 'cr help' for a list of available commands.")
+        fmt.Println("")
+        printSubcommands()
+}
+
+func main() {
+        // This has to be done after init-time to prevent an initialization loop.
+        subcommands["help"] = helpCmd
+
+        if len(os.Args) < 2 {
+                fmt.Println("No subcommand recognized.")
+                printCrHelp()
+                os.Exit(-1)
+        }
+
+        cmd = subcommands[os.Args[1]]
+        if cmd == nil {
+                fmt.Printf("Unrecognized subcommand '%v'.\n", os.Args[1])
+                printCrHelp()
+                os.Exit(-1)
+        }
+
+        flags := flag.NewFlagSet("flags", flag.ExitOnError)
+        flags.BoolVar(&help, "help", false, "print help for the given command")
+        flags.BoolVar(&verbose, "verbose", false, "print more verbose output")
+        cmd.InitFlags(flags)
+        flags.Parse(os.Args[2:])
+
+        if verbose {
+                terminal.SetDebug(verbose)
+        }
+
+        // There are two ways to get help: 'cr help foo' and 'cr foo --help'. The
+        // former is handled by setSubcommand, the latter is special-cased here.
+        if help {
+                cmd.Help(flags)
+                return
+        }
+
+        err := cmd.Run(flags)
+        if err != nil {
+                fmt.Print(err)
+                cmd.Help(flags)
+        }
+}