electerm-data-tool 1.0.0

package/src/sqlite.js

@@ -0,0 +1,132 @@
+/**
+ * sqlite api wrapper
+ * Updated to use two database files: one for 'data' table, one for others
+ */
+
+const { appPath, defaultUserName } = require('./common/app-props')
+const { resolve } = require('path')
+const uid = require('./common/uid')
+const { DatabaseSync } = require('node:sqlite')
+
+// Define paths for two database files
+const mainDbPath = resolve(appPath, 'electerm', 'users', defaultUserName, 'electerm.db')
+const dataDbPath = resolve(appPath, 'electerm', 'users', defaultUserName, 'electerm_data.db')
+
+// Create two database instances
+const mainDb = new DatabaseSync(mainDbPath)
+const dataDb = new DatabaseSync(dataDbPath)
+
+const tables = [
+  'bookmarks',
+  'bookmarkGroups',
+  'addressBookmarks',
+  'terminalThemes',
+  'lastStates',
+  'data',
+  'quickCommands',
+  'log',
+  'dbUpgradeLog',
+  'profiles'
+]
+
+// Create tables in appropriate databases
+for (const table of tables) {
+  if (table === 'data') {
+    dataDb.exec(`CREATE TABLE IF NOT EXISTS \`${table}\` (_id TEXT PRIMARY KEY, data TEXT)`)
+  } else {
+    mainDb.exec(`CREATE TABLE IF NOT EXISTS \`${table}\` (_id TEXT PRIMARY KEY, data TEXT)`)
+  }
+}
+
+// Helper function to get the appropriate database for a table
+function getDatabase (dbName) {
+  return dbName === 'data' ? dataDb : mainDb
+}
+
+function toDoc (row) {
+  if (!row) return null
+  return {
+    ...JSON.parse(row.data || '{}'),
+    _id: row._id
+  }
+}
+
+function toRow (doc) {
+  const _id = doc._id || doc.id || uid()
+  const copy = { ...doc }
+  delete copy._id
+  delete copy.id
+  return {
+    _id,
+    data: JSON.stringify(copy)
+  }
+}
+
+async function dbAction (dbName, op, ...args) {
+  if (op === 'compactDatafile') {
+    return
+  }
+  if (!tables.includes(dbName)) {
+    throw new Error(`Table ${dbName} does not exist`)
+  }
+
+  // Get the appropriate database for this table
+  const db = getDatabase(dbName)
+
+  if (op === 'find') {
+    const sql = `SELECT * FROM \`${dbName}\``
+    const stmt = db.prepare(sql)
+    const rows = stmt.all()
+    return (rows || []).map(toDoc).filter(Boolean)
+  } else if (op === 'findOne') {
+    const query = args[0] || {}
+    const sql = `SELECT * FROM \`${dbName}\` WHERE _id = ? LIMIT 1`
+    const params = [query._id]
+    const stmt = db.prepare(sql)
+    const row = stmt.get(...params)
+    return toDoc(row)
+  } else if (op === 'insert') {
+    const inserts = Array.isArray(args[0]) ? args[0] : [args[0]]
+    const inserted = []
+    for (const doc of inserts) {
+      const { _id, data } = toRow(doc)
+      const stmt = db.prepare(`INSERT INTO \`${dbName}\` (_id, data) VALUES (?, ?)`)
+      stmt.run(_id, data)
+      inserted.push({ ...doc, _id })
+    }
+    return Array.isArray(args[0]) ? inserted : inserted[0]
+  } else if (op === 'remove') {
+    const query = args[0] || {}
+    const sql = `DELETE FROM \`${dbName}\` WHERE _id = ?`
+    const params = [query._id]
+    const stmt = db.prepare(sql)
+    const res = stmt.run(...params)
+    return res.changes
+  } else if (op === 'update') {
+    const query = args[0]
+    const updateObj = args[1]
+    const options = args[2] || {}
+    const { upsert = false } = options
+    const qid = query._id || query.id
+    const newData = updateObj.$set || updateObj
+    const { _id, data } = toRow({
+      _id: qid,
+      ...newData
+    })
+    let stmt
+    let res
+    if (upsert) {
+      stmt = db.prepare(`REPLACE INTO \`${dbName}\` (_id, data) VALUES (?, ?)`)
+      res = stmt.run(_id, data)
+    } else {
+      stmt = db.prepare(`UPDATE \`${dbName}\` SET data = ? WHERE _id = ?`)
+      res = stmt.run(data, qid)
+    }
+    return res.changes
+  }
+}
+
+module.exports = {
+  dbAction,
+  tables
+}