Private
Public Access
1
0
Files
lua-prometheus-exporter/control.lua
2026-01-02 10:26:50 +01:00

188 lines
7.1 KiB
Lua

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 = tonumber(settings.startup["factorio-metrics-exporter-udp-address"].value) or 52555
isInitialized = false
sendIndex = 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
end
else
serverIndex = 1
end
isInitialized = true
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"}})