<?php

namespace App\Services\Care\Pediatrics;

/**
 * Map pediatrics clinical progress onto visit stages.
 */
class PediatricsWorkflowService
{
    public const STAGE_CHECK_IN = 'check_in';

    public const STAGE_HISTORY = 'history';

    public const STAGE_EXAM = 'exam';

    public const STAGE_INVESTIGATION = 'investigation';

    public const STAGE_PLAN = 'plan';

    public const STAGE_TREATMENT = 'treatment';

    public const STAGE_COMPLETED = 'completed';

    /**
     * @param  array<string, mixed>  $payload
     */
    public function stageFromHistory(array $payload): string
    {
        $history = trim((string) ($payload['history'] ?? ''));
        if ($history !== '') {
            return self::STAGE_HISTORY;
        }

        return self::STAGE_CHECK_IN;
    }

    /**
     * @param  array<string, mixed>  $payload
     */
    public function stageFromExam(array $payload): string
    {
        if (! empty($payload['examination']) || ! empty($payload['chief_complaint'])) {
            return self::STAGE_EXAM;
        }

        return self::STAGE_HISTORY;
    }

    /**
     * @param  array<string, mixed>  $payload
     */
    public function stageFromInvestigation(array $payload): string
    {
        $findings = trim((string) ($payload['findings'] ?? ''));
        if ($findings !== '') {
            return self::STAGE_INVESTIGATION;
        }

        return self::STAGE_EXAM;
    }

    /**
     * @param  array<string, mixed>  $payload
     */
    public function stageFromPlan(array $payload): string
    {
        if (! empty($payload['treatment_planned'])) {
            return self::STAGE_TREATMENT;
        }

        $plan = trim((string) ($payload['plan'] ?? ''));
        if ($plan !== '') {
            return self::STAGE_PLAN;
        }

        return self::STAGE_INVESTIGATION;
    }

    /**
     * @param  array<string, mixed>  $payload
     */
    public function shouldCompleteVisit(array $payload): bool
    {
        $outcome = strtolower((string) ($payload['outcome'] ?? ''));

        return str_contains($outcome, 'completed');
    }

    /**
     * @return array<string, array{next: string, label: string}>
     */
    public function stageFlow(): array
    {
        return [
            self::STAGE_CHECK_IN => ['next' => self::STAGE_HISTORY, 'label' => 'Start history'],
            self::STAGE_HISTORY => ['next' => self::STAGE_EXAM, 'label' => 'Move to exam / growth'],
            self::STAGE_EXAM => ['next' => self::STAGE_INVESTIGATION, 'label' => 'Move to investigations'],
            self::STAGE_INVESTIGATION => ['next' => self::STAGE_PLAN, 'label' => 'Move to diagnosis & plan'],
            self::STAGE_PLAN => ['next' => self::STAGE_TREATMENT, 'label' => 'Move to treatment'],
            self::STAGE_TREATMENT => ['next' => self::STAGE_COMPLETED, 'label' => 'Complete visit'],
            self::STAGE_COMPLETED => ['next' => self::STAGE_COMPLETED, 'label' => 'Completed'],
        ];
    }
}