koreader/frontend/ui/widget/filechooser.lua

require "ui/widget/menu"

FileChooser = Menu:extend{
	height = Screen:getHeight(),
	width = Screen:getWidth(),
	no_title = true,
	path = lfs.currentdir(),
	parent = nil,
	show_hidden = false,
	filter = function(filename) return true end,
}

function FileChooser:init()
	self:updateItemTableFromPath(self.path)
	Menu.init(self) -- call parent's init()
end

function FileChooser:compressPath(item_path)
	if (item_path:sub(1, 1) == ".") then
		-- ignore relative path
		return item_path
	end

	-- compress paths like "test/pdf/../epub" into "test/epub"
	local path = item_path
	while path:match("/[^/]+[/][\\.][\\.]") do
		path = path:gsub("/[^/]+[/][\\.][\\.]", "")
	end
	return path
end

function FileChooser:updateItemTableFromPath(path)
	path = self:compressPath(path)
	local dirs = {}
	local files = {}
	self.path = path

	for f in lfs.dir(self.path) do
		if self.show_hidden or not string.match(f, "^%.[^.]") then
			local filename = self.path.."/"..f
			local filemode = lfs.attributes(filename, "mode")
			if filemode == "directory" and f ~= "." and f~=".." then
				if self.dir_filter(filename) then
					table.insert(dirs, f)
				end
			elseif filemode == "file" then
				if self.file_filter(filename) then
					table.insert(files, f)
				end
			end
		end
	end
	table.sort(dirs)
	if self.path ~= "/" then table.insert(dirs, 1, "..") end
	table.sort(files)

	self.item_table = {}
	for _, dir in ipairs(dirs) do
		table.insert(self.item_table, { text = dir.."/", path = self.path.."/"..dir })
	end
	for _, file in ipairs(files) do
		table.insert(self.item_table, { text = file, path = self.path.."/"..file })
	end
end

function FileChooser:changeToPath(path)
	self:updateItemTableFromPath(path)
	self:updateItems(1)
end

function FileChooser:onMenuSelect(item)
	if lfs.attributes(item.path, "mode") == "directory" then
		self:changeToPath(item.path)
	else
		self:onFileSelect(item.path)
	end
	return true
end

function FileChooser:onFileSelect(file)
	UIManager:close(self)
	return true
end