var Readable  = require('stream').Readable;

// copy-paste from https://github.com/felixge/node-mysql/blob/master/lib/protocol/sequences/Query.js
module.exports = function (command, connectionStream) {
  command.stream = function(options) {
    var stream;

    options = options || {};
    options.objectMode = true;
    stream = new Readable(options),

    stream._read = function() {
      connectionStream.resume();
    };

    this.on("result",function(row,i) {
      if (!stream.push(row)) connectionStream.pause();
      stream.emit("result",row,i);  // replicate old emitter
    });

    this.on("error",function(err) {
      stream.emit("error",err);  // Pass on any errors
    });

    this.on("end", function() {
      stream.push(null);  // pushing null, indicating EOF
    });

    this.on("fields",function(fields,i) {
      stream.emit("fields",fields,i);  // replicate old emitter
    });

    return stream;
  };
};