Merge pull request #1 from RB387/dev

Added first version of bot

Author: RB387

.env

@@ -0,0 +1 @@
+DISCORD_TOKEN=

.gitignore

@@ -14,8 +14,6 @@ dist/
 downloads/
 eggs/
 .eggs/
-lib/
-lib64/
 parts/
 sdist/
 var/
@@ -102,7 +100,6 @@ celerybeat.pid
 *.sage.py
 
 # Environments
-.env
 .venv
 env/
 venv/
@@ -127,3 +124,5 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+.idea

Dockerfile

@@ -0,0 +1,26 @@
+FROM python:3.8-alpine
+
+COPY . ./
+
+# Install build dependencies
+RUN apk update && apk add --no-cache --virtual .build-deps \
+  build-base \
+  libffi-dev \
+  libsodium-dev
+
+# Install dependencies
+RUN apk update && apk add --no-cache \
+  ca-certificates \
+  ffmpeg \
+  opus-dev \
+  libffi \
+  libsodium \
+  gcc
+
+# Install pip dependencies
+RUN pip3 install --no-cache-dir -r requirements.txt
+
+# Clean up build dependencies
+RUN apk del .build-deps
+
+ENTRYPOINT ["python", "main.py"]

README.md

@@ -0,0 +1,60 @@
+# Python Discord Music Bot
+Easy to extend and support
+
+## Supported commands
+```
+!play [url or name] - play song from any stream/video or youtube search
+!queue - current bot song queue
+!pause - pause current song
+!resume - resume paused song
+!skip - skip current song
+!clear - clear bot queue
+!message-clean - clear bot messages
+!studio21 - play radio studio21
+!21queue - show studio21 queue 
+```
+
+## Background queue tasks
+Just add queue in config.toml
+```python
+bot: DiscordBot
+
+async def task():
+    print('done in background')
+    
+await bot.add_task('my_queue', task)
+```
+
+## Configuration
+All bot configuration in config.toml  
+The only exception is secrets. They must be set in environment or in file .env:
+```
+DISCORD_TOKEN -- bot token
+```
+
+## MacOS Installation
+```
+brew install ffmpeg
+pip intall -r requirements.txt
+```
+
+## Run
+### Docker
+```
+docker build -t discord-music-bot .
+docker run discord-music-bot
+```
+### Local
+```
+python main.py
+```
+
+## Tests
+```
+pip install -r test-requirements.txt
+pytest tests/
+```
+
+## TODO
+* Write tests on bot business logic
+* Add loop queue

config.toml

@@ -0,0 +1,30 @@
+language = "en"
+
+[common]
+messages_ttl = "5"
+playlist_queue = "playlist"
+
+[bot]
+command_prefix = "!"
+
+[[bot.queues]]
+name = "playlist"
+
+[ydl]
+format = "bestaudio/best"
+noplaylist = "true"
+source_address = "0.0.0.0"
+default_search = "auto"
+
+[ffmpeg]
+before_options = "-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5 -nostdin"
+options = "-vn"
+
+[logger]
+level = "INFO"
+
+[response]
+colour = 7506394
+footer_img = "https://i.imgur.com/gFHBoZA.png"
+footer_text = "https://github.com/RB387/py-discord-music-bot"
+author_url = "https://github.com/RB387/py-discord-music-bot"

constants.py

@@ -0,0 +1,4 @@
+from pathlib import Path
+
+CONFIG_PATH = Path(__file__).parent / 'config.toml'
+TEMPLATES_PATH = Path(__file__).parent / 'templates'

lib/__init__.py

@@ -0,0 +1,2 @@
+from .commads import *  # noqa
+from .events import *  # noqa

lib/app.py

@@ -0,0 +1,43 @@
+from typing import Type
+
+from lib.core.bot import DiscordBot
+from lib.core.injector import DependencyInjector
+from lib.core.logger import Logger
+from lib.core.router import Router, router as default_router
+
+
+class App:
+    def __init__(
+        self,
+        injector: DependencyInjector,
+        bot_cls: Type[DiscordBot] = DiscordBot,
+    ):
+
+        self._injector = injector
+        self._bot = injector.inject(bot_cls)
+        self._logger = injector.inject(Logger)
+        self._routers = [default_router]
+
+    def run(self):
+        self._setup_routes()
+
+        self._bot.listen('on_connect')(self._injector.connect)
+        self._bot.listen('on_disconnect')(self._injector.disconnect)
+
+        self._logger.info('Running bot...')
+        self._bot.run()
+
+    def add_router(self, router: Router):
+        self._routers.append(router)
+
+    def _setup_routes(self):
+        for router in self._routers:
+            routes = router.get_routes()
+
+            for event in routes.events:
+                handler = self._injector.inject(event.handler)
+                self._bot.listen(event.name)(handler.handle)
+
+            for command in routes.commands:
+                handler = self._injector.inject(command.handler)
+                self._bot.command(name=command.name)(handler.handle)

lib/commads/__init__.py

@@ -0,0 +1,8 @@
+from .clear import PlaylistCleaner
+from .message_clear import MessageCleaner
+from .pause import Pause
+from .play import Player
+from .queue import QueueStatus
+from .resume import Resume
+from .skip import Skip
+from .studio21 import Studio21Player, Studio21QueueStatus

lib/commads/clear.py

@@ -0,0 +1,19 @@
+from dataclasses import dataclass
+
+from discord.ext.commands import Context
+
+from lib.core import ClientProtocol, router
+from lib.emoji import Emoji
+from lib.messenger import Messenger
+from lib.playlist import Playlist
+
+
+@router.command('clear')
+@dataclass
+class PlaylistCleaner(ClientProtocol):
+    playlist: Playlist
+    messenger: Messenger
+
+    async def handle(self, ctx: Context):
+        await self.messenger.react(ctx, Emoji.OK_HAND)
+        await self.playlist.clear()