From 0e7340cb25e9b713579386a7e665c54be9a954fa Mon Sep 17 00:00:00 2001
From: Camila Belo <camilaibs@gmail.com>
Date: Tue, 30 Jan 2024 14:03:33 +0100
Subject: [PATCH] feat(catalog-react): accepts entity cards config schema

Signed-off-by: Camila Belo <camilaibs@gmail.com>
---
 plugins/catalog-react/src/alpha.tsx | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/plugins/catalog-react/src/alpha.tsx b/plugins/catalog-react/src/alpha.tsx
index 330cd573cd..f5f4dfdb74 100644
--- a/plugins/catalog-react/src/alpha.tsx
+++ b/plugins/catalog-react/src/alpha.tsx
@@ -17,6 +17,7 @@
 import {
   AnyExtensionInputMap,
   ExtensionBoundary,
+  PortableSchema,
   ResolvedExtensionInputs,
   RouteRef,
   coreExtensionData,
@@ -48,6 +49,7 @@ export const catalogExtensionData = {
 // TODO: Figure out how to merge with provided config schema
 /** @alpha */
 export function createEntityCardExtension<
+  TConfig extends { filter?: string },
   TInputs extends AnyExtensionInputMap,
 >(options: {
   namespace?: string;
@@ -55,13 +57,23 @@ export function createEntityCardExtension<
   attachTo?: { id: string; input: string };
   disabled?: boolean;
   inputs?: TInputs;
+  configSchema?: PortableSchema<TConfig>;
   filter?:
     | typeof catalogExtensionData.entityFilterFunction.T
     | typeof catalogExtensionData.entityFilterExpression.T;
   loader: (options: {
+    config: TConfig;
     inputs: Expand<ResolvedExtensionInputs<TInputs>>;
   }) => Promise<JSX.Element>;
 }) {
+  const configSchema =
+    'configSchema' in options
+      ? options.configSchema
+      : (createSchemaFromZod(z =>
+          z.object({
+            filter: z.string().optional(),
+          }),
+        ) as PortableSchema<TConfig>);
   return createExtension({
     kind: 'entity-card',
     namespace: options.namespace,
@@ -77,15 +89,11 @@ export function createEntityCardExtension<
       filterExpression: catalogExtensionData.entityFilterExpression.optional(),
     },
     inputs: options.inputs,
-    configSchema: createSchemaFromZod(z =>
-      z.object({
-        filter: z.string().optional(),
-      }),
-    ),
+    configSchema,
     factory({ config, inputs, node }) {
       const ExtensionComponent = lazy(() =>
         options
-          .loader({ inputs })
+          .loader({ inputs, config })
           .then(element => ({ default: () => element })),
       );