{}", + escape(cache_data.pages.get(page).unwrap_or(&"".to_string())) + ); + + bot.edit_message_text(message.chat.id, message.id, new_text) + .parse_mode(ParseMode::Html) + .reply_markup(keyboard) + .await?; + } else { + bot.edit_message_text( + message.chat.id, + message.id, + "Срок действия кеша перевода истек. Пожалуйста, переведите заново.", + ) + .reply_markup(InlineKeyboardMarkup::new(vec![vec![]])) + .await?; + } + Ok(()) +} + +async fn handle_language_menu_pagination( + bot: Bot, + message: &Message, + data: &str, +) -> Result<(), MyError> { + if let Ok(page) = data.trim_start_matches("tr_page:").parse::
{}", escape(&full_translated_text)); + let switch_lang_button = + InlineKeyboardButton::callback(lang_display_name.to_string(), "tr_show_langs"); + let mut keyboard = delete_message_button(user.id.0); + if let Some(first_row) = keyboard.inline_keyboard.get_mut(0) { + first_row.insert(0, switch_lang_button); + } else { + keyboard.inline_keyboard.push(vec![switch_lang_button]); + } + bot.edit_message_text(message.chat.id, message.id, response) + .parse_mode(ParseMode::Html) + .reply_markup(keyboard) + .await?; + } else { + let translation_id = Uuid::new_v4().to_string(); + let redis_key = format!("translation:{}", translation_id); + let cache_data = TranslationCache { + pages: display_pages.clone(), + user_id: user.id.0, + original_url: None, + target_lang: target_lang.to_string(), + }; + redis_client.set(&redis_key, &cache_data, 3600).await?; + + let switch_lang_button = + InlineKeyboardButton::callback(lang_display_name.to_string(), "tr_show_langs"); + let delete_button = delete_message_button(user.id.0) + .inline_keyboard + .remove(0) + .remove(0); + + let keyboard = Paginator::new("tr", display_pages.len()) + .current_page(0) + .set_callback_formatter(move |page| format!("tr:page:{}:{}", translation_id, page)) + .add_bottom_row(vec![switch_lang_button, delete_button]) + .build(); + + let response_text = format!("
{}", escape(&display_pages[0])); + bot.edit_message_text(message.chat.id, message.id, response_text) + .parse_mode(ParseMode::Html) + .reply_markup(keyboard) + .await?; + } + + Ok(()) +} + +async fn handle_show_languages( + bot: &Bot, + message: &Message, + user: &teloxide::types::User, + config: &Config, +) -> Result<(), MyError> { + if let Some(original_message) = message.reply_to_message() { + if let Some(text) = original_message + .text() + .or_else(|| original_message.caption()) + { + let job = TranslateJob { + text: text.to_string(), + user_id: user.id.0, + }; + let redis_key_job = format!("translate_job:{}", user.id); + config + .get_redis_client() + .set(&redis_key_job, &job, 600) + .await?; + } + } else { + bot.edit_message_text( + message.chat.id, + message.id, + "Не удалось найти исходное сообщение для смены языка.", + ) + .await?; + return Ok(()); + } + + let keyboard = create_language_keyboard(0); + bot.edit_message_text( + message.chat.id, + message.id, + "Выберите новый язык для перевода:", + ) + .reply_markup(keyboard) + .await?; + + Ok(()) +} diff --git a/src/bot/callbacks/whisper.rs b/src/bot/callbacks/whisper.rs new file mode 100644 index 0000000..5f37705 --- /dev/null +++ b/src/bot/callbacks/whisper.rs @@ -0,0 +1,91 @@ +use crate::{bot::inlines::whisper::Whisper, core::config::Config, errors::MyError}; +use teloxide::{ + Bot, + payloads::{AnswerCallbackQuerySetters, EditMessageTextSetters}, + prelude::{CallbackQuery, Requester}, + types::InlineKeyboardMarkup, +}; + +// TODO: refactor entire handler +pub async fn handle_whisper_callback( + bot: Bot, + q: CallbackQuery, + config: &Config, +) -> Result<(), MyError> { + let data = q.data.as_ref().ok_or("Callback query data is empty")?; + + let parts: Vec<&str> = data.split('_').collect(); + if parts.len() != 3 || parts[0] != "whisper" { + return Ok(()); + } + + let action = parts[1]; + let whisper_id = parts[2]; + + let user = q.from.clone(); + + let redis_key = format!("whisper:{}", whisper_id); + + let whisper: Option
\
+ > Версия: {}\n\
+ > Пинг API: {} мс\n\
+ > Нагрузка ЦП: {:.2}%\n\
+ > ОЗУ: {}/{} МБ\n\
+ ",
+ version, api_ping, cpu_usage_percent, used_ram_mb, total_ram_mb
+ );
+
+ let news_link_button =
+ InlineKeyboardButton::url("Канал с новостями", "https://t.me/fulturate".parse().unwrap());
+ let terms_of_use_link_button = InlineKeyboardButton::url(
+ "Условия использования",
+ "https://telegra.ph/Terms-Of-Use--Usloviya-ispolzovaniya-09-21"
+ .parse()
+ .unwrap(),
+ );
+
+ bot.send_message(message.chat.id, response_message)
+ .reply_parameters(ReplyParameters::new(message.id))
+ .parse_mode(ParseMode::Html)
+ .reply_markup(InlineKeyboardMarkup::new(vec![vec![
+ news_link_button,
+ terms_of_use_link_button,
+ ]]))
+ .await?;
+
+ Ok(())
+}
\ No newline at end of file
diff --git a/src/bot/commands/translate.rs b/src/bot/commands/translate.rs
new file mode 100644
index 0000000..6ec34cf
--- /dev/null
+++ b/src/bot/commands/translate.rs
@@ -0,0 +1,219 @@
+use crate::{
+ bot::keyboards::{delete::delete_message_button, translate::create_language_keyboard},
+ core::{
+ config::Config,
+ services::translation::{SUPPORTED_LANGUAGES, normalize_language_code},
+ },
+ errors::MyError,
+ util::paginator::{FrameBuild, Paginator},
+};
+use futures::future::join_all;
+use serde::{Deserialize, Serialize};
+use teloxide::{
+ prelude::*,
+ types::{InlineKeyboardButton, Message, ParseMode, ReplyParameters},
+ utils::html::escape,
+};
+use translators::{GoogleTranslator, Translator};
+use uuid::Uuid;
+
+#[derive(Serialize, Deserialize, Debug, Clone)]
+pub struct TranslationCache {
+ pub(crate) pages: Vec{}", escape(&full_translated_text)); + + let switch_lang_button = + InlineKeyboardButton::callback(lang_display_name.to_string(), "tr_show_langs"); + + let mut keyboard = delete_message_button(user.id.0); + match keyboard.inline_keyboard.get_mut(0) { + Some(first_row) => { + first_row.insert(0, switch_lang_button); + } + None => { + keyboard.inline_keyboard.push(vec![switch_lang_button]); + } + } + + bot.send_message(msg.chat.id, response) + .reply_parameters(ReplyParameters::new(replied_to_message.id)) + .parse_mode(ParseMode::Html) + .reply_markup(keyboard) + .await?; + } else { + let translation_id = Uuid::new_v4().to_string(); + let redis_key = format!("translation:{}", translation_id); + + let cache_data = TranslationCache { + pages: display_pages.clone(), + user_id: user.id.0, + original_url: None, + target_lang: target_lang.to_string(), + }; + config + .get_redis_client() + .set(&redis_key, &cache_data, 3600) + .await?; + + let switch_lang_button = + InlineKeyboardButton::callback(lang_display_name.to_string(), "tr_show_langs"); + let delete_button = delete_message_button(user.id.0) + .inline_keyboard + .remove(0) + .remove(0); + + let keyboard = Paginator::new("tr", display_pages.len()) + .current_page(0) + .set_callback_formatter(move |page| format!("tr:page:{}:{}", translation_id, page)) + .add_bottom_row(vec![switch_lang_button, delete_button]) + .build(); + + let response_text = format!("
{}", escape(&display_pages[0])); + bot.send_message(msg.chat.id, response_text) + .reply_parameters(ReplyParameters::new(replied_to_message.id)) + .parse_mode(ParseMode::Html) + .reply_markup(keyboard) + .await?; + } + + Ok(()) +} diff --git a/src/bot/dispatcher.rs b/src/bot/dispatcher.rs new file mode 100644 index 0000000..4b96019 --- /dev/null +++ b/src/bot/dispatcher.rs @@ -0,0 +1,306 @@ +use crate::bot::modules::registry::MOD_MANAGER; +use crate::bot::modules::Owner; +use crate::core::db::schemas::settings::Settings; +use crate::{ + bot::{ + callbacks::callback_query_handlers, + commander::command_handlers, + inlines::{ + cobalter::{handle_cobalt_inline, handle_inline_video, is_query_url}, + currency::{handle_currency_inline, is_currency_query}, + whisper::{handle_whisper_inline, is_whisper_query}, + }, + keyboards::delete::delete_message_button, + messager::{handle_currency, handle_speech}, + messages::chat::handle_bot_added, + }, + core::{config::Config, + db::schemas::user::User as DBUser}, + errors::MyError, + util::enums::Command, +}; +use log::{debug, error, info}; +use mongodb::bson::doc; +use oximod::{set_global_client, Model}; +use serde::Deserialize; +use std::{convert::Infallible, fmt::Write, ops::ControlFlow, sync::Arc}; +use teloxide::{ + dispatching::{ + Dispatcher, DpHandlerDescription, HandlerExt, MessageFilterExt, UpdateFilterExt, + }, + dptree, + error_handlers::LoggingErrorHandler, + payloads::{AnswerInlineQuerySetters, SendDocumentSetters}, + prelude::{ChatId, Handler, Message, Requester}, + types::{ + Chat, InlineKeyboardButton, InlineKeyboardMarkup, InlineQuery, InlineQueryResult, + InlineQueryResultArticle, InputFile, InputMessageContent, InputMessageContentText, Me, + MessageId, ParseMode, ThreadId, Update, User, + }, + update_listeners::Polling, + utils::{command::BotCommands, html}, + Bot, +}; + +async fn root_handler( + update: Update, + config: Arc
{}{}",
+ chat.id, title
+ )
+ .unwrap();
+ } else {
+ writeln!(&mut message_text, "В чате: (???)").unwrap();
+ }
+
+ if let Some(user) = user {
+ let username = user
+ .username
+ .as_ref()
+ .map_or("".to_string(), |u| format!(" (@{})", u));
+ let full_name = html::escape(&user.full_name());
+ writeln!(
+ &mut message_text,
+ "Вызвал: {} ({}){}",
+ full_name, user.id, username
+ )
+ .unwrap();
+ } else {
+ writeln!(&mut message_text, "Вызвал: (???)").unwrap();
+ }
+
+ let error_name = short_error_name(&err);
+ writeln!(
+ &mut message_text,
+ "\nОшибка:\n{}", + html::escape(&error_name) + ) + .unwrap(); + + let hashtag = "#error"; + writeln!(&mut message_text, "\n{}", hashtag).unwrap(); + + let full_error_text = format!("{:#?}", err); + let document = InputFile::memory(full_error_text.into_bytes()).file_name("error_details.txt"); + + if let (Ok(log_chat_id), Ok(error_thread_id)) = ( + config.get_log_chat_id().parse::
{}", escaped_block) + }) + .collect::
{}\nСтатус: {}", + self.name(), + self.description(), + if settings.enabled { "✅ Включен" } else { "❌ Выключен" } + ); + + let toggle_button = InlineKeyboardButton::callback( + if settings.enabled { "Выключить модуль" } else { "Включить модуль" }, + format!("{}:settings:toggle_module:{}", self.key(), commander_id), + ); + + let quality_options = [ + VideoQuality::Q720, + VideoQuality::Q1080, + VideoQuality::Q1440, + VideoQuality::Max, + ]; + let quality_buttons = quality_options + .iter() + .map(|q| { + let display_text = if settings.video_quality == *q { + format!("• {}p •", q.as_str()) + } else { + format!("{}p", q.as_str()) + }; + let cb_data = format!( + "{}:settings:set:quality:{}:{}", + self.key(), + q.as_str(), + commander_id + ); + InlineKeyboardButton::callback(display_text, cb_data) + }) + .collect::
{}\nСтатус: {}\n\nВыберите валюты для отображения.", + self.name(), + self.description(), + if settings.enabled { "✅ Включен" } else { "❌ Выключен" } + ); + + let toggle_button = InlineKeyboardButton::callback( + if settings.enabled { "Выключить модуль" } else { "Включить модуль" }, + format!("{}:settings:toggle_module:{}", self.key(), commander_id), + ); + + let all_currencies = get_all_currency_codes(CURRENCY_CONFIG_PATH.parse().unwrap())?; + + let back_button = InlineKeyboardButton::callback( + "⬅️ Назад", + format!("settings_back:{}:{}:{}", owner.r#type, owner.id, commander_id), + ); + + let mut keyboard = Paginator::from(self.key(), &all_currencies) + .per_page(12) + .columns(3) + .current_page(page) + .add_bottom_row(vec![back_button]) + .set_callback_prefix(format!("{}:settings", self.key())) + .build(|currency| { + let is_selected = settings.selected_codes.contains(¤cy.code); + let icon = if is_selected { "✅" } else { "❌" }; + let button_text = format!("{} {}", icon, currency.code); + let callback_data = format!( + "{}:settings:toggle:{}:{}", + self.key(), + currency.code, + commander_id + ); + InlineKeyboardButton::callback(button_text, callback_data) + }); + + keyboard.inline_keyboard.insert(0, vec![toggle_button]); + + Ok((text, keyboard)) + } +} + +pub async fn currency_codes_handler(bot: Bot, msg: Message, code: String) -> Result<(), MyError> { + if msg.chat.is_private() { + handle_currency_update::
{}\nСтатус: {}", + self.name(), + self.description(), + if settings.enabled { "✅ Включен" } else { "❌ Выключен" } + ); + + let toggle_button = InlineKeyboardButton::callback( + if settings.enabled { "Выключить модуль" } else { "Включить модуль" }, + format!("{}:settings:toggle_module:{}", self.key(), commander_id), + ); + + let keyboard = InlineKeyboardMarkup::new(vec![ + vec![toggle_button], + vec![InlineKeyboardButton::callback( + "⬅️ Назад", + format!("settings_back:{}:{}:{}", owner.r#type, owner.id, commander_id), + )], + ]); + + Ok((text, keyboard)) + } + + async fn handle_callback( + &self, + bot: Bot, + q: &CallbackQuery, + owner: &Owner, + data: &str, + commander_id: u64, + ) -> Result<(), MyError> { + let Some(message) = &q.message else { return Ok(()); }; + let Some(message) = message.regular_message() else { return Ok(()); }; + + let parts: Vec<_> = data.split(':').collect(); + + if parts.len() == 1 && parts[0] == "toggle_module" { + let mut settings: WhisperSettings = + Settings::get_module_settings(owner, self.key()).await?; + settings.enabled = !settings.enabled; + Settings::update_module_settings(owner, self.key(), settings).await?; + + let (text, keyboard) = self.get_settings_ui(owner, commander_id).await?; + bot.edit_message_text(message.chat.id, message.id, text) + .reply_markup(keyboard) + .parse_mode(teloxide::types::ParseMode::Html) + .await?; + return Ok(()); + } + + bot.answer_callback_query(q.id.clone()).await?; + + Ok(()) + } + + fn designed_for(&self, owner_type: &str) -> bool { + owner_type == "user" + } + + async fn is_enabled(&self, owner: &Owner) -> bool { + if !self.designed_for(&owner.r#type) { + return false; + } + let settings: WhisperSettings = Settings::get_module_settings(owner, self.key()).await.unwrap(); // god of unwraps + settings.enabled + } + + fn factory_settings(&self) -> Result
{}\n\nUsage:
/{} CURRENCY_CODE (e.g., /{} UAH) to enable/disable it.\n\nNotes:\n✅ - enabled\n❌ - disabled",
- codes_list, &command.command, &command.command
- );
- } else {
- // default fallback
- message = format!(
- "Available currencies to set up: {}\n\nNotes:\n✅ - enabled\n❌ - disabled", - codes_list - ); - } - } - - bot.send_message(msg.chat.id, message) - .parse_mode(ParseMode::Html) - .reply_parameters(ReplyParameters::new(msg.id)) - .await?; - - Ok(()) -} diff --git a/src/util/currency/converter.rs b/src/core/services/currency/converter.rs similarity index 94% rename from src/util/currency/converter.rs rename to src/core/services/currency/converter.rs index 9e5a4fe..3e00dca 100644 --- a/src/util/currency/converter.rs +++ b/src/core/services/currency/converter.rs @@ -1,25 +1,22 @@ -use std::{ - collections::HashMap, - fs, - sync::Arc, - time::{Duration, Instant}, +use crate::{ + bot::modules::{Owner, currency::CurrencySettings}, + core::db::schemas::settings::Settings, + errors::MyError, + util::currency_values::WORD_VALUES, }; - -use super::structs::WORD_VALUES; -use crate::db::schemas::CurrenciesFunctions; -use crate::db::schemas::group::Group; -use crate::db::schemas::user::User; use log::{debug, error, warn}; use once_cell::sync::Lazy; -use oximod::Model; use regex::Regex; use reqwest::Client; use serde::{Deserialize, Serialize}; -use teloxide::prelude::InlineQuery; -use teloxide::types::Chat; +use std::{ + collections::HashMap, + fs, + sync::Arc, + time::{Duration, Instant}, +}; use thiserror::Error; use tokio::sync::Mutex; -use crate::util::errors::MyError; const CACHE_DURATION_SECS: u64 = 60 * 10; pub const CURRENCY_CONFIG_PATH: &str = "currencies.json"; @@ -129,7 +126,7 @@ fn build_regex_from_config() -> Result
{}", text_parts[page]); + let new_keyboard = create_transcription_keyboard(page, text_parts.len(), query.from.id.0); + + if message.text() != Some(new_text.as_str()) || message.reply_markup() != Some(&new_keyboard) { + bot.edit_message_text(message.chat.id, message.id, new_text) + .parse_mode(ParseMode::Html) + .reply_markup(new_keyboard) + .await?; + } + + Ok(()) +} + +pub async fn back_handler(bot: Bot, query: CallbackQuery, config: &Config) -> Result<(), MyError> { + let Some(message) = query.message.and_then(|m| m.regular_message().cloned()) else { + return Ok(()); + }; + + let cache = config.get_redis_client(); + let message_cache_key = format!("message_file_map:{}", message.id); + let Some(file_unique_id) = cache.get::
{}", text_parts[0]), + ) + .parse_mode(ParseMode::Html) + .reply_markup(keyboard) + .await?; + + Ok(()) +} + +pub async fn summarization_handler( + bot: Bot, + query: CallbackQuery, + config: &Config, +) -> Result<(), MyError> { + let Some(message) = query.message.and_then(|m| m.regular_message().cloned()) else { + return Ok(()); + }; + + let cache = config.get_redis_client(); + let message_file_map_key = format!("message_file_map:{}", message.id); + let Some(file_unique_id) = cache.get::
{}", + cached_summary + ); + bot.edit_message_text(message.chat.id, message.id, final_text) + .parse_mode(ParseMode::Html) + .reply_markup(create_summary_keyboard()) + .await?; + return Ok(()); + } + + bot.edit_message_text( + message.chat.id, + message.id, + "Составляю краткое содержание...", + ) + .await?; + + let file_data = save_file_to_memory(&bot, &cache_entry.file_id).await?; + let new_summary = + summarize_audio(cache_entry.mime_type.clone(), file_data, config.clone()).await?; + + if new_summary.is_empty() || new_summary.contains("Не удалось получить") { + bot.edit_message_text( + message.chat.id, + message.id, + "❌ Не удалось составить краткое содержание.", + ) + .await?; + return Ok(()); + } + + cache_entry.summary = Some(new_summary.clone()); + cache.set(&file_cache_key, &cache_entry, 86400).await?; + + let final_text = format!( + "✨:\n
{}", + new_summary + ); + bot.edit_message_text(message.chat.id, message.id, final_text) + .parse_mode(ParseMode::Html) + .reply_markup(create_summary_keyboard()) + .await?; + + Ok(()) +} + +async fn get_cached( + bot: &Bot, + file: &AudioStruct, + config: &Config, +) -> Result
{}", text_parts[0]), + ) + .parse_mode(ParseMode::Html) + .reply_markup(keyboard) + .await?; + } + Err(e) => { + error!("Failed to get transcription: {:?}", e); + let error_text = match e { + MyError::Other(msg) if msg.contains("Не удалось преобразовать") => { + msg + } + MyError::Reqwest(_) => { + "❌ Ошибка: Не удалось скачать файл. Возможно, он слишком большой (>20MB)." + .to_string() + } + _ => "❌ Произошла неизвестная ошибка при обработке аудио.".to_string(), + }; + bot.edit_message_text(message.chat.id, message.id, error_text) + .await?; + } + } + } else { + bot.edit_message_text( + message.chat.id, + message.id, + "Не удалось найти голосовое сообщение.", + ) + .parse_mode(ParseMode::Html) + .await?; + } + Ok(()) +} + +pub async fn summarize_audio( + mime_type: String, + data: Bytes, + config: Config, +) -> Result
\
- > Version: {}\n\
- > API Ping: {} ms\n\
- > CPU Usage: {:.2}%\n\
- > RAM Usage: {}/{} MB\n\
- ",
- version, api_ping, cpu_usage_percent, used_ram_mb, total_ram_mb
- );
-
- bot.send_message(message.chat.id, response_message)
- .reply_parameters(ReplyParameters::new(message.id))
- .parse_mode(ParseMode::Html)
- .await?;
- Ok(())
-}
diff --git a/src/handlers/markups/inlines/currency.rs b/src/handlers/markups/inlines/currency.rs
deleted file mode 100644
index fa5bbc1..0000000
--- a/src/handlers/markups/inlines/currency.rs
+++ /dev/null
@@ -1,126 +0,0 @@
-use crate::config::Config;
-use crate::db::schemas::user::User;
-use log::{debug, error};
-use mongodb::bson::doc;
-use oximod::Model;
-use std::error as error_handler;
-use std::sync::Arc;
-use teloxide::Bot;
-use teloxide::payloads::AnswerInlineQuerySetters;
-use teloxide::prelude::Requester;
-use teloxide::types::{
- Chat, ChatId, ChatKind, ChatPrivate, InlineKeyboardButton, InlineKeyboardMarkup, InlineQuery,
- InlineQueryResult, InlineQueryResultArticle, InputMessageContent, InputMessageContentText, Me,
- ParseMode,
-};
-use uuid::Uuid;
-
-pub async fn handle_currency_inline(
- bot: Bot,
- q: InlineQuery,
- config: Arc{}", escaped_block) - }) - .collect::
{}", text_parts[0]), - ) - .parse_mode(ParseMode::Html) - .reply_markup(delete_message_button(original_user_id.0)) - .await?; - - for part in text_parts.iter().skip(1) { - bot.send_message( - msg.chat.id, - format!("
\n{}\n", part), - ) - .reply_parameters(ReplyParameters::new(msg.id)) - .parse_mode(ParseMode::Html) - .reply_markup(delete_message_button(original_user_id.0)) - .await?; - } - } else { - bot.edit_message_text( - msg.chat.id, - message.id, - "Не удалось найти голосовое сообщение.", - ) - .parse_mode(ParseMode::Html) - // .reply_markup(delete_message_button()) - .await?; - } - } - Ok(()) -} - -pub async fn get_file_id(msg: &Message) -> Option