jelq

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

use serde_json::json;

use super::{

    Cause, Floor, Item, Outcome, Run, Story,

    balance::{

        CAUGHT_MELONED, COCONUT_ODDS, ENERGY_ODDS, KITCHEN_ICE_ODDS, KITCHEN_TELLINGS,

        KITCHEN_TOLD, MELON_SHAME, RUPTURE_LINE, SUSPICION_LIMIT, WATERMELON,

    },

};

use crate::problem::Problem;

use rand::Rng;

impl Run {

    /// Raid the fridge. It costs the floor turn like anything else, and the household keeps count.

    pub fn raid_kitchen(&mut self) -> Result<Outcome, Problem> {

        if !self.floor.has_kitchen() {

            return Err(Problem::new(

                StatusCode::NOT_FOUND,

                "no-kitchen",

                "There is no kitchen on this floor.",

            ));

        }

        if self.kitchen_tellings >= KITCHEN_TELLINGS {

            return Err(Problem::new(

                StatusCode::GONE,

                "kitchen-off-limits",

                "The kitchen is off limits. You have been told twice.",

            ));

        }

        if self.story == Story::Pending {

            return Ok(self.branch_offer());

        }

        self.touch();

        self.turn += 1;

        self.here.turns += 1;

        self.realise();

        if self.chance(self.rupture_odds()) {

            return Err(self.rupture());

        }

        let was_below = self.suspicion < SUSPICION_LIMIT;

        // One draw of 120 keeps the marginals at the documented 1 in 3, 1 in 5 and 1 in 8.

        let draw = self.rng.random_range(0..120);

        let oil = 120 / COCONUT_ODDS;

        let energy = oil + 120 / ENERGY_ODDS;

        let ice = energy + 120 / KITCHEN_ICE_ODDS;

        let outcome = if self.eq > RUPTURE_LINE && self.rng.random_ratio(WATERMELON.0, WATERMELON.1)

        {

            self.here.melon_offered = true;

            self.melon_offer()

        } else if draw < oil {

            self.items.grant(Item::CoconutOil);

            Ok(Outcome::json(

                StatusCode::OK,

                json!({ "detail": "A jar of coconut oil, half of it already gone." }),

            ))

        } else if draw < energy {

            self.items.grant(Item::EnergyDrink);

            Ok(Outcome::json(

                StatusCode::OK,

                json!({ "detail": "An energy drink, cold and faintly alarming." }),

            ))

        } else if draw < ice {

            self.items.grant(Item::IcePack);

            Ok(Outcome::json(

                StatusCode::OK,

                json!({ "detail": "An ice pack, buried under the peas." }),

            ))

        } else {

            Ok(Outcome::empty(StatusCode::NO_CONTENT))

        };

        self.settle(was_below);

        let outcome = outcome?;

        if self.chance(KITCHEN_TOLD) {

            self.kitchen_tellings += 1;

            if self.kitchen_tellings >= KITCHEN_TELLINGS {

                return Err(Problem::new(

                    StatusCode::GONE,

                    "kitchen-off-limits",

                    "The kitchen is off limits. You have been told twice.",

                ));

            }

            self.notices.push(if self.floor == Floor::Nans {

                "Stop raiding the fridge."

            } else {

                "Stop raiding the kitchen."

            });

        }

        Ok(outcome)

    }

    fn melon_offer(&self) -> Result<Outcome, Problem> {

        let commit = self.path("/kitchen/watermelon");

        Ok(Outcome::json(

            StatusCode::MULTIPLE_CHOICES,

            json!({

                "detail": "The fridge holds a watermelon. It is exactly the right shape.",

                "choices": [

                    { "href": commit, "method": "POST", "title": "Commit to the watermelon" },

                    { "href": self.path(""), "method": "GET", "title": "Close the fridge" },

                ],

            }),

        )

        .header(

            header::LINK,

            format!(

                "<{commit}>; rel=\"alternate\"; title=\"Commit to the watermelon\"; allow=\"POST\""

            ),

        ))

    }

    /// The watermelon. It empties EQ, keeps the gains, costs shame, and the household may notice.

    pub fn commit_melon(&mut self) -> Result<Outcome, Problem> {

        if !self.here.melon_offered {

            return Err(Problem::new(

                StatusCode::CONFLICT,

                "no-melon-on-offer",

                "There is no watermelon on offer.",

            ));

        }

        self.here.melon_offered = false;

        self.reset_eq(0);

        self.shame += MELON_SHAME;

        self.touch();

        if self.chance(CAUGHT_MELONED) {

            return Err(self.end(Cause::Meloned));

        }

        Ok(Outcome::json(

            StatusCode::OK,

            json!({

                "detail": "The watermelon did not survive, and neither did the mood. EQ is gone, the gains are kept, and the shame is considerable.",

            }),

        ))

    }

}