require "defines"
require "story"

game.on_init(function()
  global.story = story_init()
end)

game.on_mod_configuration_changed(function()
  if global.story == nil then
    global.story = story_init()
  end
end)

-- The story contents
story_table =
{{
  {
    action =
    function()
      game.show_message_dialog{text = {"msg-introduction"}}
      game.show_message_dialog{text = {"msg-ask-chest"}}
    end
  },
  {
    action =
    function()
      local player = game.get_player(1)
      if player ~= nil then
        player.gui.top.add{type = "button", name="button_use_chest", caption={"button-use-chest"}}
        player.gui.top.add{type = "button", name="button_no_chest", caption={"button-no-chest"}}
      end
    end
  },
  {
    condition =
    function(event)
      if event.name == defines.events.on_gui_click then
        local player = game.get_player(event.player_index)
        if event.element.name == "button_use_chest" then
          build_starting_chest(player)
        elseif event.element.name ~= "button_no_chest" then
          return false
        end
        player.gui.top.button_use_chest.destroy()
        player.gui.top.button_no_chest.destroy()
        return true
      end
      return false
    end,
    action = function()
      game.show_message_dialog{text = {"msg-ask-technologies"}}
    end
  },
  {
    action =
    function()
      local player = game.get_player(1)
      if player ~= nil then
        player.gui.top.add{type = "button", name = "button_technologies_researched", caption = {"button-technologies-researched"}}
        player.gui.top.add{type = "button", name = "button_technologies_normal", caption = {"button-technologies-normal"}}
      end
    end
  },
  {
    condition =
    function(event)
      if event.name == defines.events.on_gui_click then
        local player = game.get_player(event.player_index)
        if event.element.name == "button_technologies_researched" then
          player.force.research_all_technologies()
        elseif event.element.name ~= "button_technologies_normal" then
          return false
        end
        player.gui.top.button_technologies_researched.destroy()
        player.gui.top.button_technologies_normal.destroy()
        return true
      end
      return false
    end,
    action = function()
      action = function() game.show_message_dialog{text = {"msg-start-play"}} end
    end
  },
  -- start the game
  {
    -- After this point we don't need events any more, so we just disable them all
    action =
    function()
      game.on_event(defines.events, nil)
    end
  },
  {
    condition = function() return false end,
    action = function() end
  }
}}

story_init_helpers(story_table)

-- We need to update the story after every event except on_player_created, so the order
-- of the tw following definitions is important.

game.on_event(defines.events, function(event)
  story_udpate(global.story, event, "")
end)

game.on_event(defines.events.on_player_created, function(event)
  local player = game.get_player(event.player_index)
  local character = player.character
  player.character = nil
  if character then character.destroy() end
end)

function build_starting_chest(player)
  local chest_position = player.surface.find_non_colliding_position("steel-chest", player.position, 0, 0.1)
  local chest = player.surface.create_entity{name="steel-chest", position=chest_position, force = game.forces.player}
  chest.insert({name="wood", count="50"})
  chest.insert({name="coal", count="100"})
  chest.insert({name="stone", count="100"})
  chest.insert({name="iron-plate", count="400"})
  chest.insert({name="copper-plate", count="400"})
  chest.insert({name="steel-plate", count="100"})
  chest.insert({name="iron-gear-wheel", count="200"})
  chest.insert({name="electronic-circuit", count="200"})
  chest.insert({name="advanced-circuit", count="200"})
  chest.insert({name="offshore-pump", count="20"})
  chest.insert({name="pipe", count="50"})
  chest.insert({name="boiler", count="50"})
  chest.insert({name="basic-mining-drill", count="50"})
  chest.insert({name="steam-engine", count="10"})
  chest.insert({name="stone-furnace", count="50"})
  chest.insert({name="basic-transport-belt", count="100"})
  chest.insert({name="fast-transport-belt", count="50"})
  chest.insert({name="express-transport-belt", count="50"})
  chest.insert({name="basic-inserter", count="50"})
  chest.insert({name="fast-inserter", count="50"})
  chest.insert({name="long-handed-inserter", count="50"})
  chest.insert({name="smart-inserter", count="50"})
  chest.insert({name="small-electric-pole", count="50"})
  chest.insert({name="assembling-machine-1", count="50"})
  chest.insert({name="assembling-machine-2", count="50"})
  chest.insert({name="straight-rail", count="150"})
  chest.insert({name="curved-rail", count="50"})
  chest.insert({name="train-stop", count="10"})
  chest.insert({name="rail-signal", count="50"})
  chest.insert({name="diesel-locomotive", count="5"})
  chest.insert({name="cargo-wagon", count="5"})
end

game.on_event(defines.events.on_rocket_launched, function(event)
  local force = event.rocket.force
  if event.rocket.get_item_count("satellite") > 0 then
    if global.satellite_sent == nil then
      global.satellite_sent = {}
    end
    if global.satellite_sent[force.name] == nil then
      game.set_game_state{game_finished=true, player_won=true, can_continue=true}
      global.satellite_sent[force.name] = 1
    else
      global.satellite_sent[force.name] = global.satellite_sent[force.name] + 1
    end
    for index, player in pairs(force.players) do
      if player.gui.left.rocket_score == nil then
        local frame = player.gui.left.add{name = "rocket_score", type = "frame", direction = "horizontal", caption={"score"}}
        frame.add{name="rocket_count_label", type = "label", caption={"", {"rockets-sent"}, ""}}
        frame.add{name="rocket_count", type = "label", caption="1"}
      else
        player.gui.left.rocket_score.rocket_count.caption = tostring(global.satellite_sent[force.name])
      end
    end
  else
    if (#game.players <= 1) then
      game.show_message_dialog{text = {"gui-rocket-silo.rocket-launched-without-satellite"}}
    else
      for index, player in pairs(force.players) do
        player.print({"gui-rocket-silo.rocket-launched-without-satellite"})
      end
    end
  end
end)