Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
build: feature(impl_trait_in_assoc_type)
- Loading branch information
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,177 +1,177 @@ | ||
| use std::ops::Deref; | ||
|
|
||
| use arcstr::ArcStr; | ||
| use color_eyre::eyre::Result; | ||
| use lateinit::LateInit; | ||
| use mesagisto_client::{net::NET, res::RES}; | ||
| use teloxide::{ | ||
| adaptors::DefaultParseMode, | ||
| payloads::{SendAnimationSetters, SendMessageSetters, SendPhotoSetters}, | ||
| prelude::Requester, | ||
| types::{File as TgFile, InputFile, Message, MessageId}, | ||
| utils::command::BotCommands, | ||
| Bot, | ||
| }; | ||
| use teloxide_core::types::ChatId; | ||
| use tracing::{instrument, warn}; | ||
|
|
||
| use crate::{ | ||
| commands::{bind::BindCommand, manage::ManageCommand}, | ||
| config::CONFIG, | ||
| }; | ||
| pub type BotRequester = DefaultParseMode<Bot>; | ||
|
|
||
| #[derive(Singleton, Default, Debug)] | ||
| pub struct TgBot { | ||
| inner: LateInit<BotRequester>, | ||
| } | ||
| impl TgBot { | ||
| pub async fn init(&self, bot: BotRequester) -> Result<()> { | ||
| let mut commands = Vec::new(); | ||
| commands.append(&mut BindCommand::bot_commands()); | ||
| commands.append(&mut ManageCommand::bot_commands()); | ||
| bot.set_my_commands(commands).await?; | ||
| self.inner.init(bot); | ||
| Ok(()) | ||
| } | ||
|
|
||
| // fixme use this-error | ||
| pub async fn file(&self, uid: &Vec<u8>, id: &Vec<u8>) -> Result<()> { | ||
| let id_str: ArcStr = base64_url::encode(id).into(); | ||
| let TgFile { path, .. } = self | ||
| .get_file(String::from_utf8_lossy(id)) | ||
| .await | ||
| .expect("获取文件失败"); | ||
| let tmp_path = RES.tmp_path(&id_str); | ||
| let url = self.get_url_by_path(path); | ||
| NET.download(&url, &tmp_path).await?; | ||
| RES.put_file(uid, &tmp_path).await?; | ||
| Ok(()) | ||
| } | ||
|
|
||
| pub fn get_url_by_path(&self, file_path: String) -> ArcStr { | ||
| format!( | ||
| "https://api.telegram.org/file/bot{token}/{file}", | ||
| token = CONFIG.telegram.token, | ||
| file = file_path | ||
| ) | ||
| .into() | ||
| } | ||
|
|
||
| #[instrument(skip(self))] | ||
| pub async fn send_text<T>( | ||
| &self, | ||
| chat_id: ChatId, | ||
| text: T, | ||
| reply: Option<i32>, | ||
| ) -> Result<teloxide::types::Message> | ||
| where | ||
| T: Into<String> + Clone + std::fmt::Debug, | ||
| { | ||
| let send = self.inner.send_message(chat_id, text.clone()); | ||
| let send = if let Some(reply) = reply { | ||
| send.reply_to_message_id(MessageId(reply)) | ||
| } else { | ||
| send | ||
| }; | ||
| match send.await { | ||
| Ok(ok) => Ok(ok), | ||
| Err(e) => match e { | ||
| teloxide::RequestError::MigrateToChatId(new_id) => { | ||
| let target = chat_id.0; | ||
| warn!("Chat migrated from {} to {}", target, new_id); | ||
| if CONFIG.migrate_chat(&target, &new_id) { | ||
| let send = TG_BOT.send_message(ChatId(new_id), text.clone()); | ||
| let receipt = if let Some(reply) = reply { | ||
| send.reply_to_message_id(MessageId(reply)).await? | ||
| } else { | ||
| send.await? | ||
| }; | ||
| Ok(receipt) | ||
| } else { | ||
| Err(e.into()) | ||
| } | ||
| } | ||
| _ => Err(e.into()), | ||
| }, | ||
| } | ||
| } | ||
|
|
||
| #[instrument(skip(self))] | ||
| pub async fn send_image( | ||
| &self, | ||
| chat_id: ChatId, | ||
| image_path: &std::path::Path, | ||
| reply: Option<i32>, | ||
| ) -> Result<teloxide::types::Message> { | ||
| let photo = InputFile::file(image_path); | ||
| let kind = infer::get_from_path(image_path) | ||
| .expect("file read failed when refering file type") | ||
| .expect("Unkown file type"); | ||
| let is_gif = "gif" == kind.extension(); | ||
| let is_webp = "webp" == kind.extension(); | ||
| let result = if is_gif { | ||
| let photo = photo.clone().file_name(format!( | ||
| "{:?}.gif", | ||
| image_path.file_name().expect("Wrong filename") | ||
| )); | ||
|
|
||
| let send = self.inner.send_animation(chat_id, photo.clone()); | ||
| if let Some(reply) = reply { | ||
| send.reply_to_message_id(MessageId(reply)).await | ||
| } else { | ||
| send.await | ||
| } | ||
| } else { | ||
| if is_webp { | ||
| self.inner.send_sticker(chat_id, photo.clone()).await | ||
| } else { | ||
| let send = self.inner.send_photo(chat_id, photo.clone()); | ||
| if let Some(reply) = reply { | ||
| send.reply_to_message_id(MessageId(reply)).await | ||
| } else { | ||
| send.await | ||
| } | ||
| } | ||
| }; | ||
|
Check warning on line 136 in src/bot.rs
|
||
|
|
||
| match result { | ||
| Ok(ok) => Ok(ok), | ||
| Err(e) => match e { | ||
| teloxide::RequestError::MigrateToChatId(new_id) => { | ||
| let target = chat_id.0; | ||
| warn!("Chat migrated from {} to {}", target, new_id); | ||
| if CONFIG.migrate_chat(&target, &new_id) { | ||
| let receipt: Message = if is_gif { | ||
| let send = self.inner.send_animation(chat_id, photo.clone()); | ||
| if let Some(reply) = reply { | ||
| send.reply_to_message_id(MessageId(reply)).await? | ||
| } else { | ||
| send.await? | ||
| } | ||
| } else { | ||
| let send = self.inner.send_photo(chat_id, photo.clone()); | ||
| if let Some(reply) = reply { | ||
| send.reply_to_message_id(MessageId(reply)).await? | ||
| } else { | ||
| send.await? | ||
| } | ||
| }; | ||
| Ok(receipt) | ||
| } else { | ||
| Err(e.into()) | ||
| } | ||
| } | ||
| _ => Err(e.into()), | ||
| }, | ||
| } | ||
| } | ||
| } | ||
|
|
||
| impl Deref for TgBot { | ||
| type Target = BotRequester; | ||
|
|
||
| fn deref(&self) -> &Self::Target { | ||
| &self.inner | ||
| } | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,8 +1,4 @@ | ||
| use std::{fs::File, io::BufReader, path::PathBuf}; | ||
|
|
||
| use image::ImageFormat; | ||
|
|
||
|
|
||
Oops, something went wrong.