2
0
mirror of https://github.com/koreader/koreader synced 2024-11-16 06:12:56 +00:00
koreader/plugins/newsdownloader.koplugin/main.lua
Roygbyte eb1e084d93
NewsDownloader: avoid some crashes (#9697)
Keep nil reference from being passed to UIManager.
2022-11-07 18:21:50 +01:00

1095 lines
39 KiB
Lua

local BD = require("ui/bidi")
local DataStorage = require("datastorage")
--local DownloadBackend = require("internaldownloadbackend")
--local DownloadBackend = require("luahttpdownloadbackend")
local DownloadBackend = require("epubdownloadbackend")
local ReadHistory = require("readhistory")
local FFIUtil = require("ffi/util")
local FeedView = require("feed_view")
local InfoMessage = require("ui/widget/infomessage")
local LuaSettings = require("frontend/luasettings")
local UIManager = require("ui/uimanager")
local KeyValuePage = require("ui/widget/keyvaluepage")
local InputDialog = require("ui/widget/inputdialog")
local MultiConfirmBox = require("ui/widget/multiconfirmbox")
local NetworkMgr = require("ui/network/manager")
local Persist = require("persist")
local WidgetContainer = require("ui/widget/container/widgetcontainer")
local dateparser = require("lib.dateparser")
local lfs = require("libs/libkoreader-lfs")
local logger = require("logger")
local util = require("util")
local _ = require("gettext")
local T = FFIUtil.template
local NewsDownloader = WidgetContainer:extend{
name = "news_downloader",
initialized = false,
feed_config_file = "feed_config.lua",
feed_config_path = nil,
news_config_file = "news_settings.lua",
settings = nil,
download_dir_name = "news",
download_dir = nil,
file_extension = ".epub",
config_key_custom_dl_dir = "custom_dl_dir",
empty_feed = {
[1] = "https://",
limit = 5,
download_full_article = true,
include_images = true,
enable_filter = false,
filter_element = ""
},
kv = nil, -- KeyValuePage
}
local FEED_TYPE_RSS = "rss"
local FEED_TYPE_ATOM = "atom"
--local initialized = false
--local feed_config_file_name = "feed_config.lua"
--local news_downloader_config_file = "news_downloader_settings.lua
-- If a title looks like <title>blabla</title> it'll just be feed.title.
-- If a title looks like <title attr="alb">blabla</title> then we get a table
-- where [1] is the title string and the attributes are also available.
local function getFeedTitle(possible_title)
if type(possible_title) == "string" then
return util.htmlEntitiesToUtf8(possible_title)
elseif possible_title[1] and type(possible_title[1]) == "string" then
return util.htmlEntitiesToUtf8(possible_title[1])
end
end
-- There can be multiple links.
-- For now we just assume the first link is probably the right one.
--- @todo Write unit tests.
-- Some feeds that can be used for unit test.
-- http://fransdejonge.com/feed/ for multiple links.
-- https://github.com/koreader/koreader/commits/master.atom for single link with attributes.
local function getFeedLink(possible_link)
local E = {}
if type(possible_link) == "string" then
return possible_link
elseif (possible_link._attr or E).href then
return possible_link._attr.href
elseif ((possible_link[1] or E)._attr or E).href then
return possible_link[1]._attr.href
end
end
function NewsDownloader:init()
self.ui.menu:registerToMainMenu(self)
end
function NewsDownloader:addToMainMenu(menu_items)
menu_items.news_downloader = {
text = _("News downloader (RSS/Atom)"),
sub_item_table_func = function()
return self:getSubMenuItems()
end,
}
end
function NewsDownloader:getSubMenuItems()
self:lazyInitialization()
local sub_item_table
sub_item_table = {
{
text = _("Go to news folder"),
callback = function()
self:openDownloadsFolder()
end,
},
{
text = _("Sync news feeds"),
keep_menu_open = true,
callback = function(touchmenu_instance)
NetworkMgr:runWhenOnline(function() self:loadConfigAndProcessFeedsWithUI(touchmenu_instance) end)
end,
},
{
text = _("Edit news feeds"),
keep_menu_open = true,
callback = function()
local Trapper = require("ui/trapper")
Trapper:wrap(function()
self:viewFeedList()
end)
end,
},
{
text = _("Settings"),
sub_item_table = {
{
text = _("Set download folder"),
keep_menu_open = true,
callback = function() self:setCustomDownloadDirectory() end,
},
{
text = _("Never download images"),
keep_menu_open = true,
checked_func = function()
return self.settings:isTrue("never_download_images")
end,
callback = function()
self.settings:toggle("never_download_images")
self.settings:flush()
end,
},
{
text = _("Delete all downloaded items"),
keep_menu_open = true,
callback = function()
local Trapper = require("ui/trapper")
Trapper:wrap(function()
local should_delete = Trapper:confirm(
_("Are you sure you want to delete all downloaded items?"),
_("Cancel"),
_("Delete")
)
if should_delete then
self:removeNewsButKeepFeedConfig()
Trapper:reset()
else
Trapper:reset()
end
end)
end,
},
},
},
{
text = _("About"),
keep_menu_open = true,
callback = function()
UIManager:show(InfoMessage:new{
text = T(_("News downloader retrieves RSS and Atom news entries and stores them to:\n%1\n\nEach entry is a separate EPUB file that can be browsed by KOReader.\nFeeds can be configured with download limits and other customization through the Edit Feeds menu item."),
BD.dirpath(self.download_dir))
})
end,
},
}
return sub_item_table
end
-- lazyInitialization sets up variables that point to the
-- Downloads folder and the feeds configuration file.
function NewsDownloader:lazyInitialization()
if not self.initialized then
logger.dbg("NewsDownloader: obtaining news folder")
self.settings = LuaSettings:open(("%s/%s"):format(DataStorage:getSettingsDir(), self.news_config_file))
-- Check to see if a custom download directory has been set.
if self.settings:has(self.config_key_custom_dl_dir) then
self.download_dir = self.settings:readSetting(self.config_key_custom_dl_dir)
else
self.download_dir =
("%s/%s/"):format(
DataStorage:getFullDataDir(),
self.download_dir_name)
end
logger.dbg("NewsDownloader: Custom directory set to:", self.download_dir)
-- If the directory doesn't exist we will create it.
if not lfs.attributes(self.download_dir, "mode") then
logger.dbg("NewsDownloader: Creating initial directory")
lfs.mkdir(self.download_dir)
end
-- Now set the path to the feed configuration file.
self.feed_config_path = self.download_dir .. self.feed_config_file
-- If the configuration file doesn't exist create it.
if not lfs.attributes(self.feed_config_path, "mode") then
logger.dbg("NewsDownloader: Creating initial feed config.")
FFIUtil.copyFile(FFIUtil.joinPath(self.path, self.feed_config_file),
self.feed_config_path)
end
self.initialized = true
end
end
function NewsDownloader:loadConfigAndProcessFeeds(touchmenu_instance)
local UI = require("ui/trapper")
logger.dbg("force repaint due to upcoming blocking calls")
local ok, feed_config = pcall(dofile, self.feed_config_path)
if not ok or not feed_config then
UI:info(T(_("Invalid configuration file. Detailed error message:\n%1"), feed_config))
return
end
-- If the file contains no table elements, then the user hasn't set any feeds.
if #feed_config <= 0 then
logger.err("NewsDownloader: empty feed list.", self.feed_config_path)
local should_edit_feed_list = UI:confirm(
T(_("Feed list is empty. If you want to download news, you'll have to add a feed first.")),
_("Close"),
_("Edit feed list")
)
if should_edit_feed_list then
-- Show the user a blank feed view so they can
-- add a feed to their list.
local feed_item_vc = FeedView:getItem(
1,
self.empty_feed,
function(id, edit_key, value)
self:editFeedAttribute(id, edit_key, value)
end
)
self:viewFeedItem(
feed_item_vc
)
end
return
end
local never_download_images = self.settings:isTrue("never_download_images")
local unsupported_feeds_urls = {}
local total_feed_entries = #feed_config
local feed_message
for idx, feed in ipairs(feed_config) do
local url = feed[1]
local limit = feed.limit
local download_full_article = feed.download_full_article == nil or feed.download_full_article
local include_images = not never_download_images and feed.include_images
local enable_filter = feed.enable_filter or feed.enable_filter == nil
local filter_element = feed.filter_element or feed.filter_element == nil
-- Check if the two required attributes are set.
if url and limit then
feed_message = T(_("Processing %1/%2:\n%3"), idx, total_feed_entries, BD.url(url))
UI:info(feed_message)
-- Process the feed source.
self:processFeedSource(
url,
tonumber(limit),
unsupported_feeds_urls,
download_full_article,
include_images,
feed_message,
enable_filter,
filter_element)
else
logger.warn("NewsDownloader: invalid feed config entry.", feed)
end
end
if #unsupported_feeds_urls <= 0 then
-- When no errors are present, we get a happy message.
feed_message = _("Downloading news finished.")
else
-- When some errors are present, we get a sour message that includes
-- information about the source of the error.
local unsupported_urls = ""
for key, value in pairs(unsupported_feeds_urls) do
-- Create the error message.
unsupported_urls = unsupported_urls .. " " .. value[1] .. " " .. value[2]
-- Not sure what this does.
if key ~= #unsupported_feeds_urls then
unsupported_urls = BD.url(unsupported_urls) .. ", "
end
end
-- Tell the user there were problems.
feed_message = _("Downloading news finished with errors.")
-- Display a dialogue that requires the user to acknowledge
-- that errors occured.
UI:confirm(
T(_([[
Could not process some feeds.
Unsupported format in: %1. Please
review your feed configuration file.]])
, unsupported_urls),
_("Continue"),
""
)
end
-- Clear the info widgets before displaying the next ui widget.
UI:clear()
-- Check to see if this method was called from the menu. If it was,
-- we will have gotten a touchmenu_instance. This will context gives the user
-- two options about what to do next, which are handled by this block.
if touchmenu_instance then
-- Ask the user if they want to go to their downloads folder
-- or if they'd rather remain at the menu.
feed_message = feed_message .. _("Go to download folder?")
local should_go_to_downloads = UI:confirm(
feed_message,
_("Close"),
_("Go to downloads")
)
if should_go_to_downloads then
-- Go to downloads folder.
UI:clear()
self:openDownloadsFolder()
touchmenu_instance:closeMenu()
NetworkMgr:afterWifiAction()
return
else
-- Return to the menu.
NetworkMgr:afterWifiAction()
return
end
end
return
end
function NewsDownloader:loadConfigAndProcessFeedsWithUI(touchmenu_instance)
local Trapper = require("ui/trapper")
Trapper:wrap(function()
self:loadConfigAndProcessFeeds(touchmenu_instance)
end)
end
function NewsDownloader:processFeedSource(url, limit, unsupported_feeds_urls, download_full_article, include_images, message, enable_filter, filter_element)
local ok, response = pcall(function()
return DownloadBackend:getResponseAsString(url)
end)
local feeds
-- Check to see if a response is available to deserialize.
if ok then
feeds = self:deserializeXMLString(response)
end
-- If the response is not available (for a reason that we don't know),
-- add the URL to the unsupported feeds list.
if not ok or not feeds then
local error_message
if not ok then
error_message = _("(Reason: Failed to download content)")
else
error_message = _("(Reason: Error during feed deserialization)")
end
table.insert(
unsupported_feeds_urls,
{
url,
error_message
}
)
return
end
-- Check to see if the feed uses RSS.
local is_rss = feeds.rss
and feeds.rss.channel
and feeds.rss.channel.title
and feeds.rss.channel.item
and feeds.rss.channel.item[1]
and feeds.rss.channel.item[1].title
and feeds.rss.channel.item[1].link
-- Check to see if the feed uses Atom.
local is_atom = feeds.feed
and feeds.feed.title
and feeds.feed.entry[1]
and feeds.feed.entry[1].title
and feeds.feed.entry[1].link
-- Process the feeds accordingly.
if is_atom then
ok = pcall(function()
return self:processFeed(
FEED_TYPE_ATOM,
feeds,
limit,
download_full_article,
include_images,
message,
enable_filter,
filter_element
)
end)
elseif is_rss then
ok = pcall(function()
return self:processFeed(
FEED_TYPE_RSS,
feeds,
limit,
download_full_article,
include_images,
message,
enable_filter,
filter_element
)
end)
end
-- If the feed can't be processed, or it is neither
-- Atom or RSS, then add it to the unsupported feeds list
-- and return an error message.
if not ok or (not is_rss and not is_atom) then
local error_message
if not ok then
error_message = _("(Reason: Failed to download content)")
elseif not is_rss then
error_message = _("(Reason: Couldn't process RSS)")
elseif not is_atom then
error_message = _("(Reason: Couldn't process Atom)")
end
table.insert(
unsupported_feeds_urls,
{
url,
error_message
}
)
end
end
function NewsDownloader:deserializeXMLString(xml_str)
-- uses LuaXML https://github.com/manoelcampos/LuaXML
-- The MIT License (MIT)
-- Copyright (c) 2016 Manoel Campos da Silva Filho
-- see: koreader/plugins/newsdownloader.koplugin/lib/LICENSE_LuaXML
local treehdl = require("lib/handler")
local libxml = require("lib/xml")
-- Instantiate the object that parses the XML file as a Lua table.
local xmlhandler = treehdl.simpleTreeHandler()
-- Remove UTF-8 byte order mark, as it will cause LuaXML to fail
xml_str = xml_str:gsub("^\xef\xbb\xbf", "", 1)
-- Instantiate the object that parses the XML to a Lua table.
local ok = pcall(function()
libxml.xmlParser(xmlhandler):parse(xml_str)
end)
if not ok then return end
return xmlhandler.root
end
function NewsDownloader:processFeed(feed_type, feeds, limit, download_full_article, include_images, message, enable_filter, filter_element)
local feed_title
local feed_item
local total_items
-- Setup the above vars based on feed type.
if feed_type == FEED_TYPE_RSS then
feed_title = util.htmlEntitiesToUtf8(feeds.rss.channel.title)
feed_item = feeds.rss.channel.item
total_items = (limit == 0)
and #feeds.rss.channel.item
or limit
else
feed_title = getFeedTitle(feeds.feed.title)
feed_item = feeds.feed.entry
total_items = (limit == 0)
and #feeds.feed.entry
or limit
end
-- Get the path to the output directory.
local feed_output_dir = ("%s%s/"):format(
self.download_dir,
util.getSafeFilename(util.htmlEntitiesToUtf8(feed_title)))
-- Create the output directory if it doesn't exist.
if not lfs.attributes(feed_output_dir, "mode") then
lfs.mkdir(feed_output_dir)
end
-- Download the feed
for index, feed in pairs(feed_item) do
-- If limit has been met, stop downloading feed.
if limit ~= 0 and index - 1 == limit then
break
end
-- Create a message to display during processing.
local article_message = T(
_("%1\n\nFetching article %2/%3:"),
message,
index,
total_items
)
-- Get the feed description.
local feed_description
if feed_type == FEED_TYPE_RSS then
feed_description = feed.description
else
feed_description = feed.summary
end
-- Download the article.
if download_full_article then
self:downloadFeed(
feed,
feed_output_dir,
include_images,
article_message,
enable_filter,
filter_element
)
else
self:createFromDescription(
feed,
feed_description,
feed_output_dir,
include_images,
article_message
)
end
end
end
local function parseDate(dateTime)
-- Uses lua-feedparser https://github.com/slact/lua-feedparser
-- feedparser is available under the (new) BSD license.
-- see: koreader/plugins/newsdownloader.koplugin/lib/LICENCE_lua-feedparser
local date = dateparser.parse(dateTime)
return os.date("%y-%m-%d_%H-%M_", date)
end
-- This appears to be used by Atom feeds in processFeed.
local function getTitleWithDate(feed)
local title = util.getSafeFilename(getFeedTitle(feed.title))
if feed.updated then
title = parseDate(feed.updated) .. title
elseif feed.pubDate then
title = parseDate(feed.pubDate) .. title
elseif feed.published then
title = parseDate(feed.published) .. title
end
return title
end
function NewsDownloader:downloadFeed(feed, feed_output_dir, include_images, message, enable_filter, filter_element)
local title_with_date = getTitleWithDate(feed)
local news_file_path = ("%s%s%s"):format(feed_output_dir,
title_with_date,
self.file_extension)
local file_mode = lfs.attributes(news_file_path, "mode")
if file_mode == "file" then
logger.dbg("NewsDownloader:", news_file_path, "already exists. Skipping")
else
logger.dbg("NewsDownloader: News file will be stored to :", news_file_path)
local article_message = T(_("%1\n%2"), message, title_with_date)
local link = getFeedLink(feed.link)
local html = DownloadBackend:loadPage(link)
DownloadBackend:createEpub(news_file_path, html, link, include_images, article_message, enable_filter, filter_element)
end
end
function NewsDownloader:createFromDescription(feed, content, feed_output_dir, include_images, message)
local title_with_date = getTitleWithDate(feed)
local news_file_path = ("%s%s%s"):format(feed_output_dir,
title_with_date,
self.file_extension)
local file_mode = lfs.attributes(news_file_path, "mode")
if file_mode == "file" then
logger.dbg("NewsDownloader:", news_file_path, "already exists. Skipping")
else
logger.dbg("NewsDownloader: News file will be stored to :", news_file_path)
local article_message = T(_("%1\n%2"), message, title_with_date)
local footer = _("This is just a description of the feed. To download the full article instead, go to the News Downloader settings and change 'download_full_article' to 'true'.")
local html = string.format([[<!DOCTYPE html>
<html>
<head><meta charset='UTF-8'><title>%s</title></head>
<body><header><h2>%s</h2></header><article>%s</article>
<br><footer><small>%s</small></footer>
</body>
</html>]], feed.title, feed.title, content, footer)
local link = getFeedLink(feed.link)
DownloadBackend:createEpub(news_file_path, html, link, include_images, article_message)
end
end
function NewsDownloader:removeNewsButKeepFeedConfig()
logger.dbg("NewsDownloader: Removing news from :", self.download_dir)
for entry in lfs.dir(self.download_dir) do
if entry ~= "." and entry ~= ".." and entry ~= self.feed_config_file then
local entry_path = self.download_dir .. "/" .. entry
local entry_mode = lfs.attributes(entry_path, "mode")
if entry_mode == "file" then
os.remove(entry_path)
elseif entry_mode == "directory" then
FFIUtil.purgeDir(entry_path)
end
end
end
UIManager:show(InfoMessage:new{
text = _("All downloaded news feed items deleted.")
})
end
function NewsDownloader:setCustomDownloadDirectory()
require("ui/downloadmgr"):new{
onConfirm = function(path)
logger.dbg("NewsDownloader: set download directory to: ", path)
self.settings:saveSetting(self.config_key_custom_dl_dir, ("%s/"):format(path))
self.settings:flush()
logger.dbg("NewsDownloader: Coping to new download folder previous self.feed_config_file from: ", self.feed_config_path)
FFIUtil.copyFile(self.feed_config_path, ("%s/%s"):format(path, self.feed_config_file))
self.initialized = false
self:lazyInitialization()
end,
}:chooseDir()
end
function NewsDownloader:viewFeedList()
local UI = require("ui/trapper")
UI:info(_("Loading news feed list…"))
-- Protected call to see if feed config path returns a file that can be opened.
local ok, feed_config = pcall(dofile, self.feed_config_path)
if not ok or not feed_config then
local change_feed_config = UI:confirm(
_("Could not open feed list. Feeds configuration file is invalid."),
_("Close"),
_("View file")
)
if change_feed_config then
self:changeFeedConfig()
end
return
end
UI:clear()
-- See if the config file contains any feed items
if #feed_config <= 0 then
logger.err("NewsDownloader: empty feed list.", self.feed_config_path)
-- Why not ask the user if they want to add one?
-- Or, in future, move along to our list UI with an entry for new feeds
--return
end
local view_content = FeedView:getList(
feed_config,
function(feed_item_vc)
self:viewFeedItem(
feed_item_vc
)
end,
function(id, edit_key, value)
self:editFeedAttribute(id, edit_key, value)
end,
function(id)
self:deleteFeed(id)
end
)
-- Add a "Add new feed" button with callback
table.insert(
view_content,
{
_("Add new feed"),
"",
callback = function()
-- Prepare the view with all the callbacks for editing the attributes
local feed_item_vc = FeedView:getItem(
#feed_config + 1,
self.empty_feed,
function(id, edit_key, value)
self:editFeedAttribute(id, edit_key, value)
end
)
self:viewFeedItem(
feed_item_vc
)
end
}
)
-- Show the list of feeds.
if self.kv then
UIManager:close(self.kv)
end
self.kv = KeyValuePage:new{
title = _("RSS/Atom feeds"),
value_overflow_align = "right",
kv_pairs = view_content,
callback_return = function()
UIManager:close(self.kv)
end
}
UIManager:show(self.kv)
end
function NewsDownloader:viewFeedItem(data)
if self.kv then
UIManager:close(self.kv)
end
self.kv = KeyValuePage:new{
title = _("Edit Feed"),
value_overflow_align = "left",
kv_pairs = data,
callback_return = function()
self:viewFeedList()
end
}
UIManager:show(self.kv)
end
function NewsDownloader:editFeedAttribute(id, key, value)
local kv = self.kv
-- There are basically two types of values: string (incl. numbers)
-- and booleans. This block chooses what type of value our
-- attribute will need and displays the corresponding dialog.
if key == FeedView.URL
or key == FeedView.LIMIT
or key == FeedView.FILTER_ELEMENT then
local title
local input_type
local description
if key == FeedView.URL then
title = _("Edit feed URL")
input_type = "string"
elseif key == FeedView.LIMIT then
title = _("Edit feed limit")
description = _("Set to 0 for no limit to how many items are downloaded")
input_type = "number"
elseif key == FeedView.FILTER_ELEMENT then
title = _("Edit filter element.")
description = _("Filter based on the given CSS selector. E.g.: name_of_css.element.class")
input_type = "string"
else
return false
end
local input_dialog
input_dialog = InputDialog:new{
title = title,
input = value,
input_type = input_type,
description = description,
buttons = {
{
{
text = _("Cancel"),
id = "close",
callback = function()
UIManager:close(input_dialog)
UIManager:show(kv)
end,
},
{
text = _("Save"),
is_enter_default = true,
callback = function()
UIManager:close(input_dialog)
self:updateFeedConfig(id, key, input_dialog:getInputValue())
end,
},
}
},
}
UIManager:show(input_dialog)
input_dialog:onShowKeyboard()
return true
else
local text
if key == FeedView.DOWNLOAD_FULL_ARTICLE then
text = _("Download full article?")
elseif key == FeedView.INCLUDE_IMAGES then
text = _("Include images?")
elseif key == FeedView.ENABLE_FILTER then
text = _("Enable CSS filter?")
end
local multi_box
multi_box= MultiConfirmBox:new{
text = text,
choice1_text = _("Yes"),
choice1_callback = function()
UIManager:close(multi_box)
self:updateFeedConfig(id, key, true)
end,
choice2_text = _("No"),
choice2_callback = function()
UIManager:close(multi_box)
self:updateFeedConfig(id, key, false)
end,
cancel_callback = function()
UIManager:close(multi_box)
UIManager:show(kv)
end,
}
UIManager:show(multi_box)
end
end
function NewsDownloader:updateFeedConfig(id, key, value)
local UI = require("ui/trapper")
-- Because this method is called at the menu,
-- we might not have an active view. So this conditional
-- statement avoids closing a null reference.
if self.kv then
UIManager:close(self.kv)
end
-- It's possible that we will get a null value.
-- This block catches that possibility.
if id ~= nil and key ~= nil and value ~= nil then
-- This logger is a bit opaque because T() wasn't playing nice with booleans
logger.dbg("Newsdownloader: attempting to update config:")
else
logger.dbg("Newsdownloader: null value supplied to update. Not updating config")
return
end
local ok, feed_config = pcall(dofile, self.feed_config_path)
if not ok or not feed_config then
UI:info(T(_("Invalid configuration file. Detailed error message:\n%1"), feed_config))
return
end
-- If the file contains no table elements, then the user hasn't set any feeds.
if #feed_config <= 0 then
logger.dbg("NewsDownloader: empty feed list.", self.feed_config_path)
end
-- Check to see if the id is larger than the number of feeds. If it is,
-- then we know this is a new add. Insert the base array.
if id > #feed_config then
table.insert(
feed_config,
self.empty_feed
)
end
local new_config = {}
-- In this loop, we cycle through the feed items. A series of
-- conditionals checks to see if we are at the right id
-- and key (i.e.: the key that triggered this function.
-- If we are at the right spot, we overrite (or create) the value
for idx, feed in ipairs(feed_config) do
-- Check to see if this is the correct feed to update.
if idx == id then
if key == FeedView.URL then
if feed[1] then
-- If the value exists, then it's been set. So all we do
-- is overwrite the value.
feed[1] = value
else
-- If the value doesn't exist, then we need to set it.
-- So we insert it into the table.
table.insert(
feed,
{
value
}
)
end
elseif key == FeedView.LIMIT then
if feed.limit then
feed.limit = value
else
table.insert(
feed,
{
"limit",
value
}
)
end
elseif key == FeedView.DOWNLOAD_FULL_ARTICLE then
if feed.download_full_article ~= nil then
feed.download_full_article = value
else
table.insert(
feed,
{
"download_full_article",
value
}
)
end
elseif key == FeedView.INCLUDE_IMAGES then
if feed.include_images ~= nil then
feed.include_images = value
else
table.insert(
feed,
{
"include_images",
value
}
)
end
elseif key == FeedView.ENABLE_FILTER then
if feed.enable_filter ~= nil then
feed.enable_filter = value
else
table.insert(
feed,
{
"enable_filter",
value
}
)
end
elseif key == FeedView.FILTER_ELEMENT then
if feed.filter_element then
feed.filter_element = value
else
table.insert(
feed,
{
"filter_element",
value
}
)
end
end
end
-- Now we insert the updated (or newly created) feed into the
-- new config feed that we're building in this loop.
table.insert(
new_config,
feed
)
end
-- Save the config
logger.dbg("NewsDownloader: config to save", new_config)
self:saveConfig(new_config)
-- Refresh the view
local feed_item_vc = FeedView:getItem(
id,
new_config[id],
function(cb_id, cb_edit_key, cb_value)
self:editFeedAttribute(cb_id, cb_edit_key, cb_value)
end
)
self:viewFeedItem(
feed_item_vc
)
end
function NewsDownloader:deleteFeed(id)
local UI = require("ui/trapper")
logger.dbg("Newsdownloader: attempting to delete feed")
-- Check to see if we can get the config file.
local ok, feed_config = pcall(dofile, self.feed_config_path)
if not ok or not feed_config then
UI:info(T(_("Invalid configuration file. Detailed error message:\n%1"), feed_config))
return
end
-- In this loop, we cycle through the feed items. A series of
-- conditionals checks to see if we are at the right id
-- and key (i.e.: the key that triggered this function.
-- If we are at the right spot, we overrite (or create) the value
local new_config = {}
for idx, feed in ipairs(feed_config) do
-- Check to see if this is the correct feed to update.
if idx ~= id then
table.insert(
new_config,
feed
)
end
end
-- Save the config
local Trapper = require("ui/trapper")
Trapper:wrap(function()
logger.dbg("NewsDownloader: config to save", new_config)
self:saveConfig(new_config)
end)
-- Refresh the view
self:viewFeedList()
end
function NewsDownloader:saveConfig(config)
local UI = require("ui/trapper")
UI:info(_("Saving news feed list…"))
local persist = Persist:new{
path = self.feed_config_path
}
local ok = persist:save(config)
if not ok then
UI:info(_("Could not save news feed config."))
else
UI:info(_("News feed config updated successfully."))
end
UI:reset()
end
function NewsDownloader:changeFeedConfig()
local feed_config_file = io.open(self.feed_config_path, "rb")
local config = feed_config_file:read("*all")
feed_config_file:close()
local config_editor
logger.info("NewsDownloader: opening configuration file", self.feed_config_path)
config_editor = InputDialog:new{
title = T(_("Config: %1"), BD.filepath(self.feed_config_path)),
input = config,
input_type = "string",
para_direction_rtl = false, -- force LTR
fullscreen = true,
condensed = true,
allow_newline = true,
cursor_at_end = false,
add_nav_bar = true,
reset_callback = function()
return config
end,
save_callback = function(content)
if content and #content > 0 then
local parse_error = util.checkLuaSyntax(content)
if not parse_error then
local syntax_okay, syntax_error = pcall(loadstring(content))
if syntax_okay then
feed_config_file = io.open(self.feed_config_path, "w")
feed_config_file:write(content)
feed_config_file:close()
return true, _("Configuration saved")
else
return false, T(_("Configuration invalid: %1"), syntax_error)
end
else
return false, T(_("Configuration invalid: %1"), parse_error)
end
end
return false, _("Configuration empty")
end,
}
UIManager:show(config_editor)
config_editor:onShowKeyboard()
end
function NewsDownloader:openDownloadsFolder()
local FileManager = require("apps/filemanager/filemanager")
if self.ui.document then
self.ui:onClose()
end
if FileManager.instance then
FileManager.instance:reinit(self.download_dir)
else
FileManager:showFiles(self.download_dir)
end
end
function NewsDownloader:onCloseDocument()
local document_full_path = self.ui.document.file
if document_full_path and self.download_dir and self.download_dir == string.sub(document_full_path, 1, string.len(self.download_dir)) then
logger.dbg("NewsDownloader: document_full_path:", document_full_path)
logger.dbg("NewsDownloader: self.download_dir:", self.download_dir)
logger.dbg("NewsDownloader: removing NewsDownloader file from history.")
ReadHistory:removeItemByPath(document_full_path)
local doc_dir = util.splitFilePathName(document_full_path)
self.ui:setLastDirForFileBrowser(doc_dir)
end
end
--
-- KeyValuePage doesn't like to get a table with sub tables.
-- This function flattens an array, moving all nested tables
-- up the food chain, so to speak
--
function NewsDownloader:flattenArray(base_array, source_array)
for key, value in pairs(source_array) do
if value[2] == nil then
-- If the value is empty, then it's probably supposed to be a line
table.insert(
base_array,
"---"
)
else
if value["callback"] then
table.insert(
base_array,
{
value[1], value[2], callback = value["callback"]
}
)
else
table.insert(
base_array,
{
value[1], value[2]
}
)
end
end
end
return base_array
end
return NewsDownloader