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;

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

        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 - (120 / COCONUT_ODDS + 120 / ENERGY_ODDS + 120 / KITCHEN_ICE_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.",

                "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 {

            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 {

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

            }),

        ))

    }

}