local launch, cfg, main = ...

local vfs = require("vfs")

local listMode = { }

listMode.controls = { 
	common.newButton(2, 2, 60, 20, "New", function()
		listMode:New()
	end),
	common.newButton(66, 2, 60, 20, "Copy", function()
		listMode:CopySel()
	end, function()
		return listMode.sel ~= nil
	end),
	common.newButton(130, 2, 60, 20, "Rename", function()
		listMode:RenameSel()
	end, function()
		return listMode.sel ~= nil
	end),
	common.newButton(194, 2, 60, 20, "Delete", function()
		listMode:DeleteSel()
	end, function()
		return listMode.sel ~= nil
	end),
}

function listMode:BuildList()
	self.list = { }
	vfs.scan(true)
	for _, file in ipairs(vfs.root.files) do
		if file.name:lower():match("%.xml$") and file.name:lower() ~= "settings.xml" then
			local build = { }
			build.fileName = file.name
			local dbXML, errMsg = common.xml.LoadXMLFile(file.name)
			if dbXML and dbXML[1].elem == "PathOfBuilding" then
				for _, node in ipairs(dbXML[1]) do
					if type(node) == "table" and node.elem == "Build" then
						build.className = node.attrib.className
						build.ascendClassName = node.attrib.ascendClassName
						build.level = tonumber(node.attrib.level) or 1
					end
				end	
			end
			table.insert(self.list, build)
		end
	end
	self:SortList()
end

function listMode:SortList()
	local oldSelFileName = self.sel and self.list[self.sel].fileName
	table.sort(self.list, function(a, b) return a.fileName:upper() < b.fileName:upper() end)
	if oldSelFileName then
		self:SelFileByName(oldSelFileName)
	end
end

function listMode:EditInit(finFunc)
	self.edit = self.sel
	self.editFinFunc = finFunc
	self.editField = common.newEditField(self.list[self.sel].fileName:gsub(".xml$",""), nil, "[%w _+.()]")
	self.editField.x = 2
	self.editField.y = 6 + self.sel * 20
	self.editField.width = cfg.screenW
	self.editField.height = 16
end

function listMode:EditFinish()
	local msg = self.editFinFunc(self.editField.buf)
	if msg then
		launch:ShowPrompt(1, 0.5, 0, msg.."\n\nEnter/Escape to dismiss")
		return
	end
	self.edit = nil
	self.editField = nil
end

function listMode:EditCancel()
	self.sel = nil
	self.edit = nil
	self.editField = nil
	self:BuildList()	
end

function listMode:New()
	table.insert(self.list, 1, { fileName = "", level = 1 })
	self.sel = 1
	self:EditInit(function(buf)
		if #buf < 1 then
			return "No name entered"
		end
		local fileName = buf .. ".xml"
		local outFile, msg = io.open(fileName, "r")
		if outFile then
			outFile:close()
			return "'"..fileName.."' already exists"
		end
		outFile, msg = io.open(fileName, "w")
		if not outFile then
			return "Couldn't create '"..fileName.."': "..msg
		end
		outFile:write('<?xml version="1.0" encoding="UTF-8"?>\n<PathOfBuilding>\n</PathOfBuilding>')
		outFile:close()
		self.list[self.sel].fileName = fileName
		self:SortList()
	end)
end

function listMode:CopySel()
	local srcName = self.list[self.sel].fileName
	table.insert(self.list, self.sel + 1, copyTable(self.list[self.sel]))
	self.sel = self.sel + 1
	self.list[self.sel].fileName = srcName:gsub(".xml$","") .. " (copy)"
	self:EditInit(function(buf)
		if #buf < 1 then
			return "No name entered"
		end
		local inFile, msg = io.open(srcName, "r")
		if not inFile then
			return "Couldn't copy '"..srcName.."': "..msg
		end
		local dstName = buf .. ".xml"
		local outFile, msg = io.open(dstName, "r")
		if outFile then
			outFile:close()
			return "'"..dstName.."' already exists"
		end
		outFile, msg = io.open(dstName, "w")
		if not outFile then
			return "Couldn't create '"..dstName.."': "..msg
		end
		outFile:write(inFile:read("*a"))
		inFile:close()
		outFile:close()
		self.list[self.sel].fileName = dstName
		self:SortList()
	end)
end

function listMode:RenameSel()
	local oldName = self.list[self.sel].fileName
	self:EditInit(function(buf)
		if #buf < 1 then
			return "No name entered"
		end
		local newName = buf .. ".xml"
		if newName == oldName then
			return
		end
		if newName:lower() ~= oldName:lower() then
			local newFile = io.open(newName, "r")
			if newFile then
				newFile:close()
				return "'"..newName.."' already exists"
			end
		end
		local res, msg = os.rename(oldName, newName)
		if not res then
			return "Couldn't rename '"..oldName.."' to '"..newName.."': "..msg
		end
		self.list[self.sel].fileName = newName
		self:SortList()
	end)
end

function listMode:DeleteSel()
	launch:ShowPrompt(1, 0, 0, "Are you sure you want to delete\n'"..self.list[self.sel].fileName.."' ? (y/n)", function(key)
		if key == "y" then
			os.remove(self.list[self.sel].fileName)
			self:BuildList()
			self.sel = nil
		end
		return true
	end)
end

function listMode:SelFileByName(selFileName)
	self.sel = nil
	for index, build in ipairs(self.list) do
		if build.fileName == selFileName then
			self.sel = index
			break
		end
	end
end

function listMode:Init(selFileName)
	self:BuildList()
	self:SelFileByName(selFileName)
end

function listMode:Shutdown()
	if self.edit then
		self:EditCancel()
	end
end

function listMode:OnFrame(inputEvents)
	for id, event in ipairs(inputEvents) do
		if event.type == "KeyDown" then
			self:OnKeyDown(event.key, event.doubleClick)
		elseif event.type == "KeyUp" then
			self:OnKeyUp(event.key)
		elseif event.type == "Char" then
			self:OnChar(event.key)
		end
	end
	common.controlsDraw(self)
	for index, build in ipairs(self.list) do
		local y = 4 + index * 20
		if self.sel == index then
			SetDrawColor(1, 1, 1)
		else
			SetDrawColor(0.5, 0.5, 0.5)
		end
		DrawImage(nil, 0, y, cfg.screenW, 20)
		if self.sel == index then
			SetDrawColor(0.33, 0.33, 0.33)
		else
			SetDrawColor(0, 0, 0)
		end
		DrawImage(nil, 0, y + 1, cfg.screenW, 18)
		if self.edit == index then
			self.editField:Draw(2, y + 2, 16)
		else
			if self.sel == index then
				SetDrawColor(1, 1, 1)
			else
				SetDrawColor(0.8, 0.8, 0.8)
			end
			DrawString(4, y + 2, "LEFT", 16, "VAR", build.fileName:gsub(".xml",""))
			DrawString(304, y + 2, "LEFT", 16, "VAR", string.format("Level %d %s", build.level, build.ascendClassName or build.className or "?"))
		end
	end
end

function listMode:OnKeyDown(key, doubleClick)
	if self.edit then
		if key == "RETURN" then
			self:EditFinish()
		elseif key == "ESCAPE" then
			self:EditCancel()
		else
			self.editField:OnKeyDown(key)
		end
	elseif key == "LEFTBUTTON" then
		self.selControl = nil
		local cx, cy = GetCursorPos()
		for _, control in pairs(self.controls) do
			if control.IsMouseOver and control:IsMouseOver() then
				control:OnKeyDown(key)
				self.selControl = control
				return
			end
		end
		self.sel = nil
		for index, fileName in ipairs(self.list) do
			local y = 4 + index * 20
			if cy >= y and cy < y + 20 then
				if doubleClick then
					main:SetMode("BUILD", self.list[index].fileName)
				else
					self.sel = index
				end
				return
			end
		end
	elseif key == "RETURN" then
		if self.sel then
			main:SetMode("BUILD", self.list[self.sel].fileName)
		end
	elseif key == "UP" then
		if not self.sel then
			self.sel = #self.list
		else
			self.sel = (self.sel - 2) % #self.list + 1
		end
	elseif key == "DOWN" then
		if not self.sel then
			self.sel = 1
		else
			self.sel = self.sel % #self.list + 1
		end
	elseif key == "F2" then
		if self.sel then
			self:RenameSel()
		end
	elseif key == "DELETE" then
		if self.sel then
			self:DeleteSel()
		end
	end
end

function listMode:OnKeyUp(key)
	if self.edit then
		self.editField:OnKeyUp(key)
	elseif self.selControl then
		self.selControl:OnKeyUp(key)
		self.selControl = nil
	end
end

function listMode:OnChar(key)
	if self.edit then
		self.editField:OnChar(key)
	end
end

return listMode