WC_API_Taxes

The WooCommerce WC API Taxes class.

Defined (1)

The class is defined in the following location(s).

/includes/api/legacy/v3/class-wc-api-taxes.php  
  1. class WC_API_Taxes extends WC_API_Resource { 
  2.  
  3.     /** @var string $base the route base */ 
  4.     protected $base = '/taxes'; 
  5.  
  6.     /** 
  7.      * Register the routes for this class 
  8.      * 
  9.      * GET /taxes 
  10.      * GET /taxes/count 
  11.      * GET /taxes/<id> 
  12.      * 
  13.      * @since 2.1 
  14.      * @param array $routes 
  15.      * @return array 
  16.      */ 
  17.     public function register_routes( $routes ) { 
  18.  
  19.         # GET/POST /taxes 
  20.         $routes[ $this->base ] = array( 
  21.             array( array( $this, 'get_taxes' ), WC_API_Server::READABLE ),  
  22.             array( array( $this, 'create_tax' ), WC_API_Server::CREATABLE | WC_API_Server::ACCEPT_DATA ),  
  23.  ); 
  24.  
  25.         # GET /taxes/count 
  26.         $routes[ $this->base . '/count' ] = array( 
  27.             array( array( $this, 'get_taxes_count' ), WC_API_Server::READABLE ),  
  28.  ); 
  29.  
  30.         # GET/PUT/DELETE /taxes/<id> 
  31.         $routes[ $this->base . '/(?P<id>\d+)' ] = array( 
  32.             array( array( $this, 'get_tax' ), WC_API_Server::READABLE ),  
  33.             array( array( $this, 'edit_tax' ), WC_API_SERVER::EDITABLE | WC_API_SERVER::ACCEPT_DATA ),  
  34.             array( array( $this, 'delete_tax' ), WC_API_SERVER::DELETABLE ),  
  35.  ); 
  36.  
  37.         # GET/POST /taxes/classes 
  38.         $routes[ $this->base . '/classes' ] = array( 
  39.             array( array( $this, 'get_tax_classes' ), WC_API_Server::READABLE ),  
  40.             array( array( $this, 'create_tax_class' ), WC_API_Server::CREATABLE | WC_API_Server::ACCEPT_DATA ),  
  41.  ); 
  42.  
  43.         # GET /taxes/classes/count 
  44.         $routes[ $this->base . '/classes/count' ] = array( 
  45.             array( array( $this, 'get_tax_classes_count' ), WC_API_Server::READABLE ),  
  46.  ); 
  47.  
  48.         # GET /taxes/classes/<slug> 
  49.         $routes[ $this->base . '/classes/(?P<slug>\w[\w\s\-]*)' ] = array( 
  50.             array( array( $this, 'delete_tax_class' ), WC_API_SERVER::DELETABLE ),  
  51.  ); 
  52.  
  53.         # POST|PUT /taxes/bulk 
  54.         $routes[ $this->base . '/bulk' ] = array( 
  55.             array( array( $this, 'bulk' ), WC_API_Server::EDITABLE | WC_API_Server::ACCEPT_DATA ),  
  56.  ); 
  57.  
  58.         return $routes; 
  59.     } 
  60.  
  61.     /** 
  62.      * Get all taxes 
  63.      * 
  64.      * @since 2.5.0 
  65.      * 
  66.      * @param string $fields 
  67.      * @param array  $filter 
  68.      * @param string $class 
  69.      * @param int    $page 
  70.      * 
  71.      * @return array 
  72.      */ 
  73.     public function get_taxes( $fields = null, $filter = array(), $class = null, $page = 1 ) { 
  74.         if ( ! empty( $class ) ) { 
  75.             $filter['tax_rate_class'] = $class; 
  76.         } 
  77.  
  78.         $filter['page'] = $page; 
  79.  
  80.         $query = $this->query_tax_rates( $filter ); 
  81.  
  82.         $taxes = array(); 
  83.  
  84.         foreach ( $query['results'] as $tax ) { 
  85.             $taxes[] = current( $this->get_tax( $tax->tax_rate_id, $fields ) ); 
  86.         } 
  87.  
  88.         // Set pagination headers 
  89.         $this->server->add_pagination_headers( $query['headers'] ); 
  90.  
  91.         return array( 'taxes' => $taxes ); 
  92.     } 
  93.  
  94.     /** 
  95.      * Get the tax for the given ID 
  96.      * 
  97.      * @since 2.5.0 
  98.      * 
  99.      * @param int $id The tax ID 
  100.      * @param string $fields fields to include in response 
  101.      * 
  102.      * @return array|WP_Error 
  103.      */ 
  104.     public function get_tax( $id, $fields = null ) { 
  105.         global $wpdb; 
  106.  
  107.         try { 
  108.             $id = absint( $id ); 
  109.  
  110.             // Permissions check 
  111.             if ( ! current_user_can( 'manage_woocommerce' ) ) { 
  112.                 throw new WC_API_Exception( 'woocommerce_api_user_cannot_read_tax', __( 'You do not have permission to read tax rate', 'woocommerce' ), 401 ); 
  113.             } 
  114.  
  115.             // Get tax rate details 
  116.             $tax = WC_Tax::_get_tax_rate( $id ); 
  117.  
  118.             if ( is_wp_error( $tax ) || empty( $tax ) ) { 
  119.                 throw new WC_API_Exception( 'woocommerce_api_invalid_tax_id', __( 'A tax rate with the provided ID could not be found', 'woocommerce' ), 404 ); 
  120.             } 
  121.  
  122.             $tax_data = array( 
  123.                 'id' => (int) $tax['tax_rate_id'],  
  124.                 'country' => $tax['tax_rate_country'],  
  125.                 'state' => $tax['tax_rate_state'],  
  126.                 'postcode' => '',  
  127.                 'city' => '',  
  128.                 'rate' => $tax['tax_rate'],  
  129.                 'name' => $tax['tax_rate_name'],  
  130.                 'priority' => (int) $tax['tax_rate_priority'],  
  131.                 'compound' => (bool) $tax['tax_rate_compound'],  
  132.                 'shipping' => (bool) $tax['tax_rate_shipping'],  
  133.                 'order' => (int) $tax['tax_rate_order'],  
  134.                 'class' => $tax['tax_rate_class'] ? $tax['tax_rate_class'] : 'standard',  
  135.  ); 
  136.  
  137.             // Get locales from a tax rate 
  138.             $locales = $wpdb->get_results( $wpdb->prepare( " 
  139.                 SELECT location_code, location_type 
  140.                 FROM {$wpdb->prefix}woocommerce_tax_rate_locations 
  141.                 WHERE tax_rate_id = %d 
  142.             ", $id ) ); 
  143.  
  144.             if ( ! is_wp_error( $tax ) && ! is_null( $tax ) ) { 
  145.                 foreach ( $locales as $locale ) { 
  146.                     $tax_data[ $locale->location_type ] = $locale->location_code; 
  147.                 } 
  148.             } 
  149.  
  150.             return array( 'tax' => apply_filters( 'woocommerce_api_tax_response', $tax_data, $tax, $fields, $this ) ); 
  151.         } catch ( WC_API_Exception $e ) { 
  152.             return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) ); 
  153.         } 
  154.     } 
  155.  
  156.     /** 
  157.      * Create a tax 
  158.      * 
  159.      * @since 2.5.0 
  160.      * 
  161.      * @param array $data 
  162.      * 
  163.      * @return array 
  164.      */ 
  165.     public function create_tax( $data ) { 
  166.         try { 
  167.             if ( ! isset( $data['tax'] ) ) { 
  168.                 throw new WC_API_Exception( 'woocommerce_api_missing_tax_data', sprintf( __( 'No %1$s data specified to create %1$s', 'woocommerce' ), 'tax' ), 400 ); 
  169.             } 
  170.  
  171.             // Check permissions 
  172.             if ( ! current_user_can( 'manage_woocommerce' ) ) { 
  173.                 throw new WC_API_Exception( 'woocommerce_api_user_cannot_create_tax', __( 'You do not have permission to create tax rates', 'woocommerce' ), 401 ); 
  174.             } 
  175.  
  176.             $data = apply_filters( 'woocommerce_api_create_tax_data', $data['tax'], $this ); 
  177.  
  178.             $tax_data = array( 
  179.                 'tax_rate_country' => '',  
  180.                 'tax_rate_state' => '',  
  181.                 'tax_rate' => '',  
  182.                 'tax_rate_name' => '',  
  183.                 'tax_rate_priority' => 1,  
  184.                 'tax_rate_compound' => 0,  
  185.                 'tax_rate_shipping' => 1,  
  186.                 'tax_rate_order' => 0,  
  187.                 'tax_rate_class' => '',  
  188.  ); 
  189.  
  190.             foreach ( $tax_data as $key => $value ) { 
  191.                 $new_key = str_replace( 'tax_rate_', '', $key ); 
  192.                 $new_key = 'tax_rate' === $new_key ? 'rate' : $new_key; 
  193.  
  194.                 if ( isset( $data[ $new_key ] ) ) { 
  195.                     if ( in_array( $new_key, array( 'compound', 'shipping' ) ) ) { 
  196.                         $tax_data[ $key ] = $data[ $new_key ] ? 1 : 0; 
  197.                     } else { 
  198.                         $tax_data[ $key ] = $data[ $new_key ]; 
  199.                     } 
  200.                 } 
  201.             } 
  202.  
  203.             // Create tax rate 
  204.             $id = WC_Tax::_insert_tax_rate( $tax_data ); 
  205.  
  206.             // Add locales 
  207.             if ( ! empty( $data['postcode'] ) ) { 
  208.                 WC_Tax::_update_tax_rate_postcodes( $id, wc_clean( $data['postcode'] ) ); 
  209.             } 
  210.  
  211.             if ( ! empty( $data['city'] ) ) { 
  212.                 WC_Tax::_update_tax_rate_cities( $id, wc_clean( $data['city'] ) ); 
  213.             } 
  214.  
  215.             do_action( 'woocommerce_api_create_tax', $id, $data ); 
  216.  
  217.             $this->server->send_status( 201 ); 
  218.  
  219.             return $this->get_tax( $id ); 
  220.         } catch ( WC_API_Exception $e ) { 
  221.             return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) ); 
  222.         } 
  223.     } 
  224.  
  225.     /** 
  226.      * Edit a tax 
  227.      * 
  228.      * @since 2.5.0 
  229.      * 
  230.      * @param int $id The tax ID 
  231.      * @param array $data 
  232.      * 
  233.      * @return array 
  234.      */ 
  235.     public function edit_tax( $id, $data ) { 
  236.         try { 
  237.             if ( ! isset( $data['tax'] ) ) { 
  238.                 throw new WC_API_Exception( 'woocommerce_api_missing_tax_data', sprintf( __( 'No %1$s data specified to edit %1$s', 'woocommerce' ), 'tax' ), 400 ); 
  239.             } 
  240.  
  241.             // Check permissions 
  242.             if ( ! current_user_can( 'manage_woocommerce' ) ) { 
  243.                 throw new WC_API_Exception( 'woocommerce_api_user_cannot_edit_tax', __( 'You do not have permission to edit tax rates', 'woocommerce' ), 401 ); 
  244.             } 
  245.  
  246.             $data = $data['tax']; 
  247.  
  248.             // Get current tax rate data 
  249.             $tax = $this->get_tax( $id ); 
  250.  
  251.             if ( is_wp_error( $tax ) ) { 
  252.                 $error_data = $tax->get_error_data(); 
  253.                 throw new WC_API_Exception( $tax->get_error_code(), $tax->get_error_message(), $error_data['status'] ); 
  254.             } 
  255.  
  256.             $current_data = $tax['tax']; 
  257.             $data = apply_filters( 'woocommerce_api_edit_tax_data', $data, $this ); 
  258.             $tax_data = array(); 
  259.             $default_fields = array( 
  260.                 'tax_rate_country',  
  261.                 'tax_rate_state',  
  262.                 'tax_rate',  
  263.                 'tax_rate_name',  
  264.                 'tax_rate_priority',  
  265.                 'tax_rate_compound',  
  266.                 'tax_rate_shipping',  
  267.                 'tax_rate_order',  
  268.                 'tax_rate_class',  
  269.  ); 
  270.  
  271.             foreach ( $data as $key => $value ) { 
  272.                 $new_key = 'rate' === $key ? 'tax_rate' : 'tax_rate_' . $key; 
  273.  
  274.                 // Check if the key is valid 
  275.                 if ( ! in_array( $new_key, $default_fields ) ) { 
  276.                     continue; 
  277.                 } 
  278.  
  279.                 // Test new data against current data 
  280.                 if ( $value === $current_data[ $key ] ) { 
  281.                     continue; 
  282.                 } 
  283.  
  284.                 // Fix compund and shipping values 
  285.                 if ( in_array( $key, array( 'compound', 'shipping' ) ) ) { 
  286.                     $value = $value ? 1 : 0; 
  287.                 } 
  288.  
  289.                 $tax_data[ $new_key ] = $value; 
  290.             } 
  291.  
  292.             // Update tax rate 
  293.             WC_Tax::_update_tax_rate( $id, $tax_data ); 
  294.  
  295.             // Update locales 
  296.             if ( ! empty( $data['postcode'] ) && $current_data['postcode'] != $data['postcode'] ) { 
  297.                 WC_Tax::_update_tax_rate_postcodes( $id, wc_clean( $data['postcode'] ) ); 
  298.             } 
  299.  
  300.             if ( ! empty( $data['city'] ) && $current_data['city'] != $data['city'] ) { 
  301.                 WC_Tax::_update_tax_rate_cities( $id, wc_clean( $data['city'] ) ); 
  302.             } 
  303.  
  304.             do_action( 'woocommerce_api_edit_tax_rate', $id, $data ); 
  305.  
  306.             return $this->get_tax( $id ); 
  307.         } catch ( WC_API_Exception $e ) { 
  308.             return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) ); 
  309.         } 
  310.     } 
  311.  
  312.     /** 
  313.      * Delete a tax 
  314.      * 
  315.      * @since 2.5.0 
  316.      * 
  317.      * @param int $id The tax ID 
  318.      * 
  319.      * @return array 
  320.      */ 
  321.     public function delete_tax( $id ) { 
  322.         global $wpdb; 
  323.  
  324.         try { 
  325.             // Check permissions 
  326.             if ( ! current_user_can( 'manage_woocommerce' ) ) { 
  327.                 throw new WC_API_Exception( 'woocommerce_api_user_cannot_delete_tax', __( 'You do not have permission to delete tax rates', 'woocommerce' ), 401 ); 
  328.             } 
  329.  
  330.             $id = absint( $id ); 
  331.  
  332.             WC_Tax::_delete_tax_rate( $id ); 
  333.  
  334.             if ( 0 === $wpdb->rows_affected ) { 
  335.                 throw new WC_API_Exception( 'woocommerce_api_cannot_delete_tax', __( 'Could not delete the tax rate', 'woocommerce' ), 401 ); 
  336.             } 
  337.  
  338.             return array( 'message' => sprintf( __( 'Deleted %s', 'woocommerce' ), 'tax' ) ); 
  339.         } catch ( WC_API_Exception $e ) { 
  340.             return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) ); 
  341.         } 
  342.     } 
  343.  
  344.     /** 
  345.      * Get the total number of taxes 
  346.      * 
  347.      * @since 2.5.0 
  348.      * 
  349.      * @param string $class 
  350.      * @param array $filter 
  351.      * 
  352.      * @return array 
  353.      */ 
  354.     public function get_taxes_count( $class = null, $filter = array() ) { 
  355.         try { 
  356.             if ( ! current_user_can( 'manage_woocommerce' ) ) { 
  357.                 throw new WC_API_Exception( 'woocommerce_api_user_cannot_read_taxes_count', __( 'You do not have permission to read the taxes count', 'woocommerce' ), 401 ); 
  358.             } 
  359.  
  360.             if ( ! empty( $class ) ) { 
  361.                 $filter['tax_rate_class'] = $class; 
  362.             } 
  363.  
  364.             $query = $this->query_tax_rates( $filter, true ); 
  365.  
  366.             return array( 'count' => (int) $query['headers']->total ); 
  367.         } catch ( WC_API_Exception $e ) { 
  368.             return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) ); 
  369.         } 
  370.     } 
  371.  
  372.     /** 
  373.      * Helper method to get tax rates objects 
  374.      * 
  375.      * @since 2.5.0 
  376.      * 
  377.      * @param  array $args 
  378.      * @param  bool  $count_only 
  379.      * 
  380.      * @return array 
  381.      */ 
  382.     protected function query_tax_rates( $args, $count_only = false ) { 
  383.         global $wpdb; 
  384.  
  385.         $results = ''; 
  386.  
  387.         // Set args 
  388.         $args = $this->merge_query_args( $args, array() ); 
  389.  
  390.         $query = " 
  391.             SELECT tax_rate_id 
  392.             FROM {$wpdb->prefix}woocommerce_tax_rates 
  393.             WHERE 1 = 1 
  394.         "; 
  395.  
  396.         // Filter by tax class 
  397.         if ( ! empty( $args['tax_rate_class'] ) ) { 
  398.             $tax_rate_class = 'standard' !== $args['tax_rate_class'] ? sanitize_title( $args['tax_rate_class'] ) : ''; 
  399.             $query .= " AND tax_rate_class = '$tax_rate_class'"; 
  400.         } 
  401.  
  402.         // Order tax rates 
  403.         $order_by = ' ORDER BY tax_rate_order'; 
  404.  
  405.         // Pagination 
  406.         $per_page = isset( $args['posts_per_page'] ) ? $args['posts_per_page'] : get_option( 'posts_per_page' ); 
  407.         $offset = 1 < $args['paged'] ? ( $args['paged'] - 1 ) * $per_page : 0; 
  408.         $pagination = sprintf( ' LIMIT %d, %d', $offset, $per_page ); 
  409.  
  410.         if ( ! $count_only ) { 
  411.             $results = $wpdb->get_results( $query . $order_by . $pagination ); 
  412.         } 
  413.  
  414.         $wpdb->get_results( $query ); 
  415.         $headers = new stdClass; 
  416.         $headers->page = $args['paged']; 
  417.         $headers->total = (int) $wpdb->num_rows; 
  418.         $headers->is_single = $per_page > $headers->total; 
  419.         $headers->total_pages = ceil( $headers->total / $per_page ); 
  420.  
  421.         return array( 
  422.             'results' => $results,  
  423.             'headers' => $headers,  
  424.  ); 
  425.     } 
  426.  
  427.     /** 
  428.      * Bulk update or insert taxes 
  429.      * Accepts an array with taxes in the formats supported by 
  430.      * WC_API_Taxes->create_tax() and WC_API_Taxes->edit_tax() 
  431.      * 
  432.      * @since 2.5.0 
  433.      * 
  434.      * @param array $data 
  435.      * 
  436.      * @return array 
  437.      */ 
  438.     public function bulk( $data ) { 
  439.         try { 
  440.             if ( ! isset( $data['taxes'] ) ) { 
  441.                 throw new WC_API_Exception( 'woocommerce_api_missing_taxes_data', sprintf( __( 'No %1$s data specified to create/edit %1$s', 'woocommerce' ), 'taxes' ), 400 ); 
  442.             } 
  443.  
  444.             $data = $data['taxes']; 
  445.             $limit = apply_filters( 'woocommerce_api_bulk_limit', 100, 'taxes' ); 
  446.  
  447.             // Limit bulk operation 
  448.             if ( count( $data ) > $limit ) { 
  449.                 throw new WC_API_Exception( 'woocommerce_api_taxes_request_entity_too_large', sprintf( __( 'Unable to accept more than %s items for this request.', 'woocommerce' ), $limit ), 413 ); 
  450.             } 
  451.  
  452.             $taxes = array(); 
  453.  
  454.             foreach ( $data as $_tax ) { 
  455.                 $tax_id = 0; 
  456.  
  457.                 // Try to get the tax rate ID 
  458.                 if ( isset( $_tax['id'] ) ) { 
  459.                     $tax_id = intval( $_tax['id'] ); 
  460.                 } 
  461.  
  462.                 if ( $tax_id ) { 
  463.  
  464.                     // Tax rate exists / edit tax rate 
  465.                     $edit = $this->edit_tax( $tax_id, array( 'tax' => $_tax ) ); 
  466.  
  467.                     if ( is_wp_error( $edit ) ) { 
  468.                         $taxes[] = array( 
  469.                             'id' => $tax_id,  
  470.                             'error' => array( 'code' => $edit->get_error_code(), 'message' => $edit->get_error_message() ),  
  471.  ); 
  472.                     } else { 
  473.                         $taxes[] = $edit['tax']; 
  474.                     } 
  475.                 } else { 
  476.  
  477.                     // Tax rate don't exists / create tax rate 
  478.                     $new = $this->create_tax( array( 'tax' => $_tax ) ); 
  479.  
  480.                     if ( is_wp_error( $new ) ) { 
  481.                         $taxes[] = array( 
  482.                             'id' => $tax_id,  
  483.                             'error' => array( 'code' => $new->get_error_code(), 'message' => $new->get_error_message() ),  
  484.  ); 
  485.                     } else { 
  486.                         $taxes[] = $new['tax']; 
  487.                     } 
  488.                 } 
  489.             } 
  490.  
  491.             return array( 'taxes' => apply_filters( 'woocommerce_api_taxes_bulk_response', $taxes, $this ) ); 
  492.         } catch ( WC_API_Exception $e ) { 
  493.             return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) ); 
  494.         } 
  495.     } 
  496.  
  497.     /** 
  498.      * Get all tax classes 
  499.      * 
  500.      * @since 2.5.0 
  501.      * 
  502.      * @param string $fields 
  503.      * 
  504.      * @return array 
  505.      */ 
  506.     public function get_tax_classes( $fields = null ) { 
  507.         try { 
  508.             // Permissions check 
  509.             if ( ! current_user_can( 'manage_woocommerce' ) ) { 
  510.                 throw new WC_API_Exception( 'woocommerce_api_user_cannot_read_tax_classes', __( 'You do not have permission to read tax classes', 'woocommerce' ), 401 ); 
  511.             } 
  512.  
  513.             $tax_classes = array(); 
  514.  
  515.             // Add standard class 
  516.             $tax_classes[] = array( 
  517.                 'slug' => 'standard',  
  518.                 'name' => __( 'Standard rate', 'woocommerce' ),  
  519.  ); 
  520.  
  521.             $classes = WC_Tax::get_tax_classes(); 
  522.  
  523.             foreach ( $classes as $class ) { 
  524.                 $tax_classes[] = apply_filters( 'woocommerce_api_tax_class_response', array( 
  525.                     'slug' => sanitize_title( $class ),  
  526.                     'name' => $class,  
  527.  ), $class, $fields, $this ); 
  528.             } 
  529.  
  530.             return array( 'tax_classes' => apply_filters( 'woocommerce_api_tax_classes_response', $tax_classes, $classes, $fields, $this ) ); 
  531.         } catch ( WC_API_Exception $e ) { 
  532.             return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) ); 
  533.         } 
  534.     } 
  535.  
  536.     /** 
  537.      * Create a tax class. 
  538.      * 
  539.      * @since 2.5.0 
  540.      * 
  541.      * @param array $data 
  542.      * 
  543.      * @return array 
  544.      */ 
  545.     public function create_tax_class( $data ) { 
  546.         try { 
  547.             if ( ! isset( $data['tax_class'] ) ) { 
  548.                 throw new WC_API_Exception( 'woocommerce_api_missing_tax_class_data', sprintf( __( 'No %1$s data specified to create %1$s', 'woocommerce' ), 'tax_class' ), 400 ); 
  549.             } 
  550.  
  551.             // Check permissions 
  552.             if ( ! current_user_can( 'manage_woocommerce' ) ) { 
  553.                 throw new WC_API_Exception( 'woocommerce_api_user_cannot_create_tax_class', __( 'You do not have permission to create tax classes', 'woocommerce' ), 401 ); 
  554.             } 
  555.  
  556.             $data = $data['tax_class']; 
  557.  
  558.             if ( empty( $data['name'] ) ) { 
  559.                 throw new WC_API_Exception( 'woocommerce_api_missing_tax_class_name', sprintf( __( 'Missing parameter %s', 'woocommerce' ), 'name' ), 400 ); 
  560.             } 
  561.  
  562.             $name = sanitize_text_field( $data['name'] ); 
  563.             $slug = sanitize_title( $name ); 
  564.             $classes = WC_Tax::get_tax_classes(); 
  565.             $exists = false; 
  566.  
  567.             // Check if class exists. 
  568.             foreach ( $classes as $key => $class ) { 
  569.                 if ( sanitize_title( $class ) === $slug ) { 
  570.                     $exists = true; 
  571.                     break; 
  572.                 } 
  573.             } 
  574.  
  575.             // Return error if tax class already exists. 
  576.             if ( $exists ) { 
  577.                 throw new WC_API_Exception( 'woocommerce_api_cannot_create_tax_class', __( 'Tax class already exists', 'woocommerce' ), 401 ); 
  578.             } 
  579.  
  580.             // Add the new class. 
  581.             $classes[] = $name; 
  582.  
  583.             update_option( 'woocommerce_tax_classes', implode( "\n", $classes ) ); 
  584.  
  585.             do_action( 'woocommerce_api_create_tax_class', $slug, $data ); 
  586.  
  587.             $this->server->send_status( 201 ); 
  588.  
  589.             return array( 
  590.                 'tax_class' => array( 
  591.                     'slug' => $slug,  
  592.                     'name' => $name,  
  593.  ),  
  594.  ); 
  595.         } catch ( WC_API_Exception $e ) { 
  596.             return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) ); 
  597.         } 
  598.     } 
  599.  
  600.     /** 
  601.      * Delete a tax class 
  602.      * 
  603.      * @since 2.5.0 
  604.      * 
  605.      * @param int $slug The tax class slug 
  606.      * 
  607.      * @return array 
  608.      */ 
  609.     public function delete_tax_class( $slug ) { 
  610.         global $wpdb; 
  611.  
  612.         try { 
  613.             // Check permissions 
  614.             if ( ! current_user_can( 'manage_woocommerce' ) ) { 
  615.                 throw new WC_API_Exception( 'woocommerce_api_user_cannot_delete_tax_class', __( 'You do not have permission to delete tax classes', 'woocommerce' ), 401 ); 
  616.             } 
  617.  
  618.             $slug = sanitize_title( $slug ); 
  619.             $classes = WC_Tax::get_tax_classes(); 
  620.             $deleted = false; 
  621.  
  622.             foreach ( $classes as $key => $class ) { 
  623.                 if ( sanitize_title( $class ) === $slug ) { 
  624.                     unset( $classes[ $key ] ); 
  625.                     $deleted = true; 
  626.                     break; 
  627.                 } 
  628.             } 
  629.  
  630.             if ( ! $deleted ) { 
  631.                 throw new WC_API_Exception( 'woocommerce_api_cannot_delete_tax_class', __( 'Could not delete the tax class', 'woocommerce' ), 401 ); 
  632.             } 
  633.  
  634.             update_option( 'woocommerce_tax_classes', implode( "\n", $classes ) ); 
  635.  
  636.             // Delete tax rate locations locations from the selected class. 
  637.             $wpdb->query( $wpdb->prepare( " 
  638.                 DELETE locations.* 
  639.                 FROM {$wpdb->prefix}woocommerce_tax_rate_locations AS locations 
  640.                 INNER JOIN 
  641.                     {$wpdb->prefix}woocommerce_tax_rates AS rates 
  642.                     ON rates.tax_rate_id = locations.tax_rate_id 
  643.                 WHERE rates.tax_rate_class = '%s' 
  644.             ", $slug ) ); 
  645.  
  646.             // Delete tax rates in the selected class. 
  647.             $wpdb->delete( $wpdb->prefix . 'woocommerce_tax_rates', array( 'tax_rate_class' => $slug ), array( '%s' ) ); 
  648.  
  649.             return array( 'message' => sprintf( __( 'Deleted %s', 'woocommerce' ), 'tax_class' ) ); 
  650.         } catch ( WC_API_Exception $e ) { 
  651.             return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) ); 
  652.         } 
  653.     } 
  654.  
  655.     /** 
  656.      * Get the total number of tax classes 
  657.      * 
  658.      * @since 2.5.0 
  659.      * 
  660.      * @return array 
  661.      */ 
  662.     public function get_tax_classes_count() { 
  663.         try { 
  664.             if ( ! current_user_can( 'manage_woocommerce' ) ) { 
  665.                 throw new WC_API_Exception( 'woocommerce_api_user_cannot_read_tax_classes_count', __( 'You do not have permission to read the tax classes count', 'woocommerce' ), 401 ); 
  666.             } 
  667.  
  668.             $total = count( WC_Tax::get_tax_classes() ) + 1; // +1 for Standard Rate 
  669.  
  670.             return array( 'count' => $total ); 
  671.         } catch ( WC_API_Exception $e ) { 
  672.             return new WP_Error( $e->getErrorCode(), $e->getMessage(), array( 'status' => $e->getCode() ) ); 
  673.         } 
  674.     } 
  675. }