jelq

pub mod balance;

mod body;

mod clinic;

mod floors;

mod forum;

mod hazards;

mod household;

mod items;

mod kitchen;

mod outcome;

mod shop;

mod state;

mod techniques;

mod turn;

mod utility;

pub use body::PatchOp;

pub use forum::{ELDER_LAST_MODIFIED, FORUM_THREAD, forum_thread};

pub use items::{Item, Items, SHOP};

pub use outcome::{Outcome, Payload};

pub use state::{

    Cause, Clinic, Conditions, Elder, Ending, Floor, FloorState, Grip, MumVisit, Observer, Story,

};

pub use techniques::{LUBRICANTS, Lube, Technique};

use std::{

    hash::{DefaultHasher, Hash, Hasher},

    time::{Duration, Instant, SystemTime},

};

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

use rand::{Rng, SeedableRng, rngs::StdRng};

use serde_json::{Value, json};

use crate::problem::Problem;

use balance::{

    HONEST_TOLERANCE, MAX_STAMINA, PULLED_STAMINA, SELF_REPORT_INFLATION, STARTING_CASH,

    STARTING_EQ,

};

use state::Pending;

pub struct Run {

    /// The private id in the run's URL. Knowing it is the only authorisation a run has.

    pub id: String,

    pub public_id: String,

    etag_salt: u64,

    pub version: u64,

    pub turn: u32,

    pub floor: Floor,

    pub here: FloorState,

    pub length: f64,

    pub girth: f64,

    pub foreskin: Option<f64>,

    pub eq: i32,

    pub stamina: i32,

    pub suspicion: u8,

    pub shame: u32,

    pub shame_reported: u32,

    pub shame_total: u32,

    pub cash: u32,

    pub grip: Grip,

    pub items: Items,

    pub conditions: Conditions,

    pub warmed_up: bool,

    pub put_applied: bool,

    pub story: Story,

    pub meditation_taught: bool,

    pub kitchen_tellings: u8,

    pub mum: Option<MumVisit>,

    pub heard: Vec<String>,

    pub elder: Elder,

    pub clinic: Clinic,

    pub seeded: bool,

    pub filmed_at: Option<SystemTime>,

    pub observed: Option<Observer>,

    pub refractory: Duration,

    pub refractory_until: Option<Instant>,

    pub pending: Vec<Pending>,

    pub notices: Vec<&'static str>,

    pub ended: Option<Ending>,

    pub last_seen: Instant,

    rng: StdRng,

}

pub fn round2(x: f64) -> f64 {

    (x * 100.0).round() / 100.0

}

fn round3(x: f64) -> f64 {

    (x * 1000.0).round() / 1000.0

}

fn honest(claimed: f64, actual: f64) -> bool {

    (claimed - actual).abs() < HONEST_TOLERANCE

}

fn not_extended() -> Problem {

    Problem::new(

        StatusCode::NOT_EXTENDED,

        "not-extended",

        "The requested extension was not applied. The server has recorded the attempt.",

    )

}

pub fn parse_credentials<'a>(authorization: &'a str, scheme: &str) -> Option<&'a str> {

    let (given, rest) = authorization.trim().split_once(' ')?;

    given

        .eq_ignore_ascii_case(scheme)

        .then(|| rest.trim().trim_matches('"'))

}

pub fn etag_matches(condition: &str, etag: &str) -> bool {

    condition

        .split(',')

        .map(str::trim)

        .any(|tag| tag == "*" || tag.trim_start_matches("W/") == etag)

}

pub fn shaft(mm: usize) -> String {

    format!("8{}D", "=".repeat(mm.saturating_sub(2)))

}

impl Run {

    pub fn new(id: String, public_id: String, seed: u64) -> Self {

        let mut rng = StdRng::seed_from_u64(seed);

        let length = round2(rng.random_range(4.5..6.5));

        let girth = round2(rng.random_range(4.0..5.0));

        let foreskin = round2(rng.random_range(0.4..2.4));

        let etag_salt = rng.random();

        Self {

            id,

            public_id,

            etag_salt,

            version: 0,

            turn: 0,

            floor: Floor::Bedroom,

            here: FloorState::default(),

            length,

            girth,

            foreskin: Some(foreskin),

            eq: STARTING_EQ,

            stamina: MAX_STAMINA,

            suspicion: 0,

            shame: 0,

            shame_reported: 0,

            shame_total: 0,

            cash: STARTING_CASH,

            grip: Grip::Overhand,

            items: Items::default(),

            conditions: Conditions::default(),

            warmed_up: false,

            put_applied: false,

            story: Story::Unopened,

            meditation_taught: false,

            kitchen_tellings: 0,

            mum: None,

            heard: Vec::new(),

            notices: Vec::new(),

            elder: Elder::Unverified { strikes: 0 },

            clinic: Clinic::default(),

            seeded: false,

            filmed_at: None,

            observed: None,

            refractory: Duration::from_secs(30),

            refractory_until: None,

            pending: Vec::new(),

            ended: None,

            last_seen: Instant::now(),

            rng,

        }

    }

    fn etag_at(&self, version: u64) -> String {

        let mut hasher = DefaultHasher::new();

        (self.etag_salt, version).hash(&mut hasher);

        format!("\"{:016x}\"", hasher.finish())

    }

    pub fn etag(&self) -> String {

        self.etag_at(self.version)

    }

    /// The ETag the run had when it was created, which the true ending asks for.

    pub fn origin_etag(&self) -> String {

        self.etag_at(0)

    }

    pub fn touch(&mut self) {

        self.version += 1;

    }

    fn fresh_nonce(&mut self) -> String {

        format!("{:016x}", self.rng.random::<u64>())

    }

    fn chance(&mut self, odds: u32) -> bool {

        self.rng.random_ratio(1, odds)

    }

    pub fn path(&self, sub: &str) -> String {

        format!("/runs/{}{sub}", self.id)

    }

    pub fn current_length(&self) -> f64 {

        self.length + self.here.temp_length

    }

    pub fn self_reported(&self) -> f64 {

        self.current_length() + SELF_REPORT_INFLATION

    }

    pub fn max_stamina(&self) -> i32 {

        if self.conditions.pulled_muscle {

            PULLED_STAMINA

        } else {

            MAX_STAMINA

        }

    }

    pub fn forum_read(&self) -> bool {

        self.story == Story::ReadForum

    }

    pub fn footage(&self) -> Option<(SystemTime, u64)> {

        let filmed = self.filmed_at?;

        let age = filmed.elapsed().map_or(0, |d| d.as_secs());

        Some((filmed, age))

    }

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

        match (&self.ended, self.clinic.cured) {

            (Some(_), _) => "gone",

            (None, true) => "normal",

            (None, false) => "alive",

        }

    }

    /// The parts of the run that its response headers do not already carry.

    pub fn view(&self) -> Value {

        let conditions: Vec<&str> = [

            (self.here.teapot, "short and stout"),

            (self.conditions.peyronie, "curved"),

            (self.items.cage, "caged"),

            (self.conditions.priapism_turns > 0, "priapism"),

            (self.conditions.pulled_muscle, "pulled muscle"),

            (self.filmed_at.is_some(), "viral"),

        ]

        .into_iter()

        .filter_map(|(active, name)| active.then_some(name))

        .collect();

        json!({

            "public_id": self.public_id,

            "status": self.status(),

            "turn": self.turn,

            "run_shame": self.shame,

            "grip": self.grip.name(),

            "warmed_up": self.warmed_up,

            "ranked": !self.seeded,

            "conditions": conditions,

        })

    }

    pub fn end(&mut self, cause: Cause) -> Problem {

        self.ended = Some(Ending {

            cause,

            turn: self.turn,

        });

        cause.problem().with("turn", self.turn)

    }

    /// Ends the refractory period once it has run out, whatever the request.

    pub fn recover(&mut self, now: Instant) {

        if self.refractory_until.is_some_and(|until| now >= until) {

            self.refractory_until = None;

            self.stamina = self.max_stamina();

            self.eq = STARTING_EQ;

            self.warmed_up = false;

            self.touch();

        }

    }

    fn check_refractory(&mut self, now: Instant) -> Result<(), Problem> {

        self.recover(now);

        let Some(until) = self.refractory_until else {

            return Ok(());

        };

        let seconds = until.saturating_duration_since(now).as_secs() + 1;

        Err(Problem::new(

            StatusCode::TOO_MANY_REQUESTS,

            "overtrained",

            "Stamina is exhausted. The refractory period is in effect.",

        )

        .header(header::RETRY_AFTER, seconds.to_string()))

    }

    fn exhaust(&mut self, now: Instant) {

        self.stamina = 0;

        self.refractory_until = Some(now + self.refractory);

    }

    fn realise(&mut self) {

        let (due, waiting): (Vec<_>, Vec<_>) = std::mem::take(&mut self.pending)

            .into_iter()

            .partition(|p| p.due <= self.turn);

        self.pending = waiting;

        for p in due {

            self.length += p.length;

            self.girth += p.girth;

        }

    }

    fn gain(&mut self, length: f64, girth: f64, foreskin: f64) {

        self.length += length;

        self.girth += girth;

        if let Some(f) = &mut self.foreskin {

            *f += foreskin;

        }

    }

    fn rupture(&mut self) -> Problem {

        let cause = Cause::Ruptured {

            ice_unused: self.items.ice_packs > 0,

        };

        self.end(cause)

    }

}