botbase.ts

/** Base file to reduce the amount of boilerplate code in each file */

import * as fs from 'fs';
import * as path from 'path';
import * as child_process from 'child_process';
export {fs, path, child_process};

/** Notify by email on facing unexpected errors, see wikitech.wikimedia.org/wiki/Help:Toolforge/Email */
export function emailOnError(err: Error, taskname: string, isFatal = true) {
    logFullError(err, isFatal);
    child_process.exec(
        `echo "Subject: ${taskname} error\n\n${taskname} task resulted in the error:\n\n${err.stack}\n" | /usr/sbin/exim -odf -i tools.sdzerobot@tools.wmflabs.org`,
        () => {} // Emailing failed, must be a non-toolforge environ
    );
    // exit normally
}
export function logFullError(err: Error, isFatal = true) {
    // datetime similar to what mwn log produces, but can't use that directly as mwn may not have loaded
    const dateTimeString = new Date().toISOString().slice(0, 19).replace('T', ' ');
    console.log(`[${dateTimeString}] ${isFatal ? '[E] Fatal error' : '[E] Error'}`);
    console.log(err);
}

// Errors occurring inside async functions are caught by emailOnError(),
// this is only for anything else, such as failing imports
process.on('uncaughtException', function(err) {
    if (process.argv[1]) {
        var taskname = path.basename(process.argv[1]);
        emailOnError(err, taskname);
    } else { // else we're probably running in the console
        console.log(err);
    }
});

import {mwn} from '../mwn';
export {mwn};

/** Colorised and dated console logging. */
export const log = mwn.log;

/** Parsed console arguments */
export const argv = require('minimist')(process.argv.slice(2));

import {updateLoggingConfig} from "../mwn/build/log";
updateLoggingConfig({
    printVerbose: !!argv.verbose
});

/** bot account and database access credentials */
const auth = require('./.auth');
export class AuthManager {
    static get(account: string) {
        return auth[account];
    }
}

export const bot = new mwn({
    apiUrl: 'https://en.wikipedia.org/w/api.php',
    ...AuthManager.get('SDZeroBot:bp1'),
    OAuthCredentials: AuthManager.get('SDZeroBot:oauth1'),
    defaultParams: {
        assert: 'bot'
    },
    maxRetries: 7, // Nov 2020: lag on the roof
    userAgent: 'w:en:User:SDZeroBot'
});

bot.initOAuth();

import TextExtractor from "./TextExtractor";
export { TextExtractor };

// Deprecated exports, import from ./db or ./utils directly
export {mysql, db, enwikidb, toolsdb} from './db';
export * as utils from './utils';