2
0
mirror of https://github.com/koreader/koreader synced 2024-11-08 07:10:27 +00:00
koreader/plugins/wallabag.koplugin/main.lua

1226 lines
47 KiB
Lua

--[[--
@module koplugin.wallabag
]]
local BD = require("ui/bidi")
local DataStorage = require("datastorage")
local Dispatcher = require("dispatcher")
local DocSettings = require("docsettings")
local DocumentRegistry = require("document/documentregistry")
local Event = require("ui/event")
local FFIUtil = require("ffi/util")
local FileManager = require("apps/filemanager/filemanager")
local InfoMessage = require("ui/widget/infomessage")
local InputDialog = require("ui/widget/inputdialog")
local JSON = require("json")
local LuaSettings = require("frontend/luasettings")
local Math = require("optmath")
local MultiConfirmBox = require("ui/widget/multiconfirmbox")
local MultiInputDialog = require("ui/widget/multiinputdialog")
local NetworkMgr = require("ui/network/manager")
local ReadHistory = require("readhistory")
local UIManager = require("ui/uimanager")
local WidgetContainer = require("ui/widget/container/widgetcontainer")
local filemanagerutil = require("apps/filemanager/filemanagerutil")
local http = require("socket.http")
local lfs = require("libs/libkoreader-lfs")
local logger = require("logger")
local ltn12 = require("ltn12")
local socket = require("socket")
local socketutil = require("socketutil")
local util = require("util")
local _ = require("gettext")
local T = FFIUtil.template
-- constants
local article_id_prefix = "[w-id_"
local article_id_postfix = "] "
local failed, skipped, downloaded = 1, 2, 3
local Wallabag = WidgetContainer:extend{
name = "wallabag",
}
function Wallabag:onDispatcherRegisterActions()
Dispatcher:registerAction("wallabag_download", { category="none", event="SynchronizeWallabag", title=_("Wallabag retrieval"), general=true,})
end
function Wallabag:init()
self.token_expiry = 0
-- default values so that user doesn't have to explicitely set them
self.is_delete_finished = true
self.is_delete_read = false
self.is_auto_delete = false
self.is_sync_remote_delete = false
self.is_archiving_deleted = true
self.send_review_as_tags = false
self.filter_tag = ""
self.ignore_tags = ""
self.auto_tags = ""
self.articles_per_sync = 30
self:onDispatcherRegisterActions()
self.ui.menu:registerToMainMenu(self)
self.wb_settings = self.readSettings()
self.server_url = self.wb_settings.data.wallabag.server_url
self.client_id = self.wb_settings.data.wallabag.client_id
self.client_secret = self.wb_settings.data.wallabag.client_secret
self.username = self.wb_settings.data.wallabag.username
self.password = self.wb_settings.data.wallabag.password
self.directory = self.wb_settings.data.wallabag.directory
if self.wb_settings.data.wallabag.is_delete_finished ~= nil then
self.is_delete_finished = self.wb_settings.data.wallabag.is_delete_finished
end
if self.wb_settings.data.wallabag.send_review_as_tags ~= nil then
self.send_review_as_tags = self.wb_settings.data.wallabag.send_review_as_tags
end
if self.wb_settings.data.wallabag.is_delete_read ~= nil then
self.is_delete_read = self.wb_settings.data.wallabag.is_delete_read
end
if self.wb_settings.data.wallabag.is_auto_delete ~= nil then
self.is_auto_delete = self.wb_settings.data.wallabag.is_auto_delete
end
if self.wb_settings.data.wallabag.is_sync_remote_delete ~= nil then
self.is_sync_remote_delete = self.wb_settings.data.wallabag.is_sync_remote_delete
end
if self.wb_settings.data.wallabag.is_archiving_deleted ~= nil then
self.is_archiving_deleted = self.wb_settings.data.wallabag.is_archiving_deleted
end
if self.wb_settings.data.wallabag.filter_tag then
self.filter_tag = self.wb_settings.data.wallabag.filter_tag
end
if self.wb_settings.data.wallabag.ignore_tags then
self.ignore_tags = self.wb_settings.data.wallabag.ignore_tags
end
if self.wb_settings.data.wallabag.auto_tags then
self.auto_tags = self.wb_settings.data.wallabag.auto_tags
end
if self.wb_settings.data.wallabag.articles_per_sync ~= nil then
self.articles_per_sync = self.wb_settings.data.wallabag.articles_per_sync
end
self.remove_finished_from_history = self.wb_settings.data.wallabag.remove_finished_from_history or false
self.download_queue = self.wb_settings.data.wallabag.download_queue or {}
-- workaround for dateparser only available if newsdownloader is active
self.is_dateparser_available = false
self.is_dateparser_checked = false
-- workaround for dateparser, only once
-- the parser is in newsdownloader.koplugin, check if it is available
if not self.is_dateparser_checked then
local res
res, self.dateparser = pcall(require, "lib.dateparser")
if res then self.is_dateparser_available = true end
self.is_dateparser_checked = true
end
if self.ui and self.ui.link then
self.ui.link:addToExternalLinkDialog("25_wallabag", function(this, link_url)
return {
text = _("Add to Wallabag"),
callback = function()
UIManager:close(this.external_link_dialog)
this.ui:handleEvent(Event:new("AddWallabagArticle", link_url))
end,
}
end)
end
end
function Wallabag:addToMainMenu(menu_items)
menu_items.wallabag = {
text = _("Wallabag"),
sub_item_table = {
{
text = _("Retrieve new articles from server"),
callback = function()
self.ui:handleEvent(Event:new("SynchronizeWallabag"))
end,
},
{
text = _("Delete finished articles remotely"),
callback = function()
local connect_callback = function()
local num_deleted = self:processLocalFiles("manual")
UIManager:show(InfoMessage:new{
text = T(_("Articles processed.\nDeleted: %1"), num_deleted)
})
self:refreshCurrentDirIfNeeded()
end
NetworkMgr:runWhenOnline(connect_callback)
end,
enabled_func = function()
return self.is_delete_finished or self.is_delete_read
end,
},
{
text = _("Go to download folder"),
callback = function()
if self.ui.document then
self.ui:onClose()
end
if FileManager.instance then
FileManager.instance:reinit(self.directory)
else
FileManager:showFiles(self.directory)
end
end,
},
{
text = _("Settings"),
callback_func = function()
return nil
end,
separator = true,
sub_item_table = {
{
text = _("Configure Wallabag server"),
keep_menu_open = true,
callback = function()
self:editServerSettings()
end,
},
{
text = _("Configure Wallabag client"),
keep_menu_open = true,
callback = function()
self:editClientSettings()
end,
},
{
text_func = function()
local path
if not self.directory or self.directory == "" then
path = _("Not set")
else
path = filemanagerutil.abbreviate(self.directory)
end
return T(_("Set download folder: %1"), BD.dirpath(path))
end,
keep_menu_open = true,
callback = function(touchmenu_instance)
self:setDownloadDirectory(touchmenu_instance)
end,
separator = true,
},
{
text_func = function()
local filter
if not self.filter_tag or self.filter_tag == "" then
filter = _("All articles")
else
filter = self.filter_tag
end
return T(_("Filter articles by tag: %1"), filter)
end,
keep_menu_open = true,
callback = function(touchmenu_instance)
self:setFilterTag(touchmenu_instance)
end,
},
{
text_func = function()
if not self.ignore_tags or self.ignore_tags == "" then
return _("Ignore tags")
end
return T(_("Ignore tags (%1)"), self.ignore_tags)
end,
keep_menu_open = true,
callback = function(touchmenu_instance)
self:setTagsDialog(touchmenu_instance,
_("Tags to ignore"),
_("Enter a comma-separated list of tags to ignore."),
self.ignore_tags,
function(tags)
self.ignore_tags = tags
end
)
end,
},
{
text_func = function()
if not self.auto_tags or self.auto_tags == "" then
return _("Automatic tags")
end
return T(_("Automatic tags (%1)"), self.auto_tags)
end,
keep_menu_open = true,
callback = function(touchmenu_instance)
self:setTagsDialog(touchmenu_instance,
_("Tags to automatically add"),
_("Enter a comma-separated list of tags to automatically add to new articles."),
self.auto_tags,
function(tags)
self.auto_tags = tags
end
)
end,
separator = true,
},
{
text = _("Article deletion"),
separator = true,
sub_item_table = {
{
text = _("Remotely delete finished articles"),
checked_func = function() return self.is_delete_finished end,
callback = function()
self.is_delete_finished = not self.is_delete_finished
self:saveSettings()
end,
},
{
text = _("Remotely delete 100% read articles"),
checked_func = function() return self.is_delete_read end,
callback = function()
self.is_delete_read = not self.is_delete_read
self:saveSettings()
end,
separator = true,
},
{
text = _("Mark as finished instead of deleting"),
checked_func = function() return self.is_archiving_deleted end,
callback = function()
self.is_archiving_deleted = not self.is_archiving_deleted
self:saveSettings()
end,
separator = true,
},
{
text = _("Process deletions when downloading"),
checked_func = function() return self.is_auto_delete end,
callback = function()
self.is_auto_delete = not self.is_auto_delete
self:saveSettings()
end,
},
{
text = _("Synchronize remotely deleted files"),
checked_func = function() return self.is_sync_remote_delete end,
callback = function()
self.is_sync_remote_delete = not self.is_sync_remote_delete
self:saveSettings()
end,
},
},
},
{
text = _("Send review as tags"),
help_text = _("This allow you to write tags in the review field, separated by commas, which can then be sent to Wallabag."),
keep_menu_open = true,
checked_func = function()
return self.send_review_as_tags or false
end,
callback = function()
self.send_review_as_tags = not self.send_review_as_tags
self:saveSettings()
end,
},
{
text = _("Remove finished articles from history"),
keep_menu_open = true,
checked_func = function()
return self.remove_finished_from_history or false
end,
callback = function()
self.remove_finished_from_history = not self.remove_finished_from_history
self:saveSettings()
end,
},
{
text = _("Remove 100% read articles from history"),
keep_menu_open = true,
checked_func = function()
return self.remove_read_from_history or false
end,
callback = function()
self.remove_read_from_history = not self.remove_read_from_history
self:saveSettings()
end,
separator = true,
},
{
text = _("Help"),
keep_menu_open = true,
callback = function()
UIManager:show(InfoMessage:new{
text = _([[Download directory: use a directory that is exclusively used by the Wallabag plugin. Existing files in this directory risk being deleted.
Articles marked as finished or 100% read can be deleted from the server. Those articles can also be deleted automatically when downloading new articles if the 'Process deletions during download' option is enabled.
The 'Synchronize remotely deleted files' option will remove local files that no longer exist on the server.]])
})
end,
}
}
},
{
text = _("Info"),
keep_menu_open = true,
callback = function()
UIManager:show(InfoMessage:new{
text = T(_([[Wallabag is an open source read-it-later service. This plugin synchronizes with a Wallabag server.
More details: https://wallabag.org
Downloads to folder: %1]]), BD.dirpath(filemanagerutil.abbreviate(self.directory)))
})
end,
},
},
}
end
function Wallabag:getBearerToken()
-- Check if the configuration is complete
local function isempty(s)
return s == nil or s == ""
end
local server_empty = isempty(self.server_url) or isempty(self.username) or isempty(self.password) or isempty(self.client_id) or isempty(self.client_secret)
local directory_empty = isempty(self.directory)
if server_empty or directory_empty then
UIManager:show(MultiConfirmBox:new{
text = _("Please configure the server settings and set a download folder."),
choice1_text_func = function()
if server_empty then
return _("Server (★)")
else
return _("Server")
end
end,
choice1_callback = function() self:editServerSettings() end,
choice2_text_func = function()
if directory_empty then
return _("Folder (★)")
else
return _("Folder")
end
end,
choice2_callback = function() self:setDownloadDirectory() end,
})
return false
end
-- Check if the download directory is valid
local dir_mode = lfs.attributes(self.directory, "mode")
if dir_mode ~= "directory" then
UIManager:show(InfoMessage:new{
text = _("The download directory is not valid.\nPlease configure it in the settings.")
})
return false
end
if string.sub(self.directory, -1) ~= "/" then
self.directory = self.directory .. "/"
end
local now = os.time()
if self.token_expiry - now > 300 then
-- token still valid for a while, no need to renew
return true
end
local login_url = "/oauth/v2/token"
local body = {
grant_type = "password",
client_id = self.client_id,
client_secret = self.client_secret,
username = self.username,
password = self.password
}
local bodyJSON = JSON.encode(body)
local headers = {
["Content-type"] = "application/json",
["Accept"] = "application/json, */*",
["Content-Length"] = tostring(#bodyJSON),
}
local result = self:callAPI("POST", login_url, headers, bodyJSON, "")
if result then
self.access_token = result.access_token
self.token_expiry = now + result.expires_in
return true
else
UIManager:show(InfoMessage:new{
text = _("Could not login to Wallabag server."), })
return false
end
end
--- Get a JSON formatted list of articles from the server.
-- The list should have self.article_per_sync item, or less if an error occured.
-- If filter_tag is set, only articles containing this tag are queried.
-- If ignore_tags is defined, articles containing either of the tags are skipped.
function Wallabag:getArticleList()
local filtering = ""
if self.filter_tag ~= "" then
filtering = "&tags=" .. self.filter_tag
end
local article_list = {}
local page = 1
-- query the server for articles until we hit our target number
while #article_list < self.articles_per_sync do
-- get the JSON containing the article list
local articles_url = "/api/entries.json?archive=0"
.. "&page=" .. page
.. "&perPage=" .. self.articles_per_sync
.. filtering
local articles_json, err, code = self:callAPI("GET", articles_url, nil, "", "", true)
if err == "http_error" and code == 404 then
-- we may have hit the last page, there are no more articles
logger.dbg("Wallabag: couldn't get page #", page)
break -- exit while loop
elseif err or articles_json == nil then
-- another error has occured. Don't proceed with downloading
-- or deleting articles
logger.warn("Wallabag: download of page #", page, "failed with", err, code)
UIManager:show(InfoMessage:new{
text = _("Requesting article list failed."), })
return
end
-- We're only interested in the actual articles in the JSON
-- build an array of those so it's easier to manipulate later
local new_article_list = {}
for _, article in ipairs(articles_json._embedded.items) do
table.insert(new_article_list, article)
end
-- Apply the filters
new_article_list = self:filterIgnoredTags(new_article_list)
-- Append the filtered list to the final article list
for _, article in ipairs(new_article_list) do
if #article_list == self.articles_per_sync then
logger.dbg("Wallabag: hit the article target", self.articles_per_sync)
break
end
table.insert(article_list, article)
end
page = page + 1
end
return article_list
end
--- Remove all the articles from the list containing one of the ignored tags.
-- article_list: array containing a json formatted list of articles
-- returns: same array, but without any articles that contain an ignored tag.
function Wallabag:filterIgnoredTags(article_list)
-- decode all tags to ignore
local ignoring = {}
if self.ignore_tags ~= "" then
for tag in util.gsplit(self.ignore_tags, "[,]+", false) do
ignoring[tag] = true
end
end
-- rebuild a list without the ignored articles
local filtered_list = {}
for _, article in ipairs(article_list) do
local skip_article = false
for _, tag in ipairs(article.tags) do
if ignoring[tag.label] then
skip_article = true
logger.dbg("Wallabag: ignoring tag", tag.label, "in article",
article.id, ":", article.title)
break -- no need to look for other tags
end
end
if not skip_article then
table.insert(filtered_list, article)
end
end
return filtered_list
end
--- Download Wallabag article.
-- @string article
-- @treturn int 1 failed, 2 skipped, 3 downloaded
function Wallabag:download(article)
local skip_article = false
local title = util.getSafeFilename(article.title, self.directory, 230, 0)
local file_ext = ".epub"
local item_url = "/api/entries/" .. article.id .. "/export.epub"
-- If the article links to a supported file, we will download it directly.
-- All webpages are HTML. Ignore them since we want the Wallabag EPUB instead!
if article.mimetype ~= "text/html" then
if DocumentRegistry:hasProvider(nil, article.mimetype) then
file_ext = "."..DocumentRegistry:mimeToExt(article.mimetype)
item_url = article.url
-- A function represents `null` in our JSON.decode, because `nil` would just disappear.
-- In that case, fall back to the file extension.
elseif type(article.mimetype) == "function" and DocumentRegistry:hasProvider(article.url) then
file_ext = ""
item_url = article.url
end
end
local local_path = self.directory .. article_id_prefix .. article.id .. article_id_postfix .. title .. file_ext
logger.dbg("Wallabag: DOWNLOAD: id: ", article.id)
logger.dbg("Wallabag: DOWNLOAD: title: ", article.title)
logger.dbg("Wallabag: DOWNLOAD: filename: ", local_path)
local attr = lfs.attributes(local_path)
if attr then
-- File already exists, skip it. Preferably only skip if the date of local file is newer than server's.
-- newsdownloader.koplugin has a date parser but it is available only if the plugin is activated.
--- @todo find a better solution
if self.is_dateparser_available then
local server_date = self.dateparser.parse(article.updated_at)
if server_date < attr.modification then
skip_article = true
logger.dbg("Wallabag: skipping file (date checked): ", local_path)
end
else
skip_article = true
logger.dbg("Wallabag: skipping file: ", local_path)
end
end
if skip_article == false then
if self:callAPI("GET", item_url, nil, "", local_path) then
return downloaded
else
return failed
end
end
return skipped
end
-- method: (mandatory) GET, POST, DELETE, PATCH, etc...
-- apiurl: (mandatory) API call excluding the server path, or full URL to a file
-- headers: defaults to auth if given nil value, provide all headers necessary if in use
-- body: empty string if not needed
-- filepath: downloads the file if provided, returns JSON otherwise
-- @treturn result or (nil, "network_error") or (nil, "json_error")
-- or (nil, "http_error", code)
---- @todo separate call to internal API from the download on external server
function Wallabag:callAPI(method, apiurl, headers, body, filepath, quiet)
local sink = {}
local request = {}
-- Is it an API call, or a regular file direct download?
if apiurl:sub(1, 1) == "/" then
-- API call to our server, has the form "/random/api/call"
request.url = self.server_url .. apiurl
if headers == nil then
headers = {
["Authorization"] = "Bearer " .. self.access_token,
}
end
else
-- regular url link to a foreign server
local file_url = apiurl
request.url = file_url
if headers == nil then
-- no need for a token here
headers = {}
end
end
request.method = method
if filepath ~= "" then
request.sink = ltn12.sink.file(io.open(filepath, "w"))
socketutil:set_timeout(socketutil.FILE_BLOCK_TIMEOUT, socketutil.FILE_TOTAL_TIMEOUT)
else
request.sink = ltn12.sink.table(sink)
socketutil:set_timeout(socketutil.LARGE_BLOCK_TIMEOUT, socketutil.LARGE_TOTAL_TIMEOUT)
end
request.headers = headers
if body ~= "" then
request.source = ltn12.source.string(body)
end
logger.dbg("Wallabag: URL ", request.url)
logger.dbg("Wallabag: method ", method)
local code, resp_headers, status = socket.skip(1, http.request(request))
socketutil:reset_timeout()
-- raise error message when network is unavailable
if resp_headers == nil then
logger.dbg("Wallabag: Server error:", status or code)
return nil, "network_error"
end
if code == 200 then
if filepath ~= "" then
logger.dbg("Wallabag: file downloaded to", filepath)
return true
else
local content = table.concat(sink)
if content ~= "" and string.sub(content, 1,1) == "{" then
local ok, result = pcall(JSON.decode, content)
if ok and result then
-- Only enable this log when needed, the output can be large
--logger.dbg("Wallabag: result ", result)
return result
else
UIManager:show(InfoMessage:new{
text = _("Server response is not valid."), })
end
else
UIManager:show(InfoMessage:new{
text = _("Server response is not valid."), })
end
return nil, "json_error"
end
else
if filepath ~= "" then
local entry_mode = lfs.attributes(filepath, "mode")
if entry_mode == "file" then
os.remove(filepath)
logger.dbg("Wallabag: Removed failed download:", filepath)
end
elseif not quiet then
UIManager:show(InfoMessage:new{
text = _("Communication with server failed."), })
end
logger.dbg("Wallabag: Request failed:", status or code)
logger.dbg("Wallabag: Response headers:", resp_headers)
return nil, "http_error", code
end
end
function Wallabag:synchronize()
local info = InfoMessage:new{ text = _("Connecting…") }
UIManager:show(info)
UIManager:forceRePaint()
UIManager:close(info)
if self:getBearerToken() == false then
return false
end
if self.download_queue and next(self.download_queue) ~= nil then
info = InfoMessage:new{ text = _("Adding articles from queue…") }
UIManager:show(info)
UIManager:forceRePaint()
for _, articleUrl in ipairs(self.download_queue) do
self:addArticle(articleUrl)
end
self.download_queue = {}
self:saveSettings()
UIManager:close(info)
end
local deleted_count = self:processLocalFiles()
info = InfoMessage:new{ text = _("Getting article list…") }
UIManager:show(info)
UIManager:forceRePaint()
UIManager:close(info)
local remote_article_ids = {}
local downloaded_count = 0
local failed_count = 0
if self.access_token ~= "" then
local articles = self:getArticleList()
if articles then
logger.dbg("Wallabag: number of articles:", #articles)
info = InfoMessage:new{ text = _("Downloading articles…") }
UIManager:show(info)
UIManager:forceRePaint()
UIManager:close(info)
for _, article in ipairs(articles) do
logger.dbg("Wallabag: processing article ID: ", article.id)
remote_article_ids[ tostring(article.id) ] = true
local res = self:download(article)
if res == downloaded then
downloaded_count = downloaded_count + 1
elseif res == failed then
failed_count = failed_count + 1
end
end
-- synchronize remote deletions
deleted_count = deleted_count + self:processRemoteDeletes(remote_article_ids)
local msg
if failed_count ~= 0 then
msg = _("Processing finished.\n\nArticles downloaded: %1\nDeleted: %2\nFailed: %3")
info = InfoMessage:new{ text = T(msg, downloaded_count, deleted_count, failed_count) }
else
msg = _("Processing finished.\n\nArticles downloaded: %1\nDeleted: %2")
info = InfoMessage:new{ text = T(msg, downloaded_count, deleted_count) }
end
UIManager:show(info)
end -- articles
end -- access_token
end
function Wallabag:processRemoteDeletes(remote_article_ids)
if not self.is_sync_remote_delete then
logger.dbg("Wallabag: Processing of remote file deletions disabled.")
return 0
end
logger.dbg("Wallabag: articles IDs from server: ", remote_article_ids)
local info = InfoMessage:new{ text = _("Synchronizing remote deletions…") }
UIManager:show(info)
UIManager:forceRePaint()
UIManager:close(info)
local deleted_count = 0
for entry in lfs.dir(self.directory) do
if entry ~= "." and entry ~= ".." then
local entry_path = self.directory .. "/" .. entry
local id = self:getArticleID(entry_path)
if not remote_article_ids[ id ] then
logger.dbg("Wallabag: Deleting local file (deleted on server): ", entry_path)
self:deleteLocalArticle(entry_path)
deleted_count = deleted_count + 1
end
end
end -- for entry
return deleted_count
end
function Wallabag:processLocalFiles(mode)
if mode then
if self.is_auto_delete == false and mode ~= "manual" then
logger.dbg("Wallabag: Automatic processing of local files disabled.")
return 0, 0
end
end
if self:getBearerToken() == false then
return 0, 0
end
local num_deleted = 0
if self.is_delete_finished or self.is_delete_read then
local info = InfoMessage:new{ text = _("Processing local files…") }
UIManager:show(info)
UIManager:forceRePaint()
UIManager:close(info)
for entry in lfs.dir(self.directory) do
if entry ~= "." and entry ~= ".." then
local entry_path = self.directory .. "/" .. entry
if DocSettings:hasSidecarFile(entry_path) then
if self.send_review_as_tags then
self:addTags(entry_path)
end
local doc_settings = DocSettings:open(entry_path)
local summary = doc_settings:readSetting("summary")
local status = summary and summary.status
local percent_finished = doc_settings:readSetting("percent_finished")
if status == "complete" or status == "abandoned" then
if self.is_delete_finished then
self:removeArticle(entry_path)
num_deleted = num_deleted + 1
end
elseif percent_finished == 1 then -- 100% read
if self.is_delete_read then
self:removeArticle(entry_path)
num_deleted = num_deleted + 1
end
end
end -- has sidecar
end -- not . and ..
end -- for entry
end -- flag checks
return num_deleted
end
function Wallabag:addArticle(article_url)
logger.dbg("Wallabag: adding article ", article_url)
if not article_url or self:getBearerToken() == false then
return false
end
local body = {
url = article_url,
tags = self.auto_tags,
}
local body_JSON = JSON.encode(body)
local headers = {
["Content-type"] = "application/json",
["Accept"] = "application/json, */*",
["Content-Length"] = tostring(#body_JSON),
["Authorization"] = "Bearer " .. self.access_token,
}
return self:callAPI("POST", "/api/entries.json", headers, body_JSON, "")
end
function Wallabag:addTags(path)
logger.dbg("Wallabag: managing tags for article ", path)
local id = self:getArticleID(path)
if id then
local doc_settings = DocSettings:open(path)
local summary = doc_settings:readSetting("summary")
local tags = summary and summary.note
if tags and tags ~= "" then
logger.dbg("Wallabag: sending tags ", tags, " for ", path)
local body = {
tags = tags,
}
local bodyJSON = JSON.encode(body)
local headers = {
["Content-type"] = "application/json",
["Accept"] = "application/json, */*",
["Content-Length"] = tostring(#bodyJSON),
["Authorization"] = "Bearer " .. self.access_token,
}
self:callAPI("POST", "/api/entries/" .. id .. "/tags.json", headers, bodyJSON, "")
else
logger.dbg("Wallabag: no tags to send for ", path)
end
end
end
function Wallabag:removeArticle(path)
logger.dbg("Wallabag: removing article ", path)
local id = self:getArticleID(path)
if id then
if self.is_archiving_deleted then
local body = {
archive = 1
}
local bodyJSON = JSON.encode(body)
local headers = {
["Content-type"] = "application/json",
["Accept"] = "application/json, */*",
["Content-Length"] = tostring(#bodyJSON),
["Authorization"] = "Bearer " .. self.access_token,
}
self:callAPI("PATCH", "/api/entries/" .. id .. ".json", headers, bodyJSON, "")
else
self:callAPI("DELETE", "/api/entries/" .. id .. ".json", nil, "", "")
end
self:deleteLocalArticle(path)
end
end
function Wallabag:deleteLocalArticle(path)
if lfs.attributes(path, "mode") == "file" then
FileManager:deleteFile(path, true)
end
end
function Wallabag:getArticleID(path)
-- extract the Wallabag ID from the file name
local offset = self.directory:len() + 2 -- skip / and advance to the next char
local prefix_len = article_id_prefix:len()
if path:sub(offset , offset + prefix_len - 1) ~= article_id_prefix then
logger.warn("Wallabag: getArticleID: no match! ", path:sub(offset , offset + prefix_len - 1))
return
end
local endpos = path:find(article_id_postfix, offset + prefix_len)
if endpos == nil then
logger.warn("Wallabag: getArticleID: no match! ")
return
end
local id = path:sub(offset + prefix_len, endpos - 1)
return id
end
function Wallabag:refreshCurrentDirIfNeeded()
if FileManager.instance then
FileManager.instance:onRefresh()
end
end
function Wallabag:setFilterTag(touchmenu_instance)
self.tag_dialog = InputDialog:new {
title = _("Set a single tag to filter articles on"),
input = self.filter_tag,
buttons = {
{
{
text = _("Cancel"),
id = "close",
callback = function()
UIManager:close(self.tag_dialog)
end,
},
{
text = _("OK"),
is_enter_default = true,
callback = function()
self.filter_tag = self.tag_dialog:getInputText()
self:saveSettings()
touchmenu_instance:updateItems()
UIManager:close(self.tag_dialog)
end,
}
}
},
}
UIManager:show(self.tag_dialog)
self.tag_dialog:onShowKeyboard()
end
function Wallabag:setTagsDialog(touchmenu_instance, title, description, value, callback)
self.tags_dialog = InputDialog:new {
title = title,
description = description,
input = value,
buttons = {
{
{
text = _("Cancel"),
id = "close",
callback = function()
UIManager:close(self.tags_dialog)
end,
},
{
text = _("Set tags"),
is_enter_default = true,
callback = function()
callback(self.tags_dialog:getInputText())
self:saveSettings()
touchmenu_instance:updateItems()
UIManager:close(self.tags_dialog)
end,
}
}
},
}
UIManager:show(self.tags_dialog)
self.tags_dialog:onShowKeyboard()
end
function Wallabag:editServerSettings()
local text_info = T(_([[
Enter the details of your Wallabag server and account.
Client ID and client secret are long strings so you might prefer to save the empty settings and edit the config file directly in your installation directory:
%1/wallabag.lua
Restart KOReader after editing the config file.]]), BD.dirpath(DataStorage:getSettingsDir()))
self.settings_dialog = MultiInputDialog:new {
title = _("Wallabag settings"),
fields = {
{
text = self.server_url,
--description = T(_("Server URL:")),
hint = _("Server URL")
},
{
text = self.client_id,
--description = T(_("Client ID and secret")),
hint = _("Client ID")
},
{
text = self.client_secret,
hint = _("Client secret")
},
{
text = self.username,
--description = T(_("Username and password")),
hint = _("Username")
},
{
text = self.password,
text_type = "password",
hint = _("Password")
},
},
buttons = {
{
{
text = _("Cancel"),
id = "close",
callback = function()
UIManager:close(self.settings_dialog)
end
},
{
text = _("Info"),
callback = function()
UIManager:show(InfoMessage:new{ text = text_info })
end
},
{
text = _("Apply"),
callback = function()
local myfields = self.settings_dialog:getFields()
self.server_url = myfields[1]:gsub("/*$", "") -- remove all trailing "/" slashes
self.client_id = myfields[2]
self.client_secret = myfields[3]
self.username = myfields[4]
self.password = myfields[5]
self:saveSettings()
UIManager:close(self.settings_dialog)
end
},
},
},
}
UIManager:show(self.settings_dialog)
self.settings_dialog:onShowKeyboard()
end
function Wallabag:editClientSettings()
self.client_settings_dialog = MultiInputDialog:new {
title = _("Wallabag client settings"),
fields = {
{
text = self.articles_per_sync,
description = _("Number of articles"),
input_type = "number",
hint = _("Number of articles to download per sync")
},
},
buttons = {
{
{
text = _("Cancel"),
id = "close",
callback = function()
UIManager:close(self.client_settings_dialog)
end
},
{
text = _("Apply"),
callback = function()
local myfields = self.client_settings_dialog:getFields()
self.articles_per_sync = math.max(1, tonumber(myfields[1]) or self.articles_per_sync)
self:saveSettings(myfields)
UIManager:close(self.client_settings_dialog)
end
},
},
},
}
UIManager:show(self.client_settings_dialog)
self.client_settings_dialog:onShowKeyboard()
end
function Wallabag:setDownloadDirectory(touchmenu_instance)
require("ui/downloadmgr"):new{
onConfirm = function(path)
logger.dbg("Wallabag: set download directory to: ", path)
self.directory = path
self:saveSettings()
if touchmenu_instance then
touchmenu_instance:updateItems()
end
end,
}:chooseDir()
end
function Wallabag:saveSettings()
local tempsettings = {
server_url = self.server_url,
client_id = self.client_id,
client_secret = self.client_secret,
username = self.username,
password = self.password,
directory = self.directory,
filter_tag = self.filter_tag,
ignore_tags = self.ignore_tags,
auto_tags = self.auto_tags,
is_delete_finished = self.is_delete_finished,
is_delete_read = self.is_delete_read,
is_archiving_deleted = self.is_archiving_deleted,
is_auto_delete = self.is_auto_delete,
is_sync_remote_delete = self.is_sync_remote_delete,
articles_per_sync = self.articles_per_sync,
send_review_as_tags = self.send_review_as_tags,
remove_finished_from_history = self.remove_finished_from_history,
remove_read_from_history = self.remove_read_from_history,
download_queue = self.download_queue,
}
self.wb_settings:saveSetting("wallabag", tempsettings)
self.wb_settings:flush()
end
function Wallabag:readSettings()
local wb_settings = LuaSettings:open(DataStorage:getSettingsDir().."/wallabag.lua")
wb_settings:readSetting("wallabag", {})
return wb_settings
end
function Wallabag:saveWBSettings(setting)
if not self.wb_settings then self.wb_settings = self:readSettings() end
self.wb_settings:saveSetting("wallabag", setting)
self.wb_settings:flush()
end
function Wallabag:onAddWallabagArticle(article_url)
if not NetworkMgr:isOnline() then
self:addToDownloadQueue(article_url)
UIManager:show(InfoMessage:new{
text = T(_("Article added to download queue:\n%1"), BD.url(article_url)),
timeout = 1,
})
return
end
local wallabag_result = self:addArticle(article_url)
if wallabag_result then
UIManager:show(InfoMessage:new{
text = T(_("Article added to Wallabag:\n%1"), BD.url(article_url)),
})
else
UIManager:show(InfoMessage:new{
text = T(_("Error adding link to Wallabag:\n%1"), BD.url(article_url)),
})
end
-- stop propagation
return true
end
function Wallabag:onSynchronizeWallabag()
local connect_callback = function()
self:synchronize()
self:refreshCurrentDirIfNeeded()
end
NetworkMgr:runWhenOnline(connect_callback)
-- stop propagation
return true
end
function Wallabag:getLastPercent()
local percent = self.ui.paging and self.ui.paging:getLastPercent() or self.ui.rolling:getLastPercent()
return Math.roundPercent(percent)
end
function Wallabag:addToDownloadQueue(article_url)
table.insert(self.download_queue, article_url)
self:saveSettings()
end
function Wallabag:onCloseDocument()
if self.remove_finished_from_history or self.remove_read_from_history then
local document_full_path = self.ui.document.file
local summary = self.ui.doc_settings:readSetting("summary")
local status = summary and summary.status
local is_finished = status == "complete" or status == "abandoned"
local is_read = self:getLastPercent() == 1
if document_full_path
and self.directory
and ( (self.remove_finished_from_history and is_finished) or (self.remove_read_from_history and is_read) )
and self.directory == string.sub(document_full_path, 1, string.len(self.directory)) then
ReadHistory:removeItemByPath(document_full_path)
self.ui:setLastDirForFileBrowser(self.directory)
end
end
end
return Wallabag