From 4a4b9ad2de274fdf47178edba54da6d74d858c31 Mon Sep 17 00:00:00 2001
From: bnechyporenko <bnechyporenko@bol.com>
Date: Tue, 2 Aug 2022 11:51:33 +0200
Subject: [PATCH] Added project pattern for GitLab provider to be able to
 filter projects on more granular level.

Signed-off-by: bnechyporenko <bnechyporenko@bol.com>
---
 plugins/catalog-backend-module-gitlab/src/lib/types.ts      | 1 +
 .../src/providers/GitlabDiscoveryEntityProvider.ts          | 4 ++++
 .../catalog-backend-module-gitlab/src/providers/config.ts   | 6 +++++-
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/plugins/catalog-backend-module-gitlab/src/lib/types.ts b/plugins/catalog-backend-module-gitlab/src/lib/types.ts
index 69de9d28c7..8a3c33c8f2 100644
--- a/plugins/catalog-backend-module-gitlab/src/lib/types.ts
+++ b/plugins/catalog-backend-module-gitlab/src/lib/types.ts
@@ -35,4 +35,5 @@ export type GitlabProviderConfig = {
   id: string;
   branch: string;
   catalogFile: string;
+  projectPattern: RegExp;
 };
diff --git a/plugins/catalog-backend-module-gitlab/src/providers/GitlabDiscoveryEntityProvider.ts b/plugins/catalog-backend-module-gitlab/src/providers/GitlabDiscoveryEntityProvider.ts
index 36c3ee90be..ffb123edf7 100644
--- a/plugins/catalog-backend-module-gitlab/src/providers/GitlabDiscoveryEntityProvider.ts
+++ b/plugins/catalog-backend-module-gitlab/src/providers/GitlabDiscoveryEntityProvider.ts
@@ -147,6 +147,10 @@ export class GitlabDiscoveryEntityProvider implements EntityProvider {
     };
 
     for await (const project of projects) {
+      if (!this.config.projectPattern.test(project.path_with_namespace ?? '')) {
+        continue;
+      }
+
       res.scanned++;
 
       if (project.archived) {
diff --git a/plugins/catalog-backend-module-gitlab/src/providers/config.ts b/plugins/catalog-backend-module-gitlab/src/providers/config.ts
index d6ad28b2ca..a316ddc3c9 100644
--- a/plugins/catalog-backend-module-gitlab/src/providers/config.ts
+++ b/plugins/catalog-backend-module-gitlab/src/providers/config.ts
@@ -15,13 +15,14 @@
  */
 
 import { Config } from '@backstage/config';
-import { GitlabProviderConfig } from '../lib/types';
+import { GitlabProviderConfig } from '../lib';
 
 /**
  * Extracts the gitlab config from a config object
  *
  * @public
  *
+ * @param id - The provider key
  * @param config - The config object to extract from
  */
 function readGitlabConfig(id: string, config: Config): GitlabProviderConfig {
@@ -30,6 +31,8 @@ function readGitlabConfig(id: string, config: Config): GitlabProviderConfig {
   const branch = config.getOptionalString('branch') ?? 'master';
   const catalogFile =
     config.getOptionalString('entityFilename') ?? 'catalog-info.yaml';
+  const projectPattern =
+    new RegExp(config.getString('projectPattern')) ?? /[\s\S]*/;
 
   return {
     id,
@@ -37,6 +40,7 @@ function readGitlabConfig(id: string, config: Config): GitlabProviderConfig {
     branch,
     host,
     catalogFile,
+    projectPattern,
   };
 }