manuel/telegram-bot-calendar
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow notifications to be sent days ahead of time

Browse Source
This commit is contained in:
Manuel Vögele
2024-01-12 14:44:14 +01:00
parent 26fd84b827
commit a2b9384d5d
8 changed files with 131 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
src/bot.rs
View File

@@ -11,7 +11,9 @@ use teloxide::dispatching::dialogue;
use teloxide::dispatching::dialogue::InMemStorage;
use teloxide::dispatching::{UpdateFilterExt, UpdateHandler};
use teloxide::dptree::{case, deps};
use teloxide::payloads::SendMessageSetters;
use teloxide::prelude::Dispatcher;
use teloxide::requests::Request;
use teloxide::requests::Requester;
use teloxide::types::ChatId;
use teloxide::types::ChatMemberKind;
@@ -32,6 +34,7 @@ pub enum Command {
#[command()]
SetCalendar,
SetLocale,
RemindDaysAhead,
}
pub async fn spawn(bot: Throttle<Bot>, db: Database) {
@@ -45,7 +48,8 @@ pub async fn spawn(bot: Throttle<Bot>, db: Database) {
fn build_handler_chain() -> UpdateHandler<Error> {
let command_handler = teloxide::filter_command::<Command, _>()
.branch(case![Command::SetCalendar].endpoint(set_calendar))
.branch(case![Command::SetLocale].endpoint(set_locale));
.branch(case![Command::SetLocale].endpoint(set_locale))
.branch(case![Command::RemindDaysAhead].endpoint(set_remind_days_ahead));
let my_chat_member_handler = Update::filter_my_chat_member().endpoint(handle_my_chat_member);
@@ -96,6 +100,7 @@ async fn set_calendar(bot: Throttle<Bot>, msg: Message, db: Database) -> Result<
last_reminder: None,
pinned_message_id: None,
locale: "de".into(),
remind_days_ahead: 0,
};
fetch_and_announce_appointment(&bot, &mut chat_info, &db).await?;
@@ -123,6 +128,61 @@ async fn set_locale(_bot: Throttle<Bot>, msg: Message, db: Database) -> Result<(
Ok(())
}
async fn set_remind_days_ahead(
bot: Throttle<Bot>,
msg: Message,
db: Database,
) -> Result<(), Error> {
let chat = db.lock().await.transaction(|db| {
use schema::chat::dsl::*;
chat.filter(telegram_id.eq(msg.chat.id.0))
.first::<DbChat>(db)
})?;
let chat = ChatInfo::<Utc>::from(chat);
let days_ahead = msg
.text()
.map(|text| text.splitn(2, " ").nth(1).map(|param| param.parse().ok()))
.flatten()
.flatten();
if days_ahead.is_none() {
bot.send_message(
msg.chat.id,
t!("errors.param_no_days", locale = &chat.locale),
)
.reply_to_message_id(msg.id)
.send()
.await?;
return Ok(());
}
let days_ahead: i64 = days_ahead.unwrap();
if days_ahead < 0 {
bot.send_message(
msg.chat.id,
t!("errors.param_no_days_nonnegative", locale = &chat.locale),
)
.reply_to_message_id(msg.id)
.send()
.await?;
return Ok(());
}
db.lock().await.transaction::<_, Error, _>(|db| {
use schema::chat::dsl::*;
diesel::update(chat)
.filter(telegram_id.eq(msg.chat.id.0))
.set(remind_days_ahead.eq(days_ahead))
.execute(db)?;
Ok(())
})?;
Ok(())
}
pub async fn fetch_and_announce_appointment(
bot: &Throttle<Bot>,
chat_info: &mut ChatInfo<Utc>,

5
src/db.rs
View File

@@ -13,6 +13,7 @@ pub struct DbChat {
last_reminder: Option<i64>,
pinned_message: Option<i32>,
locale: Option<String>,
remind_days_ahead: i64,
}
pub struct ChatInfo<Tz: TimeZone> {
@@ -22,6 +23,7 @@ pub struct ChatInfo<Tz: TimeZone> {
pub last_reminder: Option<DateTime<Tz>>,
pub pinned_message_id: Option<i32>,
pub locale: String,
pub remind_days_ahead: u64,
}
impl From<DbChat> for ChatInfo<Utc> {
@@ -41,6 +43,8 @@ impl From<DbChat> for ChatInfo<Utc> {
let locale = db_chat.locale.unwrap_or("de".into());
let remind_days_ahead = db_chat.remind_days_ahead.try_into().unwrap_or(0);
ChatInfo {
id: db_chat.telegram_id,
calendar: db_chat.calendar,
@@ -48,6 +52,7 @@ impl From<DbChat> for ChatInfo<Utc> {
last_reminder,
pinned_message_id: db_chat.pinned_message,
locale,
remind_days_ahead,
}
}
}

64
src/main.rs
View File

@@ -10,7 +10,7 @@ use std::{env, fs::File, io::BufReader, sync::Arc};
use anyhow::Result;
use async_mutex::Mutex;
use bot::fetch_and_announce_appointment;
use chrono::{DateTime, NaiveTime, TimeZone, Utc};
use chrono::{DateTime, Days, NaiveTime, TimeZone, Utc};
use chrono_tz::Europe;
use db::ChatInfo;
use diesel::result::Error::NotFound;
@@ -41,6 +41,8 @@ pub struct Config {
poll_interval: u64,
#[serde(deserialize_with = "deserialize_time")]
reminder_time: NaiveTime,
#[serde(deserialize_with = "deserialize_time")]
preceeding_day_reminder_time: NaiveTime,
}
fn deserialize_time<'de, D: Deserializer<'de>>(deserializer: D) -> Result<NaiveTime, D::Error> {
@@ -79,7 +81,7 @@ async fn main() {
db.run_pending_migrations(MIGRATIONS).unwrap();
let db = Arc::new(Mutex::new(db));
let bot = Bot::new(config.token).throttle(Limits::default());
let bot = Bot::new(&config.token).throttle(Limits::default());
{
let db = db.clone();
@@ -104,16 +106,18 @@ async fn main() {
Err(e) => Err(e).unwrap(),
};
let next_appointment =
next_appointment.map(|timestamp| Utc.timestamp_opt(timestamp, 0).unwrap());
let sleep_duration = next_appointment
.map(|timestamp| Utc.timestamp_opt(timestamp, 0).unwrap())
.map(|date_time| date_time - now)
.map(|next_appointment| next_appointment - now)
.map(|duration| duration.to_std().unwrap())
.filter(|duration| *duration < poll_duration)
.unwrap_or(poll_duration);
sleep(sleep_duration).await;
let result = check_task(&bot, config.reminder_time, &db).await;
let result = check_task(&bot, &config, &db).await;
if let Err(e) = result {
error!("{}\nBacktrace:\n{}", e, e.backtrace());
}
@@ -131,14 +135,15 @@ struct Reminder<Tz: TimeZone> {
text: String,
}
async fn check_task(bot: &Throttle<Bot>, reminder_time: NaiveTime, db: &Database) -> Result<()> {
// Checks if the date of the next appointment has changed (and announces if so)
// Additionally, checks if it is time for a reminder and sends that reminder if necessary
async fn check_task(bot: &Throttle<Bot>, config: &Config, db: &Database) -> Result<()> {
let chats = db.lock().await.transaction(|db| {
use schema::chat::dsl::*;
chat.load::<DbChat>(db)
})?;
let now = Utc::now().with_timezone(&Europe::Berlin);
let today = now.date_naive();
for chat in chats {
let mut chat_info = ChatInfo::from(chat);
@@ -150,10 +155,6 @@ async fn check_task(bot: &Throttle<Bot>, reminder_time: NaiveTime, db: &Database
};
let appointment = appointment.with_timezone(&Europe::Berlin);
if appointment.start.date_naive() != today {
continue;
}
let mut reminder = None;
if now >= appointment.start {
reminder = Some(Reminder {
@@ -161,24 +162,39 @@ async fn check_task(bot: &Throttle<Bot>, reminder_time: NaiveTime, db: &Database
text: t!("messages.starting_now", locale = &chat_info.locale),
});
} else {
let reminder_date_time = now
.date_naive()
.and_time(reminder_time)
let reminder_day =
appointment.start.date_naive() - Days::new(chat_info.remind_days_ahead);
let reminder_date_time = if chat_info.remind_days_ahead == 0 {
reminder_day.and_time(config.reminder_time)
} else {
reminder_day.and_time(config.preceeding_day_reminder_time)
};
let reminder_date_time = reminder_date_time
.and_local_timezone(now.timezone())
.unwrap();
if now >= reminder_date_time {
// TODO This can have weird effects if it's happenig around midnight, since it's not timezone aware (and may even mix multiple timezones)
let remaining_time = appointment.start.date_naive() - now.date_naive();
let remaining_days = remaining_time.num_days();
let message_id = match remaining_days {
0 => "messages.appointment_today",
1 => "messages.appointment_tomorrow",
_ => "messages.appointment_soon",
};
let reminder_text = t!(
message_id,
locale = &chat_info.locale,
no_days = remaining_days,
start_time = &appointment.start.format("%H:%M").to_string(),
uk_time = &appointment
.start
.with_timezone(&Europe::London)
.format("%H:%M")
.to_string()
);
reminder = Some(Reminder {
time: reminder_date_time,
text: t!(
"messages.appointment_today",
locale = &chat_info.locale,
start_time = &appointment.start.format("%H:%M").to_string(),
uk_time = &appointment
.start
.with_timezone(&Europe::London)
.format("%H:%M")
.to_string()
),
text: reminder_text,
})
}
}

21
src/schema.rs
View File

@@ -1,14 +1,15 @@
// @generated automatically by Diesel CLI.
diesel::table! {
chat (id) {
id -> Integer,
telegram_id -> BigInt,
calendar -> Text,
next_appointment_start -> Nullable<BigInt>,
next_appointment_end -> Nullable<BigInt>,
last_reminder -> Nullable<BigInt>,
pinned_message_id -> Nullable<Integer>,
locale -> Nullable<Text>,
}
chat (id) {
id -> Integer,
telegram_id -> BigInt,
calendar -> Text,
next_appointment_start -> Nullable<BigInt>,
next_appointment_end -> Nullable<BigInt>,
last_reminder -> Nullable<BigInt>,
pinned_message_id -> Nullable<Integer>,
locale -> Nullable<Text>,
remind_days_ahead -> BigInt,
}
}