initial

bot/bot.py

@@ -0,0 +1,61 @@
+from telebot.async_telebot import AsyncTeleBot
+from telebot.types import Message
+import asyncio
+from secret import TOKEN
+import textbook
+import re
+from urllib.parse import urlparse, urlunparse
+import random
+
+HAS_LINK_RE = r'(https?:\/\/[^\s]+|www\.[^\s]+)'
+
+bot = AsyncTeleBot(TOKEN)
+
+
+def extract_links(text: str):
+    url_pattern = r'(https?://[^\s]+|www\.[^\s]+)'
+    links = re.findall(url_pattern, text)
+    return links
+
+
+def process_text(text: str) -> str | None:
+    links = extract_links(text)
+    updated_links = []
+    for link in links:
+        if any(keyword in link for keyword in ("youtu.be", "youtube.com")):
+            parsed_link = urlparse(link)
+            if "si=" in parsed_link.query:
+                query_arr = parsed_link.query.split("&")
+                new_query_arr = [ele for ele in query_arr if "si=" not in ele]
+                new_query = "&".join(new_query_arr)
+                updated_link = parsed_link._replace(query=new_query)
+                updated_links.append(urlunparse(updated_link))
+    if updated_links:
+        return random.choice(textbook.reactions).format(link="\n".join(updated_links))
+    return None
+
+
+@bot.message_handler(commands=["start"])
+async def start(msg: Message):
+    if msg.chat.type == "private":
+        await bot.send_message(chat_id=msg.chat.id, text=textbook.start_private)
+    elif msg.chat.type in ("group", "supergroup"):
+        await bot.send_message(chat_id=msg.chat.id, text=textbook.start_group)
+
+
+@bot.message_handler(func=lambda message: True)
+async def got_message(msg: Message):
+    if re.match(string=msg.text, pattern=HAS_LINK_RE):
+        fixed_reply = process_text(msg.text)
+        if fixed_reply:
+            await bot.reply_to(msg, fixed_reply)
+
+
+async def main():
+    a = asyncio.create_task(bot.polling(non_stop=True))
+    await a
+
+
+if __name__ == "__main__":
+    print("Bot started", flush=True)
+    asyncio.run(main())