From 3a20dd41d50038da4fd05f3a38aa916c88309627 Mon Sep 17 00:00:00 2001
From: Derek McGowan <derek@mcgstyle.net>
Date: Thu, 9 Mar 2017 19:50:23 -0800
Subject: [PATCH] Add init subcommand to rootfs

Init command gets the mounts for a given chain id and
outputs a mount command.

Signed-off-by: Derek McGowan <derek@mcgstyle.net> (github: dmcgowan)
---
 cmd/dist/rootfs.go | 52 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/cmd/dist/rootfs.go b/cmd/dist/rootfs.go
index e9075c8..b77d677 100644
--- a/cmd/dist/rootfs.go
+++ b/cmd/dist/rootfs.go
@@ -3,7 +3,10 @@ package main
 import (
 	"context"
 	"encoding/json"
+	"fmt"
 	"io/ioutil"
+	"os"
+	"strings"
 
 	contentapi "github.com/docker/containerd/api/services/content"
 	rootfsapi "github.com/docker/containerd/api/services/rootfs"
@@ -21,6 +24,7 @@ var rootfsCommand = cli.Command{
 	Usage: "rootfs setups a rootfs",
 	Subcommands: []cli.Command{
 		rootfsPrepareCommand,
+		rootfsInitCommand,
 	},
 }
 
@@ -64,6 +68,54 @@ var rootfsPrepareCommand = cli.Command{
 	},
 }
 
+var rootfsInitCommand = cli.Command{
+	Name:      "init",
+	Usage:     "init gets mount commands for digest",
+	ArgsUsage: "[flags] <digest> <target>",
+	Flags:     []cli.Flag{},
+	Action: func(clicontext *cli.Context) error {
+		var (
+			ctx = background
+		)
+
+		if clicontext.NArg() != 2 {
+			return cli.ShowSubcommandHelp(clicontext)
+		}
+
+		dgst, err := digest.Parse(clicontext.Args().Get(0))
+		if err != nil {
+			return err
+		}
+		target := clicontext.Args().Get(1)
+
+		log.G(ctx).Infof("initializing mounts %s", dgst.String())
+
+		conn, err := connectGRPC(clicontext)
+		if err != nil {
+			return err
+		}
+
+		rclient := rootfsapi.NewRootFSClient(conn)
+
+		ir := &rootfsapi.InitMountsRequest{
+			Name:    target,
+			ChainID: dgst,
+		}
+
+		resp, err := rclient.InitMounts(ctx, ir)
+		if err != nil {
+			return err
+		}
+
+		for _, m := range resp.Mounts {
+			fmt.Fprintf(os.Stdout, "mount -t %s %s %s -o %s\n", m.Type, m.Source, target, strings.Join(m.Options, ","))
+		}
+		log.G(ctx).Infof("Mount response: %#v", resp)
+
+		return nil
+	},
+}
+
 func resolveManifest(ctx context.Context, provider content.Provider, dgst digest.Digest) (ocispec.Manifest, error) {
 	p, err := readAll(ctx, provider, dgst)
 	if err != nil {