jelq

use axum::http::{StatusCode, header};

use serde_json::json;

use super::{Floor, Outcome, Run, etag_matches, parse_credentials, round2};

use crate::{headers::X_DIAGNOSIS, problem::Problem};

const INTAKE: [&str; 3] = [

    "The clinic does not accept excuses.",

    "The clinic accepts one token. You have not been carrying it.",

    "Be honest.",

];

impl Run {

    pub fn diagnose(

        &mut self,

        authorization: Option<&str>,

        if_none_match: Option<&str>,

    ) -> Result<Outcome, Problem> {

        if self.floor != Floor::Clinic {

            let floors = Floor::Clinic.number() - self.floor.number();

            return Err(Problem::new(

                StatusCode::TOO_EARLY,

                "no-appointment",

                format!("The clinic's next available appointment is {floors} floors away."),

            ));

        }

        let intake = Problem::new(

            StatusCode::UNAUTHORIZED,

            "intake",

            "The clinic requires authorisation.",

        );

        let Some(authorization) = authorization else {

            return Err(intake.header(header::WWW_AUTHENTICATE, "Bearer realm=\"clinic\""));

        };

        let truthful = parse_credentials(authorization, "bearer")

            .is_some_and(|token| token.eq_ignore_ascii_case("honesty"));

        if !truthful {

            let hint = INTAKE[self.clinic.intake_attempts.min(INTAKE.len() - 1)];

            self.clinic.intake_attempts += 1;

            return Err(intake.header(

                header::WWW_AUTHENTICATE,

                format!(

                    "Bearer realm=\"clinic\", error=\"invalid_token\", error_description=\"{hint}\""

                ),

            ));

        }

        self.conditions.peyronie = false;

        self.conditions.priapism_turns = 0;

        let etag = self.origin_etag();

        if if_none_match.is_some_and(|inm| etag_matches(inm, &etag)) {

            self.clinic.cured = true;

            return Ok(Outcome::empty(StatusCode::NOT_MODIFIED)

                .header(header::ETAG, etag)

                .header(X_DIAGNOSIS, "normal"));

        }

        if !self.clinic.measured {

            self.clinic.measured = true;

            if !self.items.ruler {

                self.shame += 15;

            }

        }

        Ok(Outcome::json(

            StatusCode::OK,

            json!({

                "diagnosis": "within normal range",

                "measured_length": round2(self.length),

                "prescription": "Please stop.",

            }),

        )

        .header(header::ETAG, etag))

    }

}