From 7c427aa890d97090be63639f5a2b9dd0690ccee3 Mon Sep 17 00:00:00 2001
From: Saqib Ansari <nextchamp.saqib@gmail.com>
Date: Fri, 25 Sep 2020 23:29:45 +0530
Subject: [PATCH] feat: decrypt json data with SEK

---
 .../e_invoice_settings/e_invoice_settings.js  | 15 +++++++++++-
 .../e_invoice_settings/e_invoice_settings.py  | 24 ++++++++++++++++---
 2 files changed, 35 insertions(+), 4 deletions(-)

diff --git a/erpnext/erpnext_integrations/doctype/e_invoice_settings/e_invoice_settings.js b/erpnext/erpnext_integrations/doctype/e_invoice_settings/e_invoice_settings.js
index 9fc67ce1b98..9c3d0807355 100644
--- a/erpnext/erpnext_integrations/doctype/e_invoice_settings/e_invoice_settings.js
+++ b/erpnext/erpnext_integrations/doctype/e_invoice_settings/e_invoice_settings.js
@@ -4,6 +4,18 @@
 frappe.ui.form.on('E Invoice Settings', {
 	refresh: function(frm) {
 		frm.trigger("show_fetch_token_btn");
+		frm.add_custom_button(__("Get GSTIN Details"), 
+			() => {
+				frm.call({
+					doc: frm.doc,
+					method: 'get_gstin_details',
+					args: {
+						'gstin': '27AAACW8099E1ZX'
+					},
+					freeze: true,
+					callback: (res) => console.log(res)
+				});
+			});
 	},
 
 	show_fetch_token_btn(frm) {
@@ -16,7 +28,8 @@ frappe.ui.form.on('E Invoice Settings', {
 					frm.call({
 						doc: frm.doc,
 						method: 'make_authentication_request',
-						freeze: true
+						freeze: true,
+						callback: () => frm.refresh()
 					});
 				}
 			);
diff --git a/erpnext/erpnext_integrations/doctype/e_invoice_settings/e_invoice_settings.py b/erpnext/erpnext_integrations/doctype/e_invoice_settings/e_invoice_settings.py
index 866d75ac8c3..9a9a1b22ff6 100644
--- a/erpnext/erpnext_integrations/doctype/e_invoice_settings/e_invoice_settings.py
+++ b/erpnext/erpnext_integrations/doctype/e_invoice_settings/e_invoice_settings.py
@@ -14,7 +14,7 @@ from Crypto.Cipher import PKCS1_v1_5, AES
 from Crypto.Util.Padding import pad, unpad
 from frappe.utils.data import get_datetime
 from frappe.model.document import Document
-from frappe.integrations.utils import make_post_request
+from frappe.integrations.utils import make_post_request, make_get_request
 
 class EInvoiceSettings(Document):
 	def validate(self):
@@ -40,15 +40,18 @@ class EInvoiceSettings(Document):
 		return b64_enc_msg.decode()
 	
 	def aes_decrypt(self, enc_msg, key):
+		encode_as_b64 = True
 		if not (isinstance(key, bytes) or isinstance(key, bytearray)):
 			key = base64.b64decode(key)
+			encode_as_b64 = False
 
 		cipher = AES.new(key, AES.MODE_ECB)
 		b64_enc_msg = base64.b64decode(enc_msg)
 		msg_bytes = cipher.decrypt(b64_enc_msg)
 		msg_bytes = unpad(msg_bytes, AES.block_size) # due to ECB/PKCS5Padding
-		b64_msg_bytes = base64.b64encode(msg_bytes)
-		return b64_msg_bytes.decode()
+		if encode_as_b64:
+			msg_bytes = base64.b64encode(msg_bytes)
+		return msg_bytes.decode()
 
 	def make_authentication_request(self):
 		endpoint = 'https://einv-apisandbox.nic.in/eivital/v1.03/auth'
@@ -80,6 +83,21 @@ class EInvoiceSettings(Document):
 		self.token_expiry = get_datetime(token_expiry)
 		self.sek = sek
 		self.save()
+	
+	def get_gstin_details(self, gstin):
+		endpoint = 'https://einv-apisandbox.nic.in/eivital/v1.03/Master/gstin/{gstin}'.format(gstin=gstin)
+		headers = { 'content-type': 'application/json' }
+		headers.update(dict(client_id=self.client_id, client_secret=self.client_secret, user_name=self.username))
+		headers.update(dict(Gstin=self.gstin, AuthToken=self.auth_token))
+
+		res = make_get_request(endpoint, headers=headers)
+		self.handle_err_response(res)
+
+		enc_json = res.get('Data')
+		json_str = self.aes_decrypt(enc_json, self.sek)
+		data = json.loads(json_str)
+
+		return data
 
 	def handle_err_response(self, response):
 		if response.get('Status') == 0: