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 0a 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 ;....let mut pos
0d00: 74 65 64 3a 20 69 33 32 20 3d 20 30 3b 0a 09 09 ted: i32 = 0;...
0d10: 6c 65 74 20 69 64 20 3d 20 7b 0a 09 09 09 6c 65 let id = {....le
0d20: 74 20 6d 75 74 20 73 65 74 20 3d 20 73 65 6c 66 t mut set = self
0d30: 2e 73 6f 75 72 63 65 73 2e 6c 6f 63 6b 28 29 2e .sources.lock().
0d40: 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 6d 61 74 unwrap();....mat
0d50: 63 68 20 73 65 74 2e 67 65 74 28 69 64 29 20 7b ch set.get(id) {
0d60: 0a 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e .....Some(id) =>
0d70: 20 69 64 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 id.clone(),....
0d80: 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 .None => {......
0d90: 6c 65 74 20 69 64 20 3d 20 41 72 63 3a 3a 6e 65 let id = Arc::ne
0da0: 77 28 2a 69 64 29 3b 0a 09 09 09 09 09 73 65 74 w(*id);......set
0db0: 2e 69 6e 73 65 72 74 28 69 64 2e 63 6c 6f 6e 65 .insert(id.clone
0dc0: 28 29 29 3b 0a 09 09 09 09 09 69 64 2e 63 6c 6f ());......id.clo
0dd0: 6e 65 28 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d ne().....},....}
0de0: 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 63 6f 75 6e ...};...let coun
0df0: 74 20 3d 20 41 72 63 3a 3a 73 74 72 6f 6e 67 5f t = Arc::strong_
0e00: 63 6f 75 6e 74 28 26 69 64 29 3b 0a 09 09 69 66 count(&id);...if
0e10: 20 63 6f 75 6e 74 20 3d 3d 20 32 20 7b 0a 09 09 count == 2 {...
0e20: 09 6c 65 74 20 73 6f 75 72 63 65 20 3d 20 73 71 .let source = sq
0e30: 6c 78 3a 3a 71 75 65 72 79 21 28 22 73 65 6c 65 lx::query!("sele
0e40: 63 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 ct source_id, ch
0e50: 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 annel_id, url, i
0e60: 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 75 v_hash, owner, u
0e70: 72 6c 5f 72 65 20 66 72 6f 6d 20 72 73 73 74 67 rl_re from rsstg
0e80: 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 73 6f _source where so
0e90: 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 urce_id = $1 and
0ea0: 20 6f 77 6e 65 72 20 3d 20 24 32 22 2c 0a 09 09 owner = $2",...
0eb0: 09 09 2a 69 64 2c 20 6f 77 6e 65 72 29 2e 66 65 ..*id, owner).fe
0ec0: 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 73 65 6c tch_one(&mut sel
0ed0: 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 f.pool.acquire()
0ee0: 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 74 3f 3b .await?).await?;
0ef0: 0a 09 09 09 6c 65 74 20 64 65 73 74 69 6e 61 74 ....let destinat
0f00: 69 6f 6e 20 3d 20 6d 61 74 63 68 20 72 65 61 6c ion = match real
0f10: 20 7b 0a 09 09 09 09 74 72 75 65 20 3d 3e 20 74 {.....true => t
0f20: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 elegram_bot::Use
0f30: 72 49 64 3a 3a 6e 65 77 28 73 6f 75 72 63 65 2e rId::new(source.
0f40: 63 68 61 6e 6e 65 6c 5f 69 64 29 2c 0a 09 09 09 channel_id),....
0f50: 09 66 61 6c 73 65 20 3d 3e 20 74 65 6c 65 67 72 .false => telegr
0f60: 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a am_bot::UserId::
0f70: 6e 65 77 28 73 6f 75 72 63 65 2e 6f 77 6e 65 72 new(source.owner
0f80: 29 2c 0a 09 09 09 7d 3b 0a 09 09 09 6c 65 74 20 ),....};....let
0f90: 6d 75 74 20 74 68 69 73 5f 66 65 74 63 68 3a 20 mut this_fetch:
0fa0: 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 6d 65 3c Option<DateTime<
0fb0: 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 66 chrono::FixedOff
0fc0: 73 65 74 3e 3e 20 3d 20 4e 6f 6e 65 3b 0a 09 09 set>> = None;...
0fd0: 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 73 3a 20 .let mut posts:
0fe0: 42 54 72 65 65 4d 61 70 3c 44 61 74 65 54 69 6d BTreeMap<DateTim
0ff0: 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f e<chrono::FixedO
1000: 66 66 73 65 74 3e 2c 20 53 74 72 69 6e 67 3e 20 ffset>, String>
1010: 3d 20 42 54 72 65 65 4d 61 70 3a 3a 6e 65 77 28 = BTreeMap::new(
1020: 29 3b 0a 0a 09 09 09 6c 65 74 20 72 65 73 70 6f );.....let respo
1030: 6e 73 65 20 3d 20 73 65 6c 66 2e 68 74 74 70 5f nse = self.http_
1040: 63 6c 69 65 6e 74 2e 67 65 74 28 26 73 6f 75 72 client.get(&sour
1050: 63 65 2e 75 72 6c 29 2e 73 65 6e 64 28 29 2e 61 ce.url).send().a
1060: 77 61 69 74 3f 3b 0a 09 09 09 6c 65 74 20 73 74 wait?;....let st
1070: 61 74 75 73 20 3d 20 72 65 73 70 6f 6e 73 65 2e atus = response.
1080: 73 74 61 74 75 73 28 29 3b 0a 09 09 09 6c 65 74 status();....let
1090: 20 63 6f 6e 74 65 6e 74 20 3d 20 72 65 73 70 6f content = respo
10a0: 6e 73 65 2e 62 79 74 65 73 28 29 2e 61 77 61 69 nse.bytes().awai
10b0: 74 3f 3b 0a 09 09 09 6d 61 74 63 68 20 72 73 73 t?;....match rss
10c0: 3a 3a 43 68 61 6e 6e 65 6c 3a 3a 72 65 61 64 5f ::Channel::read_
10d0: 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e from(&content[..
10e0: 5d 29 20 7b 0a 09 09 09 09 4f 6b 28 66 65 65 64 ]) {.....Ok(feed
10f0: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 66 6f 72 20 ) => {......for
1100: 69 74 65 6d 20 69 6e 20 66 65 65 64 2e 69 74 65 item in feed.ite
1110: 6d 73 28 29 20 7b 0a 09 09 09 09 09 09 69 66 20 ms() {.......if
1120: 6c 65 74 20 53 6f 6d 65 28 6c 69 6e 6b 29 20 3d let Some(link) =
1130: 20 69 74 65 6d 2e 6c 69 6e 6b 28 29 20 7b 0a 09 item.link() {..
1140: 09 09 09 09 09 09 6c 65 74 20 64 61 74 65 20 3d ......let date =
1150: 20 6d 61 74 63 68 20 69 74 65 6d 2e 70 75 62 5f match item.pub_
1160: 64 61 74 65 28 29 20 7b 0a 09 09 09 09 09 09 09 date() {........
1170: 09 53 6f 6d 65 28 66 65 65 64 5f 64 61 74 65 29 .Some(feed_date)
1180: 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 => DateTime::pa
1190: 72 73 65 5f 66 72 6f 6d 5f 72 66 63 32 38 32 32 rse_from_rfc2822
11a0: 28 66 65 65 64 5f 64 61 74 65 29 2c 0a 09 09 09 (feed_date),....
11b0: 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 44 61 74 .....None => Dat
11c0: 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f eTime::parse_fro
11d0: 6d 5f 72 66 63 33 33 33 39 28 26 69 74 65 6d 2e m_rfc3339(&item.
11e0: 64 75 62 6c 69 6e 5f 63 6f 72 65 5f 65 78 74 28 dublin_core_ext(
11f0: 29 2e 75 6e 77 72 61 70 28 29 2e 64 61 74 65 73 ).unwrap().dates
1200: 28 29 5b 30 5d 29 2c 0a 09 09 09 09 09 09 09 7d ()[0]),........}
1210: 3f 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 75 72 ?;........let ur
1220: 6c 20 3d 20 6c 69 6e 6b 3b 0a 09 09 09 09 09 09 l = link;.......
1230: 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 64 61 .posts.insert(da
1240: 74 65 2c 20 75 72 6c 2e 74 6f 5f 73 74 72 69 6e te, url.to_strin
1250: 67 28 29 29 3b 0a 09 09 09 09 09 09 7d 0a 09 09 g());.......}...
1260: 09 09 09 7d 3b 0a 09 09 09 09 7d 2c 0a 09 09 09 ...};.....},....
1270: 09 45 72 72 28 65 72 72 29 20 3d 3e 20 6d 61 74 .Err(err) => mat
1280: 63 68 20 65 72 72 20 7b 0a 09 09 09 09 09 72 73 ch err {......rs
1290: 73 3a 3a 45 72 72 6f 72 3a 3a 49 6e 76 61 6c 69 s::Error::Invali
12a0: 64 53 74 61 72 74 54 61 67 20 3d 3e 20 7b 0a 09 dStartTag => {..
12b0: 09 09 09 09 09 6c 65 74 20 66 65 65 64 20 3d 20 .....let feed =
12c0: 61 74 6f 6d 5f 73 79 6e 64 69 63 61 74 69 6f 6e atom_syndication
12d0: 3a 3a 46 65 65 64 3a 3a 72 65 61 64 5f 66 72 6f ::Feed::read_fro
12e0: 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 0a m(&content[..]).
12f0: 09 09 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e ........with_con
1300: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1310: 22 50 72 6f 62 6c 65 6d 20 6f 70 65 6e 69 6e 67 "Problem opening
1320: 20 66 65 65 64 20 75 72 6c 3a 5c 6e 7b 7d 5c 6e feed url:\n{}\n
1330: 7b 7d 22 2c 20 26 73 6f 75 72 63 65 2e 75 72 6c {}", &source.url
1340: 2c 20 73 74 61 74 75 73 29 29 3f 3b 0a 09 09 09 , status))?;....
1350: 09 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 ...for item in f
1360: 65 65 64 2e 65 6e 74 72 69 65 73 28 29 20 7b 0a eed.entries() {.
1370: 09 09 09 09 09 09 09 6c 65 74 20 64 61 74 65 20 .......let date
1380: 3d 20 69 74 65 6d 2e 70 75 62 6c 69 73 68 65 64 = item.published
1390: 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 ().unwrap();....
13a0: 09 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 69 74 ....let url = it
13b0: 65 6d 2e 6c 69 6e 6b 73 28 29 5b 30 5d 2e 68 72 em.links()[0].hr
13c0: 65 66 28 29 3b 0a 09 09 09 09 09 09 09 70 6f 73 ef();........pos
13d0: 74 73 2e 69 6e 73 65 72 74 28 2a 64 61 74 65 2c ts.insert(*date,
13e0: 20 75 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 url.to_string()
13f0: 29 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 );.......};.....
1400: 09 7d 2c 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 .},......rss::Er
1410: 72 6f 72 3a 3a 45 6f 66 20 3d 3e 20 28 29 2c 0a ror::Eof => (),.
1420: 09 09 09 09 09 5f 20 3d 3e 20 62 61 69 6c 21 28 ....._ => bail!(
1430: 22 55 6e 73 75 70 70 6f 72 74 65 64 20 6f 72 20 "Unsupported or
1440: 6d 61 6e 67 6c 65 64 20 63 6f 6e 74 65 6e 74 3a mangled content:
1450: 5c 6e 7b 3a 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e 7b \n{:?}\n{:#?}\n{
1460: 3a 23 3f 7d 5c 6e 22 2c 20 26 73 6f 75 72 63 65 :#?}\n", &source
1470: 2e 75 72 6c 2c 20 65 72 72 2c 20 73 74 61 74 75 .url, err, statu
1480: 73 29 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 s).....}....};..
1490: 09 09 66 6f 72 20 28 64 61 74 65 2c 20 75 72 6c ..for (date, url
14a0: 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 65 72 28 ) in posts.iter(
14b0: 29 20 7b 0a 09 09 09 09 6c 65 74 20 70 6f 73 74 ) {.....let post
14c0: 5f 75 72 6c 3a 20 43 6f 77 3c 73 74 72 3e 20 3d _url: Cow<str> =
14d0: 20 6d 61 74 63 68 20 73 6f 75 72 63 65 2e 75 72 match source.ur
14e0: 6c 5f 72 65 20 7b 0a 09 09 09 09 09 53 6f 6d 65 l_re {......Some
14f0: 28 72 65 66 20 78 29 20 3d 3e 20 73 65 64 72 65 (ref x) => sedre
1500: 67 65 78 3a 3a 52 65 70 6c 61 63 65 43 6f 6d 6d gex::ReplaceComm
1510: 61 6e 64 3a 3a 6e 65 77 28 78 29 3f 2e 65 78 65 and::new(x)?.exe
1520: 63 75 74 65 28 75 72 6c 29 2c 0a 09 09 09 09 09 cute(url),......
1530: 4e 6f 6e 65 20 3d 3e 20 75 72 6c 2e 69 6e 74 6f None => url.into
1540: 28 29 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 69 (),.....};.....i
1550: 66 20 6c 65 74 20 53 6f 6d 65 28 65 78 69 73 74 f let Some(exist
1560: 73 29 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 s) = sqlx::query
1570: 21 28 22 73 65 6c 65 63 74 20 65 78 69 73 74 73 !("select exists
1580: 28 73 65 6c 65 63 74 20 74 72 75 65 20 66 72 6f (select true fro
1590: 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 77 68 65 m rsstg_post whe
15a0: 72 65 20 75 72 6c 20 3d 20 24 31 20 61 6e 64 20 re url = $1 and
15b0: 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 32 29 20 source_id = $2)
15c0: 61 73 20 65 78 69 73 74 73 3b 22 2c 0a 09 09 09 as exists;",....
15d0: 09 09 26 70 6f 73 74 5f 75 72 6c 2c 20 2a 69 64 ..&post_url, *id
15e0: 29 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 ).fetch_one(&mut
15f0: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
1600: 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 re().await?).awa
1610: 69 74 3f 2e 65 78 69 73 74 73 20 7b 0a 09 09 09 it?.exists {....
1620: 09 09 69 66 20 21 20 65 78 69 73 74 73 20 7b 0a ..if ! exists {.
1630: 09 09 09 09 09 09 69 66 20 74 68 69 73 5f 66 65 ......if this_fe
1640: 74 63 68 2e 69 73 5f 6e 6f 6e 65 28 29 20 7c 7c tch.is_none() ||
1650: 20 2a 64 61 74 65 20 3e 20 74 68 69 73 5f 66 65 *date > this_fe
1660: 74 63 68 2e 75 6e 77 72 61 70 28 29 20 7b 0a 09 tch.unwrap() {..
1670: 09 09 09 09 09 09 74 68 69 73 5f 66 65 74 63 68 ......this_fetch
1680: 20 3d 20 53 6f 6d 65 28 2a 64 61 74 65 29 3b 0a = Some(*date);.
1690: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 73 ......};.......s
16a0: 65 6c 66 2e 72 65 71 75 65 73 74 28 20 6d 61 74 elf.request( mat
16b0: 63 68 20 26 73 6f 75 72 63 65 2e 69 76 5f 68 61 ch &source.iv_ha
16c0: 73 68 20 7b 0a 09 09 09 09 09 09 09 09 53 6f 6d sh {.........Som
16d0: 65 28 68 61 73 68 29 20 3d 3e 20 74 65 6c 65 67 e(hash) => teleg
16e0: 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 ram_bot::SendMes
16f0: 73 61 67 65 3a 3a 6e 65 77 28 64 65 73 74 69 6e sage::new(destin
1700: 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 21 28 22 ation, format!("
1710: 3c 61 20 68 72 65 66 3d 5c 22 68 74 74 70 73 3a <a href=\"https:
1720: 2f 2f 74 2e 6d 65 2f 69 76 3f 75 72 6c 3d 7b 7d //t.me/iv?url={}
1730: 26 72 68 61 73 68 3d 7b 7d 5c 22 3e 20 3c 2f 61 &rhash={}\"> </a
1740: 3e 7b 30 7d 22 2c 20 26 70 6f 73 74 5f 75 72 6c >{0}", &post_url
1750: 2c 20 68 61 73 68 29 29 2c 0a 09 09 09 09 09 09 , hash)),.......
1760: 09 09 4e 6f 6e 65 20 3d 3e 20 74 65 6c 65 67 72 ..None => telegr
1770: 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 73 am_bot::SendMess
1780: 61 67 65 3a 3a 6e 65 77 28 64 65 73 74 69 6e 61 age::new(destina
1790: 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 21 28 22 7b tion, format!("{
17a0: 7d 22 2c 20 70 6f 73 74 5f 75 72 6c 29 29 2c 0a }", post_url)),.
17b0: 09 09 09 09 09 09 09 7d 2e 70 61 72 73 65 5f 6d .......}.parse_m
17c0: 6f 64 65 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 ode(telegram_bot
17d0: 3a 3a 74 79 70 65 73 3a 3a 50 61 72 73 65 4d 6f ::types::ParseMo
17e0: 64 65 3a 3a 48 74 6d 6c 29 29 2e 61 77 61 69 74 de::Html)).await
17f0: 0a 09 09 09 09 09 09 09 2e 63 6f 6e 74 65 78 74 .........context
1800: 28 22 43 61 6e 27 74 20 70 6f 73 74 20 6d 65 73 ("Can't post mes
1810: 73 61 67 65 3a 22 29 3f 3b 0a 09 09 09 09 09 09 sage:")?;.......
1820: 73 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 69 6e sqlx::query!("in
1830: 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 67 5f sert into rsstg_
1840: 70 6f 73 74 20 28 73 6f 75 72 63 65 5f 69 64 2c post (source_id,
1850: 20 70 6f 73 74 65 64 2c 20 75 72 6c 29 20 76 61 posted, url) va
1860: 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 24 33 lues ($1, $2, $3
1870: 29 3b 22 2c 0a 09 09 09 09 09 09 09 2a 69 64 2c );",........*id,
1880: 20 64 61 74 65 2c 20 26 70 6f 73 74 5f 75 72 6c date, &post_url
1890: 29 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 73 ).execute(&mut s
18a0: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
18b0: 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 74 ().await?).await
18c0: 3f 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 7d ?;......};.....}
18d0: 3b 0a 09 09 09 09 70 6f 73 74 65 64 20 2b 3d 20 ;.....posted +=
18e0: 31 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 6f 73 74 1;....};....post
18f0: 73 2e 63 6c 65 61 72 28 29 3b 0a 09 09 7d 3b 0a s.clear();...};.
1900: 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 ..sqlx::query!("
1910: 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 update rsstg_sou
1920: 72 63 65 20 73 65 74 20 6c 61 73 74 5f 73 63 72 rce set last_scr
1930: 61 70 65 20 3d 20 6e 6f 77 28 29 20 77 68 65 72 ape = now() wher
1940: 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 e source_id = $1
1950: 3b 22 2c 0a 09 09 09 2a 69 64 29 2e 65 78 65 63 ;",....*id).exec
1960: 75 74 65 28 26 6d 75 74 20 73 65 6c 66 2e 70 6f ute(&mut self.po
1970: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
1980: 69 74 3f 29 2e 61 77 61 69 74 3f 3b 0a 09 09 4f it?).await?;...O
1990: 6b 28 66 6f 72 6d 61 74 21 28 22 50 6f 73 74 65 k(format!("Poste
19a0: 64 3a 20 7b 7d 22 2c 20 26 70 6f 73 74 65 64 29 d: {}", &posted)
19b0: 2e 69 6e 74 6f 28 29 29 0a 09 7d 0a 0a 09 70 75 .into())..}...pu
19c0: 62 20 61 73 79 6e 63 20 66 6e 20 64 65 6c 65 74 b async fn delet
19d0: 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 e<S>(&self, sour
19e0: 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e ce_id: &i32, own
19f0: 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 er: S) -> Result
1a00: 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 <Cow<'_, str>>..
1a10: 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 where S: Into<i6
1a20: 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 4> {...let owner
1a30: 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b = owner.into();
1a40: 0a 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a ....match sqlx::
1a50: 71 75 65 72 79 21 28 22 64 65 6c 65 74 65 20 66 query!("delete f
1a60: 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 rom rsstg_source
1a70: 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 where source_id
1a80: 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 = $1 and owner
1a90: 3d 20 24 32 3b 22 2c 0a 09 09 09 73 6f 75 72 63 = $2;",....sourc
1aa0: 65 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 65 78 65 e_id, owner).exe
1ab0: 63 75 74 65 28 26 6d 75 74 20 73 65 6c 66 2e 70 cute(&mut self.p
1ac0: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
1ad0: 61 69 74 3f 29 2e 61 77 61 69 74 3f 2e 72 6f 77 ait?).await?.row
1ae0: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
1af0: 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 ..0 => { Ok("No
1b00: 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 data found found
1b10: 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 .".into()) },...
1b20: 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 .x => { Ok(forma
1b30: 74 21 28 22 7b 7d 20 73 6f 75 72 63 65 73 20 72 t!("{} sources r
1b40: 65 6d 6f 76 65 64 2e 22 2c 20 78 29 2e 69 6e 74 emoved.", x).int
1b50: 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a o()) },...}..}..
1b60: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 63 6c .pub async fn cl
1b70: 65 61 6e 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f ean<S>(&self, so
1b80: 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f urce_id: &i32, o
1b90: 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 wner: S) -> Resu
1ba0: 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e lt<Cow<'_, str>>
1bb0: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
1bc0: 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e i64> {...let own
1bd0: 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 er = owner.into(
1be0: 29 3b 0a 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 );....match sqlx
1bf0: 3a 3a 71 75 65 72 79 21 28 22 64 65 6c 65 74 65 ::query!("delete
1c00: 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 from rsstg_post
1c10: 20 70 20 75 73 69 6e 67 20 72 73 73 74 67 5f 73 p using rsstg_s
1c20: 6f 75 72 63 65 20 73 20 77 68 65 72 65 20 70 2e ource s where p.
1c30: 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 source_id = $1 a
1c40: 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 20 61 6e nd owner = $2 an
1c50: 64 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 d p.source_id =
1c60: 73 2e 73 6f 75 72 63 65 5f 69 64 3b 22 2c 0a 09 s.source_id;",..
1c70: 09 09 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e ..source_id, own
1c80: 65 72 29 2e 65 78 65 63 75 74 65 28 26 6d 75 74 er).execute(&mut
1c90: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
1ca0: 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 re().await?).awa
1cb0: 69 74 3f 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 it?.rows_affecte
1cc0: 64 28 29 20 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 d() {....0 => {
1cd0: 4f 6b 28 22 4e 6f 20 64 61 74 61 20 66 6f 75 6e Ok("No data foun
1ce0: 64 20 66 6f 75 6e 64 2e 22 2e 69 6e 74 6f 28 29 d found.".into()
1cf0: 29 20 7d 2c 0a 09 09 09 78 20 3d 3e 20 7b 20 4f ) },....x => { O
1d00: 6b 28 66 6f 72 6d 61 74 21 28 22 7b 7d 20 70 6f k(format!("{} po
1d10: 73 74 73 20 70 75 72 67 65 64 2e 22 2c 20 78 29 sts purged.", x)
1d20: 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a .into()) },...}.
1d30: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
1d40: 6e 20 65 6e 61 62 6c 65 3c 53 3e 28 26 73 65 6c n enable<S>(&sel
1d50: 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 f, source_id: &i
1d60: 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 32, owner: S) ->
1d70: 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 77 Result<&str>..w
1d80: 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 here S: Into<i64
1d90: 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 > {...let owner
1da0: 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a = owner.into();.
1db0: 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 ...match sqlx::q
1dc0: 75 65 72 79 21 28 22 75 70 64 61 74 65 20 72 73 uery!("update rs
1dd0: 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 stg_source set e
1de0: 6e 61 62 6c 65 64 20 3d 20 74 72 75 65 20 77 68 nabled = true wh
1df0: 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 ere source_id =
1e00: 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 $1 and owner = $
1e10: 32 22 2c 0a 09 09 09 73 6f 75 72 63 65 5f 69 64 2",....source_id
1e20: 2c 20 6f 77 6e 65 72 29 2e 65 78 65 63 75 74 65 , owner).execute
1e30: 28 26 6d 75 74 20 73 65 6c 66 2e 70 6f 6f 6c 2e (&mut self.pool.
1e40: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 3f acquire().await?
1e50: 29 2e 61 77 61 69 74 3f 2e 72 6f 77 73 5f 61 66 ).await?.rows_af
1e60: 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 31 20 fected() {....1
1e70: 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 => { Ok("Source
1e80: 65 6e 61 62 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 enabled.") },...
1e90: 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 .0 => { Ok("Sour
1ea0: 63 65 20 6e 6f 74 20 66 6f 75 6e 64 2e 22 29 20 ce not found.")
1eb0: 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 45 72 72 },...._ => { Err
1ec0: 28 61 6e 79 68 6f 77 21 28 22 44 61 74 61 62 61 (anyhow!("Databa
1ed0: 73 65 20 65 72 72 6f 72 2e 22 29 29 20 7d 2c 0a se error.")) },.
1ee0: 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ..}..}...pub asy
1ef0: 6e 63 20 66 6e 20 64 69 73 61 62 6c 65 3c 53 3e nc fn disable<S>
1f00: 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 (&self, source_i
1f10: 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 d: &i32, owner:
1f20: 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 S) -> Result<&st
1f30: 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 r>..where S: Int
1f40: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f o<i64> {...let o
1f50: 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 wner = owner.int
1f60: 6f 28 29 3b 0a 0a 09 09 6d 61 74 63 68 20 73 71 o();....match sq
1f70: 6c 78 3a 3a 71 75 65 72 79 21 28 22 75 70 64 61 lx::query!("upda
1f80: 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 te rsstg_source
1f90: 73 65 74 20 65 6e 61 62 6c 65 64 20 3d 20 66 61 set enabled = fa
1fa0: 6c 73 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 lse where source
1fb0: 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e _id = $1 and own
1fc0: 65 72 20 3d 20 24 32 22 2c 0a 09 09 09 73 6f 75 er = $2",....sou
1fd0: 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 65 rce_id, owner).e
1fe0: 78 65 63 75 74 65 28 26 6d 75 74 20 73 65 6c 66 xecute(&mut self
1ff0: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
2000: 61 77 61 69 74 3f 29 2e 61 77 61 69 74 3f 2e 72 await?).await?.r
2010: 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b ows_affected() {
2020: 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 ....1 => { Ok("S
2030: 6f 75 72 63 65 20 64 69 73 61 62 6c 65 64 2e 22 ource disabled."
2040: 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f ) },....0 => { O
2050: 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f k("Source not fo
2060: 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d und.") },...._ =
2070: 3e 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 > { Err(anyhow!(
2080: 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e "Database error.
2090: 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 ")) },...}..}...
20a0: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 75 70 64 pub async fn upd
20b0: 61 74 65 3c 53 3e 28 26 73 65 6c 66 2c 20 75 70 ate<S>(&self, up
20c0: 64 61 74 65 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 date: Option<i32
20d0: 3e 2c 20 63 68 61 6e 6e 65 6c 3a 20 26 73 74 72 >, channel: &str
20e0: 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 , channel_id: i6
20f0: 34 2c 20 75 72 6c 3a 20 26 73 74 72 2c 20 69 76 4, url: &str, iv
2100: 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 _hash: Option<&s
2110: 74 72 3e 2c 20 75 72 6c 5f 72 65 3a 20 4f 70 74 tr>, url_re: Opt
2120: 69 6f 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 72 ion<&str>, owner
2130: 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 : S) -> Result<&
2140: 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 str>..where S: I
2150: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
2160: 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 owner = owner.i
2170: 6e 74 6f 28 29 3b 0a 0a 09 09 6d 61 74 63 68 20 nto();....match
2180: 6d 61 74 63 68 20 75 70 64 61 74 65 20 7b 0a 09 match update {..
2190: 09 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 7b ...Some(id) => {
21a0: 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 ......sqlx::quer
21b0: 79 21 28 22 75 70 64 61 74 65 20 72 73 73 74 67 y!("update rsstg
21c0: 5f 73 6f 75 72 63 65 20 73 65 74 20 63 68 61 6e _source set chan
21d0: 6e 65 6c 5f 69 64 20 3d 20 24 32 2c 20 75 72 6c nel_id = $2, url
21e0: 20 3d 20 24 33 2c 20 69 76 5f 68 61 73 68 20 3d = $3, iv_hash =
21f0: 20 24 34 2c 20 6f 77 6e 65 72 20 3d 20 24 35 2c $4, owner = $5,
2200: 20 63 68 61 6e 6e 65 6c 20 3d 20 24 36 2c 20 75 channel = $6, u
2210: 72 6c 5f 72 65 20 3d 20 24 37 20 77 68 65 72 65 rl_re = $7 where
2220: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 22 source_id = $1"
2230: 2c 0a 09 09 09 09 09 09 69 64 2c 20 63 68 61 6e ,.......id, chan
2240: 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f nel_id, url, iv_
2250: 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 hash, owner, cha
2260: 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 2e 65 78 nnel, url_re).ex
2270: 65 63 75 74 65 28 26 6d 75 74 20 73 65 6c 66 2e ecute(&mut self.
2280: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
2290: 77 61 69 74 3f 29 2e 61 77 61 69 74 0a 09 09 09 wait?).await....
22a0: 09 7d 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 .},.....None =>
22b0: 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 {......sqlx::que
22c0: 72 79 21 28 22 69 6e 73 65 72 74 20 69 6e 74 6f ry!("insert into
22d0: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 28 63 rsstg_source (c
22e0: 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 hannel_id, url,
22f0: 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 iv_hash, owner,
2300: 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 channel, url_re)
2310: 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 2c values ($1, $2,
2320: 20 24 33 2c 20 24 34 2c 20 24 35 2c 20 24 36 29 $3, $4, $5, $6)
2330: 22 2c 0a 09 09 09 09 09 09 63 68 61 6e 6e 65 6c ",.......channel
2340: 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 _id, url, iv_has
2350: 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 6e 6e 65 h, owner, channe
2360: 6c 2c 20 75 72 6c 5f 72 65 29 2e 65 78 65 63 75 l, url_re).execu
2370: 74 65 28 26 6d 75 74 20 73 65 6c 66 2e 70 6f 6f te(&mut self.poo
2380: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
2390: 74 3f 29 2e 61 77 61 69 74 0a 09 09 09 09 7d 2c t?).await.....},
23a0: 0a 09 09 09 7d 20 7b 0a 09 09 09 4f 6b 28 5f 29 ....} {....Ok(_)
23b0: 20 3d 3e 20 4f 6b 28 6d 61 74 63 68 20 75 70 64 => Ok(match upd
23c0: 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 5f ate {.....Some(_
23d0: 29 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 75 70 ) => "Channel up
23e0: 64 61 74 65 64 2e 22 2c 0a 09 09 09 09 4e 6f 6e dated.",.....Non
23f0: 65 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 61 64 e => "Channel ad
2400: 64 65 64 2e 22 2c 0a 09 09 09 7d 29 2c 0a 09 09 ded.",....}),...
2410: 09 45 72 72 28 73 71 6c 78 3a 3a 45 72 72 6f 72 .Err(sqlx::Error
2420: 3a 3a 44 61 74 61 62 61 73 65 28 65 72 72 29 29 ::Database(err))
2430: 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 74 63 68 20 => {.....match
2440: 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a 3a 3c 73 err.downcast::<s
2450: 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a 3a 50 qlx::postgres::P
2460: 67 44 61 74 61 62 61 73 65 45 72 72 6f 72 3e 28 gDatabaseError>(
2470: 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b 0a 09 09 ).routine() {...
2480: 09 09 09 53 6f 6d 65 28 22 5f 62 74 5f 63 68 65 ...Some("_bt_che
2490: 63 6b 5f 75 6e 69 71 75 65 22 2c 20 29 20 3d 3e ck_unique", ) =>
24a0: 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 44 75 70 {.......Ok("Dup
24b0: 6c 69 63 61 74 65 20 6b 65 79 2e 22 29 0a 09 09 licate key.")...
24c0: 09 09 09 7d 2c 0a 09 09 09 09 09 53 6f 6d 65 28 ...},......Some(
24d0: 5f 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b _) => {.......Ok
24e0: 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 ("Database error
24f0: 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 .")......},.....
2500: 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 .None => {......
2510: 09 4f 6b 28 22 4e 6f 20 64 61 74 61 62 61 73 65 .Ok("No database
2520: 20 65 72 72 6f 72 20 65 78 74 72 61 63 74 65 64 error extracted
2530: 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 .")......},.....
2540: 7d 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 65 }....},....Err(e
2550: 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 62 61 69 rr) => {.....bai
2560: 6c 21 28 22 53 6f 72 72 79 2c 20 75 6e 6b 6e 6f l!("Sorry, unkno
2570: 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 3f 7d wn error:\n{:#?}
2580: 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 09 09 7d 2c \n", err);....},
2590: 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 ...}..}...async
25a0: 66 6e 20 61 75 74 6f 66 65 74 63 68 28 26 73 65 fn autofetch(&se
25b0: 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 73 74 lf) -> Result<st
25c0: 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f d::time::Duratio
25d0: 6e 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 64 n> {...let mut d
25e0: 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 elay = chrono::D
25f0: 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 uration::minutes
2600: 28 31 29 3b 0a 09 09 6c 65 74 20 6e 6f 77 20 3d (1);...let now =
2610: 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a chrono::Local::
2620: 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 now();...let mut
2630: 20 71 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a 71 queue = sqlx::q
2640: 75 65 72 79 21 28 72 23 22 73 65 6c 65 63 74 20 uery!(r#"select
2650: 73 6f 75 72 63 65 5f 69 64 2c 20 6e 65 78 74 5f source_id, next_
2660: 66 65 74 63 68 20 61 73 20 22 6e 65 78 74 5f 66 fetch as "next_f
2670: 65 74 63 68 3a 20 44 61 74 65 54 69 6d 65 3c 63 etch: DateTime<c
2680: 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3e 22 2c 20 hrono::Local>",
2690: 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 73 74 67 owner from rsstg
26a0: 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 6c 20 6c _order natural l
26b0: 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 67 5f 73 eft join rsstg_s
26c0: 6f 75 72 63 65 20 77 68 65 72 65 20 6e 65 78 74 ource where next
26d0: 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 29 20 2b _fetch < now() +
26e0: 20 69 6e 74 65 72 76 61 6c 20 27 31 20 6d 69 6e interval '1 min
26f0: 75 74 65 27 3b 22 23 29 0a 09 09 09 2e 66 65 74 ute';"#).....fet
2700: 63 68 5f 61 6c 6c 28 26 6d 75 74 20 73 65 6c 66 ch_all(&mut self
2710: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
2720: 61 77 61 69 74 3f 29 2e 61 77 61 69 74 3f 3b 0a await?).await?;.
2730: 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 71 75 65 ..for row in que
2740: 75 65 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 69 ue.iter() {....i
2750: 66 20 6c 65 74 20 53 6f 6d 65 28 6e 65 78 74 5f f let Some(next_
2760: 66 65 74 63 68 29 20 3d 20 72 6f 77 2e 6e 65 78 fetch) = row.nex
2770: 74 5f 66 65 74 63 68 20 7b 0a 09 09 09 09 69 66 t_fetch {.....if
2780: 20 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f next_fetch < no
2790: 77 20 7b 0a 09 09 09 09 09 69 66 20 6c 65 74 20 w {......if let
27a0: 28 53 6f 6d 65 28 6f 77 6e 65 72 29 2c 20 53 6f (Some(owner), So
27b0: 6d 65 28 73 6f 75 72 63 65 5f 69 64 29 29 20 3d me(source_id)) =
27c0: 20 28 72 6f 77 2e 6f 77 6e 65 72 2c 20 72 6f 77 (row.owner, row
27d0: 2e 73 6f 75 72 63 65 5f 69 64 29 20 7b 0a 09 09 .source_id) {...
27e0: 09 09 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 ....let clone =
27f0: 43 6f 72 65 20 7b 0a 09 09 09 09 09 09 09 6f 77 Core {........ow
2800: 6e 65 72 5f 63 68 61 74 3a 20 74 65 6c 65 67 72 ner_chat: telegr
2810: 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a am_bot::UserId::
2820: 6e 65 77 28 6f 77 6e 65 72 29 2c 0a 09 09 09 09 new(owner),.....
2830: 09 09 09 2e 2e 73 65 6c 66 2e 63 6c 6f 6e 65 28 .....self.clone(
2840: 29 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ).......};......
2850: 09 74 61 73 6b 3a 3a 73 70 61 77 6e 28 61 73 79 .task::spawn(asy
2860: 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 09 09 nc move {.......
2870: 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 .if let Err(err)
2880: 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b 28 26 = clone.check(&
2890: 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 source_id, owner
28a0: 2c 20 74 72 75 65 29 2e 61 77 61 69 74 20 7b 0a , true).await {.
28b0: 09 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 ........if let E
28c0: 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e rr(err) = clone.
28d0: 73 65 6e 64 28 26 66 6f 72 6d 61 74 21 28 22 f0 send(&format!("š
28e0: 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c {:?}", err),
28f0: 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 61 None, None).awa
2900: 69 74 20 7b 0a 09 09 09 09 09 09 09 09 09 64 62 it {..........db
2910: 67 21 28 22 43 68 65 63 6b 20 65 72 72 6f 72 3a g!("Check error:
2920: 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 09 {}", err);.....
2930: 09 09 09 09 09 2f 2f 20 63 6c 6f 6e 65 2e 64 69 .....// clone.di
2940: 73 61 62 6c 65 28 26 73 6f 75 72 63 65 5f 69 64 sable(&source_id
2950: 2c 20 6f 77 6e 65 72 29 2e 61 77 61 69 74 2e 75 , owner).await.u
2960: 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 09 09 09 nwrap();........
2970: 09 7d 3b 0a 09 09 09 09 09 09 09 7d 3b 0a 09 09 .};........};...
2980: 09 09 09 09 7d 29 3b 0a 09 09 09 09 09 7d 0a 09 ....});......}..
2990: 09 09 09 7d 20 65 6c 73 65 20 69 66 20 6e 65 78 ...} else if nex
29a0: 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 20 3c 20 t_fetch - now <
29b0: 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 64 65 6c delay {......del
29c0: 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 63 68 20 ay = next_fetch
29d0: 2d 20 6e 6f 77 3b 0a 09 09 09 09 7d 0a 09 09 09 - now;.....}....
29e0: 7d 0a 09 09 7d 3b 0a 09 09 71 75 65 75 65 2e 63 }...};...queue.c
29f0: 6c 65 61 72 28 29 3b 0a 09 09 4f 6b 28 64 65 6c lear();...Ok(del
2a00: 61 79 2e 74 6f 5f 73 74 64 28 29 3f 29 0a 09 7d ay.to_std()?)..}
2a10: 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 ...pub async fn
2a20: 6c 69 73 74 3c 53 3e 28 26 73 65 6c 66 2c 20 6f list<S>(&self, o
2a30: 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 wner: S) -> Resu
2a40: 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 72 lt<String>..wher
2a50: 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e S: Into<i64> {
2a60: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f ...let owner = o
2a70: 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 wner.into();....
2a80: 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 3a 20 56 let mut reply: V
2a90: 65 63 3c 43 6f 77 3c 73 74 72 3e 3e 20 3d 20 76 ec<Cow<str>> = v
2aa0: 65 63 21 5b 5d 3b 0a 09 09 72 65 70 6c 79 2e 70 ec![];...reply.p
2ab0: 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 2e ush("Channels:".
2ac0: 69 6e 74 6f 28 29 29 3b 0a 09 09 6c 65 74 20 72 into());...let r
2ad0: 6f 77 73 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 ows = sqlx::quer
2ae0: 79 21 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 y!("select sourc
2af0: 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 65 e_id, channel, e
2b00: 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 76 5f nabled, url, iv_
2b10: 68 61 73 68 2c 20 75 72 6c 5f 72 65 20 66 72 6f hash, url_re fro
2b20: 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 m rsstg_source w
2b30: 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 31 20 here owner = $1
2b40: 6f 72 64 65 72 20 62 79 20 73 6f 75 72 63 65 5f order by source_
2b50: 69 64 22 2c 0a 09 09 09 6f 77 6e 65 72 29 2e 66 id",....owner).f
2b60: 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 2a 73 etch_all(&mut *s
2b70: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
2b80: 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 74 ().await?).await
2b90: 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 ?;...for row in
2ba0: 72 6f 77 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 rows.iter() {...
2bb0: 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d .reply.push(form
2bc0: 61 74 21 28 22 5c 6e 5c 5c 23 ef b8 8f e2 83 a3 at!("\n\\#ļøā£
2bd0: 20 7b 7d 20 5c 5c 2a ef b8 8f e2 83 a3 20 60 7b {} \\*ļøā£ `{
2be0: 7d 60 20 7b 7d 5c 6e f0 9f 94 97 20 60 7b 7d 60 }` {}\nš `{}`
2bf0: 22 2c 20 72 6f 77 2e 73 6f 75 72 63 65 5f 69 64 ", row.source_id
2c00: 2c 20 72 6f 77 2e 63 68 61 6e 6e 65 6c 2c 0a 09 , row.channel,..
2c10: 09 09 09 6d 61 74 63 68 20 72 6f 77 2e 65 6e 61 ...match row.ena
2c20: 62 6c 65 64 20 7b 0a 09 09 09 09 09 74 72 75 65 bled {......true
2c30: 20 20 3d 3e 20 22 f0 9f 94 84 20 65 6e 61 62 6c => "š enabl
2c40: 65 64 22 2c 0a 09 09 09 09 09 66 61 6c 73 65 20 ed",......false
2c50: 3d 3e 20 22 e2 9b 94 20 64 69 73 61 62 6c 65 64 => "ā disabled
2c60: 22 2c 0a 09 09 09 09 7d 2c 20 72 6f 77 2e 75 72 ",.....}, row.ur
2c70: 6c 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 69 l).into());....i
2c80: 66 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 68 29 f let Some(hash)
2c90: 20 3d 20 26 72 6f 77 2e 69 76 5f 68 61 73 68 20 = &row.iv_hash
2ca0: 7b 0a 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 {.....reply.push
2cb0: 28 66 6f 72 6d 61 74 21 28 22 49 56 3a 20 60 7b (format!("IV: `{
2cc0: 7d 60 22 2c 20 68 61 73 68 29 2e 69 6e 74 6f 28 }`", hash).into(
2cd0: 29 29 3b 0a 09 09 09 7d 0a 09 09 09 69 66 20 6c ));....}....if l
2ce0: 65 74 20 53 6f 6d 65 28 72 65 29 20 3d 20 26 72 et Some(re) = &r
2cf0: 6f 77 2e 75 72 6c 5f 72 65 20 7b 0a 09 09 09 09 ow.url_re {.....
2d00: 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 reply.push(forma
2d10: 74 21 28 22 52 45 3a 20 60 7b 7d 60 22 2c 20 72 t!("RE: `{}`", r
2d20: 65 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 7d e).into());....}
2d30: 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 65 70 6c 79 ...};...Ok(reply
2d40: 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 0a 09 7d 0a .join("\n"))..}.
2d50: 7d 0a }.