feat: enhance catalog-plugin table with label column

Signed-off-by: Ankit Luthra <ankitluthra06@gmail.com>

plugins/catalog/api-report.md

@@ -148,6 +148,15 @@ export const CatalogTable: {
           }
         | undefined,
     ): TableColumn<CatalogTableRow>;
+    createLabelColumn(
+      key: string,
+      options?:
+        | {
+            title?: string | undefined;
+            defaultValue?: string | undefined;
+          }
+        | undefined,
+    ): TableColumn<CatalogTableRow>;
   }>;
 };
 

plugins/catalog/src/components/CatalogTable/CatalogTable.test.tsx

@@ -331,4 +331,42 @@ describe('CatalogTable component', () => {
 
     expect(getByText('Should be rendered')).toBeInTheDocument();
   });
+
+  it('should render the label column with customised title and value as specified', async () => {
+    const columns = [
+      CatalogTable.columns.createNameColumn({ defaultKind: 'API' }),
+      CatalogTable.columns.createLabelColumn('category', { title: 'Category' }),
+    ];
+    const entity = {
+      apiVersion: 'backstage.io/v1alpha1',
+      kind: 'API',
+      metadata: {
+        name: 'APIWithLabel',
+        labels: { category: 'generic' },
+      },
+    };
+    const expectedColumns = ['Name', 'Category', 'Actions'];
+
+    const { getAllByRole, getByText } = await renderInTestApp(
+      <ApiProvider apis={mockApis}>
+        <MockEntityListContextProvider value={{ entities: [entity] }}>
+          <CatalogTable columns={columns} />
+        </MockEntityListContextProvider>
+      </ApiProvider>,
+      {
+        mountedRoutes: {
+          '/catalog/:namespace/:kind/:name': entityRouteRef,
+        },
+      },
+    );
+
+    const columnHeader = getAllByRole('button').filter(
+      c => c.tagName === 'SPAN',
+    );
+    const columnHeaderLabels = columnHeader.map(c => c.textContent);
+    expect(columnHeaderLabels).toEqual(expectedColumns);
+
+    const labelCellValue = getByText('generic');
+    expect(labelCellValue).toBeInTheDocument();
+  });
 });

plugins/catalog/src/components/CatalogTable/columns.tsx

@@ -25,6 +25,31 @@ import { OverflowTooltip, TableColumn } from '@backstage/core-components';
 import { Entity } from '@backstage/catalog-model';
 import { JsonArray } from '@backstage/types';
 
+/**
+ * Renders label if exists in entities or default
+ * @param labels - Key value pairs of labels specified in entity metadata
+ * @param key - specified label key to be extracted from all labels
+ * @param defaultValue - shows default in case label key does not exist
+ * @returns Chip style component when label value exists undefined otherwise.
+ */
+const renderLabel = (
+  labels: Record<string, string> | undefined,
+  key: string,
+  defaultValue?: string,
+) => {
+  const specifiedLabelValue = (labels && labels[key]) || defaultValue;
+  return (
+    specifiedLabelValue && (
+      <Chip
+        key={specifiedLabelValue}
+        label={specifiedLabelValue}
+        size="small"
+        variant="outlined"
+      />
+    )
+  );
+};
+
 // The columnFactories symbol is not directly exported, but through the
 // CatalogTable.columns field.
 /** @public */
@@ -162,4 +187,20 @@ export const columnFactories = Object.freeze({
       searchable: true,
     };
   },
+  createLabelColumn(
+    key: string,
+    options?: { title?: string; defaultValue?: string },
+  ): TableColumn<CatalogTableRow> {
+    return {
+      title: options?.title || 'Label',
+      field: 'entity.metadata.labels',
+      cellStyle: {
+        padding: '0px 16px 0px 20px',
+      },
+      render: ({ entity }: { entity: Entity }) => (
+        <>{renderLabel(entity.metadata?.labels, key, options?.defaultValue)}</>
+      ),
+      width: 'auto',
+    };
+  },
 });