SWS.lua

--[[
### License
 This file is part of Simple Warehouse Saving.
 Copyright (C) 2023 Michael Cole

 Simple Warehouse Saving is free software: you can redistribute it and/or modify
 it under the terms of the GNU 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 General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
# README 
Simple Warehouse Saving by Pikey Aug 2023 version 1.0 (thebgpikester@hotmail.com)
Thanks to Speed and Grimes for IntegratedbasicSerialize() function 
Thanks to Eagle Dynamics for keeping the dream alive.

This software is licensed under the Lesser GNU GPL 2007 
Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>

https://www.gnu.org/licenses/lgpl-3.0.txt
This script is supplied with the full LICENSE at https://github.com/thebgpikester/SWS

## PREREQS
Requires lfs and io desanitized from missionscripting.lua.
No 3rd party SSE like Mist or Moose required.

## WHAT IT DOES
Gets the contents of all DCS Airbase warehouses in the mission and saves them to file. Reloads from the save every mission. (no prompts)

## USAGE
Load this script at mission start as a DO SCRIPT.
By default, DCS sets unlimited warehouses. Set some warehouses to have limited contents by clicking on them in the Mission Editor and unchecking 'Unlimited' next to any of aircraft, fuel and weapons. Set a fixed limit amount. Optionally, setup supply chains as per pg 104 DCS user manual EN 2020.pdf in your main DCS installation\Doc folder.
--]]

SWS={} --don't touch

SWS.filepath = lfs.writedir().."SimpleWarehouse.lua"
--You can configure a custom directory for the save file here if needs be.

SWS.updateDelaySeconds = 60
--You can change the timer interval for the saving of warehouses here. 

--COMMON FUNCTIONS
 function SWS.IntegratedbasicSerialize(s)
    if s == nil then
      return "\"\""
    else
      if ((type(s) == 'number') or (type(s) == 'boolean') or (type(s) == 'function') or (type(s) == 'table') or (type(s) == 'userdata') ) then
        return tostring(s)
      elseif type(s) == 'string' then
        return string.format('%q', s)
      end
    end
  end
  
-- imported slmod.serializeWithCycles (thanks to Speed and Grimes)
  function SWS.IntegratedserializeWithCycles(name, value, saved)
    local basicSerialize = function (o)
      if type(o) == "number" then
        return tostring(o)
      elseif type(o) == "boolean" then
        return tostring(o)
      else -- assume it is a string
        return SWS.IntegratedbasicSerialize(o)
      end
    end

    local t_str = {}
    saved = saved or {}       -- initial value
    if ((type(value) == 'string') or (type(value) == 'number') or (type(value) == 'table') or (type(value) == 'boolean')) then
      table.insert(t_str, name .. " = ")
      if type(value) == "number" or type(value) == "string" or type(value) == "boolean" then
        table.insert(t_str, basicSerialize(value) ..  "\n")
      else

        if saved[value] then    -- value already saved?
          table.insert(t_str, saved[value] .. "\n")
        else
          saved[value] = name   -- save name for next time
          table.insert(t_str, "{}\n")
          for k,v in pairs(value) do      -- save its fields
            local fieldname = string.format("%s[%s]", name, basicSerialize(k))
            table.insert(t_str, SWS.IntegratedserializeWithCycles(fieldname, v, saved))
          end
        end
      end
      return table.concat(t_str)
    else
      return ""
    end
  end

function SWS.file_exists(name) 
    if lfs.attributes(name) then
    return true
    else
    return false end 
end

function SWS.writemission(data, file)
  SWS.File = io.open(file, "w")
  SWS.File:write(data)
  SWS.File:close()
end

local Airbases = world.getAirbases()

SWS.airbaseContents = {}

function saveWarehouseContents() 
  
 for i=1, #(Airbases) do
    local w=Airbases[i]:getWarehouse()
    local Inv = w:getInventory()
    SWS.airbaseContents[Airbases[i]:getName()]=Inv
 end

return timer.getTime() + SWS.updateDelaySeconds
end

function serializeWarehouseContents()
   SWS.newMissionStr = SWS.IntegratedserializeWithCycles("SWS.SimpleWarehouse",SWS.airbaseContents)
   return timer.getTime() + SWS.updateDelaySeconds
end

function writeWarehouseContents()
  SWS.writemission(SWS.newMissionStr, SWS.filepath)
  return timer.getTime() + SWS.updateDelaySeconds
end

function loadWarehouseContents()
 dofile(SWS.filepath)
 
 local Airbases = world.getAirbases()
 
 for _, airbase in ipairs(Airbases) do

    local w=airbase:getWarehouse()
   
    local liquids = SWS.SimpleWarehouse[airbase:getName()]["liquids"]
    for liquidType, amountLiquid in pairs (liquids) do
      w:setLiquidAmount(liquidType, amountLiquid)
    end
    
    local weapons = SWS.SimpleWarehouse[airbase:getName()]["weapon"]
    for weaponName, amount in pairs(weapons) do
      w:setItem(weaponName, amount)
    end
        
    local aircraft = SWS.SimpleWarehouse[airbase:getName()]["aircraft"]
    for aircraftName, aircraftCount in pairs(aircraft) do
      w:setItem(aircraftName, aircraftCount)
    end 
 end
end

--START SCRIPT

if SWS.file_exists(SWS.filepath) then
  env.info("Loading Pikey's Simple Warehouse Saving - Save file exists, loading ...")
  loadWarehouseContents()
else
  env.info("Loading Pikey's Simple Warehouse Saving - save file does not exist, writing ...")
  saveWarehouseContents()
  serializeWarehouseContents()
  writeWarehouseContents()
end 

timer.scheduleFunction(saveWarehouseContents, {}, timer.getTime() + 1)
timer.scheduleFunction(serializeWarehouseContents, {}, timer.getTime() + 2)
timer.scheduleFunction(writeWarehouseContents, {}, timer.getTime() + 3)