-- First combat slice. Wave composition and in-game sequencing still require a live playtest.
local Encounter = require("mission_ember.encounter")

local NAMES = {
    "LANDING_MERCURY_ANCHOR_SQUAD",
    "LANDING_MERCURY_SUPPORT_A_SQUAD",
    "LANDING_MERCURY_SUPPORT_B_SQUAD",
    "LANDING_MERCURY_SUPPORT_C_SQUAD",
    "LANDING_MERCURY_RANGED_A_SQUAD",
    "LANDING_MERCURY_RANGED_B_SQUAD",
    "LANDING_MERCURY_BONUS_SUPPORT_A_SQUAD",
    "LANDING_MERCURY_BONUS_SUPPORT_B_SQUAD",
    "LANDING_MERCURY_BONUS_SUPPORT_C_SQUAD",
    "LANDING_MERCURY_BONUS_SUPPORT_D_SQUAD",
}

local BRIDGE_DEVICES = {
    "POWERHOUSE_BRIDGE_ARM_MERCURY_DEVICE",
    "POWERHOUSE_BRIDGE_GEAR_MERCURY_BOTTOM_DEVICE",
    "POWERHOUSE_BRIDGE_GEAR_MERCURY_TOP_DEVICE",
    "POWERHOUSE_BRIDGE_ARM_SUN_DEVICE",
    "POWERHOUSE_BRIDGE_GEAR_SUN_BOTTOM_DEVICE",
    "POWERHOUSE_BRIDGE_GEAR_SUN_TOP_DEVICE",
}

local function required(value, name)
    assert(value ~= nil, "mission_ember: missing SDK binding " .. name
           .. "; regenerate the SDK with the scenario-scoped squad linker")
    return value
end

return function(mission)
    local squads = {}
    for index, name in ipairs(NAMES) do
        squads[index] = {
            name = name,
            id = required(mission.Squad[name], "Squad." .. name),
            sensor = required(mission.Slot[name], "Slot." .. name),
        }
    end
    local entry = required(mission.states.STATE_80B3C09E_0008_0000_80B3C09C, "powerhouse state")
    local bridge = {}
    for index, name in ipairs(BRIDGE_DEVICES) do
        bridge[index] = required(mission.Slot[name], "Slot." .. name)
    end
    local bridge_objective = required(mission.Slot.EMBER_POWERHOUSE_BRIDGE_OBJECTIVE,
                                      "bridge objective")
    local directive = required(mission.Slot.M_DIRECTIVE_SENSOR_80B3C90A, "directive sensor")
    local clear = required(mission.Directive.FIND_AND_DISABLE_THE_ALMIGHTY_S_WEAPONS_A70DA4A6,
                           "landing clear directive")
    local controls = required(mission.Directive.FIND_AND_DISABLE_THE_ALMIGHTY_S_WEAPONS_2700C0C5,
                              "bridge controls directive")
    local encounter = Encounter.new("ember.landing", squads, function(context)
        context:slot(directive):set_directive{directive = controls}
        -- The bridge must wait for its authored interaction. Combat clear does not open it.
    end)

    return {
        initial_state = entry,
        region = entry.region_index,
        initialize = function(context)
            -- on_start runs after the initial-state roster has reached the transport.
            -- Do not repeat this on region transit or on_load: those are not mission resets.
            for _, device in ipairs(bridge) do
                context:slot(device):transition{
                    transition = context.sdk.device_transitions.close, snap = true,
                }
            end
            context:slot(bridge_objective):reset_objectives{}
        end,
        enter = function(context, state)
            if encounter:phase(state) ~= 0 then
                return
            end
            encounter:enter(context, state)
            context:slot(directive):set_directive{directive = clear}
        end,
        on_squad_state = function(context, state, event)
            encounter:on_squad_state(context, state, event)
        end,
    }
end