2020-01-04 00:18:51 +00:00
local BD = require ( " ui/bidi " )
2017-04-19 15:59:45 +00:00
local DataStorage = require ( " datastorage " )
2022-03-27 14:47:14 +00:00
--local DownloadBackend = require("internaldownloadbackend")
--local DownloadBackend = require("luahttpdownloadbackend")
local DownloadBackend = require ( " epubdownloadbackend " )
2017-10-27 21:00:49 +00:00
local ReadHistory = require ( " readhistory " )
2017-04-19 15:59:45 +00:00
local FFIUtil = require ( " ffi/util " )
2021-09-13 19:04:06 +00:00
local FeedView = require ( " feed_view " )
2017-08-11 14:28:25 +00:00
local InfoMessage = require ( " ui/widget/infomessage " )
local LuaSettings = require ( " frontend/luasettings " )
local UIManager = require ( " ui/uimanager " )
2021-09-13 19:04:06 +00:00
local KeyValuePage = require ( " ui/widget/keyvaluepage " )
local InputDialog = require ( " ui/widget/inputdialog " )
local MultiConfirmBox = require ( " ui/widget/multiconfirmbox " )
2017-10-22 07:11:23 +00:00
local NetworkMgr = require ( " ui/network/manager " )
2021-09-13 19:04:06 +00:00
local Persist = require ( " persist " )
2017-08-11 14:28:25 +00:00
local WidgetContainer = require ( " ui/widget/container/widgetcontainer " )
2022-03-27 14:47:14 +00:00
local dateparser = require ( " lib.dateparser " )
2017-08-11 14:28:25 +00:00
local logger = require ( " logger " )
local util = require ( " util " )
local _ = require ( " gettext " )
local T = FFIUtil.template
2017-04-19 15:59:45 +00:00
2018-08-17 18:54:11 +00:00
local NewsDownloader = WidgetContainer : new {
2021-09-13 19:04:06 +00:00
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 ,
2022-03-27 14:47:14 +00:00
file_extension = " .epub " ,
2021-09-13 19:04:06 +00:00
config_key_custom_dl_dir = " custom_dl_dir " ,
empty_feed = {
[ 1 ] = " https:// " ,
limit = 5 ,
download_full_article = true ,
include_images = true ,
enable_filter = false ,
2022-03-27 14:47:14 +00:00
filter_element = " "
2021-09-13 19:04:06 +00:00
} ,
kv = { }
2018-08-17 18:54:11 +00:00
}
2017-04-19 15:59:45 +00:00
2022-03-27 14:47:14 +00:00
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
2017-04-19 15:59:45 +00:00
function NewsDownloader : init ( )
self.ui . menu : registerToMainMenu ( self )
end
function NewsDownloader : addToMainMenu ( menu_items )
2017-05-08 20:15:42 +00:00
menu_items.news_downloader = {
2021-09-13 19:04:06 +00:00
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 )
2022-03-27 14:47:14 +00:00
NetworkMgr : runWhenOnline ( function ( ) self : loadConfigAndProcessFeedsWithUI ( touchmenu_instance ) end )
2021-09-13 19:04:06 +00:00
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 ,
} ,
2022-03-27 14:47:14 +00:00
{
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 ,
} ,
2021-09-13 19:04:06 +00:00
{
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 ,
2018-01-17 17:16:11 +00:00
} ,
2017-06-29 18:40:42 +00:00
} ,
2021-09-13 19:04:06 +00:00
} ,
{
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 \n Each entry is a separate EPUB file that can be browsed by KOReader. \n Feeds can be configured with download limits and other customization through the Edit Feeds menu item. " ) ,
BD.dirpath ( self.download_dir ) )
} )
end ,
2017-04-19 15:59:45 +00:00
} ,
}
2021-09-13 19:04:06 +00:00
return sub_item_table
2017-04-19 15:59:45 +00:00
end
2022-03-27 14:47:14 +00:00
-- lazyInitialization sets up variables that point to the
2021-09-14 05:42:32 +00:00
-- Downloads folder and the feeds configuration file.
2017-05-14 08:46:12 +00:00
function NewsDownloader : lazyInitialization ( )
2021-09-13 19:04:06 +00:00
if not self.initialized then
2017-05-14 08:46:12 +00:00
logger.dbg ( " NewsDownloader: obtaining news folder " )
2021-09-13 19:04:06 +00:00
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 )
2017-05-14 08:46:12 +00:00
else
2021-09-13 19:04:06 +00:00
self.download_dir =
( " %s/%s/ " ) : format (
DataStorage : getFullDataDir ( ) ,
self.download_dir_name )
2017-05-14 08:46:12 +00:00
end
2021-09-13 19:04:06 +00:00
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
2017-05-14 08:46:12 +00:00
logger.dbg ( " NewsDownloader: Creating initial directory " )
2021-09-13 19:04:06 +00:00
lfs.mkdir ( self.download_dir )
2017-05-14 08:46:12 +00:00
end
2021-09-13 19:04:06 +00:00
-- 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
2017-05-14 08:46:12 +00:00
logger.dbg ( " NewsDownloader: Creating initial feed config. " )
2021-09-13 19:04:06 +00:00
FFIUtil.copyFile ( FFIUtil.joinPath ( self.path , self.feed_config_file ) ,
self.feed_config_path )
2017-05-14 08:46:12 +00:00
end
2021-09-13 19:04:06 +00:00
self.initialized = true
2017-05-14 08:46:12 +00:00
end
end
2022-03-27 14:47:14 +00:00
function NewsDownloader : loadConfigAndProcessFeeds ( touchmenu_instance )
2019-04-30 22:37:03 +00:00
local UI = require ( " ui/trapper " )
2017-05-14 08:46:12 +00:00
logger.dbg ( " force repaint due to upcoming blocking calls " )
2022-03-27 14:47:14 +00:00
2021-09-13 19:04:06 +00:00
local ok , feed_config = pcall ( dofile , self.feed_config_path )
2017-04-19 15:59:45 +00:00
if not ok or not feed_config then
2019-04-30 22:37:03 +00:00
UI : info ( T ( _ ( " Invalid configuration file. Detailed error message: \n %1 " ) , feed_config ) )
2022-03-27 14:47:14 +00:00
return
2017-04-19 15:59:45 +00:00
end
2021-09-13 19:04:06 +00:00
-- If the file contains no table elements, then the user hasn't set any feeds.
2017-04-19 15:59:45 +00:00
if # feed_config <= 0 then
2021-09-13 19:04:06 +00:00
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
2022-03-27 14:47:14 +00:00
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
2017-05-08 20:15:42 +00:00
end
2022-03-27 14:47:14 +00:00
return
2017-04-19 15:59:45 +00:00
end
2022-03-27 14:47:14 +00:00
function NewsDownloader : loadConfigAndProcessFeedsWithUI ( touchmenu_instance )
2019-04-30 22:37:03 +00:00
local Trapper = require ( " ui/trapper " )
Trapper : wrap ( function ( )
2022-03-27 14:47:14 +00:00
self : loadConfigAndProcessFeeds ( touchmenu_instance )
end )
end
2017-04-19 15:59:45 +00:00
2022-03-27 14:47:14 +00:00
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
2018-04-12 14:05:50 +00:00
2022-03-27 14:47:14 +00:00
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 ( )
-- 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
2018-04-12 14:05:50 +00:00
2022-03-27 14:47:14 +00:00
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 \n Fetching 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
2017-08-11 14:28:25 +00:00
2022-03-27 14:47:14 +00:00
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
2017-05-08 20:15:42 +00:00
2022-03-27 14:47:14 +00:00
-- 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
2017-10-26 20:44:03 +00:00
2022-03-27 14:47:14 +00:00
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 )
2022-01-21 18:23:21 +00:00
2022-03-27 14:47:14 +00:00
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
2022-01-21 18:23:21 +00:00
2022-03-27 14:47:14 +00:00
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'. " )
2022-01-21 18:23:21 +00:00
2022-03-27 14:47:14 +00:00
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
2017-10-26 20:44:03 +00:00
end
2017-05-14 08:46:12 +00:00
function NewsDownloader : removeNewsButKeepFeedConfig ( )
2021-09-13 19:04:06 +00:00
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
2017-05-14 08:46:12 +00:00
local entry_mode = lfs.attributes ( entry_path , " mode " )
if entry_mode == " file " then
2018-08-07 19:03:43 +00:00
os.remove ( entry_path )
2017-05-14 08:46:12 +00:00
elseif entry_mode == " directory " then
FFIUtil.purgeDir ( entry_path )
end
end
end
UIManager : show ( InfoMessage : new {
2022-03-27 14:47:14 +00:00
text = _ ( " All downloaded news feed items deleted. " )
2017-05-14 08:46:12 +00:00
} )
end
function NewsDownloader : setCustomDownloadDirectory ( )
require ( " ui/downloadmgr " ) : new {
2021-09-13 19:04:06 +00:00
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 ( )
2022-03-27 14:47:14 +00:00
local UI = require ( " ui/trapper " )
UI : info ( _ ( " Loading news feed list… " ) )
2021-09-13 19:04:06 +00:00
-- 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
2021-09-14 05:42:32 +00:00
local change_feed_config = UI : confirm (
2021-09-26 10:44:43 +00:00
_ ( " Could not open feed list. Feeds configuration file is invalid. " ) ,
2021-09-13 19:04:06 +00:00
_ ( " Close " ) ,
_ ( " View file " )
)
if change_feed_config then
self : changeFeedConfig ( )
end
return
end
2022-03-27 14:47:14 +00:00
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
2021-09-13 19:04:06 +00:00
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 ,
2022-03-27 14:47:14 +00:00
function ( id )
self : deleteFeed ( id )
2021-09-13 19:04:06 +00:00
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 ~= 0 then
UIManager : close ( self.kv )
end
self.kv = KeyValuePage : new {
2021-09-26 10:44:43 +00:00
title = _ ( " RSS/Atom feeds " ) ,
2021-09-13 19:04:06 +00:00
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 ~= 0 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
2022-03-27 14:47:14 +00:00
-- 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
2021-09-13 19:04:06 +00:00
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 " ) ,
2022-03-04 20:20:00 +00:00
id = " close " ,
2021-09-13 19:04:06 +00:00
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 )
2021-09-14 05:42:32 +00:00
local UI = require ( " ui/trapper " )
2021-09-13 19:04:06 +00:00
-- 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 ~= 0 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
2022-03-27 14:47:14 +00:00
2021-09-13 19:04:06 +00:00
-- 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
2022-03-27 14:47:14 +00:00
if feed.limit then
feed.limit = value
else
table.insert (
feed ,
{
" limit " ,
value
}
)
end
2021-09-13 19:04:06 +00:00
elseif key == FeedView.DOWNLOAD_FULL_ARTICLE then
2022-03-27 14:47:14 +00:00
if feed.download_full_article ~= nil then
feed.download_full_article = value
else
table.insert (
feed ,
{
" download_full_article " ,
value
}
)
end
2021-09-13 19:04:06 +00:00
elseif key == FeedView.INCLUDE_IMAGES then
2022-03-27 14:47:14 +00:00
if feed.include_images ~= nil then
feed.include_images = value
else
table.insert (
feed ,
{
" include_images " ,
value
}
)
end
2021-09-13 19:04:06 +00:00
elseif key == FeedView.ENABLE_FILTER then
2022-03-27 14:47:14 +00:00
if feed.enable_filter ~= nil then
feed.enable_filter = value
else
table.insert (
feed ,
{
" enable_filter " ,
value
}
)
end
2021-09-13 19:04:06 +00:00
elseif key == FeedView.FILTER_ELEMENT then
2022-03-27 14:47:14 +00:00
if feed.filter_element then
feed.filter_element = value
else
table.insert (
feed ,
{
" filter_element " ,
value
}
)
end
2021-09-13 19:04:06 +00:00
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 ] ,
2021-09-14 05:42:32 +00:00
function ( cb_id , cb_edit_key , cb_value )
self : editFeedAttribute ( cb_id , cb_edit_key , cb_value )
2021-09-13 19:04:06 +00:00
end
)
self : viewFeedItem (
feed_item_vc
)
2022-03-27 14:47:14 +00:00
2021-09-13 19:04:06 +00:00
end
function NewsDownloader : deleteFeed ( id )
2021-09-14 05:42:32 +00:00
local UI = require ( " ui/trapper " )
2021-09-13 19:04:06 +00:00
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 " )
2021-09-26 10:44:43 +00:00
UI : info ( _ ( " Saving news feed list… " ) )
2021-09-13 19:04:06 +00:00
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 ( )
2017-05-14 08:46:12 +00:00
end
2018-08-07 19:03:43 +00:00
function NewsDownloader : changeFeedConfig ( )
2021-09-13 19:04:06 +00:00
local feed_config_file = io.open ( self.feed_config_path , " rb " )
2018-08-07 19:03:43 +00:00
local config = feed_config_file : read ( " *all " )
feed_config_file : close ( )
local config_editor
2021-09-13 19:04:06 +00:00
logger.info ( " NewsDownloader: opening configuration file " , self.feed_config_path )
2018-08-07 19:03:43 +00:00
config_editor = InputDialog : new {
2021-09-13 19:04:06 +00:00
title = T ( _ ( " Config: %1 " ) , BD.filepath ( self.feed_config_path ) ) ,
2018-08-07 19:03:43 +00:00
input = config ,
input_type = " string " ,
2019-12-06 21:55:35 +00:00
para_direction_rtl = false , -- force LTR
2018-08-07 19:03:43 +00:00
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
2021-09-13 19:04:06 +00:00
feed_config_file = io.open ( self.feed_config_path , " w " )
2018-08-07 19:03:43 +00:00
feed_config_file : write ( content )
feed_config_file : close ( )
return true , _ ( " Configuration saved " )
else
return false , T ( _ ( " Configuration invalid: %1 " ) , syntax_error )
end
else
2021-09-13 19:04:06 +00:00
return false , T ( _ ( " Configuration invalid: %1 " ) , parse_error )
end
2018-08-07 19:03:43 +00:00
end
return false , _ ( " Configuration empty " )
end ,
}
UIManager : show ( config_editor )
config_editor : onShowKeyboard ( )
end
2021-09-13 19:04:06 +00:00
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
2018-08-07 19:03:43 +00:00
2017-10-27 21:00:49 +00:00
function NewsDownloader : onCloseDocument ( )
local document_full_path = self.ui . document.file
2021-09-13 19:04:06 +00:00
if document_full_path and self.download_dir and self.download_dir == string.sub ( document_full_path , 1 , string.len ( self.download_dir ) ) then
2017-10-27 21:00:49 +00:00
logger.dbg ( " NewsDownloader: document_full_path: " , document_full_path )
2021-09-13 19:04:06 +00:00
logger.dbg ( " NewsDownloader: self.download_dir: " , self.download_dir )
2017-10-27 21:00:49 +00:00
logger.dbg ( " NewsDownloader: removing NewsDownloader file from history. " )
ReadHistory : removeItemByPath ( document_full_path )
2020-05-19 16:21:58 +00:00
local doc_dir = util.splitFilePathName ( document_full_path )
self.ui : setLastDirForFileBrowser ( doc_dir )
2017-10-27 21:00:49 +00:00
end
end
2022-03-27 14:47:14 +00:00
--
-- 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
2017-04-19 15:59:45 +00:00
return NewsDownloader