local Clockwork = Clockwork;

local ITEM = Clockwork.item:New("clothes_base", true);
ITEM.name = "Storm Trooper NCO Armor";
ITEM.uniqueID = "armor_clothes_base";
ITEM.model = "models/kriegsyntax/starwars/stormtrooper_cap.mdl";
ITEM.actualWeight = 2;
ITEM.invSpace = 2;
ITEM.protection = 10;
ITEM.maxArmor = 100;
ITEM.hasGasmask = false;
ITEM.hasRebreather = false;
ITEM.isPA = false;
ITEM.useText = "Wear";
ITEM.category = "Clothing";
ITEM.description = "The suit";

ITEM:AddData("armor", -1, true);
ITEM:AddData("Rarity", 1, true);
ITEM:AddData("PerceivedWeight", -1, true);
ITEM:AddData("AddInvSpace", 0, true);

ITEM:AddQueryProxy("color", ITEM.GetRarityColor);
ITEM:AddQueryProxy("weight", "PerceivedWeight");
ITEM:AddQueryProxy("addInvSpace", "AddInvSpace");

-- Called when a player wears the accessory.
function ITEM:OnChangedClothes(player, bIsWearing)
    if (bIsWearing) then
        -- Player is putting armor on
        -- Set player armor value to current remaining armor
        player:SetMaxArmor(self("maxArmor"));
        player:SetArmor(self:GetData("armor"));

        -- Set perceived weight to 0
        self:SetData("PerceivedWeight", 0);

        -- Allow armor pockets to be used for inventory weight
        self:SetData("AddInvSpace", self("invSpace"));
    else
        -- Player is taking item off
        -- Save armor on item and reset player's armor value
        self:SetData("armor", math.Clamp(player:Armor(), 0, self("maxArmor")));
        player:SetArmor(0);

        -- Set perceived weight back to the actual weight
        self:SetData("PerceivedWeight", self("actualWeight"));

        -- Remove pocket space
        self:SetData("AddInvSpace", 0);
    end;
end;

-- Called when a player has unequipped the item.
function ITEM:OnPlayerUnequipped(player, extraData)
    if (self("hasGasmask")) then
        local items = player:GetInventory();
        for k, itemList in pairs(items) do
            for k, item in pairs(itemList) do
                if (!item:IsBasedFrom("filter_base")) then
                    break;
                elseif (item:GetData("equipped")) then
                    hasEquipped = true;
                    Clockwork.player:Notify(player, "You need to unscrew your filter first!");
                    return false;
                end;
            end;
        end;
    end;

    if (player:GetInventoryWeight() + self("actualWeight") > (player:GetMaxWeight() - self("addInvSpace"))) then
        Clockwork.player:Notify(player, "You don't have enough inventory space to unequip this!");
        return false;
    end;

    player:RemoveClothes();
end;

function ITEM:CanGiveStorage(player, storageTable)
    if (player:IsWearingItem(self)) then
        Clockwork.player:Notify(player, "You cannot store this while you are wearing it!");
        return false;
    end;
end;

-- Called when a player attempts to take the item from storage.
function ITEM:CanTakeStorage(player, storageTable)
    local target = Clockwork.entity:GetPlayer(storageTable.entity);

    if (target) then
        if (target:GetInventoryWeight() > (target:GetMaxWeight() - self("addInvSpace"))) then
            return false;
        end;
    end;
end;

-- A function to get the item's rarity color.
function ITEM:GetRarityColor()
    local rarity = self:GetData("Rarity");
    if (rarity == 1) then
        return Color(248, 248, 255, 255);
    elseif (rarity == 2) then
        return Color(61, 210, 11, 255);
    elseif (rarity == 3) then
        return Color(47, 120, 255, 255);
    elseif (rarity == 4) then
        return Color(145, 50, 200, 255);
    elseif (rarity == 5) then
        return Color(255, 150, 0, 255);
    end;
end;

function ITEM:EntityHandleMenuOption(player, entity, option, argument)
    -- Armor repair
    if (option == "Repair") then
        self:RepairArmor(player);

    -- Admin armor repair
    elseif (option == "AdminRepair") then
        if (player:IsSuperAdmin()) then
            self:SetData("armor", self("maxArmor"));
            Clockwork.kernel:PrintLog(LOGTYPE_MAJOR, player:Name().." has admin-repaired a "..self("name")..".");
        else
            Clockwork.Notify(player, "You are not a super admin!");
        end;
    end;
end;

if (SERVER) then
    function ITEM:OnInstantiated()
        -- Set initial values of the data fields
        if (self:GetData("armor") == -1) then
            self:SetData("armor", self("maxArmor"));
        end;

        if (self:GetData("PerceivedWeight") == -1) then
            self:SetData("PerceivedWeight", self("actualWeight"));
        end;

        -- Set PA footstep sounds if it's PA.
        if (self("isPA")) then
            self.runSound = {
                "newvegas/fst_armorpower_01.wav",
                "newvegas/fst_armorpower_02.wav",
                "newvegas/fst_armorpower_03.wav"
            }
            self.walkSound = self.runSound;
        end;
    end;

    function ITEM:RepairArmor(player)
        if (!self("repairItem")) then
            Clockwork.player:Notify(player, "This item cannot be repaired! (Contact a dev if it should be)");
            return;
        end;
        if (self:GetData("armor") == self("maxArmor")) then
            Clockwork.player:Notify(player, "This item already has full armor!");
            return;
        end;

        -- Check if a flag is required and if the player has it
        if (self("repairFlag") and !Clockwork.player:HasFlags(player, self("repairFlag"))) then
            Clockwork.player:Notify(player, "You do not have the "..self("repairFlag").." flag!");
        end;

        local repairItem = player:FindItemByID(self("repairItem"));

        -- Check if the player has the needed item
        if (!player:HasItemByID(self("repairItem"))) then
            repairItem = Clockwork.item:CreateInstance(self("repairItem"));
            Clockwork.player:Notify(player, "You do not have enough "..repairItem("name").." to repair this!");
            return;
        end;

        local amount = self("repairAmount") or 50;

        -- Take the repair item
        player:TakeItem(repairItem);
        -- Set armor to new value
        self:SetData("armor", math.Clamp(self:GetData("armor") + amount, 0, self("maxArmor")));
        -- Notify player
        Clockwork.player:Notify(player, "You have repaired the "..self("name").." for "..tostring(amount).." armor.");
        Clockwork.kernel:PrintLog(LOGTYPE_GENERIC, player:Name().." has repaired a "..self("name").." for "..tostring(amount)..".");
    end;
else
    function ITEM:GetClientSideInfo()
        if (!self:IsInstance()) then return; end;

        local clientSideInfo = "";

        local itemRarity = self:GetData("Rarity");
        if (itemRarity == 1) then
            clientSideInfo = Clockwork.kernel:AddMarkupLine(clientSideInfo, "Common", self:GetRarityColor());
        elseif (itemRarity == 2) then
            clientSideInfo = Clockwork.kernel:AddMarkupLine(clientSideInfo, "Uncommon", self:GetRarityColor());
        elseif (itemRarity == 3) then
            clientSideInfo = Clockwork.kernel:AddMarkupLine(clientSideInfo, "Rare", self:GetRarityColor());
        elseif (itemRarity == 4) then
            clientSideInfo = Clockwork.kernel:AddMarkupLine(clientSideInfo, "Unique", self:GetRarityColor());
        elseif (itemRarity == 5) then
            clientSideInfo = Clockwork.kernel:AddMarkupLine(clientSideInfo, "Legendary", self:GetRarityColor());
        end;

        local armor = self:GetData("armor");
        if (Clockwork.player:IsWearingItem(self)) then
            armor = Clockwork.Client:Armor();
        end;

        clientSideInfo = Clockwork.kernel:AddMarkupLine(clientSideInfo, "Armor: "..math.floor(armor));
        clientSideInfo = Clockwork.kernel:AddMarkupLine(clientSideInfo, "Protection: "..math.floor(100 * self("protection")).."%");

        return (clientSideInfo != "" and clientSideInfo);
    end;

    -- Called when the item entity's menu options are needed.
    function ITEM:GetEntityMenuOptions(entity, options)
        if (!IsValid(entity)) then
            return;
        end;

        if (Clockwork.Client:IsSuperAdmin()) then
            options["AdminRepair"] = function()
                Clockwork.entity:ForceMenuOption(entity, "AdminRepair", nil);
            end;
        end;

        options["Repair"] = function()
            Clockwork.entity:ForceMenuOption(entity, "Repair", nil);
        end;
    end;
end;

Clockwork.item:Register(ITEM);