From 3622fa2b61e0ef7b1b9d76906b7f38079e84d1b3 Mon Sep 17 00:00:00 2001
From: Denis Arh <denis.arh@gmail.com>
Date: Thu, 26 Mar 2020 15:26:33 +0100
Subject: [PATCH] Add support for find-by-any on module

---
 compose/service/module.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/compose/service/module.go b/compose/service/module.go
index 70e268323..08918c0ef 100644
--- a/compose/service/module.go
+++ b/compose/service/module.go
@@ -2,6 +2,7 @@ package service
 
 import (
 	"context"
+	"strconv"
 
 	"github.com/titpetric/factory"
 	"go.uber.org/zap"
@@ -47,6 +48,7 @@ type (
 		FindByID(namespaceID, moduleID uint64) (*types.Module, error)
 		FindByName(namespaceID uint64, name string) (*types.Module, error)
 		FindByHandle(namespaceID uint64, handle string) (*types.Module, error)
+		FindByAny(namespaceID uint64, identifier interface{}) (*types.Module, error)
 		Find(filter types.ModuleFilter) (set types.ModuleSet, f types.ModuleFilter, err error)
 
 		Create(module *types.Module) (*types.Module, error)
@@ -109,6 +111,30 @@ func (svc module) FindByHandle(namespaceID uint64, handle string) (m *types.Modu
 	}
 }
 
+// FindByAny tries to find module in a particular namespace by id, handle or name
+func (svc module) FindByAny(namespaceID uint64, identifier interface{}) (r *types.Module, err error) {
+	if ID, ok := identifier.(uint64); ok {
+		r, err = svc.FindByID(namespaceID, ID)
+	} else if strIdentifier, ok := identifier.(string); ok {
+		if ID, _ := strconv.ParseUint(strIdentifier, 10, 64); ID > 0 {
+			r, err = svc.FindByID(namespaceID, ID)
+		} else {
+			r, err = svc.FindByHandle(namespaceID, strIdentifier)
+			if err == nil && r.ID == 0 {
+				r, err = svc.FindByName(namespaceID, strIdentifier)
+			}
+		}
+	} else {
+		err = ErrInvalidID.withStack()
+	}
+
+	if err != nil {
+		return
+	}
+
+	return
+}
+
 func (svc module) loader(m *types.Module, err error) (*types.Module, error) {
 	if err != nil {
 		return nil, err