<?php
/**
 * Block Toolbar AI Image Generation API
 * 
 * Handles image generation with smart fallback logic:
 * Gemini → OpenAI → Other providers (Pexels, Pixabay, Unsplash, etc.)
 */

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

// Load plugin constants
require_once plugin_dir_path( __FILE__ ) . 'constants.php';

/**
 * Wall-clock deadline for the whole generation request.
 *
 * Without this the fallback chain can hold a PHP worker for minutes while every
 * provider times out in turn.
 *
 * @return float Unix timestamp with microseconds.
 */
function ai_image_generation_deadline() {
	static $deadline = null;

	if ( null === $deadline ) {
		/**
		 * Filters the total time budget, in seconds, for one generation request.
		 *
		 * @param int $seconds Budget in seconds.
		 */
		$deadline = microtime( true ) + (int) apply_filters( 'bdthemes_ai_image_generation_budget', 45 );
	}

	return $deadline;
}

/**
 * Whether there is still time to try another provider.
 *
 * @return bool
 */
function ai_image_has_time_left() {
	return microtime( true ) < ai_image_generation_deadline();
}

/**
 * Request timeout capped by whatever is left of the budget.
 *
 * @param int $max Preferred timeout in seconds.
 * @return int
 */
function ai_image_request_timeout( $max = 10 ) {
	$remaining = ai_image_generation_deadline() - microtime( true );

	return (int) max( 5, min( $max, floor( $remaining ) ) );
}

/**
 * Remember that a provider just failed, so the next request skips it for a while.
 *
 * @param string $provider Provider id.
 */
function ai_image_mark_provider_failed( $provider ) {
	set_transient( 'bdthemes_ai_image_failed_' . $provider, 1, 5 * MINUTE_IN_SECONDS );
}

/**
 * Whether a provider failed recently and should be skipped.
 *
 * @param string $provider Provider id.
 * @return bool
 */
function ai_image_is_provider_cooling_down( $provider ) {
	return (bool) get_transient( 'bdthemes_ai_image_failed_' . $provider );
}

/**
 * Record the outcome of a provider attempt.
 *
 * "No results" means the query missed, not that the provider is unhealthy, so it
 * must not start a cooldown.
 *
 * @param string         $provider Provider id.
 * @param true|\WP_Error $result   Attempt result.
 */
function ai_image_record_provider_result( $provider, $result ) {
	if ( ! is_wp_error( $result ) ) {
		delete_transient( 'bdthemes_ai_image_failed_' . $provider );
		return;
	}

	if ( false === strpos( $result->get_error_code(), '_no_results' ) ) {
		ai_image_mark_provider_failed( $provider );
	}
}

/**
 * Log an upstream provider failure without exposing it to the browser.
 *
 * @param string $provider Provider id.
 * @param int    $code     HTTP status code.
 * @param string $body     Raw response body.
 */
function ai_image_log_upstream_error( $provider, $code, $body ) {
	if ( defined( 'WP_DEBUG' ) && WP_DEBUG ) {
		// phpcs:ignore WordPress.PHP.DevelopmentFunctions.error_log_error_log -- Debug-only diagnostics.
		error_log( sprintf( '[ai-image] %s request failed (HTTP %d): %s', $provider, (int) $code, substr( (string) $body, 0, 500 ) ) );
	}
}

/**
 * Smart AI Image Generation with Fallback or Specific Provider
 * 
 * Tries providers in order of preference:
 * 1. Gemini (if API key exists and enabled)
 * 2. OpenAI (if API key exists and enabled)
 * 3. Other free providers based on settings
 * 
 * Or uses a specific provider if requested via the 'provider' parameter.
 */
function ai_image_generate_smart_handler() {
	// Verify nonce
	if ( ! isset( $_POST['nonce'] ) || ! wp_verify_nonce( sanitize_key( $_POST['nonce'] ), 'wp_rest' ) ) {
		wp_send_json_error( array( 'message' => __( 'Invalid nonce.', 'ai-image' ) ), 403 );
	}
	
	// Check user permissions
	if ( ! current_user_can( 'edit_posts' ) ) {
		wp_send_json_error( array( 'message' => __( 'Insufficient permissions.', 'ai-image' ) ), 403 );
	}
	
	// Get the prompt
	$prompt = isset( $_POST['prompt'] ) ? sanitize_text_field( wp_unslash( $_POST['prompt'] ) ) : '';
	
	if ( empty( $prompt ) ) {
		wp_send_json_error( array( 'message' => __( 'No content provided for image generation.', 'ai-image' ) ) );
	}
	
	// Check if a specific provider was requested
	$requested_provider = isset( $_POST['provider'] ) ? sanitize_text_field( wp_unslash( $_POST['provider'] ) ) : '';
	
	if ( ! empty( $requested_provider ) ) {
		// Try the specific provider requested
		$result = ai_image_try_specific_provider( $requested_provider, $prompt );
		if ( ! is_wp_error( $result ) ) {
			wp_send_json_success( $result );
		} else {
			wp_send_json_error( array(
				'message' => sprintf(
					/* translators: 1: provider name, 2: error message */
					__( 'Failed to generate image with %1$s: %2$s', 'ai-image' ),
					ucfirst( $requested_provider ),
					$result->get_error_message()
				)
			) );
		}
		return;
	}
	
	$attempted_providers = array();
	$skipped_providers = array();
	
	// Try Gemini first
	$gemini_key = get_option( 'bdthemes_gemini_api_key' );
	$gemini_key = is_string( $gemini_key ) ? trim( $gemini_key ) : '';
	$gemini_enabled = bdthemes_ai_image_is_provider_enabled( 'gemini' );
	
	if ( ! $gemini_enabled ) {
		$skipped_providers[] = 'Gemini (disabled in settings)';
	} elseif ( empty( $gemini_key ) ) {
		$skipped_providers[] = 'Gemini (no API key)';
	} elseif ( ai_image_is_provider_cooling_down( 'gemini' ) ) {
		$skipped_providers[] = 'Gemini (failed recently)';
	} else {
		$attempted_providers[] = 'Gemini';
		$result = ai_image_try_gemini( $prompt, $gemini_key );
		ai_image_record_provider_result( 'gemini', $result );
		if ( ! is_wp_error( $result ) ) {
			wp_send_json_success( array(
				'image_url' => $result,
				'provider'  => 'Gemini',
				'message'   => __( 'Image generated with Gemini', 'ai-image' )
			) );
		}
	}
	
	// Try OpenAI second
	$openai_key = get_option( 'bdthemes_openai_api_key' );
	$openai_key = is_string( $openai_key ) ? trim( $openai_key ) : '';
	$openai_enabled = bdthemes_ai_image_is_provider_enabled( 'openai' );
	
	if ( ! $openai_enabled ) {
		$skipped_providers[] = 'OpenAI (disabled in settings)';
	} elseif ( empty( $openai_key ) ) {
		$skipped_providers[] = 'OpenAI (no API key)';
	} elseif ( ai_image_is_provider_cooling_down( 'openai' ) ) {
		$skipped_providers[] = 'OpenAI (failed recently)';
	} elseif ( ! ai_image_has_time_left() ) {
		$skipped_providers[] = 'OpenAI (time budget reached)';
	} else {
		$attempted_providers[] = 'OpenAI';
		$result = ai_image_try_openai( $prompt, $openai_key );
		ai_image_record_provider_result( 'openai', $result );
		if ( ! is_wp_error( $result ) ) {
			wp_send_json_success( array(
				'image_url' => $result,
				'provider'  => 'OpenAI',
				'message'   => __( 'Image generated with OpenAI', 'ai-image' )
			) );
		}
	}
	
	// Try other providers as fallback (Pexels, Pixabay, Unsplash, Openverse, Wikimedia)
	$result = ai_image_try_free_providers( $prompt, $attempted_providers, $skipped_providers );
	if ( ! is_wp_error( $result ) ) {
		wp_send_json_success( array(
			'image_url' => $result['url'],
			'provider'  => $result['provider'],
			/* translators: %s: provider name */
			'message'   => sprintf( __( 'Image found on %s', 'ai-image' ), $result['provider'] )
		) );
	}
	
	// If all fail, return detailed error
	$error_message = __( 'Failed to generate image with all available providers.', 'ai-image' );
	if ( ! empty( $skipped_providers ) ) {
		$error_message .= ' ' . sprintf(
			/* translators: %s: comma-separated list of skipped providers */
			__( 'Skipped: %s.', 'ai-image' ),
			implode( ', ', $skipped_providers )
		);
	}
	if ( ! empty( $attempted_providers ) ) {
		$error_message .= ' ' . sprintf(
			/* translators: %s: comma-separated list of attempted providers */
			__( 'Tried but failed: %s.', 'ai-image' ),
			implode( ', ', $attempted_providers )
		);
	}
	$error_message .= ' ' . __( 'Please check your API keys and provider settings.', 'ai-image' );
	
	wp_send_json_error( array(
		'message' => $error_message,
		'attempted' => $attempted_providers,
		'skipped' => $skipped_providers
	) );
}

/**
 * Try a specific provider
 */
function ai_image_try_specific_provider( $provider, $prompt ) {
	$provider = strtolower( $provider );
	
	// Check if provider is enabled
	if ( ! bdthemes_ai_image_is_provider_enabled( $provider ) ) {
		return new \WP_Error( 
			'provider_disabled',
			/* translators: %s: provider name */
			sprintf( __( '%s is disabled in settings', 'ai-image' ), ucfirst( $provider ) )
		);
	}
	
	// Handle AI providers (Gemini, OpenAI)
	if ( $provider === 'gemini' ) {
		$gemini_key = get_option( 'bdthemes_gemini_api_key' );
		$gemini_key = is_string( $gemini_key ) ? trim( $gemini_key ) : '';
		
		if ( empty( $gemini_key ) ) {
			return new \WP_Error( 'no_api_key', __( 'Gemini API key not configured', 'ai-image' ) );
		}
		
		$result = ai_image_try_gemini( $prompt, $gemini_key );
		if ( is_wp_error( $result ) ) {
			return $result;
		}
		
		return array(
			'image_url' => $result,
			'provider'  => 'Gemini',
			'message'   => __( 'Image generated with Gemini', 'ai-image' )
		);
	}
	
	if ( $provider === 'openai' ) {
		$openai_key = get_option( 'bdthemes_openai_api_key' );
		$openai_key = is_string( $openai_key ) ? trim( $openai_key ) : '';
		
		if ( empty( $openai_key ) ) {
			return new \WP_Error( 'no_api_key', __( 'OpenAI API key not configured', 'ai-image' ) );
		}
		
		$result = ai_image_try_openai( $prompt, $openai_key );
		if ( is_wp_error( $result ) ) {
			return $result;
		}
		
		return array(
			'image_url' => $result,
			'provider'  => 'OpenAI',
			'message'   => __( 'Image generated with OpenAI', 'ai-image' )
		);
	}
	
	// Handle free stock photo providers
	$free_providers = array( 'pexels', 'pixabay', 'unsplash', 'openverse', 'wikimedia', 'giphy' );
	if ( in_array( $provider, $free_providers, true ) ) {
		$result = ai_image_fetch_from_provider( $provider, $prompt );
		if ( is_wp_error( $result ) ) {
			return $result;
		}
		
		return array(
			'image_url' => $result,
			'provider'  => ucfirst( $provider ),
			/* translators: %s: provider name */
			'message'   => sprintf( __( 'Image found on %s', 'ai-image' ), ucfirst( $provider ) )
		);
	}

	return new \WP_Error( 'unknown_provider', __( 'Unknown provider', 'ai-image' ) );
}

/**
 * Try to generate image with Gemini
 */
function ai_image_try_gemini( $prompt, $api_key ) {
	$url = 'https://generativelanguage.googleapis.com/v1beta/models/imagen-4.0-generate-001:predict?key=' . urlencode( $api_key );
	
	$body = wp_json_encode( array(
		'instances' => array(
			array(
				'prompt' => $prompt
			)
		),
		'parameters' => array(
			'sampleCount'        => 1,
			'aspectRatio'        => '1:1',
			'safetyFilterLevel'  => 'block_some',
			'personGeneration'   => 'allow_adult'
		)
	) );
	
	$response = wp_remote_post( $url, array(
		'headers' => array(
			'Content-Type' => 'application/json',
		),
		'body'    => $body,
		'timeout' => ai_image_request_timeout( 30 ),
	) );
	
	if ( is_wp_error( $response ) ) {
		return $response;
	}
	
	$code = wp_remote_retrieve_response_code( $response );
	$response_body = wp_remote_retrieve_body( $response );
	$data = json_decode( $response_body, true );
	
	if ( $code !== 200 ) {
		$error_msg = isset( $data['error']['message'] ) ? $data['error']['message'] : __( 'Gemini API request failed.', 'ai-image' );
		return new \WP_Error( 'gemini_failed', $error_msg );
	}
	
	// Extract base64 image
	if ( isset( $data['predictions'][0]['bytesBase64Encoded'] ) ) {
		$base64_image = $data['predictions'][0]['bytesBase64Encoded'];
		return 'data:image/png;base64,' . $base64_image;
	}
	
	return new \WP_Error( 'gemini_no_image', __( 'No image returned from Gemini', 'ai-image' ) );
}

/**
 * Try to generate image with OpenAI
 */
function ai_image_try_openai( $prompt, $api_key ) {
	$url = 'https://api.openai.com/v1/images/generations';
	
	$body = wp_json_encode( array(
		'prompt'  => $prompt,
		'n'       => 1,
		'size'    => '1024x1024',
		'model'   => 'gpt-image-1',
		'quality' => 'medium',
	) );
	
	$response = wp_remote_post( $url, array(
		'headers' => array(
			'Content-Type'  => 'application/json',
			'Authorization' => 'Bearer ' . $api_key,
		),
		'body'    => $body,
		'timeout' => ai_image_request_timeout( 30 ),
	) );
	
	if ( is_wp_error( $response ) ) {
		return $response;
	}
	
	$code = wp_remote_retrieve_response_code( $response );
	$response_body = wp_remote_retrieve_body( $response );
	$data = json_decode( $response_body, true );
	
	if ( $code !== 200 ) {
		$error_msg = isset( $data['error']['message'] ) ? $data['error']['message'] : __( 'OpenAI API request failed.', 'ai-image' );
		return new \WP_Error( 'openai_failed', $error_msg );
	}
	
	// Extract image URL or base64 (GPT Image models return b64_json)
	if ( isset( $data['data'][0]['url'] ) ) {
		return $data['data'][0]['url'];
	}
	if ( isset( $data['data'][0]['b64_json'] ) ) {
		return 'data:image/png;base64,' . $data['data'][0]['b64_json'];
	}
	
	return new \WP_Error( 'openai_no_image', __( 'No image returned from OpenAI', 'ai-image' ) );
}

/**
 * Try free providers as fallback (Pexels, Pixabay, Unsplash, Openverse)
 */
function ai_image_try_free_providers( $prompt, &$attempted_providers, &$skipped_providers ) {
	// Get provider order from settings
	$default_order = array( 'pexels', 'pixabay', 'unsplash', 'openverse', 'wikimedia', 'giphy' );
	$saved_order   = get_option( 'bdthemes_ai_image_provider_order', $default_order );
	
	if ( ! is_array( $saved_order ) || empty( $saved_order ) ) {
		$saved_order = $default_order;
	}
	
	// Filter to only free providers
	$free_providers = array_intersect( $saved_order, array( 'pexels', 'pixabay', 'unsplash', 'openverse', 'wikimedia', 'giphy' ) );
	
	foreach ( $free_providers as $provider ) {
		// Check if provider is enabled
		if ( ! bdthemes_ai_image_is_provider_enabled( $provider ) ) {
			$skipped_providers[] = ucfirst( $provider ) . ' (disabled in settings)';
			continue;
		}

		// Stop before starting another request if the budget is spent.
		if ( ! ai_image_has_time_left() ) {
			$skipped_providers[] = ucfirst( $provider ) . ' (time budget reached)';
			continue;
		}

		// Skip providers that failed within the last few minutes.
		if ( ai_image_is_provider_cooling_down( $provider ) ) {
			$skipped_providers[] = ucfirst( $provider ) . ' (failed recently)';
			continue;
		}

		// Add to attempted list
		$attempted_providers[] = ucfirst( $provider );

		// Try to fetch from provider
		$result = ai_image_fetch_from_provider( $provider, $prompt );
		ai_image_record_provider_result( $provider, $result );
		if ( ! is_wp_error( $result ) ) {
			return array(
				'url'      => $result,
				'provider' => ucfirst( $provider )
			);
		}
	}
	
	return new \WP_Error( 'all_providers_failed', __( 'All image providers failed.', 'ai-image' ) );
}

/**
 * Fetch image from a specific free provider
 */
function ai_image_fetch_from_provider( $provider, $query ) {
	switch ( $provider ) {
		case 'pexels':
			return ai_image_fetch_from_pexels( $query );
		case 'pixabay':
			return ai_image_fetch_from_pixabay( $query );
		case 'unsplash':
			return ai_image_fetch_from_unsplash( $query );
		case 'openverse':
			return ai_image_fetch_from_openverse( $query );
		case 'wikimedia':
			return ai_image_fetch_from_wikimedia( $query );
		case 'giphy':
			return ai_image_fetch_from_giphy( $query );
		default:
			return new \WP_Error( 'unknown_provider', __( 'Unknown provider', 'ai-image' ) );
	}
}

/**
 * Fetch from Pexels
 */
function ai_image_fetch_from_pexels( $query ) {
	// Default Pexels API key
	$default_key = \BDT_AI_IMG\decrypt_key( AI_IMAGE_PEXELS_DEFAULT_KEY );
	
	// Try to get custom API key from settings, fallback to default
	$custom_key = get_option( 'bdthemes_pexels_api_key' );
	$custom_key = is_string( $custom_key ) ? trim( $custom_key ) : '';
	$api_key = ! empty( $custom_key ) ? $custom_key : $default_key;
	
	if ( empty( $api_key ) ) {
		return new \WP_Error( 'pexels_no_key', __( 'Pexels API key not configured', 'ai-image' ) );
	}
	
	$url = 'https://api.pexels.com/v1/search?query=' . urlencode( $query ) . '&per_page=1';
	
	$response = wp_remote_get( $url, array(
		'headers' => array(
			'Authorization' => $api_key,
		),
		'timeout' => ai_image_request_timeout( 10 ),
	) );
	
	if ( is_wp_error( $response ) ) {
		return $response;
	}
	
	$code = wp_remote_retrieve_response_code( $response );
	$response_body = wp_remote_retrieve_body( $response );
	
	// Check for HTTP errors (authentication, rate limit, etc.)
	if ( $code !== 200 ) {
		if ( $code === 401 || $code === 403 ) {
			$error_msg = __( 'Invalid Pexels API key. Please check your API key in settings.', 'ai-image' );
		} else {
			// The upstream body can be an HTML error page; log it instead of
			// relaying it to the browser.
			ai_image_log_upstream_error( 'pexels', $code, $response_body );
			/* translators: %s: HTTP error code */
			$error_msg = sprintf( __( 'Pexels API request failed with code: %s', 'ai-image' ), $code );
		}
		return new \WP_Error( 'pexels_failed', $error_msg );
	}
	
	$data = json_decode( $response_body, true );
	
	// Check for JSON decode errors
	if ( json_last_error() !== JSON_ERROR_NONE ) {
		return new \WP_Error( 'pexels_json_error', __( 'Invalid JSON response from Pexels', 'ai-image' ) );
	}
	
	if ( isset( $data['photos'][0]['src']['large'] ) ) {
		return $data['photos'][0]['src']['large'];
	}
	
	return new \WP_Error( 'pexels_no_results', __( 'No results from Pexels', 'ai-image' ) );
}

/**
 * Fetch from Pixabay
 */
function ai_image_fetch_from_pixabay( $query ) {
	// Default Pixabay API key
	$default_key = \BDT_AI_IMG\decrypt_key( AI_IMAGE_PIXABAY_DEFAULT_KEY );
	
	// Try to get custom API key from settings, fallback to default
	$custom_key = get_option( 'bdthemes_pixabay_api_key' );
	$custom_key = is_string( $custom_key ) ? trim( $custom_key ) : '';
	$api_key = ! empty( $custom_key ) ? $custom_key : $default_key;
	
	if ( empty( $api_key ) ) {
		return new \WP_Error( 'pixabay_no_key', __( 'Pixabay API key not configured', 'ai-image' ) );
	}
	
	// Ensure query is not empty
	if ( empty( $query ) || strlen( trim( $query ) ) === 0 ) {
		return new \WP_Error( 'pixabay_empty_query', __( 'Empty search query for Pixabay', 'ai-image' ) );
	}
	
	// Pixabay caps search queries at 100 characters. Use the multibyte variants
	// so a Bangla or Arabic prompt is not cut mid-character.
	$search_query = $query;
	if ( mb_strlen( $search_query ) > 100 ) {
		$search_query = mb_substr( $search_query, 0, 100 );
		// Remove partial word at the end
		$search_query = trim( preg_replace( '/\s+\S*$/u', '', $search_query ) );
	}
	
	$url = 'https://pixabay.com/api/?key=' . urlencode( $api_key ) . '&q=' . urlencode( $search_query ) . '&per_page=3&image_type=all&pretty=true&order=popular';
	
	$response = wp_remote_get( $url, array( 'timeout' => ai_image_request_timeout( 10 ) ) );
	
	if ( is_wp_error( $response ) ) {
		return $response;
	}
	
	$code = wp_remote_retrieve_response_code( $response );
	$response_body = wp_remote_retrieve_body( $response );
	
	// Check for HTTP errors first (before JSON decode)
	if ( $code !== 200 ) {
		// Pixabay returns plain text errors for 400 codes
		// The upstream body can be an HTML error page; log it instead of
		// relaying it to the browser.
		ai_image_log_upstream_error( 'pixabay', $code, $response_body );
		/* translators: %s: HTTP error code */
		$error_msg = sprintf( __( 'Pixabay API request failed with code: %s', 'ai-image' ), $code );
		return new \WP_Error( 'pixabay_failed', $error_msg );
	}
	
	$data = json_decode( $response_body, true );
	
	// Check for JSON decode errors
	if ( json_last_error() !== JSON_ERROR_NONE ) {
		return new \WP_Error( 'pixabay_json_error', __( 'Invalid JSON response from Pixabay', 'ai-image' ) );
	}
	
	// Check if we have results
	if ( isset( $data['hits'] ) && is_array( $data['hits'] ) && count( $data['hits'] ) > 0 ) {
		// Return large image URL
		if ( isset( $data['hits'][0]['largeImageURL'] ) ) {
			return $data['hits'][0]['largeImageURL'];
		}
	}
	
	/* translators: %s: search query */
	return new \WP_Error( 'pixabay_no_results', sprintf( __( 'No results from Pixabay for query: %s', 'ai-image' ), $search_query ) );
}

/**
 * Fetch from Unsplash
 */
function ai_image_fetch_from_unsplash( $query ) {
	// Default Unsplash API key
	$default_key = \BDT_AI_IMG\decrypt_key( AI_IMAGE_UNSPLASH_DEFAULT_KEY );
	
	// Try to get custom API key from settings, fallback to default
	$custom_key = get_option( 'bdthemes_unsplash_access_key' );
	$custom_key = is_string( $custom_key ) ? trim( $custom_key ) : '';
	$api_key = ! empty( $custom_key ) ? $custom_key : $default_key;
	
	if ( empty( $api_key ) ) {
		return new \WP_Error( 'unsplash_no_key', __( 'Unsplash API key not configured', 'ai-image' ) );
	}
	
	$url = 'https://api.unsplash.com/search/photos?query=' . urlencode( $query ) . '&per_page=1';
	
	$response = wp_remote_get( $url, array(
		'headers' => array(
			'Authorization' => 'Client-ID ' . $api_key,
		),
		'timeout' => ai_image_request_timeout( 10 ),
	) );
	
	if ( is_wp_error( $response ) ) {
		return $response;
	}
	
	$code = wp_remote_retrieve_response_code( $response );
	$response_body = wp_remote_retrieve_body( $response );
	
	// Check for HTTP errors (authentication, rate limit, etc.)
	if ( $code !== 200 ) {
		if ( $code === 401 || $code === 403 ) {
			$error_msg = __( 'Invalid Unsplash API key. Please check your API key in settings.', 'ai-image' );
		} else {
			// The upstream body can be an HTML error page; log it instead of
			// relaying it to the browser.
			ai_image_log_upstream_error( 'unsplash', $code, $response_body );
			/* translators: %s: HTTP error code */
			$error_msg = sprintf( __( 'Unsplash API request failed with code: %s', 'ai-image' ), $code );
		}
		return new \WP_Error( 'unsplash_failed', $error_msg );
	}
	
	$data = json_decode( $response_body, true );
	
	// Check for JSON decode errors
	if ( json_last_error() !== JSON_ERROR_NONE ) {
		return new \WP_Error( 'unsplash_json_error', __( 'Invalid JSON response from Unsplash', 'ai-image' ) );
	}
	
	if ( isset( $data['results'][0]['urls']['regular'] ) ) {
		return $data['results'][0]['urls']['regular'];
	}
	
	return new \WP_Error( 'unsplash_no_results', __( 'No results from Unsplash', 'ai-image' ) );
}

/**
 * Fetch from Openverse
 */
function ai_image_fetch_from_openverse( $query ) {
	// Ensure query is not empty
	if ( empty( $query ) || strlen( trim( $query ) ) === 0 ) {
		return new \WP_Error( 'openverse_empty_query', __( 'Empty search query for Openverse', 'ai-image' ) );
	}
	
	// Simplify query for better results - remove AI descriptors and keep core subject
	$search_query = ai_image_simplify_query_for_openverse( $query );
	
	$url = 'https://api.openverse.engineering/v1/images/?q=' . urlencode( $search_query ) . '&page_size=1';

	$response = wp_remote_get( $url, array( 'timeout' => ai_image_request_timeout( 10 ) ) );
	
	if ( is_wp_error( $response ) ) {
		return $response;
	}
	
	$code = wp_remote_retrieve_response_code( $response );
	$response_body = wp_remote_retrieve_body( $response );
	
	// Check for HTTP errors
	if ( $code !== 200 ) {
		/* translators: %s: HTTP error code */
		$error_msg = sprintf( __( 'Openverse API request failed with code: %s', 'ai-image' ), $code );
		return new \WP_Error( 'openverse_failed', $error_msg );
	}
	
	$data = json_decode( $response_body, true );
	
	// Check for JSON decode errors
	if ( json_last_error() !== JSON_ERROR_NONE ) {
		return new \WP_Error( 'openverse_json_error', __( 'Invalid JSON response from Openverse', 'ai-image' ) );
	}
	
	// Check if we have results
	if ( isset( $data['results'] ) && is_array( $data['results'] ) && count( $data['results'] ) > 0 ) {
		// Return image URL
		if ( isset( $data['results'][0]['url'] ) ) {
			return $data['results'][0]['url'];
		} 
	}
	
	/* translators: %s: search query */
	return new \WP_Error( 'openverse_no_results', sprintf( __( 'No results from Openverse for query: %s', 'ai-image' ), $search_query ) );
}

/**
 * Fetch from Wikimedia Commons
 */
function ai_image_fetch_from_wikimedia( $query ) {
	$search_query = trim( (string) $query );
	if ( '' === $search_query ) {
		return new \WP_Error( 'wikimedia_empty_query', __( 'Empty search query for Wikimedia', 'ai-image' ) );
	}

	$api_url = add_query_arg( [
		'action'        => 'query',
		'generator'     => 'search',
		'gsrsearch'     => $search_query,
		'gsrnamespace'  => 6,
		'gsrlimit'      => 1,
		'prop'          => 'imageinfo',
		'iiprop'        => 'url',
		'format'        => 'json',
		'formatversion' => 2,
		'origin'        => '*',
	], 'https://commons.wikimedia.org/w/api.php' );

	$response = wp_remote_get( $api_url, [
		'timeout' => ai_image_request_timeout( 10 ),
	] );

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

	$code = wp_remote_retrieve_response_code( $response );
	$body = wp_remote_retrieve_body( $response );

	if ( 200 !== $code ) {
		/* translators: %s: HTTP error code */
		$error_msg = sprintf( __( 'Wikimedia API request failed with code: %s', 'ai-image' ), $code );
		return new \WP_Error( 'wikimedia_failed', $error_msg );
	}

	$data = json_decode( $body, true );
	if ( JSON_ERROR_NONE !== json_last_error() || ! is_array( $data ) ) {
		return new \WP_Error( 'wikimedia_json_error', __( 'Invalid JSON response from Wikimedia', 'ai-image' ) );
	}

	if ( isset( $data['query']['pages'] ) && is_array( $data['query']['pages'] ) ) {
		foreach ( $data['query']['pages'] as $page_item ) {
			if ( isset( $page_item['imageinfo'][0]['url'] ) && ! empty( $page_item['imageinfo'][0]['url'] ) ) {
				return $page_item['imageinfo'][0]['url'];
			}
		}
	}

	/* translators: %s: search query */
	return new \WP_Error( 'wikimedia_no_results', sprintf( __( 'No results from Wikimedia for query: %s', 'ai-image' ), $search_query ) );
}

/**
 * Simplify query for Openverse by taking first 30 characters
 */
function ai_image_simplify_query_for_openverse( $query ) {
	// Openverse works better with shorter, simpler queries
	// Just take the first 30 characters (main subject only)
	$simplified = mb_substr( $query, 0, 30 );

	// Remove partial word at the end
	$simplified = trim( preg_replace( '/\s+\S*$/u', '', $simplified ) );

	// If still too short or empty, use up to 40 chars
	if ( empty( $simplified ) || mb_strlen( $simplified ) < 5 ) {
		$simplified = mb_substr( $query, 0, 40 );
		$simplified = trim( preg_replace( '/\s+\S*$/u', '', $simplified ) );
	}
	
	return trim( $simplified );
}

/**
 * Fetch from Giphy
 */
function ai_image_fetch_from_giphy( $query ) {
	// Default Giphy API key
	$default_key = \BDT_AI_IMG\decrypt_key( AI_IMAGE_GIPHY_DEFAULT_KEY );
	
	// Try to get custom API key from settings, fallback to default
	$custom_key = get_option( 'bdthemes_giphy_api_key' );
	$custom_key = is_string( $custom_key ) ? trim( $custom_key ) : '';
	$api_key = ! empty( $custom_key ) ? $custom_key : $default_key;
	
	if ( empty( $api_key ) ) {
		return new \WP_Error( 'giphy_no_key', __( 'Giphy API key not configured', 'ai-image' ) );
	}
	
	// Ensure query is not empty
	if ( empty( $query ) || strlen( trim( $query ) ) === 0 ) {
		return new \WP_Error( 'giphy_empty_query', __( 'Empty search query for Giphy', 'ai-image' ) );
	}
	
	// Simplify query for Giphy - GIFs need simple, short search terms
	$search_query = ai_image_simplify_query_for_giphy( $query );
	
	$url = 'https://api.giphy.com/v1/gifs/search?api_key=' . urlencode( $api_key ) . '&q=' . urlencode( $search_query ) . '&limit=1&rating=g';
	
	$response = wp_remote_get( $url, array( 'timeout' => ai_image_request_timeout( 10 ) ) );
	
	if ( is_wp_error( $response ) ) {
		return $response;
	}
	
	$code = wp_remote_retrieve_response_code( $response );
	$response_body = wp_remote_retrieve_body( $response );
	
	// Check for HTTP errors
	if ( $code !== 200 ) {
		/* translators: %s: HTTP error code */
		$error_msg = sprintf( __( 'Giphy API request failed with code: %s', 'ai-image' ), $code );
		return new \WP_Error( 'giphy_failed', $error_msg );
	}
	
	$data = json_decode( $response_body, true );
	
	// Check for JSON decode errors
	if ( json_last_error() !== JSON_ERROR_NONE ) {
		return new \WP_Error( 'giphy_json_error', __( 'Invalid JSON response from Giphy', 'ai-image' ) );
	}
	
	// Check if we have results
	if ( isset( $data['data'] ) && is_array( $data['data'] ) && count( $data['data'] ) > 0 ) {
		// Return GIF URL
		if ( isset( $data['data'][0]['images']['original']['url'] ) ) {
			return $data['data'][0]['images']['original']['url'];
		} 
	}
	
	/* translators: %s: search query */
	return new \WP_Error( 'giphy_no_results', sprintf( __( 'No results from Giphy for query: %s', 'ai-image' ), $search_query ) );
}

/**
 * Simplify query for Giphy by taking first 20 characters
 */
function ai_image_simplify_query_for_giphy( $query ) {
	// Giphy works best with very short, simple search terms
	// Just take the first 20 characters (main action/subject only)
	$simplified = mb_substr( $query, 0, 20 );

	// Remove partial word at the end
	$simplified = trim( preg_replace( '/\s+\S*$/u', '', $simplified ) );

	// If still too short or empty, use up to 30 chars
	if ( empty( $simplified ) || mb_strlen( $simplified ) < 3 ) {
		$simplified = mb_substr( $query, 0, 30 );
		$simplified = trim( preg_replace( '/\s+\S*$/u', '', $simplified ) );
	}
	
	return trim( $simplified );
}

// Register AJAX handler
add_action( 'wp_ajax_ai_image_generate_smart', 'ai_image_generate_smart_handler' );