chore: removing the entityClient and using a helper instead to fetch the matching Templates

Signed-off-by: blam <ben@blam.sh>

plugins/scaffolder-backend/src/lib/catalog/index.ts

@@ -1,16 +0,0 @@
-/*
- * Copyright 2020 The Backstage Authors
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-export { CatalogEntityClient } from './CatalogEntityClient';

plugins/scaffolder-backend/src/lib/index.ts

@@ -14,5 +14,4 @@
  * limitations under the License.
  */
 
-export * from './catalog';
 export * from './templating';

plugins/scaffolder-backend/src/service/helpers.ts

@@ -14,14 +14,21 @@
  * limitations under the License.
  */
 
+import { CatalogApi } from '@backstage/catalog-client';
 import {
   Entity,
   ANNOTATION_LOCATION,
   parseLocationRef,
   ANNOTATION_SOURCE_LOCATION,
+  EntityRef,
+  parseEntityRef,
 } from '@backstage/catalog-model';
 import { Config } from '@backstage/config';
-import { assertError } from '@backstage/errors';
+import { assertError, ConflictError, NotFoundError } from '@backstage/errors';
+import {
+  TemplateEntityV1beta2,
+  TemplateEntityV1beta3,
+} from '@backstage/plugin-scaffolder-common';
 import fs from 'fs-extra';
 import os from 'os';
 import { Logger } from 'winston';
@@ -78,3 +85,46 @@ export function getEntityBaseUrl(entity: Entity): string | undefined {
   // what the url is pointing to makes sense to use as a baseUrl
   return undefined;
 }
+
+/**
+ * Will use the provided CatalogApi to go find the template entity ref that is provided with and additional token
+ * Returns the first matching template
+ */
+export async function findTemplate({
+  entityRef,
+  token,
+  catalogApi,
+}: {
+  entityRef: EntityRef;
+  token?: string;
+  catalogApi: CatalogApi;
+}): Promise<TemplateEntityV1beta3 | TemplateEntityV1beta2> {
+  const parsedEntityRef = parseEntityRef(entityRef);
+  const { items } = await catalogApi.getEntities(
+    {
+      filter: {
+        kind: 'template',
+        'metadata.name': parsedEntityRef.name,
+        'metadata.namespace': parsedEntityRef.namespace,
+      },
+    },
+    {
+      token,
+    },
+  );
+
+  const templates = items.filter(
+    (entity): entity is TemplateEntityV1beta3 | TemplateEntityV1beta2 =>
+      entity.kind === 'Template',
+  );
+
+  if (templates.length !== 1) {
+    if (templates.length > 1) {
+      throw new ConflictError('Templates lookup resulted in multiple matches');
+    } else {
+      throw new NotFoundError('Template not found');
+    }
+  }
+
+  return templates[0];
+}

plugins/scaffolder-backend/src/service/router.ts

@@ -33,7 +33,7 @@ import express from 'express';
 import Router from 'express-promise-router';
 import { validate } from 'jsonschema';
 import { Logger } from 'winston';
-import { CatalogEntityClient, TemplateFilter } from '../lib';
+import { TemplateFilter } from '../lib';
 import {
   createBuiltinActions,
   DatabaseTaskStore,
@@ -44,7 +44,7 @@ import {
   TemplateActionRegistry,
 } from '../scaffolder';
 import { StorageTaskBroker } from '../scaffolder/tasks/StorageTaskBroker';
-import { getEntityBaseUrl, getWorkingDirectory } from './helpers';
+import { getEntityBaseUrl, getWorkingDirectory, findTemplate } from './helpers';
 
 /**
  * RouterOptions
@@ -93,7 +93,6 @@ export async function createRouter(
 
   const logger = parentLogger.child({ plugin: 'scaffolder' });
   const workingDirectory = await getWorkingDirectory(config, logger);
-  const entityClient = new CatalogEntityClient(catalogClient);
   const integrations = ScmIntegrations.fromConfig(config);
   let taskBroker: TaskBroker;
 
@@ -152,7 +151,9 @@ export async function createRouter(
           );
         }
 
-        const template = await entityClient.findTemplate(name, {
+        const template = await findTemplate({
+          catalogApi: catalogClient,
+          entityRef: { kind, namespace, name },
           token: getBearerToken(req.headers.authorization),
         });
         if (isSupportedTemplate(template)) {
@@ -187,8 +188,14 @@ export async function createRouter(
       const templateName: string = req.body.templateName;
       const values = req.body.values;
       const token = getBearerToken(req.headers.authorization);
-      const template = await entityClient.findTemplate(templateName, {
-        token,
+      const template = await findTemplate({
+        catalogApi: catalogClient,
+        entityRef: {
+          kind: 'template',
+          namespace: 'default',
+          name: templateName,
+        },
+        token: getBearerToken(req.headers.authorization),
       });
 
       let taskSpec: TaskSpec;