Merge pull request #1 from blueshift-labs/feature/index-type-sanitize…

…-api

Add index type sanitizer API client code

lib/apollo/clients/accounts.rb

@@ -13,6 +13,10 @@ def initialize(account)
         @account = account
       end
 
+      def sanitizer
+        handle_request("sanitizers/account/#{@account}", :get)
+      end
+
       def run_sanitizer(type:, document:, sanitizer: nil)
         handle_request("sanitize/accounts/#{@account}/types/#{type}", :get, body: {
           document: document,

lib/apollo/clients/indices.rb

@@ -19,5 +19,12 @@ def sanitizer
     def update_sanitizer(sanitizer:, override:false)
       handle_request("sanitizers/#{@index}", :put, params: {override: override}, body: sanitizer)
     end
+
+    def run_sanitizer(type:, document:, sanitizer: nil)
+      handle_request("sanitize/indices/#{@index}/types/#{type}", :get, body: {
+        document: document,
+        sanitizer: sanitizer
+      }.delete_if { |_, v| v.nil? })
+    end
   end
 end

lib/apollo/models/sanitizer.rb

@@ -105,8 +105,13 @@ def type_mapping(key:nil, value:nil)
     end
 
     def fetch
-      if @index.nil?
+      if !@index.nil?
         Apollo.indices.for_index(@index).sanitizer
+      elsif !@account.nil?
+        Apollo.accounts.for_account(@account).sanitizer(
+           type: type,
+           document: document,
+           sanitizer: self)
       else
         Apollo.indices.for_index(:default).sanitizer
       end
@@ -129,14 +134,20 @@ def override
     end
 
     def run_sanitizer(type:, document:)
-      if !@account.nil?
+      if !@index.nil?
+        Apollo.indices.for_index(@index).run_sanitizer(
+          type: type,
+          document: document,
+          sanitizer: self
+        )
+      elsif !@account.nil?
         Apollo.accounts.for_account(@account).run_sanitizer(
           type: type,
           document: document,
           sanitizer: self
         )
       else
-        raise 'missing account'
+        raise 'missing account or index'
       end
     end
   end