mirror of
https://github.com/MuRuLOSE/limoka.git
synced 2026-06-18 07:04:19 +02:00
Compare commits
53 Commits
update-sub
...
update-sub
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
430f49f345 | ||
| 54229a8266 | |||
|
|
6cb625acd8 | ||
|
|
bf03595075 | ||
|
|
8bef604170 | ||
| 61ebb7fa47 | |||
|
|
2702cd2356 | ||
|
|
37f53375bc | ||
|
|
23dade9b9e | ||
|
|
ab5aaf579b | ||
|
|
170809519a | ||
| 78a9b0cb76 | |||
|
|
c1d407f885 | ||
|
|
cf230d2922 | ||
| 7cda8f2685 | |||
|
|
92bdd1510d | ||
|
|
52f6fc53c3 | ||
|
|
5c25eaad81 | ||
| a7b8c740b4 | |||
| 44ca8633c6 | |||
| e076d4ab28 | |||
|
|
634861943a | ||
|
|
d20d019c32 | ||
|
|
041e6ec5f7 | ||
| f68d97d09b | |||
| e624861b64 | |||
|
|
d3f035ec58 | ||
|
|
1ced5efae6 | ||
|
|
3bf476c8a9 | ||
|
|
abff91c013 | ||
| 5c2de322d3 | |||
| 5b2fe145ff | |||
| 64d8ceb354 | |||
| 537ae4c485 | |||
| 52018c8e14 | |||
| 6f95292df1 | |||
| 4a5d4883a9 | |||
| ce59b13c94 | |||
| 91351cb83d | |||
|
|
118a800856 | ||
|
|
f14a99c640 | ||
|
|
c76dcc0fc1 | ||
| 51bbb98711 | |||
| edd70ad020 | |||
|
|
2c33b08ea0 | ||
|
|
e9bb89cf62 | ||
|
|
51055e6427 | ||
| 88ab265755 | |||
|
|
baada2d019 | ||
|
|
ad782a6f46 | ||
|
|
2435df880e | ||
| 2dd772b52d | |||
| c9ed00bb78 |
22
.github/workflows/ci.yml
vendored
22
.github/workflows/ci.yml
vendored
@@ -10,8 +10,10 @@ on:
|
||||
pull_request:
|
||||
branches:
|
||||
- main
|
||||
types: [opened, synchronize, reopened, closed]
|
||||
workflow_dispatch: # Allows manual triggering from GitHub UI
|
||||
|
||||
|
||||
# Environment variables available to all jobs
|
||||
env:
|
||||
BRANCH_NAME: "update-submodules_${{ github.sha }}"
|
||||
@@ -137,9 +139,8 @@ jobs:
|
||||
python3 -m venv venv
|
||||
source venv/bin/activate
|
||||
pip install --upgrade pip
|
||||
pip install requests scikit-learn tqdm
|
||||
pip install requests
|
||||
python3 parse.py
|
||||
python3 categories.py
|
||||
git add modules.json
|
||||
git commit -m "Updated modules.json after parse $(date +'%Y-%m-%d %H:%M:%S')" || echo "No changes for modules.json"
|
||||
git remote set-url origin "https://x-access-token:${GITHUB_TOKEN}@${REPO_URL}"
|
||||
@@ -208,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]"
|
||||
@@ -225,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
589
Limoka.py
@@ -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
1420
LimokaLegacy.py
Normal file
File diff suppressed because it is too large
Load Diff
@@ -23,4 +23,4 @@ class K(loader.Module):
|
||||
"""K"""
|
||||
raise Exception("Testing error handling")
|
||||
await utils.answer(message, "K")
|
||||
# why
|
||||
# why FUCK YOU BILL GATES ЛЛЛЛ
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
267
ZetGoHack/nullmod/Gradientor.py
Normal file
267
ZetGoHack/nullmod/Gradientor.py
Normal 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)
|
||||
@@ -1,3 +1,4 @@
|
||||
Chess
|
||||
HaremManager
|
||||
SchedulePlus
|
||||
SchedulePlus
|
||||
Gradientor
|
||||
|
||||
126
categories.py
126
categories.py
@@ -1,126 +0,0 @@
|
||||
import json
|
||||
from sklearn.feature_extraction.text import TfidfVectorizer
|
||||
from sklearn.linear_model import LogisticRegression
|
||||
from sklearn.multiclass import OneVsRestClassifier
|
||||
from sklearn.preprocessing import MultiLabelBinarizer
|
||||
from tqdm import tqdm
|
||||
import numpy as np
|
||||
|
||||
# Тренировочные данные (48 модулей)
|
||||
training_data = {
|
||||
"MuRuLOSE/HikkaModulesRepo/filters.py": ["Tools", "Chat"],
|
||||
"MuRuLOSE/HikkaModulesRepo/autogiveawayjoin.py": ["Automation", "Social"],
|
||||
"MuRuLOSE/HikkaModulesRepo/HTTPCat.py": ["Fun"],
|
||||
"MuRuLOSE/HikkaModulesRepo/CustomPing.py": ["Tools", "Networking"],
|
||||
"MuRuLOSE/HikkaModulesRepo/FuckTagOne.py": ["Moderation"],
|
||||
"MuRuLOSE/HikkaModulesRepo/InlineButtons.py": ["Tools", "Chat"],
|
||||
"MuRuLOSE/HikkaModulesRepo/YoutubeDL.py": ["Media"],
|
||||
"MuRuLOSE/HikkaModulesRepo/youtubesearcher.py": ["Media", "Tools"],
|
||||
"MuRuLOSE/HikkaModulesRepo/INumber.py": ["Fun", "Info"],
|
||||
"MuRuLOSE/HikkaModulesRepo/RandomDog.py": ["Fun"],
|
||||
"MuRuLOSE/HikkaModulesRepo/RemoveLinks.py": ["Moderation", "Chat"],
|
||||
"MuRuLOSE/HikkaModulesRepo/SteamClient.py": ["Games", "Tools"],
|
||||
"MuRuLOSE/HikkaModulesRepo/PinMoreChats.py": ["Chat", "Productivity"],
|
||||
"MuRuLOSE/HikkaModulesRepo/MindGameCheat.py": ["Games", "Tools"],
|
||||
"MuRuLOSE/HikkaModulesRepo/NasaImages.py": ["Media", "Info"],
|
||||
"MuRuLOSE/HikkaModulesRepo/autoreader.py": ["Automation", "Chat"],
|
||||
"MuRuLOSE/HikkaModulesRepo/K.py": ["Fun"],
|
||||
"MuRuLOSE/HikkaModulesRepo/Genshin.py": ["Games"],
|
||||
"MuRuLOSE/HikkaModulesRepo/compliments.py": ["Social", "Fun"],
|
||||
"MuRuLOSE/HikkaModulesRepo/AutoLeave.py": ["Automation", "Chat"],
|
||||
"MuRuLOSE/HikkaModulesRepo/ToTHosting.py": ["Tools", "Admin"],
|
||||
"MuRuLOSE/HikkaModulesRepo/PasswordUtils.py": ["Security", "Tools"],
|
||||
"MuRuLOSE/HikkaModulesRepo/FuckJoins.py": ["Security", "Chat"],
|
||||
"MuRuLOSE/HikkaModulesRepo/SpyEVO.py": ["Tools", "Info"],
|
||||
"MuRuLOSE/HikkaModulesRepo/FindID.py": ["Tools", "Admin"],
|
||||
"MuRuLOSE/HikkaModulesRepo/ChannelCheck.py": ["Tools", "Social"],
|
||||
"MuRuLOSE/HikkaModulesRepo/controlspam.py": ["Chat", "Tools"],
|
||||
"MuRuLOSE/HikkaModulesRepo/VKMusic.py": ["Media"],
|
||||
"MuRuLOSE/HikkaModulesRepo/morse.py": ["Tools", "Fun"],
|
||||
"MuRuLOSE/HikkaModulesRepo/YamiManager.py": ["Chat", "Tools"],
|
||||
"MuRuLOSE/HikkaModulesRepo/SearchersGenQuery.py": ["Tools", "Info"],
|
||||
"MuRuLOSE/HikkaModulesRepo/Limoka.py": ["Utilities", "Tools"],
|
||||
"MuRuLOSE/HikkaModulesRepo/CheckTime.py": ["Productivity", "Info"],
|
||||
"MuRuLOSE/HikkaModulesRepo/ReplaceWords.py": ["Chat", "Customization"],
|
||||
"MuRuLOSE/HikkaModulesRepo/TempJoinChannel.py": ["Chat", "Automation"],
|
||||
"MuRuLOSE/HikkaModulesRepo/timer.py": ["Productivity", "Tools"],
|
||||
"den4ikSuperOstryyPer4ik/astro-modules/astroafk.py": ["Automation", "Customization"],
|
||||
"den4ikSuperOstryyPer4ik/astro-modules/akinator.py": ["Games"],
|
||||
"den4ikSuperOstryyPer4ik/astro-modules/Emotions.py": ["Social", "Fun"],
|
||||
"den4ikSuperOstryyPer4ik/astro-modules/RandomStatuses.py": ["Social", "Fun"],
|
||||
"den4ikSuperOstryyPer4ik/astro-modules/RandomTrack.py": ["Media", "Fun"],
|
||||
"den4ikSuperOstryyPer4ik/astro-modules/minesweeper.py": ["Games"],
|
||||
"den4ikSuperOstryyPer4ik/astro-modules/inline_bot_manager.py": ["Tools", "Automation"],
|
||||
"MuRuLOSE/HikkaModulesRepo/ReplaceWords.py": ["Customization", "Chat"],
|
||||
"MuRuLOSE/HikkaModulesRepo/CheckTime.py": ["Productivity"],
|
||||
"MuRuLOSE/HikkaModulesRepo/SearchersGenQuery.py": ["Utilities", "Info"]
|
||||
}
|
||||
|
||||
all_categories = [
|
||||
"Utilities", "Fun", "Admin", "Media", "Games", "Tools", "Security", "Social",
|
||||
"Automation", "Info", "Chat", "Moderation", "Productivity", "Customization",
|
||||
"Networking", "Education", "Finance", "Health", "Creative", "Other"
|
||||
]
|
||||
|
||||
def get_module_text(module_path, module_data):
|
||||
name = module_data.get("name", "").lower()
|
||||
description = (module_data.get("description", "") or module_data.get("meta", {}).get("desc", "")).lower()
|
||||
if not description or description == "desc":
|
||||
description = ""
|
||||
commands_text = " ".join([f"{cmd} {desc}".lower() for func in module_data.get("commands", []) for cmd, desc in func.items()])
|
||||
new_commands_text = " ".join([f"{cmd} {data.get('doc', '')} {data.get('ru_doc', '') or ''}".lower()
|
||||
for func in module_data.get("new_commands", []) for cmd, data in func.items()])
|
||||
file_path = module_path.lower()
|
||||
file_name = file_path.split("/")[-1]
|
||||
return f"{file_name} {name} {description} {file_path} {commands_text} {new_commands_text}".strip()
|
||||
|
||||
|
||||
with open("modules.json", "r", encoding="utf-8") as f:
|
||||
modules = json.load(f)
|
||||
|
||||
# Подготовка тренировочных данных
|
||||
train_texts = [get_module_text(path, modules[path]) for path in training_data.keys() if path in modules]
|
||||
train_labels = [training_data[path] for path in training_data.keys() if path in modules]
|
||||
|
||||
# Векторизация текста
|
||||
vectorizer = TfidfVectorizer(max_features=2000)
|
||||
X_train = vectorizer.fit_transform(train_texts)
|
||||
|
||||
# Преобразование меток
|
||||
mlb = MultiLabelBinarizer(classes=all_categories)
|
||||
y_train = mlb.fit_transform(train_labels)
|
||||
|
||||
# Обучение модели с балансировкой классов
|
||||
base_clf = LogisticRegression(class_weight="balanced", max_iter=1000)
|
||||
clf = OneVsRestClassifier(base_clf)
|
||||
clf.fit(X_train, y_train)
|
||||
|
||||
# Обработка всех модулей
|
||||
print("Classifying all modules...")
|
||||
texts = [get_module_text(path, data) for path, data in modules.items()]
|
||||
X_all = vectorizer.transform(texts)
|
||||
|
||||
# Предсказание вероятностей
|
||||
probs = clf.predict_proba(X_all)
|
||||
|
||||
# Присваивание категорий
|
||||
threshold = 0.2 # Снижаем порог для большего разнообразия
|
||||
for module_path, prob_vector in tqdm(zip(modules.keys(), probs), total=len(modules), desc="Assigning categories"):
|
||||
module_data = modules[module_path]
|
||||
sorted_indices = np.argsort(prob_vector)[::-1]
|
||||
sorted_probs = prob_vector[sorted_indices]
|
||||
sorted_labels = mlb.classes_[sorted_indices]
|
||||
|
||||
selected_categories = [label for label, prob in zip(sorted_labels, sorted_probs) if prob >= threshold][:2]
|
||||
|
||||
if not selected_categories:
|
||||
selected_categories = ["Other"]
|
||||
|
||||
module_data["category"] = selected_categories
|
||||
print(f"Module: {module_path} -> Categories: {selected_categories} (top probs: {[f'{p:.2f}' for p in sorted_probs[:3]]})")
|
||||
|
||||
# Сохранение результата
|
||||
with open("modules.json", "w", encoding="utf-8") as f:
|
||||
json.dump(modules, f, ensure_ascii=False, indent=2)
|
||||
|
||||
print("Done! Check modules_categorized.json.")
|
||||
@@ -11,7 +11,7 @@ def parse_repos(file_path: str) -> list:
|
||||
with open(file_path, "r", encoding="utf-8") as f:
|
||||
data = json.load(f)
|
||||
for repo in data.get("repositories", []):
|
||||
repos.append(repo["path"])
|
||||
repos.append(repo["url"])
|
||||
return repos
|
||||
|
||||
repos = parse_repos("repositories.json")
|
||||
|
||||
@@ -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)
|
||||
)
|
||||
|
||||
@@ -17,10 +17,9 @@ figlet
|
||||
promoclaimer
|
||||
passwordgen
|
||||
send
|
||||
lastfm
|
||||
dbmod
|
||||
chatmodule
|
||||
stats
|
||||
tagwatcher
|
||||
hardspam
|
||||
YaMusic
|
||||
YaMusic
|
||||
|
||||
@@ -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}"))
|
||||
@@ -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,
|
||||
|
||||
@@ -106,7 +106,7 @@ class DeviceInfo(loader.Module):
|
||||
self.config = loader.ModuleConfig(
|
||||
loader.ConfigValue(
|
||||
"api_base_url",
|
||||
"https://mobilespecs.fiksofficial.fun",
|
||||
"https://gmsarena.vercel.app/",
|
||||
lambda: "API Url",
|
||||
validator=loader.validators.String()
|
||||
),
|
||||
|
||||
@@ -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))
|
||||
|
||||
126347
modules.json
126347
modules.json
File diff suppressed because one or more lines are too long
31
parse.py
31
parse.py
@@ -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)
|
||||
)
|
||||
|
||||
|
||||
214
radiocycle/Modules/LastFm.py
Normal file
214
radiocycle/Modules/LastFm.py
Normal 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>")
|
||||
48
radiocycle/Modules/Neofetch.py
Normal file
48
radiocycle/Modules/Neofetch.py
Normal 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"))
|
||||
|
||||
211
radiocycle/Modules/PicToStories.py
Normal file
211
radiocycle/Modules/PicToStories.py
Normal 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"))
|
||||
1436
radiocycle/Modules/SpotifyMod.py
Normal file
1436
radiocycle/Modules/SpotifyMod.py
Normal file
File diff suppressed because it is too large
Load Diff
74
radiocycle/Modules/UnbanAll.py
Normal file
74
radiocycle/Modules/UnbanAll.py
Normal 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"))
|
||||
7
radiocycle/Modules/full.txt
Normal file
7
radiocycle/Modules/full.txt
Normal file
@@ -0,0 +1,7 @@
|
||||
Neofetch
|
||||
randomanimepic
|
||||
SpotifyMod
|
||||
UnbanAll
|
||||
voicetotext
|
||||
LastFm
|
||||
PicToStories
|
||||
65
radiocycle/Modules/randomanimepic.py
Normal file
65
radiocycle/Modules/randomanimepic.py
Normal 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)
|
||||
77
radiocycle/Modules/voicetotext.py
Normal file
77
radiocycle/Modules/voicetotext.py
Normal 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)
|
||||
@@ -1,239 +1,244 @@
|
||||
{
|
||||
"repositories": [
|
||||
{
|
||||
"path": "DziruModules/hikkamods",
|
||||
"url": "https://github.com/DziruModules/hikkamods",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "kamolgks/Hikkamods",
|
||||
"url": "https://github.com/kamolgks/Hikkamods",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "thomasmod/hikkamods",
|
||||
"url": "https://github.com/thomasmod/hikkamods",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "SkillsAngels/Modules",
|
||||
"url": "https://github.com/SkillsAngels/Modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "Sad0ff/modules-ftg",
|
||||
"url": "https://github.com/Sad0ff/modules-ftg",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "Yahikoro/Modules-for-FTG",
|
||||
"url": "https://github.com/Yahikoro/Modules-for-FTG",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "KeyZenD/modules",
|
||||
"url": "https://github.com/KeyZenD/modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "AlpacaGang/ftg-modules",
|
||||
"url": "https://github.com/AlpacaGang/ftg-modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "trololo65/Modules",
|
||||
"url": "https://github.com/trololo65/Modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "Ijidishurka/modules",
|
||||
"url": "https://github.com/Ijidishurka/modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "Fl1yd/FTG-Modules",
|
||||
"url": "https://github.com/Fl1yd/FTG-Modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "D4n13l3k00/FTG-Modules",
|
||||
"url": "https://github.com/D4n13l3k00/FTG-Modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "iamnalinor/FTG-modules",
|
||||
"url": "https://github.com/iamnalinor/FTG-modules",
|
||||
"tags": ["hikkatrusted", "nonactive"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "SekaiYoneya/modules",
|
||||
"url": "https://github.com/SekaiYoneya/modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "GeekTG/FTG-Modules",
|
||||
"url": "https://github.com/GeekTG/FTG-Modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "Den4ikSuperOstryyPer4ik/Astro-modules",
|
||||
"url": "https://github.com/Den4ikSuperOstryyPer4ik/Astro-modules",
|
||||
"tags": ["hikkatrusted", "herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "vsecoder/hikka_modules",
|
||||
"url": "https://github.com/vsecoder/hikka_modules",
|
||||
"tags": ["hikkatrusted", "herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "sqlmerr/hikka_mods",
|
||||
"url": "https://github.com/sqlmerr/hikka_mods",
|
||||
"tags": ["hikkatrusted", "herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "N3rcy/modules",
|
||||
"url": "https://github.com/N3rcy/modules",
|
||||
"tags": ["hikkatrusted", "herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "KorenbZla/HikkaModules",
|
||||
"url": "https://github.com/KorenbZla/HikkaModules",
|
||||
"tags": ["hikkatrusted", "herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "MuRuLOSE/HikkaModulesRepo",
|
||||
"url": "https://github.com/MuRuLOSE/HikkaModulesRepo",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "coddrago/modules",
|
||||
"url": "https://github.com/coddrago/modules",
|
||||
"tags": ["herokutrusted", "hikkatrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "1jpshiro/hikka-modules",
|
||||
"url": "https://github.com/1jpshiro/hikka-modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "MoriSummerz/ftg-mods",
|
||||
"url": "https://github.com/MoriSummerz/ftg-mods",
|
||||
"tags": ["hikkatrusted", "nonactive"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "anon97945/hikka-mods",
|
||||
"url": "https://github.com/anon97945/hikka-mods",
|
||||
"tags": ["hikkatrusted", "nonactive"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "dorotorothequickend/DorotoroModules",
|
||||
"url": "https://github.com/dorotorothequickend/DorotoroModules",
|
||||
"tags": ["hikkatrusted", "nonlongermaintained"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "AmoreForever/amoremods",
|
||||
"url": "https://github.com/AmoreForever/amoremods",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "idiotcoders/idiotmodules",
|
||||
"url": "https://github.com/idiotcoders/idiotmodules",
|
||||
"tags": ["hikkatrusted", "herokutrusted", "nonactive"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "CakesTwix/Hikka-Modules",
|
||||
"url": "https://github.com/CakesTwix/Hikka-Modules",
|
||||
"tags": ["hikkatrusted", "nonactive"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "archquise/H.Modules",
|
||||
"url": "https://github.com/archquise/H.Modules",
|
||||
"tags": ["hikkatrusted", "nonactive"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "GD-alt/mm-hikka-mods",
|
||||
"url": "https://github.com/GD-alt/mm-hikka-mods",
|
||||
"tags": ["hikkatrusted", "herokutrusted", "nonactive"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "HitaloSama/FTG-modules-repo",
|
||||
"url": "https://github.com/HitaloSama/FTG-modules-repo",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "SekaiYoneya/Friendly-telegram",
|
||||
"url": "https://github.com/SekaiYoneya/Friendly-telegram",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "blazedzn/ftg-modules",
|
||||
"url": "https://github.com/blazedzn/ftg-modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "hikariatama/ftg",
|
||||
"url": "https://github.com/hikariatama/ftg",
|
||||
"tags": ["hikkatrusted", "nonactive"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "m4xx1m/FTG",
|
||||
"url": "https://github.com/m4xx1m/FTG",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "skillzmeow/skillzmods_hikka",
|
||||
"url": "https://github.com/skillzmeow/skillzmods_hikka",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "fajox1/famods",
|
||||
"url": "https://github.com/fajox1/famods",
|
||||
"tags": ["hikkatrusted", "herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "TheKsenon/MyHikkaModules",
|
||||
"url": "https://github.com/TheKsenon/MyHikkaModules",
|
||||
"tags": ["hikkatrusted", "herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "cryptexctl/modules-mirror",
|
||||
"url": "https://github.com/cryptexctl/modules-mirror",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "Ruslan-Isaev/modules",
|
||||
"url": "https://github.com/Ruslan-Isaev/modules",
|
||||
"tags": ["herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "shadowhikka/sh.modules",
|
||||
"url": "https://github.com/shadowhikka/sh.modules",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "fiksofficial/python-modules",
|
||||
"url": "https://github.com/fiksofficial/python-modules",
|
||||
"tags": ["herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "mead0wsss/mead0wsMods",
|
||||
"url": "https://github.com/mead0wsss/mead0wsMods",
|
||||
"tags": ["herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "SenkoGuardian/SenModules",
|
||||
"url": "https://github.com/SenkoGuardian/SenModules",
|
||||
"tags": ["herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "ZetGoHack/nullmod",
|
||||
"url": "https://github.com/ZetGoHack/nullmod",
|
||||
"tags": ["herokutrusted"],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"path": "yummy1gay/limoka",
|
||||
"url": "https://github.com/yummy1gay/limoka",
|
||||
"tags": [],
|
||||
"blacklist": []
|
||||
},
|
||||
{
|
||||
"url": "https://github.com/radiocycle/Modules",
|
||||
"tags": ["newbie"],
|
||||
"blacklist": []
|
||||
}
|
||||
]
|
||||
}
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
Reference in New Issue
Block a user