woothemes/woocommerce

View on GitHub
includes/rest-api/Controllers/Version1/class-wc-rest-customer-downloads-v1-controller.php

Summary

Maintainability
C
7 hrs
Test Coverage
<?php
/**
 * REST API Customer Downloads controller
 *
 * Handles requests to the /customers/<customer_id>/downloads endpoint.
 *
 * @author   WooThemes
 * @category API
 * @package WooCommerce\RestApi
 * @since    3.0.0
 */

if ( ! defined( 'ABSPATH' ) ) {
    exit;
}

/**
 * REST API Customers controller class.
 *
 * @package WooCommerce\RestApi
 * @extends WC_REST_Controller
 */
class WC_REST_Customer_Downloads_V1_Controller extends WC_REST_Controller {

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

    /**
     * Route base.
     *
     * @var string
     */
    protected $rest_base = 'customers/(?P<customer_id>[\d]+)/downloads';

    /**
     * Register the routes for customers.
     */
    public function register_routes() {
        register_rest_route( $this->namespace, '/' . $this->rest_base, array(
            'args' => array(
                'customer_id' => array(
                    'description' => __( 'Unique identifier for the resource.', 'woocommerce' ),
                    'type'        => 'integer',
                ),
            ),
            array(
                'methods'             => WP_REST_Server::READABLE,
                'callback'            => array( $this, 'get_items' ),
                'permission_callback' => array( $this, 'get_items_permissions_check' ),
                'args'                => $this->get_collection_params(),
            ),
            'schema' => array( $this, 'get_public_item_schema' ),
        ) );
    }

    /**
     * Check whether a given request has permission to read customers.
     *
     * @param  WP_REST_Request $request Full details about the request.
     * @return WP_Error|boolean
     */
    public function get_items_permissions_check( $request ) {
        $customer = get_user_by( 'id', (int) $request['customer_id'] );

        if ( ! $customer ) {
            return new WP_Error( 'woocommerce_rest_customer_invalid', __( 'Resource does not exist.', 'woocommerce' ), array( 'status' => 404 ) );
        }

        if ( ! wc_rest_check_user_permissions( 'read', $customer->get_id() ) ) {
            return new WP_Error( 'woocommerce_rest_cannot_view', __( 'Sorry, you cannot list resources.', 'woocommerce' ), array( 'status' => rest_authorization_required_code() ) );
        }

        return true;
    }

    /**
     * Get all customer downloads.
     *
     * @param WP_REST_Request $request
     * @return array
     */
    public function get_items( $request ) {
        $downloads = wc_get_customer_available_downloads( (int) $request['customer_id'] );

        $data = array();
        foreach ( $downloads as $download_data ) {
            $download = $this->prepare_item_for_response( (object) $download_data, $request );
            $download = $this->prepare_response_for_collection( $download );
            $data[]   = $download;
        }

        return rest_ensure_response( $data );
    }

    /**
     * Prepare a single download output for response.
     *
     * @param stdObject $download Download object.
     * @param WP_REST_Request $request Request object.
     * @return WP_REST_Response $response Response data.
     */
    public function prepare_item_for_response( $download, $request ) {
        $data = (array) $download;
        $data['access_expires']      = $data['access_expires'] ? wc_rest_prepare_date_response( $data['access_expires'] ) : 'never';
        $data['downloads_remaining'] = '' === $data['downloads_remaining'] ? 'unlimited' : $data['downloads_remaining'];

        // Remove "product_name" since it's new in 3.0.
        unset( $data['product_name'] );

        $context = ! empty( $request['context'] ) ? $request['context'] : 'view';
        $data    = $this->add_additional_fields_to_object( $data, $request );
        $data    = $this->filter_response_by_context( $data, $context );

        // Wrap the data in a response object.
        $response = rest_ensure_response( $data );

        $response->add_links( $this->prepare_links( $download, $request ) );

        /**
         * Filter customer download data returned from the REST API.
         *
         * @param WP_REST_Response $response  The response object.
         * @param stdObject        $download  Download object used to create response.
         * @param WP_REST_Request  $request   Request object.
         */
        return apply_filters( 'woocommerce_rest_prepare_customer_download', $response, $download, $request );
    }

    /**
     * Prepare links for the request.
     *
     * @param stdClass $download Download object.
     * @param WP_REST_Request $request Request object.
     * @return array Links for the given customer download.
     */
    protected function prepare_links( $download, $request ) {
        $base  = str_replace( '(?P<customer_id>[\d]+)', $request['customer_id'], $this->rest_base );
        $links = array(
            'collection' => array(
                'href' => rest_url( sprintf( '/%s/%s', $this->namespace, $base ) ),
            ),
            'product' => array(
                'href' => rest_url( sprintf( '/%s/products/%d', $this->namespace, $download->product_id ) ),
            ),
            'order' => array(
                'href' => rest_url( sprintf( '/%s/orders/%d', $this->namespace, $download->order_id ) ),
            ),
        );

        return $links;
    }

    /**
     * Get the Customer Download's schema, conforming to JSON Schema.
     *
     * @return array
     */
    public function get_item_schema() {
        $schema = array(
            '$schema'    => 'http://json-schema.org/draft-04/schema#',
            'title'      => 'customer_download',
            'type'       => 'object',
            'properties' => array(
                'download_url' => array(
                    'description' => __( 'Download file URL.', 'woocommerce' ),
                    'type'        => 'string',
                    'context'     => array( 'view' ),
                    'readonly'    => true,
                ),
                'download_id' => array(
                    'description' => __( 'Download ID (MD5).', 'woocommerce' ),
                    'type'        => 'string',
                    'context'     => array( 'view' ),
                    'readonly'    => true,
                ),
                'product_id' => array(
                    'description' => __( 'Downloadable product ID.', 'woocommerce' ),
                    'type'        => 'integer',
                    'context'     => array( 'view' ),
                    'readonly'    => true,
                ),
                'download_name' => array(
                    'description' => __( 'Downloadable file name.', 'woocommerce' ),
                    'type'        => 'string',
                    'context'     => array( 'view' ),
                    'readonly'    => true,
                ),
                'order_id' => array(
                    'description' => __( 'Order ID.', 'woocommerce' ),
                    'type'        => 'integer',
                    'context'     => array( 'view' ),
                    'readonly'    => true,
                ),
                'order_key' => array(
                    'description' => __( 'Order key.', 'woocommerce' ),
                    'type'        => 'string',
                    'context'     => array( 'view' ),
                    'readonly'    => true,
                ),
                'downloads_remaining' => array(
                    'description' => __( 'Number of downloads remaining.', 'woocommerce' ),
                    'type'        => 'string',
                    'context'     => array( 'view' ),
                    'readonly'    => true,
                ),
                'access_expires' => array(
                    'description' => __( "The date when download access expires, in the site's timezone.", 'woocommerce' ),
                    'type'        => 'string',
                    'context'     => array( 'view' ),
                    'readonly'    => true,
                ),
                'file' => array(
                    'description' => __( 'File details.', 'woocommerce' ),
                    'type'        => 'object',
                    'context'     => array( 'view' ),
                    'readonly'    => true,
                    'properties' => array(
                        'name' => array(
                            'description' => __( 'File name.', 'woocommerce' ),
                            'type'        => 'string',
                            'context'     => array( 'view' ),
                            'readonly'    => true,
                        ),
                        'file' => array(
                            'description' => __( 'File URL.', 'woocommerce' ),
                            'type'        => 'string',
                            'context'     => array( 'view' ),
                            'readonly'    => true,
                        ),
                    ),
                ),
            ),
        );

        return $this->add_additional_fields_schema( $schema );
    }

    /**
     * Get the query params for collections.
     *
     * @return array
     */
    public function get_collection_params() {
        return array(
            'context' => $this->get_context_param( array( 'default' => 'view' ) ),
        );
    }
}