Merge pull request #53 from WP-API/filter-redirect-args

Allow plugins to filter redirect arguments

Author: rmccue

inc/types/class-authorization-code.php

@@ -60,6 +60,13 @@ protected function handle_authorization_submission( $submit, Client $client, $da
 			$redirect_args['state'] = $data['state'];
 		}
 
+		$redirect_args = $this->filter_redirect_args(
+			$redirect_args,
+			$submit === 'authorize',
+			$client,
+			$data
+		);
+
 		$generated_redirect = add_query_arg( urlencode_deep( $redirect_args ), $redirect_uri );
 		wp_redirect( $generated_redirect );
 		exit;

inc/types/class-base.php

@@ -154,4 +154,34 @@ protected function render_form( Client $client, WP_Error $errors = null ) {
 	protected function get_nonce_action( Client $client ) {
 		return sprintf( 'oauth2_authorize:%s', $client->get_id() );
 	}
+
+	/**
+	 * Filter the redirection args.
+	 *
+	 * @param array $redirect_args Redirect args.
+	 * @param boolean $authorized True if authorized, false otherwise.
+	 * @param Client $client Client being authorised.
+	 * @param array $data Data for the request.
+	 */
+	protected function filter_redirect_args( $redirect_args, $authorized, Client $client, $data ) {
+		if ( ! $authorized ) {
+			/**
+			 * Filter the redirect args when the user has cancelled.
+			 *
+			 * @param array $redirect_args Redirect args.
+			 * @param Client $client Client being authorised.
+			 * @param array $data Data for the request.
+			 */
+			return apply_filters( 'oauth2.redirect_args.cancelled', $redirect_args, $client, $data );
+		}
+
+		/**
+		 * Filter the redirect args when the user has authorized.
+		 *
+		 * @param array $redirect_args Redirect args.
+		 * @param Client $client Client being authorised.
+		 * @param array $data Data for the request.
+		 */
+		return apply_filters( 'oauth2.redirect_args.authorized', $redirect_args, $client, $data );
+	}
 }

inc/types/class-implicit.php

@@ -61,6 +61,13 @@ protected function handle_authorization_submission( $submit, Client $client, $da
 			$redirect_args['state'] = $data['state'];
 		}
 
+		$redirect_args = $this->filter_redirect_args(
+			$redirect_args,
+			$submit === 'authorize',
+			$client,
+			$data
+		);
+
 		$fragment           = build_query( $redirect_args );
 		$generated_redirect = $redirect_uri . '#' . $fragment;
 		wp_redirect( $generated_redirect );