pcr-accepter.js

#!/usr/bin/env node

// Dependencies.
const argv = require('minimist')(process.argv.slice(2));
const MWBot = require('mwbot');
const mysql = require('mysql');
const util = require('util');

const credentials = require('./PCR_credentials'); // Load credentials from config.
const apiUrl = 'https://en.wikipedia.org/w/api.php';
const database = 'enwiki_p';
const acceptSummary = 'Task 65: Automatically accept, no net change';

/**
 * Log a message to stdout prepended with a timestamp.
 * @param {String} message
 */
function log(message) {
	const datestamp = new Date().toISOString().replace(/T/, ' ').replace(/\..+/, '');
	console.log(`${datestamp}: ${message}`);
}

/**
 * Retrieve bot instance
 * @returns MWBot
 */
async function getBot() {
	// Login to the bot.
	log(`Logging in to bot account`);
	const bot = new MWBot({apiUrl});
	await bot.loginGetEditToken({
		apiUrl,
		username: credentials.username,
		password: credentials.password
	});
	return bot;
}

/**
 * Accept the null changes
 * @param {int} revid
 * @param {MWBot} bot
 * @returns {Promise<void>}
 */
async function acceptNoChange( revid, bot ) {
	log(`Accepting revision ${revid}`);
	await bot.request( {
		action: 'review',
		revid: revid,
		comment: acceptSummary,
		token: bot.editToken
	} ).then( response => {
		console.log( response );
	} ).catch(err => {
		const error = err.response && err.response.error ? err.response.error.code : 'Unknown';
		log(`Failed to accept edits: ${error}`);
	});
}

/**
 * Get all pending changes
 * @param {MWBot} bot
 * @returns {Promise<Array|false>} pending changes
 */
function getPending( bot ) {
	log(`Querying pending changes via api`);
	return new Promise((resolve) => {
		bot.request( {
			action: 'query',
			list: 'oldreviewedpages',
			ormaxsize: 0,
			ornamespace: '*',
			orlimit: 'max',
			formatversion: 2,
	 	 } ).then( response => {
			console.log( response );
			resolve( response );
		  } ).catch(err => {
			const error = err.response && err.response.error ? err.response.error.code : 'Unknown';
			log(`Failed to get pending changes: ${error}`);
			resolve( false )
		});
	});
}

/**
 * Check pending changes to a page
 * @param {MWBot} bot
 * @param {int} oldid
 * @param {int} newid
 * @param {bool} dry
 */
async function checkPage( bot, oldid, newid, dry ) {
	const old_content = await getPage( bot, oldid );
	const new_content = await getPage( bot, newid );
	if ( old_content === new_content ) {
		log( `No net change between ${oldid} and ${newid}` );
		if ( dry ) {
			console.log( `Would accept ${newid}` );
		} else {
			await acceptNoChange( newid, bot );
		}
	}
}

/**
 * Get the wikitext of a page
 * @param {MWBot} bot
 * @param {int} revid
 * @returns {Promise<String|false>}
 */
function getPage( bot, revid ) {
	return new Promise((resolve) => {
		bot.request( {
			action: 'parse',
			oldid: revid,
			prop: 'wikitext'
	 	 } ).then( response => {
			//console.log( response );
			resolve( response.parse.wikitext['*'] );
		  } ).catch(err => {
			const error = err.response && err.response.error ? err.response.error.code : 'Unknown';
			log(`Failed to accept edits: ${error}`);
			resolve( false )
		});
	});
}

/**
 * Entry point for the bot task.
 * @returns {Promise<void>}
 */
async function main() {
	const bot = await getBot();
	const allPending = await getPending( bot );
	console.log( 'allPending', allPending );
	const pending = allPending.query.oldreviewedpages;
	console.log( 'pending', pending );
	for ( var iii = 0; iii < pending.length; iii++ ) {
		console.log( pending[iii].stable_revid, pending[iii].revid );
		await checkPage( bot, pending[iii].stable_revid, pending[iii].revid, argv.dry );
	}
	log('Task complete!');
	process.exit();
}

main().catch(console.error);