DogFoodIsntThatBad/Contents/mods/Dog Food Isn't That Bad/media/lua/client/DogFoodIsntThatBad.lua

local function adjustFoodNotPicky(item)
    -- Cancel out any negative effects
    if item:getUnhappyChange() > 0 then
        item:setUnhappyChange(item:getUnhappyChangeUnmodified() - item:getUnhappyChange())
    end
    if item:getBoredomChange() > 0 then
        item:setBoredomChange(item:getBoredomChangeUnmodified() - item:getBoredomChange())
    end
end

local function adjustFoodVeryPicky(item, proportion)
    local unhappyChange = item:getUnhappyChangeUnmodified()
    local boredomChange = item:getBoredomChangeUnmodified()

    -- Increase existng negative effects by 50%
    if item:getUnhappyChange() > 0 then
        unhappyChange = unhappyChange + item:getUnhappyChange() * 0.5
    end
    if item:getBoredomChange() > 0 then
        boredomChange = boredomChange + item:getBoredomChange() * 0.5
    end

    -- Microwaved food is dull
    if item:isCookedInMicrowave() then
        boredomChange = boredomChange + 10 * proportion
    end

    -- Uncooked food which ought to be cooked is disgusting
    if item:isCookable() and not item:isCooked() then
        unhappyChange = unhappyChange + 10 * proportion
    end

    -- Canned food is disgusting
    if item:getEatType() == "can" then
        unhappyChange = unhappyChange + 10 * proportion
    end

    -- Adjust food groups
    local foodGroupAdjustments = {
        ["Candy"] = 15,
        ["SoftDrink"] = 10,
        ["Sugar"] = 10,
        ["Wine"] = -10,
    }
    if foodGroupAdjustments[item:getFoodType()] then
        unhappyChange = unhappyChange + foodGroupAdjustments[item:getFoodType()] * proportion
    end

    -- Adjust specific foods
    local foodAdjustments = {
        ["BakingSoda"] = 20,
        ["Butter"] = 10,
        ["Candycane"] = 5,
        ["Cereal"] = 10,
        ["ChickenFried"] = 10,
        ["ChocoCakes"] = 15,
        ["CocoaPowder"] = 10,
        ["CookiesSugar"] = 5,
        ["Corndog"] = 10,
        ["Cornflour"] = 20,
        ["Crisps"] = 5,
        ["Crisps2"] = 5,
        ["Crisps3"] = 5,
        ["Crisps4"] = 5,
        ["FishFried"] = 10,
        ["GravyMix"] = 20,
        ["Gum"] = 10,
        ["HiHis"] = 15,
        ["Icecream"] = 10,
        ["JuiceBox"] = 10,
        ["Lollipop"] = 15,
        ["Macandcheese"] = 20,
        ["MintCandy"] = 15,
        ["OystersFried"] = 10,
        ["PancakeMix"] = 20,
        ["Plonkies"] = 15,
        ["Popcorn"] = 5,
        ["Processedcheese"] = 5,
        ["QuaggaCakes"] = 15,
        ["RefriedBeans"] = 5,
        ["Smore"] = 10,
        ["SnoGlobes"] = 15,
        ["TofuFried"] = 10,
    }
    if foodAdjustments[item:getType()] then
        unhappyChange = unhappyChange + foodAdjustments[item:getType()] * proportion
    end

    item:setUnhappyChange(unhappyChange)
    item:setBoredomChange(boredomChange)
end

local function getProportion(food, n)
    n = math.min(n, 1.0)
    n = math.max(n, 0.0)
    if food:getBaseHunger() ~= 0 and food:getHungChange() ~= 0 then
        n = math.min(food:getBaseHunger() * n / food:getHungChange(), 1.0)
        n = math.max(n, 0.0)
    end
    if food:getHungChange() < 0 and food:getHungChange() * (1.0 - n) > -0.01 then
        n = 1.0
    end
    if food:getHungChange() == 0 and food:getThirstChange() < 0 and food:getThirstChange() * (1.0 - n) > -0.01 then
        n = 1.0
    end
    return n
end

local base_tooltip_render = ISToolTipInv.render

function ISToolTipInv:render()
    local origUnhappyChange
    local origBoredomChange

    if self.item:IsFood() then
        origUnhappyChange = self.item:getUnhappyChangeUnmodified()
        origBoredomChange = self.item:getBoredomChangeUnmodified()

        -- Adjust food effects based on traits
        local traits = getPlayer():getTraits()
        if traits:contains("NotAPickyEater") then
            adjustFoodNotPicky(self.item)
        elseif traits:contains("RefinedPalate") then
            if self.item:getBaseHunger() ~= 0 then
                local percentage = self.item:getHungChange() / self.item:getBaseHunger()
                adjustFoodVeryPicky(self.item, percentage)
            else
                adjustFoodVeryPicky(self.item, 1)
            end
        end
    end

    -- Call original function
    base_tooltip_render(self)

    if self.item:IsFood() then
        -- Reset food effects
        self.item:setUnhappyChange(origUnhappyChange)
        self.item:setBoredomChange(origBoredomChange)
    end
end

local base_eat_start = ISEatFoodAction.start

function ISEatFoodAction:start()
    self.origUnhappyChange = self.item:getUnhappyChangeUnmodified()
    self.origBoredomChange = self.item:getBoredomChangeUnmodified()
    self.proportion = getProportion(self.item, self.percentage)

    -- Adjust food effects based on traits
    local traits = self.character:getTraits()
    if traits:contains("NotAPickyEater") then
        adjustFoodNotPicky(self.item)
    elseif traits:contains("RefinedPalate") then
        adjustFoodVeryPicky(self.item, self.percentage)
    end

    -- Call original function
    base_eat_start(self)
end

local base_eat_stop = ISEatFoodAction.stop

function ISEatFoodAction:stop()
    -- Call original function
    base_eat_stop(self)

    -- Reset food effects
    local percentage = self:getJobDelta()
    if percentage > 0.95 then
        percentage = 1.0
    end
    local adjust = 1 - (self.proportion * percentage)
    self.item:setUnhappyChange(self.origUnhappyChange * adjust)
    self.item:setBoredomChange(self.origBoredomChange * adjust)
end

local base_eat_perform = ISEatFoodAction.perform

function ISEatFoodAction:perform()
    -- Call original function
    base_eat_perform(self)

    -- Reset food effects
    local adjust = 1 - self.proportion
    self.item:setUnhappyChange(self.origUnhappyChange * adjust)
    self.item:setBoredomChange(self.origBoredomChange * adjust)
end