<?php

namespace App\Http\Controllers\Care;

use App\Http\Controllers\Care\Concerns\ScopesToAccount;
use App\Http\Controllers\Controller;
use App\Models\SpecialtyClinicalRecord;
use App\Models\Visit;
use App\Services\Care\Pediatrics\PediatricsAnalyticsService;
use App\Services\Care\Pediatrics\PediatricsWorkflowService;
use App\Services\Care\SpecialtyClinicalRecordService;
use App\Services\Care\SpecialtyModuleService;
use App\Services\Care\SpecialtyShellService;
use App\Services\Care\SpecialtyVisitStageService;
use Illuminate\Http\RedirectResponse;
use Illuminate\Http\Request;
use Illuminate\View\View;

class PediatricsWorkspaceController extends Controller
{
    use ScopesToAccount;

    protected function assertPediatricsAccess(Request $request, SpecialtyModuleService $modules): void
    {
        $organization = $this->organization($request);
        abort_unless($modules->memberCanAccess($organization, $this->member($request), 'pediatrics'), 403);
    }

    protected function assertPediatricsManage(Request $request, SpecialtyModuleService $modules): void
    {
        $organization = $this->organization($request);
        abort_unless($modules->memberCanManage($organization, $this->member($request), 'pediatrics'), 403);
    }

    protected function assertVisit(Request $request, Visit $visit): void
    {
        abort_unless($visit->organization_id === $this->organization($request)->id, 404);
        $this->authorizeBranch($request, (int) $visit->branch_id);
    }

    public function setStage(
        Request $request,
        Visit $visit,
        SpecialtyModuleService $modules,
        SpecialtyVisitStageService $stages,
        SpecialtyShellService $shell,
    ): RedirectResponse {
        $this->authorizeAbility($request, 'consultations.manage');
        $this->assertPediatricsManage($request, $modules);
        $this->assertVisit($request, $visit);

        $validated = $request->validate([
            'stage' => ['required', 'string', 'max:32'],
        ]);

        try {
            $stages->setStage(
                $this->organization($request),
                $visit,
                'pediatrics',
                $validated['stage'],
                $this->ownerRef($request),
                $this->actorRef($request),
            );
        } catch (\InvalidArgumentException $e) {
            return back()->with('error', $e->getMessage());
        }

        return redirect()
            ->route('care.specialty.workspace', [
                'module' => 'pediatrics',
                'visit' => $visit,
                'tab' => $shell->workspaceTabForStage('pediatrics', $validated['stage']),
            ])
            ->with('success', 'Visit stage updated.');
    }

    public function saveTreatment(
        Request $request,
        Visit $visit,
        SpecialtyModuleService $modules,
        SpecialtyClinicalRecordService $clinical,
        SpecialtyVisitStageService $stages,
        PediatricsWorkflowService $workflow,
    ): RedirectResponse {
        $this->authorizeAbility($request, 'consultations.manage');
        $this->assertPediatricsManage($request, $modules);
        $this->assertVisit($request, $visit);

        $organization = $this->organization($request);
        $owner = $this->ownerRef($request);

        $payload = (array) $request->input('payload', []);
        foreach ($clinical->fieldsFor('pediatrics', 'ped_treatment') as $field) {
            if (($field['type'] ?? '') === 'boolean') {
                $payload[$field['name']] = $request->boolean('payload.'.$field['name']);
            }
        }
        $request->merge(['payload' => $payload, 'tab' => 'treatment']);

        $validated = $request->validate(array_merge([
            'tab' => ['required', 'string'],
        ], $clinical->validationRules('pediatrics', 'ped_treatment')));

        $record = $clinical->upsert(
            $organization,
            $visit,
            'pediatrics',
            'ped_treatment',
            $clinical->payloadFromRequest($validated),
            $owner,
            $owner,
            PediatricsWorkflowService::STAGE_TREATMENT,
            SpecialtyClinicalRecord::STATUS_COMPLETED,
        );

        try {
            $stages->setStage(
                $organization,
                $visit,
                'pediatrics',
                PediatricsWorkflowService::STAGE_TREATMENT,
                $owner,
                $owner,
            );
        } catch (\InvalidArgumentException) {
        }

        if ($workflow->shouldCompleteVisit($record->payload ?? [])) {
            try {
                $stages->setStage(
                    $organization,
                    $visit,
                    'pediatrics',
                    PediatricsWorkflowService::STAGE_COMPLETED,
                    $owner,
                    $owner,
                );
            } catch (\InvalidArgumentException) {
            }

            $visit->refresh();
            if ($visit->status !== Visit::STATUS_COMPLETED) {
                $visit->update([
                    'status' => Visit::STATUS_COMPLETED,
                    'completed_at' => $visit->completed_at ?? now(),
                ]);
            }

            $appointment = $visit->appointment;
            if ($appointment && $appointment->status !== \App\Models\Appointment::STATUS_COMPLETED) {
                $appointment->update([
                    'status' => \App\Models\Appointment::STATUS_COMPLETED,
                    'completed_at' => $appointment->completed_at ?? now(),
                ]);
            }
        }

        return redirect()
            ->route('care.specialty.workspace', ['module' => 'pediatrics', 'visit' => $visit, 'tab' => 'treatment'])
            ->with('success', 'Treatment / immunization saved.');
    }

    public function reports(
        Request $request,
        SpecialtyModuleService $modules,
        SpecialtyShellService $shell,
        PediatricsAnalyticsService $analytics,
    ): View {
        $this->authorizeAbility($request, 'consultations.view');
        $this->assertPediatricsAccess($request, $modules);

        $organization = $this->organization($request);
        $branchScope = app(\App\Services\Care\OrganizationResolver::class)->branchScope($this->member($request));

        $report = $analytics->report(
            $organization,
            $this->ownerRef($request),
            $branchScope,
            $request->query('from'),
            $request->query('to'),
        );

        return view('care.specialty.pediatrics.reports', [
            'moduleKey' => 'pediatrics',
            'definition' => $modules->definition('pediatrics'),
            'shellNav' => $shell->navItems('pediatrics'),
            'report' => $report,
            'currency' => strtoupper((string) config('care.billing.currency', config('care.currency', 'GHS'))),
        ]);
    }

    public function printSummary(
        Request $request,
        Visit $visit,
        SpecialtyModuleService $modules,
        SpecialtyClinicalRecordService $clinical,
    ): View {
        $this->authorizeAbility($request, 'consultations.view');
        $this->assertPediatricsAccess($request, $modules);
        $this->assertVisit($request, $visit);

        $visit->loadMissing(['patient', 'appointment.practitioner', 'branch']);

        return view('care.specialty.pediatrics.print', [
            'visit' => $visit,
            'patient' => $visit->patient,
            'history' => $clinical->findForVisit($visit, 'pediatrics', 'ped_history'),
            'exam' => $clinical->findForVisit($visit, 'pediatrics', 'pediatric_exam'),
            'investigation' => $clinical->findForVisit($visit, 'pediatrics', 'ped_investigation'),
            'plan' => $clinical->findForVisit($visit, 'pediatrics', 'ped_plan'),
            'treatment' => $clinical->findForVisit($visit, 'pediatrics', 'ped_treatment'),
        ]);
    }
}
