-- class_perks.lua

local ClassPerks = {}

function ClassPerks.apply_perks(player)
  local class_name = global.player_classes and global.player_classes[player.index]
  if class_name then
    local class = require(class_name)
    if class.perk == "logistic_cost_reduction" then
      player.print("Ministry of Transportation: Logistic costs reduced!")
      class.apply_perks(player)
    elseif class.perk == "combat_cost_reduction" then
      player.print("Ministry of Defence: Combat costs reduced!")
      class.apply_perks(player)
    end
  end
end

-- Register the command handler for choosing a class
commands.add_command("choose_class", "Choose your class (MinistryOfTransportation or MinistryOfDefence)", function(event)
  local player = game.get_player(event.player_index)
  local class_name = event.parameter

  if class_name == "MinistryOfTransportation" or class_name == "MinistryOfDefence" then
    global.player_classes = global.player_classes or {}
    global.player_classes[player.index] = class_name:lower()
    ClassPerks.apply_perks(player)
    player.print("You have chosen the " .. class_name .. "!")
  else
    player.print("Invalid class name. Usage: /choose_class MinistryOfTransportation|MinistryOfDefence")
  end
end)

-- Initialize the global table if it doesn't exist
script.on_init(function()
  global.player_classes = global.player_classes or {}
end)

return ClassPerks