9adc69d514 2026-03-25 arcade: use crate::{
9adc69d514 2026-03-25 arcade: Arc,
9adc69d514 2026-03-25 arcade: Mutex,
3fd8c40aa8 2026-03-30 arcade: core::FeedList,
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: fmt,
be0b8602d1 2026-04-18 arcade: time::Duration,
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: };
be0b8602d1 2026-04-18 arcade: use smol::Timer;
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::{
be0b8602d1 2026-04-18 arcade: api::{
be0b8602d1 2026-04-18 arcade: Client,
be0b8602d1 2026-04-18 arcade: ExecuteError
be0b8602d1 2026-04-18 arcade: },
9c4f09193a 2026-01-09 arcade: types::{
3fd8c40aa8 2026-03-30 arcade: AnswerCallbackQuery,
9c4f09193a 2026-01-09 arcade: Bot,
9c4f09193a 2026-01-09 arcade: ChatPeerId,
be0b8602d1 2026-04-18 arcade: EditMessageResult,
be0b8602d1 2026-04-18 arcade: EditMessageText,
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 {
3fd8c40aa8 2026-03-30 arcade: // Edit one feed (version, name)
3fd8c40aa8 2026-03-30 arcade: Edit(u8, String),
9adc69d514 2026-03-25 arcade: // List all feeds (version, name to show, page number)
be0b8602d1 2026-04-18 arcade: List(u8, String, usize),
3fd8c40aa8 2026-03-30 arcade: // Show root menu (version)
3fd8c40aa8 2026-03-30 arcade: Menu(u8),
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
9adc69d514 2026-03-25 arcade: impl Callback {
3fd8c40aa8 2026-03-30 arcade: pub fn edit <S>(text: S) -> Callback
3fd8c40aa8 2026-03-30 arcade: where S: Into<String> {
3fd8c40aa8 2026-03-30 arcade: Callback::Edit(CB_VERSION, text.into())
3fd8c40aa8 2026-03-30 arcade: }
3fd8c40aa8 2026-03-30 arcade:
be0b8602d1 2026-04-18 arcade: pub fn list <S>(text: S, page: usize) -> Callback
3fd8c40aa8 2026-03-30 arcade: where S: Into<String> {
3fd8c40aa8 2026-03-30 arcade: Callback::List(CB_VERSION, text.into(), page)
3fd8c40aa8 2026-03-30 arcade: }
3fd8c40aa8 2026-03-30 arcade:
3fd8c40aa8 2026-03-30 arcade: pub fn menu () -> Callback {
3fd8c40aa8 2026-03-30 arcade: Callback::Menu(CB_VERSION)
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 {
3fd8c40aa8 2026-03-30 arcade: Callback::Edit(version, .. ) => *version,
9adc69d514 2026-03-25 arcade: Callback::List(version, .. ) => *version,
3fd8c40aa8 2026-03-30 arcade: Callback::Menu(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: }
13265e7697 2026-01-10 arcade: }
13265e7697 2026-01-10 arcade:
9adc69d514 2026-03-25 arcade: /// Produce new Keyboard Markup from current Callback
be0b8602d1 2026-04-18 arcade: pub async fn get_kb (cb: &Callback, feeds: &Arc<Mutex<FeedList>>) -> 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 {
3fd8c40aa8 2026-03-30 arcade: Callback::Edit(_, _name) => { // XXX edit missing
3fd8c40aa8 2026-03-30 arcade: let kb: Vec<Vec<InlineKeyboardButton>> = vec![];
3fd8c40aa8 2026-03-30 arcade: InlineKeyboardMarkup::from(kb)
3fd8c40aa8 2026-03-30 arcade: },
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 {
be0b8602d1 2026-04-18 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: if name.is_empty() {
be0b8602d1 2026-04-18 arcade: let feed_iter = feeds.iter().skip(start);
be0b8602d1 2026-04-18 arcade: for (id, name) in feed_iter {
374eadef45 2026-03-28 arcade: kb.push(vec![
374eadef45 2026-03-28 arcade: InlineKeyboardButton::for_callback_data(
374eadef45 2026-03-28 arcade: format!("{}. {}", id, name),
3fd8c40aa8 2026-03-30 arcade: Callback::edit(name).to_string()),
374eadef45 2026-03-28 arcade: ]);
be0b8602d1 2026-04-18 arcade: i += 1;
be0b8602d1 2026-04-18 arcade: if i == end { break }
374eadef45 2026-03-28 arcade: }
9adc69d514 2026-03-25 arcade: } else {
374eadef45 2026-03-28 arcade: let mut found = false;
374eadef45 2026-03-28 arcade: let mut first_page = None;
374eadef45 2026-03-28 arcade: for (id, feed_name) in feeds.iter() {
374eadef45 2026-03-28 arcade: if name == feed_name {
374eadef45 2026-03-28 arcade: found = true;
374eadef45 2026-03-28 arcade: }
374eadef45 2026-03-28 arcade: i += 1;
374eadef45 2026-03-28 arcade: kb.push(vec![
374eadef45 2026-03-28 arcade: InlineKeyboardButton::for_callback_data(
374eadef45 2026-03-28 arcade: format!("{}. {}", id, feed_name),
374eadef45 2026-03-28 arcade: Callback::list("xxx", *page).to_string()), // XXX edit
374eadef45 2026-03-28 arcade: ]);
be0b8602d1 2026-04-18 arcade: if i == end {
374eadef45 2026-03-28 arcade: // page complete, if found we got the right page, if not - reset and
374eadef45 2026-03-28 arcade: // continue.
374eadef45 2026-03-28 arcade: if found {
374eadef45 2026-03-28 arcade: break
374eadef45 2026-03-28 arcade: } else {
374eadef45 2026-03-28 arcade: if first_page.is_none() {
374eadef45 2026-03-28 arcade: first_page = Some(kb);
374eadef45 2026-03-28 arcade: }
374eadef45 2026-03-28 arcade: kb = vec![];
374eadef45 2026-03-28 arcade: i = 0
374eadef45 2026-03-28 arcade: }
374eadef45 2026-03-28 arcade: }
374eadef45 2026-03-28 arcade: }
374eadef45 2026-03-28 arcade: if !found {
374eadef45 2026-03-28 arcade: // name not found, showing first page
374eadef45 2026-03-28 arcade: kb = first_page.unwrap_or_default();
374eadef45 2026-03-28 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: },
3fd8c40aa8 2026-03-30 arcade: Callback::Menu(_) => {
3fd8c40aa8 2026-03-30 arcade: let kb = vec![
3fd8c40aa8 2026-03-30 arcade: vec![
3fd8c40aa8 2026-03-30 arcade: InlineKeyboardButton::for_callback_data(
3fd8c40aa8 2026-03-30 arcade: "Add new channel",
3fd8c40aa8 2026-03-30 arcade: Callback::menu().to_string()), // new XXX
3fd8c40aa8 2026-03-30 arcade: ],
3fd8c40aa8 2026-03-30 arcade: vec![
3fd8c40aa8 2026-03-30 arcade: InlineKeyboardButton::for_callback_data(
3fd8c40aa8 2026-03-30 arcade: "List channels",
3fd8c40aa8 2026-03-30 arcade: Callback::list("", 0).to_string()),
3fd8c40aa8 2026-03-30 arcade: ],
3fd8c40aa8 2026-03-30 arcade: ];
3fd8c40aa8 2026-03-30 arcade: InlineKeyboardMarkup::from(kb)
3fd8c40aa8 2026-03-30 arcade: },
9adc69d514 2026-03-25 arcade: };
13265e7697 2026-01-10 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: }
9adc69d514 2026-03-25 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: 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: })
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()?);
be0b8602d1 2026-04-18 arcade: // We don't use retries, as in async environment this will just get us stuck for extra
be0b8602d1 2026-04-18 arcade: // amount of time on simple requests. Just bail, show error and ack it in the code. In
be0b8602d1 2026-04-18 arcade: // other case we might got stuck with multiple open transactions in database.
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()
9adc69d514 2026-03-25 arcade: }
9adc69d514 2026-03-25 arcade:
3fd8c40aa8 2026-03-30 arcade: pub async fn answer_cb (&self, id: String, text: String) -> Result<bool> {
3fd8c40aa8 2026-03-30 arcade: self.client.execute(
3fd8c40aa8 2026-03-30 arcade: AnswerCallbackQuery::new(id)
3fd8c40aa8 2026-03-30 arcade: .with_text(text)
3fd8c40aa8 2026-03-30 arcade: ).await.stack()
3fd8c40aa8 2026-03-30 arcade: }
3fd8c40aa8 2026-03-30 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> {
9c4f09193a 2026-01-09 arcade: Tg {
fabcca1eaf 2026-01-09 arcade: owner: ChatPeerId::from(owner.into()),
9c4f09193a 2026-01-09 arcade: ..self.clone()
be0b8602d1 2026-04-18 arcade: }
be0b8602d1 2026-04-18 arcade: }
be0b8602d1 2026-04-18 arcade:
be0b8602d1 2026-04-18 arcade: pub async fn update_message (&self, chat_id: i64, message_id: i64, text: String, feeds: &Arc<Mutex<FeedList>>, cb: Callback) -> Result<EditMessageResult> {
be0b8602d1 2026-04-18 arcade: loop {
be0b8602d1 2026-04-18 arcade: let req = EditMessageText::for_chat_message(chat_id, message_id, &text)
be0b8602d1 2026-04-18 arcade: .with_reply_markup(get_kb(&cb, feeds).await.stack()?);
be0b8602d1 2026-04-18 arcade: let res = self.client.execute(req).await;
be0b8602d1 2026-04-18 arcade: match res {
be0b8602d1 2026-04-18 arcade: Ok(res) => return Ok(res),
be0b8602d1 2026-04-18 arcade: Err(ref err) => {
be0b8602d1 2026-04-18 arcade: if let ExecuteError::Response(resp) = err
be0b8602d1 2026-04-18 arcade: && let Some(delay) = resp.retry_after()
be0b8602d1 2026-04-18 arcade: {
be0b8602d1 2026-04-18 arcade: if delay > 60 {
be0b8602d1 2026-04-18 arcade: return res.context("Delay too big (>60), not waiting.");
be0b8602d1 2026-04-18 arcade: }
be0b8602d1 2026-04-18 arcade: Timer::after(Duration::from_secs(delay)).await;
be0b8602d1 2026-04-18 arcade: } else {
be0b8602d1 2026-04-18 arcade: return res.context("Can't update message");
be0b8602d1 2026-04-18 arcade: }
be0b8602d1 2026-04-18 arcade: },
be0b8602d1 2026-04-18 arcade: };
9c4f09193a 2026-01-09 arcade: }
9c4f09193a 2026-01-09 arcade: }
9c4f09193a 2026-01-09 arcade: }