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, KETTLE_ODDS, KITCHEN_ICE_ODDS, KITCHEN_TELLINGS,

        KITCHEN_TOLD, MELON_SHAME, RUPTURE_LINE, SUSPICION_LIMIT, WATERMELON,

    },

};

use crate::problem::Problem;

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.decision("rupture", "Does this raid of the fridge rupture something?"),

            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, 1 in 8 and 1 in 6.

        let loot = self.oracle.pick(

            self.decision("loot", "What does the fridge give up on this raid?"),

            &[

                120 / COCONUT_ODDS,

                120 / ENERGY_ODDS,

                120 / KITCHEN_ICE_ODDS,

                120 / KETTLE_ODDS,

                120 - (120 / COCONUT_ODDS

                    + 120 / ENERGY_ODDS

                    + 120 / KITCHEN_ICE_ODDS

                    + 120 / KETTLE_ODDS),

            ],

            &[

                "A jar of coconut oil, half of it already gone.",

                "An energy drink, cold and faintly alarming.",

                "An ice pack, buried under the peas.",

                "The kettle has just boiled.",

                "Nothing. The fridge is ordinary.",

            ],

        );

        let outcome = if self.eq > RUPTURE_LINE

            && self.oracle.ratio(

                self.decision(

                    "watermelon",

                    "Is there a watermelon in the fridge, exactly the right shape?",

                ),

                WATERMELON.0,

                WATERMELON.1,

            ) {

            self.here.melon_offered = true;

            self.melon_offer()

        } else if loot == 0 {

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

            Ok(Outcome::json(

                StatusCode::OK,

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

            ))

        } else if loot == 1 {

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

            Ok(Outcome::json(

                StatusCode::OK,

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

            ))

        } else if loot == 2 {

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

            Ok(Outcome::json(

                StatusCode::OK,

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

            ))

        } else if loot == 3 {

            self.here.kettle_boiled = true;

            Ok(Outcome::json(

                StatusCode::OK,

                json!({ "detail": "The kettle has just boiled. The towel can be warmed once more." }),

            ))

        } else {

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

        };

        self.settle(was_below);

        let outcome = outcome?;

        if self.chance(

            self.decision(

                "kitchen-told",

                "Has the household had enough of the raiding?",

            ),

            KITCHEN_TOLD,

        ) {

            self.kitchen_tellings += 1;

            if self.kitchen_tellings >= KITCHEN_TELLINGS {

                self.end_kitchen_visit();

                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(

            self.decision(

                "caught-meloned",

                "Is Mum standing in the doorway as you look up?",

            ),

            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.",

            }),

        ))

    }

}