career_retrievedamaged/lua/ge/extensions/career/vehicleRetrieval.lua

-- Copyright (C) 2025  snoutie
-- Authors: snoutie (copyright@achtarmig.org)
-- This program is free software: you can redistribute it and/or modify
-- it under the terms of the GNU Affero General Public License as published
-- by the Free Software Foundation, either version 3 of the License, or
-- (at your option) any later version.

-- This program is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-- GNU Affero General Public License for more details.

-- You should have received a copy of the GNU Affero General Public License
-- along with this program.  If not, see <https://www.gnu.org/licenses/>.

local M = {}

M.dependencies = { 'career_modules_inventory', 'freeroam_facilities', "career_vehicleSaveSystem",
    "career_modules_permissions", 'career_modules_valueCalculator' }

local career_modules_inventory_removeVehicleObject
local career_modules_inventory_openMenu
local vehicleObjectsToRemove = {}

local function spawnVehicle(inventoryId, callback)
    if career_modules_inventory.getVehicleIdFromInventoryId(inventoryId) then
        callback()
    else
        career_modules_inventory.spawnVehicle(inventoryId, nil, callback)
    end
end

local function Retrieve(inventoryId)
    career_vehicleSaveSystem.EnqueueVehicleToSave(inventoryId)
    if not career_vehicleSaveSystem.SaveVehicle(inventoryId) then
        career_vehicleSaveSystem.DequeueVehicleToSave(inventoryId)
    end

    local vehicleValue = career_modules_valueCalculator.getInventoryVehicleValue(inventoryId, true)
    local currentVehicleValue = career_modules_valueCalculator.getInventoryVehicleValue(inventoryId)

    career_vehicleSaveSystem.CheckSavedAsync(function()
        extensions.core_jobsystem.create(
            function(job)
                career_vehicleSaveSystem.SetFade(true)
                job.sleep(1)
                spawnVehicle(inventoryId, function()
                    career_vehicleSaveSystem.TeleportVehicle(inventoryId, false)
                    career_vehicleSaveSystem.LoadVehicle(inventoryId, currentVehicleValue < vehicleValue)
                end)
            end
        )
    end)
end

local function onComputerAddFunctions(menuData, computerFunctions)
    if menuData.computerFacility.functions["vehicleInventory"] then
        local favouriteVehicleId = career_modules_inventory.getFavoriteVehicle()

        local vehicleValue = career_modules_valueCalculator.getInventoryVehicleValue(favouriteVehicleId, true)
        local currentVehicleValue = career_modules_valueCalculator.getInventoryVehicleValue(favouriteVehicleId)
        local function_label = "Retrieve Favourite Vehicle"
        if currentVehicleValue < vehicleValue then
            function_label = function_label .. " (Damaged)"
        end

        local computerFunctionData = {
            id = "retrieve_damaged",
            label = function_label,
            callback = function() Retrieve(favouriteVehicleId) end,
            order = 1
        }

        local repairPermission = career_modules_permissions.getStatusForTag("vehicleRepair",
            { inventoryId = favouriteVehicleId })

        if not repairPermission["allow"] then
            computerFunctionData.disabled = true
            computerFunctionData.reason = { type = "text", label = "Vehicle is being repaired." }
        end

        if menuData.tutorialPartShoppingActive or menuData.tutorialTuningActive then
            computerFunctionData.disabled = true
            computerFunctionData.reason = career_modules_computer.reasons.tutorialActive
        end
        computerFunctions.general[computerFunctionData.id] = computerFunctionData
    end
end

local function RemoveVehicleObject(inventoryId)
    if career_modules_inventory.getVehicleIdFromInventoryId(inventoryId) then
        table.insert(vehicleObjectsToRemove, inventoryId)
        career_vehicleSaveSystem.EnqueueVehicleToSave(inventoryId)
        career_vehicleSaveSystem.SaveVehicle(inventoryId)
        career_vehicleSaveSystem.CheckSavedAsync(function()
            career_modules_inventory_removeVehicleObject(inventoryId)
        end)
    end
end

local function OpenMenu(_chooseButtonsData, header, _buttonsActive)
    if _chooseButtonsData then
        for _, buttonData in ipairs(_chooseButtonsData) do
            if buttonData.buttonText == "Retrieve" then
                buttonData.buttonText     = "Retrieve"
                buttonData.callback       = function(inventoryId) Retrieve(inventoryId) end
                buttonData.repairRequired = false
            end
        end
    end
    career_modules_inventory_openMenu(_chooseButtonsData, header, _buttonsActive)
end

local function onCareerActive()
    career_modules_inventory_removeVehicleObject = career_modules_inventory.removeVehicleObject
    career_modules_inventory.removeVehicleObject = RemoveVehicleObject
    career_modules_inventory_openMenu = career_modules_inventory.openMenu
    career_modules_inventory.openMenu = OpenMenu
end

M.onComputerAddFunctions = onComputerAddFunctions
M.onCareerActive = onCareerActive

return M