package.json

@@ -10,6 +10,7 @@
     "start": "node src/index.js"
   },
   "dependencies": {
+    "cron": "^1.8.2",
     "dotenv": "^8.2.0",
     "knex": "^0.21.18",
     "pg": "^8.5.1",

src/actions/setup_feed_actions.js

@@ -21,7 +21,7 @@ module.exports = ({ bot }) => {
     ctx.deleteMessage();
    
     const userID = getUserId(ctx, 'action');
-    chatsList = await listChats(chatRepository, userID);
+    const chatsList = await listChats(chatRepository, userID);
     
     const defaultMarkup = [
       [{ text: manager_feeds.action_update_list, callback_data: 'manager_feeds'},

src/commands/index.js

@@ -1,12 +1,6 @@
-const { 
-  start_bot, start_service, 
-  home, cmd_error, not_member_admin, 
-  view_chat, add_feed, remove_feed
-} = require('../messages/commands');
-const  { go_back_btn } = require('../messages/inline_keyboard');
-
+const { start_service, home, cmd_error, not_member_admin, view_chat, add_feed, remove_feed, active_feed } = require('../messages/commands');
 const { homeMarkup, timezonesMarkup, isNotMemberOrAdminMarkup, goBackManagerFeedsMarkup } = require('../markups');
-const { getChatId, isBotAdmin, isAdmin, isStillMemberAndAdmin } = require('../helpers/bot_helpers');
+const { getChatId, isBotAdmin, isAdmin, isStillMemberAndAdmin, removeCommand } = require('../helpers/bot_helpers');
 const { asyncFilter, isHashtagsValid, removeSpacesInArray, removeNotHashtagsInArray, listFeeds } = require('../helpers/features_helpers');
 
 const User = require('../domain/User');
@@ -30,32 +24,16 @@ module.exports = bot => {
 
   bot.start(async ctx => {
     const { type } = await ctx.getChat();
-    const userID = String(ctx.from.id);
+    const userID = ctx.from.id;
+    const username = ctx.from.username;
 
     if (type === 'private') {
-      !(await userRepository.existsUser(userID))
-        ? ctx.telegram.sendMessage(getChatId(ctx), start_bot.text, timezonesMarkup)
-        : ctx.telegram.sendMessage(getChatId(ctx), home.text, homeMarkup);
+      await userRepository.add({ user_id: userID, username });
+
+      ctx.telegram.sendMessage(getChatId(ctx), home.text, homeMarkup);
     } else {
       ctx.telegram.sendMessage(getChatId(ctx), 'Olá!'); 
     }
-
-    return;
-  })
-
-  bot.command('register', async ctx => {
-    const [, timezone] = ctx.message.text.split(' ');
-
-    if (!timezone) {
-      ctx.reply(cmd_error.text);
-      return;
-    }
-
-    const user = new User(ctx.from.id, ctx.from.username, timezone);
-    await userRepository.add(user.getValue());
-    
-    ctx.telegram.sendMessage(getChatId(ctx), home.text, homeMarkup);
-
     return;
   })
 
@@ -137,7 +115,7 @@ module.exports = bot => {
     }
 
     const userID = ctx.message.from.id;
-    const { id: chatID, title } = await chatRepository.getOneChatByTitle(String(userID), chatTitle);
+    const { id: chatID } = await chatRepository.getOneChatByTitle(String(userID), chatTitle);
     
     const getMessage = async () => {
       const feedsList = await listFeeds(feedRepository, chatID);
@@ -156,6 +134,7 @@ module.exports = bot => {
 
       bot.command('add', async ctx => await addFeedCommand(ctx, chatID, getMessage));
       bot.command('remove', async ctx => await removeFeedCommand(ctx, chatID, getMessage));
+      bot.command('active', async ctx => await activeChatCommand(ctx, chatID, getMessage));
 
     } else {
       ctx.telegram.sendMessage(chatID, not_member_admin.text, isNotMemberOrAdminMarkup);
@@ -166,12 +145,17 @@ module.exports = bot => {
   })
 
   async function addFeedCommand(ctx, chat_id, getMessage) {
-    let data = ctx.message.text.replace('/add', '').trim().split(' ');
+    let data = removeCommand(ctx.message.text, '/add').split(' ');
     data = removeSpacesInArray(data);
 
     const rssURL = data.slice(0, 1).toString();
     const hashtags = removeNotHashtagsInArray(data);
 
+    if (hashtags.length > 3) {
+      ctx.telegram.sendMessage(getChatId(ctx), add_feed.max_hashtags);
+      return;
+    }
+
     if (!isHashtagsValid(hashtags)) {
       ctx.telegram.sendMessage(getChatId(ctx), add_feed.cmd_error, { parse_mode: 'HTML'});
       return;
@@ -186,12 +170,8 @@ module.exports = bot => {
         return;
       }
 
-      await feedRepository.addFeed({ 
-        rss_url: rssURL,
-        hashtag: hashtags_formatted,
-        title,
-        chat_id
-      });
+      const feed = new Feed(rssURL, hashtags_formatted, title, chat_id);
+      await feedRepository.addFeed(feed.getValue());
 
       ctx.reply(add_feed.success);
       const message = await getMessage();
@@ -205,7 +185,7 @@ module.exports = bot => {
   }
 
   async function removeFeedCommand(ctx, chat_id, getMessage) {
-    const feedTitle = ctx.message.text.replace('/remove', '').trim();
+    const feedTitle = removeCommand(ctx.message.text, '/remove');
 
     if (!feedTitle) {
       ctx.telegram.sendMessage(getChatId(ctx), remove_feed.cmd_error, { parse_mode: 'HTML'});
@@ -225,5 +205,18 @@ module.exports = bot => {
     return;
   }
 
-  setupFeedActions({bot});
+  async function activeChatCommand(ctx, chat_id, getMessage) {
+    if (!(await feedRepository.containChat(chat_id))) {
+      ctx.reply(active_feed.error);
+      return;
+    }
+
+    await chatRepository.updateActiveChat(true, chat_id);
+
+    ctx.reply(active_feed.success);
+    const message = await getMessage();
+    ctx.telegram.sendMessage(...message);
+  }
+
+  setupFeedActions({ bot });
 }

src/core/index.js

@@ -0,0 +1,80 @@
+const RssParser = require('../drivers/rss-parser');
+
+const ChatRepository = require('../infra/repositories/chat_repository');
+const FeedRepository = require('../infra/repositories/feed_repository');
+const PostRepository = require('../infra/repositories/post_repository');
+
+const chatRepository = new ChatRepository();
+const feedRepository = new FeedRepository();
+const postRepository = new PostRepository();
+
+const rssParser = new RssParser();
+
+async function getChatsData(oneChatId) {
+  const chatsIds = oneChatId 
+    ? oneChatId
+    : await chatRepository.getAllChatsIdActive();
+
+  if (chatsIds.length === 0) return chatsIds;
+
+  const data = [];
+
+  await Promise.all(chatsIds.map(async chat => {
+    const items = await feedRepository.getFeeds(chat.id);
+    data.push(...items);
+
+    return items;
+  }))
+
+  async function fetchData(feed) {
+    const items = await rssParser.getFeeds(feed.rss_url);
+    return { data: items, chat_id: feed.chat_id, title: feed.title, hashtags: feed.hashtag };
+  }
+  
+  const items = data.map(feed => {
+    return new Promise((resolve) => {
+      setTimeout(async () => {
+        return resolve(await fetchData(feed));
+      }, 5000)});
+    })
+  
+  const feeds = await Promise.all(items);
+  return feeds;
+}
+
+function sendMessages({ feed, data, bot }) {
+  data.forEach(async (item, index) => {
+    const defaultObject = { title: item.title, chat_id: feed.chat_id };
+    
+    if (!(await postRepository.existsPost(defaultObject))) {
+      await postRepository.addPost(defaultObject);
+      
+      setTimeout(() => {
+        bot.telegram.sendMessage(feed.chat_id, 
+          `<strong>Novo post ✅</strong><code>\n\n${item.title}</code>\n\n<a href='${item.link}'>Ler post completo ➡️</a>\n\nDe: <i>${feed.title}</i>\n\n${feed.hashtags}`, 
+          { parse_mode: 'HTML'})
+      }, index * 6000);
+    }
+  });
+}
+
+async function start(bot) {
+  const data = await getChatsData();
+
+  if (data.length === 0) return data;
+  
+  data.forEach(async feed => {
+    const { count } = await postRepository.getPostsCount({ chat_id: feed.chat_id});
+
+    feed.data.splice(0, count);
+    const items = feed.data.slice(0, 1);
+    sendMessages({ feed, data: items, bot });
+  });
+}
+
+async function main(bot) {
+  await start(bot);
+  setInterval(async () => await start(bot), 8000);
+}
+
+module.exports = main;

src/domain/Chat.js

@@ -3,18 +3,10 @@ class Chat {
     id, 
     title, 
     user_id,
-    interval_post = "1", 
-    start_posts = "8:00", 
-    end_posts = "00:00", 
-    next_posts_time = "9:00", 
     ) {
     this.id = String(id);
     this.title = title;
     this.user_id = String(user_id);
-    this.interval_post = interval_post;
-    this.start_posts = start_posts;
-    this.end_posts = end_posts;
-    this.next_posts_time = next_posts_time;
 
     Object.freeze(this);
   }
@@ -23,11 +15,7 @@ class Chat {
     return {
       id: this.id,
       title: this.title,
-      user_id: this.user_id,
-      interval_post: this.interval_post,
-      start_posts: this.start_posts,
-      end_posts: this.end_posts,
-      next_posts_time: this.next_posts_time
+      user_id: this.user_id
     }
   }
 }

src/domain/User.js

@@ -1,8 +1,7 @@
 class User {
-  constructor(user_id, username, timezone) {
+  constructor(user_id, username) {
     this.user_id = String(user_id);
     this.username = username;
-    this.timezone = timezone;
 
     Object.freeze(this);
   }
@@ -11,7 +10,6 @@ class User {
     return {
       user_id: this.user_id,
       username: this.username,
-      timezone: this.timezone,
     }
   }
 }

src/drivers/rss-parser/index.js

@@ -2,14 +2,24 @@ const Parser = require('rss-parser');
 const parser = new Parser();
 
 class RssParser {
-  async getFeedTitle(rss_url) {
-    const feed = await parser.parseURL(rss_url);
+  async getFeedTitle(rssURL) {
+    const feed = await parser.parseURL(rssURL);
     return feed.title;
   }
 
-  async rssIsValid(rss_url) {
+  async getFeeds(rssURL) {
+    try {
+      const feeds = await parser.parseURL(rssURL);
+      
+      return feeds.items;
+    } catch (err) {
+      return [];
+    }
+  }
+
+  async rssIsValid(rssURL) {
     try { 
-      await parser.parseURL(rss_url);
+      await parser.parseURL(rssURL);
       return true;
     } catch (err) {
       console.log(err)

src/helpers/bot_helpers.js

@@ -34,10 +34,16 @@ function getUserId(ctx, type) {
   }
 }
 
+function removeCommand(message, command) {
+  const newMessage = message.replace(command, '').trim();
+  return newMessage;
+}
+
 module.exports = {
   isAdmin,
   isBotAdmin,
   getChatId,
   getUserId,
   isStillMemberAndAdmin,
+  removeCommand
 };

src/helpers/features_helpers.js

@@ -43,7 +43,7 @@ async function listFeeds(feedRepository, chatID) {
 
 async function listChats(chatRepository, userID) {
   const chats = await chatRepository.getAllChatOfUser(String(userID));
-
+  
   let chatsList = '<strong>Chats 📌</strong>\n';
   if (chats && chats.length > 0) {
     chats.forEach(chat => {
@@ -60,5 +60,5 @@ module.exports = {
   removeSpacesInArray,
   removeNotHashtagsInArray,
   listFeeds,
-  listChats
+  listChats,
 }