rakie 0.0.6 → 0.0.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6aa5c872ec3bdf47d4f33f08f12f6ba6c3521553c9411c284d01b5efc6a63431
-  data.tar.gz: 0ed904078c4aa1bdbf6d9564be3b381a5ae6ca1c2831256c2fadff5cd449c81d
+  metadata.gz: d898a3567f82f46d55e31f84ee421ef63ad4818dea4897f44b226700136a8dc9
+  data.tar.gz: 2adb4b03156ad6ea22d18a5c6e0b42a0044bf4d76949f6156f98f5ea129cfe4a
 SHA512:
-  metadata.gz: 80823efc97c0d28c64d0871149118a5df0bebd19623a7b7088b9831218e8e0cee8e8a32901a010af3ffdb5e24ce86d5892dac8f5638541427e151e60209738c2
-  data.tar.gz: 9c747c282c958a45b307eba5e9bf451f513c8337106a4c1aaf59b8b6b4809ce915607e91ca544af48149b9faa7418041fea03adf6fca06580014bd1c51765308
+  metadata.gz: ce2fdecade1fcc90be23f2a3df87512a0889662307c07ae32a3495b861de5fbe20c91f971f3924c6a5f05f307b978c71cb9f4781285d32b0d0f189565245df58
+  data.tar.gz: f7c254d194e79de3c9cd7befbacc7cf8409e258b68fce01b316af0ba46458450b0b6dde47b41f83ab93cc6783e17d0b84f400ad84bd22f0c591ce35f5f4a02f6

data/lib/rakie/channel.rb

@@ -104,15 +104,20 @@ module Rakie
     end
 
     def on_detach(io)
+      if io.closed?
+        return
+      end
+
       begin
         io.close
 
-        if @delegate
-          @delegate.on_close(self)
-        end
-
       rescue
         Log.debug("Channel is already closed")
+        return
+      end
+
+      if @delegate
+        @delegate.on_close(self)
       end
 
       Log.debug("Channel close ok")

data/lib/rakie/http_server.rb

@@ -13,10 +13,13 @@ module Rakie
       end
     end
 
-    attr_reader :channel, :opt
+    attr_reader :channel, :opt, :host, :port
 
-    def initialize(delegate=nil)
-      @channel = TCPServerChannel.new('127.0.0.1', 10086, self)
+    def initialize(host: '127.0.0.1', port: 10086, delegate: nil)
+      @host = host
+      @port = port
+
+      @channel = TCPServerChannel.new(host, port, self)
 
       # @type [Hash{Channel=>Session}]
       @sessions = {}
@@ -123,5 +126,9 @@ module Rakie
         end
       end
     end
+
+    def on_close(channel)
+      @sessions.delete(channel)
+    end
   end
 end

data/lib/rakie/version.rb

@@ -1,5 +1,5 @@
 module Rakie
-  VERSION = [0, 0, 6]
+  VERSION = [0, 0, 11]
 
   def self.version_s
     VERSION.join('.')

data/lib/rakie/websocket.rb

@@ -21,58 +21,78 @@ module Rakie
       @client_side = true
     end
 
+    # @param [String] data
     def on_recv(channel, data)
-      # Log.debug("Rakie::HTTPServer recv: #{data}")
+      Log.debug("Rakie::Websocket recv: #{data}")
 
-      # @type [WebsocketMessage] request
-      message = @recv_message
+      total_parsed = 0
 
-      if message.parse_status == ParseStatus::COMPLETE
-        message = WebsocketMessage.new
-        @recv_message = message
-      end
+      while data.length > 0
+        # @type [WebsocketMessage] request
+        message = @recv_message
 
-      len = message.parse(data)
+        if message.parse_status == ParseStatus::COMPLETE
+          message = WebsocketMessage.new
+          @recv_message = message
+        end
 
-      Log.debug("Rakie::Websocket receive message: #{message.to_s}")
+        len = message.parse(data)
+        total_parsed += len
 
-      if message.parse_status == ParseStatus::COMPLETE
-        response = WebsocketMessage.new
+        Log.debug("Rakie::Websocket receive message: #{message.to_s} parse with #{len}")
 
-        if message.op_code == WebsocketMessage::OP_PING
-          response.fin = true
-          response.op_code = WebsocketMessage::OP_PONG
-          response.payload = "pong"
+        if message.parse_status == ParseStatus::COMPLETE
+          response = WebsocketMessage.new
 
-        elsif message.op_code == WebsocketMessage::OP_PONG
-          response.fin = true
-          response.op_code = WebsocketMessage::OP_PING
-          response.payload = "ping"
+          if message.op_code == WebsocketMessage::OP_PING
+            response.fin = true
+            response.op_code = WebsocketMessage::OP_PONG
+            response.payload = "Rakie::Websocket: op pong"
 
-        elsif @delegate
-          @delegate.on_message(self, message.payload)
-          return len
+          elsif message.op_code == WebsocketMessage::OP_PONG
+            response.fin = true
+            response.op_code = WebsocketMessage::OP_PING
+            response.payload = "Rakie::Websocket: op ping"
 
-        else
-          response.fin = true
-          response.op_code = WebsocketMessage::OP_TEXT
-          response.payload = "Rakie!"
-        end
-        
-        response_data = response.to_s
+          elsif message.op_code == WebsocketMessage::OP_CLOSE
+            channel.close
+
+            Log.debug("Rakie::Websocket: op close")
+            return 0
+
+          elsif @delegate
+            @delegate.on_message(self, message.payload)
+
+          else
+            response.fin = true
+            response.op_code = WebsocketMessage::OP_TEXT
+            response.payload = "Rakie!"
+          end
+          
+          response_data = response.to_s
 
-        Log.debug("Rakie::Websocket response: #{response_data}")
+          Log.debug("Rakie::Websocket response: #{response_data}")
 
-        channel.write(response_data) # Response data
+          channel.write(response_data) # Response data
 
-      elsif message.parse_status == ParseStatus::ERROR
-        channel.close
+        elsif message.parse_status == ParseStatus::CONTINUE
+          break
+          
+        elsif message.parse_status == ParseStatus::ERROR
+          channel.close
+
+          Log.debug("Rakie::Websocket: Illegal message")
+          return 0
+        end
+
+        if len >= data.length
+          break
+        end
 
-        Log.debug("Rakie::Websocket: Illegal message")
-        return len
+        data = data[len .. -1]
       end
 
-      return len
+      return total_parsed
     end
 
     def on_send(channel)
@@ -109,6 +129,7 @@ module Rakie
       end
 
       send_message = ws_message.to_s
+      @send_messages << ws_message
 
       Log.debug("Rakie::Websocket send: #{send_message}")
 
@@ -134,4 +155,4 @@ module Rakie
       @channel.write(send_message) # Response data
     end
   end
-end
+end

data/lib/rakie/websocket_server.rb

@@ -1,13 +1,16 @@
 module Rakie
   class WebsocketServer < Websocket
     # @param [Rakie::HttpServer] http_server
-    def initialize(delegate=nil, http_server=nil)
+    def initialize(host: '127.0.0.1', port: 10086, delegate: nil, http_server: nil)
       @delegate = delegate
 
       if http_server == nil
-        http_server = HttpServer.new('127.0.0.1', 10086, self)
+        http_server = HttpServer.new(host: host, port: port)
       end
 
+      @host = http_server.host
+      @port = http_server.port
+
       http_server.opt[:websocket_delegate] = self
       @channel = http_server.channel
 
@@ -23,7 +26,6 @@ module Rakie
 
       if @delegate
         @delegate.on_connect(ws_client)
-        return
       end
 
       @clients[channel] = ws_client
@@ -49,8 +51,10 @@ module Rakie
       client = @clients[channel]
 
       if client
-        client.on_recv(channel, data)
+        return client.on_recv(channel, data)
       end
+
+      return data.length
     end
 
     def on_send(channel)
@@ -67,6 +71,8 @@ module Rakie
       if client
         client.on_close(channel)
       end
+
+      @clients.delete(channel)
     end
 
     def send(message, is_binary=false); end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rakie
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.11
 platform: ruby
 authors:
 - Jakit Liang