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