Added profiles storage and Bot.broadcast

Author: ALCC01

docs/objects/Bot.rst

@@ -7,12 +7,20 @@ This object represents a Telegram bot.
     The Bot object and its attributes should remain unchanged after it has been initialized.
 
 
-.. js:class:: Bot(token)
+.. js:class:: Bot(token[, options])
 
     :param string token: A valid token for the Telegram Bot API
+    :param object options: *Optional*
     :returns: A Bot object
 
     Creates a new Bot object and fetches basic information about it (aysncronous call to Telegram's ``getMe`` method).
+    Valid ``options`` are:
+
+    .. code-block:: javascript
+
+    {
+        profiles_path: __dirname +'/profiles.json'
+    }
 
 .. _init:
 .. js:function:: init()
@@ -215,4 +223,9 @@ This object represents a Telegram bot.
     :param string action: Must be one of the following: ``typing``, ``upload_photo``, ``record_video``, ``upload_video``, ``record_audio``, ``upload_audio``, ``upload_document``, ``find_location``
     :returns: A promise that resolves to the response (according to Telegram, true on success).
 
+.. js:function:: broadcast(filter, callback)
+
+    :param string filter: Must be one of the following: ``private``, ``group``, ``supergroup``, ``channel`` or ``all``. Can also be a function that accepts one argument (a :doc:`Chat` object without functions) and returns a boolean value
+    :param function callback: A function to which is provided a single argument, a promise that resolves to a :doc:`Chat` object.
 
+    Iterates a function through all of chats stored in the profiles storage.

lib/Bot.js

@@ -1,5 +1,6 @@
 'use strict';
 const request = require('superagent-promise')(require('superagent'), Promise),
+    extend = require('util')._extend,
     Chat = require('./Chat.js'),
     User = require('./User.js'),
     Message = require('./Message.js'),
@@ -13,7 +14,7 @@ const request = require('superagent-promise')(require('superagent'), Promise),
     events = require('events'),
     endpoint = 'https://api.telegram.org';
 
-function Bot(token) {
+function Bot(token, options) {
     this.token = token;
     this.lastUpdate = -1;
     this.limit = 100;
@@ -23,6 +24,7 @@ function Bot(token) {
     this.messageHandlers = [];
     this.me = {};
     this.emitter = new events.EventEmitter();
+    extend(this, options);
 
     this.handleAPIError = (err) => {
         if (err.response) { console.log(`An error occurred while querying the Telegram API - ${err.response.text ? JSON.parse(err.response.text).error_code : ""} ${err.response.text ? JSON.parse(err.response.text).description : ""}`)}
@@ -414,12 +416,47 @@ function Bot(token) {
         this.emitter.emit('message', message);
     };
 
+    this.handleProfiles = (message) => {
+        var edited = false;
+        if (!profiles.users[message.from.id]) {
+            profiles.users[message.from.id] = message.from;
+            profiles.users[message.from.id].bot = undefined;
+            edited = true;
+        }
+        if (!profiles.chats[message.chat.id]) {
+            profiles.chats[message.chat.id] = message.chat;
+            profiles.chats[message.chat.id].bot = undefined;
+            edited = true;
+        }
+        if (edited) fs.writeFile(this.profiles_path, JSON.stringify(profiles), (err) => {console.log(err)})
+    };
+
+    this.broadcast = (filter, callback) => {
+        if (!this.profiles_path) throw "Profiles path is not defined";
+        if (profiles.chats) {
+            for (var chat in profiles.chats) {
+                if (profiles.chats.hasOwnProperty(chat)) {
+                    chat = profiles.chats[chat];
+                    if (typeof filter == 'function') {
+                        if (filter(chat)) callback(this.getChat(chat.id));
+                    }
+                    else if (chat.type == filter || filter == 'all') callback(this.getChat(chat.id))
+                }
+            }
+        }
+    };
+
     this.init = () => {
         if (!this.token) throw "Invalid token!";
         this.messageHandlers = [
             this.handleCommands,
             this.handleMessage
         ];
+        if (this.profiles_path) {
+            this.messageHandlers.push(this.handleProfiles);
+            global.profiles = {users: {}, chats:{}};
+            try {global.profiles = require(this.profiles_path)} catch (err) {}
+        }
         this.command('help', 'Display help for this bot', (args, message) => {
             var text = 'Commands:\n';
             this.commands.forEach((entry) => {