Check Docker Compose services are running (#17)

Author: vidya381

internal/check/compose.go

@@ -0,0 +1,75 @@
+package check
+
+import (
+	"bufio"
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"os/exec"
+	"strings"
+)
+
+type ComposeCheck struct {
+	runner func() ([]byte, error)
+}
+
+func (c *ComposeCheck) Name() string {
+	return "Docker Compose services running"
+}
+
+type composeService struct {
+	Name  string `json:"Name"`
+	State string `json:"State"`
+}
+
+func (c *ComposeCheck) Run(_ context.Context) Result {
+	run := c.runner
+	if run == nil {
+		run = func() ([]byte, error) {
+			return exec.Command("docker", "compose", "ps", "--format", "json").Output()
+		}
+	}
+
+	out, err := run()
+	if err != nil {
+		return Result{
+			Name:    c.Name(),
+			Status:  StatusFail,
+			Message: "could not run docker compose ps",
+			Fix:     "make sure Docker is running and you are in the project directory",
+		}
+	}
+
+	// docker compose ps --format json outputs one JSON object per line
+	var stopped []string
+	scanner := bufio.NewScanner(bytes.NewReader(out))
+	for scanner.Scan() {
+		line := strings.TrimSpace(scanner.Text())
+		if line == "" {
+			continue
+		}
+		var svc composeService
+		if err := json.Unmarshal([]byte(line), &svc); err != nil {
+			continue
+		}
+		if svc.State != "running" {
+			stopped = append(stopped, svc.Name)
+		}
+	}
+
+	if len(stopped) > 0 {
+		return Result{
+			Name:    c.Name(),
+			Status:  StatusFail,
+			Message: fmt.Sprintf("services not running: %s", strings.Join(stopped, ", ")),
+			Fix:     "run docker compose up -d to start them",
+		}
+	}
+
+	return Result{
+		Name:    c.Name(),
+		Status:  StatusPass,
+		Message: "all services are running",
+	}
+}

internal/check/compose_test.go

@@ -0,0 +1,49 @@
+package check
+
+import (
+	"context"
+	"errors"
+	"testing"
+)
+
+func TestComposeCheck_AllRunning(t *testing.T) {
+	c := &ComposeCheck{runner: func() ([]byte, error) {
+		return []byte(`{"Name":"app-db-1","State":"running"}` + "\n" +
+			`{"Name":"app-web-1","State":"running"}` + "\n"), nil
+	}}
+	result := c.Run(context.Background())
+	if result.Status != StatusPass {
+		t.Errorf("expected pass, got %v: %s", result.Status, result.Message)
+	}
+}
+
+func TestComposeCheck_SomeStopped(t *testing.T) {
+	c := &ComposeCheck{runner: func() ([]byte, error) {
+		return []byte(`{"Name":"app-db-1","State":"running"}` + "\n" +
+			`{"Name":"app-web-1","State":"exited"}` + "\n"), nil
+	}}
+	result := c.Run(context.Background())
+	if result.Status != StatusFail {
+		t.Errorf("expected fail, got %v", result.Status)
+	}
+}
+
+func TestComposeCheck_CommandFails(t *testing.T) {
+	c := &ComposeCheck{runner: func() ([]byte, error) {
+		return nil, errors.New("docker not found")
+	}}
+	result := c.Run(context.Background())
+	if result.Status != StatusFail {
+		t.Errorf("expected fail, got %v", result.Status)
+	}
+}
+
+func TestComposeCheck_NoServices(t *testing.T) {
+	c := &ComposeCheck{runner: func() ([]byte, error) {
+		return []byte(""), nil
+	}}
+	result := c.Run(context.Background())
+	if result.Status != StatusPass {
+		t.Errorf("expected pass, got %v: %s", result.Status, result.Message)
+	}
+}

internal/check/registry.go

@@ -35,6 +35,9 @@ func Build(stack detector.DetectedStack) []Check {
 		cs = append(cs, &BinaryCheck{Binary: "docker"})
 		cs = append(cs, &DockerDaemonCheck{})
 	}
+	if stack.DockerCompose {
+		cs = append(cs, &ComposeCheck{})
+	}
 	if stack.Postgres {
 		cs = append(cs, &PostgresCheck{URL: os.Getenv("DATABASE_URL")})
 	}

internal/detector/detector.go

@@ -13,7 +13,8 @@ type DetectedStack struct {
 	Java     bool
 	Maven    bool
 	Gradle   bool
-	Docker   bool
+	Docker        bool
+	DockerCompose bool
 	Postgres bool
 	Redis    bool
 	MySQL    bool
@@ -31,9 +32,9 @@ func Detect(dir string) DetectedStack {
 	stack.Maven = fileExists(filepath.Join(dir, "pom.xml"))
 	stack.Gradle = fileExists(filepath.Join(dir, "build.gradle"))
 	stack.Java = stack.Maven || stack.Gradle
-	stack.Docker = fileExists(filepath.Join(dir, "Dockerfile")) ||
-		fileExists(filepath.Join(dir, "docker-compose.yml")) ||
+	stack.DockerCompose = fileExists(filepath.Join(dir, "docker-compose.yml")) ||
 		fileExists(filepath.Join(dir, "docker-compose.yaml"))
+	stack.Docker = fileExists(filepath.Join(dir, "Dockerfile")) || stack.DockerCompose
 
 	dbURL := os.Getenv("DATABASE_URL")
 	stack.Postgres = strings.Contains(dbURL, "postgres")