qdr.app.woo.plugins/studiou-wc-ord-print-statuses/includes/class-order-status-manager.php

<?php
// Handles custom order status registration and integration.

defined('ABSPATH') || exit;

class Order_Status_Manager {
    private $custom_statuses;
    private $print_status_slugs = array('to-print', 'in-print');

    public function __construct() {
        add_action('init', array($this, 'register_custom_order_statuses'));
        add_filter('wc_order_statuses', array($this, 'add_custom_order_statuses_to_list'));
        add_action('woocommerce_order_status_changed', array($this, 'handle_status_transitions'), 10, 3);
        add_action('woocommerce_payment_complete', array($this, 'set_order_processing_status'));
    }

    private function get_custom_status_definitions() {
        if ($this->custom_statuses === null) {
            $this->custom_statuses = array(
                'wc-to-print' => array(
                    'label'       => _x('Prepare to Printing', 'Order status', 'studiou-wc-ord-print-statuses'),
                    'label_count' => _n_noop(
                        'Prepare to Printing <span class="count">(%s)</span>',
                        'Prepare to Printing <span class="count">(%s)</span>',
                        'studiou-wc-ord-print-statuses'
                    ),
                ),
                'wc-in-print' => array(
                    'label'       => _x('In Printing', 'Order status', 'studiou-wc-ord-print-statuses'),
                    'label_count' => _n_noop(
                        'In Printing <span class="count">(%s)</span>',
                        'In Printing <span class="count">(%s)</span>',
                        'studiou-wc-ord-print-statuses'
                    ),
                ),
            );
        }
        return $this->custom_statuses;
    }

    public function register_custom_order_statuses() {
        foreach ($this->get_custom_status_definitions() as $status_slug => $status_args) {
            register_post_status($status_slug, array(
                'label'                     => $status_args['label'],
                'public'                    => true,
                'exclude_from_search'       => false,
                'show_in_admin_all_list'    => true,
                'show_in_admin_status_list' => true,
                'label_count'               => $status_args['label_count'],
            ));
        }
    }

    public function add_custom_order_statuses_to_list($order_statuses) {
        $new_order_statuses = array();
        foreach ($order_statuses as $key => $status) {
            $new_order_statuses[$key] = $status;
            if ($key === 'wc-processing') {
                foreach ($this->get_custom_status_definitions() as $status_slug => $status_args) {
                    $new_order_statuses[$status_slug] = $status_args['label'];
                }
            }
        }
        return $new_order_statuses;
    }

    /**
     * Stamp the relevant meta key when an order transitions into a print status.
     * Uses the order object (HPOS-safe) rather than update_post_meta.
     */
    public function handle_status_transitions($order_id, $old_status, $new_status) {
        UtilsLog::log('Order #' . $order_id . ' status change ' . $old_status . ' -> ' . $new_status);

        if (!in_array($new_status, $this->print_status_slugs, true)) {
            return;
        }
        $order = wc_get_order($order_id);
        if (!$order) {
            return;
        }
        $meta_key = ($new_status === 'to-print') ? 'to_print_date' : 'in_print_date';
        if (!$order->get_meta($meta_key)) {
            $order->update_meta_data($meta_key, current_time('mysql'));
            $order->save();
        }
    }

    /**
     * Move paid orders to "processing" on payment completion — except when the
     * order is already past that point (completed) or has been advanced into the
     * print pipeline (to-print, in-print). Otherwise late payments would silently
     * roll a print-state order back to processing.
     */
    public function set_order_processing_status($order_id) {
        $order = wc_get_order($order_id);
        if (!$order) {
            return;
        }
        $protected = array_merge(array('completed'), $this->print_status_slugs);
        if ($order->has_status($protected)) {
            return;
        }
        $order->update_status(
            'processing',
            __('Payment received, order is now processing.', 'studiou-wc-ord-print-statuses')
        );
        $order->add_order_note(
            __('Order automatically set to processing by Studiou WC Order Print Statuses plugin.', 'studiou-wc-ord-print-statuses')
        );
    }

    public function get_custom_statuses() {
        return $this->get_custom_status_definitions();
    }
}