From 04cbc0bf8ce0a6bfdc1cdc7909754eb9b3cf1b31 Mon Sep 17 00:00:00 2001
From: martijnvdp <m.vanderploeg@nl.ccv.eu>
Date: Fri, 18 Aug 2023 10:00:06 +0200
Subject: [PATCH] add repo not found clause

---
 pkg/lambda/check_digest_test.go | 2 +-
 pkg/lambda/ecr.go               | 1 +
 pkg/lambda/lambda.go            | 6 ++++++
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/pkg/lambda/check_digest_test.go b/pkg/lambda/check_digest_test.go
index a77685d..c5b6307 100644
--- a/pkg/lambda/check_digest_test.go
+++ b/pkg/lambda/check_digest_test.go
@@ -57,7 +57,7 @@ func Test_checkDigest(t *testing.T) {
 					"docker.io/nginx:1.23.3": {
 						name: "nginx",
 						tag:  "1.23.3",
-						hash: "sha256:942ae2dfd73088b54d7151a3c3fd5af038a51c50029bfcfd21f1e650d9579967",
+						hash: "sha256:557c9ede65655e5a70e4a32f1651638ea3bfb0802edd982810884602f700ba25",
 					},
 				},
 			},
diff --git a/pkg/lambda/ecr.go b/pkg/lambda/ecr.go
index 6b5a065..6c4b514 100644
--- a/pkg/lambda/ecr.go
+++ b/pkg/lambda/ecr.go
@@ -105,6 +105,7 @@ func (svc *ecrClient) getECRRepositories(inputRepositories []string) (repositori
 
 	if err != nil {
 		log.Printf("Error: %s", err)
+		return nil, err
 	}
 
 	return repositories, err
diff --git a/pkg/lambda/lambda.go b/pkg/lambda/lambda.go
index 90f8833..d0e0be7 100644
--- a/pkg/lambda/lambda.go
+++ b/pkg/lambda/lambda.go
@@ -214,6 +214,12 @@ func Start(ctx context.Context, event LambdaEvent) (response, error) {
 	repositories, err = svc.getinputRepositorysFromTags(names)
 
 	if err != nil {
+		if strings.Contains(err.Error(), "RepositoryNotFoundException") {
+			return response{
+				Message: "Repository not found",
+				Ok:      true,
+			}, nil
+		}
 		return returnErr(err, environmentVars.slackOAuthToken, event.SlackChannelID, errSubject,
 			"Error getting input images from tags")
 	}