WooNooW/includes/Api/ShippingController.php

<?php
/**
 * Shipping REST API Controller
 * 
 * Provides REST endpoints for shipping zones and methods management.
 * 
 * @package WooNooW
 */

namespace WooNooW\Api;

use WP_REST_Controller;
use WP_REST_Request;
use WP_REST_Response;
use WC_Shipping_Zones;

class ShippingController extends WP_REST_Controller {

	protected $namespace = 'woonoow/v1';
	protected $rest_base = 'settings/shipping';

	public function register_routes() {
		// Get shipping zones
		register_rest_route(
			$this->namespace,
			'/' . $this->rest_base . '/zones',
			array(
				array(
					'methods'             => \WP_REST_Server::READABLE,
					'callback'            => array( $this, 'get_zones' ),
					'permission_callback' => array( $this, 'check_permission' ),
				),
			)
		);

		// Toggle shipping method
		register_rest_route(
			$this->namespace,
			'/' . $this->rest_base . '/zones/(?P<zone_id>\d+)/methods/(?P<instance_id>\d+)/toggle',
			array(
				array(
					'methods'             => \WP_REST_Server::CREATABLE,
					'callback'            => array( $this, 'toggle_method' ),
					'permission_callback' => array( $this, 'check_permission' ),
					'args'                => array(
						'zone_id'     => array(
							'required' => true,
							'type'     => 'integer',
						),
						'instance_id' => array(
							'required' => true,
							'type'     => 'integer',
						),
						'enabled'     => array(
							'required' => true,
							'type'     => 'boolean',
						),
					),
				),
			)
		);
	}

	/**
	 * Get all shipping zones with their methods
	 */
	public function get_zones( WP_REST_Request $request ) {
		try {
			$zones_data = array();
			
			// Get all shipping zones
			$zones = WC_Shipping_Zones::get_zones();
			
			foreach ( $zones as $zone_data ) {
				$zone = WC_Shipping_Zones::get_zone( $zone_data['id'] );
				
				// Get zone locations (regions)
				$locations = $zone->get_zone_locations();
				$regions = array();
				
				foreach ( $locations as $location ) {
					if ( $location->type === 'country' ) {
						$countries = WC()->countries->get_countries();
						$regions[] = $countries[ $location->code ] ?? $location->code;
					} elseif ( $location->type === 'state' ) {
						$states = WC()->countries->get_states( substr( $location->code, 0, 2 ) );
						$regions[] = $states[ substr( $location->code, 3 ) ] ?? $location->code;
					} elseif ( $location->type === 'continent' ) {
						$continents = WC()->countries->get_continents();
						$regions[] = $continents[ $location->code ]['name'] ?? $location->code;
					} elseif ( $location->type === 'postcode' ) {
						$regions[] = 'Postcode: ' . $location->code;
					}
				}
				
				// Get shipping methods for this zone
				$shipping_methods = $zone->get_shipping_methods( true ); // true = enabled only
				$rates = array();
				
				foreach ( $shipping_methods as $method ) {
					$rate = array(
						'id'          => $method->id . ':' . $method->instance_id,
						'instance_id' => $method->instance_id,
						'method_id'   => $method->id,
						'name'        => $method->get_title(),
						'enabled'     => $method->enabled === 'yes',
					);
					
					// Get cost if available
					if ( isset( $method->cost ) && $method->cost !== '' ) {
						$rate['price'] = wc_price( $method->cost );
					} elseif ( $method->id === 'free_shipping' ) {
						$rate['price'] = __( 'Free', 'woonoow' );
					} else {
						$rate['price'] = __( 'Calculated', 'woonoow' );
					}
					
					// Get method description if available
					if ( method_exists( $method, 'get_method_description' ) ) {
						$rate['description'] = $method->get_method_description();
					}
					
					$rates[] = $rate;
				}
				
				$zones_data[] = array(
					'id'      => $zone_data['id'],
					'name'    => $zone->get_zone_name(),
					'order'   => $zone->get_zone_order(),
					'regions' => ! empty( $regions ) ? implode( ', ', $regions ) : __( 'No regions', 'woonoow' ),
					'rates'   => $rates,
				);
			}
			
			// Add "Rest of the World" zone (zone 0)
			$zone_0 = new \WC_Shipping_Zone( 0 );
			$shipping_methods = $zone_0->get_shipping_methods( true );
			$rates = array();
			
			foreach ( $shipping_methods as $method ) {
				$rate = array(
					'id'          => $method->id . ':' . $method->instance_id,
					'instance_id' => $method->instance_id,
					'method_id'   => $method->id,
					'name'        => $method->get_title(),
					'enabled'     => $method->enabled === 'yes',
				);
				
				if ( isset( $method->cost ) && $method->cost !== '' ) {
					$rate['price'] = wc_price( $method->cost );
				} elseif ( $method->id === 'free_shipping' ) {
					$rate['price'] = __( 'Free', 'woonoow' );
				} else {
					$rate['price'] = __( 'Calculated', 'woonoow' );
				}
				
				$rates[] = $rate;
			}
			
			if ( ! empty( $rates ) ) {
				$zones_data[] = array(
					'id'      => 0,
					'name'    => __( 'Rest of the World', 'woonoow' ),
					'order'   => 9999,
					'regions' => __( 'Locations not covered by other zones', 'woonoow' ),
					'rates'   => $rates,
				);
			}
			
			return new WP_REST_Response( $zones_data, 200 );
			
		} catch ( \Exception $e ) {
			return new WP_REST_Response(
				array(
					'error'   => 'fetch_failed',
					'message' => $e->getMessage(),
				),
				500
			);
		}
	}

	/**
	 * Toggle shipping method enabled/disabled
	 */
	public function toggle_method( WP_REST_Request $request ) {
		try {
			$zone_id     = $request->get_param( 'zone_id' );
			$instance_id = $request->get_param( 'instance_id' );
			$enabled     = $request->get_param( 'enabled' );

			// Get the zone
			$zone = \WC_Shipping_Zones::get_zone( $zone_id );
			if ( ! $zone ) {
				return new WP_REST_Response(
					array(
						'error'   => 'zone_not_found',
						'message' => __( 'Shipping zone not found', 'woonoow' ),
					),
					404
				);
			}

			// Get all shipping methods for this zone
			$shipping_methods = $zone->get_shipping_methods();
			$method_found     = false;

			foreach ( $shipping_methods as $method ) {
				if ( $method->instance_id == $instance_id ) {
					$method_found = true;
					
					// Update the enabled status
					$method->enabled = $enabled ? 'yes' : 'no';
					update_option( $method->get_instance_option_key(), $method->instance_settings );
					
					break;
				}
			}

			if ( ! $method_found ) {
				return new WP_REST_Response(
					array(
						'error'   => 'method_not_found',
						'message' => __( 'Shipping method not found', 'woonoow' ),
					),
					404
				);
			}

			// Clear shipping cache
			\WC_Cache_Helper::get_transient_version( 'shipping', true );

			return new WP_REST_Response(
				array(
					'success' => true,
					'message' => $enabled 
						? __( 'Shipping method enabled', 'woonoow' )
						: __( 'Shipping method disabled', 'woonoow' ),
				),
				200
			);

		} catch ( \Exception $e ) {
			return new WP_REST_Response(
				array(
					'error'   => 'toggle_failed',
					'message' => $e->getMessage(),
				),
				500
			);
		}
	}

	/**
	 * Check if user has permission to manage shipping
	 */
	public function check_permission() {
		return current_user_can( 'manage_woocommerce' );
	}
}