jelq

use super::{

    Run,

    balance::{DANGER_EQ, EQ_BAND, HOT_EQ, PREMATURE_EQ, RING_RUPTURE_LINE, RUPTURE_LINE},

};

/// What `X-EQ` shows in place of the number, cut on the lines the rolls already use.

#[derive(Clone, Copy, Debug, PartialEq, Eq)]

pub enum Arousal {

    Flaccid,

    Cold,

    Warm,

    Ideal,

    Swollen,

    Throbbing,

    Straining,

    Imminent,

    Priapic,

}

impl Arousal {

    pub fn name(self) -> &'static str {

        match self {

            Self::Flaccid => "Flaccid",

            Self::Cold => "Cold",

            Self::Warm => "Warm",

            Self::Ideal => "Ideal",

            Self::Swollen => "Swollen",

            Self::Throbbing => "Throbbing",

            Self::Straining => "Straining",

            Self::Imminent => "Imminent",

            Self::Priapic => "Priapic",

        }

    }

}

impl Run {

    pub(super) fn rupture_line(&self) -> i32 {

        if self.items.kit.ring {

            RING_RUPTURE_LINE

        } else {

            RUPTURE_LINE

        }

    }

    pub fn arousal(&self) -> Arousal {

        if self.body.conditions.priapism_turns > 0 {

            Arousal::Priapic

        } else if self.hotter_than(PREMATURE_EQ) {

            Arousal::Imminent

        } else if self.hotter_than(self.rupture_line()) {

            Arousal::Straining

        } else if self.hotter_than(HOT_EQ) {

            Arousal::Throbbing

        } else if self.hotter_than(DANGER_EQ) {

            Arousal::Swollen

        } else if EQ_BAND.contains(&self.body.eq) {

            Arousal::Ideal

        } else if self.body.eq == 0 {

            Arousal::Flaccid

        } else if self.cold() {

            Arousal::Cold

        } else {

            Arousal::Warm

        }

    }

}