Rename app/youtube_dns.py to app/utils.py

app/utils.py

@@ -0,0 +1,15 @@
+import re
+
+def extract_video_id(url: str) -> str:
+    patterns = [
+        r"v=([a-zA-Z0-9_-]{11})",
+        r"youtu\.be/([a-zA-Z0-9_-]{11})",
+        r"shorts/([a-zA-Z0-9_-]{11})"
+    ]
+
+    for p in patterns:
+        m = re.search(p, url)
+        if m:
+            return m.group(1)
+
+    raise ValueError("Invalid YouTube URL")

app/youtube_dns.py

@@ -1,20 +0,0 @@
-import socket
-import dns.resolver
-
-_original_getaddrinfo = socket.getaddrinfo
-
-def force_ipv4_dns():
-    def patched_getaddrinfo(host, port, family=0, type=0, proto=0, flags=0):
-        try:
-            if any(x in host for x in ("youtube.com", "googlevideo.com", "youtu.be")):
-                resolver = dns.resolver.Resolver()
-                resolver.nameservers = ["8.8.8.8", "1.1.1.1"]
-                answer = resolver.resolve(host, "A")
-                return [
-                    (socket.AF_INET, socket.SOCK_STREAM, proto, "", (str(answer[0]), port))
-                ]
-        except Exception:
-            pass
-        return _original_getaddrinfo(host, port, family, type, proto, flags)
-
-    socket.getaddrinfo = patched_getaddrinfo