feat: Add wildcard support for services name

Signed-off-by: Suleiman Dibirov <[email protected]>

Author: idsulik

types/project.go

@@ -188,16 +188,25 @@ func (p *Project) getServicesByNames(names ...string) (Services, []string) {
 	if len(names) == 0 {
 		return p.Services, nil
 	}
+
 	services := Services{}
 	var servicesNotFound []string
 	for _, name := range names {
-		service, ok := p.Services[name]
-		if !ok {
+		matched := false
+
+		for serviceName, service := range p.Services {
+			match, _ := filepath.Match(name, serviceName)
+			if match {
+				services[serviceName] = service
+				matched = true
+			}
+		}
+
+		if !matched {
 			servicesNotFound = append(servicesNotFound, name)
-			continue
 		}
-		services[name] = service
 	}
+
 	return services, servicesNotFound
 }
 

types/project_test.go

@@ -20,6 +20,7 @@ import (
 	_ "crypto/sha256"
 	"errors"
 	"fmt"
+	"sort"
 	"strings"
 	"testing"
 
@@ -158,6 +159,12 @@ func makeProject() *Project {
 				Name:  "service_6",
 				Links: []string{"service_1"},
 			},
+			"my_services_1": ServiceConfig{
+				Name: "service_6",
+			},
+			"my_services_2": ServiceConfig{
+				Name: "service_6",
+			},
 		},
 		Networks: Networks{},
 		Volumes:  Volumes{},
@@ -281,6 +288,18 @@ func TestWithServices(t *testing.T) {
 	assert.DeepEqual(t, seen, []string{"service_1", "service_2", "service_4"})
 }
 
+func TestWithServicesWithWildcard(t *testing.T) {
+	p := makeProject()
+	var seen []string
+	err := p.ForEachService([]string{"my_services_*"}, func(name string, _ *ServiceConfig) error {
+		seen = append(seen, name)
+		return nil
+	}, IgnoreDependencies)
+	assert.NilError(t, err)
+	sort.Strings(seen)
+	assert.DeepEqual(t, seen, []string{"my_services_1", "my_services_2"})
+}
+
 func TestServicesWithBuild(t *testing.T) {
 	p := makeProject()
 	assert.DeepEqual(t, []string{}, p.ServicesWithBuild())