Corrected the error related to the too large description of Tiktok video

Author: destroydevs

src/bot.rs

@@ -1,14 +1,18 @@
-use crate::data::{add_user, current_timestamp, get_user, has_user, User};
+use crate::data::{add_user, current_timestamp, get_data, get_user, has_user, set_data, User, UserColumn};
 use crate::fetcher::{parse_tiktok_content, ContentType, Media};
 use crate::parser::is_tiktok_url;
 use reqwest::Url;
 use std::cmp::PartialEq;
 use teloxide::dispatching::dialogue::GetChatId;
-use teloxide::payloads::{SendMessageSetters, SendPhotoSetters, SendVideoSetters};
+use teloxide::payloads::{SendMessageSetters, SendVideoSetters};
 use teloxide::prelude::{Message, Requester};
 use teloxide::types::{ChatId, InputFile, InputMedia, InputMediaPhoto, ParseMode};
 use teloxide::Bot;
 
+#[allow(dead_code)]
+const TEXT_LIMIT: u16 = 4096;
+const MEDIA_LIMIT: u16 = 1024;
+
 pub async fn start(bot: Bot) {
     teloxide::repl(bot, |bot: Bot, msg: Message| async move {
 
@@ -34,6 +38,8 @@ pub async fn start(bot: Bot) {
                 None => &"en".to_string()
             };
 
+            update_user(&bot,user.id.0,chat_id, &user.first_name, user_lang).await;
+
             if !is_tiktok_url(&text) {
                 send_msg(
                     &bot,
@@ -52,44 +58,6 @@ pub async fn start(bot: Bot) {
                 return Ok(());
             }
 
-            if has_user(user.id.0).await.unwrap() {
-                let data_user: User = get_user(user.id.0).await.unwrap().unwrap();
-
-                // days after last bot use
-                let time_after_last_use = (current_timestamp() - data_user.timestamp)/1000/60/60/24;
-
-                if time_after_last_use >= 3 {
-                    send_msg(
-                        &bot,
-                        chat_id,
-                        "<b>🌟 Welcome back!</b>\n🌟 We missed you! 😊",
-                        "<b>🌟 С возвращением!</b>\n🌟 Мы рады снова видеть вас! 😊",
-                        user_lang
-                    ).await;
-                }
-
-
-            } else {
-                let data_user = User {
-                    id: user.id.0,
-                    name: user.first_name.clone(),
-                    requests_amount: 1,
-                    timestamp: current_timestamp(),
-                    register_timestamp: current_timestamp(),
-                };
-                add_user(data_user).await.unwrap();
-
-                // wellcome message
-                send_msg(
-                    &bot,
-                    chat_id,
-                    "<b>🎉 Welcome aboard!</b>\n🎉 It looks like you're new here. Enjoy our bot for free - no limits! 😊",
-                    "<b>🎉 Добро пожаловать!</b>\n🎉 Похоже, вы здесь впервые. Пользуйтесь ботом бесплатно - без ограничений! 😊",
-                    user_lang
-                ).await;
-
-            }
-
             send_msg(
                 &bot,
                 chat_id,
@@ -114,14 +82,16 @@ pub async fn start(bot: Bot) {
                 },
             };
 
+            let media_title = cut_text(media.get_title(),MEDIA_LIMIT);
+
             if media.get_content_type() == ContentType::Video {
                 bot.send_video(chat_id,
                                InputFile::url(
                                    Url::parse(media.get_content_urls()[0]
                                    .as_str())
                                    .unwrap()
                                ))
-                    .caption(media.get_title())
+                    .caption(media_title)
                     .await
                     .unwrap();
             } else if media.get_content_type() == ContentType::Photo {
@@ -136,7 +106,7 @@ pub async fn start(bot: Bot) {
                     );
 
                     if !has_caption {
-                        photo_media = photo_media.caption(media.get_title());
+                        photo_media = photo_media.caption(&media_title);
                         has_caption = true;
                     }
 
@@ -173,6 +143,7 @@ pub async fn start(bot: Bot) {
         let mut text = english;
 
         if lang.eq("ru") // russia
+            || lang.eq("rs") // serbia
             || lang.eq("ua") // ukraine
             || lang.eq("by") // belarus
             || lang.eq("kz") // kazakhstan
@@ -191,4 +162,83 @@ pub async fn start(bot: Bot) {
             .await
             .unwrap();
     }
+
+    fn is_beyond_scope(text: &String, size: u16) -> bool {
+
+        text.len()>size as usize
+
+    }
+
+    fn cut_text(text: String, size: u16) -> String {
+        if is_beyond_scope(&text, size) {
+            let slice_size = size as usize;
+            text[0..slice_size].to_string()
+        } else {
+            text
+        }
+    }
+
+    async fn update_user(bot: &Bot, id: u64, chat_id: ChatId, first_name: &String, user_lang: &String) {
+        if has_user(id).await.unwrap() {
+            let data_user: User = get_user(id).await.unwrap().unwrap();
+
+            // days after last bot use
+            let time_after_last_use = (current_timestamp() - data_user.timestamp)/1000/60/60/24;
+
+            if time_after_last_use >= 3 {
+                send_msg(
+                    &bot,
+                    chat_id,
+                    "<b>🌟 Welcome back!</b>\n🌟 We missed you! 😊",
+                    "<b>🌟 С возвращением!</b>\n🌟 Мы рады снова видеть вас! 😊",
+                    user_lang
+                ).await;
+            }
+
+            if !data_user.name.eq(first_name) {
+
+                set_data(id,UserColumn::Name,first_name)
+                    .await
+                    .unwrap();
+
+            }
+
+            let data_chat_id = match get_data(id,UserColumn::ChatId).await {
+                Ok(chat_id) => chat_id,
+                _ => {
+                    None
+                }
+            };
+
+            if data_chat_id.is_none() {
+
+                set_data(id, UserColumn::ChatId, chat_id.0.to_string().as_str())
+                    .await
+                    .unwrap();
+
+            }
+
+        } else {
+            let data_user = User {
+                id,
+                chat_id: Some(chat_id.0),
+                name: first_name.to_string(),
+                requests_amount: 1,
+                timestamp: current_timestamp(),
+                register_timestamp: current_timestamp(),
+            };
+
+            add_user(data_user).await.unwrap();
+
+            // wellcome message
+            send_msg(
+                &bot,
+                chat_id,
+                "<b>🎉 Welcome aboard!</b>\n🎉 It looks like you're new here. Enjoy our bot for free - no limits! 😊",
+                "<b>🎉 Добро пожаловать!</b>\n🎉 Похоже, вы здесь впервые. Пользуйтесь ботом бесплатно - без ограничений! 😊",
+                user_lang
+            ).await;
+
+        }
+    }
 }