zot: initial commit

pkg/cli/BUILD.bazel

@@ -0,0 +1,27 @@
+load("@io_bazel_rules_go//go:def.bzl", "go_library", "go_test")
+
+go_library(
+    name = "go_default_library",
+    srcs = ["root.go"],
+    importpath = "github.com/anuvu/zot/pkg/cli",
+    visibility = ["//visibility:public"],
+    deps = [
+        "//errors:go_default_library",
+        "//pkg/api:go_default_library",
+        "//pkg/storage:go_default_library",
+        "@com_github_mitchellh_mapstructure//:go_default_library",
+        "@com_github_opencontainers_distribution_spec//:go_default_library",
+        "@com_github_rs_zerolog//log:go_default_library",
+        "@com_github_spf13_cobra//:go_default_library",
+        "@com_github_spf13_viper//:go_default_library",
+    ],
+)
+
+go_test(
+    name = "go_default_test",
+    timeout = "short",
+    srcs = ["root_test.go"],
+    embed = [":go_default_library"],
+    race = "on",
+    deps = ["@com_github_smartystreets_goconvey//convey:go_default_library"],
+)

pkg/cli/root.go

@@ -0,0 +1,99 @@
+package cli
+
+import (
+	"github.com/anuvu/zot/errors"
+	"github.com/anuvu/zot/pkg/api"
+	"github.com/anuvu/zot/pkg/storage"
+	"github.com/mitchellh/mapstructure"
+	dspec "github.com/opencontainers/distribution-spec"
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+// metadataConfig reports metadata after parsing, which we use to track
+// errors
+func metadataConfig(md *mapstructure.Metadata) viper.DecoderConfigOption {
+	return func(c *mapstructure.DecoderConfig) {
+		c.Metadata = md
+	}
+}
+
+func NewRootCmd() *cobra.Command {
+	showVersion := false
+	config := api.NewConfig()
+
+	serveCmd := &cobra.Command{
+		Use:     "serve <config>",
+		Aliases: []string{"serve"},
+		Short:   "`serve` stores and distributes OCI images",
+		Long:    "`serve` stores and distributes OCI images",
+		Run: func(cmd *cobra.Command, args []string) {
+			if len(args) > 0 {
+				viper.SetConfigFile(args[0])
+				if err := viper.ReadInConfig(); err != nil {
+					panic(err)
+				}
+
+				md := &mapstructure.Metadata{}
+				if err := viper.Unmarshal(&config, metadataConfig(md)); err != nil {
+					panic(err)
+				}
+
+				// if haven't found a single key or there were unused keys, report it as
+				// a error
+				if len(md.Keys) == 0 || len(md.Unused) > 0 {
+					panic(errors.ErrBadConfig)
+				}
+			}
+			c := api.NewController(config)
+			if err := c.Run(); err != nil {
+				panic(err)
+			}
+		},
+	}
+
+	gcDelUntagged := false
+	gcDryRun := false
+
+	gcCmd := &cobra.Command{
+		Use:     "garbage-collect <config>",
+		Aliases: []string{"gc"},
+		Short:   "`garbage-collect` deletes layers not referenced by any manifests",
+		Long:    "`garbage-collect` deletes layers not referenced by any manifests",
+		Run: func(cmd *cobra.Command, args []string) {
+			log.Info().Interface("values", config).Msg("configuration settings")
+			if config.Storage.RootDirectory != "" {
+				if err := storage.Scrub(config.Storage.RootDirectory, gcDryRun); err != nil {
+					panic(err)
+				}
+			}
+		},
+	}
+
+	gcCmd.Flags().StringVarP(&config.Storage.RootDirectory, "storage-root-dir", "r", "",
+		"Use specified directory for filestore backing image data")
+	_ = gcCmd.MarkFlagRequired("storage-root-dir")
+	gcCmd.Flags().BoolVarP(&gcDelUntagged, "delete-untagged", "m", false,
+		"delete manifests that are not currently referenced via tag")
+	gcCmd.Flags().BoolVarP(&gcDryRun, "dry-run", "d", false,
+		"do everything except remove the blobs")
+
+	rootCmd := &cobra.Command{
+		Use:   "zot",
+		Short: "`zot`",
+		Long:  "`zot`",
+		Run: func(cmd *cobra.Command, args []string) {
+			if showVersion {
+				log.Info().Str("version", dspec.Version).Msg("distribution-spec")
+			}
+			_ = cmd.Usage()
+		},
+	}
+
+	rootCmd.AddCommand(serveCmd)
+	rootCmd.AddCommand(gcCmd)
+	rootCmd.Flags().BoolVarP(&showVersion, "version", "v", false, "show the version and exit")
+
+	return rootCmd
+}

pkg/cli/root_test.go

@@ -0,0 +1,42 @@
+package cli_test
+
+import (
+	"os"
+	"testing"
+
+	"github.com/anuvu/zot/pkg/cli"
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func TestUsage(t *testing.T) {
+	oldArgs := os.Args
+	defer func() { os.Args = oldArgs }()
+
+	Convey("Test Usage", t, func(c C) {
+		os.Args = []string{"cli_test", "help"}
+		err := cli.NewRootCmd().Execute()
+		So(err, ShouldBeNil)
+	})
+}
+
+func TestServe(t *testing.T) {
+	oldArgs := os.Args
+	defer func() { os.Args = oldArgs }()
+
+	Convey("Test Usage", t, func(c C) {
+		os.Args = []string{"cli_test", "serve", "-h"}
+		err := cli.NewRootCmd().Execute()
+		So(err, ShouldBeNil)
+	})
+}
+
+func TestGC(t *testing.T) {
+	oldArgs := os.Args
+	defer func() { os.Args = oldArgs }()
+
+	Convey("Test Usage", t, func(c C) {
+		os.Args = []string{"cli_test", "garbage-collect", "-h"}
+		err := cli.NewRootCmd().Execute()
+		So(err, ShouldBeNil)
+	})
+}