<?php

declare(strict_types=1);

namespace KTXM\ProviderImap\Client\Protocol\Command;

use KTXM\ProviderImap\Client\Protocol\IdentifierMode;

use KTXM\ProviderImap\Client\Protocol\Parser\FetchResponseParser;
use Generator;
use KTXM\ProviderImap\Client\Protocol\Command\Argument\MessageTarget;
use KTXM\ProviderImap\Client\Protocol\Command\Argument\FetchOptions;
use KTXM\ProviderImap\Client\ImapException;
use KTXM\ProviderImap\Client\Message;
use KTXM\ProviderImap\Client\Protocol\RequestFrame;
use KTXM\ProviderImap\Client\Protocol\ResponseStream;
use KTXM\ProviderImap\Client\Protocol\SequenceSet;
use KTXM\ProviderImap\Client\Protocol\SessionContext;
use KTXM\ProviderImap\Client\Protocol\SessionState;

/**
 * @implements CommandInterface<Generator<int, Message>>
 */
final class FetchManyCommand implements CommandInterface
{
    private readonly MessageTarget $target;
    private readonly FetchOptions $options;

    public function __construct(MessageTarget|string|SequenceSet|null $target = null, ?FetchOptions $options = null)
    {
        $this->target = match (true) {
            $target instanceof MessageTarget => $target,
            $target instanceof SequenceSet => MessageTarget::sequence($target),
            is_string($target) => MessageTarget::sequence($target),
            default => MessageTarget::all(),
        };
        $this->options = $options ?? FetchOptions::default();
    }

    public function name(): string
    {
        return 'FETCH';
    }

    public function allowedStates(): array
    {
        return [SessionState::Selected];
    }

    public function encode(string $tag, SessionContext $context): RequestFrame
    {
        unset($tag, $context);

        return new RequestFrame(sprintf(
            '%s %s (%s)',
            $this->target->identifierMode() === IdentifierMode::Uid ? 'UID FETCH' : 'FETCH',
            $this->target->sequenceSet()->toCommand(),
            $this->options->toCommand(),
        ));
    }

    public function handle(ResponseStream $responses, SessionContext $context): Generator
    {
        if ($context->selectedMailbox() === null) {
            throw new ImapException('FETCH requires a selected mailbox.');
        }

        return (new FetchResponseParser())->parseMany($responses);
    }
}