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

2 files changed, 73 insertions, 5 deletions

diff --git a/include/rssfuncs.php b/include/rssfuncs.php
index b285f74c0..1a26e0485 100644
--- a/include/rssfuncs.php
+++ b/include/rssfuncs.php
@@ -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);
diff --git a/plugins/auto_assign_labels/init.php b/plugins/auto_assign_labels/init.php
new file mode 100644
index 000000000..36f7c3267
--- /dev/null
+++ b/plugins/auto_assign_labels/init.php
@@ -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;
+	}
+}
+?>