feature: support init db in admin portal (#658)

2026-05-12 02:20:12 +08:00 · 2025-05-20 17:45:55 +08:00
parent 272b624b9b
commit 50326bcc98
15 changed files with 1897 additions and 1809 deletions
--- a/worker/src/admin_api/db_api.ts
+++ b/worker/src/admin_api/db_api.ts
@@ -0,0 +1,156 @@
+import { Context } from "hono";
+import { CONSTANTS } from "../constants";
+import utils from "../utils";
+
+const DB_INIT_QUERIES = `
+CREATE TABLE IF NOT EXISTS raw_mails (
+    id INTEGER PRIMARY KEY,
+    message_id TEXT,
+    source TEXT,
+    address TEXT,
+    raw TEXT,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_raw_mails_address ON raw_mails(address);
+
+CREATE TABLE IF NOT EXISTS address (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    name TEXT UNIQUE,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_address_name ON address(name);
+
+CREATE TABLE IF NOT EXISTS auto_reply_mails (
+    id INTEGER PRIMARY KEY,
+    source_prefix TEXT,
+    name TEXT,
+    address TEXT UNIQUE,
+    subject TEXT,
+    message TEXT,
+    enabled INTEGER DEFAULT 1,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_auto_reply_mails_address ON auto_reply_mails(address);
+
+CREATE TABLE IF NOT EXISTS address_sender (
+    id INTEGER PRIMARY KEY,
+    address TEXT UNIQUE,
+    balance INTEGER DEFAULT 0,
+    enabled INTEGER DEFAULT 1,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_address_sender_address ON address_sender(address);
+
+CREATE TABLE IF NOT EXISTS sendbox (
+    id INTEGER PRIMARY KEY,
+    address TEXT,
+    raw TEXT,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_sendbox_address ON sendbox(address);
+
+CREATE TABLE IF NOT EXISTS settings (
+    key TEXT PRIMARY KEY,
+    value TEXT,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE TABLE IF NOT EXISTS users (
+    id INTEGER PRIMARY KEY,
+    user_email TEXT UNIQUE NOT NULL,
+    password TEXT NOT NULL,
+    user_info TEXT,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_users_user_email ON users(user_email);
+
+CREATE TABLE IF NOT EXISTS users_address (
+    id INTEGER PRIMARY KEY,
+    user_id INTEGER,
+    address_id INTEGER UNIQUE,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_users_address_user_id ON users_address(user_id);
+
+CREATE INDEX IF NOT EXISTS idx_users_address_address_id ON users_address(address_id);
+
+CREATE TABLE IF NOT EXISTS user_roles (
+    id INTEGER PRIMARY KEY,
+    user_id INTEGER UNIQUE NOT NULL,
+    role_text TEXT,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_user_roles_user_id ON user_roles(user_id);
+
+CREATE TABLE IF NOT EXISTS user_passkeys (
+    id INTEGER PRIMARY KEY,
+    user_id INTEGER NOT NULL,
+    passkey_name TEXT NOT NULL,
+    passkey_id TEXT NOT NULL,
+    passkey TEXT NOT NULL,
+    counter INTEGER DEFAULT 0,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS idx_user_passkeys_user_id ON user_passkeys(user_id);
+
+CREATE UNIQUE INDEX IF NOT EXISTS idx_user_passkeys_user_id_passkey_id ON user_passkeys(user_id, passkey_id);
+`
+
+export default {
+    initialize: async (c: Context<HonoCustomType>) => {
+        // remove all \r and \n characters from the query string
+        // split by ; and join with a ;\n
+        const query = DB_INIT_QUERIES.replace(/[\r\n]/g, "")
+            .split(";")
+            .map((query) => query.trim())
+            .join(";\n");
+        await c.env.DB.exec(query);
+
+        const version = await utils.getSetting(c, CONSTANTS.DB_VERSION_KEY);
+        if (version) {
+            return c.json({ message: "Database already initialized" });
+        }
+        await utils.saveSetting(c, CONSTANTS.DB_VERSION_KEY, CONSTANTS.DB_VERSION);
+        return c.json({ message: "Database initialized" });
+    },
+    migrate: async (c: Context<HonoCustomType>) => {
+        const version = await utils.getSetting(c, CONSTANTS.DB_VERSION_KEY);
+        if (version != CONSTANTS.DB_VERSION) {
+            // TODO: Perform migration logic here
+
+            // Update the version in the settings table
+            await utils.saveSetting(c, CONSTANTS.DB_VERSION_KEY, CONSTANTS.DB_VERSION);
+            return c.json({
+                success: true,
+                message: "Database migrated"
+            });
+        }
+        return c.json({
+            success: true,
+            message: "Database does not need migration"
+        });
+    },
+    getVersion: async (c: Context<HonoCustomType>) => {
+        const version = await utils.getSetting(c, CONSTANTS.DB_VERSION_KEY);
+        return c.json({
+            need_initialization: !version,
+            need_migration: version && version != CONSTANTS.DB_VERSION,
+            current_db_version: version,
+            code_db_version: CONSTANTS.DB_VERSION
+        });
+    },
+}