Factorio-Paranoidal_mod/helmod/core/Converter.lua

local inflate = require "lib.deflatelua"
local deflate = require "lib.zlib-deflate"
local base64 = require "lib.base64"

---
---Description of the module.
---@class Converter
local Converter = {
  ---single-line comment
  classname = "HMConverter",
  ---use gzip
  compressed = true,
  ---length of line
  line_length = 120
}

-------------------------------------------------------------------------------
---Trim string
---@param s string
---@return string
function Converter.trim(s)
  return (s:gsub("^%s*(.-)%s*$", "%1"))
end

-------------------------------------------------------------------------------
---Write table to string
---@param data_table table
---@return string
function Converter.write2(data_table)
  local data_string = serpent.dump(data_table)
  return game.encode_string(data_string)
end

-------------------------------------------------------------------------------
---Write table to string
---@param data_table table
---@return string
function Converter.write(data_table)
  local data_string = serpent.dump(data_table)
  if (Converter.compressed) then
    data_string = deflate.gzip(data_string)
    data_string = base64.enc(data_string)
    if (Converter.line_length > 0) then
      ---Add line breaks
      data_string = data_string:gsub( ("%S"):rep(Converter.line_length), "%1\n" )
    end
  end
  data_string = data_string .. "\n"
  return data_string
end

-------------------------------------------------------------------------------
---Read string to table
---@param data_string string
---@return table
function Converter.read2(data_string)
  if data_string == nil then return nil end
  data_string = Converter.trim(data_string)
  if (string.sub(data_string, 1, 8) ~= "do local") then
    local ok , err = pcall(function()
      data_string = game.decode_string(data_string)
    end)
    if not(ok) then
      return nil
    end
  end
  local status, data_table = pcall(loadstring, data_string)
  if (status) then
    return data_table()
  end
  return nil
end

-------------------------------------------------------------------------------
---Read string to table
---@param data_string string
---@return table
function Converter.read(data_string)
  if data_string == nil then return nil end
  data_string = Converter.trim(data_string)
  if (string.sub(data_string, 1, 8) ~= "do local") then
    local input = base64.dec(data_string)
    local data_table = {}
    local output = {}
    local status, result = pcall(inflate.gunzip, { input = input, output = function(byte) output[#output+1] = string.char(byte) end })
    if (status) then
      data_string = table.concat(output)
    else
      return nil
    end
  end
  local status, data_table = pcall(loadstring, data_string)
  if (status) then
    return data_table()
  end
  return nil
end

-------------------------------------------------------------------------------
---Read string to table
---@param data_string string
---@return table
function Converter.decode_string(data_string)
  if data_string == nil then return nil end
  data_string = Converter.trim(data_string)
  if (string.sub(data_string, 1, 8) ~= "do local") then
    local input = base64.dec(data_string)
    local data_table = {}
    local output = {}
    local status, result = pcall(inflate.inflate_zlib, { input = input, output = function(byte) output[#output+1] = string.char(byte) end })
    if (status) then
      return table.concat(output)
    else
      return nil
    end
  end
end

-------------------------------------------------------------------------------
---Indent string
---@param json string
---@return string
function Converter.indent(json)
  local table_value = game.json_to_table(json)
  local result = Converter.indentTable(table_value, 0)
  return result
end

-------------------------------------------------------------------------------
---Indent table
---@param input any
---@param level number
---@return string
function Converter.indentTable(input, level)
  local indent_char = "    "
  if type(input) == "table" then
    local first = true
    local is_array = true
    local temp = ""
    for key,value in pairs(input) do
      if first == false then
        temp = temp .. ",\n"
      end
      if type(key) == "string" then
        is_array = false
        temp = temp .. string.rep(indent_char, level)
        temp = temp .. "\"" .. key .. "\": "
        temp = temp .. Converter.indentTable(value, level + 1)
      else
        temp = temp .. string.rep(indent_char, level)
        temp = temp .. Converter.indentTable(value, level + 1)
      end
      first = false
    end
    if is_array == true then
      temp = "[\n" .. temp .. "\n"
      temp = temp .. string.rep(indent_char, level-1)
      temp = temp .. "]"
      return temp
    else
      temp = "{\n" .. temp .. "\n"
      temp = temp .. string.rep(indent_char, level-1)
      temp = temp .. "}"
      return temp
    end
  elseif type(input) == "string" then
    return "\"" .. input .. "\""
  elseif type(input) == "boolean" then
    if input == true then
      return "true"
    else
      return "false"
    end
  end
  return input
end

return Converter