jelq

use std::{

    sync::Arc,

    time::{Duration, Instant},

};

use axum::{

    Router,

    body::{Body, to_bytes},

    http::{Method, Request, StatusCode, Version, header},

    response::Response,

};

use serde_json::Value;

use tower::ServiceExt;

use crate::{

    AppState, Config,

    game::{

        Floor, MumVisit, Run, Story,

        balance::{STORAGE_GIRTH, SUSPICION_LIMIT},

    },

    router,

};

async fn send(

    app: &Router,

    method: &str,

    uri: &str,

    headers: &[(&str, &str)],

    body: &str,

) -> Response {

    let mut request = Request::builder()

        .method(Method::from_bytes(method.as_bytes()).unwrap())

        .uri(uri);

    for (name, value) in headers {

        request = request.header(*name, *value);

    }

    app.clone()

        .oneshot(request.body(Body::from(body.to_owned())).unwrap())

        .await

        .unwrap()

}

async fn json(response: Response) -> Value {

    serde_json::from_slice(&to_bytes(response.into_body(), usize::MAX).await.unwrap()).unwrap()

}

async fn text(response: Response) -> String {

    String::from_utf8(

        to_bytes(response.into_body(), usize::MAX)

            .await

            .unwrap()

            .to_vec(),

    )

    .unwrap()

}

fn header<'a>(response: &'a Response, name: &str) -> &'a str {

    response.headers()[name].to_str().unwrap()

}

fn fixture() -> Run {

    Run::new("cafebabe".into(), "cafe".into(), 69)

}

fn seeded_with(floor: Floor, setup: impl FnOnce(&mut Run)) -> Router {

    let state = Arc::new(AppState::default());

    let mut run = fixture();

    run.floor = floor;

    setup(&mut run);

    state.insert(run);

    router(state)

}

fn seeded(floor: Floor) -> (Router, String) {

    let state = Arc::new(AppState::default());

    let mut run = fixture();

    run.floor = floor;

    let etag = run.origin_etag();

    state.insert(run);

    (router(state), etag)

}

fn app() -> Router {

    crate::app(Config::default())

}

fn mum_at_the_door(run: &mut Run) {

    run.mum = Some(MumVisit {

        errand: "She heard a noise.",

        nonce: "abc".into(),

        strikes: 0,

        knocked: false,

    });

}

#[tokio::test]

async fn creating_a_run_returns_location_and_etag() {

    let response = send(&app(), "POST", "/runs?seed=1", &[], "").await;

    assert_eq!(response.status(), StatusCode::CREATED);

    assert!(header(&response, "location").starts_with("/runs/"));

    assert!(response.headers().contains_key(header::ETAG));

    assert_eq!(header(&response, "server"), "jelqd/0.6.9");

}

#[tokio::test]

async fn bodies_do_not_repeat_the_headers() {

    let (app, _) = seeded(Floor::Bedroom);

    let response = send(&app, "GET", "/runs/cafebabe", &[], "").await;

    assert!(response.headers().contains_key("x-eq"));

    let body = json(response).await;

    for stat in [

        "eq",

        "girth",

        "foreskin",

        "stamina",

        "cash",

        "self_reported_length",

    ] {

        assert!(body.get(stat).is_none(), "{stat} is already a header");

    }

}

#[tokio::test]

async fn jelqing_cold_fails_the_precondition() {

    let (app, _) = seeded(Floor::Bedroom);

    let response = send(&app, "JELQ", "/runs/cafebabe?technique=stretch", &[], "").await;

    assert_eq!(response.status(), StatusCode::PRECONDITION_FAILED);

    assert_eq!(

        header(&response, "content-type"),

        "application/problem+json"

    );

}

#[tokio::test]

async fn wet_without_lube_is_unsupported_media() {

    let (app, _) = seeded(Floor::Bedroom);

    send(&app, "JELQ", "/runs/cafebabe?technique=warmup", &[], "").await;

    let response = send(&app, "JELQ", "/runs/cafebabe?technique=wet", &[], "").await;

    assert_eq!(response.status(), StatusCode::UNSUPPORTED_MEDIA_TYPE);

    let images = send(

        &app,

        "JELQ",

        "/runs/cafebabe?technique=wet",

        &[("content-type", "image/jpeg")],

        "",

    )

    .await;

    assert_eq!(json(images).await["detail"], "No.");

}

#[tokio::test]

async fn unknown_methods_are_not_implemented() {

    let (app, _) = seeded(Floor::Bedroom);

    let response = send(&app, "MILK", "/runs/cafebabe", &[], "").await;

    assert_eq!(response.status(), StatusCode::NOT_IMPLEMENTED);

    let teapot = send(&app, "BREW", "/runs/cafebabe", &[], "").await;

    assert_eq!(teapot.status(), StatusCode::IM_A_TEAPOT);

}

#[tokio::test]

async fn the_clinic_does_not_advertise_jelq() {

    let (app, _) = seeded(Floor::Clinic);

    let options = send(&app, "OPTIONS", "/runs/cafebabe", &[], "").await;

    assert!(!header(&options, "allow").contains("JELQ"));

    let jelq = send(&app, "JELQ", "/runs/cafebabe?technique=wet", &[], "").await;

    assert_eq!(jelq.status(), StatusCode::METHOD_NOT_ALLOWED);

    assert!(!header(&jelq, "allow").contains("JELQ"));

    let (bedroom, _) = seeded(Floor::Bedroom);

    let options = send(&bedroom, "OPTIONS", "/runs/cafebabe", &[], "").await;

    assert!(header(&options, "allow").contains("JELQ"));

}

#[tokio::test]

async fn patch_only_accepts_the_truth() {

    let (app, _) = seeded(Floor::Bedroom);

    let patch = [("content-type", "application/json-patch+json")];

    let replace = send(

        &app,

        "PATCH",

        "/runs/cafebabe",

        &patch,

        r#"[{"op":"replace","path":"/length","value":12}]"#,

    )

    .await;

    assert_eq!(replace.status(), StatusCode::UNPROCESSABLE_ENTITY);

    let lie = send(

        &app,

        "PATCH",

        "/runs/cafebabe",

        &patch,

        r#"[{"op":"test","path":"/length","value":40}]"#,

    )

    .await;

    assert_eq!(

        json(lie).await["type"],

        "https://jelq.dev/problems/exaggeration"

    );

}

#[tokio::test]

async fn deleting_the_foreskin_requires_certainty() {

    let (app, _) = seeded(Floor::Bedroom);

    let response = send(&app, "DELETE", "/runs/cafebabe/foreskin", &[], "").await;

    assert_eq!(response.status(), StatusCode::PRECONDITION_REQUIRED);

    let look = send(&app, "GET", "/runs/cafebabe/foreskin", &[], "").await;

    assert_eq!(look.status(), StatusCode::OK);

    let etag = header(&look, "etag").to_owned();

    assert!(text(look).await.contains('~'));

    let response = send(

        &app,

        "DELETE",

        "/runs/cafebabe/foreskin",

        &[("if-match", &etag)],

        "",

    )

    .await;

    assert_eq!(response.status(), StatusCode::NO_CONTENT);

    let response = send(&app, "GET", "/runs/cafebabe/foreskin", &[], "").await;

    assert_eq!(response.status(), StatusCode::GONE);

}

#[tokio::test]

async fn length_is_chunked_until_the_ruler() {

    let (app, _) = seeded(Floor::Bedroom);

    let lie = send(&app, "GET", "/runs/cafebabe/length", &[], "").await;

    assert!(!lie.headers().contains_key(header::CONTENT_LENGTH));

    let app = seeded_with(Floor::Bedroom, |run| run.items.ruler = true);

    let truth = send(&app, "GET", "/runs/cafebabe/length", &[], "").await;

    let declared: usize = header(&truth, "content-length").parse().unwrap();

    let body = text(truth).await;

    assert_eq!(body.len(), declared);

    assert!(body.starts_with("8=") && body.ends_with('D'));

}

#[tokio::test]

async fn length_supports_inch_ranges() {

    let (app, _) = seeded(Floor::Bedroom);

    let response = send(

        &app,

        "GET",

        "/runs/cafebabe/length",

        &[("range", "inches=0-1")],

        "",

    )

    .await;

    assert_eq!(response.status(), StatusCode::PARTIAL_CONTENT);

    assert!(header(&response, "content-range").starts_with("inches 0.00-1.00/"));

    assert_eq!(text(response).await.len(), 25);

    let absurd = send(

        &app,

        "GET",

        "/runs/cafebabe/length",

        &[("range", "inches=40-50")],

        "",

    )

    .await;

    assert_eq!(absurd.status(), StatusCode::RANGE_NOT_SATISFIABLE);

    let malformed = send(

        &app,

        "GET",

        "/runs/cafebabe/length",

        &[("range", "inches=6.17")],

        "",

    )

    .await;

    assert_eq!(malformed.status(), StatusCode::OK);

    let suffix = send(

        &app,

        "GET",

        "/runs/cafebabe/length",

        &[("range", "inches=-1")],

        "",

    )

    .await;

    assert_eq!(suffix.status(), StatusCode::PARTIAL_CONTENT);

    assert!(text(suffix).await.ends_with('D'));

    for (range, status) in [

        ("inches=1-", StatusCode::PARTIAL_CONTENT),

        ("inches=40-", StatusCode::RANGE_NOT_SATISFIABLE),

        ("inches=-0", StatusCode::RANGE_NOT_SATISFIABLE),

        ("inches=2-1", StatusCode::OK),

        ("inches=NaN-2", StatusCode::OK),

        ("inches=0-inf", StatusCode::OK),

    ] {

        let response = send(

            &app,

            "GET",

            "/runs/cafebabe/length",

            &[("range", range)],

            "",

        )

        .await;

        assert_eq!(response.status(), status, "{range}");

    }

}

#[tokio::test]

async fn the_clinic_escalates_toward_honesty() {

    let (app, _) = seeded(Floor::Clinic);

    let mut hints = Vec::new();

    for _ in 0..3 {

        let response = send(

            &app,

            "GET",

            "/runs/cafebabe/diagnosis",

            &[("authorization", "Excuse \"changing\"")],

            "",

        )

        .await;

        assert_eq!(response.status(), StatusCode::UNAUTHORIZED);

        hints.push(header(&response, "www-authenticate").to_owned());

    }

    assert!(hints[2].contains("Be honest."));

}

#[tokio::test]

async fn the_true_ending_is_not_modified() {

    let (early, _) = seeded(Floor::Bedroom);

    let response = send(&early, "GET", "/runs/cafebabe/diagnosis", &[], "").await;

    assert_eq!(response.status(), StatusCode::TOO_EARLY);

    let (app, etag) = seeded(Floor::Clinic);

    assert!(

        !etag.contains("cafebabe"),

        "the origin ETag must not be derivable from the id"

    );

    let response = send(

        &app,

        "GET",

        "/runs/cafebabe/diagnosis",

        &[

            ("authorization", "Bearer honesty"),

            ("if-none-match", &etag),

        ],

        "",

    )

    .await;

    assert_eq!(response.status(), StatusCode::NOT_MODIFIED);

    assert_eq!(header(&response, "x-diagnosis"), "normal");

}

#[tokio::test]

async fn mum_accepts_a_known_excuse() {

    let app = seeded_with(Floor::Bedroom, mum_at_the_door);

    let response = send(&app, "GET", "/runs/cafebabe", &[], "").await;

    assert_eq!(response.status(), StatusCode::UNAUTHORIZED);

    assert_eq!(header(&response, "via"), "1.1 mum");

    let response = send(

        &app,

        "GET",

        "/runs/cafebabe",

        &[("authorization", "Excuse \"homework\"")],

        "",

    )

    .await;

    assert_eq!(response.status(), StatusCode::OK);

}

#[tokio::test]

async fn mum_runs_out_of_patience() {

    let app = seeded_with(Floor::Bedroom, mum_at_the_door);

    let lie = [("authorization", "Excuse \"it is for science\"")];

    for _ in 0..2 {

        let response = send(&app, "GET", "/runs/cafebabe", &lie, "").await;

        assert_eq!(response.status(), StatusCode::UNAUTHORIZED);

    }

    let caught = send(&app, "GET", "/runs/cafebabe", &lie, "").await;

    assert_eq!(caught.status(), StatusCode::GONE);

    assert_eq!(

        json(caught).await["type"],

        "https://jelq.dev/problems/caught"

    );

}

#[tokio::test]

async fn mum_remembers_excuses_by_christmas() {

    let app = seeded_with(Floor::Nans, |run| {

        mum_at_the_door(run);

        run.heard.push("homework".into());

    });

    let basic = send(

        &app,

        "GET",

        "/runs/cafebabe",

        &[("authorization", "Excuse \"changing\"")],

        "",

    )

    .await;

    assert_eq!(basic.status(), StatusCode::UNAUTHORIZED);

    assert!(header(&basic, "www-authenticate").starts_with("Digest"));

    let replay = send(

        &app,

        "GET",

        "/runs/cafebabe",

        &[(

            "authorization",

            "Digest username=\"homework\", nonce=\"abc\"",

        )],

        "",

    )

    .await;

    assert_eq!(

        json(replay).await["type"],

        "https://jelq.dev/problems/excuse-replayed"

    );

    let fresh = send(

        &app,

        "GET",

        "/runs/cafebabe",

        &[(

            "authorization",

            "Digest username=\"wrapping presents\", nonce=\"abc\"",

        )],

        "",

    )

    .await;

    assert_eq!(fresh.status(), StatusCode::OK);

}

#[tokio::test]

async fn a_pulled_muscle_is_a_true_excuse() {

    let app = seeded_with(Floor::Nans, |run| {

        mum_at_the_door(run);

        run.conditions.pulled_muscle = true;

        run.heard.push("pulled a muscle".into());

    });

    let response = send(

        &app,

        "GET",

        "/runs/cafebabe",

        &[(

            "authorization",

            "Digest username=\"pulled a muscle\", nonce=\"abc\"",

        )],

        "",

    )

    .await;

    assert_eq!(response.status(), StatusCode::OK);

    assert!(header(&response, "x-stamina").ends_with("/7"));

}

#[tokio::test]

async fn floors_end_when_they_end() {

    let app = seeded_with(Floor::Bedroom, |_| {});

    let early = send(&app, "POST", "/runs/cafebabe/floor", &[], "").await;

    assert_eq!(early.status(), StatusCode::TOO_EARLY);

    let app = seeded_with(Floor::Bedroom, |run| run.here.turns = 6);

    let next = send(&app, "POST", "/runs/cafebabe/floor", &[], "").await;

    assert!(

        next.status() == StatusCode::SEE_OTHER || next.status() == StatusCode::CONFLICT,

        "either the floor changed or the zip did"

    );

}

#[tokio::test]

async fn pump_bro_upsells() {

    let closed = seeded_with(Floor::Bedroom, |_| {});

    let response = send(&closed, "GET", "/runs/cafebabe/shop", &[], "").await;

    assert_eq!(response.status(), StatusCode::NOT_FOUND);

    let app = seeded_with(Floor::ChangingRoom, |_| {});

    let pump = send(&app, "POST", "/runs/cafebabe/shop/pump", &[], "").await;

    assert_eq!(pump.status(), StatusCode::UPGRADE_REQUIRED);

    assert_eq!(header(&pump, "upgrade"), "pump-pro/2.0");

    let pro = send(&app, "POST", "/runs/cafebabe/shop/pump-pro", &[], "").await;

    assert_eq!(pro.status(), StatusCode::CREATED);

    let broke = send(&app, "POST", "/runs/cafebabe/shop/extender", &[], "").await;

    assert_eq!(broke.status(), StatusCode::PAYMENT_REQUIRED);

}