<?php

declare(strict_types=1);

namespace RaCar\CursorAiProvider\Provider;

use RaCar\CursorAiProvider\Metadata\CursorModelMetadataDirectory;
use RaCar\CursorAiProvider\Models\CursorTextGenerationModel;
use WordPress\AiClient\AiClient;
use WordPress\AiClient\Common\Exception\RuntimeException;
use WordPress\AiClient\Providers\ApiBasedImplementation\AbstractApiProvider;
use WordPress\AiClient\Providers\ApiBasedImplementation\ListModelsApiBasedProviderAvailability;
use WordPress\AiClient\Providers\Contracts\ModelMetadataDirectoryInterface;
use WordPress\AiClient\Providers\Contracts\ProviderAvailabilityInterface;
use WordPress\AiClient\Providers\DTO\ProviderMetadata;
use WordPress\AiClient\Providers\Enums\ProviderTypeEnum;
use WordPress\AiClient\Providers\Http\Enums\RequestAuthenticationMethod;
use WordPress\AiClient\Providers\Models\Contracts\ModelInterface;
use WordPress\AiClient\Providers\Models\DTO\ModelMetadata;

/**
 * Class for the AI Provider for Cursor.
 *
 * @since 0.1.0
 */
class CursorProvider extends AbstractApiProvider
{
	/**
	 * {@inheritDoc}
	 *
	 * @since 0.1.0
	 */
	protected static function baseUrl(): string
	{
		return 'https://api.cursor.com';
	}

	/**
	 * {@inheritDoc}
	 *
	 * @since 0.1.0
	 */
	protected static function createModel(
		ModelMetadata $modelMetadata,
		ProviderMetadata $providerMetadata
	): ModelInterface {
		$capabilities = $modelMetadata->getSupportedCapabilities();
		foreach ($capabilities as $capability) {
			if ($capability->isTextGeneration()) {
				return new CursorTextGenerationModel($modelMetadata, $providerMetadata);
			}
		}

		throw new RuntimeException(
			esc_html(
				'Unsupported model capabilities: ' . implode(
					', ',
					array_map(
						static fn ($capability): string => sanitize_text_field((string) $capability),
						$capabilities
					)
				)
			)
		);
	}

	/**
	 * {@inheritDoc}
	 *
	 * @since 0.1.0
	 */
	protected static function createProviderMetadata(): ProviderMetadata
	{
		$providerMetadataArgs = array(
			'cursor',
			'Cursor',
			ProviderTypeEnum::cloud(),
			'https://cursor.com/dashboard/api',
			RequestAuthenticationMethod::apiKey(),
		);

		// Provider description support was added in 1.2.0.
		if (version_compare(AiClient::VERSION, '1.2.0', '>=')) {
			if (function_exists('__')) {
				$providerMetadataArgs[] = __(
					'Text generation via Cursor Cloud Agents for the WordPress AI Client.',
					'ai-provider-for-cursor'
				);
			} else {
				$providerMetadataArgs[] = 'Text generation via Cursor Cloud Agents for the WordPress AI Client.';
			}
		}

		// Provider logoPath support was added in 1.3.0.
		if (version_compare(AiClient::VERSION, '1.3.0', '>=')) {
			$providerMetadataArgs[] = dirname(__DIR__, 2) . '/assets/images/cursor.svg';
		}

		return new ProviderMetadata(...$providerMetadataArgs);
	}

	/**
	 * {@inheritDoc}
	 *
	 * @since 0.1.0
	 */
	protected static function createProviderAvailability(): ProviderAvailabilityInterface
	{
		return new ListModelsApiBasedProviderAvailability(
			static::modelMetadataDirectory()
		);
	}

	/**
	 * {@inheritDoc}
	 *
	 * @since 0.1.0
	 */
	protected static function createModelMetadataDirectory(): ModelMetadataDirectoryInterface
	{
		return new CursorModelMetadataDirectory();
	}
}