--[[
Quarry Mining Script v 1.0 Alpha


by Master_Jynx

*******************************************************************************************************
*   Usage:                                                          *
*   The Auto Resume feature requires you to add a command to the startUp file               *
*   <script name> -r                                                        *
*   where <script name> is what you named this script when saving it.                       *
*                                                                   *
*   Place the turtle on the bottom left of the quarry area, it will mine forward and rightward. *
*   Place 1 chest above the turtle's and only place fuel in it                          *
*   Don't use fuel that leave bukkits or other other empty objects behind.                  *
*   Place a chest behind the turtle, for it to unload into.                         *
*                                                                   *
*   Place a chest in slot 15 and a chest cart in slot 14. It will use these to collect the      *
*   contents of chests and chest carts.                                         *
*                                                                   *
*   Leave Slot 16 Empty, the turtle will fill it                                    *
*******************************************************************************************************

Directional Notes on Facing values
When facing North F=2 Z-coord decreases by 1
When facing East    F=3 X-coord increases by 1
When facing South F=0 Z-coord increases by 1
When facing West    F=1 X-coord decreases by 1
--]]

-- **********************************************
-- *    status variable is used to monitor what *
-- *    the script should be doing          *
-- *    0=script starting                   *
-- *  1=Error, stop script              *
-- *    2=Not resuming, ask for data            *
-- *    3=Trying to resume              *
-- *    4=Quarrying                     *
-- *    5=Go Home to unload or refuel           *
-- *  6=Quarry Done, head to start position *
-- **********************************************

local paramFile = "jQuarrySettings.txt"
local resumeFile = "jQuarryResume.txt"
local TrafficY = -1
local status  = 0
local turnDir = 1

local QuarryDone = false
local maxDigTry = 16
local goHome = false
local QuarrySize = 0
local QuarryD = 0
local QuarryDone = false

local StartX = 0
local StartY = 0
local StartZ = 0

local CurrentX = 0
local CurrentY = 0
local CurrentZ = 0

local resumeX = 0
local resumeY = 0
local resumeZ = 0

local BaseX = 0
local BaseY = 0
local BaseZ = 0

local BorderN = 0
local BorderW = 0
local BorderS = 0
local BorderE = 0

local LastX = 0
local LastZ = 0
local lastIsEven = nil

local Facing = nil
local startFacing = nil
local resumeFacing = nil

local minCoal = 10
local minFuel = 100

-- *************************************************
-- *    Checks for GPS, current location and checks  *
-- *    command line arguments                 *
-- *************************************************

function startUp(...)
    local x,y,z = gps.locate(3)
    if (x == nil) then
        print("GPS system not responding.")
        print("Fix it, then restart this script")
        status = 1
        return
    end
    CurrentX = x
    CurrentY = y
    CurrentZ = z
    if #arg == 0 then
        status = 2
    end
    if #arg > 1 then
        status = 1
        print("Invalid number of arguments")
        return
    end
    if (#arg == 1) then
        if (arg[1] == "-r") then
            status = 3
        else
            status = 1
            print("Invalid argument")
            print("only -r is currently supported")
            print("please correct the error and restart the script")
        end
    end
end
function intro()
    term.clear()
    term.setCursorPos(1,1)
    print("Welcome to Master_Jynx's")
    print("Quarry Script version 1.0.5")
    print("Beginning New Quarry Setup")
    sleep(3)
    while (QuarrySize < 4 or QuarrySize > 20) do
        term.clear()
        term.setCursorPos(1,1)
        print("Quarry Width & Length will be the same")
        print("Quarry Size must be 4 to 20")
        print("Enter Size: ")
        QuarrySize = math.floor(math.abs(tonumber(io.read())))
    end
    while (QuarryD < 5 or QuarryD > 200) do
        term.clear()
        term.setCursorPos(1,1)
        print("What Level to stop on?")
        print("Level 5 is the lowest.")
        print("Will stop @ bedrock")
        print("Enter Depth: ")
        QuarryD = math.floor(math.abs(tonumber(io.read())))
        if QuarryD < 5 then
            print("Depth too low.")
            print("Set Depth to Level 5 or higher")
        end
        if QuarryD > 128 then
            print("Depth too high.")
        end
    end
    sleep(1)
    term.clear()
    term.setCursorPos(1,1)
    print("Place 1 chest cart in slot 14 ")
    print("and 1 chest in slot 15")
    while ((turtle.getItemCount(14) ~= 1) or (turtle.getItemCount(15) ~= 1)) do
        sleep(0.5)
    end
    term.clear()
    term.setCursorPos(1,1)
    print("Thank You")
    print("Turlte will now calibrate itself and begin mining.")
    print("Please close this window and step away.")
    sleep(2)
end

-- **************************************************
-- *        Used to find which way turtle is facing *
-- **************************************************

function getFacing()
    local loop = 0
    while not turtle.forward() do
        turtle.attack()
        turtle.dig()
        loop = loop + 1
        sleep(0.8)
        if loop > maxDigTry then
            status = 1
            return
        end
    end
    BaseX,BaseY,BaseZ = gps.locate()
    Facing = ((BaseX < CurrentX) and 1 or ((BaseX > CurrentX) and 3 or ((BaseZ < CurrentZ) and 2 or 0)))
    turtle.turnRight()
    turtle.turnRight()
    while loop <= maxDigTry do
        if not turtle.forward() then
            turtle.attack()
            turtle.dig()
            loop = loop + 1
            sleep(0.8)
        else
            loop = 0
            break
        end
    end
    turtle.turnRight()
    turtle.turnRight()
end

-- **************************************************
-- *        Used to find which way turtle is facing *
-- **************************************************
function changeFacing(a)
    while (Facing ~= a) do
        turtle.turnRight()
        Facing = Facing + 1
        if (Facing > 3) then
            Facing = 0
        end
    end
end

--**** Fueling scripts ****
function checkFuel(count,level)
    local a = 0
    a = (turtle.getFuelLevel() < level) and (a + 1) or a
    a = (turtle.getItemCount(16) < count) and (a + 10) or a
    return a
end

function fuelUp(level)
    turtle.select(16)
    while (turtle.getFuelLevel() < level) do
        turtle.refuel(1)
    end
    turtle.select(1)
end

function fillTank(count)
    turtle.select(16)
    while (turtle.getItemCount(16) < count) do
        if not turtle.suckUp() then
            term.clear()
            term.write("Add more fuel to fuel chest")
            term.write("I'm waiting to refuel.....")
            sleep(10)
        end
    end
    turtle.select(1)
end



--****  Movement scripts ****

function moveForward()
    while not turtle.forward() do
        turtle.attack()
        sleep(0.8)
    end
    if (Facing == 0) then
        CurrentZ = CurrentZ + 1
    end
    if (Facing == 1) then
        CurrentX = CurrentX - 1
    end
    if (Facing == 2) then
        CurrentZ = CurrentZ - 1
    end
    if (Facing == 3) then
        CurrentX = CurrentX + 1
    end
end

function moveUp()
    while not turtle.up() do
        if turtle.detectUp()then
            turtle.digUp()
            sleep(0.8)
        else
            turtle.attackUp()
        end
    end
    CurrentY = CurrentY + 1
end

function moveDown()
    while not turtle.down() do
        if turtle.detectDown()then
            turtle.digDown()
            sleep(0.8)
        else
            turtle.attackDown()
        end
    end
    CurrentY = CurrentY - 1
end

function gotoHome()
    local a = 0
    moveTo(BaseX,StartY,BaseZ)
    moveTo(StartX,StartY,StartZ)
    changeFacing(startFacing)
    if (status ~=6) and (status ~= 1) then
        a = checkFuel(minCoal,minFuel)
        if (a == 10) or (a == 11) then
            fillTank(minCoal)
        end
        if (a == 1) or (a == 11) then
            fuelUp(minFuel)
        end
    end
    turtle.turnLeft()
    turtle.turnLeft()
    for i=1,13 do
        while (turtle.getItemCount(i) ~= 0) do
            turtle.select(i)
            turtle.drop()
        end
    end
    turtle.turnLeft()
    turtle.turnLeft()
    if (status == 6) or (Status == 1) then
        QuarryDone = true
        return
    else
        goHome = false
        status = 4
        moveForward()
        moveTo(resumeX,resumeY,resumeZ)
        changeFacing(resumeFacing)
    end
end


function moveTo(X1,Y1,Z1)
    local a
    if (CurrentY ~= Y1) then
        while (CurrentY ~= Y1) do
            a = (CurrentY < Y1) and moveUp() or moveDown()
        end
    end
    if (CurrentX ~= X1) then
        changeFacing((CurrentX > X1) and 1 or 3)
        while (CurrentX ~= X1) do
            moveForward()
        end
    end
    if (CurrentZ ~= Z1) then
        changeFacing((CurrentZ > Z1) and 2 or 0)
        while (CurrentZ ~= Z1) do
            moveForward()
        end
    end
end
--*****************************************************
--**    0=South 1=West  2=North 3=East  *
--** When facing North F=2 Z-coord decreases by 1   *
--** When facing East   F=3 X-coord increases by 1  *
--** When facing South F=0 Z-coord increases by 1   *
--** When facing West   F=1 X-coord decreases by 1  *
--*****************************************************

function nextLevel()
    turnDir = 1
    local depth = CurrentY - QuarryD
    if depth <= 1 then
        status = 6
        QuarryDone = true
        dataSwap1()
        gotoHome()
        return
    end
    if depth == 2 then
        moveDown()
    end
    if depth == 3 then
        moveDown()
        moveDown()
    end
    if depth >= 4 then
        moveDown()
        moveDown()
        moveDown()
    end
end

function dataSwap1()
    resumeX = CurrentX
    resumeY = CurrentY
    resumeZ = CurrentZ
    resumeFacing = Facing
end

--*****************************************************
--** Starting calibration. doesn't run on resume.   *
--** sets Borders, starting position and facing     *
--*****************************************************
--*****************************************************
--**    0=South 1=West  2=North 3=East  *
--** When facing North F=2 Z-coord decreases by 1   *
--** When facing East   F=3 X-coord increases by 1  *
--** When facing South F=0 Z-coord increases by 1   *
--** When facing West   F=1 X-coord decreases by 1  *
--*****************************************************

function calibrate()
    StartX = CurrentX
    StartY = CurrentY
    StartZ = CurrentZ
    getFacing()
    startFacing = Facing
    if (Facing == 0) then
        BorderS = StartZ + (QuarrySize)
        BorderW = StartX - (QuarrySize - 1)
        BorderN = StartZ + 1
        BorderE = StartX
    end
    if (Facing == 1) then
        BorderS = StartZ
        BorderW = StartX - (QuarrySize)
        BorderN = StartZ - (QuarrySize - 1)
        BorderE = StartX + 1
    end
    if (Facing == 2) then
        BorderS = StartZ + 1
        BorderW = StartX
        BorderN = StartZ - (QuarrySize)
        BorderE = StartX + (QuarrySize - 1)
    end
    if (Facing == 3) then
        BorderS = StartZ + (QuarrySize - 1)
        BorderW = StartX + 1
        BorderN = StartZ
        BorderE = StartX + (QuarrySize)
    end
    findEnd()
    TrafficY = StartY
end

function findEnd()
    lastIsEven = (QuarrySize == math.floor(QuarrySize/2)*2) and true or false
    if lastIsEven then
        if Facing == 0 then
            LastX = BorderW
            LastZ = BorderN
        end
        if Facing == 1 then
            LastX = BorderE
            LastZ = BorderN
        end
        if Facing == 2 then
            LastX = BorderE
            LastZ = BorderS
        end
        if Facing == 3 then
            LastX = BorderW
            LastZ = BorderS
        end
    else
        if Facing == 0 then
            LastX = BorderW
            LastZ = BorderS
        end
        if Facing == 1 then
            LastX = BorderW
            LastZ = BorderN
        end
        if Facing == 2 then
            LastX = BorderE
            LastZ = BorderN
        end
        if Facing == 3 then
            LastX = BorderE
            LastZ = BorderS
        end
    end
end

--**** File Operations ****
function saveParam()
    local file = fs.open(paramFile,"w")
    file.writeLine(QuarrySize)
    file.writeLine(QuarryD)
    file.writeLine(StartX)
    file.writeLine(StartY)
    file.writeLine(StartZ)
    file.writeLine(startFacing)
    file.writeLine(BorderN)
    file.writeLine(BorderW)
    file.writeLine(BorderS)
    file.writeLine(BorderE)
    file.writeLine(LastX)
    file.writeLine(LastZ)
    file.close()
end
function saveResume()
    local file = fs.open(resumeFile,"w")
    file.writeLine(CurrentX)
    file.writeLine(CurrentY)
    file.writeLine(CurrentZ)
    file.writeLine(Facing)
    file.writeLine(goHome)
    file.writeLine(QuarryDone)
    file.close()
end


-- *************************************************
-- *    Tries to load Resume Data from the 2 files   *
-- *************************************************

function tryResume()
    local file
    local file1 = fs.exists(paramFile)
    local file2 = fs.exists(resumeFile)
    if (file1 == false or file2 == false) then
        print("Resume files not found.")
        print("Unable to continue.")
        status = 1
        gotoHome()
        return
    end
    if (status ~= 1) then
        file = fs.open(paramFile,"r")
        local fileData = {}
        local line = file.readLine()
        while (line ~= nil) do
            table.insert(fileData,line)
            line = file.readLine()
        end
        file.close()
        for i=1,#fileData do
            if (fileData[1] == nil) then
            status = 1
            end
        end
        if (#fileData ~= 11) then
            status = 1
        end
        if (status ~= 1) then
            QuarrySize = fileData[1]
            QuarryD = fileData[2]
            StartX = fileData[3]
            StartY = fileData[4]
            StartZ = fileData[5]
            startFacing = fileData[6]
            BorderN = fileData[7]
            BorderW = fileData[8]
            BorderS = fileData[9]
            BorderE = fileData[10]
            LastX = fileData[11]
            LastZ = fileData[12]
            TrafficY = StartY
        end
        if (status ~= 1) then
            file = fs.open(resumeFile,"r")
            fileData = {}
            line = file.readLine()
            while (line ~= nil) do
                table.insert(fileData,line)
                line = file.readLine()
            end
            file.close()
            for i=1,#fileData do
                if (fileData[1] == nil) then
                status = 1
                end
            end
            if (#fileData ~= 6) then
                status = 1
            end
            if (status ~= 1) then
                resumeX = fileData[1]
                resumeY = fileData[2]
                resumeZ = fileData[3]
                resumeFacing = fileData[4]
                goHome = fileData[5]
                QuarryDOne = fileData[6]

            end
        end
    end

end

--**** Mining script ****
-- **********************************************************
-- *    0=South 1=West  2=North 3=East          *
-- **********************************************************

function doMine(loops)
    local d = CurrentY - 1
    local a,b,c = false,0,0
    if not countEmptySlots(2) then
        dataSwap1()
        gotoHome()
    end
    for i=1,loops do
        if (CurrentX == LastX and CurrentZ == LastZ) then
            if QuarryD >= d then
                status = 6
                gotoHome()
            else
                turtle.digUp()
                turtle.digDown()
                moveTo(BaseX,CurrentY,BaseZ)
                dataSwap1()
                resumeFacing = startFacing
                gotoHome()
                I = loops + 1
                nextLevel()
                return
            end
        end

        if (Facing == 0) then
            if (CurrentZ == BorderS) then
                if (turnDir == 1) then
                    a,b,c = true,1,2
                    turnDir = 2
                else
                    a,b,c = true,3,2
                    turnDir = 1
                end
            end
        elseif (Facing == 1) then
            if (CurrentX == BorderW) then
                if (turnDir == 1) then
                    a,b,c = true,2,3
                    turnDir = 2
                else
                    a,b,c = true,0,3
                    turnDir = 1
                end
            end
        elseif (Facing == 2) then
            if (CurrentZ == BorderN) then
                if (turnDir == 1) then
                    a,b,c = true,3,0
                    turnDir = 2
                else
                    a,b,c = true,1,0
                    turnDir = 1
                end
            end
        elseif (Facing == 3) then
            if (CurrentX == BorderE) then
                if (turnDir == 1) then
                    a,b,c = true,0,1
                    turnDir = 2
                else
                    a,b,c = true,2,1
                    turnDir = 1
                end
            end
        end
term.clear()
term.setCursorPos(1,1)
print("Current XYZ: "..CurrentX.."  "..CurrentY.."  "..CurrentZ)
print("Loops:   "..loops)
print("Fuel Level:   "..turtle.getFuelLevel())
print("Status:    "..status)
print("goHome:    "..tostring(goHome))
        if a then
            changeFacing(b)
        end
        tryDig()
        moveForward()
        if a then
            changeFacing(c)
        end

    end
end
function tryDig()
    loop = 0
    checkChest()
    checkChestUp()
    checkChestDown()
    turtle.digUp()
    turtle.digDown()
    while (turtle.detect() and loop <= maxDigTry) do
        turtle.dig()
        sleep(0.8)
        loop = loop + 1
    end
    if (loop > maxDigTry) then
        status = 6
        return false
    else
        return true
    end
end


-- ****************************************************
-- *    Checks for and empties chests and chest carts   *
-- *    Will unload and refuel first, if needed     *
-- ****************************************************
function checkChest()
    local slotMatch = false
    local room = countEmptySlots(6)
    if not room then
        dataSwap1()
        status = 5
        goHome = true
        gotoHome()
        status = 4
    end
    if (turtle.detect()) then
        turtle.select(14)
        slotMatch = (turtle.compare()) and true or false
        turtle.select(15)
        slotMatch = (turtle.compare()) and true or false

    end
    if slotMatch then
        while (turtle.suck()) do
        end
    end
end

function checkChestUp()
    local slotMatch = false
    local room = countEmptySlots(6)
    if not room then
        dataSwap1()
        status = 5
        goHome = true
        gotoHome()
        status = 4
    end
    if (turtle.detectUp()) then
        turtle.select(14)
        slotMatch = (turtle.compareUp()) and true or false
        turtle.select(15)
        slotMatch = (turtle.compareUp()) and true or false
    end
    if slotMatch then
        while (turtle.suckUp()) do
        end
    end
end

function checkChestDown()
    local slotMatch = false
    local room = countEmptySlots(6)
    if not room then
        dataSwap1()
        status = 5
        goHome = true
        gotoHome()
        status = 4
    end
    if (turtle.detectDown()) then
        turtle.select(14)
        slotMatch = (turtle.compareDown()) and true or false
        turtle.select(15)
        slotMatch = (turtle.compareDown()) and true or false
    end
    if slotMatch then
        while (turtle.suckDown()) do
        end
    end
end
function countEmptySlots (num)
    local a = 0
    for i=1,13 do
        a = (turtle.getItemCount(i) > 0) and a + 1 or a
    end
    return (a <= num) and true or false
end


--********************************************************************

-- **********************************************
-- *    Status Codes                    *
-- *    0=script starting                   *
-- *  1=Error, stop script              *
-- *    2=Not resuming, ask for data            *
-- *    3=Trying to resume              *
-- *    4=Quarrying                     *
-- *    5=Go Home to unload or refuel           *
-- *  6=Quarry Done, head to start position *
-- **********************************************

-- **********************************************
-- *        Main Program                        *
-- **********************************************


startUp()
if (status == 1) then return end

if (status == 3) then
    tryResume()
    if (status == 1) then
        print("Could not load Resume Data.")
        print("File corruption file errors found.")
        print("Resume not possible")
        return
    end
    if not goHome then
        moveTo(resumeX,resumeY,resumeZ)
        changeFacing(resumeFacing)
        status = 4
    else
        status = 5
        gotoHome()
    end
end


if (status == 2) then
    local a = 0
    intro()
    a = checkFuel(minCoal,minFuel)
    if (a == 10) or (a == 11) then
        fillTank(minCoal)
    end
    if (a == 1) or (a == 11) then
        fuelUp(minFuel)
    end
    calibrate()
    saveParam()
    saveResume()
    moveForward()
    status = 4
end
-- **********************************
-- *    Main Script Flow Control    *
-- **********************************

while ((status == 4) and not QuarryDone) do
    local a = 0
    a = checkFuel(minCoal,minFuel)
    if (a == 10) or (a == 11) then
        fillTank(minCoal)
    end
    if (a == 1) or (a == 11) then
        fuelUp(minFuel)
    end
    doMine(5)
    if (QuarryDone) then
        return
    end
end

-- *********************************************************************************