backend/app/plugins/wiaas/includes/admin/class-wiaas-admin-product.php

@@ -4,12 +4,13 @@ class Wiaas_Admin_Product {
 
     public static function init() {
 
-        add_action('acf/save_post', array(__CLASS__, 'wiaas_my_save_post'), 20, 1);
+        add_action('acf/save_post', array(__CLASS__, 'save_initial_product_country_and_type'), 20, 1);
         add_action('woocommerce_after_register_post_type', array(__CLASS__, 'wiaas_register_product_status'));
         add_filter('woocommerce_register_post_type_product', array(__CLASS__, 'wiaas_modify_product'));
         add_action('add_meta_boxes', array(__CLASS__, 'wiaas_maybe_remove_metaboxes'), 999);
         add_filter('wp_insert_post_data', array(__CLASS__, 'wiaas_maybe_set_no_country_status'), 999, 2);
-        add_action('acf/render_field/type=taxonomy', array(__CLASS__, 'wiaas_render_field'), 10, 1);
+
+        add_action('acf/render_field/type=group', array(__CLASS__, 'render_choose_button_for_new_product'), 10, 1);
 
     }
 
@@ -28,13 +29,12 @@ class Wiaas_Admin_Product {
         ));
     }
 
+    public static function render_choose_button_for_new_product($field) {
 
-    public static function wiaas_render_field($field) {
+        if ($field['_name'] === '_wiaas_product_general') {
-
-        if ($field['_name'] === '_wiaas_product_country') {
             ?>
-            <div id="submitpost" style="padding: 20px;">
+            <div id="submitpost" style="padding: 20px 0;">
-                <input style="float:right;" name="save" id="save-post" class="button" type="submit" value="Choose"/>
+                <input style="float:right;" name="publish" id="publish" class="button button-large button-primary" type="submit" value="Choose"/>
             </div>
             <?php
         }
@@ -53,8 +53,7 @@ class Wiaas_Admin_Product {
 
         $post_id = $post->ID;
 
-        if ($post_id === 0 ||
+        if ($post_id === 0 || $post->post_status === '_wiaas_no_country') {
-            ($post->post_status !== 'publish' && !(Wiaas_Countries::get_package_country(wc_get_product($post_id))))) {
 
             remove_meta_box('woocommerce-product-data', 'product', 'normal');
             remove_meta_box('submitdiv', 'product', 'side');
@@ -99,24 +98,54 @@ class Wiaas_Admin_Product {
      * @param $post_id
      */
 
-    public static function wiaas_my_save_post($post_id) {
+    public static function save_initial_product_country_and_type($post_id) {
 
         global $post;
-        $status = get_post_status( $post->ID);
 
-        $value = get_field('_wiaas_product_country', $post_id, true);
+        if ($post->post_type !== 'product') {
-        $type = get_field('_wiaas_product_type', $post_id, true);
+            return;
+        }
 
-        if (!empty($value) && $status === '_wiaas_no_country' ) {
+        $general = get_field('_wiaas_product_general', $post_id);
+
+        $country = $general['_wiaas_product_country'];
+        $type = $general['_wiaas_product_type'];
+
+        /**
+         * If type is missing set status to no country
+         *
+         * If this is not template product type and country is missing set status to no country
+         *
+         */
+        if (empty($type) || ( empty($country) && $type !== 'wiaastemplate' )) {
+            wp_update_post(array(
+                'ID' => $post_id,
+                'post_status' => '_wiaas_no_country'
+            ));
+
+            return;
+        }
+
+        /**
+         * Country and type are selected for simple and bundle products so
+         * link them
+         */
 
-            wp_set_object_terms($post_id, $value, 'product_country', true);
         wp_set_object_terms($post_id, $type, 'product_type', true);
 
+        if ($type !== 'wiaastemplate') {
+            wp_set_object_terms($post_id, $country, 'product_country', true);
+        }
+
+        /**
+         * If product had no country status change it to draft status
+         */
+
+        if ($post->post_status === '_wiaas_no_country') {
             wp_update_post(array(
                 'ID' => $post_id,
                 'post_status' => 'draft'
             ));
-
         }
     }
 

backend/app/plugins/wiaas/includes/db-updates/data/wiaas-ui-field-general.json

@@ -3,6 +3,21 @@
         "key": "group_5bbf694106e34",
         "title": "General",
         "fields": [
+            {
+                "key": "field_5bceef4c9771c",
+                "label": "General",
+                "name": "_wiaas_product_general",
+                "type": "group",
+                "instructions": "",
+                "required": 0,
+                "conditional_logic": 0,
+                "wrapper": {
+                    "width": "",
+                    "class": "",
+                    "id": ""
+                },
+                "layout": "block",
+                "sub_fields": [
                     {
                         "key": "field_5bc0965a35406",
                         "label": "Type",
@@ -35,7 +50,7 @@
                         "name": "_wiaas_product_country",
                         "type": "taxonomy",
                         "instructions": "",
-                "required": 0,
+                        "required": 1,
                         "conditional_logic": [
                             [
                                 {
@@ -59,6 +74,8 @@
                         "return_format": "id",
                         "multiple": 0
                     }
+                ]
+            }
         ],
         "location": [
             [
@@ -76,7 +93,7 @@
         ],
         "menu_order": 0,
         "position": "acf_after_title",
-        "style": "default",
+        "style": "seamless",
         "label_placement": "top",
         "instruction_placement": "label",
         "hide_on_screen": "",