<?php

namespace AffiliateLauncher\WooCommerce;

if (!defined('ABSPATH')) {
    exit;
}

/**
 * Writes AffiliateLauncher attribution onto Woo orders (classic + Store API / blocks).
 */
final class Checkout
{
    /** @var self|null */
    private static $instance = null;

    public static function instance(): self
    {
        if (self::$instance === null) {
            self::$instance = new self();
        }
        return self::$instance;
    }

    public function init(): void
    {
        add_action('woocommerce_checkout_create_order', [$this, 'attach_meta_to_order'], 20, 2);
        add_action(
            'woocommerce_store_api_checkout_update_order_from_request',
            [$this, 'attach_meta_from_store_api'],
            20,
            2
        );
        add_action('woocommerce_checkout_order_processed', [$this, 'attach_meta_after_classic'], 20, 1);
        add_action('woocommerce_store_api_checkout_order_processed', [$this, 'attach_meta_after_store_api'], 20, 1);
    }

    public function attach_meta_to_order($order, $data = null): void
    {
        if (!$order instanceof \WC_Order) {
            return;
        }
        $this->write_meta($order);
    }

    public function attach_meta_from_store_api($order, $request = null): void
    {
        if (!$order instanceof \WC_Order) {
            return;
        }
        $this->write_meta($order);
    }

    public function attach_meta_after_classic($order_id): void
    {
        $order = wc_get_order($order_id);
        if ($order) {
            $this->write_meta($order);
            $order->save();
        }
    }

    public function attach_meta_after_store_api($order): void
    {
        if ($order instanceof \WC_Order) {
            $this->write_meta($order);
            $order->save();
        }
    }

    private function write_meta(\WC_Order $order): void
    {
        $state = Attribution::get_state();
        if ($state['session_id'] === '') {
            $state = Attribution::set_state($state);
        }
        $order->update_meta_data('_fsession', $state['session_id']);
        if ($state['distributor_id'] !== '') {
            $order->update_meta_data('foresight-distributor-id', $state['distributor_id']);
            $order->update_meta_data('_fdist', $state['distributor_id']);
        }
        if ($state['event_id'] !== '') {
            $order->update_meta_data('foresight-event-id', $state['event_id']);
            $order->update_meta_data('_fevent', $state['event_id']);
        }
        if ($state['event_title'] !== '') {
            $order->update_meta_data('foresight-event-title', $state['event_title']);
        }
    }
}