diff --git a/.gitignore b/.gitignore index 68bc17f..e307134 100644 --- a/.gitignore +++ b/.gitignore @@ -152,6 +152,9 @@ dmypy.json # Cython debug symbols cython_debug/ +compose.yml +.idea/ + # PyCharm # JetBrains specific template is maintained in a separate JetBrains.gitignore that can # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore diff --git a/.idea/.gitignore b/.idea/.gitignore new file mode 100644 index 0000000..13566b8 --- /dev/null +++ b/.idea/.gitignore @@ -0,0 +1,8 @@ +# Default ignored files +/shelf/ +/workspace.xml +# Editor-based HTTP Client requests +/httpRequests/ +# Datasource local storage ignored files +/dataSources/ +/dataSources.local.xml diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..3421d9c --- /dev/null +++ b/Dockerfile @@ -0,0 +1,10 @@ +FROM python:3.11.6 + +WORKDIR /usr/src/units + +COPY requiremtns.txt . +RUN pip install -r requirements.com + +COPY bot.py . + +CMD ["python", "bot.py"] diff --git a/bot.py b/bot.py new file mode 100644 index 0000000..7992776 --- /dev/null +++ b/bot.py @@ -0,0 +1,51 @@ +import os +from urllib.parse import urlparse, parse_qs +from contextlib import suppress +import discord +from discord.ext import commands +import re + +intents = discord.Intents.default() +intents.members = True +intents.message_content = True + +bot = commands.Bot(command_prefix='?', intents=intents) + +PIPED_URL = os.getenv('PIPED_URL') +DISCORD_TOKEN = os.getenv('DISCORD_TOKEN') + + +@bot.event +async def on_ready(): + print("Ready!") + + +@bot.event +async def on_message(message): + regex = re.compile( + r'https://(music\.)?(www\.)?youtu(be)?\.[a-zA-Z0-9()]{1,6}\b([-a-zA-Z0-9()@:%_+.~#?&/=]*)') + + result = regex.search(message.content) + if result is not None: + youtube_url = result.group(0) + video_id = await get_youtube_id(youtube_url) + + if video_id is not None: + await message.reply(f"I think you meant {PIPED_URL}{video_id}.", mention_author=False) + + +async def get_youtube_id(url: str, ignore_playlist=True) -> str: + query = urlparse(url) + if query.hostname == 'youtu.be': return query.path[1:] + if query.hostname in {'www.youtube.com', 'youtube.com', 'music.youtube.com'}: + if not ignore_playlist: + # use case: get playlist id not current video in playlist + with suppress(KeyError): + return parse_qs(query.query)['list'][0] + if query.path == '/watch': return parse_qs(query.query)['v'][0] + if query.path[:7] == '/watch/': return query.path.split('/')[1] + if query.path[:7] == '/embed/': return query.path.split('/')[2] + if query.path[:3] == '/v/': return query.path.split('/')[2] + + +bot.run(DISCORD_TOKEN) diff --git a/requiremtns.txt b/requiremtns.txt new file mode 100644 index 0000000..69a3e12 --- /dev/null +++ b/requiremtns.txt @@ -0,0 +1,10 @@ +aiohttp==3.8.6 +aiosignal==1.3.1 +async-timeout==4.0.3 +attrs==23.1.0 +charset-normalizer==3.3.2 +discord.py==2.3.2 +frozenlist==1.4.0 +idna==3.4 +multidict==6.0.4 +yarl==1.9.2