From 58f758386151c718b034dba9d17297738e52d3e5 Mon Sep 17 00:00:00 2001
From: Emma Indal <emma.indahl@gmail.com>
Date: Mon, 6 Mar 2023 08:29:03 +0100
Subject: [PATCH] create searchIndexRegistry module + add to backend

Signed-off-by: Emma Indal <emma.indahl@gmail.com>
---
 packages/backend-next/src/index.ts | 48 ++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/packages/backend-next/src/index.ts b/packages/backend-next/src/index.ts
index 071a446cae..93161de9a2 100644
--- a/packages/backend-next/src/index.ts
+++ b/packages/backend-next/src/index.ts
@@ -22,9 +22,56 @@ import { todoPlugin } from '@backstage/plugin-todo-backend';
 import { techdocsPlugin } from '@backstage/plugin-techdocs-backend/alpha';
 import { searchPlugin } from '@backstage/plugin-search-backend';
 import { elasticSearchEngineModule } from '@backstage/plugin-search-backend-module-elasticsearch';
+import {
+  coreServices,
+  createBackendModule,
+} from '@backstage/backend-plugin-api';
+import { searchIndexRegistryExtensionPoint } from '@backstage/plugin-search-backend-node';
+import { DefaultCatalogCollatorFactory } from '@backstage/plugin-catalog-backend';
 // import { pgSearchEngineModule } from '@backstage/plugin-search-backend-module-pg';
 // import { lunrSearchEngineModule } from '@backstage/plugin-search-backend-node/';
 
+// TODO: move to search-backend-node?
+export const searchIndexRegistry = createBackendModule({
+  moduleId: 'searchIndexRegistry',
+  pluginId: 'search',
+  register(env) {
+    env.registerInit({
+      deps: {
+        indexRegistry: searchIndexRegistryExtensionPoint,
+        config: coreServices.config,
+        discovery: coreServices.discovery,
+        tokenManager: coreServices.tokenManager,
+        logger: coreServices.logger,
+        scheduler: coreServices.scheduler,
+      },
+      async init({
+        indexRegistry,
+        config,
+        discovery,
+        tokenManager,
+        scheduler,
+      }) {
+        const schedule = scheduler.createScheduledTaskRunner({
+          frequency: { minutes: 10 },
+          timeout: { minutes: 15 },
+          // A 3 second delay gives the backend server a chance to initialize before
+          // any collators are executed, which may attempt requests against the API.
+          initialDelay: { seconds: 3 },
+        });
+
+        indexRegistry.addCollator({
+          schedule,
+          factory: DefaultCatalogCollatorFactory.fromConfig(config, {
+            discovery: discovery,
+            tokenManager: tokenManager,
+          }),
+        });
+      },
+    });
+  },
+});
+
 const backend = createBackend();
 
 backend.add(catalogPlugin());
@@ -43,5 +90,6 @@ backend.add(
     },
   }),
 );
+backend.add(searchIndexRegistry());
 
 backend.start();