♻️ create new row util

Author: anaisberg

src/frontend/apps/impress/src/features/docs/doc-editor/components/custom-blocks/DatabaseBlock/DatabaseGrid.tsx

@@ -11,6 +11,7 @@ import {
 
 import { AddButtonComponent } from './AddColumnButton';
 import { AddRowButton } from './AddRowButton';
+import { createNewRow } from './utils';
 
 export const DatabaseGrid = ({
   documentId,
@@ -73,14 +74,11 @@ export const DatabaseGrid = ({
       rowData1.push(row);
     }
 
-    const addNewRow: Record<string, string> = {};
-    for (const [key] of filteredEntries) {
-      addNewRow[key] = '+ new row';
-    }
-    rowData1.push(addNewRow);
-
     setRowData(rowData1);
 
+    const addNewRow = createNewRow('+ new row', colDefs ?? []);
+    rowData1.push(addNewRow);
+
     const columnNames = Object.keys(Object.fromEntries(filteredEntries));
     const columns: ColDef[] = columnNames.map((key) => ({
       field: key,
@@ -94,9 +92,8 @@ export const DatabaseGrid = ({
             params: { values: params.data },
           };
 
-          if (Object.values(params.data)[0].includes('new')) {
+          if (Object.values(params.data)[0]?.includes('new')) {
             return addRowButton;
-            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
           }
           return undefined;
         }

src/frontend/apps/impress/src/features/docs/doc-editor/components/custom-blocks/DatabaseBlock/utils.ts

@@ -0,0 +1,17 @@
+import { ColDef } from 'ag-grid-community';
+
+export const createNewRow = (
+  value: string | undefined,
+  columns: ColDef[] | undefined,
+) => {
+  const defaultValue = value ?? '';
+  const columnNames = columns?.map((col) => col.field);
+  const addNewRow: Record<string, string> = {};
+  columnNames?.forEach((name) => {
+    if (name !== undefined) {
+      addNewRow[name] = defaultValue;
+    }
+  });
+
+  return addNewRow;
+};