0000: 75 73 65 20 61 6e 79 68 6f 77 3a 3a 7b 61 6e 79 use anyhow::{any
0010: 68 6f 77 2c 20 62 61 69 6c 2c 20 43 6f 6e 74 65 how, bail, Conte
0020: 78 74 2c 20 52 65 73 75 6c 74 7d 3b 0a 75 73 65 xt, Result};.use
0030: 20 61 73 79 6e 63 5f 73 74 64 3a 3a 74 61 73 6b async_std::task
0040: 3b 0a 75 73 65 20 63 68 72 6f 6e 6f 3a 3a 44 61 ;.use chrono::Da
0050: 74 65 54 69 6d 65 3b 0a 75 73 65 20 73 71 6c 78 teTime;.use sqlx
0060: 3a 3a 70 6f 73 74 67 72 65 73 3a 3a 50 67 50 6f ::postgres::PgPo
0070: 6f 6c 4f 70 74 69 6f 6e 73 3b 0a 75 73 65 20 74 olOptions;.use t
0080: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 7b 0a 09 elegram_bot::{..
0090: 5f 62 61 73 65 3a 3a 45 72 72 6f 72 20 61 73 20 _base::Error as
00a0: 54 67 72 45 72 72 6f 72 2c 0a 09 45 72 72 6f 72 TgrError,..Error
00b0: 20 61 73 20 54 67 45 72 72 6f 72 2c 0a 7d 3b 0a as TgError,.};.
00c0: 75 73 65 20 74 68 69 73 65 72 72 6f 72 3a 3a 45 use thiserror::E
00d0: 72 72 6f 72 3b 0a 0a 75 73 65 20 73 74 64 3a 3a rror;..use std::
00e0: 7b 0a 09 62 6f 72 72 6f 77 3a 3a 43 6f 77 2c 0a {..borrow::Cow,.
00f0: 09 63 6f 6c 6c 65 63 74 69 6f 6e 73 3a 3a 7b 0a .collections::{.
0100: 09 09 42 54 72 65 65 4d 61 70 2c 0a 09 09 48 61 ..BTreeMap,...Ha
0110: 73 68 53 65 74 2c 0a 09 7d 2c 0a 09 6e 75 6d 3a shSet,..},..num:
0120: 3a 54 72 79 46 72 6f 6d 49 6e 74 45 72 72 6f 72 :TryFromIntError
0130: 2c 0a 09 73 79 6e 63 3a 3a 7b 0a 09 09 41 72 63 ,..sync::{...Arc
0140: 2c 0a 09 09 4d 75 74 65 78 0a 09 7d 2c 0a 7d 3b ,...Mutex..},.};
0150: 0a 0a 23 5b 64 65 72 69 76 65 28 45 72 72 6f 72 ..#[derive(Error
0160: 2c 20 44 65 62 75 67 29 5d 0a 70 75 62 20 65 6e , Debug)].pub en
0170: 75 6d 20 52 73 73 45 72 72 6f 72 20 7b 0a 09 23 um RssError {..#
0180: 5b 65 72 72 6f 72 28 74 72 61 6e 73 70 61 72 65 [error(transpare
0190: 6e 74 29 5d 0a 09 54 67 28 23 5b 66 72 6f 6d 5d nt)]..Tg(#[from]
01a0: 20 54 67 45 72 72 6f 72 29 2c 0a 09 23 5b 65 72 TgError),..#[er
01b0: 72 6f 72 28 74 72 61 6e 73 70 61 72 65 6e 74 29 ror(transparent)
01c0: 5d 0a 09 49 6e 74 28 23 5b 66 72 6f 6d 5d 20 54 ]..Int(#[from] T
01d0: 72 79 46 72 6f 6d 49 6e 74 45 72 72 6f 72 29 2c ryFromIntError),
01e0: 0a 7d 0a 0a 23 5b 64 65 72 69 76 65 28 43 6c 6f .}..#[derive(Clo
01f0: 6e 65 29 5d 0a 70 75 62 20 73 74 72 75 63 74 20 ne)].pub struct
0200: 43 6f 72 65 20 7b 0a 09 6f 77 6e 65 72 5f 63 68 Core {..owner_ch
0210: 61 74 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 at: telegram_bot
0220: 3a 3a 55 73 65 72 49 64 2c 0a 09 70 75 62 20 74 ::UserId,..pub t
0230: 67 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a g: telegram_bot:
0240: 3a 41 70 69 2c 0a 09 70 75 62 20 6d 79 3a 20 74 :Api,..pub my: t
0250: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 elegram_bot::Use
0260: 72 2c 0a 09 70 6f 6f 6c 3a 20 73 71 6c 78 3a 3a r,..pool: sqlx::
0270: 50 6f 6f 6c 3c 73 71 6c 78 3a 3a 50 6f 73 74 67 Pool<sqlx::Postg
0280: 72 65 73 3e 2c 0a 09 73 6f 75 72 63 65 73 3a 20 res>,..sources:
0290: 41 72 63 3c 4d 75 74 65 78 3c 48 61 73 68 53 65 Arc<Mutex<HashSe
02a0: 74 3c 41 72 63 3c 69 33 32 3e 3e 3e 3e 2c 0a 09 t<Arc<i32>>>>,..
02b0: 68 74 74 70 5f 63 6c 69 65 6e 74 3a 20 72 65 71 http_client: req
02c0: 77 65 73 74 3a 3a 43 6c 69 65 6e 74 2c 0a 7d 0a west::Client,.}.
02d0: 0a 69 6d 70 6c 20 43 6f 72 65 20 7b 0a 09 70 75 .impl Core {..pu
02e0: 62 20 66 6e 20 6e 65 77 28 73 65 74 74 69 6e 67 b fn new(setting
02f0: 73 3a 20 63 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 s: config::Confi
0300: 67 29 20 2d 3e 20 52 65 73 75 6c 74 3c 41 72 63 g) -> Result<Arc
0310: 3c 43 6f 72 65 3e 3e 20 7b 0a 09 09 6c 65 74 20 <Core>> {...let
0320: 6f 77 6e 65 72 20 3d 20 73 65 74 74 69 6e 67 73 owner = settings
0330: 2e 67 65 74 5f 69 6e 74 28 22 6f 77 6e 65 72 22 .get_int("owner"
0340: 29 3f 3b 0a 09 09 6c 65 74 20 61 70 69 5f 6b 65 )?;...let api_ke
0350: 79 20 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 y = settings.get
0360: 5f 73 74 72 69 6e 67 28 22 61 70 69 5f 6b 65 79 _string("api_key
0370: 22 29 3f 3b 0a 09 09 6c 65 74 20 74 67 20 3d 20 ")?;...let tg =
0380: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 41 70 telegram_bot::Ap
0390: 69 3a 3a 6e 65 77 28 61 70 69 5f 6b 65 79 29 3b i::new(api_key);
03a0: 0a 09 09 6c 65 74 20 74 67 5f 63 6c 6f 6e 65 64 ...let tg_cloned
03b0: 20 3d 20 74 67 2e 63 6c 6f 6e 65 28 29 3b 0a 0a = tg.clone();..
03c0: 09 09 6c 65 74 20 6d 75 74 20 63 6c 69 65 6e 74 ..let mut client
03d0: 20 3d 20 72 65 71 77 65 73 74 3a 3a 43 6c 69 65 = reqwest::Clie
03e0: 6e 74 3a 3a 62 75 69 6c 64 65 72 28 29 3b 0a 09 nt::builder();..
03f0: 09 69 66 20 6c 65 74 20 4f 6b 28 70 72 6f 78 79 .if let Ok(proxy
0400: 29 20 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 ) = settings.get
0410: 5f 73 74 72 69 6e 67 28 22 70 72 6f 78 79 22 29 _string("proxy")
0420: 20 7b 0a 09 09 09 6c 65 74 20 70 72 6f 78 79 20 {....let proxy
0430: 3d 20 72 65 71 77 65 73 74 3a 3a 50 72 6f 78 79 = reqwest::Proxy
0440: 3a 3a 61 6c 6c 28 70 72 6f 78 79 29 3f 3b 0a 09 ::all(proxy)?;..
0450: 09 09 63 6c 69 65 6e 74 20 3d 20 63 6c 69 65 6e ..client = clien
0460: 74 2e 70 72 6f 78 79 28 70 72 6f 78 79 29 3b 0a t.proxy(proxy);.
0470: 09 09 7d 0a 09 09 6c 65 74 20 68 74 74 70 5f 63 ..}...let http_c
0480: 6c 69 65 6e 74 20 3d 20 63 6c 69 65 6e 74 2e 62 lient = client.b
0490: 75 69 6c 64 28 29 3f 3b 0a 09 09 6c 65 74 20 63 uild()?;...let c
04a0: 6f 72 65 20 3d 20 41 72 63 3a 3a 6e 65 77 28 43 ore = Arc::new(C
04b0: 6f 72 65 20 7b 0a 09 09 09 74 67 2c 0a 09 09 09 ore {....tg,....
04c0: 6d 79 3a 20 74 61 73 6b 3a 3a 62 6c 6f 63 6b 5f my: task::block_
04d0: 6f 6e 28 61 73 79 6e 63 20 7b 0a 09 09 09 09 74 on(async {.....t
04e0: 67 5f 63 6c 6f 6e 65 64 2e 73 65 6e 64 28 74 65 g_cloned.send(te
04f0: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 47 65 74 4d legram_bot::GetM
0500: 65 29 2e 61 77 61 69 74 0a 09 09 09 7d 29 3f 2c e).await....})?,
0510: 0a 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 ....owner_chat:
0520: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 telegram_bot::Us
0530: 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 erId::new(owner)
0540: 2c 0a 09 09 09 70 6f 6f 6c 3a 20 50 67 50 6f 6f ,....pool: PgPoo
0550: 6c 4f 70 74 69 6f 6e 73 3a 3a 6e 65 77 28 29 0a lOptions::new().
0560: 09 09 09 09 2e 6d 61 78 5f 63 6f 6e 6e 65 63 74 .....max_connect
0570: 69 6f 6e 73 28 35 29 0a 09 09 09 09 2e 61 63 71 ions(5)......acq
0580: 75 69 72 65 5f 74 69 6d 65 6f 75 74 28 73 74 64 uire_timeout(std
0590: 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e ::time::Duration
05a0: 3a 3a 6e 65 77 28 33 30 30 2c 20 30 29 29 0a 09 ::new(300, 0))..
05b0: 09 09 09 2e 69 64 6c 65 5f 74 69 6d 65 6f 75 74 ....idle_timeout
05c0: 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 (std::time::Dura
05d0: 74 69 6f 6e 3a 3a 6e 65 77 28 36 30 2c 20 30 29 tion::new(60, 0)
05e0: 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 74 5f 6c )......connect_l
05f0: 61 7a 79 28 26 73 65 74 74 69 6e 67 73 2e 67 65 azy(&settings.ge
0600: 74 5f 73 74 72 69 6e 67 28 22 70 67 22 29 3f 29 t_string("pg")?)
0610: 3f 2c 0a 09 09 09 73 6f 75 72 63 65 73 3a 20 41 ?,....sources: A
0620: 72 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a 3a 6e rc::new(Mutex::n
0630: 65 77 28 48 61 73 68 53 65 74 3a 3a 6e 65 77 28 ew(HashSet::new(
0640: 29 29 29 2c 0a 09 09 09 68 74 74 70 5f 63 6c 69 ))),....http_cli
0650: 65 6e 74 2c 0a 09 09 7d 29 3b 0a 09 09 6c 65 74 ent,...});...let
0660: 20 63 6c 6f 6e 65 20 3d 20 63 6f 72 65 2e 63 6c clone = core.cl
0670: 6f 6e 65 28 29 3b 0a 09 09 74 61 73 6b 3a 3a 73 one();...task::s
0680: 70 61 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 pawn(async move
0690: 7b 0a 09 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 09 {....loop {.....
06a0: 6c 65 74 20 64 65 6c 61 79 20 3d 20 6d 61 74 63 let delay = matc
06b0: 68 20 26 63 6c 6f 6e 65 2e 61 75 74 6f 66 65 74 h &clone.autofet
06c0: 63 68 28 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 ch().await {....
06d0: 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a ..Err(err) => {.
06e0: 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 ......if let Err
06f0: 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 (err) = clone.se
0700: 6e 64 28 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 nd(format!("š
0710: 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f {:?}", err), No
0720: 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 74 20 ne, None).await
0730: 7b 0a 09 09 09 09 09 09 09 65 70 72 69 6e 74 6c {........eprintl
0740: 6e 21 28 22 41 75 74 6f 66 65 74 63 68 20 65 72 n!("Autofetch er
0750: 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a ror: {}", err);.
0760: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 73 ......};.......s
0770: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
0780: 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 63 73 28 36 30 on::from_secs(60
0790: 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 4f )......},......O
07a0: 6b 28 74 69 6d 65 29 20 3d 3e 20 2a 74 69 6d 65 k(time) => *time
07b0: 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 74 61 73 ,.....};.....tas
07c0: 6b 3a 3a 73 6c 65 65 70 28 64 65 6c 61 79 29 2e k::sleep(delay).
07d0: 61 77 61 69 74 3b 0a 09 09 09 7d 0a 09 09 7d 29 await;....}...})
07e0: 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 0a 09 7d 0a ;...Ok(core)..}.
07f0: 0a 09 70 75 62 20 66 6e 20 73 74 72 65 61 6d 28 ..pub fn stream(
0800: 26 73 65 6c 66 29 20 2d 3e 20 74 65 6c 65 67 72 &self) -> telegr
0810: 61 6d 5f 62 6f 74 3a 3a 55 70 64 61 74 65 73 53 am_bot::UpdatesS
0820: 74 72 65 61 6d 20 7b 0a 09 09 73 65 6c 66 2e 74 tream {...self.t
0830: 67 2e 73 74 72 65 61 6d 28 29 0a 09 7d 0a 0a 09 g.stream()..}...
0840: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 73 65 6e pub async fn sen
0850: 64 3c 27 61 2c 20 53 3e 28 26 73 65 6c 66 2c 20 d<'a, S>(&self,
0860: 6d 73 67 3a 20 53 2c 20 74 61 72 67 65 74 3a 20 msg: S, target:
0870: 4f 70 74 69 6f 6e 3c 74 65 6c 65 67 72 61 6d 5f Option<telegram_
0880: 62 6f 74 3a 3a 55 73 65 72 49 64 3e 2c 20 6d 6f bot::UserId>, mo
0890: 64 65 3a 20 4f 70 74 69 6f 6e 3c 74 65 6c 65 67 de: Option<teleg
08a0: 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a ram_bot::types::
08b0: 50 61 72 73 65 4d 6f 64 65 3e 29 20 2d 3e 20 52 ParseMode>) -> R
08c0: 65 73 75 6c 74 3c 28 29 3e 0a 09 77 68 65 72 65 esult<()>..where
08d0: 20 53 3a 20 49 6e 74 6f 3c 43 6f 77 3c 27 61 2c S: Into<Cow<'a,
08e0: 20 73 74 72 3e 3e 20 7b 0a 09 09 6c 65 74 20 6d str>> {...let m
08f0: 6f 64 65 20 3d 20 6d 6f 64 65 2e 75 6e 77 72 61 ode = mode.unwra
0900: 70 5f 6f 72 28 74 65 6c 65 67 72 61 6d 5f 62 6f p_or(telegram_bo
0910: 74 3a 3a 74 79 70 65 73 3a 3a 50 61 72 73 65 4d t::types::ParseM
0920: 6f 64 65 3a 3a 48 74 6d 6c 29 3b 0a 09 09 6c 65 ode::Html);...le
0930: 74 20 74 61 72 67 65 74 20 3d 20 74 61 72 67 65 t target = targe
0940: 74 2e 75 6e 77 72 61 70 5f 6f 72 28 73 65 6c 66 t.unwrap_or(self
0950: 2e 6f 77 6e 65 72 5f 63 68 61 74 29 3b 0a 09 09 .owner_chat);...
0960: 73 65 6c 66 2e 72 65 71 75 65 73 74 28 74 65 6c self.request(tel
0970: 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d egram_bot::SendM
0980: 65 73 73 61 67 65 3a 3a 6e 65 77 28 74 61 72 67 essage::new(targ
0990: 65 74 2c 20 6d 73 67 29 2e 70 61 72 73 65 5f 6d et, msg).parse_m
09a0: 6f 64 65 28 6d 6f 64 65 29 29 2e 61 77 61 69 74 ode(mode)).await
09b0: 3f 3b 0a 09 09 4f 6b 28 28 29 29 0a 09 7d 0a 0a ?;...Ok(())..}..
09c0: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 72 65 .pub async fn re
09d0: 71 75 65 73 74 3c 52 65 71 3a 20 74 65 6c 65 67 quest<Req: teleg
09e0: 72 61 6d 5f 62 6f 74 3a 3a 52 65 71 75 65 73 74 ram_bot::Request
09f0: 3e 20 28 26 73 65 6c 66 2c 20 72 65 71 3a 20 52 > (&self, req: R
0a00: 65 71 29 20 2d 3e 20 52 65 73 75 6c 74 3c 3c 52 eq) -> Result<<R
0a10: 65 71 3a 3a 52 65 73 70 6f 6e 73 65 20 61 73 20 eq::Response as
0a20: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 52 65 telegram_bot::Re
0a30: 73 70 6f 6e 73 65 54 79 70 65 3e 3a 3a 54 79 70 sponseType>::Typ
0a40: 65 2c 20 52 73 73 45 72 72 6f 72 3e 20 7b 0a 09 e, RssError> {..
0a50: 09 6c 6f 6f 70 20 7b 0a 09 09 09 6c 65 74 20 72 .loop {....let r
0a60: 65 73 20 3d 20 73 65 6c 66 2e 74 67 2e 73 65 6e es = self.tg.sen
0a70: 64 28 26 72 65 71 29 2e 61 77 61 69 74 3b 0a 09 d(&req).await;..
0a80: 09 09 6d 61 74 63 68 20 72 65 73 20 7b 0a 09 09 ..match res {...
0a90: 09 09 4f 6b 28 5f 29 20 3d 3e 20 72 65 74 75 72 ..Ok(_) => retur
0aa0: 6e 20 4f 6b 28 72 65 73 3f 29 2c 0a 09 09 09 09 n Ok(res?),.....
0ab0: 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 Err(err) => {...
0ac0: 09 09 09 6d 61 74 63 68 20 26 65 72 72 20 7b 0a ...match &err {.
0ad0: 09 09 09 09 09 09 54 67 45 72 72 6f 72 3a 3a 52 ......TgError::R
0ae0: 61 77 28 54 67 72 45 72 72 6f 72 3a 3a 54 65 6c aw(TgrError::Tel
0af0: 65 67 72 61 6d 45 72 72 6f 72 20 7b 20 64 65 73 egramError { des
0b00: 63 72 69 70 74 69 6f 6e 3a 20 5f 2c 20 70 61 72 cription: _, par
0b10: 61 6d 65 74 65 72 73 3a 20 53 6f 6d 65 28 70 61 ameters: Some(pa
0b20: 72 61 6d 73 29 20 7d 29 20 3d 3e 20 7b 0a 09 09 rams) }) => {...
0b30: 09 09 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 .....if let Some
0b40: 28 64 65 6c 61 79 29 20 3d 20 70 61 72 61 6d 73 (delay) = params
0b50: 2e 72 65 74 72 79 5f 61 66 74 65 72 20 7b 0a 09 .retry_after {..
0b60: 09 09 09 09 09 09 09 70 72 69 6e 74 6c 6e 21 28 .......println!(
0b70: 22 54 68 72 6f 74 74 6c 65 64 2c 20 77 61 69 74 "Throttled, wait
0b80: 69 6e 67 20 7b 7d 20 73 65 6e 63 6f 6e 64 73 2e ing {} senconds.
0b90: 22 2c 20 64 65 6c 61 79 29 3b 0a 09 09 09 09 09 ", delay);......
0ba0: 09 09 09 74 61 73 6b 3a 3a 73 6c 65 65 70 28 73 ...task::sleep(s
0bb0: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
0bc0: 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 63 73 28 64 65 on::from_secs(de
0bd0: 6c 61 79 2e 74 72 79 5f 69 6e 74 6f 28 29 3f 29 lay.try_into()?)
0be0: 29 2e 61 77 61 69 74 3b 0a 09 09 09 09 09 09 09 ).await;........
0bf0: 7d 20 65 6c 73 65 20 7b 0a 09 09 09 09 09 09 09 } else {........
0c00: 09 72 65 74 75 72 6e 20 45 72 72 28 65 72 72 2e .return Err(err.
0c10: 69 6e 74 6f 28 29 29 3b 0a 09 09 09 09 09 09 09 into());........
0c20: 7d 0a 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 }.......},......
0c30: 09 5f 20 3d 3e 20 72 65 74 75 72 6e 20 45 72 72 ._ => return Err
0c40: 28 65 72 72 2e 69 6e 74 6f 28 29 29 2c 0a 09 09 (err.into()),...
0c50: 09 09 09 7d 0a 09 09 09 09 7d 2c 0a 09 09 09 7d ...}.....},....}
0c60: 3b 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 ;...}..}...pub a
0c70: 73 79 6e 63 20 66 6e 20 63 68 65 63 6b 3c 53 3e sync fn check<S>
0c80: 28 26 73 65 6c 66 2c 20 69 64 3a 20 26 69 33 32 (&self, id: &i32
0c90: 2c 20 6f 77 6e 65 72 3a 20 53 2c 20 72 65 61 6c , owner: S, real
0ca0: 3a 20 62 6f 6f 6c 29 20 2d 3e 20 52 65 73 75 6c : bool) -> Resul
0cb0: 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a t<Cow<'_, str>>.
0cc0: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
0cd0: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
0ce0: 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 r = owner.into()
0cf0: 3b 0a 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 ;...let mut post
0d00: 65 64 3a 20 69 33 32 20 3d 20 30 3b 0a 09 09 6c ed: i32 = 0;...l
0d10: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
0d20: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
0d30: 29 2e 61 77 61 69 74 3f 3b 0a 0a 09 09 6c 65 74 ).await?;....let
0d40: 20 69 64 20 3d 20 7b 0a 09 09 09 6c 65 74 20 6d id = {....let m
0d50: 75 74 20 73 65 74 20 3d 20 73 65 6c 66 2e 73 6f ut set = self.so
0d60: 75 72 63 65 73 2e 6c 6f 63 6b 28 29 2e 75 6e 77 urces.lock().unw
0d70: 72 61 70 28 29 3b 0a 09 09 09 6d 61 74 63 68 20 rap();....match
0d80: 73 65 74 2e 67 65 74 28 69 64 29 20 7b 0a 09 09 set.get(id) {...
0d90: 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 69 64 ..Some(id) => id
0da0: 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 09 4e 6f .clone(),.....No
0db0: 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 6c 65 74 ne => {......let
0dc0: 20 69 64 20 3d 20 41 72 63 3a 3a 6e 65 77 28 2a id = Arc::new(*
0dd0: 69 64 29 3b 0a 09 09 09 09 09 73 65 74 2e 69 6e id);......set.in
0de0: 73 65 72 74 28 69 64 2e 63 6c 6f 6e 65 28 29 29 sert(id.clone())
0df0: 3b 0a 09 09 09 09 09 69 64 2e 63 6c 6f 6e 65 28 ;......id.clone(
0e00: 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 ).....},....}...
0e10: 7d 3b 0a 09 09 6c 65 74 20 63 6f 75 6e 74 20 3d };...let count =
0e20: 20 41 72 63 3a 3a 73 74 72 6f 6e 67 5f 63 6f 75 Arc::strong_cou
0e30: 6e 74 28 26 69 64 29 3b 0a 09 09 69 66 20 63 6f nt(&id);...if co
0e40: 75 6e 74 20 3d 3d 20 32 20 7b 0a 09 09 09 6c 65 unt == 2 {....le
0e50: 74 20 73 6f 75 72 63 65 20 3d 20 73 71 6c 78 3a t source = sqlx:
0e60: 3a 71 75 65 72 79 21 28 22 73 65 6c 65 63 74 20 :query!("select
0e70: 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e source_id, chann
0e80: 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 el_id, url, iv_h
0e90: 61 73 68 2c 20 6f 77 6e 65 72 2c 20 75 72 6c 5f ash, owner, url_
0ea0: 72 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f re from rsstg_so
0eb0: 75 72 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 urce where sourc
0ec0: 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 e_id = $1 and ow
0ed0: 6e 65 72 20 3d 20 24 32 22 2c 0a 09 09 09 09 2a ner = $2",.....*
0ee0: 69 64 2c 20 6f 77 6e 65 72 29 2e 66 65 74 63 68 id, owner).fetch
0ef0: 5f 6f 6e 65 28 26 6d 75 74 20 2a 63 6f 6e 6e 29 _one(&mut *conn)
0f00: 2e 61 77 61 69 74 3f 3b 0a 09 09 09 6c 65 74 20 .await?;....let
0f10: 64 65 73 74 69 6e 61 74 69 6f 6e 20 3d 20 6d 61 destination = ma
0f20: 74 63 68 20 72 65 61 6c 20 7b 0a 09 09 09 09 74 tch real {.....t
0f30: 72 75 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f rue => telegram_
0f40: 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 bot::UserId::new
0f50: 28 73 6f 75 72 63 65 2e 63 68 61 6e 6e 65 6c 5f (source.channel_
0f60: 69 64 29 2c 0a 09 09 09 09 66 61 6c 73 65 20 3d id),.....false =
0f70: 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a > telegram_bot::
0f80: 55 73 65 72 49 64 3a 3a 6e 65 77 28 73 6f 75 72 UserId::new(sour
0f90: 63 65 2e 6f 77 6e 65 72 29 2c 0a 09 09 09 7d 3b ce.owner),....};
0fa0: 0a 09 09 09 6c 65 74 20 6d 75 74 20 74 68 69 73 ....let mut this
0fb0: 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e 3c 44 _fetch: Option<D
0fc0: 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a ateTime<chrono::
0fd0: 46 69 78 65 64 4f 66 66 73 65 74 3e 3e 20 3d 20 FixedOffset>> =
0fe0: 4e 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 6d 75 74 None;....let mut
0ff0: 20 70 6f 73 74 73 3a 20 42 54 72 65 65 4d 61 70 posts: BTreeMap
1000: 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f <DateTime<chrono
1010: 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 2c 20 ::FixedOffset>,
1020: 53 74 72 69 6e 67 3e 20 3d 20 42 54 72 65 65 4d String> = BTreeM
1030: 61 70 3a 3a 6e 65 77 28 29 3b 0a 0a 09 09 09 6c ap::new();.....l
1040: 65 74 20 72 65 73 70 6f 6e 73 65 20 3d 20 73 65 et response = se
1050: 6c 66 2e 68 74 74 70 5f 63 6c 69 65 6e 74 2e 67 lf.http_client.g
1060: 65 74 28 26 73 6f 75 72 63 65 2e 75 72 6c 29 2e et(&source.url).
1070: 73 65 6e 64 28 29 2e 61 77 61 69 74 3f 3b 0a 09 send().await?;..
1080: 09 09 6c 65 74 20 73 74 61 74 75 73 20 3d 20 72 ..let status = r
1090: 65 73 70 6f 6e 73 65 2e 73 74 61 74 75 73 28 29 esponse.status()
10a0: 3b 0a 09 09 09 6c 65 74 20 63 6f 6e 74 65 6e 74 ;....let content
10b0: 20 3d 20 72 65 73 70 6f 6e 73 65 2e 62 79 74 65 = response.byte
10c0: 73 28 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 6d s().await?;....m
10d0: 61 74 63 68 20 72 73 73 3a 3a 43 68 61 6e 6e 65 atch rss::Channe
10e0: 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f l::read_from(&co
10f0: 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 09 09 ntent[..]) {....
1100: 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 7b 0a 09 .Ok(feed) => {..
1110: 09 09 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 ....for item in
1120: 66 65 65 64 2e 69 74 65 6d 73 28 29 20 7b 0a 09 feed.items() {..
1130: 09 09 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 .....if let Some
1140: 28 6c 69 6e 6b 29 20 3d 20 69 74 65 6d 2e 6c 69 (link) = item.li
1150: 6e 6b 28 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 nk() {........le
1160: 74 20 64 61 74 65 20 3d 20 6d 61 74 63 68 20 69 t date = match i
1170: 74 65 6d 2e 70 75 62 5f 64 61 74 65 28 29 20 7b tem.pub_date() {
1180: 0a 09 09 09 09 09 09 09 09 53 6f 6d 65 28 66 65 .........Some(fe
1190: 65 64 5f 64 61 74 65 29 20 3d 3e 20 44 61 74 65 ed_date) => Date
11a0: 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d Time::parse_from
11b0: 5f 72 66 63 32 38 32 32 28 66 65 65 64 5f 64 61 _rfc2822(feed_da
11c0: 74 65 29 2c 0a 09 09 09 09 09 09 09 09 4e 6f 6e te),.........Non
11d0: 65 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 e => DateTime::p
11e0: 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 33 33 33 arse_from_rfc333
11f0: 39 28 26 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 63 9(&item.dublin_c
1200: 6f 72 65 5f 65 78 74 28 29 2e 75 6e 77 72 61 70 ore_ext().unwrap
1210: 28 29 2e 64 61 74 65 73 28 29 5b 30 5d 29 2c 0a ().dates()[0]),.
1220: 09 09 09 09 09 09 09 7d 3f 3b 0a 09 09 09 09 09 .......}?;......
1230: 09 09 6c 65 74 20 75 72 6c 20 3d 20 6c 69 6e 6b ..let url = link
1240: 3b 0a 09 09 09 09 09 09 09 70 6f 73 74 73 2e 69 ;........posts.i
1250: 6e 73 65 72 74 28 64 61 74 65 2c 20 75 72 6c 2e nsert(date, url.
1260: 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 to_string());...
1270: 09 09 09 09 7d 0a 09 09 09 09 09 7d 3b 0a 09 09 ....}......};...
1280: 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 65 72 72 ..},.....Err(err
1290: 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 72 20 7b ) => match err {
12a0: 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 6f 72 ......rss::Error
12b0: 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 74 54 61 ::InvalidStartTa
12c0: 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 6d 61 74 g => {.......mat
12d0: 63 68 20 61 74 6f 6d 5f 73 79 6e 64 69 63 61 74 ch atom_syndicat
12e0: 69 6f 6e 3a 3a 46 65 65 64 3a 3a 72 65 61 64 5f ion::Feed::read_
12f0: 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e from(&content[..
1300: 5d 29 20 7b 0a 09 09 09 09 09 09 09 4f 6b 28 66 ]) {........Ok(f
1310: 65 65 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 eed) => {.......
1320: 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 65 ..for item in fe
1330: 65 64 2e 65 6e 74 72 69 65 73 28 29 20 7b 0a 09 ed.entries() {..
1340: 09 09 09 09 09 09 09 09 6c 65 74 20 64 61 74 65 ........let date
1350: 20 3d 20 69 74 65 6d 2e 70 75 62 6c 69 73 68 65 = item.publishe
1360: 64 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 d().unwrap();...
1370: 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c 20 3d .......let url =
1380: 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 29 5b 30 5d item.links()[0]
1390: 2e 68 72 65 66 28 29 3b 0a 09 09 09 09 09 09 09 .href();........
13a0: 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 2a ..posts.insert(*
13b0: 64 61 74 65 2c 20 75 72 6c 2e 74 6f 5f 73 74 72 date, url.to_str
13c0: 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 ing());.........
13d0: 7d 3b 0a 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 };........},....
13e0: 09 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 ....Err(err) =>
13f0: 7b 0a 09 09 09 09 09 09 09 09 62 61 69 6c 21 28 {.........bail!(
1400: 22 55 6e 73 75 70 70 6f 72 74 65 64 20 6f 72 20 "Unsupported or
1410: 6d 61 6e 67 6c 65 64 20 63 6f 6e 74 65 6e 74 3a mangled content:
1420: 5c 6e 7b 3a 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e 7b \n{:?}\n{:#?}\n{
1430: 3a 23 3f 7d 5c 6e 22 2c 20 26 73 6f 75 72 63 65 :#?}\n", &source
1440: 2e 75 72 6c 2c 20 65 72 72 2c 20 73 74 61 74 75 .url, err, statu
1450: 73 29 0a 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 s)........},....
1460: 09 09 09 7d 0a 09 09 09 09 09 7d 2c 0a 09 09 09 ...}......},....
1470: 09 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 6f ..rss::Error::Eo
1480: 66 20 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f 20 f => (),......_
1490: 3d 3e 20 62 61 69 6c 21 28 22 55 6e 73 75 70 70 => bail!("Unsupp
14a0: 6f 72 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 64 orted or mangled
14b0: 20 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c content:\n{:?}\
14c0: 6e 7b 3a 23 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e 22 n{:#?}\n{:#?}\n"
14d0: 2c 20 26 73 6f 75 72 63 65 2e 75 72 6c 2c 20 65 , &source.url, e
14e0: 72 72 2c 20 73 74 61 74 75 73 29 0a 09 09 09 09 rr, status).....
14f0: 7d 0a 09 09 09 7d 3b 0a 09 09 09 66 6f 72 20 28 }....};....for (
1500: 64 61 74 65 2c 20 75 72 6c 29 20 69 6e 20 70 6f date, url) in po
1510: 73 74 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 sts.iter() {....
1520: 09 6c 65 74 20 70 6f 73 74 5f 75 72 6c 3a 20 43 .let post_url: C
1530: 6f 77 3c 73 74 72 3e 20 3d 20 6d 61 74 63 68 20 ow<str> = match
1540: 73 6f 75 72 63 65 2e 75 72 6c 5f 72 65 20 7b 0a source.url_re {.
1550: 09 09 09 09 09 53 6f 6d 65 28 72 65 66 20 78 29 .....Some(ref x)
1560: 20 3d 3e 20 73 65 64 72 65 67 65 78 3a 3a 52 65 => sedregex::Re
1570: 70 6c 61 63 65 43 6f 6d 6d 61 6e 64 3a 3a 6e 65 placeCommand::ne
1580: 77 28 78 29 3f 2e 65 78 65 63 75 74 65 28 75 72 w(x)?.execute(ur
1590: 6c 29 2c 0a 09 09 09 09 09 4e 6f 6e 65 20 3d 3e l),......None =>
15a0: 20 75 72 6c 2e 69 6e 74 6f 28 29 2c 0a 09 09 09 url.into(),....
15b0: 09 7d 3b 0a 09 09 09 09 69 66 20 6c 65 74 20 53 .};.....if let S
15c0: 6f 6d 65 28 65 78 69 73 74 73 29 20 3d 20 73 71 ome(exists) = sq
15d0: 6c 78 3a 3a 71 75 65 72 79 21 28 22 73 65 6c 65 lx::query!("sele
15e0: 63 74 20 65 78 69 73 74 73 28 73 65 6c 65 63 74 ct exists(select
15f0: 20 74 72 75 65 20 66 72 6f 6d 20 72 73 73 74 67 true from rsstg
1600: 5f 70 6f 73 74 20 77 68 65 72 65 20 75 72 6c 20 _post where url
1610: 3d 20 24 31 20 61 6e 64 20 73 6f 75 72 63 65 5f = $1 and source_
1620: 69 64 20 3d 20 24 32 29 20 61 73 20 65 78 69 73 id = $2) as exis
1630: 74 73 3b 22 2c 0a 09 09 09 09 09 26 70 6f 73 74 ts;",......&post
1640: 5f 75 72 6c 2c 20 2a 69 64 29 2e 66 65 74 63 68 _url, *id).fetch
1650: 5f 6f 6e 65 28 26 6d 75 74 20 2a 63 6f 6e 6e 29 _one(&mut *conn)
1660: 2e 61 77 61 69 74 3f 2e 65 78 69 73 74 73 20 7b .await?.exists {
1670: 0a 09 09 09 09 09 69 66 20 21 20 65 78 69 73 74 ......if ! exist
1680: 73 20 7b 0a 09 09 09 09 09 09 69 66 20 74 68 69 s {.......if thi
1690: 73 5f 66 65 74 63 68 2e 69 73 5f 6e 6f 6e 65 28 s_fetch.is_none(
16a0: 29 20 7c 7c 20 2a 64 61 74 65 20 3e 20 74 68 69 ) || *date > thi
16b0: 73 5f 66 65 74 63 68 2e 75 6e 77 72 61 70 28 29 s_fetch.unwrap()
16c0: 20 7b 0a 09 09 09 09 09 09 09 74 68 69 73 5f 66 {........this_f
16d0: 65 74 63 68 20 3d 20 53 6f 6d 65 28 2a 64 61 74 etch = Some(*dat
16e0: 65 29 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 e);.......};....
16f0: 09 09 09 73 65 6c 66 2e 72 65 71 75 65 73 74 28 ...self.request(
1700: 20 6d 61 74 63 68 20 26 73 6f 75 72 63 65 2e 69 match &source.i
1710: 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 09 v_hash {........
1720: 09 53 6f 6d 65 28 68 61 73 68 29 20 3d 3e 20 74 .Some(hash) => t
1730: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e elegram_bot::Sen
1740: 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 dMessage::new(de
1750: 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 stination, forma
1760: 74 21 28 22 3c 61 20 68 72 65 66 3d 5c 22 68 74 t!("<a href=\"ht
1770: 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 72 tps://t.me/iv?ur
1780: 6c 3d 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c 22 3e l={}&rhash={}\">
1790: 20 3c 2f 61 3e 7b 30 7d 22 2c 20 26 70 6f 73 74 </a>{0}", &post
17a0: 5f 75 72 6c 2c 20 68 61 73 68 29 29 2c 0a 09 09 _url, hash)),...
17b0: 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 65 ......None => te
17c0: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 legram_bot::Send
17d0: 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 73 Message::new(des
17e0: 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 tination, format
17f0: 21 28 22 7b 7d 22 2c 20 70 6f 73 74 5f 75 72 6c !("{}", post_url
1800: 29 29 2c 0a 09 09 09 09 09 09 09 7d 2e 70 61 72 )),........}.par
1810: 73 65 5f 6d 6f 64 65 28 74 65 6c 65 67 72 61 6d se_mode(telegram
1820: 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 61 72 _bot::types::Par
1830: 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 2e 61 seMode::Html)).a
1840: 77 61 69 74 0a 09 09 09 09 09 09 09 2e 63 6f 6e wait.........con
1850: 74 65 78 74 28 22 43 61 6e 27 74 20 70 6f 73 74 text("Can't post
1860: 20 6d 65 73 73 61 67 65 3a 22 29 3f 3b 0a 09 09 message:")?;...
1870: 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 21 ....sqlx::query!
1880: 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 ("insert into rs
1890: 73 74 67 5f 70 6f 73 74 20 28 73 6f 75 72 63 65 stg_post (source
18a0: 5f 69 64 2c 20 70 6f 73 74 65 64 2c 20 75 72 6c _id, posted, url
18b0: 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 ) values ($1, $2
18c0: 2c 20 24 33 29 3b 22 2c 0a 09 09 09 09 09 09 09 , $3);",........
18d0: 2a 69 64 2c 20 64 61 74 65 2c 20 26 70 6f 73 74 *id, date, &post
18e0: 5f 75 72 6c 29 2e 65 78 65 63 75 74 65 28 26 6d _url).execute(&m
18f0: 75 74 20 2a 63 6f 6e 6e 29 2e 61 77 61 69 74 3f ut *conn).await?
1900: 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 7d 3b ;......};.....};
1910: 0a 09 09 09 09 70 6f 73 74 65 64 20 2b 3d 20 31 .....posted += 1
1920: 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 6f 73 74 73 ;....};....posts
1930: 2e 63 6c 65 61 72 28 29 3b 0a 09 09 7d 3b 0a 09 .clear();...};..
1940: 09 73 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 75 .sqlx::query!("u
1950: 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 pdate rsstg_sour
1960: 63 65 20 73 65 74 20 6c 61 73 74 5f 73 63 72 61 ce set last_scra
1970: 70 65 20 3d 20 6e 6f 77 28 29 20 77 68 65 72 65 pe = now() where
1980: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 3b source_id = $1;
1990: 22 2c 0a 09 09 09 2a 69 64 29 2e 65 78 65 63 75 ",....*id).execu
19a0: 74 65 28 26 6d 75 74 20 2a 63 6f 6e 6e 29 2e 61 te(&mut *conn).a
19b0: 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 66 6f 72 6d wait?;...Ok(form
19c0: 61 74 21 28 22 50 6f 73 74 65 64 3a 20 7b 7d 22 at!("Posted: {}"
19d0: 2c 20 26 70 6f 73 74 65 64 29 2e 69 6e 74 6f 28 , &posted).into(
19e0: 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e ))..}...pub asyn
19f0: 63 20 66 6e 20 64 65 6c 65 74 65 3c 53 3e 28 26 c fn delete<S>(&
1a00: 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a self, source_id:
1a10: 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 &i32, owner: S)
1a20: 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 3c 27 -> Result<Cow<'
1a30: 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 65 20 _, str>>..where
1a40: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
1a50: 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e .let owner = own
1a60: 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6d 61 er.into();....ma
1a70: 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 21 tch sqlx::query!
1a80: 28 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 ("delete from rs
1a90: 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 stg_source where
1aa0: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 source_id = $1
1ab0: 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 3b 22 and owner = $2;"
1ac0: 2c 0a 09 09 09 73 6f 75 72 63 65 5f 69 64 2c 20 ,....source_id,
1ad0: 6f 77 6e 65 72 29 2e 65 78 65 63 75 74 65 28 26 owner).execute(&
1ae0: 6d 75 74 20 2a 73 65 6c 66 2e 70 6f 6f 6c 2e 61 mut *self.pool.a
1af0: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 3f 29 cquire().await?)
1b00: 2e 61 77 61 69 74 3f 2e 72 6f 77 73 5f 61 66 66 .await?.rows_aff
1b10: 65 63 74 65 64 28 29 20 7b 0a 09 09 09 30 20 3d ected() {....0 =
1b20: 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 61 74 61 20 > { Ok("No data
1b30: 66 6f 75 6e 64 20 66 6f 75 6e 64 2e 22 2e 69 6e found found.".in
1b40: 74 6f 28 29 29 20 7d 2c 0a 09 09 09 78 20 3d 3e to()) },....x =>
1b50: 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 22 7b { Ok(format!("{
1b60: 7d 20 73 6f 75 72 63 65 73 20 72 65 6d 6f 76 65 } sources remove
1b70: 64 2e 22 2c 20 78 29 2e 69 6e 74 6f 28 29 29 20 d.", x).into())
1b80: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 },...}..}...pub
1b90: 61 73 79 6e 63 20 66 6e 20 63 6c 65 61 6e 3c 53 async fn clean<S
1ba0: 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f >(&self, source_
1bb0: 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a id: &i32, owner:
1bc0: 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f S) -> Result<Co
1bd0: 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 w<'_, str>>..whe
1be0: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
1bf0: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
1c00: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 owner.into();...
1c10: 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 .match sqlx::que
1c20: 72 79 21 28 22 64 65 6c 65 74 65 20 66 72 6f 6d ry!("delete from
1c30: 20 72 73 73 74 67 5f 70 6f 73 74 20 70 20 75 73 rsstg_post p us
1c40: 69 6e 67 20 72 73 73 74 67 5f 73 6f 75 72 63 65 ing rsstg_source
1c50: 20 73 20 77 68 65 72 65 20 70 2e 73 6f 75 72 63 s where p.sourc
1c60: 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 e_id = $1 and ow
1c70: 6e 65 72 20 3d 20 24 32 20 61 6e 64 20 70 2e 73 ner = $2 and p.s
1c80: 6f 75 72 63 65 5f 69 64 20 3d 20 73 2e 73 6f 75 ource_id = s.sou
1c90: 72 63 65 5f 69 64 3b 22 2c 0a 09 09 09 73 6f 75 rce_id;",....sou
1ca0: 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 65 rce_id, owner).e
1cb0: 78 65 63 75 74 65 28 26 6d 75 74 20 2a 73 65 6c xecute(&mut *sel
1cc0: 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 f.pool.acquire()
1cd0: 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 74 3f 2e .await?).await?.
1ce0: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 rows_affected()
1cf0: 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 {....0 => { Ok("
1d00: 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f No data found fo
1d10: 75 6e 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c und.".into()) },
1d20: 0a 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f ....x => { Ok(fo
1d30: 72 6d 61 74 21 28 22 7b 7d 20 70 6f 73 74 73 20 rmat!("{} posts
1d40: 70 75 72 67 65 64 2e 22 2c 20 78 29 2e 69 6e 74 purged.", x).int
1d50: 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a o()) },...}..}..
1d60: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 65 6e .pub async fn en
1d70: 61 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 able<S>(&self, s
1d80: 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 ource_id: &i32,
1d90: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
1da0: 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 ult<&str>..where
1db0: 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a S: Into<i64> {.
1dc0: 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 ..let owner = ow
1dd0: 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6d ner.into();....m
1de0: 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 atch sqlx::query
1df0: 21 28 22 75 70 64 61 74 65 20 72 73 73 74 67 5f !("update rsstg_
1e00: 73 6f 75 72 63 65 20 73 65 74 20 65 6e 61 62 6c source set enabl
1e10: 65 64 20 3d 20 74 72 75 65 20 77 68 65 72 65 20 ed = true where
1e20: 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 source_id = $1 a
1e30: 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 2c 0a nd owner = $2",.
1e40: 09 09 09 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 ...source_id, ow
1e50: 6e 65 72 29 2e 65 78 65 63 75 74 65 28 26 6d 75 ner).execute(&mu
1e60: 74 20 2a 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 t *self.pool.acq
1e70: 75 69 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 uire().await?).a
1e80: 77 61 69 74 3f 2e 72 6f 77 73 5f 61 66 66 65 63 wait?.rows_affec
1e90: 74 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 ted() {....1 =>
1ea0: 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 65 6e 61 { Ok("Source ena
1eb0: 62 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 bled.") },....0
1ec0: 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 => { Ok("Source
1ed0: 6e 6f 74 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a not found.") },.
1ee0: 09 09 09 5f 20 3d 3e 20 7b 20 45 72 72 28 61 6e ..._ => { Err(an
1ef0: 79 68 6f 77 21 28 22 44 61 74 61 62 61 73 65 20 yhow!("Database
1f00: 65 72 72 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 7d error.")) },...}
1f10: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
1f20: 66 6e 20 64 69 73 61 62 6c 65 3c 53 3e 28 26 73 fn disable<S>(&s
1f30: 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 elf, source_id:
1f40: 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 &i32, owner: S)
1f50: 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a -> Result<&str>.
1f60: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
1f70: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
1f80: 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 r = owner.into()
1f90: 3b 0a 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a ;....match sqlx:
1fa0: 3a 71 75 65 72 79 21 28 22 75 70 64 61 74 65 20 :query!("update
1fb0: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 rsstg_source set
1fc0: 20 65 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 65 enabled = false
1fd0: 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 where source_id
1fe0: 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 = $1 and owner
1ff0: 3d 20 24 32 22 2c 0a 09 09 09 73 6f 75 72 63 65 = $2",....source
2000: 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 65 78 65 63 _id, owner).exec
2010: 75 74 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e 70 ute(&mut *self.p
2020: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
2030: 61 69 74 3f 29 2e 61 77 61 69 74 3f 2e 72 6f 77 ait?).await?.row
2040: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
2050: 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 ..1 => { Ok("Sou
2060: 72 63 65 20 64 69 73 61 62 6c 65 64 2e 22 29 20 rce disabled.")
2070: 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 },....0 => { Ok(
2080: 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e "Source not foun
2090: 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 d.") },...._ =>
20a0: 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 { Err(anyhow!("D
20b0: 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 atabase error.")
20c0: 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 ) },...}..}...pu
20d0: 62 20 61 73 79 6e 63 20 66 6e 20 75 70 64 61 74 b async fn updat
20e0: 65 3c 53 3e 28 26 73 65 6c 66 2c 20 75 70 64 61 e<S>(&self, upda
20f0: 74 65 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c te: Option<i32>,
2100: 20 63 68 61 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 channel: &str,
2110: 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c channel_id: i64,
2120: 20 75 72 6c 3a 20 26 73 74 72 2c 20 69 76 5f 68 url: &str, iv_h
2130: 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 ash: Option<&str
2140: 3e 2c 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f >, url_re: Optio
2150: 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 72 3a 20 n<&str>, owner:
2160: 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 S) -> Result<&st
2170: 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 r>..where S: Int
2180: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f o<i64> {...let o
2190: 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 wner = owner.int
21a0: 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 o();...let mut c
21b0: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e onn = self.pool.
21c0: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 3f acquire().await?
21d0: 3b 0a 0a 09 09 6d 61 74 63 68 20 6d 61 74 63 68 ;....match match
21e0: 20 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f update {.....So
21f0: 6d 65 28 69 64 29 20 3d 3e 20 7b 0a 09 09 09 09 me(id) => {.....
2200: 09 73 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 75 .sqlx::query!("u
2210: 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 pdate rsstg_sour
2220: 63 65 20 73 65 74 20 63 68 61 6e 6e 65 6c 5f 69 ce set channel_i
2230: 64 20 3d 20 24 32 2c 20 75 72 6c 20 3d 20 24 33 d = $2, url = $3
2240: 2c 20 69 76 5f 68 61 73 68 20 3d 20 24 34 2c 20 , iv_hash = $4,
2250: 6f 77 6e 65 72 20 3d 20 24 35 2c 20 63 68 61 6e owner = $5, chan
2260: 6e 65 6c 20 3d 20 24 36 2c 20 75 72 6c 5f 72 65 nel = $6, url_re
2270: 20 3d 20 24 37 20 77 68 65 72 65 20 73 6f 75 72 = $7 where sour
2280: 63 65 5f 69 64 20 3d 20 24 31 22 2c 0a 09 09 09 ce_id = $1",....
2290: 09 09 09 69 64 2c 20 63 68 61 6e 6e 65 6c 5f 69 ...id, channel_i
22a0: 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c d, url, iv_hash,
22b0: 20 6f 77 6e 65 72 2c 20 63 68 61 6e 6e 65 6c 2c owner, channel,
22c0: 20 75 72 6c 5f 72 65 29 2e 65 78 65 63 75 74 65 url_re).execute
22d0: 28 26 6d 75 74 20 2a 63 6f 6e 6e 29 2e 61 77 61 (&mut *conn).awa
22e0: 69 74 0a 09 09 09 09 7d 2c 0a 09 09 09 09 4e 6f it.....},.....No
22f0: 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c ne => {......sql
2300: 78 3a 3a 71 75 65 72 79 21 28 22 69 6e 73 65 72 x::query!("inser
2310: 74 20 69 6e 74 6f 20 72 73 73 74 67 5f 73 6f 75 t into rsstg_sou
2320: 72 63 65 20 28 63 68 61 6e 6e 65 6c 5f 69 64 2c rce (channel_id,
2330: 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f url, iv_hash, o
2340: 77 6e 65 72 2c 20 63 68 61 6e 6e 65 6c 2c 20 75 wner, channel, u
2350: 72 6c 5f 72 65 29 20 76 61 6c 75 65 73 20 28 24 rl_re) values ($
2360: 31 2c 20 24 32 2c 20 24 33 2c 20 24 34 2c 20 24 1, $2, $3, $4, $
2370: 35 2c 20 24 36 29 22 2c 0a 09 09 09 09 09 09 63 5, $6)",.......c
2380: 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 hannel_id, url,
2390: 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 iv_hash, owner,
23a0: 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 channel, url_re)
23b0: 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 2a 63 .execute(&mut *c
23c0: 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 7d onn).await.....}
23d0: 2c 0a 09 09 09 7d 20 7b 0a 09 09 09 4f 6b 28 5f ,....} {....Ok(_
23e0: 29 20 3d 3e 20 4f 6b 28 6d 61 74 63 68 20 75 70 ) => Ok(match up
23f0: 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 date {.....Some(
2400: 5f 29 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 75 _) => "Channel u
2410: 70 64 61 74 65 64 2e 22 2c 0a 09 09 09 09 4e 6f pdated.",.....No
2420: 6e 65 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 61 ne => "Channel a
2430: 64 64 65 64 2e 22 2c 0a 09 09 09 7d 29 2c 0a 09 dded.",....}),..
2440: 09 09 45 72 72 28 73 71 6c 78 3a 3a 45 72 72 6f ..Err(sqlx::Erro
2450: 72 3a 3a 44 61 74 61 62 61 73 65 28 65 72 72 29 r::Database(err)
2460: 29 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 74 63 68 ) => {.....match
2470: 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a 3a 3c err.downcast::<
2480: 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a 3a sqlx::postgres::
2490: 50 67 44 61 74 61 62 61 73 65 45 72 72 6f 72 3e PgDatabaseError>
24a0: 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b 0a 09 ().routine() {..
24b0: 09 09 09 09 53 6f 6d 65 28 22 5f 62 74 5f 63 68 ....Some("_bt_ch
24c0: 65 63 6b 5f 75 6e 69 71 75 65 22 2c 20 29 20 3d eck_unique", ) =
24d0: 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 44 75 > {.......Ok("Du
24e0: 70 6c 69 63 61 74 65 20 6b 65 79 2e 22 29 0a 09 plicate key.")..
24f0: 09 09 09 09 7d 2c 0a 09 09 09 09 09 53 6f 6d 65 ....},......Some
2500: 28 5f 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f (_) => {.......O
2510: 6b 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f k("Database erro
2520: 72 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 r.")......},....
2530: 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 ..None => {.....
2540: 09 09 4f 6b 28 22 4e 6f 20 64 61 74 61 62 61 73 ..Ok("No databas
2550: 65 20 65 72 72 6f 72 20 65 78 74 72 61 63 74 65 e error extracte
2560: 64 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 d.")......},....
2570: 09 7d 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 .}....},....Err(
2580: 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 62 61 err) => {.....ba
2590: 69 6c 21 28 22 53 6f 72 72 79 2c 20 75 6e 6b 6e il!("Sorry, unkn
25a0: 6f 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 3f own error:\n{:#?
25b0: 7d 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 09 09 7d }\n", err);....}
25c0: 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e 63 ,...}..}...async
25d0: 20 66 6e 20 61 75 74 6f 66 65 74 63 68 28 26 73 fn autofetch(&s
25e0: 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 73 elf) -> Result<s
25f0: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
2600: 6f 6e 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 on> {...let mut
2610: 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a delay = chrono::
2620: 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 Duration::minute
2630: 73 28 31 29 3b 0a 09 09 6c 65 74 20 6e 6f 77 20 s(1);...let now
2640: 3d 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a = chrono::Local:
2650: 3a 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 20 6d 75 :now();...let mu
2660: 74 20 71 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a t queue = sqlx::
2670: 71 75 65 72 79 21 28 72 23 22 73 65 6c 65 63 74 query!(r#"select
2680: 20 73 6f 75 72 63 65 5f 69 64 2c 20 6e 65 78 74 source_id, next
2690: 5f 66 65 74 63 68 20 61 73 20 22 6e 65 78 74 5f _fetch as "next_
26a0: 66 65 74 63 68 3a 20 44 61 74 65 54 69 6d 65 3c fetch: DateTime<
26b0: 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3e 22 2c chrono::Local>",
26c0: 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 73 74 owner from rsst
26d0: 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 6c 20 g_order natural
26e0: 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 67 5f left join rsstg_
26f0: 73 6f 75 72 63 65 20 77 68 65 72 65 20 6e 65 78 source where nex
2700: 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 29 20 t_fetch < now()
2710: 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 20 6d 69 + interval '1 mi
2720: 6e 75 74 65 27 3b 22 23 29 0a 09 09 09 2e 66 65 nute';"#).....fe
2730: 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 2a 73 65 tch_all(&mut *se
2740: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
2750: 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 74 3f ).await?).await?
2760: 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 71 ;...for row in q
2770: 75 65 75 65 2e 69 74 65 72 28 29 20 7b 0a 09 09 ueue.iter() {...
2780: 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 6e 65 78 .if let Some(nex
2790: 74 5f 66 65 74 63 68 29 20 3d 20 72 6f 77 2e 6e t_fetch) = row.n
27a0: 65 78 74 5f 66 65 74 63 68 20 7b 0a 09 09 09 09 ext_fetch {.....
27b0: 69 66 20 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 if next_fetch <
27c0: 6e 6f 77 20 7b 0a 09 09 09 09 09 69 66 20 6c 65 now {......if le
27d0: 74 20 28 53 6f 6d 65 28 6f 77 6e 65 72 29 2c 20 t (Some(owner),
27e0: 53 6f 6d 65 28 73 6f 75 72 63 65 5f 69 64 29 29 Some(source_id))
27f0: 20 3d 20 28 72 6f 77 2e 6f 77 6e 65 72 2c 20 72 = (row.owner, r
2800: 6f 77 2e 73 6f 75 72 63 65 5f 69 64 29 20 7b 0a ow.source_id) {.
2810: 09 09 09 09 09 09 6c 65 74 20 63 6c 6f 6e 65 20 ......let clone
2820: 3d 20 43 6f 72 65 20 7b 0a 09 09 09 09 09 09 09 = Core {........
2830: 6f 77 6e 65 72 5f 63 68 61 74 3a 20 74 65 6c 65 owner_chat: tele
2840: 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 gram_bot::UserId
2850: 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 2c 0a 09 09 ::new(owner),...
2860: 09 09 09 09 09 2e 2e 73 65 6c 66 2e 63 6c 6f 6e .......self.clon
2870: 65 28 29 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 e().......};....
2880: 09 09 09 74 61 73 6b 3a 3a 73 70 61 77 6e 28 61 ...task::spawn(a
2890: 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 sync move {.....
28a0: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
28b0: 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b r) = clone.check
28c0: 28 26 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e (&source_id, own
28d0: 65 72 2c 20 74 72 75 65 29 2e 61 77 61 69 74 20 er, true).await
28e0: 7b 0a 09 09 09 09 09 09 09 09 69 66 20 6c 65 74 {.........if let
28f0: 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e Err(err) = clon
2900: 65 2e 73 65 6e 64 28 26 66 6f 72 6d 61 74 21 28 e.send(&format!(
2910: 22 f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 "š {:?}", err
2920: 29 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 ), None, None).a
2930: 77 61 69 74 20 7b 0a 09 09 09 09 09 09 09 09 09 wait {..........
2940: 64 62 67 21 28 22 43 68 65 63 6b 20 65 72 72 6f dbg!("Check erro
2950: 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 r: {}", err);...
2960: 09 09 09 09 09 09 09 2f 2f 20 63 6c 6f 6e 65 2e .......// clone.
2970: 64 69 73 61 62 6c 65 28 26 73 6f 75 72 63 65 5f disable(&source_
2980: 69 64 2c 20 6f 77 6e 65 72 29 2e 61 77 61 69 74 id, owner).await
2990: 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 09 .unwrap();......
29a0: 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 7d 3b 0a ...};........};.
29b0: 09 09 09 09 09 09 7d 29 3b 0a 09 09 09 09 09 7d ......});......}
29c0: 0a 09 09 09 09 7d 20 65 6c 73 65 20 69 66 20 6e .....} else if n
29d0: 65 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 20 ext_fetch - now
29e0: 3c 20 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 64 < delay {......d
29f0: 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 63 elay = next_fetc
2a00: 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 7d 0a 09 h - now;.....}..
2a10: 09 09 7d 0a 09 09 7d 3b 0a 09 09 71 75 65 75 65 ..}...};...queue
2a20: 2e 63 6c 65 61 72 28 29 3b 0a 09 09 4f 6b 28 64 .clear();...Ok(d
2a30: 65 6c 61 79 2e 74 6f 5f 73 74 64 28 29 3f 29 0a elay.to_std()?).
2a40: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
2a50: 6e 20 6c 69 73 74 3c 53 3e 28 26 73 65 6c 66 2c n list<S>(&self,
2a60: 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 owner: S) -> Re
2a70: 73 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 sult<String>..wh
2a80: 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e ere S: Into<i64>
2a90: 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d {...let owner =
2aa0: 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a owner.into();..
2ab0: 09 09 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 3a ..let mut reply:
2ac0: 20 56 65 63 3c 43 6f 77 3c 73 74 72 3e 3e 20 3d Vec<Cow<str>> =
2ad0: 20 76 65 63 21 5b 5d 3b 0a 09 09 72 65 70 6c 79 vec![];...reply
2ae0: 2e 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a .push("Channels:
2af0: 22 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 6c 65 74 ".into());...let
2b00: 20 72 6f 77 73 20 3d 20 73 71 6c 78 3a 3a 71 75 rows = sqlx::qu
2b10: 65 72 79 21 28 22 73 65 6c 65 63 74 20 73 6f 75 ery!("select sou
2b20: 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c rce_id, channel,
2b30: 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 enabled, url, i
2b40: 76 5f 68 61 73 68 2c 20 75 72 6c 5f 72 65 20 66 v_hash, url_re f
2b50: 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 rom rsstg_source
2b60: 20 77 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 where owner = $
2b70: 31 20 6f 72 64 65 72 20 62 79 20 73 6f 75 72 63 1 order by sourc
2b80: 65 5f 69 64 22 2c 0a 09 09 09 6f 77 6e 65 72 29 e_id",....owner)
2b90: 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 .fetch_all(&mut
2ba0: 2a 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 *self.pool.acqui
2bb0: 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 re().await?).awa
2bc0: 69 74 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 it?;...for row i
2bd0: 6e 20 72 6f 77 73 2e 69 74 65 72 28 29 20 7b 0a n rows.iter() {.
2be0: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f ...reply.push(fo
2bf0: 72 6d 61 74 21 28 22 5c 6e 5c 5c 23 ef b8 8f e2 rmat!("\n\\#ļøā
2c00: 83 a3 20 7b 7d 20 5c 5c 2a ef b8 8f e2 83 a3 20 £ {} \\*ļøā£
2c10: 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 97 20 60 7b `{}` {}\nš `{
2c20: 7d 60 22 2c 20 72 6f 77 2e 73 6f 75 72 63 65 5f }`", row.source_
2c30: 69 64 2c 20 72 6f 77 2e 63 68 61 6e 6e 65 6c 2c id, row.channel,
2c40: 0a 09 09 09 09 6d 61 74 63 68 20 72 6f 77 2e 65 .....match row.e
2c50: 6e 61 62 6c 65 64 20 7b 0a 09 09 09 09 09 74 72 nabled {......tr
2c60: 75 65 20 20 3d 3e 20 22 f0 9f 94 84 20 65 6e 61 ue => "š ena
2c70: 62 6c 65 64 22 2c 0a 09 09 09 09 09 66 61 6c 73 bled",......fals
2c80: 65 20 3d 3e 20 22 e2 9b 94 20 64 69 73 61 62 6c e => "ā disabl
2c90: 65 64 22 2c 0a 09 09 09 09 7d 2c 20 72 6f 77 2e ed",.....}, row.
2ca0: 75 72 6c 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 url).into());...
2cb0: 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 .if let Some(has
2cc0: 68 29 20 3d 20 26 72 6f 77 2e 69 76 5f 68 61 73 h) = &row.iv_has
2cd0: 68 20 7b 0a 09 09 09 09 72 65 70 6c 79 2e 70 75 h {.....reply.pu
2ce0: 73 68 28 66 6f 72 6d 61 74 21 28 22 49 56 3a 20 sh(format!("IV:
2cf0: 60 7b 7d 60 22 2c 20 68 61 73 68 29 2e 69 6e 74 `{}`", hash).int
2d00: 6f 28 29 29 3b 0a 09 09 09 7d 0a 09 09 09 69 66 o());....}....if
2d10: 20 6c 65 74 20 53 6f 6d 65 28 72 65 29 20 3d 20 let Some(re) =
2d20: 26 72 6f 77 2e 75 72 6c 5f 72 65 20 7b 0a 09 09 &row.url_re {...
2d30: 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 ..reply.push(for
2d40: 6d 61 74 21 28 22 52 45 3a 20 60 7b 7d 60 22 2c mat!("RE: `{}`",
2d50: 20 72 65 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 re).into());...
2d60: 09 7d 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 65 70 .}...};...Ok(rep
2d70: 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 0a 09 ly.join("\n"))..
2d80: 7d 0a 7d 0a }.}.