<?php

namespace Classifai\Features;

use Classifai\Services\LanguageProcessing;
use Classifai\Providers\Watson\NLU;
use Classifai\Providers\OpenAI\Embeddings as OpenAIEmbeddings;
use Classifai\Providers\Azure\Embeddings as AzureEmbeddings;
use Classifai\Providers\Localhost\OllamaEmbeddings;
use WP_REST_Server;
use WP_REST_Request;
use WP_Error;

use function Classifai\get_post_statuses_for_language_settings;
use function Classifai\get_post_types_for_language_settings;
use function Classifai\check_term_permissions;
use function Classifai\get_classification_feature_enabled;
use function Classifai\get_classification_feature_taxonomy;
use function Classifai\get_asset_info;
use function Classifai\get_classification_mode;

/**
 * Class Classification
 */
class Classification extends Feature {
	/**
	 * ID of the current feature.
	 *
	 * @var string
	 */
	const ID = 'feature_classification';

	/**
	 * Constructor.
	 */
	public function __construct() {
		$this->label = __( 'Classification', 'classifai' );

		// Contains all providers that are registered to the service.
		$this->provider_instances = $this->get_provider_instances( LanguageProcessing::get_service_providers() );

		// Contains just the providers this feature supports.
		$this->supported_providers = [
			NLU::ID              => __( 'IBM Watson NLU', 'classifai' ),
			OpenAIEmbeddings::ID => __( 'OpenAI Embeddings', 'classifai' ),
			AzureEmbeddings::ID  => __( 'Azure OpenAI Embeddings', 'classifai' ),
			OllamaEmbeddings::ID => __( 'Ollama', 'classifai' ),
		];
	}

	/**
	 * Set up necessary hooks.
	 *
	 * We utilize this so we can register the REST route.
	 */
	public function setup() {
		parent::setup();
		add_action( 'rest_api_init', [ $this, 'register_endpoints' ] );
	}

	/**
	 * Set up necessary hooks.
	 */
	public function feature_setup() {
		$post_types = $this->get_supported_post_types();
		if ( ! empty( $post_types ) ) {
			foreach ( $post_types as $post_type ) {
				add_action( 'rest_after_insert_' . $post_type, [ $this, 'rest_after_insert' ] );
			}
		}

		add_action( 'admin_enqueue_scripts', [ $this, 'enqueue_admin_assets' ] );
		add_action( 'enqueue_block_editor_assets', [ $this, 'enqueue_editor_assets' ] );
		add_action( 'classifai_after_feature_settings_form', [ $this, 'render_previewer' ] );
		add_action( 'rest_api_init', [ $this, 'add_process_content_meta_to_rest_api' ] );
		add_action( 'wp_ajax_classifai_get_post_search_results', array( $this, 'get_post_search_results' ) );
		add_filter( 'default_post_metadata', [ $this, 'default_post_metadata' ], 10, 3 );

		// Support the Classic Editor.
		add_action( 'add_meta_boxes', [ $this, 'add_meta_box' ], 10, 2 );
		add_action( 'save_post', [ $this, 'save_meta_box' ] );
		add_action( 'admin_post_classifai_classify_post', array( $this, 'classifai_classify_post' ) );
		add_action( 'admin_notices', [ $this, 'show_error_if' ] );
		add_filter( 'removable_query_args', [ $this, 'removable_query_args' ] );
	}

	/**
	 * Register any needed endpoints.
	 */
	public function register_endpoints() {
		$post_types = $this->get_supported_post_types();
		foreach ( $post_types as $post_type ) {
			register_meta(
				$post_type,
				'_classifai_error',
				[
					'show_in_rest'  => true,
					'single'        => true,
					'auth_callback' => '__return_true',
				]
			);
		}

		register_rest_route(
			'classifai/v1',
			'classify/(?P<id>\d+)',
			[
				'methods'             => WP_REST_Server::READABLE,
				'callback'            => [ $this, 'rest_endpoint_callback' ],
				'args'                => array(
					'id'        => array(
						'required'          => true,
						'type'              => 'integer',
						'sanitize_callback' => 'absint',
						'description'       => esc_html__( 'Post ID to classify.', 'classifai' ),
					),
					'linkTerms' => array(
						'type'        => 'boolean',
						'description' => esc_html__( 'Whether to link terms or not.', 'classifai' ),
						'default'     => true,
					),
				),
				'permission_callback' => [ $this, 'classify_permissions_check' ],
			]
		);
	}

	/**
	 * Check if a given request has access to run classification.
	 *
	 * @param WP_REST_Request $request Full data about the request.
	 * @return WP_Error|bool
	 */
	public function classify_permissions_check( WP_REST_Request $request ) {
		$post_id = $request->get_param( 'id' );

		// Ensure we have a logged in user that can edit the item.
		if ( empty( $post_id ) || ! current_user_can( 'edit_post', $post_id ) ) {
			return false;
		}

		$post_type     = get_post_type( $post_id );
		$post_type_obj = get_post_type_object( $post_type );

		// Ensure the post type is allowed in REST endpoints.
		if ( ! $post_type || empty( $post_type_obj ) || empty( $post_type_obj->show_in_rest ) ) {
			return false;
		}

		// For all enabled features, ensure the user has proper permissions to add/edit terms.
		$provider_instance = $this->get_feature_provider_instance();
		if ( empty( $provider_instance->nlu_features ) ) {
			return new WP_Error( 'not_enabled', esc_html__( 'Classification not configured correctly for the selected provider.', 'classifai' ) );
		}

		foreach ( $provider_instance->nlu_features as $feature_name => $feature ) {
			if ( ! get_classification_feature_enabled( $feature_name ) ) {
				continue;
			}

			$taxonomy   = get_classification_feature_taxonomy( $feature_name );
			$permission = check_term_permissions( $taxonomy );

			if ( is_wp_error( $permission ) ) {
				return $permission;
			}
		}

		$post_status   = get_post_status( $post_id );
		$supported     = $this->get_supported_post_types();
		$post_statuses = $this->get_supported_post_statuses();

		// Check if processing allowed.
		if (
			! in_array( $post_status, $post_statuses, true ) ||
			! in_array( $post_type, $supported, true ) ||
			! $this->is_feature_enabled()
		) {
			return new WP_Error( 'not_enabled', esc_html__( 'Classification not enabled for current item.', 'classifai' ) );
		}

		return true;
	}

	/**
	 * Generic request handler for all our custom routes.
	 *
	 * @param WP_REST_Request $request The full request object.
	 * @return \WP_REST_Response
	 */
	public function rest_endpoint_callback( WP_REST_Request $request ) {
		$route = $request->get_route();

		if ( strpos( $route, '/classifai/v1/classify' ) === 0 ) {
			$results = $this->run(
				$request->get_param( 'id' ),
				'classify',
				[
					'link_terms' => $request->get_param( 'linkTerms' ),
				]
			);

			// Save results or return the results that need saved.
			if ( ! is_wp_error( $results ) ) {
				$results = $this->save( $request->get_param( 'id' ), $results, $request->get_param( 'linkTerms' ) ?? true );
			}

			return rest_ensure_response(
				[
					'terms'              => $results,
					'feature_taxonomies' => $this->get_all_feature_taxonomies(),
				]
			);
		}

		return parent::rest_endpoint_callback( $request );
	}

	/**
	 * Save or return the classification results.
	 *
	 * If $link is true, we link the terms to the item. If
	 * it is false, we just return the terms that need linked
	 * so they can show in the UI.
	 *
	 * @param int   $post_id The post ID.
	 * @param array $results Term results
	 * @param bool  $link Whether to link the terms or not.
	 * @return array|WP_Error
	 */
	public function save( int $post_id, array $results, bool $link = true ) {
		$provider_instance = $this->get_feature_provider_instance();

		/**
		 * Filter results to be saved.
		 *
		 * @since 3.1.0
		 * @hook classifai_feature_classification_pre_save_results
		 *
		 * @param array  $supported Term results.
		 * @param int    $post_id   Post ID.
		 * @param bool   $link      Whether to link the terms or not.
		 * @param object $this      Current instance of the class.
		 *
		 * @return array Term results.
		 */
		$results = apply_filters( 'classifai_' . static::ID . '_pre_save_results', $results, $post_id, $link, $this );

		switch ( $provider_instance::ID ) {
			case NLU::ID:
				$results = $provider_instance->link( $post_id, $results, $link );
				break;
			case AzureEmbeddings::ID:
			case OpenAIEmbeddings::ID:
			case OllamaEmbeddings::ID:
				$results = $provider_instance->set_terms( $post_id, $results, $link );
				break;
		}

		return $results;
	}

	/**
	 * Run classification after an item has been inserted via REST.
	 *
	 * @param \WP_Post $post Post object.
	 */
	public function rest_after_insert( \WP_Post $post ) {
		$supported_post_types = $this->get_supported_post_types();
		$post_statuses        = $this->get_supported_post_statuses();

		// Ensure the post type and status is allowed.
		if (
			! in_array( $post->post_type, $supported_post_types, true ) ||
			! in_array( $post->post_status, $post_statuses, true )
		) {
			return;
		}

		// Check if processing on save is disabled.
		if ( 'no' === get_post_meta( $post->ID, '_classifai_process_content', true ) ) {
			return;
		}

		$results = $this->run( $post->ID, 'classify' );

		if ( ! empty( $results ) && ! is_wp_error( $results ) ) {
			$this->save( $post->ID, $results );
			delete_post_meta( $post->ID, '_classifai_error' );
		} elseif ( is_wp_error( $results ) ) {
			update_post_meta(
				$post->ID,
				'_classifai_error',
				wp_json_encode(
					[
						'code'    => $results->get_error_code(),
						'message' => $results->get_error_message(),
					]
				)
			);
		}
	}

	/**
	 * Enqueue the admin scripts.
	 */
	public function enqueue_admin_assets() {
		wp_enqueue_script(
			'classifai-plugin-classification-previewer-js',
			CLASSIFAI_PLUGIN_URL . 'dist/classifai-plugin-classification-previewer.js',
			get_asset_info( 'classifai-plugin-classification-previewer', 'dependencies' ),
			get_asset_info( 'classifai-plugin-classification-previewer', 'version' ),
			true
		);

		wp_enqueue_style(
			'classifai-plugin-classification-previewer-css',
			CLASSIFAI_PLUGIN_URL . 'dist/classifai-plugin-classification-previewer.css',
			array(),
			get_asset_info( 'classifai-plugin-classification-previewer', 'version' ),
			'all'
		);
	}

	/**
	 * Enqueue editor assets.
	 */
	public function enqueue_editor_assets() {
		global $post;

		wp_enqueue_script(
			'classifai-plugin-classification-ibm-watson-js',
			CLASSIFAI_PLUGIN_URL . 'dist/classifai-plugin-classification-ibm-watson.js',
			get_asset_info( 'classifai-plugin-classification-ibm-watson', 'dependencies' ),
			get_asset_info( 'classifai-plugin-classification-ibm-watson', 'version' ),
			true
		);

		if ( empty( $post ) ) {
			return;
		}

		wp_enqueue_script(
			'classifai-plugin-classification-js',
			CLASSIFAI_PLUGIN_URL . 'dist/classifai-plugin-classification.js',
			array_merge( get_asset_info( 'classifai-plugin-classification', 'dependencies' ), array( 'lodash' ), array( Feature::PLUGIN_AREA_SCRIPT ) ),
			get_asset_info( 'classifai-plugin-classification', 'version' ),
			true
		);

		wp_enqueue_script(
			'classifai-plugin-classification-pre-publish-js',
			CLASSIFAI_PLUGIN_URL . 'dist/classifai-plugin-classification-pre-publish.js',
			array_merge( get_asset_info( 'classifai-plugin-classification-pre-publish', 'dependencies' ), array( 'lodash' ), array( Feature::PLUGIN_AREA_SCRIPT ) ),
			get_asset_info( 'classifai-plugin-classification-pre-publish', 'version' ),
			true
		);

		wp_add_inline_script(
			'classifai-plugin-classification-js',
			sprintf(
				'var classifaiPostData = %s;',
				wp_json_encode(
					[
						'NLUEnabled'           => $this->is_feature_enabled(),
						'supportedPostTypes'   => $this->get_supported_post_types(),
						'supportedPostStatues' => $this->get_supported_post_statuses(),
						'noPermissions'        => ! is_user_logged_in() || ! current_user_can( 'edit_post', $post->ID ),
					]
				)
			),
			'before'
		);
	}

	/**
	 * Add `classifai_process_content` to the REST API for view/edit.
	 */
	public function add_process_content_meta_to_rest_api() {
		$supported_post_types = $this->get_supported_post_types();

		register_rest_field(
			$supported_post_types,
			'classifai_process_content',
			[
				'get_callback'    => function ( $data ) {
					$process_content = get_post_meta( $data['id'], '_classifai_process_content', true );
					return ( 'no' === $process_content ) ? 'no' : 'yes';
				},
				'update_callback' => function ( $value, $data ) {
					$value = ( 'no' === $value ) ? 'no' : 'yes';
					return update_post_meta( $data->ID, '_classifai_process_content', $value );
				},
				'schema'          => [
					'type'    => 'string',
					'context' => [ 'view', 'edit' ],
				],
			]
		);
	}

	/**
	 * Searches and returns posts.
	 */
	public function get_post_search_results() {
		$nonce = isset( $_POST['nonce'] ) ? sanitize_text_field( wp_unslash( $_POST['nonce'] ) ) : false;

		if ( ! ( $nonce && wp_verify_nonce( $nonce, 'classifai-previewer-action' ) ) ) {
			wp_send_json_error( esc_html__( 'Failed nonce check.', 'classifai' ) );
		}

		$search_term   = isset( $_POST['search'] ) ? sanitize_text_field( wp_unslash( $_POST['search'] ) ) : '';
		$post_types    = isset( $_POST['post_types'] ) ? explode( ',', sanitize_text_field( wp_unslash( $_POST['post_types'] ) ) ) : 'post';
		$post_statuses = isset( $_POST['post_status'] ) ? explode( ',', sanitize_text_field( wp_unslash( $_POST['post_status'] ) ) ) : 'publish';

		$posts = get_posts(
			array(
				'post_type'   => $post_types,
				'post_status' => $post_statuses,
				's'           => $search_term,
			)
		);

		wp_send_json_success( $posts );
	}

	/**
	 * Add metabox to enable/disable language processing.
	 *
	 * @param string   $post_type Post type.
	 * @param \WP_Post $post WP_Post object.
	 */
	public function add_meta_box( string $post_type, $post ) {
		$supported_post_types = $this->get_supported_post_types();
		$post_statuses        = $this->get_supported_post_statuses();
		$post_status          = get_post_status( $post );

		if (
			in_array( $post_type, $supported_post_types, true ) &&
			in_array( $post_status, $post_statuses, true )
		) {
			add_meta_box(
				'classifai_language_processing_metabox',
				__( 'ClassifAI Language Processing', 'classifai' ),
				[ $this, 'render_meta_box' ],
				null,
				'side',
				'high',
				array( '__back_compat_meta_box' => true )
			);
		}
	}

	/**
	 * Render metabox content.
	 *
	 * @param \WP_Post $post WP_Post object.
	 */
	public function render_meta_box( \WP_Post $post ) {
		wp_nonce_field( 'classifai_language_processing_meta_action', 'classifai_language_processing_meta' );
		$process_content = get_post_meta( $post->ID, '_classifai_process_content', true );
		$process_content = ( 'no' === $process_content ) ? 'no' : 'yes';
		?>

		<p>
			<label for="classifai-process-content">
				<input type="checkbox" value="yes" name="_classifai_process_content" id="classifai-process-content" <?php checked( $process_content, 'yes' ); ?> />
				<?php esc_html_e( 'Automatically tag content on update', 'classifai' ); ?>
			</label>
		</p>

		<div class="classifai-classify-post-wrapper" style="display: none;">
			<a href="<?php echo esc_url( wp_nonce_url( admin_url( 'admin-post.php?action=classifai_classify_post&post_id=' . $post->ID ), 'classifai_classify_post_action', 'classifai_classify_post_nonce' ) ); ?>" class="button button-classify-post">
				<?php esc_html_e( 'Suggest terms & tags', 'classifai' ); ?>
			</a>
		</div>
		<?php
	}

	/**
	 * Handles saving the metabox.
	 *
	 * @param int $post_id Current post ID.
	 */
	public function save_meta_box( int $post_id ) {
		if (
			wp_is_post_autosave( $post_id ) ||
			wp_is_post_revision( $post_id ) ||
			! current_user_can( 'edit_post', $post_id )
		) {
			return;
		}

		if (
			empty( $_POST['classifai_language_processing_meta'] ) ||
			! wp_verify_nonce( sanitize_text_field( wp_unslash( $_POST['classifai_language_processing_meta'] ) ), 'classifai_language_processing_meta_action' )
		) {
			return;
		}

		$classifai_process_content = isset( $_POST['_classifai_process_content'] ) ? sanitize_key( wp_unslash( $_POST['_classifai_process_content'] ) ) : '';

		if ( 'yes' !== $classifai_process_content ) {
			update_post_meta( $post_id, '_classifai_process_content', 'no' );
		} else {
			update_post_meta( $post_id, '_classifai_process_content', 'yes' );

			$results = $this->run( $post_id, 'classify' );

			if ( ! empty( $results ) && ! is_wp_error( $results ) ) {
				$this->save( $post_id, $results );
				delete_post_meta( $post_id, '_classifai_error' );
			} elseif ( is_wp_error( $results ) ) {
				update_post_meta(
					$post_id,
					'_classifai_error',
					wp_json_encode(
						[
							'code'    => $results->get_error_code(),
							'message' => $results->get_error_message(),
						]
					)
				);
			}
		}
	}

	/**
	 * Classify post manually.
	 *
	 * Fires when the Classify button is clicked
	 * in the Classic Editor.
	 */
	public function classifai_classify_post() {
		if (
			empty( $_GET['classifai_classify_post_nonce'] ) ||
			! wp_verify_nonce( sanitize_text_field( wp_unslash( $_GET['classifai_classify_post_nonce'] ) ), 'classifai_classify_post_action' )
		) {
			wp_die( esc_html__( 'You don\'t have permission to perform this operation.', 'classifai' ) );
		}

		$post_id = isset( $_GET['post_id'] ) ? absint( $_GET['post_id'] ) : 0;

		if ( ! $post_id ) {
			exit();
		}

		// Check to see if processing is disabled and overwrite that.
		// Since we are manually classifying, we want to force this.
		$enabled = get_post_meta( $post_id, '_classifai_process_content', true );
		if ( 'yes' !== $enabled ) {
			update_post_meta( $post_id, '_classifai_process_content', 'yes' );
		}

		$results = $this->run( $post_id, 'classify' );

		// Ensure the processing value is changed back to what it was.
		if ( 'yes' !== $enabled ) {
			update_post_meta( $post_id, '_classifai_process_content', 'no' );
		}

		$classified = array();

		if ( ! empty( $results ) && ! is_wp_error( $results ) ) {
			$this->save( $post_id, $results );
			$classified = array( 'classifai_classify' => 1 );
			delete_post_meta( $post_id, '_classifai_error' );
		} elseif ( is_wp_error( $results ) ) {
			update_post_meta(
				$post_id,
				'_classifai_error',
				wp_json_encode(
					[
						'code'    => $results->get_error_code(),
						'message' => $results->get_error_message(),
					]
				)
			);
		}

		wp_safe_redirect( esc_url_raw( add_query_arg( $classified, get_edit_post_link( $post_id, 'edit' ) ) ) );
		exit();
	}

	/**
	 * Outputs an admin notice with the error message if needed.
	 */
	public function show_error_if() {
		global $post;

		if ( empty( $post ) ) {
			return;
		}

		$post_id = $post->ID;

		if ( empty( $post_id ) ) {
			return;
		}

		$error = get_post_meta( $post_id, '_classifai_error', true );

		if ( ! empty( $error ) ) {
			delete_post_meta( $post_id, '_classifai_error' );
			$error   = (array) json_decode( $error );
			$code    = ! empty( $error['code'] ) ? $error['code'] : 500;
			$message = ! empty( $error['message'] ) ? $error['message'] : 'Unknown API error';

			?>
			<div class="notice notice-error is-dismissible">
				<p>
					<?php esc_html_e( 'Error: Failed to classify content.', 'classifai' ); ?>
				</p>
				<p>
					<?php echo esc_html( $code ); ?>
					-
					<?php echo esc_html( $message ); ?>
				</p>
			</div>
			<?php
		}

		// Display classify post success message for manually classified post.
		// phpcs:ignore WordPress.Security.NonceVerification.Recommended
		$classified = isset( $_GET['classifai_classify'] ) ? intval( wp_unslash( $_GET['classifai_classify'] ) ) : 0;

		if ( 1 === $classified ) {
			$post_type       = get_post_type_object( get_post_type( $post ) );
			$post_type_label = esc_html__( 'Post', 'classifai' );
			if ( $post_type ) {
				$post_type_label = $post_type->labels->singular_name;
			}
			?>

			<div class="notice notice-success is-dismissible">
				<p>
					<?php
					// translators: %s is post type label.
					printf( esc_html__( '%s classified successfully.', 'classifai' ), esc_html( $post_type_label ) );
					?>
				</p>
			</div>

			<?php
		}
	}

	/**
	 * Sets the default value for the _classifai_process_content meta key.
	 *
	 * @param mixed  $value     The value get_metadata() should return - a single metadata value,
	 *                          or an array of values.
	 * @param int    $object_id Object ID.
	 * @param string $meta_key  Meta key.
	 * @return mixed
	 */
	public function default_post_metadata( $value, int $object_id, string $meta_key ) {
		if ( '_classifai_process_content' === $meta_key ) {
			if ( 'automatic_classification' === get_classification_mode() ) {
				return 'yes';
			} else {
				return 'no';
			}
		}

		return $value;
	}

	/**
	 * Add "classifai_classify" in list of query variable names to remove.
	 *
	 * @param array $removable_query_args An array of query variable names to remove from a URL.
	 * @return array
	 */
	public function removable_query_args( array $removable_query_args ): array {
		$removable_query_args[] = 'classifai_classify';
		return $removable_query_args;
	}

	/**
	 * Renders the previewer window for the feature.
	 *
	 * @param string $active_feature The active feature.
	 */
	public function render_previewer( string $active_feature ) {
		if ( self::ID !== $active_feature || ! $this->is_feature_enabled() ) {
			return;
		}

		$provider_instance       = $this->get_feature_provider_instance();
		$nlu_features            = array();
		$supported_post_statuses = $this->get_supported_post_statuses();
		$supported_post_types    = $this->get_supported_post_types();

		$posts_to_preview = get_posts(
			array(
				'post_type'      => $supported_post_types,
				'post_status'    => $supported_post_statuses,
				'posts_per_page' => 10,
			)
		);

		if ( ! empty( $provider_instance->nlu_features ) ) {
			$nlu_features = $provider_instance->nlu_features;
		}
		?>

		<div id="classifai-post-preview-app">
			<h2><?php esc_html_e( 'Preview Language Processing', 'classifai' ); ?></h2>

			<div id="classifai-post-preview-controls">
				<select id="classifai-preview-post-selector">
					<?php foreach ( $posts_to_preview as $post ) : ?>
						<option value="<?php echo esc_attr( $post->ID ); ?>"><?php echo esc_html( $post->post_title ); ?></option>
					<?php endforeach; ?>
				</select>

				<?php wp_nonce_field( 'classifai-previewer-action', 'classifai-previewer-nonce' ); ?>

				<button type="button" class="button" id="get-classifier-preview-data-btn">
					<span><?php esc_html_e( 'Preview', 'classifai' ); ?></span>
				</button>
			</div>

			<div id="classifai-post-preview-wrapper">
				<?php
				foreach ( $nlu_features as $feature_slug => $feature ) :
					if ( ! get_classification_feature_enabled( $feature_slug ) ) {
						continue;
					}
					?>

					<div class="tax-row tax-row--<?php echo esc_attr( $feature_slug ); ?>">
						<div class="tax-type"><?php echo esc_html( $feature['feature'] ); ?></div>
					</div>

					<?php
				endforeach;
				?>
			</div>
		</div>

		<?php
	}

	/**
	 * Get the description for the enable field.
	 *
	 * @return string
	 */
	public function get_enable_description(): string {
		return esc_html__( 'Classify your content automatically or when manually triggered.', 'classifai' );
	}

	/**
	 * Add any needed custom fields.
	 */
	public function add_custom_settings_fields() {
		$settings          = $this->get_settings();
		$provider_instance = $this->get_feature_provider_instance();
		$nlu_features      = array();
		$post_statuses     = get_post_statuses_for_language_settings();
		$post_types        = get_post_types_for_language_settings();
		$post_type_options = array();

		if ( ! empty( $provider_instance->nlu_features ) ) {
			$nlu_features = $provider_instance->nlu_features;
		}

		foreach ( $post_types as $post_type ) {
			$post_type_options[ $post_type->name ] = $post_type->label;
		}

		add_settings_field(
			'classification_mode',
			esc_html__( 'Classification mode', 'classifai' ),
			[ $this, 'render_radio_group' ],
			$this->get_option_name(),
			$this->get_option_name() . '_section',
			[
				'label_for'     => 'classification_mode',
				'default_value' => $settings['classification_mode'],
				'options'       => array(
					'manual_review'            => __( 'Manual review', 'classifai' ),
					'automatic_classification' => __( 'Automatic classification', 'classifai' ),
				),
			]
		);

		$method_options = array(
			'recommended_terms' => __( 'Recommend terms even if they do not exist on the site', 'classifai' ),
			'existing_terms'    => __( 'Only recommend terms that already exist on the site', 'classifai' ),
		);

		// Embeddings only supports existing terms.
		if ( isset( $settings['provider'] ) && ( OpenAIEmbeddings::ID === $settings['provider'] || AzureEmbeddings::ID === $settings['provider'] || OllamaEmbeddings::ID === $settings['provider'] ) ) {
			unset( $method_options['recommended_terms'] );
			$settings['classification_method'] = 'existing_terms';
		}

		add_settings_field(
			'classification_method',
			esc_html__( 'Classification method', 'classifai' ),
			[ $this, 'render_radio_group' ],
			$this->get_option_name(),
			$this->get_option_name() . '_section',
			[
				'label_for'     => 'classification_method',
				'default_value' => $settings['classification_method'],
				'options'       => $method_options,
			]
		);

		foreach ( $nlu_features as $classify_by => $labels ) {
			add_settings_field(
				$classify_by,
				esc_html( $labels['feature'] ),
				[ $this, 'render_nlu_feature_settings' ],
				$this->get_option_name(),
				$this->get_option_name() . '_section',
				[
					'feature'       => $classify_by,
					'labels'        => $labels,
					'default_value' => $settings[ $classify_by ],
					'post_types'    => $settings['post_types'],
				]
			);
		}

		add_settings_field(
			'post_statuses',
			esc_html__( 'Post statuses', 'classifai' ),
			[ $this, 'render_checkbox_group' ],
			$this->get_option_name(),
			$this->get_option_name() . '_section',
			[
				'label_for'      => 'post_statuses',
				'options'        => $post_statuses,
				'default_values' => $settings['post_statuses'],
				'description'    => __( 'Choose which post statuses are allowed to use this feature.', 'classifai' ),
			]
		);

		add_settings_field(
			'post_types',
			esc_html__( 'Post types', 'classifai' ),
			[ $this, 'render_checkbox_group' ],
			$this->get_option_name(),
			$this->get_option_name() . '_section',
			[
				'label_for'      => 'post_types',
				'options'        => $post_type_options,
				'default_values' => $settings['post_types'],
				'description'    => __( 'Choose which post types are allowed to use this feature.', 'classifai' ),
			]
		);
	}

	/**
	 * Returns the default settings for the feature.
	 *
	 * @return array
	 */
	public function get_feature_default_settings(): array {
		return [
			'post_statuses'         => [
				'publish' => 'publish',
			],
			'post_types'            => [
				'post' => 'post',
			],
			'classification_mode'   => 'manual_review',
			'classification_method' => 'recommended_terms',
			'provider'              => NLU::ID,
		];
	}

	/**
	 * Sanitizes the default feature settings.
	 *
	 * @param array $new_settings Settings being saved.
	 * @return array
	 */
	public function sanitize_default_feature_settings( array $new_settings ): array {
		$settings          = $this->get_settings();
		$provider_instance = $this->get_feature_provider_instance();

		$new_settings['classification_mode'] = sanitize_text_field( $new_settings['classification_mode'] ?? $settings['classification_mode'] );

		$new_settings['classification_method'] = sanitize_text_field( $new_settings['classification_method'] ?? $settings['classification_method'] );

		// Embeddings only supports existing terms.
		if ( isset( $new_settings['provider'] ) && ( OpenAIEmbeddings::ID === $new_settings['provider'] || AzureEmbeddings::ID === $new_settings['provider'] || OllamaEmbeddings::ID === $settings['provider'] ) ) {
			$new_settings['classification_method'] = 'existing_terms';
		}

		$new_settings['post_statuses'] = isset( $new_settings['post_statuses'] ) ? array_map( 'sanitize_text_field', $new_settings['post_statuses'] ) : $settings['post_statuses'];

		$new_settings['post_types'] = isset( $new_settings['post_types'] ) ? array_map( 'sanitize_text_field', $new_settings['post_types'] ) : $settings['post_types'];

		if ( ! empty( $provider_instance->nlu_features ) ) {
			foreach ( array_keys( $provider_instance->nlu_features ) as $feature_name ) {
				$new_settings[ $feature_name ]               = absint( $new_settings[ $feature_name ] ?? $settings[ $feature_name ] );
				$new_settings[ "{$feature_name}_threshold" ] = floatval( $new_settings[ "{$feature_name}_threshold" ] ?? $settings[ "{$feature_name}_threshold" ] );
				$new_settings[ "{$feature_name}_taxonomy" ]  = sanitize_text_field( $new_settings[ "{$feature_name}_taxonomy" ] ?? $settings[ "{$feature_name}_taxonomy" ] );
			}
		}

		return $new_settings;
	}

	/**
	 * Get all feature taxonomies.
	 *
	 * @return array
	 */
	public function get_all_feature_taxonomies(): array {
		$feature_taxonomies = [];
		$provider_instance  = $this->get_feature_provider_instance();

		if ( empty( $provider_instance->nlu_features ) ) {
			return $feature_taxonomies;
		}

		foreach ( array_keys( $provider_instance->nlu_features ) as $feature_name ) {
			if ( ! get_classification_feature_enabled( $feature_name ) ) {
				continue;
			}

			$taxonomy   = get_classification_feature_taxonomy( $feature_name );
			$permission = check_term_permissions( $taxonomy );

			if ( is_wp_error( $permission ) ) {
				continue;
			}

			if ( 'post_tag' === $taxonomy ) {
				$taxonomy = 'tags';
			}

			if ( 'category' === $taxonomy ) {
				$taxonomy = 'categories';
			}

			$feature_taxonomies[] = $taxonomy;
		}

		return $feature_taxonomies;
	}

	/**
	 * Render the NLU features settings.
	 *
	 * @param array $args Settings for the inputs
	 */
	public function render_nlu_feature_settings( array $args ) {
		$feature = $args['feature'];
		$labels  = $args['labels'];

		$taxonomies = $this->get_supported_taxonomies( $args['post_types'] );
		$features   = $this->get_settings();
		$taxonomy   = isset( $features[ "{$feature}_taxonomy" ] ) ? $features[ "{$feature}_taxonomy" ] : $labels['taxonomy_default'];

		// Enable classification type
		$feature_args = [
			'label_for'  => $feature,
			'input_type' => 'checkbox',
		];

		$threshold_args = [
			'label_for'     => "{$feature}_threshold",
			'input_type'    => 'number',
			'default_value' => $labels['threshold_default'],
			'min'           => 0,
			'max'           => 100,
			'step'          => 0.01,
		];
		?>

		<legend class="screen-reader-text">
			<?php esc_html_e( 'Classification Taxonomy Settings', 'classifai' ); ?>
		</legend>

		<p>
			<?php $this->render_input( $feature_args ); ?>
			<label for="<?php echo esc_attr( $feature ); ?>">
				<?php esc_html_e( 'Enable', 'classifai' ); ?>
			</label>
		</p>

		<p>
			<label for="<?php echo esc_attr( "{$feature}_threshold" ); ?>">
				<?php echo esc_html( $labels['threshold'] ); ?>
			</label><br/>
			<?php $this->render_input( $threshold_args ); ?>
		</p>

		<?php if ( NLU::ID === $features['provider'] ) : ?>
		<p>
			<label for="classifai-settings-<?php echo esc_attr( "{$feature}_taxonomy" ); ?>">
				<?php echo esc_html( $labels['taxonomy'] ); ?>
			</label><br/>
			<select id="classifai-settings-<?php echo esc_attr( "{$feature}_taxonomy" ); ?>" name="<?php echo esc_attr( $this->get_option_name() ); ?>[<?php echo esc_attr( "{$feature}_taxonomy" ); ?>]">
				<?php foreach ( $taxonomies as $name => $singular_name ) : ?>
					<option value="<?php echo esc_attr( $name ); ?>" <?php selected( $taxonomy, esc_attr( $name ) ); ?>>
						<?php echo esc_html( $singular_name ); ?>
					</option>
				<?php endforeach; ?>
			</select>
		</p>
		<?php endif; ?>
		<?php
	}

	/**
	 * Return the list of supported taxonomies
	 *
	 * @param array $post_types Array of supported post types.
	 * @return array
	 */
	public function get_supported_taxonomies( array $post_types = [] ): array {
		$supported_post_types = [];

		if ( ! empty( $post_types ) ) {
			foreach ( $post_types as $post_type => $enabled ) {
				if ( ! empty( $enabled ) ) {
					$supported_post_types[] = $post_type;
				}
			}
		}

		return $this->get_taxonomies( $supported_post_types );
	}

	/**
	 * Generates feature setting data required for migration from
	 * ClassifAI < 3.0.0 to 3.0.0
	 *
	 * @return array
	 */
	public function migrate_settings() {
		$old_settings = get_option( 'classifai_watson_nlu', array() );
		$new_settings = $this->get_default_settings();

		if ( isset( $old_settings['authenticated'] ) && $old_settings['authenticated'] ) {
			$new_settings['provider'] = 'ibm_watson_nlu';

			// Status
			if ( isset( $old_settings['enable_content_classification'] ) ) {
				$new_settings['status'] = $old_settings['enable_content_classification'];
			}

			// Post types
			if ( isset( $old_settings['post_types'] ) ) {
				if ( is_array( $old_settings['post_types'] ) ) {
					foreach ( $old_settings['post_types'] as $post_type => $value ) {
						if ( 1 === $value ) {
							$new_settings['post_types'][ $post_type ] = $post_type;
							continue;
						} elseif ( is_null( $value ) ) {
							$new_settings['post_types'][ $post_type ] = '0';
							continue;
						}
						$new_settings['post_types'][ $post_type ] = $value;
					}
				}

				unset( $new_settings['post_types']['attachment'] );
			}

			// Post statuses
			if ( isset( $old_settings['post_statuses'] ) ) {
				if ( is_array( $old_settings['post_statuses'] ) ) {
					foreach ( $old_settings['post_statuses'] as $post_status => $value ) {
						if ( 1 === $value ) {
							$new_settings['post_statuses'][ $post_status ] = $post_status;
							continue;
						} elseif ( is_null( $value ) ) {
							$new_settings['post_statuses'][ $post_status ] = '0';
							continue;
						}
						$new_settings['post_statuses'][ $post_status ] = $value;
					}
				}
			}

			// Roles
			if ( isset( $old_settings['content_classification_roles'] ) ) {
				$new_settings['roles'] = $old_settings['content_classification_roles'];
			}

			// Users
			if ( isset( $old_settings['content_classification_users'] ) ) {
				$new_settings['users'] = $old_settings['content_classification_users'];
			}

			// Provider.
			if ( isset( $old_settings['credentials'] ) && isset( $old_settings['credentials']['watson_url'] ) ) {
				$new_settings['ibm_watson_nlu']['endpoint_url'] = $old_settings['credentials']['watson_url'];
			}

			if ( isset( $old_settings['credentials'] ) && isset( $old_settings['credentials']['watson_username'] ) ) {
				$new_settings['ibm_watson_nlu']['username'] = $old_settings['credentials']['watson_username'];
			}

			if ( isset( $old_settings['credentials'] ) && isset( $old_settings['credentials']['watson_password'] ) ) {
				$new_settings['ibm_watson_nlu']['password'] = $old_settings['credentials']['watson_password'];
			}

			if ( isset( $old_settings['classification_mode'] ) ) {
				$new_settings['classification_mode'] = $old_settings['classification_mode'];
			}

			if ( isset( $old_settings['classification_method'] ) ) {
				$new_settings['classification_method'] = $old_settings['classification_method'];
			}

			if ( isset( $old_settings['features'] ) ) {
				foreach ( $old_settings['features'] as $feature => $value ) {
					$new_settings[ $feature ] = $value;
				}
			}

			if ( isset( $old_settings['authenticated'] ) ) {
				$new_settings['ibm_watson_nlu']['authenticated'] = $old_settings['authenticated'];
			}

			if ( isset( $old_settings['content_classification_user_based_opt_out'] ) ) {
				$new_settings['user_based_opt_out'] = $old_settings['content_classification_user_based_opt_out'];
			}
		} else {
			$old_settings = get_option( 'classifai_openai_embeddings', array() );

			if ( isset( $old_settings['enable_classification'] ) ) {
				$new_settings['status'] = $old_settings['enable_classification'];
			}

			$new_settings['provider'] = 'openai_embeddings';

			if ( isset( $old_settings['api_key'] ) ) {
				$new_settings['openai_embeddings']['api_key'] = $old_settings['api_key'];
			}

			if ( isset( $old_settings['taxonomies'] ) ) {
				foreach ( $old_settings['taxonomies'] as $feature => $value ) {
					$new_settings[ $feature ] = $value;
				}
			}

			if ( isset( $old_settings['authenticated'] ) ) {
				$new_settings['openai_embeddings']['authenticated'] = $old_settings['authenticated'];
			}

			if ( isset( $old_settings['post_statuses'] ) ) {
				$new_settings['post_statuses'] = $old_settings['post_statuses'];
			}

			if ( isset( $old_settings['post_types'] ) ) {
				$new_settings['post_types'] = $old_settings['post_types'];
			}

			if ( isset( $old_settings['classification_roles'] ) ) {
				$new_settings['roles'] = $old_settings['classification_roles'];
			}

			if ( isset( $old_settings['classification_users'] ) ) {
				$new_settings['users'] = $old_settings['classification_users'];
			}

			if ( isset( $old_settings['classification_user_based_opt_out'] ) ) {
				$new_settings['user_based_opt_out'] = $old_settings['classification_user_based_opt_out'];
			}
		}

		return $new_settings;
	}

	/**
	 * Get status of embeddings generation process.
	 *
	 * @return bool
	 */
	public function is_embeddings_generation_in_progress(): bool {
		$is_in_progress    = false;
		$provider_instance = $this->get_feature_provider_instance();
		if ( $provider_instance && method_exists( $provider_instance, 'is_embeddings_generation_in_progress' ) ) {
			$is_in_progress = $provider_instance->is_embeddings_generation_in_progress();
		}
		return $is_in_progress;
	}
}
