-- informatron_integration.lua

local function create_class_selection_tab(player)
  local informatron = player.gui.screen.informatron
  local tab = informatron.add {
    type = "tab",
    name = "class_selection_tab",
    caption = "Class Selection",
  }

  local frame = informatron.add {
    type = "frame",
    name = "class_selection_frame",
    caption = "Choose Your Class",
    direction = "vertical",
    style = "inside_shallow_frame_with_padding",
  }
  frame.style.maximal_height = 200

  frame.add {
    type = "button",
    name = "ministry_of_transportation_button",
    caption = "Ministry of Transportation",
  }

  frame.add {
    type = "button",
    name = "ministry_of_defence_button",
    caption = "Ministry of Defence",
  }

  tab.content = frame
end

script.on_event(defines.events.on_player_joined_game, function(event)
  local player = game.players[event.player_index]
  create_class_selection_tab(player)
end)