first commit
This commit is contained in:
127
wp-content/plugins/elementor/includes/user-data.php
Normal file
127
wp-content/plugins/elementor/includes/user-data.php
Normal file
@@ -0,0 +1,127 @@
|
||||
<?php
|
||||
namespace Elementor;
|
||||
|
||||
if ( ! defined( 'ABSPATH' ) ) {
|
||||
exit; // Exit if accessed directly.
|
||||
}
|
||||
|
||||
class User_Data {
|
||||
const API_NAMESPACE = 'elementor/v1';
|
||||
const API_BASE = '/user-data/current-user';
|
||||
|
||||
public static function init() {
|
||||
add_action( 'rest_api_init', fn() => self::register_routes() );
|
||||
}
|
||||
|
||||
private static function register_routes() {
|
||||
register_rest_route( self::API_NAMESPACE, self::API_BASE, [
|
||||
[
|
||||
'methods' => 'GET',
|
||||
'callback' => fn( $request ) => self::route_wrapper( fn() => self::get_current_user( $request ) ),
|
||||
'permission_callback' => fn() => is_user_logged_in(),
|
||||
],
|
||||
[
|
||||
'methods' => 'PATCH',
|
||||
'callback' => fn( $request ) => self::route_wrapper( fn() => self::update_current_user( $request ) ),
|
||||
'permission_callback' => fn() => is_user_logged_in(),
|
||||
'args' => [
|
||||
'suppressedMessages' => [
|
||||
'required' => false,
|
||||
'type' => 'array',
|
||||
'description' => 'Array of suppressed message keys',
|
||||
'items' => [
|
||||
'type' => 'string',
|
||||
],
|
||||
'validate_callback' => function( $param, $request, $key ) {
|
||||
return is_array( $param );
|
||||
},
|
||||
'sanitize_callback' => fn( $param, $request, $key ) => self::sanitize_suppressed_messages( $param, $request, $key ),
|
||||
],
|
||||
],
|
||||
],
|
||||
] );
|
||||
}
|
||||
|
||||
/**
|
||||
* @param \WP_REST_Request $request The request object.
|
||||
* @return \WP_REST_Response|\WP_Error Response object or error.
|
||||
*/
|
||||
public static function get_current_user( $request ) {
|
||||
$current_user = wp_get_current_user();
|
||||
$introduction_meta = User::get_introduction_meta();
|
||||
|
||||
$suppressed_messages = [];
|
||||
if ( is_array( $introduction_meta ) ) {
|
||||
foreach ( $introduction_meta as $key => $value ) {
|
||||
if ( $value ) {
|
||||
$suppressed_messages[] = $key;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
$capabilities = array_keys( $current_user->allcaps );
|
||||
|
||||
$data = [
|
||||
'suppressedMessages' => $suppressed_messages,
|
||||
'capabilities' => $capabilities,
|
||||
];
|
||||
|
||||
return new \WP_REST_Response( $data, 200 );
|
||||
}
|
||||
|
||||
/**
|
||||
* @param \WP_REST_Request $request The request object.
|
||||
* @return \WP_REST_Response|\WP_Error Response object or error.
|
||||
*/
|
||||
public static function update_current_user( $request ) {
|
||||
$user_id = get_current_user_id();
|
||||
|
||||
$suppressed_messages = $request->get_param( 'suppressedMessages' );
|
||||
|
||||
if ( $request->has_param( 'suppressedMessages' ) && is_array( $suppressed_messages ) ) {
|
||||
$introduction_meta = [];
|
||||
foreach ( $suppressed_messages as $message ) {
|
||||
$introduction_meta[ $message ] = true;
|
||||
}
|
||||
|
||||
update_user_meta( $user_id, User::INTRODUCTION_KEY, $introduction_meta );
|
||||
}
|
||||
|
||||
return self::get_current_user( $request );
|
||||
}
|
||||
|
||||
/**
|
||||
* @param array $param The parameter value.
|
||||
* @param \WP_REST_Request $request The request object.
|
||||
* @param string $key The parameter key.
|
||||
* @return array|null The sanitized array or null.
|
||||
*/
|
||||
public static function sanitize_suppressed_messages( $param, $request, $key ) {
|
||||
if ( ! is_array( $param ) ) {
|
||||
return null;
|
||||
}
|
||||
|
||||
$sanitized_messages = [];
|
||||
foreach ( $param as $message ) {
|
||||
if ( is_string( $message ) ) {
|
||||
$sanitized_message = sanitize_text_field( $message );
|
||||
|
||||
if ( ! empty( $sanitized_message ) ) {
|
||||
$sanitized_messages[] = $sanitized_message;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return $sanitized_messages;
|
||||
}
|
||||
|
||||
private static function route_wrapper( callable $cb ) {
|
||||
try {
|
||||
$response = $cb();
|
||||
} catch ( \Exception $e ) {
|
||||
return new \WP_Error( 'unexpected_error', 'Something went wrong', [ 'status' => 500 ] );
|
||||
}
|
||||
|
||||
return $response;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user