wechaty-puppet-official-account/dist/esm/src/official-account/payload-store.js

Wechaty Puppet for WeChat Official Accounts

import path from 'path';
import os from 'os';
import fs from 'fs';
import { log } from 'wechaty-puppet';
import { FlashStore } from 'flash-store';
import LRU from 'lru-cache';
import { VERSION } from '../config.js';
import semverPkg from 'semver';
const { major, minor } = semverPkg;
class PayloadStore {
    appId;
    cacheOAContactPayload;
    cacheOAMessagePayload;
    constructor(appId) {
        this.appId = appId;
        log.verbose('PayloadStore', 'constructor(%s)', appId);
    }
    async start() {
        log.verbose('PayloadStore', 'start()');
        if (this.cacheOAMessagePayload) {
            throw new Error('PayloadStore should be stop() before start() again.');
        }
        /**
         * FlashStore
         */
        const baseDir = path.join(os.homedir(), '.wechaty', 'wechaty-puppet-official-account', `v${major(VERSION)}.${minor(VERSION)}`, this.appId);
        if (!fs.existsSync(baseDir)) {
            fs.mkdirSync(baseDir, { recursive: true });
        }
        this.cacheOAContactPayload = new FlashStore(path.join(baseDir, 'oa-contact-raw-payload'));
        /**
         * LRU
         */
        const lruOptions = {
            dispose(key, val) {
                log.silly('PayloadStore', `constructor() lruOptions.dispose(${key}, ${JSON.stringify(val)})`);
            },
            max: 1000,
            maxAge: 1000 * 60 * 60,
        };
        this.cacheOAMessagePayload = new LRU(lruOptions);
    }
    async stop() {
        log.verbose('PayloadStore', 'stop()');
        if (this.cacheOAMessagePayload) {
            this.cacheOAMessagePayload = undefined;
        }
        if (this.cacheOAContactPayload) {
            await this.cacheOAContactPayload.close();
            this.cacheOAContactPayload = undefined;
        }
    }
    async getMessagePayload(id) {
        log.verbose('PayloadStore', 'getMessagePayload(%s)', id);
        return this.cacheOAMessagePayload?.get(id);
    }
    async setMessagePayload(id, payload) {
        log.verbose('PayloadStore', 'setMessagePayload(%s, %s)', id, JSON.stringify(payload));
        await this.cacheOAMessagePayload?.set(id, payload);
    }
    async getContactPayload(id) {
        log.verbose('PayloadStore', 'getContactPayload(%s)', id);
        return this.cacheOAContactPayload?.get(id);
    }
    async setContactPayload(id, payload) {
        log.verbose('PayloadStore', 'setContactPayload(%s, %s)', id, JSON.stringify(payload));
        await this.cacheOAContactPayload?.set(id, payload);
    }
}
export { PayloadStore };
//# sourceMappingURL=payload-store.js.map