v1.3 [ YukkiMusic ]

Author: Vivekkumar-IN

CHANGELOG.md

@@ -1,7 +1,23 @@
-# Changelog
+# CHANGELOG
 
 All notable changes to YukkiMusic will be documented in this file.
 
+
+## [v1.3] - 2024-11-15
+
+### Added
+- Introduced the `COOKIE_LINK` variable for pasting cookies in the `batbin` link.
+- Added the `asyncify` decorator to convert synchronous functions to asynchronous.
+- Added the `OWNER` command that will show in the log group or bot PM/DM.
+
+### Removed
+- Deprecated the `oauth2` method, which is no longer supported. This functionality has been merged into `yt-dlp`.
+
+### Fixed
+- Various bug fixes and performance improvements.
+
+**Full Changelog:** [`v1.2...v1.3`](https://github.com/TheTeamVivek/YukkiMusic/compare/v1.2...v1.3)
+
 ## [v1.2] - 2024-11-03
 
 ### Added

YukkiMusic/__init__.py

@@ -8,13 +8,17 @@
 # All rights reserved.
 
 from YukkiMusic.core.bot import YukkiBot
+from YukkiMusic.core.cookies import save_cookies
 from YukkiMusic.core.dir import dirr
 from YukkiMusic.core.git import git
 from YukkiMusic.core.userbot import Userbot
 from YukkiMusic.misc import dbb, heroku, sudo
 
 from .logging import LOGGER
 
+# Save cookies in txt
+save_cookies()
+
 # Directories
 dirr()
 

YukkiMusic/core/bot.py

@@ -16,18 +16,16 @@
 
 from pyrogram import Client
 from pyrogram.enums import ChatMemberStatus
-from pyrogram.errors import (
-    ChatSendPhotosForbidden,
-    ChatWriteForbidden,
-    FloodWait,
-    MessageIdInvalid,
-)
-from pyrogram.types import (
-    BotCommand,
-    BotCommandScopeAllChatAdministrators,
-    BotCommandScopeAllGroupChats,
-    BotCommandScopeAllPrivateChats,
-)
+from pyrogram.types import BotCommand
+from pyrogram.types import BotCommandScopeAllChatAdministrators
+from pyrogram.types import BotCommandScopeAllGroupChats
+from pyrogram.types import BotCommandScopeAllPrivateChats
+from pyrogram.types import BotCommandScopeChat
+from pyrogram.types import BotCommandScopeChatMember
+from pyrogram.errors import ChatSendPhotosForbidden
+from pyrogram.errors import ChatWriteForbidden
+from pyrogram.errors import FloodWait
+from pyrogram.errors import MessageIdInvalid
 
 import config
 
@@ -98,57 +96,108 @@ async def start(self):
         get_me = await self.get_me()
         self.username = get_me.username
         self.id = get_me.id
-        self.name = self.me.first_name + " " + (self.me.last_name or "")
-        self.mention = self.me.mention
+        self.name = f"{get_me.first_name} {get_me.last_name or ''}"
+        self.mention = get_me.mention
 
         try:
             await self.send_message(
                 config.LOG_GROUP_ID,
-                text=f"<u><b>{self.mention} Bot Started :</b><u>\n\nId : <code>{self.id}</code>\nName : {self.name}\nUsername : @{self.username}",
+                text=(
+                    f"<u><b>{self.mention} Bot Started :</b></u>\n\n"
+                    f"Id : <code>{self.id}</code>\n"
+                    f"Name : {self.name}\n"
+                    f"Username : @{self.username}"
+                ),
             )
-        except:
+        except Exception as e:
             LOGGER(__name__).error(
-                "Bot has failed to access the log group. Make sure that you have added your bot to your log channel and promoted as admin!"
+                "Bot failed to access the log group. Ensure the bot is added and promoted as admin."
             )
-            LOGGER(__name__).error("An error occurred", exc_info=True)
+            LOGGER(__name__).error("Error details:", exc_info=True)
             sys.exit()
+
         if config.SET_CMDS == str(True):
+            try:
+                await self._set_default_commands()
+            except Exception as e:
+                LOGGER(__name__).warning("Failed to set commands:", exc_info=True)
+
+    async def _set_default_commands(self):
+        private_commands = [
+            BotCommand("start", "Start the bot"),
+            BotCommand("help", "Get the help menu"),
+            BotCommand("ping", "Check if the bot is alive or dead"),
+        ]
+        group_commands = [BotCommand("play", "Start playing requested song")]
+        admin_commands = [
+            BotCommand("play", "Start playing requested song"),
+            BotCommand("skip", "Move to next track in queue"),
+            BotCommand("pause", "Pause the current playing song"),
+            BotCommand("resume", "Resume the paused song"),
+            BotCommand("end", "Clear the queue and leave voice chat"),
+            BotCommand("shuffle", "Randomly shuffle the queued playlist"),
+            BotCommand("playmode", "Change the default playmode for your chat"),
+            BotCommand("settings", "Open bot settings for your chat"),
+        ]
+        owner_commands = [
+            BotCommand("update", "Update the bot"),
+            BotCommand("restart", "Restart the bot"),
+            BotCommand("logs", "Get logs"),
+            BotCommand("export", "Export all data of mongodb"),
+            BotCommand("import", "Import all data in mongodb"),
+            BotCommand("addsudo", "Add a user as a sudoer"),
+            BotCommand("delsudo", "Remove a user from sudoers"),
+            BotCommand("sudolist", "List all sudo users"),
+            BotCommand("log", "Get the bot logs"),
+            BotCommand("getvar", "Get a specific environment variable"),
+            BotCommand("delvar", "Delete a specific environment variable"),
+            BotCommand("setvar", "Set a specific environment variable"),
+            BotCommand("usage", "Get dyno usage information"),
+            BotCommand("maintenance", "Enable or disable maintenance mode"),
+            BotCommand("logger", "Enable or disable logging"),
+            BotCommand("block", "Block a user"),
+            BotCommand("unblock", "Unblock a user"),
+            BotCommand("blacklist", "Blacklist a chat"),
+            BotCommand("whitelist", "Whitelist a chat"),
+            BotCommand("blacklisted", "List all blacklisted chats"),
+            BotCommand("autoend", "Enable or disable auto end for streams"),
+            BotCommand("reboot", "Reboot the bot"),
+            BotCommand("restart", "Restart the bot"),
+        ]
+
+        await self.set_bot_commands(
+            private_commands, scope=BotCommandScopeAllPrivateChats()
+        )
+        await self.set_bot_commands(
+            group_commands, scope=BotCommandScopeAllGroupChats()
+        )
+        await self.set_bot_commands(
+            admin_commands, scope=BotCommandScopeAllChatAdministrators()
+        )
+
+        LOG_GROUP_ID = (
+            f"@{config.LOG_GROUP_ID}"
+            if isinstance(config.LOG_GROUP_ID, str)
+            and not config.LOG_GROUP_ID.startswith("@")
+            else config.LOG_GROUP_ID
+        )
+
+        for owner_id in config.OWNER_ID:
             try:
                 await self.set_bot_commands(
-                    commands=[
-                        BotCommand("start", "Start the bot"),
-                        BotCommand("help", "Get the help menu"),
-                        BotCommand("ping", "Check if the bot is alive or dead"),
-                    ],
-                    scope=BotCommandScopeAllPrivateChats(),
+                    owner_commands,
+                    scope=BotCommandScopeChatMember(
+                        chat_id=LOG_GROUP_ID, user_id=owner_id
+                    ),
                 )
                 await self.set_bot_commands(
-                    commands=[
-                        BotCommand("play", "Start playing requested song"),
-                    ],
-                    scope=BotCommandScopeAllGroupChats(),
+                    owner_commands, scope=BotCommandScopeChat(chat_id=owner_id)
                 )
-                await self.set_bot_commands(
-                    commands=[
-                        BotCommand("play", "Start playing requested song"),
-                        BotCommand("skip", "Move to next track in queue"),
-                        BotCommand("pause", "Pause the current playing song"),
-                        BotCommand("resume", "Resume the paused song"),
-                        BotCommand("end", "Clear the queue and leave voicechat"),
-                        BotCommand("shuffle", "Randomly shuffles the queued playlist."),
-                        BotCommand(
-                            "playmode",
-                            "Allows you to change the default playmode for your chat",
-                        ),
-                        BotCommand(
-                            "settings",
-                            "Open the settings of the music bot for your chat.",
-                        ),
-                    ],
-                    scope=BotCommandScopeAllChatAdministrators(),
+            except Exception as e:
+                LOGGER(__name__).warning(
+                    "Failed to set owner commands for user %s:", owner_id, exc_info=True
                 )
-            except:
-                pass
+
         else:
             pass
         try:
@@ -163,6 +212,3 @@ async def start(self):
         else:
             self.name = get_me.first_name
         LOGGER(__name__).info(f"MusicBot started as {self.name}")
-
-    async def stop(self):
-        await super().stop()