Merge pull request #9 from SetuHQ/support-oauth

Support for OAuth

Author: nithinkashyapn

README.md

@@ -33,7 +33,8 @@ dl = deeplink.Deeplink(
     "YOUR SCHEME ID",
     "YOUR JWT SECRET",
     "YOUR PRODUCT INSTANCE ID",
-    mode="PRODUCTION | SANDBOX" # default SANDBOX
+    mode="PRODUCTION | SANDBOX" # default SANDBOX,
+    authType= "JWT | OAUTH" # default JWT
 )
 ```
 

setu/auth.py

@@ -1,6 +1,7 @@
 import jwt
 import datetime
 import uuid
+import requests
 
 
 def generate_JWT(schemeId, secret):
@@ -24,8 +25,38 @@ def verify_JWT(token, schemeId, secret):
         raise
 
 
-def generate_setu_headers(schemeId, secret, setuProductInstanceID):
-    return {
-        "Authorization": generate_bearer_JWT(schemeId, secret),
+def generate_oAuth_token(clientId, secret, url):
+    url = url + "/v2/auth/token"
+    payload = {
+        "clientID": clientId,
+        "secret": secret,
+    }
+    headers = {
+        "Content-Type": "application/json"
+    }
+    response = requests.post(
+        url,
+        json=payload,
+        headers=headers
+    )
+
+    if response.status_code == 200:
+        data = response.json()
+        return "Bearer {}".format(data["data"]["token"])
+    else:
+        raise
+
+
+def generate_setu_headers(schemeId, secret, setuProductInstanceID, url,  authType="JWT"):
+
+    headers = {
         "X-Setu-Product-Instance-ID": setuProductInstanceID
     }
+
+    if authType == "JWT":
+        auth = generate_bearer_JWT(schemeId, secret)
+    else:
+        auth = generate_oAuth_token(schemeId, secret, url)
+
+    headers["Authorization"] = auth
+    return headers

setu/deeplink.py

@@ -15,12 +15,22 @@ class Prod:
 
 class Deeplink:
 
-    def __init__(self, schemeId, secret, productInstance, mode="SANDBOX"):
+    def __init__(self, schemeId, secret, productInstance, authType="JWT", mode="SANDBOX"):
         self.schemeId = schemeId
         self.secret = secret
         self.productInstance = productInstance
         self.url = URLS.Sandbox.url if mode != "PRODUCTION" else URLS.Prod.url
         self.mode = mode
+        self.authType = authType
+        
+        
+        if self.authType != 'JWT':
+            # Generate required headers
+            self.headers = generate_setu_headers(
+                self.schemeId, self.secret, self.productInstance, self.url, self.authType
+            )
+            # Set the url to v2
+            self.url = self.url + '/v2'
 
     # Generate UPI payment link method
     def create_payment_link(
@@ -60,14 +70,15 @@ def create_payment_link(
         if validationRules is not None:
             payload.update({"validationRules": validationRules})
 
-        # Generate required headers
-        headers = generate_setu_headers(
-            self.schemeId, self.secret, self.productInstance
-        )
+        # In case of JWT the token expires often enough to warrant creation of new tokens
+        if self.authType == 'JWT':
+            self.headers = generate_setu_headers(
+                self.schemeId, self.secret, self.productInstance, self.url, self.authType
+            )
 
         # Call API with required parameters
         response = requests.post(
-            self.url + path, json=payload, headers=headers
+            self.url + path, json=payload, headers=self.headers
         )
 
         # Handle errors
@@ -81,15 +92,16 @@ def check_payment_status(
         platformBillID
     ):
         path = "/payment-links/{}".format(platformBillID)
-
-        # Generate required headers
-        headers = generate_setu_headers(
-            self.schemeId, self.secret, self.productInstance
-        )
+        
+        # In case of JWT the token expires often enough to warrant creation of new tokens
+        if self.authType == 'JWT':
+            self.headers = generate_setu_headers(
+                self.schemeId, self.secret, self.productInstance, self.url, self.authType
+            )
 
         # Call API with required parameters
         response = requests.get(
-            self.url + path, headers=headers
+            self.url + path, headers=self.headers
         )
 
         return response.json()
@@ -115,14 +127,15 @@ def trigger_mock_payment(self, amountValue, upiID):
             "type": "UPI",
         }
 
-        # Generate required headers
-        headers = generate_setu_headers(
-            self.schemeId, self.secret, self.productInstance
-        )
+        # In case of JWT the token expires often enough to warrant creation of new tokens
+        if self.authType == 'JWT':
+            self.headers = generate_setu_headers(
+                self.schemeId, self.secret, self.productInstance, self.url, self.authType
+            )
 
         # Call API with required parameters
         response = requests.post(
-            self.url + path, json=payload, headers=headers
+            self.url + path, json=payload, headers=self.headers
         )
 
         return response