killingdruid/telegram-tui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: eliminate code duplication - extract helpers and use retry utils

Browse Source 
Extracted duplicate code and unified timeout handling across the codebase.

Changes:
- Extracted open_chat_and_load_data() function (eliminates 52 lines of duplication)
- Replaced manual y/н/Enter handling with handle_yes_no() from modal_handler (2 places)
- Replaced 7 direct tokio::time::timeout calls with retry utils (auth, main_input, main)
- Added with_timeout_ignore() for non-critical operations
- Fixed modal_handler.rs bug: corrected Russian 'y' key (д → н)
- Removed unused imports in handlers/mod.rs and utils/mod.rs

Impact:
- main_input.rs: 1164 → 958 lines (-206 lines, -18%)
- Code duplication: 52 lines eliminated
- Direct timeout calls: 7 → 1 (-86%)
- DRY principle applied throughout

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
Mikhail Kilin
2026-02-02 14:20:33 +03:00
parent 8e48d076de
commit 0768283e8a
8 changed files with 150 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

152
src/input/main_input.rs
View File

@@ -6,7 +6,8 @@ use crate::input::handlers::{
};
use crate::tdlib::ChatAction;
use crate::types::{ChatId, MessageId};
use crate::utils::{with_timeout, with_timeout_msg};
use crate::utils::{with_timeout, with_timeout_msg, with_timeout_ignore};
use crate::utils::modal_handler::handle_yes_no;
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
use std::time::{Duration, Instant};
@@ -23,8 +24,9 @@ pub async fn handle<T: TdClientTrait>(app: &mut App<T>, key: KeyEvent) {
// Обработка подтверждения выхода из группы
let confirmation_step = app.get_leave_group_confirmation_step();
if confirmation_step > 0 {
match key.code {
KeyCode::Char('y') | KeyCode::Char('н') | KeyCode::Enter => {
match handle_yes_no(key.code) {
Some(true) => {
// Подтверждение
if confirmation_step == 1 {
// Первое подтверждение - показываем второе
app.show_leave_group_final_confirmation();
@@ -46,11 +48,13 @@ pub async fn handle<T: TdClientTrait>(app: &mut App<T>, key: KeyEvent) {
}
}
}
KeyCode::Char('n') | KeyCode::Char('т') | KeyCode::Esc => {
Some(false) => {
// Отмена
app.cancel_leave_group();
}
_ => {}
None => {
// Другая клавиша - игнорируем
}
}
return;
}
@@ -324,8 +328,8 @@ pub async fn handle<T: TdClientTrait>(app: &mut App<T>, key: KeyEvent) {
// Модалка подтверждения удаления
if app.is_confirm_delete_shown() {
match key.code {
KeyCode::Char('y') | KeyCode::Char('н') | KeyCode::Enter => {
match handle_yes_no(key.code) {
Some(true) => {
// Подтверждение удаления
if let Some(msg_id) = app.chat_state.selected_message_id() {
if let Some(chat_id) = app.get_selected_chat_id() {
@@ -366,11 +370,13 @@ pub async fn handle<T: TdClientTrait>(app: &mut App<T>, key: KeyEvent) {
// Закрываем модалку
app.chat_state = crate::app::ChatState::Normal;
}
KeyCode::Char('n') | KeyCode::Char('т') | KeyCode::Esc => {
Some(false) => {
// Отмена удаления
app.chat_state = crate::app::ChatState::Normal;
}
_ => {}
None => {
// Другая клавиша - игнорируем
}
}
return;
}
@@ -435,42 +441,7 @@ pub async fn handle<T: TdClientTrait>(app: &mut App<T>, key: KeyEvent) {
// Выбрать чат из отфильтрованного списка
app.select_filtered_chat();
if let Some(chat_id) = app.get_selected_chat_id() {
app.status_message = Some("Загрузка сообщений...".to_string());
app.message_scroll_offset = 0;
match with_timeout_msg(
Duration::from_secs(10),
app.td_client.get_chat_history(ChatId::new(chat_id), 100),
"Таймаут загрузки сообщений",
)
.await
{
Ok(messages) => {
// Сохраняем загруженные сообщения
app.td_client.set_current_chat_messages(messages);
// ВАЖНО: Устанавливаем current_chat_id ТОЛЬКО ПОСЛЕ сохранения истории
// Это предотвращает race condition с Update::NewMessage
app.td_client.set_current_chat_id(Some(ChatId::new(chat_id)));
// Загружаем недостающие reply info
let _ = tokio::time::timeout(
Duration::from_secs(5),
app.td_client.fetch_missing_reply_info(),
)
.await;
// Загружаем последнее закреплённое сообщение
let _ = tokio::time::timeout(
Duration::from_secs(2),
app.td_client.load_current_pinned_message(ChatId::new(chat_id)),
)
.await;
// Загружаем черновик
app.load_draft();
app.status_message = None;
}
Err(e) => {
app.error_message = Some(e);
app.status_message = None;
}
}
open_chat_and_load_data(app, chat_id).await;
}
}
KeyCode::Backspace => {
@@ -620,42 +591,7 @@ pub async fn handle<T: TdClientTrait>(app: &mut App<T>, key: KeyEvent) {
if app.selected_chat_id != prev_selected {
if let Some(chat_id) = app.get_selected_chat_id() {
app.status_message = Some("Загрузка сообщений...".to_string());
app.message_scroll_offset = 0;
match with_timeout_msg(
Duration::from_secs(10),
app.td_client.get_chat_history(ChatId::new(chat_id), 100),
"Таймаут загрузки сообщений",
)
.await
{
Ok(messages) => {
// Сохраняем загруженные сообщения
app.td_client.set_current_chat_messages(messages);
// ВАЖНО: Устанавливаем current_chat_id ТОЛЬКО ПОСЛЕ сохранения истории
// Это предотвращает race condition с Update::NewMessage
app.td_client.set_current_chat_id(Some(ChatId::new(chat_id)));
// Загружаем недостающие reply info
let _ = tokio::time::timeout(
Duration::from_secs(5),
app.td_client.fetch_missing_reply_info(),
)
.await;
// Загружаем последнее закреплённое сообщение
let _ = tokio::time::timeout(
Duration::from_secs(2),
app.td_client.load_current_pinned_message(ChatId::new(chat_id)),
)
.await;
// Загружаем черновик
app.load_draft();
app.status_message = None;
}
Err(e) => {
app.error_message = Some(e);
app.status_message = None;
}
}
open_chat_and_load_data(app, chat_id).await;
}
}
}
@@ -966,3 +902,57 @@ pub async fn handle<T: TdClientTrait>(app: &mut App<T>, key: KeyEvent) {
}
}
}
/// Открывает чат и загружает все необходимые данные.
///
/// Выполняет:
/// - Загрузку истории сообщений (с timeout)
/// - Установку current_chat_id (после загрузки, чтобы избежать race condition)
/// - Загрузку reply info (с timeout)
/// - Загрузку закреплённого сообщения (с timeout)
/// - Загрузку черновика
///
/// При ошибке устанавливает error_message и очищает status_message.
async fn open_chat_and_load_data<T: TdClientTrait>(app: &mut App<T>, chat_id: i64) {
app.status_message = Some("Загрузка сообщений...".to_string());
app.message_scroll_offset = 0;
match with_timeout_msg(
Duration::from_secs(10),
app.td_client.get_chat_history(ChatId::new(chat_id), 100),
"Таймаут загрузки сообщений",
)
.await
{
Ok(messages) => {
// Сохраняем загруженные сообщения
app.td_client.set_current_chat_messages(messages);
// ВАЖНО: Устанавливаем current_chat_id ТОЛЬКО ПОСЛЕ сохранения истории
// Это предотвращает race condition с Update::NewMessage
app.td_client.set_current_chat_id(Some(ChatId::new(chat_id)));
// Загружаем недостающие reply info (игнорируем ошибки)
with_timeout_ignore(
Duration::from_secs(5),
app.td_client.fetch_missing_reply_info(),
)
.await;
// Загружаем последнее закреплённое сообщение (игнорируем ошибки)
with_timeout_ignore(
Duration::from_secs(2),
app.td_client.load_current_pinned_message(ChatId::new(chat_id)),
)
.await;
// Загружаем черновик
app.load_draft();
app.status_message = None;
}
Err(e) => {
app.error_message = Some(e);
app.status_message = None;
}
}
}