#!/usr/bin/env php
<?php
- set_include_path(get_include_path() . PATH_SEPARATOR .
- dirname(__FILE__) . "/include");
+ set_include_path(dirname(__FILE__) ."/include" . PATH_SEPARATOR .
+ get_include_path());
define('DISABLE_SESSIONS', true);
chdir(dirname(__FILE__));
+ require_once "autoload.php";
require_once "functions.php";
- require_once "rssfuncs.php";
- require_once "sanity_check.php";
require_once "config.php";
+ require_once "sanity_check.php";
require_once "db.php";
require_once "db-prefs.php";
- require_once "update_self.php";
- if (!defined('STDIN')) {
+ if (!defined('PHP_EXECUTABLE'))
+ define('PHP_EXECUTABLE', '/usr/bin/php');
+
+ $pdo = Db::pdo();
+
+ init_plugins();
+
+ $longopts = array("feeds",
+ "feedbrowser",
+ "daemon",
+ "daemon-loop",
+ "task:",
+ "cleanup-tags",
+ "quiet",
+ "log:",
+ "indexes",
+ "pidlock:",
+ "update-schema",
+ "convert-filters",
+ "force-update",
+ "gen-search-idx",
+ "list-plugins",
+ "debug-feed:",
+ "force-refetch",
+ "force-rehash",
+ "decrypt-feeds",
+ "help");
+
+ foreach (PluginHost::getInstance()->get_commands() as $command => $data) {
+ array_push($longopts, $command . $data["suffix"]);
+ }
+
+ $options = getopt("", $longopts);
+
+ if (!is_array($options)) {
+ die("error: getopt() failed. ".
+ "Most probably you are using PHP CGI to run this script ".
+ "instead of required PHP CLI. Check tt-rss wiki page on updating feeds for ".
+ "additional information.\n");
+ }
+
+ if (count($options) == 0 && !defined('STDIN')) {
?> <html>
<head>
- <title>Update Tool</title>
+ <title>Tiny Tiny RSS data update script.</title>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
- <link rel="stylesheet" type="text/css" href="utility.css">
</head>
<body>
- <div class="floatingLogo"><img src="images/logo_wide.png"></div>
- <h1><?php echo __("Update") ?></h1>
+ <div class="floatingLogo"><img src="images/logo_small.png"></div>
+ <h1><?php echo __("Tiny Tiny RSS data update script.") ?></h1>
- <?php print_error("Please run this script from the command line."); ?>
+ <?php print_error("Please run this script from the command line. Use option \"--help\" to display command help if this error is displayed erroneously."); ?>
</body></html>
<?php
exit;
}
- if (!defined('PHP_EXECUTABLE'))
- define('PHP_EXECUTABLE', '/usr/bin/php');
-
- $op = $argv;
-
- if (count($argv) == 1 || in_array("-help", $op) ) {
+ if (count($options) == 0 || isset($options["help"]) ) {
print "Tiny Tiny RSS data update script.\n\n";
print "Options:\n";
- print " -feeds - update feeds\n";
- print " -feedbrowser - update feedbrowser\n";
- print " -daemon - start single-process update daemon\n";
- print " -cleanup-tags - perform tags table maintenance\n";
- print " -get-feeds - receive popular feeds from linked instances\n";
- print " -import USER FILE - import articles from XML\n";
- print " -update-self - update tt-rss installation to latest version\n";
- print " -quiet - don't show messages\n";
- print " -indexes - recreate missing schema indexes\n";
- print " -convert-filters - convert type1 filters to type2\n";
- print " -help - show this help\n";
+ print " --feeds - update feeds\n";
+ print " --feedbrowser - update feedbrowser\n";
+ print " --daemon - start single-process update daemon\n";
+ print " --task N - create lockfile using this task id\n";
+ print " --cleanup-tags - perform tags table maintenance\n";
+ print " --quiet - don't output messages to stdout\n";
+ print " --log FILE - log messages to FILE\n";
+ print " --indexes - recreate missing schema indexes\n";
+ print " --update-schema - update database schema\n";
+ print " --gen-search-idx - generate basic PostgreSQL fulltext search index\n";
+ print " --convert-filters - convert type1 filters to type2\n";
+ print " --force-update - force update of all feeds\n";
+ print " --list-plugins - list all available plugins\n";
+ print " --debug-feed N - perform debug update of feed N\n";
+ print " --force-refetch - debug update: force refetch feed data\n";
+ print " --force-rehash - debug update: force rehash articles\n";
+ print " --decrypt-feeds - decrypt feed passwords\n";
+ print " --help - show this help\n";
+ print "Plugin options:\n";
+
+ foreach (PluginHost::getInstance()->get_commands() as $command => $data) {
+ $args = $data['arghelp'];
+ printf(" --%-19s - %s\n", "$command $args", $data["description"]);
+ }
+
return;
}
- define('QUIET', in_array("-quiet", $op));
+ if (!isset($options['daemon'])) {
+ require_once "errorhandler.php";
+ }
- if (!in_array("-daemon", $op)) {
+ if (!isset($options['update-schema'])) {
+ $schema_version = get_schema_version();
+
+ if ($schema_version != SCHEMA_VERSION) {
+ die("Schema version is wrong, please upgrade the database.\n");
+ }
+ }
+
+ define('QUIET', isset($options['quiet']));
+
+ if (isset($options["log"])) {
+ _debug("Logging to " . $options["log"]);
+ define('LOGFILE', $options["log"]);
+ }
+
+ if (!isset($options["daemon"])) {
$lock_filename = "update.lock";
} else {
$lock_filename = "update_daemon.lock";
}
+ if (isset($options["task"])) {
+ _debug("Using task id " . $options["task"]);
+ $lock_filename = $lock_filename . "-task_" . $options["task"];
+ }
+
+ if (isset($options["pidlock"])) {
+ $my_pid = $options["pidlock"];
+ $lock_filename = "update_daemon-$my_pid.lock";
+
+ }
+
+ _debug("Lock: $lock_filename");
+
$lock_handle = make_lockfile($lock_filename);
$must_exit = false;
+ if (isset($options["task"]) && isset($options["pidlock"])) {
+ $waits = $options["task"] * 5;
+ _debug("Waiting before update ($waits)");
+ sleep($waits);
+ }
+
// Try to lock a file in order to avoid concurrent update.
if (!$lock_handle) {
die("error: Can't create lockfile ($lock_filename). ".
"Maybe another update process is already running.\n");
}
- // Create a database connection.
- $link = db_connect(DB_HOST, DB_USER, DB_PASS, DB_NAME);
-
- init_connection($link);
+ if (isset($options["force-update"])) {
+ _debug("marking all feeds as needing update...");
- if (in_array("-feeds", $op)) {
- // Update all feeds needing a update.
- update_daemon_common($link);
-
- // Update feedbrowser
- $count = update_feedbrowser_cache($link);
- _debug("Feedbrowser updated, $count feeds processed.");
-
- // Purge orphans and cleanup tags
- purge_orphans($link, true);
+ $pdo->query( "UPDATE ttrss_feeds SET
+ last_update_started = '1970-01-01', last_updated = '1970-01-01'");
+ }
- $rc = cleanup_tags($link, 14, 50000);
- _debug("Cleaned $rc cached tags.");
+ if (isset($options["feeds"])) {
+ RSSUtils::update_daemon_common();
+ RSSUtils::housekeeping_common(true);
- get_linked_feeds($link);
+ PluginHost::getInstance()->run_hooks(PluginHost::HOOK_UPDATE_TASK, "hook_update_task", $op);
}
- if (in_array("-feedbrowser", $op)) {
- $count = update_feedbrowser_cache($link);
+ if (isset($options["feedbrowser"])) {
+ $count = RSSUtils::update_feedbrowser_cache();
print "Finished, $count feeds processed.\n";
}
- if (in_array("-daemon", $op)) {
- $op = array_diff($op, array("-daemon"));
+ if (isset($options["daemon"])) {
while (true) {
- passthru(PHP_EXECUTABLE . " " . implode(' ', $op) . " -daemon-loop");
- _debug("Sleeping for " . DAEMON_SLEEP_INTERVAL . " seconds...");
- sleep(DAEMON_SLEEP_INTERVAL);
- }
- }
+ $quiet = (isset($options["quiet"])) ? "--quiet" : "";
+ $log = isset($options['log']) ? '--log '.$options['log'] : '';
- if (in_array("-daemon-loop", $op)) {
- if (!make_stampfile('update_daemon.stamp')) {
- die("error: unable to create stampfile\n");
- }
+ passthru(PHP_EXECUTABLE . " " . $argv[0] ." --daemon-loop $quiet $log");
- // Call to the feed batch update function
- // or regenerate feedbrowser cache
-
- if (rand(0,100) > 30) {
- update_daemon_common($link);
- } else {
- $count = update_feedbrowser_cache($link);
- _debug("Feedbrowser updated, $count feeds processed.");
+ // let's enforce a minimum spawn interval as to not forkbomb the host
+ $spawn_interval = max(60, DAEMON_SLEEP_INTERVAL);
- purge_orphans($link, true);
-
- $rc = cleanup_tags($link, 14, 50000);
-
- _debug("Cleaned $rc cached tags.");
-
- get_linked_feeds($link);
+ _debug("Sleeping for $spawn_interval seconds...");
+ sleep($spawn_interval);
}
-
}
- if (in_array("-cleanup-tags", $op)) {
- $rc = cleanup_tags($link, 14, 50000);
- _debug("$rc tags deleted.\n");
- }
-
- if (in_array("-get-feeds", $op)) {
- get_linked_feeds($link);
- }
-
- if (in_array("-import",$op)) {
- $username = $argv[count($argv) - 2];
- $filename = $argv[count($argv) - 1];
-
- if (!$username) {
- print "error: please specify username.\n";
- return;
- }
-
- if (!is_file($filename)) {
- print "error: input filename ($filename) doesn't exist.\n";
- return;
+ if (isset($options["daemon-loop"])) {
+ if (!make_stampfile('update_daemon.stamp')) {
+ _debug("warning: unable to create stampfile\n");
}
- _debug("importing $filename for user $username...\n");
-
- $result = db_query($link, "SELECT id FROM ttrss_users WHERE login = '$username'");
+ RSSUtils::update_daemon_common(isset($options["pidlock"]) ? 50 : DAEMON_FEED_LIMIT);
- if (db_num_rows($result) == 0) {
- print "error: could not find user $username.\n";
- return;
- }
+ if (!isset($options["pidlock"]) || $options["task"] == 0)
+ RSSUtils::housekeeping_common(true);
- $owner_uid = db_fetch_result($result, 0, "id");
-
- perform_data_import($link, $filename, $owner_uid);
+ PluginHost::getInstance()->run_hooks(PluginHost::HOOK_UPDATE_TASK, "hook_update_task", $op);
+ }
+ if (isset($options["cleanup-tags"])) {
+ $rc = cleanup_tags( 14, 50000);
+ _debug("$rc tags deleted.\n");
}
- if (in_array("-indexes", $op)) {
+ if (isset($options["indexes"])) {
_debug("PLEASE BACKUP YOUR DATABASE BEFORE PROCEEDING!");
_debug("Type 'yes' to continue.");
_debug("clearing existing indexes...");
if (DB_TYPE == "pgsql") {
- $result = db_query($link, "SELECT relname FROM
+ $sth = $pdo->query( "SELECT relname FROM
pg_catalog.pg_class WHERE relname LIKE 'ttrss_%'
AND relname NOT LIKE '%_pkey'
AND relkind = 'i'");
} else {
- $result = db_query($link, "SELECT index_name,table_name FROM
+ $sth = $pdo->query( "SELECT index_name,table_name FROM
information_schema.statistics WHERE index_name LIKE 'ttrss_%'");
}
- while ($line = db_fetch_assoc($result)) {
+ while ($line = $sth->fetch()) {
if (DB_TYPE == "pgsql") {
$statement = "DROP INDEX " . $line["relname"];
_debug($statement);
$line['table_name']." DROP INDEX ".$line['index_name'];
_debug($statement);
}
- db_query($link, $statement, false);
+ $pdo->query($statement);
}
_debug("reading indexes from schema for: " . DB_TYPE);
$statement = "CREATE INDEX $index ON $table";
_debug($statement);
- db_query($link, $statement);
+ $pdo->query($statement);
}
}
fclose($fp);
_debug("all done.");
}
- if (in_array("-update-self", $op)) {
- _debug("Warning: self-updating is experimental. Use at your own risk.");
- _debug("Please backup your tt-rss directory before continuing. Your database will not be modified.");
- _debug("Type 'yes' to continue.");
-
- if (read_stdin() != 'yes')
- exit;
-
- update_self($link, in_array("-force", $op));
- }
-
- if (in_array("-convert-filters", $op)) {
+ if (isset($options["convert-filters"])) {
_debug("WARNING: this will remove all existing type2 filters.");
_debug("Type 'yes' to continue.");
_debug("converting filters...");
- db_query($link, "DELETE FROM ttrss_filters2");
+ $pdo->query("DELETE FROM ttrss_filters2");
- $result = db_query($link, "SELECT * FROM ttrss_filters ORDER BY id");
+ $res = $pdo->query("SELECT * FROM ttrss_filters ORDER BY id");
- while ($line = db_fetch_assoc($result)) {
+ while ($line = $res->fetch()) {
$owner_uid = $line["owner_uid"];
// date filters are removed
$_REQUEST = $filter;
$_SESSION["uid"] = $owner_uid;
- $filters = new Pref_Filters($link, $_REQUEST);
+ $filters = new Pref_Filters($_REQUEST);
$filters->add();
}
}
}
- db_close($link);
+ if (isset($options["update-schema"])) {
+ _debug("checking for updates (" . DB_TYPE . ")...");
+
+ $updater = new DbUpdater(Db::pdo(), DB_TYPE, SCHEMA_VERSION);
+
+ if ($updater->isUpdateRequired()) {
+ _debug("schema update required, version " . $updater->getSchemaVersion() . " to " . SCHEMA_VERSION);
+ _debug("WARNING: please backup your database before continuing.");
+ _debug("Type 'yes' to continue.");
+
+ if (read_stdin() != 'yes')
+ exit;
+
+ for ($i = $updater->getSchemaVersion() + 1; $i <= SCHEMA_VERSION; $i++) {
+ _debug("performing update up to version $i...");
+
+ $result = $updater->performUpdateTo($i, false);
+
+ _debug($result ? "OK!" : "FAILED!");
+
+ if (!$result) return;
+
+ }
+ } else {
+ _debug("update not required.");
+ }
+
+ }
+
+ if (isset($options["gen-search-idx"])) {
+ echo "Generating search index (stemming set to English)...\n";
+
+ $res = $pdo->query("SELECT COUNT(id) AS count FROM ttrss_entries WHERE tsvector_combined IS NULL");
+ $row = $res->fetch();
+ $count = $row['count'];
+
+ print "Articles to process: $count.\n";
+
+ $limit = 500;
+ $processed = 0;
+
+ $sth = $pdo->prepare("SELECT id, title, content FROM ttrss_entries WHERE
+ tsvector_combined IS NULL ORDER BY id LIMIT ?");
+ $sth->execute([$limit]);
+
+ $usth = $pdo->prepare("UPDATE ttrss_entries
+ SET tsvector_combined = to_tsvector('english', ?) WHERE id = ?");
+
+ while (true) {
+
+ while ($line = $sth->fetch()) {
+ $tsvector_combined = mb_substr($line['title'] . ' ' .
+ preg_replace('/[<\?\:]/', ' ', strip_tags($line['content'])),
+ 0, 1000000);
+
+ $usth->execute([$tsvector_combined, $line['id']]);
+
+ $processed++;
+ }
+
+ print "Processed $processed articles...\n";
+
+ if ($processed < $limit) {
+ echo "All done.\n";
+ break;
+ }
+ }
+ }
+
+ if (isset($options["list-plugins"])) {
+ $tmppluginhost = new PluginHost();
+ $tmppluginhost->load_all($tmppluginhost::KIND_ALL, false);
+ $enabled = array_map("trim", explode(",", PLUGINS));
+
+ echo "List of all available plugins:\n";
+
+ foreach ($tmppluginhost->get_plugins() as $name => $plugin) {
+ $about = $plugin->about();
+
+ $status = $about[3] ? "system" : "user";
+
+ if (in_array($name, $enabled)) $name .= "*";
+
+ printf("%-50s %-10s v%.2f (by %s)\n%s\n\n",
+ $name, $status, $about[0], $about[2], $about[1]);
+ }
+
+ echo "Plugins marked by * are currently enabled for all users.\n";
+
+ }
+
+ if (isset($options["debug-feed"])) {
+ $feed = $options["debug-feed"];
+
+ if (isset($options["force-refetch"])) $_REQUEST["force_refetch"] = true;
+ if (isset($options["force-rehash"])) $_REQUEST["force_rehash"] = true;
- if ($lock_handle != false) {
- fclose($lock_handle);
+ $_REQUEST['xdebug'] = 1;
+
+ $rc = RSSUtils::update_rss_feed($feed) != false ? 0 : 1;
+
+ exit($rc);
}
+ if (isset($options["decrypt-feeds"])) {
+
+ if (!function_exists("mcrypt_decrypt")) {
+ _debug("mcrypt functions not available.");
+ return;
+ }
+
+ $res = $pdo->query("SELECT id, auth_pass FROM ttrss_feeds WHERE auth_pass_encrypted = true");
+
+ require_once "crypt.php";
+
+ $total = 0;
+
+ $pdo->beginTransaction();
+
+ $usth = $pdo->prepare("UPDATE ttrss_feeds SET auth_pass_encrypted = false, auth_pass = ?
+ WHERE id = ?");
+
+ while ($line = $res->fetch()) {
+ _debug("processing feed id " . $line["id"]);
+
+ $auth_pass = decrypt_string($line["auth_pass"]);
+
+ $usth->execute([$auth_pass, $line['id']]);
+
+ ++$total;
+ }
+
+ $pdo->commit();
+
+ _debug("$total feeds processed.");
+ }
+
+ PluginHost::getInstance()->run_commands($options);
+
if (file_exists(LOCK_DIRECTORY . "/$lock_filename"))
+ if (strtoupper(substr(PHP_OS, 0, 3)) == 'WIN')
+ fclose($lock_handle);
unlink(LOCK_DIRECTORY . "/$lock_filename");
?>