From dcc8d342358f155fe665433be5a6b7088403e6c3 Mon Sep 17 00:00:00 2001
From: artie <artiehime@gmail.com>
Date: Thu, 3 Oct 2024 19:30:13 +0200
Subject: [PATCH] add user-friendly error message for youtube ban

---
 artemis/cogs/media.py | 24 +++++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/artemis/cogs/media.py b/artemis/cogs/media.py
index 7a83d54..1e30b7f 100644
--- a/artemis/cogs/media.py
+++ b/artemis/cogs/media.py
@@ -1,7 +1,6 @@
 from __future__ import annotations, unicode_literals
 
 import asyncio
-import html
 import re
 import shlex
 import struct
@@ -19,6 +18,7 @@ from bs4 import BeautifulSoup
 from discord.ext import commands
 from pycaption import SRTWriter, WebVTTReader
 from yt_dlp.utils import parse_duration
+from yt_dlp.extractor.youtube import YoutubeIE
 
 from .. import utils
 from ..utils.common import ArtemisError, compress_image
@@ -56,7 +56,25 @@ def format_ytdlp_error(error: str) -> str:
     return ret
 
 
+YOUTUBE_BANNED_MESSAGE = """
+# Artemis has been banned from YouTube. 
+## Here are some trusted alternatives:
+
+**Cross-platform**
+**[Cobalt](<https://cobalt.tools/>)** - user-friendly web application
+
+**Android**
+**[Seal](<https://github.com/JunkFood02/Seal>)** - pretty open-source app that wraps yt-dlp
+
+**Desktop (Advanced)**
+**[yt-dlp](<https://github.com/yt-dlp/yt-dlp>)** - the best command-line downloader
+"""
+
+
 async def run_ytdlp(query: str, opts: dict, download: bool = True) -> dict:
+    if YoutubeIE.suitable(query):
+        raise ArtemisError(YOUTUBE_BANNED_MESSAGE)
+
     try:
         with yt_dlp.YoutubeDL(opts) as ytdl:
             return await asyncio.to_thread(ytdl.extract_info, query, download=download)
@@ -339,7 +357,7 @@ class Media(commands.Cog):
         trim = flags.trim
         ss, to = flags.ss, None
 
-        async def monitor_download():
+        async def _monitor_download():
             nonlocal msg, state
             while not finished:
                 content = "Processing..."
@@ -424,7 +442,7 @@ class Media(commands.Cog):
                 ytdl_opts["format"] = format
 
             info_dict = None
-            asyncio.create_task(monitor_download())
+            # asyncio.create_task(monitor_download())
             async with ctx.typing():
                 info_dict = await run_ytdlp(url, ytdl_opts)
             state = "uploading"