Add shard eval, broadcastEval, and fix fetchGuildCount response

src/client/Client.js

@@ -141,7 +141,16 @@ class Client extends EventEmitter {
 
     if (process.send) {
       process.on('message', message => {
-        if (message === '_guildCount') process.send(this.guilds.size);
+        if (!message) return;
+        if (message === '_guildCount') {
+          process.send({ _guildCount: this.guilds.size });
+        } else if (message._eval) {
+          try {
+            process.send({ _evalResult: eval(message._eval) });
+          } catch (err) {
+            process.send({ _evalError: err });
+          }
+        }
       });
     }
   }

src/sharding/Shard.js

@@ -54,6 +54,36 @@ class Shard {
       if (!sent) throw new Error('Failed to send message to shard\'s process.');
     });
   }
+
+  /**
+   * Evaluates a script on the shard, in the context of the Client.
+   * @param {string} script JavaScript to run on the shard
+   * @returns {Promise<*>} Result of the script
+   */
+  eval(script) {
+    return new Promise((resolve, reject) => {
+      const listener = message => {
+        if (!message) return;
+        if (message._evalResult) {
+          this.process.removeListener('message', listener);
+          resolve(message._evalResult);
+        } else if (message._evalError) {
+          this.process.removeListener('message', listener);
+          const err = new Error(message._evalError.message, message._evalError.fileName, message._evalError.lineNumber);
+          err.name = message._evalError.name;
+          err.columnNumber = message._evalError.columnNumber;
+          err.stack = message._evalError.stack;
+          reject(err);
+        }
+      };
+      this.process.on('message', listener);
+
+      this.send({ _eval: script }).catch(err => {
+        this.process.removeListener('message', listener);
+        reject(err);
+      });
+    });
+  }
 }
 
 module.exports = Shard;

src/sharding/ShardingManager.js

@@ -114,6 +114,17 @@ class ShardingManager extends EventEmitter {
     return Promise.all(promises);
   }
 
+  /**
+   * Evaluates a script on all shards, in the context of the Clients.
+   * @param {string} script JavaScript to run on each shard
+   * @returns {Promise<Array>} Results of the script
+   */
+  broadcastEval(script) {
+    const promises = [];
+    for (const shard of this.shards.values()) promises.push(shard.eval(script));
+    return Promise.all(promises);
+  }
+
   /**
    * Obtains the total guild count across all shards.
    * @param {number} [timeout=3000] Time to automatically fail after (in milliseconds)