mirror of
https://github.com/koreader/koreader
synced 2024-11-13 19:11:25 +00:00
264 lines
6.8 KiB
Lua
264 lines
6.8 KiB
Lua
require "rendertext"
|
|
require "keys"
|
|
require "graphics"
|
|
require "font"
|
|
|
|
FileSearcher = {
|
|
-- title height
|
|
title_H = 45,
|
|
-- spacing between lines
|
|
spacing = 40,
|
|
-- foot height
|
|
foot_H = 27,
|
|
|
|
-- state buffer
|
|
dirs = {},
|
|
files = {},
|
|
result = {},
|
|
items = 0,
|
|
page = 0,
|
|
current = 1,
|
|
oldcurrent = 1,
|
|
}
|
|
|
|
function FileSearcher:readDir()
|
|
self.dirs = {self.path}
|
|
self.files = {}
|
|
while #self.dirs ~= 0 do
|
|
new_dirs = {}
|
|
-- handle each dir
|
|
for __, d in pairs(self.dirs) do
|
|
-- handle files in d
|
|
for f in lfs.dir(d) do
|
|
local file_type = string.lower(string.match(f, ".+%.([^.]+)") or "")
|
|
if lfs.attributes(d.."/"..f, "mode") == "directory"
|
|
and f ~= "." and f~= ".." and not string.match(f, "^%.[^.]") then
|
|
table.insert(new_dirs, d.."/"..f)
|
|
elseif file_type == "djvu" or file_type == "pdf"
|
|
or file_type == "xps" or file_type == "cbz"
|
|
or file_type == "epub" or file_type == "txt"
|
|
or file_type == "rtf" or file_type == "htm"
|
|
or file_type == "fb2" or file_type == "chm" then
|
|
file_entry = {dir=d, name=f,}
|
|
table.insert(self.files, file_entry)
|
|
--print("file:"..d.."/"..f)
|
|
end
|
|
end
|
|
end
|
|
self.dirs = new_dirs
|
|
end
|
|
end
|
|
|
|
function FileSearcher:setPath(newPath)
|
|
self.path = newPath
|
|
self:readDir()
|
|
self.items = #self.files
|
|
--@TODO check none found 19.02 2012
|
|
if self.items == 0 then
|
|
return nil
|
|
end
|
|
self.page = 1
|
|
self.current = 1
|
|
return true
|
|
end
|
|
|
|
function FileSearcher:setSearchResult(keywords)
|
|
self.result = {}
|
|
if keywords == " " then -- one space to show all files
|
|
self.result = self.files
|
|
else
|
|
for __,f in pairs(self.files) do
|
|
if string.find(string.lower(f.name), keywords) then
|
|
table.insert(self.result, f)
|
|
end
|
|
end
|
|
end
|
|
self.items = #self.result
|
|
self.page = 1
|
|
self.current = 1
|
|
end
|
|
|
|
function FileSearcher:init(search_path)
|
|
if search_path then
|
|
self:setPath(search_path)
|
|
else
|
|
self:setPath("/mnt/us/documents")
|
|
end
|
|
end
|
|
|
|
|
|
function FileSearcher:choose(ypos, height, keywords)
|
|
local perpage = math.floor(height / self.spacing) - 2
|
|
local pagedirty = true
|
|
local markerdirty = false
|
|
|
|
local prevItem = function ()
|
|
if self.current == 1 then
|
|
if self.page > 1 then
|
|
self.current = perpage
|
|
self.page = self.page - 1
|
|
pagedirty = true
|
|
end
|
|
else
|
|
self.current = self.current - 1
|
|
markerdirty = true
|
|
end
|
|
end
|
|
|
|
local nextItem = function ()
|
|
if self.current == perpage then
|
|
if self.page < (self.items / perpage) then
|
|
self.current = 1
|
|
self.page = self.page + 1
|
|
pagedirty = true
|
|
end
|
|
else
|
|
if self.page ~= math.floor(self.items / perpage) + 1
|
|
or self.current + (self.page-1)*perpage < self.items then
|
|
self.current = self.current + 1
|
|
markerdirty = true
|
|
end
|
|
end
|
|
end
|
|
|
|
-- if given keywords, set new result according to keywords.
|
|
-- Otherwise, display the previous search result.
|
|
if keywords then
|
|
self:setSearchResult(keywords)
|
|
end
|
|
|
|
while true do
|
|
local cface, cfhash = Font:getFaceAndHash(22)
|
|
local tface, tfhash = Font:getFaceAndHash(25, Font.tfont)
|
|
local fface, ffhash = Font:getFaceAndHash(16, Font.ffont)
|
|
|
|
if pagedirty then
|
|
markerdirty = true
|
|
fb.bb:paintRect(0, ypos, fb.bb:getWidth(), height, 0)
|
|
|
|
-- draw menu title
|
|
renderUtf8Text(fb.bb, 30, ypos + self.title_H, tface, tfhash,
|
|
"Search Result for: "..keywords, true)
|
|
|
|
-- draw results
|
|
local c
|
|
if self.items == 0 then -- nothing found
|
|
y = ypos + self.title_H + self.spacing * 2
|
|
renderUtf8Text(fb.bb, 20, y, cface, cfhash,
|
|
"Sorry, no match found.", true)
|
|
renderUtf8Text(fb.bb, 20, y + self.spacing, cface, cfhash,
|
|
"Please try a different keyword.", true)
|
|
markerdirty = false
|
|
else -- found something, draw it
|
|
for c = 1, perpage do
|
|
local i = (self.page - 1) * perpage + c
|
|
if i <= self.items then
|
|
y = ypos + self.title_H + (self.spacing * c)
|
|
renderUtf8Text(fb.bb, 50, y, cface, cfhash,
|
|
self.result[i].name, true)
|
|
end
|
|
end
|
|
end
|
|
|
|
-- draw footer
|
|
y = ypos + self.title_H + (self.spacing * perpage) + self.foot_H
|
|
x = (fb.bb:getWidth() / 2) - 50
|
|
all_page = (math.floor(self.items / perpage)+1)
|
|
renderUtf8Text(fb.bb, x, y, fface, ffhash,
|
|
"Page "..self.page.." of "..all_page, true)
|
|
end
|
|
|
|
if markerdirty then
|
|
if not pagedirty then
|
|
if self.oldcurrent > 0 then
|
|
y = ypos + self.title_H + (self.spacing * self.oldcurrent) + 10
|
|
fb.bb:paintRect(30, y, fb.bb:getWidth() - 60, 3, 0)
|
|
fb:refresh(1, 30, y, fb.bb:getWidth() - 60, 3)
|
|
end
|
|
end
|
|
-- draw new marker line
|
|
y = ypos + self.title_H + (self.spacing * self.current) + 10
|
|
fb.bb:paintRect(30, y, fb.bb:getWidth() - 60, 3, 15)
|
|
if not pagedirty then
|
|
fb:refresh(1, 30, y, fb.bb:getWidth() - 60, 3)
|
|
end
|
|
self.oldcurrent = self.current
|
|
markerdirty = false
|
|
end
|
|
|
|
if pagedirty then
|
|
fb:refresh(0, 0, ypos, fb.bb:getWidth(), height)
|
|
pagedirty = false
|
|
end
|
|
|
|
local ev = input.waitForEvent()
|
|
ev.code = adjustKeyEvents(ev)
|
|
if ev.type == EV_KEY and ev.value == EVENT_VALUE_KEY_PRESS then
|
|
if ev.code == KEY_FW_UP then
|
|
prevItem()
|
|
elseif ev.code == KEY_FW_DOWN then
|
|
nextItem()
|
|
elseif ev.code == KEY_PGFWD then
|
|
if self.page < (self.items / perpage) then
|
|
if self.current + self.page*perpage > self.items then
|
|
self.current = self.items - self.page*perpage
|
|
end
|
|
self.page = self.page + 1
|
|
pagedirty = true
|
|
else
|
|
self.current = self.items - (self.page-1)*perpage
|
|
markerdirty = true
|
|
end
|
|
elseif ev.code == KEY_PGBCK then
|
|
if self.page > 1 then
|
|
self.page = self.page - 1
|
|
pagedirty = true
|
|
else
|
|
self.current = 1
|
|
markerdirty = true
|
|
end
|
|
elseif ev.code == KEY_S then
|
|
old_keywords = keywords
|
|
keywords = InputBox:input(height-100, 100, "Search:", old_keywords)
|
|
if keywords then
|
|
self:setSearchResult(keywords)
|
|
else
|
|
keywords = old_keywords
|
|
end
|
|
pagedirty = true
|
|
elseif ev.code == KEY_F then -- invoke fontchooser menu
|
|
fonts_menu = SelectMenu:new{
|
|
menu_title = "Fonts Menu",
|
|
item_array = Font.fonts,
|
|
}
|
|
local re = fonts_menu:choose(0, height)
|
|
if re then
|
|
Font.cfont = Font.fonts[re]
|
|
Font:update()
|
|
end
|
|
pagedirty = true
|
|
elseif ev.code == KEY_ENTER or ev.code == KEY_FW_PRESS then
|
|
file_entry = self.result[perpage*(self.page-1)+self.current]
|
|
file_full_path = file_entry.dir .. "/" .. file_entry.name
|
|
|
|
-- rotation mode might be changed while reading, so
|
|
-- record height_percent here
|
|
local height_percent = height/fb.bb:getHeight()
|
|
openFile(file_full_path)
|
|
|
|
--reset height and item index if screen has been rotated
|
|
local old_perpage = perpage
|
|
height = math.floor(fb.bb:getHeight()*height_percent)
|
|
perpage = math.floor(height / self.spacing) - 2
|
|
self.current = (old_perpage * (self.page - 1) +
|
|
self.current) % perpage
|
|
self.page = math.floor(self.items / perpage) + 1
|
|
|
|
pagedirty = true
|
|
elseif ev.code == KEY_BACK or ev.code == KEY_HOME then
|
|
return nil
|
|
end
|
|
end
|
|
end
|
|
end
|