blallo/tt-rss
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

add auto_assign_labels plugin; allow article filter plugins to add labels to articles

Browse source
This commit is contained in:
Andrew Dolgov 2015-03-04 00:26:52 +03:00
parent 0c6f7b314a
commit a29fe12195
2 changed files with 73 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
include/rssfuncs.php
View file

@ -671,9 +671,11 @@
if (db_num_rows($result) != 0) {
$base_entry_id = db_fetch_result($result, 0, "id");
$entry_stored_hash = db_fetch_result($result, 0, "content_hash");
$article_labels = get_article_labels($base_entry_id, $owner_uid);
} else {
$base_entry_id = false;
$entry_stored_hash = "";
$article_labels = array();
}
$article = array("owner_uid" => $owner_uid, // read only
@ -681,6 +683,7 @@
"title" => $entry_title,
"content" => $entry_content,
"link" => $entry_link,
"labels" => $article_labels, // current limitation: can add labels to article, can't remove them
"tags" => $entry_tags,
"author" => $entry_author,
"force_catchup" => false, // ugly hack for the time being
@ -739,6 +742,12 @@
$entry_link = db_escape_string($article["link"]);
$entry_content = $article["content"]; // escaped below
$entry_force_catchup = $article["force_catchup"];
$article_labels = $article["labels"];
if ($debug_enabled) {
_debug("article labels:", $debug_enabled);
print_r($article_labels);
}
_debug("force catchup: $entry_force_catchup");
@ -790,12 +799,8 @@
'$entry_language',
'$entry_author')");
$article_labels = array();
} else {
$base_entry_id = db_fetch_result($result, 0, "id");
$article_labels = get_article_labels($base_entry_id, $owner_uid);
}
// now it should exist, if not - bad luck then
@ -962,7 +967,13 @@
db_query("COMMIT");
_debug("assigning labels...", $debug_enabled);
_debug("assigning labels [other]...", $debug_enabled);
foreach ($article_labels as $label) {
label_add_article($entry_ref_id, $label[1], $owner_uid);
}
_debug("assigning labels [filters]...", $debug_enabled);
assign_article_to_label_filters($entry_ref_id, $article_filters,
$owner_uid, $article_labels);

57
plugins/auto_assign_labels/init.php Normal file
View file

@ -0,0 +1,57 @@
<?php
class Auto_Assign_Labels extends Plugin {
private $host;
function about() {
return array(1.0,
"Assign labels automatically based on article title, content, and tags",
"fox");
}
function init($host) {
$this->host = $host;
$host->add_hook($host::HOOK_ARTICLE_FILTER, $this);
}
function get_all_labels_filter_format($owner_uid) {
$rv = array();
$result = db_query("SELECT id, fg_color, bg_color, caption FROM ttrss_labels2 WHERE owner_uid = " . $owner_uid);
while ($line = db_fetch_assoc($result)) {
array_push($rv, array(label_to_feed_id($line["id"]),
$line["caption"], $line["fg_color"], $line["bg_color"]));
}
return $rv;
}
function hook_article_filter($article) {
$owner_uid = $article["owner_uid"];
$labels = $this->get_all_labels_filter_format($owner_uid);
$tags_str = join(",", $article["tags"]);
foreach ($labels as $label) {
$caption = preg_quote($label[1]);
if ($caption && preg_match("/\b$caption\b/i", "$tags_str " . strip_tags($article["content"]) . " " . $article["title"])) {
# defined in rssfuncs.php
if (!labels_contains_caption($article["labels"], $caption)) {
array_push($article["labels"], $label);
}
}
}
return $article;
}
function api_version() {
return 2;
}
}
?>