<?php

declare(strict_types=1);

namespace RaCar\CursorAiProvider\Metadata;

use RaCar\CursorAiProvider\Provider\CursorProvider;
use WordPress\AiClient\Messages\Enums\ModalityEnum;
use WordPress\AiClient\Providers\Http\DTO\Request;
use WordPress\AiClient\Providers\Http\DTO\Response;
use WordPress\AiClient\Providers\Http\Enums\HttpMethodEnum;
use WordPress\AiClient\Providers\Http\Exception\ResponseException;
use WordPress\AiClient\Providers\Models\DTO\ModelMetadata;
use WordPress\AiClient\Providers\Models\DTO\SupportedOption;
use WordPress\AiClient\Providers\Models\Enums\CapabilityEnum;
use WordPress\AiClient\Providers\Models\Enums\OptionEnum;
use WordPress\AiClient\Providers\OpenAiCompatibleImplementation\AbstractOpenAiCompatibleModelMetadataDirectory;

/**
 * Model metadata directory for Cursor Cloud Agents.
 *
 * @since 0.1.0
 *
 * @phpstan-type ModelsResponseData array{
 *     items?: list<array{id: string, displayName?: string, description?: string}>,
 *     models?: list<string|array{id?: string, displayName?: string}>
 * }
 */
class CursorModelMetadataDirectory extends AbstractOpenAiCompatibleModelMetadataDirectory
{
	/**
	 * {@inheritDoc}
	 *
	 * @since 0.1.0
	 */
	protected function createRequest(HttpMethodEnum $method, string $path, array $headers = [], $data = null): Request
	{
		return new Request(
			$method,
			CursorProvider::url('v1/' . ltrim($path, '/')),
			$headers,
			$data
		);
	}

	/**
	 * {@inheritDoc}
	 *
	 * @since 0.1.0
	 */
	protected function parseResponseToModelMetadataList(Response $response): array
	{
		/** @var ModelsResponseData $responseData */
		$responseData = $response->getData();

		$items = array();
		if (isset($responseData['items']) && is_array($responseData['items'])) {
			$items = $responseData['items'];
		} elseif (isset($responseData['models']) && is_array($responseData['models'])) {
			$items = $responseData['models'];
		} else {
			throw ResponseException::fromMissingData('Cursor', 'items');
		}

		$capabilities = array(
			CapabilityEnum::textGeneration(),
			CapabilityEnum::chatHistory(),
		);

		// Declare options commonly set by the WordPress AI plugin so model
		// matching succeeds. Cursor Cloud Agents ignore sampling params at
		// the HTTP layer; system instructions are folded into the prompt text.
		$options = array(
			new SupportedOption(OptionEnum::systemInstruction()),
			new SupportedOption(OptionEnum::temperature()),
			new SupportedOption(OptionEnum::maxTokens()),
			new SupportedOption(OptionEnum::outputMimeType(), array('text/plain', 'application/json')),
			new SupportedOption(OptionEnum::outputSchema()),
			new SupportedOption(OptionEnum::customOptions()),
			new SupportedOption(OptionEnum::inputModalities(), array(array(ModalityEnum::text()))),
			new SupportedOption(OptionEnum::outputModalities(), array(array(ModalityEnum::text()))),
		);

		$models = array();
		foreach ($items as $item) {
			if (is_string($item)) {
				$modelId   = $item;
				$modelName = $item;
			} elseif (is_array($item) && isset($item['id']) && is_string($item['id'])) {
				$modelId   = $item['id'];
				$modelName = isset($item['displayName']) && is_string($item['displayName'])
					? $item['displayName']
					: $modelId;
			} else {
				continue;
			}

			$models[] = new ModelMetadata(
				$modelId,
				$modelName,
				$capabilities,
				$options
			);
		}

		return array_values($models);
	}
}