From 397391221afcd61d2d6debe8d7cd6c66c2aabad3 Mon Sep 17 00:00:00 2001
From: neverbiasu <1751162157@qq.com>
Date: Fri, 3 May 2024 22:40:59 +0800
Subject: [PATCH] addCheckPort

---
 phi/cli/auth_server.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/phi/cli/auth_server.py b/phi/cli/auth_server.py
index e33e348e9..f4afed740 100644
--- a/phi/cli/auth_server.py
+++ b/phi/cli/auth_server.py
@@ -3,6 +3,7 @@
 
 from phi.cli.settings import phi_cli_settings
 
+import socket
 
 class CliAuthRequestHandler(BaseHTTPRequestHandler):
     """Request Handler to accept the CLI auth token after the web based auth flow.
@@ -82,9 +83,14 @@ def shut_down(self):
         self._thread.close()  # type: ignore
 
 
+def check_port(port):
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+        return s.connect_ex(('localhost', port)) == 0
+    
+
 def get_port_for_auth_server():
-    # TODO: Check if port is available
-    return 9191
+    if check_port(9191):
+        return 9191
 
 
 def get_auth_token_from_web_flow(port) -> Optional[str]: