initial commit
This commit is contained in:
253
includes/wccom-site/class-wc-wccom-site.php
Normal file
253
includes/wccom-site/class-wc-wccom-site.php
Normal file
@ -0,0 +1,253 @@
|
||||
<?php
|
||||
/**
|
||||
* WooCommerce.com Product Installation.
|
||||
*
|
||||
* @package WooCommerce\WCCom
|
||||
* @since 3.7.0
|
||||
*/
|
||||
|
||||
defined( 'ABSPATH' ) || exit;
|
||||
|
||||
/**
|
||||
* WC_WCCOM_Site Class
|
||||
*
|
||||
* Main class for WooCommerce.com connected site.
|
||||
*/
|
||||
class WC_WCCOM_Site {
|
||||
|
||||
const AUTH_ERROR_FILTER_NAME = 'wccom_auth_error';
|
||||
|
||||
/**
|
||||
* Load the WCCOM site class.
|
||||
*
|
||||
* @since 3.7.0
|
||||
*/
|
||||
public static function load() {
|
||||
self::includes();
|
||||
|
||||
add_action( 'woocommerce_wccom_install_products', array( 'WC_WCCOM_Site_Installer', 'install' ) );
|
||||
add_filter( 'determine_current_user', array( __CLASS__, 'authenticate_wccom' ), 14 );
|
||||
add_action( 'woocommerce_rest_api_get_rest_namespaces', array( __CLASS__, 'register_rest_namespace' ) );
|
||||
}
|
||||
|
||||
/**
|
||||
* Include support files.
|
||||
*
|
||||
* @since 3.7.0
|
||||
*/
|
||||
protected static function includes() {
|
||||
require_once WC_ABSPATH . 'includes/admin/helper/class-wc-helper.php';
|
||||
require_once WC_ABSPATH . 'includes/wccom-site/class-wc-wccom-site-installer.php';
|
||||
require_once WC_ABSPATH . 'includes/wccom-site/class-wc-wccom-site-installer-requirements-check.php';
|
||||
}
|
||||
|
||||
/**
|
||||
* Authenticate WooCommerce.com request.
|
||||
*
|
||||
* @since 3.7.0
|
||||
* @param int|false $user_id User ID.
|
||||
* @return int|false
|
||||
*/
|
||||
public static function authenticate_wccom( $user_id ) {
|
||||
if ( ! empty( $user_id ) || ! self::is_request_to_wccom_site_rest_api() ) {
|
||||
return $user_id;
|
||||
}
|
||||
|
||||
$auth_header = trim( self::get_authorization_header() );
|
||||
|
||||
if ( stripos( $auth_header, 'Bearer ' ) === 0 ) {
|
||||
$access_token = trim( substr( $auth_header, 7 ) );
|
||||
} elseif ( ! empty( $_GET['token'] ) && is_string( $_GET['token'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended
|
||||
$access_token = trim( $_GET['token'] ); // phpcs:ignore WordPress.Security.NonceVerification.Recommended, WordPress.Security.ValidatedSanitizedInput.MissingUnslash, WordPress.Security.ValidatedSanitizedInput.InputNotSanitized
|
||||
} else {
|
||||
add_filter(
|
||||
self::AUTH_ERROR_FILTER_NAME,
|
||||
function() {
|
||||
return new WP_Error(
|
||||
WC_REST_WCCOM_Site_Installer_Errors::NO_ACCESS_TOKEN_CODE,
|
||||
WC_REST_WCCOM_Site_Installer_Errors::NO_ACCESS_TOKEN_MESSAGE,
|
||||
array( 'status' => WC_REST_WCCOM_Site_Installer_Errors::NO_ACCESS_TOKEN_HTTP_CODE )
|
||||
);
|
||||
}
|
||||
);
|
||||
return false;
|
||||
}
|
||||
|
||||
if ( ! empty( $_SERVER['HTTP_X_WOO_SIGNATURE'] ) ) {
|
||||
$signature = trim( $_SERVER['HTTP_X_WOO_SIGNATURE'] ); // phpcs:ignore WordPress.Security.ValidatedSanitizedInput.MissingUnslash,WordPress.Security.ValidatedSanitizedInput.InputNotSanitized
|
||||
} elseif ( ! empty( $_GET['signature'] ) && is_string( $_GET['signature'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended
|
||||
$signature = trim( $_GET['signature'] ); // phpcs:ignore WordPress.Security.NonceVerification.Recommended, WordPress.Security.ValidatedSanitizedInput.MissingUnslash, WordPress.Security.ValidatedSanitizedInput.InputNotSanitized
|
||||
} else {
|
||||
add_filter(
|
||||
self::AUTH_ERROR_FILTER_NAME,
|
||||
function() {
|
||||
return new WP_Error(
|
||||
WC_REST_WCCOM_Site_Installer_Errors::NO_SIGNATURE_CODE,
|
||||
WC_REST_WCCOM_Site_Installer_Errors::NO_SIGNATURE_MESSAGE,
|
||||
array( 'status' => WC_REST_WCCOM_Site_Installer_Errors::NO_SIGNATURE_HTTP_CODE )
|
||||
);
|
||||
}
|
||||
);
|
||||
return false;
|
||||
}
|
||||
|
||||
require_once WC_ABSPATH . 'includes/admin/helper/class-wc-helper-options.php';
|
||||
$site_auth = WC_Helper_Options::get( 'auth' );
|
||||
|
||||
if ( empty( $site_auth['access_token'] ) ) {
|
||||
add_filter(
|
||||
self::AUTH_ERROR_FILTER_NAME,
|
||||
function() {
|
||||
return new WP_Error(
|
||||
WC_REST_WCCOM_Site_Installer_Errors::SITE_NOT_CONNECTED_CODE,
|
||||
WC_REST_WCCOM_Site_Installer_Errors::SITE_NOT_CONNECTED_MESSAGE,
|
||||
array( 'status' => WC_REST_WCCOM_Site_Installer_Errors::SITE_NOT_CONNECTED_HTTP_CODE )
|
||||
);
|
||||
}
|
||||
);
|
||||
return false;
|
||||
}
|
||||
|
||||
if ( ! hash_equals( $access_token, $site_auth['access_token'] ) ) {
|
||||
add_filter(
|
||||
self::AUTH_ERROR_FILTER_NAME,
|
||||
function() {
|
||||
return new WP_Error(
|
||||
WC_REST_WCCOM_Site_Installer_Errors::INVALID_TOKEN_CODE,
|
||||
WC_REST_WCCOM_Site_Installer_Errors::INVALID_TOKEN_MESSAGE,
|
||||
array( 'status' => WC_REST_WCCOM_Site_Installer_Errors::INVALID_TOKEN_HTTP_CODE )
|
||||
);
|
||||
}
|
||||
);
|
||||
return false;
|
||||
}
|
||||
|
||||
$body = WP_REST_Server::get_raw_data();
|
||||
|
||||
if ( ! self::verify_wccom_request( $body, $signature, $site_auth['access_token_secret'] ) ) {
|
||||
add_filter(
|
||||
self::AUTH_ERROR_FILTER_NAME,
|
||||
function() {
|
||||
return new WP_Error(
|
||||
WC_REST_WCCOM_Site_Installer_Errors::REQUEST_VERIFICATION_FAILED_CODE,
|
||||
WC_REST_WCCOM_Site_Installer_Errors::REQUEST_VERIFICATION_FAILED_MESSAGE,
|
||||
array( 'status' => WC_REST_WCCOM_Site_Installer_Errors::REQUEST_VERIFICATION_FAILED_HTTP_CODE )
|
||||
);
|
||||
}
|
||||
);
|
||||
return false;
|
||||
}
|
||||
|
||||
$user = get_user_by( 'id', $site_auth['user_id'] );
|
||||
if ( ! $user ) {
|
||||
add_filter(
|
||||
self::AUTH_ERROR_FILTER_NAME,
|
||||
function() {
|
||||
return new WP_Error(
|
||||
WC_REST_WCCOM_Site_Installer_Errors::USER_NOT_FOUND_CODE,
|
||||
WC_REST_WCCOM_Site_Installer_Errors::USER_NOT_FOUND_MESSAGE,
|
||||
array( 'status' => WC_REST_WCCOM_Site_Installer_Errors::USER_NOT_FOUND_HTTP_CODE )
|
||||
);
|
||||
}
|
||||
);
|
||||
return false;
|
||||
}
|
||||
|
||||
return $user;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the authorization header.
|
||||
*
|
||||
* On certain systems and configurations, the Authorization header will be
|
||||
* stripped out by the server or PHP. Typically this is then used to
|
||||
* generate `PHP_AUTH_USER`/`PHP_AUTH_PASS` but not passed on. We use
|
||||
* `getallheaders` here to try and grab it out instead.
|
||||
*
|
||||
* @since 3.7.0
|
||||
* @return string Authorization header if set.
|
||||
*/
|
||||
protected static function get_authorization_header() {
|
||||
if ( ! empty( $_SERVER['HTTP_AUTHORIZATION'] ) ) {
|
||||
return wp_unslash( $_SERVER['HTTP_AUTHORIZATION'] ); // phpcs:ignore WordPress.Security.ValidatedSanitizedInput.InputNotSanitized
|
||||
}
|
||||
|
||||
if ( function_exists( 'getallheaders' ) ) {
|
||||
$headers = getallheaders();
|
||||
// Check for the authoization header case-insensitively.
|
||||
foreach ( $headers as $key => $value ) {
|
||||
if ( 'authorization' === strtolower( $key ) ) {
|
||||
return $value;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return '';
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if this is a request to WCCOM Site REST API.
|
||||
*
|
||||
* @since 3.7.0
|
||||
* @return bool
|
||||
*/
|
||||
protected static function is_request_to_wccom_site_rest_api() {
|
||||
|
||||
if ( isset( $_REQUEST['rest_route'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended
|
||||
$route = wp_unslash( $_REQUEST['rest_route'] ); // phpcs:ignore WordPress.Security.ValidatedSanitizedInput.InputNotSanitized, WordPress.Security.NonceVerification.Recommended
|
||||
$rest_prefix = '';
|
||||
} else {
|
||||
$route = wp_unslash( add_query_arg( array() ) );
|
||||
$rest_prefix = trailingslashit( rest_get_url_prefix() );
|
||||
}
|
||||
|
||||
return false !== strpos( $route, $rest_prefix . 'wccom-site/' );
|
||||
}
|
||||
|
||||
/**
|
||||
* Verify WooCommerce.com request from a given body and signature request.
|
||||
*
|
||||
* @since 3.7.0
|
||||
* @param string $body Request body.
|
||||
* @param string $signature Request signature found in X-Woo-Signature header.
|
||||
* @param string $access_token_secret Access token secret for this site.
|
||||
* @return bool
|
||||
*/
|
||||
protected static function verify_wccom_request( $body, $signature, $access_token_secret ) {
|
||||
// phpcs:disable WordPress.Security.ValidatedSanitizedInput.InputNotValidated, WordPress.Security.ValidatedSanitizedInput.MissingUnslash, WordPress.Security.ValidatedSanitizedInput.InputNotSanitized
|
||||
$data = array(
|
||||
'host' => $_SERVER['HTTP_HOST'],
|
||||
'request_uri' => urldecode( remove_query_arg( array( 'token', 'signature' ), $_SERVER['REQUEST_URI'] ) ),
|
||||
'method' => strtoupper( $_SERVER['REQUEST_METHOD'] ),
|
||||
);
|
||||
// phpcs:enable
|
||||
|
||||
if ( ! empty( $body ) ) {
|
||||
$data['body'] = $body;
|
||||
}
|
||||
|
||||
$expected_signature = hash_hmac( 'sha256', wp_json_encode( $data ), $access_token_secret );
|
||||
|
||||
return hash_equals( $expected_signature, $signature );
|
||||
}
|
||||
|
||||
/**
|
||||
* Register wccom-site REST namespace.
|
||||
*
|
||||
* @since 3.7.0
|
||||
* @param array $namespaces List of registered namespaces.
|
||||
* @return array Registered namespaces.
|
||||
*/
|
||||
public static function register_rest_namespace( $namespaces ) {
|
||||
require_once WC_ABSPATH . 'includes/wccom-site/rest-api/class-wc-rest-wccom-site-installer-errors.php';
|
||||
require_once WC_ABSPATH . 'includes/wccom-site/rest-api/endpoints/class-wc-rest-wccom-site-installer-controller.php';
|
||||
|
||||
$namespaces['wccom-site/v1'] = array(
|
||||
'installer' => 'WC_REST_WCCOM_Site_Installer_Controller',
|
||||
);
|
||||
|
||||
return $namespaces;
|
||||
}
|
||||
}
|
||||
|
||||
WC_WCCOM_Site::load();
|
||||
Reference in New Issue
Block a user