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

<?php
// Handles custom bulk actions for orders.

defined('ABSPATH') || exit;

class Bulk_Actions_Manager {
    public function __construct() {
        add_filter('bulk_actions-woocommerce_page_wc-orders', array($this, 'register_bulk_actions'));
        add_filter('handle_bulk_actions-woocommerce_page_wc-orders', array($this, 'handle_bulk_actions'), 10, 3);
    }

    public function register_bulk_actions($bulk_actions) {
        $bulk_actions['prepare_to_printing_export']       = __('Prepare to Printing Export', 'studiou-wc-ord-print-statuses');
        $bulk_actions['set_status_to_prepare_to_printing'] = __('Set Status to Prepare to Printing', 'studiou-wc-ord-print-statuses');
        $bulk_actions['set_status_to_in_printing']         = __('Set Status to In Printing', 'studiou-wc-ord-print-statuses');
        return $bulk_actions;
    }

    public function handle_bulk_actions($redirect_to, $action, $post_ids) {
        $known = array(
            'prepare_to_printing_export',
            'set_status_to_prepare_to_printing',
            'set_status_to_in_printing',
        );
        if (!in_array($action, $known, true)) {
            return $redirect_to;
        }
        if (!current_user_can('edit_shop_orders')) {
            wp_die(esc_html__('You do not have sufficient permissions to perform this action.', 'studiou-wc-ord-print-statuses'));
        }
        $post_ids = array_filter(array_map('intval', (array) $post_ids));
        if (empty($post_ids)) {
            return $redirect_to;
        }

        switch ($action) {
            case 'prepare_to_printing_export':
                return $this->prepare_to_printing_export($post_ids, $redirect_to);
            case 'set_status_to_prepare_to_printing':
                return $this->set_status_to_prepare_to_printing($post_ids, $redirect_to);
            case 'set_status_to_in_printing':
                return $this->set_status_to_in_printing($post_ids, $redirect_to);
        }
        return $redirect_to;
    }

    /**
     * Atomic order: pull rows, build the CSV in memory, then flip statuses, then
     * stream the file. If the export query returns nothing, statuses are still
     * advanced (the spec requires it) but the user is told no CSV was produced.
     */
    private function prepare_to_printing_export($post_ids, $redirect_to) {
        $orders_data = Studiou_DB_Manager::get_orders_for_printing($post_ids);
        $csv_data    = !empty($orders_data) ? $this->generate_csv($orders_data) : '';

        $updated = Studiou_DB_Manager::set_orders_to_prepare_printing($post_ids);
        UtilsLog::message(
            sprintf(
                _n(
                    '%d order moved to "Prepare to Printing".',
                    '%d orders moved to "Prepare to Printing".',
                    $updated,
                    'studiou-wc-ord-print-statuses'
                ),
                $updated
            ),
            'success'
        );

        if ($csv_data === '') {
            UtilsLog::message(
                __('No exportable line items were found for the selected orders. CSV was not generated.', 'studiou-wc-ord-print-statuses'),
                'warning'
            );
            return $redirect_to;
        }

        $this->output_csv($csv_data, 'prepare_to_printing_export.csv'); // exits
    }

    private function set_status_to_prepare_to_printing($post_ids, $redirect_to) {
        $updated = Studiou_DB_Manager::set_orders_to_prepare_printing($post_ids);
        UtilsLog::message(
            sprintf(
                _n(
                    '%d order moved to "Prepare to Printing".',
                    '%d orders moved to "Prepare to Printing".',
                    $updated,
                    'studiou-wc-ord-print-statuses'
                ),
                $updated
            ),
            'success'
        );
        return add_query_arg('bulk_action', 'marked_prepare_to_printing', $redirect_to);
    }

    private function set_status_to_in_printing($post_ids, $redirect_to) {
        $updated = Studiou_DB_Manager::set_orders_to_in_printing($post_ids);
        UtilsLog::message(
            sprintf(
                _n(
                    '%d order moved to "In Printing".',
                    '%d orders moved to "In Printing".',
                    $updated,
                    'studiou-wc-ord-print-statuses'
                ),
                $updated
            ),
            'success'
        );
        return add_query_arg('bulk_action', 'marked_in_printing', $redirect_to);
    }

    private function generate_csv($data) {
        $output = fopen('php://temp', 'w+');
        fputcsv($output, array_keys((array) $data[0]), ',');
        foreach ($data as $row) {
            fputcsv($output, (array) $row, ',');
        }
        rewind($output);
        $csv = stream_get_contents($output);
        fclose($output);
        return $csv;
    }

    /**
     * Stream CSV with UTF-8 BOM so Excel on Windows reads diacritics correctly.
     */
    private function output_csv($csv_data, $filename) {
        if (!headers_sent()) {
            nocache_headers();
            header('Content-Type: text/csv; charset=utf-8');
            header('Content-Disposition: attachment; filename="' . $filename . '"');
        }
        echo "\xEF\xBB\xBF" . $csv_data; // UTF-8 BOM
        exit;
    }
}