Turtle Quarry Mk3

Args = {...}
fileName = "quarrytemp"

function printData()
	local path = shell.resolve(fileName)
	data = { ["dir"] = textutils.serialize(dir), ["nDir"] = nDir, ["x"] = x, ["y"] = y, ["z"] = z, ["atBedrock"] = atBedrock, ["dirDown"] = dirDown, ["dirFor"] = dirFor, ["dirBack"] = dirBack, ["tRight"] = tRight, ["qDirRight"] = qDirRight, ["demx"] = demx, ["demy"] = demy }
	if fs.exists(fileName) then
		fs.delete(fileName)
	end
	local file = fs.open(path,"w")
	file.write(textutils.serialize(data))
	file.close()
end

function getData()
	local path = shell.resolve(fileName)
	if fs.exists(fileName) then
		local file = fs.open(path,"r")
		data = textutils.unserialize(file.readAll())
		dir = textutils.unserialize(data["dir"])
		nDir = data["nDir"]
		x,y,z = data["x"],data["y"],data["z"]
		atBedrock = data["atBedrock"]
		dirDown = data["dirDown"]
		dirFor = data["dirFor"]
		dirBack = data["dirBack"]
		tRight = data["tRight"]
		qDirRight = data["qDirRight"]
		demx = data["demx"]
		demy = data["demy"]
		file.close()
		return true
	else
		print("Cannot continue, no data found.")
		return false
	end
end

function printUsage()
	print("Usage:")
	print("quarry <LENGTH> <WIDE> <left/right>")
	print("Quarries a <LENGTH> long and <WIDE> wide hole to bedrock.")
	print("quarry continue")
	print("Continues from where it left off.")
	print("Place enderchest in slot 1.")
end

function getFuel()
	return turtle.getFuelLevel()
end

function dropOff()
	print("Dropping off items, fuel: "..getFuel())
	turtle.digUp()
	turtle.select(1)
	if turtle.placeUp() then
		for i=2,16 do
			turtle.select(i)
			turtle.dropUp()
		end
		turtle.select(1)
		turtle.digUp()
	else
		print("Failed to drop off.")
	end
end

function hasSpace()
	local space = false
	for i=1,16 do
		if turtle.getItemCount(i) == 0 then
			space = true
		end
	end
	if not space then dropOff() end
end

function goDirFor(bool)
	if bool then
		if dir[nDir] == "+X" then x = x + 1 elseif
		dir[nDir] == "-X" then x = x - 1 elseif
		dir[nDir] == "+Y" then y = y + 1 elseif
		dir[nDir] == "-Y" then y = y - 1 end
	else
		if dir[nDir] == "+X" then x = x - 1 elseif
		dir[nDir] == "-X" then x = x + 1 elseif
		dir[nDir] == "+Y" then y = y - 1 elseif
		dir[nDir] == "-Y" then y = y + 1 end
	end
end

function mineFor()
	while turtle.detect() do
		turtle.dig()
		os.sleep(0.5)
	end
	hasSpace()
	return true
end

function mineUp()
	while turtle.detectUp() do
		turtle.digUp()
		os.sleep(0.5)
	end
	hasSpace()
	return true
end

function mineDown()
	turtle.digDown()
	hasSpace()
	return true
end

function goFor()
	if turtle.forward() then
		goDirFor(true)
		printData()
		return true
	else
		return false
	end
end

function goBack()
	if turtle.back() then
		goDirFor(false)
		printData()
		return true
	else
		return false
	end
end

function goUp()
	if turtle.up() then
		z = z + 1
		printData()
		return true
	else
		return false
	end
end

function goDown()
	if turtle.down() then
		z = z - 1
		printData()
		return true
	else
		return false
	end
end

function turnRight()
	turtle.turnRight()
	if nDir < 4 then
		nDir = nDir + 1
	else
		nDir = 1
	end
end

function turnLeft()
	turtle.turnLeft()
	if nDir > 1 then
		nDir = nDir - 1
	else
		nDir = 4
	end
end

function stripFor()
	mineUp()
	if turtle.detectUp() then
		while turtle.detectUp() do
			goBack()
			mineUp()
		end
		goUp()
	end
	mineFor()
	while turtle.detect() do
		goUp()
		mineFor()
	end
	local n = 0
	while turtle.detectDown() do
		mineDown()
		n = n + 1
		if n >= 10 then break end
	end
	if turtle.detectDown() and not atBedrock then
		atBedrock = true
		print("Hit bedrock.")
	end
	return goFor()
end

function turn()
	print("Turning around, fuel: "..getFuel())
	if tRight then
		turnRight()
		while not stripFor() do
			os.sleep(0.5)
		end
		turnRight()
	else
		turnLeft()
		while not stripFor() do
			os.sleep(0.5)
		end
		turnLeft()
	end
	tRight = not tRight
	dirFor = not dirFor
end

function stripDown()
	for i=1,3 do
		mineUp()
		local n = 0
		while turtle.detectDown() do
			mineDown()
			n = n + 1
			if n >= 10 then break end
		end
		if turtle.detectDown() and not atBedrock then
			atBedrock = true
			print("Hit bedrock.")
			break
		end
		goDown()
	end
	turnRight()
	turnRight()
	dirFor = not dirFor
	dirBack = not dirBack
end

function turnTo(str)
	while dir[nDir] ~= str do
		turnRight()
	end
end

function resetPos()
	while z ~= 0 do
		goUp()
	end
	turnTo("-X")
	while x ~= 0 do
		goFor()
	end
	if y < 0 then
		turnTo("+Y")
	else
		turnTo("-Y")
	end
	while y ~= 0 do
		goFor()
	end
	turnTo("+X")
end

function quarry()
	while not atBedrock do
		if qDirRight then
			if dirFor then
				while x < demx do
					stripFor()
				end
			else
				while x > 0 do
					stripFor()
				end
			end
			if dirBack then
				if y > 0 then
					turn()
				else
					stripDown()
				end
			else
				if y < demy then
					turn()
				else
					stripDown()
				end
			end
		else
			if dirFor then
				while x < demx do
					stripFor()
				end
			else
				while x > 0 do
					stripFor()
				end
			end
			if dirBack then
				if y > 0 then
					turn()
				else
					stripDown()
				end
			else
				if y < demy then
					turn()
				else
					stripDown()
				end
			end
		end
	end
	print("Going back to start, fuel: "..getFuel())
	mineDown()
	mineUp()
	resetPos()
	fs.delete(fileName)
	dropOff()
	print("Job done, fuel left: "..getFuel())
end

if #Args == 0 then
	printUsage()
else
	if Args[1] == "continue" then
		if getData() then
			print("Continue with quarry dimensions "..(demx+1).."x"..(demy+1)..".")
			quarry()
		end
	elseif tonumber(Args[1]) == 0 or tonumber(Args[2]) == 0 then
		printUsage()
	else
		if Args[3] == "right" or Args[3] == "left" then
			nDir = 1
			x,y,z = 0,0,0
			atBedrock = false
			dirDown = true
			dirFor = true
			dirBack = false
			if Args[3] == "right" then
				tRight = true
				qDirRight = true
				dir = { [1]="+X", [2]="+Y", [3]="-X", [4]="-Y" }
			else
				tRight = false
				qDirRight = false
				dir = { [1]="+X", [2]="-Y", [3]="-X", [4]="+Y" }
			end
			demx = tonumber(Args[1]) - 1
			demy = tonumber(Args[2]) - 1
			quarry()
		else
			printUsage()
		end
	end
end