<?php
namespace BavarianRankEngine\Providers;

class OpenAIProvider implements ProviderInterface {
	private const API_URL = 'https://api.openai.com/v1/chat/completions';

	public function getId(): string {
		return 'openai'; }
	public function getName(): string {
		return 'OpenAI'; }

	public function getModels(): array {
		return array(
			'gpt-4.1'       => 'GPT-4.1 (Empfohlen)',
			'gpt-4o'        => 'GPT-4o',
			'gpt-4o-mini'   => 'GPT-4o Mini (Günstig)',
			'gpt-3.5-turbo' => 'GPT-3.5 Turbo (Sehr günstig)',
		);
	}

	public function testConnection( string $api_key ): array {
		try {
			$this->generateText( 'Say "ok"', $api_key, 'gpt-4o-mini', 5 );
			return array(
				'success' => true,
				'message' => 'Verbindung erfolgreich',
			);
		} catch ( \RuntimeException $e ) {
			return array(
				'success' => false,
				'message' => $e->getMessage(),
			);
		}
	}

	public function generateText( string $prompt, string $api_key, string $model, int $max_tokens = 300 ): string {
		$response = wp_remote_post(
			self::API_URL,
			array(
				'timeout' => 30,
				'headers' => array(
					'Authorization' => 'Bearer ' . $api_key,
					'Content-Type'  => 'application/json',
				),
				'body'    => wp_json_encode(
					array(
						'model'      => $model,
						'messages'   => array(
							array(
								'role'    => 'user',
								'content' => $prompt,
							),
						),
						'max_tokens' => $max_tokens,
					)
				),
			)
		);

		return $this->parseResponse( $response );
	}

	private function parseResponse( $response ): string {
		if ( is_wp_error( $response ) ) {
			throw new \RuntimeException( esc_html( $response->get_error_message() ) );
		}

		$code = wp_remote_retrieve_response_code( $response );
		$body = json_decode( wp_remote_retrieve_body( $response ), true );

		if ( $code !== 200 ) {
			$msg = $body['error']['message'] ?? "HTTP $code";
			throw new \RuntimeException( esc_html( $msg ) );
		}

		return trim( $body['choices'][0]['message']['content'] ?? '' );
	}
}