add support for TinyURL

extension/src/background.js

@@ -1,5 +1,6 @@
 const shortenerDomains = [
-    "t.co"
+    "t.co",
+    "tinyurl.com"
 ];
 
 browser.contextMenus.create({

extension/src/manifest.json

@@ -1,7 +1,7 @@
 {
   "manifest_version": 2,
   "name": "Unshortener",
-  "version": "0.2",
+  "version": "0.3",
 
   "description": "Unshorten links from Twitter.",
 

server/src/main.py

@@ -3,10 +3,11 @@ from typing import Optional
 from urllib.parse import urlparse
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from unshorteners import unshorten_twitter
+from unshorteners import unshorten_twitter, unshorten_tinyurl
 
 UNSHORTEN = {
-    't.co': unshorten_twitter
+    't.co': unshorten_twitter,
+    'tinyurl.com': unshorten_tinyurl
 }
 
 CACHE = {}
@@ -32,9 +33,11 @@ async def receive_url(url: Optional[str] = None):
         return {"error": f"cannot unshorten {domain}"}
 
     if url in CACHE:
-        unshortened = CACHE[url]
-    else:
-        unshortened = UNSHORTEN[domain](url)
-        CACHE[url] = unshortened
+        return CACHE[url]
 
-    return unshortened
+    result = UNSHORTEN[domain](url)
+    if result:
+        CACHE[url] = result
+        return result
+
+    return {"error": f"server error"}

server/src/unshorteners.py

@@ -1,12 +1,27 @@
 """Unshortening functions"""
 import re
 import requests
+from typing import Optional
 
 
-def unshorten_twitter(url: str):
+def unshorten_tinyurl(url: str) -> Optional[str]:
+    """Retrieve the actual URL behind a TinyURL."""
+    try:
+        response = requests.get(url, timeout=4, allow_redirects=False)
+    except requests.RequestException:
+        return None
+
+    if response.status_code == 301:
+        return response.headers.get("location", None)
+
+    return None
+
+
+def unshorten_twitter(url: str) -> Optional[str]:
     """Retrieve the actual URL behind a Twitter URL."""
     pattern = re.compile(r"<title>(.*?)<\/title>")
 
+    try:
         response = requests.get(
             url=url,
             headers={
@@ -14,6 +29,8 @@ def unshorten_twitter(url: str):
             },
             timeout=4
         )
+    except requests.RequestException:
+        return None
 
     match = pattern.search(response.text)
     if match: