Move NewClient and AppContext to commands pkg

Signed-off-by: Michael Crosby <crosbymichael@gmail.com>

cmd/ctr/apply.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/containerd/containerd/archive"
 	"github.com/containerd/containerd/archive/compression"
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/containerd/containerd/log"
 	"github.com/urfave/cli"
 )
@@ -18,7 +19,7 @@ var applyCommand = cli.Command{
 		var (
 			dir = context.Args().First()
 		)
-		ctx, cancel := appContext(context)
+		ctx, cancel := commands.AppContext(context)
 		defer cancel()
 
 		log.G(ctx).Info("applying layer from stdin")

cmd/ctr/attach.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/containerd/console"
 	"github.com/containerd/containerd"
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/sirupsen/logrus"
 	"github.com/urfave/cli"
 )
@@ -14,7 +15,7 @@ var taskAttachCommand = cli.Command{
 	Usage:     "attach to the IO of a running container",
 	ArgsUsage: "CONTAINER",
 	Action: func(context *cli.Context) error {
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/checkpoint.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 
 	"github.com/containerd/containerd"
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/pkg/errors"
 	"github.com/urfave/cli"
 )
@@ -23,7 +24,7 @@ var taskCheckpointCommand = cli.Command{
 		if id == "" {
 			return errors.New("container id must be provided")
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/commands/client.go

@@ -0,0 +1,40 @@
+package commands
+
+import (
+	gocontext "context"
+
+	"github.com/containerd/containerd"
+	"github.com/containerd/containerd/namespaces"
+	"github.com/urfave/cli"
+)
+
+// AppContext returns the context for a command. Should only be called once per
+// command, near the start.
+//
+// This will ensure the namespace is picked up and set the timeout, if one is
+// defined.
+func AppContext(context *cli.Context) (gocontext.Context, gocontext.CancelFunc) {
+	var (
+		ctx       = gocontext.Background()
+		timeout   = context.GlobalDuration("timeout")
+		namespace = context.GlobalString("namespace")
+		cancel    gocontext.CancelFunc
+	)
+	ctx = namespaces.WithNamespace(ctx, namespace)
+	if timeout > 0 {
+		ctx, cancel = gocontext.WithTimeout(ctx, timeout)
+	} else {
+		ctx, cancel = gocontext.WithCancel(ctx)
+	}
+	return ctx, cancel
+}
+
+// NewClient returns a new containerd client
+func NewClient(context *cli.Context) (*containerd.Client, gocontext.Context, gocontext.CancelFunc, error) {
+	client, err := containerd.New(context.GlobalString("address"))
+	if err != nil {
+		return nil, nil, nil, err
+	}
+	ctx, cancel := AppContext(context)
+	return client, ctx, cancel, nil
+}

cmd/ctr/container_delete.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 
 	"github.com/containerd/containerd"
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/containerd/containerd/log"
 	"github.com/urfave/cli"
 )
@@ -23,7 +24,7 @@ var containersDeleteCommand = cli.Command{
 	},
 	Action: func(context *cli.Context) error {
 		var exitErr error
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/containers.go

@@ -5,6 +5,7 @@ import (
 	"os"
 	"text/tabwriter"
 
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/urfave/cli"
 )
 
@@ -29,7 +30,7 @@ var containersCommand = cli.Command{
 			filters = context.Args()
 			quiet   = context.Bool("quiet")
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/content.go

@@ -46,7 +46,7 @@ var (
 			if err != nil {
 				return err
 			}
-			client, ctx, cancel, err := newClient(context)
+			client, ctx, cancel, err := commands.NewClient(context)
 			if err != nil {
 				return err
 			}
@@ -90,7 +90,7 @@ var (
 			if ref == "" {
 				return errors.New("must specify a transaction reference")
 			}
-			client, ctx, cancel, err := newClient(context)
+			client, ctx, cancel, err := commands.NewClient(context)
 			if err != nil {
 				return err
 			}
@@ -124,7 +124,7 @@ var (
 		},
 		Action: func(context *cli.Context) error {
 			match := context.Args().First()
-			client, ctx, cancel, err := newClient(context)
+			client, ctx, cancel, err := commands.NewClient(context)
 			if err != nil {
 				return err
 			}
@@ -164,7 +164,7 @@ var (
 				quiet = context.Bool("quiet")
 				args  = []string(context.Args())
 			)
-			client, ctx, cancel, err := newClient(context)
+			client, ctx, cancel, err := commands.NewClient(context)
 			if err != nil {
 				return err
 			}
@@ -214,7 +214,7 @@ var (
 		Flags:       []cli.Flag{},
 		Action: func(context *cli.Context) error {
 			object, labels := commands.ObjectWithLabelArgs(context)
-			client, ctx, cancel, err := newClient(context)
+			client, ctx, cancel, err := commands.NewClient(context)
 			if err != nil {
 				return err
 			}
@@ -289,7 +289,7 @@ var (
 			if err != nil {
 				return err
 			}
-			client, ctx, cancel, err := newClient(context)
+			client, ctx, cancel, err := commands.NewClient(context)
 			if err != nil {
 				return err
 			}
@@ -338,7 +338,7 @@ var (
 				args      = []string(context.Args())
 				exitError error
 			)
-			client, ctx, cancel, err := newClient(context)
+			client, ctx, cancel, err := commands.NewClient(context)
 			if err != nil {
 				return err
 			}

cmd/ctr/events.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 
 	eventsapi "github.com/containerd/containerd/api/services/events/v1"
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/containerd/typeurl"
 	"github.com/urfave/cli"
 )
@@ -13,7 +14,7 @@ var eventsCommand = cli.Command{
 	Name:  "events",
 	Usage: "display containerd events",
 	Action: func(context *cli.Context) error {
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/exec.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/containerd/console"
 	"github.com/containerd/containerd"
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/sirupsen/logrus"
 	"github.com/urfave/cli"
 )
@@ -36,7 +37,7 @@ var taskExecCommand = cli.Command{
 		if id == "" {
 			return errors.New("container id must be provided")
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/export.go

@@ -4,6 +4,7 @@ import (
 	"io"
 	"os"
 
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/containerd/containerd/reference"
 	digest "github.com/opencontainers/go-digest"
 	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
@@ -38,7 +39,7 @@ var imagesExportCommand = cli.Command{
 			local = context.Args().Get(1)
 			desc  ocispec.Descriptor
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/fetch.go

@@ -51,7 +51,7 @@ Most of this is experimental and there are few leaps to make this work.`,
 }
 
 func fetch(ref string, cliContext *cli.Context) (containerd.Image, error) {
-	client, ctx, cancel, err := newClient(cliContext)
+	client, ctx, cancel, err := commands.NewClient(cliContext)
 	if err != nil {
 		return nil, err
 	}

cmd/ctr/fetchobject.go

@@ -23,7 +23,7 @@ var fetchObjectCommand = cli.Command{
 		var (
 			ref = context.Args().First()
 		)
-		ctx, cancel := appContext(context)
+		ctx, cancel := commands.AppContext(context)
 		defer cancel()
 
 		resolver, err := getResolver(ctx, context)

cmd/ctr/images.go

@@ -47,7 +47,7 @@ var imagesListCommand = cli.Command{
 			filters = context.Args()
 			quiet   = context.Bool("quiet")
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -130,7 +130,7 @@ var imagesSetLabelsCommand = cli.Command{
 			replaceAll   = context.Bool("replace-all")
 			name, labels = commands.ObjectWithLabelArgs(context)
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -183,7 +183,7 @@ var imagesCheckCommand = cli.Command{
 		var (
 			exitErr error
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -263,7 +263,7 @@ var imageRemoveCommand = cli.Command{
 	Description: `Remove one or more images by reference.`,
 	Flags:       []cli.Flag{},
 	Action: func(context *cli.Context) error {
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/import.go

@@ -31,7 +31,7 @@ var imagesImportCommand = cli.Command{
 			refObject = context.String("ref-object")
 			labels    = commands.LabelArgs(context.StringSlice("label"))
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/info.go

@@ -15,7 +15,7 @@ var containerInfoCommand = cli.Command{
 		if id == "" {
 			return errors.New("container id must be provided")
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/kill.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/pkg/errors"
 	"github.com/urfave/cli"
 )
@@ -41,7 +42,7 @@ var taskKillCommand = cli.Command{
 		if pid > 0 && all {
 			return errors.New("enter a pid or all; not both")
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/labels.go

@@ -20,7 +20,7 @@ var containersSetLabelsCommand = cli.Command{
 		if containerID == "" {
 			return errors.New("please specify a container")
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/namespaces.go

@@ -35,7 +35,7 @@ var namespacesCreateCommand = cli.Command{
 		if namespace == "" {
 			return errors.New("please specify a namespace")
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -56,7 +56,7 @@ var namespacesSetLabelsCommand = cli.Command{
 		if namespace == "" {
 			return errors.New("please specify a namespace")
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -85,7 +85,7 @@ var namespacesListCommand = cli.Command{
 	},
 	Action: func(context *cli.Context) error {
 		quiet := context.Bool("quiet")
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -131,7 +131,7 @@ var namespacesRemoveCommand = cli.Command{
 	Description: "Remove one or more namespaces. For now, the namespace must be empty.",
 	Action: func(context *cli.Context) error {
 		var exitErr error
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/pause.go

@@ -1,13 +1,16 @@
 package main
 
-import "github.com/urfave/cli"
+import (
+	"github.com/containerd/containerd/cmd/ctr/commands"
+	"github.com/urfave/cli"
+)
 
 var taskPauseCommand = cli.Command{
 	Name:      "pause",
 	Usage:     "pause an existing container",
 	ArgsUsage: "CONTAINER",
 	Action: func(context *cli.Context) error {
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/plugins.go

@@ -9,6 +9,7 @@ import (
 
 	introspection "github.com/containerd/containerd/api/services/introspection/v1"
 	"github.com/containerd/containerd/api/types"
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/containerd/containerd/platforms"
 	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
 	"github.com/urfave/cli"
@@ -33,7 +34,7 @@ var pluginsCommand = cli.Command{
 			quiet    = context.Bool("quiet")
 			detailed = context.Bool("detailed")
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/ps.go

@@ -5,6 +5,7 @@ import (
 	"os"
 	"text/tabwriter"
 
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/containerd/containerd/windows/hcsshimtypes"
 	"github.com/pkg/errors"
 	"github.com/urfave/cli"
@@ -19,7 +20,7 @@ var taskPsCommand = cli.Command{
 		if id == "" {
 			return errors.New("container id must be provided")
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/pull.go

@@ -27,7 +27,7 @@ command. As part of this process, we do the following:
 			ref = context.Args().First()
 		)
 
-		ctx, cancel := appContext(context)
+		ctx, cancel := commands.AppContext(context)
 		defer cancel()
 
 		img, err := fetch(ref, context)

cmd/ctr/push.go

@@ -52,7 +52,7 @@ var pushCommand = cli.Command{
 			local = context.Args().Get(1)
 			desc  ocispec.Descriptor
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/pushobject.go

@@ -27,7 +27,7 @@ var pushObjectCommand = cli.Command{
 		if err != nil {
 			return err
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/resume.go

@@ -1,13 +1,16 @@
 package main
 
-import "github.com/urfave/cli"
+import (
+	"github.com/containerd/containerd/cmd/ctr/commands"
+	"github.com/urfave/cli"
+)
 
 var taskResumeCommand = cli.Command{
 	Name:      "resume",
 	Usage:     "resume a paused container",
 	ArgsUsage: "CONTAINER",
 	Action: func(context *cli.Context) error {
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/rootfs.go

@@ -28,7 +28,7 @@ var rootfsUnpackCommand = cli.Command{
 		if err != nil {
 			return err
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/run.go

@@ -118,7 +118,7 @@ var runCommand = cli.Command{
 		if id == "" {
 			return errors.New("container id must be provided")
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/snapshot.go

@@ -38,7 +38,7 @@ var listSnapshotCommand = cli.Command{
 	Aliases: []string{"ls"},
 	Usage:   "list snapshots",
 	Action: func(context *cli.Context) error {
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -83,7 +83,7 @@ var usageSnapshotCommand = cli.Command{
 				return progress.Bytes(s).String()
 			}
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -124,7 +124,7 @@ var removeSnapshotCommand = cli.Command{
 	ArgsUsage: "<key> [<key>, ...]",
 	Usage:     "remove snapshots",
 	Action: func(context *cli.Context) error {
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -160,7 +160,7 @@ var prepareSnapshotCommand = cli.Command{
 			key    = context.Args().Get(0)
 			parent = context.Args().Get(1)
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -199,7 +199,7 @@ var viewSnapshotCommand = cli.Command{
 			key    = context.Args().Get(0)
 			parent = context.Args().Get(1)
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -232,7 +232,7 @@ var mountSnapshotCommand = cli.Command{
 			target = context.Args().Get(0)
 			key    = context.Args().Get(1)
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -261,7 +261,7 @@ var commitSnapshotCommand = cli.Command{
 			key    = context.Args().Get(0)
 			active = context.Args().Get(1)
 		)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -275,7 +275,7 @@ var treeSnapshotCommand = cli.Command{
 	Name:  "tree",
 	Usage: "display tree view of snapshot branches",
 	Action: func(context *cli.Context) error {
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -315,7 +315,7 @@ var infoSnapshotCommand = cli.Command{
 		}
 
 		key := context.Args().Get(0)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
@@ -340,7 +340,7 @@ var labelSnapshotCommand = cli.Command{
 	Flags:       []cli.Flag{},
 	Action: func(context *cli.Context) error {
 		key, labels := commands.ObjectWithLabelArgs(context)
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/start.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"github.com/containerd/console"
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/pkg/errors"
 	"github.com/sirupsen/logrus"
 	"github.com/urfave/cli"
@@ -25,7 +26,7 @@ var taskStartCommand = cli.Command{
 		if id == "" {
 			return errors.New("container id must be provided")
 		}
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/task.go

@@ -6,6 +6,7 @@ import (
 	"text/tabwriter"
 
 	tasks "github.com/containerd/containerd/api/services/tasks/v1"
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/urfave/cli"
 )
 
@@ -32,7 +33,7 @@ var tasksCommand = cli.Command{
 	},
 	Action: func(context *cli.Context) error {
 		quiet := context.Bool("quiet")
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/task_delete.go

@@ -1,13 +1,16 @@
 package main
 
-import "github.com/urfave/cli"
+import (
+	"github.com/containerd/containerd/cmd/ctr/commands"
+	"github.com/urfave/cli"
+)
 
 var taskDeleteCommand = cli.Command{
 	Name:      "delete",
 	Usage:     "delete a task",
 	ArgsUsage: "CONTAINER",
 	Action: func(context *cli.Context) error {
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}

cmd/ctr/utils.go

@@ -16,8 +16,6 @@ import (
 	"time"
 
 	"github.com/containerd/console"
-	"github.com/containerd/containerd"
-	"github.com/containerd/containerd/namespaces"
 	"github.com/containerd/containerd/remotes"
 	"github.com/containerd/containerd/remotes/docker"
 	specs "github.com/opencontainers/runtime-spec/specs-go"
@@ -26,39 +24,6 @@ import (
 	"github.com/urfave/cli"
 )
 
-// appContext returns the context for a command. Should only be called once per
-// command, near the start.
-//
-// This will ensure the namespace is picked up and set the timeout, if one is
-// defined.
-func appContext(clicontext *cli.Context) (gocontext.Context, gocontext.CancelFunc) {
-	var (
-		ctx       = gocontext.Background()
-		timeout   = clicontext.GlobalDuration("timeout")
-		namespace = clicontext.GlobalString("namespace")
-		cancel    gocontext.CancelFunc
-	)
-
-	ctx = namespaces.WithNamespace(ctx, namespace)
-
-	if timeout > 0 {
-		ctx, cancel = gocontext.WithTimeout(ctx, timeout)
-	} else {
-		ctx, cancel = gocontext.WithCancel(ctx)
-	}
-
-	return ctx, cancel
-}
-
-func newClient(context *cli.Context) (*containerd.Client, gocontext.Context, gocontext.CancelFunc, error) {
-	client, err := containerd.New(context.GlobalString("address"))
-	if err != nil {
-		return nil, nil, nil, err
-	}
-	ctx, cancel := appContext(context)
-	return client, ctx, cancel, nil
-}
-
 func passwordPrompt() (string, error) {
 	c := console.Current()
 	defer c.Reset()

cmd/ctr/version.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"os"
 
+	"github.com/containerd/containerd/cmd/ctr/commands"
 	"github.com/containerd/containerd/version"
 	"github.com/urfave/cli"
 )
@@ -19,7 +20,7 @@ var versionCommand = cli.Command{
 		fmt.Printf("  Version: %s\n", version.Version)
 		fmt.Printf("  Revision: %s\n", version.Revision)
 		fmt.Println("")
-		client, ctx, cancel, err := newClient(context)
+		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}