newznab-tmux/app/Services/TvProcessing/Pipes/AbstractTvProviderPipe.php

<?php

declare(strict_types=1);

namespace App\Services\TvProcessing\Pipes;

use App\Services\TvProcessing\TvProcessingPassable;
use App\Services\TvProcessing\TvProcessingResult;
use Closure;

/**
 * Base class for TV processing pipe handlers.
 *
 * Each pipe is responsible for processing releases through a specific provider.
 */
abstract class AbstractTvProviderPipe
{
    protected int $priority = 50;

    protected bool $echoOutput = true;

    /**
     * @var array<string, mixed>
     */
    protected array $titleCache = [];

    public function __construct() {}

    /**
     * Handle the TV processing request.
     */
    public function handle(TvProcessingPassable $passable, Closure $next): TvProcessingPassable
    {
        // If we already have a match, skip processing
        if ($passable->shouldStopProcessing()) {
            return $next($passable);
        }

        // Skip if we don't have valid parsed info
        if (! $passable->hasValidParsedInfo()) {
            return $next($passable);
        }

        // Skip if this provider shouldn't process this release
        if ($this->shouldSkip($passable)) {
            $passable->updateResult(
                TvProcessingResult::skipped('Provider skipped', $this->getName()),
                $this->getName()
            );

            return $next($passable);
        }

        // Attempt to process with this provider
        $result = $this->process($passable);

        // Update the result
        $passable->updateResult($result, $this->getName());

        return $next($passable);
    }

    /**
     * Get the priority of this provider (lower = higher priority).
     */
    public function getPriority(): int
    {
        return $this->priority;
    }

    /**
     * Get the name of this provider for debugging/logging.
     */
    abstract public function getName(): string;

    /**
     * Get the status code for releases awaiting this provider.
     */
    abstract public function getStatusCode(): int;

    /**
     * Attempt to process the release through this provider.
     */
    abstract protected function process(TvProcessingPassable $passable): TvProcessingResult;

    /**
     * Check if this provider should be skipped for the given passable.
     */
    protected function shouldSkip(TvProcessingPassable $passable): bool
    {
        return false;
    }

    /**
     * Set echo output flag.
     */
    public function setEchoOutput(bool $echo): self
    {
        $this->echoOutput = $echo;

        return $this;
    }

    /**
     * Check if a title is in the failure cache.
     */
    protected function isInTitleCache(string $title): bool
    {
        return in_array($title, $this->titleCache, true);
    }

    /**
     * Add a title to the failure cache.
     */
    protected function addToTitleCache(string $title): void
    {
        $this->titleCache[] = $title;
    }

    /**
     * Clear the title cache.
     */
    public function clearTitleCache(): void
    {
        $this->titleCache = [];
    }

    /**
     * Truncate title for display purposes.
     */
    protected function truncateTitle(string $title, int $maxLength = 45): string
    {
        if (mb_strlen($title) <= $maxLength) {
            return $title;
        }

        return mb_substr($title, 0, $maxLength - 3).'...';
    }

    /**
     * Output match success message.
     */
    protected function outputMatch(string $title, ?int $season = null, ?int $episode = null, ?string $airdate = null): void
    {
        if (! $this->echoOutput) {
            return;
        }

        cli()->primaryOver('    → ');
        cli()->headerOver($this->truncateTitle($title));

        if ($airdate !== null) {
            cli()->primaryOver(' | ');
            cli()->warningOver($airdate);
        } elseif ($season !== null && $episode !== null) {
            cli()->primaryOver(' S');
            cli()->warningOver(sprintf('%02d', $season));
            cli()->primaryOver('E');
            cli()->warningOver(sprintf('%02d', $episode));
        }

        cli()->primaryOver(' ✓ ');
        cli()->primary('MATCHED ('.$this->getName().')');
    }

    /**
     * Output not found message.
     */
    protected function outputNotFound(string $title): void
    {
        if (! $this->echoOutput) {
            return;
        }

        cli()->primaryOver('    → ');
        cli()->alternateOver($this->truncateTitle($title));
        cli()->primaryOver(' → ');
        cli()->alternate('Not found in '.$this->getName());
    }

    /**
     * Output skipped message.
     */
    protected function outputSkipped(string $title): void
    {
        if (! $this->echoOutput) {
            return;
        }

        cli()->primaryOver('    → ');
        cli()->alternateOver($this->truncateTitle($title));
        cli()->primaryOver(' → ');
        cli()->alternate('Skipped (previously failed)');
    }

    /**
     * Output searching message.
     */
    protected function outputSearching(string $title): void
    {
        if (! $this->echoOutput) {
            return;
        }

        cli()->primaryOver('    → ');
        cli()->headerOver($this->truncateTitle($title));
        cli()->primaryOver(' → ');
        cli()->info('Searching '.$this->getName().'...');
    }

    /**
     * Output found in DB message.
     */
    protected function outputFoundInDb(string $title): void
    {
        if (! $this->echoOutput) {
            return;
        }

        cli()->primaryOver('    → ');
        cli()->headerOver($this->truncateTitle($title));
        cli()->primaryOver(' → ');
        cli()->info('Found in DB');
    }
}