rack-csrf-detector 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c0dea88d24a9de1ca64da770e521fe4be8195b76
-  data.tar.gz: 9cd14e8a69e1ff336b6cbfe0dcee0cdd83e8d895
+  metadata.gz: aa4ba5ae22cdd0b48bce43cb38428101a2aefa77
+  data.tar.gz: 0885fd8ab62bc67a1a838f63a5fe70023602d737
 SHA512:
-  metadata.gz: 0734528033885051d452cc22e3581306ce98f0855b814a6a5836e195c495916cfe7c54716cd04ae3e52817f86c386557bb29f8a7ff0b7e9763020e8e03399e4c
-  data.tar.gz: 36f3281606e24d29e4bc706e1355b48e493114986f2e689b23a01728a04201f26f776359380bbc871fe5683a0643191b6b26cc1c0297d9b5cdee9022c57d44f4
+  metadata.gz: bf54867a0be468b15e0b78ee83eb02dd8220169eaa71dae2bfa3b7c8e21d5fdcb98ba44eb3a519de24ef3a98a72ec275744c64be0750aa1d31525f1874b77e3d
+  data.tar.gz: 713f5e6db6d30e829bdc50f038525b28dd45e3f90d0974792e8ea7707d39ac710c962eb170b90e5e56b7f2dc03dbc9841607e1d1fcd15aadd1e3710148dfff99

data/lib/rack/csrf_detector/active_record_instrument.rb

@@ -0,0 +1,35 @@
+require 'active_record/connection_adapters/abstract/transaction'
+
+class Rack::CsrfDetector::ActiveRecordInstrument
+  def use!
+    if ActiveRecord::VERSION::STRING.match(/^4.2/)
+      🙉_activerecord_4_2!
+    else
+      🙉_activerecord_4_0!
+    end
+  end
+
+  private
+
+  def 🙉_activerecord_4_0!
+    ActiveRecord::ConnectionAdapters::OpenTransaction.class_eval do
+      commit_method = instance_method(:commit)
+
+      define_method :commit do
+        Rack::CsrfDetector.more_bad!
+        commit_method.bind(self).call
+      end
+    end
+  end
+
+  def 🙉_activerecord_4_2!
+    ActiveRecord::ConnectionAdapters::Transaction.class_eval do
+      commit_method = instance_method(:commit)
+
+      define_method :commit do
+        Rack::CsrfDetector.more_bad!
+        commit_method.bind(self).call
+      end
+    end
+  end
+end

data/lib/rack/csrf_detector/sidekiq_instrument.rb

@@ -0,0 +1,18 @@
+require 'sidekiq'
+
+class Rack::CsrfDetector::SidekiqInstrument
+  def use!
+    Sidekiq.configure_client do |config|
+      config.client_middleware do |chain|
+        chain.add Middleware
+      end
+    end
+  end
+
+  class Middleware
+    def call(worker_class, msg, queue, redis_pool)
+      Rack::CsrfDetector.more_bad!
+      yield
+    end
+  end
+end

data/lib/rack/csrf_detector.rb

@@ -2,9 +2,16 @@ module Rack
   class CsrfDetector
     @@bad_count = 0
 
-    def initialize(app)
+    def initialize(app, opts={}, &block)
       @app = app
-      🙉_activerecord!
+
+      if block_given?
+        if block.arity == 1
+          block.call(self)
+        else
+          instance_eval(&block)
+        end
+      end
     end
 
     def call(env)
@@ -24,38 +31,8 @@ module Rack
 
     private
 
-    def 🙉_activerecord!
-      if ActiveRecord::VERSION::STRING.match(/^4.2/)
-        🙉_activerecord_4_2!
-      else
-        🙉_activerecord_4_0!
-      end
-    end
-
-    def 🙉_activerecord_4_0!
-      require 'active_record/connection_adapters/abstract/transaction'
-
-      ActiveRecord::ConnectionAdapters::OpenTransaction.class_eval do
-        commit_method = instance_method(:commit)
-
-        define_method :commit do
-          Rack::CsrfDetector.more_bad!
-          commit_method.bind(self).call
-        end
-      end
-    end
-
-    def 🙉_activerecord_4_2!
-      require 'active_record/connection_adapters/abstract/transaction'
-
-      ActiveRecord::ConnectionAdapters::Transaction.class_eval do
-        commit_method = instance_method(:commit)
-
-        define_method :commit do
-          Rack::CsrfDetector.more_bad!
-          commit_method.bind(self).call
-        end
-      end
+    def use(klass)
+      klass.new.use!
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rack-csrf-detector
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Tommy Murphy
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-07-04 00:00:00.000000000 Z
+date: 2015-07-06 00:00:00.000000000 Z
 dependencies: []
 description: |-
   This middleware helps you identify when a GET request results in an
@@ -19,6 +19,8 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/rack/csrf_detector.rb
+- lib/rack/csrf_detector/active_record_instrument.rb
+- lib/rack/csrf_detector/sidekiq_instrument.rb
 homepage: https://github.com/tam7t/rack-csrf-detector
 licenses:
 - MIT