local Config   = require('opus.config')
local Event    = require('opus.event')
local itemDB   = require('core.itemDB')
local Socket   = require('opus.socket')
local Terminal = require('opus.terminal')
local UI       = require('opus.ui')
local Util     = require('opus.util')

local colors     = _G.colors
local fs         = _G.fs
local multishell = _ENV.multishell
local network    = _G.network
local os         = _G.os
local shell      = _ENV.shell
local term       = _G.term

--UI.Button.defaults.focusIndicator = ' '
UI:configure('Turtles', ...)

local config = { }
Config.load('Turtles', config)

local options = {
	turtle      = { arg = 'i', type = 'number', value = config.id or -1,
								 desc = 'Turtle ID' },
	tab         = { arg = 's', type = 'string', value = config.tab or 'Sel',
								 desc = 'Selected tab to display' },
	help        = { arg = 'h', type = 'flag',   value = false,
								 desc = 'Displays the options' },
}

local SCRIPTS_PATH = 'packages/common/etc/scripts'

local nullTerm = Terminal.getNullTerm(term.current())
local socket

local page = UI.Page {
	coords = UI.Window {
		backgroundColor = colors.black,
		height = 3,
	},
	tabs = UI.Tabs {
		x = 1, y = 4, ey = -2,
		scripts = UI.ScrollingGrid {
			tabTitle = 'Run',
			backgroundColor = colors.cyan,
			columns = {
				{ heading = '', key = 'label' },
			},
			disableHeader = true,
			sortColumn = 'label',
			autospace = true,
		},
		turtles = UI.ScrollingGrid {
			tabTitle = 'Select',
			backgroundColor = colors.cyan,
			columns = {
				{ heading = 'label',  key = 'label'    },
				{ heading = 'Dist',   key = 'distance' },
				{ heading = 'Status', key = 'status'   },
				{ heading = 'Fuel',   key = 'fuel'     },
			},
			disableHeader = true,
			sortColumn = 'label',
			autospace = true,
		},
		inventory = UI.ScrollingGrid {
			backgroundColor = colors.cyan,
			tabTitle = 'Inv',
			columns = {
				{ heading = '',          key = 'index', width = 2 },
				{ heading = '',          key = 'count', width = 2 },
				{ heading = 'Inventory', key = 'key',  width = UI.term.width - 7 },
			},
			disableHeader = true,
			sortColumn = 'index',
		},
--[[
		policy = UI.ScrollingGrid {
			tabTitle = 'Mod',
			backgroundColor = UI.TabBar.defaults.selectedBackgroundColor,
			columns = {
				{ heading = 'label', key = 'label' },
			},
			values = policies,
			disableHeader = true,
			sortColumn = 'label',
			autospace = true,
		},
		]]
		action = UI.Window {
			tabTitle = 'Action',
			backgroundColor = colors.cyan,
			moveUp = UI.Button {
				x = 5, y = 2,
				text = 'up',
				fn = 'turtle.up',
			},
			moveDown = UI.Button {
				x = 5, y = 4,
				text = 'dn',
				fn = 'turtle.down',
			},
			moveForward = UI.Button {
				x = 9, y = 3,
				text = 'f',
				fn = 'turtle.forward',
			},
			moveBack = UI.Button {
				x = 2, y = 3,
				text = 'b',
				fn = 'turtle.back',
			},
			turnLeft = UI.Button {
				x = 2, y = 6,
				text = 'lt',
				fn = 'turtle.turnLeft',
			},
			turnRight = UI.Button {
				x = 8, y = 6,
				text = 'rt',
				fn = 'turtle.turnRight',
			},
			info = UI.TextArea {
				x = 15, y = 2,
				inactive = true,
			}
		},
	},
	statusBar = UI.StatusBar {
		values = { },
		columns = {
			{ key = 'status'              },
			{ key = 'distance', width = 6 },
			{ key = 'fuel',     width = 6 },
		},
	},
	notification = UI.Notification(),
	accelerators = {
		[ 'control-q' ] = 'quit',
	},
}

function page:enable(turtle)
	self.turtle = turtle
	UI.Page.enable(self)
end

function page:runFunction(script, nowrap)
	for _ = 1, 2 do
		if not socket then
			socket = Socket.connect(self.turtle.id, 161)
		end

		if socket then
			if not nowrap then
				script = 'turtle.run(' .. script .. ')'
			end
			if socket:write({ type = 'scriptEx', args = script }) then
				local t = socket:read(3)
				if t then
					return table.unpack(t)
				end
				return false, 'Socket timeout'
			end
		end
		socket = nil
	end
	self.notification:error('Unable to connect')
end

function page:runScript(scriptName)
	if self.turtle then
		self.notification:info('Connecting')
		self:sync()

		local cmd = string.format('Script %d %s', self.turtle.id, scriptName)
		local ot = term.redirect(nullTerm)
		pcall(function() shell.run(cmd) end)
		term.redirect(ot)
		self.notification:success('Sent')
	end
end

function page.coords:draw()
	local t = self.parent.turtle
	self:clear()
	if t then
		self:setCursorPos(2, 2)
		local ind = 'GPS'
		if not t.point.gps then
			ind = 'REL'
		end
		self:print(string.format('%s : %d,%d,%d',
			ind, t.point.x, t.point.y, t.point.z))
	end
end

--[[ Inventory Tab ]]--
function page.tabs.inventory:getRowTextColor(row, selected)
	if page.turtle and row.selected then
		return colors.yellow
	end
	return UI.ScrollingGrid.getRowTextColor(self, row, selected)
end

function page.tabs.inventory:draw()
	local t = page.turtle
	Util.clear(self.values)
	if t then
		for k,v in pairs(t.inv or { }) do -- new method (less data)
			local index, count = k:match('(%d+),(%d+)')
			v = {
				index = tonumber(index),
				key = v,
				count = tonumber(count),
			}
			table.insert(self.values, v)
		end

		for _,v in pairs(t.inventory or { }) do
			if v.count > 0 then
				table.insert(self.values, v)
			end
		end

		for _,v in pairs(self.values) do
			if v.index == t.slotIndex then
				v.selected = true
			end
			if v.key then
				v.key = itemDB:getName(v.key)
			end
		end
	end
	self:adjustWidth()
	self:update()
	UI.ScrollingGrid.draw(self)
end

function page.tabs.inventory:eventHandler(event)
	if event.type == 'grid_select' then
		local fn = string.format('turtle.select(%d)', event.selected.index)
		page:runFunction(fn)
	else
		return UI.ScrollingGrid.eventHandler(self, event)
	end
	return true
end

function page.tabs.scripts:draw()
	Util.clear(self.values)
	local files = fs.list(SCRIPTS_PATH)
	for _,path in pairs(files) do
		table.insert(self.values, { label = path, path = fs.combine(SCRIPTS_PATH, path) })
	end
	self:update()
	UI.ScrollingGrid.draw(self)
end

function page.tabs.scripts:eventHandler(event)
	if event.type == 'grid_select' then
		page:runScript(event.selected.label)
	else
		return UI.ScrollingGrid.eventHandler(self, event)
	end
	return true
end

function page.tabs.turtles:getDisplayValues(row)
	row = Util.shallowCopy(row)
	if row.fuel then
		row.fuel = Util.toBytes(row.fuel)
	end
	if row.distance then
		row.distance = Util.round(row.distance, 1)
	end
	return row
end

function page.tabs.turtles:draw()
	Util.clear(self.values)
	for _,v in pairs(network) do
		if v.fuel then
			table.insert(self.values, v)
		end
	end
	self:update()
	UI.ScrollingGrid.draw(self)
end

function page.tabs.turtles:eventHandler(event)
	if event.type == 'grid_select' then
		page.turtle = event.selected
		config.id = event.selected.id
		Config.update('Turtles', config)
		multishell.setTitle(multishell.getCurrent(), page.turtle.label)
		if socket then
			socket:close()
			socket = nil
		end
	else
		return UI.ScrollingGrid.eventHandler(self, event)
	end
	return true
end

function page.statusBar:draw()
	local t = self.parent.turtle
	if t then
		self.values.status = t.status
		self.values.distance = t.distance and Util.round(t.distance, 2)
		self.values.fuel = Util.toBytes(t.fuel)
	end
	UI.StatusBar.draw(self)
end

function page:showBlocks()
	local script = [[
		local function inspect(direction)
			local s,b = turtle['inspect' .. (direction or '')]()
			if not s then
				return 'minecraft:air:0'
			end
			return string.format('%s:%d', b.name, b.metadata)
		end

		local bu, bf, bd = inspect('Up'), inspect(), inspect('Down')
		return string.format('%s\n%s\n%s', bu, bf, bd)
	]]

	local s, m = self:runFunction(script, true)
	self.tabs.action.info:setText(s or m)
end

function page:eventHandler(event)
	if event.type == 'quit' then
		UI:exitPullEvents()

	elseif event.type == 'tab_select' then
		config.tab = event.button.text
		Config.update('Turtles', config)

	elseif event.type == 'button_press' then
		if event.button.fn then
			self:runFunction(event.button.fn, event.button.nowrap)
			self:showBlocks()
		elseif event.button.script then
			self:runScript(event.button.script)
		end
	else
		return UI.Page.eventHandler(self, event)
	end
	return true
end

function page:enable()
	UI.Page.enable(self)
--  self.tabs:activateTab(page.tabs.turtles)
end

if not Util.getOptions(options, { ... }, true) then
	return
end

if options.turtle.value >= 0 then
	for _ = 1, 10 do
		page.turtle = _G.network[options.turtle.value]
		if page.turtle then
			break
		end
		os.sleep(1)
	end
end

Event.onInterval(1, function()
	if page.turtle then
		local t = _G.network[page.turtle.id]
		page.turtle = t
		page:draw()
		page:sync()
	end
end)

if config.tab then
	page.tabs.tabBar:selectTab(config.tab)
end

UI:setPage(page)

UI:pullEvents()