feat: allow auth everywhere

Author: robertandreimoldoveanu

src/uipath_sdk/_cli/_auth/_auth_server.py

@@ -24,7 +24,7 @@ def __init__(self, token_data):
         super().__init__("Token received successfully")
 
 
-def make_request_handler_class(state, code_verifier, token_callback):
+def make_request_handler_class(state, code_verifier, token_callback, domain):
     class SimpleHTTPSRequestHandler(http.server.SimpleHTTPRequestHandler):
         """Simple HTTPS request handler that serves static files."""
 
@@ -85,6 +85,10 @@ def do_GET(self):
                 content = content.replace("__PY_REPLACE_EXPECTED_STATE__", state)
                 content = content.replace("__PY_REPLACE_CODE_VERIFIER__", code_verifier)
                 content = content.replace("__PY_REPLACE_REDIRECT_URI__", redirect_uri)
+                content = content.replace(
+                    "__PY_REPLACE_CLIENT_ID__", auth_config["client_id"]
+                )
+                content = content.replace("__PY_REPLACE_DOMAIN__", domain)
 
                 self.send_response(200)
                 self.send_header("Content-Type", "text/html")
@@ -123,25 +127,25 @@ def token_received_callback(self, token_data):
         self.token_data = token_data
         self.should_shutdown = True
 
-    def create_server(self, state, code_verifier):
+    def create_server(self, state, code_verifier, domain):
         """Create and configure the HTTPS server."""
         # Create SSL context
         context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
         context.load_cert_chain(self.cert_file, self.key_file)
 
         # Create server
         handler = make_request_handler_class(
-            state, code_verifier, self.token_received_callback
+            state, code_verifier, self.token_received_callback, domain
         )
         self.httpd = socketserver.TCPServer(("", self.port), handler)
         self.httpd.socket = context.wrap_socket(self.httpd.socket, server_side=True)
 
         return self.httpd
 
-    def start(self, state, code_verifier):
+    def start(self, state, code_verifier, domain):
         """Start the server."""
         if not self.httpd:
-            self.create_server(state, code_verifier)
+            self.create_server(state, code_verifier, domain)
 
         try:
             if self.httpd:

src/uipath_sdk/_cli/_auth/_oidc_utils.py

@@ -28,11 +28,15 @@ def get_auth_config() -> AuthConfig:
     with open(os.path.join(os.path.dirname(__file__), "auth_config.json"), "r") as f:
         auth_config = json.load(f)
 
+    port = auth_config.get("port", 8104)
+
+    redirect_uri = auth_config["redirect_uri"].replace("__PY_REPLACE_PORT__", str(port))
+
     return AuthConfig(
         client_id=auth_config["client_id"],
-        redirect_uri=auth_config["redirect_uri"],
+        redirect_uri=redirect_uri,
         scope=auth_config["scope"],
-        port=auth_config.get("port", 6234),
+        port=port,
     )
 
 

src/uipath_sdk/_cli/_auth/index.html

@@ -39,11 +39,11 @@ <h1>UiPath CLI Authentication</h1>
                 formData.append('grant_type', 'authorization_code');
                 formData.append('code', code);
                 formData.append('redirect_uri', '__PY_REPLACE_REDIRECT_URI__');
-                formData.append('client_id', '1119a927-10ab-4543-bd1a-ad6bfbbc27f4');
+                formData.append('client_id', '__PY_REPLACE_CLIENT_ID__');
                 formData.append('code_verifier', codeVerifier);
 
                 // Make token request
-                const response = await fetch('https://alpha.uipath.com/identity_/connect/token', {
+                const response = await fetch('https://__PY_REPLACE_DOMAIN__.uipath.com/identity_/connect/token', {
                     method: 'POST',
                     headers: {
                         'Content-Type': 'application/x-www-form-urlencoded'
@@ -164,4 +164,4 @@ <h1>UiPath CLI Authentication</h1>
     </script>
 </body>
 
-</html>
+</html>