whoosh 1.3.2 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '058f9e32f51902477b54086f0b13891fb7dd4b3a33fbcef71387d2556af05ebe'
-  data.tar.gz: abdfab2e7f266bd8d32f7357d531a7847a7087c30a508d1de955edddd1564383
+  metadata.gz: cf87392b23cb5b6a3cdc3fc27c06df4d62d7dd3b780e8695f9479cf869a0922a
+  data.tar.gz: 249803c4606645c7032deae70836954c3f1255ed5c9bf44342d07cf50afb2214
 SHA512:
-  metadata.gz: 4e2f5bc134e5e2f32eb0c21258f240646abb0d140dfc8834e113d1f52a43d9919266ede8f36ba0750791ee98641d94d6059380d82274a171a8fb610ad9b0f963
-  data.tar.gz: 3e0ce60dd666d57c6aa84804a2f5a0cf5af34bd2a215faf3d31705d4d535489de8c01a979c03144d1784a61526af8ea3e3941cacd1fee623ef9aae0bf3e7d276
+  metadata.gz: c871107c868e72d7f27596a48719dd752155d91ff114937b7448a7474e1a0ac6bdfa39dec8db23fe857e9728e3e2a0db3517013a70f820590386ebc7876b2707
+  data.tar.gz: 6ebad6447841c6587c0b3853fdb4f2666680db1e6e1eaf01a1fb1edb854af0a13ebd5e5a205b5866686346caf747a0aa49cbba30b2587afec681dc409829de23

data/README.md

@@ -13,7 +13,7 @@
   <img src="https://img.shields.io/badge/ruby-%3E%3D%203.4.0-red" alt="Ruby">
   <img src="https://img.shields.io/badge/rack-3.0-blue" alt="Rack">
   <img src="https://img.shields.io/badge/license-MIT-green" alt="License">
-  <img src="https://img.shields.io/badge/tests-540%20passing-brightgreen" alt="Tests">
+  <img src="https://img.shields.io/badge/tests-542%20passing-brightgreen" alt="Tests">
   <img src="https://img.shields.io/badge/overhead-2.5%C2%B5s-orange" alt="Performance">
 </p>
 

data/lib/whoosh/streaming/websocket.rb

@@ -1,29 +1,25 @@
 # frozen_string_literal: true
 
 require "json"
-require "digest"
-require "base64"
+require "faye/websocket"
 
 module Whoosh
   module Streaming
     class WebSocket
-      GUID = "258EAFA5-E914-47DA-95CA-5AB5DC65C3E5"
-
       attr_reader :env
 
       def initialize(env)
         @env = env
-        @io = nil
+        @ws = nil
         @closed = false
+        @on_open = nil
         @on_message = nil
         @on_close = nil
-        @on_open = nil
       end
 
       # Check if the request is a WebSocket upgrade
       def self.websocket?(env)
-        env["HTTP_UPGRADE"]&.downcase == "websocket" &&
-          env["HTTP_CONNECTION"]&.downcase&.include?("upgrade")
+        Faye::WebSocket.websocket?(env)
       end
 
       # Register callbacks
@@ -41,154 +37,58 @@ module Whoosh
 
       # Send data to the client
       def send(data)
-        return if @closed
+        return if @closed || @ws.nil?
         formatted = data.is_a?(String) ? data : JSON.generate(data)
-        write_frame(formatted)
+        @ws.send(formatted)
       end
 
-      def close
+      def close(code = nil, reason = nil)
         return if @closed
         @closed = true
-        write_close_frame
-        @io&.close rescue nil
-        @on_close&.call
+        @ws&.close(code || 1000, reason || "")
       end
 
       def closed?
         @closed
       end
 
-      # Returns a Rack response that hijacks the connection
+      # Returns a Rack response — hijacks the connection via faye-websocket
       def rack_response
         unless self.class.websocket?(@env)
           return [400, { "content-type" => "text/plain" }, ["Not a WebSocket request"]]
         end
 
-        # WebSocket handshake
-        key = @env["HTTP_SEC_WEBSOCKET_KEY"]
-        accept = Base64.strict_encode64(Digest::SHA1.digest(key + GUID))
-
-        headers = {
-          "Upgrade" => "websocket",
-          "Connection" => "Upgrade",
-          "Sec-WebSocket-Accept" => accept
-        }
+        @ws = Faye::WebSocket.new(@env)
 
-        # Use rack.hijack to take over the connection
-        if @env["rack.hijack"]
-          @env["rack.hijack"].call
-          @io = @env["rack.hijack_io"]
-
-          # Send handshake response manually
-          @io.write("HTTP/1.1 101 Switching Protocols\r\n")
-          headers.each { |k, v| @io.write("#{k}: #{v}\r\n") }
-          @io.write("\r\n")
-          @io.flush
-
-          # Notify open
+        @ws.on :open do |_event|
           @on_open&.call
-
-          # Start reading frames in a thread
-          Thread.new { read_loop }
-
-          # Return a dummy response (connection is hijacked)
-          [-1, {}, []]
-        else
-          # Fallback: return 101 and let the server handle hijack
-          [101, headers, []]
         end
-      end
-
-      # For testing — simulate without real socket
-      def trigger_message(msg)
-        @on_message&.call(msg)
-      end
-
-      def trigger_close
-        @on_close&.call
-        @closed = true
-      end
 
-      private
-
-      def read_loop
-        while !@closed && @io && !@io.closed?
-          frame = read_frame
-          break unless frame
-
-          case frame[:opcode]
-          when 0x1 # Text frame
-            @on_message&.call(frame[:data])
-          when 0x8 # Close frame
-            close
-            break
-          when 0x9 # Ping
-            write_pong(frame[:data])
-          end
+        @ws.on :message do |event|
+          @on_message&.call(event.data)
         end
-      rescue IOError, Errno::ECONNRESET, Errno::EPIPE
-        @closed = true
-        @on_close&.call
-      end
-
-      def read_frame
-        return nil unless @io && !@io.closed?
-
-        first_byte = @io.readbyte
-        fin = (first_byte & 0x80) != 0
-        opcode = first_byte & 0x0F
-
-        second_byte = @io.readbyte
-        masked = (second_byte & 0x80) != 0
-        length = second_byte & 0x7F
-
-        if length == 126
-          length = @io.read(2).unpack1("n")
-        elsif length == 127
-          length = @io.read(8).unpack1("Q>")
-        end
-
-        mask_key = masked ? @io.read(4).bytes : nil
-        payload = @io.read(length)&.bytes || []
 
-        if masked && mask_key
-          payload = payload.each_with_index.map { |b, i| b ^ mask_key[i % 4] }
+        @ws.on :close do |event|
+          @closed = true
+          @on_close&.call(event.code, event.reason)
+          @ws = nil
         end
 
-        { opcode: opcode, data: payload.pack("C*"), fin: fin }
-      rescue EOFError, IOError
-        nil
+        @ws.rack_response
       end
 
-      def write_frame(data, opcode: 0x1)
-        return unless @io && !@io.closed?
-
-        bytes = data.encode("UTF-8").bytes
-        frame = [0x80 | opcode] # FIN + opcode
-
-        if bytes.length < 126
-          frame << bytes.length
-        elsif bytes.length < 65536
-          frame << 126
-          frame += [bytes.length].pack("n").bytes
-        else
-          frame << 127
-          frame += [bytes.length].pack("Q>").bytes
-        end
-
-        frame += bytes
-        @io.write(frame.pack("C*"))
-        @io.flush
-      rescue IOError, Errno::EPIPE
-        @closed = true
+      # For testing without real socket
+      def trigger_message(msg)
+        @on_message&.call(msg)
       end
 
-      def write_close_frame
-        write_frame("", opcode: 0x8) rescue nil
+      def trigger_close(code = 1000, reason = "")
+        @on_close&.call(code, reason)
+        @closed = true
       end
 
-      def write_pong(data)
-        write_frame(data, opcode: 0xA) rescue nil
+      def trigger_open
+        @on_open&.call
       end
     end
   end

data/lib/whoosh/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Whoosh
-  VERSION = "1.3.2"
+  VERSION = "1.4.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: whoosh
 version: !ruby/object:Gem::Version
-  version: 1.3.2
+  version: 1.4.0
 platform: ruby
 authors:
 - Johannes Dwi Cahyo
@@ -107,6 +107,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.8'
+- !ruby/object:Gem::Dependency
+  name: faye-websocket
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.11'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.11'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement