1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

<?php
/**
 * REST API Experiment Controller
 *
 * Handles requests to /experiment
 */

namespace Automattic\WooCommerce\Admin\API;

defined( 'ABSPATH' ) || exit;

/**
 * Data controller.
 *
 * @extends WC_REST_Data_Controller
 */
class Experiments extends \WC_REST_Data_Controller {

    /**
     * Endpoint namespace.
     *
     * @var string
     */
    protected $namespace = 'wc-admin';

    /**
     * Route base.
     *
     * @var string
     */
    protected $rest_base = 'experiments';

    /**
     * Register routes.
     */
    public function register_routes() {
        register_rest_route(
            $this->namespace,
            '/' . $this->rest_base . '/assignment',
            array(
                array(
                    'methods'             => \WP_REST_Server::READABLE,
                    'callback'            => array( $this, 'get_assignment' ),
                    'permission_callback' => array( $this, 'get_item_permissions_check' ),
                ),
                'schema' => array( $this, 'get_public_item_schema' ),
            )
        );
    }


    /**
     * Forward the experiment request to WP.com and return the WP.com response.
     *
     * @param \WP_REST_Request $request Request data.
     *
     * @return \WP_Error|\WP_REST_Response
     */
    public function get_assignment( $request ) {
        $args = $request->get_query_params();

        if ( ! isset( $args['experiment_name'] ) ) {
            return new \WP_Error(
                'woocommerce_rest_experiment_name_required',
                __( 'Sorry, experiment_name is required.', 'woocommerce' ),
                array( 'status' => 400 )
            );
        }

        unset( $args['rest_route'] );

        $abtest   = new \WooCommerce\Admin\Experimental_Abtest(
            $request->get_param( 'anon_id' ) ?? '',
            'woocommerce',
            true, // set consent to true here since frontend has checked it already.
            true  // set true to send request as auth user.
        );
        $response = $abtest->request_assignment( $args );
        if ( is_wp_error( $response ) ) {
            return $response;
        }

        return json_decode( $response['body'], true );
    }
}