--[[
Author: TheOriginalBIT
Version: 1.0
Created: 13 Dec 2012
Last Update: 13 Dec 2012

License:

COPYRIGHT NOTICE
Copyright © 2012-2013 Joshua Asbury a.k.a TheOriginalBIT [theoriginalbit@gmail.com]

Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
associated documentation files (the "Software"), to deal in the Software without restriction,
including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense,
copies of the Software, and to permit persons to whom the Software is furnished to do so,
subject to the following conditions:

-The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
-Visible credit is given to the original author.
-The software is distributed in a non-profit way.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
]]--


--[[

Variables you may change

]]

saplingSlot		= 1
bonemealSlot	= 2 -- This is only compatible with TurtleOS 1.4 and above
woodSlot		= 3 -- This is just starting dig slot, used to compare

deployBonemealSide = "bottom"	--[[ Should be: any side without peripheral or the front. Must have wire connecting to something that can deploy bonemeal i.e. RedPower Deployer
								This setting is for TurtleOS 1.3 only. ]]
								 

retainItems		= false		-- For TurtleOS 1.3 compatibilit

itemDropOffSide = "left"	--[[ Should be: left, right, up, down, back (cannot be front as thats where tree must grow)
								 For TurtleOS 1.4 can be chest or RedPower Relay, etc, 1.3 must be something such as Buildcraft obsidian pipe or transposer NOT directly infront ]]

--[[

Program Code

Note: Anything you change beyond this point may cause undesireable effects.

I (TheOriginalBIT) in no way take responsibility for any damage done by
misuse or changing of this code. But please do feel free to change if you
wish.

]]

tArgs = {...}

local function checkInventory()
	x, y = term.getCursorPos()
	while turtle.getItemCount(saplingSlot) == 0 do
		term.setCursorPos(x,y)
		print("Waiting for saplings in slot 1")
		sleep(0.1)
	end
end

local function place(slot)
	turtle.select(slot)
	turtle.place()
end

local function deployBonemeal()
	rs.setOutput(deployBonemealSide, true)
	sleep(0.1)
	rs.setOutput(deployBonemealSide, false)
end

local function waitForSaplingGrowth()
	turtle.select(saplingSlot)
	x, y = term.getCursorPos()
	if (turtle.getItemCount(saplingSlot) == 0) then
		print("No sapling to compare to.\nNo way to check growth.\nQuiting")
		error()
	end
	
	while turtle.compare() do
		if (turtle.getItemCount(bonemealSlot) > 0) then
			if (osVersion >= 1.4) then
				place(bonemealSlot)
			elseif (osVersion == 1.3) then
				deployBonemeal()
			end
			turtle.select(saplingSlot)
		end
		term.setCursorPos(x,y)
		print("Waiting for sapling to grow.")
		sleep(0.1)
	end
	print("Sapling has grown.")
end

local function forward()
	turtle.select(woodSlot)
	turtle.dig()
	if (not turtle.forward() and not turtle.detect()) then
		print("I think I'm out of fuel and for OS compatibility reasons I cant check.")
		error()
	end
	
end

local function up()
	turtle.select(woodSlot)
	while turtle.compareUp() do
		turtle.digUp()
		if (not turtle.up() and not turtle.detectUp()) then
			print("I think I'm out of fuel and for OS compatibility reasons I cant check. On next run usage: "..shell.getRunningProgram().." home")
			error()
		end
		sleep(0.1)
	end
	if (turtle.detectUp()) then -- this makes sure the top leaves are removed to allow big tree growth
		turtle.digUp()
	end
end

local function removeTree()
	print("Removing tree.")
	forward()
	up()
end

local function returnHome()
	while not turtle.detectDown() do
		if (not turtle.down()) then
			print("I think I'm out of fuel and for OS compatibility reasons I cant check. On next run usage: "..shell.getRunningProgram().." home")
			error()
		end
		sleep(0.1)
	end
	if (not turtle.back()) then
		print("I think I'm out of fuel and for OS compatibility reasons I cant check. On next run usage: "..shell.getRunningProgram().." home")
		error()
	end
end

local function emptyInventory()
	for i = 1, tonumber(slotCount), 1 do
		if (i ~= saplingSlot or (osVersion == 1.3 and i ~= bonemealSlot)) then
			turtle.select(i)
			turtle.drop()
		end
	end
end

local function dropOffItems()
	if (itemDropOffSide == "up") or (itemDropOffSide == "down") then
		emptyInventory()
		return
	end
	
	funcTo		= turtle.turnLeft
	funcBack	= turtle.turnRight
	 
	if (itemDropOffSide == "right") then
		funcTo = turtle.turnRight
		funcBack = turtle.turnLeft
	end
	
	if (itemDropOffSide == "back") then funcTo() end
	funcTo()
	emptyInventory()
	if (itemDropOffSide == "back") then funcFrom() end
	funcFrom()
end

local function run()
	while true do
		checkInventory()
		place(saplingSlot)
		if (osVersion >= 1.4) then
			place(bonemealSlot)
		elseif (osVersion == 1.3) then
			deployBonemeal()
		end
		
		waitForSaplingGrowth() -- This is just incase its out of bonemeal
		
		removeTree()
		
		returnHome()
		
		if (not retainItems) then
			dropOffItems()
		end
		
		sleep(0.1)
	end
end

if (not turtle) then
	print("Sorry but this can only run on a turtle.")
	return
end

osVersion = os.version()
osVersion = tonumber(string.sub(osVersion, string.len(osVersion) - 3, string.len(osVersion)))
print("Running on TurtleOS "..osVersion)
if (osVersion < 1.3) then
	print("Unable to run on this OS. Requires TurtleOS 1.3 or higher.")
	return
elseif (osVersion > 1.4) then
	print("Sorry TurtleOS "..osVersion.." is not supported by this version. Please check for an update.")
	return
end

if ((#tArgs == 1) and (tArgs[1] == "home")) then
	returnHome()
elseif ((#tArgs == 1) and (tArgs[1] ~= "home")) then
	print("Usage: "..shell.getRunningProgram())
	print("Usage: "..shell.getRunningProgram().." home")
	return
end

term.clear()
term.setCursorPos(1,1)

slotCount = 9

if (osVersion == 1.4) then
	slotCount = 16
	write("Have you put fuel in me? ")
	answer = read()
	if (tolower(answer) ~= "yes") then
		print("\nPlease refuel me and run again.")
		return
	end
end
print(slotCount)
run()