WheatFarm

local switch_slot = 1
local blocksLeft, blocksRight, blocksForward = 0, 0, 0
local x, y, z, direction
local home_x, home_y, home_z, home_direction
local currentSide = "left"

local task = "start_wheatfarming"
local seeds_chest1, seeds_chest2, harvest_chest, planted, harvested = 0,0,0,0,0

-- [MAIN FUNCTIONS]
function start_wheatfarming()
    while true do
        change_task("start_wheatfarming")
        if currentSide == "left" then
            fixInv()
            start_left()
        elseif currentSide == "right" then
            fixInv()
            start_right()
        end
        change_side()
        home()
        emptyInv()
        iddle()
    end
end

function left_side()
    change_task("left_side")
    printInfo()
    plant()
    if home_direction == "S" then
        repeat
            if direction == "E" then
                repeat forward() plant() until x == home_x + blocksLeft
                change_backwards()
            elseif direction == "W" then
                repeat forward() plant() until x == home_x - 1
                change_backwards()
            end
        until y == home_y + blocksForward
        if direction == "E" then
            repeat forward() plant() until x == home_x + blocksLeft
        elseif direction == "W" then
            repeat forward() plant() until x == home_x - 1
        end
    elseif home_direction == "N" then
        repeat
            if direction == "W" then
                repeat forward() plant() until x == home_x - blocksLeft
                change_backwards()
            elseif direction == "E" then
                repeat forward() plant() until x == home_x - 1
                change_backwards()
            end
        until y == home_y - blocksForward
        if direction == "W" then
            repeat forward() plant() until x == home_x - blocksLeft
        elseif direction == "E" then
            repeat forward() plant() until x == home_x - 1
        end
    elseif home_direction == "W" then -- WEST SIDE
        repeat
            if direction == "S" then
                repeat forward() plant() until y == home_y + blocksLeft
                change_backwards()
            elseif direction == "N" then
                repeat forward() plant() until y == home_y + 1
                change_backwards()
            end
        until x == home_x - blocksForward
        if direction == "S" then
            repeat forward() plant() until y == home_y + blocksLeft
        elseif direction == "N" then
            repeat forward() plant() until y == home_y + 1
        end
    elseif home_direction == "E" then  -- EAST SIDE
        repeat
            if direction == "N" then
                repeat forward() plant() until y == home_y - blocksLeft
                change_backwards()
            elseif direction == "S" then
                repeat forward() plant() until y == home_y - 1
                change_backwards()
            end
        until x == home_x + blocksForward
        if direction == "N" then
            repeat forward() plant() until y == home_y - blocksLeft
        elseif direction == "S" then
            repeat forward() plant() until y >= home_y - 1
        end
    end
end

function right_side()
    change_task("right_side")
    printInfo()
    plant()
    if home_direction == "S" then
        repeat
            if direction == "W" then
                repeat forward() plant() until x == home_x - blocksRight
                change_backwards()
            elseif direction == "E" then
                repeat forward() plant() until x == home_x
                change_backwards()
            end
        until y == home_y + blocksForward
        if direction == "W" then
            repeat forward() plant() until x == home_x - blocksRight
        elseif direction == "E" then
            repeat forward() plant() until x == home_x
        end
    elseif home_direction == "N" then
        repeat
            if direction == "E" then
                repeat forward() plant() until x == home_x + blocksRight
                change_backwards()
            elseif direction == "W" then
                repeat forward() plant() until x == home_x
                change_backwards()
            end
        until y == home_y - blocksForward
        if direction == "E" then
            repeat forward() plant() until x == home_x + blocksRight
        elseif direction == "W" then
            repeat forward() plant() until x == home_x
        end
    elseif home_direction == "W" then
        repeat
            if direction == "N" then
                repeat forward() plant() until y == home_y - blocksRight
                change_backwards()
            elseif direction == "S" then
                repeat forward() plant() until y == home_y
                change_backwards()
            end
        until x == home_x - blocksForward
        if direction == "N" then
            repeat forward() plant() until y == home_y - blocksRight
        elseif direction == "S" then
            repeat forward() plant() until y == home_y
        end
    elseif home_direction == "E" then
        repeat
            if direction == "S" then
                repeat forward() plant() until y == home_y + blocksRight
                change_backwards()
            elseif direction == "N" then
                repeat forward() plant() until y == home_y
                change_backwards()
            end
        until x == home_x + blocksForward
        if direction == "S" then
            repeat forward() plant() until y == home_y + blocksRight
        elseif direction == "N" then
            repeat forward() plant() until y == home_y
        end
    end
end

function home()
    change_task("home")
    if z ~= home_z then
        if z > home_z then
            repeat
                down()
            until z == home_z
        elseif z < home_z then
            repeat
                up()
            until z == home_z
        end
    end
    if home_x ~= x or home_y ~= y then
        if home_direction == "N" or  home_direction == "S" then
            if x < home_x then
                while(direction ~= "E") do right() end
            elseif x > home_x then
                while(direction ~= "W") do right() end
            end
            while(x ~= home_x) do
                forward()
            end

            if home_direction == "N" then
                while(direction ~= "S") do right() end
            elseif home_direction == "S" then
                while(direction ~= "N") do right() end
            end
            while(y ~= home_y) do
                forward()
            end

            left()
            left()

        -- W AND E
        elseif home_direction == "W" or home_direction == "E" then
            if y > home_y then
                while(direction ~= "N") do right() end
            elseif y < home_y then
                while(direction ~= "S") do right() end
            end
            while(y ~= home_y) do
                forward()
            end

            if home_direction == "W" then
                while(direction ~= "E") do right() end
            elseif home_direction == "E" then
                while(direction ~= "W") do right() end
            end
            while(x ~= home_x) do
                forward()
            end

            while(direction ~= home_direction) do
                left()
            end
        end
    else
        while(direction ~= home_direction) do
            left()
        end
    end
end

function emptyInv()
    change_task("emptyInv")
    local empty_wheat = true
    local empty_seeds = true
    local empty_seeds2 = true

    local total_seeds_counter = 0

    for i=16,2,-1 do
        turtle.select(i)
        if turtle.getItemCount(i) ~= 0 then
            local data = turtle.getItemDetail(i)
            if data and string.find(data.name, "wheat_seeds") then
                total_seeds_counter = total_seeds_counter + turtle.getItemCount(i)
            end
        end
    end

    left()
    forward()

    local left_seeds_count = math.floor(total_seeds_counter / 2)

    for i=16,2,-1 do
        turtle.select(i)
        if turtle.getItemCount(i) ~= 0 then
            local data = turtle.getItemDetail(i)
            local seeds = turtle.getItemCount(i)
            if data and string.find(data.name, "wheat_seeds") then
                if left_seeds_count > turtle.getItemCount(i) then
                    local turtle_slot_seeds = turtle.getItemCount(i)
                    if turtle.dropDown() == false then
                        empty_seeds = false
                    else
                        seeds_chest1 = seeds_chest1 + seeds
                        left_seeds_count = left_seeds_count - turtle_slot_seeds
                    end
                else
                    repeat
                        if turtle.dropDown(left_seeds_count) == false then
                            empty_seeds = false
                        else
                            left_seeds_count = left_seeds_count - left_seeds_count
                            seeds_chest1 = seeds_chest1 + 1
                        end
                    until turtle.getItemDetail(i) == 0 or left_seeds_count == 0 or empty_seeds == false
                end
            end
        end
    end

    back()
    back()

    for i=16,2,-1 do
        turtle.select(i)
        if turtle.getItemCount(i) ~= 0 then
            local data = turtle.getItemDetail(i)
            local seeds = turtle.getItemCount(i)
            if data and string.find(data.name, "wheat_seeds") then
                if turtle.dropDown() == false then
                    empty_seeds2 = false
                else
                    seeds_chest2 = seeds_chest2 + 1
                end
            end
        end
    end

    forward()
    right()

    -- empty wheat
    for i=16,2,-1 do
        turtle.select(i)
        if turtle.getItemCount(i) ~= 0 then
            local data = turtle.getItemDetail(i)
            local wheat = turtle.getItemCount(i)
            if data then
                if turtle.dropDown() == false then
                    empty_wheat = false
                else
                    harvest_chest = harvest_chest + wheat
                end
            end
        end
    end

    if empty_seeds2 == false and empty_wheat == false then
        clearScreen()
        print("Second seed chest and wheat chest are full.")
        print("")
        print("Press enter when the problem is fixed")
        local chest_full = false
        while (chest_full == false) do
            local event, param = os.pullEvent()
            if event == "key" and param == 28 then -- key 28 is the 'enter' key
                chest_full = true
            end
        end
        emptyInv()
    elseif empty_wheat == false then
        clearScreen()
        print("Wheat chest is full.")
        print("")
        print("Press enter when the problem is fixed")
        local chest_full = false
        while (chest_full == false) do
            local event, param = os.pullEvent()
            if event == "key" and param == 28 then -- key 28 is the 'enter' key
                chest_full = true
            end
        end
        emptyInv()
    end
    saveCounters()

end

function iddle()
    change_task("iddle")
    local sleep_timer = math.random(100,300)

    while sleep_timer > 0 do
        sleep_timer = sleep_timer - 1
        printIddleInfo(sleep_timer)
        sleep(1)
    end
end

function fixInv()
    change_task("fixInv")
    printInfo()

    left()
    forward()

    local seeds_needed = 0

    if currentSide == "left" then
        seeds_needed = math.floor((blocksLeft * blocksForward)/100*25)
    else
        seeds_needed = math.floor(((blocksRight + 1) * blocksForward)/100*25)
    end

    local total_seeds_needed = seeds_needed

    turtle.select(2)
    local this_selected_slot = 2
    while seeds_needed > 1 do
        turtle.select(this_selected_slot)
        if seeds_needed > 64 then
            turtle.suckDown()
            seeds_needed = seeds_needed - seeds_needed
        elseif seeds_needed ~= 0 then
            turtle.suckDown(seeds_needed)
            seeds_needed = seeds_needed - seeds_needed
        end
        this_selected_slot = this_selected_slot + 1
        if this_selected_slot > 16 then
            break
        end
    end

    local total_seeds_counter = 0

    for i=16,2,-1 do
        turtle.select(i)
        if turtle.getItemCount(i) ~= 0 then
            local data = turtle.getItemDetail(i)
            if data and string.find(data.name, "wheat_seeds") then
                total_seeds_counter = total_seeds_counter + turtle.getItemCount(i)
            end
        end
    end


    for i=16,2,-1 do
        if total_seeds_counter > total_seeds_needed then
            turtle.select(i)
            if turtle.getItemCount(i) ~= 0 then
                if (total_seeds_counter - total_seeds_needed) > 64 then
                    turtle.dropDown(64)
                    total_seeds_counter = total_seeds_counter - 64
                elseif (total_seeds_counter - total_seeds_needed) > 0 then
                    turtle.dropDown(total_seeds_counter - total_seeds_needed)
                    total_seeds_counter = total_seeds_counter - total_seeds_needed
                end
            end
        end
    end

    back()
    right()
end


-- [SUB FUNCTIONS]

function change_side()
    if currentSide == "left" then
        currentSide = "right"
    elseif currentSide == "right" then
        currentSide = "left"
    end
    local h = fs.open("wheatfarmFolder/currentSide", "w")
        h.writeLine(currentSide)
        h.close()
end

function change_backwards()
    clearScreen()

    if home_direction == "N" then
        if direction == "W" then
            changeRight()
        elseif direction == "E" then
            changeLeft()
        end
    elseif home_direction == "S" then
        if direction == "E" then
            changeRight()
        elseif direction == "W" then
            changeLeft()
        end
    elseif home_direction == "W" then
        if direction == "S" then
            changeRight()
        elseif direction == "N" then
            changeLeft()
        end
    elseif home_direction == "E" then
        if direction == "N" then
            changeRight()
        elseif direction == "S" then
            changeLeft()
        end
    end
    plant()
end

function start_left()
    forward()
    left()
    forward()
    left_side()
end

function start_right()
    forward()
    right()
    right_side()
end

function changeLeft()
    left()
    forward()
    left()
end

function changeRight()
    right()
    forward()
    right()
end

function change_task(new_task)
    task = new_task
    local h = fs.open("wheatfarmFolder/task", "w")
    h.writeLine(task)
    h.close()
end
-- [FARM FUNCTIONS]

function  plant()
    local slot = 2

    local success, plant_data = turtle.inspectDown()
    if success and string.find(plant_data.name, "wheat") and plant_data.metadata == 7 then
        turtle.select(2)
        if turtle.digDown() then
            harvested = harvested + 1
        end
        for i=16,2,-1 do
            local p = turtle.getItemDetail(i)
            if p and string.find(p.name, "seed") ~= nil then
                slot = i
            end
        end
        turtle.select(slot)
        if turtle.placeDown() then
            planted = planted + 1
        end
    elseif not success then
        turtle.select(2)
        turtle.digDown()
        for i=16,2,-1 do
            local p = turtle.getItemDetail(i)
            if p and string.find(p.name, "seed") ~= nil then
                slot = i
            end
        end
        turtle.select(slot)
        if turtle.placeDown() then
            planted = planted + 1
        end
    end
    turtle.select(2)
    saveCounters()
end

-- [TURTLE FUNCTIONS]

function forward()
    if turtle.forward() then
        if direction == "W" then
            x = x - 1
        elseif direction == "E" then
            x = x + 1
        elseif direction == "S" then
            y = y + 1
        elseif direction == "N" then
            y = y - 1
        end
    end
    printInfo()
end

function back()
    if turtle.back() then
        if direction == "W" then
            x = x + 1
        elseif direction == "E" then
            x = x - 1
        elseif direction == "S" then
            y = y - 1
        elseif direction == "N" then
            y = y + 1
        end
    end
    printInfo()
end

function left()
    turtle.turnLeft()
    if direction == "N" then
        direction = "W"
    elseif direction == "W" then
        direction = "S"
    elseif direction == "S" then
        direction = "E"
    elseif direction == "E" then
        direction = "N"
    end
    printInfo()
end

function right()
    turtle.turnRight()
    if direction == "N" then
        direction = "E"
    elseif direction == "E" then
        direction = "S"
    elseif direction == "S" then
        direction = "W"
    elseif direction == "W" then
        direction = "N"
    end
    printInfo()
end

function down()
    turtle.down()
    z = z - 1
end

function up()
    turtle.up()
    z = z + 1
end


-- [OTHER FUNCTIONS]

function saveCounters()
    local h = fs.open("wheatfarmFolder/harvested", "w")
    h.writeLine(seeds_chest1)
    h.writeLine(seeds_chest2)
    h.writeLine(harvest_chest)
    h.writeLine(planted)
    h.writeLine(harvested)
    h.close()
end

function init()
    if fs.exists("wheatfarmFolder") then
        print("reading first input, probably a crash occured")
        print("---------------------------------------------")
        print("Reading first input")
        local h = fs.open("wheatfarmFolder/firstInput", "r")
        blocksLeft = tonumber(h.readLine())
        blocksRight = tonumber(h.readLine())
        blocksForward = tonumber(h.readLine())
        h.close()

        print("Reading currentSide and task  variable")
        local h = fs.open("wheatfarmFolder/currentSide", "r")
        currentSide = h.readLine()
        h.close()

        local h = fs.open("wheatfarmFolder/task", "r")
        task = h.readLine()
        h.close()

        print("Reading counter variables")
        local h = fs.open("wheatfarmFolder/harvested", "r")
        seeds_chest1 = tonumber(h.readLine())
        seeds_chest2 = tonumber(h.readLine())
        harvest_chest = tonumber(h.readLine())
        planted = tonumber(h.readLine())
        harvested = tonumber(h.readLine())
        h.close()

        print("Reading home position")
        local h = fs.open("wheatfarmFolder/home", "r")
        home_x = tonumber(h.readLine())
        home_y = tonumber(h.readLine())
        home_z = tonumber(h.readLine())
        home_direction = h.readLine()
        h.close()

        print("Checking my x y z coordinates and direction")
        checkGps()

        print(" ")
        print("My X: " .. x)
        print("My Y: " .. y)
        print("My Z: " .. z)
        print("My direction: " .. direction)
        print("")
        print("Home X: " .. home_x)
        print("Home Y: " .. home_y)
        print("Home Z: " .. home_z)
        print("Home direction: " .. home_direction)
        print("")
        print("Task: " .. task)
        sleep(2)
        printInfo()

        if task == "emptyInv" then
            home()
            emptyInv()
            fixInv()
            start_wheatfarming()
        elseif task == "fixInv" then
            home()
            fixInv()
            start_wheatfarming()
        elseif task == "start_wheatfarming" then
            start_wheatfarming()
        elseif task == "left_side" then
            start_left()
            change_side()
            home()
            emptyInv()
            -- right side
            fixInv()
            start_right()
            change_side()
            home()
            emptyInv()
            start_wheatfarming()
        elseif task == "right_side" then
            start_right()
            change_side()
            home()
            emptyInv()
            start_wheatfarming()
        elseif task == "iddle" then
            home()
            fixInv()
            start_wheatfarming()
        elseif task == "home" then
            home()
            fixInv()
            start_wheatfarming()
        end

    else
        startup()
    end
end

function startup()
    clearScreen()
    print("Place a chest to the left, under me and to the right")
    print("Chest to the left will be filled with seeds till it's full")
    print("Chest under me will be filled with wheat")
    print("Chest to the right will be filled with seeds when the first one is full")
    print(" ")
    print(" ")
    print("Press enter when chest setup is done")
    local chest_setup = false
    while (chest_setup == false) do
        local event, param = os.pullEvent()
        if event == "key" and param == 28 then -- key 28 is the 'enter' key
            chest_setup = true
        end
        sleep(0.2)
    end

    clearScreen()

    print("Put a modem in slot 1")
    repeat
        local data = turtle.getItemDetail(switch_slot)
        sleep(0.2)
    until data ~= nil and string.find(data.name, "ComputerCraft")

    clearScreen()

    print("How many blocks can I go left?")
    blocksLeft = tonumber(read())
    print(" ")
    print("How many blocks can I go right?")
    blocksRight = tonumber(read())
    print(" ")
    print("How many blocks can I go forward?")
    blocksForward = tonumber(read())
    print(" ")

    clearScreen()

    print("Is this the right input: ")
    print("Left space: " .. blocksLeft)
    print("Right space: " .. blocksRight)
    print("Length: : " .. blocksForward)
    print("Total field: " .. (blocksLeft + blocksRight + 1)*blocksForward)
    print(" ")
    print(" ")
    print("Press enter if it's correct, otherwhise: Ctrl+R for reboot")

    local rightInput = false
    while (rightInput == false) do
        local event, param = os.pullEvent()
        if event == "key" and param == 28 then -- key 28 is the 'enter' key
            rightInput = true
        end
        sleep(0.2)
    end

    clearScreen()
    print("Saving settings")
    print("---------------")
    print("Making folder")
    fs.makeDir("wheatfarmFolder")

    print("Saving first input")
    local h = fs.open("wheatfarmFolder/firstInput", "w")
    h.writeLine(blocksLeft)
    h.writeLine(blocksRight)
    h.writeLine(blocksForward)
    h.close()

    print("Saving currentSide and task  variable")
    local h = fs.open("wheatfarmFolder/currentSide", "w")
    h.writeLine(currentSide)
    h.close()
    local h = fs.open("wheatfarmFolder/task", "w")
    h.writeLine("start_wheatfarming")
    h.close()

    print("Saving counter variables")
    local h = fs.open("wheatfarmFolder/harvested", "w")
    h.writeLine(seeds_chest1)
    h.writeLine(seeds_chest2)
    h.writeLine(harvest_chest)
    h.writeLine(planted)
    h.writeLine(harvested)
    h.close()

    print("Checking my x y z coordinates and direction")
    checkGps()
    print("Saving home position")
    local h = fs.open("wheatfarmFolder/home", "w")
    h.writeLine(x)
    h.writeLine(y)
    h.writeLine(z)
    h.writeLine(direction)
    h.close()

    home_direction = direction

    print(" ")
    print("My X: " .. x)
    print("My Y: " .. y)
    print("My Z: " .. z)
    print("My direction: " .. direction)
    print("")

    home_x = x
    home_y = y
    home_z = z

    print("Making startup file")
    if fs.exists("startup") then
        fs.delete("startup")
        local h = fs.open("startup", "w")
        h.writeLine("sleep(5)")
        h.writeLine("shell.run('" .. shell.getRunningProgram() .. "')")
        h.close()
    else
        h.writeLine("sleep(5)")
        local h = fs.open("startup", "w")
        h.writeLine("shell.run('" .. shell.getRunningProgram() .. "')")
        h.close()
    end

    print("Starting with wheat farm")
    start_wheatfarming()
end

function printInfo()
    clearScreen()
    print("X: " .. x )
    print("Y: " .. y)
    print("Z: " .. z)
    print("Facing: " .. direction)
    print("--------------------------------")
    if task == "start_wheatfarming" then
        print("Currently starting up to farm")
    elseif task == "left_side" then
        print("Currently farming the left side of the farm")
    elseif task == "right_side" then
        print("Currently farming the right side of the farm")
    elseif task == "emptyInv" then
        print("Currenlty emptying inventory")
    elseif task == "home" then
        print("Going to home position")
    elseif task == "fixInv" then
        print("Fixing inventory")
    end
    print("--------------------------------")
    print("Home X: " .. home_x)
    print("Home Y: " .. home_y)
    print("Home Z: " .. home_z)
    print("Home direction: " .. home_direction)

end

function printIddleInfo(sleep_timer)
    clearScreen()
    print("Waiting " .. sleep_timer .. " seconds to start farming again")
    print("---------------------------------------")
    print("Seeds put in left chest: " .. seeds_chest1)
    print("Seeds put in right chest: " .. seeds_chest2)
    print("Wheat put in center chest: " .. harvest_chest)
    print("Wheat planted: " .. planted)
    print("Wheat harvested: " .. harvested)
    print("")
    print("(CTRL+R) Reboot to start farming.")

end

function clearScreen()
    term.clear()
    term.setCursorPos(1,1)
end

function checkGps()
    -- switch modem and check if slot is now a hoe
    turtle.select(switch_slot)
    local data = turtle.getItemDetail(switch_slot)
    if data == nil then
        print("Put a modem in slot 1")
        repeat
            data = turtle.getItemDetail(switch_slot)
            sleep(0.2)
        until data ~= nil and string.find(data.name, "ComputerCraft")
    elseif string.find(data.name, "ComputerCraft") == nil then
        turtle.equipLeft()
        data = turtle.getItemDetail(switch_slot)
        if string.find(data.name, "ComputerCraft") == nil then
            turtle.equipLeft()
            print("Put a modem in slot 1")
            repeat
                data = turtle.getItemDetail(switch_slot)
                sleep(0.2)
            until data ~= nil and string.find(data.name, "ComputerCraft")
        end
    end

    turtle.equipLeft()
    data = turtle.getItemDetail(switch_slot)
    if data == nil or string.find(data.name, "hoe") == nil then
        print("Put a diamond hoe in slot 1")
        repeat
            data = turtle.getItemDetail(switch_slot)
            sleep(0.2)
        until data ~= nil and string.find(data.name, "hoe")
    end

    -- check gps
    gps_locate()
    turtle.equipLeft()
end

function gps_locate()
    local new_x, new_y, new_z = gps.locate(5)
    if not new_x then
        gps_locate_broken()
    else
        x = math.floor(new_x)
        y = math.floor(new_y)
        z = math.floor(new_z)
    end
    -- check direction
    if turtle.forward() == true then
        local second_x, second_y, second_z = gps.locate(5)
        if not second_x then
            gps_locate_broken()
        else
            if math.floor(second_y) ~= y then
                if y < math.floor(second_y) then
                    direction = "S"
                else
                    direction = "N"
                end
            elseif math.floor(second_x) ~= x then
                if x > math.floor(second_x) then
                    direction = "W"
                else
                    direction = "E"
                end
            end
            turtle.back()
        end
    elseif turtle.back() == true then
        local second_x, second_y, second_z = gps.locate(5)
        if not second_x then
            gps_locate_broken()
        else
            if math.floor(second_y) ~= y then
                if y > math.floor(second_y) then
                    direction = "S"
                else
                    direction = "N"
                end
            elseif math.floor(second_x) ~= x then
                if x < math.floor(second_x) then
                    direction = "W"
                else
                    direction = "E"
                end
            end
            turtle.back()
        end
    end

end

function gps_locate_broken()
    clearScreen()
    print("Failed to get my location!")
    print("Press enter to retry")
    local checked_gps = false
    while (checked_gps == false) do
        local event, param = os.pullEvent()
        if event == "key" and param == 28 then -- key 28 is the 'enter' key
            checked_gps = true
        end
        second_x, second_y, second_z = gps.locate(5)
        if second_x then
            checked_gps = true
        end
        sleep(0.2)
    end
    clearScreen()
end

init()