Code Preview Copy All

local tArgs = { ... }


--[[
    Release version of Lattix, Version:
	]]
local version = "a0.3"
--[[
    A lot of features from Lattice weren't adopted yet :/ but at least you can enjoy the extra stability
   
    Made by Konlab
   
    If you wonder why the screen is so stable (no flickering) then it's because of the awesomeness of CrazedProgrammmer's Surface API, go check it out here: http://www.computercraft.info/forums2/index.php?/topic/22397-surface-api-162/

	Big thanks to:
	EveryOS (thanks for pointing out the command thing)
]]


--CONFIG
--colors of lattix
local lat_theme = {
	menu_back = colors.red;
	menu_text = colors.white; 
	main_back = colors.white;
	main_text = colors.black;
	selected_text = colors.cyan;
	dir_text = colors.lime;
	multiselect_back = colors.lightGray;
	blank_back = colors.white;
	blank_text = colors.black;
}
--colors of lattix on basic comps
local lat_basic_theme = {
	menu_back = colors.gray;
	menu_text = colors.white; 
	main_back = colors.white;
	main_text = colors.black;
	selected_text = colors.lightGray;
	dir_text = colors.black;
	multiselect_back = colors.white;
	blank_back = colors.white;
	blank_text = colors.black;
}
local dirprefix = "" --this text gets added before dirs
local basic_dirprefix = "[] " --this text gets added before dirs on non-advanced computers
local selectprefix = "" --this text gets added before selected things (even in multiselect)
local basic_selectprefix = "> " --this text gets added before selected things (even in multiselect) on non-advanced computers
--which programs to use to open .<xtension> files. Special codes: "run" and "api" and special extension "def" - def will be used for every extension not listed (def must be something), run runs it without args, api loads it with os.loadAPI (yes ikr very useful rn since os.loadAPI doesn't like weird extensions). The empty string indicates no extension
--the file is opened by running the program chosen below with arg #1 being the path to the file
local defaultPrograms = {
	[""] = "/rom/programs/edit.lua",
	["txt"] = "/rom/programs/edit.lua",
	["cfg"] = "/rom/programs/edit.lua",
	["config"] = "/rom/programs/edit.lua",
	["nfp"] = "/rom/programs/fun/advanced/paint.lua",
	["lua"] = "run",
--	["api"] = "api", --if a custom OS sets up os.loadAPI to work with .api and stores its apis in .api then this could work
	["def"] = "/rom/programs/edit.lua"
}
--which programs can be used to create files (think clicking the New... button) Special code: choose means the program is chosen manually through a dialog
--these programs are run with the path to the file that is to be created as first argument
local editors = {
	["Other"] = "choose",
	["NFP painting"] = "/rom/programs/fun/advanced/paint.lua",
	["Lua script"] = "/rom/programs/edit.lua",
	["Text file"] = "/rom/programs/edit.lua"
	
	
	
}
--Extensions for the above editors
local editors_extensions = {
	["NFP painting"] = "nfp",
	["Lua script"] = "lua",
	["Text file"] = "txt"
	
	
}
--paths and links to apis/third party software
local surfacepath = "surface"

--code adopted and modified from CC's source code
local function pastebinGet(paste, path)
	
	if not http then
		error("HTTP disabled", 2)
	end
	--- Attempts to guess the pastebin ID from the given code or URL
	local extractId = function (paste)
		local patterns = {
			"^([%a%d]+){{code}}quot;,
			"^https?://pastebin.com/([%a%d]+){{code}}quot;,
			"^pastebin.com/([%a%d]+){{code}}quot;,
			"^https?://pastebin.com/raw/([%a%d]+){{code}}quot;,
			"^pastebin.com/raw/([%a%d]+){{code}}quot;,
		}

		for i = 1, #patterns do
			local code = paste:match( patterns[i] )
			if code then return code end
		end

		return nil
	end

	local get = function (url)
		local paste = extractId( url )
		if not paste then
			error( "Invalid pastebin code.", 0 )
			return
		end

		-- Add a cache buster so that spam protection is re-checked
		local cacheBuster = ("%x"):format(math.random(0, 2^30))
		local response, err = http.get(
			"https://pastebin.com/raw/"..textutils.urlEncode( paste ).."?cb="..cacheBuster
		)

		if response then
			-- If spam protection is activated, we get redirected to /paste with Content-Type: text/html
			local headers = response.getResponseHeaders()
			if not headers["Content-Type"] or not headers["Content-Type"]:find( "^text/plain" ) then
				error( "Pastebin blocked the download due to spam protection. Please complete the captcha in a web browser: https://pastebin.com/" .. textutils.urlEncode( paste ) , 0)
				return
			end

			local sResponse = response.readAll()
			response.close()
			return sResponse
		else
			error (err, 0)
		end
	end

	-- Determine file to download
	local sCode = paste
	local sPath = path
	if fs.exists( sPath ) then
		error( "File already exists", 0 )
		return
	end

	-- GET the contents from pastebin
	local res = get(sCode)
	if res then
		local file = fs.open( sPath, "w" )
		file.write( res )
		file.close()
	end
end
local function wGet(url, sPath)

	if not http then
		error( "HTTP disabled", 2 )
	end

	local function getFilename( sUrl )
		sUrl = sUrl:gsub( "[#?].*" , "" ):gsub( "/+{{code}}quot; , "" )
		return sUrl:match( "/([^/]+){{code}}quot; )
	end

	local function get( sUrl )
		-- Check if the URL is valid
		local ok, err = http.checkURL( url )
		if not ok then
			error( err or "Invalid URL.", 0 )
			return
		end

		local response = http.get( sUrl , nil , true )
		if not response then
			error( "Failed." )
		end

		local sResponse = response.readAll()
		response.close()
		return sResponse
	end

	if fs.exists( sPath ) then
		error( "File already exists", 0 )
		return
	end

	local res = get(url)
	if not res then return end

	local file = fs.open( sPath, "wb" )
	file.write( res )
	file.close()
end


local function program()

--api loading
if not surface then
	if fs.exists(surfacepath) then
		os.loadAPI(surfacepath)
	else
		print("installing missing component: " .. surfacepath)
		pastebinGet("J2Y288mW", surfacepath)
		os.loadAPI(surfacepath)
	end
end
--term args
local home --the directory Lattix is opened in
local root --the root directory that you're not allowed to go above

if #tArgs > 0 then
	home = tArgs[1]
else
	home = ""
end
if #tArgs > 1 then
	root = tArgs[2]
else
	root = ""
end

root = fs.combine("", root)
home = fs.combine("", home)

--variables
local clipboard --a file path/table of files
local clip_cut = false -- original file(s) will be deleted if this is true
local history = {}
local w,h = term.getSize()



--setting up path
if not term.isColor() then
	lat_theme = lat_basic_theme
	dirprefix = basic_dirprefix
	selectprefix = basic_selectprefix
end
local path = root

if fs.isDir(fs.combine(root,home)) then
	path = fs.combine(root,home)
elseif fs.isDir(root) then
	path = root
else
	error("Not valid root folder",0)
end
local scroll = 0
local selected = 0
local endprogram = false
local isCtrlDown = false
local isShiftDown = false
local selection = {}
local isMultiSelect = false

local index = 0
local empty = function() end
local shell_input = ""

local itemNames = {} --displayed texts
local itemPaths = {} --paths to items
local customTitle --if not nil then the title of the program is replaced with this

local prewrite = {} --for fancy command stuff

--setting up the second session
local alt_histories = {
{path}, {path}, {path}, {path}
}
local alt_paths = {
	[1] = path,
	[2] = path,
	[3] = path,
	[4] = path
}
local alt_scrolls = {
	0,0,0,0
}
local alt_selected = {
	0,0,0,0
}
local cSession = 1

local surf = surface.create(w,h," ",lat_theme.blank_back, lat_theme.blank_text) 

local pathValid = true

--functions yet undefined:
local redraw --so that any gui drawing can call a redraw
local remap --so that redraw and remap can be overwritten if neccessary
local restore --so that special popups that use the main loop can restore the file browser functionality
--helpers
local function mathdown(num)
	if num % 1 ~= 0 then
		return math.floor(num)
	end
	return num - 1
end
local function sort(path) --returns a sorted table of folder names and file names at path with folders being first
	local tbl = {}
	items = fs.list(path)
	table.sort(items)
	--insert dirs
	for i=1,#items do
		if fs.isDir(fs.combine(path,items[i])) then
			tbl[#tbl+1] = fs.combine(path, items[i])
		end
	end
	for i=1,#items do
		if not fs.isDir(fs.combine(path,items[i])) then
			tbl[#tbl+1] = fs.combine(path, items[i])
		end
	end
	return tbl
end
local function formatNumber(num)
	local extra = {"B","KB","MB"}
	local eindex = 1
	while num > 1024 do
		num = num / 1024
		num = math.floor(num)
		eindex = eindex + 1
	end
	return tostring(num..extra[eindex])
end
local function posToIndex(maxn,x,y,scroll)
	return y+scroll-2 < maxn and y+scroll-1 or 0
end
local function IndexToPos(index,scroll)
	return 2,index-scroll
end
local function clear()
	surf:clear()
	surf:render()
	term.setCursorPos(1,1)
end

local function readFolderRaw(folder)
	customTitle = nil --to quit all special view modes
	pathValid = true
	path = folder
	itemNames = {}
	itemPaths = {}
	
	local t = sort(folder)
	for i=1,#t do
		itemNames[i] = fs.getName(t[i])
		itemPaths[i] = t[i]
	end
	
end

local function switchFolderRaw(folder) --folder is in absolute path, does not add to history
	readFolderRaw(folder)
	selected = 0
	scroll = 0
	shell_input = ""
end

local function switchFolder(folder) --folder is in absolute path
	history[#history + 1] = path
	switchFolderRaw(folder)
end

local function waitForKeyOrClick()
	while true do
		local e = os.pullEvent()
		if e == "key" or e == "mouse_click" then
			break
		end
	end
end
local function split(inputstr, sep)
        if sep == nil then
                sep = "%s"
        end
        local t={} ; i=1
        for str in string.gmatch(inputstr, "([^"..sep.."]+)") do
                t[i] = str
                i = i + 1
        end
        return t
end
local function switchSessions(newI)
	--if newI == cSession then return end
	alt_histories[cSession] = history
	alt_paths[cSession] = path
	alt_scrolls[cSession] = scroll
	alt_selected[cSession] = selected
	history = alt_histories[newI]
	--after all we need to update the item list
	switchFolderRaw(alt_paths[newI])
	scroll = alt_scrolls[newI]
	selected = alt_selected[newI]
	cSession = newI
end
local function extensionRead(extension, width, bcolor, tcolor, dtext)
	if #prewrite > 0 then
		local tmp = prewrite[1]
		table.remove(prewrite, 1)
		return tmp
	end
	local x,y = term.getCursorPos()
	term.setCursorBlink(true)
	local t = dtext or ""
	local i = #t
	local scroll = 0
	local tbasis
	
	while true do
		surf:drawLine(x,y,x+width,y," ", bcolor,tcolor)
		tbasis = t .. extension
		if i < scroll then
			scroll = i
		end
		if i > scroll + width - #extension then
			scroll = i - width + #extension
		end
		if #tbasis > width then
			if scroll + width > #tbasis then
				scroll = #tbasis - width
			end
			tbasis = tbasis:sub(1+scroll, width+scroll)
		else
			scroll = 0
		end
		surf:drawText(x,y,tbasis,bcolor,tcolor)
		surf:render()
		term.setCursorPos(x+i-scroll, y)
		local ev = {os.pullEvent()}
		local e,k = ev[1], ev[2]
		repeat
			if e == "paste" then
				e = "char"
				table.remove(ev, 1)
				k = table.concat(ev, " ")
			end
			if e == "char" then
				if i == #t then
					t = t .. k
					i = i + #k
				elseif i == 0 then
					t = k .. t
					i = i + #k
				else
					t = t:sub(1,i) .. k .. t:sub(i+1, #t)
					i = i + #k
				end
			end
			
			if e == "key" then
				local mini = 0
				local maxi = #t
				if k == keys.left and i > mini then
					i = i - 1
					break
				end
				if k == keys.right and i < maxi then
					i = i + 1
					break
				end
				
				if k == keys.up then
					i = mini
					break
				end
				if k == keys.down then
					i = maxi
					break
				end
				if k == keys.enter then
					term.setCursorBlink(false)
					return t .. extension
				end
				if k == keys.delete and i < #t then
					t = t:sub(1,i) .. t:sub(i+2,#t)
					break
				end
				if k == keys.backspace and i > 0 then
					t = t:sub(1,i-1) .. t:sub(i+1, #t)
					i = i - 1
					break
				end
			end
		until true
	end
end
--custom dialoges (just the GUI, it also draws it)
local function drawInfobox(txt) --just info text, no buttons
	surf:fillRect(math.floor(w/2-(#txt+2)/2), math.floor(h/2)-2, math.floor(w/2+(#txt+2)/2),math.floor(h/2)+2, " ", lat_theme.menu_back, lat_theme.menu_text)
	surf:drawText(math.floor(w/2-(#txt+2)/2)+1,math.floor(h/2)-1, txt, lat_theme.menu_back, lat_theme.menu_text)
	surf:render()
end
local function drawTextbox(txt) --text feedback
	surf:fillRect(math.floor(w/2-(#txt+2)/2),math.floor(h/2)-2,math.floor(w/2+(#txt+2)/2),math.floor(h/2)+2, " ", lat_theme.menu_back, lat_theme.menu_text)
	surf:drawText(math.floor(w/2-(#txt+2)/2)+1,math.floor(h/2)-1, txt, lat_theme.menu_back, lat_theme.menu_text)
	surf:render()
	term.setCursorPos(math.floor(w/2-(#txt+2)/2)+1,math.floor(h/2)+1)
end
local function drawButtonBox(txt,buttons) --multiple buttons, configurable
	if type(buttons) ~= "table" then
		buttons = {{" Yes ",colors.red,colors.white},{" No ",colors.gray,colors.white}}
	end
	if txt == "" or txt == nil then
		txt = " Are you sure? "
	end
	
	drawTextbox(txt) --reuse the rectangle
	local x,y = term.getCursorPos()
	
	for i=1,#buttons do
		x = math.floor((w/2-(#txt+2)/2)+((#txt+2)/i)*(i-1)) + 1
		if term.isColor() then
			surf:drawText(x,y, buttons[i][1], buttons[i][2], buttons[i][3])
		else
			surf:drawText(x,y,buttons[i][1], lat_theme.main_back, lat_theme.main_text)
		end
	end
	surf:render()
	return y
end
local function drawPopupBox(x,y,buttons) --x and y are the corners
	local ydir = y < h/2 and 1 or -1
	surf:fillRect(x,y,x+15,y+(#buttons+1)*ydir, " ", lat_theme.menu_back, lat_theme.menu_text)
	for k,v in pairs(buttons) do
		surf:drawText(x+1,y+k*ydir, v, lat_theme.menu_back, lat_theme.menu_text)
	end
	surf:render()
end
--custom dialogue functionality (difference from popups is that they stop the main program flow, program cannot close, etc. they are allowed to handle events)
local function infoBox(txt, noredraw)
	drawInfobox(txt)
	waitForKeyOrClick()
	if not noredraw then
		redraw()
	end
end
local function textBox(txt, dtext)
	drawTextbox(txt)
	local resp = extensionRead("", #txt, lat_theme.menu_back, lat_theme.menu_text, dtext)
	redraw()
	return resp
end
local function xtensionTextBox(txt, xts)
	drawTextbox(txt)
	local resp = extensionRead(xts, #txt, lat_theme.menu_back, lat_theme.menu_text)
	redraw()
	return resp
end
local function fileChooseBox(txt)
	return textBox(txt)
end
local function dirChooseBox(txt)
	return textBox(txt)
end
local function buttonBox(txt, buttons)
	local y = drawButtonBox(txt, buttons)
	while true do
		local _,b,x2,y2 = os.pullEvent("mouse_click")
		if b == 1 and y == y2 then
			for i=1,#buttons do
				local x = math.floor((w/2-(#txt+2)/2)+((#txt+2)/i)*(i-1)) + 1
				if x2 > x - 1 and x2 < x + #buttons[i][1] then
					redraw()
					return i
				end
			end
		end
	end
end
--these were moved above popupbox:
local function safeRun(func, ...)
	local succ, msg = pcall(func, ...)
	if not succ then
		infoBox(msg)
	end
end
local function refresh()
	if pathValid then
		readFolderRaw(path)
	else
		local t = {}
		for k,v in pairs(itemPaths) do
			if fs.exists(v) then
				t[#t+1] = v
			end
		end
		itemPaths = t
	end
end
local function popupBox(x,y,buttons,functions) --popups defined to have width 15
	drawPopupBox(x,y,buttons)
	local ydir = y < h/2 and 1 or -1
	while true do
		local _,b,cx,cy = os.pullEvent("mouse_click")
		if b == 1 then
			if cx < x or cx > x + 15 then
				os.queueEvent("mouse_click", b, cx, cy)
				break
			end
			if not (cy*ydir > y*ydir and cy*ydir - y*ydir < #buttons+1) then
				os.queueEvent("mouse_click", b, cx, cy)
				break
			end
			--for menus inside popup boxes
			redraw()
			safeRun(functions[ydir*cy-ydir*y],x,y)
			refresh()
			break
		else
			os.queueEvent("mouse_click", b, cx, cy)
			break
		end
	end
	
end

local findFileBox --for later
local findDirBox

--file editing functionality
local function run(path, ...)
	local tArgs = {...}
	local function box()
		clear()
		
		shell.run(fs.combine("/", path), unpack(tArgs)) --better alternative to shell.run?
		print("Press any key or click to return to Lattix")
		waitForKeyOrClick()
	end
	local cor = coroutine.create(box)
	coroutine.resume(cor)
	--later better sandboxing required that doesn't allow modification of _G or some file defense
	while (coroutine.status(cor) ~= "dead") do
		local event = {os.pullEventRaw()} --terminate works only on the sub-program
		coroutine.resume(cor, unpack(event))
	end
end
local function mkdir(path, name)
	fs.makeDir(fs.combine(path, name))
end
local function paste(path, clipboard, cutEnabled) --i wonder what happens if you try to move something inside of itself
	if type(clipboard) == "table" then
		for i=1,#clipboard do
			paste(path, clipboard[i], cutEnabled)
		end
		return
	end
	if not clipboard then error("Clipboard is empty",0) end
	if not fs.exists(clipboard) then error("File copied doesn't exist",0) end
	local func = cutEnabled and fs.move or fs.copy
	local goal = fs.combine(path, fs.getName(clipboard))
	local i = 1
	while fs.exists(goal) do
		goal = fs.combine(path, "Copy" .. (i>1 and " " .. tostring(i) or "") .. " of " .. fs.getName(clipboard))
		i = i + 1
	end
	func(clipboard, goal)
end
local function advFind(path, wildcard, results)
	if not results then results = {} end
	local t = fs.find(fs.combine(path, wildcard))
	for i=1,#t do
		results[#results+1] = t[i]
	end
	local dirs = fs.list(path)
	for i=1,#dirs do
		if fs.isDir(fs.combine(path, dirs[i])) then
			results = advFind(fs.combine(path, dirs[i]), wildcard, results)
		end
	end
	
	return results
end
--GUI core functionality

local function doubleClick(path) --contains a reference to refresh
	selected = 0
	if fs.isDir(path) then
		switchFolder(path)
	else
		local elements = split(fs.getName(path), ".")
		if #elements == 1 then
			elements[2] = "" --no extension
		end
		if not defaultPrograms[elements[#elements]] then
			elements[#elements] = "def" --unknown extension
		end
		if (defaultPrograms[elements[#elements]]) == "api" then
			os.loadAPI(path)
			refresh()
		elseif defaultPrograms[elements[#elements]] == "run" then
			run(path)
			refresh()
		else
			shell.run(defaultPrograms[elements[#elements]], path)
			refresh()
		end
	end
end
local function gotoFolder()
	local target
		target = textBox("Please specify target folder")
		if target == "" then return end --cancel if empty
		if not target or not fs.exists(target) or not fs.isDir(target) then
			infoBox("    Not a valid directory   ")
			return
		end
		switchFolder(fs.combine("", target))
end
local function makeDirPrompt()
	local name = textBox("Enter the name of the new directory")
	if name == "" then return end
	if fs.exists(fs.combine(path, name)) then
		infoBox("Failure - File name already used")
	end
	mkdir(path, name)
end
local function pastebin()
	local link = textBox("Enter the pastebin link")
	if link == "" then return end
	local name = textBox("Enter the name of the file")
	if name == "" then return end
	if fs.exists(fs.combine(path,name)) then
		infoBox("Failure - File name already used")
		return
	end
	local succ, err = pcall(pastebinGet,link, fs.combine(path, name))
	if not succ then
		infoBox(err)
	end
end
local function wgetPrompt()
	local link = textBox("Enter the url")
	if link == "" then return end
	local name = textBox("Enter the name of the file")
	if name == "" then return end
	if fs.exists(fs.combine(path,name)) then
		infoBox("Failure - File name already used")
		return
	end
	local succ, err = pcall(wGet,link, fs.combine(path, name))
	if not succ then
		infoBox(err)
	end
end
local function runPrompt(appath)
	if not appath then
		appath = fs.combine(path, textBox("Enter the name of the script"))
	end
	if not fs.exists(appath) then
		infoBox("Script doesn't exist: " .. appath)
		return
	end
	if fs.isDir(appath) then
		infoBox("Cannot run a directory: " .. appath)
		return
	end
	args = textBox("Please enter the arguments")
	run(appath, split(args, " "))
end
local function copy(name)--name actually means full path
	clip_cut = false
	clipboard = name
end
local function cut(name) --name actually means full path here
	copy(name)
	clip_cut = true
end
local function renamePrompt(path)
	if not fs.exists(path) then infoBox("Nothing to rename") return end
	local name = textBox("Enter the new name", fs.getName(path))
	if name == "" then return end
	if fs.exists(fs.combine(fs.getDir(path), name)) then
		infoBox("Failure - File already exists")
		return
	end
	fs.move(path, fs.combine(fs.getDir(path), name))
	selected = 0
end
local function deletePrompt(path)
	if not fs.exists(path) then infoBox("Nothing to delete") return end
	local response = buttonBox("Do you really want to delete " .. fs.getName(path) .. "?", {{" Delete ", colors.white, colors.red}, {" Cancel ", colors.yellow, colors.black }})
	if response == 1 then
		fs.delete(path)
	end
	selected = 0
end
local function findFilesPrompt()
	local wildCard = textBox("Enter filename or part of it")
	if wildCard == "" then return end --cancel option
	local finds = advFind(path, "*" .. wildCard .. "*")
	if #finds == 0 then
		infoBox("No files found")
		return
	end
	itemNames = {}
	itemPaths = {}
	selected = 0
	scroll = 0
	for i=1,#finds do
		itemNames[i] = finds[i]
		itemPaths[i] = finds[i]
		customTitle = "Search results"
		pathValid = false
	end
	
end
--GUI functionality - event mapping
local buttons = {}
local keymap = {}
local eventmap = {}

for i=1,w do
	buttons[i] = {}
	for j=1,h do
	
		buttons[i][j] = {function() end, function() end}
	end
end
local function newButton(x, y, w, h, func_left, func_right)
	for i=x, x+w-1 do
		for j=y, y+h-1 do
			buttons[i][j] = {func_left, func_right}
		end
	end
end
local function clearButtons(x, y, w, h)
	newButton(x,y,w,h, empty, empty)
end

local function clearAllEvents()
	keymap = {}
	eventmap = {}
	clearButtons(1,1,w,h)
end
--mappings:
local popup_newmenu_names = {
	"New ...",
	"New dir",
	"Paste",
	"Pastebin",
	"wget"
}
local popup_newmenu_functions = {
	function(x,y) 
		local options = {}
		local functions = {}
		for k,v in pairs(editors) do
			local i = #options+1 
			options[i] = k
			functions[i] = function()
				local ext = ""
				if editors_extensions[k] then
					ext = editors_extensions[k]
				end
				local app
				if v == "choose" then
					
				else
					app = v
				end
				local item = xtensionTextBox("What should the new file be called?", "." .. ext)
				local target = fs.combine(path, item)
				if not fs.exists(target) then
					shell.run(app, target)
				else
					infoBox("Failure - file already exists")
				end
			end
		end
		popupBox(x,y,options,functions)
	end,
	function() makeDirPrompt() end,
	function() paste(path, clipboard, clip_cut) end,
	function() pastebin() end,
	function() wgetPrompt() end,
	
}





local popup_lockednewmenu_names = {
	"Refresh"
}
local popup_lockednewmenu_functions = {
	function() refresh() end
}

local popup_menu_names = {
	"Go to dir",
	"Find file",
	"Version: "..version
}
local popup_menu_functions = {
	function() gotoFolder() end,
	function() findFilesPrompt() end,
	function() infoBox("Lattix version " .. version) end,
}
local menu_functions = {
	back = function() 
		if not pathValid then
			pathValid = true
			refresh()
			
			return
		end
		if #history > 1 then
			switchFolderRaw(history[#history])
			table.remove(history, #history)
		end
	end,
	up = function()
		if not pathValid then
			pathValid = true
			refresh()
			return
		end
		if path == "" or path == "/" then return end
		switchFolder(fs.combine(path, ".."))
	end,
	menu = function()
		--open advanced menu
		popupBox(9,2,popup_menu_names, popup_menu_functions)
	end,
	root = function()
		if not pathValid then
			pathValid = true
		end
		switchFolder(root)
	end,
	plus = function()
		--open new menu
		if pathValid then
			popupBox(5,2,popup_newmenu_names, popup_newmenu_functions)
		else
			popupBox(5,2,popup_lockednewmenu_names, popup_lockednewmenu_functions)
		end
	end,
	quit = function()
		clear()
		endprogram = true
	end
}
local filePopupNames = {
--	"Edit", --opens default editor, as double click would, expect that for .lua it opens edit too
	"Run", --runs
	"Run w/ args", --runs with args
--	"Open With", --select a program from a list and use it to open this, config file will be huge i see
	"Rename",
	"Copy",
	"Cut",
	"Delete"
}
local function getSelectedPath()
	return itemPaths[selected]
end
local filePopupFunctions = {
	function(x,y) run(getSelectedPath()) end,--run
	function(x,y) runPrompt(getSelectedPath()) end,--run w args
	function(x,y) renamePrompt(getSelectedPath()) end, --rename,
	function(x,y) copy(getSelectedPath()) end, --copy
	function(x,y) cut (getSelectedPath()) end, --cut
	function(x,y) deletePrompt(getSelectedPath()) end --delete
}
local folderPopupNames = {
	"Open",
--	"Open in ...", --1,2,3,4 and program, TODO later bc open in and open with need some design decisions
--	"Unpack",
	"Rename",
	"Copy",
	"Cut",
	"Delete"
}
local folderPopupFunctions = {
	function(x,y) switchFolder(getSelectedPath()) end, --open
	function(x,y) renamePrompt(getSelectedPath()) end, --rename
	function(x,y) copy(getSelectedPath()) end, --copy
	function(x,y) cut(getSelectedPath()) end, --cut
	function(x,y) deletePrompt(getSelectedPath()) end --delete
}
local multiPopupNames = {}
local multiPopupFunctions = {}--for multiselect, copy, cut, pack into folder, delete

local commands = { --table of functions, arg: list of words typed, including the command itself
	["run"] = function(words)
		if #words == 1 then
			runPrompt()
			return
		end
		
		local torun
		for i=1,#itemNames do
			if words[2] == itemNames[i] then
				torun = i
				break
			end
		end
		if not torun then return end
		local args = {}
		if #words > 2 then
			for i=3, #words do
				args[i-2] = words[i]
			end
		end
		if fs.exists(itemPaths[torun]) and not fs.isDir(itemPaths[torun]) then
			run(itemPaths[torun], unpack(args))
		end
	end,
	["goto"] = function(words)
		local target
		if #words < 2 then
			gotoFolder()
			return
		else
			target = words[2]
		end
		--from local path
		if target:sub(1,1) ~= "/" and fs.exists(fs.combine(path, target)) and fs.isDir(fs.combine(path, target)) then
			switchFolder(fs.combine(path, target))
			return
		end
		--from absolute path
		if not target or not fs.exists(target) or not fs.isDir(target) then
			infoBox("Not a valid directory")
			return
		end
		switchFolder(fs.combine("", target))
	end,
	["deselect"] = function(words)
		selected = 0
	end,
	["select"] = function(words)
		--deselect cuz no arg
		if #words < 2 then
			selected = 0
			return
		end
		--select by name
		for i=1, #itemNames do
			if itemNames[i] == words[2] then
				selected = i
				return
			end
		end
		
		--select by index
		local index = tonumber(words[2])
		if index and index == math.floor(index) and index <= #itemPaths and index >= 0 then
			selected = index
			return
		end
		
	end,
	["count"] = function(words)
		infoBox("Number of items: " .. #itemPaths)
	end,
	["up"] = function(words)
		menu_functions.up()
	end,
	["root"] = function(words)
		menu_functions.root()
	end,
	["quit"] = function(words)
		menu_functions.quit()
	end,
	["back"] = function(words)
		menu_functions.back()
	end,
	["switch"] = function(words)
		if #words < 2 then
			local nses = cSession + 1
			switchSessions(nses < 5 and nses or 1)
			return
		end
		local index = tonumber(words[2])
		if index and index == math.floor(index) and index <= 4 and index >= 1 then
			switchSessions(index)
		end
	end,
	["help"] = function(words)
		infoBox(" Read the Lattix CC forum post, commands section ")
	end,
}
local function raw(str)
	str = str:lower()
	str = str:gsub(' ','')
	str = str:gsub('_', '')
	return str
end
local function evaulate(nametbl, functbl, name)
	for i=1, #nametbl do
		if raw(nametbl[i]) == raw(name) then
			return functbl[i]
		end
	end
end
local alias = {
	["cd"] = "goto",
	["mkdir"] = "newdir",
	["find"] = "findfile",
	["search"] = "find",
	["about"] = "version:" .. version,
	["version"] = "version:"..version,
	["exit"] = "quit",
	[".."] = "back",
	["/"] = "root",
	["x"] = "quit",
	["#"] = "count"
}

local mt = {
	__index = function(tbl, i)
		local function fullEval()
			if raw(i) == raw("new...") then
				return nil
			end
			local func
			if pathValid then
				func = evaulate(popup_newmenu_names, popup_newmenu_functions, i)
				if func then return func end
			else
				func = evaulate(popup_lockednewmenu_names, popup_lockednewmenu_functions, i)
				if func then return func end
			end
			func = evaulate(popup_menu_names, popup_menu_functions, i)
			if func then return func end
			--if multiselection...
			if selected ~= 0 and itemPaths[selected] and fs.exists(itemPaths[selected]) then
				if fs.isDir(itemPaths[selected]) then
					func = evaulate(folderPopupNames, folderPopupFunctions, i)
					if func then return func end
				else
					func = evaulate(filePopupNames, filePopupFunctions, i)
					if func then return func end
				end
			end
		end
		local final
		while true do
			final = fullEval()
			if final then break end
			if not alias[raw(i)] then
				break
			end
			i = alias[raw(i)]
			if commands[i] then final = commands[i] break end
		end
		return final
	end
}
setmetatable(commands, mt)



local function mapMenu()
	
	local switch = function(e)
		switchSessions(e[3]-w+6)
	end
	newButton(1,1,1,1, menu_functions.back, empty)
	newButton(3,1,1,1, menu_functions.up, empty)
	newButton(7,1,1,1, menu_functions.root, empty)
	newButton(5,1,1,1, menu_functions.plus, empty)
	newButton(9,1,1,1, menu_functions.menu, empty)
	newButton(w,1,1,1, menu_functions.quit, empty)
	newButton(w-5,1,4,1, switch, empty) 
	keymap[keys.left] = up
end
local drawFiles

local function enterPress()
	words = split(shell_input, " ")
	if words and #words > 0 then
		if commands[words[1]:lower()] then
			prewrite = {}
			if #words > 1 then
				for i=2, #words do
					prewrite[i-1] = words[i]
				end
			end
			commands[words[1]:lower()](words)
			refresh()
			prewrite = {}
		else
			infoBox("Unknown command: " .. words[1]:lower())
		end
	elseif selected > 0 and selected <= #itemPaths then
		doubleClick(itemPaths[selected])
	end
	shell_input = ""
end
local function filePopup(x,y,path)
	if fs.isDir(path) then
		--directory
		popupBox(x,y,folderPopupNames, folderPopupFunctions)
	else
		--file
		popupBox(x,y,filePopupNames, filePopupFunctions)
	end
end
local function mapFiles()
	--popup menu implementation (right click)
	local file_rightclick = function(e)
		local i = posToIndex(#itemNames,e[3],e[4],scroll) --get index
		if itemPaths[i] then
			if selected ~= i then --select if not selected
				selected = i --just for aesthetics
				redraw()
			end
			--show file/folder/multiselect relevant popup
			if selection and #selection > 0 then
				--multiselect
			else
				filePopup(e[3],e[4],getSelectedPath())
			end
		else
			selected = 0
			--show the same popup as the + button in the menu
			if pathValid then
				popupBox(e[3],e[4],popup_newmenu_names, popup_newmenu_functions)
			else
				popupBox(e[3], e[4], popup_lockednewmenu_names, popup_lockednewmenu_functions)
			end
		end
	end
	--select implementation (left click)
	local file_leftclick = function(e)
		local i = posToIndex(#itemNames,e[3],e[4],scroll)
		if itemPaths[i] then
			if selected == i then
				doubleClick(itemPaths[i])
			else
				selected = i
			end
		else
			selected = 0
		end
	end
	
	newButton(1,2,w,h-2,file_leftclick, file_rightclick)
	--multiselect stuff
	
	--scrolling
	eventmap["mouse_scroll"] = function(e)
		local b = e[2]
		if b == 1 and #itemPaths - scroll > h-2 then
			scroll = scroll + 1
		end
		if b == -1 and scroll > 0 then
			scroll = scroll - 1
		end
	end
	
	keymap[keys.enter] = enterPress
	keymap[keys.up] = function()
		if selected > 1 then
			selected = selected - 1
			if scroll >= selected then
				scroll = selected - 1
			end
		end
		if selected == 0 then
			selected = #itemPaths
			if scroll + h - 2 < selected then
				scroll = selected - h + 2
			end
		end
	end
	keymap[keys.down] = function()
		if selected < #itemPaths then
			selected = selected + 1
			if scroll + h - 2 < selected then
				scroll = selected - h + 2
			end
		end
	end
	keymap[keys.right] = function()
		if selected > 0 and selected <= #itemPaths then
			local x,y = IndexToPos(selected, scroll)
			if x < 2 then
				x = 2
			end
			if x > h-1 then
				x = h-1
			end
			filePopup(x,y, getSelectedPath()) 
		end
	end
end

local function mapBar()
	eventmap.char = function(e) 
		shell_input = shell_input .. e[2]
	end
	keymap[keys.backspace] = function()
		shell_input = shell_input:sub(1, #shell_input-1)
	end
	--enter is mapped in mapFiles for the bar too
end
--draw components
local function drawMenu()
	surf:drawLine(1,1,w,1," ", lat_theme.menu_back, lat_theme.menu_text)
	if term.isColor() then
		surf:drawText(1,1,"< ^ + / m", lat_theme.menu_back, lat_theme.menu_text)
	end
	local str
	if customTitle then
		str = customTitle
	else
		if path ~= "" then
			str = cSession .. " - " .. path
		else
			str = tostring(cSession)
		end
	end
	str = #str < w/2 and str or str:sub(1,math.floor(w/2)) .. "..."
	surf:drawText(8+math.floor((w-12)/2-#str/2),1,str, lat_theme.menu_back, lat_theme.menu_text)
	
	if term.isColor() then
		surf:drawText(w-5,1,"1234", lat_theme.menu_back, lat_theme.menu_text)
		surf:drawPixel(w,1,"x", colors.red, colors.white)
	end
	
end

drawFiles = function()
	if selection == nil then selection = {} end --just in case selection was nilled
	local cy = 2 --current y pos of drawing
	local i = scroll + 1 --index in "items"
	local tcol
	local bcol
	while IndexToPos(i,scroll) < h do
		if not itemNames[i] then break end --because the while condition checks for screen size, this checks for file count
		local twrite = ""
		bcol = (lat_theme.main_back)
		if selection[i] then
			bcol = (lat_theme.multiselect_back)
			twrite = twrite .. selectprefix
		end
		if i ~= selected then
			if not fs.isDir(itemPaths[i]) then
				tcol = (lat_theme.main_text)
			else
				tcol = (lat_theme.dir_text)
				
			end
		else
			tcol = (lat_theme.selected_text)
		end
		if fs.isDir(itemPaths[i]) then
			twrite = twrite .. dirprefix
		end
		twrite = twrite .. itemNames[i]
		surf:drawLine(1, cy, w, cy, " ", bcol, tcol)
		surf:drawText(2, cy, twrite, bcol, tcol)
		local mwrite = "-"
		if not fs.isDir(itemPaths[i]) then
			mwrite = formatNumber(fs.getSize(itemPaths[i]))
		end
		local startX = w-6
		surf:drawLine(startX-1, cy, w, cy, " ", bcol, tcol)
		surf:drawText(startX,cy, mwrite, bcol, tcol)
		i = i + 1
		cy = cy + 1 --up both indexes
	end
	return itemNames
end

local function drawBar()
	surf:drawLine(1,h,w,h, " ", lat_theme.menu_back, lat_theme.menu_text)
	surf:drawText(1,h, path .. "> " .. shell_input, lat_theme.menu_back, lat_theme.menu_text)
end


--GUI drawing functions
redraw = function()
	if not fs.exists(path) or not fs.isDir(path) then
		infoBox("Error: " .. path .. " is not a valid directory", true)
		path = root
	end
	
	surf:clear(" ", lat_theme.blank_back, lat_theme.blank_text)
	drawMenu()
	
	drawFiles()
	
	drawBar()
	
	surf:render()
end

remap = function()
	clearAllEvents()
	mapMenu()
	mapFiles()
	mapBar()
end
local oRedraw = redraw --orginal backup
local oRemap = remap
restore = function()
	redraw = oRedraw
	remap = oRemap
end

do --choose file/folder dialog

end

switchFolder(path)

--main loop
while not endprogram do
	redraw()
	remap()
	local e = { os.pullEvent() }
	if e[1] == "mouse_click" then
		buttons[e[3]][e[4]][e[2]](e)
	elseif e[1] == "key" then
		if keymap[e[2]] then
			keymap[e[2]](e)
		end
	elseif eventmap[e[1]] then
		eventmap[e[1]](e)
	end	
end

end

local succ, msg = pcall(program)

term.setBackgroundColor(colors.black)
term.setTextColor(colors.white)
term.clear()
term.setCursorPos(1,1)
	
if not succ and msg ~= "Terminated" then
	print("Congratulations, Lattix has crashed")
	print()
	print("Please report with steps to reproduce to the forum post to get your name added to the credits")
	print()
	print(msg)
end
if succ or msg == "Terminated" then
	print("[Lattix]: bye")
	print()
	print(msg or "version " .. version)
end