135 lines
3.8 KiB
Lua
135 lines
3.8 KiB
Lua
local enums = {}
|
|
|
|
---@type LuaEntityPrototype[]
|
|
local cached_miners = {}
|
|
---@type table<string, boolean>
|
|
local cached_resource_categories = {}
|
|
|
|
local invalid_resource = { --fluid or otherwise
|
|
["se-core-mining"] = true,
|
|
}
|
|
local miner_blacklist = {
|
|
["se-core-miner-drill"] = true
|
|
}
|
|
|
|
function enums.get_default_miner()
|
|
if game.active_mods["nullius"] then
|
|
return "nullius-medium-miner-1"
|
|
end
|
|
return "electric-mining-drill"
|
|
end
|
|
|
|
---Get mining drills and resource categories
|
|
---@return LuaEntityPrototype[], table
|
|
function enums.get_available_miners()
|
|
enums.get_available_miners = function() return cached_miners, cached_resource_categories end
|
|
|
|
local all_miners = game.get_filtered_entity_prototypes{{filter="type", type="mining-drill"}}
|
|
---@type table<string, LuaEntityPrototype>
|
|
--local all_fluids = game.get_filtered_item_prototypes({filter="type", type="
|
|
local all_resources = game.get_filtered_entity_prototypes{{filter="type", type="resource"}}
|
|
---@type table<string, LuaResourceCategoryPrototype>
|
|
|
|
for name, proto in pairs(all_resources) do
|
|
---@cast proto LuaEntityPrototype
|
|
local mineable_properties = proto.mineable_properties
|
|
|
|
if mineable_properties.products then
|
|
for _, product in ipairs(mineable_properties.products) do
|
|
if product.type == "fluid" then
|
|
invalid_resource[name] = true
|
|
break
|
|
end
|
|
end
|
|
else
|
|
invalid_resource[name] = true
|
|
break
|
|
end
|
|
end
|
|
|
|
for miner_name, miner_proto in pairs(all_miners) do
|
|
if miner_blacklist[miner_name] then goto continue_miner end
|
|
if miner_proto.resource_categories then
|
|
for resource_cat, bool in pairs(miner_proto.resource_categories) do
|
|
if invalid_resource[resource_cat] then
|
|
miner_blacklist[miner_name] = true
|
|
end
|
|
end
|
|
else
|
|
miner_blacklist[miner_name] = true
|
|
goto continue_miner
|
|
end
|
|
|
|
---@cast miner_proto LuaEntityPrototype
|
|
local fluidboxes = miner_proto.fluidbox_prototypes
|
|
for _, fluidbox in pairs(fluidboxes) do
|
|
---@cast fluidbox LuaFluidBoxPrototype
|
|
if fluidbox.production_type == "output" then
|
|
miner_blacklist[miner_name] = true
|
|
end
|
|
end
|
|
::continue_miner::
|
|
end
|
|
|
|
if game.active_mods["Cursed-FMD"] then
|
|
local mangled_categories = {}
|
|
local miners = {}
|
|
for name, proto in pairs(all_miners) do
|
|
if string.find(name, ";") then -- Cursed-FMD hack
|
|
for resource_name, _ in pairs(proto.resource_categories) do
|
|
if not invalid_resource[resource_name] and not string.find(resource_name, "core-fragment") then
|
|
mangled_categories[resource_name] = true
|
|
end
|
|
end
|
|
else
|
|
if proto.flags and proto.flags.hidden then goto continue_miner end
|
|
if miner_blacklist[name] then goto continue_miner end
|
|
|
|
for resource_name, _ in pairs(proto.resource_categories) do
|
|
if not invalid_resource[resource_name] and not string.find(resource_name, "core-fragment") then
|
|
miners[name] = proto
|
|
end
|
|
end
|
|
end
|
|
::continue_miner::
|
|
end
|
|
cached_miners = miners
|
|
cached_resource_categories = mangled_categories
|
|
else
|
|
local miners = {}
|
|
local resource_categories = {
|
|
["basic-solid"] = true,
|
|
["hard-resource"] = true,
|
|
}
|
|
for name, proto in pairs(all_miners) do
|
|
if proto.flags and proto.flags.hidden then goto continue_miner end
|
|
if miner_blacklist[name] then goto continue_miner end
|
|
--if not proto.resource_categories["basic-solid"] then goto continue_miner end
|
|
for resource_category, bool in pairs(proto.resource_categories) do
|
|
resource_categories[resource_category] = bool
|
|
end
|
|
|
|
miners[name] = proto
|
|
|
|
::continue_miner::
|
|
end
|
|
|
|
cached_miners = miners
|
|
cached_resource_categories = resource_categories
|
|
--[[ {
|
|
["basic-solid"] = true,
|
|
["hard-resource"] = true,
|
|
}]]
|
|
end
|
|
return enums.get_available_miners()
|
|
end
|
|
|
|
enums.space_surfaces = {
|
|
["asteroid-belt"] = true,
|
|
["asteroid-field"] = true,
|
|
["orbit"] = true,
|
|
["anomaly"] = true,
|
|
}
|
|
|
|
return enums
|