active_record_api-request 0.2.1 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 50362697da6b9fa6cb9f864ccf8445d699751c3ac6186ac51b397a1a7c8d96b5
-  data.tar.gz: c329f7ea4c3501dd6696c3bf35a6727dd417da3841904f47e655b3be6e9e7836
+  metadata.gz: 1bb4855eb0cc1e05c36a6cbf9f854e5e9f5d7295273c8ffdb2b3a818c010b1b8
+  data.tar.gz: 262c7a98bcf70e50279c2c19e28710780259fd3518168e6b1169b1f8616e6e93
 SHA512:
-  metadata.gz: a64b9a39dc79bf9aec4d44d3c1f702648840e1ce4af75e4b3d69388b4a0f6524e12fd107dc88cbb60d46b7c121bcce4e7aa79f33d76e47a6438828930ff14098
-  data.tar.gz: 62c66f196fed51c485e665ed89c155316dd6fefbbdb8fa7997c381a3e33f86d2bb2289ffe1bcc90540440327e379458cdce0136bd00fd44b4aa8ca32bdfdaf00
+  metadata.gz: 4863d724c7155740e6873cae1d1f0d1e51b8947810cbb31d57c99effd0c18c5fcefec930628402deaed14be4ea157d88eb7064ee89b55a87c998217b930242a6
+  data.tar.gz: bb3949e1090164cd04be0f4e06f9e26e0b01f88c065196e0fedfa7ef32cf941b0533349fd376ed9330c3c165de8033dd7eea412bf05fe332ba345099638c6554

data/lib/active_record_api/request.rb

@@ -10,6 +10,9 @@ module ActiveRecordApi
     autoload :Connection
     autoload :FaradayFollowNextLinks
     autoload :FaradayCacheServiceDown
-    autoload :MicroServices
+    autoload :AuthError
+    autoload :FaradayAuthTokenRetry
+    autoload :TokenRetriever
+    autoload :TokenCache
   end
 end

data/lib/active_record_api/request/auth_error.rb

@@ -0,0 +1,9 @@
+require 'faraday_middleware'
+#:nocov:#
+module ActiveRecordApi
+  module Request
+    class AuthError < Faraday::ClientError
+    end
+  end
+end
+#:nocov:#

data/lib/active_record_api/request/connection.rb

@@ -19,11 +19,12 @@ module ActiveRecordApi
           builder.options[:timeout] = 5
           builder.request :json
           builder.request :url_encoded
-          builder.request :retry, max: 5, interval: 0.05, interval_randomness: 0.5, backoff_factor: 2
+          builder.request :retry, max: 5, interval: 0.05, interval_randomness: 0.5, backoff_factor: 2, exceptions: [ActiveRecordApi::Request::AuthError]
           builder.use FaradayCacheServiceDown, cache_store
           builder.use :http_cache, http_cache_options
           builder.use FaradayFollowNextLinks, 5
           builder.use FaradayMiddleware::FollowRedirects, standards_compliant: true
+          builder.use FaradayAuthTokenRetry, credentials: credentials, host: host, token_path: token_path, debug: debug
           builder.response :json, content_type: /\bjson$/
           builder.response :raise_error
           builder.response :logger if debug

data/lib/active_record_api/request/credentials.rb

@@ -1,4 +1,5 @@
 require 'active_attr'
+#:nocov:#
 
 module ActiveRecordApi
   module Request
@@ -7,7 +8,7 @@ module ActiveRecordApi
       attr_writer :host, :token_path, :token, :email, :password, :headers
 
       def token
-        @token ||= config(:token) || connection.post("#{host}/#{token_path}", credentials).body['token']
+        @token ||= config(:token)
       end
 
       def host
@@ -27,9 +28,13 @@ module ActiveRecordApi
       end
 
       def headers
-        @headers ||= {
-          'Authorization' => "Bearer #{token}"
-        }
+        @headers ||= {}.tap do |headers|
+          if token.present?
+            headers['Authorization'] = "Bearer #{token}"
+          else
+            headers['InjectToken'] = 'true'
+          end
+        end
       end
 
       def credentials
@@ -41,8 +46,14 @@ module ActiveRecordApi
       end
 
       def config(name)
-        ENV["ACTIVE_RECORD_API_REQUEST_#{name.upcase}"] || Rails.application.credentials.try(:active_record_api_request).try(:[], name)
+        ENV["ACTIVE_RECORD_API_REQUEST_#{name.upcase}"] || credentials_config(name)
+      end
+
+      def credentials_config(name)
+        return unless defined? Rails
+        Rails.application.try(:credentials).try(:active_record_api_request).try(:[], name)
       end
     end
   end
 end
+#:nocov:#

data/lib/active_record_api/request/faraday_auth_token_retry.rb

@@ -0,0 +1,38 @@
+require 'faraday_middleware'
+#:nocov:#
+
+module ActiveRecordApi
+  module Request
+    class FaradayAuthTokenRetry < Faraday::Middleware
+      AUTH_KEY = 'Authorization'.freeze
+
+      attr_reader :credentials, :host, :token_path, :debug
+
+      def initialize(app, credentials: nil, host: nil, token_path: nil, debug: false)
+        super(app)
+        @credentials = credentials
+        @host = host
+        @token_path = token_path
+        @debug = debug
+      end
+
+      def call(env)
+        if env.request_headers['InjectToken'].present? && env.request_headers['InjectToken'] == 'true'
+          env.request_headers[AUTH_KEY] = "Bearer #{retrieve_token}"
+        end
+        @app.call(env)
+      rescue Faraday::ClientError => error
+        if error.response.try(:[], 'code') == 401
+          raise AuthError, error.response
+        else
+          raise error
+        end
+      end
+
+      def retrieve_token
+        TokenCache.new(debug: debug).fetch_token Proc.new { TokenRetriever.new(credentials: credentials, host: host, token_path: token_path, debug: debug).fetch_token }
+      end
+    end
+  end
+end
+#:nocov:#

data/lib/active_record_api/request/token_cache.rb

@@ -0,0 +1,65 @@
+#:nocov:#
+module ActiveRecordApi
+  module Request
+    class TokenCache
+      include ActiveAttr::Model
+      attr_accessor  :debug
+
+      def fetch_token(token_proc)
+        retrieve_token_from_redis(token_proc)
+      end
+
+      def retrieve_token_from_redis(token_proc)
+        debug_log 'looking?'
+        return token_proc.call() unless defined? AUTH_REDIS_POOL
+        debug_log 'looking for token in cache'
+        AUTH_REDIS_POOL.with do |client|
+          @redis_client = client
+          @token = nil
+          @retry = 0
+          while @token.nil? && @retry < 5
+            @retry += 1
+            @token = @redis_client.get('auth_token')
+            debug_log "retry ##{@retry}"
+            debug_log "found token: #{@token}"
+            if @token.nil?
+              if acquire_lock
+                begin
+                  debug_log "acquired lock"
+                  @token = token_proc.call()
+                  @redis_client.set 'auth_token', @token
+                ensure
+                  debug_log 'releasing lock'
+                  release_lock
+                end
+              else
+                debug_log 'waiting for token to be cached'
+                sleep 3
+              end
+            end
+          end
+        end
+      end
+
+      def debug_log(message)
+        Rails.logger.info message if debug
+      end
+
+      def acquire_lock
+        @redis_client.set 'auth_token_lock', unique_key, nx: true, ex: 60
+        current_lock_value = @redis_client.get 'auth_token_lock'
+        Rails.logger.info "#{current_lock_value} == #{unique_key}" if debug
+        current_lock_value == unique_key
+      end
+
+      def release_lock
+        @redis_client.del 'auth_token_lock'
+      end
+
+      def unique_key
+        @unique_key ||= rand.to_s
+      end
+    end
+  end
+end
+#:nocov:#

data/lib/active_record_api/request/token_retriever.rb

@@ -0,0 +1,19 @@
+#:nocov:#
+module ActiveRecordApi
+  module Request
+    class TokenRetriever < Methods
+      attr_accessor :credentials, :host, :token_path
+
+      alias_method :path, :token_path
+
+      def fetch_token
+        post(payload: credentials)
+      end
+
+      def headers
+        @headers ||= {}
+      end
+    end
+  end
+end
+#:nocov:#

data/lib/active_record_api/request/version.rb

@@ -1,5 +1,5 @@
 module ActiveRecordApi
   module Request
-    VERSION = '0.2.1'.freeze
+    VERSION = '0.3.2'.freeze
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_record_api-request
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.2
 platform: ruby
 authors:
 - Full Measure Education
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-04-19 00:00:00.000000000 Z
+date: 2019-04-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: active_attr
@@ -176,12 +176,15 @@ files:
 - README.md
 - Rakefile
 - lib/active_record_api/request.rb
+- lib/active_record_api/request/auth_error.rb
 - lib/active_record_api/request/connection.rb
 - lib/active_record_api/request/credentials.rb
+- lib/active_record_api/request/faraday_auth_token_retry.rb
 - lib/active_record_api/request/faraday_cache_service_down.rb
 - lib/active_record_api/request/faraday_follow_next_links.rb
 - lib/active_record_api/request/methods.rb
-- lib/active_record_api/request/micro_services.rb
+- lib/active_record_api/request/token_cache.rb
+- lib/active_record_api/request/token_retriever.rb
 - lib/active_record_api/request/version.rb
 homepage: https://gitlab.com/fullmeasure/public/gems/active_record_api-request
 licenses:

data/lib/active_record_api/request/micro_services.rb

@@ -1,15 +0,0 @@
-module ActiveRecordApi
-  module Request
-    module MicroServices
-      MICRO_SERVICE_PATHS.each do |path_hash|
-        klass = Class.new(Methods) do
-          define_method :path do
-            "#{path_hash[:service_path]}/#{path_hash[:model_path]}"
-          end
-        end
-        klass_name = klass.new.path.tr('/', '_').gsub(/[^a-z_]/i, '').camelcase
-        const_set klass_name, klass
-      end
-    end
-  end
-end