minecraft-cc-tools/inferium-server.lua

local net = require('libs/net')
local utils = require('libs/utils')
local inferium = require('config/inferium')

local DEFAULT_HARVESTER_NAME = 'harvester'
local VERSION = "2.0.3"
local INFERIUM_SERVER = 'inferium.com'

local PERSISTED_CONFIGS = '/data/inferium-configs'
local UPGRADE_SCRIPT = '/upgrade.lua'

local defaultConfig = inferium.defaultConfig or error('no default config provided in config', 0)

-- Persistance utils

local savePersistedConfigs = function(configs)
  local file = fs.open(PERSISTED_CONFIGS, 'w')

  if not file then
    error('savePersistedConfigs: cannot open ' .. PERSISTED_CONFIGS .. ' file!')
  end

  file.write(textutils.serialize(configs))
  file.close()
end

local readPersistedConfigs = function()
  local file = fs.open(PERSISTED_CONFIGS, 'r')

  if not file then
    local initialConfig = {}
    savePersistedConfigs(initialConfig)
    return initialConfig
  end

  local serializedConfigs = file.readAll()
  file.close()

  return textutils.unserialize(serializedConfigs)
end

-- State
local LOCAL_CONFIGS = readPersistedConfigs()

local function getConfigForComputer(computerId)
  return LOCAL_CONFIGS[tostring(computerId)]
end

local function saveConfigForComputer(computerId, config)
  LOCAL_CONFIGS[tostring(computerId)] = config
  savePersistedConfigs(LOCAL_CONFIGS)
end

-- Utils
local function formatSeedName(essenceName)
  if not essenceName then
    return false
  end

  return 'mysticalagriculture:' .. essenceName .. '_seeds'
end

local function getConfigWithLength(config)
  if not config then
    return config
  end

  local configWithLength = utils.shallowClone(config)
  configWithLength.length = utils.sizeof(config.plan)

  return configWithLength
end

local function isValidConfig(config)
  if not config or type(config.plan) ~= 'table' then
    return false
  end

  if config.firstCropZ == nil or config.firstCropZ < 1 then
    return false
  end

  if config.fertilizedBoost == nil then
    return false
  end

  return true
end

-- Routes

local function getConfig(message)
  local id = message and message.payload and message.payload.id

  if not id then
    return {
      type = "get-config/error",
      payload = nil,
      errorMessage = "no id provided in message payload"
    }
  end

  local config = getConfigForComputer(id)

  if not config then
    return {
      type = "get-config/error",
      payload = nil,
      errorMessage = "cannot retrieve configuration for given id"
    }
  end

  return {
    type = "get-config/response",
    payload = getConfigWithLength(config)
  }
end

local function registerAndGetConfig(_, computerId)
  if not computerId == nil then
    print('get-config error: no computerId found')
    return nil
  end

  local config = getConfigForComputer(computerId)

  if not config then
    print('new harvester detected: ' .. tostring(computerId))
    saveConfigForComputer(computerId, defaultConfig)
    config = defaultConfig
  end

  return {
    type = "register-and-get-config/response",
    payload = getConfigWithLength(config)
  }
end

local function setConfig(message)
  local payload = message and message.payload
  local harvesterId = payload and payload.id
  local config = payload and payload.config

  if not isValidConfig(config) or not harvesterId then
    return false
  end

  saveConfigForComputer(harvesterId, config)
  return true
end

local function deleteConfig(message)
  local payload = message and message.payload
  local harvesterId = payload and payload.id

  if not harvesterId then
    return false
  end

  if LOCAL_CONFIGS[tostring(harvesterId)] then
    saveConfigForComputer(harvesterId, nil)
    return true
  end

  return false
end

-- return a table of harvesters { id: string, name: string }[]
local function listHarvesters()
  local result = {}

  for k, v in pairs(LOCAL_CONFIGS) do
    table.insert(result, { id = k, name = v.name or DEFAULT_HARVESTER_NAME })
  end

  return result
end

local function listAvailableSeeds()
  return {
    formatSeedName('inferium'),
    formatSeedName('experience'),
    formatSeedName('soulium'),
    formatSeedName('diamond'),
    formatSeedName('redstone'),
    formatSeedName('coal'),
    formatSeedName('iron'),
    formatSeedName('dye'),
    formatSeedName('nether_quartz'),
    formatSeedName('yellorium'),
    formatSeedName('glowstone'),
    formatSeedName('fire'),
    formatSeedName('water'),
    formatSeedName('dirt')
  }
end

local function exitServer(_, _, stopServer)
  stopServer();
  return true
end

local function upgradeServer()
  shell.execute(UPGRADE_SCRIPT)
  return true
end

local ROUTES = {
  ['register-and-get-config'] = registerAndGetConfig,
  ['get-config'] = getConfig,
  ['set-config'] = setConfig,
  ['delete-config'] = deleteConfig,
  ['list-harvesters'] = listHarvesters,
  ['list-seeds'] = listAvailableSeeds,
  ['exit-server'] = exitServer,
  ['upgrade-server'] = upgradeServer,
}

net.openRednet()
print('> inferium server v' .. VERSION .. ' started')

net.listenQuery(INFERIUM_SERVER, function(message, computerId, stopServer)
  if type(message) ~= 'table' then
    print('error: malformed message received', textutils.serialize(message))
    return {}
  end

  local router = ROUTES[message.type]

  if not router then
    print('warning: unknown type of message received', message.type)
    return {}
  end

  return router(message, computerId, stopServer)
end)
print('> server stopped')

net.closeRednet()
print('> rednet closed')