From e3781965054ab125b8037c1ffd6379d3d1834bc6 Mon Sep 17 00:00:00 2001
From: Derek McGowan <derek@mcgstyle.net>
Date: Mon, 16 Oct 2017 11:54:01 -0700
Subject: [PATCH] Add snapshot label command to ctr

Allows easier debugging of snapshots using labels

Signed-off-by: Derek McGowan <derek@mcgstyle.net>
---
 cmd/ctr/snapshot.go | 53 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/cmd/ctr/snapshot.go b/cmd/ctr/snapshot.go
index f8a2b5ecf..3b990cdb5 100644
--- a/cmd/ctr/snapshot.go
+++ b/cmd/ctr/snapshot.go
@@ -28,6 +28,7 @@ var snapshotCommand = cli.Command{
 		mountSnapshotCommand,
 		commitSnapshotCommand,
 		infoSnapshotCommand,
+		labelSnapshotCommand,
 	},
 }
 
@@ -343,6 +344,58 @@ var infoSnapshotCommand = cli.Command{
 	},
 }
 
+var labelSnapshotCommand = cli.Command{
+	Name:        "label",
+	Usage:       "add labels to content",
+	ArgsUsage:   "[flags] <name> [<label>=<value> ...]",
+	Description: `Labels snapshots in the snapshotter`,
+	Flags:       []cli.Flag{},
+	Action: func(clicontext *cli.Context) error {
+		var (
+			key, labels = objectWithLabelArgs(clicontext)
+		)
+		ctx, cancel := appContext(clicontext)
+		defer cancel()
+
+		snapshotter, err := getSnapshotter(clicontext)
+		if err != nil {
+			return err
+		}
+
+		info := snapshot.Info{
+			Name:   key,
+			Labels: map[string]string{},
+		}
+
+		var paths []string
+		for k, v := range labels {
+			paths = append(paths, fmt.Sprintf("labels.%s", k))
+			if v != "" {
+				info.Labels[k] = v
+			}
+		}
+
+		// Nothing updated, do no clear
+		if len(paths) == 0 {
+			info, err = snapshotter.Stat(ctx, info.Name)
+		} else {
+			info, err = snapshotter.Update(ctx, info, paths...)
+		}
+		if err != nil {
+			return err
+		}
+
+		var labelStrings []string
+		for k, v := range info.Labels {
+			labelStrings = append(labelStrings, fmt.Sprintf("%s=%s", k, v))
+		}
+
+		fmt.Println(strings.Join(labelStrings, ","))
+
+		return nil
+	},
+}
+
 type snapshotTreeNode struct {
 	Name     string
 	Parent   string