Skip to content

Instantly share code, notes, and snippets.

@WProbot

WProbot/cc-wc-orders-controller

Forked from LosArtos/cc-wc-orders-controller
Last active November 25, 2021 03:56
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save WProbot/083d565693ab6794e8e2cb4de50d0bf9 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save WProbot/083d565693ab6794e8e2cb4de50d0bf9 to your computer and use it in GitHub Desktop.
Download ZIP
orders-controller
Raw
cc-wc-orders-controller
<?php
if (!defined('ABSPATH')) {
exit;
}
class CC_WC_Orders_Controller
{
public function Init()
{
/* set to 20+ (e.g 100) for UNI CPO plugin compatibility */
add_filter('woocommerce_admin_order_item_thumbnail', array($this, 'GetOrderImage'), 100, 3);
add_filter('woocommerce_order_item_get_formatted_meta_data', array($this, 'FilterMetaData'));
add_filter('wp_kses_allowed_html', array($this, 'FilterAllowedHtml'), 10, 2);
add_filter('woocommerce_display_item_meta', array($this, 'FilterItemMeta'), 10, 3);
add_filter('bulk_actions-edit-shop_order', array($this, 'AddExportBulkAction'), 20, 1);
add_action('woocommerce_checkout_create_order_line_item', array($this, 'CreateOrderItemMeta'), 10, 3);
add_action('handle_bulk_actions-edit-shop_order', array($this, 'HandleExportOrders'), 10, 3);
add_action('admin_notices', array($this, 'OutputExportNotice'));
add_action('woocommerce_after_order_itemmeta', array($this, 'DisplayOrderItemMeta'), 10, 3);
}
public function DisplayOrderItemMeta($item_id, $item, $product)
{
$data = $item->get_meta('cc-urls');
if (!empty($data)) {
foreach ($data as $url) {
$fileName = $this->PrepareFilename($item);
if ($fileName) {
$url .= '/' . $fileName;
}
printf('<div><a href="%s">' . __('Download design', 'customers-canvas-for-wc') . '</a></div>', $url);
}
}
$data = $item->get_meta('cc-data');
if (!empty($data)) {
foreach ($data as $k => $v) {
if ($k == "snapshot")
continue;
if ($this->isValidURL($v)) {
if (is_numeric($k)) {
printf('<div><a href="%s">' . $v . '</a></div>', $v);
} else {
printf('<div><b>' . $k . ': </b><a href="%s">' . $v . '</a></div>', $v);
}
} else {
echo '<div><b>' . $k . ':</b> ' . $v . '</div>';
}
}
} else {
/*
* Added for UNI CPO compatibility bc the only way to get it is to let UNI CPO plugin do all submit work
* so we can't create cc-data, cc-urls and cc-images meta objects
*/
$data = $item->get_meta('_download_urls');
if (!empty($data)) {
foreach ($data as $k => $v) {
$cur_url = str_replace('"', '', $v);
echo $cur_url;
if (is_numeric($k)) {
printf('<div><b>_download_url_' . $k . ': </b><a href="%s">' . $cur_url . '</a></div>', $cur_url);
} else {
printf('<div><b>' . $k . ': </b><a href="%s">' . $cur_url . '</a></div>', $cur_url);
}
}
}
$data = $item->get_meta('_data');
if (!empty($data)) {
foreach ($data as $k => $v) {
if ($k == "snapshot")
continue;
if ($this->isValidURL($v)) {
if (is_numeric($k)) {
printf('<div><b>' . $k . ': </b><a href="%s">' . $v . '</a></div>', $v);
} else {
printf('<div><b>' . $k . ': </b><a href="%s">' . $v . '</a></div>', $v);
}
} else {
echo '<div><b>' . $k . ':</b> ' . $v . '</div>';
}
}
}
}
}
protected function startsWith($str, $prefix)
{
$length = strlen($prefix);
return (substr($str, 0, $length) === $prefix);
}
protected function isValidURL($url)
{
return $this->startsWith($url, "http");
// return preg_match('/^(http|https):\\/\\/[a-z0-9_]+([\\-\\.]{1}[a-z_0-9]+)*\\.[_a-z]{2,5}' . '((:[0-9]{1,5})?\\/.*)?$/i', $url);
}
public function AddExportBulkAction($actions)
{
if (get_option('cc_wc_order_export') == 'true') {
$actions['export_designs'] = __('Export PDF designs', 'woocommerce');
}
return $actions;
}
public function HandleExportOrders($redirect_to, $action, $post_ids)
{
if (get_option('cc_wc_order_export') != 'true') {
return $redirect_to;
}
$exportPath = stripslashes(get_option('cc_wc_order_export_path'));
$fileNamePattern = get_option('cc_wc_filename_pattern');
$redirect_to = remove_query_arg(array(
'export_designs',
'error',
'processed_count',
'processed_ids',
), $redirect_to);
if (empty($exportPath)) {
$redirect_to = add_query_arg(array(
'export_designs' => '1',
'error' => 'Export path is empty.',
), $redirect_to);
}
if ($action !== 'export_designs') {
return $redirect_to;
}
$processed_ids = array();
$error = '';
foreach ($post_ids as $post_id) {
$order = wc_get_order($post_id);
$order_data = $order->get_data();
$items = $order->get_items();
foreach ($items as $item) {
$data = $item->get_meta('cc-urls');
$url = $data[0];
$dir = $this->PrepareNameFromPattern($exportPath, $item);
if (!file_exists($dir)) {
mkdir($dir, 0777, true);
}
$fileName = $this->PrepareNameFromPattern($fileNamePattern, $item);
if (!$fileName) {
$fileName = basename($url, '.pdf');
}
$mainName = cc_wc_join_paths($dir, $fileName);
$newPath = $mainName . '.pdf';
$copyNum = 1;
while (file_exists(($newPath))) {
$newPath = $mainName . '(' . $copyNum++ . ').pdf';
}
$local_error = cc_wc_download_file($url, $newPath);
if ($local_error) {
$error .= $local_error . '|';
} else {
$processed_ids[] = $post_id;
}
}
}
if (!empty($error)) {
$redirect_to = add_query_arg(array(
'error' => $error,
), $redirect_to);
}
return $redirect_to = add_query_arg(array(
'export_designs' => '1',
'processed_count' => count($processed_ids),
'processed_ids' => implode(',', $processed_ids),
), $redirect_to);
}
private function PrepareFilename($item)
{
$str = get_option('cc_wc_filename_pattern');
return $this->PrepareNameFromPattern($str, $item);
}
private function PrepareNameFromPattern($pattern, $item)
{
if (empty($pattern)) {
return false;
}
$sep = "<>";
$tokens = array();
$next = strtok($pattern, $sep);
while ($next !== false) {
$tokens[] = $next;
$next = strtok($sep);
}
$result = '';
foreach ($tokens as $token) {
$result .= $this->MapFilenameToken($item, $token);
}
return $result;
}
private function MapFilenameToken($item, $token)
{
switch ($token) {
case 'order_id':
return get_post_meta( $item->get_order_id(), '_wcj_order_number', true );
case 'product_id':
return $item->get_product_id();
case 'product_name':
return $item->get_name();
case 'product_sku':
$product = $item->get_product();
return $product->get_sku();
case 'item_quantity':
return $item->get_quantity();
}
// It's not a supported token. We will treat it as a constant part of pattern
return $token;
}
public function OutputExportNotice()
{
if (empty($_REQUEST['export_designs'])) {
return;
}
if (!empty($_REQUEST['error'])) {
$errors = explode('|', $_REQUEST['error']);
$msg = '';
foreach ($errors as $error) {
if ($error) {
$msg .= '<p>' . esc_html__('Export failed', 'customers-canvas-for-wc') . ': ' . esc_html($error) . '</p>';
}
}
printf('<div id="message" class="error">%s</div>', $msg);
}
$count = intval($_REQUEST['processed_count']);
if ($count > 0) {
printf('<div id="message" class="updated"><p>' .
_n(esc_html__('Exported %s Order design.', 'customers-canvas-for-wc'),
esc_html__('Exported %s Orders designs.', 'customers-canvas-for-wc'),
$count,
'write_downloads'
) . '</p></div>', $count);
}
}
public function FilterAllowedHtml($allowed, $context)
{
if (is_array($context)) {
return $allowed;
}
if ($context === 'post') {
$allowed['a']['data-fancybox'] = true;
}
return $allowed;
}
public function FilterItemMeta($html, $item, $args)
{
$html = '';
$strings = array();
foreach ($item->get_formatted_meta_data() as $meta_id => $meta) {
$query = 'Download #';
if (isset($meta->key) && substr($meta->key, 0, strlen($query)) !== $query) {
$value = $args['autop'] ? wp_kses_post($meta->display_value) : wp_kses_post(make_clickable(trim($meta->display_value)));
$strings[] = '<strong class="wc-item-meta-label">' . wp_kses_post($meta->display_key) . ':</strong> ' . $value;
}
}
if ($strings) {
$html = $args['before'] . implode($args['separator'], $strings) . $args['after'];
}
return $html;
}
public function FilterMetaData($formatted_meta)
{
$temp_metas = [];
foreach ($formatted_meta as $key => $meta) {
if (isset($meta->key) && !in_array($meta->key, [
'cc-images',
])) {
$temp_metas[$key] = $meta;
}
}
return $temp_metas;
}
public function CreateOrderItemMeta($item, $cart_item_key, $values)
{
if (isset($values['cc-data'])) {
$item->update_meta_data('cc-images', $values['cc-data']['images']);
$item->update_meta_data('cc-urls', $values['cc-data']['urls']);
$item->update_meta_data('cc-data', $values['cc-data']['data']);
}
}
public function GetOrderImage($image, $id, $item)
{
$data = $item->get_meta_data();
$result = array_filter(
$data,
function ($e) {
return $e->key == 'cc-images';
});
if ($result) {
$images = current($result)->value;
$html = '';
$i = 0;
foreach ($images as $image) {
if ($i < 1) {
$html .= '<a data-fancybox="' . $id . '" href="' . $image . '"><img src="' . $image . '"></a>';
} else {
$html .= '<a data-fancybox="' . $id . '" href="' . $image . '" style="display:none"></a>';
}
$i++;
}
return $html;
} else {
/*
* Added for UNI CPO compatibility bc the only way to get it is to let UNI CPO plugin do all submit work
* so we can't create cc-data, cc-urls and cc-images meta objects
*/
$result = array_filter(
$data,
function ($e) {
return $e->key == '_images';
});
if ($result) {
$images = current($result)->value;
$html = '';
$i = 0;
foreach ($images as $image) {
if ($i < 1) {
$html .= '<a data-fancybox="' . $id . '" href="' . $image . '"><img src="' . $image . '"></a>';
} else {
$html .= '<a data-fancybox="' . $id . '" href="' . $image . '" style="display:none"></a>';
}
$i++;
}
return $html;
} else {
return $image;
}
}
}
}
Raw
wcj-order-numbers
<?php
/**
* Booster for WooCommerce - Module - Order Numbers
*
* @version 5.4.6
* @author Pluggabl LLC.
*/
if ( ! defined( 'ABSPATH' ) ) exit; // Exit if accessed directly
if ( ! class_exists( 'WCJ_Order_Numbers' ) ) :
class WCJ_Order_Numbers extends WCJ_Module {
/**
* Constructor.
*
* @version 5.2.0
* @todo (maybe) rename "Orders Renumerate" to "Renumerate orders"
* @todo (maybe) use `woocommerce_new_order` hook instead of `wp_insert_post`
*/
function __construct() {
$this->id = 'order_numbers';
$this->short_desc = __( 'Order Numbers', 'woocommerce-jetpack' );
$this->desc = __( 'Sequential order numbering, custom order number prefix, suffix and number width. Prefix Options (Order Number Custom Prefix available in free version). Suffix options (Plus). ', 'woocommerce-jetpack' );
$this->desc_pro = __( 'Sequential order numbering, custom order number prefix, suffix and number width.', 'woocommerce-jetpack' );
$this->link_slug = 'woocommerce-custom-order-numbers';
parent::__construct();
$this->add_tools( array(
'renumerate_orders' => array(
'title' => __( 'Orders Renumerate', 'woocommerce-jetpack' ),
'desc' => __( 'Tool renumerates all orders.', 'woocommerce-jetpack' ),
),
) );
if ( $this->is_enabled() ) {
// Add & display custom order number
add_action( 'wp_insert_post', array( $this, 'add_new_order_number' ), PHP_INT_MAX );
add_filter( 'woocommerce_order_number', array( $this, 'display_order_number' ), PHP_INT_MAX, 2 );
// Order tracking
if ( 'yes' === wcj_get_option( 'wcj_order_number_order_tracking_enabled', 'yes' ) ) {
add_filter( 'woocommerce_shortcode_order_tracking_order_id', array( $this, 'add_order_number_to_tracking' ), PHP_INT_MAX );
add_action( 'init', array( $this, 'remove_order_tracking_sanitize_order_id_filter' ) );
}
// Search by custom number
if ( 'yes' === wcj_get_option( 'wcj_order_number_search_by_custom_number_enabled', 'yes' ) ) {
add_action( 'pre_get_posts', array( $this, 'search_by_custom_number' ) );
}
// "WooCommerce Subscriptions" plugin
$woocommerce_subscriptions_types = array( 'subscription', 'renewal_order', 'resubscribe_order', 'copy_order' );
foreach ( $woocommerce_subscriptions_types as $woocommerce_subscriptions_type ) {
add_filter( 'wcs_' . $woocommerce_subscriptions_type . '_meta', array( $this, 'woocommerce_subscriptions_remove_meta_copy' ), PHP_INT_MAX, 3 );
}
// Editable order number
if ( 'yes' === apply_filters( 'booster_option', 'no', wcj_get_option( 'wcj_order_number_editable_order_number_meta_box_enabled', 'no' ) ) ) {
$this->meta_box_screen = 'shop_order';
$this->meta_box_context = 'side';
$this->meta_box_priority = 'high';
add_action( 'add_meta_boxes', array( $this, 'maybe_add_meta_box' ), PHP_INT_MAX, 2 );
add_action( 'save_post_shop_order', array( $this, 'save_meta_box' ), PHP_INT_MAX, 2 );
}
// Compatibility with WPNotif plugin
add_action( 'admin_init', function () {
if ( 'yes' === wcj_get_option( 'wcj_order_numbers_compatibility_wpnotif', 'no' ) ) {
remove_filter( 'wpnotif_filter_message', 'wpnotif_add_tracking_details', 10, 2 );
add_filter( 'wpnotif_filter_message', array( $this, 'wpnotif_add_tracking_details' ), 10, 2 );
}
} );
}
}
/**
* wpnotif_add_tracking_details.
*
* @see wpnotif_add_tracking_details() from WPNotif/inclues/filters.php
*
* @version 5.1.0
* @since 5.1.0
*
* @param $msg
* @param $order
*
* @return mixed
*/
function wpnotif_add_tracking_details( $msg, $order ) {
$tracking_link_string = '';
if ( class_exists( 'YITH_Tracking_Data' ) ) {
$values = array();
$yith_placeholders = apply_filters( 'ywsn_sms_placeholders', array(), $order );
if ( isset( $yith_placeholders['{tracking_url}'] ) ) {
$tracking_link_string = $yith_placeholders['{tracking_url}'];
}
} else {
$tracking_links = $this->wpnotif_get_wc_tracking_links( $order );
if ( ! empty( $tracking_links ) ) {
$tracking_link_string = implode( ',', $tracking_links );
}
}
$msg = str_replace( '{{wc-tracking-link}}', $tracking_link_string, $msg );
return $msg;
}
/**
* wpnotif_get_wc_tracking_links.
*
* @see wpnotif_get_wc_tracking_links() from WPNotif/inclues/filters.php
*
* @version 5.1.0
* @since 5.1.0
*
* @param $order
*
* @return array
*/
function wpnotif_get_wc_tracking_links( $order ) {
if ( class_exists( 'WC_Shipment_Tracking_Actions' ) ) {
$st = WC_Shipment_Tracking_Actions::get_instance();
} else if ( is_plugin_active( 'woo-advanced-shipment-tracking/woocommerce-advanced-shipment-tracking.php' ) ) {
$st = WC_Advanced_Shipment_Tracking_Actions::get_instance();
} else {
return array();
}
$order_id = $order->get_id();
$tracking_items = $st->get_tracking_items( $order_id );
if ( ! empty( $tracking_items ) ) {
$tracking_links = array();
foreach ( $tracking_items as $tracking_item ) {
$formated_item = $st->get_formatted_tracking_item( $order_id, $tracking_item );
$tracking_links[] = htmlspecialchars_decode( $formated_item['formatted_tracking_link'] );
break;
}
}
return $tracking_links;
}
/**
* maybe_add_meta_box.
*
* @version 3.5.0
* @since 3.5.0
* @todo re-think if setting number for yet not-numbered order should be allowed (i.e. do not check for `( '' !== get_post_meta( $post->ID, '_wcj_order_number', true ) )`)
*/
function maybe_add_meta_box( $post_type, $post ) {
if ( '' !== get_post_meta( $post->ID, '_wcj_order_number', true ) ) {
parent::add_meta_box();
}
}
/**
* woocommerce_subscriptions_remove_meta_copy.
*
* @version 3.2.3
* @since 3.2.3
*/
function woocommerce_subscriptions_remove_meta_copy( $meta, $to_order, $from_order ) {
foreach ( $meta as $meta_id => $meta_item ) {
if ( '_wcj_order_number' === $meta_item['meta_key'] ) {
unset( $meta[ $meta_id ] );
}
}
return $meta;
}
/**
* remove_order_tracking_sanitize_order_id_filter.
*
* @version 3.1.0
* @since 3.1.0
*/
function remove_order_tracking_sanitize_order_id_filter() {
remove_filter( 'woocommerce_shortcode_order_tracking_order_id', 'wc_sanitize_order_id' );
}
/**
* search_by_custom_number.
*
* @version 4.2.0
* @since 2.6.0
* @todo `_wcj_order_number` is used for `sequential` and `hash` only
*/
function search_by_custom_number( $query ) {
if (
! is_admin() ||
! property_exists( $query, "query" ) ||
! isset( $query->query['s'] ) ||
empty( $search = trim( $query->query['s'] ) ) ||
'shop_order' !== $query->query['post_type']
) {
return;
}
remove_action( 'pre_get_posts', array( $this, 'search_by_custom_number' ) );
// Get prefix and suffix options
$prefix = do_shortcode( wcj_get_option( 'wcj_order_number_prefix', '' ) );
$prefix .= date_i18n( wcj_get_option( 'wcj_order_number_date_prefix', '' ) );
$suffix = do_shortcode( wcj_get_option( 'wcj_order_number_suffix', '' ) );
$suffix .= date_i18n( wcj_get_option( 'wcj_order_number_date_suffix', '' ) );
// Ignore suffix and prefix from search input
$search_no_suffix = preg_replace( "/\A{$prefix}/i", '', $search );
$search_no_suffix_and_prefix = preg_replace( "/{$suffix}\z/i", '', $search_no_suffix );
$final_search = empty( $search_no_suffix_and_prefix ) ? $search : $search_no_suffix_and_prefix;
if($search == $final_search){
$final_search = substr($final_search,strlen( $prefix ));
$final_search = ltrim($final_search,0);
if(strlen( $suffix ) > 0)
{
$final_search = substr($final_search,0,-strlen( $suffix ));
}
}
// Post Status
$post_status = isset( $_GET['post_status'] ) ? $_GET['post_status'] : 'any';
// Try to search post by '_wcj_order_number' meta key
$meta_query_args = array(
array(
'key' => '_wcj_order_number',
'value' => $final_search,
'compare' => '='
)
);
$search_query = new WP_Query( array(
'fields' => 'ids',
'update_post_meta_cache' => false,
'update_post_term_cache' => false,
'post_status' => $post_status,
'post_type' => 'shop_order',
'meta_query' => $meta_query_args
) );
// If found, create the query by meta_key
if ( 1 == $search_query->found_posts ) {
$query->set( 'post_status', $post_status );
$query->set( 's', '' );
$query->set( 'post__in', array() );
$current_meta_query = empty( $query->get( 'meta_query' ) ) ? array() : $query->get( 'meta_query' );
$query->set( 'meta_query', array_merge( $current_meta_query, $meta_query_args ) );
} // If not found search by post_id
else {
$query->set( 'post_status', $post_status );
$query->set( 's', '' );
$current_post_in = empty( $query->get( 'post__in' ) ) ? array() : $query->get( 'post__in' );
$query->set( 'post__in', array_merge( $current_post_in, array( $final_search ) ) );
}
}
/**
* add_order_number_to_tracking.
*
* @version 3.1.0
* @since 2.5.2
*/
function add_order_number_to_tracking( $order_number ) {
$offset = 0;
$block_size = 512;
while( true ) {
$args = array(
'post_type' => 'shop_order',
'post_status' => 'any',
'posts_per_page' => $block_size,
'orderby' => 'ID',
'order' => 'DESC',
'offset' => $offset,
'fields' => 'ids',
);
$loop = new WP_Query( $args );
if ( ! $loop->have_posts() ) {
break;
}
foreach ( $loop->posts as $_order_id ) {
$_order = wc_get_order( $_order_id );
$_order_number = $this->display_order_number( $_order_id, $_order );
if ( $_order_number === $order_number ) {
return $_order_id;
}
}
$offset += $block_size;
}
return $order_number;
}
/**
* Display order number.
*
* @version 3.5.0
*/
function display_order_number( $order_number, $order ) {
$order_id = wcj_get_order_id( $order );
$order_number_meta = get_post_meta( $order_id , '_wcj_order_number', true );
if ( '' == $order_number_meta || 'no' === wcj_get_option( 'wcj_order_number_sequential_enabled', 'yes' ) ) {
$order_number_meta = $order_id;
}
$order_timestamp = strtotime( ( WCJ_IS_WC_VERSION_BELOW_3 ? $order->post->post_date : $order->get_date_created() ) );
$order_number = apply_filters( 'booster_option',
sprintf( '%s%s', do_shortcode( wcj_get_option( 'wcj_order_number_prefix', '' ) ), $order_number_meta ),
sprintf( '%s%s%0' . wcj_get_option( 'wcj_order_number_min_width', 0 ) . 's%s%s',
do_shortcode( wcj_get_option( 'wcj_order_number_prefix', '' ) ),
date_i18n( wcj_get_option( 'wcj_order_number_date_prefix', '' ), $order_timestamp ),
$order_number_meta,
do_shortcode( wcj_get_option( 'wcj_order_number_suffix', '' ) ),
date_i18n( wcj_get_option( 'wcj_order_number_date_suffix', '' ), $order_timestamp )
)
);
if ( false !== strpos( $order_number, '%order_items_skus%' ) ) {
$order_number = str_replace( '%order_items_skus%', do_shortcode( '[wcj_order_items order_id="' . $order_id . '" field="_sku" sep="-"]' ), $order_number );
}
return $order_number;
}
/**
* Add Renumerate Orders tool to WooCommerce menu (the content).
*
* @version 3.3.0
* @todo restyle
* @todo add more result info (e.g. number of regenerated orders etc.)
*/
function create_renumerate_orders_tool() {
$result_message = '';
if ( isset( $_POST['renumerate_orders'] ) ) {
$this->renumerate_orders();
$result_message = '<p><div class="updated"><p><strong>' . __( 'Orders successfully renumerated!', 'woocommerce-jetpack' ) . '</strong></p></div></p>';
} else {
if ( 'yes' === wcj_get_option( 'wcj_order_number_sequential_enabled', 'yes' ) ) {
$result_message .= '<p>' . sprintf( __( 'Sequential number generation is enabled. Next order number will be %s.', 'woocommerce-jetpack' ),
'<code>' . wcj_get_option( 'wcj_order_number_counter', 1 ) . '</code>' ) . '</p>';
}
}
$html = '';
$html .= '<div class="wrap">';
$html .= $this->get_tool_header_html( 'renumerate_orders' );
$html .= '<p>';
$html .= sprintf(
__( 'Press the button below to renumerate all existing orders starting from order counter settings in <a href="%s">Order Numbers</a> module.',
'woocommerce-jetpack' ),
admin_url( 'admin.php?page=wc-settings&tab=jetpack&wcj-cat=shipping_and_orders&section=order_numbers' )
);
$html .= '</p>';
$html .= $result_message;
$html .= '<form method="post" action="">';
$html .= '<input class="button-primary" type="submit" name="renumerate_orders" value="' . __( 'Renumerate orders', 'woocommerce-jetpack' ) . '">';
$html .= '</form>';
$html .= '</div>';
echo $html;
}
/**
* add_new_order_number.
*/
function add_new_order_number( $order_id ) {
$this->add_order_number_meta( $order_id, false );
}
/**
* maybe_reset_sequential_counter.
*
* @version 3.3.0
* @since 3.3.0
* @todo use transactions on `wcj_order_number_use_mysql_transaction_enabled`
*/
function maybe_reset_sequential_counter( $current_order_number, $order_id ) {
if ( 'no' != ( $reset_period = wcj_get_option( 'wcj_order_number_counter_reset_enabled', 'no' ) ) ) {
$previous_order_date = wcj_get_option( 'wcj_order_number_counter_previous_order_date', 0 );
$current_order_date = strtotime( wcj_get_order_date( wc_get_order( $order_id ) ) );
update_option( 'wcj_order_number_counter_previous_order_date', $current_order_date );
if ( 0 != $previous_order_date ) {
$do_reset = false;
switch ( $reset_period ) {
case 'daily':
$do_reset = (
date( 'Y', $current_order_date ) != date( 'Y', $previous_order_date ) ||
date( 'm', $current_order_date ) != date( 'm', $previous_order_date ) ||
date( 'd', $current_order_date ) != date( 'd', $previous_order_date )
);
break;
case 'monthly':
$do_reset = (
date( 'Y', $current_order_date ) != date( 'Y', $previous_order_date ) ||
date( 'm', $current_order_date ) != date( 'm', $previous_order_date )
);
break;
case 'yearly':
$do_reset = (
date( 'Y', $current_order_date ) != date( 'Y', $previous_order_date )
);
break;
}
if ( $do_reset ) {
return 1;
}
}
}
return $current_order_number;
}
/**
* Add/update order_number meta to order.
*
* @version 5.4.6
* @todo (maybe) save order ID instead of `$current_order_number = ''` (if `'no' === wcj_get_option( 'wcj_order_number_sequential_enabled', 'yes' )`)
*/
function add_order_number_meta( $order_id, $do_overwrite ) {
if ( 'shop_order' !== get_post_type( $order_id ) || 'auto-draft' === get_post_status( $order_id ) ) {
return;
}
if ( true === $do_overwrite || 0 == get_post_meta( $order_id, '_wcj_order_number', true ) || '' == get_post_meta( $order_id, '_wcj_order_number', true ) ) {
if ( $order_id < wcj_get_option( 'wcj_order_numbers_min_order_id', 0 ) ) {
return;
}
if ( 'yes' === wcj_get_option( 'wcj_order_number_sequential_enabled', 'yes' ) && 'yes' === wcj_get_option( 'wcj_order_number_use_mysql_transaction_enabled', 'yes' ) ) {
global $wpdb;
$wpdb->query( 'START TRANSACTION' );
$wp_options_table = $wpdb->prefix . 'options';
$result_select = $wpdb->get_row( "SELECT * FROM $wp_options_table WHERE option_name = 'wcj_order_number_counter'" );
if ( NULL != $result_select ) {
$current_order_number = $this->maybe_reset_sequential_counter( $result_select->option_value, $order_id );
$result_update = $wpdb->update( $wp_options_table, array( 'option_value' => ( $current_order_number + 1 ) ), array( 'option_name' => 'wcj_order_number_counter' ) );
if ( NULL != $result_update || $result_select->option_value == ( $current_order_number + 1 ) ) {
$wpdb->query( 'COMMIT' ); // all ok
update_post_meta( $order_id, '_wcj_order_number', apply_filters( 'wcj_order_number_meta', $current_order_number, $order_id ) );
} else {
$wpdb->query( 'ROLLBACK' ); // something went wrong, Rollback
}
} else {
$wpdb->query( 'ROLLBACK' ); // something went wrong, Rollback
}
} else {
if ( 'hash_crc32' === wcj_get_option( 'wcj_order_number_sequential_enabled', 'yes' ) ) {
$current_order_number = sprintf( "%u", crc32( $order_id ) );
} elseif ( 'yes' === wcj_get_option( 'wcj_order_number_sequential_enabled', 'yes' ) ) {
$current_order_number = $this->maybe_reset_sequential_counter( wcj_get_option( 'wcj_order_number_counter', 1 ), $order_id );
update_option( 'wcj_order_number_counter', ( $current_order_number + 1 ) );
} else { // 'no' === wcj_get_option( 'wcj_order_number_sequential_enabled', 'yes' ) // order ID
$current_order_number = '';
}
update_post_meta( $order_id, '_wcj_order_number', apply_filters( 'wcj_order_number_meta', $current_order_number, $order_id ) );
}
}
}
/**
* Renumerate orders function.
*
* @version 3.3.0
* @todo renumerate in date range only
* @todo (maybe) selectable `post_status`
* @todo (maybe) set default value for `wcj_order_numbers_renumerate_tool_orderby` to `ID` (instead of `date`)
*/
function renumerate_orders() {
if ( 'yes' === wcj_get_option( 'wcj_order_number_sequential_enabled', 'yes' ) && 'no' != wcj_get_option( 'wcj_order_number_counter_reset_enabled', 'no' ) ) {
update_option( 'wcj_order_number_counter_previous_order_date', 0 );
}
$offset = 0;
$block_size = 512;
while( true ) {
$args = array(
'post_type' => 'shop_order',
'post_status' => 'any',
'posts_per_page' => $block_size,
'orderby' => wcj_get_option( 'wcj_order_numbers_renumerate_tool_orderby', 'date' ),
'order' => wcj_get_option( 'wcj_order_numbers_renumerate_tool_order', 'ASC' ),
'offset' => $offset,
'fields' => 'ids',
);
$loop = new WP_Query( $args );
if ( ! $loop->have_posts() ) {
break;
}
foreach ( $loop->posts as $order_id ) {
$this->add_order_number_meta( $order_id, true );
}
$offset += $block_size;
}
}
}
endif;
return new WCJ_Order_Numbers();
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment