<?php

class Groq_AI_Ajax_Controller {
	/** @var Groq_AI_Product_Text_Plugin */
	private $plugin;

	public function __construct( Groq_AI_Product_Text_Plugin $plugin ) {
		$this->plugin = $plugin;

		add_action( 'wp_ajax_groq_ai_generate_text', [ $this, 'handle_generate_text' ] );
		add_action( 'wp_ajax_groq_ai_refresh_models', [ $this, 'handle_refresh_models' ] );
		add_action( 'wp_ajax_groq_ai_generate_term_text', [ $this, 'handle_generate_term_text' ] );
	}

	public function handle_generate_term_text() {
		if ( ! current_user_can( 'manage_options' ) ) {
			wp_send_json_error( [ 'message' => __( 'Je hebt geen toestemming voor deze actie.', GROQ_AI_PRODUCT_TEXT_DOMAIN ) ], 403 );
		}

		check_ajax_referer( 'groq_ai_generate_term', 'nonce' );

		$prompt   = isset( $_POST['prompt'] ) ? sanitize_textarea_field( wp_unslash( $_POST['prompt'] ) ) : '';
		$taxonomy = isset( $_POST['taxonomy'] ) ? sanitize_key( wp_unslash( $_POST['taxonomy'] ) ) : '';
		$term_id  = isset( $_POST['term_id'] ) ? absint( $_POST['term_id'] ) : 0;
		$include_top_products = ! empty( $_POST['include_top_products'] );
		$top_products_limit   = isset( $_POST['top_products_limit'] ) ? absint( $_POST['top_products_limit'] ) : 10;
		$top_products_limit   = max( 1, min( 25, $top_products_limit ) );

		if ( '' === $prompt || '' === $taxonomy || ! taxonomy_exists( $taxonomy ) || ! $term_id ) {
			wp_send_json_error( [ 'message' => __( 'Prompt, taxonomy en term_id zijn verplicht.', GROQ_AI_PRODUCT_TEXT_DOMAIN ) ], 400 );
		}

		$term = get_term( $term_id, $taxonomy );
		if ( ! $term || is_wp_error( $term ) ) {
			wp_send_json_error( [ 'message' => __( 'Term niet gevonden.', GROQ_AI_PRODUCT_TEXT_DOMAIN ) ], 404 );
		}

		$settings         = $this->plugin->get_settings();
		$provider_manager = $this->plugin->get_provider_manager();
		$provider_key     = $settings['provider'];
		$provider         = $provider_manager->get_provider( $provider_key );

		if ( ! $provider ) {
			$provider     = $provider_manager->get_provider( 'groq' );
			$provider_key = 'groq';
		}

		$conversation_id = $this->plugin->get_conversation_manager()->ensure_id( $provider_key, $settings['store_context'] );
		$prompt_builder  = $this->plugin->get_prompt_builder();
		$system_prompt   = method_exists( $prompt_builder, 'build_term_system_prompt' )
			? $prompt_builder->build_term_system_prompt( $settings, $conversation_id, $term )
			: $prompt_builder->build_system_prompt( $settings, $conversation_id );

		$model = $this->plugin->get_selected_model( $provider, $settings );
		$context_block = '';
		if ( method_exists( $prompt_builder, 'build_term_context_block' ) ) {
			$context_block = $prompt_builder->build_term_context_block(
				$term,
				[
					'include_top_products' => $include_top_products,
					'top_products_limit' => $top_products_limit,
				],
				$settings
			);
		}
		$prompt_with_context = method_exists( $prompt_builder, 'prepend_term_context_to_prompt' )
			? $prompt_builder->prepend_term_context_to_prompt( $prompt, $context_block )
			: $prompt_builder->prepend_context_to_prompt( $prompt, $context_block );

		$response_format  = null;
		$use_response_format = $this->plugin->should_use_response_format( $provider, $settings );
		if ( $use_response_format && method_exists( $prompt_builder, 'get_term_response_format_definition' ) ) {
			$response_format = $prompt_builder->get_term_response_format_definition( $settings );
			$final_prompt    = $prompt_with_context;
		} elseif ( method_exists( $prompt_builder, 'append_term_response_instructions' ) ) {
			$final_prompt = $prompt_builder->append_term_response_instructions( $prompt_with_context, $settings );
		} else {
			$final_prompt = $prompt_builder->append_response_instructions( $prompt_with_context, $settings );
		}

		$result = $provider->generate_content(
			[
				'prompt'          => $final_prompt,
				'system_prompt'   => $system_prompt,
				'model'           => $model,
				'settings'        => $settings,
				'temperature'     => 0.7,
				'conversation_id' => $conversation_id,
				'response_format' => $response_format,
			]
		);

		if ( is_wp_error( $result ) ) {
			wp_send_json_error( [ 'message' => $result->get_error_message() ], 500 );
		}

		$response_text = $this->extract_content_text( $result );
		$parsed = null;
		if ( method_exists( $prompt_builder, 'parse_term_structured_response' ) ) {
			$parsed = $prompt_builder->parse_term_structured_response( $response_text, $settings );
		}
		if ( ! is_array( $parsed ) ) {
			$parsed = [
				'description' => trim( (string) $response_text ),
			];
		}

		$default_bottom_key = isset( $settings['term_bottom_description_meta_key'] ) ? sanitize_key( (string) $settings['term_bottom_description_meta_key'] ) : '';
		$bottom_meta_key    = apply_filters( 'groq_ai_term_bottom_description_meta_key', $default_bottom_key, $term, $settings );
		$bottom_meta_key    = sanitize_key( (string) $bottom_meta_key );
		$has_bottom_field   = ( '' !== $bottom_meta_key );

		$top_description = isset( $parsed['description'] ) ? (string) $parsed['description'] : '';
		$bottom_description = isset( $parsed['bottom_description'] ) ? (string) $parsed['bottom_description'] : '';
		$apply_text = $has_bottom_field ? ( '' !== $bottom_description ? $bottom_description : $top_description ) : $top_description;

		wp_send_json_success(
			[
				'top_description' => $top_description,
				'bottom_description' => $has_bottom_field ? $apply_text : $bottom_description,
				'description' => $apply_text,
				'raw' => $response_text,
			]
		);
	}

	public function handle_generate_text() {
		if ( ! current_user_can( 'edit_products' ) ) {
			wp_send_json_error( [ 'message' => __( 'Je hebt geen toestemming voor deze actie.', GROQ_AI_PRODUCT_TEXT_DOMAIN ) ], 403 );
		}

		check_ajax_referer( 'groq_ai_generate', 'nonce' );

		$prompt = isset( $_POST['prompt'] ) ? sanitize_textarea_field( wp_unslash( $_POST['prompt'] ) ) : '';
		$post_id = isset( $_POST['post_id'] ) ? absint( $_POST['post_id'] ) : 0;

		$settings      = $this->plugin->get_settings();
		$provider_manager = $this->plugin->get_provider_manager();
		$provider_key  = $settings['provider'];
		$provider      = $provider_manager->get_provider( $provider_key );

		if ( ! $provider ) {
			$provider      = $provider_manager->get_provider( 'groq' );
			$provider_key  = 'groq';
		}

		$conversation_id      = $this->plugin->get_conversation_manager()->ensure_id( $provider_key, $settings['store_context'] );
		$prompt_builder       = $this->plugin->get_prompt_builder();
		$system_prompt        = $prompt_builder->build_system_prompt( $settings, $conversation_id );
		$model                = $this->plugin->get_selected_model( $provider, $settings );
		$context_fields       = $prompt_builder->parse_context_fields_from_request( isset( $_POST['context_fields'] ) ? $_POST['context_fields'] : '', $settings );
		$image_context_mode   = $this->plugin->get_image_context_mode( $settings );
		$image_context_limit  = $this->plugin->get_image_context_limit( $settings );

		if ( 'none' === $image_context_mode ) {
			$context_fields['images'] = false;
		}

		$image_context_enabled = ! empty( $context_fields['images'] );
		$use_base64_payloads   = $image_context_enabled && 'base64' === $image_context_mode && $provider->supports_image_context();
		$total_image_count     = $image_context_enabled ? $prompt_builder->get_product_image_count( $post_id ) : 0;
		$image_context_count   = $image_context_enabled ? min( $image_context_limit, $total_image_count ) : 0;

		$prompt_image_mode = 'none';
		if ( $image_context_enabled ) {
			if ( $use_base64_payloads ) {
				$prompt_image_mode = 'base64';
			} else {
				$prompt_image_mode = 'url';
			}

			if ( 'base64' === $image_context_mode && ! $provider->supports_image_context() ) {
				$prompt_image_mode = 'url';
			}
		}

		$product_context_text = $prompt_builder->build_product_context_block( $post_id, $context_fields, $prompt_image_mode, $image_context_limit );
		$image_context_payloads = [];
		if ( $use_base64_payloads ) {
			$image_context_payloads = $prompt_builder->get_product_image_payloads( $post_id, $image_context_limit );
		}
		$prompt_with_context  = $prompt_builder->prepend_context_to_prompt( $prompt, $product_context_text );

		$image_context_meta = [
			'requested_mode' => $image_context_mode,
			'effective_mode' => $prompt_image_mode,
			'limit'          => $image_context_limit,
			'available'      => $total_image_count,
			'used'           => $image_context_count,
			'base64_sent'    => $use_base64_payloads ? count( $image_context_payloads ) : 0,
		];

		$response_format  = null;
		$use_response_format = $this->plugin->should_use_response_format( $provider, $settings );
		if ( $use_response_format ) {
			$response_format = $prompt_builder->get_response_format_definition( $settings );
			$final_prompt    = $prompt_with_context;
		} else {
			$final_prompt = $prompt_builder->append_response_instructions( $prompt_with_context, $settings );
		}

		$result = $provider->generate_content(
			[
				'prompt'          => $final_prompt,
				'system_prompt'   => $system_prompt,
				'model'           => $model,
				'settings'        => $settings,
				'temperature'     => 0.7,
				'conversation_id' => $conversation_id,
				'response_format' => $response_format,
				'image_context'   => $image_context_payloads,
			]
		);

		if ( is_wp_error( $result ) ) {
			$this->plugin->get_generation_logger()->log_generation_event(
				[
					'provider'      => $provider_key,
					'model'         => $model,
					'prompt'        => $final_prompt,
					'response'      => '',
					'usage'         => [
						'image_context' => $image_context_meta,
					],
					'post_id'       => $post_id,
					'status'        => 'error',
					'error_message' => $result->get_error_message(),
				]
			);
			wp_send_json_error( [ 'message' => $result->get_error_message() ], 500 );
		}

		$response_text = $this->extract_content_text( $result );
		$response_usage = is_array( $result ) && isset( $result['usage'] ) ? $result['usage'] : [];
		if ( ! is_array( $response_usage ) ) {
			$response_usage = [];
		}
		$response_usage['image_context'] = $image_context_meta;

		$response = $prompt_builder->parse_structured_response( $response_text, $settings );

		if ( is_wp_error( $response ) ) {
			$this->plugin->get_generation_logger()->log_generation_event(
				[
					'provider'      => $provider_key,
					'model'         => $model,
					'prompt'        => $final_prompt,
					'response'      => $response_text,
					'usage'         => $response_usage,
					'post_id'       => $post_id,
					'status'        => 'error',
					'error_message' => $response->get_error_message(),
				]
			);
			wp_send_json_error( [ 'message' => $response->get_error_message() ], 500 );
		}

		$this->plugin->get_generation_logger()->log_generation_event(
			[
				'provider' => $provider_key,
				'model'    => $model,
				'prompt'   => $final_prompt,
				'response' => $response_text,
				'usage'    => $response_usage,
				'post_id'  => $post_id,
				'status'   => 'success',
			]
		);

		wp_send_json_success(
			[
				'fields' => $response,
				'raw'    => $response_text,
			]
		);
	}

	public function handle_refresh_models() {
		if ( ! current_user_can( 'manage_options' ) ) {
			wp_send_json_error( [ 'message' => __( 'Geen toestemming.', GROQ_AI_PRODUCT_TEXT_DOMAIN ) ], 403 );
		}

		check_ajax_referer( 'groq_ai_refresh_models', 'nonce' );

		$provider_key = isset( $_POST['provider'] ) ? sanitize_text_field( wp_unslash( $_POST['provider'] ) ) : '';
		$api_key      = isset( $_POST['apiKey'] ) ? sanitize_text_field( wp_unslash( $_POST['apiKey'] ) ) : '';

		if ( empty( $provider_key ) || empty( $api_key ) ) {
			wp_send_json_error( [ 'message' => __( 'Provider en API-sleutel zijn verplicht.', GROQ_AI_PRODUCT_TEXT_DOMAIN ) ], 400 );
		}

		$provider = $this->plugin->get_provider_manager()->get_provider( $provider_key );

		if ( ! $provider || ! $provider->supports_live_models() ) {
			wp_send_json_error( [ 'message' => __( 'Deze aanbieder ondersteunt het ophalen van modellen niet.', GROQ_AI_PRODUCT_TEXT_DOMAIN ) ], 400 );
		}

		$result = $provider->fetch_live_models( $api_key );

		if ( is_wp_error( $result ) ) {
			wp_send_json_error( [ 'message' => $result->get_error_message() ], 500 );
		}

		$models = Groq_AI_Model_Exclusions::filter_models( $provider_key, array_values( array_unique( $result ) ) );
		$models = $this->plugin->update_cached_models_for_provider( $provider_key, $models );

		wp_send_json_success( [ 'models' => $models ] );
	}

	private function extract_content_text( $result ) {
		if ( is_array( $result ) && isset( $result['content'] ) ) {
			return (string) $result['content'];
		}

		return (string) $result;
	}
}