bot.js

require("dotenv").config()
const {Telegraf} = require("telegraf")
const {Topics} = require("./topics")
const wait = hours => new Promise(resolve => setTimeout(resolve, hours * 60 * 60 * 1000))
const mongoose = require("mongoose")
const bot = new Telegraf(process.env.BOT_TOKEN)
const chatId = process.env.CHAT_ID
const dayjs = require("dayjs")
const Article = require("./models/Article")
const WebsiteService = require("./service/WebsiteService")
const NewsService = require("./service/NewsService")

const conf = require("./config");
const cron = require("node-cron");

const initDB = async () => {
    try {
        await mongoose.connect(process.env.MONGODB, {
            useNewUrlParser: true,
            useUnifiedTopology: true
        })

        console.log("Mongo connected...")
    } catch (error) {
        console.error("PROBLEMES DE CONEXIO")
        console.error(error)
        process.exit(0)
    }
}

const notificationNews = async () => {
    const websites = await WebsiteService.getAll()
    for (let website of websites) {
        let news_items = []
        try {
            console.log(`${dayjs().format("YYYY-MM-DD HH:mm:ss")} -> SCRAPPING: ${website.name}`)
            news_items = await NewsService.parseUrl(website.url)
            for (let item of news_items) {
                try {
                    let url = item.link.trim().toLowerCase()
                    console.log(`${dayjs().format("YYYY-MM-DD HH:mm:ss")} -> ${url}`)
                    await Article.create({url})
                    await bot.telegram.sendMessage(chatId, url)
                    console.log(`${dayjs().format("YYYY-MM-DD HH:mm:ss")} : Enviat: ${item.link}`)
                } catch (error) {
                    console.log(`${dayjs().format("YYYY-MM-DD HH:mm:ss")} : Exists: ${item.link}`)
                    console.log(error)
                }
                await wait(0.1)
            }
        } catch (error) {
            console.log(`${dayjs().format("YYYY-MM-DD HH:mm:ss")} : URL: ${website.name} ${error}`)
        }
    }
}

function hasArguments() {
    return process.argv.length > 2;
}

const init = async () => {
    await initDB();
    if (hasArguments()) {
        const args = process.argv.slice(2);
        if (args[0] === '-h') {
            console.log("TODO: HELP")
        } else if (args[0] === '-add') {
            const elementsToAdd = args.slice(1);

            let index = 0;
            while (index < elementsToAdd.length) {
                let name = elementsToAdd[index++];
                let url = elementsToAdd[index++];
                if (name && url) {
                    name = name.trim();
                    url = url.trim();
                    await WebsiteService.addWebsite(name, url);
                }
            }
            await mongoose.disconnect();
        } else if (args[0] === '-rem') {
            const elementsToRemove = args.slice(1);
            for (let index = 0; index < elementsToRemove.length; index++) {
                let name = elementsToRemove[index];
                if (name) {
                    name = name.trim();
                    await WebsiteService.removeWebsite(name);
                }
            }
            await mongoose.disconnect();
        } else if (args[0] === '-list') {
            const websites = await WebsiteService.getAll();
            if (websites.length > 0) {
                console.log("Registered websites:");
                websites.map((website => {
                    console.log(`${website.name} - ${website.url}`)
                }))
            }
            await mongoose.disconnect();
        }
    } else {
        bot.start(ctx => ctx.reply("Bot iniciat!"))

        Topics(bot)
        bot
            .launch()
            .then(() => console.log(`TELEGRAM BOT: "${chatId}" running...`))
            .catch(error => console.error(error))

        process.once("SIGINT", () => bot.stop("SIGINT"))
        process.once("SIGTERM", () => bot.stop("SIGTERM"))

        for (let cronConf of conf.crons) {
            cron.schedule(cronConf, function () {
                notificationNews()
            })
        }
    }
}

init();