#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace newsbeuter { #define LOCK_SUFFIX ".lock" std::string lock_file; int ctrl_c_hit = 0; void ctrl_c_action(int sig) { LOG(LOG_DEBUG,"caught signal %d",sig); if (SIGINT == sig) { ctrl_c_hit = 1; } else { stfl::reset(); utils::remove_fs_lock(lock_file); ::exit(EXIT_FAILURE); } } void ignore_signal(int sig) { LOG(LOG_WARN, "caught signal %d but ignored it", sig); } void omg_a_child_died(int /* sig */) { pid_t pid; int stat; while ((pid = waitpid(-1,&stat,WNOHANG)) > 0) { } ::signal(SIGCHLD, omg_a_child_died); /* in case of unreliable signals */ } controller::controller() : v(0), urlcfg(0), rsscache(0), url_file("urls"), cache_file("cache.db"), config_file("config"), queue_file("queue"), refresh_on_start(false), api(0), offline_mode(false) { } /** * \brief Try to setup XDG style dirs. * * returns false, if that fails */ bool controller::setup_dirs_xdg(const char *env_home) { const char *env_xdg_config; const char *env_xdg_data; std::string xdg_config_dir; std::string xdg_data_dir; env_xdg_config = ::getenv("XDG_CONFIG_HOME"); if (env_xdg_config) { xdg_config_dir = env_xdg_config; } else { xdg_config_dir = env_home; xdg_config_dir.append(NEWSBEUTER_PATH_SEP); xdg_config_dir.append(".config"); } env_xdg_data = ::getenv("XDG_DATA_HOME"); if (env_xdg_data) { xdg_data_dir = env_xdg_data; } else { xdg_data_dir = env_home; xdg_data_dir.append(NEWSBEUTER_PATH_SEP); xdg_data_dir.append(".local"); xdg_data_dir.append(NEWSBEUTER_PATH_SEP); xdg_data_dir.append("share"); } xdg_config_dir.append(NEWSBEUTER_PATH_SEP); xdg_config_dir.append(NEWSBEUTER_SUBDIR_XDG); xdg_data_dir.append(NEWSBEUTER_PATH_SEP); xdg_data_dir.append(NEWSBEUTER_SUBDIR_XDG); if (access(xdg_config_dir.c_str(), R_OK | X_OK) != 0) { std::cout << utils::strprintf(_("XDG: configuration directory '%s' not accessible, using '%s' instead."), xdg_config_dir.c_str(), config_dir.c_str()) << std::endl; return false; } if (access(xdg_data_dir.c_str(), R_OK | X_OK | W_OK) != 0) { std::cout << utils::strprintf(_("XDG: data directory '%s' not accessible, using '%s' instead."), xdg_data_dir.c_str(), config_dir.c_str()) << std::endl; return false; } config_dir = xdg_config_dir; /* in config */ url_file = config_dir + std::string(NEWSBEUTER_PATH_SEP) + url_file; config_file = config_dir + std::string(NEWSBEUTER_PATH_SEP) + config_file; /* in data */ cache_file = xdg_data_dir + std::string(NEWSBEUTER_PATH_SEP) + cache_file; lock_file = cache_file + LOCK_SUFFIX; queue_file = xdg_data_dir + std::string(NEWSBEUTER_PATH_SEP) + queue_file; searchfile = utils::strprintf("%s%shistory.search", xdg_data_dir.c_str(), NEWSBEUTER_PATH_SEP); cmdlinefile = utils::strprintf("%s%shistory.cmdline", xdg_data_dir.c_str(), NEWSBEUTER_PATH_SEP); return true; } void controller::setup_dirs() { const char * env_home; if (!(env_home = ::getenv("HOME"))) { struct passwd * spw = ::getpwuid(::getuid()); if (spw) { env_home = spw->pw_dir; } else { std::cout << _("Fatal error: couldn't determine home directory!") << std::endl; std::cout << utils::strprintf(_("Please set the HOME environment variable or add a valid user for UID %u!"), ::getuid()) << std::endl; ::exit(EXIT_FAILURE); } } config_dir = env_home; config_dir.append(NEWSBEUTER_PATH_SEP); config_dir.append(NEWSBEUTER_CONFIG_SUBDIR); if (setup_dirs_xdg(env_home)) return; mkdir(config_dir.c_str(),0700); // create configuration directory if it doesn't exist url_file = config_dir + std::string(NEWSBEUTER_PATH_SEP) + url_file; cache_file = config_dir + std::string(NEWSBEUTER_PATH_SEP) + cache_file; lock_file = cache_file + LOCK_SUFFIX; config_file = config_dir + std::string(NEWSBEUTER_PATH_SEP) + config_file; queue_file = config_dir + std::string(NEWSBEUTER_PATH_SEP) + queue_file; searchfile = utils::strprintf("%s%shistory.search", config_dir.c_str(), NEWSBEUTER_PATH_SEP); cmdlinefile = utils::strprintf("%s%shistory.cmdline", config_dir.c_str(), NEWSBEUTER_PATH_SEP); } controller::~controller() { delete rsscache; delete urlcfg; delete api; scope_mutex feedslock(&feeds_mutex); for (std::vector >::iterator it=feeds.begin();it!=feeds.end();it++) { scope_mutex lock(&((*it)->item_mutex)); (*it)->items().clear(); } feeds.clear(); } void controller::set_view(view * vv) { v = vv; } void controller::run(int argc, char * argv[]) { int c; setup_dirs(); ::signal(SIGINT, ctrl_c_action); ::signal(SIGPIPE, ignore_signal); ::signal(SIGHUP, ctrl_c_action); ::signal(SIGCHLD, omg_a_child_died); bool do_import = false, do_export = false, cachefile_given_on_cmdline = false, do_vacuum = false; bool real_offline_mode = false; std::string importfile; bool do_read_import = false, do_read_export = false; std::string readinfofile; unsigned int show_version = 0; bool silent = false; bool execute_cmds = false; do { if((c = ::getopt(argc,argv,"i:erhqu:c:C:d:l:vVoxXI:E:"))<0) continue; switch (c) { case ':': /* fall-through */ case '?': /* missing option */ usage(argv[0]); break; case 'i': if (do_export) usage(argv[0]); do_import = true; silent = true; importfile = optarg; break; case 'r': refresh_on_start = true; break; case 'e': if (do_import) usage(argv[0]); do_export = true; silent = true; break; case 'h': usage(argv[0]); break; case 'u': url_file = optarg; break; case 'c': cache_file = optarg; lock_file = std::string(cache_file) + LOCK_SUFFIX; cachefile_given_on_cmdline = true; break; case 'C': config_file = optarg; break; case 'X': do_vacuum = true; break; case 'v': case 'V': show_version++; break; case 'o': offline_mode = true; break; case 'x': execute_cmds = true; silent = true; break; case 'q': silent = true; break; case 'd': // this is an undocumented debug commandline option! GetLogger().set_logfile(optarg); break; case 'l': // this is an undocumented debug commandline option! { loglevel level = static_cast(atoi(optarg)); if (level > LOG_NONE && level <= LOG_DEBUG) GetLogger().set_loglevel(level); } break; case 'I': if (do_read_export) usage(argv[0]); do_read_import = true; readinfofile = optarg; break; case 'E': if (do_read_import) usage(argv[0]); do_read_export = true; readinfofile = optarg; break; default: std::cout << utils::strprintf(_("%s: unknown option - %c"), argv[0], static_cast(c)) << std::endl; usage(argv[0]); break; } } while (c != -1); if (show_version) { version_information(argv[0], show_version); } if (do_import) { LOG(LOG_INFO,"Importing OPML file from %s",importfile.c_str()); urlcfg = new file_urlreader(url_file); urlcfg->reload(); import_opml(importfile.c_str()); return; } LOG(LOG_INFO, "nl_langinfo(CODESET): %s", nl_langinfo(CODESET)); if (!do_export) { if (!silent) std::cout << utils::strprintf(_("Starting %s %s..."), PROGRAM_NAME, PROGRAM_VERSION) << std::endl; pid_t pid; if (!utils::try_fs_lock(lock_file, pid)) { if (pid > 0) { LOG(LOG_ERROR,"an instance is already running: pid = %u",pid); } else { LOG(LOG_ERROR,"something went wrong with the lock: %s", strerror(errno)); } if (!execute_cmds) { std::cout << utils::strprintf(_("Error: an instance of %s is already running (PID: %u)"), PROGRAM_NAME, pid) << std::endl; } return; } } if (!silent) std::cout << _("Loading configuration..."); std::cout.flush(); cfg.register_commands(cfgparser); colorman.register_commands(cfgparser); keymap keys(KM_NEWSBEUTER); cfgparser.register_handler("bind-key",&keys); cfgparser.register_handler("unbind-key",&keys); cfgparser.register_handler("macro", &keys); cfgparser.register_handler("ignore-article",&ign); cfgparser.register_handler("always-download",&ign); cfgparser.register_handler("reset-unread-on-update",&ign); cfgparser.register_handler("define-filter",&filters); cfgparser.register_handler("highlight", &rxman); cfgparser.register_handler("highlight-article", &rxman); try { cfgparser.parse("/etc/" PROGRAM_NAME "/config"); cfgparser.parse(config_file); } catch (const configexception& ex) { LOG(LOG_ERROR,"an exception occured while parsing the configuration file: %s",ex.what()); std::cout << ex.what() << std::endl; utils::remove_fs_lock(lock_file); return; } update_config(); if (!silent) std::cout << _("done.") << std::endl; // create cache object std::string cachefilepath = cfg.get_configvalue("cache-file"); if (cachefilepath.length() > 0 && !cachefile_given_on_cmdline) { cache_file = cachefilepath.c_str(); // ok, we got another cache file path via the configuration // that means we need to remove the old lock file, assemble // the new lock file's name, and then try to lock it. utils::remove_fs_lock(lock_file); lock_file = std::string(cache_file) + LOCK_SUFFIX; pid_t pid; if (!utils::try_fs_lock(lock_file, pid)) { if (pid > 0) { LOG(LOG_ERROR,"an instance is already running: pid = %u",pid); } else { LOG(LOG_ERROR,"something went wrong with the lock: %s", strerror(errno)); } std::cout << utils::strprintf(_("Error: an instance of %s is already running (PID: %u)"), PROGRAM_NAME, pid) << std::endl; return; } } if (!silent) { std::cout << _("Opening cache..."); std::cout.flush(); } try { rsscache = new cache(cache_file,&cfg); } catch (const dbexception& e) { std::cout << utils::strprintf(_("Error: opening the cache file `%s' failed: %s"), cache_file.c_str(), e.what()) << std::endl; utils::remove_fs_lock(lock_file); ::exit(EXIT_FAILURE); } if (!silent) { std::cout << _("done.") << std::endl; } std::string type = cfg.get_configvalue("urls-source"); if (type == "local") { urlcfg = new file_urlreader(url_file); } else if (type == "opml") { urlcfg = new opml_urlreader(&cfg); real_offline_mode = offline_mode; } else if (type == "googlereader") { api = new googlereader_api(&cfg); urlcfg = new googlereader_urlreader(&cfg, url_file, api); real_offline_mode = offline_mode; } else { LOG(LOG_ERROR,"unknown urls-source `%s'", urlcfg->get_source().c_str()); } if (real_offline_mode) { if (!do_export) { std::cout << _("Loading URLs from local cache..."); std::cout.flush(); } urlcfg->set_offline(true); urlcfg->get_urls() = rsscache->get_feed_urls(); if (!do_export) { std::cout << _("done.") << std::endl; } } else { if (!do_export && !silent) { std::cout << utils::strprintf(_("Loading URLs from %s..."), urlcfg->get_source().c_str()); std::cout.flush(); } if (api) { if (!api->authenticate()) { std::cout << "Authentication failed." << std::endl; utils::remove_fs_lock(lock_file); return; } } urlcfg->reload(); if (!do_export && !silent) { std::cout << _("done.") << std::endl; } if (api && type == "googlereader") { // ugly hack! std::vector actions = rsscache->get_google_replay(); if (actions.size() > 0) { std::cout << _("Updating Google Reader unread states..."); std::cout.flush(); std::vector successful_guids = dynamic_cast(api)->bulk_mark_articles_read(actions); rsscache->delete_google_replay_by_guid(successful_guids); std::cout << _("done.") << std::endl; } } } if (urlcfg->get_urls().size() == 0) { LOG(LOG_ERROR,"no URLs configured."); std::string msg; if (type == "local") { msg = utils::strprintf(_("Error: no URLs configured. Please fill the file %s with RSS feed URLs or import an OPML file."), url_file.c_str()); } else if (type == "opml") { msg = utils::strprintf(_("It looks like the OPML feed you subscribed contains no feeds. Please fill it with feeds, and try again.")); } else if (type == "googlereader") { msg = utils::strprintf(_("It looks like you haven't configured any feeds in your Google Reader account. Please do so, and try again.")); } else { assert(0); // shouldn't happen } std::cout << msg << std::endl << std::endl; usage(argv[0]); } if (!do_export && !do_vacuum && !silent) std::cout << _("Loading articles from cache..."); if (do_vacuum) std::cout << _("Opening cache..."); std::cout.flush(); if (do_vacuum) { std::cout << _("done.") << std::endl; std::cout << _("Cleaning up cache thoroughly..."); std::cout.flush(); rsscache->do_vacuum(); std::cout << _("done.") << std::endl; utils::remove_fs_lock(lock_file); return; } unsigned int i=0; for (std::vector::const_iterator it=urlcfg->get_urls().begin(); it != urlcfg->get_urls().end(); ++it, ++i) { std::tr1::shared_ptr feed(new rss_feed(rsscache)); try { feed->set_rssurl(*it); feed->set_tags(urlcfg->get_tags(*it)); bool ignore_disp = (cfg.get_configvalue("ignore-mode") == "display"); rsscache->internalize_rssfeed(feed, ignore_disp ? &ign : NULL); } catch(const dbexception& e) { std::cout << _("Error while loading feeds from database: ") << e.what() << std::endl; utils::remove_fs_lock(lock_file); return; } catch(const std::string& str) { std::cout << utils::strprintf(_("Error while loading feed '%s': %s"), it->c_str(), str.c_str()) << std::endl; utils::remove_fs_lock(lock_file); return; } feed->set_order(i); scope_mutex feedslock(&feeds_mutex); feeds.push_back(feed); } sort_feeds(); std::vector tags = urlcfg->get_alltags(); if (!do_export && !silent) std::cout << _("done.") << std::endl; // if configured, we fill all query feeds with some data; no need to sort it, it will be refilled when actually opening it. if (cfg.get_configvalue_as_bool("prepopulate-query-feeds")) { std::cout << _("Prepopulating query feeds..."); std::cout.flush(); scope_mutex feedslock(&feeds_mutex); for (std::vector >::iterator it=feeds.begin();it!=feeds.end();it++) { if ((*it)->rssurl().substr(0,6) == "query:") { (*it)->update_items(get_all_feeds_unlocked()); } } std::cout << _("done.") << std::endl; } if (do_export) { export_opml(); utils::remove_fs_lock(lock_file); return; } if (do_read_import) { LOG(LOG_INFO,"Importing read information file from %s",readinfofile.c_str()); std::cout << _("Importing list of read articles..."); std::cout.flush(); import_read_information(readinfofile); std::cout << _("done.") << std::endl; return; } if (do_read_export) { LOG(LOG_INFO,"Exporting read information file to %s",readinfofile.c_str()); std::cout << _("Exporting list of read articles..."); std::cout.flush(); export_read_information(readinfofile); std::cout << _("done.") << std::endl; return; } if (execute_cmds) { execute_commands(argv, optind); utils::remove_fs_lock(lock_file); return; } // if the user wants to refresh on startup via configuration file, then do so, // but only if -r hasn't been supplied. if (!refresh_on_start && cfg.get_configvalue_as_bool("refresh-on-startup")) { refresh_on_start = true; } // hand over the important objects to the view v->set_config_container(&cfg); v->set_keymap(&keys); v->set_tags(tags); formaction::load_histories(searchfile, cmdlinefile); // run the view v->run(); unsigned int history_limit = cfg.get_configvalue_as_int("history-limit"); LOG(LOG_DEBUG, "controller::run: history-limit = %u", history_limit); formaction::save_histories(searchfile, cmdlinefile, history_limit); if (!silent) { std::cout << _("Cleaning up cache..."); std::cout.flush(); } try { scope_mutex feedslock(&feeds_mutex); rsscache->cleanup_cache(feeds); if (!silent) { std::cout << _("done.") << std::endl; } } catch (const dbexception& e) { LOG(LOG_USERERROR, "Cleaning up cache failed: %s", e.what()); if (!silent) { std::cout << _("failed: ") << e.what() << std::endl; } } utils::remove_fs_lock(lock_file); } void controller::update_feedlist() { scope_mutex feedslock(&feeds_mutex); v->set_feedlist(feeds); } void controller::update_visible_feeds() { scope_mutex feedslock(&feeds_mutex); v->update_visible_feeds(feeds); } void controller::catchup_all() { try { rsscache->catchup_all(); } catch (const dbexception& e) { v->show_error(utils::strprintf(_("Error: couldn't mark all feeds read: %s"), e.what())); return; } scope_mutex feedslock(&feeds_mutex); for (std::vector >::iterator it=feeds.begin();it!=feeds.end();++it) { scope_mutex lock(&(*it)->item_mutex); if ((*it)->items().size() > 0) { if (api) { api->mark_all_read((*it)->rssurl()); } for (std::vector >::iterator jt=(*it)->items().begin();jt!=(*it)->items().end();++jt) { (*jt)->set_unread_nowrite(false); } } } } void controller::mark_article_read(const std::string& guid, bool read) { if (api) { if (offline_mode) { if (dynamic_cast(api) != NULL) { LOG(LOG_DEBUG, "controller::mark_article_read: recording %s", guid.c_str()); record_google_replay(guid, read); } else { LOG(LOG_DEBUG, "not on googlereader_api"); } } else { api->mark_article_read(guid, read); } } } void controller::record_google_replay(const std::string& guid, bool read) { rsscache->record_google_replay(guid, read ? GOOGLE_MARK_READ : GOOGLE_MARK_UNREAD); } void controller::mark_all_read(unsigned int pos) { if (pos < feeds.size()) { scope_measure m("controller::mark_all_read"); scope_mutex feedslock(&feeds_mutex); std::tr1::shared_ptr feed = feeds[pos]; if (feed->rssurl().substr(0,6) == "query:") { rsscache->catchup_all(feed); } else { rsscache->catchup_all(feed->rssurl()); if (api) { api->mark_all_read(feed->rssurl()); } } m.stopover("after rsscache->catchup_all, before iteration over items"); scope_mutex lock(&feed->item_mutex); std::vector >& items = feed->items(); std::vector >::iterator begin = items.begin(), end = items.end(); if (items.size() > 0) { bool notify = items[0]->feedurl() != feed->rssurl(); LOG(LOG_DEBUG, "controller::mark_all_read: notify = %s", notify ? "yes" : "no"); for (std::vector >::iterator it=begin;it!=end;++it) { (*it)->set_unread_nowrite_notify(false, notify); } } } } void controller::reload(unsigned int pos, unsigned int max, bool unattended) { LOG(LOG_DEBUG, "controller::reload: pos = %u max = %u", pos, max); if (pos < feeds.size()) { std::tr1::shared_ptr oldfeed = feeds[pos]; std::string errmsg; if (!unattended) v->set_status(utils::strprintf(_("%sLoading %s..."), prepare_message(pos+1, max).c_str(), utils::censor_url(oldfeed->rssurl()).c_str())); bool ignore_dl = (cfg.get_configvalue("ignore-mode") == "download"); rss_parser parser(oldfeed->rssurl(), rsscache, &cfg, ignore_dl ? &ign : NULL, api); LOG(LOG_DEBUG, "controller::reload: created parser"); try { oldfeed->set_status(DURING_DOWNLOAD); std::tr1::shared_ptr newfeed = parser.parse(); if (newfeed->items().size() > 0) { scope_mutex feedslock(&feeds_mutex); save_feed(newfeed, pos); enqueue_items(newfeed); if (!unattended) { v->set_feedlist(feeds); } } else { LOG(LOG_DEBUG, "controller::reload: feed is empty"); } oldfeed->set_status(SUCCESS); v->set_status(""); } catch (const dbexception& e) { errmsg = utils::strprintf(_("Error while retrieving %s: %s"), utils::censor_url(oldfeed->rssurl()).c_str(), e.what()); } catch (const std::string& emsg) { errmsg = utils::strprintf(_("Error while retrieving %s: %s"), utils::censor_url(oldfeed->rssurl()).c_str(), emsg.c_str()); } catch (rsspp::exception& e) { errmsg = utils::strprintf(_("Error while retrieving %s: %s"), utils::censor_url(oldfeed->rssurl()).c_str(), e.what()); } if (errmsg != "") { oldfeed->set_status(DL_ERROR); v->set_status(errmsg); LOG(LOG_USERERROR, "%s", errmsg.c_str()); } } else { v->show_error(_("Error: invalid feed!")); } } std::tr1::shared_ptr controller::get_feed(unsigned int pos) { scope_mutex feedslock(&feeds_mutex); if (pos >= feeds.size()) { throw std::out_of_range(_("invalid feed index (bug)")); } std::tr1::shared_ptr feed = feeds[pos]; return feed; } void controller::reload_indexes(const std::vector& indexes, bool unattended) { scope_measure m1("controller::reload_indexes"); unsigned int unread_feeds, unread_articles; compute_unread_numbers(unread_feeds, unread_articles); unsigned long size; { scope_mutex feedslock(&feeds_mutex); size = feeds.size(); } for (std::vector::const_iterator it=indexes.begin();it!=indexes.end();++it) { this->reload(*it, size, unattended); } unsigned int unread_feeds2, unread_articles2; compute_unread_numbers(unread_feeds2, unread_articles2); bool notify_always = cfg.get_configvalue_as_bool("notify-always"); if (notify_always || unread_feeds2 != unread_feeds || unread_articles2 != unread_articles) { fmtstr_formatter fmt; fmt.register_fmt('f', utils::to_s(unread_feeds2)); fmt.register_fmt('n', utils::to_s(unread_articles2)); fmt.register_fmt('d', utils::to_s(unread_articles2 - unread_articles)); fmt.register_fmt('D', utils::to_s(unread_feeds2 - unread_feeds)); this->notify(fmt.do_format(cfg.get_configvalue("notify-format"))); } if (!unattended) v->set_status(""); } void controller::reload_range(unsigned int start, unsigned int end, unsigned int size, bool unattended) { for (unsigned int i=start;i<=end;i++) { LOG(LOG_DEBUG, "controller::reload_range: reloading feed #%u", i); this->reload(i, size, unattended); } } void controller::reload_all(bool unattended) { unsigned int unread_feeds, unread_articles; compute_unread_numbers(unread_feeds, unread_articles); unsigned int num_threads = cfg.get_configvalue_as_int("reload-threads"); time_t t1, t2, dt; unsigned int size; { scope_mutex feedlock(&feeds_mutex); for (std::vector >::iterator it=feeds.begin();it!=feeds.end();it++) { (*it)->reset_status(); } size = feeds.size(); } if (num_threads < 1) num_threads = 1; if (num_threads > size) { num_threads = size; } t1 = time(NULL); LOG(LOG_DEBUG,"controller::reload_all: starting with reload all..."); if (num_threads <= 1) { this->reload_range(0, size-1, size, unattended); } else { std::vector > partitions = utils::partition_indexes(0, size-1, num_threads); std::vector threads; LOG(LOG_DEBUG, "controller::reload_all: starting reload threads..."); for (unsigned int i=0;istart()); } LOG(LOG_DEBUG, "controller::reload_all: starting my own reload..."); this->reload_range(partitions[num_threads-1].first, partitions[num_threads-1].second, size, unattended); LOG(LOG_DEBUG, "controller::reload_all: joining other threads..."); for (std::vector::iterator it=threads.begin();it!=threads.end();it++) { ::pthread_join(*it, NULL); } } sort_feeds(); update_feedlist(); t2 = time(NULL); dt = t2 - t1; LOG(LOG_INFO, "controller::reload_all: reload took %d seconds", dt); unsigned int unread_feeds2, unread_articles2; compute_unread_numbers(unread_feeds2, unread_articles2); bool notify_always = cfg.get_configvalue_as_bool("notify-always"); if (notify_always || unread_feeds2 != unread_feeds || unread_articles2 != unread_articles) { fmtstr_formatter fmt; fmt.register_fmt('f', utils::to_s(unread_feeds2)); fmt.register_fmt('n', utils::to_s(unread_articles2)); fmt.register_fmt('d', utils::to_s(unread_articles2 - unread_articles)); fmt.register_fmt('D', utils::to_s(unread_feeds2 - unread_feeds)); this->notify(fmt.do_format(cfg.get_configvalue("notify-format"))); } } void controller::notify(const std::string& msg) { if (cfg.get_configvalue_as_bool("notify-screen")) { LOG(LOG_DEBUG, "controller:notify: notifying screen"); std::cout << "\033^" << msg << "\033\\"; std::cout.flush(); } if (cfg.get_configvalue_as_bool("notify-xterm")) { LOG(LOG_DEBUG, "controller:notify: notifying xterm"); std::cout << "\033]2;" << msg << "\033\\"; std::cout.flush(); } if (cfg.get_configvalue_as_bool("notify-beep")) { LOG(LOG_DEBUG, "controller:notify: notifying beep"); ::beep(); } if (cfg.get_configvalue("notify-program").length() > 0) { std::string prog = cfg.get_configvalue("notify-program"); LOG(LOG_DEBUG, "controller:notify: notifying external program `%s'", prog.c_str()); utils::run_command(prog, msg); } } void controller::compute_unread_numbers(unsigned int& unread_feeds, unsigned int& unread_articles) { unread_feeds = 0; unread_articles = 0; for (std::vector >::iterator it=feeds.begin();it!=feeds.end();++it) { unsigned int items = (*it)->unread_item_count(); if (items > 0) { ++unread_feeds; unread_articles += items; } } } bool controller::trylock_reload_mutex() { if (reload_mutex.trylock()) { LOG(LOG_DEBUG, "controller::trylock_reload_mutex succeeded"); return true; } LOG(LOG_DEBUG, "controller::trylock_reload_mutex failed"); return false; } void controller::start_reload_all_thread(std::vector * indexes) { LOG(LOG_INFO,"starting reload all thread"); thread * dlt = new downloadthread(this, indexes); dlt->start(); } void controller::version_information(const char * argv0, unsigned int level) { if (level<=1) { std::cout << PROGRAM_NAME << " " << PROGRAM_VERSION << " - " << PROGRAM_URL << std::endl; std::cout << "Copyright (C) 2006-2010 Andreas Krennmair" << std::endl << std::endl; std::cout << _("newsbeuter is free software and licensed under the MIT/X Consortium License.") << std::endl; std::cout << utils::strprintf(_("Type `%s -vv' for more information."), argv0) << std::endl << std::endl; struct utsname xuts; uname(&xuts); std::cout << "Compilation date/time: " << __DATE__ << " " << __TIME__ << std::endl; std::cout << "System: " << xuts.sysname << " " << xuts.release << " (" << xuts.machine << ")" << std::endl; #if defined(__GNUC__) && defined(__VERSION__) std::cout << "Compiler: g++ " << __VERSION__ << std::endl; #endif std::cout << "ncurses: " << curses_version() << " (compiled with " << NCURSES_VERSION << ")" << std::endl; std::cout << "libcurl: " << curl_version() << " (compiled with " << LIBCURL_VERSION << ")" << std::endl; std::cout << "SQLite: " << sqlite3_libversion() << " (compiled with " << SQLITE_VERSION << ")" << std::endl; std::cout << "libxml2: compiled with " << LIBXML_DOTTED_VERSION << std::endl << std::endl; } else { std::cout << LICENSE_str << std::endl; } ::exit(EXIT_SUCCESS); } static unsigned int gentabs(const std::string& str) { int tabcount = 2 - ((3 + utils::strwidth(str)) / 8); if (tabcount <= 0) { tabcount = 1; } return tabcount; } void controller::usage(char * argv0) { std::cout << utils::strprintf(_("%s %s\nusage: %s [-i |-e] [-u ] [-c ] [-x ...] [-h]\n"), PROGRAM_NAME, PROGRAM_VERSION, argv0); struct { char arg; const char * params; const char * desc; } args[] = { { 'e', "", _("export OPML feed to stdout") }, { 'r', "", _("refresh feeds on start") }, { 'i', _(""), _("import OPML file") }, { 'u', _(""), _("read RSS feed URLs from ") }, { 'c', _(""), _("use as cache file") }, { 'C', _(""), _("read configuration from ") }, { 'X', "", _("clean up cache thoroughly") }, { 'x', _("..."), _("execute list of commands") }, { 'o', "", _("activate offline mode (only applies to Google Reader synchronization mode)") }, { 'q', "", _("quiet startup") }, { 'v', "", _("get version information") }, { 'l', _(""), _("write a log with a certain loglevel (valid values: 1 to 6)") }, { 'd', _(""), _("use as output log file") }, { 'E', _(""), _("export list of read articles to ") }, { 'I', _(""), _("import list of read articles from ") }, { 'h', "", _("this help") }, { '\0', NULL, NULL } }; for (unsigned int i=0;args[i].arg != '\0';i++) { unsigned int tabs = gentabs(args[i].params); std::cout << "\t-" << args[i].arg << " " << args[i].params; for (unsigned int j=0;jchildren; node != NULL; node = node->next) { if (strcmp((const char *)node->name, "body")==0) { LOG(LOG_DEBUG, "import_opml: found body"); rec_find_rss_outlines(node->children, ""); urlcfg->write_config(); } } xmlFreeDoc(doc); std::cout << utils::strprintf(_("Import of %s finished."), filename) << std::endl; } void controller::export_opml() { xmlDocPtr root = xmlNewDoc((const xmlChar *)"1.0"); xmlNodePtr opml_node = xmlNewDocNode(root, NULL, (const xmlChar *)"opml", NULL); xmlSetProp(opml_node, (const xmlChar *)"version", (const xmlChar *)"1.0"); xmlDocSetRootElement(root, opml_node); xmlNodePtr head = xmlNewTextChild(opml_node, NULL, (const xmlChar *)"head", NULL); xmlNewTextChild(head, NULL, (const xmlChar *)"title", (const xmlChar *)PROGRAM_NAME " - Exported Feeds"); xmlNodePtr body = xmlNewTextChild(opml_node, NULL, (const xmlChar *)"body", NULL); for (std::vector >::iterator it=feeds.begin(); it != feeds.end(); ++it) { if (!utils::is_special_url((*it)->rssurl())) { std::string rssurl = (*it)->rssurl(); std::string link = (*it)->link(); std::string title = (*it)->title(); xmlNodePtr outline = xmlNewTextChild(body, NULL, (const xmlChar *)"outline", NULL); xmlSetProp(outline, (const xmlChar *)"type", (const xmlChar *)"rss"); xmlSetProp(outline, (const xmlChar *)"xmlUrl", (const xmlChar *)rssurl.c_str()); xmlSetProp(outline, (const xmlChar *)"htmlUrl", (const xmlChar *)link.c_str()); xmlSetProp(outline, (const xmlChar *)"title", (const xmlChar *)title.c_str()); } } xmlSaveCtxtPtr savectx = xmlSaveToFd(1, NULL, 1); xmlSaveDoc(savectx, root); xmlSaveClose(savectx); xmlFreeNode(opml_node); } void controller::rec_find_rss_outlines(xmlNode * node, std::string tag) { while (node) { std::string newtag = tag; if (strcmp((const char *)node->name, "outline")==0) { char * url = (char *)xmlGetProp(node, (const xmlChar *)"xmlUrl"); if (!url) { url = (char *)xmlGetProp(node, (const xmlChar *)"url"); } if (url) { LOG(LOG_DEBUG,"OPML import: found RSS outline with url = %s",url); std::string nurl = std::string(url); // Liferea uses a pipe to signal feeds read from the output of // a program in its OPMLs. Convert them to our syntax. if (*url == '|') { nurl = utils::strprintf("exec:%s", url+1); LOG(LOG_DEBUG,"OPML import: liferea-style url %s converted to %s", url, nurl.c_str()); } // Handle OPML filters. char * filtercmd = (char *)xmlGetProp(node, (const xmlChar *)"filtercmd"); if (filtercmd) { LOG(LOG_DEBUG,"OPML import: adding filter command %s to url %s", filtercmd, nurl.c_str()); nurl.insert(0, utils::strprintf("filter:%s:", filtercmd)); xmlFree(filtercmd); } xmlFree(url); // Filters and scripts may have arguments, so, quote them when needed. url = (char*) xmlStrdup((const xmlChar*)utils::quote_if_necessary(nurl).c_str()); assert(url); bool found = false; LOG(LOG_DEBUG, "OPML import: size = %u", urlcfg->get_urls().size()); if (urlcfg->get_urls().size() > 0) { for (std::vector::iterator it = urlcfg->get_urls().begin(); it != urlcfg->get_urls().end(); ++it) { if (*it == url) { found = true; } } } if (!found) { LOG(LOG_DEBUG,"OPML import: added url = %s",url); urlcfg->get_urls().push_back(std::string(url)); if (tag.length() > 0) { LOG(LOG_DEBUG, "OPML import: appending tag %s to url %s", tag.c_str(), url); urlcfg->get_tags(url).push_back(tag); } } else { LOG(LOG_DEBUG,"OPML import: url = %s is already in list",url); } xmlFree(url); } else { char * text = (char *)xmlGetProp(node, (const xmlChar *)"text"); if (!text) text = (char *)xmlGetProp(node, (const xmlChar *)"title"); if (text) { if (newtag.length() > 0) { newtag.append("/"); } newtag.append(text); xmlFree(text); } } } rec_find_rss_outlines(node->children, newtag); node = node->next; } } std::vector > controller::search_for_items(const std::string& query, const std::string& feedurl) { std::vector > items = rsscache->search_for_items(query, feedurl); LOG(LOG_DEBUG, "controller::search_for_items: setting feed pointers"); for (std::vector >::iterator it=items.begin();it!=items.end();++it) { (*it)->set_feedptr(get_feed_by_url((*it)->feedurl())); } return items; } std::tr1::shared_ptr controller::get_feed_by_url(const std::string& feedurl) { for (std::vector >::iterator it=feeds.begin();it!=feeds.end();++it) { if (feedurl == (*it)->rssurl()) return *it; } return std::tr1::shared_ptr(); } bool controller::is_valid_podcast_type(const std::string& /* mimetype */) { return true; } void controller::enqueue_url(const std::string& url, std::tr1::shared_ptr feed) { bool url_found = false; std::fstream f; f.open(queue_file.c_str(), std::fstream::in); if (f.is_open()) { do { std::string line; getline(f, line); if (!f.eof() && line.length() > 0) { std::vector fields = utils::tokenize_quoted(line); if (fields.size() > 0 && fields[0] == url) { url_found = true; break; } } } while (!f.eof()); f.close(); } if (!url_found) { f.open(queue_file.c_str(), std::fstream::app | std::fstream::out); std::string filename = generate_enqueue_filename(url, feed); f << url << " " << stfl::quote(filename) << std::endl; f.close(); } } void controller::reload_urls_file() { urlcfg->reload(); std::vector > new_feeds; unsigned int i = 0; for (std::vector::const_iterator it=urlcfg->get_urls().begin();it!=urlcfg->get_urls().end();++it,++i) { bool found = false; for (std::vector >::iterator jt=feeds.begin();jt!=feeds.end();++jt) { if (*it == (*jt)->rssurl()) { found = true; (*jt)->set_tags(urlcfg->get_tags(*it)); (*jt)->set_order(i); new_feeds.push_back(*jt); break; } } if (!found) { std::tr1::shared_ptr new_feed(new rss_feed(rsscache)); try { new_feed->set_rssurl(*it); } catch (const std::string& str) { LOG(LOG_USERERROR, "ignored invalid RSS URL '%s'", it->c_str()); continue; } new_feed->set_tags(urlcfg->get_tags(*it)); new_feed->set_order(i); try { bool ignore_disp = (cfg.get_configvalue("ignore-mode") == "display"); rsscache->internalize_rssfeed(new_feed, ignore_disp ? &ign : NULL); } catch(const dbexception& e) { LOG(LOG_ERROR, "controller::reload_urls_file: caught exception: %s", e.what()); throw e; } new_feeds.push_back(new_feed); } } { scope_mutex feedslock(&feeds_mutex); feeds = new_feeds; } sort_feeds(); update_feedlist(); } void controller::edit_urls_file() { const char * editor; editor = getenv("VISUAL"); if (!editor) editor = getenv("EDITOR"); if (!editor) editor = "vi"; std::string cmdline = utils::strprintf("%s \"%s\"", editor, utils::replace_all(url_file,"\"","\\\"").c_str()); v->push_empty_formaction(); stfl::reset(); LOG(LOG_DEBUG, "controller::edit_urls_file: running `%s'", cmdline.c_str()); ::system(cmdline.c_str()); v->pop_current_formaction(); reload_urls_file(); } std::string controller::bookmark(const std::string& url, const std::string& title, const std::string& description) { std::string bookmark_cmd = cfg.get_configvalue("bookmark-cmd"); bool is_interactive = cfg.get_configvalue_as_bool("bookmark-interactive"); if (bookmark_cmd.length() > 0) { std::string cmdline = utils::strprintf("%s '%s' %s %s", bookmark_cmd.c_str(), utils::replace_all(url,"'", "%27").c_str(), stfl::quote(title).c_str(), stfl::quote(description).c_str()); LOG(LOG_DEBUG, "controller::bookmark: cmd = %s", cmdline.c_str()); if (is_interactive) { v->push_empty_formaction(); stfl::reset(); ::system(cmdline.c_str()); v->pop_current_formaction(); return ""; } else { char * my_argv[4]; my_argv[0] = const_cast("/bin/sh"); my_argv[1] = const_cast("-c"); my_argv[2] = const_cast(cmdline.c_str()); my_argv[3] = NULL; return utils::run_program(my_argv, ""); } } else { return _("bookmarking support is not configured. Please set the configuration variable `bookmark-cmd' accordingly."); } } void controller::execute_commands(char ** argv, unsigned int i) { if (v->formaction_stack_size() > 0) v->pop_current_formaction(); for (;argv[i];++i) { LOG(LOG_DEBUG, "controller::execute_commands: executing `%s'", argv[i]); std::string cmd(argv[i]); if (cmd == "reload") { reload_all(true); } else if (cmd == "print-unread") { std::cout << utils::strprintf(_("%u unread articles"), rsscache->get_unread_count()) << std::endl; } } } std::string controller::write_temporary_item(std::tr1::shared_ptr item) { char filename[1024]; snprintf(filename, sizeof(filename), "/tmp/newsbeuter-article.XXXXXX"); int fd = mkstemp(filename); if (fd != -1) { write_item(item, filename); close(fd); return std::string(filename); } else { return ""; } } void controller::write_item(std::tr1::shared_ptr item, const std::string& filename) { std::fstream f; f.open(filename.c_str(),std::fstream::out); if (!f.is_open()) throw exception(errno); write_item(item, f); } void controller::write_item(std::tr1::shared_ptr item, std::ostream& ostr) { std::vector lines; std::vector links; // not used std::string title(_("Title: ")); title.append(item->title()); lines.push_back(title); std::string author(_("Author: ")); author.append(item->author()); lines.push_back(author); std::string date(_("Date: ")); date.append(item->pubDate()); lines.push_back(date); std::string link(_("Link: ")); link.append(item->link()); lines.push_back(link); if (item->enclosure_url() != "") { std::string dlurl(_("Podcast Download URL: ")); dlurl.append(item->enclosure_url()); lines.push_back(dlurl); } lines.push_back(std::string("")); unsigned int width = cfg.get_configvalue_as_int("text-width"); if (width == 0) width = 80; htmlrenderer rnd(width, true); rnd.render(item->description(), lines, links, item->feedurl()); for (std::vector::iterator it=lines.begin();it!=lines.end();++it) { ostr << *it << std::endl; } } void controller::mark_deleted(const std::string& guid, bool b) { rsscache->mark_item_deleted(guid, b); } std::string controller::prepare_message(unsigned int pos, unsigned int max) { if (max > 0) { return utils::strprintf("(%u/%u) ", pos, max); } return ""; } void controller::save_feed(std::tr1::shared_ptr feed, unsigned int pos) { if (!feed->is_empty()) { LOG(LOG_DEBUG, "controller::reload: feed is nonempty, saving"); rsscache->externalize_rssfeed(feed, ign.matches_resetunread(feed->rssurl())); LOG(LOG_DEBUG, "controller::reload: after externalize_rssfeed"); bool ignore_disp = (cfg.get_configvalue("ignore-mode") == "display"); rsscache->internalize_rssfeed(feed, ignore_disp ? &ign : NULL); LOG(LOG_DEBUG, "controller::reload: after internalize_rssfeed"); feed->set_tags(urlcfg->get_tags(feed->rssurl())); { unsigned int order = feeds[pos]->get_order(); scope_mutex itemlock(&feeds[pos]->item_mutex); feeds[pos]->items().clear(); feed->set_order(order); } feeds[pos] = feed; v->notify_itemlist_change(feeds[pos]); } else { LOG(LOG_DEBUG, "controller::reload: feed is empty, not saving"); } } void controller::enqueue_items(std::tr1::shared_ptr feed) { if (!cfg.get_configvalue_as_bool("podcast-auto-enqueue")) return; scope_mutex lock(&feed->item_mutex); for (std::vector >::iterator it=feed->items().begin();it!=feed->items().end();++it) { if (!(*it)->enqueued() && (*it)->enclosure_url().length() > 0) { LOG(LOG_DEBUG, "controller::reload: enclosure_url = `%s' enclosure_type = `%s'", (*it)->enclosure_url().c_str(), (*it)->enclosure_type().c_str()); if (is_valid_podcast_type((*it)->enclosure_type()) && utils::is_http_url((*it)->enclosure_url())) { LOG(LOG_INFO, "controller::reload: enqueuing `%s'", (*it)->enclosure_url().c_str()); enqueue_url((*it)->enclosure_url(), feed); (*it)->set_enqueued(true); rsscache->update_rssitem_unread_and_enqueued(*it, feed->rssurl()); } } } } std::string controller::generate_enqueue_filename(const std::string& url, std::tr1::shared_ptr feed) { std::string dlformat = cfg.get_configvalue("download-path"); if (dlformat[dlformat.length()-1] != NEWSBEUTER_PATH_SEP[0]) dlformat.append(NEWSBEUTER_PATH_SEP); fmtstr_formatter fmt; fmt.register_fmt('n', feed->title()); fmt.register_fmt('h', get_hostname_from_url(url)); std::string dlpath = fmt.do_format(dlformat); char buf[2048]; snprintf(buf, sizeof(buf), "%s", url.c_str()); char * base = basename(buf); if (!base || strlen(base) == 0) { char lbuf[128]; time_t t = time(NULL); strftime(lbuf, sizeof(lbuf), "%Y-%b-%d-%H%M%S.unknown", localtime(&t)); dlpath.append(lbuf); } else { dlpath.append(base); } return dlpath; } std::string controller::get_hostname_from_url(const std::string& url) { xmlURIPtr uri = xmlParseURI(url.c_str()); std::string hostname; if (uri) { hostname = uri->server; xmlFreeURI(uri); } return hostname; } void controller::import_read_information(const std::string& readinfofile) { std::vector guids; std::ifstream f(readinfofile.c_str()); std::string line; getline(f,line); if (!f.is_open()) { return; } while (f.is_open() && !f.eof()) { guids.push_back(line); getline(f, line); } rsscache->mark_items_read_by_guid(guids); } void controller::export_read_information(const std::string& readinfofile) { std::vector guids = rsscache->get_read_item_guids(); std::fstream f; f.open(readinfofile.c_str(), std::fstream::out); if (f.is_open()) { for (std::vector::iterator it=guids.begin();it!=guids.end();it++) { f << *it << std::endl; } } } struct sort_feeds_by_firsttag : public std::binary_function, std::tr1::shared_ptr, bool> { sort_feeds_by_firsttag() { } bool operator()(std::tr1::shared_ptr a, std::tr1::shared_ptr b) { if (a->get_firsttag().length() == 0 || b->get_firsttag().length() == 0) { return a->get_firsttag().length() > b->get_firsttag().length(); } return strcasecmp(a->get_firsttag().c_str(), b->get_firsttag().c_str()) < 0; } }; struct sort_feeds_by_title : public std::binary_function, std::tr1::shared_ptr, bool> { sort_feeds_by_title() { } bool operator()(std::tr1::shared_ptr a, std::tr1::shared_ptr b) { return strcasecmp(a->title().c_str(), b->title().c_str()) < 0; } }; struct sort_feeds_by_articles : public std::binary_function, std::tr1::shared_ptr, bool> { sort_feeds_by_articles() { } bool operator()(std::tr1::shared_ptr a, std::tr1::shared_ptr b) { return a->total_item_count() < b->total_item_count(); } }; struct sort_feeds_by_unread_articles : public std::binary_function, std::tr1::shared_ptr, bool> { sort_feeds_by_unread_articles() { } bool operator()(std::tr1::shared_ptr a, std::tr1::shared_ptr b) { return a->unread_item_count() < b->unread_item_count(); } }; struct sort_feeds_by_order : public std::binary_function, std::tr1::shared_ptr, bool> { sort_feeds_by_order() { } bool operator()(std::tr1::shared_ptr a, std::tr1::shared_ptr b) { return a->get_order() < b->get_order(); } }; void controller::sort_feeds() { scope_mutex feedslock(&feeds_mutex); std::vector sortmethod_info = utils::tokenize(cfg.get_configvalue("feed-sort-order"), "-"); std::string sortmethod = sortmethod_info[0]; std::string direction = "desc"; if (sortmethod_info.size() > 1) direction = sortmethod_info[1]; if (sortmethod == "none") { std::stable_sort(feeds.begin(), feeds.end(), sort_feeds_by_order()); } else if (sortmethod == "firsttag") { std::stable_sort(feeds.begin(), feeds.end(), sort_feeds_by_firsttag()); } else if (sortmethod == "title") { std::stable_sort(feeds.begin(), feeds.end(), sort_feeds_by_title()); } else if (sortmethod == "articlecount") { std::stable_sort(feeds.begin(), feeds.end(), sort_feeds_by_articles()); } else if (sortmethod == "unreadarticlecount") { std::stable_sort(feeds.begin(), feeds.end(), sort_feeds_by_unread_articles()); } if (direction == "asc") { std::reverse(feeds.begin(), feeds.end()); } } void controller::update_config() { v->set_regexmanager(&rxman); v->update_bindings(); if (colorman.colors_loaded()) { v->set_colors(colorman.get_fgcolors(), colorman.get_bgcolors(), colorman.get_attributes()); v->apply_colors_to_all_formactions(); } if (cfg.get_configvalue("error-log").length() > 0) { GetLogger().set_errorlogfile(cfg.get_configvalue("error-log").c_str()); } } void controller::load_configfile(const std::string& filename) { if (cfgparser.parse(filename, true)) { update_config(); } else { v->show_error(utils::strprintf(_("Error: couldn't open configuration file `%s'!"), filename.c_str())); } } void controller::dump_config(const std::string& filename) { std::vector configlines; cfg.dump_config(configlines); if (v) { v->get_keys()->dump_config(configlines); } ign.dump_config(configlines); filters.dump_config(configlines); colorman.dump_config(configlines); rxman.dump_config(configlines); std::fstream f; f.open(filename.c_str(), std::fstream::out); if (f.is_open()) { for (std::vector::iterator it=configlines.begin();it!=configlines.end();it++) { f << *it << std::endl; } } } unsigned int controller::get_pos_of_next_unread(unsigned int pos) { scope_mutex feedslock(&feeds_mutex); for (pos++;pos < feeds.size();pos++) { if (feeds[pos]->unread_item_count() > 0) break; } return pos; } void controller::update_flags(std::tr1::shared_ptr item) { if (api) { api->update_article_flags(item->oldflags(), item->flags(), item->guid()); } item->update_flags(); } std::vector > controller::get_all_feeds() { std::vector > tmpfeeds; { scope_mutex feedslock(&feeds_mutex); tmpfeeds = feeds; } return tmpfeeds; } std::vector > controller::get_all_feeds_unlocked() { return feeds; } }