9adc69d514 2026-03-25 arcade: use crate::{
9adc69d514 2026-03-25 arcade: Arc,
9adc69d514 2026-03-25 arcade: Mutex,
9adc69d514 2026-03-25 arcade: };
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: use std::{
9adc69d514 2026-03-25 arcade: borrow::Cow,
9adc69d514 2026-03-25 arcade: collections::HashMap,
9adc69d514 2026-03-25 arcade: fmt,
9adc69d514 2026-03-25 arcade: };
9adc69d514 2026-03-25 arcade:
13265e7697 2026-01-10 arcade: use serde::{
13265e7697 2026-01-10 arcade: Deserialize,
13265e7697 2026-01-10 arcade: Serialize,
13265e7697 2026-01-10 arcade: };
9c4f09193a 2026-01-09 arcade: use stacked_errors::{
9adc69d514 2026-03-25 arcade: bail,
9c4f09193a 2026-01-09 arcade: Result,
9c4f09193a 2026-01-09 arcade: StackableErr,
9c4f09193a 2026-01-09 arcade: };
9c4f09193a 2026-01-09 arcade: use tgbot::{
9c4f09193a 2026-01-09 arcade: api::Client,
9c4f09193a 2026-01-09 arcade: types::{
9c4f09193a 2026-01-09 arcade: Bot,
9c4f09193a 2026-01-09 arcade: ChatPeerId,
9c4f09193a 2026-01-09 arcade: GetBot,
13265e7697 2026-01-10 arcade: InlineKeyboardButton,
13265e7697 2026-01-10 arcade: InlineKeyboardMarkup,
9c4f09193a 2026-01-09 arcade: Message,
9c4f09193a 2026-01-09 arcade: ParseMode,
9c4f09193a 2026-01-09 arcade: SendMessage,
9c4f09193a 2026-01-09 arcade: },
9c4f09193a 2026-01-09 arcade: };
13265e7697 2026-01-10 arcade:
9adc69d514 2026-03-25 arcade: const CB_VERSION: u8 = 0;
9adc69d514 2026-03-25 arcade:
13265e7697 2026-01-10 arcade: #[derive(Serialize, Deserialize, Debug)]
9adc69d514 2026-03-25 arcade: pub enum Callback {
9adc69d514 2026-03-25 arcade: // List all feeds (version, name to show, page number)
9adc69d514 2026-03-25 arcade: List(u8, String, u8),
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: impl Callback {
9adc69d514 2026-03-25 arcade: pub fn list (text: &str, page: u8) -> Callback {
9adc69d514 2026-03-25 arcade: Callback::List(CB_VERSION, text.to_owned(), page)
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: fn version (&self) -> u8 {
9adc69d514 2026-03-25 arcade: match self {
9adc69d514 2026-03-25 arcade: Callback::List(version, .. ) => *version,
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: impl fmt::Display for Callback {
9adc69d514 2026-03-25 arcade: fn fmt (&self, f: &mut fmt::Formatter) -> fmt::Result {
9adc69d514 2026-03-25 arcade: f.write_str(&toml::to_string(self).map_err(|_| fmt::Error)?)
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: /// Produce new Keyboard Markup from current Callback
9adc69d514 2026-03-25 arcade: pub async fn get_kb (cb: &Callback, feeds: Arc<Mutex<HashMap<i32, String>>>) -> Result<InlineKeyboardMarkup> {
9adc69d514 2026-03-25 arcade: if cb.version() != CB_VERSION {
9adc69d514 2026-03-25 arcade: bail!("Wrong callback version.");
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade: let mark = match cb {
9adc69d514 2026-03-25 arcade: Callback::List(_, name, page) => {
9adc69d514 2026-03-25 arcade: let mut kb = vec![];
9adc69d514 2026-03-25 arcade: let feeds = feeds.lock_arc().await;
9adc69d514 2026-03-25 arcade: let long = feeds.len() > 6;
9adc69d514 2026-03-25 arcade: let (start, end) = if long {
9adc69d514 2026-03-25 arcade: (page * 5, 5 + page * 5)
9adc69d514 2026-03-25 arcade: } else {
9adc69d514 2026-03-25 arcade: (0, 6)
9adc69d514 2026-03-25 arcade: };
9adc69d514 2026-03-25 arcade: let mut i = 0;
9adc69d514 2026-03-25 arcade: for (id, name) in feeds.iter() {
9adc69d514 2026-03-25 arcade: if i < start { continue }
9adc69d514 2026-03-25 arcade: if i > end { break }
9adc69d514 2026-03-25 arcade: i += 1;
9adc69d514 2026-03-25 arcade: kb.push(vec![
9adc69d514 2026-03-25 arcade: InlineKeyboardButton::for_callback_data(
9adc69d514 2026-03-25 arcade: format!("{}. {}", id, name),
9adc69d514 2026-03-25 arcade: Callback::list("xxx", *page).to_string()), // XXX edit
9adc69d514 2026-03-25 arcade: ])
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade: if name.is_empty() {
9adc69d514 2026-03-25 arcade: // no name - reverting to pages, any unknown number means last page
9adc69d514 2026-03-25 arcade: kb.push(vec![
9adc69d514 2026-03-25 arcade: InlineKeyboardButton::for_callback_data("1",
9adc69d514 2026-03-25 arcade: Callback::list("xxx", 0).to_string()),
9adc69d514 2026-03-25 arcade: ])
9adc69d514 2026-03-25 arcade: } else {
9adc69d514 2026-03-25 arcade: kb.push(vec![
9adc69d514 2026-03-25 arcade: InlineKeyboardButton::for_callback_data("1",
9adc69d514 2026-03-25 arcade: Callback::list("xxx", 0).to_string()),
9adc69d514 2026-03-25 arcade: ])
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade: if long {
9adc69d514 2026-03-25 arcade: kb.push(vec![
9adc69d514 2026-03-25 arcade: InlineKeyboardButton::for_callback_data("<<",
9adc69d514 2026-03-25 arcade: Callback::list("", if *page == 0 { *page } else { page - 1 } ).to_string()),
9adc69d514 2026-03-25 arcade: InlineKeyboardButton::for_callback_data(">>",
9adc69d514 2026-03-25 arcade: Callback::list("", page + 1).to_string()),
9adc69d514 2026-03-25 arcade: ]);
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade: InlineKeyboardMarkup::from(kb)
9adc69d514 2026-03-25 arcade: },
9adc69d514 2026-03-25 arcade: };
9adc69d514 2026-03-25 arcade: Ok(mark)
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: pub enum MyMessage <'a> {
9adc69d514 2026-03-25 arcade: Html { text: Cow<'a, str> },
9adc69d514 2026-03-25 arcade: HtmlTo { text: Cow<'a, str>, to: ChatPeerId },
9adc69d514 2026-03-25 arcade: HtmlToKb { text: Cow<'a, str>, to: ChatPeerId, kb: InlineKeyboardMarkup },
9adc69d514 2026-03-25 arcade: Text { text: Cow<'a, str> },
9adc69d514 2026-03-25 arcade: TextTo { text: Cow<'a, str>, to: ChatPeerId },
13265e7697 2026-01-10 arcade: }
13265e7697 2026-01-10 arcade:
9adc69d514 2026-03-25 arcade: impl MyMessage <'_> {
9adc69d514 2026-03-25 arcade: pub fn html <'a, S> (text: S) -> MyMessage<'a>
9adc69d514 2026-03-25 arcade: where S: Into<Cow<'a, str>> {
9adc69d514 2026-03-25 arcade: let text = text.into();
9adc69d514 2026-03-25 arcade: MyMessage::Html { text }
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: pub fn html_to <'a, S> (text: S, to: ChatPeerId) -> MyMessage<'a>
9adc69d514 2026-03-25 arcade: where S: Into<Cow<'a, str>> {
9adc69d514 2026-03-25 arcade: let text = text.into();
9adc69d514 2026-03-25 arcade: MyMessage::HtmlTo { text, to }
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: pub fn html_to_kb <'a, S> (text: S, to: ChatPeerId, kb: InlineKeyboardMarkup) -> MyMessage<'a>
9adc69d514 2026-03-25 arcade: where S: Into<Cow<'a, str>> {
9adc69d514 2026-03-25 arcade: let text = text.into();
9adc69d514 2026-03-25 arcade: MyMessage::HtmlToKb { text, to, kb }
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: pub fn text <'a, S> (text: S) -> MyMessage<'a>
9adc69d514 2026-03-25 arcade: where S: Into<Cow<'a, str>> {
9adc69d514 2026-03-25 arcade: let text = text.into();
9adc69d514 2026-03-25 arcade: MyMessage::Text { text }
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: pub fn text_to <'a, S> (text: S, to: ChatPeerId) -> MyMessage<'a>
9adc69d514 2026-03-25 arcade: where S: Into<Cow<'a, str>> {
9adc69d514 2026-03-25 arcade: let text = text.into();
9adc69d514 2026-03-25 arcade: MyMessage::TextTo { text, to }
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: fn req (&self, tg: &Tg) -> Result<SendMessage> {
9adc69d514 2026-03-25 arcade: Ok(match self {
9adc69d514 2026-03-25 arcade: MyMessage::Html { text } =>
9adc69d514 2026-03-25 arcade: SendMessage::new(tg.owner, text.as_ref())
9adc69d514 2026-03-25 arcade: .with_parse_mode(ParseMode::Html),
9adc69d514 2026-03-25 arcade: MyMessage::HtmlTo { text, to } =>
9adc69d514 2026-03-25 arcade: SendMessage::new(*to, text.as_ref())
9adc69d514 2026-03-25 arcade: .with_parse_mode(ParseMode::Html),
9adc69d514 2026-03-25 arcade: MyMessage::HtmlToKb { text, to, kb } =>
9adc69d514 2026-03-25 arcade: SendMessage::new(*to, text.as_ref())
9adc69d514 2026-03-25 arcade: .with_parse_mode(ParseMode::Html)
9adc69d514 2026-03-25 arcade: .with_reply_markup(kb.clone()),
9adc69d514 2026-03-25 arcade: MyMessage::Text { text } =>
9adc69d514 2026-03-25 arcade: SendMessage::new(tg.owner, text.as_ref())
9adc69d514 2026-03-25 arcade: .with_parse_mode(ParseMode::MarkdownV2),
9adc69d514 2026-03-25 arcade: MyMessage::TextTo { text, to } =>
9adc69d514 2026-03-25 arcade: SendMessage::new(*to, text.as_ref())
9adc69d514 2026-03-25 arcade: .with_parse_mode(ParseMode::MarkdownV2),
9adc69d514 2026-03-25 arcade: })
9adc69d514 2026-03-25 arcade: }
13265e7697 2026-01-10 arcade: }
9c4f09193a 2026-01-09 arcade:
9c4f09193a 2026-01-09 arcade: #[derive(Clone)]
9c4f09193a 2026-01-09 arcade: pub struct Tg {
9c4f09193a 2026-01-09 arcade: pub me: Bot,
9c4f09193a 2026-01-09 arcade: pub owner: ChatPeerId,
9c4f09193a 2026-01-09 arcade: pub client: Client,
9c4f09193a 2026-01-09 arcade: }
9c4f09193a 2026-01-09 arcade:
9c4f09193a 2026-01-09 arcade: impl Tg {
fabcca1eaf 2026-01-09 arcade: /// Construct a new `Tg` instance from configuration.
fabcca1eaf 2026-01-09 arcade: ///
fabcca1eaf 2026-01-09 arcade: /// The `settings` must provide the following keys:
fabcca1eaf 2026-01-09 arcade: /// - `"api_key"` (string),
fabcca1eaf 2026-01-09 arcade: /// - `"owner"` (integer chat id),
fabcca1eaf 2026-01-09 arcade: /// - `"api_gateway"` (string).
fabcca1eaf 2026-01-09 arcade: ///
fabcca1eaf 2026-01-09 arcade: /// The function initialises the client, configures the gateway and fetches the bot identity
fabcca1eaf 2026-01-09 arcade: /// before returning the constructed `Tg`.
9c4f09193a 2026-01-09 arcade: pub async fn new (settings: &config::Config) -> Result<Tg> {
9c4f09193a 2026-01-09 arcade: let api_key = settings.get_string("api_key").stack()?;
9c4f09193a 2026-01-09 arcade:
9c4f09193a 2026-01-09 arcade: let owner = ChatPeerId::from(settings.get_int("owner").stack()?);
9c4f09193a 2026-01-09 arcade: let client = Client::new(&api_key).stack()?
9c4f09193a 2026-01-09 arcade: .with_host(settings.get_string("api_gateway").stack()?)
9c4f09193a 2026-01-09 arcade: .with_max_retries(0);
9c4f09193a 2026-01-09 arcade: let me = client.execute(GetBot).await.stack()?;
9c4f09193a 2026-01-09 arcade: Ok(Tg {
9c4f09193a 2026-01-09 arcade: me,
9c4f09193a 2026-01-09 arcade: owner,
9c4f09193a 2026-01-09 arcade: client,
9c4f09193a 2026-01-09 arcade: })
9c4f09193a 2026-01-09 arcade: }
9c4f09193a 2026-01-09 arcade:
fabcca1eaf 2026-01-09 arcade: /// Send a text message to a chat, using an optional target and parse mode.
fabcca1eaf 2026-01-09 arcade: ///
fabcca1eaf 2026-01-09 arcade: /// # Returns
fabcca1eaf 2026-01-09 arcade: /// The sent `Message` on success.
9adc69d514 2026-03-25 arcade: pub async fn send (&self, msg: MyMessage<'_>) -> Result<Message> {
9adc69d514 2026-03-25 arcade: self.client.execute(msg.req(self)?).await.stack()
fabcca1eaf 2026-01-09 arcade: }
fabcca1eaf 2026-01-09 arcade:
fabcca1eaf 2026-01-09 arcade: /// Create a copy of this `Tg` with the owner replaced by the given chat ID.
fabcca1eaf 2026-01-09 arcade: ///
fabcca1eaf 2026-01-09 arcade: /// # Parameters
fabcca1eaf 2026-01-09 arcade: /// - `owner`: The Telegram chat identifier to set as the new owner (expressed as an `i64`).
fabcca1eaf 2026-01-09 arcade: ///
fabcca1eaf 2026-01-09 arcade: /// # Returns
fabcca1eaf 2026-01-09 arcade: /// A new `Tg` instance identical to the original except its `owner` field is set to the provided chat ID.
fabcca1eaf 2026-01-09 arcade: pub fn with_owner <O>(&self, owner: O) -> Tg
fabcca1eaf 2026-01-09 arcade: where O: Into<i64> {
fabcca1eaf 2026-01-09 arcade: Tg {
fabcca1eaf 2026-01-09 arcade: owner: ChatPeerId::from(owner.into()),
9c4f09193a 2026-01-09 arcade: ..self.clone()
9c4f09193a 2026-01-09 arcade: }
9c4f09193a 2026-01-09 arcade: }
9c4f09193a 2026-01-09 arcade: }