koreader/frontend/ui/reader/readertoc.lua

ReaderToc = InputContainer:new{
	key_events = {
		ShowToc = { {"T"}, doc = "show Table of Content menu"},
	},
	dimen = Geom:new{ w = G_width-20, h = G_height-20},
	current_page = 0,
	current_pos = 0,
}

function ReaderToc:cleanUpTocTitle(title)
	return title:gsub("\13", "")
end

function ReaderToc:onSetDimensions(dimen)
	self.dimen = dimen
end

--function ReaderToc:fillToc()
	--self.toc = self.doc:getToc()
--end

-- getTocTitleByPage wrapper, so specific reader
-- can tranform pageno according its need
function ReaderToc:getTocTitleByPage(pageno)
	return self:_getTocTitleByPage(pageno)
end

function ReaderToc:_getTocTitleByPage(pageno)
	if not self.toc then
	-- build toc when needed.
	self:fillToc()
	end

	-- no table of content
	if #self.toc == 0 then
		return ""
	end

	local pre_entry = self.toc[1]
	for _k,_v in ipairs(self.toc) do
		if _v.page > pageno then
			break
		end
		pre_entry = _v
	end
	return self:cleanUpTocTitle(pre_entry.title)
end

function ReaderToc:getTocTitleOfCurrentPage()
	return self:getTocTitleByPage(self.pageno)
end

function ReaderToc:onShowToc()
	function callback(item)
		self.ui:handleEvent(Event:new("PageUpdate", item.page))
	end

	local items = self.ui.document:getToc()
	-- build menu items
	for _,v in ipairs(items) do
		v.text = ("        "):rep(v.depth-1)..self:cleanUpTocTitle(v.title)
	end
	toc_menu = Menu:new{
		title = "Table of Contents",
		item_table = items,
		width = self.dimen.w,
		height = self.dimen.h,
		on_select_callback = callback,
	}

	UIManager:show(toc_menu)
end

function ReaderToc:onPageUpdate(new_page_no)
	self.current_page = new_page_no
end

function ReaderToc:onPosUpdate(new_pos)
	self.current_pos = new_pos
end