Upgrade to Telegraf 4.3

C0rn3j · May 7, 2021 · 96e6a3c · 96e6a3c
1 parent a71292a
commit 96e6a3c
Show file tree

Hide file tree

Showing 15 changed files with 59 additions and 56 deletions.
diff --git a/handlers/commands/admin.js b/handlers/commands/admin.js
@@ -13,41 +13,41 @@ const {
 } = require('../../stores/user');
 
 /** @param { import('../../typings/context').ExtendedContext } ctx */
-const adminHandler = async ({ from, message, replyWithHTML }) => {
-	if (!isMaster(from)) return null;
+const adminHandler = async (ctx) => {
+	if (!isMaster(ctx.from)) return null;
 
-	const { targets } = parse(message);
+	const { targets } = parse(ctx.message);
 
 	if (targets.length > 1) {
-		return replyWithHTML(
+		return ctx.replyWithHTML(
 			'ℹ️ <b>Specify one user to promote.</b>',
 		).then(scheduleDeletion());
 	}
 
 	const userToAdmin = targets.length
 		? await getUser(strip(targets[0]))
-		: from;
+		: ctx.from;
 
 	if (!userToAdmin) {
-		return replyWithHTML(
+		return ctx.replyWithHTML(
 			'❓ <b>User unknown.</b>\n' +
 			'Please forward their message, then try again.',
 		).then(scheduleDeletion());
 	}
 
 	if (userToAdmin.status === 'banned') {
-		return replyWithHTML('ℹ️ <b>Can\'t admin banned user.</b>');
+		return ctx.replyWithHTML('ℹ️ <b>Can\'t admin banned user.</b>');
 	}
 
 	if (userToAdmin.status === 'admin') {
-		return replyWithHTML(
+		return ctx.replyWithHTML(
 			html`⭐️ ${link(userToAdmin)} <b>is already admin.</b>`,
 		);
 	}
 
 	await admin(userToAdmin);
 
-	return replyWithHTML(html`⭐️ ${link(userToAdmin)} <b>is now admin.</b>`);
+	return ctx.replyWithHTML(html`⭐️ ${link(userToAdmin)} <b>is now admin.</b>`);
 };
 
 module.exports = adminHandler;
diff --git a/handlers/commands/groups.js b/handlers/commands/groups.js
@@ -30,15 +30,15 @@ const emojiRegex = XRegExp.tag('gx')`
 const stripEmoji = s => s.replace(emojiRegex, '');
 
 /** @param { import('../../typings/context').ExtendedContext } ctx */
-const groupsHandler = async ({ replyWithHTML }) => {
+const groupsHandler = async (ctx) => {
 	const groups = await listVisibleGroups();
 
 	groups.sort((a, b) =>
 		stripEmoji(a.title).localeCompare(stripEmoji(b.title)));
 
 	const entries = TgHtml.join('\n', groups.map(entry));
 
-	return replyWithHTML(TgHtml.tag`🛠 <b>Groups I manage</b>:\n\n${entries}`, {
+	return ctx.replyWithHTML(TgHtml.tag`🛠 <b>Groups I manage</b>:\n\n${entries}`, {
 		disable_web_page_preview: true,
 		reply_markup,
 	}).then(scheduleDeletion());

diff --git a/handlers/commands/help.js b/handlers/commands/help.js
@@ -19,10 +19,10 @@ might be a better choice for you.
 `;
 
 /** @param { import('../../typings/context').ExtendedContext } ctx */
-const helpHandler = ({ chat, replyWithHTML }) => {
-	if (chat.type !== 'private') return null;
+const helpHandler = (ctx) => {
+	if (ctx.chat.type !== 'private') return null;
 
-	return replyWithHTML(
+	return ctx.replyWithHTML(
 		message,
 		Markup.inlineKeyboard([
 			Markup.urlButton('🛠 Setup a New Bot', homepage)

diff --git a/handlers/commands/link.js b/handlers/commands/link.js
@@ -7,14 +7,14 @@ const { scheduleDeletion } = require('../../utils/tg');
 const { managesGroup } = require('../../stores/group');
 
 /** @param { import('../../typings/context').ExtendedContext } ctx */
-const linkHandler = async ({ chat, replyWithHTML }, next) => {
-	if (chat.type === 'private') {
+const linkHandler = async (ctx, next) => {
+	if (ctx.chat.type === 'private') {
 		return next();
 	}
 
-	const group = await managesGroup({ id: chat.id });
+	const group = await managesGroup({ id: ctx.chat.id });
 
-	return replyWithHTML(group.link || '️ℹ️ <b>No link to this group</b>', {
+	return ctx.replyWithHTML(group.link || '️ℹ️ <b>No link to this group</b>', {
 		disable_web_page_preview: false,
 	}).then(scheduleDeletion());
 };

diff --git a/handlers/commands/removeCommand.js b/handlers/commands/removeCommand.js
@@ -6,39 +6,39 @@ const { getCommand, removeCommand } = require('../../stores/command');
 const { isMaster } = require('../../utils/config');
 
 /** @param { import('../../typings/context').ExtendedContext } ctx */
-const removeCommandHandler = async ({ from, chat, message, replyWithHTML }) => {
-	const { text } = message;
-	if (chat.type !== 'private') return null;
+const removeCommandHandler = async (ctx) => {
+	const { text } = ctx.message;
+	if (ctx.chat.type !== 'private') return null;
 
-	if (from.status !== 'admin') {
-		return replyWithHTML(
+	if (ctx.from.status !== 'admin') {
+		return ctx.replyWithHTML(
 			'ℹ️ <b>Sorry, only admins access this command.</b>',
 		);
 	}
 	const [ , commandName ] = text.split(' ');
 	if (!commandName) {
-		return replyWithHTML(
+		return ctx.replyWithHTML(
 			'<b>Send a valid command.</b>\n\nExample:\n' +
 			'<code>/removecommand rules</code>',
 		);
 	}
 
 	const command = await getCommand({ name: commandName.toLowerCase() });
 	if (!command) {
-		return replyWithHTML(
+		return ctx.replyWithHTML(
 			'ℹ️ <b>Command couldn\'t be found.</b>',
 		);
 	}
 
 	const role = command.role.toLowerCase();
-	if (role === 'master' && !isMaster(from)) {
-		return replyWithHTML(
+	if (role === 'master' && !isMaster(ctx.from)) {
+		return ctx.replyWithHTML(
 			'ℹ️ <b>Sorry, only master can remove this command.</b>',
 		);
 	}
 
 	await removeCommand({ name: commandName.toLowerCase() });
-	return replyWithHTML(
+	return ctx.replyWithHTML(
 		`✅ <code>!${commandName}</code> ` +
 		'<b>has been removed successfully.</b>',
 	);

diff --git a/handlers/commands/unadmin.js b/handlers/commands/unadmin.js
@@ -29,27 +29,27 @@ const tgUnadmin = async (userToUnadmin) => {
 };
 
 /** @param { import('../../typings/context').ExtendedContext } ctx */
-const unAdminHandler = async ({ from, message, replyWithHTML }) => {
-	if (!isMaster(from)) return null;
+const unAdminHandler = async (ctx) => {
+	if (!isMaster(ctx.from)) return null;
 
-	const { targets } = parse(message);
+	const { targets } = parse(ctx.message);
 
 	if (targets.length !== 1) {
-		return replyWithHTML(
+		return ctx.replyWithHTML(
 			'ℹ️ <b>Specify one user to unadmin.</b>',
 		).then(scheduleDeletion());
 	}
 
 	const userToUnadmin = await getUser(strip(targets[0]));
 
 	if (!userToUnadmin) {
-		return replyWithHTML(
+		return ctx.replyWithHTML(
 			'❓ <b>User unknown.</b>',
 		).then(scheduleDeletion());
 	}
 
 	if (userToUnadmin.status !== 'admin') {
-		return replyWithHTML(
+		return ctx.replyWithHTML(
 			html`ℹ️ ${link(userToUnadmin)} <b>is not admin.</b>`,
 		);
 	}
@@ -58,7 +58,7 @@ const unAdminHandler = async ({ from, message, replyWithHTML }) => {
 
 	await unadmin(userToUnadmin);
 
-	return replyWithHTML(
+	return ctx.replyWithHTML(
 		html`❗️ ${link(userToUnadmin)} <b>is no longer admin.</b>`,
 	);
 };

diff --git a/handlers/commands/user.js b/handlers/commands/user.js
@@ -50,33 +50,33 @@ const title = user => {
 };
 
 /** @param { import('../../typings/context').ExtendedContext } ctx */
-const getWarnsHandler = async ({ from, message, replyWithHTML }) => {
-	if (!from) {
-		return replyWithHTML(
+const getWarnsHandler = async (ctx) => {
+	if (!ctx.from) {
+		return ctx.replyWithHTML(
 			'ℹ️ <b>This command is not available in channels.</b>',
 		).then(scheduleDeletion());
 	}
 
-	const { flags, targets } = parse(message);
+	const { flags, targets } = parse(ctx.message);
 
 	if (targets.length > 1) {
-		return replyWithHTML(
+		return ctx.replyWithHTML(
 			'ℹ️ <b>Specify one user.</b>',
 		).then(scheduleDeletion());
 	}
 
-	const theUser = targets.length && from.status === 'admin'
+	const theUser = targets.length && ctx.from.status === 'admin'
 		? await getUser(strip(targets[0]))
-		: from;
+		: ctx.from;
 
 	if (!theUser) {
-		return replyWithHTML(
+		return ctx.replyWithHTML(
 			'❓ <b>User unknown.</b>',
 		).then(scheduleDeletion());
 	}
 
-	if (flags.has('raw') && from.status === 'admin') {
-		return replyWithHTML(
+	if (flags.has('raw') && ctx.from.status === 'admin') {
+		return ctx.replyWithHTML(
 			TgHtml.pre(inspect(theUser)),
 		).then(scheduleDeletion());
 	}
@@ -111,7 +111,7 @@ const getWarnsHandler = async ({ from, message, replyWithHTML }) => {
 		permitS,
 	].filter(isNotEmpty));
 
-	return replyWithHTML(TgHtml.join('\n\n', [
+	return ctx.replyWithHTML(TgHtml.join('\n\n', [
 		oneliners,
 		userWarns,
 		banReason,

diff --git a/handlers/middlewares/commandButtons.js b/handlers/middlewares/commandButtons.js
@@ -24,7 +24,7 @@ module.exports = (ctx, next) => {
 	Object.assign(cbCtx, contextCustomizations);
 	cbCtx.botInfo = ctx.botInfo;
 
-	cbCtx.reply = ctx.editMessageText;
+	cbCtx.reply = ctx.editMessageText.bind(ctx);
 
 	return next(cbCtx);
 };
diff --git a/handlers/middlewares/removeChannelForwards.js b/handlers/middlewares/removeChannelForwards.js
@@ -1,8 +1,7 @@
 'use strict';
 
 const R = require('ramda');
-const { optional, passThru } = require('telegraf');
-
+const { Telegraf: { optional, passThru } } = require('telegraf');
 const { permit } = require('../../stores/user');
 
 const { html, lrm } = require('../../utils/html');

diff --git a/handlers/regex/groupLinker.js b/handlers/regex/groupLinker.js
@@ -1,6 +1,6 @@
 'use strict';
 
-const { hears } = require('telegraf');
+const { Telegraf: { hears } } = require('telegraf');
 const XRegExp = require('xregexp');
 
 const { managesGroup } = require.main.require('./stores/group');

diff --git a/handlers/regex/index.js b/handlers/regex/index.js
@@ -1,6 +1,6 @@
 'use strict';
 
-const { compose, hears } = require('telegraf');
+const { Telegraf: { compose, hears } } = require('telegraf');
 
 /* eslint-disable global-require */
 

diff --git a/handlers/regex/runCustomCmd.js b/handlers/regex/runCustomCmd.js
@@ -1,6 +1,6 @@
 'use strict';
 
-const { hears } = require('telegraf');
+const { Telegraf: { hears } } = require('telegraf');
 const R = require('ramda');
 
 // DB
@@ -63,7 +63,11 @@ const runCustomCmdHandler = async (ctx, next) => {
 	};
 
 	return ctx[typeToMethod(type)](content, options)
-		.then(scheduleDeletion(autoDelete(command) && deleteCustom.after));
+		.then(({ message_id }) =>
+			scheduleDeletion(
+				autoDelete(command) && deleteCustom.after)({
+					chat: ctx.chat, message_id
+				}));
 };
 
 module.exports = hears(/^! ?(\w+)/, runCustomCmdHandler);
diff --git a/package.json b/package.json
@@ -41,7 +41,7 @@
     "require-directory": "^2.1.1",
     "spamwatch": "^0.2.0",
     "string-replace-async": "^1.2.1",
-    "telegraf": "^3.38.0",
+    "telegraf": "^4.3.0",
     "ts-node": "^8.9.1",
     "typescript": "^3.8.3",
     "xregexp": "^4.2.0"

diff --git a/plugins/index.js b/plugins/index.js
@@ -1,6 +1,6 @@
 'use strict';
 
-const { compose } = require('telegraf');
+const { Telegraf: { compose } } = require('telegraf');
 
 const { config } = require('../utils/config');
 const names = config.plugins || [];

diff --git a/utils/tg.js b/utils/tg.js
@@ -33,7 +33,7 @@ const displayUser = user =>
 /** @param {number | string | false} ms */
 const deleteAfter = ms => (ctx, next) => {
 	if (ms !== false) {
-		setTimeout(ctx.deleteMessage, millisecond(ms));
+		setTimeout(ctx.deleteMessage.bind(ctx), millisecond(ms));
 	}
 	next();
 };