From 297c403248f026b4dc0c91e56a6f867ae007c1ea Mon Sep 17 00:00:00 2001
From: jesopo
Date: Wed, 26 Jun 2019 18:09:39 +0100
Subject: add !unshorten

---
 modules/shorturl.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'modules')

diff --git a/modules/shorturl.py b/modules/shorturl.py
index 43b57c55..49b9f3cd 100644
--- a/modules/shorturl.py
+++ b/modules/shorturl.py
@@ -70,3 +70,21 @@ class Module(ModuleManager.BaseModule):
 
         event["stdout"].write("Shortened URL: %s" % self._shorturl(
             event["server"], url))
+
+    @utils.hook("received.command.unshorten", min_args=1)
+    def unshorten(self, event):
+        url = event["args_split"][0]
+        if not re.match(utils.http.REGEX_URL, url):
+            url = "http://%s" % url
+
+        try:
+            response = utils.http.request(url, method="HEAD",
+                allow_redirects=False)
+        except:
+            response = None
+
+        if response and "location" in response.headers:
+            event["stdout"].write("Unshortened: %s" %
+                response.headers["location"])
+        else:
+            event["stderr"].write("Failed to unshorten URL")
-- 
cgit v1.3.1-10-gc9f91