<?php
if ( ! defined( 'ABSPATH' ) ) { exit; }

class AIT_AIAA_Guidance_CPT {

    const CPT = 'ait_admin_guidance';

    public function __construct() {
        add_action( 'init', array( $this, 'register_cpt' ) );
        add_action( 'add_meta_boxes', array( $this, 'add_meta_boxes' ) );
        add_action( 'save_post', array( $this, 'save_meta' ), 10, 2 );
        add_filter( 'manage_edit-' . self::CPT . '_columns', array( $this, 'register_table_columns' ) );
        add_action( 'manage_' . self::CPT . '_posts_custom_column', array( $this, 'render_table_column' ), 10, 2 );

    }

    public function register_cpt() {

        $labels = array(
            'name'               => __( 'Admin Guidance', 'ait-ai-admin-assistance' ),
            'singular_name'      => __( 'Guidance Item', 'ait-ai-admin-assistance' ),
            'add_new'            => __( 'Add New', 'ait-ai-admin-assistance' ),
            'add_new_item'       => __( 'Add New Guidance Item', 'ait-ai-admin-assistance' ),
            'edit_item'          => __( 'Edit Guidance Item', 'ait-ai-admin-assistance' ),
            'new_item'           => __( 'New Guidance Item', 'ait-ai-admin-assistance' ),
            'view_item'          => __( 'View Guidance Item', 'ait-ai-admin-assistance' ),
            'search_items'       => __( 'Search Guidance Items', 'ait-ai-admin-assistance' ),
            'not_found'          => __( 'No guidance items found', 'ait-ai-admin-assistance' ),
            'not_found_in_trash' => __( 'No guidance items found in Trash', 'ait-ai-admin-assistance' ),
            'menu_name'          => __( 'Admin Guidance', 'ait-ai-admin-assistance' ),
        );

        register_post_type( self::CPT, array(
            'labels'             => $labels,
            'public'             => false,
            'show_ui'            => true,
            'show_in_menu'       => true,
            'menu_position'      => 59, // above Appearance (60)
            'menu_icon'          => 'dashicons-editor-help',
            'supports'           => array( 'title', 'editor' ),
            'capability_type'    => 'post',
            'map_meta_cap'       => true,
            'has_archive'        => false,
        ) );
    }

    
    public function register_table_columns( $columns ) {
        // Keep checkbox + title at the front.
        $new = array();
        if ( isset( $columns['cb'] ) ) { $new['cb'] = $columns['cb']; }
        if ( isset( $columns['title'] ) ) { $new['title'] = $columns['title']; }

        $new['ait_aiaa_priority']    = __( 'Priority', 'ait-ai-admin-assistance' );
        $new['ait_aiaa_target_type'] = __( 'Target Type', 'ait-ai-admin-assistance' );
        $new['ait_aiaa_field_level'] = __( 'Field-level', 'ait-ai-admin-assistance' );
        $new['ait_aiaa_hide_roles']  = __( 'Hide for Roles', 'ait-ai-admin-assistance' );

        // Append any remaining columns (e.g. date).
        foreach ( $columns as $k => $v ) {
            if ( isset( $new[ $k ] ) ) { continue; }
            $new[ $k ] = $v;
        }
        return $new;
    }

    public function render_table_column( $column, $post_id ) {
        if ( 'ait_aiaa_priority' === $column ) {
            $priority = get_post_meta( $post_id, '_ait_ag_priority', true );
            if ( '' === $priority ) { $priority = '1'; }
            echo esc_html( $priority );
            return;
        }

        if ( 'ait_aiaa_target_type' === $column ) {
            $target_type = get_post_meta( $post_id, '_ait_ag_target_type', true );
            if ( '' === $target_type ) { $target_type = 'global'; }

            $map = array(
                'global'    => __( 'Global', 'ait-ai-admin-assistance' ),
                'screen'    => __( 'Screen', 'ait-ai-admin-assistance' ),
                'post_type' => __( 'Post Type', 'ait-ai-admin-assistance' ),
                'taxonomy'  => __( 'Taxonomy', 'ait-ai-admin-assistance' ),
            );

            $label = isset( $map[ $target_type ] ) ? $map[ $target_type ] : $target_type;

            // Add a small hint value where helpful.
            if ( 'screen' === $target_type ) {
                $sid = get_post_meta( $post_id, '_ait_ag_screen_id', true );
                if ( $sid ) { $label .= ' — ' . $sid; }
            } elseif ( 'post_type' === $target_type ) {
                $pts = get_post_meta( $post_id, '_ait_ag_post_types', true );
                if ( is_array( $pts ) && ! empty( $pts ) ) { $label .= ' — ' . implode( ', ', $pts ); }
            } elseif ( 'taxonomy' === $target_type ) {
                $tax = get_post_meta( $post_id, '_ait_ag_taxonomies', true );
                if ( is_array( $tax ) && ! empty( $tax ) ) { $label .= ' — ' . implode( ', ', $tax ); }
            }

            echo esc_html( $label );
            return;
        }

        if ( 'ait_aiaa_field_level' === $column ) {
            $selectors = get_post_meta( $post_id, '_ait_ag_field_selectors', true );
            $has = ( is_string( $selectors ) && trim( $selectors ) !== '' );
            echo $has ? esc_html__( 'Yes', 'ait-ai-admin-assistance' ) : '&#8212;';
            return;
        }

        if ( 'ait_aiaa_hide_roles' === $column ) {
            $hide_roles = get_post_meta( $post_id, '_ait_ag_hide_roles', true );
            if ( ! is_array( $hide_roles ) ) { $hide_roles = array(); }
            if ( empty( $hide_roles ) ) { echo '&#8212;'; return; }

            $wp_roles = function_exists( 'wp_roles' ) ? wp_roles() : null;
            $all_roles = ( $wp_roles && isset( $wp_roles->roles ) && is_array( $wp_roles->roles ) ) ? $wp_roles->roles : array();

            $names = array();
            foreach ( $hide_roles as $rk ) {
                if ( isset( $all_roles[ $rk ]['name'] ) ) { $names[] = $all_roles[ $rk ]['name']; }
                else { $names[] = $rk; }
            }
            echo esc_html( implode( ', ', $names ) );
            return;
        }
    }

public function add_meta_boxes() {
        add_meta_box(
            'ait-aiaa-guidance-meta',
            __( 'Guidance Targeting', 'ait-ai-admin-assistance' ),
            array( $this, 'render_meta_box' ),
            self::CPT,
            'side',
            'default'
        );
    }

    public function render_meta_box( $post ) {

        wp_nonce_field( 'ait_aiaa_guidance_meta', 'ait_aiaa_guidance_meta_nonce' );

        $enabled     = get_post_meta( $post->ID, '_ait_ag_enabled', true );
        $priority    = get_post_meta( $post->ID, '_ait_ag_priority', true );
        $target_type = get_post_meta( $post->ID, '_ait_ag_target_type', true );
        $screen_id   = get_post_meta( $post->ID, '_ait_ag_screen_id', true );

        // New multi-select fields.
        $post_types  = get_post_meta( $post->ID, '_ait_ag_post_types', true );
        $taxonomies  = get_post_meta( $post->ID, '_ait_ag_taxonomies', true );

        
        $field_selectors = get_post_meta( $post->ID, '_ait_ag_field_selectors', true );
        $field_selectors = is_string( $field_selectors ) ? $field_selectors : '';
// Back-compat with old single post_type field.
        if ( ( empty( $post_types ) || ! is_array( $post_types ) ) ) {
            $old_post_type = get_post_meta( $post->ID, '_ait_ag_post_type', true );
            if ( ! empty( $old_post_type ) && is_string( $old_post_type ) ) {
                $post_types = array( $old_post_type );
            }
        }

        if ( ! is_array( $post_types ) ) { $post_types = array(); }
        if ( ! is_array( $taxonomies ) ) { $taxonomies = array(); }

        // Hide for roles.
        $hide_roles = get_post_meta( $post->ID, '_ait_ag_hide_roles', true );
        if ( ! is_array( $hide_roles ) ) { $hide_roles = array(); }

        $wp_roles = function_exists( 'wp_roles' ) ? wp_roles() : null;
        $all_roles = ( $wp_roles && isset( $wp_roles->roles ) && is_array( $wp_roles->roles ) ) ? $wp_roles->roles : array();


        if ( '' === $target_type ) { $target_type = 'global'; }
        if ( '' === $enabled ) { $enabled = '1'; }
        if ( '' === $priority ) { $priority = '10'; }

        $all_post_types = get_post_types( array( 'show_ui' => true ), 'objects' );
        $all_taxonomies = get_taxonomies( array( 'show_ui' => true ), 'objects' );

        ?>
        <p>
            <label>
                <input type="checkbox" name="ait_ag_enabled" value="1" <?php checked( $enabled, '1' ); ?> />
                <?php esc_html_e( 'Enabled', 'ait-ai-admin-assistance' ); ?>
            </label>
        </p>

        <p>
            <label for="ait_ag_priority"><strong><?php esc_html_e( 'Priority', 'ait-ai-admin-assistance' ); ?></strong></label><br/>
            <input type="number" name="ait_ag_priority" id="ait_ag_priority" value="<?php echo esc_attr( $priority ); ?>" min="1" max="999" style="width:100%;" />
        </p>

        <p>
            <label for="ait_ag_target_type"><strong><?php esc_html_e( 'Target Type', 'ait-ai-admin-assistance' ); ?></strong></label><br/>
            <select name="ait_ag_target_type" id="ait_ag_target_type" style="width:100%;">
                <option value="global" <?php selected( $target_type, 'global' ); ?>><?php esc_html_e( 'Global', 'ait-ai-admin-assistance' ); ?></option>
                <option value="screen" <?php selected( $target_type, 'screen' ); ?>><?php esc_html_e( 'Admin screen', 'ait-ai-admin-assistance' ); ?></option>
                <option value="post_type" <?php selected( $target_type, 'post_type' ); ?>><?php esc_html_e( 'Post type', 'ait-ai-admin-assistance' ); ?></option>
                <option value="taxonomy" <?php selected( $target_type, 'taxonomy' ); ?>><?php esc_html_e( 'Taxonomy', 'ait-ai-admin-assistance' ); ?></option>
            </select>
        </p>

        <div class="ait-aiaa-target-screen">
            <p>
                <label for="ait_ag_screen_id"><strong><?php esc_html_e( 'Screen ID', 'ait-ai-admin-assistance' ); ?></strong></label><br/>
                <input type="text" name="ait_ag_screen_id" id="ait_ag_screen_id" value="<?php echo esc_attr( $screen_id ); ?>" style="width:100%;" />
                <span class="description"><?php esc_html_e( 'Example: edit-post, post, widgets, nav-menus.', 'ait-ai-admin-assistance' ); ?></span>
            </p>
        </div>

        <div class="ait-aiaa-target-post-type">
            <p><strong><?php esc_html_e( 'Post type', 'ait-ai-admin-assistance' ); ?></strong></p>
            <?php foreach ( $all_post_types as $pt_key => $pt_obj ) : ?>
                <label style="display:block; margin: 2px 0;">
                    <input type="checkbox" name="ait_ag_post_types[]" value="<?php echo esc_attr( $pt_key ); ?>" <?php checked( in_array( $pt_key, $post_types, true ) ); ?> />
                    <?php echo esc_html( $pt_obj->labels->singular_name ); ?>
                    <span class="description">(<?php echo esc_html( $pt_key ); ?>)</span>
                </label>
            <?php endforeach; ?>
            <p class="description"><?php esc_html_e( 'Shown only when Target Type is "Post type".', 'ait-ai-admin-assistance' ); ?></p>
        </div>

        <div class="ait-aiaa-target-taxonomy">
            <p><strong><?php esc_html_e( 'Taxonomy type', 'ait-ai-admin-assistance' ); ?></strong></p>
            <?php foreach ( $all_taxonomies as $tx_key => $tx_obj ) : ?>
                <label style="display:block; margin: 2px 0;">
                    <input type="checkbox" name="ait_ag_taxonomies[]" value="<?php echo esc_attr( $tx_key ); ?>" <?php checked( in_array( $tx_key, $taxonomies, true ) ); ?> />
                    <?php echo esc_html( $tx_obj->labels->singular_name ); ?>
                    <span class="description">(<?php echo esc_html( $tx_key ); ?>)</span>
                </label>
            <?php endforeach; ?>
            <p class="description"><?php esc_html_e( 'Shown only when Target Type is "Taxonomy".', 'ait-ai-admin-assistance' ); ?></p>
        </div>
        <div class="ait-aiaa-target-fields" style="margin-top:10px;">
            <p><strong><?php esc_html_e( 'Field-level guidance selectors (Pro)', 'ait-ai-admin-assistance' ); ?></strong></p>
            <textarea name="ait_ag_field_selectors" rows="4" style="width:100%;" placeholder="#titlewrap, .postbox"><?php echo esc_textarea( $field_selectors ); ?></textarea>
            <p class="description"><?php esc_html_e( 'Optional. Newline-separated CSS selectors. When Pro is active, matching fields will show this guidance inline on the page.', 'ait-ai-admin-assistance' ); ?></p>
        </div>


        <div class="ait-aiaa-target-fields" style="margin-top:10px;">
            <p><strong><?php esc_html_e( 'Hide for Roles', 'ait-ai-admin-assistance' ); ?></strong></p>
            <div class="ait-aiaa-hide-roles-checkboxes" style="max-height:160px; overflow:auto; border:1px solid #ccd0d4; padding:8px; background:#fff;">
                <?php foreach ( $all_roles as $role_key => $role_data ) : ?>
                    <?php $role_name = isset( $role_data['name'] ) ? $role_data['name'] : $role_key; ?>
                    <label style="display:block; margin:0 0 6px 0;">
                        <input type="checkbox" name="ait_ag_hide_roles[]" value="<?php echo esc_attr( $role_key ); ?>" <?php checked( in_array( $role_key, $hide_roles, true ) ); ?> />
                        <?php echo esc_html( $role_name ); ?>
                    </label>
                <?php endforeach; ?>
            </div>
            <p class="description"><?php esc_html_e( 'Optional. If selected, this guidance item will not be shown to users with those roles.', 'ait-ai-admin-assistance' ); ?></p>
        </div>
        <?php
    }

    public function save_meta( $post_id, $post ) {

        if ( self::CPT !== $post->post_type ) { return; }
        if ( defined( 'DOING_AUTOSAVE' ) && DOING_AUTOSAVE ) { return; }
        if ( ! isset( $_POST['ait_aiaa_guidance_meta_nonce'] ) || ! wp_verify_nonce( sanitize_text_field( wp_unslash( $_POST['ait_aiaa_guidance_meta_nonce'] ) ), 'ait_aiaa_guidance_meta' ) ) { return; }
        if ( ! current_user_can( 'edit_post', $post_id ) ) { return; }

        $enabled     = isset( $_POST['ait_ag_enabled'] ) ? '1' : '0';
        $priority    = isset( $_POST['ait_ag_priority'] ) ? absint( wp_unslash( $_POST['ait_ag_priority'] ) ) : 10;
        $target_type = isset( $_POST['ait_ag_target_type'] ) ? sanitize_text_field( wp_unslash( $_POST['ait_ag_target_type'] ) ) : 'global';
        $screen_id   = isset( $_POST['ait_ag_screen_id'] ) ? sanitize_text_field( wp_unslash( $_POST['ait_ag_screen_id'] ) ) : '';

        $post_types = isset( $_POST['ait_ag_post_types'] ) ? (array) wp_unslash( $_POST['ait_ag_post_types'] ) : array();
        $post_types = array_values( array_filter( array_map( 'sanitize_text_field', $post_types ) ) );

        $taxonomies = isset( $_POST['ait_ag_taxonomies'] ) ? (array) wp_unslash( $_POST['ait_ag_taxonomies'] ) : array();
        $taxonomies = array_values( array_filter( array_map( 'sanitize_text_field', $taxonomies ) ) );    

        $hide_roles = isset( $_POST['ait_ag_hide_roles'] ) ? (array) wp_unslash( $_POST['ait_ag_hide_roles'] ) : array();
        $hide_roles = array_values( array_filter( array_map( 'sanitize_text_field', $hide_roles ) ) );

        $field_selectors = isset( $_POST['ait_ag_field_selectors'] ) ? sanitize_textarea_field( wp_unslash( $_POST['ait_ag_field_selectors'] ) ) : '';
        
        update_post_meta( $post_id, '_ait_ag_enabled', $enabled );
        update_post_meta( $post_id, '_ait_ag_priority', (string) $priority );
        update_post_meta( $post_id, '_ait_ag_target_type', $target_type );

        // Screen ID is only meaningful for target_type=screen.
        update_post_meta( $post_id, '_ait_ag_screen_id', $screen_id );

        // Store arrays for multi-targeting.
        update_post_meta( $post_id, '_ait_ag_post_types', $post_types );
        update_post_meta( $post_id, '_ait_ag_taxonomies', $taxonomies );
        update_post_meta( $post_id, '_ait_ag_hide_roles', $hide_roles );
    
        update_post_meta( $post_id, '_ait_ag_field_selectors', $field_selectors );
}
}