<?php

if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

class MetaForge_AISettings {

	public function __construct() {
		add_action( 'admin_menu', array( $this, 'admin_menu' ), 50 );
		add_action( 'admin_init', array( $this, 'register_settings' ) );
		add_action( 'wp_ajax_metaforge_ai_test_connection', array( $this, 'test_connection' ) );
	}

	public static function is_wp_ai_connector_connected() {
		if ( function_exists( 'wp_ai_client_prompt' ) ) {
			return true;
		}
		return false;
	}

	public function test_connection() {
		check_ajax_referer( 'metaforge_ai_test_connection_nonce', 'nonce' );

		if ( ! current_user_can( 'manage_metaforge' ) ) {
			wp_send_json_error( array( 'message' => 'Unauthorized' ) );
		}

		$test_key   = sanitize_text_field( wp_unslash( $_POST['api_key'] ?? '' ) );
		$test_model = sanitize_text_field( wp_unslash( $_POST['model'] ?? '' ) );

		$is_connector_connected = self::is_wp_ai_connector_connected();

		if ( $test_model === 'auto_wp_ai' ) {
			if ( ! $is_connector_connected ) {
				wp_send_json_error(
					array(
						'message' => 'WordPress AI Connector has no active provider connected. Please install/connect a provider under Settings > Connectors, or select a direct AI model from the dropdown.',
					)
				);
			}
			$test_key = 'wp_ai_core';
		}

		if ( empty( $test_key ) ) {
			$saved_settings = get_option( 'metaforge_ai_settings', array() );
			if ( ! empty( $saved_settings['api_key'] ) ) {
				$test_key = $saved_settings['api_key'];
			} elseif ( $is_connector_connected ) {
				$test_key = 'wp_ai_core';
			} else {
				wp_send_json_error( array( 'message' => 'Please enter an API key to test, or connect an AI provider under Settings > Connectors.' ) );
			}
		}

		$clean_test_key = class_exists( 'MetaForge_APIClient' ) ? MetaForge_APIClient::decrypt_key( $test_key ) : $test_key;
		$trimmed_key    = trim( $clean_test_key );
		$model_lower    = strtolower( $test_model );

		// Only validate key prefixes if not testing through WordPress Core AI Client
		if ( $clean_test_key !== 'wp_ai_core' ) {
			if ( strpos( $trimmed_key, 'gsk_' ) === 0 && strpos( $model_lower, 'llama' ) === false && strpos( $model_lower, 'distill' ) === false && strpos( $model_lower, 'qwen' ) === false && strpos( $model_lower, 'gpt-oss' ) === false && strpos( $model_lower, 'groq' ) === false ) {
				wp_send_json_error( array( 'message' => 'This key starts with "gsk_", which is a Groq API key. Please select a Groq model from the dropdown.' ) );
			}
			if ( ( strpos( $trimmed_key, 'AIzaSy' ) === 0 || strpos( $trimmed_key, 'AQ.' ) === 0 ) && strpos( $model_lower, 'gemini' ) === false ) {
				wp_send_json_error( array( 'message' => 'This key is a Google Gemini API key. Please select a Google Gemini model from the dropdown.' ) );
			}
			if ( strpos( $trimmed_key, 'sk-ant-' ) === 0 && strpos( $model_lower, 'claude' ) === false ) {
				wp_send_json_error( array( 'message' => 'This key starts with "sk-ant-", which is an Anthropic Claude API key. Please select a Claude model from the dropdown.' ) );
			}
			if ( strpos( $trimmed_key, 'sk-or-v1-' ) === 0 && strpos( $model_lower, 'openrouter' ) === false ) {
				wp_send_json_error( array( 'message' => 'This key starts with "sk-or-v1-", which is an OpenRouter key. Please select OpenRouter from the dropdown.' ) );
			}
			if ( strpos( $trimmed_key, 'sk-proj-' ) === 0 && strpos( $model_lower, 'gpt' ) === false && strpos( $model_lower, 'o1' ) === false && strpos( $model_lower, 'o3' ) === false ) {
				wp_send_json_error( array( 'message' => 'This key starts with "sk-proj-", which is an OpenAI key. Please select an OpenAI model (e.g. GPT-4o Mini) from the dropdown.' ) );
			}
		}

		// Temporarily override the saved options so the client uses the unsaved ones
		$filter_key = ( $clean_test_key === 'wp_ai_core' ) ? '' : $clean_test_key;
		add_filter(
			'pre_option_metaforge_ai_settings',
			function () use ( $filter_key, $test_model ) {
				return array(
					'api_key' => $filter_key,
					'model'   => $test_model,
				);
			}
		);

		// This is a dry run with minimal tokens
		$client = new MetaForge_APIClient();

		$system = "You are a test script. Respond with exactly 'OK'.";
		$user   = 'Test connection.';

		$response = $client->generate( $system, $user, false );

		if ( is_wp_error( $response ) ) {
			$error_data = $response->get_error_data();
			$retry      = isset( $error_data['retry_after'] ) ? intval( $error_data['retry_after'] ) : 0;
			wp_send_json_error(
				array(
					'message'     => $response->get_error_message(),
					'code'        => $response->get_error_code(),
					'retry_after' => $retry,
				)
			);
		}

		wp_send_json_success( array( 'message' => 'Connection successful! Engine is ready.' ) );
	}

	public function admin_menu() {
		add_submenu_page(
			'metaforge',
			'AI Settings',
			'AI Settings',
			'manage_metaforge',
			'metaforge_ai_settings',
			array( $this, 'admin_page' )
		);
	}

	public function register_settings() {
		register_setting( 'metaforge_ai_settings_group', 'metaforge_ai_settings', array( $this, 'sanitize_settings' ) );
	}

	public function sanitize_settings( $input ) {
		$sanitized          = array();
		$model              = isset( $input['model'] ) ? sanitize_text_field( $input['model'] ) : 'gemini-2.0-flash';
		$sanitized['model'] = $model;

		if ( isset( $input['api_key'] ) ) {
			$raw_key              = sanitize_text_field( $input['api_key'] );
			$sanitized['api_key'] = ! empty( $raw_key ) && class_exists( 'MetaForge_APIClient' ) ? MetaForge_APIClient::encrypt_key( $raw_key ) : $raw_key;
		} else {
			$sanitized['api_key'] = '';
		}
		$sanitized['detect_layout']   = isset( $input['detect_layout'] ) ? '1' : '0';
		$sanitized['generate_schema'] = isset( $input['generate_schema'] ) ? '1' : '0';
		$sanitized['fallback_ai']     = isset( $input['fallback_ai'] ) ? '1' : '0';

		return $sanitized;
	}

	public function admin_page() {
		if ( ! current_user_can( 'manage_metaforge' ) ) {
			return;
		}

        // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- Read-only notice query flag.
		if ( isset( $_GET['settings-updated'] ) && sanitize_text_field( wp_unslash( $_GET['settings-updated'] ) ) ) {
			add_settings_error( 'metaforge_ai_messages', 'metaforge_ai_message', 'Settings Saved Successfully', 'updated' );
		}

		$options = get_option(
			'metaforge_ai_settings',
			array(
				'api_key'         => '',
				'model'           => 'gemini-3.5-flash-lite',
				'detect_layout'   => '0',
				'generate_schema' => '0',
				'fallback_ai'     => '0',
			)
		);

		?>
		<div class="wrap mfe-wrap">
			<?php MetaForge_AdminHeader::render_top_bar( 'ai_settings' ); ?>
			<?php
			$docs_link = '<a href="https://searchnscore.com/" target="_blank" class="mfe-btn-secondary"><span class="dashicons dashicons-book-alt"></span> Documentation</a>';
			MetaForge_AdminHeader::render_sub_header( 'AI Settings', '', $docs_link );
			?>

			<?php
			$errors = get_settings_errors( 'metaforge_ai_messages' );
			foreach ( $errors as $error ) {
				$class = ( $error['type'] === 'error' ) ? 'mfe-notice-error' : 'mfe-notice-success';
				echo '<div class="' . esc_attr( $class ) . '"><p>' . esc_html( $error['message'] ) . '</p></div>';
			}
			if ( ! defined( 'AUTH_KEY' ) ) {
				echo '<div class="notice notice-warning inline mfe-mb-10"><p><strong>🔒 Security Recommendation:</strong> <code>AUTH_KEY</code> is not explicitly set in <code>wp-config.php</code>. Define a unique <code>AUTH_KEY</code> or <code>MetaForge_AI_API_KEY</code> in <code>wp-config.php</code> for maximum API key encryption security.</p></div>';
			}
			?>

			<div class="mfe-builder-wrap">
				<form action="options.php" method="post">
					<?php settings_fields( 'metaforge_ai_settings_group' ); ?>

					<!-- AI API Key Configuration Card -->
					<div class="mfe-card mfe-mb-20">
						<div class="mfe-card-header">
							<div class="mfe-card-header-left">
								<span class="dashicons dashicons-admin-network"></span> AI Provider API Configuration
							</div>
						</div>
						<div class="mfe-card-body">
							<?php if ( function_exists( 'wp_ai_client_prompt' ) ) : ?>
								<div class="mfe-ai-core-notice">
									<span class="dashicons dashicons-admin-generic mfe-ai-core-notice-icon"></span>
									<div>
										<strong>WordPress Core AI Client Active (WordPress 7.0+)</strong><br>
										Your site supports centralized AI providers via <strong>Settings &gt; Connectors</strong>.
										If using centralized AI, ensure MetaForge is approved under <a
											href="<?php echo esc_url( admin_url( 'options-connectors.php' ) ); ?>">Settings
											&gt; Connectors</a> (click <strong>"Edit"</strong> on your connected provider and enable
										the checkbox for MetaForge Engine). Direct API keys below are also fully supported as direct
										integrations.
									</div>
								</div>
							<?php endif; ?>
							<table class="form-table">
								<tr valign="top">
									<th scope="row">Provider API Key</th>
									<td>
										<?php
										$is_wp_ai              = function_exists( 'wp_ai_client_prompt' );
										$is_connector_selected = ( $options['model'] === 'auto_wp_ai' );
										$connector_connected   = self::is_wp_ai_connector_connected();
										$has_key               = ! empty( $options['api_key'] );
										$display_api_key       = class_exists( 'MetaForge_APIClient' ) ? MetaForge_APIClient::decrypt_key( $options['api_key'] ) : $options['api_key'];
										?>

										<?php if ( $is_wp_ai ) : ?>
											<?php
											$connector_box_classes   = array( 'mfe-connector-connected-box' );
											$connector_box_classes[] = $connector_connected ? 'is-connected' : 'is-disconnected';
											if ( ! ( $is_connector_selected || ( ! $has_key && $connector_connected ) ) ) {
												$connector_box_classes[] = 'mfe-hidden';
											}
											?>
											<div id="mfe-wp-connector-notice" class="<?php echo esc_attr( implode( ' ', $connector_box_classes ) ); ?>"
												data-connected="<?php echo $connector_connected ? '1' : '0'; ?>">
												<?php if ( $connector_connected ) : ?>
													<span class="dashicons dashicons-yes-alt mfe-connector-icon"></span>
													<div>
														<strong class="mfe-connector-title"><?php esc_html_e( 'Connected via WordPress AI Connector', 'metaforge-engine' ); ?></strong>
														<p class="mfe-connector-desc">
															<?php esc_html_e( 'Your site uses centralized AI credentials configured in Settings > Connectors. When no direct API key is entered, MetaForge automatically uses your connected AI provider.', 'metaforge-engine' ); ?>
															<a href="<?php echo esc_url( admin_url( 'options-connectors.php' ) ); ?>" target="_blank" class="mfe-connector-link">
																<?php esc_html_e( 'Settings > Connectors &rarr;', 'metaforge-engine' ); ?>
															</a>
														</p>
													</div>
												<?php else : ?>
													<span class="dashicons dashicons-warning mfe-connector-icon"></span>
													<div>
														<strong class="mfe-connector-title"><?php esc_html_e( 'WordPress AI Connector Not Connected Yet', 'metaforge-engine' ); ?></strong>
														<p class="mfe-connector-desc">
															<?php esc_html_e( 'Default Site AI Provider is selected, but no AI connector is connected yet in WordPress.', 'metaforge-engine' ); ?>
															<a href="<?php echo esc_url( admin_url( 'options-connectors.php' ) ); ?>" target="_blank" class="mfe-connector-link">
																<?php esc_html_e( 'Connect a provider in Settings > Connectors &rarr;', 'metaforge-engine' ); ?>
															</a>
															<?php esc_html_e( 'or enter a direct API key below.', 'metaforge-engine' ); ?>
														</p>
													</div>
												<?php endif; ?>
											</div>
										<?php endif; ?>

										<div class="mfe-ai-key-container">
											<div class="mfe-ai-key-input-wrap">
												<input type="password" name="metaforge_ai_settings[api_key]" id="mfe-ai-api-key"
													value="<?php echo ( $options['model'] === 'auto_wp_ai' ) ? '' : esc_attr( $display_api_key ); ?>"
													class="regular-text mfe-ai-key-input <?php echo $has_key ? 'is-configured' : ''; ?>"
													<?php echo ( $options['model'] === 'auto_wp_ai' ) ? 'disabled="disabled"' : ( $has_key ? 'readonly="readonly"' : '' ); ?>
													placeholder="<?php echo ( $options['model'] === 'auto_wp_ai' ) ? esc_attr__( 'WordPress AI Connector mode (Direct key disabled)', 'metaforge-engine' ) : esc_attr__( 'Enter your API key', 'metaforge-engine' ); ?>" />

												<button type="button" class="mfe-toggle-eye-btn mfe-ai-eye-btn"
													data-target="#mfe-ai-api-key" title="Toggle visibility">
													<span class="dashicons dashicons-visibility mfe-ai-eye-icon"></span>
												</button>
											</div>

											<?php if ( $options['model'] === 'auto_wp_ai' ) : ?>
												<button type="button" class="button mfe-edit-api-btn is-hidden">Edit</button>
												<button type="button" class="button mfe-ai-test-btn">Test Connector</button>
											<?php elseif ( $has_key ) : ?>
												<button type="button" class="button mfe-edit-api-btn">Edit</button>
												<button type="button" class="button mfe-ai-test-btn is-hidden">Test Connection</button>
											<?php else : ?>
												<button type="button" class="button mfe-edit-api-btn is-hidden">Edit</button>
												<button type="button" class="button mfe-ai-test-btn">Test Connection</button>
											<?php endif; ?>
										</div>

										<div id="mfe-ai-test-result"
											class="mfe-ai-test-status <?php echo $has_key ? '' : 'is-hidden'; ?>">
											<?php if ( $has_key ) : ?>
												<span class="mfe-ai-test-success"><span class="dashicons dashicons-yes-alt"></span>
													API Key Configured</span>
											<?php endif; ?>
										</div>
										<p class="description">
											Your API key is stored securely using AES-256-CBC encryption in the database.<br>
											<?php if ( function_exists( 'wp_ai_client_prompt' ) ) : ?>
												<strong>WordPress AI Client:</strong> Centrally managed under <a
													href="<?php echo esc_url( admin_url( 'options-connectors.php' ) ); ?>">Settings
													&gt; Connectors</a>.<br>
											<?php endif; ?>
											<strong>DeepSeek:</strong> Get your key at <a
												href="https://platform.deepseek.com/api_keys"
												target="_blank">platform.deepseek.com</a><br>
											<strong>Google Gemini:</strong> Get a free API key at <a
												href="https://aistudio.google.com/app/apikey"
												target="_blank">aistudio.google.com</a><br>
											<strong>Groq (Fast / Free Llama 3):</strong> Get a free tier key at <a
												href="https://console.groq.com/keys" target="_blank">console.groq.com</a><br>
											<strong>OpenAI:</strong> Get an API key at <a
												href="https://platform.openai.com/api-keys"
												target="_blank">platform.openai.com</a><br>
											<strong>Anthropic (Claude):</strong> Get a key at <a
												href="https://console.anthropic.com/settings/keys"
												target="_blank">console.anthropic.com</a><br>
											<strong>OpenRouter:</strong> Universal gateway key at <a
												href="https://openrouter.ai/keys" target="_blank">openrouter.ai</a>
										</p>
									</td>
								</tr>
								<tr valign="top">
									<th scope="row">AI Model Selection</th>
									<td>
										<select name="metaforge_ai_settings[model]" class="mfe-ai-model-select"
											id="mfe-ai-model-select">
											<?php if ( function_exists( 'wp_ai_client_prompt' ) ) : ?>
												<optgroup label="— WordPress Core AI Client (WordPress 7.0+)">
													<option value="auto_wp_ai" <?php selected( $options['model'], 'auto_wp_ai' ); ?>>
														Default Site AI Provider (Settings &gt; Connectors)</option>
												</optgroup>
											<?php endif; ?>
											<optgroup label="— DeepSeek Direct API">
												<option value="deepseek-chat" <?php selected( $options['model'], 'deepseek-chat' ); ?>>DeepSeek V3 (High Intelligence, Low Cost)</option>
												<option value="deepseek-reasoner" <?php selected( $options['model'], 'deepseek-reasoner' ); ?>>DeepSeek R1 (Deep Reasoning Engine)</option>
											</optgroup>
											<optgroup label="— Groq API">
												<option value="openai/gpt-oss-120b" <?php selected( $options['model'], 'openai/gpt-oss-120b' ); ?>>GPT-OSS 120B (Groq Fast Reasoning - Recommended
													⚡)</option>
												<option value="openai/gpt-oss-20b" <?php selected( $options['model'], 'openai/gpt-oss-20b' ); ?>>GPT-OSS 20B (Groq Ultra-Fast)</option>
												<option value="qwen/qwen3.8-27b" <?php selected( $options['model'], 'qwen/qwen3.8-27b' ); ?>>Qwen 3.8 27B (Groq Vision &amp; Text)</option>
												<option value="llama-3.1-8b-instant" <?php selected( $options['model'], 'llama-3.1-8b-instant' ); ?>>Llama 3.1 8B Instant (Legacy)</option>
												<option value="llama-3.3-70b-versatile" <?php selected( $options['model'], 'llama-3.3-70b-versatile' ); ?>>Llama 3.3 70B Versatile (Legacy)</option>
											</optgroup>
											<optgroup label="— Google Gemini">
												<option value="gemini-3.5-flash-lite" <?php selected( $options['model'], 'gemini-3.5-flash-lite' ); ?>>Gemini 3.5 Flash Lite (Fast &amp; Free Tier -
													Recommended ⚡)</option>
												<option value="gemini-2.0-flash" <?php selected( $options['model'], 'gemini-2.0-flash' ); ?>>Gemini 2.0 Flash (Recommended ⚡)</option>
												<option value="gemini-2.5-flash" <?php selected( $options['model'], 'gemini-2.5-flash' ); ?>>Gemini 2.5 Flash</option>
												<option value="gemini-1.5-flash" <?php selected( $options['model'], 'gemini-1.5-flash' ); ?>>Gemini 1.5 Flash (Fast)</option>
												<option value="gemini-1.5-pro" <?php selected( $options['model'], 'gemini-1.5-pro' ); ?>>Gemini 1.5 Pro</option>
											</optgroup>
											<optgroup label="— OpenAI">
												<option value="gpt-4o-mini" <?php selected( $options['model'], 'gpt-4o-mini' ); ?>>GPT-4o Mini (Fast &amp; Cost-effective)</option>
												<option value="gpt-4o" <?php selected( $options['model'], 'gpt-4o' ); ?>>GPT-4o
													(Omni Benchmark Leader)</option>
												<option value="o3-mini" <?php selected( $options['model'], 'o3-mini' ); ?>>o3-mini
													(Reasoning Model)</option>
												<option value="o1" <?php selected( $options['model'], 'o1' ); ?>>o1 (Full
													Reasoning Model)</option>
											</optgroup>
											<optgroup label="— Anthropic Claude">
												<option value="claude-3-5-sonnet-20241022" <?php selected( $options['model'], 'claude-3-5-sonnet-20241022' ); ?>>Claude 3.5 Sonnet (Recommended ⚡)</option>
												<option value="claude-3-7-sonnet-20250219" <?php selected( $options['model'], 'claude-3-7-sonnet-20250219' ); ?>>Claude 3.7 Sonnet (Hybrid Reasoning)
												</option>
												<option value="claude-3-5-haiku-20241022" <?php selected( $options['model'], 'claude-3-5-haiku-20241022' ); ?>>Claude 3.5 Haiku (Fast)</option>
												<option value="claude-3-opus-20240229" <?php selected( $options['model'], 'claude-3-opus-20240229' ); ?>>Claude 3 Opus</option>
											</optgroup>
											<optgroup label="— OpenRouter Gateway">
												<option value="openrouter/auto" <?php selected( $options['model'], 'openrouter/auto' ); ?>>OpenRouter Auto Route</option>
											</optgroup>
										</select>
										<p class="description">Select the target AI model. Ensure your API Key corresponds to
											your chosen provider key.</p>
									</td>
								</tr>
							</table>
						</div>
					</div>

					<div class="mfe-card mfe-mb-20">
						<div class="mfe-card-header">
							<div class="mfe-card-header-left">
								<span class="dashicons dashicons-admin-generic"></span> Feature Toggles
							</div>
						</div>
						<div class="mfe-card-body">
							<table class="form-table">
								<tr valign="top">
									<th scope="row">Smart Layout Detection</th>
									<td>
										<label>
											<input type="checkbox" name="metaforge_ai_settings[detect_layout]" value="1" <?php checked( $options['detect_layout'], '1' ); ?> />
											Enable AI-suggested UI layouts based on field types
										</label>
									</td>
								</tr>
								<tr valign="top">
									<th scope="row">Schema Generation</th>
									<td>
										<label>
											<input type="checkbox" name="metaforge_ai_settings[generate_schema]" value="1" <?php checked( $options['generate_schema'], '1' ); ?> />
											Automatically generate structured schema (JSON-LD) for SEO
										</label>
									</td>
								</tr>
								<tr valign="top">
									<th scope="row">Fallback AI Generation</th>
									<td>
										<label>
											<input type="checkbox" name="metaforge_ai_settings[fallback_ai]" value="1" <?php checked( $options['fallback_ai'], '1' ); ?> />
											Allow AI to generate placeholder content when fields are left blank
										</label>
									</td>
								</tr>
							</table>
						</div>
					</div>

					<p class="submit">
						<button type="submit" class="mfe-btn-primary">Save Settings</button>
					</p>
				</form>
			</div>
			<?php MetaForge_AdminHeader::render_footer(); ?>
		</div>
		<?php
	}
}