work on logout

app/main/views/sign_in.py

@@ -27,20 +27,15 @@ from app.utils.login import is_safe_redirect_url
 
 
 def _get_access_token(code, state):
-    current_app.logger.info(
-        f"HURRAY!  THIS IS REDIRECT FROM LOGIN DOT GOV AND WE HAVE CODE {code} and STATE {state}"
-    )
-    # TODO use the code to get the access_token with jwt
-    # Using the access_token get the email from the user info
-    # Use the call five lines down to look up the user from the email
-    # activate the user and redirect as five lines down
+    client_id = os.getenv("LOGIN_DOT_GOV_CLIENT_ID")
+    access_token_url = os.getenv("LOGIN_DOT_GOV_ACCESS_TOKEN_URL")
     pemfile = open("./private.pem", "r")
     keystring = pemfile.read()
     pemfile.close()
     payload = {
-        "iss": "urn:gov:gsa:openidconnect.profiles:sp:sso:gsa:test_notify_gov",
-        "sub": "urn:gov:gsa:openidconnect.profiles:sp:sso:gsa:test_notify_gov",
-        "aud": "https://idp.int.identitysandbox.gov/api/openid_connect/token",
+        "iss": client_id,
+        "sub": client_id,
+        "aud": access_token_url,
         "jti": str(uuid.uuid4()),
         # JWT expiration time (10 minute maximum)
         "exp": int(time.time()) + (10 * 60),
@@ -48,7 +43,7 @@ def _get_access_token(code, state):
 
     jwt_instance = jwt.PyJWT()
     token = jwt_instance.encode(payload, keystring, algorithm="RS256")
-    base_url = "https://idp.int.identitysandbox.gov/api/openid_connect/token?"
+    base_url = f"{access_token_url}?"
     cli_assert = f"client_assertion={token}"
     cli_assert_type = "client_assertion_type=urn%3Aietf%3Aparams%3Aoauth%3Aclient-assertion-type%3Ajwt-bearer"
     code_param = f"code={code}"
@@ -57,17 +52,16 @@ def _get_access_token(code, state):
     response = requests.post(url, headers=headers)
     current_app.logger.info(f"GOT A RESPONSE {response.json()}")
     access_token = response.json()["access_token"]
-    current_app.logger.info(f"HURRAY GOT THE ACCESS TOKEN! {access_token}")
     return access_token
 
 
 def _get_user_email(access_token):
     headers = {"Authorization": "Bearer %s" % access_token}
+    user_info_url = os.getenv("LOGIN_DOT_GOV_USER_INFO_URL")
     user_attributes = requests.get(
-        "https://idp.int.identitysandbox.gov/api/openid_connect/userinfo",
+        user_info_url,
         headers=headers,
     )
-    current_app.logger.info(f"HURRAY GOT USER ATTRIBUTES {user_attributes.json()}")
     user_email = user_attributes.json()["email"]
     return user_email
 

app/main/views/sign_out.py

@@ -1,12 +1,34 @@
-from flask import redirect, url_for
+import os
+import uuid
+
+import requests
+from flask import redirect, url_for, current_app
 from flask_login import current_user
 
 from app.main import main
 
 
+def _sign_out_at_login_dot_gov():
+
+    base_url = "https://idp.int.identitysandbox.gov/openid_connect/logout?"
+    client_id = (
+        f"client_id={os.getenv('LOGIN_DOT_GOV_CLIENT_ID')}"
+    )
+    post_logout_redirect_uri = "post_logout_redirect_api=http://localhost:6012/sign-in"
+    state = f"state={str(uuid.uuid4())}"
+
+    # TODO If I take this url and put it in the browser, login.gov sign out works properly
+    # TODO But with this code it results in a 404 error message and we don't sign out from login.gov
+    url = f"{base_url}&{client_id}&{post_logout_redirect_uri}&{state}"
+    current_app.logger.info(f"URL={url}")
+    response = requests.post(url)
+    current_app.logger.info(f"GOT A RESPONSE {response}")
+
+
 @main.route("/sign-out", methods=(["GET"]))
 def sign_out():
     # An AnonymousUser does not have an id
     if current_user.is_authenticated:
         current_user.sign_out()
+        _sign_out_at_login_dot_gov()
     return redirect(url_for("main.index"))