Merge pull request #57 from donovandb/master

Add Remember Me option

Author: pitbulk

onelogin-saml-sso/php/configuration.php

@@ -1,5 +1,4 @@
 <?php
-
 // Make sure we don't expose any info if called directly
 if ( !function_exists( 'add_action' ) ) {
 	echo 'Hi there!  I\'m just a plugin, not much I can do when called directly.';
@@ -86,7 +85,8 @@ function onelogin_saml_configuration() {
 			'onelogin_saml_attr_mapping_mail' => __('E-mail', 'onelogin-saml-sso') . ' *',
 			'onelogin_saml_attr_mapping_firstname' => __('First Name', 'onelogin-saml-sso'),
 			'onelogin_saml_attr_mapping_lastname' => __('Last Name', 'onelogin-saml-sso'),
-			'onelogin_saml_attr_mapping_role' => __('Role', 'onelogin-saml-sso')
+			'onelogin_saml_attr_mapping_role' => __('Role', 'onelogin-saml-sso'),
+			'onelogin_saml_attr_mapping_rememberme' => __('Remember Me', 'onelogin-saml-sso')
 		);
 		foreach ($mapping_fields as $name => $description) {
 			register_setting($option_group, $name);
@@ -286,6 +286,11 @@ function plugin_setting_string_onelogin_saml_attr_mapping_lastname() {
 			  value= "'.esc_attr(get_option('onelogin_saml_attr_mapping_lastname')).'" size="30">';
 	}
 
+	function plugin_setting_string_onelogin_saml_attr_mapping_rememberme() {
+		echo '<input type="text" name="onelogin_saml_attr_mapping_rememberme" id="onelogin_saml_attr_mapping_rememberme"
+			  value= "'.esc_html(get_option('onelogin_saml_attr_mapping_rememberme')).'" size="30">';
+	}
+
 	function plugin_setting_string_onelogin_saml_attr_mapping_role() {
 		echo '<input type="text" name="onelogin_saml_attr_mapping_role" id="onelogin_saml_attr_mapping_role"
 			  value= "'.esc_attr(get_option('onelogin_saml_attr_mapping_role')).'" size="30">'.

onelogin-saml-sso/php/functions.php

@@ -280,10 +280,15 @@ function saml_acs() {
 		exit();
 	} else if ($user_id) {
 		wp_set_current_user($user_id);
-		wp_set_auth_cookie($user_id);
+		
+		$rememberme = false;
+		$remembermeMapping = get_option('onelogin_saml_attr_mapping_rememberme');
+		if (!empty($remembermeMapping) && isset($attrs[$remembermeMapping]) && !empty($attrs[$remembermeMapping][0])) {
+    			$rememberme = in_array($attrs[$remembermeMapping][0], array(1, true, '1', 'yes', 'on')) ? true : false;
+		}
+		wp_set_auth_cookie($user_id, $rememberme);
+
 		setcookie(SAML_LOGIN_COOKIE, 1, time() + YEAR_IN_SECONDS, SITECOOKIEPATH );
-		#do_action('wp_login', $user_id);
-		#wp_signon($user_id);
 	}
 
 	if (isset($_REQUEST['RelayState'])) {