init auth and main function
This commit is contained in:
Yuriy Yuriev
@@ -0,0 +1,924 @@
|
||||
"""
|
||||
Обработчики команд с удобным интерфейсом управления задачами.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import logging
|
||||
from datetime import datetime
|
||||
from typing import Optional, Dict
|
||||
|
||||
from aiogram import Dispatcher, F, Bot
|
||||
from aiogram.filters import Command
|
||||
from aiogram.types import (
|
||||
Message, CallbackQuery, InlineKeyboardMarkup,
|
||||
InlineKeyboardButton, ReplyKeyboardMarkup, KeyboardButton
|
||||
)
|
||||
from aiogram.utils.keyboard import InlineKeyboardBuilder, ReplyKeyboardBuilder
|
||||
|
||||
from config.settings import settings
|
||||
from managers.background_tasks import BackgroundTaskManager
|
||||
from managers.proxy_manager import ProxyManager
|
||||
from managers.task_manager import TaskManager, TaskParams
|
||||
from services.browser_service import BrowserService
|
||||
from services.visit_service import VisitScheduler
|
||||
from utils.helpers import parse_range, format_range, extract_domain
|
||||
from utils.telegram import send_message_safe, send_visit_result
|
||||
from managers.storage import TaskStorage, ChatStorage
|
||||
from auth.manager import AuthManager
|
||||
from auth.storage import AuthStorage
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class BotInterface:
|
||||
"""Интерфейс бота с управлением задачами."""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
background_tasks: BackgroundTaskManager,
|
||||
proxy_manager: ProxyManager,
|
||||
browser_service: BrowserService,
|
||||
storage: TaskStorage = None,
|
||||
chat_storage: ChatStorage = None,
|
||||
):
|
||||
self.background_tasks = background_tasks
|
||||
self.proxy_manager = proxy_manager
|
||||
self.browser_service = browser_service
|
||||
self.task_manager = TaskManager()
|
||||
self.storage = storage
|
||||
self.chat_storage = chat_storage
|
||||
self.auth_manager = AuthManager()
|
||||
self.auth_storage = AuthStorage()
|
||||
self._user_input_state: Dict[int, dict] = {}
|
||||
|
||||
def register(self, dp: Dispatcher):
|
||||
"""Регистрация всех обработчиков."""
|
||||
|
||||
# Сохраняем ссылку на self для использования в обработчиках
|
||||
interface = self
|
||||
|
||||
# --- АВТОРИЗАЦИЯ ---
|
||||
|
||||
@dp.message(Command("register"))
|
||||
async def cmd_register(message: Message):
|
||||
"""Регистрация: /register <пароль>"""
|
||||
parts = message.text.split(maxsplit=1)
|
||||
if len(parts) < 2 or len(parts[1]) < 4:
|
||||
await message.answer(
|
||||
"❌ Использование: /register <пароль от 4 символов>"
|
||||
)
|
||||
return
|
||||
password = parts[1]
|
||||
user_id = message.from_user.id
|
||||
exists = await interface.auth_storage.user_exists(user_id)
|
||||
if exists:
|
||||
await message.answer(
|
||||
"⚠️ Вы уже зарегистрированы. Используйте /login <пароль>"
|
||||
)
|
||||
return
|
||||
password_hash, salt = interface.auth_manager._hash_password(password)
|
||||
success = await interface.auth_storage.register(
|
||||
user_id, password_hash, salt
|
||||
)
|
||||
if success:
|
||||
interface.auth_manager.login(user_id)
|
||||
await message.answer(
|
||||
"✅ Регистрация прошла успешно!\n"
|
||||
"Теперь вы можете использовать бота."
|
||||
)
|
||||
else:
|
||||
await message.answer("❌ Ошибка регистрации")
|
||||
|
||||
@dp.message(Command("login"))
|
||||
async def cmd_login(message: Message):
|
||||
"""Вход: /login <пароль>"""
|
||||
parts = message.text.split(maxsplit=1)
|
||||
if len(parts) < 2:
|
||||
await message.answer("❌ Использование: /login <пароль>")
|
||||
return
|
||||
password = parts[1]
|
||||
user_id = message.from_user.id
|
||||
user_data = await interface.auth_storage.get_user(user_id)
|
||||
if not user_data:
|
||||
await message.answer(
|
||||
"❌ Вы не зарегистрированы.\n"
|
||||
"Используйте /register <пароль>"
|
||||
)
|
||||
return
|
||||
locked = interface.auth_manager.is_locked_out(user_id)
|
||||
if locked:
|
||||
await message.answer("🔒 Слишком много попыток. Попробуйте позже.")
|
||||
return
|
||||
if interface.auth_manager.verify_password(
|
||||
password, user_data["password_hash"], user_data["salt"]
|
||||
):
|
||||
interface.auth_manager.login(user_id)
|
||||
await message.answer(
|
||||
"✅ Вы вошли в систему!\n"
|
||||
"Используйте /start для главного меню."
|
||||
)
|
||||
else:
|
||||
interface.auth_manager.record_failed_attempt(user_id)
|
||||
await message.answer("❌ Неверный пароль")
|
||||
|
||||
@dp.message(Command("logout"))
|
||||
async def cmd_logout(message: Message):
|
||||
"""Выйти из аккаунта."""
|
||||
interface.auth_manager.logout(message.from_user.id)
|
||||
await message.answer("🔑 Вы вышли из системы.")
|
||||
|
||||
# Декоратор: доступ только для авторизованных
|
||||
async def require_auth(handler_fn, event):
|
||||
user_id = event.from_user.id
|
||||
if not interface.auth_manager.is_authenticated(user_id):
|
||||
await event.answer(
|
||||
"🔒 Требуется авторизация!\n\n"
|
||||
"/register <пароль> — зарегистрироваться\n"
|
||||
"/login <пароль> — войти"
|
||||
)
|
||||
return False
|
||||
return True
|
||||
|
||||
# === Главное меню ===
|
||||
@dp.message(Command("start"))
|
||||
@dp.message(F.text == "📋 Главное меню")
|
||||
async def cmd_start(message: Message):
|
||||
if not await require_auth(message):
|
||||
return
|
||||
await interface._show_main_menu(message)
|
||||
|
||||
@dp.callback_query(F.data == "menu_main")
|
||||
async def cb_main_menu(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
await interface._show_main_menu(callback.message)
|
||||
await callback.answer()
|
||||
|
||||
# === Стримеры ===
|
||||
@dp.message(F.text == "📺 Стримеры")
|
||||
@dp.message(Command("streamers"))
|
||||
async def btn_streamers(message: Message):
|
||||
if not await require_auth(message):
|
||||
return
|
||||
await interface._show_streamers_list(message)
|
||||
|
||||
@dp.callback_query(F.data == "menu_streamers")
|
||||
async def cb_streamers(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
await interface._show_streamers_list(callback.message, edit=True)
|
||||
await callback.answer()
|
||||
|
||||
# === Задачи ===
|
||||
@dp.message(F.text == "📊 Задачи")
|
||||
@dp.message(Command("tasks"))
|
||||
async def btn_tasks(message: Message):
|
||||
if not await require_auth(message):
|
||||
return
|
||||
await interface._show_tasks_list(message)
|
||||
|
||||
@dp.callback_query(F.data == "menu_tasks")
|
||||
async def cb_tasks(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
await interface._show_tasks_list(callback.message, edit=True)
|
||||
await callback.answer()
|
||||
|
||||
# === Статус ===
|
||||
@dp.message(F.text == "📈 Статус")
|
||||
async def btn_status(message: Message):
|
||||
if not await require_auth(message):
|
||||
return
|
||||
await interface._show_status(message)
|
||||
|
||||
@dp.callback_query(F.data == "menu_status")
|
||||
async def cb_status(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
await interface._show_status(callback.message, edit=True)
|
||||
await callback.answer()
|
||||
|
||||
# === Стоп всё ===
|
||||
@dp.message(F.text == "🛑 Остановить всё")
|
||||
@dp.callback_query(F.data == "stop_all")
|
||||
async def handle_stop_all(event):
|
||||
if not await require_auth(event):
|
||||
return
|
||||
count = await interface.background_tasks.cancel_all()
|
||||
for task_id in list(interface.task_manager._tasks.keys()):
|
||||
await interface.task_manager.remove_task(task_id)
|
||||
|
||||
text = f"🛑 Остановлено задач: {count}"
|
||||
kb = InlineKeyboardBuilder()
|
||||
kb.row(InlineKeyboardButton(text="🔙 В меню", callback_data="menu_main"))
|
||||
|
||||
if isinstance(event, CallbackQuery):
|
||||
await event.message.edit_text(text, reply_markup=kb.as_markup())
|
||||
await event.answer()
|
||||
else:
|
||||
await event.answer(text, reply_markup=kb.as_markup())
|
||||
|
||||
# === Быстрый старт ===
|
||||
@dp.message(F.text == "🚀 Быстрый старт")
|
||||
async def btn_quick(message: Message):
|
||||
if not await require_auth(message):
|
||||
return
|
||||
await message.answer(
|
||||
"🚀 **БЫСТРЫЙ СТАРТ**\n\n"
|
||||
"**Twitch мониторинг:**\n"
|
||||
"`<канал> <username> [посещений] [минут] [задержка] [чтение] [домены]`\n\n"
|
||||
"**Пример:** `streamer user123 5 30 10-30 10-30`\n\n"
|
||||
"**Посещение URL:**\n"
|
||||
"`<url> [задержка] [повторов] [чтение]`\n\n"
|
||||
"**Пример:** `https://example.com 10-30 100 15-30`"
|
||||
)
|
||||
|
||||
# === Управление задачей ===
|
||||
@dp.callback_query(F.data.startswith("tdetail_"))
|
||||
async def cb_task_detail(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
task_id = callback.data.replace("tdetail_", "", 1)
|
||||
await interface._show_task_detail(callback, task_id)
|
||||
await callback.answer()
|
||||
|
||||
@dp.callback_query(F.data.startswith("tpause_"))
|
||||
async def cb_task_pause(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
task_id = callback.data.replace("tpause_", "", 1)
|
||||
await interface.task_manager.pause_task(task_id)
|
||||
await callback.answer("⏸️ Пауза")
|
||||
await interface._show_task_detail(callback, task_id)
|
||||
|
||||
@dp.callback_query(F.data.startswith("tresume_"))
|
||||
async def cb_task_resume(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
task_id = callback.data.replace("tresume_", "", 1)
|
||||
await interface.task_manager.resume_task(task_id)
|
||||
await callback.answer("▶️ Возобновлено")
|
||||
await interface._show_task_detail(callback, task_id)
|
||||
|
||||
@dp.callback_query(F.data.startswith("tstop_"))
|
||||
async def tstop(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
tid = callback.data.replace("tstop_", "", 1)
|
||||
await interface.task_manager.stop_task(tid)
|
||||
await interface.background_tasks.cancel_task(tid)
|
||||
await callback.answer("🛑 Остановлена")
|
||||
await interface._show_task_detail(callback, tid)
|
||||
|
||||
@dp.callback_query(F.data.startswith("tskip_"))
|
||||
async def cb_task_skip(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
task_id = callback.data.replace("tskip_", "", 1)
|
||||
await interface.task_manager.update_task(task_id, skip_next=True)
|
||||
await callback.answer("⏭️ Пропущено")
|
||||
|
||||
@dp.callback_query(F.data.startswith("trestart_"))
|
||||
async def cb_task_restart(callback: CallbackQuery):
|
||||
data = callback.data
|
||||
|
||||
# Проверяем, что это именно перезапуск задачи (trestart_task_)
|
||||
if data.startswith("trestart_task_"):
|
||||
tid = data.replace("trestart_task_", "", 1)
|
||||
logger.info(f"Перезапуск задачи: {tid}")
|
||||
|
||||
params = await interface.task_manager.get_task(tid)
|
||||
if not params:
|
||||
await callback.answer("❌ Не найдена")
|
||||
return
|
||||
|
||||
await interface.task_manager.update_task(
|
||||
tid,
|
||||
completed=False, stopped=False, paused=False,
|
||||
current_visit=0, total_visits=0,
|
||||
successful_visits=0, links_found=0,
|
||||
started_at=datetime.now()
|
||||
)
|
||||
|
||||
ok = await interface.background_tasks.start_task(
|
||||
task_id=tid,
|
||||
coro=interface._run_twitch(tid, params, callback.message),
|
||||
task_type="twitch_irc",
|
||||
metadata={'type': 'twitch_irc', 'channel': params.channel, 'chat_id': params.chat_id}
|
||||
)
|
||||
|
||||
if ok:
|
||||
await callback.answer("🔄 Перезапущено!")
|
||||
else:
|
||||
await callback.answer("❌ Ошибка запуска")
|
||||
|
||||
await interface._show_task_detail(callback, tid)
|
||||
else:
|
||||
# Обычный сброс статистики (trestart_ без task)
|
||||
task_id = data.replace("trestart_", "", 1)
|
||||
await interface.task_manager.restart_task(task_id)
|
||||
await callback.answer("🔄 Статистика сброшена")
|
||||
await interface._show_task_detail(callback, task_id)
|
||||
|
||||
@dp.callback_query(F.data.startswith("tset_"))
|
||||
async def cb_task_set(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
"""Установка параметра задачи."""
|
||||
data = callback.data
|
||||
|
||||
logger.info(f"SET callback: {data}")
|
||||
|
||||
# Формат: tset_TASKID_PARAM
|
||||
# Убираем префикс
|
||||
data = data.replace("tset_", "", 1)
|
||||
|
||||
# Ищем последнее подчеркивание для разделения task_id и param
|
||||
idx = data.rfind("_")
|
||||
if idx == -1:
|
||||
await callback.answer("❌ Неверный формат")
|
||||
return
|
||||
|
||||
task_id = data[:idx]
|
||||
param = data[idx + 1:]
|
||||
|
||||
logger.info(f"Parsed: task_id={task_id[:20]}, param={param}")
|
||||
|
||||
# Сохраняем состояние ожидания ввода
|
||||
interface._user_input_state[callback.from_user.id] = {
|
||||
"task_id": task_id,
|
||||
"param": param
|
||||
}
|
||||
|
||||
prompts = {
|
||||
"delay": "⏱️ Задержка (сек): `10-30` или `15`",
|
||||
"reading": "📖 Чтение (сек): `10-30` или `20`",
|
||||
"perlink": "🔗 Посещений на ссылку: `5`",
|
||||
"minutes": "⏰ Длительность (мин): `30`",
|
||||
"domains": "🌐 Домены через запятую: `site1.com, site2.ru`\nИли `all` для всех",
|
||||
}
|
||||
|
||||
builder = InlineKeyboardBuilder()
|
||||
builder.row(InlineKeyboardButton(text="❌ Отмена", callback_data=f"tdetail_{task_id}"))
|
||||
|
||||
prompt = prompts.get(param, f"Введите значение для {param}:")
|
||||
await callback.message.answer(prompt, reply_markup=builder.as_markup())
|
||||
await callback.answer(f"Ожидаю ввод: {param}")
|
||||
|
||||
# === Ввод параметров ===
|
||||
@dp.message()
|
||||
async def handle_all_text(message: Message):
|
||||
"""Обрабатывает ВСЕ текстовые сообщения."""
|
||||
text = message.text.strip() if message.text else ""
|
||||
user_id = message.from_user.id
|
||||
|
||||
if not interface.auth_manager.is_authenticated(user_id):
|
||||
await message.answer(
|
||||
"🔒 Требуется авторизация!\n\n"
|
||||
"/register <пароль> — зарегистрироваться\n"
|
||||
"/login <пароль> — войти"
|
||||
)
|
||||
return
|
||||
|
||||
logger.info(f"Handle text: '{text[:50]}' from user {user_id}, waiting: {user_id in interface._user_input_state}")
|
||||
|
||||
# 1. Проверяем - ожидаем ввод параметра
|
||||
if user_id in interface._user_input_state:
|
||||
await interface._process_param_input(message)
|
||||
return
|
||||
|
||||
# 2. Проверяем - это URL
|
||||
if text.startswith('http://') or text.startswith('https://'):
|
||||
parts = text.split()
|
||||
await interface._add_visit(message)
|
||||
return
|
||||
|
||||
# 3. Проверяем - это команда добавления стримера с параметрами
|
||||
# Формат: channel [visits] [minutes] [delay] [reading] [domains]
|
||||
parts = text.split()
|
||||
if len(parts) >= 1 and not text.startswith('/'):
|
||||
# Проверяем что первый параметр похож на канал (буквы/цифры/_)
|
||||
import re
|
||||
if re.match(r'^[a-zA-Z0-9_]+$', parts[0]):
|
||||
await interface._add_streamer(message, parts)
|
||||
return
|
||||
|
||||
# 4. Если ничего не подошло
|
||||
logger.info(f"Unhandled message: {text[:50]}")
|
||||
|
||||
@dp.callback_query(F.data.startswith("tdelete_"))
|
||||
async def tdelete(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
tid = callback.data.replace("tdelete_", "", 1)
|
||||
await interface.task_manager.remove_task(tid)
|
||||
await callback.answer("🗑️ Удалена")
|
||||
await interface._show_streamers_list(callback.message, edit=True)
|
||||
|
||||
|
||||
@dp.callback_query(F.data.startswith("treset_"))
|
||||
async def treset(callback: CallbackQuery):
|
||||
if not await require_auth(callback):
|
||||
return
|
||||
"""Сброс статистики активной задачи."""
|
||||
tid = callback.data.replace("treset_", "", 1)
|
||||
await interface.task_manager.update_task(
|
||||
tid,
|
||||
current_visit=0,
|
||||
total_visits=0,
|
||||
successful_visits=0,
|
||||
links_found=0
|
||||
)
|
||||
await callback.answer("🔄 Статистика сброшена")
|
||||
await interface._show_task_detail(callback, tid)
|
||||
|
||||
|
||||
# =========================================================================
|
||||
# ДОБАВЛЕНИЕ ЗАДАЧ
|
||||
# =========================================================================
|
||||
|
||||
async def _add_streamer(self, message: Message, parts: list):
|
||||
channel = parts[0].lower()
|
||||
|
||||
# Проверяем, что второй параметр - число (посещений), а не username
|
||||
vl = settings.DEFAULT_VISITS_PER_LINK
|
||||
mm = settings.DEFAULT_MONITOR_MINUTES
|
||||
mind = settings.DEFAULT_MIN_DELAY
|
||||
maxd = settings.DEFAULT_MAX_DELAY
|
||||
minr = settings.DEFAULT_MIN_READING
|
||||
maxr = settings.DEFAULT_MAX_READING
|
||||
domains = None
|
||||
|
||||
try:
|
||||
if len(parts) > 1: vl = int(parts[1])
|
||||
if len(parts) > 2: mm = int(parts[2])
|
||||
if len(parts) > 3 and '-' in parts[3]: mind, maxd = parse_range(parts[3])
|
||||
if len(parts) > 4 and '-' in parts[4]: minr, maxr = parse_range(parts[4], default_min=1)
|
||||
if len(parts) > 5: domains = [d.strip() for d in parts[5].split(',')]
|
||||
except ValueError as e:
|
||||
await message.answer(f"❌ {e}")
|
||||
return
|
||||
|
||||
tid = f"twitch_{channel}_{datetime.now().strftime('%H%M%S')}"
|
||||
|
||||
params = TaskParams(
|
||||
task_type="twitch_irc",
|
||||
channel=channel,
|
||||
target_username="*", # Все пользователи
|
||||
visits_per_link=vl,
|
||||
monitor_minutes=mm,
|
||||
min_delay=mind, max_delay=maxd,
|
||||
min_reading=minr, max_reading=maxr,
|
||||
allowed_domains=domains,
|
||||
started_at=datetime.now(),
|
||||
chat_id=message.chat.id
|
||||
)
|
||||
|
||||
await self.task_manager.add_task(tid, params)
|
||||
|
||||
success = await self.background_tasks.start_task(
|
||||
task_id=tid,
|
||||
coro=self._run_twitch(tid, params, message),
|
||||
task_type="twitch_irc",
|
||||
metadata={'type': 'twitch_irc', 'channel': channel, 'chat_id': message.chat.id}
|
||||
)
|
||||
|
||||
if success:
|
||||
if self.storage:
|
||||
await self.storage.save_task(tid, params)
|
||||
if self.chat_storage:
|
||||
await self.chat_storage.add_streamer(message.chat.id, channel)
|
||||
domains_str = ", ".join(domains) if domains else "все"
|
||||
text = (
|
||||
f"✅ **МОНИТОРИНГ ЗАПУЩЕН**\n\n"
|
||||
f"📺 Канал: `{channel}`\n"
|
||||
f"👤 Отслеживание: все пользователи\n"
|
||||
f"🔗 На ссылку: {vl} визитов\n"
|
||||
f"⏱️ Мониторинг: {mm} мин\n"
|
||||
f"⏱️ Задержка: {format_range(mind, maxd)}с\n"
|
||||
f"📖 Чтение: {format_range(minr, maxr)}с\n"
|
||||
f"🌐 Домены: {domains_str}\n"
|
||||
)
|
||||
text += f"\n🆔 `{tid[:12]}...`\n📺 /streamers — управление"
|
||||
await message.answer(text)
|
||||
|
||||
async def _add_visit(self, message: Message):
|
||||
"""Добавление URL для посещения."""
|
||||
parts = message.text.strip().split()
|
||||
url = parts[0]
|
||||
|
||||
min_delay = settings.DEFAULT_MIN_DELAY
|
||||
max_delay = settings.DEFAULT_MAX_DELAY
|
||||
min_reading = settings.DEFAULT_MIN_READING
|
||||
max_reading = settings.DEFAULT_MAX_READING
|
||||
max_visits = None
|
||||
|
||||
try:
|
||||
if len(parts) > 1 and '-' in parts[1]:
|
||||
min_delay, max_delay = parse_range(parts[1])
|
||||
elif len(parts) > 1:
|
||||
min_delay = max_delay = int(parts[1])
|
||||
if len(parts) > 2:
|
||||
if '-' in parts[2]:
|
||||
min_reading, max_reading = parse_range(parts[2], default_min=1)
|
||||
else:
|
||||
max_visits = int(parts[2])
|
||||
if len(parts) > 3 and '-' in parts[3]:
|
||||
min_reading, max_reading = parse_range(parts[3], default_min=1)
|
||||
except ValueError as e:
|
||||
await message.answer(f"❌ Ошибка: {e}")
|
||||
return
|
||||
|
||||
domain = extract_domain(url)[:20]
|
||||
task_id = f"visit_{domain}_{datetime.now().strftime('%H%M%S')}"
|
||||
|
||||
params = TaskParams(
|
||||
url=url, task_type="visit",
|
||||
min_delay=min_delay, max_delay=max_delay,
|
||||
min_reading=min_reading, max_reading=max_reading,
|
||||
max_visits=max_visits,
|
||||
started_at=datetime.now(), chat_id=message.chat.id
|
||||
)
|
||||
|
||||
await self.task_manager.add_task(task_id, params)
|
||||
|
||||
scheduler = VisitScheduler(
|
||||
browser_service=self.browser_service,
|
||||
url=url,
|
||||
min_delay=min_delay, max_delay=max_delay,
|
||||
min_reading=min_reading, max_reading=max_reading,
|
||||
max_visits=max_visits,
|
||||
)
|
||||
|
||||
async def on_complete(visit_num, result, delay, reading):
|
||||
await send_visit_result(message.bot, message.chat.id, visit_num, result, delay, reading, scheduler.successful, scheduler.failed)
|
||||
|
||||
scheduler.on_visit_complete = on_complete
|
||||
|
||||
success = await self.background_tasks.start_task(
|
||||
task_id=task_id, coro=scheduler.run(), task_type="visit",
|
||||
metadata={'type': 'visit', 'url': url, 'chat_id': message.chat.id}
|
||||
)
|
||||
|
||||
if success:
|
||||
await message.answer(
|
||||
f"✅ **ПОСЕЩЕНИЕ ЗАПУЩЕНО**\n\n"
|
||||
f"🌐 `{url}`\n"
|
||||
f"⏱️ Задержка: {format_range(min_delay, max_delay)}с\n"
|
||||
f"📖 Чтение: {format_range(min_reading, max_reading)}с\n"
|
||||
f"🔢 Повторов: {'∞' if max_visits is None else max_visits}\n"
|
||||
f"🆔 `{task_id}`"
|
||||
)
|
||||
|
||||
# =========================================================================
|
||||
# ОТОБРАЖЕНИЕ
|
||||
# =========================================================================
|
||||
|
||||
async def _show_main_menu(self, message: Message):
|
||||
"""Главное меню."""
|
||||
stats = await self.task_manager.get_stats()
|
||||
|
||||
text = (
|
||||
"👋 **ГЛАВНОЕ МЕНЮ**\n\n"
|
||||
f"📺 Twitch: **{stats['twitch']}**\n"
|
||||
f"🌐 Визитов: **{stats['visit']}**\n"
|
||||
f"⏸️ Пауза: **{stats['paused']}**\n\n"
|
||||
"Выберите раздел:"
|
||||
)
|
||||
|
||||
builder = InlineKeyboardBuilder()
|
||||
builder.row(InlineKeyboardButton(text=f"📺 Стримеры ({stats['twitch']})", callback_data="menu_streamers"))
|
||||
builder.row(InlineKeyboardButton(text=f"📊 Задачи ({stats['total']})", callback_data="menu_tasks"))
|
||||
builder.row(InlineKeyboardButton(text="📈 Статус", callback_data="menu_status"))
|
||||
builder.row(InlineKeyboardButton(text="🛑 Остановить всё", callback_data="stop_all"))
|
||||
|
||||
reply = ReplyKeyboardBuilder()
|
||||
reply.row(KeyboardButton(text="📺 Стримеры"), KeyboardButton(text="📊 Задачи"))
|
||||
reply.row(KeyboardButton(text="📈 Статус"), KeyboardButton(text="🚀 Быстрый старт"))
|
||||
reply.row(KeyboardButton(text="🛑 Остановить всё"), KeyboardButton(text="📋 Главное меню"))
|
||||
|
||||
await message.answer(text, reply_markup=reply.as_markup(resize_keyboard=True))
|
||||
await message.answer("💡 Кнопки управления:", reply_markup=builder.as_markup())
|
||||
|
||||
async def _show_streamers_list(self, message: Message, edit: bool = False):
|
||||
active = await self.task_manager.get_active_tasks()
|
||||
completed = await self.task_manager.get_completed_tasks()
|
||||
|
||||
twitch_active = {tid: p for tid, p in active.items() if p.task_type == "twitch_irc"}
|
||||
twitch_done = {tid: p for tid, p in completed.items() if p.task_type == "twitch_irc"}
|
||||
|
||||
if not twitch_active and not twitch_done:
|
||||
text = "📺 **СТРИМЕРЫ**\n\nНет задач.\nОтправьте `<канал>` для добавления."
|
||||
builder = InlineKeyboardBuilder()
|
||||
builder.row(InlineKeyboardButton(text="🔙 В меню", callback_data="menu_main"))
|
||||
else:
|
||||
text = "📺 **СТРИМЕРЫ**\n\n"
|
||||
builder = InlineKeyboardBuilder()
|
||||
|
||||
# Активные
|
||||
if twitch_active:
|
||||
text += "**🔄 Активные:**\n"
|
||||
for tid, p in list(twitch_active.items())[:10]:
|
||||
emoji = p.get_status_emoji()
|
||||
domains = ", ".join(p.allowed_domains) if p.allowed_domains else "все"
|
||||
|
||||
text += (
|
||||
f"{emoji} **{p.channel}**\n"
|
||||
f" 🔗{p.links_found} | 📊{p.total_visits} | 🌐{domains}\n"
|
||||
)
|
||||
builder.row(InlineKeyboardButton(
|
||||
text=f"{emoji} {p.channel} (🔗{p.links_found})",
|
||||
callback_data=f"tdetail_{tid}"
|
||||
))
|
||||
|
||||
# Завершенные
|
||||
if twitch_done:
|
||||
text += "\n**📁 Завершенные:**\n"
|
||||
for tid, p in list(twitch_done.items())[:5]:
|
||||
emoji = p.get_status_emoji()
|
||||
domains = ", ".join(p.allowed_domains) if p.allowed_domains else "все"
|
||||
|
||||
text += (
|
||||
f"{emoji} **{p.channel}**\n"
|
||||
f" 🔗{p.links_found} | 📊{p.total_visits} | 🌐{domains}\n"
|
||||
)
|
||||
builder.row(InlineKeyboardButton(
|
||||
text=f"{emoji} {p.channel} (завершена)",
|
||||
callback_data=f"tdetail_{tid}"
|
||||
))
|
||||
|
||||
builder.row(InlineKeyboardButton(text="🔙 В меню", callback_data="menu_main"))
|
||||
|
||||
if edit:
|
||||
try:
|
||||
await message.edit_text(text, reply_markup=builder.as_markup())
|
||||
except Exception as e:
|
||||
logger.debug(f"Edit message skipped: {e}")
|
||||
else:
|
||||
await message.answer(text, reply_markup=builder.as_markup())
|
||||
|
||||
async def _show_tasks_list(self, message: Message, edit: bool = False):
|
||||
"""Все задачи."""
|
||||
text = await self.task_manager.format_task_list()
|
||||
builder = InlineKeyboardBuilder()
|
||||
builder.row(InlineKeyboardButton(text="🔄 Обновить", callback_data="menu_tasks"))
|
||||
builder.row(InlineKeyboardButton(text="🔙 В меню", callback_data="menu_main"))
|
||||
|
||||
if edit:
|
||||
try:
|
||||
await message.edit_text(text, reply_markup=builder.as_markup())
|
||||
except Exception as e:
|
||||
logger.debug(f"Edit message skipped: {e}")
|
||||
else:
|
||||
await message.answer(text, reply_markup=builder.as_markup())
|
||||
|
||||
async def _show_status(self, message: Message, edit: bool = False):
|
||||
"""Статус."""
|
||||
stats = await self.task_manager.get_stats()
|
||||
proxy_stats = self.proxy_manager.get_stats()
|
||||
|
||||
text = (
|
||||
"📈 **СТАТУС**\n\n"
|
||||
f"📺 Twitch: **{stats['twitch']}**\n"
|
||||
f"🌐 Визиты: **{stats['visit']}**\n"
|
||||
f"⏸️ Пауза: **{stats['paused']}**\n"
|
||||
f"🔄 Активно: **{stats['active']}**\n\n"
|
||||
f"🔌 Прокси: {proxy_stats['total']} всего, {proxy_stats['available']} доступно"
|
||||
)
|
||||
|
||||
builder = InlineKeyboardBuilder()
|
||||
builder.row(InlineKeyboardButton(text="🔄 Обновить", callback_data="menu_status"))
|
||||
builder.row(InlineKeyboardButton(text="🔙 В меню", callback_data="menu_main"))
|
||||
|
||||
if edit:
|
||||
try:
|
||||
await message.edit_text(text, reply_markup=builder.as_markup())
|
||||
except Exception as e:
|
||||
logger.debug(f"Edit message skipped: {e}")
|
||||
else:
|
||||
await message.answer(text, reply_markup=builder.as_markup())
|
||||
|
||||
# =========================================================================
|
||||
# УПРАВЛЕНИЕ ЗАДАЧЕЙ
|
||||
# =========================================================================
|
||||
|
||||
async def _show_task_detail(self, callback: CallbackQuery, task_id: str):
|
||||
params = await self.task_manager.get_task(task_id)
|
||||
if not params:
|
||||
await callback.answer("❌ Задача не найдена")
|
||||
return
|
||||
|
||||
text = f"📋 **УПРАВЛЕНИЕ**\n\n{params.get_stats_text()}"
|
||||
|
||||
builder = InlineKeyboardBuilder()
|
||||
|
||||
# Для активных задач
|
||||
if not params.completed and not params.stopped:
|
||||
if params.paused:
|
||||
builder.row(InlineKeyboardButton(text="▶️ Возобновить", callback_data=f"tresume_{task_id}"))
|
||||
else:
|
||||
builder.row(InlineKeyboardButton(text="⏸️ Пауза", callback_data=f"tpause_{task_id}"))
|
||||
|
||||
builder.row(
|
||||
InlineKeyboardButton(text="⏱️ Задержка", callback_data=f"tset_{task_id}_delay"),
|
||||
InlineKeyboardButton(text="📖 Чтение", callback_data=f"tset_{task_id}_reading")
|
||||
)
|
||||
builder.row(
|
||||
InlineKeyboardButton(text="🔗 На ссылку", callback_data=f"tset_{task_id}_perlink"),
|
||||
InlineKeyboardButton(text="⏰ Минуты", callback_data=f"tset_{task_id}_minutes")
|
||||
)
|
||||
builder.row(
|
||||
InlineKeyboardButton(text="🌐 Домены", callback_data=f"tset_{task_id}_domains"),
|
||||
InlineKeyboardButton(text="⏭️ Пропустить", callback_data=f"tskip_{task_id}")
|
||||
)
|
||||
builder.row(
|
||||
InlineKeyboardButton(text="🔄 Сброс", callback_data=f"treset_{task_id}"),
|
||||
InlineKeyboardButton(text="🛑 Стоп", callback_data=f"tstop_{task_id}")
|
||||
)
|
||||
else:
|
||||
# Для завершенных/остановленных
|
||||
builder.row(
|
||||
InlineKeyboardButton(text="🔄 Перезапустить", callback_data=f"trestart_task_{task_id}"),
|
||||
)
|
||||
builder.row(
|
||||
InlineKeyboardButton(text="🗑️ Удалить", callback_data=f"tdelete_{task_id}")
|
||||
)
|
||||
|
||||
builder.row(InlineKeyboardButton(text="🔙 К списку", callback_data="menu_streamers"))
|
||||
|
||||
try:
|
||||
await callback.message.edit_text(text, reply_markup=builder.as_markup())
|
||||
except Exception as e:
|
||||
if "not modified" not in str(e):
|
||||
logger.error(f"Edit error: {e}")
|
||||
|
||||
|
||||
|
||||
async def _process_param_input(self, message: Message):
|
||||
"""Обработка ввода параметров."""
|
||||
user_id = message.from_user.id
|
||||
state = self._user_input_state.pop(user_id)
|
||||
|
||||
task_id = state["task_id"]
|
||||
param = state["param"]
|
||||
|
||||
# Получаем текст и очищаем от кавычек
|
||||
value = message.text.strip() if message.text else ""
|
||||
value = value.replace("'", "").replace('"', "").replace("`", "").strip()
|
||||
|
||||
logger.info(f"Process param: {param} = {value}")
|
||||
|
||||
if not value:
|
||||
await message.answer("❌ Пустое значение. Попробуйте ещё раз.")
|
||||
return
|
||||
|
||||
try:
|
||||
if param == "delay":
|
||||
if "-" in value:
|
||||
mn, mx = parse_range(value)
|
||||
else:
|
||||
mn = mx = int(value)
|
||||
await self.task_manager.update_task(task_id, min_delay=mn, max_delay=mx)
|
||||
await message.answer(f"✅ Задержка: {format_range(mn, mx)}с")
|
||||
|
||||
elif param == "reading":
|
||||
if "-" in value:
|
||||
mn, mx = parse_range(value, default_min=1)
|
||||
else:
|
||||
mn = mx = int(value)
|
||||
await self.task_manager.update_task(task_id, min_reading=mn, max_reading=mx)
|
||||
await message.answer(f"✅ Чтение: {format_range(mn, mx)}с")
|
||||
|
||||
elif param == "perlink":
|
||||
await self.task_manager.update_task(task_id, visits_per_link=int(value))
|
||||
await message.answer(f"✅ На ссылку: {value}")
|
||||
|
||||
elif param == "minutes":
|
||||
await self.task_manager.update_task(task_id, monitor_minutes=int(value))
|
||||
await message.answer(f"✅ Мониторинг: {value} мин")
|
||||
|
||||
elif param == "domains":
|
||||
if value.lower() in ["all", "все"]:
|
||||
await self.task_manager.update_task(task_id, allowed_domains=None)
|
||||
await message.answer("✅ Домены: все (фильтр отключен)")
|
||||
else:
|
||||
domains = [d.strip().lower() for d in value.split(",") if d.strip()]
|
||||
await self.task_manager.update_task(task_id, allowed_domains=domains)
|
||||
await message.answer(f"✅ Домены: {', '.join(domains)}")
|
||||
else:
|
||||
await message.answer(f"❌ Неизвестный параметр: {param}")
|
||||
|
||||
except ValueError as e:
|
||||
await message.answer(f"❌ Ошибка: {e}")
|
||||
|
||||
async def _run_twitch_monitor(self, task_id: str, params: TaskParams, message: Message):
|
||||
"""Запуск Twitch мониторинга."""
|
||||
from services.irc_service import TwitchIRCClient
|
||||
|
||||
irc = TwitchIRCClient(params.channel, params.target_username)
|
||||
|
||||
async def on_url(url: str, username: str):
|
||||
params.links_found += 1
|
||||
try:
|
||||
from urllib.parse import urlparse
|
||||
domain = urlparse(url).netloc
|
||||
await send_message_safe(
|
||||
message.bot, message.chat.id,
|
||||
f"🔗 @{username}: `{url[:60]}`\n🌐 Домен: `{domain}`"
|
||||
)
|
||||
except Exception as e:
|
||||
logger.warning(f"Failed to send URL with domain: {e}")
|
||||
await send_message_safe(message.bot, message.chat.id, f"🔗 @{username}: `{url[:60]}`")
|
||||
|
||||
for i in range(params.visits_per_link):
|
||||
while params.paused:
|
||||
await asyncio.sleep(1)
|
||||
|
||||
reading = params.get_reading_time()
|
||||
result = await self.browser_service.visit_page(url, reading)
|
||||
|
||||
params.total_visits += 1
|
||||
if result.success:
|
||||
params.successful_visits += 1
|
||||
|
||||
if i < params.visits_per_link - 1:
|
||||
await asyncio.sleep(params.get_delay())
|
||||
|
||||
try:
|
||||
await irc.listen_for_messages(on_url, params.monitor_duration, params.allowed_domains)
|
||||
except asyncio.CancelledError:
|
||||
raise
|
||||
finally:
|
||||
await irc.disconnect()
|
||||
await self.task_manager.remove_task(task_id)
|
||||
|
||||
async def _run_twitch(self, task_id: str, params: TaskParams, message: Message):
|
||||
from services.irc_service import TwitchIRCClient
|
||||
|
||||
irc = TwitchIRCClient(params.channel, params.target_username)
|
||||
|
||||
async def on_url(url, username):
|
||||
"""Запускаем обработку ссылки в отдельной задаче."""
|
||||
task = asyncio.create_task(self._process_url(url, username, params, message))
|
||||
task.add_done_callback(
|
||||
lambda t: logger.error(f"URL processing failed: {t.exception()}") if not t.done() and t.exception() else None
|
||||
)
|
||||
|
||||
def is_active():
|
||||
return not params.paused and not params.stopped
|
||||
|
||||
logger.info(f"🚀 IRC monitor: {params.channel}")
|
||||
|
||||
try:
|
||||
await irc.listen_for_messages(on_url, params.monitor_duration, params.allowed_domains, is_active=is_active)
|
||||
await self.task_manager.complete_task(task_id)
|
||||
except asyncio.CancelledError:
|
||||
raise
|
||||
finally:
|
||||
await irc.disconnect()
|
||||
if self.storage:
|
||||
await self.storage.save_task(task_id, params)
|
||||
|
||||
|
||||
|
||||
async def _process_url(self, url: str, username: str, params: TaskParams, message: Message):
|
||||
"""Обрабатывает найденную ссылку (выполняется параллельно)."""
|
||||
try:
|
||||
params.links_found += 1
|
||||
|
||||
from urllib.parse import urlparse
|
||||
domain = urlparse(url).netloc
|
||||
|
||||
logger.info(f"🔗 @{username}: {url}")
|
||||
|
||||
await send_message_safe(
|
||||
message.bot, message.chat.id,
|
||||
f"🔗 @{username}: `{url[:60]}`"
|
||||
)
|
||||
|
||||
for i in range(params.visits_per_link):
|
||||
while params.paused:
|
||||
await asyncio.sleep(1)
|
||||
|
||||
reading = params.get_reading_time()
|
||||
|
||||
try:
|
||||
result = await self.browser_service.visit_page_from_twitch(
|
||||
url=url,
|
||||
channel=params.channel,
|
||||
reading_time=reading
|
||||
)
|
||||
params.total_visits += 1
|
||||
if result.success:
|
||||
params.successful_visits += 1
|
||||
except Exception as e:
|
||||
logger.error(f"Visit error: {e}")
|
||||
|
||||
if i < params.visits_per_link - 1:
|
||||
await asyncio.sleep(params.get_delay())
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Process URL error: {e}")
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user