1
0
mirror of https://github.com/ComfyFactory/ComfyFactorio.git synced 2025-01-24 03:47:58 +02:00
2020-05-13 08:18:14 +02:00

545 lines
17 KiB
Lua

local Event = require 'utils.event'
local Map_score = require 'comfy_panel.map_score'
local BiterRolls = require 'modules.wave_defense.biter_rolls'
local unearthing_worm = require 'functions.unearthing_worm'
local unearthing_biters = require 'functions.unearthing_biters'
local Loot = require 'maps.lumberjack.loot'
local Pets = require 'maps.lumberjack.biter_pets'
local tick_tack_trap = require 'functions.tick_tack_trap'
local RPG = require 'maps.lumberjack.rpg'
local Mining = require 'maps.lumberjack.mining'
local Terrain = require 'maps.lumberjack.terrain'
local BiterHealthBooster = require 'modules.biter_health_booster'
-- tables
local WPT = require 'maps.lumberjack.table'
local WD = require 'modules.wave_defense.table'
-- module
local Public = {}
local math_random = math.random
local math_floor = math.floor
local math_abs = math.abs
local grandmaster = '[color=blue]Grandmaster:[/color]'
local treasure_chest_messages = {
"You notice an old crate within the rubble. It's filled with treasure!",
"You find a chest underneath the broken rocks. It's filled with goodies!",
'We has found the precious!'
}
local rare_treasure_chest_messages = {
'Your magic improves. You have found a chest that is filled with rare treasures!',
"Oh wonderful magic. You found a chest underneath the broken rocks. It's filled with rare goodies!",
"You're a wizard Harry! We has found the rare precious!"
}
local disabled_entities = {
'gun-turret',
'laser-turret',
'flamethrower-turret',
'land-mine'
}
local disabled_threats = {
['entity-ghost'] = true,
['raw-fish'] = true
}
local defeated_messages = {
"Oh no, the biters nom'ed the train away!",
"I'm not 100% sure, but - apparently the train was chewed away.",
'You had one objective - defend the train *-*',
"Looks like we're resetting cause you did not defend the train ._.",
'ohgodno-why-must-you-do-this-to-me'
}
local function shuffle(tbl)
local size = #tbl
for i = size, 1, -1 do
local rand = math_random(size)
tbl[i], tbl[rand] = tbl[rand], tbl[i]
end
return tbl
end
local function set_objective_health(entity, final_damage_amount)
local this = WPT.get_table()
if final_damage_amount == 0 then
return
end
this.locomotive_health = math_floor(this.locomotive_health - final_damage_amount)
this.cargo_health = math_floor(this.cargo_health - final_damage_amount)
if this.locomotive_health > this.locomotive_max_health then
this.locomotive_health = this.locomotive_max_health
end
if this.cargo_health > this.cargo_max_health then
this.cargo_health = this.cargo_max_health
end
if this.locomotive_health <= 0 then
Public.loco_died()
end
local m
if entity == this.locomotive then
m = this.locomotive_health / this.locomotive_max_health
entity.health = 1000 * m
elseif entity == this.locomotive_cargo then
m = this.cargo_health / this.cargo_max_health
entity.health = 600 * m
end
rendering.set_text(this.health_text, 'HP: ' .. this.locomotive_health .. ' / ' .. this.locomotive_max_health)
end
local function is_protected(entity)
local this = WPT.get_table()
local map_name = 'lumberjack'
if string.sub(entity.surface.name, 0, #map_name) ~= map_name then
return true
end
local protected = {this.locomotive, this.locomotive_cargo}
for i = 1, #protected do
if protected[i] == entity then
return true
end
end
return false
end
local function protect_train(event)
local this = WPT.get_table()
if event.entity.force.index ~= 1 then
return
end --Player Force
if is_protected(event.entity) then
if event.entity == this.locomotive_cargo or event.entity == this.locomotive then
if event.cause then
if
event.cause.force.index == 2 or event.cause.force.name == 'lumber_defense' or
event.cause.force.name == 'defenders'
then
if this.locomotive_health <= 0 then
goto continue
end
set_objective_health(event.entity, event.final_damage_amount)
end
end
::continue::
end
if not event.entity.valid then
return
end
event.entity.health = event.entity.health + event.final_damage_amount
end
end
local function hidden_biter(entity)
local surface = entity.surface
local h = math_floor(math_abs(entity.position.y))
local m = 1 / Terrain.level_depth
local count = math_floor(math_random(0, h + Terrain.level_depth) * m) + 1
local position = surface.find_non_colliding_position('small-biter', entity.position, 16, 0.5)
if not position then
position = entity.position
end
BiterRolls.wave_defense_set_unit_raffle(h * 0.20)
for _ = 1, count, 1 do
local unit
if math_random(1, 3) == 1 then
unit = surface.create_entity({name = BiterRolls.wave_defense_roll_spitter_name(), position = position})
else
unit = surface.create_entity({name = BiterRolls.wave_defense_roll_biter_name(), position = position})
end
if math_random(1, 64) == 1 then
BiterHealthBooster.add_boss_unit(unit, m * h * 5 + 1, 0.38)
end
end
end
local function hidden_worm(entity)
BiterRolls.wave_defense_set_worm_raffle(math.sqrt(entity.position.x ^ 2 + entity.position.y ^ 2) * 0.20)
entity.surface.create_entity({name = BiterRolls.wave_defense_roll_worm_name(), position = entity.position})
end
local function hidden_biter_pet(event)
if math_random(1, 2048) ~= 1 then
return
end
BiterRolls.wave_defense_set_unit_raffle(math.sqrt(event.entity.position.x ^ 2 + event.entity.position.y ^ 2) * 0.25)
local unit
if math_random(1, 3) == 1 then
unit =
event.entity.surface.create_entity(
{name = BiterRolls.wave_defense_roll_spitter_name(), position = event.entity.position}
)
else
unit =
event.entity.surface.create_entity(
{name = BiterRolls.wave_defense_roll_biter_name(), position = event.entity.position}
)
end
Pets.biter_pets_tame_unit(game.players[event.player_index], unit, true)
end
local function hidden_treasure(event)
local player = game.players[event.player_index]
local rpg_t = RPG.get_table()
local magic = rpg_t[player.index].magic
if math.random(1, 320) ~= 1 then
return
end
if magic > 50 then
player.print(
rare_treasure_chest_messages[math.random(1, #rare_treasure_chest_messages)],
{r = 0.98, g = 0.66, b = 0.22}
)
Loot.add_rare(event.entity.surface, event.entity.position, 'wooden-chest', magic)
return
end
player.print(treasure_chest_messages[math.random(1, #treasure_chest_messages)], {r = 0.98, g = 0.66, b = 0.22})
Loot.add(event.entity.surface, event.entity.position, 'wooden-chest')
end
local function biters_chew_rocks_faster(event)
if event.entity.force.index ~= 3 then
return
end --Neutral Force
if not event.cause then
return
end
if not event.cause.valid then
return
end
if event.cause.force.index ~= 2 then
return
end --Enemy Force
event.entity.health = event.entity.health - event.final_damage_amount * 2.5
end
local function give_coin(player)
player.insert({name = 'coin', count = 1})
end
local function on_player_mined_entity(event)
local this = WPT.get_table()
local entity = event.entity
local player = game.players[event.player_index]
if not player.valid then
return
end
if not entity.valid then
return
end
local map_name = 'lumberjack'
if string.sub(entity.surface.name, 0, #map_name) ~= map_name then
return
end
if disabled_threats[entity.name] then
return
end
Mining.on_player_mined_entity(event)
if entity.type == 'unit' or entity.type == 'unit-spawner' then
if math_random(1, 160) == 1 then
tick_tack_trap(entity.surface, entity.position)
return
end
if math.random(1, 32) == 1 then
hidden_biter(event.entity)
return
end
end
if entity.type ~= 'simple-entity' or entity.type ~= 'tree' then
this.mined_scrap = this.mined_scrap + 1
give_coin(player)
if math.random(1, 32) == 1 then
hidden_biter(event.entity)
return
end
if math.random(1, 512) == 1 then
hidden_worm(event.entity)
return
end
hidden_biter_pet(event)
hidden_treasure(event)
if math_random(1, 160) == 1 then
tick_tack_trap(entity.surface, entity.position)
return
end
end
if entity.force.name ~= 'defenders' then
return
end
local positions = {}
local r = math.ceil(entity.prototype.max_health / 32)
for x = r * -1, r, 1 do
for y = r * -1, r, 1 do
positions[#positions + 1] = {x = entity.position.x + x, y = entity.position.y + y}
end
end
positions = shuffle(positions)
for i = 1, math.ceil(entity.prototype.max_health / 32), 1 do
if not positions[i] then
return
end
if math_random(1, 3) ~= 1 then
unearthing_biters(entity.surface, positions[i], math_random(5, 10))
else
unearthing_worm(entity.surface, positions[i])
end
end
end
local function on_entity_damaged(event)
if not event.entity then
return
end
if not event.entity.valid then
return
end
if not event.entity.health then
return
end
protect_train(event)
biters_chew_rocks_faster(event)
end
local function on_player_repaired_entity(event)
local this = WPT.get_table()
if not event.entity then
return
end
if not event.entity.valid then
return
end
if not event.entity.health then
return
end
local entity = event.entity
if entity == this.locomotive_cargo or entity == this.locomotive then
set_objective_health(entity, -1)
end
end
local function on_entity_died(event)
local this = WPT.get_table()
local entity = event.entity
if not entity.valid then
return
end
local map_name = 'lumberjack'
if string.sub(entity.surface.name, 0, #map_name) ~= map_name then
return
end
if disabled_threats[entity.name] then
return
end
Mining.on_entity_died(event)
if entity.type == 'unit' or entity.type == 'unit-spawner' then
this.biters_killed = this.biters_killed + 1
if math_random(1, 512) == 1 then
tick_tack_trap(entity.surface, entity.position)
return
end
if math.random(1, 32) == 1 then
hidden_biter(event.entity)
return
end
end
if entity.type ~= 'simple-entity' or entity.type ~= 'tree' then
if math.random(1, 32) == 1 then
hidden_biter(event.entity)
return
end
if math.random(1, 512) == 1 then
hidden_worm(event.entity)
return
end
if math_random(1, 512) == 1 then
tick_tack_trap(entity.surface, entity.position)
return
end
end
if entity.force.name ~= 'defenders' then
return
end
local positions = {}
local r = math.ceil(entity.prototype.max_health / 32)
for x = r * -1, r, 1 do
for y = r * -1, r, 1 do
positions[#positions + 1] = {x = entity.position.x + x, y = entity.position.y + y}
end
end
positions = shuffle(positions)
for i = 1, math.ceil(entity.prototype.max_health / 32), 1 do
if not positions[i] then
return
end
if math_random(1, 3) ~= 1 then
unearthing_biters(entity.surface, positions[i], math_random(5, 10))
else
unearthing_worm(entity.surface, positions[i])
end
end
end
local function on_robot_built_entity(event)
local map_name = 'lumberjack'
if string.sub(event.created_entity.surface.name, 0, #map_name) ~= map_name then
return
end
local y = event.created_entity.position.y
local ent = event.created_entity
if y >= 150 then
game.print(grandmaster .. ' I do not approve, ' .. ent.name .. ' was obliterated.', {r = 1, g = 0.5, b = 0.1})
ent.die()
return
else
for _, e in pairs(disabled_entities) do
if e == event.created_entity.name then
if y >= 0 then
ent.active = false
if event.player_index then
game.print(
grandmaster .. " Can't build here. I disabled your " .. ent.name .. '.',
{r = 1, g = 0.5, b = 0.1}
)
return
end
end
end
end
end
end
local function on_built_entity(event)
local map_name = 'lumberjack'
if string.sub(event.created_entity.surface.name, 0, #map_name) ~= map_name then
return
end
local player = game.players[event.player_index]
local y = event.created_entity.position.y
local ent = event.created_entity
if y >= 150 then
player.print(grandmaster .. ' I do not approve, ' .. ent.name .. ' was obliterated.', {r = 1, g = 0.5, b = 0.1})
ent.die()
return
else
for _, e in pairs(disabled_entities) do
if e == event.created_entity.name then
if y >= 0 then
ent.active = false
if event.player_index then
player.print(
grandmaster .. " Can't build here. I disabled your " .. ent.name .. '.',
{r = 1, g = 0.5, b = 0.1}
)
return
end
end
end
end
end
end
function Public.set_scores()
local this = WPT.get_table()
local wagon = this.locomotive_cargo
if not wagon then
return
end
if not wagon.valid then
return
end
local score = math_floor(wagon.position.y * -1)
for _, player in pairs(game.connected_players) do
if score > Map_score.get_score(player) then
Map_score.set_score(player, score)
end
end
end
function Public.loco_died()
local this = WPT.get_table()
local surface = game.surfaces[this.active_surface_index]
local wave_defense_table = WD.get_table()
Public.set_scores()
if not this.locomotive.valid then
wave_defense_table.game_lost = true
wave_defense_table.target = nil
game.print(
grandmaster .. ' ' .. defeated_messages[math.random(1, #defeated_messages)],
{r = 1, g = 0.5, b = 0.1}
)
game.print(grandmaster .. ' Better luck next time.', {r = 1, g = 0.5, b = 0.1})
Public.reset_map()
return
end
this.locomotive_health = 0
this.locomotive.color = {0.49, 0, 255, 1}
rendering.set_text(this.health_text, 'HP: ' .. this.locomotive_health .. ' / ' .. this.locomotive_max_health)
wave_defense_table.game_lost = true
wave_defense_table.target = nil
game.print(grandmaster .. ' ' .. defeated_messages[math.random(1, #defeated_messages)], {r = 1, g = 0.5, b = 0.1})
game.print(grandmaster .. ' Better luck next time.', {r = 1, g = 0.5, b = 0.1})
game.print(grandmaster .. ' Game will soft-reset shortly.', {r = 1, g = 0.5, b = 0.1})
game.forces.enemy.set_friend('player', true)
game.forces.player.set_friend('enemy', true)
local fake_shooter =
surface.create_entity({name = 'character', position = this.locomotive.position, force = 'enemy'})
surface.create_entity(
{
name = 'atomic-rocket',
position = this.locomotive.position,
force = 'enemy',
speed = 1,
max_range = 1200,
target = this.locomotive,
source = fake_shooter
}
)
surface.spill_item_stack(this.locomotive.position, {name = 'coin', count = 512}, false)
surface.spill_item_stack(this.locomotive_cargo.position, {name = 'coin', count = 512}, false)
this.game_reset_tick = game.tick + 1800
for _, player in pairs(game.connected_players) do
player.play_sound {path = 'utility/game_lost', volume_modifier = 0.75}
end
end
Event.add(defines.events.on_entity_damaged, on_entity_damaged)
Event.add(defines.events.on_player_repaired_entity, on_player_repaired_entity)
Event.add(defines.events.on_player_mined_entity, on_player_mined_entity)
Event.add(defines.events.on_entity_died, on_entity_died)
Event.add(defines.events.on_built_entity, on_built_entity)
Event.add(defines.events.on_robot_built_entity, on_robot_built_entity)
return Public