From 2f08467a832e7a59f62558b37144e13f3c0a277e Mon Sep 17 00:00:00 2001
From: simon987 <me@simon987.net>
Date: Mon, 12 Aug 2019 09:19:46 -0400
Subject: [PATCH] Add post-process pass

---
 monitoring.py   |  2 +-
 post_process.py | 45 +++++++++++++++++++++++++++++++++++++++++++++
 run.py          | 14 ++++++++------
 3 files changed, 54 insertions(+), 7 deletions(-)
 create mode 100644 post_process.py

diff --git a/monitoring.py b/monitoring.py
index f21647e..7325d93 100644
--- a/monitoring.py
+++ b/monitoring.py
@@ -1,6 +1,6 @@
 from influxdb import InfluxDBClient
 
-client = InfluxDBClient('localhost', 8086, 'root', 'root', 'reddit_feed')
+client = InfluxDBClient("localhost", 8086, "root", "root", "reddit_feed")
 
 
 def init():
diff --git a/post_process.py b/post_process.py
new file mode 100644
index 0000000..6e4d4f5
--- /dev/null
+++ b/post_process.py
@@ -0,0 +1,45 @@
+import re
+
+LINK_RE = re.compile(r'href="([^"]+)"')
+INTERNAL_RE = re.compile(r"^https?://(reddit.com|redd.it|old.reddit.com|www.reddit.com|np.reddit.com)/(.*)")
+
+
+def post_process(thing):
+    thing["v"] = 1.0
+
+    thing["urls"] = []
+    if "body_html" in thing and thing["body_html"]:
+        thing["urls"].extend(get_links_from_body(thing["body_html"]))
+    elif "selftext_html" in thing and thing["selftext_html"]:
+        thing["urls"].extend(get_links_from_body(thing["selftext_html"]))
+
+    if "url" in thing and thing["url"] and is_external(thing["url"]):
+        thing["urls"].append(thing["url"])
+
+    return thing
+
+
+def get_links_from_body(body):
+    result = set()
+
+    for match in LINK_RE.finditer(body):
+        url = match.group(1)
+        if is_external(url):
+            result.add(url)
+
+    return list(result)
+
+
+def is_external(url):
+
+    if INTERNAL_RE.match(url):
+        return False
+
+    if "message/compose" in url:
+        return False
+
+    if url.startswith(("/", "#", "</")):
+        return False
+
+    return True
+
diff --git a/run.py b/run.py
index 838975c..6dc7443 100755
--- a/run.py
+++ b/run.py
@@ -18,6 +18,7 @@ from praw.endpoints import API_PATH
 from praw.models import Comment
 
 import monitoring
+from post_process import post_process
 from rate_limiter import GoodRateLimiter
 from util import update_cursor, read_cursor, reddit_ids
 
@@ -43,7 +44,7 @@ logger.addHandler(StreamHandler(sys.stdout))
 
 def serialize(thing):
     if isinstance(thing, Comment):
-        return json.dumps({
+        return {
             "author": str(thing.author),
             "author_flair_text": thing.author_flair_text,
             "body": thing.body,
@@ -65,9 +66,9 @@ def serialize(thing):
             "subreddit_id": thing.subreddit_id,
             "subreddit_type": thing.subreddit_type,
             "ups": thing.ups,
-        })
+        }
     else:
-        return json.dumps({
+        return {
             "archived": thing.archived,
             "author": str(thing.author),
             "author_flair_text": thing.author_flair_text,
@@ -121,17 +122,18 @@ def serialize(thing):
             "title": thing.title,
             "ups": thing.ups,
             "url": thing.url,
-        })
+        }
 
 
 def publish(thing):
     thing_type = type(thing).__name__.lower()
     j = serialize(thing)
+    post_process(j)
 
     reddit_channel.basic_publish(
         exchange='reddit',
-        routing_key="%s.%s" % (thing_type, str(thing.subreddit)),
-        body=j
+        routing_key="%s.%s" % (thing_type, str(thing.subreddit).lower()),
+        body=json.dumps(j)
     )