f2e-server3/lib/middlewares/auth/user_store.js

f2e-server 3.0

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
    __setModuleDefault(result, mod);
    return result;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.UserStore = void 0;
const node_crypto_1 = require("node:crypto");
const utils_1 = require("../../utils");
const fs = __importStar(require("node:fs"));
const dbfile_1 = require("../../utils/dbfile");
class UserStore {
    db_user;
    deleteCallbacks = [];
    mock_user;
    /**
     * @param db_path 用户密码存储文件路径
     * 用户密码文件格式（密码为md5密文）: 每行一个用户
     * ```
     *      username1:password1:nickname1
     *      username2:password2:nickname2
     *      username3:password3:nickname3
     * ```
     */
    constructor(db_path, mock_user) {
        this.mock_user = mock_user;
        this.db_user = new dbfile_1.DBFile({
            filePath: db_path,
            watchFile: true,
            initData: new Map([
                ['admin:e10adc3949ba59abbe56e057f20f883e', { username: 'admin', nickname: '管理员' }],
                ['user:e10adc3949ba59abbe56e057f20f883e', { username: 'user', nickname: '用户' }],
            ]),
            fileToData: (filePath) => {
                const lines = fs.readFileSync(filePath, 'utf8').split(/[\r\n]+/);
                const user_map = new Map();
                for (const line of lines) {
                    const [username, password, nickname] = line.split(':');
                    user_map.set(`${username}:${password}`, { username, nickname });
                }
                return user_map;
            },
            dataToFile: (data) => {
                const lines = [];
                data.forEach((v, key) => {
                    lines.push(`${key}:${v.nickname}`);
                });
                return lines.join('\n');
            },
            onFileChange: (newData, oldData) => {
                const deleteCallbacks = this.deleteCallbacks;
                oldData?.forEach(function (v, key) {
                    if (!newData?.has(key)) {
                        const username = key.split(':')[0];
                        utils_1.logger.info(`user: "${username}" deleted`);
                        deleteCallbacks.forEach(callback => callback(username));
                    }
                });
            }
        });
    }
    async getUser(username, password) {
        const pwd = (0, node_crypto_1.createHash)('md5').update(password).digest('hex');
        return this.db_user.data?.get(`${username}:${pwd}`);
    }
    onDeleteUser(callback) {
        this.deleteCallbacks.push(callback);
    }
    async getLoginUser() {
        return this.mock_user;
    }
}
exports.UserStore = UserStore;