blob: 72c23a96a0e6674e845813a24c2b94e8e77a3027 [file] [log] [blame]
// 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 lib
import (
"context"
"flag"
"github.com/bazelbuild/remote-apis-sdks/go/pkg/client"
"github.com/maruel/subcommands"
"go.chromium.org/luci/client/cas"
"go.chromium.org/luci/common/cli"
"go.chromium.org/luci/common/errors"
"go.chromium.org/luci/common/logging"
"go.chromium.org/luci/common/runtime/profiling"
)
// AuthFlags is an interface to register auth flags and create RBE Client.
type AuthFlags interface {
// Register registers auth flags to the given flag set. e.g. -service-account-json.
Register(f *flag.FlagSet)
// Parse parses auth flags.
Parse() error
// NewClient creates an authroised RBE Client.
NewClient(ctx context.Context, instance string, readOnly bool) (*client.Client, error)
}
var _ cli.ContextModificator = (*commonFlags)(nil)
type commonFlags struct {
subcommands.CommandRunBase
casFlags cas.Flags
logConfig logging.Config // for -log-level, used by ModifyContext
profiler profiling.Profiler
authFlags AuthFlags
}
func (c *commonFlags) Init(authFlags AuthFlags) {
c.casFlags.Init(&c.Flags)
c.authFlags = authFlags
c.authFlags.Register(&c.Flags)
c.logConfig.Level = logging.Warning
c.logConfig.AddFlags(&c.Flags)
c.profiler.AddFlags(&c.Flags)
}
func (c *commonFlags) Parse() error {
if c.logConfig.Level == logging.Debug {
// extract glog flag used in remote-apis-sdks
logtostderr := flag.Lookup("logtostderr")
if logtostderr == nil {
return errors.Reason("logtostderr flag for glog not found").Err()
}
v := flag.Lookup("v")
if v == nil {
return errors.Reason("v flag for glog not found").Err()
}
logtostderr.Value.Set("true")
v.Value.Set("9")
}
if err := c.profiler.Start(); err != nil {
return err
}
if err := c.authFlags.Parse(); err != nil {
return err
}
return c.casFlags.Parse()
}
// ModifyContext implements cli.ContextModificator.
func (c *commonFlags) ModifyContext(ctx context.Context) context.Context {
return c.logConfig.Set(ctx)
}