From 6cd6cc126a44419903642e19804f1a466788262c Mon Sep 17 00:00:00 2001 From: Robert Staddon Date: Wed, 18 May 2016 11:36:57 -0500 Subject: [PATCH] Record the URL of the current page when creating a login button --- includes/openid-connect-generic-login-form.php | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/includes/openid-connect-generic-login-form.php b/includes/openid-connect-generic-login-form.php index e580cbe..00e3eb1 100644 --- a/includes/openid-connect-generic-login-form.php +++ b/includes/openid-connect-generic-login-form.php @@ -86,6 +86,16 @@ class OpenID_Connect_Generic_Login_Form { $text = apply_filters( 'openid-connect-generic-login-button-text', __( 'Login with OpenID Connect' ) ); $href = $this->client_wrapper->get_authentication_url(); + // record the URL of this page if set to redirect back to origin page + if( $this->settings->redirect_user_back ) { + $redirect_expiry = time() + DAY_IN_SECONDS; + if ( $GLOBALS['pagenow'] == 'wp-login.php' ) + $redirect_url = admin_url(); + else + $redirect_url = home_url( esc_url( add_query_arg( NULL, NULL ) ) ); + setcookie( $this->client_wrapper->cookie_redirect_key, $redirect_url, $redirect_expiry, COOKIEPATH, COOKIE_DOMAIN, is_ssl() ); + } + ob_start(); ?>