From ffdc13fb4beb810e5d700384bf47ba3fb1b0049e Mon Sep 17 00:00:00 2001 From: Brent Shepherd Date: Thu, 23 Jul 2015 22:15:44 -0700 Subject: [PATCH] Remove duplication in WC_Ajax::save_variations() And use the original WC_Meta_Box_Product_Data::save_variations() method instead. --- includes/class-wc-ajax.php | 305 +------------------------------------ 1 file changed, 1 insertion(+), 304 deletions(-) diff --git a/includes/class-wc-ajax.php b/includes/class-wc-ajax.php index 33078e8ece5..62efb1f8503 100644 --- a/includes/class-wc-ajax.php +++ b/includes/class-wc-ajax.php @@ -2588,311 +2588,8 @@ class WC_AJAX { die( -1 ); } - global $wpdb; - $product_id = absint( $_POST['product_id'] ); - $attributes = (array) maybe_unserialize( get_post_meta( $product_id, '_product_attributes', true ) ); - - if ( isset( $_POST['variable_sku'] ) ) { - $variable_post_id = $_POST['variable_post_id']; - $variable_sku = $_POST['variable_sku']; - $variable_regular_price = $_POST['variable_regular_price']; - $variable_sale_price = $_POST['variable_sale_price']; - $upload_image_id = $_POST['upload_image_id']; - $variable_download_limit = $_POST['variable_download_limit']; - $variable_download_expiry = $_POST['variable_download_expiry']; - $variable_shipping_class = $_POST['variable_shipping_class']; - $variable_tax_class = isset( $_POST['variable_tax_class'] ) ? $_POST['variable_tax_class'] : array(); - $variable_menu_order = $_POST['variation_menu_order']; - $variable_sale_price_dates_from = $_POST['variable_sale_price_dates_from']; - $variable_sale_price_dates_to = $_POST['variable_sale_price_dates_to']; - - $variable_weight = isset( $_POST['variable_weight'] ) ? $_POST['variable_weight'] : array(); - $variable_length = isset( $_POST['variable_length'] ) ? $_POST['variable_length'] : array(); - $variable_width = isset( $_POST['variable_width'] ) ? $_POST['variable_width'] : array(); - $variable_height = isset( $_POST['variable_height'] ) ? $_POST['variable_height'] : array(); - $variable_enabled = isset( $_POST['variable_enabled'] ) ? $_POST['variable_enabled'] : array(); - $variable_is_virtual = isset( $_POST['variable_is_virtual'] ) ? $_POST['variable_is_virtual'] : array(); - $variable_is_downloadable = isset( $_POST['variable_is_downloadable'] ) ? $_POST['variable_is_downloadable'] : array(); - - $variable_manage_stock = isset( $_POST['variable_manage_stock'] ) ? $_POST['variable_manage_stock'] : array(); - $variable_stock = isset( $_POST['variable_stock'] ) ? $_POST['variable_stock'] : array(); - $variable_backorders = isset( $_POST['variable_backorders'] ) ? $_POST['variable_backorders'] : array(); - $variable_stock_status = isset( $_POST['variable_stock_status'] ) ? $_POST['variable_stock_status'] : array(); - - $variable_description = isset( $_POST['variable_description'] ) ? $_POST['variable_description'] : array(); - - $max_loop = max( array_keys( $_POST['variable_post_id'] ) ); - - for ( $i = 0; $i <= $max_loop; $i ++ ) { - - if ( ! isset( $variable_post_id[ $i ] ) ) { - continue; - } - - $variation_id = absint( $variable_post_id[ $i ] ); - - // Checkboxes - $is_virtual = isset( $variable_is_virtual[ $i ] ) ? 'yes' : 'no'; - $is_downloadable = isset( $variable_is_downloadable[ $i ] ) ? 'yes' : 'no'; - $post_status = isset( $variable_enabled[ $i ] ) ? 'publish' : 'private'; - $manage_stock = isset( $variable_manage_stock[ $i ] ) ? 'yes' : 'no'; - - // Generate a useful post title - $variation_post_title = sprintf( __( 'Variation #%s of %s', 'woocommerce' ), absint( $variation_id ), esc_html( get_the_title( $product_id ) ) ); - - // Update or Add post - if ( ! $variation_id ) { - - $variation = array( - 'post_title' => $variation_post_title, - 'post_content' => '', - 'post_status' => $post_status, - 'post_author' => get_current_user_id(), - 'post_parent' => $product_id, - 'post_type' => 'product_variation', - 'menu_order' => $variable_menu_order[ $i ] - ); - - $variation_id = wp_insert_post( $variation ); - - do_action( 'woocommerce_create_product_variation', $variation_id ); - - } else { - - $wpdb->update( $wpdb->posts, array( 'post_status' => $post_status, 'post_title' => $variation_post_title, 'menu_order' => $variable_menu_order[ $i ] ), array( 'ID' => $variation_id ) ); - - do_action( 'woocommerce_update_product_variation', $variation_id ); - - } - - // Only continue if we have a variation ID - if ( ! $variation_id ) { - continue; - } - - // Unique SKU - $sku = get_post_meta( $variation_id, '_sku', true ); - $new_sku = wc_clean( stripslashes( $variable_sku[ $i ] ) ); - - if ( '' == $new_sku ) { - update_post_meta( $variation_id, '_sku', '' ); - } elseif ( $new_sku !== $sku ) { - - if ( ! empty( $new_sku ) ) { - $unique_sku = wc_product_has_unique_sku( $variation_id, $new_sku ); - - if ( ! $unique_sku ) { - WC_Admin_Meta_Boxes::add_error( __( 'Variation SKU must be unique.', 'woocommerce' ) ); - } else { - update_post_meta( $variation_id, '_sku', $new_sku ); - } - } else { - update_post_meta( $variation_id, '_sku', '' ); - } - } - - // Update post meta - update_post_meta( $variation_id, '_thumbnail_id', absint( $upload_image_id[ $i ] ) ); - update_post_meta( $variation_id, '_virtual', wc_clean( $is_virtual ) ); - update_post_meta( $variation_id, '_downloadable', wc_clean( $is_downloadable ) ); - - if ( isset( $variable_weight[ $i ] ) ) { - update_post_meta( $variation_id, '_weight', ( '' === $variable_weight[ $i ] ) ? '' : wc_format_decimal( $variable_weight[ $i ] ) ); - } - - if ( isset( $variable_length[ $i ] ) ) { - update_post_meta( $variation_id, '_length', ( '' === $variable_length[ $i ] ) ? '' : wc_format_decimal( $variable_length[ $i ] ) ); - } - - if ( isset( $variable_width[ $i ] ) ) { - update_post_meta( $variation_id, '_width', ( '' === $variable_width[ $i ] ) ? '' : wc_format_decimal( $variable_width[ $i ] ) ); - } - - if ( isset( $variable_height[ $i ] ) ) { - update_post_meta( $variation_id, '_height', ( '' === $variable_height[ $i ] ) ? '' : wc_format_decimal( $variable_height[ $i ] ) ); - } - - // Stock handling - update_post_meta( $variation_id, '_manage_stock', $manage_stock ); - - // Only update stock status to user setting if changed by the user, but do so before looking at stock levels at variation level - if ( ! empty( $variable_stock_status[ $i ] ) ) { - wc_update_product_stock_status( $variation_id, $variable_stock_status[ $i ] ); - } - - if ( 'yes' === $manage_stock ) { - update_post_meta( $variation_id, '_backorders', wc_clean( $variable_backorders[ $i ] ) ); - wc_update_product_stock( $variation_id, wc_stock_amount( $variable_stock[ $i ] ) ); - } else { - delete_post_meta( $variation_id, '_backorders' ); - delete_post_meta( $variation_id, '_stock' ); - } - - // Price handling - $regular_price = wc_format_decimal( $variable_regular_price[ $i ] ); - $sale_price = $variable_sale_price[ $i ] === '' ? '' : wc_format_decimal( $variable_sale_price[ $i ] ); - $date_from = wc_clean( $variable_sale_price_dates_from[ $i ] ); - $date_to = wc_clean( $variable_sale_price_dates_to[ $i ] ); - - update_post_meta( $variation_id, '_regular_price', $regular_price ); - update_post_meta( $variation_id, '_sale_price', $sale_price ); - - // Save Dates - update_post_meta( $variation_id, '_sale_price_dates_from', $date_from ? strtotime( $date_from ) : '' ); - update_post_meta( $variation_id, '_sale_price_dates_to', $date_to ? strtotime( $date_to ) : '' ); - - if ( $date_to && ! $date_from ) { - update_post_meta( $variation_id, '_sale_price_dates_from', strtotime( 'NOW', current_time( 'timestamp' ) ) ); - } - - // Update price if on sale - if ( '' !== $sale_price && '' === $date_to && '' === $date_from ) { - update_post_meta( $variation_id, '_price', $sale_price ); - } else { - update_post_meta( $variation_id, '_price', $regular_price ); - } - - if ( '' !== $sale_price && $date_from && strtotime( $date_from ) < strtotime( 'NOW', current_time( 'timestamp' ) ) ) { - update_post_meta( $variation_id, '_price', $sale_price ); - } - - if ( $date_to && strtotime( $date_to ) < strtotime( 'NOW', current_time( 'timestamp' ) ) ) { - update_post_meta( $variation_id, '_price', $regular_price ); - update_post_meta( $variation_id, '_sale_price_dates_from', '' ); - update_post_meta( $variation_id, '_sale_price_dates_to', '' ); - } - - if ( isset( $variable_tax_class[ $i ] ) && $variable_tax_class[ $i ] !== 'parent' ) { - update_post_meta( $variation_id, '_tax_class', wc_clean( $variable_tax_class[ $i ] ) ); - } else { - delete_post_meta( $variation_id, '_tax_class' ); - } - - if ( 'yes' == $is_downloadable ) { - update_post_meta( $variation_id, '_download_limit', wc_clean( $variable_download_limit[ $i ] ) ); - update_post_meta( $variation_id, '_download_expiry', wc_clean( $variable_download_expiry[ $i ] ) ); - - $files = array(); - $file_names = isset( $_POST['_wc_variation_file_names'][ $variation_id ] ) ? array_map( 'wc_clean', $_POST['_wc_variation_file_names'][ $variation_id ] ) : array(); - $file_urls = isset( $_POST['_wc_variation_file_urls'][ $variation_id ] ) ? array_map( 'wc_clean', $_POST['_wc_variation_file_urls'][ $variation_id ] ) : array(); - $file_url_size = sizeof( $file_urls ); - $allowed_file_types = get_allowed_mime_types(); - - for ( $ii = 0; $ii < $file_url_size; $ii ++ ) { - if ( ! empty( $file_urls[ $ii ] ) ) { - // Find type and file URL - if ( 0 === strpos( $file_urls[ $ii ], 'http' ) ) { - $file_is = 'absolute'; - $file_url = esc_url_raw( $file_urls[ $ii ] ); - } elseif ( '[' === substr( $file_urls[ $ii ], 0, 1 ) && ']' === substr( $file_urls[ $ii ], -1 ) ) { - $file_is = 'shortcode'; - $file_url = wc_clean( $file_urls[ $ii ] ); - } else { - $file_is = 'relative'; - $file_url = wc_clean( $file_urls[ $ii ] ); - } - - $file_name = wc_clean( $file_names[ $ii ] ); - $file_hash = md5( $file_url ); - - // Validate the file extension - if ( in_array( $file_is, array( 'absolute', 'relative' ) ) ) { - $file_type = wp_check_filetype( strtok( $file_url, '?' ) ); - $parsed_url = parse_url( $file_url, PHP_URL_PATH ); - $extension = pathinfo( $parsed_url, PATHINFO_EXTENSION ); - - if ( ! empty( $extension ) && ! in_array( $file_type['type'], $allowed_file_types ) ) { - WC_Admin_Meta_Boxes::add_error( sprintf( __( 'The downloadable file %s cannot be used as it does not have an allowed file type. Allowed types include: %s', 'woocommerce' ), '' . basename( $file_url ) . '', '' . implode( ', ', array_keys( $allowed_file_types ) ) . '' ) ); - continue; - } - } - - // Validate the file exists - if ( 'relative' === $file_is && ! apply_filters( 'woocommerce_downloadable_file_exists', file_exists( $file_url ), $file_url ) ) { - WC_Admin_Meta_Boxes::add_error( sprintf( __( 'The downloadable file %s cannot be used as it does not exist on the server.', 'woocommerce' ), '' . $file_url . '' ) ); - continue; - } - - $files[ $file_hash ] = array( - 'name' => $file_name, - 'file' => $file_url - ); - } - } - - // grant permission to any newly added files on any existing orders for this product prior to saving - do_action( 'woocommerce_process_product_file_download_paths', $product_id, $variation_id, $files ); - - update_post_meta( $variation_id, '_downloadable_files', $files ); - } else { - update_post_meta( $variation_id, '_download_limit', '' ); - update_post_meta( $variation_id, '_download_expiry', '' ); - update_post_meta( $variation_id, '_downloadable_files', '' ); - } - - update_post_meta( $variation_id, '_variation_description', wp_kses_post( $variable_description[ $i ] ) ); - - // Save shipping class - $variable_shipping_class[ $i ] = ! empty( $variable_shipping_class[ $i ] ) ? (int) $variable_shipping_class[ $i ] : ''; - wp_set_object_terms( $variation_id, $variable_shipping_class[ $i ], 'product_shipping_class'); - - // Update Attributes - $updated_attribute_keys = array(); - foreach ( $attributes as $attribute ) { - if ( $attribute['is_variation'] ) { - $attribute_key = 'attribute_' . sanitize_title( $attribute['name'] ); - $updated_attribute_keys[] = $attribute_key; - - if ( $attribute['is_taxonomy'] ) { - // Don't use wc_clean as it destroys sanitized characters - $value = isset( $_POST[ $attribute_key ][ $i ] ) ? sanitize_title( stripslashes( $_POST[ $attribute_key ][ $i ] ) ) : ''; - } else { - $value = isset( $_POST[ $attribute_key ][ $i ] ) ? wc_clean( stripslashes( $_POST[ $attribute_key ][ $i ] ) ) : ''; - } - - update_post_meta( $variation_id, $attribute_key, $value ); - } - } - - // Remove old taxonomies attributes so data is kept up to date - first get attribute key names - $delete_attribute_keys = $wpdb->get_col( $wpdb->prepare( "SELECT meta_key FROM {$wpdb->postmeta} WHERE meta_key LIKE 'attribute_%%' AND meta_key NOT IN ( '" . implode( "','", $updated_attribute_keys ) . "' ) AND post_id = %d;", $variation_id ) ); - - foreach ( $delete_attribute_keys as $key ) { - delete_post_meta( $variation_id, $key ); - } - - do_action( 'woocommerce_save_product_variation', $variation_id, $i ); - } - } - - // Update parent if variable so price sorting works and stays in sync with the cheapest child - WC_Product_Variable::sync( $product_id ); - - // Update default attribute options setting - $default_attributes = array(); - - foreach ( $attributes as $attribute ) { - - if ( $attribute['is_variation'] ) { - $value = ''; - - if ( isset( $_POST[ 'default_attribute_' . sanitize_title( $attribute['name'] ) ] ) ) { - if ( $attribute['is_taxonomy'] ) { - // Don't use wc_clean as it destroys sanitized characters - $value = sanitize_title( trim( stripslashes( $_POST[ 'default_attribute_' . sanitize_title( $attribute['name'] ) ] ) ) ); - } else { - $value = wc_clean( trim( stripslashes( $_POST[ 'default_attribute_' . sanitize_title( $attribute['name'] ) ] ) ) ); - } - } - - if ( $value ) { - $default_attributes[ sanitize_title( $attribute['name'] ) ] = $value; - } - } - } - - update_post_meta( $product_id, '_default_attributes', $default_attributes ); + WC_Meta_Box_Product_Data::save_variations( $product_id, get_post( $product_id ) ); do_action( 'woocommerce_ajax_save_product_variations', $product_id );