Compare commits

...

45 Commits

Author SHA1 Message Date
github-actions[bot]
430f49f345 Updated modules.json after parse 2026-02-14 06:58:27 2026-02-14 06:58:27 +00:00
54229a8266 nvm 2026-02-14 09:54:25 +03:00
Zahar Vanilovv
6cb625acd8 Merge pull request #183 from MuRuLOSE/update-submodules_61ebb7fa472caa2abd4d6c7523bcf8830258167c
Update of repositories 2026-02-09 01:29:55
2026-02-09 17:25:41 +03:00
github-actions[bot]
bf03595075 Updated modules.json after parse 2026-02-09 01:29:33 2026-02-09 01:29:33 +00:00
github-actions[bot]
8bef604170 Added and updated repositories 2026-02-09 01:29:03 2026-02-09 01:29:03 +00:00
61ebb7fa47 Legacy version is here! 2026-02-08 20:17:50 +03:00
Zahar Vanilovv
2702cd2356 Merge pull request #182 from MuRuLOSE/update-submodules_ab5aaf579b3117a79d44591859b2cd30a6fcd5a1
Update of repositories 2026-02-08 01:59:39
2026-02-08 08:01:47 +03:00
github-actions[bot]
37f53375bc Updated modules.json after parse 2026-02-08 01:59:16 2026-02-08 01:59:16 +00:00
github-actions[bot]
23dade9b9e Added and updated repositories 2026-02-08 01:58:46 2026-02-08 01:58:46 +00:00
Macsim
ab5aaf579b Merge pull request #181 from MuRuLOSE/update-submodules_78a9b0cb76b2d495a5ff23a25614d91699a8757d
Update of repositories 2026-02-07 07:57:35
2026-02-07 11:05:32 +03:00
github-actions[bot]
170809519a Updated modules.json after parse 2026-02-07 07:57:13 2026-02-07 07:57:13 +00:00
78a9b0cb76 added full tds attribute 2026-02-07 10:54:30 +03:00
Macsim
c1d407f885 Merge pull request #180 from MuRuLOSE/update-submodules_7cda8f2685159651f9c2c9ff238215f2b5894d6c
Update of repositories 2026-02-07 07:41:22
2026-02-07 10:43:12 +03:00
github-actions[bot]
cf230d2922 Updated modules.json after parse 2026-02-07 07:41:02 2026-02-07 07:41:02 +00:00
7cda8f2685 fix f-strings while parsing 2026-02-07 10:35:19 +03:00
Macsim
92bdd1510d Merge pull request #179 from MuRuLOSE/update-submodules_a7b8c740b42879d9b3c0d577d275d5b42e4db254
Update of repositories 2026-02-07 01:23:55
2026-02-07 10:34:25 +03:00
github-actions[bot]
52f6fc53c3 Updated modules.json after parse 2026-02-07 01:23:38 2026-02-07 01:23:38 +00:00
github-actions[bot]
5c25eaad81 Added and updated repositories 2026-02-07 01:23:08 2026-02-07 01:23:08 +00:00
a7b8c740b4 Now index locking should be not a problem 2026-02-06 12:27:31 +03:00
44ca8633c6 Merge branch 'main' of https://github.com/MuRuLOSE/limoka 2026-02-06 12:03:00 +03:00
e076d4ab28 lang and dev_username hotfix 2026-02-06 12:02:48 +03:00
Macsim
634861943a Merge pull request #177 from MuRuLOSE/update-submodules_f68d97d09b6ca907775228791cde9446509954c8
Update of repositories 2026-02-06 01:25:04
2026-02-06 07:43:55 +03:00
github-actions[bot]
d20d019c32 Updated modules.json after parse 2026-02-06 01:24:40 2026-02-06 01:24:40 +00:00
github-actions[bot]
041e6ec5f7 Added and updated repositories 2026-02-06 01:24:12 2026-02-06 01:24:12 +00:00
f68d97d09b bolded some text 2026-02-05 20:30:35 +03:00
e624861b64 Blacklisted some module 2026-02-05 20:17:09 +03:00
Macsim
d3f035ec58 Merge pull request #176 from MuRuLOSE/update-submodules_5c2de322d31171b50a220dcbb9862aea611ab15a
Update of repositories 2026-02-05 01:23:39
2026-02-05 20:07:37 +03:00
Macsim
1ced5efae6 Update SpotifyMod.py 2026-02-05 20:01:13 +03:00
github-actions[bot]
3bf476c8a9 Updated modules.json after parse 2026-02-05 01:23:15 2026-02-05 01:23:15 +00:00
github-actions[bot]
abff91c013 Added and updated repositories 2026-02-05 01:22:45 2026-02-05 01:22:45 +00:00
5c2de322d3 new developer 2026-01-30 20:29:03 +03:00
5b2fe145ff b 2026-01-30 17:17:10 +03:00
64d8ceb354 clear histroy 2026-01-30 16:58:22 +03:00
537ae4c485 againfix 2026-01-30 16:50:55 +03:00
52018c8e14 fix not found PeerUser bcs of id 2026-01-30 15:42:57 +03:00
6f95292df1 hotfix again 2026-01-30 15:31:04 +03:00
4a5d4883a9 version bump 2026-01-30 15:22:55 +03:00
ce59b13c94 Merge branch 'main' of https://github.com/MuRuLOSE/limoka 2026-01-30 15:22:26 +03:00
91351cb83d hotfix (i am scared...) 2026-01-30 15:22:01 +03:00
Macsim
118a800856 Merge pull request #170 from MuRuLOSE/update-submodules_51bbb98711dbacc9d018d87706b7f28c77463cc4
Update of repositories 2026-01-30 11:59:59
2026-01-30 15:00:33 +03:00
github-actions[bot]
f14a99c640 Updated modules.json after parse 2026-01-30 11:59:36 2026-01-30 11:59:36 +00:00
github-actions[bot]
c76dcc0fc1 Added and updated repositories 2026-01-30 11:59:07 2026-01-30 11:59:07 +00:00
51bbb98711 Merge branch 'main' of https://github.com/MuRuLOSE/limoka 2026-01-30 14:56:57 +03:00
edd70ad020 бь 2026-01-30 14:56:48 +03:00
Macsim
2c33b08ea0 Merge pull request #169 from MuRuLOSE/update-submodules_88ab2657552874370ef8600b9655bbfe90c2decd
Update of repositories 2026-01-30 11:50:56
2026-01-30 14:52:28 +03:00
25 changed files with 70867 additions and 65605 deletions

View File

@@ -209,16 +209,14 @@ jobs:
notify_diffs:
runs-on: ubuntu-latest
if: |
(github.event_name == 'push' && github.ref == 'refs/heads/main') ||
(github.event_name == 'pull_request' && github.event.action == 'closed' && github.event.pull_request.merged == true)
if: github.event_name == 'push' && github.ref == 'refs/heads/main'
needs: parse
steps:
- name: Checkout repository
uses: actions/checkout@v4
with:
fetch-depth: ${{ env.GIT_DEPTH }}
- name: Configure Git for github-actions[bot]
fetch-depth: 0
- name: Configure Git
run: |
git config --global user.email "github-actions[bot]@users.noreply.github.com"
git config --global user.name "github-actions[bot]"
@@ -226,15 +224,18 @@ jobs:
uses: actions/setup-python@v5
with:
python-version: '3.9'
- name: Install Python dependencies
- name: Install dependencies
run: pip install aiohttp
- name: Send module diffs to channel
env:
TELEGRAM_BOT_TOKEN: ${{ secrets.TELEGRAM_BOT_TOKEN }}
TELEGRAM_CHAT_ID: ${{ secrets.TELEGRAM_CHAT_ID_UPDATE }}
run: |
git fetch origin main
python3 update_diffs.py --token ${{ secrets.TELEGRAM_BOT_TOKEN }} --chat_id ${{ secrets.TELEGRAM_CHAT_ID_UPDATE }} --base_commit HEAD~1
LAST_COMMIT_MESSAGE=$(git log -1 --pretty=%B)
if [[ "$LAST_COMMIT_MESSAGE" == Merge* ]]; then
python3 update_diffs.py --token ${{ secrets.TELEGRAM_BOT_TOKEN }} --chat_id ${{ secrets.TELEGRAM_CHAT_ID_UPDATE }} --base_commit HEAD~1
fi
backup:
runs-on: ubuntu-latest

589
Limoka.py
View File

@@ -2,10 +2,13 @@
# requires: whoosh cryptography
from collections import Counter, defaultdict
import shutil
from whoosh.index import create_in, open_dir
from whoosh.fields import Schema, TEXT, ID
from whoosh.qparser import QueryParser, OrGroup
from whoosh.query import FuzzyTerm, Wildcard
from whoosh.writing import LockError
import aiohttp
import random
import logging
@@ -14,26 +17,43 @@ import html
import json
import re
import asyncio
from typing import Union, List, Dict, Any, Optional
from typing import Iterable, Union, List, Dict, Any, Optional
import hashlib
from telethon.types import Message
from telethon.errors.rpcerrorlist import WebpageMediaEmptyError
from telethon import TelegramClient
from telethon.errors.rpcerrorlist import YouBlockedUserError
from telethon import functions
try:
from aiogram.utils.exceptions import BadRequest
except ImportError:
from aiogram.exceptions import TelegramBadRequest as BadRequest
from .. import utils, loader
from ..types import InlineCall
logger = logging.getLogger("Limoka")
__version__ = (1, 3, 1)
__version__ = (1, 4, 2)
WEIGHTS = {
"inline.token_obtainment": 15,
"main": 10,
"inline": 7,
"translations": 5,
"security": 3,
}
DEFAULT_WEIGHT = 1
BASE_DIR = os.getcwd()
def _get_lang_value(data: Dict[str, Any], lang: str) -> str:
if not isinstance(data, dict):
return str(data) if data else ""
return data.get(lang, data.get("default", data.get("en", "")))
class Search:
def __init__(self, query, ix):
self.schema = Schema(
@@ -54,6 +74,7 @@ class Search:
return list(set(result["path"] for result in results))
return []
class LimokaAPI:
async def fetch_json(self, base_url, path):
url = f"{base_url}{path}"
@@ -61,9 +82,11 @@ class LimokaAPI:
async with session.get(url) as response:
return json.loads(await response.text())
@loader.tds
class Limoka(loader.Module):
"""Modules are now in one place with easy searching!"""
strings = {
"name": "Limoka",
"wait": (
@@ -79,9 +102,7 @@ class Limoka(loader.Module):
"<b><emoji document_id=5418299289141004396>🧑‍💻</emoji> Developer:</b> {username}\n\n"
"<b><emoji document_id=5418376169055602355>🏷</emoji> Tags:</b> {tags}\n\n"
),
"found_body": (
"{commands}"
),
"found_body": ("{commands}"),
"found_footer": (
"\n<emoji document_id=5411143117711624172>🪄</emoji> <code>{prefix}dlm {url}{module_path}</code>"
),
@@ -166,8 +187,15 @@ class Limoka(loader.Module):
"hikkatrusted": "Hikka Trusted",
"nonactive": "Non-Active Repository",
"nonlongermaintained": "No Longer Maintained Repository",
"newbie": "Newbie"
}
"newbie": "Newbie",
},
"indexing_in_progress": (
"⚠️ Database is busy, "
"try again later. "
"If issue persists, try "
"removing limoka_index in the userbot's root folder. "
"If error persists again, report to developers"
),
}
strings_ru = {
"name": "Limoka",
@@ -183,9 +211,7 @@ class Limoka(loader.Module):
"<b><emoji document_id=5418299289141004396>🧑‍💻</emoji> Разработчик:</b> {username}\n\n"
"<b><emoji document_id=5418376169055602355>🏷</emoji> Теги:</b> {tags}\n\n"
),
"found_body": (
"{commands}"
),
"found_body": ("{commands}"),
"found_footer": (
"\n<emoji document_id=5411143117711624172>🪄</emoji> <code>{prefix}dlm {url}{module_path}</code>"
),
@@ -218,7 +244,7 @@ class Limoka(loader.Module):
(
"<emoji document_id=5188311512791393083>🔎</emoji> Limoka имеет лучший поиск*!\n"
"<i>* В сравнении с предыдущей версией Limoka</i>"
)
),
],
"inline404": "Не найдено",
"inline?": "Запрос слишком короткий / не найден",
@@ -274,8 +300,15 @@ class Limoka(loader.Module):
"hikkatrusted": "Hikka Trusted",
"nonactive": "Неактивный репозиторий",
"nonlongermaintained": "Неподдерживаемый репозиторий",
"newbie": "Новичок"
"newbie": "Новичок",
},
"indexing_in_progress": (
"⚠️ База данных занята, "
"попробуйте снова через несколько секунд. "
"Если ошибка сохраняется, попробуйте "
"удалить limoka_index в корневой папке юзербота. "
"Если ошибка сохраняется снова, сообщите разработчикам"
),
"_cls_doc": "Модули теперь в одном месте с простым и удобным поиском!",
}
@@ -303,24 +336,24 @@ class Limoka(loader.Module):
)
self.name = self.strings["name"]
self._invalid_banners = set()
self._service_bot_id = 8581621390
self._bot_username = "limoka_bbot"
self._base_url = self.config["limokaurl"]
# Search session states
self.SEARCH_STATES = {
"no_banner": "no_banner", # 404 - Нет баннера
"global_search": "global_search", # Глобальный поиск
"not_found": "not_found", # Не найдено (модуль)
"filter_select": "filter_select", # Выбор категорий (фильтров)
"no_banner": "no_banner", # 404 - No banner
"global_search": "global_search", # Global search
"not_found": "not_found", # Not found (module)
"filter_select": "filter_select", # Select categories (filters)
}
# State banners - placeholders for now
# State banners
self.state_banners = {
"no_banner": "https://raw.githubusercontent.com/MuRuLOSE/hikka-assets/refs/heads/main/Limoka%20-%20No%20banner.png",
"global_search": "https://raw.githubusercontent.com/MuRuLOSE/hikka-assets/main/Limoka%20-%20Global%20Search.png",
"not_found": "https://raw.githubusercontent.com/MuRuLOSE/hikka-assets/main/Limoka%20-%20Not%20Found.png",
"filter_select": "https://raw.githubusercontent.com/MuRuLOSE/hikka-assets/main/Limoka%20-%20Categories.png",
}
}
def _filter_newbies(self, modules: Dict[str, Any]) -> Dict[str, Any]:
"""Filter out modules which belong to repositories tagged as 'newbie'.
@@ -355,7 +388,7 @@ class Limoka(loader.Module):
current_index: int = 0,
) -> Dict[str, Any]:
"""Create a search session dictionary to track state across callbacks.
Args:
state: Current search state (one of SEARCH_STATES values)
query: Current search query
@@ -363,7 +396,7 @@ class Limoka(loader.Module):
results: Search results list
current_index: Index of current result being displayed
banner_url: Banner image URL for current state
Returns:
Dictionary containing the complete session state
"""
@@ -391,38 +424,49 @@ class Limoka(loader.Module):
else:
self.ix = open_dir("limoka_search")
self._history = self.pointer("history", [])
self.modules = await self.api.fetch_json(
self._base_url, "modules.json"
self.modules = (await self.api.fetch_json(self._base_url, "modules.json")).get(
"modules", {}
)
raw = (await self.api.fetch_json(
self._base_url, "repositories.json"
)).get("repositories", [])
self.repositories = {
repo["path"]: repo
for repo in raw
}
raw = (await self.api.fetch_json(self._base_url, "repositories.json")).get(
"repositories", []
)
self.repositories = {repo["url"]: repo for repo in raw}
# Apply newbie filter if enabled
try:
self.modules = self._filter_newbies(self.modules)
except Exception:
pass
self._bot_username = (await self.client.get_entity(self._service_bot_id)).username
await self._update_index()
self._service_bot_id = (await self.client.get_entity(self._bot_username)).id
loop = asyncio.get_running_loop()
self.ix_task = loop.run_in_executor(None, lambda: asyncio.run(self._update_index()))
if self.config["external_install_allowed"]:
try:
message = await self.client.get_messages(self._bot_username, limit=1)
if not message:
message = await self.client.send_message(self._service_bot_id, "/start")
message = await self.client.send_message(
self._bot_username, "/start"
)
await message.delete()
await self.client(
functions.messages.DeleteHistoryRequest(
peer=self._bot_username,
max_id=0,
just_clear=True,
revoke=True,
)
)
except YouBlockedUserError:
logger.warning(f"Please unblock {self._bot_username} to enable external installation feature. Or disable external_install_allowed in Limoka settings to get rid of this message.")
logger.warning(
f"Please unblock {self._bot_username} to enable external installation feature. Or disable external_install_allowed in Limoka settings to get rid of this message."
)
self._userbot_bot_username = (await self.inline.bot.get_me()).username
@loader.loop(interval=3600)
async def _update_modules_loop(self):
self.modules = await self.api.fetch_json(
self._base_url, "modules.json"
)
self.modules = await self.api.fetch_json(self._base_url, "modules.json")
# Re-apply newbie filter after modules refresh
try:
self.modules = self._filter_newbies(self.modules)
@@ -431,29 +475,56 @@ class Limoka(loader.Module):
await self._update_index()
async def _update_index(self):
writer = self.ix.writer()
modules_to_index = self._filter_newbies(self.modules)
for module_path, module_data in modules_to_index.items():
writer.add_document(
title=module_data["name"],
path=module_path,
content=module_data["name"] + " " + (module_data.get("description") or "" + " " + ((module_data.get("meta").get("developer") or "") if module_data.get("meta") else "")),
)
for func in module_data.get("commands", []):
for command, description in func.items():
writer.add_document(
title=module_data["name"],
path=module_path,
content=f"{command} {description}",
try:
writer = self.ix.writer()
modules_to_index = self._filter_newbies(self.modules)
for module_path, module_data in modules_to_index.items():
writer.add_document(
title=module_data["name"],
path=module_path,
content=module_data["name"]
+ " "
+ (
module_data.get("description")
or ""
+ " "
+ (
(module_data.get("meta").get("developer") or "")
if module_data.get("meta")
else ""
)
),
)
for func in module_data.get("commands", []):
for command, description in func.items():
writer.add_document(
title=module_data["name"],
path=module_path,
content=f"{command} {description}",
)
writer.commit()
except LockError:
folder = os.path.join(BASE_DIR, "limoka_search")
if os.path.commonpath([folder, BASE_DIR]) == BASE_DIR and os.path.exists(folder):
shutil.rmtree(folder)
await self._update_index()
else:
logger.error(
(
f"Skipping unsafe rmtree for {folder}. Please, report this to developer. ",
f"Debug info: folder={folder}, base_dir={BASE_DIR}, common_path={os.path.commonpath([folder, BASE_DIR])}, exists={os.path.exists(folder)}",
)
writer.commit()
)
async def _validate_url(self, url: str) -> Optional[str]:
if not url or url in self._invalid_banners:
return None
try:
async with aiohttp.ClientSession() as session:
async with session.head(url, timeout=5, allow_redirects=True) as response:
async with session.head(
url, timeout=5, allow_redirects=True
) as response:
if response.status != 200:
self._invalid_banners.add(url)
return None
@@ -466,24 +537,58 @@ class Limoka(loader.Module):
if url:
self._invalid_banners.add(url)
return None
def find_userbot(self, keys: Iterable[str]) -> str | None:
scores = defaultdict(int)
for key in keys:
parts = key.split(".")
for i in range(1, len(parts)):
prefix = ".".join(parts[:i])
suffix = ".".join(parts[i:])
weight = WEIGHTS.get(suffix, DEFAULT_WEIGHT)
scores[prefix] += weight
if not scores:
return None
return max(scores, key=scores.get)
@property
def user_lang(self) -> str:
self.db.get("heroku.translations", "lang")
userbot = self.find_userbot(self.db.keys())
if not userbot:
logger.warning(
"Cannot determine userbot type. "
"Probably not FTG-like Userbot? "
"Defaulting language to English. "
"If this is unexpected, please report to the module developer."
)
return "en"
return self.db.get(f"{userbot}.translations", "lang")
def generate_commands(self, module_info, lang: str = "en"):
commands = []
for i, func in enumerate(module_info.get("commands", []), 1):
for command, description in func.items():
emoji = self.strings["emojis"].get(i, "")
desc = description or self.strings["no_info"]
commands.append(
self.strings["command_template"].format(
prefix=self.get_prefix(),
command=html.escape(command.replace("cmd", "")),
emoji=emoji,
description=html.escape(desc),
)
for i, cmd in enumerate(module_info.get("new_commands", []), 1):
name = cmd.get("name", "")
desc_map = cmd.get("description", {})
emoji = self.strings["emojis"].get(i, "")
desc = _get_lang_value(desc_map, lang) or self.strings["no_info"]
commands.append(
self.strings["command_template"].format(
prefix=self.get_prefix(),
command=html.escape(name),
emoji=emoji,
description=html.escape(desc),
)
)
for i, handler in enumerate(module_info.get("inline_handlers", []), 1):
name = handler.get("name", "")
desc_map = handler.get("description", {})
@@ -509,12 +614,14 @@ class Limoka(loader.Module):
name = html.escape(module_info.get("name") or self.strings["no_info"])
cls_doc = module_info.get("cls_doc", {})
description = html.escape(
_get_lang_value(cls_doc, lang) or
_get_lang_value(module_info.get("description", ""), lang) or
self.strings["no_info"]
_get_lang_value(cls_doc, lang)
or _get_lang_value(module_info.get("description", ""), lang)
or self.strings["no_info"]
)
dev_username = html.escape(module_info["meta"].get("developer") or "Unknown")
raw_path = (
module_path if module_path is not None else module_info.get("path", "")
)
dev_username = html.escape(module_info["meta"].get("developer", "Unknown"))
raw_path = module_path if module_path is not None else module_info.get("path", "")
clean_module_path = (raw_path or "").replace("\\", "/")
commands = self.generate_commands(module_info, lang)
categories_text = ""
@@ -526,10 +633,12 @@ class Limoka(loader.Module):
)
if len(description) > 300:
description = description[:297] + ""
repo_key = "/".join(module_path.split("/")[:2]) if "/" in module_path else module_path
repo_key = (
"/".join(module_path.split("/")[:2]) if "/" in module_path else module_path
)
tags_list = []
for x in self.repositories:
if x == repo_key:
if x.replace("https://github.com/", "") == repo_key:
tags_list = self.repositories.get(x, {}).get("tags", [])
break
tags_text = ", ".join(self.strings["tags"].get(tag, tag) for tag in tags_list)
@@ -566,14 +675,12 @@ class Limoka(loader.Module):
)
return header, body_pages, footer, categories_text
def _build_navigation_markup(
self, session: Dict[str, Any]
) -> list:
def _build_navigation_markup(self, session: Dict[str, Any]) -> list:
result = session["results"]
index = session["current_index"]
query = session["query"]
filters = session["filters"]
page = index + 1
markup = [
[
@@ -585,7 +692,11 @@ class Limoka(loader.Module):
{"text": f"{page}/{len(result)}", "callback": self._inline_void},
{
"text": "" if index + 1 < len(result) else "🚫",
"callback": self._next_page if index + 1 < len(result) else self._inline_void,
"callback": (
self._next_page
if index + 1 < len(result)
else self._inline_void
),
"args": (session,) if index + 1 < len(result) else (),
},
],
@@ -614,60 +725,93 @@ class Limoka(loader.Module):
return markup
def _build_module_markup(
self, session: Dict[str, Any], body_pages: List[str], page_body: int, module_path: str
self,
session: Dict[str, Any],
body_pages: List[str],
page_body: int,
module_path: str,
) -> list:
result = session["results"]
index = session["current_index"]
query = session["query"]
filters = session["filters"]
markup = []
if len(body_pages) > 1:
markup.append([
{
"text": "◀️" if page_body > 0 else "🚫",
"callback": self._previous_body_page if page_body > 0 else self._inline_void,
"args": (session, module_path, page_body) if page_body > 0 else (),
},
{"text": f"Body {page_body + 1}/{len(body_pages)}", "callback": self._inline_void},
{
"text": "▶️" if page_body + 1 < len(body_pages) else "🚫",
"callback": self._next_body_page if page_body + 1 < len(body_pages) else self._inline_void,
"args": (session, module_path, page_body) if page_body + 1 < len(body_pages) else (),
},
])
markup.append(
[
{
"text": "◀️" if page_body > 0 else "🚫",
"callback": (
self._previous_body_page
if page_body > 0
else self._inline_void
),
"args": (
(session, module_path, page_body) if page_body > 0 else ()
),
},
{
"text": f"Body {page_body + 1}/{len(body_pages)}",
"callback": self._inline_void,
},
{
"text": "▶️" if page_body + 1 < len(body_pages) else "🚫",
"callback": (
self._next_body_page
if page_body + 1 < len(body_pages)
else self._inline_void
),
"args": (
(session, module_path, page_body)
if page_body + 1 < len(body_pages)
else ()
),
},
]
)
page = index + 1
markup.append([
{
"text": "" if index > 0 else "🚫",
"callback": self._previous_page if index > 0 else self._inline_void,
"args": (session,) if index > 0 else (),
},
{"text": f"{page}/{len(result)}", "callback": self._inline_void},
{
"text": "" if index + 1 < len(result) else "🚫",
"callback": self._next_page if index + 1 < len(result) else self._inline_void,
"args": (session,) if index + 1 < len(result) else (),
},
])
markup.append([
{
"text": "🔍 " + self.strings["filter_menu"].split(":")[0],
"callback": self._display_filter_menu,
"args": (session,),
},
{
"text": "🔄 " + self.strings["change_query"],
"callback": self._enter_query,
},
])
markup.append([
{
"text": self.strings["global_button"],
"callback": self._show_global_results,
"args": (session,),
},
])
markup.append(
[
{
"text": "" if index > 0 else "🚫",
"callback": self._previous_page if index > 0 else self._inline_void,
"args": (session,) if index > 0 else (),
},
{"text": f"{page}/{len(result)}", "callback": self._inline_void},
{
"text": "" if index + 1 < len(result) else "🚫",
"callback": (
self._next_page
if index + 1 < len(result)
else self._inline_void
),
"args": (session,) if index + 1 < len(result) else (),
},
]
)
markup.append(
[
{
"text": "🔍 " + self.strings["filter_menu"].split(":")[0],
"callback": self._display_filter_menu,
"args": (session,),
},
{
"text": "🔄 " + self.strings["change_query"],
"callback": self._enter_query,
},
]
)
markup.append(
[
{
"text": self.strings["global_button"],
"callback": self._show_global_results,
"args": (session,),
},
]
)
markup.append(
[{"text": self.strings.get("close", "❌ Close"), "action": "close"}]
)
@@ -698,7 +842,9 @@ class Limoka(loader.Module):
)
else:
if photo is not None:
await message_or_call.edit(text=text, reply_markup=markup, photo=photo)
await message_or_call.edit(
text=text, reply_markup=markup, photo=photo
)
else:
await message_or_call.edit(text=text, reply_markup=markup)
except (BadRequest, WebpageMediaEmptyError) as e:
@@ -719,8 +865,11 @@ class Limoka(loader.Module):
try:
query = session["query"]
filters = session["filters"]
lang = self.user_lang()
lang = self.user_lang
logger.info(
f"Displaying module: {module_path} for query: {query} with filters: {filters} in language: {lang}"
)
module_banner_raw = module_info.get("meta", {}).get("banner")
photo = await self._validate_url(module_banner_raw)
@@ -739,11 +888,11 @@ class Limoka(loader.Module):
current_body = body_pages[min(page_body, len(body_pages) - 1)]
full_message = header + current_body + footer + categories_text
markup = self._build_module_markup(session, body_pages, page_body, module_path)
await self._safe_display(
message_or_call, full_message, markup, photo
markup = self._build_module_markup(
session, body_pages, page_body, module_path
)
await self._safe_display(message_or_call, full_message, markup, photo)
except Exception as e:
logger.exception(f"Error in _display_module: {e}")
if isinstance(message_or_call, Message):
@@ -752,30 +901,45 @@ class Limoka(loader.Module):
await message_or_call.edit(self.strings["error_occurred"])
async def _previous_body_page(
self, call: InlineCall, session: Dict[str, Any], module_path: str, page_body: int
self,
call: InlineCall,
session: Dict[str, Any],
module_path: str,
page_body: int,
):
module_info = self.modules[module_path]
new_page_body = max(page_body - 1, 0)
await self._display_module(call, module_info, module_path, session, page_body=new_page_body)
await self._display_module(
call, module_info, module_path, session, page_body=new_page_body
)
async def _next_body_page(
self, call: InlineCall, session: Dict[str, Any], module_path: str, page_body: int
self,
call: InlineCall,
session: Dict[str, Any],
module_path: str,
page_body: int,
):
module_info = self.modules[module_path]
query = session["query"]
filters = session["filters"]
header, body_pages, footer, categories_text = self._format_module_content(
module_info, query, filters, include_categories=True, module_path=module_path, lang=self.user_lang()
module_info,
query,
filters,
include_categories=True,
module_path=module_path,
lang=self.user_lang,
)
new_page_body = min(page_body + 1, len(body_pages) - 1)
await self._display_module(call, module_info, module_path, session, page_body=new_page_body)
await self._display_module(
call, module_info, module_path, session, page_body=new_page_body
)
async def _display_filter_menu(
self, call: InlineCall, session: Dict[str, Any]
):
async def _display_filter_menu(self, call: InlineCall, session: Dict[str, Any]):
query = session["query"]
current_filters = session["filters"]
categories = current_filters.get("category", [])
filters_text = self.strings["selected_categories"].format(
categories=(
@@ -812,14 +976,14 @@ class Limoka(loader.Module):
[{"text": self.strings.get("close", "❌ Close"), "action": "close"}],
]
text = self.strings["filter_menu"].format(query=query) + f"\n{filters_text}"
await call.edit(text, reply_markup=markup, photo=self._get_banner_for_state("filter_select"))
await call.edit(
text, reply_markup=markup, photo=self._get_banner_for_state("filter_select")
)
async def _select_category(
self, call: InlineCall, session: Dict[str, Any]
):
async def _select_category(self, call: InlineCall, session: Dict[str, Any]):
query = session["query"]
current_filters = session["filters"]
all_categories = set()
for module_data in self.modules.values():
all_categories.update(module_data.get("category", ["No category"]))
@@ -849,7 +1013,7 @@ class Limoka(loader.Module):
)
if cat in selected_categories:
button_text = "" + button_text
# Create new session with updated filters
new_session = session.copy()
row.append(
@@ -882,7 +1046,7 @@ class Limoka(loader.Module):
):
query = session["query"]
current_filters = session["filters"]
new_filters = current_filters.copy()
selected_categories = new_filters.get("category", [])
if category in selected_categories:
@@ -910,7 +1074,7 @@ class Limoka(loader.Module):
):
query = session["query"]
filters = session["filters"]
searcher = Search(query.lower(), self.ix)
try:
result = searcher.search_module()
@@ -972,7 +1136,7 @@ class Limoka(loader.Module):
return
module_path = filtered_result[0]
module_info = self.modules[module_path]
# Create session for displaying module
display_session = self._create_search_session(
state=self.SEARCH_STATES["global_search"],
@@ -981,9 +1145,7 @@ class Limoka(loader.Module):
results=filtered_result,
current_index=0,
)
await self._display_module(
call, module_info, module_path, display_session, 0
)
await self._display_module(call, module_info, module_path, display_session, 0)
async def _enter_query_handler(
self, call_or_query, query: Optional[str] = None, *args, **kwargs
@@ -1063,7 +1225,7 @@ class Limoka(loader.Module):
return
module_path = result[0]
module_info = self.modules[module_path]
# Create session for displaying module
display_session = self._create_search_session(
state=self.SEARCH_STATES["global_search"],
@@ -1087,11 +1249,13 @@ class Limoka(loader.Module):
{
"text": self.strings["back_to_results"],
"callback": self._show_results,
"args": (self._create_search_session(
state=self.SEARCH_STATES["global_search"],
query=query or "",
filters={},
),),
"args": (
self._create_search_session(
state=self.SEARCH_STATES["global_search"],
query=query or "",
filters={},
),
),
}
],
[
@@ -1105,7 +1269,7 @@ class Limoka(loader.Module):
async def _show_global_results(self, call: InlineCall, session: Dict[str, Any]):
query = session["query"]
searcher = Search(query.lower(), self.ix)
try:
result = searcher.search_module()
@@ -1134,7 +1298,7 @@ class Limoka(loader.Module):
if not info:
continue
name = info.get("name", "Unknown")
global_session = self._create_search_session(
state=self.SEARCH_STATES["global_search"],
query=query,
@@ -1160,47 +1324,37 @@ class Limoka(loader.Module):
self, call: InlineCall, module_path: str, session: Dict[str, Any]
):
module_info = self.modules[module_path]
await self._display_module(
call, module_info, module_path, session, 0
)
await self._display_module(call, module_info, module_path, session, 0)
async def _next_page(
self, call: InlineCall, session: Dict[str, Any]
):
async def _next_page(self, call: InlineCall, session: Dict[str, Any]):
result = session["results"]
index = session["current_index"]
if index + 1 >= len(result):
await call.answer(self.strings["last_page"])
return
index += 1
module_path = result[index]
module_info = self.modules[module_path]
new_session = session.copy()
new_session["current_index"] = index
await self._display_module(
call, module_info, module_path, new_session, 0
)
await self._display_module(call, module_info, module_path, new_session, 0)
async def _previous_page(
self, call: InlineCall, session: Dict[str, Any]
):
async def _previous_page(self, call: InlineCall, session: Dict[str, Any]):
result = session["results"]
index = session["current_index"]
if index - 1 < 0:
await call.answer(self.strings["first_page"])
return
index -= 1
module_path = result[index]
module_info = self.modules[module_path]
new_session = session.copy()
new_session["current_index"] = index
await self._display_module(
call, module_info, module_path, new_session, 0
)
await self._display_module(call, module_info, module_path, new_session, 0)
async def _inline_void(self, call: InlineCall):
await call.answer()
@@ -1209,6 +1363,11 @@ class Limoka(loader.Module):
async def limokacmd(self, message: Message):
"""[query / nothing] - Search modules"""
args = utils.get_args_raw(message)
lock_path = os.path.join(BASE_DIR, "limoka_search", "index.lock")
if os.path.exists(lock_path):
await utils.answer(message, self.strings["indexing_in_progress"])
return
if not args:
markup = [
[
@@ -1233,7 +1392,7 @@ class Limoka(loader.Module):
text=self.strings["start_search_form"],
message=message,
reply_markup=markup,
photo=self._get_banner_for_state("global_search")
photo=self._get_banner_for_state("global_search"),
)
return
history = self.get("history", [])
@@ -1258,7 +1417,7 @@ class Limoka(loader.Module):
return await utils.answer(message, self.strings["404"].format(query=args))
module_path = result[0]
module_info = self.modules[module_path]
# Create session for displaying module
display_session = self._create_search_session(
state=self.SEARCH_STATES["global_search"],
@@ -1267,7 +1426,9 @@ class Limoka(loader.Module):
results=result,
current_index=0,
)
await self._display_module(message, module_info, module_path, display_session, 0)
await self._display_module(
message, module_info, module_path, display_session, 0
)
async def _show_global_form(self, call: InlineCall, message: Message):
markup = [
@@ -1298,12 +1459,12 @@ class Limoka(loader.Module):
self, call: InlineCall, query: str, message: Message, *args, **kwargs
):
global_session = self._create_search_session(
state=self.SEARCH_STATES["global_search"],
query=query,
filters={},
results=[],
current_index=0,
) # idk what is that crap but it works lol
state=self.SEARCH_STATES["global_search"],
query=query,
filters={},
results=[],
current_index=0,
) # idk what is that crap but it works lol
if len(query) <= 1:
await call.edit(
self.strings["?"],
@@ -1431,16 +1592,20 @@ class Limoka(loader.Module):
elif hasattr(message.from_id, "channel_id"):
sender_id = message.from_id.channel_id
if sender_id != self._service_bot_id:
logger.debug("Message not from official bot, ignoring")
# logger.debug("Message not from official bot, ignoring")
return
if not self.config["external_install_allowed"]:
return
try:
clean_text = getattr(message, "raw_text", None) or getattr(
message, "message", None
) or message.text or ""
clean_text = (
getattr(message, "raw_text", None)
or getattr(message, "message", None)
or message.text
or ""
)
if message.entities:
from html import unescape
clean_text = unescape(clean_text)
clean_text = re.sub(r"<[^>]+>", "", clean_text)
match = re.search(r"#limoka:([^\s\"'<>]+)", clean_text)
@@ -1469,25 +1634,37 @@ class Limoka(loader.Module):
if not found:
logger.warning(f"Module not found after cleanup: {module_path}")
await utils.answer(
message, self.strings["watcher_module_not_found"].format(path=html.escape(module_path))
message,
self.strings["watcher_module_not_found"].format(
path=html.escape(module_path)
),
)
return
try:
import base64
from cryptography.hazmat.primitives.asymmetric import ed25519
PUB_KEY_B64 = "MCowBQYDK2VwAyEA1ltSnqtf3pGBuctuAYqHivCXsaRtKOVxavai7yin7ZE="
PUB_KEY_B64 = (
"MCowBQYDK2VwAyEA1ltSnqtf3pGBuctuAYqHivCXsaRtKOVxavai7yin7ZE="
)
der_bytes = base64.b64decode(PUB_KEY_B64)
raw_pubkey = der_bytes[-32:]
module_url = self.config["limokaurl"] + module_path
async with aiohttp.ClientSession() as session:
async with session.get(module_url, timeout=10) as resp:
if resp.status != 200:
logger.error(f"Failed to fetch module for verification: {module_url} (HTTP {resp.status})")
await utils.answer(message, self.strings["watcher_loader_missing"])
logger.error(
f"Failed to fetch module for verification: {module_url} (HTTP {resp.status})"
)
await utils.answer(
message, self.strings["watcher_loader_missing"]
)
return
module_bytes = await resp.read()
sha256 = hashlib.sha256(module_bytes).hexdigest()
public_key = ed25519.Ed25519PublicKey.from_public_bytes(raw_pubkey)
public_key = ed25519.Ed25519PublicKey.from_public_bytes(
raw_pubkey
)
signature = bytes.fromhex(signature_hex)
signed_payload = f"{module_path}|{sha256}".encode()
public_key.verify(signature, signed_payload)
@@ -1511,21 +1688,27 @@ class Limoka(loader.Module):
if status:
try:
bot_peer = await self.client.get_entity(self._service_bot_id)
await self.client.send_message(bot_peer, f"#limoka:sucsess:{message.id}")
await self.client.send_message(
bot_peer, f"#limoka:sucsess:{message.id}"
)
except Exception as e:
logger.error(f"Failed to send success confirmation: {e}")
else:
logger.error(f"Installation failed with status: {status}")
try:
bot_peer = await self.client.get_entity(self._service_bot_id)
await self.client.send_message(bot_peer, f"#limoka:failed:{message.id}")
await self.client.send_message(
bot_peer, f"#limoka:failed:{message.id}"
)
except Exception as e:
logger.error(f"Failed to send failure notification: {e}")
except Exception as e:
logger.exception(f"CRITICAL ERROR in secure_install_watcher: {e}")
try:
await utils.answer(message, self.strings["watcher_critical"].format(error=str(e)[:100]))
await utils.answer(
message, self.strings["watcher_critical"].format(error=str(e)[:100])
)
await asyncio.sleep(5)
await message.delete()
except Exception:
pass
pass

1420
LimokaLegacy.py Normal file

File diff suppressed because it is too large Load Diff

View File

@@ -23,4 +23,4 @@ class K(loader.Module):
"""K"""
raise Exception("Testing error handling")
await utils.answer(message, "K")
# why FUCK YOU BILL GATES
# why FUCK YOU BILL GATES ЛЛЛЛ

View File

@@ -3,7 +3,7 @@
# This software is released under the MIT License.
# https://opensource.org/licenses/MIT
__version__ = (5, 8, 1) #фыр
__version__ = (6, 1, 0) #фыр
# meta developer: @SenkoGuardianModules
@@ -22,12 +22,10 @@ import socket
import base64
import uuid
import json
from PIL import Image
import asyncio
import logging
import tempfile
import aiohttp
from datetime import datetime
from markdown_it import MarkdownIt
import pytz
@@ -41,6 +39,8 @@ except ImportError:
GOOGLE_AVAILABLE = False
google_exceptions = None
from PIL import Image
from datetime import datetime
from telethon import types as tg_types
from telethon.tl.types import Message, DocumentAttributeFilename, DocumentAttributeSticker
from telethon.utils import get_display_name, get_peer_id
@@ -59,6 +59,7 @@ logger = logging.getLogger(__name__)
DB_HISTORY_KEY = "gemini_conversations_v4"
DB_GAUTO_HISTORY_KEY = "gemini_gauto_conversations_v1"
DB_IMPERSONATION_KEY = "gemini_impersonation_chats"
DB_PRESETS_KEY = "gemini_prompt_presets"
GEMINI_TIMEOUT = 840
MAX_FFMPEG_SIZE = 90 * 1024 * 1024
DB_KEY_MAP_KEY = "gemini_key_model_map"
@@ -84,7 +85,15 @@ class Gemini(loader.Module):
"cfg_google_search_doc": "Включить поиск Google (Grounding) для актуальной информации.",
"cfg_image_model_doc": "Модель Gemini для генерации изображений (например: gemini-2.5-flash-image).",
"cfg_inline_pagination_doc": "Использовать инлайн-кнопки для длинных ответов.",
"no_api_key": '❗️ <b>Api ключ(и) не настроен(ы).</b>\nПолучить Api ключ можно <a href="https://aistudio.google.com/app/apikey">здесь</a>.\n<b>Добавьте ключ(и) в конфиге модуля:</b> <code>.cfg gemini api_key</code>',
"no_api_key": (
'❗️ <b>Api ключ(и) не настроен(ы).</b>\nПолучить Api ключ можно <a href="https://aistudio.google.com/app/apikey">здесь</a>.\n'
'<b>Добавьте ключ(и) в конфиге модуля:</b> <code>.cfg gemini api_key</code>\n'
'Так же можно использовать провайдера Openrouter <code>.cfg gemini provider</code>\n'
' Получить Openrouter ключ можно <a href="https://openrouter.ai/settings/keys">здесь</a>'
),
"no_api_key_Openrouter": '❗️ <b>API ключ для OpenRouter не настроен.</b>\nПолучить ключ можно <a href="https://openrouter.ai/settings/keys">здесь</a>.\n<b>Добавьте ключ в конфиге модуля:</b> <code>.cfg gemini Openrouter_api_key</code>',
"invalid_api_key_Openrouter": '❗️ <b>Предоставленный API ключ OpenRouter недействителен.</b>\nУбедитесь, что он правильно скопирован из <a href="https://openrouter.ai/settings/keys">OpenRouter</a>.',
"gmodel_list_title_Openrouter": "📋 <b>Доступные модели OpenRouter:</b>",
"invalid_api_key": '❗️ <b>Предоставленный API ключ недействителен.</b>\nУбедитесь, что он правильно скопирован из <a href="https://aistudio.google.com/app/apikey">Google AI Studio</a> и что для него включен Gemini API.',
"all_keys_exhausted": "❗️ <b>Все доступные API ключи ({}) исчерпали свою квоту.</b>\nПопробуйте позже или добавьте новые ключи в конфиге: <code>.cfg gemini api_key</code>",
"no_prompt_or_media": "⚠️ <i>Нужен текст или ответ на медиа/файл.</i>",
@@ -150,6 +159,21 @@ class Gemini(loader.Module):
"gmodel_no_models": "⚠️ Не удалось получить список моделей.",
"gmodel_list_error": "❗️ Ошибка получения списка: {}",
"gimg_process": "<emoji document_id=5325547803936572038>✨</emoji> <b>Генерация...</b>\n🧠 <i>Модель: {model}</i>",
"gprompt_usage": " <b>Использование:</b>\n<code>.gprompt <текст/пресет></code> — установить.\n<code>.gprompt -c</code> — очистить.\n<code>.gpresets</code> — база пресетов.",
"gpresets_usage": (
" <b>Управление пресетами:</b>\n"
"• <code>.gpresets save [Имя] текст</code> — сохранить (имя в скобках, если с пробелами).\n"
"• <code>.gpresets load 1</code> или <code>имя</code> — загрузить по номеру/имени.\n"
"• <code>.gpresets del 1</code> или <code>имя</code> — удалить.\n"
"• <code>.gpresets list</code> — список."
),
"gpreset_loaded": "✅ <b>Установлен пресет:</b> [<code>{}</code>]\nДлина: {} симв.",
"gpreset_saved": "💾 <b>Пресет сохранен!</b>\n🏷 <b>Имя:</b> {}\n№ <b>Индекс:</b> {}",
"gpreset_deleted": "🗑 <b>Пресет удален:</b> {}",
"gpreset_not_found": "🚫 Пресет с таким именем или индексом не найден.",
"gpreset_list_head": "📋 <b>Ваши пресеты:</b>\n",
"gpreset_empty": "📂 Список пресетов пуст.",
}
TEXT_MIME_TYPES = {
"text/plain", "text/markdown", "text/html", "text/css", "text/csv",
@@ -159,6 +183,8 @@ class Gemini(loader.Module):
def __init__(self):
self.config = loader.ModuleConfig(
loader.ConfigValue("api_key", "", self.strings["cfg_api_key_doc"], validator=loader.validators.Hidden()),
loader.ConfigValue("Openrouter_api_key", "", "API Key от OpenRouter (получить <a href='https://openrouter.ai/settings/keys'>тут</a>).", validator=loader.validators.Hidden()),
loader.ConfigValue("provider", "google", "Провайдер API: 'google' или 'openrouter'.", validator=loader.validators.Choice(["google", "openrouter"])),
loader.ConfigValue("model_name", "gemini-2.5-flash", self.strings["cfg_model_name_doc"]),
loader.ConfigValue("interactive_buttons", True, self.strings["cfg_buttons_doc"], validator=loader.validators.Boolean()),
loader.ConfigValue("system_instruction", "", self.strings["cfg_system_instruction_doc"], validator=loader.validators.String()),
@@ -184,6 +210,7 @@ class Gemini(loader.Module):
loader.ConfigValue("inline_pagination", False, self.strings["cfg_inline_pagination_doc"], validator=loader.validators.Boolean()),
loader.ConfigValue("image_model_name", "gemini-2.5-flash-image", self.strings["cfg_image_model_doc"]),
)
self.prompt_presets = []
self.conversations = {}
self.gauto_conversations = {}
self.last_requests = {}
@@ -211,6 +238,9 @@ class Gemini(loader.Module):
return
self.current_api_key_index = 0
self.conversations = self._load_history_from_db(DB_HISTORY_KEY)
self.prompt_presets = self.db.get(self.strings["name"], DB_PRESETS_KEY, [])
if isinstance(self.prompt_presets, dict):
self.prompt_presets = [{"name": k, "content": v} for k, v in self.prompt_presets.items()]
self.gauto_conversations = self._load_history_from_db(DB_GAUTO_HISTORY_KEY)
self.impersonation_chats = set(self.db.get(self.strings["name"], DB_IMPERSONATION_KEY, []))
if not self.api_keys:
@@ -315,7 +345,7 @@ class Gemini(loader.Module):
final_parts.insert(0, types.Part(text=full_prompt_text))
return final_parts, warnings
async def _send_to_gemini(self, message, parts: list, regeneration: bool=False, call: InlineCall=None, status_msg=None, chat_id_override: int=None, impersonation_mode: bool=False, use_url_context: bool=False, display_prompt: str=None):
async def _send_to_gemini(self, message, parts: list, regeneration: bool=False, call: InlineCall=None, status_msg=None, chat_id_override: int=None, impersonation_mode: bool=False, use_url_context: bool=False, display_prompt: str=None):
msg_obj = None
if regeneration:
chat_id = chat_id_override; base_message_id = message
@@ -323,6 +353,74 @@ class Gemini(loader.Module):
except Exception: msg_obj = None
else:
chat_id = utils.get_chat_id(message); base_message_id = message.id; msg_obj = message
if self.config["provider"] == "openrouter":
if regeneration:
current_turn_parts, request_text_for_display = self.last_requests.get(f"{chat_id}:{base_message_id}", (parts, "[регенерация]"))
else:
current_turn_parts = parts
user_text_from_parts = " ".join([p.text for p in parts if hasattr(p, "text") and p.text])
request_text_for_display = display_prompt or user_text_from_parts or "[медиа-запрос]"
self.last_requests[f"{chat_id}:{base_message_id}"] = (current_turn_parts, request_text_for_display)
try:
sys_instruct = self.config["system_instruction"] or None
if impersonation_mode:
my_name = get_display_name(self.me)
chat_history_text = await self._get_recent_chat_text(chat_id)
sys_instruct = self.config["impersonation_prompt"].format(my_name=my_name, chat_history=chat_history_text)
history_key = "global_context" if (self.config.get("global_memory") and not impersonation_mode) else str(chat_id)
raw_hist = self._get_structured_history(history_key, gauto=impersonation_mode)
if regeneration and raw_hist: raw_hist = raw_hist[:-2]
openai_messages = self._convert_google_history_to_openai(raw_hist, sys_instruct)
content_list = []
for p in current_turn_parts:
if hasattr(p, "text") and p.text:
content_list.append({"type": "text", "text": p.text})
elif hasattr(p, "inline_data") and p.inline_data:
mime = p.inline_data.mime_type
data = p.inline_data.data
if mime.startswith("image/"):
b64_img = base64.b64encode(data).decode("utf-8")
content_list.append({
"type": "image_url",
"image_url": {"url": f"data:{mime};base64,{b64_img}"}
})
if not content_list:
content_list = request_text_for_display
openai_messages.append({"role": "user", "content": content_list})
target_model = self.config["model_name"]
result_text = await self._send_to_Openrouter_api(target_model, openai_messages, self.config["temperature"])
if self._is_memory_enabled(str(chat_id)):
self._update_history(history_key, current_turn_parts, result_text, regeneration, msg_obj, gauto=impersonation_mode)
if impersonation_mode: return result_text
hist_len = len(self._get_structured_history(history_key)) // 2
mem_ind_fmt = self.strings.get("memory_status_global", self.strings["memory_status"])
if self.config.get("global_memory"):
mem_ind = mem_ind_fmt.format(hist_len)
else:
mem_ind = self.strings["memory_status"].format(hist_len, self.config["max_history_length"])
model_info = f"<i>OpenRouter: <code>{target_model}</code></i>"
response_html = self._markdown_to_html(result_text)
formatted_body = self._format_response_with_smart_separation(response_html)
question_html = f"<blockquote>{utils.escape_html(request_text_for_display[:200])}</blockquote>"
text_to_send = f"{mem_ind}\n{model_info}\n\n{self.strings['question_prefix']}\n{question_html}\n\n{self.strings['response_prefix']}\n{formatted_body}"
buttons = self._get_inline_buttons(chat_id, base_message_id) if self.config["interactive_buttons"] else None
if len(text_to_send) > 4096:
file = io.BytesIO(result_text.encode("utf-8")); file.name = "Gemini_response.txt"
if call: await self.client.send_file(call.chat_id, file, caption="Response too long", reply_to=call.message_id)
elif status_msg:
await status_msg.delete()
await self.client.send_file(chat_id, file, caption="Response too long", reply_to=base_message_id)
else:
if call: await call.edit(text_to_send, reply_markup=buttons)
elif status_msg: await utils.answer(status_msg, text_to_send, reply_markup=buttons)
return ""
except Exception as e:
error_text = self._handle_error(e)
if impersonation_mode: logger.error(f"Gauto/Openrouter error: {error_text}")
elif call: await call.edit(error_text)
elif status_msg: await utils.answer(status_msg, error_text)
return None
api_key_str = self.config["api_key"]
self.api_keys = [k.strip() for k in api_key_str.split(",") if k.strip()] if api_key_str else []
if not self.api_keys:
@@ -616,32 +714,39 @@ class Gemini(loader.Module):
@loader.command()
async def gprompt(self, message: Message):
"""[текст / -c / ответ на файл][-c (очистить)] / (ничего. увидеть промпт) Установить системный промпт (инструкцию/system_instruction)."""
"""<текст/-c/ответ на файл>Установить промпт."""
args = utils.get_args_raw(message)
reply = await message.get_reply_message()
if args == "-c":
self.config["system_instruction"] = ""
return await utils.answer(message, self.strings["gprompt_cleared"])
new_p = None
if reply and reply.file:
new_prompt = None
preset = self._find_preset(args)
if preset:
new_prompt = preset['content']
elif reply and reply.file:
if reply.file.size > 1024 * 1024:
return await utils.answer(message, self.strings["gprompt_file_too_big"])
try:
data = await self.client.download_file(reply.media, bytes)
try: new_p = data.decode("utf-8")
file_data = await self.client.download_file(reply.media, bytes)
try: new_prompt = file_data.decode("utf-8")
except UnicodeDecodeError: return await utils.answer(message, self.strings["gprompt_not_text"])
except Exception as e: return await utils.answer(message, self.strings["gprompt_file_error"].format(e))
elif args: new_p = args
if new_p:
self.config["system_instruction"] = new_p
return await utils.answer(message, self.strings["gprompt_updated"].format(len(new_p)))
cur = self.config["system_instruction"]
if not cur: return await utils.answer(message, self.strings["gprompt_usage"])
if len(cur) > 4000:
file = io.BytesIO(cur.encode("utf-8")); file.name = "system_instruction.txt"
except Exception as e:
return await utils.answer(message, self.strings["gprompt_file_error"].format(e))
elif args:
new_prompt = args
if new_prompt is not None:
self.config["system_instruction"] = new_prompt
return await utils.answer(message, self.strings["gprompt_updated"].format(len(new_prompt)))
current_prompt = self.config["system_instruction"]
if not current_prompt:
return await utils.answer(message, self.strings["gprompt_usage"])
if len(current_prompt) > 4000:
file = io.BytesIO(current_prompt.encode("utf-8"))
file.name = "system_instruction.txt"
await utils.answer(message, self.strings["gprompt_current"], file=file)
else:
await utils.answer(message, f"{self.strings['gprompt_current']}\n<code>{utils.escape_html(cur)}</code>")
await utils.answer(message, f"{self.strings['gprompt_current']}\n<code>{utils.escape_html(current_prompt)}</code>")
@loader.command()
async def gauto(self, message: Message):
@@ -700,6 +805,64 @@ class Gemini(loader.Module):
else:
await utils.answer(message, self.strings["gclear_usage"])
@loader.command()
async def gpresets(self, message: Message):
"""<save/load/del/list> — Управление пресетами (профилями)."""
args = utils.get_args_raw(message)
if not args: return await utils.answer(message, self.strings["gpresets_usage"])
match = re.match(r"^(\w+)(?:\s+\[(.+?)\]|\s+(\S+))?(?:\s+(.*))?$", args, re.DOTALL)
if not match: return await utils.answer(message, self.strings["gpresets_usage"])
action = match.group(1).lower()
name = match.group(2) or match.group(3)
content = match.group(4)
if action == "list":
if not self.prompt_presets: return await utils.answer(message, self.strings["gpreset_empty"])
text = self.strings["gpreset_list_head"]
for idx, p in enumerate(self.prompt_presets, 1):
text += f"<b>{idx}.</b> <code>{p['name']}</code> ({len(p['content'])} симв.)\n"
return await utils.answer(message, text)
if action == "save":
if not name: return await utils.answer(message, "❌ Укажите имя: <code>.gpresets save [Имя] текст</code>")
reply = await message.get_reply_message()
if not content and reply:
if reply.text: content = reply.text
elif reply.file:
try: content = (await self.client.download_file(reply.media, bytes)).decode("utf-8", errors="ignore")
except: pass
if not content: return await utils.answer(message, "❌ Нет текста для сохранения.")
existing = self._find_preset(name)
if existing:
existing['content'] = content
else:
self.prompt_presets.append({"name": name, "content": content})
self.db.set(self.strings["name"], DB_PRESETS_KEY, self.prompt_presets)
await utils.answer(message, self.strings["gpreset_saved"].format(name, len(self.prompt_presets)))
elif action == "load":
target = self._find_preset(name)
if not target: return await utils.answer(message, self.strings["gpreset_not_found"])
self.config["system_instruction"] = target['content']
await utils.answer(message, self.strings["gpreset_loaded"].format(target['name'], len(target['content'])))
elif action == "del":
target = self._find_preset(name)
if not target: return await utils.answer(message, self.strings["gpreset_not_found"])
self.prompt_presets.remove(target)
self.db.set(self.strings["name"], DB_PRESETS_KEY, self.prompt_presets)
await utils.answer(message, self.strings["gpreset_deleted"].format(target['name']))
else:
await utils.answer(message, self.strings["gpresets_usage"])
def _find_preset(self, query):
"Ищет пресет по номеру (строка '1') или имени."
if not query: return None
if str(query).isdigit():
idx = int(query) - 1
if 0 <= idx < len(self.prompt_presets):
return self.prompt_presets[idx]
for p in self.prompt_presets:
if p['name'].lower() == str(query).lower():
return p
return None
@loader.command()
async def gmemdel(self, message: Message):
"""[N] — удалить последние N пар сообщений из памяти."""
@@ -814,25 +977,66 @@ class Gemini(loader.Module):
@loader.command()
async def gmodel(self, message: Message):
"""[model или пусто] — Узнать/сменить модель. -s — список доступных моделей в файле."""
args = utils.get_args_raw(message).strip().lower()
if '-s' in args:
if not self.api_keys: return await utils.answer(message, self.strings['no_api_key'])
sts = await utils.answer(message, self.strings["processing"])
"""[model] [-s] — Узнать/сменить модель. -s — список."""
args_raw = utils.get_args_raw(message).strip()
args_list = args_raw.split()
is_list_request = "-s" in [arg.lower() for arg in args_list]
provider = self.config["provider"]
if is_list_request:
status_msg = await utils.answer(message, self.strings["processing"])
try:
client = genai.Client(api_key=self.api_keys[0])
models = await asyncio.to_thread(client.models.list)
txt = "\n".join([f"• <code>{m.name.split('/')[-1]}</code> ({m.display_name})" for m in models])
f = io.BytesIO((self.strings["gmodel_list_title"] + "\n" + txt).encode('utf-8'))
f.name = "models_list.txt"
await self.client.send_file(message.chat_id, file=f, caption="📋 Список доступных моделей", reply_to=message.id)
await sts.delete()
except Exception as e: await utils.answer(sts, self.strings["gmodel_list_error"].format(self._handle_error(e)))
if provider == "openrouter":
api_key = self.config["Openrouter_api_key"]
if not api_key: return await utils.answer(status_msg, self.strings['no_api_key_Openrouter'])
async with aiohttp.ClientSession() as session:
async with session.get(
"https://openrouter.ai/api/v1/models",
headers={"Authorization": f"Bearer {api_key}"}
) as resp:
if resp.status != 200: raise ValueError(f"HTTP {resp.status}")
data = await resp.json()
models_data = data.get("data", [])
models_data.sort(key=lambda x: x["id"])
top_list = []
other_list = []
favs = ["google/gemini-2.0-flash-001", "openai/gpt-4o", "anthropic/claude-3.5-sonnet", "deepseek/deepseek-r1"]
for m in models_data:
mid = m["id"]
line = f"• <code>{mid}</code>"
if mid in favs: top_list.append(line)
elif any(x in mid for x in ["gemini", "gpt", "claude", "deepseek"]): other_list.append(line)
text = self.strings.get("gmodel_list_title_Openrouter", "📋 Models:") + "\n" + "\n".join(top_list) + "\n\n" + "\n".join(other_list[:50])
file = io.BytesIO(text.encode("utf-8")); file.name = "openrouter_models.txt"
await self.client.send_file(message.chat_id, file=file, caption="📋 OpenRouter Models", reply_to=message.id)
await status_msg.delete()
else:
if not self.api_keys: return await utils.answer(status_msg, self.strings['no_api_key'])
client = genai.Client(api_key=self.api_keys[0])
models = await asyncio.to_thread(client.models.list)
txt = "\n".join([f"• <code>{m.name.split('/')[-1]}</code> ({m.display_name})" for m in models])
f = io.BytesIO((self.strings["gmodel_list_title"] + "\n" + txt).encode('utf-8'))
f.name = "models_list.txt"
await self.client.send_file(message.chat_id, file=f, caption="📋 Список доступных моделей", reply_to=message.id)
await status_msg.delete()
except Exception as e:
await utils.answer(status_msg, self.strings["gmodel_list_error"].format(self._handle_error(e)))
return
if not args: return await utils.answer(message, f"Текущая модель: <code>{self.config['model_name']}</code>")
self.config["model_name"] = args
await utils.answer(message, f"Модель Gemini установлена: <code>{args}</code>")
if not args_raw:
return await utils.answer(message, f"🔮 <b>Провайдер:</b> {provider}\n🧠 <b>Модель:</b> <code>{self.config['model_name']}</code>")
self.config["model_name"] = args_raw
warning = ""
if provider == "google" and ("/" in args_raw or any(x in args_raw.lower() for x in ["gpt", "claude", "deepseek", "llama"])):
warning = (
"\n\n⚠️ <b>Конфликт настроек!</b>\n"
f"Вы установили модель <code>{args_raw}</code>, но провайдер остался <b>Google</b>.\n"
"Смените провайдера командой:\n<code>.cfg gemini provider openrouter</code>"
)
elif provider == "openrouter" and "/" not in args_raw and "gemini" in args_raw.lower():
warning = (
"\n\n⚠️ <b>Совет:</b> Для OpenRouter лучше использовать полные ID.\n"
f"Например: <code>google/{args_raw}</code>"
)
await utils.answer(message, f"✅ Модель установлена: <code>{args_raw}</code>{warning}")
@loader.command()
async def gres(self, message: Message):
@@ -1107,16 +1311,26 @@ class Gemini(loader.Module):
msgs = await self.client.get_messages(cid, limit=lim)
if skip_last and msgs: msgs = msgs[1:]
for m in msgs:
if not m or (not m.text and not m.media): continue
if not m: continue
if not (m.text or m.sticker or m.photo or m.file or m.media):
continue
name = get_display_name(await m.get_sender()) or "Unknown"
txt = m.text or ("[Media]" if m.media else "")
txt = m.text or ""
if m.sticker:
alt = next((a.alt for a in m.sticker.attributes if isinstance(a, DocumentAttributeSticker)), "?")
alt = "?"
if hasattr(m.sticker, 'attributes'):
alt = next((a.alt for a in m.sticker.attributes if isinstance(a, DocumentAttributeSticker)), "?")
txt += f" [Стикер: {alt}]"
elif m.photo: txt += " [Фото]"
elif m.document and not hasattr(m.media, "webpage"): txt += "айл]"
if txt.strip(): lines.append(f"{name}: {txt.strip()}")
except: pass
elif m.photo:
txt += "ото]"
elif m.file:
txt += " [Файл]"
elif m.media and not txt:
txt += " [Медиа]"
if txt.strip():
lines.append(f"{name}: {txt.strip()}")
except Exception as e:
pass
return "\n".join(reversed(lines))
def _handle_error(self, e: Exception) -> str:
@@ -1202,35 +1416,6 @@ class Gemini(loader.Module):
self._clear_history(chat_id, gauto=False)
await call.edit(self.strings["memory_cleared"], reply_markup=None)
async def _get_recent_chat_text(self, chat_id: int, count: int = None, skip_last: bool = False) -> str:
history_limit = count or self.config["impersonation_history_limit"]
fetch_limit = history_limit + 1 if skip_last else history_limit
chat_history_lines = []
try:
messages = await self.client.get_messages(chat_id, limit=fetch_limit)
if skip_last and messages:
messages = messages[1:]
for msg in messages:
if not msg: continue
if not msg.text and not msg.sticker and not msg.photo and not (msg.media and not hasattr(msg.media, "webpage")):
continue
sender = await msg.get_sender()
sender_name = get_display_name(sender) if sender else "Unknown"
text_content = msg.text or ""
if msg.sticker and hasattr(msg.sticker, 'attributes'):
alt_text = next((attr.alt for attr in msg.sticker.attributes if isinstance(attr, DocumentAttributeSticker)), None)
text_content += f" [Стикер: {alt_text or '?'}]"
elif msg.photo:
text_content += " [Фото]"
elif msg.document and not hasattr(msg.media, "webpage"):
text_content += " [Файл]"
if text_content.strip():
chat_history_lines.append(f"{sender_name}: {text_content.strip()}")
except Exception as e:
logger.warning(f"Не удалось получить историю для авто-ответа: {e}")
return "\n".join(reversed(chat_history_lines))
async def _scan_keys(self, force=False):
"""
Сканирует ключи на валидность.
@@ -1336,6 +1521,54 @@ class Gemini(loader.Module):
return out.getvalue()
except: return img_bytes
async def _send_to_Openrouter_api(self, model, messages, temperature):
"""Отправка запроса в OpenRouter (OpenAI format)"""
api_key = self.config["Openrouter_api_key"]
if not api_key:
raise ValueError("Не указан OpenRouter API Key! Установите его в .cfg")
url = "https://openrouter.ai/api/v1/chat/completions"
headers = {
"Authorization": f"Bearer {api_key}",
"Content-Type": "application/json",
"HTTP-Referer": "https://github.com/SenkoGuardian",
"X-Title": "Gemini Module for Heroku Telegram-userbot",
}
payload = {
"model": model,
"messages": messages,
"temperature": min(temperature, 1.0)
}
async with aiohttp.ClientSession() as session:
async with session.post(url, headers=headers, json=payload, timeout=GEMINI_TIMEOUT) as resp:
text = await resp.text()
if resp.status != 200:
try:
err_json = json.loads(text)
err_msg = err_json.get('error', {}).get('message', text)
except:
err_msg = text
raise ConnectionError(f"OpenRouter API Error {resp.status}: {err_msg}")
try:
result = json.loads(text)
except json.JSONDecodeError:
raise ValueError(f"OpenRouter вернул не JSON: {text[:100]}...")
if "choices" not in result or not result["choices"]:
if "error" in result:
raise ValueError(f"OpenRouter Logic Error: {result['error']}")
raise ValueError(f"Пустой ответ (нет 'choices'). Raw: {text}")
return result["choices"][0]["message"]["content"]
def _convert_google_history_to_openai(self, history: list, system_prompt: str) -> list:
"""Конвертирует историю из формата Google в формат OpenAI."""
messages = []
if system_prompt:
messages.append({"role": "system", "content": system_prompt})
for item in history:
role = "assistant" if item['role'] == "model" else "user"
content = item.get("content", "")
messages.append({"role": role, "content": content})
return messages
def _is_memory_enabled(self, chat_id: str) -> bool: return chat_id not in self.memory_disabled_chats
def _disable_memory(self, chat_id: int): self.memory_disabled_chats.add(str(chat_id))
def _enable_memory(self, chat_id: int): self.memory_disabled_chats.discard(str(chat_id))

View File

@@ -0,0 +1,267 @@
#░░░███░███░███░███░███
#░░░░░█░█░░░░█░░█░░░█░█
#░░░░█░░███░░█░░█░█░█░█
#░░░█░░░█░░░░█░░█░█░█░█
#░░░███░███░░█░░███░███
# meta developer: @nullmod
__version__ = (1, 0, 1)
import io
import math
from PIL import Image, ImageDraw
from herokutl.tl.custom import Message
from herokutl.tl.functions.help import (
GetPeerProfileColorsRequest
)
from herokutl.tl.types import (
EmojiStatusCollectible
)
from .. import loader, utils
def resize_image(image: Image.Image, max_size: int = 1280) -> Image.Image:
w, h = image.size
if max(w, h) <= max_size:
return image
else:
scale = max_size / max(w, h)
new_w = int(w * scale)
new_h = int(h * scale)
return image.resize((new_w, new_h), Image.LANCZOS)
# Source: https://gist.github.com/weihanglo/1e754ec47fdd683a42fdf6a272904535#file-draw_gradient_pillow-py
def get_gradient(size: tuple, color1: tuple, color2: tuple, gradient_type: str = "linear") -> Image.Image:
def interpolate(f_co, t_co, interval):
if interval <= 1:
yield list(t_co)
return
det_co = [(t - f) / (interval - 1) for f, t in zip(f_co, t_co)]
for i in range(interval):
yield [round(f + det * i) for f, det in zip(f_co, det_co)]
gradient = Image.new('RGB', size, color=(0, 0, 0))
draw = ImageDraw.Draw(gradient)
if gradient_type == "linear":
top_color, bottom_color = color1, color2
for y, color in enumerate(interpolate(top_color, bottom_color, max(1, size[1]))):
draw.line([(0, y), (size[0], y)], fill=tuple(color), width=1)
elif gradient_type == "radial":
center_color, edge_color = color1, color2
max_radius = math.hypot(size[0], size[1]) / 2.0
interval = max(1, int(math.ceil(max_radius)) + 1)
colors = list(interpolate(center_color, edge_color, interval))
cx = size[0] / 2
cy = size[1] / 2
for r_index, color in enumerate(colors):
r = interval - 1 - r_index
if r < 0:
continue
bbox = [
int(round(cx - r)),
int(round(cy - r)),
int(round(cx + r)),
int(round(cy + r))
]
draw.ellipse(bbox, fill=tuple(color))
return gradient
def set_gradient(im: io.BytesIO, gradient: Image.Image) -> io.BytesIO:
img = resize_image(Image.open(im).convert('RGBA'))
max_size = max(img.width, img.height)
gradient = gradient.resize((max_size, max_size), Image.LANCZOS).convert('RGBA')
left = (max_size - img.width) // 2
top = (max_size - img.height) // 2
gradient.paste(img, (left, top), img)
buffer = io.BytesIO()
gradient.save(buffer, format='PNG')
buffer.seek(0)
return buffer
def crop_by_bbox(img: Image.Image, bbox: tuple = None):
img_w, img_h = img.size
x, y, w, h = bbox or BBOX_TGA_TGD
left = int(round(x * img_w))
top = int(round(y * img_h))
right = int(round((x + w) * img_w))
bottom = int(round((y + h) * img_h))
return img.crop((left, top, right, bottom))
def hex_to_rgb(value: int):
return ((value >> 16) & 255, (value >> 8) & 255, value & 255)
def hexes_to_rgbs(value: list):
if len(value) > 1:
res = list()
for i in value:
res.append(hex_to_rgb(i))
return tuple(res)
else:
res = hex_to_rgb(value[0])
return (res, res)
SHAPES = {
# TODO: фигуры для создания масок на авы
}
BBOX_TGA_TGD = (
2894 / 8268,
1260 / 8268,
2504 / 8268,
2504 / 8268,
)
@loader.tds
class Gradientor(loader.Module):
strings = {
"name": "Gradientor",
"_cls_doc": "A module to create your profile picture with a background from your profile",
"gradient_creating": "<tg-emoji emoji-id=5886667040432853038>🔁</tg-emoji> Creating gradient...",
"gradient_created": "<tg-emoji emoji-id=5818804345247894731>✅</tg-emoji> Gradient created!",
}
strings_ru = {
"_cls_doc": "Модуль для создания вашей аватарки на фоне из вашего профиля",
"gradient_creating": "<tg-emoji emoji-id=5886667040432853038>🔁</tg-emoji> Создание градиента...",
"gradient_created": "<tg-emoji emoji-id=5818804345247894731>✅</tg-emoji> Градиент создан!",
}
async def client_ready(self):
self.colors = self.get("PROFILE_COLORS", None)
if not self.colors or not self.colors.get("light", None):
raw_colors = (await self.client(GetPeerProfileColorsRequest(0))).colors
self.colors = {
"dark": {
str(col.color_id): hexes_to_rgbs(col.dark_colors.bg_colors) for col
in raw_colors
},
"light": {
str(col.color_id): hexes_to_rgbs(col.colors.bg_colors) for col
in raw_colors
},
}
self.set("PROFILE_COLORS", self.colors)
@loader.command(
ru_doc="[фотография/reply] - создать аватарку с градиентом из цвета профиля\n"
"--update-cache - обновить кеш профиля, если вы только что сменили фон профиля\n"
"--linear - использовать линейный градиент\n"
"--light - использовать светлую тему"
)
async def makepp(self, message: Message):
"""[photo/reply] - create a profile picture with a gradient from profile color\n
--update-cache - update profile cache if you just changed profile background\n
--linear - use linear gradient\n
--light - use light theme"""
reply: Message = await message.get_reply_message()
args = utils.get_args(message)
if "--update-cache" in args:
upd_cache = True
args.remove("--update-cache")
else:
upd_cache = False
if "--linear" in args:
force_linear = True
args.remove("--linear")
else:
force_linear = False
if "--light" in args:
theme = "light"
args.remove("--light")
else:
theme = "dark"
if "--full" in args:
_full = True
args.remove("--full")
else:
_full = False
user = None
background_only = False
if args:
user = await self.client.get_entity(int(args[0]) if args[0].isdigit() else args[0])
photo_source = (
message
if (not reply or not (reply.photo or reply.document and "image/" in getattr(reply.document, "mime_type", "")))
else reply
)
if not (photo_source.photo or photo_source.document and "image/" in getattr(photo_source.document, "mime_type", "")):
background_only = True
if not user:
if upd_cache:
user = self.client.hikka_me = await self.client.get_me()
elif reply:
user = reply.sender
else:
user = self.client.hikka_me
if not user.premium:
color1, color2 = (28, 28, 28), (28, 28, 28)
elif user.emoji_status and isinstance(user.emoji_status, EmojiStatusCollectible):
color1, color2 = (
user.emoji_status.edge_color, user.emoji_status.center_color
)
color1 = hex_to_rgb(color1)
color2 = hex_to_rgb(color2)
elif user.profile_color:
color_variant = user.profile_color.color
color1, color2 = self.colors.get(theme).get(
str(color_variant),
((28, 28, 28), (28, 28, 28))
)
else:
color1, color2 = (28, 28, 28), (28, 28, 28)
await utils.answer(message, self.strings["gradient_creating"])
gradient = get_gradient((1280, 1280), color1, color2, "linear" if force_linear else "radial")
if not _full:
gradient = crop_by_bbox(gradient)
if not background_only and not _full:
p_b = await photo_source.download_media(bytes)
p_b_io = io.BytesIO(p_b)
p_b_io.seek(0)
result = set_gradient(p_b_io, gradient)
else:
result = io.BytesIO()
gradient.save(result, format='PNG')
result.seek(0)
result.name = "grad @nullmod.png"
await utils.answer(message, self.strings["gradient_created"], file=result, force_document=True)

View File

@@ -1,3 +1,4 @@
Chess
HaremManager
SchedulePlus
SchedulePlus
Gradientor

View File

@@ -1,8 +1,8 @@
__version__ = (3, 1, 1)
# meta banner: https://raw.githubusercontent.com/kamekuro/hikka-mods/main/banners/yamusic.png
# packurl: https://raw.githubusercontent.com/coddrago/assets/refs/heads/main/modules/yamusic.yml
# meta pic: https://raw.githubusercontent.com/kamekuro/hikka-mods/main/icons/yamusic.png
# meta developer: @codrago
# meta banner: https://raw.githubusercontent.com/coddrago/modules/refs/heads/main/banner.png
# meta developer: @codrago_m
# old meta dev: @kamekuro xuesos
# scope: heroku_only
# scope: heroku_min 1.7.2
@@ -41,6 +41,7 @@ class Banners:
meta_info: str = "Music",
is_liked: bool = False,
repeat_mode: str = "NONE",
blur: int = 0,
):
self.title = title
self.artists = artists
@@ -52,6 +53,7 @@ class Banners:
self.meta_info = meta_info
self.is_liked = is_liked
self.repeat_mode = repeat_mode
self.blur = blur
def ultra(self) -> io.BytesIO:
WIDTH, HEIGHT = 2560, 1220
@@ -96,7 +98,9 @@ class Banners:
background = background.crop((0, offset, bg_w, offset + new_h))
background = background.resize((WIDTH, HEIGHT), Image.Resampling.LANCZOS)
background = background.filter(ImageFilter.GaussianBlur(radius=0))
if self.blur > 0:
background = background.filter(ImageFilter.GaussianBlur(radius=self.blur))
dark_overlay = Image.new("RGBA", (WIDTH, HEIGHT), (0, 0, 0, 180))
background = Image.alpha_composite(background, dark_overlay)
@@ -296,30 +300,6 @@ class Banners:
(heart_x, icon_y_center + heart_size + 5),
]
if self.is_liked:
draw.ellipse(c1_box, fill="red", outline="red")
draw.ellipse(c2_box, fill="red", outline="red")
draw.polygon(tri_points, fill="red", outline="red")
else:
draw.ellipse(c1_box, fill=None, outline="red", width=3)
draw.ellipse(c2_box, fill=None, outline="red", width=3)
draw.line(
[
(heart_x - c_r * 2 + 1, icon_y_center),
(heart_x, icon_y_center + heart_size + 5),
],
fill="red",
width=3,
)
draw.line(
[
(heart_x + c_r * 2 - 1, icon_y_center),
(heart_x, icon_y_center + heart_size + 5),
],
fill="red",
width=3,
)
by = io.BytesIO()
background.save(by, format="PNG")
by.seek(0)
@@ -378,8 +358,13 @@ class YaMusicMod(loader.Module):
option="banner_version",
default="ultra",
doc=lambda: self.strings["_cfg"]["banner_version"],
validator=loader.validators.Choice(["old", "new", "ultra"]),
),)
validator=loader.validators.Choice(["ultra"]),
),
loader.ConfigValue(
option="blur",
default=0,
),
)
self.ym_client = None
self.device_id = "".join(random.choices(string.ascii_lowercase, k=16))
@@ -392,6 +377,7 @@ class YaMusicMod(loader.Module):
#"now_play", self._now_play_placeholder, "placeholder for nowplay music"
# Heroku 2.0.0 feature
#)
#utils.register_placeholder("duration", self._duration_placeholder, "progress bar")
if not self.get("guide_sent", False):
await self.inline.bot.send_message(self._tg_id, self.strings("iguide"))
@@ -437,7 +423,7 @@ class YaMusicMod(loader.Module):
me = await self._client.get_me()
self._premium = me.premium if hasattr(me, "premium") else False
@loader.loop(30)
@loader.loop(15)
async def autobio(self):
if not self.config["token"]:
self.autobio.stop()
@@ -547,6 +533,88 @@ class YaMusicMod(loader.Module):
),
)
async def _duration_placeholder(self):
"""Placeholder for {duration} with custom emoji bar"""
if not self.config["token"]:
return "No Token"
try:
now = await self.__get_now_playing()
if not now or now.get("paused"):
return "<code>Not Playing</code>"
duration = now.get("duration_ms", 0)
progress = now.get("progress_ms", 0)
if duration == 0:
return "0%"
percent = (progress / duration) * 100
s_less_10 = (
"<emoji document_id=5454137780454067986></emoji>"
"<emoji document_id=6158923355173949539>⭐</emoji>"
"<emoji document_id=6159012102083188132>⭐</emoji>"
"<emoji document_id=6159012102083188132>⭐</emoji>"
"<emoji document_id=6158753257289158944>⭐</emoji>"
"<emoji document_id=6156700344526049665>⭐</emoji>"
)
s_10_to_20 = (
"<emoji document_id=5454137780454067986></emoji>"
"<emoji document_id=6159095673556840262>⭐</emoji>"
"<emoji document_id=6159012102083188132>⭐</emoji>"
"<emoji document_id=6156933677214341691>⭐</emoji>"
"<emoji document_id=6158753257289158944>⭐</emoji>"
"<emoji document_id=6156700344526049665>⭐</emoji>"
)
s_30_to_40 = (
"<emoji document_id=5454137780454067986></emoji>"
"<emoji document_id=5454397458471750662></emoji>"
"<emoji document_id=5454397458471750662></emoji>"
"<emoji document_id=6158923355173949539>⭐</emoji>"
"<emoji document_id=6159012102083188132>⭐</emoji>"
"<emoji document_id=6156700344526049665>⭐</emoji>"
)
s_over_50 = (
"<emoji document_id=5454137780454067986></emoji>"
"<emoji document_id=5454397458471750662></emoji>"
"<emoji document_id=5454397458471750662></emoji>"
"<emoji document_id=5454397458471750662></emoji>"
"<emoji document_id=6156933677214341691>⭐</emoji>"
"<emoji document_id=6156700344526049665>⭐</emoji>"
)
s_over_80 = (
"<emoji document_id=5454137780454067986></emoji>"
"<emoji document_id=5454397458471750662></emoji>"
"<emoji document_id=5454397458471750662></emoji>"
"<emoji document_id=5454397458471750662></emoji>"
"<emoji document_id=5454397458471750662></emoji>"
"<emoji document_id=6156700344526049665>⭐</emoji>"
)
if percent < 10:
return s_less_10
elif percent < 20:
return s_10_to_20
elif percent < 30:
return s_10_to_20
elif percent < 40:
return s_30_to_40
elif percent < 50:
return s_30_to_40
elif percent < 80:
return s_over_50
else:
return s_over_80
except Exception as e:
return f"Error: {e}"
async def _download_bytes(self, url: str) -> typing.Optional[bytes]:
try:
async with aiohttp.ClientSession() as session:
@@ -686,8 +754,10 @@ class YaMusicMod(loader.Module):
meta_info=meta_info,
is_liked=is_liked,
repeat_mode=repeat_mode,
blur=self.config["blur"],
)
file = await utils.run_sync(
getattr(banners, self.config["banner_version"], banners.ultra)
)

View File

@@ -17,10 +17,9 @@ figlet
promoclaimer
passwordgen
send
lastfm
dbmod
chatmodule
stats
tagwatcher
hardspam
YaMusic
YaMusic

View File

@@ -1,124 +0,0 @@
# ---------------------------------------------------------------------------------
#░█▀▄░▄▀▀▄░█▀▄░█▀▀▄░█▀▀▄░█▀▀▀░▄▀▀▄░░░█▀▄▀█
#░█░░░█░░█░█░█░█▄▄▀░█▄▄█░█░▀▄░█░░█░░░█░▀░█
#░▀▀▀░░▀▀░░▀▀░░▀░▀▀░▀░░▀░▀▀▀▀░░▀▀░░░░▀░░▒▀
# Name: LastFM
# Description: Module for music from different services
# Author: @codrago_m
# ---------------------------------------------------------------------------------
# 🔒 Licensed under the GNU AGPLv3
# 🌐 https://www.gnu.org/licenses/agpl-3.0.html
# ---------------------------------------------------------------------------------
# Author: @codrago
# Commands: nowplay
# scope: heroku_only
# meta developer: @codrago_m
# meta banner: https://raw.githubusercontent.com/coddrago/modules/refs/heads/main/banner.png
# meta pic: https://envs.sh/Hob.webp
# ---------------------------------------------------------------------------------
from .. import loader, utils
from herokutl import events
import requests
import asyncio
@loader.tds
class lastfmmod(loader.Module):
"""Module for music from different services"""
def __init__(self):
self.config = loader.ModuleConfig(
loader.ConfigValue(
"username_lastfm",
None,
lambda: self.strings["_doc_username_lastfm"],
),
loader.ConfigValue(
"text",
"<emoji document_id=6007938409857815902>🎧</emoji> <b>now playing...</b>\n"
"<emoji document_id=5915480455603295660>🎶</emoji><b> playlist: </b><code>{song_album}</code>\n"
"<emoji document_id=5891249688933305846>🎵</emoji> <b>track:</b> <code>{song_name}</code>\n"
"<emoji document_id=5897554554894946515>🎤</emoji> <b>artist:</b> <code>{song_artist}</code>",
lambda: self.strings["_doc_text"],
),
)
strings = {
"name": "LastFm",
"loading":"<emoji document_id=5873204392429096339>⌨️</emoji> Loading song...",
"bot_no_result": "<emoji document_id=5465665476971471368>❌</emoji> Nothing found.\nTitle: {song_name}\nAuthor: {song_artist}\nAlbum:{song_album}",
"_doc_text": "The text that will be written next to the file",
"_doc_username_lastfm": "Your username from last.fm",
"nick_error": "<emoji document_id=5465665476971471368>❌</emoji> Put your nickname from last.fm",
"tutorial": "Go to last.fm and register.\nBE SURE to remember the username and password, they will come in handy later.\nLet's look at the VK version\nAfter that, go to the @vkxci channel, download VK X and log in to your VK account, then go to settings and click «Integrations», select Last FM.\nEnter the username and password.\nThen you're almost done!\nWrite <code>{prefix}fcfg lastfm username_lastfm</code> {username}\nUse the <code>{prefix}nowplay</code> command and enjoy life!",
}
strings_ru = {
"name": "LastFm",
"loading": "<emoji document_id=5873204392429096339>⌨️</emoji> Загрузка трека...",
"bot_no_result": "<emoji document_id=5465665476971471368>❌</emoji> Ничего не найдено.\nНазвание: {song_name}\nИсполнитель: {song_artist}\nАльбом: {song_album}",
"_doc_text": "Текст, который будет написан рядом с файлом",
"_doc_username_lastfm": "Ваш username с last.fm",
"nick_error": "<emoji document_id=5465665476971471368>❌</emoji> Укажите ваш никнейм с last.fm",
"tutorial": "Зайдите на last.fm и зарегистрируйтесь.\nОБЯЗАТЕЛЬНО запомните логин и пароль, они пригодятся позже.\nРассмотрим вариант для VK\nПосле этого зайдите в канал @vkxci, скачайте VK X и авторизуйтесь в своём аккаунте VK, затем зайдите в настройки и нажмите «Интеграции», выберите Last FM.\nВведите логин и пароль.\nЗатем вы почти закончили!\nНапишите <code>{prefix}fcfg lastfm username_lastfm</code> {username}\nИспользуйте команду <code>{prefix}nowplay</code> и наслаждайтесь жизнью!",
}
@loader.command(alias="np")
async def nowplay(self, message):
"""| send playing track"""
lastfm_username = self.config["username_lastfm"]
API_KEY = "460cda35be2fbf4f28e8ea7a38580730" # Облегчение жизни школьникам
if not lastfm_username:
response_text = self.strings["nick_error"]
await self.invoke("config", "lastfm", message=message)
await utils.answer(message, response_text)
else:
try:
current_track_url = f'http://ws.audioscrobbler.com/2.0/?method=user.getrecenttracks&nowplaying=true&user={lastfm_username}&api_key={API_KEY}&format=json'
response = requests.get(current_track_url)
data = response.json()
if 'recenttracks' in data and 'track' in data['recenttracks'] and data['recenttracks']['track']:
nowplaying_track = None
for track in data['recenttracks']['track']:
if '@attr' in track and 'nowplaying' in track['@attr']:
nowplaying_track = track
break
if nowplaying_track:
song_name = nowplaying_track.get('name', 'Unknown song')
song_artist = nowplaying_track.get('artist', {}).get('#text', 'Unknown Artist')
if nowplaying_track.get('album', {}).get('#text') == nowplaying_track.get('name'):
song_album = "single"
else:
song_album = nowplaying_track.get('album', {}).get('#text', 'Unknown Album')
response_text = f"/search {song_name} - {song_artist}"
try:
async with message.client.conversation("@LyaDownbot") as conv:
await conv.send_message(response_text)
while True:
response_bot = await conv.get_response()
if "Не удалось найти трек" in response_bot.text:
await utils.answer(message, self.strings["bot_no_result"])
return
if "Ищем треки..." in response_bot.text:
await utils.answer(message, self.strings["loading"])
if response_bot.media:
await message.client.send_file(message.chat_id, response_bot.media, caption = self.config["text"].format(song_artist=song_artist, song_album=song_album, song_name=song_name))
await message.delete()
return
except Exception as e:
await utils.answer(message, f"<pre><code class='language-python'>{e}</code></pre>")
except Exception as e:
await utils.answer(message, f"<pre><code class='language-python'>{e}</code></pre>")
@loader.command()
async def tutorl(self, message):
"""| tutorial"""
await utils.answer(message, self.strings['tutorial'].format(prefix = self.get_prefix(), username="{username}"))

View File

@@ -1,10 +1,13 @@
# meta developer: @codrago_m
# scope: heroku_min 2.0.0
import logging
from .. import utils, loader, main
from telethon.tl.functions.messages import MarkDialogUnreadRequest
from .. import loader, main, utils
logger = logging.getLogger("TagWatcher")
@@ -124,6 +127,7 @@ class TagWatcher(loader.Module):
description="Here will be notifications about mentions in chats.",
icon_emoji_id=5409025823388741707,
)
self.xdlib = await self.import_lib(
"https://raw.githubusercontent.com/coddrago/modules/refs/heads/main/libs/xdlib.py",
suspend_on_error=True,

View File

@@ -1,5 +1,5 @@
# -- version --
__version__ = (1, 2, 1)
__version__ = (1, 2, 2)
# -- version --
@@ -9,7 +9,7 @@ __version__ = (1, 2, 1)
# ██║╚██╔╝██║██╔══╝░░██╔══██║██║░░██║██║░░██║░░████╔═████║░░╚═══██╗░╚═══██╗
# ██║░╚═╝░██║███████╗██║░░██║██████╔╝╚█████╔╝░░╚██╔╝░╚██╔╝░██████╔╝██████╔╝
# ╚═╝░░░░░╚═╝╚══════╝╚═╝░░╚═╝╚═════╝░░╚════╝░░░░╚═╝░░░╚═╝░░╚═════╝░╚═════╝░
# © Copyright 2025
# © Copyright 2026
# ✈ https://t.me/mead0wssMods
@@ -33,14 +33,17 @@ class SenderGifts(loader.Module):
"checking_user": "<emoji document_id=5206634672204829887>🔍</emoji> Проверка пользователя...",
"checking_balance": "<emoji document_id=5206634672204829887>🔍</emoji> Проверка баланса...",
"user_not_found": "<emoji document_id=4958526153955476488>❌</emoji> Пользователь не найден",
"gift_menu": "<emoji document_id=5931696400982088015>🎁</emoji> Выберите категорию подарков.\n\n<emoji document_id=6032693626394382504>👤</emoji> Пользователь: {}\n<emoji document_id=5873153278023307367>📄</emoji> Текст: {}\n<emoji document_id=5951810621887484519>⭐</emoji> Баланс: {} звезд",
"category_menu": "<emoji document_id=5931696400982088015>🎁</emoji> Подарки за {}\n\n<emoji document_id=6032693626394382504>👤</emoji> Пользователь: {}\n<emoji document_id=5873153278023307367>📄</emoji> Текст: {}",
"gift_menu": "<tg-emoji emoji-id=5370781982886220096>🎁</tg-emoji> Выберите категорию подарков.\n\n<tg-emoji emoji-id=6048471184461271609>👤</tg-emoji> Пользователь: {}\n<tg-emoji emoji-id=6048762138430803961>📂</tg-emoji> Текст: {}\n<tg-emoji emoji-id=5321485469249198987>⭐</tg-emoji> Баланс: {} звезд",
"category_menu": "<tg-emoji emoji-id=5370781982886220096>🎁</tg-emoji> Подарки за {}\n\n<tg-emoji emoji-id=6048471184461271609>👤</tg-emoji> Пользователь: {}\n<tg-emoji emoji-id=6048762138430803961>📂</tg-emoji> Текст: {}",
"privacy_menu": "<tg-emoji emoji-id=5370781982886220096>🎁</tg-emoji> Выбран подарок: {}\n\nКак отправить подарок?",
"sending_gift": "<emoji document_id=5201691993775818138>🛫</emoji> Отправка подарка...",
"gift_sent": "<emoji document_id=5021905410089550576>✅</emoji> Подарок успешно отправлен!",
"not_enough_stars": "<emoji document_id=4958526153955476488>❌</emoji> Недостаточно звезд для отправки подарка {}!",
"min_stars_error": "<emoji document_id=4958526153955476488>❌</emoji> Недостаточно звезд для отправки минимального подарка!",
"no_available_gifts": "<emoji document_id=4958526153955476488>❌</emoji> Нет доступных подарков для вашего баланса",
"balance_error": "<emoji document_id=4958526153955476488>❌</emoji> Ошибка при проверке баланса",
"btn_public": "📢 Публично",
"btn_anon": "🕵️ Анонимно",
}
gift_categories = {
@@ -57,6 +60,7 @@ class SenderGifts(loader.Module):
{"id": 5170314324215857265, "emoji": "💐", "name": "Цветы"},
{"id": 5170564780938756245, "emoji": "🚀", "name": "Ракета"},
{"id": 5922558454332916696, "emoji": "🎄", "name": "Ёлка"},
{"id": 5956217000635139069, "emoji": "🧸", "name": "Новогодний мишка"}
],
100: [
{"id": 5168043875654172773, "emoji": "🏆", "name": "Кубок"},
@@ -135,9 +139,11 @@ class SenderGifts(loader.Module):
if row:
buttons.append(row)
helper_msg = await self.inline.form("🪐", balance_msg)
await utils.answer(
balance_msg,
helper_msg,
self.strings["gift_menu"].format(
f"@{user.username}" if user.username else user.first_name,
text if text else "-",
@@ -153,8 +159,8 @@ class SenderGifts(loader.Module):
for gift in gifts:
row.append({
"text": gift["emoji"],
"callback": self._send_gift,
"args": (user_id, gift["id"], text, gift["emoji"], msg_id, balance),
"callback": self._select_privacy,
"args": (user_id, gift["id"], text, gift["emoji"], msg_id, balance, price),
})
if len(row) == 3:
buttons.append(row)
@@ -183,6 +189,34 @@ class SenderGifts(loader.Module):
reply_markup=buttons
)
async def _select_privacy(self, call, user_id, gift_id, text, gift_emoji, msg_id, balance, price):
buttons = [
[
{
"text": self.strings["btn_public"],
"callback": self._send_gift,
"args": (user_id, gift_id, text, gift_emoji, msg_id, balance, False) # hide_name=False публично
},
{
"text": self.strings["btn_anon"],
"callback": self._send_gift,
"args": (user_id, gift_id, text, gift_emoji, msg_id, balance, True) # hide_name=True анонимно
}
],
[
{
"text": "⬅️ Назад",
"callback": self._show_category,
"args": (user_id, price, text, balance, msg_id)
}
]
]
await call.edit(
self.strings["privacy_menu"].format(gift_emoji),
reply_markup=buttons
)
async def _back_to_categories(self, call, user_id, text, balance, msg_id):
try:
user = await self.client.get_entity(user_id)
@@ -216,7 +250,7 @@ class SenderGifts(loader.Module):
reply_markup=buttons
)
async def _send_gift(self, call, user_id, gift_id, text, gift_emoji, msg_id, balance):
async def _send_gift(self, call, user_id, gift_id, text, gift_emoji, msg_id, balance, hide_name):
try:
await call.edit(
self.strings["sending_gift"],
@@ -227,11 +261,11 @@ class SenderGifts(loader.Module):
self.client.parse_mode,
)
text, entities = parse_mode.parse(text)
user = await self.client.get_input_entity(user_id)
inv = InputInvoiceStarGift(
user,
gift_id,
hide_name=hide_name,
message=TextWithEntities(text, entities) if text else TextWithEntities("", [])
)
form = await self.client(GetPaymentFormRequest(inv))

130503
modules.json

File diff suppressed because it is too large Load Diff

View File

@@ -10,8 +10,10 @@ logger = logging.getLogger(__name__)
def safe_unparse(node: ast.AST) -> str:
try:
return ast.unparse(node)
except AttributeError:
return getattr(node, 'id', str(type(node).__name__))
except Exception:
if hasattr(node, "id"):
return str(node.id)
return str(node)
def load_blacklist(file_path):
with open(file_path, "r", encoding="utf-8") as f:
@@ -27,17 +29,38 @@ def load_blacklist(file_path):
return blacklisted_modules
def is_loader_tds(deco: ast.AST) -> bool:
return (
isinstance(deco, ast.Attribute)
and isinstance(deco.value, ast.Name)
and deco.value.id == "loader"
and deco.attr in {"tds", "translatable_docstring"}
)
def extract_string_value(node: ast.AST) -> Optional[str]:
try:
if isinstance(node, ast.Constant) and isinstance(node.value, str):
return node.value
if isinstance(node, ast.Str):
return node.s
if isinstance(node, ast.JoinedStr):
parts = []
for v in node.values:
if isinstance(v, ast.Constant) and isinstance(v.value, str):
parts.append(v.value)
elif isinstance(v, ast.FormattedValue):
parts.append("{" + safe_unparse(v.value) + "}")
return "".join(parts)
if isinstance(node, ast.Name):
return node.id
if isinstance(node, ast.Attribute):
return f"{safe_unparse(node.value)}.{node.attr}"
return str(node)
return None
except Exception:
return None
@@ -111,7 +134,7 @@ def get_module_info(module_path: str) -> Optional[Dict[str, Any]]:
is_module_class = (
"Mod" in node.name or
any(isinstance(d, ast.Attribute) and safe_unparse(d).startswith("loader.tds") for d in node.decorator_list) or
any(is_loader_tds(d) for d in node.decorator_list) or
any(isinstance(d, ast.Name) and d.id == "loader" for d in node.decorator_list)
)

View File

@@ -0,0 +1,214 @@
# =======================================
# _ __ __ __ _
# | |/ /___ | \/ | ___ __| |___
# | ' // _ \ | |\/| |/ _ \ / _` / __|
# | . \ __/ | | | | (_) | (_| \__ \
# |_|\_\___| |_| |_|\___/ \__,_|___/
# @ke_mods
# =======================================
#
# LICENSE: CC BY-ND 4.0 (Attribution-NoDerivatives 4.0 International)
# --------------------------------------
# https://creativecommons.org/licenses/by-nd/4.0/legalcode
# =======================================
# meta developer: @ke_mods
from .. import loader, utils
import requests
import io
import textwrap
from PIL import Image, ImageDraw, ImageEnhance, ImageFilter, ImageFont, ImageOps
class Banners:
def __init__(
self,
title: str,
artists: list,
track_cover: bytes,
font
):
self.title = title
self.artists = ", ".join(artists) if isinstance(artists, list) else artists
self.track_cover = track_cover
self.font_url = font
def _get_font(self, size, font_bytes):
return ImageFont.truetype(io.BytesIO(font_bytes), size)
def _prepare_cover(self, size, radius):
cover = Image.open(io.BytesIO(self.track_cover)).convert("RGBA")
cover = cover.resize((size, size), Image.Resampling.LANCZOS)
mask = Image.new("L", (size, size), 0)
draw = ImageDraw.Draw(mask)
draw.rounded_rectangle((0, 0, size, size), radius=radius, fill=255)
output = Image.new("RGBA", (size, size), (0, 0, 0, 0))
output.paste(cover, (0, 0), mask=mask)
return output
def _prepare_background(self, w, h):
bg = Image.open(io.BytesIO(self.track_cover)).convert("RGBA")
bg = bg.resize((w, h), Image.Resampling.BICUBIC)
bg = bg.filter(ImageFilter.GaussianBlur(radius=20))
bg = ImageEnhance.Brightness(bg).enhance(0.4)
return bg
def horizontal(self):
W, H = 1500, 600
padding = 60
cover_size = 480
font_bytes = requests.get(self.font_url).content
title_font = self._get_font(55, font_bytes)
artist_font = self._get_font(45, font_bytes)
lfm_font = self._get_font(35, font_bytes)
img = self._prepare_background(W, H)
draw = ImageDraw.Draw(img)
cover = self._prepare_cover(cover_size, 30)
img.paste(cover, (padding, (H - cover_size) // 2), cover)
text_x = padding + cover_size + 60
text_y_start = 100
text_width_limit = W - text_x - padding
display_title = self.title
while title_font.getlength(display_title) > text_width_limit and len(display_title) > 0:
display_title = display_title[:-1]
if len(display_title) < len(self.title): display_title += ""
display_artist = self.artists
while artist_font.getlength(display_artist) > text_width_limit and len(display_artist) > 0:
display_artist = display_artist[:-1]
if len(display_artist) < len(self.artists): display_artist += ""
draw.text((text_x, text_y_start), display_title, font=title_font, fill="white")
draw.text((text_x, text_y_start + 70), display_artist, font=artist_font, fill="#B3B3B3")
bar_y = 480
draw.text((text_x, bar_y), "last.fm", font=lfm_font, fill="white")
by = io.BytesIO()
img.save(by, format="PNG")
by.seek(0)
by.name = "banner.png"
return by
def vertical(self):
W, H = 1000, 1500
padding = 80
cover_size = 800
font_bytes = requests.get(self.font_url).content
title_font = self._get_font(60, font_bytes)
artist_font = self._get_font(45, font_bytes)
lfm_font = self._get_font(35, font_bytes)
img = self._prepare_background(W, H)
draw = ImageDraw.Draw(img)
cover = self._prepare_cover(cover_size, 40)
cover_x = (W - cover_size) // 2
cover_y = 120
img.paste(cover, (cover_x, cover_y), cover)
text_area_y = cover_y + cover_size + 120
text_width_limit = W - (padding * 2)
display_title = self.title
while title_font.getlength(display_title) > text_width_limit and len(display_title) > 0:
display_title = display_title[:-1]
if len(display_title) < len(self.title): display_title += ""
display_artist = self.artists
while artist_font.getlength(display_artist) > text_width_limit and len(display_artist) > 0:
display_artist = display_artist[:-1]
if len(display_artist) < len(self.artists): display_artist += ""
title_w = title_font.getlength(display_title)
draw.text(((W - title_w) / 2, text_area_y), display_title, font=title_font, fill="white")
artist_w = artist_font.getlength(display_artist)
draw.text(((W - artist_w) / 2, text_area_y + 75), display_artist, font=artist_font, fill="#B3B3B3")
bar_y = text_area_y + 260
lfm_w = lfm_font.getlength("last.fm")
draw.text(((W - lfm_w) / 2, bar_y), "last.fm", font=lfm_font, fill="white")
by = io.BytesIO()
img.save(by, format="PNG")
by.seek(0)
by.name = "banner.png"
return by
@loader.tds
class lastfmmod(loader.Module):
"""Module for music from different services"""
strings = {
"name": "LastFm",
"no_track": "<emoji document_id=5465665476971471368>❌</emoji> <b>No track is currently playing</b>",
"_doc_text": "The text that will be written next to the file",
"_doc_username": "Your username from last.fm",
"nick_error": "<emoji document_id=5465665476971471368>❌</emoji> <b>Put your nickname from last.fm</b>",
"uploading": "<emoji document_id=5841359499146825803>🕔</emoji> <i>Uploading banner...</i>",
}
strings_ru = {
"name": "LastFm",
"no_track": "<emoji document_id=5465665476971471368>❌</emoji> <b>Сейчас ничего не играет</b>",
"_doc_text": "Текст, который будет написан рядом с файлом",
"_doc_username": "Ваш username с last.fm",
"nick_error": "<emoji document_id=5465665476971471368>❌</emoji> <b>Укажите ваш никнейм с last.fm</b>",
"uploading": "<emoji document_id=5841359499146825803>🕔</emoji> <i>Загрузка баннера...</i>",
}
strings_jp = {
"name": "LastFm",
"no_track": "<emoji document_id=5465665476971471368>❌</emoji> <b>現在再生中のトラックはありません</b>",
"_doc_text": "ファイルの横に表示されるテキスト",
"_doc_username": "Last.fmのユーザー名",
"nick_error": "<emoji document_id=5465665476971471368>❌</emoji> <b>Last.fmのニックネームを入力してください</b>",
"uploading": "<emoji document_id=5841359499146825803>🕔</emoji> <i>バナーをアップロード中...</i>",
}
def __init__(self):
self.config = loader.ModuleConfig(
loader.ConfigValue("username", None, lambda: self.strings["_doc_username"]),
loader.ConfigValue("custom_text", "<emoji document_id=5413612466208799435>🤩</emoji> <b>{song_name}</b> — <b>{song_artist}</b>", lambda: self.strings["_doc_text"]),
loader.ConfigValue("font", "https://raw.githubusercontent.com/kamekuro/assets/master/fonts/Onest-Bold.ttf", "Custom font URL (ttf)"),
loader.ConfigValue("banner_version", "horizontal", lambda: "Banner version", validator=loader.validators.Choice(["horizontal", "vertical"])),
)
@loader.command(alias="np")
async def nowplay(self, message):
"""| send playing track info"""
user = self.config["username"]
if not user:
await self.invoke("config", "lastfm", message=message)
return await utils.answer(message, self.strings["nick_error"])
try:
url = f'http://ws.audioscrobbler.com/2.0/?method=user.getrecenttracks&nowplaying=true&user={user}&api_key=460cda35be2fbf4f28e8ea7a38580730&format=json'
data = requests.get(url).json()
track = next((t for t in data.get('recenttracks', {}).get('track', []) if t.get('@attr', {}).get('nowplaying')), None)
if not track:
return await utils.answer(message, self.strings["no_track"])
name = track.get('name', 'Unknown')
artist = track.get('artist', {}).get('#text', 'Unknown')
caption = self.config["custom_text"].format(song_artist=artist, song_name=name)
imgs = track.get('image', [])
cov_url = next((i['#text'] for i in imgs if i['size'] == 'extralarge'), imgs[-1]['#text'] if imgs else None)
if not cov_url:
return await utils.answer(message, caption)
msg = await utils.answer(message, self.strings["uploading"])
cov_bytes = await utils.run_sync(requests.get, cov_url)
banners = Banners(name, artist, cov_bytes.content, self.config["font"])
file = await utils.run_sync(getattr(banners, self.config["banner_version"]))
await utils.answer(msg, caption, file=file)
except Exception as e:
await utils.answer(message, f"<pre><code class='language-python'>{e}</code></pre>")

View File

@@ -0,0 +1,48 @@
# =======================================
# _ __ __ __ _
# | |/ /___ | \/ | ___ __| |___
# | ' // _ \ | |\/| |/ _ \ / _` / __|
# | . \ __/ | | | | (_) | (_| \__ \
# |_|\_\___| |_| |_|\___/ \__,_|___/
# @ke_mods
# =======================================
#
# LICENSE: CC BY-ND 4.0 (Attribution-NoDerivatives 4.0 International)
# --------------------------------------
# https://creativecommons.org/licenses/by-nd/4.0/legalcode
# =======================================
# meta developer: @ke_mods
import subprocess
from .. import loader, utils
@loader.tds
class NeofetchMod(loader.Module):
strings = {
"name": "Neofetch",
"not_installed": "<b>Please, install</b> <i>Neofetch</i> <b>package</b>",
}
strings_ru = {
"not_installed": "<b>Пожалуйста, установите пакет</b> <i>Neofetch</i>",
}
strings_ua = {
"not_installed": "<b>Будь ласка, встановіть пакет<b> <i>Neofetch</i>",
}
@loader.command(
ru_doc="- запустить команду neofetch",
ua_doc="- запустити команду neofetch",
)
async def neofetchcmd(self, message):
"""- run neofetch command"""
try:
result = subprocess.run(["neofetch", "--stdout"], capture_output=True, text=True)
output = result.stdout
await utils.answer(message, f"<pre>{utils.escape_html(output)}</pre>")
except FileNotFoundError:
await utils.answer(message, self.strings("not_installed"))

View File

@@ -0,0 +1,211 @@
# =======================================
# _ __ __ __ _
# | |/ /___ | \/ | ___ __| |___
# | ' // _ \ | |\/| |/ _ \ / _` / __|
# | . \ __/ | | | | (_) | (_| \__ \
# |_|\_\___| |_| |_|\___/ \__,_|___/
# @ke_mods
# =======================================
#
# LICENSE: CC BY-ND 4.0 (Attribution-NoDerivatives 4.0 International)
# --------------------------------------
# https://creativecommons.org/licenses/by-nd/4.0/legalcode
# =======================================
# meta developer: @ke_mods
# requires: pillow
import io
import asyncio
from telethon import functions, types
from PIL import Image
from .. import loader, utils
@loader.tds
class PicToStoriesMod(loader.Module):
"""Grid for stories"""
strings = {
"name": "PicToStories",
"no_rep": (
"<emoji document_id=5879813604068298387>❗️</emoji> "
"<b>Reply to photo!</b>"
),
"work": (
"<emoji document_id=5841359499146825803>🕔</emoji> "
"<b>Processing...</b>"
),
"done": (
"<emoji document_id=5776375003280838798>✅</emoji> "
"<b>Done! Check your profile.</b>"
),
"err": (
"<emoji document_id=5778527486270770928>❌</emoji> "
"<b>Error:</b> {}"
),
}
strings_ru = {
"no_rep": (
"<emoji document_id=5879813604068298387>❗️</emoji> "
"<b>Реплай на фото!</b>"
),
"work": (
"<emoji document_id=5841359499146825803>🕔</emoji> "
"<b>Обрабатываю...</b>"
),
"done": (
"<emoji document_id=5776375003280838798>✅</emoji> "
"<b>Готово! Проверяй профиль.</b>"
),
"err": (
"<emoji document_id=5778527486270770928>❌</emoji> "
"<b>Ошибка:</b> {}"
),
}
def __init__(self):
self.config = loader.ModuleConfig(
loader.ConfigValue(
"period",
48,
lambda: "Visibility period in hours",
validator=loader.validators.Integer(),
),
loader.ConfigValue(
"blacklist",
[],
lambda: "Blacklisted user IDs",
validator=loader.validators.Series(loader.validators.Integer()),
),
loader.ConfigValue(
"cooldown",
0,
lambda: "Cooldown between stories in seconds",
validator=loader.validators.Integer(minimum=0),
),
)
@loader.command(ru_doc="<реплай на фото> [название альбома] - сделать сетку")
async def ptscmd(self, message):
"""<reply to photo> [album name] - make grid"""
args = utils.get_args_raw(message)
reply = await message.get_reply_message()
if not reply or not reply.media:
await utils.answer(message, self.strings("no_rep"))
return
try:
image_bytes = await reply.download_media(file=bytes)
img = Image.open(io.BytesIO(image_bytes))
except Exception as e:
await utils.answer(message, self.strings("err").format(e))
return
await utils.answer(message, self.strings("work"))
w, h = img.size
curr_ratio = w / h
variants = [
(5 / 4, 2),
(4 / 5, 3),
(3 / 5, 4),
(9 / 16, 5)
]
best_ratio, rows = min(variants, key=lambda x: abs(curr_ratio - x[0]))
new_h = int(w / best_ratio)
img = img.resize((w, new_h), Image.LANCZOS)
w, h = img.size
parts = []
pw, ph = w // 3, h // rows
for r in range(rows):
for c in range(3):
x, y = c * pw, r * ph
parts.append(img.crop((x, y, x + pw, y + ph)))
parts.reverse()
privacy = [types.InputPrivacyValueAllowAll()]
if self.config["blacklist"]:
entities = []
for uid in self.config["blacklist"]:
try:
entities.append(await self.client.get_input_entity(uid))
except Exception:
continue
if entities:
privacy.append(types.InputPrivacyValueDisallowUsers(users=entities))
story_ids = []
for i, p in enumerate(parts):
out = io.BytesIO()
p.save(out, "JPEG", quality=95)
out.seek(0)
uploaded_file = await self.client.upload_file(out, file_name="s.jpg")
res = await self.client(
functions.stories.SendStoryRequest(
peer=types.InputPeerSelf(),
media=types.InputMediaUploadedPhoto(uploaded_file),
privacy_rules=privacy,
period=self.config["period"] * 3600,
)
)
sid = next(
(
u.story_id if hasattr(u, "story_id") else u.id
for u in res.updates
if hasattr(u, "story_id") or hasattr(u, "id")
),
None,
)
if sid:
story_ids.append(sid)
if self.config["cooldown"] > 0 and i < len(parts) - 1:
await asyncio.sleep(self.config["cooldown"])
if not story_ids:
return
if args:
all_albums = await self.client(
functions.stories.GetAlbumsRequest(peer=types.InputPeerSelf(), hash=0)
)
target = next(
(a for a in all_albums.albums if getattr(a, 'title', '') == args),
None
)
if target:
await self.client(
functions.stories.UpdateAlbumRequest(
peer=types.InputPeerSelf(),
album_id=target.album_id,
add_stories=story_ids,
)
)
else:
await self.client(
functions.stories.CreateAlbumRequest(
peer=types.InputPeerSelf(),
stories=story_ids,
title=args,
)
)
else:
await self.client(
functions.stories.TogglePinnedRequest(
peer=types.InputPeerSelf(), id=story_ids, pinned=True
)
)
await utils.answer(message, self.strings("done"))

File diff suppressed because it is too large Load Diff

View File

@@ -0,0 +1,74 @@
# =======================================
# _ __ __ __ _
# | |/ /___ | \/ | ___ __| |___
# | ' // _ \ | |\/| |/ _ \ / _` / __|
# | . \ __/ | | | | (_) | (_| \__ \
# |_|\_\___| |_| |_|\___/ \__,_|___/
# @ke_mods
# =======================================
#
# LICENSE: CC BY-ND 4.0 (Attribution-NoDerivatives 4.0 International)
# --------------------------------------
# https://creativecommons.org/licenses/by-nd/4.0/legalcode
# =======================================
# meta developer: @ke_mods
from .. import loader, utils
from telethon.tl.types import ChatBannedRights
from telethon.tl.functions.channels import EditBannedRequest
from telethon.tl.types import ChannelParticipantsKicked
@loader.tds
class UnbanAllMod(loader.Module):
strings = {
"name": "UnbanAll",
"no_rights": "<b>❌ I don't have administrator rights to remove restrictions.</b>",
"success": "<b>✅ All banned chat members have been unbanned.</b>",
"unban_in_process": "<b>👀 Unbanning users...</b>",
"no_banned": "<b> There are no banned members in this chat.</b>",
"error_occured": "<b>💢 An error occurred while unbanning user <code>{}</code>:</b>\n<code>{}</code>",
}
strings_ru = {
"no_rights": "<b>❌ У меня нет прав администратора для снятия ограничений.</b>",
"success": "<b>✅ Все забаненные участники чата были разблокированы.</b>",
"unban_in_process": "<b>👀 Разбаниваю пользователей...</b>",
"no_banned": "<b> В этом чате нет забаненных участников.</b>",
"error_occured": "<b>💢 Произошла ошибка при разблокировке пользователя <code>{}</code>:</b>\n<code>{}</code>",
}
@loader.command(ru_doc="- Разбанить всех забаненных пользователей")
async def unbanallcmd(self, message):
"""- Unban all banned users"""
chat = await message.get_chat()
if not chat.admin_rights and not chat.creator:
await utils.answer(message, self.strings("no_rights"))
return
await utils.answer(message, self.strings("unban_in_process"))
no_banned = True
async for user in self.client.iter_participants(
message.chat_id, filter=ChannelParticipantsKicked
):
no_banned = False
try:
await self.client(EditBannedRequest(
message.chat_id,
user.id,
ChatBannedRights(until_date=0)
))
except Exception as e:
await utils.answer(message, self.strings("error_occured").format(user.id, e))
pass
if no_banned:
await utils.answer(message, self.strings("no_banned"))
return
await utils.answer(message, self.strings("success"))

View File

@@ -0,0 +1,7 @@
Neofetch
randomanimepic
SpotifyMod
UnbanAll
voicetotext
LastFm
PicToStories

View File

@@ -0,0 +1,65 @@
# =======================================
# _ __ __ __ _
# | |/ /___ | \/ | ___ __| |___
# | ' // _ \ | |\/| |/ _ \ / _` / __|
# | . \ __/ | | | | (_) | (_| \__ \
# |_|\_\___| |_| |_|\___/ \__,_|___/
# @ke_mods
# =======================================
#
# LICENSE: CC BY-ND 4.0 (Attribution-NoDerivatives 4.0 International)
# --------------------------------------
# https://creativecommons.org/licenses/by-nd/4.0/legalcode
# =======================================
# meta developer: @ke_mods
import requests
import asyncio
import logging
import traceback
from logging import basicConfig
from .. import loader, utils
basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
@loader.tds
class RandomAnimePicMod(loader.Module):
strings = {
"name": "RandomAnimePic",
"img": "<emoji document_id=4916036072560919511>✅</emoji> <b>Your anime pic</b>\n<emoji document_id=5877465816030515018>🔗</emoji> <b>URL:</b> {}",
"loading": "<emoji document_id=4911241630633165627>✨</emoji> <b>Loading image...</b>",
"error": "<emoji document_id=5116151848855667552>🚫</emoji> <b>An unexpected error occurred...</b>",
}
strings_ru = {
"img": "<emoji document_id=4916036072560919511>✅</emoji> <b>Ваша аниме-картинка</b>\n<emoji document_id=5877465816030515018>🔗</emoji> <b>Ссылка:</b> {}",
"loading": "<emoji document_id=4911241630633165627>✨</emoji> <b>Загрузка изображения...</b>",
"error": "<emoji document_id=5116151848855667552>🚫</emoji> <b>Произошла непредвиденная ошибка...</b>",
}
@loader.command(
ru_doc="- получить рандомную аниме-картинку 👀"
)
async def rapiccmd(self, message):
"""- fetch random anime-pic 👀"""
await utils.answer(message, self.strings("loading"))
try:
res = requests.get("https://api.nekosia.cat/api/v1/images/cute?count=1")
res.raise_for_status()
data = res.json()
image_url = data['image']['original']['url']
await asyncio.sleep(2)
await utils.answer(message, self.strings("img").format(image_url), file=image_url, reply_to=message.reply_to_msg_id)
except Exception:
logger.error("Error fetching random anime pic: %s", traceback.format_exc())
await utils.answer(message, self.strings("error"))
await asyncio.sleep(5)

View File

@@ -0,0 +1,77 @@
# =======================================
# _ __ __ __ _
# | |/ /___ | \/ | ___ __| |___
# | ' // _ \ | |\/| |/ _ \ / _` / __|
# | . \ __/ | | | | (_) | (_| \__ \
# |_|\_\___| |_| |_|\___/ \__,_|___/
# @ke_mods
# =======================================
#
# LICENSE: CC BY-ND 4.0 (Attribution-NoDerivatives 4.0 International)
# --------------------------------------
# https://creativecommons.org/licenses/by-nd/4.0/legalcode
# =======================================
# meta developer: @ke_mods
# scope: ffmpeg
# requires: pydub SpeechRecognition
from .. import loader, utils
import os
import speech_recognition as sr
from pydub import AudioSegment
@loader.tds
class VoiceToTextMod(loader.Module):
strings = {
"name": "VoiceToText",
"process_text": "<emoji document_id=4911241630633165627>✨</emoji> <b>Recognizing the message text...</b>",
"vtt_success": "<emoji document_id=5116110535565247270>🔥</emoji> <b>Recognized text:</b>\n<blockquote expandable>{}</blockquote>",
"vtt_failure": "<emoji document_id=5116151848855667552>🚫</emoji> <b>Failed to recognize the message.</b>",
"vtt_request_error": "<emoji document_id=5116151848855667552>🚫</emoji> <b>Error when contacting the recognition service:</b>\n<code>{}</code>",
"vtt_invalid": "<emoji document_id=5116151848855667552>🚫</emoji> <b>Please reply to a voice or video message with the command</b> <code>{}vtt</code>",
"vtt_successful": "<emoji document_id=4916036072560919511>✅</emoji> <b>Text recognized successfully</b>",
}
strings_ru = {
"process_text": "<emoji document_id=4911241630633165627>✨</emoji> <b>Распознаю текст сообщения...</b>",
"vtt_success": "<emoji document_id=5116110535565247270>🔥</emoji> <b>Распознанный текст:</b>\n<blockquote expandable>{}</blockquote>",
"vtt_failure": "<emoji document_id=5116151848855667552>🚫</emoji> <b>Не удалось распознать сообщение.</b>",
"vtt_request_error": "<emoji document_id=5116151848855667552>🚫</emoji> <b>Ошибка при обращении к сервису распознавания:</b>\n<code>{}</code>",
"vtt_invalid": "<emoji document_id=5116151848855667552>🚫</emoji> <b>Пожалуйста, ответьте на голосовое или видеосообщение командой</b> <code>{}vtt</code>",
"vtt_successful": "<emoji document_id=4916036072560919511>✅</emoji> <b>Текст успешно распознан</b>",
}
@loader.command(
ru_doc="- распознает текст из голосового или видеосообщения.",
)
async def vttcmd(self, message):
"""- recognizes text from voice or video messages."""
reply = await message.get_reply_message()
if not reply or not (reply.voice or reply.video_note):
await utils.answer(message, self.strings["vtt_invalid"].format(self.get_prefix()))
return
msg = await utils.answer(
message, self.strings["process_text"], reply_to=message.id
)
media_file = await reply.download_media()
wav_file = media_file.replace('.mp4', '.wav') if reply.video_note else media_file.replace('.oga', '.wav')
try:
AudioSegment.from_file(media_file).export(wav_file, format='wav')
recognizer = sr.Recognizer()
with sr.AudioFile(wav_file) as source:
audio_data = recognizer.record(source)
try:
text = recognizer.recognize_google(audio_data, language='ru-RU')
await utils.answer(msg, self.strings["vtt_success"].format(text))
except sr.UnknownValueError:
await utils.answer(msg, self.strings["vtt_failure"])
except sr.RequestError as e:
await utils.answer(msg, self.strings["vtt_request_error"].format(e))
finally:
os.remove(media_file)
os.remove(wav_file)

View File

@@ -234,6 +234,11 @@
"url": "https://github.com/yummy1gay/limoka",
"tags": [],
"blacklist": []
},
{
"url": "https://github.com/radiocycle/Modules",
"tags": ["newbie"],
"blacklist": []
}
]
}

View File

@@ -53,6 +53,7 @@ async def send_document(session, file_path, caption=None):
data = aiohttp.FormData()
data.add_field('chat_id', arguments.chat_id)
data.add_field('document', f, filename=os.path.basename(file_path))
data.add_field('parse_mode', 'HTML')
if caption:
data.add_field('caption', caption)
data.add_field('parse_mode', 'Markdown')
@@ -133,8 +134,9 @@ async def main():
diff_url = f"https://github.com/MuRuLOSE/limoka/compare/{old_hash}...{new_hash}.diff"
message = (
f"🪼 Module {module_name} changes approved\n\n"
f"[File URL]({github_url}) | [Diff URL]({diff_url})\n\n"
f"🪼 <b>Module <code>{module_name}</code> changes approved</b>\n\n"
f"<b><a href=\"{github_url}\">File URL</a></b> | "
f"<b><a href=\"{diff_url}\">Diff URL</a></b>"
)
# Get diff

View File

@@ -1,396 +1,409 @@
__version__ = (1, 1, 1, 1)
# This file is a part of Hikka Userbot!
# This product includes software developed by t.me/Fl1yd and t.me/spypm.
# Based on the "SQuotes" module.
# 🌐 https://github.com/hikariatama/Hikka
# You CAN edit this file without direct permission from the author.
# You can redistribute this file with any modifications.
# thx to t.me/LyoSU for github.com/LyoSU/quote-api
# meta developer: @yg_modules
# scope: hikka_only
# scope: hikka_min 1.6.3
# █▄█ █░█ █▀▄▀█ █▀▄▀█ █▄█   █▀▄▀█ █▀█ █▀▄ █▀
# ░█░ █▄█ █░▀░█ █░▀░█ ░█░   █░▀░█ █▄█ █▄▀ ▄█
import base64, io, requests, telethon
from time import gmtime
from typing import List, Optional, Tuple, Union
from PIL import Image, ImageDraw
from telethon.tl import types
from telethon.extensions import html
from telethon.tl.patched import Message
from .. import loader, utils
class Dick:
@staticmethod
def ents(es: types.TypeMessageEntity) -> List[dict]:
out: List[dict] = []
if not es: return out
for e in es:
try:
d = e.to_dict(); t = d.pop("_","").replace("MessageEntity","").lower()
if not t: continue
mt = {"bold": "bold","italic": "italic","underline": "underline","strikethrough": "strikethrough",
"code": "code","pre": "pre","texturl": "text_link","url": "url","email": "email",
"phone": "phone_number","mention": "mention",
"mentionname": "text_mention","hashtag": "hashtag","cashtag": "cashtag",
"botcommand": "bot_command","spoiler": "spoiler","customemoji": "custom_emoji"}.get(t,t)
it = {"type": mt,"offset": d.get("offset",0),"length": d.get("length",0)}
if t=="texturl": it["url"]=d.get("url","")
elif t=="mentionname": it["user"]={"id": d.get("user_id",0)}
elif t=="customemoji": it["custom_emoji_id"]=str(d.get("document_id",""))
elif t=="pre": it["language"]=d.get("language","")
out.append(it)
except Exception: continue
return out
@staticmethod
def dur(s: Union[int,float]) -> str:
t=gmtime(s); return (f"{t.tm_hour:02d}:" if t.tm_hour>0 else "")+f"{t.tm_min:02d}:{t.tm_sec:02d}"
@staticmethod
def desc(m: Message, rep: bool=False) -> str:
return (
"📷 Фото" if m.photo and rep else
(m.file.emoji+" Стикер") if m.sticker and rep else
"📹 Видеосообщение" if m.video_note and rep else
"📹 Видео" if m.video and rep else
"🖼 GIF" if m.gif else
"📊 Опрос" if m.poll else
"📍 Местоположение" if m.geo else
"👤 Контакт" if m.contact else
(f"🎵 Голосовое сообщение: {Dick.dur(m.voice.attributes[0].duration)}" if m.voice else
(f"🎧 Музыка: {Dick.dur(m.audio.attributes[0].duration)} | {m.audio.attributes[0].performer} - {m.audio.attributes[0].title}" if m.audio else
(f"💾 Файл: {m.file.name}" if isinstance(m.media, types.MessageMediaDocument) and not Dick.pick(m) else
(f"{m.media.emoticon} Кость: {m.media.value}" if isinstance(m.media, types.MessageMediaDice) else
(f"Сервисное сообщение: {m.action.to_dict().get('_')}" if isinstance(m, types.MessageService) else "")))))) #)))
@staticmethod
def split(name: Optional[str]) -> Tuple[str,str]:
if not name: return "",""
p=name.split(); return (p[0], " ".join(p[1:]) if len(p)>1 else "")
@staticmethod
def pick(m: Message):
if m and m.media:
return m.photo or m.sticker or m.video or m.video_note or m.gif or m.web_preview
return None
@staticmethod
def wf(b: Optional[bytes]) -> List[int]:
if not b: return []
n=(len(b)*8)//5
if not n: return []
out: List[int]=[]
last=n-1
for i in range(last):
j=i*5; bi,sh=j//8,j%8
v=int.from_bytes(b[bi:bi+2],"little") if bi+1<len(b) else b[bi]
out.append((v>>sh)&0b11111)
j=last*5; bi,sh=j//8,j%8
v=int.from_bytes(b[bi:bi+2],"little") if bi+1<len(b) else b[bi]
out.append((v>>sh)&0b11111)
return out
@staticmethod
async def img(b: bytes, circle: bool=False) -> Optional[str]:
try:
im=Image.open(io.BytesIO(b))
if im.mode!="RGBA": im=im.convert("RGBA")
if circle:
size=min(im.size)
mask=Image.new("L",(size,size),0); ImageDraw.Draw(mask).ellipse((0,0,size,size),fill=255)
sq=Image.new("RGBA",(size,size),(0,0,0,0))
off=((size-im.width)//2,(size-im.height)//2); sq.paste(im,off)
im=Image.composite(sq,Image.new("RGBA",(size,size),(0,0,0,0)),mask)
o=io.BytesIO(); im.save(o,format="PNG")
return f"data:image/png;base64,{base64.b64encode(o.getvalue()).decode()}"
except Exception:
return None
@staticmethod
async def stc(b: bytes) -> Optional[str]:
try:
im=Image.open(io.BytesIO(b))
if im.mode not in ("RGBA","LA"): im=im.convert("RGBA")
elif im.mode=="LA": im=im.convert("RGBA")
o=io.BytesIO(); im.save(o,format="PNG")
return f"data:image/png;base64,{base64.b64encode(o.getvalue()).decode()}"
except Exception:
return None
@staticmethod
async def proc(cli, obj, m: Message) -> Optional[dict]:
try:
if m.voice:
for a in m.voice.attributes or []:
if getattr(a,"voice",False) and hasattr(a,"waveform"):
return {"voice":{"waveform":Dick.wf(a.waveform)}}
b: bytes = await cli.download_media(obj, bytes, thumb=-1)
if not b: return None
if m.sticker:
u=await Dick.stc(b); return {"url": u} if u else None
u=await Dick.img(b, circle=bool(m.video_note))
return {"url": u} if u else None
except Exception:
return None
@staticmethod
async def ava(cli, uid: int) -> Optional[str]:
try:
b=await cli.download_profile_photo(uid, bytes)
if b: return f"data:image/jpeg;base64,{base64.b64encode(b).decode()}"
except Exception: pass
return None
@staticmethod
async def post(url: str, data: dict):
try:
return await utils.run_sync(requests.post, url, json=data, timeout=30)
except Exception:
return None
@loader.tds
class Quotes(loader.Module):
"""Модуль для создания цитат из сообщений"""
strings = {"name": "yg_quotes",
"no_reply": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Нет реплая на сообщение",
"processing": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Обработка…",
"api_processing": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Ожидание ответа API…",
"api_error": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Ошибка API: {}",
"loading_media": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Отправка…",
"no_args_or_reply": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Нет аргументов или реплая",
"args_error": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Ошибка разбора аргументов. Запрос: <code>{}</code>",
"too_many_messages": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Слишком много сообщений. Максимум: <code>{}</code>"}
def __init__(self):
self.config=loader.ModuleConfig(
loader.ConfigValue("type","quote",
lambda:"Тип цитаты",
validator=loader.validators.Choice(["quote", "stories"])),
loader.ConfigValue("bg_color","#162330",
lambda:"Цвет фона цитаты (например, #1a1a1a или red)"),
loader.ConfigValue("width",512,
lambda:"Ширина цитаты (px)",
validator=loader.validators.Integer(minimum=200,maximum=2000)),
loader.ConfigValue("height",768,
lambda:"Высота цитаты (px)",
validator=loader.validators.Integer(minimum=200,maximum=2000)),
loader.ConfigValue("scale",2,
lambda:"Масштаб рендера",
validator=loader.validators.Choice([1, 2, 3])),
loader.ConfigValue("emoji_brand","apple",
lambda:"Стиль эмодзи (apple, google, twitter и т.д.)"),
loader.ConfigValue("max_messages",15,
lambda:"Максимальное число сообщений в цитате",
validator=loader.validators.Integer(minimum=1,maximum=50)),
loader.ConfigValue("endpoint","https://kok.gay/gayotes/generate",
lambda:"URL API-эндпоинта (можешь поднять локально - github.com/yummy1gay/quote-api)",
validator=loader.validators.Link()))
async def client_ready(self, client, db):
self.client=client; self.db=db
async def qcmd(self, m: Message):
"""
Обычные цитаты:
• .q — процитировать одно сообщение из реплая
• .q 2 — процитировать 2 сообщения
• .q 3 #2d2d2d — 3 сообщения на тёмном фоне
• .q pink — фон по имени цвета
• .q !file — отправить как файл (PNG)
"""
try:
args=utils.get_args(m); rep=await m.get_reply_message()
if not rep: return await utils.answer(m,self.strings["no_reply"])
st=await utils.answer(m,self.strings["processing"])
doc="!file" in args
n=next((int(a) for a in args if a.isdigit() and int(a)>0),1)
bg=next((a for a in args if a!="!file" and not a.isdigit()), self.config["bg_color"])
if n>self.config["max_messages"]:
return await utils.answer(st,self.strings["too_many_messages"].format(self.config["max_messages"]))
js=await self.parse(m,n)
if not js: return await utils.answer(st,self.strings["api_error"].format("Не удалось собрать сообщения"))
pay={"backgroundColor":bg,"width":self.config["width"],"height":self.config["height"],
"scale":self.config["scale"],"emojiBrand":self.config["emoji_brand"],"messages":js,
"format": "webp" if not doc else "png", "type": self.config["type"]}
await utils.answer(st,self.strings["api_processing"])
r=await Dick.post(f"{self.config['endpoint']}.webp",pay)
if not r or r.status_code!=200:
try: err=r.json().get("error",f"HTTP {r.status_code}") if r else "Нетворк еррорь"
except Exception: err=f"HTTP {r.status_code}" if r else "Нетворк еррорь"
return await utils.answer(st,self.strings["api_error"].format(err))
buf=io.BytesIO(r.content); buf.name="YgQuote"+(".png" if doc else ".webp")
await utils.answer(st,buf,force_document=doc)
except Exception as e:
return await utils.answer(m,f"<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Ошибка: {e}")
async def fqcmd(self, m: Message):
"""
Фейковые цитаты:
• .fq <@ или ID> <текст> — цитата от пользователя
• .fq <reply> <текст> — цитата от автора реплая
• .fq <@/ID> <текст> -r <@/ID> <текст> — с ответом
• .fq user1 текст; user2 текст — несколько сообщений
"""
try:
raw=utils.get_args_html(m); rep=await m.get_reply_message()
if not (raw or rep): return await utils.answer(m,self.strings["no_args_or_reply"])
st= await utils.answer(m,self.strings["processing"])
try: js=await self.fake(raw,rep)
except (IndexError,ValueError): return await utils.answer(st,self.strings["args_error"].format(m.text))
if len(js)>self.config["max_messages"]:
return await utils.answer(st,self.strings["too_many_messages"].format(self.config["max_messages"]))
dickk={"backgroundColor":self.config["bg_color"],"width":self.config["width"],"height":self.config["height"],
"scale":self.config["scale"],"emojiBrand":self.config["emoji_brand"],"messages":js,
"format": "webp","type":self.config["type"]}
await utils.answer(st,self.strings["api_processing"])
r=await Dick.post(f"{self.config['endpoint']}.webp",dickk)
if not r or r.status_code!=200:
try: err=r.json().get("error",f"HTTP {r.status_code}") if r else "Нетворк еррорь"
except Exception: err=f"HTTP {r.status_code}" if r else "Нетворк еррорь"
return await utils.answer(st,self.strings["api_error"].format(err))
buf=io.BytesIO(r.content); buf.name="YgQuote.webp"
await utils.answer(st,buf)
except Exception as e:
return await utils.answer(m,f"<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Ошибка: {e}")
async def parse(self, trg: Message, n: int) -> Optional[List[dict]]:
try:
rep= await trg.get_reply_message()
lst: List[Message]=[mm async for mm in self.client.iter_messages(trg.chat_id,limit=n,reverse=True,add_offset=1,offset_id=rep.id if rep else None)]
except Exception:
return None
out: List[dict]=[]
for mm in lst:
try:
u=await self.who(mm)
if not u: continue
name=telethon.utils.get_display_name(u); f,l=Dick.split(name)
ava=await Dick.ava(self.client,getattr(u,"id",0)) if getattr(u,"id",None) else None
rb=None
try:
r=await mm.get_reply_message()
if r:
rname=telethon.utils.get_display_name(r.sender)
rtxt=Dick.desc(r,True)
if r.raw_text: rtxt=(rtxt+". "+r.raw_text) if rtxt else r.raw_text
rb={"name":rname,"text":rtxt or "","entities":Dick.ents(r.entities),
"chatId": r.sender.id if r.sender else mm.chat_id,"from":{"name":rname}}
except Exception: rb=None
med=None; obj=Dick.pick(mm)
if obj: med=await Dick.proc(self.client,obj,mm)
txt=mm.raw_text or ""; ad=Dick.desc(mm)
if ad: txt=f"{txt}\n\n{ad}" if txt else ad
item={"from":{"id":getattr(u,"id", 0),"first_name":getattr(u,"first_name","") or f,"last_name":getattr(u,"last_name","") or l,
"username":getattr(u,"username",None),"name":name,"photo":{"url":ava} if ava else {}},
"text":txt,"entities":Dick.ents(mm.entities),"avatar":True}
try:
if mm.voice:
a = next((a for a in mm.voice.attributes or []
if getattr(a, "voice", False) and hasattr(a, "waveform")), None)
if a: item["voice"] = {"waveform": Dick.wf(a.waveform)}
except Exception: pass
if med: item["voice" if "voice" in med else "media"] = med.get("voice", med)
es=getattr(u,"emoji_status",None)
if getattr(es,"document_id",None): item["from"]["emoji_status"]=str(es.document_id)
if rb: item["replyMessage"]=rb
out.append(item)
except Exception: continue
return out
async def who(self, m: Message):
try:
if m.fwd_from:
if m.fwd_from.from_id:
pid=m.fwd_from.from_id
uid=pid.channel_id if isinstance(pid, types.PeerChannel) else pid.user_id
try: return await self.client.get_entity(uid)
except Exception: return m.sender
if m.fwd_from.from_name:
return types.User(
id=hash(m.fwd_from.from_name)%2147483647, first_name=m.fwd_from.from_name,
username=None, phone=None, bot=False, verified=False, restricted=False,
scam=False, fake=False, premium=False)
return m.sender
except Exception:
return m.sender
async def fake(self, args: str, rep: Optional[Message]) -> List[dict]:
async def tok(ch: str):
p=ch.split()
if not p: return None,""
who=p[0]; tx=ch.split(maxsplit=1)[1] if len(p)>1 else ""
try:
u=await self.client.get_entity(int(who) if who.isdigit() else who)
return u,tx
except Exception:
return None,tx
if rep and not args:
u=rep.sender; name=telethon.utils.get_display_name(u); f,l=Dick.split(name)
ava=await Dick.ava(self.client,u.id) if getattr(u,"id",None) else None
msg={"from":{"id":u.id,"first_name":getattr(u,"first_name","") or f,"last_name":getattr(u,"last_name","") or l,
"username":getattr(u,"username",None),"name":name,"photo":{"url":ava} if ava else {}},
"text":"","entities":[], "avatar":True}
es=getattr(u,"emoji_status",None)
if getattr(es,"document_id", None): msg["from"]["emoji_status"]=str(es.document_id)
return [msg]
if rep and args:
u=rep.sender
return await self.fake(f"{getattr(u,'id','')} {args}", None)
out: List[dict]=[]
for part in args.split("; "):
try:
rb=None
if " -r " in part:
a,b=part.split(" -r ",1); u1,t1=await tok(a); u2,t2=await tok(b)
else:
u1,t1=await tok(part); u2,t2=None,None
if not u1: continue
txt1, ents1 = html.parse(t1) if t1 else ("", [])
name=telethon.utils.get_display_name(u1); f,l=Dick.split(name)
ava=await Dick.ava(self.client,u1.id)
if u2:
txt2, ents2 = html.parse(t2) if t2 else ("", [])
name2=telethon.utils.get_display_name(u2); ava2=await Dick.ava(self.client,u2.id)
rb={"name":name2,"text":txt2,"entities":Dick.ents(ents2),"chatId":u2.id,"from":{"name":name2,"photo":{"url":ava2} if ava2 else {}}}
msg={"from":{"id":u1.id,"first_name":getattr(u1,"first_name","") or f,"last_name":getattr(u1,"last_name","") or l,
"username":getattr(u1,"username",None),"name":name,"photo":{"url":ava} if ava else {}},
"text":txt1,"entities":Dick.ents(ents1), "avatar":True}
es=getattr(u1,"emoji_status",None)
if getattr(es,"document_id",None): msg["from"]["emoji_status"]=str(es.document_id)
if rb: msg["replyMessage"]=rb
out.append(msg)
except Exception: continue
__version__ = (1, 2, 0, 0)
# This file is a part of Hikka Userbot!
# This product includes software developed by t.me/Fl1yd and t.me/spypm.
# Based on the "SQuotes" module.
# 🌐 https://github.com/hikariatama/Hikka
# You CAN edit this file without direct permission from the author.
# You can redistribute this file with any modifications.
# thx to t.me/LyoSU for github.com/LyoSU/quote-api
# meta developer: @yg_modules
# scope: hikka_only
# scope: hikka_min 1.6.3
# Changelog v1.2:
# - Added: Proxy for users from RF
# - Fixed: Correct reply author resolving for forwarded messages
# █▄█ █░█ █▀▄▀█ █▀▄▀█ █▄█   █▀▄▀█ █▀█ █▀▄ █▀
# ░█░ █▄█ █░▀░█ █░▀░█ ░█░   █░▀░█ █▄█ █▄▀ ▄█
import base64, io, requests, telethon
from time import gmtime
from typing import List, Optional, Tuple, Union
from PIL import Image, ImageDraw
from telethon.tl import types
from telethon.extensions import html
from telethon.tl.patched import Message
from .. import loader, utils
class Dick:
@staticmethod
def ents(es: types.TypeMessageEntity) -> List[dict]:
out: List[dict] = []
if not es: return out
for e in es:
try:
d = e.to_dict(); t = d.pop("_","").replace("MessageEntity","").lower()
if not t: continue
mt = {"bold": "bold","italic": "italic","underline": "underline","strikethrough": "strikethrough",
"code": "code","pre": "pre","texturl": "text_link","url": "url","email": "email",
"phone": "phone_number","mention": "mention",
"mentionname": "text_mention","hashtag": "hashtag","cashtag": "cashtag",
"botcommand": "bot_command","spoiler": "spoiler","customemoji": "custom_emoji"}.get(t,t)
it = {"type": mt,"offset": d.get("offset",0),"length": d.get("length",0)}
if t=="texturl": it["url"]=d.get("url","")
elif t=="mentionname": it["user"]={"id": d.get("user_id",0)}
elif t=="customemoji": it["custom_emoji_id"]=str(d.get("document_id",""))
elif t=="pre": it["language"]=d.get("language","")
out.append(it)
except Exception: continue
return out
@staticmethod
def dur(s: Union[int,float]) -> str:
t=gmtime(s); return (f"{t.tm_hour:02d}:" if t.tm_hour>0 else "")+f"{t.tm_min:02d}:{t.tm_sec:02d}"
@staticmethod
def desc(m: Message, rep: bool=False) -> str:
return (
"📷 Фото" if m.photo and rep else
(m.file.emoji+" Стикер") if m.sticker and rep else
"📹 Видеосообщение" if m.video_note and rep else
"📹 Видео" if m.video and rep else
"🖼 GIF" if m.gif else
"📊 Опрос" if m.poll else
"📍 Местоположение" if m.geo else
"👤 Контакт" if m.contact else
(f"🎵 Голосовое сообщение: {Dick.dur(m.voice.attributes[0].duration)}" if m.voice else
(f"🎧 Музыка: {Dick.dur(m.audio.attributes[0].duration)} | {m.audio.attributes[0].performer} - {m.audio.attributes[0].title}" if m.audio else
(f"💾 Файл: {m.file.name}" if isinstance(m.media, types.MessageMediaDocument) and not Dick.pick(m) else
(f"{m.media.emoticon} Кость: {m.media.value}" if isinstance(m.media, types.MessageMediaDice) else
(f"Сервисное сообщение: {m.action.to_dict().get('_')}" if isinstance(m, types.MessageService) else "")))))) #)))
@staticmethod
def split(name: Optional[str]) -> Tuple[str,str]:
if not name: return "",""
p=name.split(); return (p[0], " ".join(p[1:]) if len(p)>1 else "")
@staticmethod
def pick(m: Message):
if m and m.media:
return m.photo or m.sticker or m.video or m.video_note or m.gif or m.web_preview
return None
@staticmethod
def wf(b: Optional[bytes]) -> List[int]:
if not b: return []
n=(len(b)*8)//5
if not n: return []
out: List[int]=[]
last=n-1
for i in range(last):
j=i*5; bi,sh=j//8,j%8
v=int.from_bytes(b[bi:bi+2],"little") if bi+1<len(b) else b[bi]
out.append((v>>sh)&0b11111)
j=last*5; bi,sh=j//8,j%8
v=int.from_bytes(b[bi:bi+2],"little") if bi+1<len(b) else b[bi]
out.append((v>>sh)&0b11111)
return out
@staticmethod
async def img(b: bytes, circle: bool=False) -> Optional[str]:
try:
im=Image.open(io.BytesIO(b))
if im.mode!="RGBA": im=im.convert("RGBA")
if circle:
size=min(im.size)
mask=Image.new("L",(size,size),0); ImageDraw.Draw(mask).ellipse((0,0,size,size),fill=255)
sq=Image.new("RGBA",(size,size),(0,0,0,0))
off=((size-im.width)//2,(size-im.height)//2); sq.paste(im,off)
im=Image.composite(sq,Image.new("RGBA",(size,size),(0,0,0,0)),mask)
o=io.BytesIO(); im.save(o,format="PNG")
return f"data:image/png;base64,{base64.b64encode(o.getvalue()).decode()}"
except Exception:
return None
@staticmethod
async def stc(b: bytes) -> Optional[str]:
try:
im=Image.open(io.BytesIO(b))
if im.mode not in ("RGBA","LA"): im=im.convert("RGBA")
elif im.mode=="LA": im=im.convert("RGBA")
o=io.BytesIO(); im.save(o,format="PNG")
return f"data:image/png;base64,{base64.b64encode(o.getvalue()).decode()}"
except Exception:
return None
@staticmethod
async def proc(cli, obj, m: Message) -> Optional[dict]:
try:
if m.voice:
for a in m.voice.attributes or []:
if getattr(a,"voice",False) and hasattr(a,"waveform"):
return {"voice":{"waveform":Dick.wf(a.waveform)}}
b: bytes = await cli.download_media(obj, bytes, thumb=-1)
if not b: return None
if m.sticker:
u=await Dick.stc(b); return {"url": u} if u else None
u=await Dick.img(b, circle=bool(m.video_note))
return {"url": u} if u else None
except Exception:
return None
@staticmethod
async def ava(cli, uid: int) -> Optional[str]:
try:
b=await cli.download_profile_photo(uid, bytes)
if b: return f"data:image/jpeg;base64,{base64.b64encode(b).decode()}"
except Exception: pass
return None
@staticmethod
async def post(url: str, data: dict):
try:
return await utils.run_sync(requests.post, url, json=data, timeout=30)
except Exception:
return None
@loader.tds
class Quotes(loader.Module):
"""Модуль для создания цитат из сообщений"""
strings = {"name": "yg_quotes",
"no_reply": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Нет реплая на сообщение",
"processing": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Обработка…",
"api_processing": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Ожидание ответа API…",
"api_error": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Ошибка API: {}",
"loading_media": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Отправка…",
"no_args_or_reply": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Нет аргументов или реплая",
"args_error": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Ошибка разбора аргументов. Запрос: <code>{}</code>",
"too_many_messages": "<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Слишком много сообщений. Максимум: <code>{}</code>"}
def __init__(self):
self.config=loader.ModuleConfig(
loader.ConfigValue("type","quote",
lambda:"Тип цитаты",
validator=loader.validators.Choice(["quote", "stories"])),
loader.ConfigValue("bg_color","#162330",
lambda:"Цвет фона цитаты (например, #1a1a1a или red)"),
loader.ConfigValue("width",512,
lambda:"Ширина цитаты (px)",
validator=loader.validators.Integer(minimum=200,maximum=2000)),
loader.ConfigValue("height",768,
lambda:"Высота цитаты (px)",
validator=loader.validators.Integer(minimum=200,maximum=2000)),
loader.ConfigValue("scale",2,
lambda:"Масштаб рендера",
validator=loader.validators.Choice([1, 2, 3])),
loader.ConfigValue("emoji_brand","apple",
lambda:"Стиль эмодзи (apple, google, twitter и т.д.)"),
loader.ConfigValue("max_messages",15,
lambda:"Максимальное число сообщений в цитате",
validator=loader.validators.Integer(minimum=1,maximum=50)),
loader.ConfigValue("endpoint","https://kok.gay/gayotes/generate",
lambda:"URL API-эндпоинта (можешь поднять локально - github.com/yummy1gay/quote-api)",
validator=loader.validators.Link()),
loader.ConfigValue("use_rf_proxy", False,
lambda:'Включает прокси для РФ, если основной эндпоинт возвращает ошибку "Нетворк еррорь", и при этом сервер с юзерботом находится в России или ты сам сидишь в России с ограниченным доступом к зарубежным ресурсам (Termux / UserLAnd)',
validator=loader.validators.Boolean()),
loader.ConfigValue("rf_endpoint", "https://ru.kok.gay/gayotes/generate",
lambda:"URL API-эндпоинта для РФ",
validator=loader.validators.Link()))
async def client_ready(self, client, db):
self.client=client; self.db=db
async def qcmd(self, m: Message):
"""
Обычные цитаты:
• .q — процитировать одно сообщение из реплая
• .q 2 — процитировать 2 сообщения
• .q 3 #2d2d2d — 3 сообщения на тёмном фоне
• .q pink — фон по имени цвета
• .q !file — отправить как файл (PNG)
"""
try:
args=utils.get_args(m); rep=await m.get_reply_message()
if not rep: return await utils.answer(m,self.strings["no_reply"])
st=await utils.answer(m,self.strings["processing"])
doc="!file" in args
n=next((int(a) for a in args if a.isdigit() and int(a)>0),1)
bg=next((a for a in args if a!="!file" and not a.isdigit()), self.config["bg_color"])
if n>self.config["max_messages"]:
return await utils.answer(st,self.strings["too_many_messages"].format(self.config["max_messages"]))
js=await self.parse(m,n)
if not js: return await utils.answer(st,self.strings["api_error"].format("Не удалось собрать сообщения"))
pay={"backgroundColor":bg,"width":self.config["width"],"height":self.config["height"],
"scale":self.config["scale"],"emojiBrand":self.config["emoji_brand"],"messages":js,
"format": "webp" if not doc else "png", "type": self.config["type"]}
await utils.answer(st,self.strings["api_processing"])
endpoint=self.config['rf_endpoint'] if self.config['use_rf_proxy'] else self.config['endpoint']
r=await Dick.post(f"{endpoint}.webp",pay)
if not r or r.status_code!=200:
try: err=r.json().get("error",f"HTTP {r.status_code}") if r else "Нетворк еррорь (попробуй включить <code>use_rf_proxy</code> в конфиге)"
except Exception: err=f"HTTP {r.status_code}" if r else "Нетворк еррорь (попробуй включить <code>use_rf_proxy</code> в конфиге)"
return await utils.answer(st,self.strings["api_error"].format(err))
buf=io.BytesIO(r.content); buf.name="YgQuote"+(".png" if doc else ".webp")
await utils.answer(st,buf,force_document=doc)
except Exception as e:
return await utils.answer(m,f"<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Ошибка: {e}")
async def fqcmd(self, m: Message):
"""
Фейковые цитаты:
• .fq <@ или ID> <текст> — цитата от пользователя
• .fq <reply> <текст> — цитата от автора реплая
• .fq <@/ID> <текст> -r <@/ID> <текст> — с ответом
• .fq user1 текст; user2 текст — несколько сообщений
"""
try:
raw=utils.get_args_html(m); rep=await m.get_reply_message()
if not (raw or rep): return await utils.answer(m,self.strings["no_args_or_reply"])
st= await utils.answer(m,self.strings["processing"])
try: js=await self.fake(raw,rep)
except (IndexError,ValueError): return await utils.answer(st,self.strings["args_error"].format(m.text))
if len(js)>self.config["max_messages"]:
return await utils.answer(st,self.strings["too_many_messages"].format(self.config["max_messages"]))
dickk={"backgroundColor":self.config["bg_color"],"width":self.config["width"],"height":self.config["height"],
"scale":self.config["scale"],"emojiBrand":self.config["emoji_brand"],"messages":js,
"format": "webp","type":self.config["type"]}
await utils.answer(st,self.strings["api_processing"])
endpoint=self.config['rf_endpoint'] if self.config['use_rf_proxy'] else self.config['endpoint']
r=await Dick.post(f"{endpoint}.webp",dickk)
if not r or r.status_code!=200:
try: err=r.json().get("error",f"HTTP {r.status_code}") if r else "Нетворк еррорь (попробуй включить <code>use_rf_proxy</code> в конфиге)"
except Exception: err=f"HTTP {r.status_code}" if r else "Нетворк еррорь (попробуй включить <code>use_rf_proxy</code> в конфиге)"
return await utils.answer(st,self.strings["api_error"].format(err))
buf=io.BytesIO(r.content); buf.name="YgQuote.webp"
await utils.answer(st,buf)
except Exception as e:
return await utils.answer(m,f"<emoji document_id=6321272741005624970>🏳️‍🌈</emoji> Ошибка: {e}")
async def parse(self, trg: Message, n: int) -> Optional[List[dict]]:
try:
rep= await trg.get_reply_message()
lst: List[Message]=[mm async for mm in self.client.iter_messages(trg.chat_id,limit=n,reverse=True,add_offset=1,offset_id=rep.id if rep else None)]
except Exception:
return None
out: List[dict]=[]
for mm in lst:
try:
u=await self.who(mm)
if not u: continue
name=telethon.utils.get_display_name(u); f,l=Dick.split(name)
ava=await Dick.ava(self.client,getattr(u,"id",0)) if getattr(u,"id",None) else None
rb=None
try:
r=await mm.get_reply_message()
if r:
ruser = await self.who(r)
rname=telethon.utils.get_display_name(ruser)
rtxt=Dick.desc(r,True)
if r.raw_text: rtxt=(rtxt+". "+r.raw_text) if rtxt else r.raw_text
rb={"name":rname,"text":rtxt or "","entities":Dick.ents(r.entities),
"chatId": r.sender.id if r.sender else mm.chat_id,"from":{"name":rname}}
except Exception: rb=None
med=None; obj=Dick.pick(mm)
if obj: med=await Dick.proc(self.client,obj,mm)
txt=mm.raw_text or ""; ad=Dick.desc(mm)
if ad: txt=f"{txt}\n\n{ad}" if txt else ad
item={"from":{"id":getattr(u,"id", 0),"first_name":getattr(u,"first_name","") or f,"last_name":getattr(u,"last_name","") or l,
"username":getattr(u,"username",None),"name":name,"photo":{"url":ava} if ava else {}},
"text":txt,"entities":Dick.ents(mm.entities),"avatar":True}
try:
if mm.voice:
a = next((a for a in mm.voice.attributes or []
if getattr(a, "voice", False) and hasattr(a, "waveform")), None)
if a: item["voice"] = {"waveform": Dick.wf(a.waveform)}
except Exception: pass
if med: item["voice" if "voice" in med else "media"] = med.get("voice", med)
es=getattr(u,"emoji_status",None)
if getattr(es,"document_id",None): item["from"]["emoji_status"]=str(es.document_id)
if rb: item["replyMessage"]=rb
out.append(item)
except Exception: continue
return out
async def who(self, m: Message):
try:
if m.fwd_from:
if m.fwd_from.from_id:
pid=m.fwd_from.from_id
uid=pid.channel_id if isinstance(pid, types.PeerChannel) else pid.user_id
try: return await self.client.get_entity(uid)
except Exception: return m.sender
if m.fwd_from.from_name:
return types.User(
id=hash(m.fwd_from.from_name)%2147483647, first_name=m.fwd_from.from_name,
username=None, phone=None, bot=False, verified=False, restricted=False,
scam=False, fake=False, premium=False)
return m.sender
except Exception:
return m.sender
async def fake(self, args: str, rep: Optional[Message]) -> List[dict]:
async def tok(ch: str):
p=ch.split()
if not p: return None,""
who=p[0]; tx=ch.split(maxsplit=1)[1] if len(p)>1 else ""
try:
u=await self.client.get_entity(int(who) if who.isdigit() else who)
return u,tx
except Exception:
return None,tx
if rep and not args:
u=rep.sender; name=telethon.utils.get_display_name(u); f,l=Dick.split(name)
ava=await Dick.ava(self.client,u.id) if getattr(u,"id",None) else None
msg={"from":{"id":u.id,"first_name":getattr(u,"first_name","") or f,"last_name":getattr(u,"last_name","") or l,
"username":getattr(u,"username",None),"name":name,"photo":{"url":ava} if ava else {}},
"text":"","entities":[], "avatar":True}
es=getattr(u,"emoji_status",None)
if getattr(es,"document_id", None): msg["from"]["emoji_status"]=str(es.document_id)
return [msg]
if rep and args:
u=rep.sender
return await self.fake(f"{getattr(u,'id','')} {args}", None)
out: List[dict]=[]
for part in args.split("; "):
try:
rb=None
if " -r " in part:
a,b=part.split(" -r ",1); u1,t1=await tok(a); u2,t2=await tok(b)
else:
u1,t1=await tok(part); u2,t2=None,None
if not u1: continue
txt1, ents1 = html.parse(t1) if t1 else ("", [])
name=telethon.utils.get_display_name(u1); f,l=Dick.split(name)
ava=await Dick.ava(self.client,u1.id)
if u2:
txt2, ents2 = html.parse(t2) if t2 else ("", [])
name2=telethon.utils.get_display_name(u2); ava2=await Dick.ava(self.client,u2.id)
rb={"name":name2,"text":txt2,"entities":Dick.ents(ents2),"chatId":u2.id,"from":{"name":name2,"photo":{"url":ava2} if ava2 else {}}}
msg={"from":{"id":u1.id,"first_name":getattr(u1,"first_name","") or f,"last_name":getattr(u1,"last_name","") or l,
"username":getattr(u1,"username",None),"name":name,"photo":{"url":ava} if ava else {}},
"text":txt1,"entities":Dick.ents(ents1), "avatar":True}
es=getattr(u1,"emoji_status",None)
if getattr(es,"document_id",None): msg["from"]["emoji_status"]=str(es.document_id)
if rb: msg["replyMessage"]=rb
out.append(msg)
except Exception: continue
return out