mirror of https://github.com/koreader/koreader
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
2.0 KiB
Lua
64 lines
2.0 KiB
Lua
local ReaderDictionary = require("apps/reader/modules/readerdictionary")
|
|
local Translator = require("ui/translator")
|
|
local Wikipedia = require("ui/wikipedia")
|
|
local Screen = require("device").screen
|
|
local DEBUG = require("dbg")
|
|
local _ = require("gettext")
|
|
|
|
-- Wikipedia as a special dictionary
|
|
local ReaderWikipedia = ReaderDictionary:extend{
|
|
-- identify itself
|
|
wiki = true,
|
|
no_page = _("No wiki page found."),
|
|
}
|
|
|
|
-- the super "class" ReaderDictionary has already registers a menu entry
|
|
-- we should override the init function in ReaderWikipedia
|
|
function ReaderWikipedia:init()
|
|
end
|
|
|
|
function ReaderWikipedia:onLookupWikipedia(word, box)
|
|
-- detect language of the text
|
|
local ok, lang = pcall(Translator.detect, Translator, word)
|
|
if not ok then return end
|
|
-- convert "zh-CN" and "zh-TW" to "zh"
|
|
lang = lang:match("(.*)-") or lang
|
|
-- strip punctuation characters around selected word
|
|
word = string.gsub(word, "^%p+", '')
|
|
word = string.gsub(word, "%p+$", '')
|
|
-- seems lower case phrase has higher hit rate
|
|
word = string.lower(word)
|
|
local results = {}
|
|
local ok, pages = pcall(Wikipedia.wikintro, Wikipedia, word, lang)
|
|
if ok and pages then
|
|
for pageid, page in pairs(pages) do
|
|
local result = {
|
|
dict = _("Wikipedia"),
|
|
word = page.title,
|
|
definition = page.extract or self.no_page,
|
|
}
|
|
table.insert(results, result)
|
|
end
|
|
DEBUG("lookup result:", word, results)
|
|
self:showDict(word, results, box)
|
|
else
|
|
DEBUG("error:", pages)
|
|
-- dummy results
|
|
results = {
|
|
{
|
|
dict = _("Wikipedia"),
|
|
word = word,
|
|
definition = self.no_page,
|
|
}
|
|
}
|
|
DEBUG("dummy result table:", word, results)
|
|
self:showDict(word, results, box)
|
|
end
|
|
end
|
|
|
|
-- override onSaveSettings in ReaderDictionary
|
|
function ReaderWikipedia:onSaveSettings()
|
|
end
|
|
|
|
return ReaderWikipedia
|