jelq

public

an http based roguelike rpg about jelqing

Star0

Watch0

24 commits 1 branch 0 tags 0 releases1.3 MiB

pub mod balance;
mod body;
mod clinic;
mod floors;
mod forum;
mod hazards;
mod household;
mod items;
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};
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 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 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,
            mum: None,
            heard: 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)
    }
}

1	pub mod balance;
2	mod body;
3	mod clinic;
4	mod floors;
5	mod forum;
6	mod hazards;
7	mod household;
8	mod items;
9	mod outcome;
10	mod shop;
11	mod state;
12	mod techniques;
13	mod turn;
14	mod utility;
15
16	pub use body::PatchOp;
17	pub use forum::{ELDER_LAST_MODIFIED, FORUM_THREAD, forum_thread};
18	pub use items::{Item, Items};
19	pub use outcome::{Outcome, Payload};
20	pub use state::{
21	Cause, Clinic, Conditions, Elder, Ending, Floor, FloorState, Grip, MumVisit, Observer, Story,
22	};
23	pub use techniques::{LUBRICANTS, Lube, Technique};
24
25	use std::{
26	hash::{DefaultHasher, Hash, Hasher},
27	time::{Duration, Instant, SystemTime},
28	};
29
30	use axum::http::{StatusCode, header};
31	use rand::{Rng, SeedableRng, rngs::StdRng};
32	use serde_json::{Value, json};
33
34	use crate::problem::Problem;
35	use balance::{
36	HONEST_TOLERANCE, MAX_STAMINA, PULLED_STAMINA, SELF_REPORT_INFLATION, STARTING_CASH,
37	STARTING_EQ,
38	};
39	use state::Pending;
40
41	pub struct Run {
42	/// The private id in the run's URL. Knowing it is the only authorisation a run has.
43	pub id: String,
44	pub public_id: String,
45	etag_salt: u64,
46	pub version: u64,
47	pub turn: u32,
48	pub floor: Floor,
49	pub here: FloorState,
50	pub length: f64,
51	pub girth: f64,
52	pub foreskin: Option<f64>,
53	pub eq: i32,
54	pub stamina: i32,
55	pub suspicion: u8,
56	pub shame: u32,
57	pub shame_reported: u32,
58	pub shame_total: u32,
59	pub cash: u32,
60	pub grip: Grip,
61	pub items: Items,
62	pub conditions: Conditions,
63	pub warmed_up: bool,
64	pub put_applied: bool,
65	pub story: Story,
66	pub mum: Option<MumVisit>,
67	pub heard: Vec<String>,
68	pub elder: Elder,
69	pub clinic: Clinic,
70	pub seeded: bool,
71	pub filmed_at: Option<SystemTime>,
72	pub observed: Option<Observer>,
73	pub refractory: Duration,
74	pub refractory_until: Option<Instant>,
75	pub pending: Vec<Pending>,
76	pub ended: Option<Ending>,
77	pub last_seen: Instant,
78	rng: StdRng,
79	}
80
81	pub fn round2(x: f64) -> f64 {
82	(x * 100.0).round() / 100.0
83	}
84
85	fn round3(x: f64) -> f64 {
86	(x * 1000.0).round() / 1000.0
87	}
88
89	fn honest(claimed: f64, actual: f64) -> bool {
90	(claimed - actual).abs() < HONEST_TOLERANCE
91	}
92
93	fn not_extended() -> Problem {
94	Problem::new(
95	StatusCode::NOT_EXTENDED,
96	"not-extended",
97	"The requested extension was not applied. The server has recorded the attempt.",
98	)
99	}
100
101	pub fn parse_credentials<'a>(authorization: &'a str, scheme: &str) -> Option<&'a str> {
102	let (given, rest) = authorization.trim().split_once(' ')?;
103	given
104	.eq_ignore_ascii_case(scheme)
105	.then(\|\| rest.trim().trim_matches('"'))
106	}
107
108	pub fn etag_matches(condition: &str, etag: &str) -> bool {
109	condition
110	.split(',')
111	.map(str::trim)
112	.any(\|tag\| tag == "*" \|\| tag.trim_start_matches("W/") == etag)
113	}
114
115	pub fn shaft(mm: usize) -> String {
116	format!("8{}D", "=".repeat(mm.saturating_sub(2)))
117	}
118
119	impl Run {
120	pub fn new(id: String, public_id: String, seed: u64) -> Self {
121	let mut rng = StdRng::seed_from_u64(seed);
122	let length = round2(rng.random_range(4.5..6.5));
123	let girth = round2(rng.random_range(4.0..5.0));
124	let foreskin = round2(rng.random_range(0.4..2.4));
125	let etag_salt = rng.random();
126	Self {
127	id,
128	public_id,
129	etag_salt,
130	version: 0,
131	turn: 0,
132	floor: Floor::Bedroom,
133	here: FloorState::default(),
134	length,
135	girth,
136	foreskin: Some(foreskin),
137	eq: STARTING_EQ,
138	stamina: MAX_STAMINA,
139	suspicion: 0,
140	shame: 0,
141	shame_reported: 0,
142	shame_total: 0,
143	cash: STARTING_CASH,
144	grip: Grip::Overhand,
145	items: Items::default(),
146	conditions: Conditions::default(),
147	warmed_up: false,
148	put_applied: false,
149	story: Story::Unopened,
150	mum: None,
151	heard: Vec::new(),
152	elder: Elder::Unverified { strikes: 0 },
153	clinic: Clinic::default(),
154	seeded: false,
155	filmed_at: None,
156	observed: None,
157	refractory: Duration::from_secs(30),
158	refractory_until: None,
159	pending: Vec::new(),
160	ended: None,
161	last_seen: Instant::now(),
162	rng,
163	}
164	}
165
166	fn etag_at(&self, version: u64) -> String {
167	let mut hasher = DefaultHasher::new();
168	(self.etag_salt, version).hash(&mut hasher);
169	format!("\"{:016x}\"", hasher.finish())
170	}
171
172	pub fn etag(&self) -> String {
173	self.etag_at(self.version)
174	}
175
176	/// The ETag the run had when it was created, which the true ending asks for.
177	pub fn origin_etag(&self) -> String {
178	self.etag_at(0)
179	}
180
181	pub fn touch(&mut self) {
182	self.version += 1;
183	}
184
185	fn fresh_nonce(&mut self) -> String {
186	format!("{:016x}", self.rng.random::<u64>())
187	}
188
189	fn chance(&mut self, odds: u32) -> bool {
190	self.rng.random_ratio(1, odds)
191	}
192
193	pub fn path(&self, sub: &str) -> String {
194	format!("/runs/{}{sub}", self.id)
195	}
196
197	pub fn current_length(&self) -> f64 {
198	self.length + self.here.temp_length
199	}
200
201	pub fn self_reported(&self) -> f64 {
202	self.current_length() + SELF_REPORT_INFLATION
203	}
204
205	pub fn max_stamina(&self) -> i32 {
206	if self.conditions.pulled_muscle {
207	PULLED_STAMINA
208	} else {
209	MAX_STAMINA
210	}
211	}
212
213	pub fn forum_read(&self) -> bool {
214	self.story == Story::ReadForum
215	}
216
217	pub fn footage(&self) -> Option<(SystemTime, u64)> {
218	let filmed = self.filmed_at?;
219	let age = filmed.elapsed().map_or(0, \|d\| d.as_secs());
220	Some((filmed, age))
221	}
222
223	pub fn status(&self) -> &'static str {
224	match (&self.ended, self.clinic.cured) {
225	(Some(_), _) => "gone",
226	(None, true) => "normal",
227	(None, false) => "alive",
228	}
229	}
230
231	/// The parts of the run that its response headers do not already carry.
232	pub fn view(&self) -> Value {
233	let conditions: Vec<&str> = [
234	(self.here.teapot, "short and stout"),
235	(self.conditions.peyronie, "curved"),
236	(self.items.cage, "caged"),
237	(self.conditions.priapism_turns > 0, "priapism"),
238	(self.conditions.pulled_muscle, "pulled muscle"),
239	(self.filmed_at.is_some(), "viral"),
240	]
241	.into_iter()
242	.filter_map(\|(active, name)\| active.then_some(name))
243	.collect();
244	json!({
245	"public_id": self.public_id,
246	"status": self.status(),
247	"turn": self.turn,
248	"run_shame": self.shame,
249	"grip": self.grip.name(),
250	"warmed_up": self.warmed_up,
251	"ranked": !self.seeded,
252	"conditions": conditions,
253	})
254	}
255
256	pub fn end(&mut self, cause: Cause) -> Problem {
257	self.ended = Some(Ending {
258	cause,
259	turn: self.turn,
260	});
261	cause.problem().with("turn", self.turn)
262	}
263
264	/// Ends the refractory period once it has run out, whatever the request.
265	pub fn recover(&mut self, now: Instant) {
266	if self.refractory_until.is_some_and(\|until\| now >= until) {
267	self.refractory_until = None;
268	self.stamina = self.max_stamina();
269	self.eq = STARTING_EQ;
270	self.warmed_up = false;
271	self.touch();
272	}
273	}
274
275	fn check_refractory(&mut self, now: Instant) -> Result<(), Problem> {
276	self.recover(now);
277	let Some(until) = self.refractory_until else {
278	return Ok(());
279	};
280	let seconds = until.saturating_duration_since(now).as_secs() + 1;
281	Err(Problem::new(
282	StatusCode::TOO_MANY_REQUESTS,
283	"overtrained",
284	"Stamina is exhausted. The refractory period is in effect.",
285	)
286	.header(header::RETRY_AFTER, seconds.to_string()))
287	}
288
289	fn exhaust(&mut self, now: Instant) {
290	self.stamina = 0;
291	self.refractory_until = Some(now + self.refractory);
292	}
293
294	fn realise(&mut self) {
295	let (due, waiting): (Vec<_>, Vec<_>) = std::mem::take(&mut self.pending)
296	.into_iter()
297	.partition(\|p\| p.due <= self.turn);
298	self.pending = waiting;
299	for p in due {
300	self.length += p.length;
301	self.girth += p.girth;
302	}
303	}
304
305	fn gain(&mut self, length: f64, girth: f64, foreskin: f64) {
306	self.length += length;
307	self.girth += girth;
308	if let Some(f) = &mut self.foreskin {
309	*f += foreskin;
310	}
311	}
312
313	fn rupture(&mut self) -> Problem {
314	let cause = Cause::Ruptured {
315	ice_unused: self.items.ice_packs > 0,
316	};
317	self.end(cause)
318	}
319	}