Utilise WordPress's default 'remember me' function

Author: donovandb

onelogin-saml-sso/php/functions.php

@@ -1,4 +1,12 @@
 <?php
+/* 
+ * includes capabilities for a "remember me" login flag passed as a SAML Attribute.
+ * To use, pass a value of 'yes' in a SAML Attribute, then map that attribute
+ * name in the WordPress Dashboard SSO Settings here:
+ * Wordpress Settings => SSO/SAML Settings => Attribute Mapping => Remember Me
+ 
+ * Other affected files: /php/configuration.php
+*/
 
 // Make sure we don't expose any info if called directly
 if ( !function_exists( 'add_action' ) ) {
@@ -280,10 +288,24 @@ function saml_acs() {
 		exit();
 	} else if ($user_id) {
 		wp_set_current_user($user_id);
-		wp_set_auth_cookie($user_id);
-		setcookie(SAML_LOGIN_COOKIE, 1, time() + YEAR_IN_SECONDS, SITECOOKIEPATH );
-		#do_action('wp_login', $user_id);
-		#wp_signon($user_id);
+		
+		$remembermeMapping = get_option('onelogin_saml_attr_mapping_rememberme');
+		if ( !empty($remembermeMapping) && isset($attrs[$remembermeMapping]) && !empty($attrs[$remembermeMapping][0])) {
+			$rememberme = $attrs[$remembermeMapping][0];
+		}
+		   
+		if ( isset($rememberme) ) {
+			// ** Value of 'yes' is assumed to be passed here, if other, need to edit this. 
+			if ( $rememberme == 'yes' ) {
+				wp_set_auth_cookie($user_id, true);
+			} else {
+				wp_set_auth_cookie($user_id);
+			}
+		} else {
+			wp_set_auth_cookie($user_id);
+		}
+
+		setcookie('saml_login', 1, time() + YEAR_IN_SECONDS, SITECOOKIEPATH );
 	}
 
 	if (isset($_REQUEST['RelayState'])) {