feat: add story downloader, poll move, and various plugin improvements

DragMusic/plugins/kuri/poll.py

@@ -0,0 +1,61 @@
+"""
+/poll <question> | <option1> | <option2> | ... [| key=value ...]
+
+Create a poll in the group. Only admins/owner can use this command.
+Supports all Pyrogram send_poll options via key=value pairs after the options.
+Example:
+/poll What's your favorite color? | Red | Blue | Green | is_anonymous=False | allows_multiple_answers=True
+"""
+
+from pyrogram import filters
+from pyrogram.types import Message
+from DragMusic import app
+
+async def is_admin(client, chat_id, user_id):
+    member = await client.get_chat_member(chat_id, user_id)
+    return member.status in ("administrator", "creator")
+
+@app.on_message(filters.command("poll", prefixes=["/", "!"]) & filters.group)
+async def poll_command(client, message: Message):
+    if not await is_admin(client, message.chat.id, message.from_user.id):
+        return await message.reply_text("Only group admins and the owner can use this command.")
+    if len(message.command) == 1:
+        return await message.reply_text(
+            "Usage:\n/poll <question> | <option1> | <option2> | ... [| key=value ...]\n"
+            "Example:\n/poll What's your favorite color? | Red | Blue | Green | is_anonymous=False | allows_multiple_answers=True"
+        )
+    # Join the rest of the message and split by |
+    parts = message.text.split(None, 1)[1].split("|")
+    parts = [p.strip() for p in parts if p.strip()]
+    if len(parts) < 3:
+        return await message.reply_text("You must provide a question and at least two options.")
+    question = parts[0]
+    options = []
+    kwargs = {}
+    for part in parts[1:]:
+        if "=" in part:
+            key, value = part.split("=", 1)
+            key = key.strip()
+            value = value.strip()
+            # Convert value to correct type
+            if value.lower() == "true":
+                value = True
+            elif value.lower() == "false":
+                value = False
+            elif value.isdigit():
+                value = int(value)
+            kwargs[key] = value
+        else:
+            options.append(part)
+    if len(options) < 2:
+        return await message.reply_text("You must provide at least two options.")
+    try:
+        poll_msg = await app.send_poll(
+            chat_id=message.chat.id,
+            question=question,
+            options=options,
+            **kwargs
+        )
+        await message.reply_text(f"Poll created: [Jump to poll]({poll_msg.link})", disable_web_page_preview=True)
+    except Exception as e:
+        await message.reply_text(f"Failed to create poll: {e}") 

DragMusic/plugins/kuri/storydl.py

@@ -0,0 +1,47 @@
+"""
+/story <username or userid> <story_id>
+
+Download a Telegram story (video or photo) from a user by username/userid and story id, using the userbot. The story will be sent to the chat where the command was used.
+
+Example:
+/story @Itzabhi16 1
+/story 123456789 2
+"""
+
+import tempfile
+import os
+from pyrogram import filters
+from pyrogram.types import Message
+from DragMusic import userbot, app
+
+@app.on_message(filters.command("story", prefixes=["/", "!"]))
+async def story_download_handler(client, message: Message):
+    if len(message.command) < 3:
+        return await message.reply_text("Usage: /story <username or userid> <story_id>")
+    target = message.command[1]
+    try:
+        story_id = int(message.command[2])
+    except ValueError:
+        return await message.reply_text("Story ID must be an integer.")
+    try:
+        stories = await userbot.one.get_stories(target, [story_id])
+        # If only one story is returned, wrap in a list for uniformity
+        if not isinstance(stories, list):
+            stories = [stories]
+        sent = False
+        for s in stories:
+            temp_dir = tempfile.gettempdir()
+            if hasattr(s, "video") and s.video:
+                file = await userbot.one.download_media(s.video.file_id, file_name=os.path.join(temp_dir, f"story_{s.id}.mp4"))
+                await userbot.one.send_video(message.chat.id, file, caption=f"Story ID: {s.id}")
+                sent = True
+            elif hasattr(s, "photo") and s.photo:
+                file = await userbot.one.download_media(s.photo.file_id, file_name=os.path.join(temp_dir, f"story_{s.id}.jpg"))
+                await userbot.one.send_photo(message.chat.id, file, caption=f"Story ID: {s.id}")
+                sent = True
+        if sent:
+            await message.reply_text("Story sent to this chat!")
+        else:
+            await message.reply_text("No video or photo found in the specified story.")
+    except Exception as e:
+        await message.reply_text(f"Failed to download story: {e}") 

DragMusic/plugins/kuri/title.py

@@ -0,0 +1,38 @@
+"""
+Plugin to set a custom admin title in a supergroup using /title <user> <title>.
+Only group owner and admins can use this command.
+Usable by the bot (not userbot).
+"""
+
+from pyrogram import filters
+from pyrogram.types import Message
+from DragMusic import app
+
+async def is_admin(client, chat_id, user_id):
+    member = await client.get_chat_member(chat_id, user_id)
+    return member.status in ("administrator", "creator")
+
+@app.on_message(filters.command("title", prefixes=["/", "!"]) & filters.group)
+async def set_admin_title(client, message: Message):
+    if not await is_admin(client, message.chat.id, message.from_user.id):
+        await message.reply_text("Only group admins and the owner can use this command.")
+        return
+    if len(message.command) < 3:
+        await message.reply_text("Usage: /title <user> <title>")
+        return
+    user = message.command[1]
+    title = " ".join(message.command[2:])
+    try:
+        # Try to get user_id from mention, username, or ID
+        if message.reply_to_message:
+            user_id = message.reply_to_message.from_user.id
+        else:
+            try:
+                user_id = int(user)
+            except ValueError:
+                user_obj = await client.get_users(user)
+                user_id = user_obj.id
+        await client.set_administrator_title(message.chat.id, user_id, title)
+        await message.reply_text(f"Successfully set admin title to: {title}")
+    except Exception as e:
+        await message.reply_text(f"Failed to set admin title: {e}")