<?php
/**
 * Address Fields Override for Block Checkout
 * 
 * This mirrors how ThemeHigh does it - override woocommerce_default_address_fields
 * and inject custom fields with priority BEFORE WooCommerce processes them
 */

if (!defined('ABSPATH')) {
    exit;
}

class CFEP_Address_Fields_Override {
    
    private static $instance = null;
    
    public static function instance() {
        if (is_null(self::$instance)) {
            self::$instance = new self();
        }
        return self::$instance;
    }
    
    private function __construct() {
        // Hook at priority 1 - BEFORE WooCommerce processes (they use priority 10)
        add_filter('woocommerce_default_address_fields', array($this, 'inject_custom_fields'), 1);
        
        // Also hook the locale system
        add_filter('woocommerce_get_country_locale_base', array($this, 'add_to_locale'), 1);
        add_filter('woocommerce_get_country_locale_default', array($this, 'add_to_locale'), 1);
        
    }
    
    /**
     * Inject custom fields into default address fields
     * This is the KEY - these become "native" address fields that Block Checkout will sort by priority
     */
    public function inject_custom_fields($fields) {
        global $wpdb;
        
        $table_name = $wpdb->prefix . 'cfep_checkout_fields';
        
        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching, WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared, PluginCheck.Security.DirectDB.UnescapedDBParameter
        if ($wpdb->get_var("SHOW TABLES LIKE '$table_name'") != $table_name) {
            return $fields;
        }
        
        $cache_key = 'cfep_custom_address_fields';
        $custom_fields = wp_cache_get($cache_key, 'cfep');
        if ($custom_fields === false) {
            // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching, WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared, PluginCheck.Security.DirectDB.UnescapedDBParameter
            $custom_fields = $wpdb->get_results( "SELECT field_key, field_type, field_label, position, required, placeholder FROM {$table_name} WHERE enabled = 1 AND section IN ('billing', 'shipping') ORDER BY position ASC" );
            wp_cache_set($cache_key, $custom_fields, 'cfep', 300);
        }
        
        if (empty($custom_fields)) {
            return $fields;
        }
        
        
        foreach ($custom_fields as $field) {
            // Remove prefixes to get clean field key
            $clean_key = str_replace(array('billing_', 'shipping_', 'cfep_', 'cfep-'), '', $field->field_key);
            $clean_key = str_replace('-', '_', $clean_key);
            
            // Build field configuration EXACTLY like WooCommerce native fields
            $field_config = array(
                'label' => $field->field_label,
                'required' => (bool) $field->required,
                'priority' => (int) $field->position, // THIS IS THE KEY!
                'class' => array('form-row-wide'),
                'autocomplete' => 'off',
            );
            
            if (!empty($field->placeholder)) {
                $field_config['placeholder'] = $field->placeholder;
            }
            
            // Map field types
            switch ($field->field_type) {
                case 'select':
                    $field_config['type'] = 'select';
                    // Get options from database
                    $options = $this->get_field_options($field->field_key);
                    if (!empty($options)) {
                        $field_config['options'] = $options;
                    }
                    break;
                case 'textarea':
                    $field_config['type'] = 'textarea';
                    break;
                default:
                    $field_config['type'] = 'text';
                    break;
            }
            
            // Add to fields array - it will be sorted by priority later
            $fields[$clean_key] = $field_config;
            
        }
        
        return $fields;
    }
    
    /**
     * Add custom fields to locale array as well
     */
    public function add_to_locale($locale) {
        global $wpdb;
        
        $table_name = $wpdb->prefix . 'cfep_checkout_fields';
        
        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching, WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared, PluginCheck.Security.DirectDB.UnescapedDBParameter
        if ($wpdb->get_var("SHOW TABLES LIKE '$table_name'") != $table_name) {
            return $locale;
        }
        
        $cache_key = 'cfep_locale_address_fields';
        $custom_fields = wp_cache_get($cache_key, 'cfep');
        if ($custom_fields === false) {
            // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching, WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared, PluginCheck.Security.DirectDB.UnescapedDBParameter
            $custom_fields = $wpdb->get_results( "SELECT field_key, field_label, position, required FROM {$table_name} WHERE enabled = 1 AND section IN ('billing', 'shipping')" );
            wp_cache_set($cache_key, $custom_fields, 'cfep', 300);
        }
        
        foreach ($custom_fields as $field) {
            $clean_key = str_replace(array('billing_', 'shipping_', 'cfep_', 'cfep-'), '', $field->field_key);
            $clean_key = str_replace('-', '_', $clean_key);
            
            $locale[$clean_key] = array(
                'label' => $field->field_label,
                'required' => (bool) $field->required,
                'priority' => (int) $field->position,
            );
        }
        
        return $locale;
    }
    
    /**
     * Get field options for select fields
     */
    private function get_field_options($field_key) {
        global $wpdb;
        
        $table_name = $wpdb->prefix . 'cfep_checkout_fields';
        
        $cache_key = 'cfep_field_options_' . $field_key;
        $field_options = wp_cache_get($cache_key, 'cfep');
        if ($field_options === false) {
            // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching, WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared, PluginCheck.Security.DirectDB.UnescapedDBParameter
            $field_options = $wpdb->get_var( $wpdb->prepare( "SELECT field_options FROM {$table_name} WHERE field_key = %s", $field_key ) );
            wp_cache_set($cache_key, $field_options, 'cfep', 300);
        }
        
        if (empty($field_options)) {
            return array();
        }
        
        $options = array();
        $lines = explode("\n", $field_options);
        
        foreach ($lines as $line) {
            $line = trim($line);
            if (empty($line)) continue;
            
            $options[$line] = $line;
        }
        
        return $options;
    }
}