client/casclient/flag.go

// Copyright 2020 The LUCI Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package casclient

import (
	"flag"
	"regexp"

	"go.chromium.org/luci/common/errors"
)

// Flags contains values parsed from command line arguments for RBE-CAS.
type Flags struct {
	Addr     string
	Instance string
}

// Init initializes flag.FlagSet.
func (c *Flags) Init(f *flag.FlagSet) {
	f.StringVar(&c.Addr, "cas-addr", AddrProd, "CAS address.")
	f.StringVar(&c.Instance, "cas-instance", "", "CAS instance (GCP). Format is either a project ID, or \"projects/<project_id>/instances/<instance_id>\"")
}

// Parse applies changes specified by command line flags.
func (c *Flags) Parse() error {
	if c.Instance == "" {
		if c.Addr == AddrProd {
			return errors.Reason("CAS instance or local CAS address must be specified").Err()
		}
		// Use local CAS address in this case.
		return nil
	}
	ins, err := parseCASInstance(c.Instance)
	if err != nil {
		return err
	}
	c.Instance = ins
	return nil
}

// GCP project ID format: https://cloud.google.com/resource-manager/docs/creating-managing-projects
// Not the most accurate regexp, but let's just assume most people know what they are doing...
var (
	projectRE  = regexp.MustCompile(`^[a-z0-9\-]+$`)
	instanceRE = regexp.MustCompile(`^projects/[a-z0-9\-]+/instances/[^/]+$`)
)

func parseCASInstance(ins string) (string, error) {
	if projectRE.MatchString(ins) {
		return "projects/" + ins + "/instances/default_instance", nil
	}
	if instanceRE.MatchString(ins) {
		return ins, nil
	}
	return "", errors.Reason("invalid CAS instance: %s", ins).Err()
}