543 lines
18 KiB
PHP
543 lines
18 KiB
PHP
<?php
|
|
/**
|
|
* REST API Coupons controller
|
|
*
|
|
* Handles requests to the /coupons endpoint.
|
|
*
|
|
* @author WooThemes
|
|
* @category API
|
|
* @package WooCommerce/API
|
|
* @since 2.6.0
|
|
*/
|
|
|
|
if ( ! defined( 'ABSPATH' ) ) {
|
|
exit;
|
|
}
|
|
|
|
/**
|
|
* REST API Coupons controller class.
|
|
*
|
|
* @package WooCommerce/API
|
|
* @extends WC_REST_Posts_Controller
|
|
*/
|
|
class WC_REST_Coupons_Controller extends WC_REST_Posts_Controller {
|
|
|
|
/**
|
|
* Endpoint namespace.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $namespace = 'wc/v1';
|
|
|
|
/**
|
|
* Route base.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $rest_base = 'coupons';
|
|
|
|
/**
|
|
* Post type.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $post_type = 'shop_coupon';
|
|
|
|
/**
|
|
* Order refunds actions.
|
|
*/
|
|
public function __construct() {
|
|
add_filter( "woocommerce_rest_{$this->post_type}_query", array( $this, 'query_args' ), 10, 2 );
|
|
}
|
|
|
|
/**
|
|
* Register the routes for coupons.
|
|
*/
|
|
public function register_routes() {
|
|
register_rest_route( $this->namespace, '/' . $this->rest_base, array(
|
|
array(
|
|
'methods' => WP_REST_Server::READABLE,
|
|
'callback' => array( $this, 'get_items' ),
|
|
'permission_callback' => array( $this, 'get_items_permissions_check' ),
|
|
'args' => $this->get_collection_params(),
|
|
),
|
|
array(
|
|
'methods' => WP_REST_Server::CREATABLE,
|
|
'callback' => array( $this, 'create_item' ),
|
|
'permission_callback' => array( $this, 'create_item_permissions_check' ),
|
|
'args' => array_merge( $this->get_endpoint_args_for_item_schema( WP_REST_Server::CREATABLE ), array(
|
|
'code' => array(
|
|
'required' => true,
|
|
),
|
|
) ),
|
|
),
|
|
'schema' => array( $this, 'get_public_item_schema' ),
|
|
) );
|
|
|
|
register_rest_route( $this->namespace, '/' . $this->rest_base . '/(?P<id>[\d]+)', array(
|
|
array(
|
|
'methods' => WP_REST_Server::READABLE,
|
|
'callback' => array( $this, 'get_item' ),
|
|
'permission_callback' => array( $this, 'get_item_permissions_check' ),
|
|
'args' => array(
|
|
'context' => $this->get_context_param( array( 'default' => 'view' ) ),
|
|
),
|
|
),
|
|
array(
|
|
'methods' => WP_REST_Server::EDITABLE,
|
|
'callback' => array( $this, 'update_item' ),
|
|
'permission_callback' => array( $this, 'update_item_permissions_check' ),
|
|
'args' => $this->get_endpoint_args_for_item_schema( WP_REST_Server::EDITABLE ),
|
|
),
|
|
array(
|
|
'methods' => WP_REST_Server::DELETABLE,
|
|
'callback' => array( $this, 'delete_item' ),
|
|
'permission_callback' => array( $this, 'delete_item_permissions_check' ),
|
|
'args' => array(
|
|
'force' => array(
|
|
'default' => false,
|
|
'description' => __( 'Whether to bypass trash and force deletion.', 'woocommerce' ),
|
|
),
|
|
),
|
|
),
|
|
'schema' => array( $this, 'get_public_item_schema' ),
|
|
) );
|
|
|
|
register_rest_route( $this->namespace, '/' . $this->rest_base . '/batch', array(
|
|
array(
|
|
'methods' => WP_REST_Server::EDITABLE,
|
|
'callback' => array( $this, 'batch_items' ),
|
|
'permission_callback' => array( $this, 'batch_items_permissions_check' ),
|
|
'args' => $this->get_endpoint_args_for_item_schema( WP_REST_Server::EDITABLE ),
|
|
),
|
|
'schema' => array( $this, 'get_public_batch_schema' ),
|
|
) );
|
|
}
|
|
|
|
/**
|
|
* Query args.
|
|
*
|
|
* @param array $args
|
|
* @param WP_REST_Request $request
|
|
* @return array
|
|
*/
|
|
public function query_args( $args, $request ) {
|
|
global $wpdb;
|
|
|
|
if ( ! empty( $request['code'] ) ) {
|
|
$id = $wpdb->get_var( $wpdb->prepare( "SELECT id FROM $wpdb->posts WHERE post_title = %s AND post_type = 'shop_coupon' AND post_status = 'publish'", $request['code'] ) );
|
|
$args['post__in'] = array( $id );
|
|
}
|
|
|
|
return $args;
|
|
}
|
|
|
|
/**
|
|
* Prepare a single coupon output for response.
|
|
*
|
|
* @param WP_Post $post Post object.
|
|
* @param WP_REST_Request $request Request object.
|
|
* @return WP_REST_Response $data
|
|
*/
|
|
public function prepare_item_for_response( $post, $request ) {
|
|
$code = wc_get_coupon_code_by_id( $post->ID );
|
|
$coupon = new WC_Coupon( $code );
|
|
$data = $coupon->get_data();
|
|
$format_decimal = array( 'amount', 'minimum_amount', 'maximum_amount' );
|
|
$format_date = array( 'date_created', 'date_modified', 'date_expires' );
|
|
$format_null = array( 'usage_limit', 'usage_limit_per_user', 'limit_usage_to_x_items' );
|
|
|
|
// Format decimal values.
|
|
foreach ( $format_decimal as $key ) {
|
|
$data[ $key ] = wc_format_decimal( $data[ $key ], 2 );
|
|
}
|
|
|
|
// Format date values.
|
|
foreach ( $format_date as $key ) {
|
|
$data[ $key ] = $data[ $key ] ? wc_rest_prepare_date_response( get_gmt_from_date( date( 'Y-m-d H:i:s', $data[ $key ] ) ) ) : null;
|
|
}
|
|
|
|
// Format null values.
|
|
foreach ( $format_null as $key ) {
|
|
$data[ $key ] = $data[ $key ] ? $data[ $key ] : null;
|
|
}
|
|
|
|
$context = ! empty( $request['context'] ) ? $request['context'] : 'view';
|
|
$data = $this->add_additional_fields_to_object( $data, $request );
|
|
$data = $this->filter_response_by_context( $data, $context );
|
|
$response = rest_ensure_response( $data );
|
|
$response->add_links( $this->prepare_links( $post, $request ) );
|
|
|
|
/**
|
|
* Filter the data for a response.
|
|
*
|
|
* The dynamic portion of the hook name, $this->post_type, refers to post_type of the post being
|
|
* prepared for the response.
|
|
*
|
|
* @param WP_REST_Response $response The response object.
|
|
* @param WP_Post $post Post object.
|
|
* @param WP_REST_Request $request Request object.
|
|
*/
|
|
return apply_filters( "woocommerce_rest_prepare_{$this->post_type}", $response, $post, $request );
|
|
}
|
|
|
|
/**
|
|
* Only reutrn writeable props from schema.
|
|
* @param array $schema
|
|
* @return bool
|
|
*/
|
|
protected function filter_writable_props( $schema ) {
|
|
return empty( $schema['readonly'] );
|
|
}
|
|
|
|
/**
|
|
* Prepare a single coupon for create or update.
|
|
*
|
|
* @param WP_REST_Request $request Request object.
|
|
* @return WP_Error|stdClass $data Post object.
|
|
*/
|
|
protected function prepare_item_for_database( $request ) {
|
|
global $wpdb;
|
|
|
|
$id = isset( $request['id'] ) ? absint( $request['id'] ) : 0;
|
|
$coupon = new WC_Coupon( $id );
|
|
$schema = $this->get_item_schema();
|
|
$data_keys = array_keys( array_filter( $schema['properties'], array( $this, 'filter_writable_props' ) ) );
|
|
|
|
// BW compat
|
|
if ( $request['exclude_product_ids'] ) {
|
|
$request['excluded_product_ids'] = $request['exclude_product_ids'];
|
|
}
|
|
if ( $request['expiry_date'] ) {
|
|
$request['date_expires'] = $request['expiry_date'];
|
|
}
|
|
|
|
// Validate required POST fields.
|
|
if ( 'POST' === $request->get_method() && 0 === $coupon->get_id() ) {
|
|
if ( empty( $request['code'] ) ) {
|
|
return new WP_Error( 'woocommerce_rest_empty_coupon_code', sprintf( __( 'The coupon code cannot be empty.', 'woocommerce' ), 'code' ), array( 'status' => 400 ) );
|
|
}
|
|
}
|
|
|
|
// Handle all writable props
|
|
foreach ( $data_keys as $key ) {
|
|
$value = $request[ $key ];
|
|
|
|
if ( ! is_null( $value ) ) {
|
|
switch ( $key ) {
|
|
case 'code' :
|
|
$coupon_code = apply_filters( 'woocommerce_coupon_code', $value );
|
|
$id = $coupon->get_id() ? $coupon->get_id() : 0;
|
|
$id_from_code = wc_get_coupon_id_by_code( $coupon_code, $id );
|
|
|
|
if ( $id_from_code ) {
|
|
return new WP_Error( 'woocommerce_rest_coupon_code_already_exists', __( 'The coupon code already exists', 'woocommerce' ), array( 'status' => 400 ) );
|
|
}
|
|
|
|
$coupon->set_code( $coupon_code );
|
|
break;
|
|
case 'meta_data' :
|
|
if ( is_array( $value ) ) {
|
|
foreach ( $value as $meta ) {
|
|
$coupon->update_meta_data( $meta['key'], $meta['value'], $meta['id'] );
|
|
}
|
|
}
|
|
break;
|
|
case 'description' :
|
|
$coupon->set_description( wp_filter_post_kses( $value ) );
|
|
break;
|
|
default :
|
|
if ( is_callable( array( $coupon, "set_{$key}" ) ) ) {
|
|
$coupon->{"set_{$key}"}( $value );
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Filter the query_vars used in `get_items` for the constructed query.
|
|
*
|
|
* The dynamic portion of the hook name, $this->post_type, refers to post_type of the post being
|
|
* prepared for insertion.
|
|
*
|
|
* @param WC_Coupon $coupon The coupon object.
|
|
* @param WP_REST_Request $request Request object.
|
|
*/
|
|
return apply_filters( "woocommerce_rest_pre_insert_{$this->post_type}", $coupon, $request );
|
|
}
|
|
|
|
/**
|
|
* Create a single item.
|
|
*
|
|
* @param WP_REST_Request $request Full details about the request.
|
|
* @return WP_Error|WP_REST_Response
|
|
*/
|
|
public function create_item( $request ) {
|
|
if ( ! empty( $request['id'] ) ) {
|
|
/* translators: %s: post type */
|
|
return new WP_Error( "woocommerce_rest_{$this->post_type}_exists", sprintf( __( 'Cannot create existing %s.', 'woocommerce' ), $this->post_type ), array( 'status' => 400 ) );
|
|
}
|
|
|
|
$coupon_id = $this->save_coupon( $request );
|
|
if ( is_wp_error( $coupon_id ) ) {
|
|
return $coupon_id;
|
|
}
|
|
|
|
$post = get_post( $coupon_id );
|
|
$this->update_additional_fields_for_object( $post, $request );
|
|
|
|
$this->add_post_meta_fields( $post, $request );
|
|
|
|
/**
|
|
* Fires after a single item is created or updated via the REST API.
|
|
*
|
|
* @param object $post Inserted object (not a WP_Post object).
|
|
* @param WP_REST_Request $request Request object.
|
|
* @param boolean $creating True when creating item, false when updating.
|
|
*/
|
|
do_action( "woocommerce_rest_insert_{$this->post_type}", $post, $request, true );
|
|
$request->set_param( 'context', 'edit' );
|
|
$response = $this->prepare_item_for_response( $post, $request );
|
|
$response = rest_ensure_response( $response );
|
|
$response->set_status( 201 );
|
|
$response->header( 'Location', rest_url( sprintf( '/%s/%s/%d', $this->namespace, $this->rest_base, $post->ID ) ) );
|
|
|
|
return $response;
|
|
}
|
|
|
|
/**
|
|
* Update a single coupon.
|
|
*
|
|
* @param WP_REST_Request $request Full details about the request.
|
|
* @return WP_Error|WP_REST_Response
|
|
*/
|
|
public function update_item( $request ) {
|
|
try {
|
|
$post_id = (int) $request['id'];
|
|
|
|
if ( empty( $post_id ) || get_post_type( $post_id ) !== $this->post_type ) {
|
|
return new WP_Error( "woocommerce_rest_{$this->post_type}_invalid_id", __( 'ID is invalid.', 'woocommerce' ), array( 'status' => 400 ) );
|
|
}
|
|
|
|
$coupon_id = $this->save_coupon( $request );
|
|
if ( is_wp_error( $coupon_id ) ) {
|
|
return $coupon_id;
|
|
}
|
|
|
|
$post = get_post( $coupon_id );
|
|
$this->update_additional_fields_for_object( $post, $request );
|
|
|
|
/**
|
|
* Fires after a single item is created or updated via the REST API.
|
|
*
|
|
* @param object $post Inserted object (not a WP_Post object).
|
|
* @param WP_REST_Request $request Request object.
|
|
* @param boolean $creating True when creating item, false when updating.
|
|
*/
|
|
do_action( "woocommerce_rest_insert_{$this->post_type}", $post, $request, false );
|
|
$request->set_param( 'context', 'edit' );
|
|
$response = $this->prepare_item_for_response( $post, $request );
|
|
return rest_ensure_response( $response );
|
|
|
|
} catch ( Exception $e ) {
|
|
return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) );
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Saves a coupon to the database.
|
|
*/
|
|
public function save_coupon( $request ) {
|
|
try {
|
|
$coupon = $this->prepare_item_for_database( $request );
|
|
$coupon->save();
|
|
return $coupon->get_id();
|
|
} catch ( WC_Data_Exception $e ) {
|
|
return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) );
|
|
} catch ( WC_REST_Exception $e ) {
|
|
return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) );
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get the Coupon's schema, conforming to JSON Schema.
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_item_schema() {
|
|
$schema = array(
|
|
'$schema' => 'http://json-schema.org/draft-04/schema#',
|
|
'title' => $this->post_type,
|
|
'type' => 'object',
|
|
'properties' => array(
|
|
'id' => array(
|
|
'description' => __( 'Unique identifier for the object.', 'woocommerce' ),
|
|
'type' => 'integer',
|
|
'context' => array( 'view', 'edit' ),
|
|
'readonly' => true,
|
|
),
|
|
'code' => array(
|
|
'description' => __( 'Coupon code.', 'woocommerce' ),
|
|
'type' => 'string',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'date_created' => array(
|
|
'description' => __( "The date the coupon was created, in the site's timezone.", 'woocommerce' ),
|
|
'type' => 'date-time',
|
|
'context' => array( 'view', 'edit' ),
|
|
'readonly' => true,
|
|
),
|
|
'date_modified' => array(
|
|
'description' => __( "The date the coupon was last modified, in the site's timezone.", 'woocommerce' ),
|
|
'type' => 'date-time',
|
|
'context' => array( 'view', 'edit' ),
|
|
'readonly' => true,
|
|
),
|
|
'description' => array(
|
|
'description' => __( 'Coupon description.', 'woocommerce' ),
|
|
'type' => 'string',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'discount_type' => array(
|
|
'description' => __( 'Determines the type of discount that will be applied.', 'woocommerce' ),
|
|
'type' => 'string',
|
|
'default' => 'fixed_cart',
|
|
'enum' => array_keys( wc_get_coupon_types() ),
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'amount' => array(
|
|
'description' => __( 'The amount of discount.', 'woocommerce' ),
|
|
'type' => 'string',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'date_expires' => array(
|
|
'description' => __( 'UTC DateTime when the coupon expires.', 'woocommerce' ),
|
|
'type' => 'string',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'usage_count' => array(
|
|
'description' => __( 'Number of times the coupon has been used already.', 'woocommerce' ),
|
|
'type' => 'integer',
|
|
'context' => array( 'view', 'edit' ),
|
|
'readonly' => true,
|
|
),
|
|
'individual_use' => array(
|
|
'description' => __( 'Whether coupon can only be used individually.', 'woocommerce' ),
|
|
'type' => 'boolean',
|
|
'default' => false,
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'product_ids' => array(
|
|
'description' => __( "List of product ID's the coupon can be used on.", 'woocommerce' ),
|
|
'type' => 'array',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'excluded_product_ids' => array(
|
|
'description' => __( "List of product ID's the coupon cannot be used on.", 'woocommerce' ),
|
|
'type' => 'array',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'usage_limit' => array(
|
|
'description' => __( 'How many times the coupon can be used.', 'woocommerce' ),
|
|
'type' => 'integer',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'usage_limit_per_user' => array(
|
|
'description' => __( 'How many times the coupon can be used per customer.', 'woocommerce' ),
|
|
'type' => 'integer',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'limit_usage_to_x_items' => array(
|
|
'description' => __( 'Max number of items in the cart the coupon can be applied to.', 'woocommerce' ),
|
|
'type' => 'integer',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'free_shipping' => array(
|
|
'description' => __( 'Define if can be applied for free shipping.', 'woocommerce' ),
|
|
'type' => 'boolean',
|
|
'default' => false,
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'product_categories' => array(
|
|
'description' => __( "List of category ID's the coupon applies to.", 'woocommerce' ),
|
|
'type' => 'array',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'excluded_product_categories' => array(
|
|
'description' => __( "List of category ID's the coupon does not apply to.", 'woocommerce' ),
|
|
'type' => 'array',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'exclude_sale_items' => array(
|
|
'description' => __( 'Define if should not apply when have sale items.', 'woocommerce' ),
|
|
'type' => 'boolean',
|
|
'default' => false,
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'minimum_amount' => array(
|
|
'description' => __( 'Minimum order amount that needs to be in the cart before coupon applies.', 'woocommerce' ),
|
|
'type' => 'string',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'maximum_amount' => array(
|
|
'description' => __( 'Maximum order amount allowed when using the coupon.', 'woocommerce' ),
|
|
'type' => 'string',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'email_restrictions' => array(
|
|
'description' => __( 'List of email addresses that can use this coupon.', 'woocommerce' ),
|
|
'type' => 'array',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'used_by' => array(
|
|
'description' => __( 'List of user IDs who have used the coupon.', 'woocommerce' ),
|
|
'type' => 'array',
|
|
'context' => array( 'view', 'edit' ),
|
|
'readonly' => true,
|
|
),
|
|
'meta_data' => array(
|
|
'description' => __( 'Order meta data.', 'woocommerce' ),
|
|
'type' => 'array',
|
|
'context' => array( 'view', 'edit' ),
|
|
'properties' => array(
|
|
'id' => array(
|
|
'description' => __( 'Meta ID.', 'woocommerce' ),
|
|
'type' => 'int',
|
|
'context' => array( 'view', 'edit' ),
|
|
'readonly' => true,
|
|
),
|
|
'key' => array(
|
|
'description' => __( 'Meta key.', 'woocommerce' ),
|
|
'type' => 'string',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
'value' => array(
|
|
'description' => __( 'Meta value.', 'woocommerce' ),
|
|
'type' => 'string',
|
|
'context' => array( 'view', 'edit' ),
|
|
),
|
|
),
|
|
),
|
|
),
|
|
);
|
|
return $this->add_additional_fields_schema( $schema );
|
|
}
|
|
|
|
/**
|
|
* Get the query params for collections of attachments.
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_collection_params() {
|
|
$params = parent::get_collection_params();
|
|
$params['code'] = array(
|
|
'description' => __( 'Limit result set to resources with a specific code.', 'woocommerce' ),
|
|
'type' => 'string',
|
|
'sanitize_callback' => 'sanitize_text_field',
|
|
'validate_callback' => 'rest_validate_request_arg',
|
|
);
|
|
return $params;
|
|
}
|
|
}
|