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['param']; $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')."\">"; if (DB_TYPE != "pgsql") { print_error("Database type not supported."); } else { $res = $this->pdo->query("select 'similarity'::regproc"); if (!$res->fetch()) { print_error("pg_trgm extension not found."); } $similarity = $this->host->get($this, "similarity"); $min_title_length = $this->host->get($this, "min_title_length"); $enable_globally = $this->host->get($this, "enable_globally"); if (!$similarity) $similarity = '0.75'; if (!$min_title_length) $min_title_length = '32'; print "
"; print ""; print_hidden("op", "pluginhandler"); print_hidden("method", "save"); print_hidden("plugin", "af_psql_trgm"); 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_button("submit", __("Save"), "class='alt-primary'"); print "
"; $enabled_feeds = $this->host->get($this, "enabled_feeds"); if (!array($enabled_feeds)) $enabled_feeds = array(); $enabled_feeds = $this->filter_unknown_feeds($enabled_feeds); $this->host->set($this, "enabled_feeds", $enabled_feeds); if (count($enabled_feeds) > 0) { print "

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

"; print ""; } } print "
"; } function hook_prefs_edit_feed($feed_id) { print "
".__("Similarity (pg_trgm)")."
"; print "
"; $enabled_feeds = $this->host->get($this, "enabled_feeds"); if (!array($enabled_feeds)) $enabled_feeds = array(); $key = array_search($feed_id, $enabled_feeds); $checked = $key !== FALSE ? "checked" : ""; print "
"; print ""; print "
"; print "
"; } function hook_prefs_save_feed($feed_id) { $enabled_feeds = $this->host->get($this, "enabled_feeds"); if (!is_array($enabled_feeds)) $enabled_feeds = array(); $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->fetch()) return $article; $enable_globally = $this->host->get($this, "enable_globally"); if (!$enable_globally) { $enabled_feeds = $this->host->get($this, "enabled_feeds"); $key = array_search($article["feed"]["id"], $enabled_feeds); if ($key === FALSE) return $article; } $similarity = (float) $this->host->get($this, "similarity"); if ($similarity < 0.01) return $article; $min_title_length = (int) $this->host->get($this, "min_title_length"); if (mb_strlen($article["title"]) < $min_title_length) 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 $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: $similarity_result", Debug::$LOG_EXTENDED); if ($similarity_result >= $similarity) { $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; } }