--[[
Sprint mod for Minetest by GunshipPenguin

To the extent possible under law, the author(s)
have dedicated all copyright and related and neighboring rights
to this software to the public domain worldwide. This software is
distributed without any warranty.
]]

sprint.players = {}
local staminaHud = {}

-- Lil' helping functions
sprint.set_maxstamina = function(pname, mstamina)
	if sprint.players[pname] and mstamina > 0 then
		sprint.players[pname].maxStamina = mstamina
		if sprint.players[pname].stamina > sprint.players[pname].maxStamina then
			sprint.players[pname].stamina = sprint.players[pname].maxStamina
		end
		hb.change_hudbar(minetest.get_player_by_name(pname), "sprint", sprint.players[pname].stamina, sprint.players[pname].maxStamina)
	end
end

sprint.get_maxstamina = function(pname)
	if sprint.players[pname] then
		return sprint.players[pname].maxStamina
	end
end

sprint.increase_maxstamina = function(pname, sincrease)
	local stamina = sprint.get_maxstamina(pname)
	if stamina then
		sprint.set_maxstamina(pname, stamina + sincrease)
	end
end

sprint.decrease_maxstamina = function(pname, sdicrease)
	local stamina = sprint.get_maxstamina(pname)
	if stamina then
		sprint.set_maxstamina(pname, stamina - sdicrease)
	end
end
-- When you write shit english, don't write shit code.
-- Writing it in your native language would be less of a pain.
--   - gravgun
sprint.dicrease_maxstamina = sprint.decrease_maxstamina

sprint.set_default_maxstamina = function(pname)
	sprint.set_maxstamina(pname, SPRINT_STAMINA)
end



minetest.register_on_joinplayer(function(player)
	local playerName = player:get_player_name()

	sprint.players[playerName] = {
		sprinting = false,
		timeOut = 0,
		stamina = SPRINT_STAMINA,
		shouldSprint = false,
		maxStamina = SPRINT_STAMINA
	}
	if SPRINT_HUDBARS_USED then
		hb.init_hudbar(player, "sprint")
	else
		sprint.players[playerName].hud = player:hud_add({
			hud_elem_type = "statbar",
			position = {x=0.5,y=1},
			size = {x=24, y=24},
			text = "stamina.png",
			number = 20,
			alignment = {x=0,y=1},
			offset = {x=-320, y=-186},
			}
		)
	end
end)
minetest.register_on_leaveplayer(function(player)
	local playerName = player:get_player_name()
	sprint.players[playerName] = nil
end)
local gameTime = 0
minetest.register_globalstep(function(dtime)
	--Get the gametime
	gameTime = gameTime + dtime

	--Loop through all connected sprint.players
	for playerName,playerInfo in pairs(sprint.players) do
		local player = minetest.get_player_by_name(playerName)
		if player ~= nil then
			--Check if the player should be sprinting
			if player:get_player_control()["aux1"] and player:get_player_control()["up"] then
				sprint.players[playerName]["shouldSprint"] = true
			else
				sprint.players[playerName]["shouldSprint"] = false
			end
			--Stop sprinting if the player is pressing the LMB or RMB
			if player:get_player_control()["LMB"] or player:get_player_control()["RMB"] then
				setSprinting(playerName, false)
				playerInfo["timeOut"] = 3
			end

			if gameTime > 0.4 then
				local pos = player:getpos()
				-- From playerplus :
				-- am I near a spiky cactus?
				pos.y = pos.y + 0.1
				if minetest.find_node_near(pos, 1, "default:cactus_spiky") and player:get_hp() > 0 then
					player:set_hp(player:get_hp()-1)
				end

				--If the player is sprinting, create particles behind him/her
				if playerInfo["sprinting"] == true then
					local numParticles = math.random(1, 2)
					local playerPos = player:getpos()
					local playerNode = minetest.get_node({x=playerPos["x"], y=playerPos["y"]-1, z=playerPos["z"]})
					if playerNode["name"] ~= "air" then
						for i=1, numParticles, 1 do
							minetest.add_particle({
								pos = {x=playerPos["x"]+math.random(-1,1)*math.random()/2,y=playerPos["y"]+0.1,z=playerPos["z"]+math.random(-1,1)*math.random()/2},
								velocity = {x=0, y=5, z=0},
								acc = {x=0, y=-13, z=0},
								expirationtime = math.random(),
								size = math.random()+0.5,
								collisiondetection = true,
								vertical = false,
								texture = "sprint_particle.png",
							})
						end
					end
				end
			end
			--Adjust player states
			if sprint.players[playerName]["shouldSprint"] == true and playerInfo["timeOut"] == 0 then --Stopped
				setSprinting(playerName, true)
			elseif sprint.players[playerName]["shouldSprint"] == false then
				setSprinting(playerName, false)
			end

			if playerInfo["timeOut"] > 0 then
				playerInfo["timeOut"] = playerInfo["timeOut"] - dtime
				if playerInfo["timeOut"] < 0 then
					playerInfo["timeOut"] = 0
				end
			else
				--Lower the player's stamina by dtime if he/she is sprinting and set his/her state to 0 if stamina is zero
				if playerInfo["sprinting"] == true then
					playerInfo["stamina"] = playerInfo["stamina"] - dtime
					if playerInfo["stamina"] <= 0 then
						playerInfo["stamina"] = 0
						setSprinting(playerName, false)
						playerInfo["timeOut"] = 1
						minetest.sound_play("default_breathless",{object=player})
					end
				end
			end

			--Increase player's stamina if he/she is not sprinting and his/her stamina is less than SPRINT_STAMINA
			if playerInfo["sprinting"] == false and playerInfo["stamina"] < playerInfo["maxStamina"] then
				playerInfo["stamina"] = playerInfo["stamina"] + dtime
			end
			-- Cap stamina at SPRINT_STAMINA
			if playerInfo["stamina"] > playerInfo["maxStamina"] then
				playerInfo["stamina"] = playerInfo["maxStamina"]
			end

			--Update the sprint.players's hud sprint stamina bar

			if SPRINT_HUDBARS_USED then
				hb.change_hudbar(player, "sprint", playerInfo["stamina"])
			else
				local numBars = (playerInfo["stamina"]/playerInfo["maxStamina"])*20
				player:hud_change(playerInfo["hud"], "number", numBars)
			end
		end
	end
	if gameTime > 0.4 then
		gameTime = 0
	end
end)

function setSprinting(playerName, sprinting) --Sets the state of a player (0=stopped/moving, 1=sprinting)
	local player = minetest.get_player_by_name(playerName)
	if sprint.players[playerName] then
		sprint.players[playerName]["sprinting"] = sprinting
		if sprinting == true then
			player_physics.set_stats(player, "sprint", {speed=SPRINT_SPEED, jump=SPRINT_JUMP})
		elseif sprinting == false then
			player_physics.set_stats(player, "sprint", {speed=0, jump=0})
		end
		return true
	end
	return false
end