@oml/markdown 0.10.0 → 0.11.0

package/src/renderers/table-renderer.ts

@@ -27,6 +27,14 @@ type TableEditorActionContext = {
     pointerY?: number;
 };
 
+type TableRootWithAiContext = HTMLDivElement & {
+    __omlAiContext?: {
+        columns: string[];
+        rows: Array<{ iri: string; cells: string[] }>;
+        blockSource?: string;
+    };
+};
+
 export class TableMarkdownBlockRenderer extends QueryMarkdownBlockRenderer {
     protected readonly tableKinds: ReadonlyArray<TableBlockKind> = ['table'];
 
@@ -92,7 +100,7 @@ export class TableMarkdownBlockRenderer extends QueryMarkdownBlockRenderer {
         options: Record<string, unknown> | undefined,
         blockSource?: string
     ): HTMLElement {
-        const root = document.createElement('div');
+        const root = document.createElement('div') as TableRootWithAiContext;
         root.className = 'table-root graph-root';
         const isTree = this.tableKinds.includes('tree');
         const allRowsWithIndex = rows.map((cells, index) => ({ index, cells }));
@@ -114,6 +122,16 @@ export class TableMarkdownBlockRenderer extends QueryMarkdownBlockRenderer {
             ...row,
             cells: visibleColumnIndexes.map((columnIndex) => row.cells[columnIndex] ?? ''),
         }));
+        root.__omlAiContext = {
+            columns: visibleColumns.slice(),
+            rows: rowsWithIndex
+                .map((row) => ({
+                    iri: (row.cells[0] ?? '').trim(),
+                    cells: row.cells.slice(),
+                }))
+                .filter((row) => row.iri.length > 0),
+            blockSource,
+        };
         const columnContexts = createColumnContexts(visibleColumns, rowsWithIndex);
         const treeModel = isTree
             ? this.createTreeModel(columns, allRowsWithIndex, visibleColumnIndexes, options)
@@ -218,10 +236,16 @@ export class TableMarkdownBlockRenderer extends QueryMarkdownBlockRenderer {
                 return;
             }
             const filtered = this.applyFiltersAndSorting(columns, sourceRows, state);
-            this.requestCsvDownload(visibleColumns, filtered.map((entry) => entry.cells));
+            const csvExport = this.transformCsvDownloadData(visibleColumns, filtered.map((entry) => entry.cells));
+            this.requestCsvDownload(csvExport.columns, csvExport.rows);
         });
         controlsRight.appendChild(downloadButton);
 
+        const uploadButton = this.createUploadCsvButton(visibleColumns, rowsWithIndex, isTree, blockSource);
+        if (uploadButton) {
+            controlsRight.appendChild(uploadButton);
+        }
+
         controls.appendChild(controlsLeft);
         controls.appendChild(controlsRight);
         root.appendChild(controls);
@@ -957,6 +981,19 @@ export class TableMarkdownBlockRenderer extends QueryMarkdownBlockRenderer {
         return value;
     }
 
+    protected createUploadCsvButton(
+        _columns: string[],
+        _rows: Array<{ index: number; cells: string[] }>,
+        _isTree: boolean,
+        _blockSource: string | undefined
+    ): HTMLElement | undefined {
+        return undefined;
+    }
+
+    protected transformCsvDownloadData(columns: string[], rows: string[][]): { columns: string[]; rows: string[][] } {
+        return { columns, rows };
+    }
+
     private requestCsvDownload(columns: string[], rows: string[][]): void {
         if (columns.length === 0) {
             return;