Private
Public Access
1
0
Files
lua-prometheus-exporter/control.lua

356 lines
14 KiB
Lua

require("game-stats")
require("production-stats")
require("pollution-stats")
require("research-stats")
require("power-stats")
require("logistic-network-stats")
require("train-stats")
tickInterval = tonumber(settings.global["factorio-metrics-exporter-tick-interval"].value) or 300
udpAddress = 52555
isInitialized = false
sendIndex = 0
serverIndex = 0
scannedGrids = false
scannedLabs = false
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 = {}
storage.playerKillCount = {}
storage.representativePoles = {}
storage.playerDeathCause = {}
storage.constructedEntites = {}
storage.deconstructedEntities = {}
storage.trainStats = {}
---@type LuaTrain[]
storage.trains = {}
storage.scannedGrids = false
storage.scannedLabs = false
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
storage.playerKillCount = storage.playerKillCount or {}
storage.representativePoles = storage.representativePoles or {}
storage.scannedGrids = storage.scannedGrids or false
storage.scannedLabs = storage.scannedLabs or false
storage.playerDeathCause = storage.playerDeathCause or {}
storage.constructedEntites = storage.constructedEntites or {}
storage.deconstructedEntities = storage.deconstructedEntities or{}
storage.trains = storage.trains or {}
---@type table<uint, trainStat>
storage.trainStats = storage.trainStats or {}
ScanNetworks()
ScanLabs()
ScanTrains()
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)
--Log player cause by player
if event.cause and event.cause.type == "character" then
local killer = event.cause.player
if killer then
local killer_index = killer.index
local victim_index = event.player_index
local killerName = killer.name
local victimName = game.players[victim_index].name
log(("Player ID %d:%s killed player ID %d:%s"):format(killer_index,killerName,victim_index,victimName))
storage.playerKillCount[killer_index] =
storage.playerKillCount[killer_index] or {}
storage.playerKillCount[killer_index][victim_index] =
(storage.playerKillCount[killer_index][victim_index] or 0) + 1
end
end
--Log cause of player death
if event.cause and event.cause.type then
storage.playerDeathCause[event.player_index] =
storage.playerDeathCause[event.player_index] or {}
storage.playerDeathCause[event.player_index][event.cause.type] =
(storage.playerDeathCause[event.player_index][event.cause.type] or 0) + 1
log(("Player %s died from type %s"):format(game.players[event.player_index].name,event.cause.type))
end
--Log player death count
storage.playerDeathCount[event.player_index] = (storage.playerDeathCount[event.player_index] or 0) + 1
end)
function SendGameStats()
if options.enablePlayers then
local returnParts = {}
GetMods()
returnParts[#returnParts+1] = GetMapSeed()
returnParts[#returnParts+1] = GetRocketsLaunched()
returnParts[#returnParts+1] = GetPlayerTime()
returnParts[#returnParts+1] = GetPlayerDeaths()
returnParts[#returnParts+1] = GetPlayerKills()
returnParts[#returnParts+1] = GetPlayerEntityStats()
helpers.send_udp(udpAddress, table.concat(returnParts, "\n"), serverIndex)
end
end
function SendAll(event)
if(isInitialized == false) then
if game.is_multiplayer() then
serverIndex = 0
log("Loaded game as mulitplayer")
else
log("Loaded game as singleplayer")
serverIndex = 1
end
end
isInitialized = true
if scannedGrids == false then
ScanNetworks()
scannedGrids = true
end
if scannedLabs == false then
ScanLabs()
scannedLabs = true
end
if options.enableMod==true then
local interval = math.max(1, math.floor(tickInterval / 10))
if event.tick % interval ~= 0 then return end
sendIndex = (sendIndex % 10) + 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 sendIndex == 8 then SendPowerStats() end
if sendIndex == 9 then SendGameStats() end
if sendIndex == 10 then SendTrainStats() end
end
end
function UpdateStorage(event)
if not event then return end
if event.entity.type == "lab" then
UpdateLabs(event)
end
if event.entity.type == "electric-pole" then
AddPowerPole(event)
end
end
function RemoveStorage(event)
if not event then return end
if event.entity.type =="lab" then
RemoveLab(event)
end
if event.entity.type == "electric-pole" then
RemovePowerPole(event)
end
--log(event.entity.name)
if event.entity.name == "crash-site-spaceship" then
--log(event.name)
if event.name == defines.events.on_player_mined_entity then
--log("in ban call")
if settings.global["factorio-metrics-exporter-enable_denkmalschutz"].value == true then
game.ban_player(event.player_index,"You violated the rules of DENKMALSCHUTZ!!!")
end
end
end
end
function CreateEntity(event)
if not event then return end
--Event is PlayerPlaced
if event.name == defines.events.on_built_entity then
if event.entity.name ~= "entity-ghost" then
storage.constructedEntites[event.player_index] = storage.constructedEntites[event.player_index] or {}
storage.constructedEntites[event.player_index][event.entity.name] = (storage.constructedEntites[event.player_index][event.entity.name] or 0) + 1
end
end
--Event is RobotPlaced
if event.name == defines.events.on_robot_built_entity then
if event.entity.name ~= "entity-ghost" then
if event.entity.last_user then
local lastUser = event.entity.last_user.index
storage.constructedEntites[lastUser] = storage.constructedEntites[lastUser] or {}
storage.constructedEntites[lastUser][event.entity.name] = (storage.constructedEntites[lastUser][event.entity.name] or 0) + 1
end
end
end
--Event is spaceplatform build
if event.name == defines.events.on_space_platform_built_entity then
if event.entity.name ~= "entity-ghost" then
if event.entity.last_user then
local lastUser = event.entity.last_user.index
storage.constructedEntites[lastUser] = storage.constructedEntites[lastUser] or {}
storage.constructedEntites[lastUser][event.entity.name] = (storage.constructedEntites[lastUser][event.entity.name] or 0) + 1
end
end
end
UpdateStorage(event)
end
function RemoveEntity(event)
if event.name == defines.events.on_player_mined_entity then
storage.deconstructedEntities[event.player_index] = storage.deconstructedEntities[event.player_index] or {}
storage.deconstructedEntities[event.player_index][event.entity.name] = (storage.deconstructedEntities[event.player_index][event.entity.name] or 0) + 1
end
if event.name == defines.events.on_robot_mined_entity then
if event.entity.name ~= "entity-ghost" then
if event.entity.last_user then
local lastUser = event.entity.last_user.index
storage.deconstructedEntities[lastUser] = storage.deconstructedEntities[lastUser] or {}
storage.deconstructedEntities[lastUser][event.entity.name] = (storage.deconstructedEntities[lastUser][event.entity.name] or 0) + 1
end
end
end
if event.name == defines.events.on_space_platform_mined_entity then
if event.entity.name ~= "entity-ghost" then
if event.entity.last_user then
local lastUser = event.entity.last_user.index
storage.deconstructedEntities[lastUser] = storage.deconstructedEntities[lastUser] or {}
storage.deconstructedEntities[lastUser][event.entity.name] = (storage.deconstructedEntities[lastUser][event.entity.name] or 0) + 1
end
end
end
if event.name == defines.events.on_entity_died then
end
RemoveStorage(event)
end
script.on_event(defines.events.on_tick, SendAll)
--Script hooks for power and lab stats
--script.on_event(defines.events.on_built_entity,UpdateStorage,{{filter = "type", type = "electric-pole"},{filter ="type", type="lab"}})
--script.on_event(defines.events.on_player_mined_entity, RemoveStorage,{{filter = "type", type = "electric-pole"},{filter ="type", type="lab"},{filter = "type", type="container"}})
--script.on_event(defines.events.on_robot_built_entity,UpdateStorage,{{filter = "type", type = "electric-pole"},{filter ="type", type="lab"}})
--script.on_event(defines.events.on_robot_mined_entity,RemoveStorage,{{filter = "type", type = "electric-pole"},{filter ="type", type="lab"},{filter = "type", type="container"}})
--script.on_event(defines.events.on_entity_died,RemoveStorage,{{filter = "type", type = "electric-pole"},{filter ="type", type="lab"},{filter = "type", type="container"}})
script.on_event(defines.events.on_built_entity,CreateEntity)
script.on_event(defines.events.on_robot_built_entity,CreateEntity)
script.on_event(defines.events.on_space_platform_built_entity,CreateEntity)
script.on_event(defines.events.on_player_mined_entity,RemoveEntity)
script.on_event(defines.events.on_robot_mined_entity,RemoveEntity)
script.on_event(defines.events.on_space_platform_mined_entity,RemoveEntity)
script.on_event(defines.events.on_entity_died,RemoveEntity)
script.on_event(defines.events.on_train_changed_state,onTrainStateChange)