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

refactor: create timeout/retry utilities to reduce code duplication (P1.1)

Browse Source 
Created new utility modules to eliminate repeated timeout/retry patterns:
- src/utils/retry.rs: with_timeout() and with_timeout_msg() helpers
- src/utils/formatting.rs: timestamp formatting utilities (from utils.rs)
- src/utils/tdlib.rs: TDLib log configuration utilities (from utils.rs)

Refactored src/input/main_input.rs:
- Replaced 18+ instances of timeout(Duration, op).await pattern
- Simplified error handling from nested Ok(Ok(...))/Ok(Err(...))/Err(...)
  to cleaner Ok(...)/Err(...) with custom timeout messages
- Added type annotations for compiler type inference

Benefits:
- Reduced code duplication from ~20 instances to 2 utility functions
- Cleaner, more readable error handling
- Easier to maintain timeout logic in one place
- All 59 tests passing

Progress: REFACTORING_OPPORTUNITIES.md #1 (Дублирование кода) - Частично выполнено

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
Mikhail Kilin
2026-02-01 19:56:33 +03:00
parent 2b04b785c0
commit c6beea5608
6 changed files with 832 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

145
src/input/main_input.rs
View File

@@ -1,9 +1,9 @@
use crate::app::App;
use crate::tdlib::ChatAction;
use crate::types::{ChatId, MessageId};
use crate::utils::{with_timeout, with_timeout_msg};
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
use std::time::{Duration, Instant};
use tokio::time::timeout;
pub async fn handle(app: &mut App, key: KeyEvent) {
let has_ctrl = key.modifiers.contains(KeyModifiers::CONTROL);
@@ -12,7 +12,7 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
match key.code {
KeyCode::Char('r') if has_ctrl => {
app.status_message = Some("Обновление чатов...".to_string());
let _ = timeout(Duration::from_secs(5), app.td_client.load_chats(50)).await;
let _ = with_timeout(Duration::from_secs(5), app.td_client.load_chats(50)).await;
app.status_message = None;
return;
}
@@ -28,13 +28,15 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
if app.selected_chat_id.is_some() && !app.is_pinned_mode() {
if let Some(chat_id) = app.get_selected_chat_id() {
app.status_message = Some("Загрузка закреплённых...".to_string());
match timeout(
match with_timeout_msg(
Duration::from_secs(5),
app.td_client.get_pinned_messages(ChatId::new(chat_id)),
"Таймаут загрузки",
)
.await
{
Ok(Ok(messages)) => {
Ok(messages) => {
let messages: Vec<crate::tdlib::MessageInfo> = messages;
if messages.is_empty() {
app.status_message = Some("Нет закреплённых сообщений".to_string());
} else {
@@ -42,14 +44,10 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
app.status_message = None;
}
}
Ok(Err(e)) => {
Err(e) => {
app.error_message = Some(e);
app.status_message = None;
}
Err(_) => {
app.error_message = Some("Таймаут загрузки".to_string());
app.status_message = None;
}
}
}
}
@@ -219,7 +217,7 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
// Выполняем поиск при изменении запроса
if let Some(chat_id) = app.get_selected_chat_id() {
if !query.is_empty() {
if let Ok(Ok(results)) = timeout(
if let Ok(results) = with_timeout(
Duration::from_secs(3),
app.td_client.search_messages(ChatId::new(chat_id), &query),
)
@@ -240,7 +238,7 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
app.update_search_query(query.clone());
// Выполняем поиск при изменении запроса
if let Some(chat_id) = app.get_selected_chat_id() {
if let Ok(Ok(results)) = timeout(
if let Ok(results) = with_timeout(
Duration::from_secs(3),
app.td_client.search_messages(ChatId::new(chat_id), &query),
)
@@ -340,27 +338,22 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
app.status_message = Some("Отправка реакции...".to_string());
app.needs_redraw = true;
match timeout(
match with_timeout_msg(
Duration::from_secs(5),
app.td_client
.toggle_reaction(chat_id, message_id, emoji.clone()),
"Таймаут отправки реакции",
)
.await
{
Ok(Ok(_)) => {
Ok(_) => {
app.status_message =
Some(format!("Реакция {} добавлена", emoji));
app.exit_reaction_picker_mode();
app.needs_redraw = true;
}
Ok(Err(e)) => {
app.error_message = Some(format!("Ошибка: {}", e));
app.status_message = None;
app.needs_redraw = true;
}
Err(_) => {
app.error_message =
Some("Таймаут отправки реакции".to_string());
Err(e) => {
app.error_message = Some(e);
app.status_message = None;
app.needs_redraw = true;
}
@@ -394,17 +387,18 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
.map(|m| m.can_be_deleted_for_all_users())
.unwrap_or(false);
match timeout(
match with_timeout_msg(
Duration::from_secs(5),
app.td_client.delete_messages(
ChatId::new(chat_id),
vec![msg_id],
can_delete_for_all,
),
"Таймаут удаления",
)
.await
{
Ok(Ok(_)) => {
Ok(_) => {
// Удаляем из локального списка
app.td_client
.current_chat_messages_mut()
@@ -412,12 +406,9 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
// Сбрасываем состояние
app.chat_state = crate::app::ChatState::Normal;
}
Ok(Err(e)) => {
Err(e) => {
app.error_message = Some(e);
}
Err(_) => {
app.error_message = Some("Таймаут удаления".to_string());
}
}
}
}
@@ -447,26 +438,24 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
let to_chat_id = chat.id;
if let Some(msg_id) = app.chat_state.selected_message_id() {
if let Some(from_chat_id) = app.get_selected_chat_id() {
match timeout(
match with_timeout_msg(
Duration::from_secs(5),
app.td_client.forward_messages(
to_chat_id,
ChatId::new(from_chat_id),
vec![msg_id],
),
"Таймаут пересылки",
)
.await
{
Ok(Ok(_)) => {
Ok(_) => {
app.status_message =
Some("Сообщение переслано".to_string());
}
Ok(Err(e)) => {
Err(e) => {
app.error_message = Some(e);
}
Err(_) => {
app.error_message = Some("Таймаут пересылки".to_string());
}
}
}
}
@@ -497,26 +486,27 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
if let Some(chat_id) = app.get_selected_chat_id() {
app.status_message = Some("Загрузка сообщений...".to_string());
app.message_scroll_offset = 0;
match timeout(
match with_timeout_msg(
Duration::from_secs(10),
app.td_client.get_chat_history(ChatId::new(chat_id), 100),
"Таймаут загрузки сообщений",
)
.await
{
Ok(Ok(messages)) => {
Ok(messages) => {
// Сохраняем загруженные сообщения
*app.td_client.current_chat_messages_mut() = messages;
// ВАЖНО: Устанавливаем current_chat_id ТОЛЬКО ПОСЛЕ сохранения истории
// Это предотвращает race condition с Update::NewMessage
app.td_client.set_current_chat_id(Some(ChatId::new(chat_id)));
// Загружаем недостающие reply info
let _ = timeout(
let _ = tokio::time::timeout(
Duration::from_secs(5),
app.td_client.fetch_missing_reply_info(),
)
.await;
// Загружаем последнее закреплённое сообщение
let _ = timeout(
let _ = tokio::time::timeout(
Duration::from_secs(2),
app.td_client.load_current_pinned_message(ChatId::new(chat_id)),
)
@@ -525,14 +515,10 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
app.load_draft();
app.status_message = None;
}
Ok(Err(e)) => {
Err(e) => {
app.error_message = Some(e);
app.status_message = None;
}
Err(_) => {
app.error_message = Some("Таймаут загрузки сообщений".to_string());
app.status_message = None;
}
}
}
}
@@ -596,13 +582,14 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
return;
}
match timeout(
match with_timeout_msg(
Duration::from_secs(5),
app.td_client.edit_message(ChatId::new(chat_id), msg_id, text),
"Таймаут редактирования",
)
.await
{
Ok(Ok(mut edited_msg)) => {
Ok(mut edited_msg) => {
// Сохраняем reply_to из старого сообщения (если есть)
let messages = app.td_client.current_chat_messages_mut();
if let Some(pos) = messages.iter().position(|m| m.id() == msg_id) {
@@ -623,14 +610,8 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
app.chat_state = crate::app::ChatState::Normal;
app.needs_redraw = true; // ВАЖНО: перерисовываем UI
}
Ok(Err(e)) => {
app.error_message = Some(format!(
"Редактирование (chat={}, msg={}): {}",
chat_id, msg_id.as_i64(), e
));
}
Err(_) => {
app.error_message = Some("Таймаут редактирования".to_string());
Err(e) => {
app.error_message = Some(e);
}
}
}
@@ -662,25 +643,23 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
.send_chat_action(ChatId::new(chat_id), ChatAction::Cancel)
.await;
match timeout(
match with_timeout_msg(
Duration::from_secs(5),
app.td_client
.send_message(ChatId::new(chat_id), text, reply_to_id, reply_info),
"Таймаут отправки",
)
.await
{
Ok(Ok(sent_msg)) => {
Ok(sent_msg) => {
// Добавляем отправленное сообщение в список (с лимитом)
app.td_client.push_message(sent_msg);
// Сбрасываем скролл чтобы видеть новое сообщение
app.message_scroll_offset = 0;
}
Ok(Err(e)) => {
Err(e) => {
app.error_message = Some(e);
}
Err(_) => {
app.error_message = Some("Таймаут отправки".to_string());
}
}
}
}
@@ -694,26 +673,27 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
if let Some(chat_id) = app.get_selected_chat_id() {
app.status_message = Some("Загрузка сообщений...".to_string());
app.message_scroll_offset = 0;
match timeout(
match with_timeout_msg(
Duration::from_secs(10),
app.td_client.get_chat_history(ChatId::new(chat_id), 100),
"Таймаут загрузки сообщений",
)
.await
{
Ok(Ok(messages)) => {
Ok(messages) => {
// Сохраняем загруженные сообщения
*app.td_client.current_chat_messages_mut() = messages;
// ВАЖНО: Устанавливаем current_chat_id ТОЛЬКО ПОСЛЕ сохранения истории
// Это предотвращает race condition с Update::NewMessage
app.td_client.set_current_chat_id(Some(ChatId::new(chat_id)));
// Загружаем недостающие reply info
let _ = timeout(
let _ = tokio::time::timeout(
Duration::from_secs(5),
app.td_client.fetch_missing_reply_info(),
)
.await;
// Загружаем последнее закреплённое сообщение
let _ = timeout(
let _ = tokio::time::timeout(
Duration::from_secs(2),
app.td_client.load_current_pinned_message(ChatId::new(chat_id)),
)
@@ -722,14 +702,10 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
app.load_draft();
app.status_message = None;
}
Ok(Err(e)) => {
Err(e) => {
app.error_message = Some(e);
app.status_message = None;
}
Err(_) => {
app.error_message = Some("Таймаут загрузки сообщений".to_string());
app.status_message = None;
}
}
}
}
@@ -823,14 +799,16 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
app.needs_redraw = true;
// Запрашиваем доступные реакции
match timeout(
match with_timeout_msg(
Duration::from_secs(5),
app.td_client
.get_message_available_reactions(chat_id, message_id),
"Таймаут загрузки реакций",
)
.await
{
Ok(Ok(reactions)) => {
Ok(reactions) => {
let reactions: Vec<String> = reactions;
if reactions.is_empty() {
app.error_message =
Some("Реакции недоступны для этого сообщения".to_string());
@@ -842,13 +820,8 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
app.needs_redraw = true;
}
}
Ok(Err(e)) => {
app.error_message = Some(format!("Ошибка загрузки реакций: {}", e));
app.status_message = None;
app.needs_redraw = true;
}
Err(_) => {
app.error_message = Some("Таймаут загрузки реакций".to_string());
Err(e) => {
app.error_message = Some(e);
app.status_message = None;
app.needs_redraw = true;
}
@@ -864,20 +837,21 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
if key.code == KeyCode::Char('u') && has_ctrl {
if let Some(chat_id) = app.selected_chat_id {
app.status_message = Some("Загрузка профиля...".to_string());
match timeout(Duration::from_secs(5), app.td_client.get_profile_info(chat_id)).await
match with_timeout_msg(
Duration::from_secs(5),
app.td_client.get_profile_info(chat_id),
"Таймаут загрузки профиля",
)
.await
{
Ok(Ok(profile)) => {
Ok(profile) => {
app.enter_profile_mode(profile);
app.status_message = None;
}
Ok(Err(e)) => {
Err(e) => {
app.error_message = Some(e);
app.status_message = None;
}
Err(_) => {
app.error_message = Some("Таймаут загрузки профиля".to_string());
app.status_message = None;
}
}
}
return;
@@ -991,13 +965,14 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
if app.message_scroll_offset
> app.td_client.current_chat_messages().len().saturating_sub(10)
{
if let Ok(Ok(older)) = timeout(
if let Ok(older) = with_timeout(
Duration::from_secs(3),
app.td_client
.load_older_messages(ChatId::new(chat_id), oldest_msg_id),
)
.await
{
let older: Vec<crate::tdlib::MessageInfo> = older;
if !older.is_empty() {
// Добавляем старые сообщения в начало
let msgs = app.td_client.current_chat_messages_mut();
@@ -1033,7 +1008,7 @@ pub async fn handle(app: &mut App, key: KeyEvent) {
app.selected_folder_id = Some(folder_id);
// Загружаем чаты папки
app.status_message = Some("Загрузка чатов папки...".to_string());
let _ = timeout(
let _ = with_timeout(
Duration::from_secs(5),
app.td_client.load_folder_chats(folder_id, 50),
)