fix missing document icon in reusable components

Author: vieiralucas

apps/web/src/components/v2Editor/customBlocks/python/index.tsx

@@ -316,7 +316,8 @@ function PythonBlock(props: Props) {
           title,
           type: 'python',
         },
-        props.document.title
+        props.document.title,
+        props.document.icon
       )
     } else if (!isComponentInstance) {
       // can only update component if it is not an instance

apps/web/src/components/v2Editor/customBlocks/sql/index.tsx

@@ -445,7 +445,8 @@ function SQLBlock(props: Props) {
           title,
           type: 'sql',
         },
-        props.document.title
+        props.document.title,
+        props.document.icon
       )
     } else if (!isComponentInstance) {
       // can only update component if it is not an instance

apps/web/src/hooks/useReusableComponents.tsx

@@ -21,7 +21,8 @@ type API = {
   create: (
     workspaceId: string,
     data: Omit<NewReusableComponent, 'id'> & { id: string },
-    documenTitle: string
+    documenTitle: string,
+    documentIcon: string
   ) => void
   update: (
     workspaceId: string,
@@ -161,7 +162,8 @@ export function ReusableComponentsProvider(props: Props) {
     async (
       workspaceId: string,
       data: Omit<NewReusableComponent, 'id'> & { id: string },
-      documentTitle: string
+      documentTitle: string,
+      documentIcon: string
     ) => {
       // optimistic update
       setState((state) => {
@@ -176,6 +178,7 @@ export function ReusableComponentsProvider(props: Props) {
             document: {
               id: data.documentId,
               title: documentTitle,
+              icon: documentIcon,
             },
             createdAt: new Date().toISOString(),
             updatedAt: new Date().toISOString(),

packages/database/src/components.ts

@@ -71,6 +71,7 @@ export async function createReusableComponent(
         select: {
           id: true,
           title: true,
+          icon: true,
         },
       },
     },
@@ -105,6 +106,7 @@ export async function updateReusableComponent(
         select: {
           id: true,
           title: true,
+          icon: true,
         },
       },
     },