go/src/infra/cros/cmd/branch_util/cli.go

// Copyright 2021 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
package main

import (
	"context"
	"fmt"
	"log"
	"os"

	"infra/cros/internal/branch"

	"github.com/maruel/subcommands"
	"go.chromium.org/luci/auth"
	"go.chromium.org/luci/auth/client/authcli"
	"go.chromium.org/luci/common/api/gerrit"
	"go.chromium.org/luci/hardcoded/chromeinfra"
)

var (
	// Context used by CLI.
	Context            = context.Background()
	workerCount        int
	defaultManifestURL = "https://chrome-internal.googlesource.com/chromeos/manifest-internal"
)

func getApplication(authOpts auth.Options) *subcommands.DefaultApplication {
	return &subcommands.DefaultApplication{
		Name:  "branch_util",
		Title: "cros branch tool",
		Commands: []*subcommands.Command{
			subcommands.CmdHelp,
			getCmdRenameBranch(authOpts),
			getCmdDeleteBranch(authOpts),
			getCmdCreateBranch(authOpts),
			authcli.SubcommandInfo(authOpts, "auth-info", false),
			authcli.SubcommandLogin(authOpts, "auth-login", false),
			authcli.SubcommandLogout(authOpts, "auth-logout", false),
		},
	}
}

type branchApplication struct {
	*subcommands.DefaultApplication
	stdoutLog *log.Logger
	stderrLog *log.Logger
}

type branchCommand interface {
	validate([]string) (bool, string)
	getRoot() string
	getManifestURL() string
}

// CommonFlags contains flags shared amongst all sub commands.
type CommonFlags struct {
	subcommands.CommandRunBase
	Push           bool
	Force          bool
	Root           string
	ManifestURL    string
	SkipGroupCheck bool
	authFlags      authcli.Flags
}

// InitFlags initializes the common flags.
func (c *CommonFlags) InitFlags(authOpts auth.Options) {
	// Common flags
	c.Flags.BoolVar(&c.Push, "push", false,
		"Push branch modifications to remote repos. Before setting this flag, "+
			"ensure that you have the proper permissions and that you know what "+
			"you are doing. Ye be warned.")
	c.Flags.BoolVar(&c.Force, "force", false,
		"Required for any remote operation that would delete an existing "+
			"branch. Also required when trying to branch from a previously "+
			"branched manifest version.")
	// Sync CheckoutOptions
	c.Flags.StringVar(&c.ManifestURL, "manifest-url", defaultManifestURL,
		"URL of the manifest to be checked out. Defaults to googlesource URL "+
			"for manifest-internal.")
	c.Flags.IntVar(&workerCount, "j", 1, "Number of jobs to run for parallel operations.")
	c.Flags.BoolVar(&c.SkipGroupCheck, "skip-group-check", false,
		"If set, skips checking if the invoker is in mdb/chromeos-branch-creators. "+
			"ACLs will still be enforced.")
	c.authFlags.Register(c.GetFlags(), authOpts)
}

// Run runs the CLI.
func Run(c branchCommand, a subcommands.Application, args []string, env subcommands.Env) int {
	branch.StdoutLog = a.(*branchApplication).stdoutLog
	branch.StderrLog = a.(*branchApplication).stderrLog

	// Validate flags/arguments.
	ok, errMsg := c.validate(args)
	if !ok {
		fmt.Fprintf(a.GetErr(), errMsg+"\n")
		return 1
	}

	return 0
}

func main() {
	opts := chromeinfra.DefaultAuthOptions()
	opts.Scopes = []string{gerrit.OAuthScope, auth.OAuthScopeEmail}
	s := &branchApplication{
		getApplication(opts),
		log.New(os.Stdout, "", log.LstdFlags|log.Lmicroseconds),
		log.New(os.Stderr, "", log.LstdFlags|log.Lmicroseconds)}
	os.Exit(subcommands.Run(s, nil))
}