feat(iceberg): add native SQL support across Studio

release/app/package.json

@@ -16,7 +16,7 @@
   },
   "dependencies": {
     "@databricks/sql": "^1.11.0",
-    "@duckdb/node-api": "^1.5.2-r.1",
+    "@duckdb/node-api": "1.5.5-r.1",
     "better-sqlite3": "^12.2.0",
     "drizzle-orm": "^0.44.4",
     "fs-extra": "^11.4.0",

src/main/ipcHandlers/icebergDatalake.ipcHandlers.ts

@@ -46,6 +46,29 @@ export const registerIcebergDatalakeHandlers = () => {
     IcebergDatalakeService.testInstanceConnection(id),
   );
 
+  ipcMain.handle('iceberg:sqlCapability', (_e, id: string) =>
+    IcebergDatalakeService.getSqlCapability(id),
+  );
+
+  ipcMain.handle('iceberg:sqlSchema', (_e, id: string) =>
+    IcebergDatalakeService.getSqlSchema(id),
+  );
+
+  ipcMain.handle('iceberg:verifySqlAccess', (_e, id: string, draft) =>
+    IcebergDatalakeService.verifySqlAccess(
+      id,
+      IcebergDatalakeService.validateSqlVerificationPayload(id, draft),
+    ),
+  );
+
+  ipcMain.handle('iceberg:executeSql', (_e, params) =>
+    IcebergDatalakeService.executeSql(params),
+  );
+
+  ipcMain.handle('iceberg:cancelSql', (_e, executionId: string) =>
+    IcebergDatalakeService.cancelSql(executionId),
+  );
+
   ipcMain.handle('iceberg:listNamespaces', (_e, id: string, parent?) =>
     IcebergDatalakeService.listNamespaces(id, parent),
   );

src/main/ipcHandlers/notebooks.ipcHandlers.ts

@@ -4,6 +4,10 @@
  */
 
 import { ipcMain } from 'electron';
+import type {
+  NotebookExecutionOptions,
+  NotebookRunAllCell,
+} from '../../types/notebooks';
 import { NotebooksService } from '../services/notebooks.service';
 
 export function registerNotebooksHandlers() {
@@ -119,6 +123,7 @@ export function registerNotebooksHandlers() {
       sql: string,
       limit?: number,
       offset?: number,
+      options?: NotebookExecutionOptions,
     ) => {
       return NotebooksService.runCell(
         connectionId,
@@ -127,6 +132,7 @@ export function registerNotebooksHandlers() {
         sql,
         limit,
         offset,
+        options,
       );
     },
   );
@@ -157,11 +163,20 @@ export function registerNotebooksHandlers() {
   // Run all cells
   ipcMain.handle(
     'notebooks:runAll',
-    async (_event, connectionId: string, notebookId: string) => {
-      return NotebooksService.runAllCells(connectionId, notebookId);
+    async (
+      _event,
+      connectionId: string,
+      notebookId: string,
+      cellRun?: NotebookRunAllCell,
+    ) => {
+      return NotebooksService.runAllCells(connectionId, notebookId, cellRun);
     },
   );
 
+  ipcMain.handle('notebooks:cancelIcebergCell', (_event, executionId: string) =>
+    NotebooksService.cancelIcebergCell(executionId),
+  );
+
   // List archived notebooks
   ipcMain.handle('notebooks:archived:list', async () => {
     return NotebooksService.listArchivedNotebooks();