feat(docker): list & start

unshade · unshade · commit 2ff6af148dd1 · 2025-11-11T23:36:18.000+01:00
diff --git a/cmd/docker.go b/cmd/docker.go
@@ -19,19 +19,19 @@ This application is a tool to generate the needed files
 to quickly create a Cobra application.`,
 	Run: func(cmd *cobra.Command, args []string) {
 		switch args[0] {
+		case "list":
+			unraidClient.Docker.ListContainers(cmd.Context())
 		case "stop":
 			unraidClient.Docker.StopContainer(cmd.Context(), args[1])
 		case "start":
-			// startContainer(cmd, args)
-		case "restart":
-			// restartContainer(cmd, args)
+			unraidClient.Docker.StartContainer(cmd.Context(), args[1])
 		}
 	},
 }
 
 func init() {
 	rootCmd.AddCommand(dockerCmd)
 
-	dockerCmd.Args = cobra.MinimumNArgs(2)
-	dockerCmd.ValidArgs = []string{"stop", "restart", "start"}
+	dockerCmd.Args = cobra.MinimumNArgs(1)
+	dockerCmd.ValidArgs = []string{"stop", "restart", "start", "list"}
 }
diff --git a/pkg/client/docker.go b/pkg/client/docker.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"log"
+	"strings"
 
 	"github.com/machinebox/graphql"
 )
@@ -20,6 +21,79 @@ func NewDockerClient(apiKey string, graphqlClient *graphql.Client) *DockerClient
 	}
 }
 
+func (c *DockerClient) ListContainers(ctx context.Context) {
+	query := `
+	query Query {
+		docker {
+			containers {
+				id
+				image
+				state
+			}
+		}
+	}`
+
+	req := graphql.NewRequest(query)
+	req.Header.Set(ApiKeyHeader, c.ApiKey)
+
+	var respData struct {
+		Docker struct {
+			Containers []struct {
+				ID    string `json:"id"`
+				Image string `json:"image"`
+				State string `json:"state"`
+			} `json:"containers"`
+		} `json:"docker"`
+	}
+
+	if err := c.GraphQLClient.Run(ctx, req, &respData); err != nil {
+		log.Printf("Error query list containers: %v", err)
+		return
+	}
+
+	for _, container := range respData.Docker.Containers {
+		idParts := strings.Split(container.ID, ":")
+		compactID := idParts[len(idParts)-1]
+
+		if len(compactID) > 12 {
+			compactID = compactID[:12]
+		}
+
+		fmt.Printf("ID: %s | Image: %s | State: %s\n", compactID, container.Image, container.State)
+	}
+}
+
+func (c *DockerClient) StartContainer(ctx context.Context, startId string) {
+	mutation := `
+	mutation Mutation($startId: PrefixedID!) {
+		docker {
+			start(id: $startId) {
+				id
+			}
+		}
+	}`
+
+	req := graphql.NewRequest(mutation)
+	req.Header.Set(ApiKeyHeader, c.ApiKey)
+
+	req.Var("startId", startId)
+
+	var respData struct {
+		Docker struct {
+			Start struct {
+				ID string `json:"id"`
+			} `json:"start"`
+		} `json:"docker"`
+	}
+
+	if err := c.GraphQLClient.Run(ctx, req, &respData); err != nil {
+		log.Printf("Error mutation start: %v", err)
+		return
+	}
+
+	fmt.Printf("Container started with ID: %s\n", respData.Docker.Start.ID)
+}
+
 func (c *DockerClient) StopContainer(ctx context.Context, stopId string) {
 	mutation := `
 	mutation Stop($stopId: PrefixedID!) {
