From 8d171ffdbe028ba6dbb78abfe91daa0eb8879356 Mon Sep 17 00:00:00 2001
From: iou1name <iou1name@national.shitposting.agency>
Date: Wed, 19 Sep 2018 13:01:19 -0400
Subject: [PATCH] added puu.sh filename support

---
 modules/url.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/modules/url.py b/modules/url.py
index d0127f3..6a7b429 100755
--- a/modules/url.py
+++ b/modules/url.py
@@ -8,10 +8,28 @@ from html.parser import HTMLParser
 
 import requests
 
-from module import hook
+from module import hook, url_callback
 
 HEADERS = {"User-Agent": "bix nood gimme the title", "Range": "bytes=0-4096"}
 
+@url_callback('puu.sh/')
+def get_puush_fname(bot, url):
+	"""
+	Callback puu.sh links. Gets the filename and displays it.
+	"""
+	try:
+		res = requests.head(url)
+	except requests.exceptions.RequestException:
+		return
+
+	fname = res.headers.get('Content-Disposition', '')
+	fname = re.search(r'\"(.*)\"', fname).group(1)
+	if not fname:
+		return
+
+	bot.msg(f"[ \x0310puu.sh \x03] - \x0304{fname}")
+
+
 @hook(True)
 def title_auto(bot, trigger):
 	"""