image/layout/tar: Add a CreateTarFile helper

The NewEngine commands for the tar-backed image-layout engines (both
the CAS and refs engines) open files O_RDWR and expect image-layout
compatible content in the tarball.  That makes sense, but for folks
who *don't* have such a tarball, a helper like CreateTarFile makes it
easy to explicitly create an empty one.

The 0666 file modes and 0777 directory modes rely on the caller's
umask to appropriately limit user/group/other permissions for the
tarball itself and any content extracted to the filesystem from the
tarball.

Expose this new functionality on the command line as:

  $ oci-image-tool init image-layout PATH

where 'init' and 'image-layout' are two separate levels in case we
support initializing additional types of repositories in the future.

Signed-off-by: W. Trevor King <[email protected]>

Author: wking

cmd/oci-image-tool/init.go

@@ -0,0 +1,33 @@
+// Copyright 2016 The Linux Foundation
+//
+// 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 (
+	"io"
+	"log"
+
+	"github.com/spf13/cobra"
+)
+
+func newInitCmd(stdout io.Writer, stderr *log.Logger) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "init",
+		Short: "Initialize an OCI image",
+	}
+
+	cmd.AddCommand(newInitImageLayoutCmd(stdout, stderr))
+
+	return cmd
+}

cmd/oci-image-tool/init_image_layout.go

@@ -0,0 +1,62 @@
+// Copyright 2016 The Linux Foundation
+//
+// 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 (
+	"io"
+	"log"
+	"os"
+
+	"github.com/opencontainers/image-spec/image/layout"
+	"github.com/spf13/cobra"
+	"golang.org/x/net/context"
+)
+
+type initImageLayout struct {
+	stderr *log.Logger
+}
+
+func newInitImageLayoutCmd(stdout io.Writer, stderr *log.Logger) *cobra.Command {
+	state := &initImageLayout{
+		stderr: stderr,
+	}
+
+	return &cobra.Command{
+		Use:   "image-layout PATH",
+		Short: "Initialize an OCI image-layout repository",
+		Run:   state.Run,
+	}
+}
+
+func (state *initImageLayout) Run(cmd *cobra.Command, args []string) {
+	if len(args) != 1 {
+		if err := cmd.Usage(); err != nil {
+			state.stderr.Println(err)
+		}
+		os.Exit(1)
+	}
+
+	path := args[0]
+
+	ctx := context.Background()
+
+	err := layout.CreateTarFile(ctx, path)
+	if err != nil {
+		state.stderr.Println(err)
+		os.Exit(1)
+	}
+
+	os.Exit(0)
+}

cmd/oci-image-tool/main.go

@@ -30,6 +30,7 @@ func main() {
 	stdout := log.New(os.Stdout, "", 0)
 	stderr := log.New(os.Stderr, "", 0)
 
+	cmd.AddCommand(newInitCmd(os.Stdout, stderr))
 	cmd.AddCommand(newValidateCmd(stdout, stderr))
 	cmd.AddCommand(newUnpackCmd(stdout, stderr))
 	cmd.AddCommand(newBundleCmd(stdout, stderr))

image/layout/tar.go

@@ -187,3 +187,50 @@ func CheckTarVersion(ctx context.Context, reader io.ReadSeeker) (err error) {
 
 	return nil
 }
+
+// CreateTarFile creates a new image-layout tar file at the given path.
+func CreateTarFile(ctx context.Context, path string) (err error) {
+	file, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_EXCL, 0666)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	tarWriter := tar.NewWriter(file)
+	defer tarWriter.Close()
+
+	now := time.Now()
+	for _, name := range []string{"./blobs/", "./refs/"} {
+		header := &tar.Header{
+			Name:     name,
+			Mode:     0777,
+			ModTime:  now,
+			Typeflag: tar.TypeDir,
+		}
+		err = tarWriter.WriteHeader(header)
+		if err != nil {
+			return err
+		}
+	}
+
+	imageLayoutVersion := specs.ImageLayoutVersion{
+		Version: "1.0.0",
+	}
+	imageLayoutVersionBytes, err := json.Marshal(imageLayoutVersion)
+	if err != nil {
+		return err
+	}
+	header := &tar.Header{
+		Name:     "./oci-layout",
+		Mode:     0666,
+		Size:     int64(len(imageLayoutVersionBytes)),
+		ModTime:  now,
+		Typeflag: tar.TypeReg,
+	}
+	err = tarWriter.WriteHeader(header)
+	if err != nil {
+		return err
+	}
+	_, err = tarWriter.Write(imageLayoutVersionBytes)
+	return err
+}