Added account level sanitizer fetch API

blueshift-labs · Sep 15, 2022 · 872be62 · 872be62
1 parent 417cdb2
commit 872be62
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/lib/apollo/clients/accounts.rb b/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,

diff --git a/lib/apollo/models/sanitizer.rb b/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