From d37799f56b01aef374fdecc664f40005603041d0 Mon Sep 17 00:00:00 2001
From: Johan Haals <johan.haals@gmail.com>
Date: Mon, 13 Sep 2021 14:59:47 +0200
Subject: [PATCH] catalog-backend: Add refresh endpoint to createRouter

Co-authored-by: Patrik Oldsberg <poldsberg@gmail.com>
Signed-off-by: Johan Haals <johan.haals@gmail.com>
---
 plugins/catalog-backend/src/service/router.ts | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/plugins/catalog-backend/src/service/router.ts b/plugins/catalog-backend/src/service/router.ts
index 71b8614a99..6a521b4475 100644
--- a/plugins/catalog-backend/src/service/router.ts
+++ b/plugins/catalog-backend/src/service/router.ts
@@ -28,7 +28,11 @@ import { Logger } from 'winston';
 import yn from 'yn';
 import { EntitiesCatalog, LocationsCatalog } from '../catalog';
 import { HigherOrderOperation, LocationAnalyzer } from '../ingestion/types';
-import { LocationService } from '../next/types';
+import {
+  LocationService,
+  CatalogProcessingEngine,
+  EntityRefreshOptions,
+} from '../next/types';
 import {
   basicEntityFilter,
   parseEntityFilterParams,
@@ -47,6 +51,7 @@ export interface RouterOptions {
   higherOrderOperation?: HigherOrderOperation;
   locationAnalyzer?: LocationAnalyzer;
   locationService?: LocationService;
+  processingEngine?: CatalogProcessingEngine;
   logger: Logger;
   config: Config;
 }
@@ -60,6 +65,7 @@ export async function createRouter(
     higherOrderOperation,
     locationAnalyzer,
     locationService,
+    processingEngine,
     config,
     logger,
   } = options;
@@ -73,6 +79,14 @@ export async function createRouter(
     logger.info('Catalog is running in readonly mode');
   }
 
+  if (processingEngine) {
+    router.post('/refresh', async (req, res) => {
+      const refreshOptions: EntityRefreshOptions = req.body;
+      await processingEngine.refresh(refreshOptions);
+      res.status(200).send();
+    });
+  }
+
   if (entitiesCatalog) {
     router
       .get('/entities', async (req, res) => {