Code Preview Copy All

--[[
The MIT License (MIT)
 
Copyright (c) 2014 Lyqyd
 
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
 
The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.
 
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
--]]

local places = {}
local destination, current, heading, distance
local placesOffset = 1
local screenHandler, screenDraw
local newPlace, newPlaceFocus = {x = "", y = "", z = "", name = ""}, "name"
local goto, gotoFocus = {x = "", y = "", z = ""}, "x"
local tracing, traces = false, {}
local openDoors = false
local unpack = unpack or table.unpack

if fs.exists(".places") then
	local handle = io.open(".places", "r")
	if handle then
		for line in handle:lines() do
			local entry = {}
			entry.x, entry.y, entry.z, entry.name = string.match(line, "(%-?%d+),(%-?%d+),(%-?%d+),(.*)")
			if entry.x and entry.y and entry.z and entry.name then
				table.insert(places, entry)
			end
		end
		handle:close()
	end
end

local function savePlaces()
	local handle = io.open(".places", "w")
	if handle then
		for i = 1, #places do
			handle:write(places[i].x..","..places[i].y..","..places[i].z..","..places[i].name.."\n")
		end
		handle:close()
	else
		error("Couldn't open file for writing!")
	end
end

local function saveTrace()
	local name
	if fs.exists("trace") then
		local count = 1
		repeat
			count = count + 1
		until not fs.exists("trace"..count)
		name = "trace"..count
	else
		name = "trace"
	end
	local handle = io.open(name, "w")
	if handle then
		for i = 1, #traces do
			handle:write(traces[i].x..","..traces[i].y..","..traces[i].z.."\n")
		end
		handle:close()
	end
end


local function printCenter(text)
	text = tostring(text)
	local x, y = term.getSize()
	local xCur, yCur = term.getCursorPos()
	term.setCursorPos((x - #text) / 2 + 1, yCur)
	term.write(text)
end

local function menuBar()
	term.setCursorPos(1,1)
	term.setBackgroundColor(colors.gray)
	term.setTextColor(colors.white)
	term.write("  Places    Goto    Exit  ")
end

local function clearState()
	term.setCursorBlink(false)
	placesOffset = 1
	newPlace = {x = "", y = "", z = "", name = ""}
	newPlaceFocus = "name"
end

local function drawDestinationLine(index)
	local x, y = term.getSize()
	term.setBackgroundColor(colors.red)
	term.setTextColor(colors.white)
	term.write("x")
	term.setBackgroundColor(index % 2 == 0 and colors.lightGray or colors.white)
	term.setTextColor(colors.black)
	local destCoords = tostring(places[index].x)..","..tostring(places[index].y)..","..tostring(places[index].z)
	local nameLen = x - (#destCoords + 4) --4 for spacing and initial X and scrollbar
	local nameStr = string.sub(places[index].name..string.rep(" ", x), 1, nameLen)
	term.write(" "..nameStr.." "..destCoords)
end


local function mainScreen()
	local x, y = term.getSize()
	term.setBackgroundColor(colors.lightGray)
	term.setTextColor(colors.black)
	if heading then
		term.setCursorPos(3, 4)
		term.write("Heading: "..heading)
	end
	if current then
		term.setCursorPos(3, 6)
		term.write("Current Location:")
		term.setCursorPos(1, 7)
		printCenter(current)
	end
	if destination then
		term.setCursorPos(3, 9)
		term.write("Destination: ")
		if destinationName then
			term.write(destinationName)
		end
		term.setCursorPos(1, 10)
		printCenter(destination)
	end
	if distance then
		term.setCursorPos(3, 12)
		term.write("Distance: "..distance.."m")
	end
	term.setCursorPos(3, y - 1)
	term.setTextColor(tracing and colors.lime or colors.red)
	term.write("Tracing    ")
	if fs.exists("/.fingerprint") then
		term.setTextColor(openDoors and colors.lime or colors.red)
		term.write("Open Doors")
	end
	term.setTextColor(colors.black)
	term.setCursorPos(1, y)
end

local function handleMain(event)
	local x, y = term.getSize()
	if event[1] == "char" then
		if event[2] == "t" then
			tracing = not tracing
			if not tracing then
				--toggled off, save traces.
				saveTrace()
			end
		elseif event[2] == "f" then
			if not fs.exists("/.fingerprint") then
				local handle = io.open("/.fingerprint", "w")
				if handle then
					for i = 1, 256 do
						handle:write(string.char(math.random(32, 126)))
					end
					handle:close()
				end
			end
		end
	elseif event[1] == "mouse_click" then
		if event[4] == y - 1 and event[3] >= 3 and event[3] <= 9 then
			tracing = not tracing
			if not tracing then
				saveTrace()
			end
		elseif event[4] == y - 1 and event[3] >= 14 and event[3] <= 23 and fs.exists("/.fingerprint") then
			openDoors = not openDoors
		end
	end
end

local placesScreen, handlePlaces

local function newPlaceScreen()
	local x, y = term.getSize()
	local curX, curY
	term.setBackgroundColor(colors.lightGray)
	term.setTextColor(colors.black)
	term.setCursorPos(3, 4)
	term.write("Name: ")
	if newPlace.name then
		term.write(newPlace.name)
	end
	if newPlaceFocus == "name" then
		curX, curY = term.getCursorPos()
	end
	term.setCursorPos(3, 6)
	term.write("X: ")
	if newPlace.x then
		term.write(string.sub(tostring(newPlace.x), 0-(x - 7)))
	end
	if newPlaceFocus == "x" then
		curX, curY = term.getCursorPos()
	end
	term.setCursorPos(3, 7)
	term.write("Y: ")
	if newPlace.y then
		term.write(string.sub(tostring(newPlace.y), 0-(x - 7)))
	end
	if newPlaceFocus == "y" then
		curX, curY = term.getCursorPos()
	end
	term.setCursorPos(3, 8)
	term.write("Z: ")
	if newPlace.z then
		term.write(string.sub(tostring(newPlace.z), 0-(x - 7)))
	end
	if newPlaceFocus == "z" then
		curX, curY = term.getCursorPos()
	end
	term.setCursorPos(x - 5, 10)
	term.write("Done")
	if curX and curY then
		term.setCursorPos(curX, curY)
		term.setCursorBlink(true)
	end
end

local function handleNewPlace(event)
	local x, y = term.getSize()
	if event[1] == "mouse_click" then
		if event[4] == 4 then
			newPlaceFocus = "name"
		elseif event[4] == 6 then
			newPlaceFocus = "x"
		elseif event[4] == 7 then
			newPlaceFocus = "y"
		elseif event[4] == 8 then
			newPlaceFocus = "z"
		elseif event[4] == 10 and event[3] >= x - 5 and event[3] <= x - 1 then
			--clicked Done
			if tonumber(newPlace.x) and tonumber(newPlace.y) and tonumber(newPlace.z) and #newPlace.name > 0 then
				local entry = {x = newPlace.x, y = newPlace.y, z = newPlace.z, name = newPlace.name}
				table.insert(places, entry)
				savePlaces()
				clearState()
				screenHandler = handlePlaces
				screenDraw = placesScreen
			end
		end
	elseif event[1] == "char" then
		if event[2] == "-" and (newPlaceFocus == "x" or newPlaceFocus == "y" or newPlaceFocus == "z") then
			if string.sub(newPlace[newPlaceFocus], 1, 1) == "-" then
				newPlace[newPlaceFocus] = string.sub(newPlace[newPlaceFocus], 2)
			else
				newPlace[newPlaceFocus] = "-"..newPlace[newPlaceFocus]
			end
		else
			newPlace[newPlaceFocus] = newPlace[newPlaceFocus]..event[2]
		end
	elseif event[1] == "key" and event[2] == 14 then
		--backspace
		newPlace[newPlaceFocus] = string.sub(newPlace[newPlaceFocus], 1, #newPlace[newPlaceFocus] - 1)
	end
end

function placesScreen()
	term.setBackgroundColor(colors.lightGray)
	term.setTextColor(colors.black)
	term.setCursorPos(1, 3)
	term.write("  New:   Manual  ")
	if current and type(current) == "table" then
		term.write("Current")
	end
	term.setCursorPos(1, 5)
	local x, y = term.getSize()
	for i = 1, math.min(#places, y - 4) do
		term.setCursorPos(1, i + 4)
		drawDestinationLine(i + placesOffset - 1)
	end
	term.setCursorPos(x, 5)
	term.setBackgroundColor(colors.gray)
	term.setTextColor(colors.white)
	term.write("^")
	term.setCursorPos(x, y)
	term.write("v")
end

function handlePlaces(event)
	if event[1] == "mouse_click" then
		local x, y = term.getSize()
		if event[4] >= 5 and event[4] <= y then
			--clicked somewhere in the places list
			if event[3] == 1 then
				--removing a place from the list, clicked the x
				local index = placesOffset + event[4] - 5 --5 due to the starting y coord of the list
				if index <= #places then
					if event[2] == 1 then
						table.remove(places, index)
						savePlaces()
						placesScreen()
					elseif event[2] == 2 then
						--right-clicked, open place for editing!
						clearState()
						newPlace.x = tostring(places[index].x)
						newPlace.y = tostring(places[index].y)
						newPlace.z = tostring(places[index].z)
						newPlace.name = places[index].name
						table.remove(places, index)
						savePlaces()
						screenHandler = handleNewPlace
						screenDraw = newPlaceScreen
					end
				end
			elseif event[3] == x then
				--clicked the scroll bar.
				if event[4] == 5 then
					--up
					if placesOffset > 1 then
						placesOffset = placesOffset - 1
					end
				elseif event[4] == y then
					if placesOffset + y - 5 < #places then
						placesOffset = placesOffset + 1
					end
				end
			else
				--clicked on a place!
				local index = placesOffset + event[4] - 5
				if index <= #places then
					destination = vector.new(places[index].x, places[index].y, places[index].z)
					destinationName = places[index].name
					screenHandler = handleMain
					screenDraw = mainScreen
				end			
			end
		elseif event[4] == 3 then
			--clicked one of our new places options
			if event[3] >= 10 and event[3] <= 15 then
				--clicked manual
				clearState()
				screenHandler = handleNewPlace
				screenDraw = newPlaceScreen
			elseif event[3] >= 18 and event[3] <= 24 then
				--clicked current
				if current and type(current) == "table" then
					clearState()
					newPlace.x = tostring(current.x)
					newPlace.y = tostring(current.y)
					newPlace.z = tostring(current.z)
					screenHandler = handleNewPlace
					screenDraw = newPlaceScreen
				end
			end
		end
	end
end

local function gotoScreen()
	local x, y = term.getSize()
	local curX, curY
	term.setBackgroundColor(colors.lightGray)
	term.setTextColor(colors.black)
	term.setCursorPos(3, 4)
	term.write("X: ")
	if goto.x then
		term.write(string.sub(tostring(goto.x), 0-(x - 7)))
	end
	if gotoFocus == "x" then
		curX, curY = term.getCursorPos()
	end
	term.setCursorPos(3, 5)
	term.write("Y: ")
	if goto.y then
		term.write(string.sub(tostring(goto.y), 0-(x - 7)))
	end
	if gotoFocus == "y" then
		curX, curY = term.getCursorPos()
	end
	term.setCursorPos(3, 6)
	term.write("Z: ")
	if goto.z then
		term.write(string.sub(tostring(goto.z), 0-(x - 7)))
	end
	if gotoFocus == "z" then
		curX, curY = term.getCursorPos()
	end
	term.setCursorPos(x - 5, 8)
	term.write("Done")
	if curX and curY then
		term.setCursorPos(curX, curY)
		term.setCursorBlink(true)
	end
end

local function handleGoto(event)
	local x, y = term.getSize()
	if event[1] == "mouse_click" then
		if event[4] == 4 then
			gotoFocus = "x"
		elseif event[4] == 5 then
			gotoFocus = "y"
		elseif event[4] == 6 then
			gotoFocus = "z"
		elseif event[4] == 8 and event[3] >= x - 5 and event[3] <= x - 1 then
			--clicked Done
			if tonumber(goto.x) and tonumber(goto.y) and tonumber(goto.z) then
				destinationName = nil
				destination = vector.new(tonumber(goto.x), tonumber(goto.y), tonumber(goto.z))
				clearState()
				screenHandler = handleMain
				screenDraw = mainScreen
			end
		end
	elseif event[1] == "char" then
		if event[2] == "-" and (gotoFocus == "x" or gotoFocus == "y" or gotoFocus == "z") then
			if string.sub(goto[gotoFocus], 1, 1) == "-" then
				goto[gotoFocus] = string.sub(goto[gotoFocus], 2)
			else
				goto[gotoFocus] = "-"..goto[gotoFocus]
			end
		else
			goto[gotoFocus] = goto[gotoFocus]..event[2]
		end
	elseif event[1] == "key" and event[2] == 14 then
		--backspace
		goto[gotoFocus] = string.sub(goto[gotoFocus], 1, #goto[gotoFocus] - 1)
	end
end

local function handleEvents()
	while true do
		term.setBackgroundColor(colors.lightGray)
		term.clear()
		menuBar()
		screenDraw()
		local event = {os.pullEvent()}
		if event[1] == "mouse_click" then
			if event[4] == 1 then
				--clicked the menu bar at the top
				if event[3] >= 3 and event[3] <= 8 then
					--clicked places
					clearState()
					screenHandler = handlePlaces
					screenDraw = placesScreen
				elseif event[3] >= 13 and event[3] <= 16 then
					--clicked goto
					clearState()
					screenHandler = handleGoto
					screenDraw = gotoScreen
				elseif event[3] >= 21 and event[3] <= 24 then
					--clicked exit
					if screenHandler ~= handleMain then
						--not on main screen, return to main screen
						clearState()
						screenHandler = handleMain
						screenDraw = mainScreen
					else
						if tracing then
							saveTrace()
						end
						return
					end
				end
			else
				screenHandler(event)
			end
		elseif event[1] == "key" or event[1] == "char" or event[1] == "mouse_scroll" or event[1] == "mouse_drag" then
			screenHandler(event)
		end
	end
end

local function doGPS()
	local loc, oldLoc
	while true do
		oldLoc = loc
		loc = nil
		local result = {gps.locate()}
		if #result == 3 then
			loc = vector.new(unpack(result))
		end
		if loc then
			current = loc
			if destination then
				distance = math.sqrt((destination.x - current.x)^2 + (destination.y - current.y)^2 + (destination.z - current.z)^2)
			end
			if tracing then
				if not oldLoc or (oldLoc and (loc.x ~= oldLoc.x or loc.y ~= oldLoc.y or loc.z ~= oldLoc.z)) then
					table.insert(traces, loc)
				end
			end
		elseif current then
			current = "GPS Signal Lost"
		end
		if loc and oldLoc then
			local head = loc - oldLoc
			if math.abs(head.z) > math.abs(head.x) and math.abs(head.z) > 1 then
				if head.z == math.abs(head.z) then
					--positive z, south
					heading = "South"
				else
					heading = "North"
				end
			elseif math.abs(head.x) > 1 then
				if head.x == math.abs(head.x) then
					--positive x, east
					heading = "East"
				else
					heading = "West"
				end
			end
		end
		sleep(1)
	end
end

local function openDoor()
	local modem = peripheral.wrap("back")
	modem.open(4210)
	local handle = io.open("/.fingerprint")
	local fingerprint
	if handle then
		fingerprint = handle:read("*a")
		handle:close()
	end
	while true do
		local event = {os.pullEvent()}
		if openDoors then
			if event[1] == "modem_message" and event[3] == 4210 then
				modem.transmit(4211, 4212, textutils.serialize({content = textutils.serialize(fingerprint), senderID = os.getComputerID(), senderName = os.getComputerLabel(), channel = 4211, replyChannel = 4212, messageID = messageID or math.random(10000), destinationID = event[5].senderID}))
			end
		end
	end
end
				

screenHandler = handleMain
screenDraw = mainScreen
parallel.waitForAny(handleEvents, doGPS, openDoor)