Squashed 'userbot/' content from commit 7fb0a0e

git-subtree-dir: userbot
git-subtree-split: 7fb0a0e179

images/icons.py

@@ -0,0 +1,110 @@
+import random
+import re
+from io import BytesIO
+
+import requests
+from bs4 import BeautifulSoup as BS
+from pyrogram import Client, filters
+from pyrogram.errors import RPCError
+from pyrogram.types import InputMediaPhoto, Message
+from utils.misc import modules_help, prefix
+
+
+@Client.on_message(filters.command('icon', prefix) & filters.me)
+async def search_icon(_, message: Message):
+    if len(message.command) != 2:
+        return await message.edit_text('Please provide some text to search icons from Flaticon.com.')
+    query = message.text.split(maxsplit=1)[1]
+
+    await message.edit_text('Searching for icons...')
+    search_query = query.replace(' ', '%20')
+    url = f'https://www.flaticon.com/search?word={search_query}'
+
+    try:
+        html_content = requests.get(url, timeout=10).text
+        soup = BS(html_content, 'html.parser')
+        results = soup.find_all(
+            'img',
+            src=re.compile(r'https://cdn-icons-png.flaticon.com/128/[0-9]+/[0-9]+.png'),
+        )
+
+        if not results:
+            return await message.edit('No results found.')
+
+        random.shuffle(results)
+        icons = []
+        for i in range(5):
+            icons.append(results[i]['src'].replace('128', '512'))
+
+        for icon in icons:
+            await message.reply_document(icon)
+
+        return await message.delete()
+    except Exception as e:
+        await message.edit(f'An error occurred: {e}')
+        print(f'Error: {e}')
+
+
+@Client.on_message(filters.command('freepik', prefix) & filters.me)
+async def freepik_search(client: Client, message: Message):
+    parts = message.text.split(' ', 1)
+    if len(parts) < 2:
+        await message.edit_text('Please provide a search query!')
+        return
+
+    query = parts[1]
+    limit = 5
+    if ' ; ' in query:
+        match, limit_str = query.split(' ; ', 1)
+        try:
+            limit = int(limit_str)
+        except ValueError:
+            await message.edit_text('Invalid limit! Using the default value of 5.')
+    else:
+        match = query
+
+    match = match.replace(' ', '%20')
+    await message.edit_text('Searching Freepik...')
+
+    try:
+        url = f'https://www.freepik.com/api/regular/search?locale=en&term={match}'
+        json_content = requests.get(url, timeout=10).json()
+        results = []
+        for i in json_content['items']:
+            results.append(i['preview']['url'])
+
+        if results is None:
+            return await message.edit_text('No results found.')
+
+        random.shuffle(results)
+        img_urls = results[:limit]
+
+        media_group = []
+        for img_url in img_urls:
+            icon = requests.get(img_url, timeout=10)
+            if icon.status_code == 200:
+                media_group.append(InputMediaPhoto(media=BytesIO(icon.content)))
+
+        if not media_group:
+            await message.edit_text('No images could be downloaded.')
+            return
+
+        try:
+            await client.send_media_group(chat_id=message.chat.id, media=media_group)
+        except RPCError:
+            await message.edit_text('Failed to send some images. Retrying individually...')
+            for media in media_group:
+                try:
+                    await message.reply_photo(photo=media.media)
+                except Exception as e:
+                    await message.edit_text(f'Error sending image: {e}')
+
+    except Exception as e:
+        await message.edit_text(f'Failed to fetch data: {e}')
+        print(f'Error: {e}')
+
+
+modules_help['icons'] = {
+    'icon [query]': 'Search for icons on Flaticon.',
+    'freepik [query] [limit]': 'Search for images on Freepik. Limit is optional and defaults to 5.',
+}