setup

--[[
Elevator Control Setup by Hamish Robertson
Aliases: OminousPenguin, MC username: Peren

Thanks to Jonnathon Clayton for identifying a replacement for Rednet Cables (Project Red)

Licence: Creative Commons Attribution-NonCommercial-ShareAlike (http://creativecommons.org/licenses/by-nc-sa/3.0/)

This is a LUA program for ComputerCraft, a Minecraft mod by Daniel Ratcliffe aka dan200 (http://www.computercraft.info/)
This program is for controlling elevators from the Railcraft mod (http://www.railcraft.info/)
For help and feedback please use this forum thread: http://www.computercraft.info/forums2/index.php?/topic/1302-railcraft-mod-elevator-control/

You are permitted to modify and or build upon this work.
If you make your own version publicly available, then please also publish it at the above forum thread. Thank you and happy coding!

NOTE: This is the setup program. It will create a file called elevator.cfg and download the main program to elevator-main.lua

Use Ctrl+T to terminate the program
--]]

CHANNEL_ELEVATOR = 34080

function writeToPos(x, y, str)
    term.setCursorPos(x, y)
    term.write(str)
end
function clearArea(x1,y1,x2,y2)
    for i=y1,y2 do
        writeToPos(x1,i,string.rep(" ",x2-x1))
    end
end
function eventHandle(tHandlers, f, ...)
    f = tHandlers[f]
    if type(f) == "function" then return f(...) end
end
function debug(msg)
    print(msg)
    os.pullEvent("key")
end
function printSetupHeader(saving, title)
    term.clear()
    writeToPos(32,1,"Hold Ctrl+T to exit")
    title = title or "Setup"
    eqs = ("="):rep((46 - title:len())/2).."  "
    writeToPos(2,2,eqs..title..eqs:reverse())
    if saving then writeToPos(5,5,"Saving data... "); term.setCursorBlink(true) end
end

function getModemSide()
    -- Find all attached modems
    local sides = {"left","right","top","bottom","back","front"}
    local i = 1
    while i <= #sides do
        if peripheral.getType(sides[i]) ~= "modem" then
            table.remove(sides,i)
        else
            i = i + 1
        end
    end

    if #sides == 0 then
        return nil
    elseif #sides == 1 then
        return sides[1]
    end

    -- If there is more than one modem attached, ask user to choose one
    local selected = 1
    local function redraw()
        printSetupHeader()
        writeToPos(7,9,"Use the modem on the          side")
        for i=1,#sides do
            if (i == selected) then
                writeToPos(29,9,sides[i])
            else
                if (9-selected+i)<9 then
                    term.setCursorPos(29,(8-selected+i))
                else
                    term.setCursorPos(29,(10-selected+i))
                end
                term.write(sides[i])
            end
        end--for
        writeToPos(4,18,"[ Arrow keys to select, Enter to confirm ]")
    end--redraw()

    redraw(); -- First draw

    while true do
        local _, keycode = os.pullEvent ("key")
        if (keycode == 200) then -- up
            if selected > 1 then selected = selected-1 end
        elseif (keycode == 208) then -- down
            if selected < 6 then selected = selected+1 end
        elseif (keycode == 28) then -- enter
            return sides[selected]
        end
        redraw()
    end
end--getModemSide()

function getElevatorID()
    local elevatorID
    printSetupHeader()
    writeToPos(3, 4, "Are you using the Bundled Connector")
    writeToPos(6, 6, "from the Extra Peripherals mod? [Y/n] ")
    bundledConnector = string.lower(string.sub(io.read(),1,1))
    if (bundledConnector=='y') then
        bundledConnector = 'bundledConnector:yes'
    else
        bundledConnector = 'bundledConnector:no'
    end
    printSetupHeader()
    writeToPos(3, 4, "Searching for other elevator shafts.")
    writeToPos(3, 6, "(Remember that rain reduces modem range)")
    writeToPos(5, 10, "This may take a few seconds...")
    local modem = peripheral.wrap(modemSide)
    modem.open(os.getComputerID())
    modem.transmit(CHANNEL_ELEVATOR, os.getComputerID(), "ELEV\030ALL\030DISCOVER") -- Ask all elevator terminals in range to tell us their elevator ID
    local tElevatorIDs = {}
    local mt = {
        tMeta = {},
        iIndex = 1,
        __newindex = function(t,k,v)
            this = getmetatable(t)
            if this.tMeta[k] ~= true then
                this.tMeta[k] = true
                rawset(t,this.iIndex,k)
                this.iIndex = this.iIndex + 1
            end
        end,
    }
    setmetatable(tElevatorIDs,mt)
    local _, _, sChannel, sReplyChannel, sMessage
    local discoveryTimer
    local tHandlers = {
        modem_message = function (_, sChannel, sReplyChannel, sMessage, nDistance)
            discoveryTimer = os.startTimer(3) -- Reset timer
            local iter = string.gmatch(sMessage, "([^\030]+)")
            if sChannel == os.getComputerID() and iter() == "ELEV" then
                local eID = iter()
                if iter() == "ID" then
                    tElevatorIDs[eID] = true
                end
            end
        end,
        timer = function (timerID)
            if timerID == discoveryTimer then   return true end
        end
    }
    discoveryTimer = os.startTimer(3) -- Loop will timeout after three seconds of not receiving an event
    while true do
        local v = eventHandle(tHandlers, os.pullEvent())
        if v then break end
    end

    local function getCustomID()
        writeToPos(5, 7, "Please specify a shaft ID: ")
        writeToPos(2, 9, "IDs can have letters, numbers and other characters")
        writeToPos(2, 11, "Note that this is an identifier for the whole")
        writeToPos(2, 12, "elevator shaft, not just this floor.")
        writeToPos(2, 14, "Eg. The elevator shaft with ID 'House1' might have")
        writeToPos(2, 15, "3 floors: 'Entrance', 'Workshop' and 'Mine'")
        term.setCursorPos(32, 7); term.setCursorBlink(true)
        return io.read()
    end

    printSetupHeader(false,"READ ME")
    if #tElevatorIDs < 2 then
        writeToPos(2, 4, "Note the difference between elevators and floors.")
        writeToPos(2, 5, "An elevator shaft has multiple floors.")
        writeToPos(2, 6, "You can have multiple separate elevator shafts.")
        writeToPos(2, 8, "Because they communicate wirelessly, one shaft")
        writeToPos(2, 9, "might overhear another nearby shaft.")
        writeToPos(2, 11, "To prevent interference you must tell each floor")
        writeToPos(2, 12, "which elevator shaft it is part of.")
        writeToPos(2, 14, "Eg. The elevator shaft with ID 'House1' might have")
        writeToPos(2, 15, "3 floors: 'Entrance', 'Workshop' and 'Mine'")
        writeToPos(6, 17, "Press any key to continue...")
        os.pullEvent("key")
    end
    printSetupHeader()
    if #tElevatorIDs == 0 then
        writeToPos(2, 5, "No nearby elevator shafts detected")
        elevatorID = getCustomID()
    else
        writeToPos(2, 5, "The following elevator shafts were detected:")
        writeToPos(3, 18, "Enter: Add this floor to the selected shaft")
        writeToPos(3, 19, "Tab: Start a new elevator shaft")
        local selected = 1
        for k,v in ipairs(tElevatorIDs) do
            writeToPos(5, 6+k, v)
        end
        writeToPos(3,7,">")
        local function updateSelection(previousSelection)
            writeToPos(3,6+previousSelection," ")
            writeToPos(3,6+selected,">")
        end
        while true do
            previousSelection = selected
            local _, keycode = os.pullEvent ("key")
            if (keycode == 200) then -- up
                if selected > 1 then selected = selected-1 end
            elseif (keycode == 208) then -- down
                previousSelection = selected
                if selected < #tElevatorIDs then selected = selected+1 end
            elseif (keycode == 28) then -- enter
                elevatorID = tElevatorIDs[selected]
                break
            elseif (keycode == 15) then -- tab
                printSetupHeader()
                writeToPos(2, 5, "You have chosen to create a new elevator shaft")
                elevatorID = getCustomID()
                break
            end
            updateSelection(previousSelection)
        end
    end
    return elevatorID
end--getElevatorID()

function getFloorName()
    printSetupHeader()
    writeToPos(7,9,"Enter a name/label for this floor: ")
    term.setCursorPos(14,11); term.setCursorBlink(true)
    local name
    repeat
        name = io.read()
    until name ~= ""
    return name
end--getFloorName()

function getGPS()
    printSetupHeader()
    writeToPos(5, 5, "Waiting for GPS response... ")
    term.setCursorPos(5,7)
    local x, y, z = gps.locate(2)
    if x ~= nil then
        term.write("... coordinates received")
        writeToPos(5, 9, "Start GPS host in the background?")
        local _, keycode, selected = nil, nil, 1
        writeToPos(14,11,"[ Yes ]             No")
        while true do
            term.setCursorPos(14,11)
            _, keycode = os.pullEvent ("key")
            if (keycode == 205) or (keycode == 49) then -- right
                if selected == 1 then
                    selected = 2
                    term.write("  Yes             [ No ]")
                end
            elseif (keycode == 203) or (keys.getName(keycode) == n) then -- left
                if selected == 2 then
                    selected = 1
                    term.write("[ Yes ]             No  ")
                end
            elseif (keycode == 28) then -- enter
                if selected == 1 then
                    return x,y,z
                else
                    return nil, y
                end
            end
        end--while
    else -- We didn't get coords from GPS
        term.write("... could not determine coordinates")
        return
    end
end--getGPS()

function getCoordsInput()
    printSetupHeader()
    writeToPos(8,7,"x coordinate:")
    writeToPos(8,8,"y coordinate:")
    writeToPos(8,9,"z coordinate:")
    writeToPos(4,13,"Important: If your GPS hosts are all in a")
    writeToPos(4,14,"vertical line (same x & y coordinates), then")
    writeToPos(4,15,"you need to place another host off to the side")
    writeToPos(4,16,"to enable an accurate GPS fix")
    writeToPos(4,17,"(Usage: gps host x y z")
    term.setCursorPos(22,7); term.setCursorBlink(true)

    local selected, inputs = 1, {"","",""}

    local function redraw()
        term.setCursorPos(22+inputs[selected]:len(),6+selected)
    end

    local eventHandlers = {
        key = function(keycode)
            if (keycode == 200) and (selected > 1) then -- up
                selected = selected-1
                redraw(selected);
            elseif (selected < 3) and ((keycode == 208) or (keycode == 28)) then -- down
                selected = selected+1
                redraw(selected);
            elseif (keycode == 28) then -- enter
                for i,input in ipairs(inputs) do
                    if inputs[selected] == "" then return end
                end
                return inputs[1], inputs[2], inputs[3]
            elseif (keycode == 14) and inputs[selected] ~= "" then -- backspace
                    inputs[selected] = inputs[selected]:sub(1,-2)
                    term.setCursorPos(22+inputs[selected]:len(),6+selected)
                    term.write(" ")
                    term.setCursorPos(22+inputs[selected]:len(),6+selected)
            end
        end,

        char = function(c)
            if (inputs[selected]:len() < 6) and (string.find(c, "[%d\-]") ~= nil) then -- number entered
                inputs[selected] = inputs[selected]..c
                term.write(c)
            end
        end,

        handle = function (self, f, ...)
            f = self[f]
            if type(f) == "function" then return f(...) end
        end
    }
    while true do
        local x,y,z = eventHandlers:handle(os.pullEvent())
        if x then return x,y,z end
    end
end--getCoordsInput

function getFloorCoords()
    -- We need at least the y (height) coordinate so we can know what order the floors are in
    printSetupHeader()
    writeToPos(4,4,"The y (height) coordinate is required so that")
    writeToPos(4,5,"floors can be listed in the correct order.")
    writeToPos(4,7,"You have a few options:")
    writeToPos(6,10,"1. Enter y coordinate:")
    writeToPos(6,12,"2. Use GPS (requires 4 GPS hosts)")
    writeToPos(6,14,"3. Enter x,y,z and then this computer")
    writeToPos(13,15,"can become a GPS host.")
    writeToPos(7,16,"(recommended if you have over 4 floors)")
    writeToPos(2,19,"(Press F3 to view player coordinates)")

    local selected, line, y = 1, {10,12,14}, ""

    local function redraw(i, clear)
        term.setCursorPos(3,8+(2*i))
        local t = (clear and write(" ")) or write("[")
        term.setCursorPos(49,8+(2*i))
        local t = clear and write(" ") or write("]")
        term.setCursorPos(29,10)
    end

    redraw(1)
    term.setCursorBlink(true)

    local eventHandlers = {
        key = function(keycode)
            if (keycode == 200) then -- up
                if selected > 1 then
                    redraw(selected, true)
                    selected = selected-1
                    if selected == 1 then term.setCursorBlink(true) end
                    redraw(selected);
                end
            elseif (keycode == 208) then -- down
                if selected < 3 then
                    redraw(selected, true)
                    selected = selected+1
                    term.setCursorBlink(false)
                    redraw(selected);
                end
            elseif (keycode == 28) then -- enter
                if ((selected == 1) and (y ~= "")) then
                    return true, nil, y
                elseif selected == 2 then
                    return true, getGPS()
                elseif selected == 3 then
                    return true, getCoordsInput() -- input all 3
                end
            elseif (keycode == 14) then -- backspace
                if ((selected == 1) and (y ~= "")) then
                    y = y:sub(1,-2)
                    term.setCursorPos(29+y:len(),10)
                    term.write(" ")
                    term.setCursorPos(29+y:len(),10)
                end
            end
        end,

        char = function(c)
            if (selected == 1) and ((y:len() < 6) and (string.find(c, "[%d\-]") ~= nil)) then -- number entered
                y = y..c
                term.setCursorPos(28+y:len(),10)
                term.write(c)
            end
        end,

        handle = function (self, f, ...)
            f = self[f]
            if type(f) == "function" then return f(...) end
        end
    }
    while true do
        local RETURN,x,y,z = eventHandlers:handle(os.pullEvent())
        if RETURN then return x,y,z end
    end
end--getFloorCoords()

fs.delete("elevator-main.lua")
if not shell.run("pastebin", "get", "iJWyUQVr", "elevator-main.lua") then
    print("Failed to download main program. Try manually from pastebin: iJWyUQVr\n(this is just the setup component)")
    return
end

modemSide = getModemSide()
if modemSide == nil then
    printSetupHeader()
    writeToPos(15,6,"! No modem detected !")
    writeToPos(8,9,"A modem is required for")
    writeToPos(8,11,"communication with the other floors")
    writeToPos(4,13,"Please attach a modem then try again")
    return -- terminate program
end

local elevatorID = getElevatorID()

local floorName = getFloorName()

local x,y,z

while true do
    x,y,z = getFloorCoords()
    if y==nil then
        writeToPos(5, 11, "Press any key to continue...")
        os.pullEvent("key")
    else
        break
    end
end
local coords = y
if x and y then
    coords = x.."\31"..y.."\31"..z
end

printSetupHeader(true)
configPath = shell.resolve(".").."/elevator.cfg"
fs.delete(configPath)
local file = io.open(configPath, "w")
file:write(bundledConnector.."\n")
file:write(modemSide.."\n")
file:write(elevatorID.."\n")
file:write(coords.."\n")
file:write(y.."\31"..floorName)
file:close()
term.write("Done")

writeToPos(5,7,"Setting elevator-main.lua to run on startup... ")

file = io.open("/startup", "w")
file:write("shell.run(\"/"..shell.resolve(".").."/elevator-main.lua\")")
file:close()
term.write("Done")

writeToPos(5, 9, "Press any key to reboot...")
os.pullEvent("key")
os.reboot()