fix QUIT processing with flood protection

Closes #138.
martynsmith · May 29, 2014 · 574f2b0 · 574f2b0
1 parent f8e6b65
commit 574f2b0
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/lib/irc.js b/lib/irc.js
@@ -699,7 +699,14 @@ Client.prototype.disconnect = function ( message, callback ) { // {{{
     message = message || "node-irc says goodbye";
     var self = this;
     if ( self.conn.readyState == 'open' ) {
-        self.send( "QUIT", message );
+        var sendFunction;
+        if (self.opt.floodProtection) {
+            sendFunction = self._sendImmediate;
+            self._clearCmdQueue();
+        } else {
+            sendFunction = self.send;
+        }
+        sendFunction.call(self, "QUIT", message);
     }
     self.conn.requestedDisconnect = true;
     if (typeof(callback) === 'function') {
@@ -737,6 +744,14 @@ Client.prototype.activateFloodProtection = function(interval) { // {{{
         cmdQueue.push(arguments);
     };
 
+    this._sendImmediate = function () {
+        origSend.apply(self, arguments);
+    }
+
+    this._clearCmdQueue = function () {
+        cmdQueue = [];
+    }
+
     dequeue = function() {
         var args = cmdQueue.shift();
         if (args) {
@@ -750,6 +765,7 @@ Client.prototype.activateFloodProtection = function(interval) { // {{{
 
 
 }; // }}}
+
 Client.prototype.join = function(channel, callback) { // {{{
     var channelName =  channel.split(' ')[0];
     this.once('join' + channelName, function () {