require("game-stats")
require("production-stats")
require("pollution-stats")
require("research-stats")
require("power-stats")
require("logistic-network-stats")

tickInterval = tonumber(settings.global["factorio-metrics-exporter-tick-interval"].value) or 300
udpAddress = 52555
isInitialized = false
sendIndex = 0
serverIndex = 0

options = {
  enableMod = false,
  enablePlayers = false, 
  enableProduction = false,
  enableFluid = false,
  enableKills = false,
  enablePollution = false,
  enableRobots = false,
  enableResearch = false,
  enablePower = false,
  enableTrains = false
}


script.on_init(function ()
  storage.electricGrids = {}
  storage.researchedTechnologies = {}
  storage.playerDeathCount = {}
  storage.totalLabCount = 0
  storage.totalResearchSpeed = 0
  storage.totalReseachProductivity = 0
  storage.labs = {}
  sendIndex = 0

  options.enableMod = settings.global["factorio-metrics-exporter-enable"].value
  options.enableProduction = settings.global["factorio-metrics-exporter-export_production_stats"].value
  options.enablePollution = settings.global["factorio-metrics-exporter-export_pollution_stats"].value
  options.enableFluid = settings.global["factorio-metrics-exporter-export_fluid_stats"].value
  options.enablePlayers = settings.global["factorio-metrics-exporter-export_player_stats"].value
  options.enableKills = settings.global["factorio-metrics-exporter-export_kill_stats"].value
  options.enablePower = settings.global["factorio-metrics-exporter-export_power_stats"].value
  options.enableResearch = settings.global["factorio-metrics-exporter-export_research_stats"].value
  options.enableRobots = settings.global["factorio-metrics-exporter-export_logistic_stats"].value


end)

script.on_load(function ()
  log("factorio-metrics-exporter: on_load")
  log("tickInterval: "..tickInterval)
  log("udpAddress: "..udpAddress)
  
  options.enableMod = settings.global["factorio-metrics-exporter-enable"].value
  options.enableProduction = settings.global["factorio-metrics-exporter-export_production_stats"].value
  options.enablePollution = settings.global["factorio-metrics-exporter-export_pollution_stats"].value
  options.enableFluid = settings.global["factorio-metrics-exporter-export_fluid_stats"].value
  options.enablePlayers = settings.global["factorio-metrics-exporter-export_player_stats"].value
  options.enableKills = settings.global["factorio-metrics-exporter-export_kill_stats"].value
  options.enablePower = settings.global["factorio-metrics-exporter-export_power_stats"].value
  options.enableResearch = settings.global["factorio-metrics-exporter-export_research_stats"].value
  options.enableRobots = settings.global["factorio-metrics-exporter-export_logistic_stats"].value
end)

script.on_configuration_changed(function()
  storage.electricGrids = storage.electricGrids or {}
  storage.labs = storage.labs or {}
  storage.playerDeathCount = storage.playerDeathCount or {}
  storage.researchedTechnologies = storage.researchedTechnologies or {}
  storage.totalLabCount = storage.totalLabCount or 0
  storage.totalReseachProductivity = storage.totalReseachProductivity or 0
  storage.totalResearchSpeed = storage.totalResearchSpeed or 0
end
)



script.on_event(defines.events.on_runtime_mod_setting_changed, function(event)
  log("Mod setting changed: "..event.setting)
  if event.setting == "factorio-metrics-exporter-tick-interval" then
    tickInterval = settings.global["factorio-metrics-exporter-tick-interval"].value
  end 
  if event.setting == "factorio-metrics-exporter-enable" then
    options.enableMod = settings.global["factorio-metrics-exporter-enable"].value
  end
  if event.setting == "factorio-metrics-exporter-export_production_stats" then
    options.enableProduction = settings.global["factorio-metrics-exporter-export_production_stats"].value
  end
  
  if event.setting == "factorio-metrics-exporter-export_fluid_stats" then
    options.enableFluid = settings.global["factorio-metrics-exporter-export_fluid_stats"].value
  end
  
  if event.setting == "factorio-metrics-exporter-export_pollution_stats" then
    options.enablePollution = settings.global["factorio-metrics-exporter-export_pollution_stats"].value
  end
  
  if event.setting == "factorio-metrics-exporter-export_power_stats" then
    options.enablePower = settings.global["factorio-metrics-exporter-export_power_stats"].value
  end
  
  if event.setting == "factorio-metrics-exporter-export_logistic_stats" then
    options.enableRobots = settings.global["factorio-metrics-exporter-export_logistic_stats"].value
  end
  
  if event.setting == "factorio-metrics-exporter-export_player_stats" then
    options.enablePlayers = settings.global["factorio-metrics-exporter-export_player_stats"].value
  end

  if event.setting == "factorio-metrics-exporter-export_kill_stats" then
    options.enableKills = settings.global["factorio-metrics-exporter-export_kill_stats"].value
  end
  
  if event.setting == "factorio-metrics-exporter-export_research_stats" then
    options.enableResearch = settings.global["factorio-metrics-exporter-export_research_stats"].value
  end

end)


script.on_event(defines.events.on_player_died, function(event)
  --local player = game.get_player(event.player_index)
  storage.playerDeathCount[event.player_index] = (storage.playerDeathCount[event.player_index] or 0) + 1
end)

function SendGameStats(event)
  local returnParts = {}
  returnParts[#returnParts+1] = GetMods()
  returnParts[#returnParts+1] = GetMapSeed()
  returnParts[#returnParts+1] = GetRocketsLaunched()
  returnParts[#returnParts+1] = GetPlayerTime()
  returnParts[#returnParts+1] = GetPlayerDeaths()
  helpers.send_udp(udpAddress, table.concat(returnParts, "\n"), serverIndex)
end

function SendAll(event)

  if(isInitialized == false) then
    if game.is_multiplayer() then
      serverIndex = 0
      log("Loaded game as mulitplayer")
    end
  else
    log("Loaded game as singleplayer")
    serverIndex = 1
  end
  isInitialized = true

  log("ServerIndex is "..serverIndex.." now")

  if PowerGridScanned == false then
    GenerateNetworks()
  end

  if LabsScanned == false then
    GenerateLabInfo()
end


  if options.enableMod==true then

  if (event.tick % math.floor(tickInterval/7) ~= 0) then return end

  sendIndex = (sendIndex % 7) + 1
  if sendIndex == 1 then SendProductionStats() end
  if sendIndex == 2 then SendPollutionStats() end
  if sendIndex == 3 then SendKillStats() end
  if sendIndex == 4 then SendFluidProductionStats() end
  if sendIndex == 5 then SendBuildStats() end
  if sendIndex == 6 then SendResearchStats() end
  if sendIndex == 7 then SendLogisticStats() end

  if(event.tick % tickInterval*2 == 0) then 
    if options.enablePlayers then 
      SendGameStats(event) end
    end
  end
end

script.on_event(defines.events.on_tick, SendAll)

--Script hooks for power stats
script.on_event(defines.events.on_built_entity,UpdateGrids,{{filter = "type", type = "electric-pole"}})
script.on_event(defines.events.on_player_mined_entity, PurgeNetworks,{{filter = "type", type = "electric-pole"}})
script.on_event(defines.events.on_robot_built_entity,UpdateGrids,{{filter = "type", type = "electric-pole"}})
script.on_event(defines.events.on_robot_mined_entity,PurgeNetworks,{{filter = "type", type = "electric-pole"}})