f5ed284f8c 2025-06-21 arcade: use crate::utils::{
f5ed284f8c 2025-06-21 arcade: Attachment,
f5ed284f8c 2025-06-21 arcade: RE_SPECIAL,
f5ed284f8c 2025-06-21 arcade: };
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: use std::{
f5ed284f8c 2025-06-21 arcade: borrow::Cow,
f5ed284f8c 2025-06-21 arcade: collections::HashMap,
f5ed284f8c 2025-06-21 arcade: fmt::Debug,
f5ed284f8c 2025-06-21 arcade: };
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: use anyhow::{
f5ed284f8c 2025-06-21 arcade: Context,
f5ed284f8c 2025-06-21 arcade: Result,
f5ed284f8c 2025-06-21 arcade: };
f5ed284f8c 2025-06-21 arcade: use tgbot::{
f5ed284f8c 2025-06-21 arcade: api::Client,
f5ed284f8c 2025-06-21 arcade: types::{
f5ed284f8c 2025-06-21 arcade: ChatPeerId,
f5ed284f8c 2025-06-21 arcade: InputFile,
f5ed284f8c 2025-06-21 arcade: InputFileReader,
f5ed284f8c 2025-06-21 arcade: InputMediaDocument,
f5ed284f8c 2025-06-21 arcade: MediaGroup,
f5ed284f8c 2025-06-21 arcade: MediaGroupItem,
f5ed284f8c 2025-06-21 arcade: Message,
f5ed284f8c 2025-06-21 arcade: ParseMode::MarkdownV2,
f5ed284f8c 2025-06-21 arcade: SendMediaGroup,
f5ed284f8c 2025-06-21 arcade: SendMessage,
f5ed284f8c 2025-06-21 arcade: SendDocument,
f5ed284f8c 2025-06-21 arcade: },
f5ed284f8c 2025-06-21 arcade: };
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: /// Encodes special HTML entities to prevent them interfering with Telegram HTML
f5ed284f8c 2025-06-21 arcade: pub fn encode (text: &str) -> Cow<'_, str> {
f5ed284f8c 2025-06-21 arcade: RE_SPECIAL.replace_all(text, "\\$1")
f5ed284f8c 2025-06-21 arcade: }
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: #[derive(Debug)]
f5ed284f8c 2025-06-21 arcade: pub struct TelegramTransport {
f5ed284f8c 2025-06-21 arcade: tg: Client,
f5ed284f8c 2025-06-21 arcade: recipients: HashMap<String, ChatPeerId>,
f5ed284f8c 2025-06-21 arcade: pub default: ChatPeerId,
f5ed284f8c 2025-06-21 arcade: }
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: impl TelegramTransport {
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: pub fn new (api_key: String, recipients: HashMap<String, i64>, default: i64) -> Result<TelegramTransport> {
f5ed284f8c 2025-06-21 arcade: let tg = Client::new(api_key)
f5ed284f8c 2025-06-21 arcade: .context("Failed to create API.\n")?;
f5ed284f8c 2025-06-21 arcade: let recipients = recipients.into_iter()
f5ed284f8c 2025-06-21 arcade: .map(|(a, b)| (a, ChatPeerId::from(b))).collect();
f5ed284f8c 2025-06-21 arcade: let default = ChatPeerId::from(default);
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: Ok(TelegramTransport {
f5ed284f8c 2025-06-21 arcade: tg,
f5ed284f8c 2025-06-21 arcade: recipients,
f5ed284f8c 2025-06-21 arcade: default,
f5ed284f8c 2025-06-21 arcade: })
f5ed284f8c 2025-06-21 arcade: }
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: /// Send message to default user, used for debug/log/info purposes
f5ed284f8c 2025-06-21 arcade: pub async fn debug (&self, msg: &str) -> Result<Message> {
f5ed284f8c 2025-06-21 arcade: self.send(&self.default, encode(msg)).await
f5ed284f8c 2025-06-21 arcade: }
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: /// Get recipient by address
f5ed284f8c 2025-06-21 arcade: pub fn get (&self, name: &str) -> Result<&ChatPeerId> {
f5ed284f8c 2025-06-21 arcade: self.recipients.get(name)
f5ed284f8c 2025-06-21 arcade: .with_context(|| format!("Recipient \"{name}\" not found in configuration"))
f5ed284f8c 2025-06-21 arcade: }
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: /// Send message to specified user
f5ed284f8c 2025-06-21 arcade: pub async fn send <S> (&self, to: &ChatPeerId, msg: S) -> Result<Message>
f5ed284f8c 2025-06-21 arcade: where S: Into<String> + Debug{
f5ed284f8c 2025-06-21 arcade: Ok(self.tg.execute(
f5ed284f8c 2025-06-21 arcade: SendMessage::new(*to, msg)
f5ed284f8c 2025-06-21 arcade: .with_parse_mode(MarkdownV2)
f5ed284f8c 2025-06-21 arcade: ).await?)
f5ed284f8c 2025-06-21 arcade: }
f5ed284f8c 2025-06-21 arcade:
f5ed284f8c 2025-06-21 arcade: /// Send media to specified user
f5ed284f8c 2025-06-21 arcade: pub async fn sendgroup (&self, to: &ChatPeerId, media: Vec<Attachment>, msg: &str) -> Result<()> {
f5ed284f8c 2025-06-21 arcade: if media.len() > 1 {
f5ed284f8c 2025-06-21 arcade: let mut attach = vec![];
f5ed284f8c 2025-06-21 arcade: let mut pos = media.len();
f5ed284f8c 2025-06-21 arcade: for file in media {
f5ed284f8c 2025-06-21 arcade: let mut caption = InputMediaDocument::default();
f5ed284f8c 2025-06-21 arcade: if pos == 1 {
f5ed284f8c 2025-06-21 arcade: caption = caption.with_caption(msg)
f5ed284f8c 2025-06-21 arcade: .with_caption_parse_mode(MarkdownV2);
f5ed284f8c 2025-06-21 arcade: }
f5ed284f8c 2025-06-21 arcade: pos -= 1;
f5ed284f8c 2025-06-21 arcade: attach.push(
f5ed284f8c 2025-06-21 arcade: MediaGroupItem::for_document(
f5ed284f8c 2025-06-21 arcade: InputFile::from(
f5ed284f8c 2025-06-21 arcade: InputFileReader::from(file.data)
f5ed284f8c 2025-06-21 arcade: .with_file_name(file.name)
f5ed284f8c 2025-06-21 arcade: ),
f5ed284f8c 2025-06-21 arcade: caption
f5ed284f8c 2025-06-21 arcade: )
f5ed284f8c 2025-06-21 arcade: );
f5ed284f8c 2025-06-21 arcade: }
f5ed284f8c 2025-06-21 arcade: self.tg.execute(SendMediaGroup::new(*to, MediaGroup::new(attach)?)).await?;
f5ed284f8c 2025-06-21 arcade: } else {
f5ed284f8c 2025-06-21 arcade: self.tg.execute(
f5ed284f8c 2025-06-21 arcade: SendDocument::new(
f5ed284f8c 2025-06-21 arcade: *to,
f5ed284f8c 2025-06-21 arcade: InputFileReader::from(media[0].data.clone())
f5ed284f8c 2025-06-21 arcade: .with_file_name(media[0].name.clone())
f5ed284f8c 2025-06-21 arcade: ).with_caption(msg)
f5ed284f8c 2025-06-21 arcade: .with_caption_parse_mode(MarkdownV2)
f5ed284f8c 2025-06-21 arcade: ).await?;
f5ed284f8c 2025-06-21 arcade: }
f5ed284f8c 2025-06-21 arcade: Ok(())
f5ed284f8c 2025-06-21 arcade: }
f5ed284f8c 2025-06-21 arcade: }