cable_x 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a00a68c2fd12c796e0c65e951579307e93edb3faa6467747dbc55a2abbd744d2
-  data.tar.gz: 7b7a074b2674f927807fed9861d482598ec03a44fbaff7197e58b96eacc99bf4
+  metadata.gz: 73434793887fad45097125d4a9b449cee07ed4f7c99ded1173b620017906d4b4
+  data.tar.gz: 87731a5da2804aacb83abebfc6c64b647e7db7d987ed6d11104ee190b2acf48f
 SHA512:
-  metadata.gz: cf0a175541d6b23fc81b03445ef5d7ebf7f556f3ed5b179defb4ba0455979573b1bc3b83c4168f43b99871f2fa46a2ff0fd33e0dc0558db6ed55dd62bda23fe8
-  data.tar.gz: 977170bef821fc56a6f08b08887f09146cbd79f112d160cb1a98808d53e436ca9f8b063eb046bc3cf2352c572e296fc634219e57312513b81a3ac140698545b5
+  metadata.gz: 43f1f5576068478a6ea0c0a02f5f42b3f822b28a937297fd46cf0fbbebb9c4b695d9dfe85ace6e924680875ae6aea941fb8cb37fa69f94f33c2e7b1c45838ed1
+  data.tar.gz: 3c571c23fc5a40288110468f971ba411356b0c5a4451aa9fb2aaabdd519211a0e8e9c12484aedac84186c195a92a4ea2a8a3f9e34a37cce5d10fd6db45ddec5a

data/README.md

@@ -5,7 +5,7 @@
 [![Build Status](https://travis-ci.com/code-vedas/rails-cable-x.svg?branch=master)](https://travis-ci.com/code-vedas/rails-cable-x)
 
 # CableX
-Standard MVC over cable for realtime applications to enjoy seamless experience over cable. Use specialized clients for Android, IOS, Angular, React, JS
+Standard MVC over cable for realtime applications to enjoy seamless experience over cable. Use specialized clients for [Angular](https://github.com/Code-Vedas/ngx-cable-x),[JS](https://github.com/Code-Vedas/cable-x-js),[React](https://github.com/Code-Vedas/cable-x-js),Android, IOS
 
     Clients for Android, IOS, Angular, React, JS coming soon.
 ## Usage

data/lib/cable_x/cable/connection.rb

@@ -1,15 +1,74 @@
 # frozen_string_literal: true
 
+require_relative './rate_limit/block_check'
+require_relative './rate_limit/rate_limit'
+require_relative './rate_limit/rate_limit_redis'
+
 module CableX
   module Cable
     ##
     # Connection class to serve entry point
     class Connection < ActionCable::Connection::Base
-      attr_accessor :device_id
+      include RateLimit::RateLimitRedis
+      include RateLimit::CheckRateLimit
+      include RateLimit::BlockCheck
+
+      class << self
+        attr_accessor :rate_limit, :redis_config
+      end
+
+      attr_accessor :device_id, :rate_limit, :redis_config, :redis
       identified_by :device_id
 
       def connect
-        self.device_id = SecureRandom.hex(20)
+        self.device_id = Digest::SHA1.hexdigest(request_str)
+        setup_rate_limit
+      rescue StandardError => _e
+        reject_unauthorized_connection
+      end
+
+      def receive(message)
+        if (blocked_unit = check_block) || (blocked_unit = do_rate_limit)
+          disconnect_client(blocked_unit)
+        elsif check_message(message)
+          super(message)
+        else
+          malformed_message_response
+        end
+      end
+
+      def close(reason, reconnect = false)
+        transmit(type: ActionCable::INTERNAL[:message_types][:disconnect],
+                 reason: reason,
+                 reconnect: reconnect)
+        websocket.close
+      end
+
+      def malformed_message_response
+        transmit(type: 'error',
+                 message: 'malformed_message')
+      end
+
+      def check_message(message)
+        JSON.parse(message) rescue false
+      end
+
+      def setup_rate_limit
+        self.rate_limit = self.class.rate_limit
+        self.redis_config = self.class.redis_config
+        if rate_limit
+          self.redis = Redis.new(url: redis_config[:url])
+          raise StandardError, 'Rate limiting: Slow down!. Try after some time' if check_block
+        end
+        true
+      end
+
+      def request_str
+        headers_to_get = %w[SERVER_PROTOCOL SERVER_SOFTWARE REQUEST_METHOD REQUEST_PATH HTTP_USER_AGENT\
+                            HTTP_CONNECTION HTTP_ORIGIN REMOTE_ADDR ORIGINAL_FULLPATH ORIGINAL_SCRIPT_NAME\
+                            action_dispatch.authorized_host]
+        headers = request.headers.to_h
+        headers_to_get.map { |key| headers[key] }.join('-').gsub(' ', '')
       end
     end
   end

data/lib/cable_x/cable/rate_limit/block_check.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module CableX
+  module Cable
+    module RateLimit
+      ##
+      # Process Rate Limit
+      module BlockCheck
+        def check_block
+          check_block_unit('second') || check_block_unit('minute') || check_block_unit('hour') if rate_limit
+        end
+
+        def check_block_unit(unit)
+          unit if redis_exists block_device_key(unit)
+        end
+
+        def block_device(unit)
+          redis_set block_device_key(unit), true, rate_limit[:cool_down].send('second') * 1000
+          unit
+        end
+
+        def disconnect_client(unit)
+          close("Rate limiting: Limit exceeded for per #{unit} Slow down!. Try after some time", false)
+        end
+      end
+    end
+  end
+end
+# "Rate limiting: Slow down!. Try after some time"

data/lib/cable_x/cable/rate_limit/rate_limit.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module CableX
+  module Cable
+    module RateLimit
+      ##
+      # Process Rate Limit
+      module CheckRateLimit
+        def do_rate_limit
+          check_rate_limit if rate_limit
+        end
+
+        def check_rate_limit
+          check_unit('second') || check_unit('minute') || check_unit('hour')
+        end
+
+        def check_unit(unit)
+          check_value = rate_limit[unit.to_s.to_sym]
+          return unless check_value
+
+          key = device_key unit
+          redis_set key, 0, 1.send(unit) * 1000 unless redis_exists(key)
+          redis_incr key
+          block_connection = (redis_get(key) || -1).to_i >= check_value
+          return unless block_connection
+
+          block_device unit
+        end
+      end
+    end
+  end
+end

data/lib/cable_x/cable/rate_limit/rate_limit_redis.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module CableX
+  module Cable
+    module RateLimit
+      ##
+      # Process Rate Limit
+      module RateLimitRedis
+        def redis_set(key, value, expire_millisecond)
+          redis.set key, value, px: expire_millisecond
+        end
+
+        def redis_incr(key)
+          redis.incr key
+        end
+
+        def redis_exists(key)
+          redis.exists key
+        end
+
+        def redis_get(key)
+          redis.get key
+        end
+
+        def device_key(schedule)
+          "#{redis_config[:channel_prefix]}-rl-#{schedule}-#{device_id}"
+        end
+
+        def block_device_key(schedule)
+          "#{redis_config[:channel_prefix]}-rl-block-#{schedule}-#{device_id}"
+        end
+      end
+    end
+  end
+end

data/lib/cable_x/engine.rb

@@ -12,12 +12,21 @@ module CableX
     isolate_namespace CableX
 
     def self.server
-      config = Rails.application.config_for(:cable_x) rescue nil
-      self.allowed_request_origins = config[:allowed_request_origins] if config
+      setup_rate_limiting
       server = CableX::Server.server
       server.config.connection_class = -> { CableX::Cable::Connection }
       server.config.allowed_request_origins = allowed_request_origins
       server
     end
+
+    def self.setup_rate_limiting
+      cable_config = Rails.application.config_for(:cable) rescue nil
+      cable_x_config = Rails.application.config_for(:cable_x) rescue nil
+      self.allowed_request_origins = cable_x_config[:allowed_request_origins]
+      return unless cable_x_config && cable_config && cable_config[:adapter] == 'redis'
+
+      CableX::Cable::Connection.redis_config = cable_config
+      CableX::Cable::Connection.rate_limit = cable_x_config[:rate_limit]
+    end
   end
 end

data/lib/cable_x/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module CableX
-  VERSION = '0.1.1'
+  VERSION = '0.1.2'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cable_x
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Nitesh Purohit
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-04-20 00:00:00.000000000 Z
+date: 2020-05-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -66,6 +66,9 @@ files:
 - lib/cable_x.rb
 - lib/cable_x/cable/channel.rb
 - lib/cable_x/cable/connection.rb
+- lib/cable_x/cable/rate_limit/block_check.rb
+- lib/cable_x/cable/rate_limit/rate_limit.rb
+- lib/cable_x/cable/rate_limit/rate_limit_redis.rb
 - lib/cable_x/channel/cable_x_channel.rb
 - lib/cable_x/channel/process/controller.rb
 - lib/cable_x/channel/process/request.rb