Read files from the volume

Signed-off-by: David Gageot <[email protected]>

Author: dgageot

src/extension/host-binary/cmd/commands/connection_other.go

@@ -0,0 +1,21 @@
+//go:build !windows
+// +build !windows
+
+package commands
+
+import (
+	"context"
+	"net"
+
+	"github.com/docker/labs-ai-tools-for-devs/pkg/paths"
+)
+
+func dialVolumeContents(ctx context.Context) (net.Conn, error) {
+	path, err := paths.GetVolumeContentsSocketPath()
+	if err != nil {
+		return nil, err
+	}
+
+	dialer := net.Dialer{}
+	return dialer.DialContext(ctx, "unix", path)
+}

src/extension/host-binary/cmd/commands/connection_windows.go

@@ -0,0 +1,18 @@
+package commands
+
+import (
+	"context"
+	"net"
+
+	"github.com/Microsoft/go-winio"
+	"github.com/docker/labs-ai-tools-for-devs/pkg/paths"
+)
+
+func dialVolumeContents(ctx context.Context) (net.Conn, error) {
+	path, err := paths.GetVolumeContentsSocketPath()
+	if err != nil {
+		return nil, err
+	}
+
+	return winio.DialPipeContext(ctx, path)
+}

src/extension/host-binary/cmd/commands/volume.go

@@ -0,0 +1,71 @@
+package commands
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net"
+	"net/http"
+
+	"github.com/spf13/cobra"
+)
+
+func ReadFromVolume(ctx context.Context) *cobra.Command {
+	return &cobra.Command{
+		Use:   "read-from-volume",
+		Short: "Read a file from the extension's volume",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			content, err := readConfig(ctx, args[0])
+			if err != nil {
+				return err
+			}
+			fmt.Print(content)
+			return nil
+		},
+	}
+}
+
+func readConfig(ctx context.Context, filename string) (string, error) {
+	httpClient := &http.Client{
+		Transport: &http.Transport{
+			DialContext: func(ctx context.Context, _, _ string) (conn net.Conn, err error) {
+				return dialVolumeContents(ctx)
+			},
+		},
+	}
+
+	var content struct {
+		Contents string `json:"contents"`
+	}
+	if err := query(ctx, httpClient, "GET", "/volume-file-content?volumeId=docker-prompts&targetPath="+filename, &content); err != nil {
+		return "", err
+	}
+
+	return content.Contents, nil
+}
+
+func query(ctx context.Context, httpClient *http.Client, method string, endpoint string, v any) error {
+	req, err := http.NewRequestWithContext(ctx, method, "http://localhost"+endpoint, nil)
+	if err != nil {
+		return err
+	}
+	req.Header.Set("X-DockerDesktop-Host", "vm.docker.internal")
+
+	response, err := httpClient.Do(req)
+	if err != nil {
+		return err
+	}
+	defer response.Body.Close()
+
+	buf, err := io.ReadAll(response.Body)
+	if err != nil {
+		return err
+	}
+
+	if err := json.Unmarshal(buf, &v); err != nil {
+		return err
+	}
+	return nil
+}

src/extension/host-binary/cmd/main.go

@@ -28,6 +28,7 @@ func main() {
 	cmd.AddCommand(ListOAuthApps(ctx))
 	cmd.AddCommand(DeriveSecret(ctx))
 	cmd.AddCommand(commands.CurrentUser(ctx))
+	cmd.AddCommand(commands.ReadFromVolume(ctx))
 	if err := cmd.Execute(); err != nil {
 		fmt.Println(err)
 		os.Exit(1)

src/extension/host-binary/pkg/paths/paths_unix.go

@@ -19,3 +19,11 @@ func GetToolsApiSocketPath() (string, error) {
 	}
 	return filepath.Join(dir, "tools.sock"), nil
 }
+
+func GetVolumeContentsSocketPath() (string, error) {
+	dir, err := dockerDesktopSocketDir()
+	if err != nil {
+		return "", err
+	}
+	return filepath.Join(dir, "volume-contents.sock"), nil
+}

src/extension/host-binary/pkg/paths/paths_windows.go

@@ -9,3 +9,7 @@ func GetSecretsApiSocketPath() (string, error) {
 func GetToolsApiSocketPath() (string, error) {
 	return `//./pipe/dockerTools`, nil
 }
+
+func GetVolumeContentsSocketPath() (string, error) {
+	return `//./pipe/dockerVolumeContents`, nil
+}

src/extension/ui/src/utils/Files.ts

@@ -55,21 +55,14 @@ export const readFileInPromptsVolume = async (
   client: v1.DockerDesktopClient,
   path: string
 ) => {
-  return tryRunImageSync(
-    client,
-    [
-      "--rm",
-      "-v",
-      "docker-prompts:/docker-prompts:ro",
-      "--network=none",
-      "-w",
-      "/docker-prompts",
-      BUSYBOX,
-      "/bin/cat",
-      `${path}`,
-    ],
-    true
-  );
+  try {
+    const result = await client.extension.host?.cli.exec("host-binary", ["read-from-volume", path]);
+    if (result) {
+      return result.stdout;
+    }
+  } catch { }
+
+  return ""
 };
 
 export const writeToPromptsVolume = async (