database.js

const crypto = require("crypto");
require("dotenv").config();

const dbConfig = {
    type: process.env.DB_TYPE || "sqlite",
    sqlite: {
        path: process.env.DBPATH || "./test.db"
    },
    mysql: {
        host: process.env.DB_HOST || "localhost",
        user: process.env.DB_USER || "root",
        password: process.env.DB_PASSWORD || "",
        database: process.env.DB_NAME || "xjauth",
        port: process.env.DB_PORT || 3306
    }
};

function initDatabase() {
    if (dbConfig.type === "mysql") {
        return initMySQLDatabase();
    } else {
        return initSQLiteDatabase();
    }
}

function initSQLiteDatabase() {
    const Database = require("better-sqlite3");
    const db = new Database(dbConfig.sqlite.path);
    
    db.exec(`
        CREATE TABLE IF NOT EXISTS user (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            username TEXT UNIQUE NOT NULL,
            email TEXT UNIQUE NOT NULL,
            password TEXT NOT NULL,
            salt TEXT NOT NULL,
            created_at DATETIME DEFAULT CURRENT_TIMESTAMP
        )
    `);
    
    console.log(`SQLite database "${dbConfig.sqlite.path}" initialized.`);
    
    return {
        type: "sqlite",
        db: db,
        get: (sql, params = []) => db.prepare(sql).get(params),
        run: (sql, params = []) => db.prepare(sql).run(params)
    };
}

function initMySQLDatabase() {
    const mysql = require("mysql2/promise");
    const pool = mysql.createPool({
        ...dbConfig.mysql,
        waitForConnections: true,
        connectionLimit: 10,
        queueLimit: 0
    });
    async function createTable() {
        const connection = await pool.getConnection();
        try {
            await connection.execute(`
                CREATE TABLE IF NOT EXISTS user (
                    id INT AUTO_INCREMENT PRIMARY KEY,
                    username VARCHAR(255) UNIQUE NOT NULL,
                    email VARCHAR(255) UNIQUE NOT NULL,
                    password VARCHAR(255) NOT NULL,
                    salt VARCHAR(255) NOT NULL,
                    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
                )
            `);
            console.log(`MySQL database "${dbConfig.mysql.database}" initialized.`);
        } catch (error) {
            console.error("Error creating table:", error);
            throw error;
        } finally {
            connection.release();
        }
    }
    createTable();
    return {
        type: "mysql",
        pool: pool,
        get: async (sql, params = []) => {
            const [rows] = await pool.execute(sql, params);
            return rows[0];
        },
        run: async (sql, params = []) => {
            const [result] = await pool.execute(sql, params);
            return result;
        }
    };
}

function md5hash(password, salt = "") {
    return crypto.createHash("md5").update(password + salt).digest("hex");
}

function generateSalt(length = 16) {
    return crypto.randomBytes(Math.ceil(length / 2)).toString("hex").slice(0, length);
}

module.exports = { initDatabase, md5hash, generateSalt };