--[[
	Efficient miner

	GPS is required.

	Miner Requires:
		Diamond pick
		Ender Modem
		Plethora scanner
		Bucket
--]]
local Event    = require('opus.event')
local Equipper = require('turtle.equipper')
local GPS      = require('opus.gps')
local Point    = require('opus.point')
local UI       = require('opus.ui')
local Util     = require('opus.util')

local colors     = _G.colors
local fs         = _G.fs
local os         = _G.os
local read       = _G.read
local term       = _G.term
local turtle     = _G.turtle

UI:configure('scanningMiner', ...)

local args = { ... }
local options = {
	chunks      = { arg = 'c', type = 'number', value = -1,
								 desc = 'Number of chunks to mine' },
	setTrash    = { arg = 's', type = 'flag',   value = false,
								 desc = 'Add ignore items' },
	help        = { arg = 'h', type = 'flag',   value = false,
								 desc = 'Displays the options' },
}

local MIN_FUEL = 7500
local LOW_FUEL = 1500
local MAX_FUEL = turtle.getFuelLimit()

local HOME_PT = { x = 0, y = 0, z = 0 }

local DICTIONARY_FILE = 'usr/config/mining.dictionary'
local PROGRESS_FILE   = 'usr/config/scanning_mining.progress'
local STARTUP_FILE = 'usr/autorun/scanningMiner.lua'

local mining
local ignores = {
	ignore = true,
	retain = true,
}

local dictionary = {
	data = Util.readTable(DICTIONARY_FILE) or {
		[ 'minecraft:chest'              ] = 'suck',
		[ 'minecraft:lava'               ] = 'liquid_fuel',
		[ 'minecraft:flowing_lava'       ] = 'liquid_fuel',
		[ 'minecraft:mob_spawner'        ] = 'ignore',
		[ 'minecraft:bedrock'            ] = 'ignore',
		[ 'minecraft:flowing_water'      ] = 'ignore',
		[ 'minecraft:water'              ] = 'ignore',
		[ 'minecraft:air'                ] = 'ignore',
		[ 'minecraft:bucket'             ] = 'retain',
		[ 'computercraft:advanced_modem' ] = 'retain',
		[ 'minecraft:diamond_pickaxe'    ] = 'retain',
		[ 'plethora:module'              ] = 'retain',
	},
}

function dictionary:write()
	Util.writeTable(DICTIONARY_FILE, self.data)
end
function dictionary:mineable(name, damage)
	self.data[name .. ':' .. damage] = nil
end
function dictionary:ignore(name, damage)
	if damage then
		self.data[name .. ':' .. damage] = 'ignore'
	else
		self.data[name] = 'ignore'
	end
end
function dictionary:get(name, damage)
	return self.data[name] or self.data[name .. ':' .. damage]
end
function dictionary:isTrash(name, damage)
	return self:get(name, damage) == 'ignore'
end

local page = UI.Page {
	menuBar = UI.MenuBar {
		buttons = {
			--{ text = 'Mine',   event = 'mine' },
			{ text = 'Ignore', event = 'ignore' },
			{ text = 'Abort',  event = 'abort', x = -8 },
		},
	},
	grid = UI.Grid {
		y = 2, ey = -2,
		sortColumn = 'name',
		columns = {
			{ heading = 'Count',    key = 'count', width = 5, align = 'right' },
			{ heading = 'Resource', key = 'displayName' },
		},
	},
	statusBar = UI.StatusBar {
		columns = {
			{ key = 'status' },
			{ key = 'distance', width = 4 },
			{ key = 'fuel', width = 5 },
		},
	},
	accelerators = {
		[ 'control-q' ] = 'cancel',
	}
}

function page:eventHandler(event)
	local t = self.grid:getSelected()
	if t then
		if event.type == 'mine' then
			dictionary:mineable(t.name, t.damage)
			dictionary:write()

		elseif event.type == 'ignore' then
			dictionary:ignore(t.name)
			dictionary:write()
			self.grid:draw()

		end
	end

	if event.type == 'abort' then
		turtle.abort(true)

	elseif event.type == 'quit' then
		turtle.abort(true)
	end
	UI.Page.eventHandler(self, event)
end

function page.grid:getRowTextColor(row, selected)
	if dictionary:get(row.name, row.damage) == 'ignore' then
		return colors.lightGray
	end
	if row.displayName == self.nextBlock then
		return colors.yellow
	end
	return UI.Grid.getRowTextColor(self, row, selected)
end

local function getChunkCoordinates(diameter, index, x, z)
	local dirs = { -- circumference of grid
		{ xd =  0, zd =  1, heading = 1 }, -- south
		{ xd = -1, zd =  0, heading = 2 },
		{ xd =  0, zd = -1, heading = 3 },
		{ xd =  1, zd =  0, heading = 0 }  -- east
	}
	-- always move east when entering the next diameter
	if index == 0 then
		dirs[4].x = x + 16
		dirs[4].z = z
		return dirs[4]
	end
	local dir = dirs[math.floor(index / (diameter - 1)) + 1]
	dir.x = x + dir.xd * 16
	dir.z = z + dir.zd * 16
	return dir
end

local function getCornerOf(c)
	return math.floor(c.x / 16) * 16, math.floor(c.z / 16) * 16
end

local function isFinished()
	if mining.chunks ~= -1 then
		local chunks = math.pow(mining.diameter-2, 2) + mining.chunkIndex
		if chunks >= mining.chunks then
			return true
		end
	end
end

local function nextChunk()
	local x, z = getCornerOf({ x = mining.x, z = mining.z })
	local points = math.pow(mining.diameter, 2) - math.pow(mining.diameter-2, 2)
	mining.chunkIndex = mining.chunkIndex + 1

	if mining.chunkIndex >= points then
		mining.diameter = mining.diameter + 2
		mining.chunkIndex = 0
	end

	local nc = getChunkCoordinates(mining.diameter, mining.chunkIndex, x, z)

	-- enter next chunk
	mining.x = nc.x
	mining.z = nc.z

	Util.writeTable(PROGRESS_FILE, mining)

	return not isFinished()
end

local function status(newStatus)
	turtle.setStatus(newStatus)
	page.statusBar:setValue('status', newStatus)
	page.statusBar:draw()
	page:sync()
end

local function refuel()
	if turtle.getFuelLevel() < MIN_FUEL then
		local oldStatus = turtle.getStatus()
		status('refueling')

		turtle.refuel('minecraft:coal:0', 32)
		if turtle.getFuelLevel() < MIN_FUEL then
			turtle.eachFilledSlot(function(slot)
				if turtle.getFuelLevel() < MIN_FUEL then
					turtle.select(slot.index)
					turtle.refuel(64)
				end
			end)
		end
		status(oldStatus)
	end

	turtle.select(1)
end

local function safeGoto(x, z, y, h)
	local oldStatus = turtle.getStatus()

	while not turtle.go({ x = x, z = z, y = y or turtle.point.y, heading = h }) do
		status('stuck')
		if turtle.isAborted() then
			return false
		end
		os.sleep(3)
	end
	turtle.setStatus(oldStatus)
	return true
end

local function safeGotoY(y)
	local oldStatus = turtle.getStatus()
	while not turtle.gotoY(y) do
		status('stuck')
		if turtle.isAborted() then
			return false
		end
		os.sleep(1)
	end
	turtle.setStatus(oldStatus)
	return true
end

local function unload()
	local oldStatus = turtle.getStatus()
	status('unloading')
	local pt = Util.shallowCopy(turtle.point)
	safeGotoY(0)

	safeGoto(0, 0, 0)
	if not turtle.detectUp() then
		error('no chest')
	end
	local slots = turtle.getFilledSlots()
	for _,slot in pairs(slots) do
		local action = dictionary:get(slot.name, slot.damage)
		if not ignores[action] then
			turtle.select(slot.index)
			turtle.dropUp(64)
		end
	end
	turtle.condense()
	turtle.select(1)
	safeGoto(pt.x, pt.z, 0, pt.heading)

	safeGotoY(pt.y)
	status(oldStatus)
end

local function ejectTrash()
	turtle.eachFilledSlot(function(slot)
		if dictionary:isTrash(slot.name, slot.damage) then
			turtle.select(slot.index)
			turtle.dropDown(64)
		end
	end)
end

local function checkSpace()
	if turtle.getItemCount(15) > 0 then
		refuel()
		local oldStatus = turtle.getStatus()
		status('condensing')
		ejectTrash()
		turtle.condense()
		if turtle.getItemCount(15) > 0 then
			unload()
		end
		status(oldStatus)
		turtle.select(1)
	end
end

local function collectDrops(suckAction)
	for _ = 1, 50 do
		checkSpace()
		if not suckAction() then
			break
		end
	end
end

local function scan()
	local scanner = Equipper.equipLeft('plethora:scanner')
	local blocks = scanner.scan()
	Equipper.equipLeft('minecraft:diamond_pickaxe')
	local throttle = Util.throttle()

	local bedrock = -256
	local counts = { }

	for _, b in pairs(blocks) do
		if b.x == 0 and b.y == 0 and b.z == 0 then
			b.name = 'minecraft:air'
		end
		b.x = b.x + turtle.point.x
		b.y = b.y + turtle.point.y
		b.z = b.z + turtle.point.z
		throttle()

		if b.name == 'minecraft:bedrock' then
			if b.y > bedrock then
				bedrock = b.y
			end
		end
	end

	Util.filterInplace(blocks, function(b)
		throttle()
		if b.y >= bedrock then
			b.action = dictionary:get(b.name, b.metadata) or 'mine'

			if ignores[b.action] then
				return false
			end

			if b.action == 'liquid_fuel' and (b.y <= bedrock or b.metadata > 0) then
				return false
			end

			local key = b.name .. ':' .. b.metadata
			if not counts[key] then
				counts[key] = {
					displayName = key,
					name = b.name,
					damage = b.metadata,
					count = 1
			}
			else
				counts[key].count = counts[key].count + 1
			end
			return true
		end
	end)

	turtle.select(1)

	local dirty = true

	local function display()
		if dirty then
			page.grid:draw()
			page:sync()
		end
		dirty = false
	end

	page.grid:setValues(counts)
	page.grid:draw()
	display()

	status('mining')

	local i = #blocks
	Point.eachClosest(turtle.point, blocks, function(b)
		if turtle.isAborted() then
			error('aborted')
		end

		page.grid.nextBlock = b.name .. ':' .. b.metadata

		-- Get the action again in case the user has ignored via UI
		b.action = dictionary:get(b.name, b.metadata) or 'mine'
		if b.action == 'suck' or b.action == 'mine' then
			status('mining: ' .. table.concat({ b.x, b.y, b.z }, ', '))

			if b.action == 'suck' then
				local pt = turtle.moveAgainst(b)
				collectDrops(turtle.getAction(pt.direction).suck)
			end
			checkSpace()
			local s, m
			if b.y == bedrock then
				s, m = turtle.digDownAt(b)
			else
				s, m = turtle.digAt(b)
			end
			if not s then
				page.statusBar:setValue('status', b.name .. ' ' .. m)
				page.statusBar:draw()
				page:sync()
			else
				page.statusBar:setValue('mining', m)
			end
			dirty = true
		elseif b.action == 'liquid_fuel' then
			if turtle.getFuelLevel() < (MAX_FUEL - 1000) then
				if turtle.placeAt(b, 'minecraft:bucket:0') then
					turtle.refuel()
					turtle.select(1)
					dirty = true
				end
			end
		end
		local key = b.name .. ':' .. b.metadata
		counts[key].count = counts[key].count - 1
		i = i - 1
		display()
	end)
end

local function mineChunk()
	local pts = { }

	for i = 1, math.ceil(mining.home.y / 16) do
		pts[i] = { x = mining.x + 8, z = mining.z + 8, y = (i - 1) * 16 + 8 }
		if pts[i].y > mining.home.y - 8 then
			pts[i].y = mining.home.y - 8
		end
		pts[i].y = math.max(pts[i].y, 8)
		pts[i].y = pts[i].y - mining.home.y -- abs to rel
	end

	Point.eachClosest(turtle.point, pts, function(pt)
		if turtle.isAborted() then
			error('aborted')
		end
		local chunks = math.pow(mining.diameter-2, 2) + mining.chunkIndex
		status(string.format('scanning %d %d-%d',
			chunks,
			pt.y + mining.home.y - 8,
			pt.y + mining.home.y + 8))

		turtle.select(1)
		safeGoto(pt.x, pt.z, pt.y)
		scan()

		if turtle.getFuelLevel() < LOW_FUEL then
			refuel()
			local veryMinFuel = Point.turtleDistance(turtle.point, HOME_PT) + 512
			if turtle.getFuelLevel() < veryMinFuel then
				error('Not enough fuel to continue')
			end
		end
	end)
end

local function addTrash()
	local slots = turtle.getFilledSlots()

	for _,slot in pairs(slots) do
		local e = dictionary:get(slot.name, slot.damage)
		if not e or e ~= 'retain' then
			dictionary:ignore(slot.name, slot.damage)
		end
	end

	dictionary:write()
end

-- Startup logic
if not Util.getOptions(options, args) then
	return
end

-- in plethora code, we can override initialize with a scanner version
turtle.initialize = function()
	Equipper.equipModem('right')
	Equipper.equipLeft('minecraft:diamond_pickaxe')

	local function verify(item)
		if not turtle.has(item) then
			error('Missing: ' .. item)
		end
	end

	local items = { 'minecraft:bucket', 'plethora:module' }
	for _,v in pairs(items) do
		verify(v)
	end

	--os.sleep(5)
	local pt = GPS.getPoint(2) or error('GPS not found')
	local scanner = Equipper.equipLeft('plethora:scanner')
	local facing = scanner.getBlockMeta(0, 0, 0).state.facing
	pt.heading = Point.facings[facing].heading
	turtle.setPoint(pt, true)
	Equipper.equipLeft('minecraft:diamond_pickaxe')
end

local function main()
	repeat
		mineChunk()
	until not nextChunk()
end

local success, msg

if not fs.exists(DICTIONARY_FILE) or options.setTrash.value then
	print('Place blocks into the turtles inventory to ignore, such as cobble, stone, gravel, etc.')
	print('\nPress enter when ready')
	read()
	addTrash()
end

if not fs.exists(STARTUP_FILE) then
	Util.writeFile(STARTUP_FILE,
		[[print('sleeping for 10 seconds')
os.sleep(10)
shell.openForegroundTab('scanningMiner.lua')]])
	print('Autorun program created: ' .. STARTUP_FILE)
end

Event.addRoutine(function()
	turtle.reset()

	ejectTrash()

	turtle.initialize {
		right = 'computercraft:advanced_modem',
		left  = 'minecraft:diamond_pickaxe',
		required = {
			'minecraft:bucket',
			'plethora:module',
		},
		GPS = true,
		minFuel = 100,
		-- searchFor = 'ironchest:iron_shulker_box_white'
	}

	turtle.setMoveCallback(function()
		page.statusBar:setValue('fuel', Util.toBytes(turtle.getFuelLevel()))
		page.statusBar:setValue('distance', math.floor(Point.distance(turtle.point, HOME_PT)))
		page.statusBar:draw()
		page:sync()
	end)

	mining = Util.readTable(PROGRESS_FILE) or {
		diameter = 1,
		chunkIndex = 0,
		x = 0, z = 0,
		chunks = options.chunks.value,
		home = Point.copy(turtle.point),
		heading = turtle.point.heading, -- always using east for now
	}

	if options.chunks.value ~= -1 then
		mining.chunks = options.chunks.value
	end

	-- use coordinates relative to initial starting point
	turtle.setPoint({
		x = turtle.point.x - mining.home.x,
		y = turtle.point.y - mining.home.y,
		z = turtle.point.z - mining.home.z,
	})

	local distance = Point.distance(
		{ x = turtle.point.x, y = 0, z = turtle.point.z },
		HOME_PT
	)
	local maxDistance = Point.distance(
		HOME_PT,
		{ x = mining.x, y = 0, z = mining.z }
	)

	_G._syslog({ distance = distance, maxDistance = maxDistance })

	if distance > maxDistance + 32 then
		term.clear()
		term.setCursorPos(1, 1)
		_G.printError('WARNING\n\nTurtle is outside the mining area\n')
		print('Max distance:     ' .. math.floor(maxDistance))
		print('Distance to home: ' .. math.floor(distance))
		print('\nIf in a new location, delete usr/config/scanning_miner.progress')
		print('\nWaiting for 30 seconds to begin')
		os.sleep(30)
		page:sync()
	end

	if not fs.exists(PROGRESS_FILE) then
		Util.writeTable(PROGRESS_FILE, mining)
	end

	turtle.set({
		attackPolicy = 'attack',
		digPolicy = 'turtleSafe',
		movementStrategy = 'goto',
	})
	status('mining')

	Event.onTerminate(function()
		turtle.abort(true)
	end)

	if isFinished() then
		success = false
		msg = 'Mining complete'
	else
		success, msg = pcall(main)
	end

	status(success and 'finished' or turtle.isAborted() and 'aborting' or 'error')
	turtle.gotoY(0)
	if turtle.go(HOME_PT) then
		unload()
	end
	turtle.reset()

	Event.exitPullEvents()
end)

UI:setPage(page)
UI:start()

turtle.reset()

if not success and msg then
	_G.printError(msg)
end