gocqhttp-twitter-bot/src/main.ts

#!/usr/bin/env node

import * as fs from 'fs';
import * as path from 'path';

import * as commandLineUsage from 'command-line-usage';

import * as exampleConfig from '../config.example.json';
import { list, sub, unsub } from './command';
import { getLogger, setLogLevels } from './loggers';
import QQBot from './koishi';
import Worker from './twitter';

const logger = getLogger();

const sections: commandLineUsage.Section[] = [
  {
    header: 'GoCQHTTP Instagram Bot',
    content: 'The QQ Bot that forwards Instagram.',
  },
  {
    header: 'Synopsis',
    content: [
      '$ cq-instagram-bot {underline config.json}',
      '$ cq-instagram-bot {bold --help}',
    ],
  },
  {
    header: 'Documentation',
    content: [
      'Project home: {underline https://github.com/CL-Jeremy/mirai-twitter-bot}',
      'Example config: {underline https://git.io/JJ0jN}',
    ],
  },
];

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];

type Config = typeof exampleConfig;
let config: Config;
try {
  config = JSON.parse(fs.readFileSync(path.resolve(configPath), 'utf8')) as Config;
} catch (e) {
  console.log('Failed to parse config file: ', configPath);
  console.log(usage);
  process.exit(1);
}

const requiredFields = [
  'ig_username', 'ig_password',
  'cq_bot_qq', ...(config.mode || exampleConfig.mode) === 0 ? ['playwright_ws_spec_endpoint'] : [],
];

const warningFields = [
  'cq_ws_host', 'cq_ws_port', 'cq_access_token',
];

const optionalFields = [
  'lockfile', 'inactive_hours', 'work_interval', 'webshot_delay', 'loglevel', 'mode', 'resume_on_start', 'ig_socks_proxy',
].concat(warningFields);

if (requiredFields.some((value) => config[value] === undefined)) {
  console.log(`${requiredFields.join(', ')} are required`);
  process.exit(1);
}

optionalFields.forEach(key => {
  if (config[key] === undefined || typeof(config[key]) !== typeof (exampleConfig[key])) {
    if (warningFields.includes(key)) logger.warn(`${key} is undefined, use ${exampleConfig[key] || 'empty string'} as default`);
    config[key] = exampleConfig[key as keyof Config];
  }
});

['ig_session_lockfile', ...(config.mode || exampleConfig.mode) === 0 ? ['webshot_cookies_lockfile'] : []].forEach(key => {
  if (!config[key]) {
    logger.warn(`${key} is undefined, use <username>.${key.replace('_lockfile', '.lock')} as default`);
    config[key] = `${config.ig_username}.${key.replace('_lockfile', '.lock')}`;
  }
});

const k = 'ig_2fa_code_receiver_port';
if (!config[k] || config[k] < 2048 || config[k] > 65536) {
  logger.warn(`invalid value of config.${k}, use ${exampleConfig[k]} as default`);
  config[k] = exampleConfig[k];
}

setLogLevels(config.loglevel);

let lock: ILock;
if (fs.existsSync(path.resolve(config.lockfile))) {
  try {
    lock = JSON.parse(fs.readFileSync(path.resolve(config.lockfile), 'utf8')) as ILock;
  } 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);
  }
}

if (!config.resume_on_start) {
  Object.keys(lock.threads).forEach(key => {
    lock.threads[key].offset = '-1';
  });
}

const qq = new QQBot({
  access_token: config.cq_access_token,
  host: config.cq_ws_host,
  port: config.cq_ws_port,
  bot_id: config.cq_bot_qq,
  list: (c, a, cb) => list(c, a, cb, lock),
  sub: (c, a, cb) => sub(c, a, cb, lock, config.lockfile),
  unsub: (c, a, cb) => unsub(c, a, cb, lock, config.lockfile),
});

const worker = new Worker({
  sessionLockfile: config.ig_session_lockfile,
  credentials: [config.ig_username, config.ig_password],
  codeServicePort: config.ig_2fa_code_receiver_port,
  proxyUrl: config.ig_socks_proxy,
  lock,
  lockfile: config.lockfile,
  inactiveHours: config.inactive_hours,
  workInterval: config.work_interval,
  bot: qq,
  webshotDelay: config.webshot_delay,
  webshotCookiesLockfile: config.webshot_cookies_lockfile,
  mode: config.mode,
  wsUrl: config.playwright_ws_spec_endpoint,
});
worker.session.init().then(worker.launch);

qq.connect();