local sg = peripheral.find("stargate")

local event, p1, p2, p3 = nil
local screenSelection = 1
local screen = nil
local prevX, prevY = nil
local state, engaged, direction = sg.stargateState()
local favAddresses = {}

function resetScreen()
	term.setBackgroundColor(colors.gray)
	term.setTextColor(colors.black)
	term.clear()
	term.setCursorPos(1, 1)
end

function PrintInline(text, x, y)
	term.setCursorPos(prevX + x, prevY + y)
	term.write(text)
	prevX = prevX + x
	prevY = prevY + y
end

function Print(text, x, y)
	term.setCursorPos(x, y)
	term.clearLine()
	term.write(text)
	prevX = x
	prevY = y
end

function drawScreen(screenSettings)
	resetScreen()

	term.setBackgroundColor(colors.white)
	for i = 1, 3, 1 do
		term.setCursorPos(1, i)
		term.clearLine()
	end
	Print(screenSettings["title"], getCenterX(screenSettings["title"]), 2)
	term.setBackgroundColor(colors.gray)

	--For each option of a screen
	local line = getCenterY() - math.floor(tablelength(screenSettings["options"]) / 2)
	local curSel = 1
	for k, v in ipairs(screenSettings["options"]) do
		local label = v["option"]
		if curSel == screenSelection then
			term.setTextColor(colors.white)
			Print("[ ", getCenterX("[ " .. label .. " ]"), line)
			term.setTextColor(colors.black)
			PrintInline(label, string.len("[ "), 0, true)
			term.setTextColor(colors.white)
			PrintInline(" ]", string.len(label), 0, true)
			term.setTextColor(colors.black)
		else
			Print(label, getCenterX(label), line)
		end
		line = line + 1
		curSel = curSel + 1
	end
end

function getCenterX(text)
	local x,_ = term.getSize()
	return (x - string.len(text)) / 2
end

function getCenterY()
	local _,y = term.getSize()
	return y / 2
end

function tablelength(T)
  local count = 0
  for _ in pairs(T) do count = count + 1 end
  return count
end

function tableAt(T, index)
	local i = 0
	for _, v in pairs(T) do
		if i == index then
			return v
		end
		i = i + 1
	end
end

function tableAtKey(T, key, value)
	for _, v in pairs(T) do
		if v[key] == value then
			return v
		end
	end
end

function startup()
	screenSelection = 1
	screen = {
		title = "Stargate Control Program",
		options = {
			{
				option = "Dial Favorite",
				func = dialFavoriteScreen
			},
			{
				option = "Dial Manually",
				func = dialManually
			},
			{
				option = "Exit",
				func = exitProgram
			}
		}
	}
end

function dialFavoriteScreen()
	screenSelection = 1
	screen = {
		title = "Dial Favorites",
		options = {
			{
				option = "Add Favorite",
				func = function()
					if not fs.exists("favorites") then
						fs.open("favorites", "w").close()
					end
					f = fs.open("favorites","r")
					if f then
						local _, y = term.getSize()

						Print("Name:", 1, y - 1)
						term.setCursorPos(1, y)
						term.clearLine()
						local name = read()
						drawScreen(screen)

						Print("Address:", 1, y - 1)
						term.setCursorPos(1, y)
						term.clearLine()
						local gaddress = read()
						local firstLine = f.readLine()
						f.close()
						f = fs.open("favorites","a")
						if firstLine ~= "" or firstLine ~= nil then
							f.writeLine("\n\"" ..name.."\"{"..gaddress.."}")
						else
							f.writeLine("\"" ..name.."\"{"..gaddress.."}")
						end
						table.insert(favAddresses,
						{
							title = name,
							address = gaddress
						})
						f.close()

					end
					dialFavoriteScreen()
				end
			}
		}
	}

	for k, v in ipairs(favAddresses) do
		table.insert(screen["options"],
		{
			option = v["title"],
			func = dialFav
		})
	end

	table.insert(screen["options"],
	{
		option = "Back",
		func = startup
	})
end

function dialManually()
	screenSelection = 1
	screen = {
		title = "Manual Dial",
		options = {
			{
				option = "Enter",
				func = function()
					local _, y = term.getSize()
					Print("ADDRESS:", 1, y - 1)
					term.setCursorPos(1, y)
					dial(read())
					startup()
				end
			},
			{
				option = "Back",
				func = startup
			}
		}
	}
end

function parseTableFavorites()
	if not fs.exists("favorites") then
		fs.open("favorites", "w").close()
	end
	f = fs.open("favorites", "r")
	if f then
		local text = f.readAll()
		for i in string.gmatch(text, "[^\r\n]+") do
			if i ~= "" then
				local title, address = nil
				local j = string.match(i, "%b\"\"")
				j = string.gsub(j, "%s$", "")
				j = string.gsub(j, "^%s", "")
				j = string.gsub(j, "\"", "")
				ftitle = j

				local j = string.match(i, "%b{}")
				j = string.gsub(j, "%s$", "")
				j = string.gsub(j, "^%s", "")
				j = string.gsub(j, "{", "")
				j = string.gsub(j, "}", "")
				faddress = j

				table.insert(favAddresses,
				{
					title = ftitle,
					address = faddress
				})
			end
		end
		f.close()
	end
end

function dialFav()
	local destination = tableAt(favAddresses, screenSelection - 2)
	dial(destination["address"])
	startup()
end

function dial(address)
	local _, err = sg.dial(address)
	if err == "targetBusy" then
		local _, y = term.getSize()
		Print("Stargate is in use!", 1, y)
		sleep(2)
	elseif sg.energyAvailable() < sg.energyToDial(address) then
		local _, y = term.getSize()
		Print("Stargate has not enough energy!", 1, y)
		sleep(2)
	end
end

function exitProgram()
	os.reboot()
end

function awaitEvent()
	event, p1, p2, p3 = os.pullEvent()
end

function isSelectionChanged()
	if event == "key" then
		if p1 == keys.up then
			if screenSelection == 1 then
				screenSelection = tablelength(screen["options"])
			else
				screenSelection = screenSelection - 1
			end
		elseif p1 == keys.down then
			if screenSelection == tablelength(screen["options"]) then
				screenSelection = 1
			else
				screenSelection = screenSelection + 1
			end
		end
	end
end

function checkForGateControls()
	if state ~= "Idle" and screen["title"] == "Stargate Control Program" then
		table.insert(screen["options"], 3,
		{
			option = "Close Gate",
			func = function()
				sg.disconnect()
				startup()
			end
		})
	end
end

function isSelctionSelected()
	if event == "key" then
		if p1 == keys.enter then
			tableAt(screen["options"], screenSelection - 1)["func"]()
		elseif p1 == keys.backspace then
			local back = tableAtKey(screen["options"], "option", "Back")
			if back then
				back["func"]()
			end
		end
	end
end

parseTableFavorites()
startup()

while true do
	awaitEvent()
	isSelectionChanged()
	isSelctionSelected()
	state, engaged, direction = sg.stargateState()
	checkForGateControls()
	drawScreen(screen)
end