tshield 0.6.0.0 → 0.6.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 19c70680e289c5c8b514feeca2fc6a276fc7edc8
-  data.tar.gz: 68c61ad023d98d721bf2c88570e1f3170823b968
+  metadata.gz: 566af7ee3e491ed0f05f870c08e7581d90a210f9
+  data.tar.gz: 7f41265085e90389f8832b0132c158672e9b6751
 SHA512:
-  metadata.gz: 257777b038cf582aaf4d62e354a11cf976e0bdf55f3f018b3f8c60e1c18d7be81331570bd70a26e0ce461b9c5082f7840371fd7b5bf83187cdbc7865a2147ef1
-  data.tar.gz: 12df75edf9427185e53f67b12d01fab27b015735d78b62a0cf8489b1108f6e8d3af009f25035272d36502c000930b9b78780b59705983e0043da72b13d21fd18
+  metadata.gz: 3132976dff0b4c21318f2961383779d0f918a5cb1a766ed9be706d4abd3cd602731ef0563f7b54a536d78f82ff0105b5835b8c8cd42f8378010b055f78b235ea
+  data.tar.gz: b1a2bcc07892b7624971f7e51c7a994307d0bdf6199a041297d503857d2e1c72be90db604ec486fb893745e245048eb92e8e38dc08e6e58552d82a3a03ee1f8f

data/lib/tshield/after_filter.rb

@@ -0,0 +1,12 @@
+module TShield
+
+  # Example:
+  # def filter(response)
+  #   response
+  # end
+  class AfterFilter
+    def filter(response)
+      raise 'should implement method filter and returns response'
+    end
+  end
+end

data/lib/tshield/before_filter.rb

@@ -0,0 +1,12 @@
+module TShield
+
+  # Example:
+  # def filter(method, url, options)
+  #   [method, url, options]
+  # end
+  class BeforeFilter
+    def filter(method, url, options)
+      raise 'should implement method filter and returns method, url, options'
+    end
+  end
+end

data/lib/tshield/configuration.rb

@@ -1,5 +1,8 @@
 require 'yaml'
 
+require 'tshield/after_filter'
+require 'tshield/before_filter'
+
 module TShield
   class Configuration
 
@@ -12,6 +15,15 @@ module TShield
       attributes.each do |key, value|
         send("#{key}=", value)
       end
+
+      if File.exists?('filters')
+        Dir.entries('filters').each do |entry|
+          next if entry =~ /^\.\.?$/
+          puts "loading filter #{entry}"
+          entry.gsub!('.rb', '')
+          require File.join('.', 'filters', entry)
+        end
+      end
     end
 
     def self.singleton
@@ -20,7 +32,7 @@ module TShield
 
     def get_domain_for(path)
       domains.each do |url, config|
-        config['paths'].each { |p| return url if path =~ Regexp.new(p)  }
+        config['paths'].each { |p| return url if path =~ Regexp.new(p) }
       end
       nil
     end
@@ -33,6 +45,26 @@ module TShield
       domains[domain]['name'] || domain.gsub(/.*:\/\//, '')
     end
 
+    def get_before_filters(domain)
+      get_filters(domain)
+        .select { |k| k.ancestors.include?(TShield::BeforeFilter) }
+    end
+
+    def get_after_filters(domain)
+      get_filters(domain)
+        .select { |k| k.ancestors.include?(TShield::AfterFilter) }
+    end
+
+    def cache_request?(domain)
+      return true unless domains[domain].include?('cache_request')
+      domains[domain]['cache_request']
+    end
+
+    def get_filters(domain)
+      (domains[domain]['filters'] || [])
+        .collect { |f| Class.const_get(f) }
+    end
+
     def get_excluded_headers(domain)
       domains[domain]['excluded_headers'] || []
     end

data/lib/tshield/request.rb

@@ -20,7 +20,7 @@ module TShield
       @options = options 
       @configuration = TShield::Configuration.singleton
       @options[:timeout] =  @configuration.request['timeout']
-      @options[:verify] =  @configuration.request['verify_ssl'] == 'true'
+      @options[:verify] =  @configuration.request['verify_ssl']
       request
     end
 
@@ -35,8 +35,19 @@ module TShield
         @response = get_current_response  
         @response.original = false
       else
-        raw = HTTParty.send("#{method}", @url, @options)
+        @method = method
+        @configuration.get_before_filters(domain).each do |filter|
+          @method, @url, @options = filter.new.filter(@method, @url, @options)
+        end
+
+        raw = HTTParty.send("#{@method}", @url, @options)
+
+        @configuration.get_after_filters(domain).each do |filter|
+          raw = filter.new.filter(raw)
+        end
+
         @response = save(raw)
+
         @response.original = true
       end
       current_session[:counter].add(@path, method) if current_session
@@ -87,7 +98,7 @@ module TShield
     end
 
     def exists
-      file_exists
+      file_exists && @configuration.cache_request?(domain)
     end
 
     def get_current_response

data/lib/tshield/version.rb

@@ -2,7 +2,7 @@ module TShield
   class Version
     MAJOR = 0
     MINOR = 6
-    PATCH = 0
+    PATCH = 1
     PRE = 0
 
     class << self

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tshield
 version: !ruby/object:Gem::Version
-  version: 0.6.0.0
+  version: 0.6.1.0
 platform: ruby
 authors:
 - Diego Rubin
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-07-10 00:00:00.000000000 Z
+date: 2018-07-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
@@ -182,6 +182,7 @@ files:
 - Rakefile
 - bin/tshield
 - lib/tshield.rb
+- lib/tshield/after_filter.rb
 - lib/tshield/assets/favicon.ico
 - lib/tshield/assets/javascripts/application.js
 - lib/tshield/assets/javascripts/bootstrap.min.js
@@ -189,6 +190,7 @@ files:
 - lib/tshield/assets/stylesheets/application.css
 - lib/tshield/assets/stylesheets/bootstrap-theme.min.css
 - lib/tshield/assets/stylesheets/bootstrap.min.css
+- lib/tshield/before_filter.rb
 - lib/tshield/configuration.rb
 - lib/tshield/controller.rb
 - lib/tshield/controllers/admin/requests.rb