Initial clean project import

2026-06-19 18:41:41 +08:00
commit a13b804c7a
1306 changed files with 220568 additions and 0 deletions
@@ -0,0 +1,289 @@
+import sqlite3, {Database} from "better-sqlite3";
+import path from "node:path";
+import migration from "./migration";
+import {AppEnv} from "../env";
+import {Log} from "../log/main";
+import {ipcMain} from "electron";
+import fs from "node:fs";
+import {Files} from "../file/main";
+
+let dbPath: string | null = null;
+let dbConn: Database | null = null;
+let dbSuccess = false;
+
+const db = {
+    /**
+     * 检查数据库连接是否已初始化
+     * @throws {string} 如果数据库未初始化则抛出异常
+     */
+    _check() {
+        if (!dbSuccess) {
+            const errorMsg = `DBNotInitialized: dbPath=${dbPath}, dbConn=${dbConn !== null}, dbSuccess=${dbSuccess}`;
+            console.error("[DB:_check]", errorMsg);
+            throw errorMsg;
+        }
+    },
+    /**
+     * 执行SQL语句（无返回值）
+     * @param {string} sql - SQL语句
+     * @param {any[]} params - 参数数组
+     * @returns {Promise<void>}
+     */
+    async execute(sql: string, params: any = []): Promise<void> {
+        db._check();
+        try {
+            dbConn.prepare(sql).run(...params);
+        } catch (err) {
+            throw err;
+        }
+    },
+    /**
+     * 插入数据并返回插入的行ID
+     * @param {string} sql - SQL语句
+     * @param {any[]} params - 参数数组
+     * @returns {Promise<string | number>} 插入的行ID
+     */
+    async insert(sql: string, params: any = []): Promise<string | number> {
+        db._check();
+        try {
+            const result = dbConn.prepare(sql).run(...params);
+            return result.lastInsertRowid;
+        } catch (err) {
+            throw err;
+        }
+    },
+    /**
+     * 查询单行数据
+     * @param {string} sql - SQL语句
+     * @param {any[]} params - 参数数组
+     * @returns {Promise<any>} 查询结果
+     */
+    async first(sql: string, params: any = []): Promise<any> {
+        db._check();
+        try {
+            return dbConn.prepare(sql).get(...params);
+        } catch (err) {
+            throw err;
+        }
+    },
+    /**
+     * 查询多行数据
+     * @param {string} sql - SQL语句
+     * @param {any[]} params - 参数数组
+     * @returns {Promise<any[]>} 查询结果数组
+     */
+    async select(sql: string, params: any = []): Promise<any[]> {
+        db._check();
+        try {
+            return dbConn.prepare(sql).all(...params);
+        } catch (err) {
+            throw err;
+        }
+    },
+    /**
+     * 更新数据并返回影响的行数
+     * @param {string} sql - SQL语句
+     * @param {any[]} params - 参数数组
+     * @returns {Promise<number>} 影响的行数
+     */
+    async update(sql: string, params: any = []): Promise<number> {
+        db._check();
+        try {
+            const result = dbConn.prepare(sql).run(...params);
+            return result.changes;
+        } catch (err) {
+            throw err;
+        }
+    },
+    /**
+     * 删除数据并返回影响的行数
+     * @param {string} sql - SQL语句
+     * @param {any[]} params - 参数数组
+     * @returns {Promise<number>} 影响的行数
+     */
+    async delete(sql: string, params: any = []): Promise<number> {
+        db._check();
+        try {
+            const result = dbConn.prepare(sql).run(...params);
+            return result.changes;
+        } catch (err) {
+            throw err;
+        }
+    },
+};
+
+const migrate = async () => {
+    await db.execute(`CREATE TABLE IF NOT EXISTS migrate
+                      (
+                          id
+                          INTEGER
+                          PRIMARY
+                          KEY,
+                          version
+                          INTEGER
+                      )`);
+    for (const version of migration.versions) {
+        const result = await db.first(
+            `SELECT *
+             FROM migrate
+             WHERE version = ?`,
+            [version.version]
+        );
+        if (!result) {
+            Log.info(`DB.Migrate`, {version: version.version});
+            await version.up(db);
+            await db.execute(
+                `INSERT INTO migrate (version)
+                 VALUES (?)`,
+                [version.version]
+            );
+        }
+    }
+};
+
+/**
+ * 初始化数据库连接
+ * @returns {Promise<void>}
+ */
+const init = async () => {
+    console.log("[DB:init] Starting database initialization");
+    console.log("[DB:init] AppEnv.dataRoot =", AppEnv.dataRoot);
+    console.log("[DB:init] AppEnv.userData =", AppEnv.userData);
+
+    dbPath = path.join(AppEnv.dataRoot, "database.db");
+    const userDbPath = path.join(AppEnv.userData, "database.db");
+
+    console.log("[DB:init] Default dbPath =", dbPath);
+    console.log("[DB:init] User dbPath =", userDbPath);
+    console.log("[DB:init] User dbPath exists =", fs.existsSync(userDbPath));
+
+    if (fs.existsSync(userDbPath)) {
+        dbPath = userDbPath;
+        console.log("[DB:init] Using user dbPath instead");
+    }
+
+    console.log("[DB:init] Final dbPath =", dbPath);
+
+    try {
+        console.log("[DB:init] Creating SQLite3 connection...");
+        dbConn = new sqlite3(dbPath);
+        dbSuccess = true;
+        console.log("[DB:init] SQLite3 connection created successfully, dbSuccess =", dbSuccess);
+
+        console.log("[DB:init] Running migrations...");
+        await migrate();
+        console.log("[DB:init] Migrations completed successfully");
+
+        Log.info("Database connected successfully");
+    } catch (err) {
+        console.error("[DB:init] ERROR:", err);
+        Log.error("DBConnect SQLite database failed:", err.message);
+        throw err;
+    }
+};
+
+ipcMain.handle("db:execute", (event, sql: string, params: any) => {
+    try {
+        return db.execute(sql, params);
+    } catch (error) {
+        console.error("[db:execute] Error:", {
+            sql,
+            params,
+            paramCount: Array.isArray(params) ? params.length : 0,
+            placeholderCount: (sql.match(/\?/g) || []).length,
+            error: error.message
+        });
+        throw error;
+    }
+});
+ipcMain.handle("db:insert", (event, sql: string, params: any) => {
+    try {
+        return db.insert(sql, params);
+    } catch (error) {
+        console.error("[db:insert] Error:", {
+            sql,
+            params,
+            paramCount: Array.isArray(params) ? params.length : 0,
+            placeholderCount: (sql.match(/\?/g) || []).length,
+            error: error.message
+        });
+        throw error;
+    }
+});
+ipcMain.handle("db:first", (event, sql: string, params: any) => {
+    try {
+        return db.first(sql, params);
+    } catch (error) {
+        console.error("[db:first] Error:", {
+            sql,
+            params,
+            paramCount: Array.isArray(params) ? params.length : 0,
+            placeholderCount: (sql.match(/\?/g) || []).length,
+            error: error.message
+        });
+        throw error;
+    }
+});
+ipcMain.handle("db:select", (event, sql: string, params: any) => {
+    try {
+        return db.select(sql, params);
+    } catch (error) {
+        console.error("[db:select] Error:", {
+            sql,
+            params,
+            paramCount: Array.isArray(params) ? params.length : 0,
+            placeholderCount: (sql.match(/\?/g) || []).length,
+            error: error.message
+        });
+        throw error;
+    }
+});
+ipcMain.handle("db:update", (event, sql: string, params: any) => {
+    try {
+        return db.update(sql, params);
+    } catch (error) {
+        console.error("[db:update] Error:", {
+            sql,
+            params,
+            paramCount: Array.isArray(params) ? params.length : 0,
+            placeholderCount: (sql.match(/\?/g) || []).length,
+            error: error.message
+        });
+        throw error;
+    }
+});
+ipcMain.handle("db:delete", (event, sql: string, params: any) => {
+    try {
+        return db.delete(sql, params);
+    } catch (error) {
+        console.error("[db:delete] Error:", {
+            sql,
+            params,
+            paramCount: Array.isArray(params) ? params.length : 0,
+            placeholderCount: (sql.match(/\?/g) || []).length,
+            error: error.message
+        });
+        throw error;
+    }
+});
+
+export const DBMain = {
+    init,
+    execute: db.execute,
+    insert: db.insert,
+    first: db.first,
+    select: db.select,
+    update: db.update,
+    delete: db.delete,
+    // 诊断函数
+    getStatus() {
+        return {
+            dbSuccess,
+            dbPath,
+            dbConnected: dbConn !== null,
+            timestamp: new Date().toISOString(),
+        };
+    },
+};
+
+export default DBMain;