From 2a84f04bedcf98060f2808b93caee7a26da58605 Mon Sep 17 00:00:00 2001
From: Christophe Robillard <christophe.robillard@beta.gouv.fr>
Date: Fri, 26 May 2023 12:16:44 +0200
Subject: [PATCH] add sentry instrumentation for entreprise adapters

---
 .../attestation_sociale_adapter.rb            | 19 +++++++++++++------
 .../api_entreprise/extrait_kbis_adapter.rb    | 13 +++++++++----
 app/lib/api_entreprise/tva_adapter.rb         | 14 ++++++++++----
 3 files changed, 32 insertions(+), 14 deletions(-)

diff --git a/app/lib/api_entreprise/attestation_sociale_adapter.rb b/app/lib/api_entreprise/attestation_sociale_adapter.rb
index bae32627b..b9a5c47a6 100644
--- a/app/lib/api_entreprise/attestation_sociale_adapter.rb
+++ b/app/lib/api_entreprise/attestation_sociale_adapter.rb
@@ -14,12 +14,19 @@ class APIEntreprise::AttestationSocialeAdapter < APIEntreprise::Adapter
   end
 
   def process_params
-    if data_source[:data] && data_source[:data][:document_url].present?
-      {
-        entreprise_attestation_sociale_url: data_source[:data][:document_url]
-      }
-    else
-      {}
+    data = data_source[:data]
+
+    Sentry.with_scope do |scope|
+      scope.set_tags(siret: @siret)
+      scope.set_extras(source: data)
+
+      if data && data[:document_url].present?
+        {
+          entreprise_attestation_sociale_url: data[:document_url]
+        }
+      else
+        {}
+      end
     end
   end
 end
diff --git a/app/lib/api_entreprise/extrait_kbis_adapter.rb b/app/lib/api_entreprise/extrait_kbis_adapter.rb
index 06d3f2c7d..afa4559b1 100644
--- a/app/lib/api_entreprise/extrait_kbis_adapter.rb
+++ b/app/lib/api_entreprise/extrait_kbis_adapter.rb
@@ -11,10 +11,15 @@ class APIEntreprise::ExtraitKbisAdapter < APIEntreprise::Adapter
   def process_params
     result = {}
     data = data_source[:data]
-    if data
-      result[:entreprise_capital_social] = data[:capital][:montant] if data[:capital]
-      result[:entreprise_nom_commercial] = data[:nom_commercial]
+
+    Sentry.with_scope do |scope|
+      scope.set_tags(siret: @siret)
+      scope.set_extras(source: data)
+      if data
+        result[:entreprise_capital_social] = data[:capital][:montant] if data[:capital]
+        result[:entreprise_nom_commercial] = data[:nom_commercial]
+      end
+      result
     end
-    result
   end
 end
diff --git a/app/lib/api_entreprise/tva_adapter.rb b/app/lib/api_entreprise/tva_adapter.rb
index eb48f1790..cfe6dfc56 100644
--- a/app/lib/api_entreprise/tva_adapter.rb
+++ b/app/lib/api_entreprise/tva_adapter.rb
@@ -9,10 +9,16 @@ class APIEntreprise::TvaAdapter < APIEntreprise::Adapter
   end
 
   def process_params
-    result = {}
-    if data_source[:data]
-      result[:entreprise_numero_tva_intracommunautaire] = data_source[:data][:tva_number]
+    Sentry.with_scope do |scope|
+      data = data_source[:data]
+      scope.set_tags(siret: @siret)
+      scope.set_extras(source: data)
+
+      result = {}
+      if data
+        result[:entreprise_numero_tva_intracommunautaire] = data[:tva_number]
+      end
+      result
     end
-    result
   end
 end