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 7b 0a 09 70 6f 73 74 67 72 65 73 3a 3a 50 ::{..postgres::P
0070: 67 50 6f 6f 6c 4f 70 74 69 6f 6e 73 2c 0a 09 52 gPoolOptions,..R
0080: 6f 77 2c 0a 7d 3b 0a 75 73 65 20 73 74 64 3a 3a ow,.};.use std::
0090: 7b 0a 09 62 6f 72 72 6f 77 3a 3a 43 6f 77 2c 0a {..borrow::Cow,.
00a0: 09 63 6f 6c 6c 65 63 74 69 6f 6e 73 3a 3a 7b 0a .collections::{.
00b0: 09 09 42 54 72 65 65 4d 61 70 2c 0a 09 09 48 61 ..BTreeMap,...Ha
00c0: 73 68 53 65 74 2c 0a 09 7d 2c 0a 09 73 79 6e 63 shSet,..},..sync
00d0: 3a 3a 7b 41 72 63 2c 20 4d 75 74 65 78 7d 2c 0a ::{Arc, Mutex},.
00e0: 7d 3b 0a 0a 23 5b 64 65 72 69 76 65 28 43 6c 6f };..#[derive(Clo
00f0: 6e 65 29 5d 0a 70 75 62 20 73 74 72 75 63 74 20 ne)].pub struct
0100: 43 6f 72 65 20 7b 0a 09 6f 77 6e 65 72 5f 63 68 Core {..owner_ch
0110: 61 74 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 at: telegram_bot
0120: 3a 3a 55 73 65 72 49 64 2c 0a 09 70 75 62 20 74 ::UserId,..pub t
0130: 67 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a g: telegram_bot:
0140: 3a 41 70 69 2c 0a 09 70 75 62 20 6d 79 3a 20 74 :Api,..pub my: t
0150: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 elegram_bot::Use
0160: 72 2c 0a 09 70 6f 6f 6c 3a 20 73 71 6c 78 3a 3a r,..pool: sqlx::
0170: 50 6f 6f 6c 3c 73 71 6c 78 3a 3a 50 6f 73 74 67 Pool<sqlx::Postg
0180: 72 65 73 3e 2c 0a 09 73 6f 75 72 63 65 73 3a 20 res>,..sources:
0190: 41 72 63 3c 4d 75 74 65 78 3c 48 61 73 68 53 65 Arc<Mutex<HashSe
01a0: 74 3c 41 72 63 3c 69 33 32 3e 3e 3e 3e 2c 0a 09 t<Arc<i32>>>>,..
01b0: 68 74 74 70 5f 63 6c 69 65 6e 74 3a 20 72 65 71 http_client: req
01c0: 77 65 73 74 3a 3a 43 6c 69 65 6e 74 2c 0a 7d 0a west::Client,.}.
01d0: 0a 69 6d 70 6c 20 43 6f 72 65 20 7b 0a 09 70 75 .impl Core {..pu
01e0: 62 20 66 6e 20 6e 65 77 28 73 65 74 74 69 6e 67 b fn new(setting
01f0: 73 3a 20 63 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 s: config::Confi
0200: 67 29 20 2d 3e 20 52 65 73 75 6c 74 3c 41 72 63 g) -> Result<Arc
0210: 3c 43 6f 72 65 3e 3e 20 7b 0a 09 09 6c 65 74 20 <Core>> {...let
0220: 6f 77 6e 65 72 20 3d 20 73 65 74 74 69 6e 67 73 owner = settings
0230: 2e 67 65 74 5f 69 6e 74 28 22 6f 77 6e 65 72 22 .get_int("owner"
0240: 29 3f 3b 0a 09 09 6c 65 74 20 61 70 69 5f 6b 65 )?;...let api_ke
0250: 79 20 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 y = settings.get
0260: 5f 73 74 72 69 6e 67 28 22 61 70 69 5f 6b 65 79 _string("api_key
0270: 22 29 3f 3b 0a 09 09 6c 65 74 20 74 67 20 3d 20 ")?;...let tg =
0280: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 41 70 telegram_bot::Ap
0290: 69 3a 3a 6e 65 77 28 61 70 69 5f 6b 65 79 29 3b i::new(api_key);
02a0: 0a 09 09 6c 65 74 20 74 67 5f 63 6c 6f 6e 65 64 ...let tg_cloned
02b0: 20 3d 20 74 67 2e 63 6c 6f 6e 65 28 29 3b 0a 0a = tg.clone();..
02c0: 09 09 6c 65 74 20 6d 75 74 20 63 6c 69 65 6e 74 ..let mut client
02d0: 20 3d 20 72 65 71 77 65 73 74 3a 3a 43 6c 69 65 = reqwest::Clie
02e0: 6e 74 3a 3a 62 75 69 6c 64 65 72 28 29 3b 0a 09 nt::builder();..
02f0: 09 69 66 20 6c 65 74 20 4f 6b 28 70 72 6f 78 79 .if let Ok(proxy
0300: 29 20 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 ) = settings.get
0310: 5f 73 74 72 69 6e 67 28 22 70 72 6f 78 79 22 29 _string("proxy")
0320: 20 7b 0a 09 09 09 6c 65 74 20 70 72 6f 78 79 20 {....let proxy
0330: 3d 20 72 65 71 77 65 73 74 3a 3a 50 72 6f 78 79 = reqwest::Proxy
0340: 3a 3a 61 6c 6c 28 70 72 6f 78 79 29 3f 3b 0a 09 ::all(proxy)?;..
0350: 09 09 63 6c 69 65 6e 74 20 3d 20 63 6c 69 65 6e ..client = clien
0360: 74 2e 70 72 6f 78 79 28 70 72 6f 78 79 29 3b 0a t.proxy(proxy);.
0370: 09 09 7d 0a 09 09 6c 65 74 20 68 74 74 70 5f 63 ..}...let http_c
0380: 6c 69 65 6e 74 20 3d 20 63 6c 69 65 6e 74 2e 62 lient = client.b
0390: 75 69 6c 64 28 29 3f 3b 0a 09 09 6c 65 74 20 63 uild()?;...let c
03a0: 6f 72 65 20 3d 20 41 72 63 3a 3a 6e 65 77 28 43 ore = Arc::new(C
03b0: 6f 72 65 20 7b 0a 09 09 09 74 67 2c 0a 09 09 09 ore {....tg,....
03c0: 6d 79 3a 20 74 61 73 6b 3a 3a 62 6c 6f 63 6b 5f my: task::block_
03d0: 6f 6e 28 61 73 79 6e 63 20 7b 0a 09 09 09 09 74 on(async {.....t
03e0: 67 5f 63 6c 6f 6e 65 64 2e 73 65 6e 64 28 74 65 g_cloned.send(te
03f0: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 47 65 74 4d legram_bot::GetM
0400: 65 29 2e 61 77 61 69 74 0a 09 09 09 7d 29 3f 2c e).await....})?,
0410: 0a 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 ....owner_chat:
0420: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 telegram_bot::Us
0430: 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 erId::new(owner)
0440: 2c 0a 09 09 09 70 6f 6f 6c 3a 20 50 67 50 6f 6f ,....pool: PgPoo
0450: 6c 4f 70 74 69 6f 6e 73 3a 3a 6e 65 77 28 29 0a lOptions::new().
0460: 09 09 09 09 2e 6d 61 78 5f 63 6f 6e 6e 65 63 74 .....max_connect
0470: 69 6f 6e 73 28 35 29 0a 09 09 09 09 2e 61 63 71 ions(5)......acq
0480: 75 69 72 65 5f 74 69 6d 65 6f 75 74 28 73 74 64 uire_timeout(std
0490: 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e ::time::Duration
04a0: 3a 3a 6e 65 77 28 33 30 30 2c 20 30 29 29 0a 09 ::new(300, 0))..
04b0: 09 09 09 2e 69 64 6c 65 5f 74 69 6d 65 6f 75 74 ....idle_timeout
04c0: 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 (std::time::Dura
04d0: 74 69 6f 6e 3a 3a 6e 65 77 28 36 30 2c 20 30 29 tion::new(60, 0)
04e0: 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 74 5f 6c )......connect_l
04f0: 61 7a 79 28 26 73 65 74 74 69 6e 67 73 2e 67 65 azy(&settings.ge
0500: 74 5f 73 74 72 69 6e 67 28 22 70 67 22 29 3f 29 t_string("pg")?)
0510: 3f 2c 0a 09 09 09 73 6f 75 72 63 65 73 3a 20 41 ?,....sources: A
0520: 72 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a 3a 6e rc::new(Mutex::n
0530: 65 77 28 48 61 73 68 53 65 74 3a 3a 6e 65 77 28 ew(HashSet::new(
0540: 29 29 29 2c 0a 09 09 09 68 74 74 70 5f 63 6c 69 ))),....http_cli
0550: 65 6e 74 2c 0a 09 09 7d 29 3b 0a 09 09 6c 65 74 ent,...});...let
0560: 20 63 6c 6f 6e 65 20 3d 20 63 6f 72 65 2e 63 6c clone = core.cl
0570: 6f 6e 65 28 29 3b 0a 09 09 74 61 73 6b 3a 3a 73 one();...task::s
0580: 70 61 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 pawn(async move
0590: 7b 0a 09 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 09 {....loop {.....
05a0: 6c 65 74 20 64 65 6c 61 79 20 3d 20 6d 61 74 63 let delay = matc
05b0: 68 20 26 63 6c 6f 6e 65 2e 61 75 74 6f 66 65 74 h &clone.autofet
05c0: 63 68 28 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 ch().await {....
05d0: 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a ..Err(err) => {.
05e0: 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 ......if let Err
05f0: 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 (err) = clone.se
0600: 6e 64 28 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 nd(format!("š
0610: 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f {:?}", err), No
0620: 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 74 20 ne, None).await
0630: 7b 0a 09 09 09 09 09 09 09 65 70 72 69 6e 74 6c {........eprintl
0640: 6e 21 28 22 41 75 74 6f 66 65 74 63 68 20 65 72 n!("Autofetch er
0650: 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a ror: {}", err);.
0660: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 73 ......};.......s
0670: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
0680: 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 63 73 28 36 30 on::from_secs(60
0690: 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 4f )......},......O
06a0: 6b 28 74 69 6d 65 29 20 3d 3e 20 2a 74 69 6d 65 k(time) => *time
06b0: 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 74 61 73 ,.....};.....tas
06c0: 6b 3a 3a 73 6c 65 65 70 28 64 65 6c 61 79 29 2e k::sleep(delay).
06d0: 61 77 61 69 74 3b 0a 09 09 09 7d 0a 09 09 7d 29 await;....}...})
06e0: 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 0a 09 7d 0a ;...Ok(core)..}.
06f0: 0a 09 70 75 62 20 66 6e 20 73 74 72 65 61 6d 28 ..pub fn stream(
0700: 26 73 65 6c 66 29 20 2d 3e 20 74 65 6c 65 67 72 &self) -> telegr
0710: 61 6d 5f 62 6f 74 3a 3a 55 70 64 61 74 65 73 53 am_bot::UpdatesS
0720: 74 72 65 61 6d 20 7b 0a 09 09 73 65 6c 66 2e 74 tream {...self.t
0730: 67 2e 73 74 72 65 61 6d 28 29 0a 09 7d 0a 0a 09 g.stream()..}...
0740: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 73 65 6e pub async fn sen
0750: 64 3c 27 61 2c 20 53 3e 28 26 73 65 6c 66 2c 20 d<'a, S>(&self,
0760: 6d 73 67 3a 20 53 2c 20 74 61 72 67 65 74 3a 20 msg: S, target:
0770: 4f 70 74 69 6f 6e 3c 74 65 6c 65 67 72 61 6d 5f Option<telegram_
0780: 62 6f 74 3a 3a 55 73 65 72 49 64 3e 2c 20 6d 6f bot::UserId>, mo
0790: 64 65 3a 20 4f 70 74 69 6f 6e 3c 74 65 6c 65 67 de: Option<teleg
07a0: 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a ram_bot::types::
07b0: 50 61 72 73 65 4d 6f 64 65 3e 29 20 2d 3e 20 52 ParseMode>) -> R
07c0: 65 73 75 6c 74 3c 28 29 3e 0a 09 77 68 65 72 65 esult<()>..where
07d0: 20 53 3a 20 49 6e 74 6f 3c 43 6f 77 3c 27 61 2c S: Into<Cow<'a,
07e0: 20 73 74 72 3e 3e 20 7b 0a 09 09 6c 65 74 20 6d str>> {...let m
07f0: 6f 64 65 20 3d 20 6d 6f 64 65 2e 75 6e 77 72 61 ode = mode.unwra
0800: 70 5f 6f 72 28 74 65 6c 65 67 72 61 6d 5f 62 6f p_or(telegram_bo
0810: 74 3a 3a 74 79 70 65 73 3a 3a 50 61 72 73 65 4d t::types::ParseM
0820: 6f 64 65 3a 3a 48 74 6d 6c 29 3b 0a 09 09 6c 65 ode::Html);...le
0830: 74 20 74 61 72 67 65 74 20 3d 20 74 61 72 67 65 t target = targe
0840: 74 2e 75 6e 77 72 61 70 5f 6f 72 28 73 65 6c 66 t.unwrap_or(self
0850: 2e 6f 77 6e 65 72 5f 63 68 61 74 29 3b 0a 09 09 .owner_chat);...
0860: 73 65 6c 66 2e 74 67 2e 73 65 6e 64 28 74 65 6c self.tg.send(tel
0870: 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d egram_bot::SendM
0880: 65 73 73 61 67 65 3a 3a 6e 65 77 28 74 61 72 67 essage::new(targ
0890: 65 74 2c 20 6d 73 67 29 2e 70 61 72 73 65 5f 6d et, msg).parse_m
08a0: 6f 64 65 28 6d 6f 64 65 29 29 2e 61 77 61 69 74 ode(mode)).await
08b0: 3f 3b 0a 09 09 4f 6b 28 28 29 29 0a 09 7d 0a 0a ?;...Ok(())..}..
08c0: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 63 68 .pub async fn ch
08d0: 65 63 6b 3c 53 3e 28 26 73 65 6c 66 2c 20 69 64 eck<S>(&self, id
08e0: 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 : &i32, owner: S
08f0: 2c 20 72 65 61 6c 3a 20 62 6f 6f 6c 29 20 2d 3e , real: bool) ->
0900: 20 52 65 73 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 Result<Cow<'_,
0910: 73 74 72 3e 3e 0a 09 77 68 65 72 65 20 53 3a 20 str>>..where S:
0920: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
0930: 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e t owner = owner.
0940: 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d into();....let m
0950: 75 74 20 70 6f 73 74 65 64 3a 20 69 33 32 20 3d ut posted: i32 =
0960: 20 30 3b 0a 09 09 6c 65 74 20 69 64 20 3d 20 7b 0;...let id = {
0970: 0a 09 09 09 6c 65 74 20 6d 75 74 20 73 65 74 20 ....let mut set
0980: 3d 20 73 65 6c 66 2e 73 6f 75 72 63 65 73 2e 6c = self.sources.l
0990: 6f 63 6b 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a ock().unwrap();.
09a0: 09 09 09 6d 61 74 63 68 20 73 65 74 2e 67 65 74 ...match set.get
09b0: 28 69 64 29 20 7b 0a 09 09 09 09 53 6f 6d 65 28 (id) {.....Some(
09c0: 69 64 29 20 3d 3e 20 69 64 2e 63 6c 6f 6e 65 28 id) => id.clone(
09d0: 29 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b ),.....None => {
09e0: 0a 09 09 09 09 09 6c 65 74 20 69 64 20 3d 20 41 ......let id = A
09f0: 72 63 3a 3a 6e 65 77 28 2a 69 64 29 3b 0a 09 09 rc::new(*id);...
0a00: 09 09 09 73 65 74 2e 69 6e 73 65 72 74 28 69 64 ...set.insert(id
0a10: 2e 63 6c 6f 6e 65 28 29 29 3b 0a 09 09 09 09 09 .clone());......
0a20: 69 64 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 7d id.clone().....}
0a30: 2c 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 6c 65 ,....}...};...le
0a40: 74 20 63 6f 75 6e 74 20 3d 20 41 72 63 3a 3a 73 t count = Arc::s
0a50: 74 72 6f 6e 67 5f 63 6f 75 6e 74 28 26 69 64 29 trong_count(&id)
0a60: 3b 0a 09 09 69 66 20 63 6f 75 6e 74 20 3d 3d 20 ;...if count ==
0a70: 32 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 63 2 {....let mut c
0a80: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e onn = self.pool.
0a90: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a acquire().await.
0aa0: 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
0ab0: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 51 75 t(|| format!("Qu
0ac0: 65 72 79 20 71 75 65 75 65 20 66 65 74 63 68 20 ery queue fetch
0ad0: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
0ae0: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 elf.pool))?;....
0af0: 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c 78 3a 3a let row = sqlx::
0b00: 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f query("select so
0b10: 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c urce_id, channel
0b20: 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 _id, url, iv_has
0b30: 68 2c 20 6f 77 6e 65 72 2c 20 75 72 6c 5f 72 65 h, owner, url_re
0b40: 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 from rsstg_sour
0b50: 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f ce where source_
0b60: 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 id = $1 and owne
0b70: 72 20 3d 20 24 32 22 29 0a 09 09 09 09 2e 62 69 r = $2")......bi
0b80: 6e 64 28 2a 69 64 29 0a 09 09 09 09 2e 62 69 6e nd(*id)......bin
0b90: 64 28 6f 77 6e 65 72 29 0a 09 09 09 09 2e 66 65 d(owner)......fe
0ba0: 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 63 6f 6e tch_one(&mut con
0bb0: 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 n).await......wi
0bc0: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
0bd0: 72 6d 61 74 21 28 22 51 75 65 72 79 20 73 6f 75 rmat!("Query sou
0be0: 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 rce:\n{:?}", &se
0bf0: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 64 lf.pool))?;....d
0c00: 72 6f 70 28 63 6f 6e 6e 29 3b 0a 0a 09 09 09 6c rop(conn);.....l
0c10: 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 et channel_id: i
0c20: 36 34 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 64 = row.try_get
0c30: 28 22 63 68 61 6e 6e 65 6c 5f 69 64 22 29 3f 3b ("channel_id")?;
0c40: 0a 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 73 74 ....let url: &st
0c50: 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 r = row.try_get(
0c60: 22 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 "url")?;....let
0c70: 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c iv_hash: Option<
0c80: 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f &str> = row.try_
0c90: 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f 3b get("iv_hash")?;
0ca0: 0a 09 09 09 6c 65 74 20 75 72 6c 5f 72 65 20 3d ....let url_re =
0cb0: 20 6d 61 74 63 68 20 72 6f 77 2e 74 72 79 5f 67 match row.try_g
0cc0: 65 74 28 22 75 72 6c 5f 72 65 22 29 3f 20 7b 0a et("url_re")? {.
0cd0: 09 09 09 09 53 6f 6d 65 28 78 29 20 3d 3e 20 53 ....Some(x) => S
0ce0: 6f 6d 65 28 73 65 64 72 65 67 65 78 3a 3a 52 65 ome(sedregex::Re
0cf0: 70 6c 61 63 65 43 6f 6d 6d 61 6e 64 3a 3a 6e 65 placeCommand::ne
0d00: 77 28 78 29 3f 29 2c 0a 09 09 09 09 4e 6f 6e 65 w(x)?),.....None
0d10: 20 3d 3e 20 4e 6f 6e 65 2c 0a 09 09 09 7d 3b 0a => None,....};.
0d20: 09 09 09 6c 65 74 20 64 65 73 74 69 6e 61 74 69 ...let destinati
0d30: 6f 6e 20 3d 20 6d 61 74 63 68 20 72 65 61 6c 20 on = match real
0d40: 7b 0a 09 09 09 09 74 72 75 65 20 3d 3e 20 74 65 {.....true => te
0d50: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 legram_bot::User
0d60: 49 64 3a 3a 6e 65 77 28 63 68 61 6e 6e 65 6c 5f Id::new(channel_
0d70: 69 64 29 2c 0a 09 09 09 09 66 61 6c 73 65 20 3d id),.....false =
0d80: 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a > telegram_bot::
0d90: 55 73 65 72 49 64 3a 3a 6e 65 77 28 72 6f 77 2e UserId::new(row.
0da0: 74 72 79 5f 67 65 74 28 22 6f 77 6e 65 72 22 29 try_get("owner")
0db0: 3f 29 2c 0a 09 09 09 7d 3b 0a 09 09 09 6c 65 74 ?),....};....let
0dc0: 20 6d 75 74 20 74 68 69 73 5f 66 65 74 63 68 3a mut this_fetch:
0dd0: 20 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 6d 65 Option<DateTime
0de0: 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 <chrono::FixedOf
0df0: 66 73 65 74 3e 3e 20 3d 20 4e 6f 6e 65 3b 0a 09 fset>> = None;..
0e00: 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 73 3a ..let mut posts:
0e10: 20 42 54 72 65 65 4d 61 70 3c 44 61 74 65 54 69 BTreeMap<DateTi
0e20: 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 me<chrono::Fixed
0e30: 4f 66 66 73 65 74 3e 2c 20 53 74 72 69 6e 67 3e Offset>, String>
0e40: 20 3d 20 42 54 72 65 65 4d 61 70 3a 3a 6e 65 77 = BTreeMap::new
0e50: 28 29 3b 0a 0a 09 09 09 6c 65 74 20 72 65 73 70 ();.....let resp
0e60: 6f 6e 73 65 20 3d 20 73 65 6c 66 2e 68 74 74 70 onse = self.http
0e70: 5f 63 6c 69 65 6e 74 2e 67 65 74 28 75 72 6c 29 _client.get(url)
0e80: 2e 73 65 6e 64 28 29 2e 61 77 61 69 74 3f 3b 0a .send().await?;.
0e90: 09 09 09 6c 65 74 20 73 74 61 74 75 73 20 3d 20 ...let status =
0ea0: 72 65 73 70 6f 6e 73 65 2e 73 74 61 74 75 73 28 response.status(
0eb0: 29 3b 0a 09 09 09 6c 65 74 20 63 6f 6e 74 65 6e );....let conten
0ec0: 74 20 3d 20 72 65 73 70 6f 6e 73 65 2e 62 79 74 t = response.byt
0ed0: 65 73 28 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 es().await?;....
0ee0: 6d 61 74 63 68 20 72 73 73 3a 3a 43 68 61 6e 6e match rss::Chann
0ef0: 65 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 el::read_from(&c
0f00: 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 09 ontent[..]) {...
0f10: 09 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 7b 0a ..Ok(feed) => {.
0f20: 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 69 6e .....for item in
0f30: 20 66 65 65 64 2e 69 74 65 6d 73 28 29 20 7b 0a feed.items() {.
0f40: 09 09 09 09 09 09 69 66 20 6c 65 74 20 53 6f 6d ......if let Som
0f50: 65 28 6c 69 6e 6b 29 20 3d 20 69 74 65 6d 2e 6c e(link) = item.l
0f60: 69 6e 6b 28 29 20 7b 0a 09 09 09 09 09 09 09 6c ink() {........l
0f70: 65 74 20 64 61 74 65 20 3d 20 6d 61 74 63 68 20 et date = match
0f80: 69 74 65 6d 2e 70 75 62 5f 64 61 74 65 28 29 20 item.pub_date()
0f90: 7b 0a 09 09 09 09 09 09 09 09 53 6f 6d 65 28 66 {.........Some(f
0fa0: 65 65 64 5f 64 61 74 65 29 20 3d 3e 20 44 61 74 eed_date) => Dat
0fb0: 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f eTime::parse_fro
0fc0: 6d 5f 72 66 63 32 38 32 32 28 66 65 65 64 5f 64 m_rfc2822(feed_d
0fd0: 61 74 65 29 2c 0a 09 09 09 09 09 09 09 09 4e 6f ate),.........No
0fe0: 6e 65 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a ne => DateTime::
0ff0: 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 33 33 parse_from_rfc33
1000: 33 39 28 26 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 39(&item.dublin_
1010: 63 6f 72 65 5f 65 78 74 28 29 2e 75 6e 77 72 61 core_ext().unwra
1020: 70 28 29 2e 64 61 74 65 73 28 29 5b 30 5d 29 2c p().dates()[0]),
1030: 0a 09 09 09 09 09 09 09 7d 3f 3b 0a 09 09 09 09 ........}?;.....
1040: 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 6c 69 6e ...let url = lin
1050: 6b 3b 0a 09 09 09 09 09 09 09 70 6f 73 74 73 2e k;........posts.
1060: 69 6e 73 65 72 74 28 64 61 74 65 2c 20 75 72 6c insert(date, url
1070: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 .to_string());..
1080: 09 09 09 09 09 7d 0a 09 09 09 09 09 7d 3b 0a 09 .....}......};..
1090: 09 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 65 72 ...},.....Err(er
10a0: 72 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 72 20 r) => match err
10b0: 7b 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 6f {......rss::Erro
10c0: 72 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 74 54 r::InvalidStartT
10d0: 61 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 6c 65 ag => {.......le
10e0: 74 20 66 65 65 64 20 3d 20 61 74 6f 6d 5f 73 79 t feed = atom_sy
10f0: 6e 64 69 63 61 74 69 6f 6e 3a 3a 46 65 65 64 3a ndication::Feed:
1100: 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 :read_from(&cont
1110: 65 6e 74 5b 2e 2e 5d 29 0a 09 09 09 09 09 09 09 ent[..])........
1120: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
1130: 20 66 6f 72 6d 61 74 21 28 22 50 72 6f 62 6c 65 format!("Proble
1140: 6d 20 6f 70 65 6e 69 6e 67 20 66 65 65 64 20 75 m opening feed u
1150: 72 6c 3a 5c 6e 7b 7d 5c 6e 7b 7d 22 2c 20 26 75 rl:\n{}\n{}", &u
1160: 72 6c 2c 20 73 74 61 74 75 73 29 29 3f 3b 0a 09 rl, status))?;..
1170: 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 69 6e .....for item in
1180: 20 66 65 65 64 2e 65 6e 74 72 69 65 73 28 29 20 feed.entries()
1190: 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 64 61 74 {........let dat
11a0: 65 20 3d 20 69 74 65 6d 2e 70 75 62 6c 69 73 68 e = item.publish
11b0: 65 64 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 ed().unwrap();..
11c0: 09 09 09 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 ......let url =
11d0: 69 74 65 6d 2e 6c 69 6e 6b 73 28 29 5b 30 5d 2e item.links()[0].
11e0: 68 72 65 66 28 29 3b 0a 09 09 09 09 09 09 09 70 href();........p
11f0: 6f 73 74 73 2e 69 6e 73 65 72 74 28 2a 64 61 74 osts.insert(*dat
1200: 65 2c 20 75 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 e, url.to_string
1210: 28 29 29 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 ());.......};...
1220: 09 09 09 7d 2c 0a 09 09 09 09 09 72 73 73 3a 3a ...},......rss::
1230: 45 72 72 6f 72 3a 3a 45 6f 66 20 3d 3e 20 28 29 Error::Eof => ()
1240: 2c 0a 09 09 09 09 09 5f 20 3d 3e 20 62 61 69 6c ,......_ => bail
1250: 21 28 22 55 6e 73 75 70 70 6f 72 74 65 64 20 6f !("Unsupported o
1260: 72 20 6d 61 6e 67 6c 65 64 20 63 6f 6e 74 65 6e r mangled conten
1270: 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b 3a 23 3f 7d 5c t:\n{:?}\n{:#?}\
1280: 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 26 75 72 6c 2c n{:#?}\n", &url,
1290: 20 65 72 72 2c 20 73 74 61 74 75 73 29 0a 09 09 err, status)...
12a0: 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 66 6f 72 ..}....};....for
12b0: 20 28 64 61 74 65 2c 20 75 72 6c 29 20 69 6e 20 (date, url) in
12c0: 70 6f 73 74 73 2e 69 74 65 72 28 29 20 7b 0a 09 posts.iter() {..
12d0: 09 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
12e0: 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 = self.pool.acqu
12f0: 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 09 ire().await.....
1300: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
1310: 7c 20 66 6f 72 6d 61 74 21 28 22 43 68 65 63 6b | format!("Check
1320: 20 70 6f 73 74 20 66 65 74 63 68 20 63 6f 6e 6e post fetch conn
1330: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1340: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 09 6c 65 74 pool))?;.....let
1350: 20 70 6f 73 74 5f 75 72 6c 3a 20 43 6f 77 3c 73 post_url: Cow<s
1360: 74 72 3e 20 3d 20 6d 61 74 63 68 20 75 72 6c 5f tr> = match url_
1370: 72 65 20 7b 0a 09 09 09 09 09 53 6f 6d 65 28 72 re {......Some(r
1380: 65 66 20 78 29 20 3d 3e 20 78 2e 65 78 65 63 75 ef x) => x.execu
1390: 74 65 28 75 72 6c 29 2c 0a 09 09 09 09 09 4e 6f te(url),......No
13a0: 6e 65 20 3d 3e 20 75 72 6c 2e 69 6e 74 6f 28 29 ne => url.into()
13b0: 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 6c 65 74 ,.....};.....let
13c0: 20 72 6f 77 20 3d 20 73 71 6c 78 3a 3a 71 75 65 row = sqlx::que
13d0: 72 79 28 22 73 65 6c 65 63 74 20 65 78 69 73 74 ry("select exist
13e0: 73 28 73 65 6c 65 63 74 20 74 72 75 65 20 66 72 s(select true fr
13f0: 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 77 68 om rsstg_post wh
1400: 65 72 65 20 75 72 6c 20 3d 20 24 31 20 61 6e 64 ere url = $1 and
1410: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 32 29 source_id = $2)
1420: 20 61 73 20 65 78 69 73 74 73 3b 22 29 0a 09 09 as exists;")...
1430: 09 09 09 2e 62 69 6e 64 28 26 2a 70 6f 73 74 5f ....bind(&*post_
1440: 75 72 6c 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 url).......bind(
1450: 2a 69 64 29 0a 09 09 09 09 09 2e 66 65 74 63 68 *id).......fetch
1460: 5f 6f 6e 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e _one(&mut conn).
1470: 61 77 61 69 74 0a 09 09 09 09 09 2e 77 69 74 68 await.......with
1480: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
1490: 61 74 21 28 22 43 68 65 63 6b 20 70 6f 73 74 3a at!("Check post:
14a0: 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 \n{:?}", &conn))
14b0: 3f 3b 0a 09 09 09 09 6c 65 74 20 65 78 69 73 74 ?;.....let exist
14c0: 73 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 s: bool = row.tr
14d0: 79 5f 67 65 74 28 22 65 78 69 73 74 73 22 29 3f y_get("exists")?
14e0: 3b 0a 09 09 09 09 69 66 20 21 20 65 78 69 73 74 ;.....if ! exist
14f0: 73 20 7b 0a 09 09 09 09 09 69 66 20 74 68 69 73 s {......if this
1500: 5f 66 65 74 63 68 2e 69 73 5f 6e 6f 6e 65 28 29 _fetch.is_none()
1510: 20 7c 7c 20 2a 64 61 74 65 20 3e 20 74 68 69 73 || *date > this
1520: 5f 66 65 74 63 68 2e 75 6e 77 72 61 70 28 29 20 _fetch.unwrap()
1530: 7b 0a 09 09 09 09 09 09 74 68 69 73 5f 66 65 74 {.......this_fet
1540: 63 68 20 3d 20 53 6f 6d 65 28 2a 64 61 74 65 29 ch = Some(*date)
1550: 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 73 ;......};......s
1560: 65 6c 66 2e 74 67 2e 73 65 6e 64 28 20 6d 61 74 elf.tg.send( mat
1570: 63 68 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 ch iv_hash {....
1580: 09 09 09 09 53 6f 6d 65 28 68 61 73 68 29 20 3d ....Some(hash) =
1590: 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a > telegram_bot::
15a0: 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 SendMessage::new
15b0: 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f (destination, fo
15c0: 72 6d 61 74 21 28 22 3c 61 20 68 72 65 66 3d 5c rmat!("<a href=\
15d0: 22 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 "https://t.me/iv
15e0: 3f 75 72 6c 3d 7b 7d 26 72 68 61 73 68 3d 7b 7d ?url={}&rhash={}
15f0: 5c 22 3e 20 3c 2f 61 3e 7b 30 7d 22 2c 20 26 70 \"> </a>{0}", &p
1600: 6f 73 74 5f 75 72 6c 2c 20 68 61 73 68 29 29 2c ost_url, hash)),
1610: 0a 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 ........None =>
1620: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 telegram_bot::Se
1630: 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 ndMessage::new(d
1640: 65 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d estination, form
1650: 61 74 21 28 22 7b 7d 22 2c 20 70 6f 73 74 5f 75 at!("{}", post_u
1660: 72 6c 29 29 2c 0a 09 09 09 09 09 09 7d 2e 70 61 rl)),.......}.pa
1670: 72 73 65 5f 6d 6f 64 65 28 74 65 6c 65 67 72 61 rse_mode(telegra
1680: 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 61 m_bot::types::Pa
1690: 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 2e rseMode::Html)).
16a0: 61 77 61 69 74 0a 09 09 09 09 09 09 2e 63 6f 6e await........con
16b0: 74 65 78 74 28 22 43 61 6e 27 74 20 70 6f 73 74 text("Can't post
16c0: 20 6d 65 73 73 61 67 65 3a 22 29 3f 3b 0a 09 09 message:")?;...
16d0: 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ...sqlx::query("
16e0: 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 insert into rsst
16f0: 67 5f 70 6f 73 74 20 28 73 6f 75 72 63 65 5f 69 g_post (source_i
1700: 64 2c 20 70 6f 73 74 65 64 2c 20 75 72 6c 29 20 d, posted, url)
1710: 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 values ($1, $2,
1720: 24 33 29 3b 22 29 0a 09 09 09 09 09 09 2e 62 69 $3);")........bi
1730: 6e 64 28 2a 69 64 29 0a 09 09 09 09 09 09 2e 62 nd(*id)........b
1740: 69 6e 64 28 64 61 74 65 29 0a 09 09 09 09 09 09 ind(date).......
1750: 2e 62 69 6e 64 28 26 2a 70 6f 73 74 5f 75 72 6c .bind(&*post_url
1760: 29 0a 09 09 09 09 09 09 2e 65 78 65 63 75 74 65 )........execute
1770: 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 (&mut conn).awai
1780: 74 0a 09 09 09 09 09 09 2e 77 69 74 68 5f 63 6f t........with_co
1790: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
17a0: 28 22 52 65 63 6f 72 64 20 70 6f 73 74 3a 5c 6e ("Record post:\n
17b0: 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f 3b {:?}", &conn))?;
17c0: 0a 09 09 09 09 09 64 72 6f 70 28 63 6f 6e 6e 29 ......drop(conn)
17d0: 3b 0a 09 09 09 09 09 74 61 73 6b 3a 3a 73 6c 65 ;......task::sle
17e0: 65 70 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 ep(std::time::Du
17f0: 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 34 2c 20 30 ration::new(4, 0
1800: 29 29 2e 61 77 61 69 74 3b 0a 09 09 09 09 7d 3b )).await;.....};
1810: 0a 09 09 09 09 70 6f 73 74 65 64 20 2b 3d 20 31 .....posted += 1
1820: 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 6f 73 74 73 ;....};....posts
1830: 2e 63 6c 65 61 72 28 29 3b 0a 09 09 7d 3b 0a 09 .clear();...};..
1840: 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 .let mut conn =
1850: 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 self.pool.acquir
1860: 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 e().await.....wi
1870: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
1880: 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 73 63 rmat!("Update sc
1890: 72 61 70 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a rape fetch conn:
18a0: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
18b0: 6f 6f 6c 29 29 3f 3b 0a 09 09 73 71 6c 78 3a 3a ool))?;...sqlx::
18c0: 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 73 query("update rs
18d0: 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 6c stg_source set l
18e0: 61 73 74 5f 73 63 72 61 70 65 20 3d 20 6e 6f 77 ast_scrape = now
18f0: 28 29 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f () where source_
1900: 69 64 20 3d 20 24 31 3b 22 29 0a 09 09 09 2e 62 id = $1;").....b
1910: 69 6e 64 28 2a 69 64 29 0a 09 09 09 2e 65 78 65 ind(*id).....exe
1920: 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e cute(&mut conn).
1930: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
1940: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1950: 21 28 22 55 70 64 61 74 65 20 73 63 72 61 70 65 !("Update scrape
1960: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 :\n{:?}", &conn)
1970: 29 3f 3b 0a 09 09 4f 6b 28 66 6f 72 6d 61 74 21 )?;...Ok(format!
1980: 28 22 50 6f 73 74 65 64 3a 20 7b 7d 22 2c 20 26 ("Posted: {}", &
1990: 70 6f 73 74 65 64 29 2e 69 6e 74 6f 28 29 29 0a posted).into()).
19a0: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
19b0: 6e 20 64 65 6c 65 74 65 3c 53 3e 28 26 73 65 6c n delete<S>(&sel
19c0: 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 f, source_id: &i
19d0: 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 32, owner: S) ->
19e0: 20 52 65 73 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 Result<Cow<'_,
19f0: 73 74 72 3e 3e 0a 09 77 68 65 72 65 20 53 3a 20 str>>..where S:
1a00: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
1a10: 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e t owner = owner.
1a20: 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d into();....let m
1a30: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 ut conn = self.p
1a40: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
1a50: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
1a60: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1a70: 22 44 65 6c 65 74 65 20 66 65 74 63 68 20 63 6f "Delete fetch co
1a80: 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c nn:\n{:?}", &sel
1a90: 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 f.pool))?;...mat
1aa0: 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ch sqlx::query("
1ab0: 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 74 delete from rsst
1ac0: 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 73 g_source where s
1ad0: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e ource_id = $1 an
1ae0: 64 20 6f 77 6e 65 72 20 3d 20 24 32 3b 22 29 0a d owner = $2;").
1af0: 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f ....bind(source_
1b00: 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e id).....bind(own
1b10: 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 er).....execute(
1b20: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
1b30: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
1b40: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 65 t(|| format!("De
1b50: 6c 65 74 65 20 73 6f 75 72 63 65 20 72 75 6c 65 lete source rule
1b60: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1b70: 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 pool))?.....rows
1b80: 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 _affected() {...
1b90: 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 .0 => { Ok("No d
1ba0: 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 2e ata found found.
1bb0: 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 09 ".into()) },....
1bc0: 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 x => { Ok(format
1bd0: 21 28 22 7b 7d 20 73 6f 75 72 63 65 73 20 72 65 !("{} sources re
1be0: 6d 6f 76 65 64 2e 22 2c 20 78 29 2e 69 6e 74 6f moved.", x).into
1bf0: 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 ()) },...}..}...
1c00: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 63 6c 65 pub async fn cle
1c10: 61 6e 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 an<S>(&self, sou
1c20: 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 rce_id: &i32, ow
1c30: 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c ner: S) -> Resul
1c40: 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a t<Cow<'_, str>>.
1c50: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
1c60: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
1c70: 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 r = owner.into()
1c80: 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e ;....let mut con
1c90: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
1ca0: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
1cb0: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
1cc0: 7c 20 66 6f 72 6d 61 74 21 28 22 43 6c 65 61 6e | format!("Clean
1cd0: 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a fetch conn:\n{:
1ce0: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
1cf0: 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 )?;...match sqlx
1d00: 3a 3a 71 75 65 72 79 28 22 64 65 6c 65 74 65 20 ::query("delete
1d10: 66 72 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 from rsstg_post
1d20: 70 20 75 73 69 6e 67 20 72 73 73 74 67 5f 73 6f p using rsstg_so
1d30: 75 72 63 65 20 73 20 77 68 65 72 65 20 70 2e 73 urce s where p.s
1d40: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e ource_id = $1 an
1d50: 64 20 6f 77 6e 65 72 20 3d 20 24 32 20 61 6e 64 d owner = $2 and
1d60: 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 73 p.source_id = s
1d70: 2e 73 6f 75 72 63 65 5f 69 64 3b 22 29 0a 09 09 .source_id;")...
1d80: 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 ..bind(source_id
1d90: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
1da0: 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d ).....execute(&m
1db0: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 ut conn).await..
1dc0: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1dd0: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 6c 65 61 || format!("Clea
1de0: 6e 20 73 65 65 6e 20 70 6f 73 74 73 3a 5c 6e 7b n seen posts:\n{
1df0: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
1e00: 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 ))?.....rows_aff
1e10: 65 63 74 65 64 28 29 20 7b 0a 09 09 09 30 20 3d ected() {....0 =
1e20: 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 61 74 61 20 > { Ok("No data
1e30: 66 6f 75 6e 64 20 66 6f 75 6e 64 2e 22 2e 69 6e found found.".in
1e40: 74 6f 28 29 29 20 7d 2c 0a 09 09 09 78 20 3d 3e to()) },....x =>
1e50: 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 22 7b { Ok(format!("{
1e60: 7d 20 70 6f 73 74 73 20 70 75 72 67 65 64 2e 22 } posts purged."
1e70: 2c 20 78 29 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a , x).into()) },.
1e80: 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ..}..}...pub asy
1e90: 6e 63 20 66 6e 20 65 6e 61 62 6c 65 3c 53 3e 28 nc fn enable<S>(
1ea0: 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 &self, source_id
1eb0: 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 : &i32, owner: S
1ec0: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 ) -> Result<&str
1ed0: 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f >..where S: Into
1ee0: 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 <i64> {...let ow
1ef0: 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f ner = owner.into
1f00: 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 ();....let mut c
1f10: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e onn = self.pool.
1f20: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a acquire().await.
1f30: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
1f40: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 45 6e 61 (|| format!("Ena
1f50: 62 6c 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c ble fetch conn:\
1f60: 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f n{:?}", &self.po
1f70: 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 ol))?;...match s
1f80: 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 qlx::query("upda
1f90: 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 te rsstg_source
1fa0: 73 65 74 20 65 6e 61 62 6c 65 64 20 3d 20 74 72 set enabled = tr
1fb0: 75 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f ue where source_
1fc0: 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 id = $1 and owne
1fd0: 72 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e r = $2").....bin
1fe0: 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 d(source_id)....
1ff0: 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 .bind(owner)....
2000: 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f .execute(&mut co
2010: 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 nn).await.....wi
2020: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
2030: 72 6d 61 74 21 28 22 45 6e 61 62 6c 65 20 73 6f rmat!("Enable so
2040: 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 urce:\n{:?}", &s
2050: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e elf.pool))?.....
2060: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 rows_affected()
2070: 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 {....1 => { Ok("
2080: 53 6f 75 72 63 65 20 65 6e 61 62 6c 65 64 2e 22 Source enabled."
2090: 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f ) },....0 => { O
20a0: 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f k("Source not fo
20b0: 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d und.") },...._ =
20c0: 3e 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 > { Err(anyhow!(
20d0: 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e "Database error.
20e0: 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 ")) },...}..}...
20f0: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 64 69 73 pub async fn dis
2100: 61 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 able<S>(&self, s
2110: 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 ource_id: &i32,
2120: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
2130: 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 ult<&str>..where
2140: 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a S: Into<i64> {.
2150: 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 ..let owner = ow
2160: 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c ner.into();....l
2170: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
2180: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
2190: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
21a0: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
21b0: 61 74 21 28 22 44 69 73 61 62 6c 65 20 66 65 74 at!("Disable fet
21c0: 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c ch conn:\n{:?}",
21d0: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a &self.pool))?;.
21e0: 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 ..match sqlx::qu
21f0: 65 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 ery("update rsst
2200: 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 6e 61 g_source set ena
2210: 62 6c 65 64 20 3d 20 66 61 6c 73 65 20 77 68 65 bled = false whe
2220: 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 re source_id = $
2230: 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 1 and owner = $2
2240: 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 ").....bind(sour
2250: 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 ce_id).....bind(
2260: 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 owner).....execu
2270: 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 te(&mut conn).aw
2280: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
2290: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
22a0: 22 44 69 73 61 62 6c 65 20 73 6f 75 72 63 65 3a "Disable source:
22b0: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
22c0: 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f ool))?.....rows_
22d0: 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 affected() {....
22e0: 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 1 => { Ok("Sourc
22f0: 65 20 64 69 73 61 62 6c 65 64 2e 22 29 20 7d 2c e disabled.") },
2300: 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 ....0 => { Ok("S
2310: 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 2e ource not found.
2320: 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 ") },...._ => {
2330: 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 61 74 Err(anyhow!("Dat
2340: 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 29 20 abase error."))
2350: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 },...}..}...pub
2360: 61 73 79 6e 63 20 66 6e 20 75 70 64 61 74 65 3c async fn update<
2370: 53 3e 28 26 73 65 6c 66 2c 20 75 70 64 61 74 65 S>(&self, update
2380: 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c 20 63 : Option<i32>, c
2390: 68 61 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 63 68 hannel: &str, ch
23a0: 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c 20 75 annel_id: i64, u
23b0: 72 6c 3a 20 26 73 74 72 2c 20 69 76 5f 68 61 73 rl: &str, iv_has
23c0: 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c h: Option<&str>,
23d0: 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f 6e 3c url_re: Option<
23e0: 26 73 74 72 3e 2c 20 6f 77 6e 65 72 3a 20 53 29 &str>, owner: S)
23f0: 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e -> Result<&str>
2400: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
2410: 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e i64> {...let own
2420: 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 er = owner.into(
2430: 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f );....let mut co
2440: 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 nn = self.pool.a
2450: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 cquire().await..
2460: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
2470: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 || format!("Upda
2480: 74 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e te fetch conn:\n
2490: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
24a0: 6c 29 29 3f 3b 0a 0a 09 09 6d 61 74 63 68 20 6d l))?;....match m
24b0: 61 74 63 68 20 75 70 64 61 74 65 20 7b 0a 09 09 atch update {...
24c0: 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 7b 0a ..Some(id) => {.
24d0: 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 .....sqlx::query
24e0: 28 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 ("update rsstg_s
24f0: 6f 75 72 63 65 20 73 65 74 20 63 68 61 6e 6e 65 ource set channe
2500: 6c 5f 69 64 20 3d 20 24 32 2c 20 75 72 6c 20 3d l_id = $2, url =
2510: 20 24 33 2c 20 69 76 5f 68 61 73 68 20 3d 20 24 $3, iv_hash = $
2520: 34 2c 20 6f 77 6e 65 72 20 3d 20 24 35 2c 20 63 4, owner = $5, c
2530: 68 61 6e 6e 65 6c 20 3d 20 24 36 2c 20 75 72 6c hannel = $6, url
2540: 5f 72 65 20 3d 20 24 37 20 77 68 65 72 65 20 73 _re = $7 where s
2550: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 22 29 2e ource_id = $1").
2560: 62 69 6e 64 28 69 64 29 0a 09 09 09 09 7d 2c 0a bind(id).....},.
2570: 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 ....None => {...
2580: 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ...sqlx::query("
2590: 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 insert into rsst
25a0: 67 5f 73 6f 75 72 63 65 20 28 63 68 61 6e 6e 65 g_source (channe
25b0: 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 l_id, url, iv_ha
25c0: 73 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 6e 6e sh, owner, chann
25d0: 65 6c 2c 20 75 72 6c 5f 72 65 29 20 76 61 6c 75 el, url_re) valu
25e0: 65 73 20 28 24 31 2c 20 24 32 2c 20 24 33 2c 20 es ($1, $2, $3,
25f0: 24 34 2c 20 24 35 2c 20 24 36 29 22 29 0a 09 09 $4, $5, $6)")...
2600: 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 09 2e 62 69 ..},....}.....bi
2610: 6e 64 28 63 68 61 6e 6e 65 6c 5f 69 64 29 0a 09 nd(channel_id)..
2620: 09 09 2e 62 69 6e 64 28 75 72 6c 29 0a 09 09 09 ...bind(url)....
2630: 2e 62 69 6e 64 28 69 76 5f 68 61 73 68 29 0a 09 .bind(iv_hash)..
2640: 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 ...bind(owner)..
2650: 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c 29 ...bind(channel)
2660: 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c 5f 72 65 .....bind(url_re
2670: 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d ).....execute(&m
2680: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 20 7b ut conn).await {
2690: 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 4f 6b 28 ....Ok(_) => Ok(
26a0: 6d 61 74 63 68 20 75 70 64 61 74 65 20 7b 0a 09 match update {..
26b0: 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 22 43 ...Some(_) => "C
26c0: 68 61 6e 6e 65 6c 20 75 70 64 61 74 65 64 2e 22 hannel updated."
26d0: 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 22 43 ,.....None => "C
26e0: 68 61 6e 6e 65 6c 20 61 64 64 65 64 2e 22 2c 0a hannel added.",.
26f0: 09 09 09 7d 29 2c 0a 09 09 09 45 72 72 28 73 71 ...}),....Err(sq
2700: 6c 78 3a 3a 45 72 72 6f 72 3a 3a 44 61 74 61 62 lx::Error::Datab
2710: 61 73 65 28 65 72 72 29 29 20 3d 3e 20 7b 0a 09 ase(err)) => {..
2720: 09 09 09 6d 61 74 63 68 20 65 72 72 2e 64 6f 77 ...match err.dow
2730: 6e 63 61 73 74 3a 3a 3c 73 71 6c 78 3a 3a 70 6f ncast::<sqlx::po
2740: 73 74 67 72 65 73 3a 3a 50 67 44 61 74 61 62 61 stgres::PgDataba
2750: 73 65 45 72 72 6f 72 3e 28 29 2e 72 6f 75 74 69 seError>().routi
2760: 6e 65 28 29 20 7b 0a 09 09 09 09 09 53 6f 6d 65 ne() {......Some
2770: 28 22 5f 62 74 5f 63 68 65 63 6b 5f 75 6e 69 71 ("_bt_check_uniq
2780: 75 65 22 2c 20 29 20 3d 3e 20 7b 0a 09 09 09 09 ue", ) => {.....
2790: 09 09 4f 6b 28 22 44 75 70 6c 69 63 61 74 65 20 ..Ok("Duplicate
27a0: 6b 65 79 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 key.")......},..
27b0: 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 7b ....Some(_) => {
27c0: 0a 09 09 09 09 09 09 4f 6b 28 22 44 61 74 61 62 .......Ok("Datab
27d0: 61 73 65 20 65 72 72 6f 72 2e 22 29 0a 09 09 09 ase error.")....
27e0: 09 09 7d 2c 0a 09 09 09 09 09 4e 6f 6e 65 20 3d ..},......None =
27f0: 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 4e 6f > {.......Ok("No
2800: 20 64 61 74 61 62 61 73 65 20 65 72 72 6f 72 20 database error
2810: 65 78 74 72 61 63 74 65 64 2e 22 29 0a 09 09 09 extracted.")....
2820: 09 09 7d 2c 0a 09 09 09 09 7d 0a 09 09 09 7d 2c ..},.....}....},
2830: 0a 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 ....Err(err) =>
2840: 7b 0a 09 09 09 09 62 61 69 6c 21 28 22 53 6f 72 {.....bail!("Sor
2850: 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 72 72 6f ry, unknown erro
2860: 72 3a 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 65 72 r:\n{:#?}\n", er
2870: 72 29 3b 0a 09 09 09 7d 2c 0a 09 09 7d 0a 09 7d r);....},...}..}
2880: 0a 0a 09 61 73 79 6e 63 20 66 6e 20 61 75 74 6f ...async fn auto
2890: 66 65 74 63 68 28 26 73 65 6c 66 29 20 2d 3e 20 fetch(&self) ->
28a0: 52 65 73 75 6c 74 3c 73 74 64 3a 3a 74 69 6d 65 Result<std::time
28b0: 3a 3a 44 75 72 61 74 69 6f 6e 3e 20 7b 0a 09 09 ::Duration> {...
28c0: 6c 65 74 20 6d 75 74 20 64 65 6c 61 79 20 3d 20 let mut delay =
28d0: 63 68 72 6f 6e 6f 3a 3a 44 75 72 61 74 69 6f 6e chrono::Duration
28e0: 3a 3a 6d 69 6e 75 74 65 73 28 31 29 3b 0a 09 09 ::minutes(1);...
28f0: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
2900: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
2910: 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 ().await.....wit
2920: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
2930: 6d 61 74 21 28 22 41 75 74 6f 66 65 74 63 68 20 mat!("Autofetch
2940: 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f fetch conn:\n{:?
2950: 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 }", &self.pool))
2960: 3f 3b 0a 09 09 6c 65 74 20 6e 6f 77 20 3d 20 63 ?;...let now = c
2970: 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f hrono::Local::no
2980: 77 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 71 w();...let mut q
2990: 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a 71 75 65 ueue = sqlx::que
29a0: 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 ry("select sourc
29b0: 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 63 68 e_id, next_fetch
29c0: 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 73 , owner from rss
29d0: 74 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 6c tg_order natural
29e0: 20 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 67 left join rsstg
29f0: 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 6e 65 _source where ne
2a00: 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 29 xt_fetch < now()
2a10: 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 20 6d + interval '1 m
2a20: 69 6e 75 74 65 27 3b 22 29 0a 09 09 09 2e 66 65 inute';").....fe
2a30: 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 6f 6e tch_all(&mut con
2a40: 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 66 6f 72 n).await?;...for
2a50: 20 72 6f 77 20 69 6e 20 71 75 65 75 65 2e 69 74 row in queue.it
2a60: 65 72 28 29 20 7b 0a 09 09 09 6c 65 74 20 73 6f er() {....let so
2a70: 75 72 63 65 5f 69 64 3a 20 69 33 32 20 3d 20 72 urce_id: i32 = r
2a80: 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 6f 75 72 ow.try_get("sour
2a90: 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 6c 65 74 ce_id")?;....let
2aa0: 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 72 6f owner: i64 = ro
2ab0: 77 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e 65 72 w.try_get("owner
2ac0: 22 29 3f 3b 0a 09 09 09 6c 65 74 20 6e 65 78 74 ")?;....let next
2ad0: 5f 66 65 74 63 68 3a 20 44 61 74 65 54 69 6d 65 _fetch: DateTime
2ae0: 3c 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3e 20 <chrono::Local>
2af0: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 6e = row.try_get("n
2b00: 65 78 74 5f 66 65 74 63 68 22 29 3f 3b 0a 09 09 ext_fetch")?;...
2b10: 09 69 66 20 6e 65 78 74 5f 66 65 74 63 68 20 3c .if next_fetch <
2b20: 20 6e 6f 77 20 7b 0a 09 09 09 09 6c 65 74 20 63 now {.....let c
2b30: 6c 6f 6e 65 20 3d 20 43 6f 72 65 20 7b 0a 09 09 lone = Core {...
2b40: 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 74 ...owner_chat: t
2b50: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 elegram_bot::Use
2b60: 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 2c rId::new(owner),
2b70: 0a 09 09 09 09 09 2e 2e 73 65 6c 66 2e 63 6c 6f ........self.clo
2b80: 6e 65 28 29 0a 09 09 09 09 7d 3b 0a 09 09 09 09 ne().....};.....
2b90: 74 61 73 6b 3a 3a 73 70 61 77 6e 28 61 73 79 6e task::spawn(asyn
2ba0: 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 09 69 66 c move {......if
2bb0: 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 let Err(err) =
2bc0: 63 6c 6f 6e 65 2e 63 68 65 63 6b 28 26 73 6f 75 clone.check(&sou
2bd0: 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 2c 20 74 rce_id, owner, t
2be0: 72 75 65 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 rue).await {....
2bf0: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
2c00: 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 r) = clone.send(
2c10: 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b &format!("š {
2c20: 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 :?}", err), None
2c30: 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 74 20 7b 0a , None).await {.
2c40: 09 09 09 09 09 09 09 65 70 72 69 6e 74 6c 6e 21 .......eprintln!
2c50: 28 22 43 68 65 63 6b 20 65 72 72 6f 72 3a 20 7b ("Check error: {
2c60: 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 09 09 09 }", err);.......
2c70: 7d 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 7d };......};.....}
2c80: 29 3b 0a 09 09 09 7d 20 65 6c 73 65 20 69 66 20 );....} else if
2c90: 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 next_fetch - now
2ca0: 20 3c 20 64 65 6c 61 79 20 7b 0a 09 09 09 09 64 < delay {.....d
2cb0: 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 63 elay = next_fetc
2cc0: 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 7d 0a 09 09 h - now;....}...
2cd0: 7d 3b 0a 09 09 71 75 65 75 65 2e 63 6c 65 61 72 };...queue.clear
2ce0: 28 29 3b 0a 09 09 4f 6b 28 64 65 6c 61 79 2e 74 ();...Ok(delay.t
2cf0: 6f 5f 73 74 64 28 29 3f 29 0a 09 7d 0a 0a 09 70 o_std()?)..}...p
2d00: 75 62 20 61 73 79 6e 63 20 66 6e 20 6c 69 73 74 ub async fn list
2d10: 3c 53 3e 28 26 73 65 6c 66 2c 20 6f 77 6e 65 72 <S>(&self, owner
2d20: 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 : S) -> Result<S
2d30: 74 72 69 6e 67 3e 0a 09 77 68 65 72 65 20 53 3a tring>..where S:
2d40: 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c Into<i64> {...l
2d50: 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 et owner = owner
2d60: 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 .into();....let
2d70: 6d 75 74 20 72 65 70 6c 79 3a 20 56 65 63 3c 43 mut reply: Vec<C
2d80: 6f 77 3c 73 74 72 3e 3e 20 3d 20 76 65 63 21 5b ow<str>> = vec![
2d90: 5d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e ];...let mut con
2da0: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
2db0: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
2dc0: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
2dd0: 7c 20 66 6f 72 6d 61 74 21 28 22 4c 69 73 74 20 | format!("List
2de0: 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f fetch conn:\n{:?
2df0: 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 }", &self.pool))
2e00: 3f 3b 0a 09 09 72 65 70 6c 79 2e 70 75 73 68 28 ?;...reply.push(
2e10: 22 43 68 61 6e 6e 65 6c 73 3a 22 2e 69 6e 74 6f "Channels:".into
2e20: 28 29 29 3b 0a 09 09 6c 65 74 20 72 6f 77 73 20 ());...let rows
2e30: 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 = sqlx::query("s
2e40: 65 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 64 2c elect source_id,
2e50: 20 63 68 61 6e 6e 65 6c 2c 20 65 6e 61 62 6c 65 channel, enable
2e60: 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c d, url, iv_hash,
2e70: 20 75 72 6c 5f 72 65 20 66 72 6f 6d 20 72 73 73 url_re from rss
2e80: 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 tg_source where
2e90: 6f 77 6e 65 72 20 3d 20 24 31 20 6f 72 64 65 72 owner = $1 order
2ea0: 20 62 79 20 73 6f 75 72 63 65 5f 69 64 22 29 0a by source_id").
2eb0: 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a ....bind(owner).
2ec0: 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d ....fetch_all(&m
2ed0: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b ut conn).await?;
2ee0: 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 72 6f ...for row in ro
2ef0: 77 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 6c ws.iter() {....l
2f00: 65 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 et source_id: i3
2f10: 32 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 2 = row.try_get(
2f20: 22 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 "source_id")?;..
2f30: 09 09 6c 65 74 20 75 73 65 72 6e 61 6d 65 3a 20 ..let username:
2f40: 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 &str = row.try_g
2f50: 65 74 28 22 63 68 61 6e 6e 65 6c 22 29 3f 3b 0a et("channel")?;.
2f60: 09 09 09 6c 65 74 20 65 6e 61 62 6c 65 64 3a 20 ...let enabled:
2f70: 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 5f 67 bool = row.try_g
2f80: 65 74 28 22 65 6e 61 62 6c 65 64 22 29 3f 3b 0a et("enabled")?;.
2f90: 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 73 74 72 ...let url: &str
2fa0: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
2fb0: 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 69 url")?;....let i
2fc0: 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 v_hash: Option<&
2fd0: 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 str> = row.try_g
2fe0: 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f 3b 0a et("iv_hash")?;.
2ff0: 09 09 09 6c 65 74 20 75 72 6c 5f 72 65 3a 20 4f ...let url_re: O
3000: 70 74 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 72 6f ption<&str> = ro
3010: 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 5f 72 w.try_get("url_r
3020: 65 22 29 3f 3b 0a 09 09 09 72 65 70 6c 79 2e 70 e")?;....reply.p
3030: 75 73 68 28 66 6f 72 6d 61 74 21 28 22 5c 6e 5c ush(format!("\n\
3040: 5c 23 ef b8 8f e2 83 a3 20 7b 7d 20 5c 5c 2a ef \#ļøā£ {} \\*ļ
3050: b8 8f e2 83 a3 20 60 7b 7d 60 20 7b 7d 5c 6e f0 øā£ `{}` {}\nš
3060: 9f 94 97 20 60 7b 7d 60 22 2c 20 73 6f 75 72 63 `{}`", sourc
3070: 65 5f 69 64 2c 20 75 73 65 72 6e 61 6d 65 2c 20 e_id, username,
3080: 20 0a 09 09 09 09 6d 61 74 63 68 20 65 6e 61 62 .....match enab
3090: 6c 65 64 20 7b 0a 09 09 09 09 09 74 72 75 65 20 led {......true
30a0: 20 3d 3e 20 22 f0 9f 94 84 20 65 6e 61 62 6c 65 => "š enable
30b0: 64 22 2c 0a 09 09 09 09 09 66 61 6c 73 65 20 3d d",......false =
30c0: 3e 20 22 e2 9b 94 20 64 69 73 61 62 6c 65 64 22 > "ā disabled"
30d0: 2c 0a 09 09 09 09 7d 2c 20 75 72 6c 29 2e 69 6e ,.....}, url).in
30e0: 74 6f 28 29 29 3b 0a 09 09 09 69 66 20 6c 65 74 to());....if let
30f0: 20 53 6f 6d 65 28 68 61 73 68 29 20 3d 20 69 76 Some(hash) = iv
3100: 5f 68 61 73 68 20 7b 0a 09 09 09 09 72 65 70 6c _hash {.....repl
3110: 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 y.push(format!("
3120: 49 56 3a 20 60 7b 7d 60 22 2c 20 68 61 73 68 29 IV: `{}`", hash)
3130: 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 7d 0a 09 .into());....}..
3140: 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 72 65 ..if let Some(re
3150: 29 20 3d 20 75 72 6c 5f 72 65 20 7b 0a 09 09 09 ) = url_re {....
3160: 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d .reply.push(form
3170: 61 74 21 28 22 52 45 3a 20 60 7b 7d 60 22 2c 20 at!("RE: `{}`",
3180: 72 65 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 re).into());....
3190: 7d 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 65 70 6c }...};...Ok(repl
31a0: 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 0a 09 7d y.join("\n"))..}
31b0: 0a 7d 0a .}.