Merge pull request #4 from laurentiu-rocketgate/RGOPS-5517-EmbeddedField

Implemented the functionality to enable the usage of the Embedded Fields Token

Author: doug-rocketgate

RocketGate.py

@@ -26,7 +26,8 @@
 import random
 import errno
 import socket
-import ssl 
+import ssl
+from urllib.parse import urlsplit
 
 class GatewayRequest:
 
@@ -86,6 +87,8 @@ class GatewayRequest:
 
     GATEWAY_CONNECT_TIMEOUT = "gatewayConnectTimeout"
     GATEWAY_SERVER = "gatewayServer"
+    GATEWAY_SERVLET = "gatewayServlet"
+    GATEWAY_PORTNO = "gatewayPortNo"
     GATEWAY_READ_TIMEOUT = "gatewayReadTimeout"
     GENERATE_POSTBACK = "generatePostback"
 
@@ -618,8 +621,8 @@ def SendTransaction(self, serverName, request, response):
 #
 #	Gather overrides for transaction.
 #
-        urlServlet = request.Get("gatewayServlet")
-        urlPortNo = request.Get("portNo")
+        urlServlet = request.Get(GatewayRequest.GATEWAY_SERVLET)
+        urlPortNo = request.Get(GatewayRequest.GATEWAY_PORTNO)
 
 #
 #	Determine the final servlet name.
@@ -776,11 +779,29 @@ def SendTransaction(self, serverName, request, response):
 #
     def PerformTransaction(self, request, response):
 
+#
+#	If EMBEDDED_FIELDS_TOKEN is provided, send the request to the corresponding endpoint
+#
+        fullUrl = request.Get(GatewayRequest.EMBEDDED_FIELDS_TOKEN)
+        if fullUrl is not None:
+            try:
+                parsedUrl = urlsplit(fullUrl)
+                request.Set(GatewayRequest.GATEWAY_SERVER, parsedUrl.hostname)
+                request.Set(GatewayRequest.GATEWAY_SERVLET,
+                            parsedUrl.path + ("?" + parsedUrl.query if parsedUrl.query is not None else ""))
+                if parsedUrl.port is not None:
+                    request.Set(GatewayRequest.GATEWAY_PORTNO, parsedUrl.port)
+            except Exception as ex:
+                response.Set(GatewayResponse.EXCEPTION, str(ex))
+                response.Set(GatewayResponse.RESPONSE_CODE, 4)
+                response.Set(GatewayResponse.REASON_CODE, 401)
+                return 4    # Validation error: Invalid URL
+
 #
 #	If the request specifies a server name, use it.
 #	Otherwise, use the default.
 #
-        serverName = request.Get("gatewayServer")
+        serverName = request.Get(GatewayRequest.GATEWAY_SERVER)
         if serverName != None:		# Override?
             serverList = [ serverName ]		# Use this name
         else: