--***********************************************************
--**                    ROBERT JOHNSON                     **
--***********************************************************

require "TimedActions/ISBaseTimedAction"

ISTakeWaterAction = ISBaseTimedAction:derive("ISTakeWaterAction");

function ISTakeWaterAction:isValid()
	-- If the player is very thirsty, the destination item may get drained while filling it.
	-- When drained, the item may turn into another "empty" item, removing the item we're filling
	-- from it's container.
    if self.oldItem ~= nil then
        if self.oldItem and not self.oldItem:getContainer() then return false end
        return self.waterObject:hasWater()
    end
	if self.item and not self.item:getContainer() then return false end
	return self.waterObject:hasWater()
end

function ISTakeWaterAction:waitToStart()
    if self.waterObject then
        self.character:faceThisObject(self.waterObject)
    end
    return self.character:shouldBeTurning()
end

function ISTakeWaterAction:update()
    if self.item ~= nil then
        self.item:setJobDelta(self:getJobDelta());
        self.item:setUsedDelta(self.startUsedDelta + (self.endUsedDelta - self.startUsedDelta) * self:getJobDelta());
    end
    if self.waterObject then
        self.character:faceThisObject(self.waterObject);
    end
end

function ISTakeWaterAction:start()
    local waterAvailable = self.waterObject:getWaterAmount()

    if self.oldItem ~= nil then
        self.character:getInventory():AddItem(self.item);
        if self.character:isPrimaryHandItem(self.oldItem) then
            self.character:setPrimaryHandItem(self.item);
        end
        if self.character:isSecondaryHandItem(self.oldItem) then
            self.character:setSecondaryHandItem(self.item);
        end
        self.character:getInventory():Remove(self.oldItem);
        self.oldItem = nil
    end
    self.item = self.item;
    local props = self.waterObject:getProperties()
    local hasWaterFlag = (props ~= nil) and props:Is(IsoFlagType.water)
    local isLakeOrRiver = not instanceof(self.waterObject, "IsoWorldInventoryObject") and (props ~= nil) and luautils.stringStarts(self.waterObject:getSprite():getName(), 'blends_natural_02')
    local isPuddle = not hasWaterFlag and not isLakeOrRiver and (props ~= nil) and props:Is(IsoFlagType.solidfloor)
    if self.item ~= nil then
		self.item:setBeingFilled(true)
	    self.item:setJobType(getText("ContextMenu_Fill") .. self.item:getName());
	    self.item:setJobDelta(0.0);
        if (props ~= nil) and (props:Val("CustomName") == "Dispenser") then
            self.sound = self.character:playSound(self.item:getFillFromDispenserSound() or "GetWaterFromTap");
        elseif instanceof(self.waterObject, "IsoThumpable") or hasWaterFlag or isLakeOrRiver or isPuddle then -- play the drink sound for rain barrel
            self.sound = self.character:playSound("GetWaterFromLake");
        else
            self.sound = self.character:playSound(self.item:getFillFromTapSound() or "GetWaterFromTap");
        end
		local destCapacity = (1 - self.item:getUsedDelta()) / self.item:getUseDelta()
		self.waterUnit = math.min(math.ceil(destCapacity - 0.001), waterAvailable)
		self.startUsedDelta = self.item:getUsedDelta()
		self.endUsedDelta = math.min(self.item:getUsedDelta() + self.waterUnit * self.item:getUseDelta(), 1.0)
		self.action:setTime((self.waterUnit * 10) + 30)
	
		self:setAnimVariable("FoodType", self.item:getEatType());
		self:setActionAnim("fill_container_tap");
		if not self.item:getEatType() then
			self:setOverrideHandModels(nil, self.item:getStaticModel())
		else
			self:setOverrideHandModels(self.item:getStaticModel(), nil)
		end
    else
        if isLakeOrRiver or isPuddle then
            self.sound = self.character:playSound("DrinkingFromRiver");
        elseif instanceof(self.waterObject, "IsoThumpable") or hasWaterFlag or isLakeOrRiver then -- play the drink sound for rain barrel
            self.sound = self.character:playSound("DrinkingFromPool");
--            getSoundManager():PlayWorldSoundWav("PZ_DrinkingFromBottle", self.character:getCurrentSquare(), 0, 2, 0.8, true);
        else
            self.sound = self.character:playSound("DrinkingFromTap");
--            getSoundManager():PlayWorldSound("PZ_DrinkingFromTap", self.character:getCurrentSquare(), 0, 2, 0.8, true);
        end
		local thirst = self.character:getStats():getThirst()
		local waterNeeded = math.min(math.ceil(thirst / 0.1), 10)
		self.waterUnit = math.min(waterNeeded, waterAvailable)
		self.action:setTime((self.waterUnit * 10) + 15)
	
		self:setActionAnim("drink_tap")
		self:setOverrideHandModels(nil, nil)
	end
	
	self.character:reportEvent("EventTakeWater");
end

function ISTakeWaterAction.SendTakeWaterCommand(playerObj, object, units)
    if instanceof(object, "IsoWorldInventoryObject") then
        local itemID = object:getItem():getID()
        local args = {x=object:getX(), y=object:getY(), z=object:getZ(), units=units, itemID=itemID}
        sendClientCommand(playerObj, 'object', 'takeWaterFromItem', args)
        return
    end
    local index = object:getObjectIndex()
    local args = {x=object:getX(), y=object:getY(), z=object:getZ(), units=units, index=index}
    sendClientCommand(playerObj, 'object', 'takeWater', args)
end

function ISTakeWaterAction:stopSound()
	if self.sound and self.character:getEmitter():isPlaying(self.sound) then
		self.character:stopOrTriggerSound(self.sound);
	end
end

function ISTakeWaterAction:stop()
	self:stopSound();
    local used = self:getJobDelta() * self.waterUnit
    if used >= 1 then
        ISTakeWaterAction.SendTakeWaterCommand(self.character, self.waterObject, used)
    end
    if self.item ~= nil then
		self.item:setBeingFilled(false)
        self.item:setJobDelta(0.0);
        if self.waterObject:isTaintedWater() then
            self.item:setTaintedWater(true);
        end
    end
    ISBaseTimedAction.stop(self);
end

function ISTakeWaterAction:perform()
	self:stopSound();
    if self.item ~= nil then
        self.item:setBeingFilled(false)
        self.item:getContainer():setDrawDirty(true);
        self.item:setJobDelta(0.0);
        if self.waterObject:isTaintedWater() then
            self.item:setTaintedWater(true);
        end
    --Without this setUsedDelta call, the final tick never goes through.
    -- the item's UsedDelta value is set at like .99
    --This means that the option to fill that container never goes away.
    --		if self.item:getUsedDelta() > 0.91 then
    --			self.item:setUsedDelta(1.0);
    --		end
    else
        local thirst = self.character:getStats():getThirst() - (self.waterUnit / 10)
        self.character:getStats():setThirst(math.max(thirst, 0.0));
        if self.waterObject:isTaintedWater() then
			--tainted water shouldn't kill the player but make them sick - dangerous when sick
			local bodyDamage	= self.character:getBodyDamage();
			local stats			= self.character:getStats();
			if bodyDamage:getPoisonLevel() < 20 and stats:getSickness() < 0.3 then
				bodyDamage:setPoisonLevel(math.min(bodyDamage:getPoisonLevel() + 10 + self.waterUnit, 20));
			end
        end
    end

    ISTakeWaterAction.SendTakeWaterCommand(self.character, self.waterObject, self.waterUnit)

    -- needed to remove from queue / start next.
    ISBaseTimedAction.perform(self);
end

function ISTakeWaterAction:new (character, item, waterUnit, waterObject, time, oldItem)
	local o = {}
	setmetatable(o, self)
	self.__index = self
	o.character = character;
	o.item = item;
    o.oldItem = oldItem;
	o.stopOnWalk = true;
	o.stopOnRun = true;
	o.maxTime = 10 -- will set this in start()
	o.waterUnit = waterUnit; -- will set this in start()
	o.waterObject = waterObject;
	return o
end