From 3a7e5dd6ccae254415c221aedeed49973ac26255 Mon Sep 17 00:00:00 2001
From: Laura Brehm <laurabrehm@hey.com>
Date: Tue, 2 Aug 2022 21:08:15 +0200
Subject: [PATCH] Filter `compose ps` output by provided compose model

Signed-off-by: Laura Brehm <laurabrehm@hey.com>
---
 cmd/compose/ps.go |  5 +++--
 pkg/api/api.go    |  1 +
 pkg/compose/ps.go | 13 +++++++++++++
 3 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/cmd/compose/ps.go b/cmd/compose/ps.go
index 0c29499856d..ca005ec1c55 100644
--- a/cmd/compose/ps.go
+++ b/cmd/compose/ps.go
@@ -91,11 +91,12 @@ func psCommand(p *projectOptions, backend api.Service) *cobra.Command {
 }
 
 func runPs(ctx context.Context, backend api.Service, services []string, opts psOptions) error {
-	projectName, err := opts.toProjectName()
+	project, name, err := opts.projectOrName()
 	if err != nil {
 		return err
 	}
-	containers, err := backend.Ps(ctx, projectName, api.PsOptions{
+	containers, err := backend.Ps(ctx, name, api.PsOptions{
+		Project:  project,
 		All:      opts.All,
 		Services: services,
 	})
diff --git a/pkg/api/api.go b/pkg/api/api.go
index 947522854c7..9c3fc240b19 100644
--- a/pkg/api/api.go
+++ b/pkg/api/api.go
@@ -280,6 +280,7 @@ type ListOptions struct {
 
 // PsOptions group options of the Ps API
 type PsOptions struct {
+	Project  *types.Project
 	All      bool
 	Services []string
 }
diff --git a/pkg/compose/ps.go b/pkg/compose/ps.go
index de4b25f2eb6..7df826a08ae 100644
--- a/pkg/compose/ps.go
+++ b/pkg/compose/ps.go
@@ -37,6 +37,19 @@ func (s *composeService) Ps(ctx context.Context, projectName string, options api
 		return nil, err
 	}
 
+	project := options.Project
+	if project == nil {
+		project, err = s.getProjectWithResources(ctx, containers, projectName)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	if len(options.Services) == 0 {
+		options.Services = project.ServiceNames()
+	}
+
+	containers = containers.filter(isService(options.Services...))
 	summary := make([]api.ContainerSummary, len(containers))
 	eg, ctx := errgroup.WithContext(ctx)
 	for i, container := range containers {