gocqhttp-twitter-bot/dist/main.js

#!/usr/bin/env node
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const commandLineUsage = require("command-line-usage");
const fs = require("fs");
const log4js = require("log4js");
const path = require("path");
const logger = log4js.getLogger();
logger.level = 'info';
const sections = [
    {
        header: 'CQHTTP Twitter Bot',
        content: 'The QQ Bot that forwards twitters.',
    },
    {
        header: 'Synopsis',
        content: [
            '$ cqhttp-twitter-bot {underline config.json}',
            '$ cqhttp-twitter-bot {bold --help}',
        ],
    },
    {
        header: 'Documentation',
        content: [
            'Project home: {underline https://github.com/rikakomoe/cqhttp-twitter-bot}',
            'Example config: {underline https://qwqq.pw/qpfhg}',
        ],
    },
];
const usage = commandLineUsage(sections);
const args = process.argv.slice(2);
if (args.length === 0 || args[0] === 'help' || args[0] === '-h' || args[0] === '--help') {
    console.log(usage);
    process.exit(0);
}
const configPath = args[0];
let config;
try {
    config = require(path.resolve(configPath));
}
catch (e) {
    console.log('Failed to parse config file: ', configPath);
    console.log(usage);
    process.exit(1);
}
if (config.twitter_consumer_key === undefined ||
    config.twitter_consumer_secret === undefined ||
    config.twitter_access_token_key === undefined ||
    config.twitter_access_token_secret === undefined) {
    console.log('twitter_consumer_key twitter_consumer_secret twitter_access_token_key twitter_access_token_secret are required');
    process.exit(1);
}
if (config.cq_ws_host === undefined) {
    config.cq_ws_host = '127.0.0.1';
    logger.warn('cq_ws_host is undefined, use 127.0.0.1 as default');
}
if (config.cq_ws_port === undefined) {
    config.cq_ws_port = 6700;
    logger.warn('cq_ws_port is undefined, use 6700 as default');
}
if (config.cq_access_token === undefined) {
    config.cq_access_token = '';
    logger.warn('cq_access_token is undefined, use empty string as default');
}
if (config.lockfile === undefined) {
    config.lockfile = 'subscriber.lock';
}
if (config.work_interval === undefined) {
    config.work_interval = 60;
}
if (config.webshot_delay === undefined) {
    config.webshot_delay = 5000;
}
if (config.loglevel === undefined) {
    config.loglevel = 'info';
}
let redisConfig;
if (config.redis) {
    redisConfig = {
        redisHost: config.redis_host || "127.0.0.1",
        redisPort: config.redis_port || 6379,
        redisExpireTime: config.redis_expire_time || 43200,
    };
}
global.loglevel = config.loglevel;
const command_1 = require("./command");
const cqhttp_1 = require("./cqhttp");
const twitter_1 = require("./twitter");
let lock;
if (fs.existsSync(path.resolve(config.lockfile))) {
    try {
        lock = JSON.parse(fs.readFileSync(path.resolve(config.lockfile), 'utf8'));
    }
    catch (err) {
        logger.error(`Failed to parse lockfile ${config.lockfile}: `, err);
        lock = {
            workon: 0,
            feed: [],
            threads: {},
        };
    }
    fs.access(path.resolve(config.lockfile), fs.constants.W_OK, err => {
        if (err) {
            logger.fatal(`cannot write lockfile ${path.resolve(config.lockfile)}, permission denied`);
            process.exit(1);
        }
    });
}
else {
    lock = {
        workon: 0,
        feed: [],
        threads: {},
    };
    try {
        fs.writeFileSync(path.resolve(config.lockfile), JSON.stringify(lock));
    }
    catch (err) {
        logger.fatal(`cannot write lockfile ${path.resolve(config.lockfile)}, permission denied`);
        process.exit(1);
    }
}
Object.keys(lock.threads).forEach(key => {
    lock.threads[key].offset = -1;
});
const qq = new cqhttp_1.default({
    access_token: config.cq_access_token,
    host: config.cq_ws_host,
    port: config.cq_ws_port,
    list: (c, a) => command_1.list(c, a, lock),
    sub: (c, a) => command_1.sub(c, a, lock, config.lockfile),
    unsub: (c, a) => command_1.unsub(c, a, lock, config.lockfile),
});
const worker = new twitter_1.default({
    consumer_key: config.twitter_consumer_key,
    consumer_secret: config.twitter_consumer_secret,
    access_token_key: config.twitter_access_token_key,
    access_token_secret: config.twitter_access_token_secret,
    lock,
    lockfile: config.lockfile,
    workInterval: config.work_interval,
    bot: qq,
    webshotDelay: config.webshot_delay,
    redis: redisConfig,
});
worker.launch();
qq.connect();