View difference between Paste ID: y2Tx81zz and rHk4THuz

 /**
- * WooCommerce Prices by Country - Export prices by country in separate columns.
+ * Prices by Country - Use billing or shipping country from manual orders
- * This example shows how to "explode" the prices by country assigned to each
+ * The following code snippet takes the  billing or shipping country from manual orders,
- * product into separate columns (one column for each combination of region ID,
+ * so that the prices for such country are taken when adding items to the order.
- * price type and currency).
+ *
  * IMPORTANT
- * Note
+ * For this code snippet to work, the order must have the billing and/or shipping address
- * This code will export the raw prices by country stored in the database. Prices
+ * entered and saved into the database. To do so, please enter the address(es), then
- * left as "Auto" won't be calculated on the fly, as calculated data is outside
+ * save the order before adding products to it.
- * the scope of the export function.
+ *
  * GPL DISCLAIMER
- * HOW TO USE THIS CODE
+
- * Simply add the code to the bottom of your theme's functions.php file, and it
+
- * will run automatically. For more information: https://www.skyverge.com/blog/add-custom-code-to-wordpress/
+
  * merchantability and fitness for a particular purpose. The entire risk as to the quality and performance of the program
  * is with you. should the program prove defective, you assume the cost of all necessary servicing, repair or correction.
  *
  * Need a consultation, or assistance to customise this code? Find us on Codeable: https://aelia.co/hire_us
  *
  */
 
 /**
  * During the creation of manual orders, this function replaces the active country used by the
  * Prices by Country plugin with the billing or shipping country from the order.
  *
  * @param string $country
- * Adds one new column to the exported file, for each combination of the following elements:
+
- * - Region ID
+add_filter('wc_aelia_pbc_customer_country', function($country) {
- * - Region Name
+  // Intercept the "add order item" action. That's the action used to add products
- * - Price type (regular price, sale price)
+  // to manual orders
- * - Currency
+  if(defined('DOING_AJAX') && isset($_REQUEST['action']) && isset($_REQUEST['order_id']) && ($_REQUEST['action'] === 'woocommerce_add_order_item')) {
     $order = wc_get_order($_REQUEST['order_id']);
- * @param array columns
+    // Safeguard to prevent dealing with invalid orders
- * @param object exporter
+    if($order instanceof \WC_Order) {
- * @return array
+      // Take the shipping country, if the Prices by Country plugin is configured to do so
       if(Aelia\WC\PricesByCountry\WC_Aelia_Prices_By_Country::settings()->customer_country_to_use()  === 'shipping_country') {
-add_filter('woocommerce_product_export_column_names', function($columns, $exporter) {
+        $order_country = $order->get_shipping_country();
-  // @var An array of price keys (region ID, currency, price type)
+
-  global $price_keys;
+
-  $price_keys = array();
+      // If the shipping country is empty, or the Prices by Country plugin is configured to use
       // the billing country, take that one instead
-  // Safety check. We can skip the logic if the Prices by Country plugin has not
+      if(empty($order_country) || (Aelia\WC\PricesByCountry\WC_Aelia_Prices_By_Country::settings()->customer_country_to_use()  === 'billing_country')) {
-  // been initialised
+        $order_country = $order->get_billing_country();
-  if(empty($GLOBALS['wc-aelia-prices-by-country'])) {
+
-    return $columns;
+
   }
 
-  $price_regions = \Aelia\WC\PricesByCountry\WC_Aelia_Prices_By_Country::settings()->get_regions();
+  // Replace the active country with the order country, if set
-  $currencies = apply_filters('wc_aelia_cs_enabled_currencies', get_woocommerce_currency());
+  return !empty($order_country) ? $order_country : $country;
-  $price_types = array('regular_price', 'sale_price');
+});