qws

os = require 'options-stream' zlib = require 'zlib' {EventEmitter} = require 'events' {unpack, inflate, Frame} = require './frame' class Message extends EventEmitter constructor : (@socket, options)-> # http can only be use one time @options = os { deflate : true min_deflate_length : 32 close_timeout : 100 }, options frame = null { socket } = @ @inflate = zlib.createInflateRaw chunkSize : 128 * 1024 socket.on 'data', (chunk)=> # first chunk # if frame is null while chunk and chunk.length [frame, chunk] = unpack chunk, frame # 2+ chunks # else # [frame, buf] = unpack chunk, frame # if done if frame.done # decompress inflate frame, @inflate, (err, f) => return @emit 'error', err if err # emit event @onFrame f # reset frame frame = null socket.on 'error', (err)=> @emit 'error', err socket.on 'close', => @emit 'close' write : (data, opcode, mask, cb) -> # mask if typeof mask is 'function' cb = mask mask = null # opcode switch typeof opcode when 'function' cb = opcode opcode = null when 'boolean' mask = opcode opcode = null # text frame as defalut opcode ?= 'text' # no mask as defalut mask ?= false # data is frame # if data instanceof Frame # frame = data # else frame = new Frame data : data opcode : opcode fin : true mask : mask minDeflateLength : @options.min_deflate_length # pack frame.pack @options.deflate, (err, bin) => if err cb err if cb return # write @socket.write bin cb null if cb return ping : (cb)-> @write '', 'ping', false, cb return pong : (cb)-> @write '', 'pong', false, cb return continue : (cb)-> @write '', 'continue', false, cb return writeRaw : (bin) -> @socket.write bin end : (data, opcode, mask)-> if data? @write data, opcode, mask, => @close() else @close() return close : -> closed = false @write '', 'close', false, => @socket.on 'close', (err)=> clearTimeout timer closed = true @emit 'closed', err timer = setTimeout => return if closed @socket.end() , @options.close_timeout onFrame : (frame) -> switch frame.opcode when 'text' then @emit 'message', frame.data.toString(), @ when 'binary' then @emit 'message', frame.data, @ when 'ping' then @emit 'ping' when 'pong' then @emit 'pong' when 'close' @socket.end() @emit 'close' when 'continue' then @emit 'continue' # else @emit 'control', frame.opcode return reset : (options) -> @options = os @options, options @__QWS_CB = options.cb errorHandle : (msg) -> @write msg if msg @close() exports.Message = Message