feat(app,main): initial support for postgres & mysql

Author: nekomeowww

.vscode/settings.json

@@ -46,7 +46,9 @@
     "editor.formatOnSave": true
   },
   "cSpell.words": [
+    "airi",
     "Deditor",
+    "duckdb",
     "Splitpanes",
     "vueuse"
   ]

electron.vite.config.ts

@@ -8,7 +8,9 @@ export default defineConfig({
       include: [
         'drizzle-orm',
         'drizzle-orm/postgres-js',
+        'drizzle-orm/mysql2',
         'postgres',
+        'mysql2',
       ],
     })],
   },
@@ -17,7 +19,9 @@ export default defineConfig({
       include: [
         'drizzle-orm',
         'drizzle-orm/postgres-js',
+        'drizzle-orm/mysql2',
         'postgres',
+        'mysql2',
       ],
     })],
   },

package.json

@@ -44,6 +44,8 @@
     "@xsai-transformers/shared": "^0.0.6",
     "@xsai/embed": "^0.2.2",
     "@xsai/shared": "^0.2.2",
+    "mysql2": "^3.14.1",
+    "nanoid": "^5.1.5",
     "ofetch": "^1.4.1",
     "splitpanes": "^4.0.3",
     "std-env": "^3.9.0",

pnpm-lock.yaml

@@ -0,0 +1,42 @@
+import type { MySql2Database } from 'drizzle-orm/mysql2'
+import type { BrowserWindow } from 'electron'
+
+import { drizzle } from 'drizzle-orm/mysql2'
+import { ipcMain } from 'electron'
+
+import { nanoid } from '../../utils/nanoid'
+
+const databaseSessions = new Map<string, MySql2Database>()
+
+export function registerMySQL2DatabaseDialect(window: BrowserWindow) {
+  ipcMain.on('request:connect-remote-database-mysql2', (_, databaseDsn: string) => {
+    try {
+      const dbSession = drizzle(databaseDsn)
+      const dbSessionId = nanoid()
+      databaseSessions.set(dbSessionId, dbSession)
+
+      // eslint-disable-next-line no-console
+      dbSession.execute('SELECT 1').then(res => console.log(res))
+      window.webContents.send('response:connect-remote-database-mysql2', { databaseSessionId: dbSessionId, dialect: 'mysql2' })
+    }
+    catch (err) {
+      window.webContents.send('response:error:connect-remote-database-mysql2', err)
+    }
+  })
+
+  ipcMain.on('request:query-remote-database-mysql2', (_, { databaseSessionId, statement }: { databaseSessionId: string, statement: string, params: any[] }) => {
+    if (!databaseSessions.has(databaseSessionId)) {
+      window.webContents.send('response:error:query-remote-database-mysql2', new Error('Database session ID not found in session map, please connect to the database first.'))
+      return
+    }
+
+    const dbSession = databaseSessions.get(databaseSessionId)!
+    dbSession.execute(statement)
+      .then((res) => {
+        window.webContents.send('response:query-remote-database-mysql2', { databaseSessionId, results: res })
+      })
+      .catch((err) => {
+        window.webContents.send('response:error:query-remote-database-mysql2', { databaseSessionId, error: err })
+      })
+  })
+}

src/main/databases/remote/mysql.ts

@@ -0,0 +1,42 @@
+import type { PostgresJsDatabase } from 'drizzle-orm/postgres-js'
+import type { BrowserWindow } from 'electron'
+
+import { drizzle } from 'drizzle-orm/postgres-js'
+import { ipcMain } from 'electron'
+
+import { nanoid } from '../../utils/nanoid'
+
+const databaseSessions = new Map<string, PostgresJsDatabase>()
+
+export function registerPostgresJsDatabaseDialect(window: BrowserWindow) {
+  ipcMain.on('request:connect-remote-database', (_, databaseDsn: string) => {
+    try {
+      const dbSession = drizzle(databaseDsn)
+      const dbSessionId = nanoid()
+      databaseSessions.set(dbSessionId, dbSession)
+
+      // eslint-disable-next-line no-console
+      dbSession.execute('SELECT 1').then(res => console.log(res))
+      window.webContents.send('response:connect-remote-database', { databaseSessionId: dbSessionId, dialect: 'postgres' })
+    }
+    catch (err) {
+      window.webContents.send('response:error:connect-remote-database', err)
+    }
+  })
+
+  ipcMain.on('request:query-remote-database', (_, { databaseSessionId, statement }: { databaseSessionId: string, statement: string, params: any[] }) => {
+    if (!databaseSessions.has(databaseSessionId)) {
+      window.webContents.send('response:error:query-remote-database', new Error('Database session ID not found in session map, please connect to the database first.'))
+      return
+    }
+
+    const dbSession = databaseSessions.get(databaseSessionId)!
+    dbSession.execute(statement)
+      .then((res) => {
+        window.webContents.send('response:query-remote-database', { databaseSessionId, results: res })
+      })
+      .catch((err) => {
+        window.webContents.send('response:error:query-remote-database', { databaseSessionId, error: err })
+      })
+  })
+}