fix: configurable claims and relay state validity

Author: sattvikc

src/main/java/io/supertokens/storage/postgresql/Start.java

@@ -4494,9 +4494,9 @@ public List<SAMLClient> getSAMLClients(TenantIdentifier tenantIdentifier) throws
     }
 
     @Override
-    public void saveRelayStateInfo(TenantIdentifier tenantIdentifier, SAMLRelayStateInfo relayStateInfo) throws StorageQueryException {
+    public void saveRelayStateInfo(TenantIdentifier tenantIdentifier, SAMLRelayStateInfo relayStateInfo, long relayStateValidity) throws StorageQueryException {
         try {
-            SAMLQueries.saveRelayStateInfo(this, tenantIdentifier, relayStateInfo.relayState, relayStateInfo.clientId, relayStateInfo.state, relayStateInfo.redirectURI);
+            SAMLQueries.saveRelayStateInfo(this, tenantIdentifier, relayStateInfo.relayState, relayStateInfo.clientId, relayStateInfo.state, relayStateInfo.redirectURI, relayStateValidity);
         } catch (SQLException e) {
             throw new StorageQueryException(e);
         }
@@ -4512,9 +4512,9 @@ public SAMLRelayStateInfo getRelayStateInfo(TenantIdentifier tenantIdentifier, S
     }
 
     @Override
-    public void saveSAMLClaims(TenantIdentifier tenantIdentifier, String clientId, String code, JsonObject claims) throws StorageQueryException {
+    public void saveSAMLClaims(TenantIdentifier tenantIdentifier, String clientId, String code, JsonObject claims, long claimsValidity) throws StorageQueryException {
         try {
-            SAMLQueries.saveSAMLClaims(this, tenantIdentifier, clientId, code, claims.toString());
+            SAMLQueries.saveSAMLClaims(this, tenantIdentifier, clientId, code, claims.toString(), claimsValidity);
         } catch (SQLException e) {
             throw new StorageQueryException(e);
         }

src/main/java/io/supertokens/storage/postgresql/queries/SAMLQueries.java

@@ -139,7 +139,7 @@ public static String getQueryToCreateSAMLClaimsExpiresAtIndex(Start start) {
     }
 
     public static void saveRelayStateInfo(Start start, TenantIdentifier tenantIdentifier,
-                                          String relayState, String clientId, String state, String redirectURI)
+                                          String relayState, String clientId, String state, String redirectURI, long relayStateValidity)
             throws StorageQueryException, SQLException {
         String QUERY = "INSERT INTO " + getConfig(start).getSAMLRelayStateTable()
                 + "(app_id, tenant_id, relay_state, client_id, state, redirect_uri, created_at, expires_at)"
@@ -154,7 +154,7 @@ public static void saveRelayStateInfo(Start start, TenantIdentifier tenantIdenti
             pst.setString(6, redirectURI);
             long now = System.currentTimeMillis();
             pst.setLong(7, now);
-            pst.setLong(8, now + 300000);
+            pst.setLong(8, now + relayStateValidity);
         });
     }
 
@@ -181,7 +181,7 @@ public static SAMLRelayStateInfo getRelayStateInfo(Start start, TenantIdentifier
         });
     }
 
-    public static void saveSAMLClaims(Start start, TenantIdentifier tenantIdentifier, String clientId, String code, String claimsJson)
+    public static void saveSAMLClaims(Start start, TenantIdentifier tenantIdentifier, String clientId, String code, String claimsJson, long claimsValidity)
             throws StorageQueryException, SQLException {
         String QUERY = "INSERT INTO " + getConfig(start).getSAMLClaimsTable()
                 + "(app_id, tenant_id, client_id, code, claims, created_at, expires_at)"
@@ -195,7 +195,7 @@ public static void saveSAMLClaims(Start start, TenantIdentifier tenantIdentifier
             pst.setString(5, claimsJson);
             long now = System.currentTimeMillis();
             pst.setLong(6, now);
-            pst.setLong(7, now + 300000);
+            pst.setLong(7, now + claimsValidity);
         });
     }