Merge 63d90f0 into f032491

fence/blueprints/login/ras.py

@@ -58,3 +58,13 @@ def post_login(self, user, token_result):
 
             current_session.add(visa)
             current_session.commit()
+
+        # Store refresh token in db
+        refresh_token = flask.g.tokens.get("refresh_token")
+        id_token = flask.g.tokens.get("id_token")
+        decoded_id = jwt.decode(id_token, verify=False)
+        # Add 15 days to iat to calculate refresh token expiration time
+        expires = int(decoded_id.get("iat")) + (15 * 24 * 60 * 60)
+        flask.current_app.ras_client.store_refresh_token(
+            user=user, refresh_token=refresh_token, expires=expires
+        )

fence/models.py

@@ -1,10 +1,8 @@
 """
 Define sqlalchemy models.
-
 The models here inherit from the `Base` in userdatamodel, so when the fence app
 is initialized, the resulting db session includes everything from userdatamodel
 and this file.
-
 The `migrate` function in this file is called every init and can be used for
 database migrations.
 """
@@ -71,11 +69,9 @@ def query_for_user(session, username):
 class ClientAuthType(Enum):
     """
     List the possible types of OAuth client authentication, which are
-
     - None (no authentication).
     - Basic (using basic HTTP authorization header to include the client ID & secret).
     - POST (the client ID & secret are included in the body of a POST request).
-
     These all have a corresponding string which identifies them to authlib.
     """
 
@@ -555,6 +551,26 @@ class GA4GHVisaV1(Base):
     expires = Column(BigInteger, nullable=False)
 
 
+class UpstreamRefreshToken(Base):
+    # General table to store any refresh_token sent from any oidc client
+
+    __tablename__ = "upstream_refresh_token"
+
+    id = Column(BigInteger, primary_key=True)
+
+    user_id = Column(Integer, ForeignKey(User.id, ondelete="CASCADE"), nullable=False)
+    user = relationship(
+        "User",
+        backref=backref(
+            "upstream_refresh_tokens",
+            cascade="all, delete-orphan",
+            passive_deletes=True,
+        ),
+    )
+    refresh_token = Column(Text, nullable=False)
+    expires = Column(BigInteger, nullable=False)
+
+
 to_timestamp = (
     "CREATE OR REPLACE FUNCTION pc_datetime_to_timestamp(datetoconvert timestamp) "
     "RETURNS BIGINT AS "
@@ -1041,7 +1057,6 @@ def _remove_policy(driver, md):
 def _add_google_project_id(driver, md):
     """
     Add new unique not null field to GoogleServiceAccount.
-
     In order to do this without errors, we have to:
         - add the field and allow null (for all previous rows)
         - update all null entries to be unique

fence/resources/openid/idp_oauth2.py

@@ -2,6 +2,10 @@
 from cached_property import cached_property
 from jose import jwt
 import requests
+import time
+from fence.errors import AuthError
+from fence.models import UpstreamRefreshToken
+from flask_sqlalchemy_session import current_session
 
 
 class Oauth2ClientBase(object):
@@ -128,3 +132,51 @@ def get_user_id(self, code):
         for successfully logged in user OR "error" field with details of the error.
         """
         raise NotImplementedError()
+
+    def get_access_token(self, user, token_endpoint):
+
+        """
+        Get access_token using a refresh_token
+        """
+        refresh_token = None
+        expires = None
+
+        # get refresh_token and expiration from db
+        for row in user.upstream_refresh_tokens:
+            refresh_token = row.refresh_token
+            expires = row.expires
+
+        if not refresh_token:
+            raise AuthError("User doesnt have a refresh token")
+        if time.time() > expires:
+            raise AuthError("Refresh token expired. Please login again.")
+
+        try:
+            token_response = self.session.refresh_token(
+                url=token_endpoint,
+                proxies=self.get_proxies(),
+                refresh_token=refresh_token,
+            )
+            new_refresh_token = token_response["refresh_token"]
+
+            self.store_refresh_token(
+                user, refresh_token=new_refresh_token, expires=expires
+            )
+
+        except Exception as e:
+            err_msg = "Failed to get refresh token"
+            self.logger.exception("{} : {}".format(err_msg, e))
+        return token_response
+
+    def store_refresh_token(self, user, refresh_token, expires):
+        """
+        Store refresh token in db.
+        """
+        user.upstream_refresh_tokens = []
+        current_session.commit()
+        upstream_refresh_token = UpstreamRefreshToken(
+            user=user, refresh_token=refresh_token, expires=expires,
+        )
+        current_db_session = current_session.object_session(upstream_refresh_token)
+        current_db_session.add(upstream_refresh_token)
+        current_session.commit()

fence/resources/openid/ras_oauth2.py

@@ -1,4 +1,3 @@
-# need new RAS
 import flask
 from .idp_oauth2 import Oauth2ClientBase
 from jose import jwt
@@ -9,7 +8,6 @@ class RASOauth2Client(Oauth2ClientBase):
     """
     client for interacting with RAS oauth 2,
     as openid connect is supported under oauth2
-
     """
 
     RAS_DISCOVERY_URL = "https://stsstg.nih.gov/.well-known/openid-configuration"
@@ -57,34 +55,40 @@ def get_user_id(self, code):
 
             token = self.get_token(token_endpoint, code)
             keys = self.get_jwt_keys(jwks_endpoint)
+            userinfo = self.get_userinfo(token, userinfo_endpoint)
 
             claims = jwt.decode(
                 token["id_token"],
                 keys,
                 options={"verify_aud": False, "verify_at_hash": False},
             )
 
-            userinfo = self.get_userinfo(token, userinfo_endpoint)
+            username = None
+            if userinfo.get("UserID"):
+                username = userinfo["UserID"]
+                field_name = "UserID"
+            elif userinfo.get("preferred_username"):
+                username = userinfo["preferred_username"]
+                field_name = "preferred_username"
+            elif claims.get("sub"):
+                username = claims["sub"]
+                field_name = "sub"
+            if not username:
+                self.logger.error(
+                    "{}, received claims: {} and userinfo: {}".format(
+                        err_msg, claims, userinfo
+                    )
+                )
+                return {"error": err_msg}
 
-            # Save userinfo in flask.g.user for later use in post_login
+            self.logger.info("Using {} field as username.".format(field_name))
+
+            # Save userinfo and token in flask.g for later use in post_login
             flask.g.userinfo = userinfo
+            flask.g.tokens = token
 
         except Exception as e:
             self.logger.exception("{}: {}".format(err_msg, e))
             return {"error": err_msg}
 
-        username = None
-        if userinfo.get("UserID"):
-            username = userinfo["UserID"]
-        elif userinfo.get("preferred_username"):
-            username = userinfo["preferred_username"]
-        elif claims.get("sub"):
-            username = claims["sub"]
-        if not username:
-            logger.error(
-                "{}, received claims: {} and userinfo: {}".format(
-                    err_msg, claims, userinfo
-                )
-            )
-            return {"error": err_msg}
         return {"username": username}