1) $similarity = 1; if ($min_title_length < 0) $min_title_length = 0; $similarity = sprintf("%.2f", $similarity); $this->host->set($this, "similarity", $similarity); $this->host->set($this, "min_title_length", $min_title_length); $this->host->set($this, "enable_globally", $enable_globally); echo T_sprintf("Data saved (%s, %d)", $similarity, $enable_globally); } function init($host) { $this->host = $host; $host->add_hook($host::HOOK_ARTICLE_FILTER, $this); $host->add_hook($host::HOOK_PREFS_TAB, $this); $host->add_hook($host::HOOK_PREFS_EDIT_FEED, $this); $host->add_hook($host::HOOK_PREFS_SAVE_FEED, $this); $host->add_hook($host::HOOK_ARTICLE_BUTTON, $this); } function get_js() { return file_get_contents(__DIR__ . "/init.js"); } function showrelated() { $id = (int) $_REQUEST['id']; $owner_uid = $_SESSION["uid"]; $sth = $this->pdo->prepare("SELECT title FROM ttrss_entries, ttrss_user_entries WHERE ref_id = id AND id = ? AND owner_uid = ?"); $sth->execute([$id, $owner_uid]); if ($row = $sth->fetch()) { $title = $row['title']; print "

$title

"; $sth = $this->pdo->prepare("SELECT ttrss_entries.id AS id, feed_id, ttrss_entries.title AS title, updated, link, ttrss_feeds.title AS feed_title, SIMILARITY(ttrss_entries.title, ?) AS sm FROM ttrss_entries, ttrss_user_entries LEFT JOIN ttrss_feeds ON (ttrss_feeds.id = feed_id) WHERE ttrss_entries.id = ref_id AND ttrss_user_entries.owner_uid = ? AND ttrss_entries.id != ? AND date_entered >= NOW() - INTERVAL '2 weeks' ORDER BY sm DESC, date_entered DESC LIMIT 10"); $sth->execute([$title, $owner_uid, $id]); print ""; } print ""; } function hook_article_button($line) { return "bookmark_outline"; } function hook_prefs_tab($args) { if ($args != "prefFeeds") return; print "
extension ".__('Mark similar articles as read (af_psql_trgm)')."\">"; if (DB_TYPE != "pgsql") { print_error("Database type not supported."); } else { $res = $this->pdo->query("select 'similarity'::regproc"); if (!$res || !$res->fetch()) { print_error("pg_trgm extension not found."); } $similarity = $this->host->get($this, "similarity", $this->default_similarity); $min_title_length = $this->host->get($this, "min_title_length", $this->default_min_length); $enable_globally = sql_bool_to_bool($this->host->get($this, "enable_globally")); print "
"; print ""; print \Controls\pluginhandler_tags($this, "save"); print "

" . __("Global settings") . "

"; print_notice("Enable for specific feeds in the feed editor."); print "
"; print " "; print ""; print "
" . __("PostgreSQL trigram extension returns string similarity as a floating point number (0-1). Setting it too low might produce false positives, zero disables checking.") . "
"; print "
"; print " "; print ""; print "
"; print ""; print "
"; print "
"; print \Controls\submit_tag(__("Save")); print "
"; /* cleanup */ $enabled_feeds = $this->filter_unknown_feeds( $this->get_stored_array("enabled_feeds")); $this->host->set($this, "enabled_feeds", $enabled_feeds); if (count($enabled_feeds) > 0) { print "
"; print "

" . __("Currently enabled for (click to edit):") . "

"; print ""; } } print "
"; } function hook_prefs_edit_feed($feed_id) { print "
".__("Similarity (af_psql_trgm)")."
"; print "
"; $enabled_feeds = $this->get_stored_array("enabled_feeds"); $checked = in_array($feed_id, $enabled_feeds) ? "checked" : ""; print "
"; print ""; print "
"; print "
"; } function hook_prefs_save_feed($feed_id) { $enabled_feeds = $this->get_stored_array("enabled_feeds"); $enable = checkbox_to_sql_bool($_POST["trgm_similarity_enabled"] ?? ""); $key = array_search($feed_id, $enabled_feeds); if ($enable) { if ($key === false) { array_push($enabled_feeds, $feed_id); } } else { if ($key !== false) { unset($enabled_feeds[$key]); } } $this->host->set($this, "enabled_feeds", $enabled_feeds); } function hook_article_filter($article) { if (DB_TYPE != "pgsql") return $article; $res = $this->pdo->query("select 'similarity'::regproc"); if (!$res || !$res->fetch()) return $article; $enable_globally = $this->host->get($this, "enable_globally"); if (!$enable_globally && !in_array($article["feed"]["id"], $this->get_stored_array("enabled_feeds"))) { return $article; } $similarity = (float) $this->host->get($this, "similarity", $this->default_similarity); if ($similarity < 0.01) { Debug::log("af_psql_trgm: similarity is set too low ($similarity)", Debug::$LOG_EXTENDED); return $article; } $min_title_length = (int) $this->host->get($this, "min_title_length", $this->default_min_length); if (mb_strlen($article["title"]) < $min_title_length) { Debug::log("af_psql_trgm: article title is too short (min: $min_title_length)", Debug::$LOG_EXTENDED); return $article; } $owner_uid = $article["owner_uid"]; $entry_guid = $article["guid_hashed"]; $title_escaped = $article["title"]; // trgm does not return similarity=1 for completely equal strings // this seems to be no longer the case (fixed in upstream?) /* $sth = $this->pdo->prepare("SELECT COUNT(id) AS nequal FROM ttrss_entries, ttrss_user_entries WHERE ref_id = id AND date_entered >= NOW() - interval '3 days' AND title = ? AND guid != ? AND owner_uid = ?"); $sth->execute([$title_escaped, $entry_guid, $owner_uid]); $row = $sth->fetch(); $nequal = $row['nequal']; Debug::log("af_psql_trgm: num equals: $nequal", Debug::$LOG_EXTENDED); if ($nequal != 0) { $article["force_catchup"] = true; return $article; } */ $sth = $this->pdo->prepare("SELECT MAX(SIMILARITY(title, ?)) AS ms FROM ttrss_entries, ttrss_user_entries WHERE ref_id = id AND date_entered >= NOW() - interval '1 day' AND guid != ? AND owner_uid = ?"); $sth->execute([$title_escaped, $entry_guid, $owner_uid]); $row = $sth->fetch(); $similarity_result = $row['ms']; Debug::log("af_psql_trgm: similarity result for $title_escaped: $similarity_result", Debug::$LOG_EXTENDED); if ($similarity_result >= $similarity) { Debug::log("af_psql_trgm: marking article as read ($similarity_result >= $similarity)", Debug::$LOG_EXTENDED); $article["force_catchup"] = true; } return $article; } function api_version() { return 2; } private function filter_unknown_feeds($enabled_feeds) { $tmp = array(); foreach ($enabled_feeds as $feed) { $sth = $this->pdo->prepare("SELECT id FROM ttrss_feeds WHERE id = ? AND owner_uid = ?"); $sth->execute([$feed, $_SESSION['uid']]); if ($row = $sth->fetch()) { array_push($tmp, $feed); } } return $tmp; } private function get_stored_array($name) { $tmp = $this->host->get($this, $name); if (!is_array($tmp)) $tmp = []; return $tmp; } }