feat: handle dockercompat inspect for devcontainers

Signed-off-by: Shubharanshu Mahapatra <[email protected]>
runfinch · Oct 4, 2024 · d54a2ff · d54a2ff
1 parent 5a47eec
commit d54a2ff
Show file tree

Hide file tree

Showing 8 changed files with 333 additions and 19 deletions.
diff --git a/cmd/finch/devcontainer_patch.go b/cmd/finch/devcontainer_patch.go
@@ -0,0 +1,107 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+// Package main denotes the entry point of finch CLI.
+// TODO: Remove all instances of these calls once supported upstream
+package main
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"strings"
+
+	"github.com/containerd/nerdctl/v2/pkg/inspecttypes/dockercompat"
+	"github.com/docker/go-connections/nat"
+	"github.com/sirupsen/logrus"
+
+	"github.com/runfinch/finch/pkg/command"
+)
+
+// Config is from https://github.com/moby/moby/blob/8dbd90ec00daa26dc45d7da2431c965dec99e8b4/api/types/container/config.go#L37-L69
+type Config struct {
+	Hostname     string              `json:",omitempty"` // Hostname
+	User         string              `json:",omitempty"` // User that will run the command(s) inside the container, also support user:group
+	AttachStdin  bool                // Attach the standard input, makes possible user interaction
+	ExposedPorts nat.PortSet         `json:",omitempty"` // List of exposed ports
+	Env          []string            `json:",omitempty"` // List of environment variable to set in the container
+	Cmd          []string            `json:",omitempty"` // Command to run when starting the container
+	Volumes      map[string]struct{} `json:",omitempty"` // List of volumes (mounts) used for the container
+	WorkingDir   string              `json:",omitempty"` // Current directory (PWD) in the command will be launched
+	Entrypoint   []string            `json:",omitempty"` // Entrypoint to run when starting the container
+	Labels       map[string]string   `json:",omitempty"` // List of labels set to this containerT
+	Image        string              `json:",omitempty"`
+}
+
+// Container mimics a `docker container inspect` object.
+// From https://github.com/moby/moby/blob/v20.10.1/api/types/types.go#L340-L374
+type Container struct {
+	ID              string `json:"Id"`
+	Created         string
+	Path            string
+	Args            []string
+	State           *dockercompat.ContainerState
+	Image           string
+	ResolvConfPath  string
+	HostnamePath    string
+	LogPath         string
+	Name            string
+	RestartCount    int
+	Driver          string
+	Platform        string
+	AppArmorProfile string
+	SizeRw          *int64 `json:",omitempty"`
+	SizeRootFs      *int64 `json:",omitempty"`
+	Mounts          []dockercompat.MountPoint
+	Config          *Config
+	NetworkSettings *dockercompat.NetworkSettings
+}
+
+func prettyPrintJSON(input string) {
+	var mergedData []Container
+	jsonObjects := strings.Split(input, "\n")
+
+	for i, jsonStr := range jsonObjects {
+		if len(jsonStr) == 0 {
+			continue
+		}
+		var container Container
+		err := json.Unmarshal([]byte(jsonStr), &container)
+		if err != nil {
+			logrus.Error("Error parsing JSON at index: ", i, err)
+			continue
+		}
+
+		if container.Config != nil {
+			container.Config.Image = container.Image
+		}
+
+		if container.State != nil {
+			container.State.StartedAt = "0001-01-01T00:00:00Z"
+		}
+
+		if container.NetworkSettings == nil {
+			container.NetworkSettings = &dockercompat.NetworkSettings{
+				Ports: &nat.PortMap{},
+			}
+		}
+
+		mergedData = append(mergedData, container)
+	}
+
+	finalJSON, err := json.MarshalIndent(mergedData, "", "  ")
+	if err != nil {
+		logrus.Error("Error marshaling final JSON: ", err)
+		return
+	}
+
+	fmt.Println(string(finalJSON))
+}
+
+func inspectContainerOutputHandler(cmd command.Command) error {
+	var stdoutBuf bytes.Buffer
+	cmd.SetStdout(&stdoutBuf)
+	err := cmd.Run()
+	prettyPrintJSON(stdoutBuf.String())
+	return err
+}
diff --git a/cmd/finch/nerdctl.go b/cmd/finch/nerdctl.go
@@ -45,7 +45,7 @@ type nerdctlCommandCreator struct {
 
 type (
 	argHandler     func(systemDeps NerdctlCommandSystemDeps, fc *config.Finch, args []string, index int) error
-	commandHandler func(systemDeps NerdctlCommandSystemDeps, fc *config.Finch, cmdName *string, args *[]string) error
+	commandHandler func(systemDeps NerdctlCommandSystemDeps, fc *config.Finch, cmdName *string, args *[]string, inspectType *string) error
 )
 
 func newNerdctlCommandCreator(
@@ -256,7 +256,7 @@ func handleDockerBuildLoad(_ NerdctlCommandSystemDeps, fc *config.Finch, nerdctl
 	return nil
 }
 
-func handleBuildx(_ NerdctlCommandSystemDeps, fc *config.Finch, cmdName *string, args *[]string) error {
+func handleBuildx(_ NerdctlCommandSystemDeps, fc *config.Finch, cmdName *string, args *[]string, _ *string) error {
 	if fc == nil || !fc.DockerCompat {
 		return nil
 	}
@@ -279,7 +279,7 @@ func handleBuildx(_ NerdctlCommandSystemDeps, fc *config.Finch, cmdName *string,
 	return nil
 }
 
-func handleDockerCompatInspect(_ NerdctlCommandSystemDeps, fc *config.Finch, cmdName *string, args *[]string) error {
+func handleDockerCompatInspect(_ NerdctlCommandSystemDeps, fc *config.Finch, cmdName *string, args *[]string, inspectType *string) error {
 	if fc == nil || !fc.DockerCompat {
 		return nil
 	}
@@ -289,10 +289,10 @@ func handleDockerCompatInspect(_ NerdctlCommandSystemDeps, fc *config.Finch, cmd
 	}
 
 	modeDockerCompat := `--mode=dockercompat`
-	inspectType := ""
 	sizeArg := ""
 	savedArgs := []string{}
 	skip := false
+	*inspectType = ""
 
 	for idx, arg := range *args {
 		if skip {
@@ -301,13 +301,13 @@ func handleDockerCompatInspect(_ NerdctlCommandSystemDeps, fc *config.Finch, cmd
 		}
 
 		if (arg == "--type") && (idx < len(*args)-1) {
-			inspectType = (*args)[idx+1]
+			*inspectType = (*args)[idx+1]
 			skip = true
 			continue
 		}
 
 		if strings.Contains(arg, "--type") && strings.Contains(arg, "=") {
-			inspectType = strings.Split(arg, "=")[1]
+			*inspectType = strings.Split(arg, "=")[1]
 			continue
 		}
 
@@ -319,7 +319,7 @@ func handleDockerCompatInspect(_ NerdctlCommandSystemDeps, fc *config.Finch, cmd
 		savedArgs = append(savedArgs, arg)
 	}
 
-	switch inspectType {
+	switch *inspectType {
 	case "image":
 		*cmdName = "image inspect"
 		*args = append([]string{modeDockerCompat}, savedArgs...)
@@ -336,8 +336,9 @@ func handleDockerCompatInspect(_ NerdctlCommandSystemDeps, fc *config.Finch, cmd
 	case "":
 		*cmdName = "inspect"
 		*args = append([]string{modeDockerCompat}, savedArgs...)
+		*inspectType = "container"
 	default:
-		return fmt.Errorf("unsupported inspect type: %s", inspectType)
+		return fmt.Errorf("unsupported inspect type: %s", *inspectType)
 	}
 
 	return nil

diff --git a/cmd/finch/nerdctl_darwin_test.go b/cmd/finch/nerdctl_darwin_test.go
@@ -6,6 +6,7 @@
 package main
 
 import (
+	"bytes"
 	"errors"
 	"fmt"
 	"os"
@@ -1306,7 +1307,9 @@ func TestNerdctlCommand_run_inspectCommand(t *testing.T) {
 				ncsd.EXPECT().LookupEnv("COSIGN_PASSWORD").Return("", false)
 				ncsd.EXPECT().LookupEnv("COMPOSE_FILE").Return("", false)
 				c := mocks.NewCommand(ctrl)
-				lcc.EXPECT().Create("shell", limaInstanceName, "sudo", "-E", nerdctlCmdName, "inspect", "--mode=dockercompat", "da24").Return(c)
+				var stdoutBuf bytes.Buffer
+				c.EXPECT().SetStdout(&stdoutBuf)
+				lcc.EXPECT().Create("shell", limaInstanceName, "sudo", "-E", nerdctlCmdName, "inspect", "--mode=dockercompat", "da24", "--format", "{{json .}}").Return(c)
 				c.EXPECT().Run()
 			},
 		},
@@ -1339,7 +1342,9 @@ func TestNerdctlCommand_run_inspectCommand(t *testing.T) {
 				ncsd.EXPECT().LookupEnv("COSIGN_PASSWORD").Return("", false)
 				ncsd.EXPECT().LookupEnv("COMPOSE_FILE").Return("", false)
 				c := mocks.NewCommand(ctrl)
-				lcc.EXPECT().Create("shell", limaInstanceName, "sudo", "-E", nerdctlCmdName, "inspect", "--mode=dockercompat", "44de").Return(c)
+				var stdoutBuf bytes.Buffer
+				c.EXPECT().SetStdout(&stdoutBuf)
+				lcc.EXPECT().Create("shell", limaInstanceName, "sudo", "-E", nerdctlCmdName, "inspect", "--mode=dockercompat", "44de", "--format", "{{json .}}").Return(c)
 				c.EXPECT().Run()
 			},
 		},
@@ -1448,7 +1453,9 @@ func TestNerdctlCommand_run_inspectCommand(t *testing.T) {
 				ncsd.EXPECT().LookupEnv("COSIGN_PASSWORD").Return("", false)
 				ncsd.EXPECT().LookupEnv("COMPOSE_FILE").Return("", false)
 				c := mocks.NewCommand(ctrl)
-				lcc.EXPECT().Create("shell", limaInstanceName, "sudo", "-E", nerdctlCmdName, "inspect", "--mode=dockercompat", "44de").Return(c)
+				var stdoutBuf bytes.Buffer
+				c.EXPECT().SetStdout(&stdoutBuf)
+				lcc.EXPECT().Create("shell", limaInstanceName, "sudo", "-E", nerdctlCmdName, "inspect", "--mode=dockercompat", "44de", "--format", "{{json .}}").Return(c)
 				c.EXPECT().Run()
 			},
 		},

diff --git a/cmd/finch/nerdctl_native.go b/cmd/finch/nerdctl_native.go
@@ -21,6 +21,7 @@ func (nc *nerdctlCommand) run(cmdName string, args []string) error {
 		cmdHandler                   commandHandler
 		aMap                         map[string]argHandler
 		err                          error
+		inspectType                  string
 	)
 
 	// eat the debug arg, and set the log level to avoid nerdctl parsing this flag
@@ -48,7 +49,7 @@ func (nc *nerdctlCommand) run(cmdName string, args []string) error {
 
 	// First check if the command has a command handler
 	if hasCmdHandler {
-		err := cmdHandler(nc.systemDeps, nc.fc, &cmdName, &args)
+		err := cmdHandler(nc.systemDeps, nc.fc, &cmdName, &args, &inspectType)
 		if err != nil {
 			return err
 		}
@@ -81,7 +82,14 @@ func (nc *nerdctlCommand) run(cmdName string, args []string) error {
 		)
 	}
 
+	if inspectType == "container" && nc.fc.DockerCompat && !slices.Contains(cmdArgs, "--format") {
+		cmdArgs = append(cmdArgs, "--format", "{{json .}}")
+		cmd := nc.ncc.Create(cmdArgs...)
+		return inspectContainerOutputHandler(cmd)
+	}
+
 	return nc.ncc.Create(cmdArgs...).Run()
+
 }
 
 var osAliasMap = map[string]string{}

diff --git a/cmd/finch/nerdctl_remote.go b/cmd/finch/nerdctl_remote.go
@@ -37,6 +37,7 @@ func (nc *nerdctlCommand) run(cmdName string, args []string) error {
 		cmdHandler                                    commandHandler
 		aMap                                          map[string]argHandler
 		firstOptPos                                   int
+		inspectType                                   string
 	)
 
 	// accumulate distributed map entities
@@ -80,7 +81,7 @@ func (nc *nerdctlCommand) run(cmdName string, args []string) error {
 
 	// First check if the command has command handler
 	if hasCmdHandler {
-		err := cmdHandler(nc.systemDeps, nc.fc, &cmdName, &args)
+		err := cmdHandler(nc.systemDeps, nc.fc, &cmdName, &args, &inspectType)
 		if err != nil {
 			return err
 		}
@@ -333,6 +334,12 @@ func (nc *nerdctlCommand) run(cmdName string, args []string) error {
 		return nc.ncc.RunWithReplacingStdout([]command.Replacement{{Source: "nerdctl", Target: "finch"}}, runArgs...)
 	}
 
+	if inspectType == "container" && nc.fc.DockerCompat && !slices.Contains(runArgs, "--format") {
+		runArgs = append(runArgs, "--format", "{{json .}}")
+		cmd := nc.ncc.Create(runArgs...)
+		return inspectContainerOutputHandler(cmd)
+	}
+
 	return nc.ncc.Create(runArgs...).Run()
 }
 

diff --git a/cmd/finch/nerdctl_windows.go b/cmd/finch/nerdctl_windows.go
@@ -262,7 +262,7 @@ func handleSecretOption(systemDeps NerdctlCommandSystemDeps, _ *config.Finch, ne
 }
 
 // cp command handler, takes command arguments and converts hostpath to wsl path in place. It ignores all other arguments.
-func cpHandler(systemDeps NerdctlCommandSystemDeps, _ *config.Finch, _ *string, nerdctlCmdArgs *[]string) error {
+func cpHandler(systemDeps NerdctlCommandSystemDeps, _ *config.Finch, _ *string, nerdctlCmdArgs *[]string, _ *string) error {
 	for i, arg := range *nerdctlCmdArgs {
 		// -L and --follow-symlink don't have to be processed
 		if strings.HasPrefix(arg, "-") || arg == "cp" {
@@ -285,7 +285,7 @@ func cpHandler(systemDeps NerdctlCommandSystemDeps, _ *config.Finch, _ *string,
 }
 
 // this is the handler for image build command. It translates build context to wsl path.
-func imageBuildHandler(systemDeps NerdctlCommandSystemDeps, _ *config.Finch, _ *string, nerdctlCmdArgs *[]string) error {
+func imageBuildHandler(systemDeps NerdctlCommandSystemDeps, _ *config.Finch, _ *string, nerdctlCmdArgs *[]string, _ *string) error {
 	var err error
 	argLen := len(*nerdctlCmdArgs) - 1
 	// -h/--help don't have buildcontext, just return