ComfyFactorio/modules/towny/town_center.lua

--luacheck: ignore
local Team = require 'modules.towny.team'
local Table = require 'modules.towny.table'
local Public = {}

local math_random = math.random
local table_insert = table.insert
local math_floor = math.floor

local min_distance_to_spawn = 128
local square_min_distance_to_spawn = min_distance_to_spawn ^ 2
local town_radius = 27
local radius_between_towns = 160
local ore_amount = 750

local colors = {}
local c1 = 250
local c2 = 210
local c3 = -40
for v = c1, c2, c3 do
    table.insert(colors, {0, 0, v})
end
for v = c1, c2, c3 do
    table.insert(colors, {0, v, 0})
end
for v = c1, c2, c3 do
    table.insert(colors, {v, 0, 0})
end
for v = c1, c2, c3 do
    table.insert(colors, {0, v, v})
end
for v = c1, c2, c3 do
    table.insert(colors, {v, v, 0})
end
for v = c1, c2, c3 do
    table.insert(colors, {v, 0, v})
end

local town_wall_vectors = {}
for x = 2, town_radius, 1 do
    table_insert(town_wall_vectors, {x, town_radius})
    table_insert(town_wall_vectors, {x * -1, town_radius})
    table_insert(town_wall_vectors, {x, town_radius * -1})
    table_insert(town_wall_vectors, {x * -1, town_radius * -1})
end
for y = 2, town_radius - 1, 1 do
    table_insert(town_wall_vectors, {town_radius, y})
    table_insert(town_wall_vectors, {town_radius, y * -1})
    table_insert(town_wall_vectors, {town_radius * -1, y})
    table_insert(town_wall_vectors, {town_radius * -1, y * -1})
end

local gate_vectors_horizontal = {}
for x = -1, 1, 1 do
    table_insert(gate_vectors_horizontal, {x, town_radius})
    table_insert(gate_vectors_horizontal, {x, town_radius * -1})
end
local gate_vectors_vertical = {}
for y = -1, 1, 1 do
    table_insert(gate_vectors_vertical, {town_radius, y})
    table_insert(gate_vectors_vertical, {town_radius * -1, y})
end

local resource_vectors = {}
resource_vectors[1] = {}
for x = 7, 24, 1 do
    for y = 7, 24, 1 do
        table_insert(resource_vectors[1], {x, y})
    end
end
resource_vectors[2] = {}
for _, vector in pairs(resource_vectors[1]) do
    table_insert(resource_vectors[2], {vector[1] * -1, vector[2]})
end
resource_vectors[3] = {}
for _, vector in pairs(resource_vectors[1]) do
    table_insert(resource_vectors[3], {vector[1] * -1, vector[2] * -1})
end
resource_vectors[4] = {}
for _, vector in pairs(resource_vectors[1]) do
    table_insert(resource_vectors[4], {vector[1], vector[2] * -1})
end

local additional_resource_vectors = {}
additional_resource_vectors[1] = {}
for x = 10, 22, 1 do
    for y = -4, 4, 1 do
        table_insert(additional_resource_vectors[1], {x, y})
    end
end
additional_resource_vectors[2] = {}
for _, vector in pairs(additional_resource_vectors[1]) do
    table_insert(additional_resource_vectors[2], {vector[1] * -1, vector[2]})
end
additional_resource_vectors[3] = {}
for y = 10, 22, 1 do
    for x = -4, 4, 1 do
        table_insert(additional_resource_vectors[3], {x, y})
    end
end
additional_resource_vectors[4] = {}
for _, vector in pairs(additional_resource_vectors[3]) do
    table_insert(additional_resource_vectors[4], {vector[1], vector[2] * -1})
end

local market_collide_vectors = {{-1, 1}, {0, 1}, {1, 1}, {1, 0}, {1, -1}}

local clear_blacklist_types = {
    ['simple-entity'] = true,
    ['resource'] = true,
    ['cliff'] = true
}

local starter_supplies = {
    {name = 'raw-fish', count = 3},
    {name = 'grenade', count = 3},
    {name = 'stone', count = 32},
    {name = 'land-mine', count = 4},
    {name = 'iron-gear-wheel', count = 16},
    {name = 'iron-plate', count = 32},
    {name = 'copper-plate', count = 16},
    {name = 'shotgun', count = 1},
    {name = 'shotgun-shell', count = 8},
    {name = 'firearm-magazine', count = 16},
    {name = 'firearm-magazine', count = 16},
    {name = 'gun-turret', count = 2}
}

local function count_nearby_ore(surface, position, ore_name)
    local count = 0
    local r = town_radius + 8
    for _, e in pairs(surface.find_entities_filtered({area = {{position.x - r, position.y - r}, {position.x + r, position.y + r}}, force = 'neutral', name = ore_name})) do
        count = count + e.amount
    end
    return count
end

local function draw_town_spawn(player_name)
    local townytable = Table.get_table()
    local market = townytable.town_centers[player_name].market
    local position = market.position
    local surface = market.surface

    local area = {{position.x - (town_radius + 1), position.y - (town_radius + 1)}, {position.x + (town_radius + 1), position.y + (town_radius + 1)}}

    for _, e in pairs(surface.find_entities_filtered({area = area, force = 'neutral'})) do
        if not clear_blacklist_types[e.type] then
            e.destroy()
        end
    end

    for _, vector in pairs(gate_vectors_horizontal) do
        local p = {position.x + vector[1], position.y + vector[2]}
        p = surface.find_non_colliding_position('gate', p, 64, 1)
        if p then
            surface.create_entity({name = 'gate', position = p, force = player_name, direction = 2})
        end
    end
    for _, vector in pairs(gate_vectors_vertical) do
        local p = {position.x + vector[1], position.y + vector[2]}
        p = surface.find_non_colliding_position('gate', p, 64, 1)
        if p then
            surface.create_entity({name = 'gate', position = p, force = player_name, direction = 0})
        end
    end

    for _, vector in pairs(town_wall_vectors) do
        local p = {position.x + vector[1], position.y + vector[2]}
        p = surface.find_non_colliding_position('stone-wall', p, 64, 1)
        if p then
            surface.create_entity({name = 'stone-wall', position = p, force = player_name})
        end
    end

    local ores = {'iron-ore', 'copper-ore', 'stone', 'coal'}
    table.shuffle_table(ores)

    for i = 1, 4, 1 do
        if count_nearby_ore(surface, position, ores[i]) < 200000 then
            for _, vector in pairs(resource_vectors[i]) do
                local p = {position.x + vector[1], position.y + vector[2]}
                p = surface.find_non_colliding_position(ores[i], p, 64, 1)
                if p then
                    surface.create_entity({name = ores[i], position = p, amount = ore_amount})
                end
            end
        end
    end

    for _, item_stack in pairs(starter_supplies) do
        local m1 = -8 + math_random(0, 16)
        local m2 = -8 + math_random(0, 16)
        local p = {position.x + m1, position.y + m2}
        p = surface.find_non_colliding_position('wooden-chest', p, 64, 1)
        if p then
            local e = surface.create_entity({name = 'wooden-chest', position = p, force = player_name})
            local inventory = e.get_inventory(defines.inventory.chest)
            inventory.insert(item_stack)
        end
    end

    local vector_indexes = {1, 2, 3, 4}
    table.shuffle_table(vector_indexes)

    local tree = 'tree-0' .. math_random(1, 9)
    for _, vector in pairs(additional_resource_vectors[vector_indexes[1]]) do
        if math_random(1, 6) == 1 then
            local p = {position.x + vector[1], position.y + vector[2]}
            p = surface.find_non_colliding_position(tree, p, 64, 1)
            if p then
                surface.create_entity({name = tree, position = p})
            end
        end
    end

    local area = {{position.x - town_radius * 1.5, position.y - town_radius * 1.5}, {position.x + town_radius * 1.5, position.y + town_radius * 1.5}}
    if surface.count_tiles_filtered({name = {'water', 'deepwater'}, area = area}) < 8 then
        for _, vector in pairs(additional_resource_vectors[vector_indexes[2]]) do
            local p = {position.x + vector[1], position.y + vector[2]}
            if surface.get_tile(p).name ~= 'out-of-map' then
                surface.set_tiles({{name = 'water', position = p}})
            end
        end
    end
    if count_nearby_ore(surface, position, 'uranium-ore') < 100000 then
        for _, vector in pairs(additional_resource_vectors[vector_indexes[3]]) do
            local p = {position.x + vector[1], position.y + vector[2]}
            p = surface.find_non_colliding_position('uranium-ore', p, 64, 1)
            if p then
                surface.create_entity({name = 'uranium-ore', position = p, amount = ore_amount * 2})
            end
        end
    end
    local vectors = additional_resource_vectors[vector_indexes[4]]
    for _ = 1, 3, 1 do
        local vector = vectors[math_random(1, #vectors)]
        local p = {position.x + vector[1], position.y + vector[2]}
        p = surface.find_non_colliding_position('crude-oil', p, 64, 1)
        if p then
            surface.create_entity({name = 'crude-oil', position = p, amount = 500000})
        end
    end
end

local function flying_text(surface, position, text, color)
    surface.create_entity(
        {
            name = 'flying-text',
            position = {position.x, position.y - 0.5},
            text = text,
            color = color
        }
    )
end

local function is_valid_location(surface, entity)
    local townytable = Table.get_table()
    for _, vector in pairs(market_collide_vectors) do
        local p = {entity.position.x + vector[1], entity.position.y + vector[2]}
        if not surface.can_place_entity({name = 'iron-chest', position = p}) then
            flying_text(surface, entity.position, {'modules_towny.message_error_obstructed'}, {r = 0.77, g = 0.0, b = 0.0})
            return
        end
    end

    if townytable.size_of_town_centers > 48 then
        flying_text(surface, entity.position, {'modules_towny.message_error_toomany'}, {r = 0.77, g = 0.0, b = 0.0})
        return
    end

    if entity.position.x ^ 2 + entity.position.y ^ 2 < square_min_distance_to_spawn then
        flying_text(surface, entity.position, {'modules_towny.message_error_close_spawn_town'}, {r = 0.77, g = 0.0, b = 0.0})
        return
    end

    local area = {
        {entity.position.x - radius_between_towns, entity.position.y - radius_between_towns},
        {entity.position.x + radius_between_towns, entity.position.y + radius_between_towns}
    }
    if surface.count_entities_filtered({area = area, name = 'market'}) > 0 then
        flying_text(surface, entity.position, {'modules_towny.message_error_close_other_town'}, {r = 0.77, g = 0.0, b = 0.0})
        return
    end

    local area = {{entity.position.x - town_radius, entity.position.y - town_radius}, {entity.position.x + town_radius, entity.position.y + town_radius}}
    local count = 0
    for _, e in pairs(surface.find_entities_filtered({area = area})) do
        if e.force.index ~= 3 then
            count = count + 1
        end
    end

    if count > 2 then
        flying_text(surface, entity.position, {'modules_towny.message_error_neutral_entities'}, {r = 0.77, g = 0.0, b = 0.0})
        return
    end

    return true
end

function Public.set_market_health(entity, final_damage_amount)
    local townytable = Table.get_table()
    local town_center = townytable.town_centers[entity.force.name]
    town_center.health = math_floor(town_center.health - final_damage_amount)
    if town_center.health > town_center.max_health then
        town_center.health = town_center.max_health
    end
    local m = town_center.health / town_center.max_health
    entity.health = 150 * m
    rendering.set_text(town_center.health_text, 'HP: ' .. town_center.health .. ' / ' .. town_center.max_health)
end

local function is_color_used(color, town_centers)
    for k, center in pairs(town_centers) do
        if center.color then
            if center.color.r == color.r and center.color.g == color.g and center.color.b == color.b then
                return true
            end
        end
    end
end

local function get_color()
    local townytable = Table.get_table()
    local town_centers = townytable.town_centers
    local c

    local shuffle_index = {}
    for i = 1, #colors, 1 do
        shuffle_index[i] = i
    end
    table.shuffle_table(shuffle_index)

    for i = 1, #colors, 1 do
        c = {r = colors[shuffle_index[i]][1], g = colors[shuffle_index[i]][2], b = colors[shuffle_index[i]][3]}
        if not is_color_used(c, town_centers) then
            return c
        end
    end

    return c
end

function Public.found(event)
    local townytable = Table.get_table()
    local entity = event.created_entity
    if entity.force.index ~= 1 then
        return
    end
    if entity.name ~= 'stone-furnace' then
        return
    end

    local player = game.players[event.player_index]
    local player_name = tostring(player.name)

    if game.forces[player_name] then
        return
    end
    if not townytable.town_buttons[player.index] then
        return
    end

    local surface = entity.surface

    if townytable.cooldowns[player.index] then
        if game.tick < townytable.cooldowns[player.index] then
            flying_text(
                surface,
                entity.position,
                {'modules_towny.message_error_cooldown', math.ceil((townytable.cooldowns[player.index] - game.tick) / 3600)},
                {r = 0.77, g = 0.0, b = 0.0}
            )
            player.insert({name = 'stone-furnace', count = 1})
            entity.destroy()
            return true
        end
    end

    if not is_valid_location(surface, entity) then
        player.insert({name = 'stone-furnace', count = 1})
        entity.destroy()
        return true
    end

    Team.add_new_force(player_name)

    townytable.town_centers[player_name] = {}
    local town_center = townytable.town_centers[player_name]
    town_center.market = surface.create_entity({name = 'market', position = entity.position, force = player_name})
    town_center.chunk_position = {math.floor(town_center.market.position.x / 32), math.floor(town_center.market.position.y / 32)}
    town_center.max_health = 1000
    town_center.health = town_center.max_health
    town_center.color = get_color()
    town_center.research_counter = 1
    town_center.upgrades = {}
    town_center.upgrades.mining_prod = 0

    town_center.health_text =
        rendering.draw_text {
        text = 'HP: ' .. town_center.health .. ' / ' .. town_center.max_health,
        surface = surface,
        target = town_center.market,
        target_offset = {0, -2.5},
        color = {200, 200, 200},
        scale = 1.00,
        font = 'default-game',
        alignment = 'center',
        scale_with_zoom = false
    }

    town_center.town_caption =
        rendering.draw_text {
        text = player.name .. "'s Town",
        surface = surface,
        target = town_center.market,
        target_offset = {0, -3.25},
        color = town_center.color,
        scale = 1.30,
        font = 'default-game',
        alignment = 'center',
        scale_with_zoom = false
    }

    townytable.size_of_town_centers = townytable.size_of_town_centers + 1

    entity.destroy()

    draw_town_spawn(player_name)

    Team.add_player_to_town(player, town_center)
    Team.add_chart_tag(game.forces.player, town_center.market)

    local force = player.force
    force.set_spawn_position({x = town_center.market.position.x, y = town_center.market.position.y + 4}, surface)

    townytable.cooldowns[player.index] = game.tick + 3600 * 15

    townytable.town_buttons[player.index] = false

    game.print({'modules_towny.message_town_found', player.name}, {255, 255, 0})
    return true
end

return Public