1 files changed, 72 insertions, 0 deletions
diff --git a/cmd/gcrane/main.go b/cmd/gcrane/main.go
new file mode 100644
index 0000000..98871e7
--- /dev/null
+++ b/cmd/gcrane/main.go
@@ -0,0 +1,72 @@
+// Copyright 2018 Google LLC All Rights Reserved.
+//
+// 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 main
+
+import (
+	"context"
+	"os"
+	"os/signal"
+
+	"github.com/google/go-containerregistry/cmd/crane/cmd"
+	gcmd "github.com/google/go-containerregistry/cmd/gcrane/cmd"
+	"github.com/google/go-containerregistry/pkg/crane"
+	"github.com/google/go-containerregistry/pkg/gcrane"
+	"github.com/google/go-containerregistry/pkg/logs"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	logs.Warn.SetOutput(os.Stderr)
+	logs.Progress.SetOutput(os.Stderr)
+}
+
+const (
+	use   = "gcrane"
+	short = "gcrane is a tool for managing container images on gcr.io and pkg.dev"
+)
+
+func main() {
+	options := []crane.Option{crane.WithAuthFromKeychain(gcrane.Keychain)}
+	// Same as crane, but override usage and keychain.
+	root := cmd.New(use, short, options)
+
+	// Add or override commands.
+	gcraneCmds := []*cobra.Command{gcmd.NewCmdList(), gcmd.NewCmdGc(), gcmd.NewCmdCopy(), cmd.NewCmdAuth(options, "gcrane", "auth")}
+
+	// Maintain a map of google-specific commands that we "override".
+	used := make(map[string]bool)
+	for _, cmd := range gcraneCmds {
+		used[cmd.Use] = true
+	}
+
+	// Remove those from crane's set of commands.
+	for _, cmd := range root.Commands() {
+		if _, ok := used[cmd.Use]; ok {
+			root.RemoveCommand(cmd)
+		}
+	}
+
+	// Add our own.
+	for _, cmd := range gcraneCmds {
+		root.AddCommand(cmd)
+	}
+
+	ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
+	defer cancel()
+	if err := root.ExecuteContext(ctx); err != nil {
+		cancel()
+		os.Exit(1)
+	}
+}