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