<?php

declare(strict_types=1);

// phpcs:disable WordPress.Security.EscapeOutput -- No HTML output in this file; exception messages are not rendered.

namespace WordPress\CloudflareWorkersAiProvider\Provider;

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;
use WordPress\CloudflareWorkersAiProvider\Metadata\CloudflareWorkersAiModelMetadataDirectory;
use WordPress\CloudflareWorkersAiProvider\Models\CloudflareWorkersAiImageGenerationModel;
use WordPress\CloudflareWorkersAiProvider\Models\CloudflareWorkersAiTextGenerationModel;
use WordPress\CloudflareWorkersAiProvider\Models\CloudflareWorkersAiTranslationModel;

class CloudflareWorkersAiProvider extends AbstractApiProvider
{
    public const ID = 'cloudflare-workers-ai';

    protected static function baseUrl(): string
    {
        $accountId = '';
        if (function_exists('WordPress\\CloudflareWorkersAiProvider\\get_account_id')) {
            $accountId = \WordPress\CloudflareWorkersAiProvider\get_account_id();
        }
        return 'https://api.cloudflare.com/client/v4/accounts/' . rawurlencode($accountId) . '/ai/v1';
    }

    protected static function createModel(
        ModelMetadata $modelMetadata,
        ProviderMetadata $providerMetadata
    ): ModelInterface {
        $capabilities = $modelMetadata->getSupportedCapabilities();
        foreach ($capabilities as $capability) {
            if ($capability->isImageGeneration()) {
                return new CloudflareWorkersAiImageGenerationModel($modelMetadata, $providerMetadata);
            }
            if ($capability->isTextGeneration()) {
                if (CloudflareWorkersAiTranslationModel::supports($modelMetadata->getId())) {
                    return new CloudflareWorkersAiTranslationModel($modelMetadata, $providerMetadata);
                }
                return new CloudflareWorkersAiTextGenerationModel($modelMetadata, $providerMetadata);
            }
        }

        throw new RuntimeException(
            'Unsupported model capabilities: ' . implode(', ', $capabilities)
        );
    }

    protected static function createProviderMetadata(): ProviderMetadata
    {
        return new ProviderMetadata(
            self::ID,
            'Cloudflare Workers AI',
            ProviderTypeEnum::cloud(),
            'https://dash.cloudflare.com/?to=/:account/ai/workers-ai/api-quick-start',
            RequestAuthenticationMethod::apiKey(),
            __('Text, image and translation models on Cloudflare Workers AI, with a free daily allowance.', 'aipcf-ai-provider-for-cloudflare'),
            plugin_dir_path(\AIPCF_FILE) . 'assets/images/cloudflare.svg'
        );
    }

    protected static function createProviderAvailability(): ProviderAvailabilityInterface
    {
        return new ListModelsApiBasedProviderAvailability(
            static::modelMetadataDirectory()
        );
    }

    protected static function createModelMetadataDirectory(): ModelMetadataDirectoryInterface
    {
        return new CloudflareWorkersAiModelMetadataDirectory();
    }
}