add openid login endpoints

BSC-ES · Nov 19, 2024 · ebd8ebc · ebd8ebc
1 parent 2579e8b
commit ebd8ebc
Show file tree

Hide file tree

Showing 2 changed files with 78 additions and 2 deletions.
diff --git a/autosubmit_api/config/__init__.py b/autosubmit_api/config/__init__.py
@@ -27,9 +27,18 @@
 
 GITHUB_OAUTH_CLIENT_ID = os.environ.get("GITHUB_OAUTH_CLIENT_ID")
 GITHUB_OAUTH_CLIENT_SECRET = os.environ.get("GITHUB_OAUTH_CLIENT_SECRET")
-GITHUB_OAUTH_WHITELIST_ORGANIZATION = os.environ.get("GITHUB_OAUTH_WHITELIST_ORGANIZATION")
+GITHUB_OAUTH_WHITELIST_ORGANIZATION = os.environ.get(
+    "GITHUB_OAUTH_WHITELIST_ORGANIZATION"
+)
 GITHUB_OAUTH_WHITELIST_TEAM = os.environ.get("GITHUB_OAUTH_WHITELIST_TEAM")
 
+# OpenID Connect
+
+OIDC_TOKEN_URL = os.environ.get("OIDC_TOKEN_URL")
+OIDC_USERINFO_URL = os.environ.get("OIDC_USERINFO_URL")
+OIDC_CLIENT_ID = os.environ.get("OIDC_CLIENT_ID")
+OIDC_CLIENT_SECRET = os.environ.get("OIDC_CLIENT_SECRET")
+
 
 # Startup options
 def get_run_background_tasks_on_start():
@@ -47,4 +56,5 @@ def get_disable_background_tasks():
         "true",
     ]  # Default false
 
-AS_API_ROOT_PATH=os.environ.get("AS_API_ROOT_PATH", "")
+
+AS_API_ROOT_PATH = os.environ.get("AS_API_ROOT_PATH", "")
diff --git a/autosubmit_api/routers/v4/auth.py b/autosubmit_api/routers/v4/auth.py
@@ -198,3 +198,69 @@ async def github_oauth2_login(code: Optional[str] = None) -> LoginResponse:
             },
             status_code=HTTPStatus.UNAUTHORIZED,
         )
+
+
+@router.get("/openid/login", name="OpenID Connect login")
+async def openid_login(code: Optional[str] = None) -> LoginResponse:
+    """
+    Authenticate user using a configured OpenID Connect provider.
+    Is used as a callback URL for the OpenID Connect provider /authorize?response_type=code endpoint.
+    This internally exchanges the code for an access token and then uses the access token to get user info.
+    The user info must contain a `sub` field (following the OIDC spec) which is used as the user id.
+    """
+    if not code:
+        return JSONResponse(
+            content={
+                "authenticated": False,
+                "user": None,
+                "token": None,
+                "message": "Can't verify user",
+            },
+            status_code=HTTPStatus.UNAUTHORIZED,
+        )
+
+    resp_obj: dict = requests.post(
+        config.OIDC_TOKEN_URL,
+        data={
+            "client_id": config.OIDC_CLIENT_ID,
+            "client_secret": config.OIDC_CLIENT_SECRET,
+            "code": code,
+            "grant_type": "authorization_code",
+        },
+        headers={"Accept": "application/json"},
+    ).json()
+    access_token = resp_obj.get("access_token")
+
+    user_info: dict = requests.get(
+        config.OIDC_USERINFO_URL,
+        headers={"Authorization": f"Bearer {access_token}"},
+    ).json()
+    username = user_info.get("sub")
+
+    if not username:
+        return JSONResponse(
+            content={
+                "authenticated": False,
+                "user": None,
+                "token": None,
+                "message": "Couldn't find user on OpenID Connect provider",
+            },
+            status_code=HTTPStatus.UNAUTHORIZED,
+        )
+    else:
+        payload = {
+            "user_id": username,
+            "sub": username,
+            "iat": int(datetime.now().timestamp()),
+            "exp": (datetime.now() + timedelta(seconds=config.JWT_EXP_DELTA_SECONDS)),
+        }
+        jwt_token = jwt.encode(payload, config.JWT_SECRET, config.JWT_ALGORITHM)
+        return JSONResponse(
+            content={
+                "authenticated": True,
+                "user": username,
+                "token": f"Bearer {jwt_token}",
+                "message": "Token generated",
+            },
+            status_code=HTTPStatus.OK,
+        )