<?php
/**
 * Class ATFPP\AI_Translate\Deepl\Deepl_AI_API_Client
 *
 * @since 0.1.0
 * @package ai-services
 */

namespace ATFPP\AI_Translate\Deepl;

use ATFPP\AI_Translate\Services\Base\Generic_AI_API_Client;
use ATFPP\AI_Translate\Services\Contracts\Authentication;
use Felix_Arntz\ATFPP\WP_OOP_Plugin_Lib\HTTP\Contracts\Request_Handler;

/**
 * Class to interact directly with the Deepl API.
 *
 * @since 0.1.0
 * @since 0.7.0 Now extends `Generic_AI_API_Client`.
 */
class Deepl_AI_API_Client extends Generic_AI_API_Client {

	/**
	 * Constructor.
	 *
	 * @since 0.7.0
	 *
	 * @param string              $default_base_url    The default base URL for the API.
	 * @param string              $default_api_version The default API version.
	 * @param string              $api_name            The (human-readable) API name.
	 * @param Request_Handler     $request_handler     The request handler instance.
	 * @param Authentication|null $authentication      Optional. The authentication instance. Default null.
	 */
	public function __construct(
		string $default_base_url,
		string $default_api_version,
		string $api_name,
		Request_Handler $request_handler,
		?Authentication $authentication = null
	) {
		// Set custom header name for Deepl API key authentication.
		if ( $authentication ) {
			$authentication->set_authencation_scheme( 'DeepL-Auth-Key' );
		}

		$key_name=$authentication->get_option_definitions('deepl');

		$key_value=get_option(array_keys($key_name)[0], '');

		$key_free=str_ends_with($key_value, ':fx');
		
		if($key_free){
			$default_base_url = 'https://api-free.deepl.com';
		}

		parent::__construct( $default_base_url, $default_api_version, $api_name, $request_handler, $authentication );
	}

	/**
	 * Gets the request URL for the specified model and task.
	 *
	 * @since 0.1.0
	 *
	 * @param string               $path            The path to the API endpoint, relative to the base URL and version.
	 * @param array<string, mixed> $request_options Optional. The request options. Default empty array.
	 * @return string The request URL.
	 */
	protected function get_request_url( string $path, array $request_options = array() ): string {
		if ( isset( $request_options['stream'] ) && $request_options['stream'] && ! str_ends_with( $path, '?alt=sse' ) ) {
			$path .= '?alt=sse';
		}

		return parent::get_request_url( $path, $request_options );
	}

	/**
	 * Adds additional default request options to the given request options.
	 *
	 * @since 0.7.0
	 *
	 * @param array<string, mixed> $request_options The request options.
	 * @return array<string, mixed> The updated request options.
	 */
	protected function add_default_options( array $request_options ): array {
		$request_options = parent::add_default_options( $request_options );

		return $request_options;
	}
}
