local component = require("component")
local robot = require("robot")
local computer = require("computer")
local inv = component.inventory_controller
local nav = component.navigation
local geo = component.geolyzer
local vector = require("vector")
local side = require("sides")
local slot = 1
local startx, starty, startz = nav.getPosition()

local function harvest()
	if geo.analyze(0).growth == 1 then
		robot.swingDown()
		robot.suckDown()
		return true
	else
		return false
	end
end
local function plant()
print("trying to plant")
local slot = 1
if inv.getStackInInternalSlot() then
	while not inv.getStackInInternalSlot().label == "Seeds" do
			if not slot == 16 then
      slot = slot + 1
   else
       slot = 1
   end
			robot.select(slot)
	end
end
if inv.getStackInInternalSlot(slot) then
	if inv.getStackInInternalSlot(slot).label == "Seeds" then
		inv.equip()
		robot.placeDown()
return true

	else
		return false
	end
end
	return true
end
local function faceNorth()
	local face = nav.getFacing()
	if not face == 2 then
		elseif face == 3 then
			robot.turnRight()
      robot.turnRight()
		elseif face == 4 then
			robot.turnRight()
		elseif face == 5 then
			robotapi.turnLeft()
		else
		return false
		end
	return true
end
local function faceSouth()
	local face = nav.getFacing()
	if not face == 3 then
		elseif face == 2 then
			robot.turnAround()
		elseif face == 5 then
			robot.turnRight()
		elseif face == 4 then
			robot.turnLeft()
		else
		return false
		end
	return true
end
local function faceEast()
	local face = nav.getFacing()
	if not face == 5 then
		print("Faces East")
		elseif face == 4 then
			robot.turnAround()
		elseif face == 2 then
			robot.turnRight()
		elseif face == 3 then
			robot.turnLeft()
		else
		return false
	end
	return true
end
local function faceWest()
	local face = nav.getFacing()
	if not face == 4 then
		elseif face == 5 then
			robot.turnAround()
		elseif face == 3 then
			robot.turnRight()
		elseif face == 2 then
			robot.turnLeft()
		else
			return false
		end
	return true
end
local function goThere(gox, goz)
	local beginx, beginy, beginz = nav.getPosition()
	if beginx < gox then
		repeat
			faceEast()
			robot.forward()
		until beginx == gox
	elseif beginx > gox then
		repeat
			faceWest()
			robot.forward()
		until beginx == gox
	end
	if beginz < goz then
		repeat
			faceSouth()
			robot.forward()
		until beginz == goz
	elseif beginz > goz then
		repeat
			faceNorth()
			robot.forward()
		until beginz == goz
	end
	return true
end
local function charge()
	gothere(startx, startz)
	faceNorth()
	local shestside = checkChestSide()
	local chestsize = getInventorySize(chestside)
	local slot = 0
	while not slot == chestsize do
		slot = slot + 1
		select(slot)
		dropIntoSlot(chestside, slot)
	end
	while not computer.energy == computer.maxEnergy do
		os.sleep(100)
	end
end
local function chargeTest()
	local charge = tonumber(computer.energy())
	local invslot = 0
	local freeslots = robot.inventorySize()
  local slot = 1
	if charge / tonumber(computer.maxEnergy()) <= 0.1 then
		charge()
		return true
	else
		repeat
			slot = slot + 1
			robot.select(slot)
			if robot.count(slot) ~= 0 then
				freeslots = freeslots -1
			end

		until slot == robot.inventorySize()
		if freeslots <= 2 then
			charge()
			return true
		end
	end
end
local function checkChestSide()
	local chestSide
	faceNorth()
	if not getInventorySize(0) == "no inventory" then
		chestSide = 0
	elseif not getInventorySize(1) == "no inventory" then
		chestSide = 1
	elseif not getInventorySize(2) == "no inventory" then
		chestSide = 2
	elseif not getInventorySize(3) == "no inventory" then
		chestSide = 3
	elseif not getInventorySize(4) == "no inventory" then
		chestSide = 4
	elseif not getInventorySize(5) == "no inventory" then
		chestSide = 5
	end
	return chestSide
end
local function farm(fieldsize)
	faceNorth()
	local endx = startx + fieldsize -1
	local endz = startz - fieldsize -1
	local currentx = startx
	local currentz = startz
	local currenty = 0

	repeat
		if not currentx == startx then
			faceNorth()
			robot.forward()
		end
		while not currentx == startx do
			faceEast()
			robot.forward()
			currentx, currenty, currentz = nav.getPosition()
		end
		repeat
			harvest()
			plant()
			faceEast()
			robot.forward()
			currentx, currenty, currentz = nav.getPosition()
		until currentx == endx
		chargeTest()
	until currentz == endz
	charge()
end

print("Enter Field Size: ")
local fieldsizein = tonumber(io.read())
while true do
    farm(fieldsizein)
end