Function useDB

src/main/db/index.ts:14–114

(customDbPath?: string)

let currentDbPath: string | null = null

export function useDB(customDbPath?: string) {
  const dbPath
    = customDbPath || `${store.preferences.get('storage.rootPath')}/${DB_NAME}`

  if (db && currentDbPath === dbPath)
    return db

  if (db) {
    db.close()
    db = null
    currentDbPath = null
  }

  const dbDir = path.dirname(dbPath)

  if (!fs.existsSync(dbDir)) {
    fs.mkdirSync(dbDir, { recursive: true })
  }

  try {
    db = new Database(dbPath, {
      // eslint-disable-next-line no-console
      verbose: isDev ? console.log : undefined,
    })
    currentDbPath = dbPath

    db.pragma('journal_mode = WAL')
    db.pragma('foreign_keys = ON')

    db.function('unicode_lower', (str: unknown) => {
      if (typeof str !== 'string')
        return str
      return str.toLowerCase()
    })

    db.exec(`
      CREATE TABLE IF NOT EXISTS folders (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT NOT NULL,
        defaultLanguage TEXT NOT NULL,
        parentId INTEGER,
        isOpen INTEGER NOT NULL,
        orderIndex INTEGER NOT NULL DEFAULT 0,
        icon TEXT,
        createdAt INTEGER NOT NULL,
        updatedAt INTEGER NOT NULL,
        FOREIGN KEY(parentId) REFERENCES folders(id)
      )
    `)

    db.exec(`
      CREATE TABLE IF NOT EXISTS snippets (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT NOT NULL,
        description TEXT,
        folderId INTEGER,
        isDeleted INTEGER NOT NULL,
        isFavorites INTEGER NOT NULL,