local api_base_url = "https://ipod-2to6magyna-uc.a.run.app/"
local version = "2.1"

local width, height = term.getSize()
local tab = 1

local waiting_for_input = false
local last_search = nil
local last_search_url = nil
local search_results = nil
local search_error = false
local in_search_result = false
local clicked_result = nil

local playing = false
local queue = {}
local now_playing = nil
local looping = 0
local volume = 1.5

local playing_id = nil
local last_download_url = nil
local playing_status = 0
local is_loading = false
local is_error = false;

local player_handle = nil
local start = nil
local pcm = nil
local size = nil
local decoder = require "cc.audio.dfpwm".make_decoder()
local needs_next_chunk = 0
local buffer

local speakers = { peripheral.find("speaker") }
if #speakers == 0 then
	error("No speakers attached. You need to connect a speaker to this computer. If this is an Advanced Noisy Pocket Computer, then this is a bug, and you should try restarting your Minecraft game.", 0)
end

function drawHeader()
	term.setCursorPos(1,1)
	term.setBackgroundColor(colors.blue)
	term.clearLine()

	local tabs = {" Now Playing ", " Search "}

	for i=1,#tabs,1 do
		if tab == i then
			term.setTextColor(colors.white)
			term.setBackgroundColor(colors.lightBlue)
		else
			term.setTextColor(colors.lightGray)
			term.setBackgroundColor(colors.blue)
		end

		term.setCursorPos((math.floor((width/#tabs)*(i-0.5)))-math.ceil(#tabs[i]/2)+1, 1)
		term.write(tabs[i])
	end
end

function redrawScreen()
	if waiting_for_input then
		return
	end

	term.setCursorBlink(false)
	term.setBackgroundColor(colors.black)
	term.clear()

	drawHeader()

	if tab == 1 then
		drawNowPlaying()
	elseif tab == 2 then
		drawSearch()
	end
end

function drawNowPlaying()
	-- Status Info
	if now_playing ~= nil then
		term.setBackgroundColor(colors.black)
		term.setTextColor(colors.lightBlue)
		term.setCursorPos(2,3)
		term.write(string.sub(now_playing.name, 1, width - 2))
		term.setTextColor(colors.white)
		term.setCursorPos(2,4)
		term.write(string.sub(now_playing.artist, 1, width - 2))
	else
		term.setBackgroundColor(colors.black)
		term.setTextColor(colors.gray)
		term.setCursorPos(2,3)
		term.write("Not playing")
	end

	if is_loading == true then
		term.setTextColor(colors.cyan)
		term.setBackgroundColor(colors.black)
		term.setCursorPos(2,5)
		term.write("Loading...")
	elseif is_error == true then
		term.setTextColor(colors.red)
		term.setBackgroundColor(colors.black)
		term.setCursorPos(2,5)
		term.write("Network error")
	end

	-- Control Buttons
	local btn_bg = colors.gray
	local btn_fg = colors.white
	local btn_active_bg = colors.lightBlue

	-- Play/Stop Button
	if playing then
		term.setBackgroundColor(btn_active_bg)
		term.setTextColor(colors.white)
		term.setCursorPos(2, 6)
		term.write(" Stop ")
	else
		if now_playing ~= nil or #queue > 0 then
			term.setBackgroundColor(btn_bg)
			term.setTextColor(btn_fg)
		else
			term.setBackgroundColor(colors.black)
			term.setTextColor(colors.gray)
		end
		term.setCursorPos(2, 6)
		term.write(" Play ")
	end

	-- Skip Button
	if now_playing ~= nil or #queue > 0 then
		term.setBackgroundColor(btn_bg)
		term.setTextColor(btn_fg)
	else
		term.setBackgroundColor(colors.black)
		term.setTextColor(colors.gray)
	end
	term.setCursorPos(2 + 7, 6)
	term.write(" Skip ")

	-- Loop Button
	if looping ~= 0 then
		term.setBackgroundColor(btn_active_bg)
		term.setTextColor(colors.white)
	else
		term.setBackgroundColor(btn_bg)
		term.setTextColor(btn_fg)
	end
	term.setCursorPos(2 + 7 + 7, 6)
	if looping == 0 then
		term.write(" Loop Off ")
	elseif looping == 1 then
		term.write(" Loop Queue ")
	else
		term.write(" Loop Song ")
	end

	-- Volume Slider
	term.setCursorPos(2,8)
	term.setBackgroundColor(colors.gray)
	term.write(string.rep(" ", 24))

	local vol_width = math.floor(24 * (volume / 3) + 0.5)
	if vol_width > 0 then
		term.setCursorPos(2,8)
		term.setBackgroundColor(colors.lightBlue)
		term.write(string.rep(" ", vol_width))
	end

	term.setBackgroundColor(colors.black)
	term.setTextColor(colors.lightGray)
	term.setCursorPos(2 + 25, 8)
	term.write(math.floor(100 * (volume / 3) + 0.5) .. "%")

	-- Queue Display
	if #queue > 0 then
		term.setBackgroundColor(colors.black)
		for i=1,math.min(#queue, math.floor((height - 9) / 2)) do
			term.setTextColor(colors.white)
			term.setCursorPos(2,10 + (i-1)*2)
			term.write(string.sub(queue[i].name, 1, width - 2))
			term.setTextColor(colors.gray)
			term.setCursorPos(2,11 + (i-1)*2)
			term.write(string.sub(queue[i].artist, 1, width - 2))
		end
	end
end

function drawSearch()
	-- Search bar
	term.setBackgroundColor(colors.gray)
	term.setCursorPos(2, 3)
	term.write(string.rep(" ", width - 2))
	term.setCursorPos(2, 4)
	term.write(string.rep(" ", width - 2))
	term.setCursorPos(2, 5)
	term.write(string.rep(" ", width - 2))

	term.setBackgroundColor(colors.gray)
	term.setTextColor(colors.white)
	term.setCursorPos(3,4)

	-- Corrige o visual para telas pequenas
	if last_search then
		local display_text = last_search
		if string.len(display_text) > width - 4 then
			display_text = string.sub(display_text, 1, width - 7) .. "..."
		end
		term.write(display_text)
	else
		term.setTextColor(colors.lightGray)
		term.write("Search URL...")
	end

	-- Search results
	if search_results ~= nil then
		term.setBackgroundColor(colors.black)

		-- Bloqueia aquela propaganda do patreon caso ela apareça
		if search_results[1] and search_results[1].artist == "patreon.com/exclamator" then
			term.setTextColor(colors.red)
			term.setCursorPos(2,7)
			term.write("Playlist n. encontrada")
			term.setCursorPos(2,8)
			term.write("(Deve estar publica)")
		elseif #search_results == 0 then
			term.setTextColor(colors.red)
			term.setCursorPos(2,7)
			term.write("No results found.")
		else
			for i=1,math.min(#search_results, math.floor((height - 8) / 2)) do
				term.setTextColor(colors.lightBlue)
				term.setCursorPos(2,7 + (i-1)*2)
				term.write(string.sub(search_results[i].name, 1, width - 2))
				term.setTextColor(colors.white)
				term.setCursorPos(2,8 + (i-1)*2)
				term.write(string.sub(search_results[i].artist, 1, width - 2))
			end
		end
	else
		term.setCursorPos(2,7)
		term.setBackgroundColor(colors.black)
		if search_error == true then
			term.setTextColor(colors.red)
			term.write("Network error")
		elseif last_search_url ~= nil then
			term.setTextColor(colors.cyan)
			term.write("Searching...")
		else
			term.setCursorPos(2,7)
			term.setTextColor(colors.gray)
			term.write("Paste video or")
			term.setCursorPos(2,8)
			term.write("playlist links.")
		end
	end

	-- Assinatura do Criador
	term.setCursorPos(2, height - 1)
	term.setBackgroundColor(colors.black)
	term.setTextColor(colors.lightGray)
	term.write("Criado por ")
	term.setTextColor(colors.yellow)
	term.write("Luke Arwolf")

	-- Fullscreen song options
	if in_search_result == true then
		term.setBackgroundColor(colors.black)
		term.clear()
		drawHeader()

		term.setCursorPos(2,3)
		term.setTextColor(colors.lightBlue)
		term.write(string.sub(search_results[clicked_result].name, 1, width - 2))
		term.setCursorPos(2,4)
		term.setTextColor(colors.white)
		term.write(string.sub(search_results[clicked_result].artist, 1, width - 2))

		term.setBackgroundColor(colors.blue)
		term.setTextColor(colors.white)

		term.setCursorPos(2,6)
		term.write(" Play now ")

		term.setCursorPos(2,8)
		term.write(" Play next ")

		term.setCursorPos(2,10)
		term.write(" Add to queue ")

		term.setBackgroundColor(colors.gray)
		term.setCursorPos(2,13)
		term.write(" Cancel ")
	end
end

function uiLoop()
	redrawScreen()

	while true do
		if waiting_for_input then
			parallel.waitForAny(
				function()
					term.setCursorPos(3,4)
					term.setBackgroundColor(colors.white)
					term.setTextColor(colors.black)
					term.write(string.rep(" ", width - 4))
					term.setCursorPos(3,4)
					local input = read()

					if string.len(input) > 0 then
						last_search = input
						local query = input

						-- Remove espaços em branco
						query = query:match("^%s*(.-)%s*$")

						-- A NOVA INTELIGÊNCIA DE BUSCA:
						if string.sub(query, 1, 2) == "PL" then
							-- Se começa com PL, sabemos que é o ID de uma playlist! Montamos o link.
							query = "https://www.youtube.com/playlist?list=" .. query
						else
							local list_id = query:match("[?&]list=([a-zA-Z0-9_-]+)")
							if list_id then
								query = "https://www.youtube.com/playlist?list=" .. list_id
							else
								local vid_id = query:match("[?&]v=([a-zA-Z0-9_-]+)") or query:match("youtu%.be/([a-zA-Z0-9_-]+)")
								if vid_id then
									query = "https://www.youtube.com/watch?v=" .. vid_id
								end
							end
						end

						last_search_url = api_base_url .. "?v=" .. version .. "&search=" .. textutils.urlEncode(query)
						http.request(last_search_url)
						search_results = nil
						search_error = false
					else
						last_search = nil
						last_search_url = nil
						search_results = nil
						search_error = false
					end

					waiting_for_input = false
					os.queueEvent("redraw_screen")
				end,
				function()
					while waiting_for_input do
						local event, button, x, y = os.pullEvent("mouse_click")
						if y < 3 or y > 5 or x < 2 or x > width-1 then
							waiting_for_input = false
							os.queueEvent("redraw_screen")
							break
						end
					end
				end
			)
		else
			parallel.waitForAny(
				function()
					local event, button, x, y = os.pullEvent("mouse_click")

					if button == 1 then
						-- Tabs
						if in_search_result == false then
							if y == 1 then
								if x < width/2 then
									tab = 1
								else
									tab = 2
								end
								redrawScreen()
							end
						end

						if tab == 2 and in_search_result == false then
							-- Search box click
							if y >= 3 and y <= 5 and x >= 1 and x <= width-1 then
								term.setBackgroundColor(colors.white)
								term.setCursorPos(2, 3)
								term.write(string.rep(" ", width - 2))
								term.setCursorPos(2, 4)
								term.write(string.rep(" ", width - 2))
								term.setCursorPos(2, 5)
								term.write(string.rep(" ", width - 2))
								waiting_for_input = true
							end

							-- Search result click
							if search_results and not (search_results[1] and search_results[1].artist == "patreon.com/exclamator") then
								for i=1,math.min(#search_results, math.floor((height - 8) / 2)) do
									if y == 7 + (i-1)*2 or y == 8 + (i-1)*2 then
										term.setBackgroundColor(colors.lightBlue)
										term.setTextColor(colors.white)
										term.setCursorPos(2,7 + (i-1)*2)
										term.clearLine()
										term.write(string.sub(search_results[i].name, 1, width - 2))
										term.setCursorPos(2,8 + (i-1)*2)
										term.clearLine()
										term.write(string.sub(search_results[i].artist, 1, width - 2))
										sleep(0.1)
										in_search_result = true
										clicked_result = i
										redrawScreen()
									end
								end
							end
						elseif tab == 2 and in_search_result == true then
							-- Search result menu clicks

							if y == 6 then
								term.setBackgroundColor(colors.lightBlue)
								term.setCursorPos(2,6)
								term.write(" Play now ")
								sleep(0.1)
								in_search_result = false
								for _, speaker in ipairs(speakers) do
									speaker.stop()
									os.queueEvent("playback_stopped")
								end
								playing = true
								is_error = false
								playing_id = nil
								if search_results[clicked_result].type == "playlist" then
									now_playing = search_results[clicked_result].playlist_items[1]
									queue = {}
									if #search_results[clicked_result].playlist_items > 1 then
										for i=2, #search_results[clicked_result].playlist_items do
											table.insert(queue, search_results[clicked_result].playlist_items[i])
										end
									end
								else
									now_playing = search_results[clicked_result]
								end
								os.queueEvent("audio_update")
							end

							if y == 8 then
								term.setBackgroundColor(colors.lightBlue)
								term.setCursorPos(2,8)
								term.write(" Play next ")
								sleep(0.1)
								in_search_result = false
								if search_results[clicked_result].type == "playlist" then
									for i = #search_results[clicked_result].playlist_items, 1, -1 do
										table.insert(queue, 1, search_results[clicked_result].playlist_items[i])
									end
								else
									table.insert(queue, 1, search_results[clicked_result])
								end
								os.queueEvent("audio_update")
							end

							if y == 10 then
								term.setBackgroundColor(colors.lightBlue)
								term.setCursorPos(2,10)
								term.write(" Add to queue ")
								sleep(0.1)
								in_search_result = false
								if search_results[clicked_result].type == "playlist" then
									for i = 1, #search_results[clicked_result].playlist_items do
										table.insert(queue, search_results[clicked_result].playlist_items[i])
									end
								else
									table.insert(queue, search_results[clicked_result])
								end
								os.queueEvent("audio_update")
							end

							if y == 13 then
								term.setBackgroundColor(colors.white)
								term.setTextColor(colors.black)
								term.setCursorPos(2,13)
								term.write(" Cancel ")
								sleep(0.1)
								in_search_result = false
							end

							redrawScreen()
						elseif tab == 1 and in_search_result == false then
							-- Now playing tab clicks

							if y == 6 then
								-- Play/stop button
								if x >= 2 and x < 2 + 6 then
									if playing or now_playing ~= nil or #queue > 0 then
										term.setBackgroundColor(colors.white)
										term.setTextColor(colors.black)
										term.setCursorPos(2, 6)
										if playing then
											term.write(" Stop ")
										else
											term.write(" Play ")
										end
										sleep(0.1)
									end
									if playing then
										playing = false
										for _, speaker in ipairs(speakers) do
											speaker.stop()
											os.queueEvent("playback_stopped")
										end
										playing_id = nil
										is_loading = false
										is_error = false
										os.queueEvent("audio_update")
									elseif now_playing ~= nil then
										playing_id = nil
										playing = true
										is_error = false
										os.queueEvent("audio_update")
									elseif #queue > 0 then
										now_playing = queue[1]
										table.remove(queue, 1)
										playing_id = nil
										playing = true
										is_error = false
										os.queueEvent("audio_update")
									end
								end

								-- Skip button
								if x >= 2 + 7 and x < 2 + 7 + 6 then
									if now_playing ~= nil or #queue > 0 then
										term.setBackgroundColor(colors.white)
										term.setTextColor(colors.black)
										term.setCursorPos(2 + 7, 6)
										term.write(" Skip ")
										sleep(0.1)

										is_error = false
										if playing then
											for _, speaker in ipairs(speakers) do
												speaker.stop()
												os.queueEvent("playback_stopped")
											end
										end
										if #queue > 0 then
											if looping == 1 then
												table.insert(queue, now_playing)
											end
											now_playing = queue[1]
											table.remove(queue, 1)
											playing_id = nil
										else
											now_playing = nil
											playing = false
											is_loading = false
											is_error = false
											playing_id = nil
										end
										os.queueEvent("audio_update")
									end
								end

								-- Loop button
								if x >= 2 + 7 + 7 and x < 2 + 7 + 7 + 12 then
									if looping == 0 then
										looping = 1
									elseif looping == 1 then
										looping = 2
									else
										looping = 0
									end
								end
							end

							if y == 8 then
								-- Volume slider
								if x >= 1 and x < 2 + 24 then
									volume = (x - 1) / 24 * 3
								end
							end

							redrawScreen()
						end
					end
				end,
				function()
					local event, button, x, y = os.pullEvent("mouse_drag")

					if button == 1 then

						if tab == 1 and in_search_result == false then

							if y >= 7 and y <= 9 then
								-- Volume slider
								if x >= 1 and x < 2 + 24 then
									volume = (x - 1) / 24 * 3
								end
							end

							redrawScreen()
						end
					end
				end,
				function()
					local event = os.pullEvent("redraw_screen")
					redrawScreen()
				end
			)
		end
	end
end

function audioLoop()
	while true do
		-- AUDIO
		if playing and now_playing then
			local thisnowplayingid = now_playing.id
			if playing_id ~= thisnowplayingid then
				playing_id = thisnowplayingid
				last_download_url = api_base_url .. "?v=" .. version .. "&id=" .. textutils.urlEncode(playing_id)
				playing_status = 0
				needs_next_chunk = 1

				http.request({url = last_download_url, binary = true})
				is_loading = true

				os.queueEvent("redraw_screen")
				os.queueEvent("audio_update")
			elseif playing_status == 1 and needs_next_chunk == 1 then

				while true do
					local chunk = player_handle.read(size)
					if not chunk then
						if looping == 2 or (looping == 1 and #queue == 0) then
							playing_id = nil
						elseif looping == 1 and #queue > 0 then
							table.insert(queue, now_playing)
							now_playing = queue[1]
							table.remove(queue, 1)
							playing_id = nil
						else
							if #queue > 0 then
								now_playing = queue[1]
								table.remove(queue, 1)
								playing_id = nil
							else
								now_playing = nil
								playing = false
								playing_id = nil
								is_loading = false
								is_error = false
							end
						end

						os.queueEvent("redraw_screen")

						player_handle.close()
						needs_next_chunk = 0
						break
					else
						if start then
							chunk, start = start .. chunk, nil
							size = size + 4
						end

						buffer = decoder(chunk)

						local fn = {}
						for i, speaker in ipairs(speakers) do
							fn[i] = function()
								local name = peripheral.getName(speaker)
								if #speakers > 1 then
									if speaker.playAudio(buffer, volume) then
										parallel.waitForAny(
											function()
												repeat until select(2, os.pullEvent("speaker_audio_empty")) == name
											end,
											function()
												local event = os.pullEvent("playback_stopped")
												return
											end
										)
										if not playing or playing_id ~= thisnowplayingid then
											return
										end
									end
								else
									while not speaker.playAudio(buffer, volume) do
										parallel.waitForAny(
											function()
												repeat until select(2, os.pullEvent("speaker_audio_empty")) == name
											end,
											function()
												local event = os.pullEvent("playback_stopped")
												return
											end
										)
										if not playing or playing_id ~= thisnowplayingid then
											return
										end
									end
								end
								if not playing or playing_id ~= thisnowplayingid then
									return
								end
							end
						end

						local ok, err = pcall(parallel.waitForAll, table.unpack(fn))
						if not ok then
							needs_next_chunk = 2
							is_error = true
							break
						end

						if not playing or playing_id ~= thisnowplayingid then
							break
						end
					end
				end
				os.queueEvent("audio_update")
			end
		end

		os.pullEvent("audio_update")
	end
end

function httpLoop()
	while true do
		parallel.waitForAny(
			function()
				local event, url, handle = os.pullEvent("http_success")

				if url == last_search_url then
					search_results = textutils.unserialiseJSON(handle.readAll())
					os.queueEvent("redraw_screen")
				end
				if url == last_download_url then
					is_loading = false
					player_handle = handle
					start = handle.read(4)
					size = 16 * 1024 - 4
					playing_status = 1
					os.queueEvent("redraw_screen")
					os.queueEvent("audio_update")
				end
			end,
			function()
				local event, url = os.pullEvent("http_failure")

				if url == last_search_url then
					search_error = true
					os.queueEvent("redraw_screen")
				end
				if url == last_download_url then
					is_loading = false
					is_error = true
					playing = false
					playing_id = nil
					os.queueEvent("redraw_screen")
					os.queueEvent("audio_update")
				end
			end
		)
	end
end

parallel.waitForAny(uiLoop, audioLoop, httpLoop)