From 872be62a6d98cd1fb9d61ba7c4417e8de90ff5f6 Mon Sep 17 00:00:00 2001
From: vishwajitparasar-bsft <vishwajit.parasar@getblueshift.com>
Date: Thu, 15 Sep 2022 16:00:02 +0530
Subject: [PATCH] Added account level sanitizer fetch API

---
 lib/apollo/clients/accounts.rb | 4 ++++
 lib/apollo/models/sanitizer.rb | 7 ++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/lib/apollo/clients/accounts.rb b/lib/apollo/clients/accounts.rb
index b430c52..cd80dff 100644
--- 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
index 9b3d1a7..8850971 100644
--- 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