<?php
    /*
     * Ajax functions
     */

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

    // Get states by country code
    function acfcs_get_states_call() {
        if ( isset( $_POST[ 'acfcs_state_nonce' ] ) && wp_verify_nonce( sanitize_text_field( wp_unslash( $_POST[ 'acfcs_state_nonce' ] ) ), 'acfcs-state-nonce' ) ) {
            if ( isset( $_POST[ 'country_code' ] ) ) {
                $field   = [];
                $items   = array();
                $post_id = ( isset( $_POST[ 'post_id' ] ) ) ? (int) $_POST[ 'post_id' ] : false;

                if ( is_string( $_POST[ 'country_code' ] ) ) {
                    $country_code = sanitize_text_field( wp_unslash( $_POST[ 'country_code' ] ) );
                }

                if ( false != $post_id ) {
                    $fields = get_field_objects( $post_id );
                    if ( is_array( $fields ) && ! empty( $fields ) ) {
                        $field = acfcs_get_field_settings( $fields );
                    }
                }

                if ( ! isset( $field[ 'show_labels' ] ) && isset( $_POST[ 'show_labels' ] ) ) {
                    $field[ 'show_labels' ] = ( '1' == sanitize_text_field( wp_unslash( $_POST[ 'show_labels' ] ) ) ) ? true : false;
                }

                if ( isset( $country_code ) ) {
                    $states_transient = acfcs_get_states( $country_code, true, $field );
                }

                if ( isset( $states_transient ) && ! empty( $states_transient ) ) {
                    foreach ( $states_transient as $key => $label ) {
                        if ( $label != 'N/A' ) {
                            $items[] = [
                                'state_name'    => $label,
                                'country_state' => $key,
                            ];
                        } else {
                            $items[] = [
                                'state_name'    => $country_code,
                                'country_state' => $key,
                            ];
                        }
                    }
                    echo wp_json_encode( $items );
                    wp_die();
                }
            }
        }
    }
    add_action( 'wp_ajax_get_states_call', 'acfcs_get_states_call' );
    add_action( 'wp_ajax_nopriv_get_states_call', 'acfcs_get_states_call' );

    // Get cities by state code and/or country code
    function acfcs_get_cities_call() {
        if ( isset( $_POST[ 'acfcs_city_nonce' ] ) && wp_verify_nonce( sanitize_text_field( wp_unslash( $_POST[ 'acfcs_city_nonce' ] ) ), 'acfcs-city-nonce' ) ) {
            if ( isset( $_POST[ 'state_code' ] ) ) {
                $country_code      = false;
                $field             = false;
                $items             = array();
                $post_id           = ( isset( $_POST[ 'post_id' ] ) ) ? (int) $_POST[ 'post_id' ] : false;
                $posted_state_code = sanitize_text_field( wp_unslash( $_POST[ 'state_code' ] ) );
                $state_code        = false;

                if ( false != $post_id ) {
                    $fields = get_field_objects( $post_id );
                    if ( ! empty( $fields ) ) {
                        $field = acfcs_get_field_settings( $fields );
                    }
                }

                if ( ! isset( $field[ 'show_labels' ] ) && isset( $_POST[ 'show_labels' ] ) ) {
                    $show_labels = sanitize_text_field( wp_unslash( $_POST[ 'show_labels' ] ) );
                    if ( '1' == $show_labels ) {
                        $field[ 'show_labels' ] = true;
                    } elseif ( '0' == $show_labels ) {
                        $field[ 'show_labels' ] = false;
                    }
                }

                if ( 6 <= strlen( $posted_state_code ) ) {
                    $codes        = explode( '-', $posted_state_code );
                    $country_code = $codes[ 0 ];
                    $state_code   = $codes[ 1 ];
                } elseif ( strpos( $posted_state_code, 'FR-' ) !== false ) {
                    $country_code = substr( $posted_state_code, 0, 2 );
                    $state_code   = substr( $posted_state_code, 3 );
                } elseif ( 2 == strlen( $posted_state_code ) ) {
                    // if 2 == strlen( $posted_state_code ) then it's probably a country code
                    // this is probably never reached, but just in case...
                    $country_code = $posted_state_code;
                    $state_code   = false;
                } elseif ( ! empty( $posted_state_code ) ) {
                    $codes        = explode( '-', $posted_state_code );
                    $country_code = $codes[ 0 ];
                    $state_code   = $codes[ 1 ];
                } else {
                    // fallback if all else fails
                    if ( isset( $field[ 'default_country' ] ) && ! empty( $field[ 'default_country' ] ) ) {
                        $country_code = $field[ 'default_country' ];
                    }
                }

                $cities_transient = acfcs_get_cities( $country_code, $state_code, $field );

                if ( ! empty( $cities_transient ) ) {
                    foreach ( $cities_transient as $city ) {
                        $items[] = [
                            'id'        => $city,
                            'city_name' => $city,
                        ];
                    }
                    echo wp_json_encode( $items );
                    wp_die();
                }
            }
        }
    }
    add_action( 'wp_ajax_get_cities_call', 'acfcs_get_cities_call' );
    add_action( 'wp_ajax_nopriv_get_cities_call', 'acfcs_get_cities_call' );