jelq

use super::balance::STARTING_ICE_PACKS;

#[derive(Clone, Copy, PartialEq, Eq, Debug)]

pub enum Item {

    WarmTowel,

    IcePack,

    Ruler,

    DoorLock,

    DndSign,

    Blinds,

    Pump,

    PumpPro,

    Extender,

    Ring,

    Supplement,

    Cage,

    CageKey,

    Spreadsheet,

    TeapotSpout,

    BabyOil,

    CoconutOil,

    Lotion,

    Vaseline,

    EnergyDrink,

}

pub const SHOP: [Item; 15] = [

    Item::Pump,

    Item::PumpPro,

    Item::Extender,

    Item::Ring,

    Item::Supplement,

    Item::DoorLock,

    Item::DndSign,

    Item::Blinds,

    Item::Ruler,

    Item::IcePack,

    Item::CageKey,

    Item::BabyOil,

    Item::CoconutOil,

    Item::Lotion,

    Item::Vaseline,

];

impl Item {

    const ALL: [Self; 20] = [

        Self::WarmTowel,

        Self::IcePack,

        Self::Ruler,

        Self::DoorLock,

        Self::DndSign,

        Self::Blinds,

        Self::Pump,

        Self::PumpPro,

        Self::Extender,

        Self::Ring,

        Self::Supplement,

        Self::Cage,

        Self::CageKey,

        Self::Spreadsheet,

        Self::TeapotSpout,

        Self::BabyOil,

        Self::CoconutOil,

        Self::Lotion,

        Self::Vaseline,

        Self::EnergyDrink,

    ];

    pub fn slug(self) -> &'static str {

        match self {

            Self::WarmTowel => "warm-towel",

            Self::IcePack => "ice-pack",

            Self::Ruler => "ruler",

            Self::DoorLock => "door-lock",

            Self::DndSign => "do-not-disturb-sign",

            Self::Blinds => "window-blinds",

            Self::Pump => "pump",

            Self::PumpPro => "pump-pro",

            Self::Extender => "extender",

            Self::Ring => "cock-ring",

            Self::Supplement => "supplement",

            Self::Cage => "chastity-cage",

            Self::CageKey => "cage-key",

            Self::Spreadsheet => "spreadsheet",

            Self::TeapotSpout => "teapot-spout",

            Self::BabyOil => "baby-oil",

            Self::CoconutOil => "coconut-oil",

            Self::Lotion => "lotion",

            Self::Vaseline => "vaseline",

            Self::EnergyDrink => "energy-drink",

        }

    }

    pub fn parse(slug: &str) -> Option<Self> {

        Self::ALL.into_iter().find(|item| item.slug() == slug)

    }

    pub fn description(self) -> &'static str {

        match self {

            Self::WarmTowel => {

                "Reusable. Satisfies the warm-up precondition. Use technique=warmup."

            }

            Self::IcePack => "Lowers EQ by 40. Use technique=ice.",

            Self::Ruler => "Measures honestly. The length resource now carries a Content-Length.",

            Self::DoorLock => "Keeps your brother out once.",

            Self::DndSign => "Works on housekeeping. Mum does not read signs.",

            Self::Blinds => "The neighbour has a good camera. The blinds are better.",

            Self::Pump => "A pump.",

            Self::PumpPro => "Use technique=pump. Gains are temporary.",

            Self::Extender => "Passive gains each turn. Takes up room. Can be removed with DELETE.",

            Self::Ring => {

                "Girth gains doubled. Rupture risk raised. Responses are compressed if you accept gzip. Can be removed with DELETE."

            }

            Self::Supplement => "Use technique=supplement.",

            Self::Cage => "Locked.",

            Self::CageKey => "Pump-Bro does not ask.",

            Self::Spreadsheet => "Charts your progress. Every line is flat.",

            Self::TeapotSpout => "Does nothing. Cannot be dropped.",

            Self::BabyOil => "Smooth. +1. One unit per jelq.",

            Self::CoconutOil => "+3. Solid until it matters. Suspicious at Nan's.",

            Self::Lotion => "Handy. +0, and it chafes, so each jelq costs an extra stamina.",

            Self::Vaseline => "+2. Raises suspicion everywhere.",

            Self::EnergyDrink => "technique=energy. +3 stamina. Tastes like a battery.",

        }

    }

    pub fn pitch(self) -> &'static str {

        match self {

            Self::PumpPro => "A pump, but professional.",

            Self::Extender => "Passive gains every turn. Takes up room.",

            Self::Ring => "Doubles girth gains. Pump-Bro does not discuss the risks.",

            Self::Supplement => "Clinically formulated. Ships from overseas.",

            Self::DndSign => "Housekeeping respects it.",

            Self::Blinds => "The neighbour has a good camera.",

            Self::Ruler => "Pump-Bro does not recommend it.",

            Self::IcePack => "Cold.",

            _ => self.description(),

        }

    }

    pub fn price(self) -> Option<u32> {

        match self {

            Self::Pump => Some(10),

            Self::PumpPro | Self::Extender | Self::CageKey => Some(15),

            Self::Ring => Some(8),

            Self::Supplement | Self::DoorLock | Self::DndSign | Self::Blinds | Self::Ruler => {

                Some(5)

            }

            Self::BabyOil | Self::CoconutOil | Self::Lotion | Self::Vaseline => Some(4),

            Self::IcePack => Some(3),

            Self::WarmTowel

            | Self::Cage

            | Self::Spreadsheet

            | Self::TeapotSpout

            | Self::EnergyDrink => None,

        }

    }

}

pub struct Items {

    pub ruler: bool,

    pub ice_packs: u32,

    pub door_locks: u32,

    pub dnd_sign: bool,

    pub blinds: bool,

    pub pump_pro: bool,

    pub extender: bool,

    pub ring: bool,

    pub supplements: u32,

    pub cage: bool,

    pub spreadsheet: bool,

    pub teapot_spout: bool,

    pub baby_oil: u32,

    pub coconut_oil: u32,

    pub lotion: u32,

    pub vaseline: u32,

    pub energy_drinks: u32,

}

impl Default for Items {

    fn default() -> Self {

        Self {

            ruler: false,

            ice_packs: STARTING_ICE_PACKS,

            door_locks: 0,

            dnd_sign: false,

            blinds: false,

            pump_pro: false,

            extender: false,

            ring: false,

            supplements: 0,

            cage: false,

            spreadsheet: false,

            teapot_spout: false,

            baby_oil: 1,

            coconut_oil: 0,

            lotion: 1,

            vaseline: 0,

            energy_drinks: 0,

        }

    }

}

impl Items {

    pub fn has(&self, item: Item) -> bool {

        match item {

            Item::WarmTowel => true,

            Item::IcePack => self.ice_packs > 0,

            Item::Ruler => self.ruler,

            Item::DoorLock => self.door_locks > 0,

            Item::DndSign => self.dnd_sign,

            Item::Blinds => self.blinds,

            Item::PumpPro => self.pump_pro,

            Item::Extender => self.extender,

            Item::Ring => self.ring,

            Item::Supplement => self.supplements > 0,

            Item::Cage => self.cage,

            Item::Spreadsheet => self.spreadsheet,

            Item::TeapotSpout => self.teapot_spout,

            Item::BabyOil => self.baby_oil > 0,

            Item::CoconutOil => self.coconut_oil > 0,

            Item::Lotion => self.lotion > 0,

            Item::Vaseline => self.vaseline > 0,

            Item::EnergyDrink => self.energy_drinks > 0,

            Item::Pump | Item::CageKey => false,

        }

    }

    pub fn owned(&self) -> Vec<Item> {

        Item::ALL

            .into_iter()

            .filter(|&item| self.has(item))

            .collect()

    }

    pub fn grant(&mut self, item: Item) {

        match item {

            Item::IcePack => self.ice_packs += 1,

            Item::Ruler => self.ruler = true,

            Item::DoorLock => self.door_locks += 1,

            Item::DndSign => self.dnd_sign = true,

            Item::Blinds => self.blinds = true,

            Item::PumpPro => self.pump_pro = true,

            Item::Extender => self.extender = true,

            Item::Ring => self.ring = true,

            Item::Supplement => self.supplements += 1,

            Item::Cage => self.cage = true,

            Item::CageKey => self.cage = false,

            Item::Spreadsheet => self.spreadsheet = true,

            Item::TeapotSpout => self.teapot_spout = true,

            Item::BabyOil => self.baby_oil += 1,

            Item::CoconutOil => self.coconut_oil += 1,

            Item::Lotion => self.lotion += 1,

            Item::Vaseline => self.vaseline += 1,

            Item::EnergyDrink => self.energy_drinks += 1,

            Item::WarmTowel | Item::Pump => {}

        }

    }

}