-- Vars

local speed         = tonumber(minetest.settings:get ("sprint_speed")) or 1.3
local jump          = tonumber(minetest.settings:get ("sprint_jump")) or 1.1
local key           = minetest.settings:get ("sprint_key") or "Use"
local dir           = minetest.settings:get_bool("sprint_forward_only") ~= false
local particles     = tonumber(minetest.settings:get ("sprint_particles")) or 2
local stamina       = minetest.settings:get_bool("sprint_stamina") ~= false
local stamina_drain = tonumber(minetest.settings:get ("sprint_stamina_drain")) or 2
local replenish     = tonumber(minetest.settings:get ("sprint_stamina_replenish")) or 2
local starve        = minetest.settings:get_bool("sprint_starve") ~= false
local starve_drain  = tonumber(minetest.settings:get ("sprint_starve_drain")) or 0.5
local starve_limit  = tonumber(minetest.settings:get ("sprint_starve_limit")) or 6
local breath        = minetest.settings:get_bool("sprint_breath") ~= false
local breath_drain  = tonumber(minetest.settings:get ("sprint_breath_drain")) or 1
local autohide      = minetest.settings:get_bool("hudbars_autohide_stamina") ~= false

local sprint_timer_step = 0.5
local sprint_timer = 0
local stamina_timer = 0
local breath_timer = 0

local hudbars = minetest.get_modpath("hudbars") or false
local monoids = minetest.get_modpath("player_monoids") or false
local starve
if minetest.get_modpath("hbhunger") then
  starve = "hbhunger"
elseif minetest.get_modpath("hunger_ng") then
  starve = "hunger_ng"
  --starve_drain = starve_drain * -1
else
  starve = false
end
if minetest.settings:get_bool("creative_mode") then
  starve = false
end
-- Functions

local function start_sprint(player)
  if player:get_attribute("hbsprint:sprinting") == "false" then
    if monoids then
      player_monoids.speed:add_change(player, speed, "hbsprint:speed")
      player_monoids.jump:add_change(player, jump, "hbsprint:jump")
    else
      player:set_physics_override({speed = speed, jump = jump})
    end
  end
end

local function stop_sprint(player)
  if player:get_attribute("hbsprint:sprinting") == "true" then
    if monoids then
      player_monoids.speed:del_change(player, "hbsprint:speed")
      player_monoids.jump:del_change(player, "hbsprint:jump")
    else
      player:set_physics_override({speed = 1, jump = 1})
    end
  end
end

local function drain_stamina(player)
  local player_stamina = tonumber(player:get_attribute("hbsprint:stamina"))
  if player_stamina > 0 then
    player:set_attribute("hbsprint:stamina", player_stamina - stamina_drain)
  end
  if hudbars then
    if autohide and player_stamina < 20 then hb.unhide_hudbar(player, "stamina") end
    hb.change_hudbar(player, "stamina", player_stamina)
  end
end

local function replenish_stamina(player)
  local player_stamina = tonumber(player:get_attribute("hbsprint:stamina"))
  if player_stamina < 20 then
    player:set_attribute("hbsprint:stamina", player_stamina + stamina_drain)
  end
  if hudbars then
    hb.change_hudbar(player, "stamina", player_stamina)
    if autohide and player_stamina == 20 then hb.hide_hudbar(player, "stamina") end
  end
end

local function drain_hunger(player, hunger, name)
  if hunger > 0 then
    local newhunger = hunger - starve_drain
    if starve == "hbhunger" then
      hbhunger.hunger[name] = newhunger
      hbhunger.set_hunger_raw(player)
    elseif starve == "hunger_ng" then
      player:set_attribute("hunger_ng:hunger", newhunger)
    end
  end
end

local function drain_breath(player)
  local player_breath = player:get_breath()
  if player_breath < 11 then
    player_breath = player_breath - breath_drain
    if player_breath > 0 then
      player:set_breath(player_breath)
    end
  end
end

local function create_particles(player, name, pos, ground)
  if ground and ground.name ~= "air" and ground.name ~= "ignore" then
    local def = minetest.registered_nodes[ground.name]
    local tile = def.tiles[1] or def.inventory_image or ""
    if type(tile) == "string" then
      for i = 1, particles do
        minetest.add_particle({
          pos = {x = pos.x + math.random(-1,1) * math.random() / 2, y = pos.y + 0.1, z = pos.z + math.random(-1,1) * math.random() / 2},
          velocity = {x = 0, y = 5, z = 0},
          acceleration = {x = 0, y = -13, z = 0},
          expirationtime = math.random(),
          size = math.random() + 0.5,
          vertical = false,
          texture = tile,
        })
      end
    end
  end
end

-- Registrations

if minetest.get_modpath("hudbars") ~= nil and stamina then
  hb.register_hudbar(
    "stamina",
    0xFFFFFF,
    "Stamina",
    {
      bar = "sprint_stamina_bar.png",
      icon = "sprint_stamina_icon.png",
      bgicon = "sprint_stamina_bgicon.png"
    },
    20,
    20,
    autohide)
end

minetest.register_on_joinplayer(function(player)
  if hudbars and stamina then hb.init_hudbar(player, "stamina", 20, 20, autohide) end
  player:set_attribute("hbsprint:stamina", 20)
end)

minetest.register_globalstep(function(dtime)
  sprint_timer = sprint_timer + dtime
  stamina_timer = stamina_timer + dtime
  breath_timer = breath_timer + dtime
  if sprint_timer >= sprint_timer_step then
    for _,player in ipairs(minetest.get_connected_players()) do
      local ctrl = player:get_player_control()
      local key_press = false
      if key == "Use" and dir then
        key_press = ctrl.aux1 and ctrl.up and not ctrl.left and not ctrl.right
      elseif key == "Use" and not dir then
        key_press = ctrl.aux1
      end
      -- if key == "W" and dir then
      --   key_press = ctrl.aux1 and ctrl.up or key_press and ctrl.up
      -- elseif key == "W" then
      --   key_press = ctrl.aux1 or key_press and key_tap
      -- end

      if key_press then
        local name = player:get_player_name()
        local hunger = 30
        local pos = player:get_pos()
        local ground = minetest.get_node_or_nil({x=pos.x, y=pos.y-1, z=pos.z})
        local walkable = false
        local player_stamina = tonumber(player:get_attribute("hbsprint:stamina"))
        if starve == "hbhunger" then
          hunger = tonumber(hbhunger.hunger[name])
        elseif starve == "hunger_ng" then
          hunger = tonumber(player:get_attribute("hunger_ng:hunger"))
        end
        if ground ~= nil then
          local ground_def = minetest.registered_nodes[ground.name]
          if ground_def then
            walkable = minetest.registered_nodes[ground.name].walkable
          end
        end
        if player_stamina > 0 and hunger > starve_limit and walkable then
          start_sprint(player)
          player:set_attribute("hbsprint:sprinting", "true")
          if stamina then drain_stamina(player) end
          if starve then drain_hunger(player, hunger, name) end
          if breath then
            if breath_timer >= 2 then
              drain_breath(player)
              breath_timer = 0
            end
          end
          if particles then create_particles(player, name, pos, ground) end
        else
          stop_sprint(player)
          player:set_attribute("hbsprint:sprinting", "false")
        end
      else
        stop_sprint(player)
        player:set_attribute("hbsprint:sprinting", "false")
        if stamina_timer >= replenish then
          if stamina then replenish_stamina(player) end
          stamina_timer = 0
        end
      end
    end
    sprint_timer = 0
  end
end)

minetest.log("action", "[hbsprint] loaded.")