9171c791eb 2021-11-13 1: mod command;
9171c791eb 2021-11-13 2: mod core;
b2297258c7 2021-11-06 3:
b2297258c7 2021-11-06 4: use futures::StreamExt;
9171c791eb 2021-11-13 5: use anyhow::Result;
b2b8005309 2020-12-18 6:
b2b8005309 2020-12-18 7: #[tokio::main]
b2b8005309 2020-12-18 8: async fn main() -> Result<()> {
28da2e2a00 2022-03-12 9: let settings = config::Config::builder()
28da2e2a00 2022-03-12 10: .add_source(config::File::with_name("rsstg"))
28da2e2a00 2022-03-12 11: .build()?;
b2b8005309 2020-12-18 12:
9171c791eb 2021-11-13 13: let core = core::Core::new(settings).await?;
b2b8005309 2020-12-18 14:
b2b8005309 2020-12-18 15: let mut stream = core.stream();
a7f91033c0 2021-11-13 16: stream.allowed_updates(&[telegram_bot::AllowedUpdate::Message]);
a7f91033c0 2021-11-13 17: let mut reply_to: Option<telegram_bot::UserId>;
b2b8005309 2020-12-18 18:
b2b8005309 2020-12-18 19: loop {
b2b8005309 2020-12-18 20: reply_to = None;
b2b8005309 2020-12-18 21: match stream.next().await {
b2b8005309 2020-12-18 22: Some(update) => {
f988dfd28f 2022-02-13 23: if let Err(err) = handle(update?, &core, &reply_to).await {
26339860ce 2022-02-13 24: core.send(&format!("🛑 {:?}", err), reply_to, None).await?;
a7f91033c0 2021-11-13 25: };
a7f91033c0 2021-11-13 26: },
a7f91033c0 2021-11-13 27: None => {
26339860ce 2022-02-13 28: core.send("🛑 None error.", None, None).await?;
a7f91033c0 2021-11-13 29: }
a7f91033c0 2021-11-13 30: };
a7f91033c0 2021-11-13 31: }
a7f91033c0 2021-11-13 32: }
a7f91033c0 2021-11-13 33:
a7f91033c0 2021-11-13 34: async fn handle(update: telegram_bot::Update, core: &core::Core, mut _reply_to: &Option<telegram_bot::UserId>) -> Result<()> {
f988dfd28f 2022-02-13 35: if let telegram_bot::UpdateKind::Message(message) = update.kind {
f988dfd28f 2022-02-13 36: if let telegram_bot::MessageKind::Text { ref data, .. } = message.kind {
f988dfd28f 2022-02-13 37: let sender = message.from.id;
f988dfd28f 2022-02-13 38: let words: Vec<&str> = data.split_whitespace().collect();
f988dfd28f 2022-02-13 39: match match words[0] {
f988dfd28f 2022-02-13 40: "/check" | "/clean" | "/enable" | "/delete" | "/disable" => command::command(core, sender, words).await,
f988dfd28f 2022-02-13 41: "/start" => command::start(core, sender).await,
f988dfd28f 2022-02-13 42: "/list" => command::list(core, sender).await,
f988dfd28f 2022-02-13 43: "/add" | "/update" => command::update(core, sender, words).await,
f988dfd28f 2022-02-13 44: _ => Ok(()),
f988dfd28f 2022-02-13 45: } {
26339860ce 2022-02-13 46: Err(err) => core.send(format!("🛑 {:?}", err), Some(sender), None).await?,
f988dfd28f 2022-02-13 47: Ok(()) => {},
f988dfd28f 2022-02-13 48: };
f988dfd28f 2022-02-13 49: };
4acdad1942 2020-11-26 50: };
61df933942 2020-11-18 51:
61df933942 2020-11-18 52: Ok(())
61df933942 2020-11-18 53: }