rss-telegram-bot/bot.js

RSSBot 是一个能帮你订阅 RSSHub 的 Telegram Bot

const TelegramBot = require('node-telegram-bot-api');
const config = require('./config');
const logger = require('./utils/logger');
const user = require('./user');

// replace the value below with the Telegram token you receive from @BotFather
const token = config.telegram_bot_token;

// Create a bot that uses 'polling' to fetch new updates
const bot = new TelegramBot(token, { polling: true });

bot._sendMessage = bot.sendMessage
bot.sendMessage = function (chatId, text, options) {
    logger.info(`Message back ${chatId}: ${text}`);
    return bot._sendMessage(chatId, text, options)
}
// Listen for any kind of message. There are different kinds of
// messages.
bot.on('message', (msg) => {
    const chatId = msg.chat.id;
    const msgId = msg.message_id;
    const text = msg.text;
    logger.info(`Message from ${chatId}: ${text}`)
    if (config.reply_received !== "false") {
        // send a message to the chat acknowledging receipt of their message
        bot._sendMessage(chatId, 'Received your message', {
            disable_web_page_preview: true,
            reply_to_message_id: msgId
        });
    }
});

let username;
let fetching = false;
async function getUsername() {
    if (!username) {
        if (!fetching) {
            fetching = true;
            const me = await bot.getMe();
            username = me.username;
            fetching = false;
        } else {
            while (fetching)
                await new Promise(resolve => setTimeout(resolve, 0));
        }
    }
    if (!username) process.exit(1);
    return username;
}

bot.onValidText = function (regexp, callback) {
    bot.onText(regexp, (msg, ...args) => {
        if (!user.validate(msg)) {
            bot._sendMessage(msg.chat.id, '下去！这是私家车');
            return;
        }
        callback(msg, ...args)
    });
}

let onQueryInit = 0;
bot.onQuery = async function (regexp, callback) {
    onQueryInit += 1;
    logger.info(`Methods "onQuery" initializing: ${onQueryInit}`);
    const username = await getUsername();
    bot.onValidText(new RegExp(`^@${username} (.+)`), (msg, match) => {
        if (!user.validate(msg)) {
            return;
        }
        if (match && match.length > 1) {
            const result = regexp.exec(match[1]);
            if (!result) {
                return false;
            }
            // reset index so we start at the beginning of the regex each time
            regexp.lastIndex = 0;
            callback(msg, result);
        }
    });
    onQueryInit -= 1;
    if (onQueryInit <= 0) {
        logger.info('All Methods "onQuery" initialized');
    }
}

bot.sendDeleteMessage = async function (chatId, delMsgId, text, options) {
    await bot.sendMessage(chatId, text, options);
    await bot.deleteMessage(chatId, delMsgId); // 阅后即焚
}

bot.setMyCommands([
    { command: 'ping', description: 'return you something' }
])

bot.onPing = (callback) => bot.onValidText(new RegExp(`^/ping`), callback)

module.exports = bot;