<?php

declare(strict_types=1);

namespace App\Services\Ai;

use App\Models\AiProvider;
use GuzzleHttp\Client;
use GuzzleHttp\Exception\GuzzleException;
use Illuminate\Support\Facades\Log;

class StepFunService implements AiServiceInterface
{
    /**
     * @param AiProvider $provider
     */
    public function __construct(private AiProvider $provider)
    {
        $this->client = new Client([
            'base_uri' => rtrim((string) $provider->base_url, '/') . '/',
            'timeout' => 120,
        ]);
    }

    public function chat(array $messages, array $data = []): array
    {
        $apiKey = $data['api_key'] ?? $this->provider->api_key;

        if (empty($apiKey)) {
            throw new \InvalidArgumentException('StepFun API key is not configured.');
        }

        try {
            $payload = [
                'model' => (string) ($data['model'] ?? ''),
                'messages' => $messages,
            ];

            if (isset($data['temperature'])) {
                $payload['temperature'] = (float) $data['temperature'];
            }

            if (isset($data['max_tokens'])) {
                $payload['max_tokens'] = (int) $data['max_tokens'];
            }

            if (!empty($data['reasoning_effort'])) {
                $payload['reasoning_effort'] = (string) $data['reasoning_effort'];
            }

            $response = $this->client->post('chat/completions', [
                'headers' => [
                    'Authorization' => 'Bearer ' . $apiKey,
                    'Content-Type' => 'application/json',
                ],
                'json' => $payload,
            ]);

            $body = json_decode($response->getBody()->getContents(), true);

            if (!isset($body['choices'][0]['message']['content'])) {
                throw new \RuntimeException('Invalid response structure from StepFun.');
            }

            return [
                'content' => $body['choices'][0]['message']['content'],
                'tokens_used' => $body['usage']['total_tokens'] ?? null,
                'model' => $body['model'] ?? (string) ($data['model'] ?? ''),
            ];
        } catch (GuzzleException $e) {
            Log::error('StepFun API error: ' . $e->getMessage());
            throw new \RuntimeException('Failed to generate prompt with StepFun: ' . $e->getMessage(), (int) $e->getCode(), $e);
        }
    }

    /**
     * {@inheritdoc}
     */
    public function generatePrompt(array $data): array
    {
        return $this->chat(
            [['role' => 'user', 'content' => (string) ($data['prompt'] ?? '')]],
            $data
        );
    }
}