From 7946d17e6ca070fa4cab079191a00214e9c46769 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Th=C3=A9o=20DRAPPIER?= <theo.drappier@horanet.com>
Date: Tue, 7 Dec 2021 11:34:05 +0100
Subject: [PATCH] [Imp #6] 3DSv2 Integration

---
 .../models/inherited_payment_acquirer.py      | 44 ++++++++++++++++++-
 requirements.txt                              |  1 +
 2 files changed, 44 insertions(+), 1 deletion(-)

diff --git a/payment_paybox/models/inherited_payment_acquirer.py b/payment_paybox/models/inherited_payment_acquirer.py
index 1d55f6b..208eab6 100644
--- a/payment_paybox/models/inherited_payment_acquirer.py
+++ b/payment_paybox/models/inherited_payment_acquirer.py
@@ -7,6 +7,7 @@
 import logging
 import hmac
 import binascii
+import pycountry
 
 _logger = logging.getLogger(__name__)
 
@@ -89,6 +90,45 @@ def paybox_get_authentication_key(self):
         else:
             return self.paybox_test_authentication_key
 
+    @api.model
+    def paybox_generate_shopping_cart_xml(self, reference):
+        tx = self.env['payment.transaction'].search([('reference', '=', reference)])
+        qty = 1
+        if tx.sale_order_id:
+            qty = tx.sale_order_id.cart_quantity
+        elif tx.account_invoice_id:
+            qty = int(sum(tx.account_invoice_id.mapped('invoice_line_ids.quantity')))
+        return '<?xml version="1.0" encoding="utf-8"?>' \
+               '<shoppingcart>' \
+               '<total>' \
+               '<totalQuantity>%d</totalQuantity>' \
+               '</total>' \
+               '</shoppingcart>' % qty
+
+    @api.model
+    def paybox_generate_billing_xml(self, partner):
+        code_iso_num = pycountry.countries.get(alpha_2=partner.country_id.code)
+        return '<?xml version="1.0" encoding="utf-8"?>' \
+               '<Billing>' \
+               '<Address>' \
+               '<FirstName>%s</FirstName>' \
+               '<LastName>%s</LastName>' \
+               '<Address1>%s</Address1>' \
+               '<Address2>%s</Address2>' \
+               '<ZipCode>%s</ZipCode>' \
+               '<City>%s</City>' \
+               '<CountryCode>%s</CountryCode>' \
+               '</Address>' \
+               '</Billing>' % \
+               (
+                   partner.firstname,
+                   partner.lastname,
+                   partner.street,
+                   partner.street2,
+                   partner.zip_id.name,
+                   partner.city_id.name,
+                   code_iso_num.numeric
+               )
 
     @api.multi
     def paybox_form_generate_values(self, values):
@@ -123,7 +163,9 @@ def paybox_form_generate_values(self, values):
             ('PBX_REFUSE', urllib.parse.urljoin(base_url, '/payment/paybox/dpn?return_url=%s' % values.get('return_url'))),
             ('PBX_ANNULE', urllib.parse.urljoin(base_url, '/payment/paybox/dpn?return_url=%s' % values.get('return_url'))),
             ('PBX_ATTENTE', urllib.parse.urljoin(base_url, '/payment/paybox/dpn?return_url=%s' % values.get('return_url'))),
-            ('PBX_REPONDRE_A', urllib.parse.urljoin(base_url, '/payment/paybox/ipn'))
+            ('PBX_REPONDRE_A', urllib.parse.urljoin(base_url, '/payment/paybox/ipn')),
+            ('PBX_SHOPPINGCART', self.paybox_generate_shopping_cart_xml(values.get('reference'))),
+            ('PBX_BILLING', self.paybox_generate_billing_xml(values.get('partner')))
         ]
 
         signature = self.paybox_generate_message_hmac(vals)
diff --git a/requirements.txt b/requirements.txt
index acdfd20..a770b24 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1,2 @@
 pycryptodome
+pycountry
\ No newline at end of file