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 7d t<Arc<i32>>>>,.}
01b0: 0a 0a 69 6d 70 6c 20 43 6f 72 65 20 7b 0a 09 70 ..impl Core {..p
01c0: 75 62 20 66 6e 20 6e 65 77 28 73 65 74 74 69 6e ub fn new(settin
01d0: 67 73 3a 20 63 6f 6e 66 69 67 3a 3a 43 6f 6e 66 gs: config::Conf
01e0: 69 67 29 20 2d 3e 20 52 65 73 75 6c 74 3c 41 72 ig) -> Result<Ar
01f0: 63 3c 43 6f 72 65 3e 3e 20 7b 0a 09 09 6c 65 74 c<Core>> {...let
0200: 20 6f 77 6e 65 72 20 3d 20 73 65 74 74 69 6e 67 owner = setting
0210: 73 2e 67 65 74 5f 69 6e 74 28 22 6f 77 6e 65 72 s.get_int("owner
0220: 22 29 3f 3b 0a 09 09 6c 65 74 20 61 70 69 5f 6b ")?;...let api_k
0230: 65 79 20 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 ey = settings.ge
0240: 74 5f 73 74 72 69 6e 67 28 22 61 70 69 5f 6b 65 t_string("api_ke
0250: 79 22 29 3f 3b 0a 09 09 6c 65 74 20 74 67 20 3d y")?;...let tg =
0260: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 41 telegram_bot::A
0270: 70 69 3a 3a 6e 65 77 28 26 61 70 69 5f 6b 65 79 pi::new(&api_key
0280: 29 3b 0a 09 09 6c 65 74 20 74 67 5f 63 6c 6f 6e );...let tg_clon
0290: 65 64 20 3d 20 74 67 2e 63 6c 6f 6e 65 28 29 3b ed = tg.clone();
02a0: 0a 09 09 6c 65 74 20 63 6f 72 65 20 3d 20 41 72 ...let core = Ar
02b0: 63 3a 3a 6e 65 77 28 43 6f 72 65 20 7b 0a 09 09 c::new(Core {...
02c0: 09 74 67 2c 0a 09 09 09 6d 79 3a 20 74 61 73 6b .tg,....my: task
02d0: 3a 3a 62 6c 6f 63 6b 5f 6f 6e 28 61 73 79 6e 63 ::block_on(async
02e0: 20 7b 0a 09 09 09 09 74 67 5f 63 6c 6f 6e 65 64 {.....tg_cloned
02f0: 2e 73 65 6e 64 28 74 65 6c 65 67 72 61 6d 5f 62 .send(telegram_b
0300: 6f 74 3a 3a 47 65 74 4d 65 29 2e 61 77 61 69 74 ot::GetMe).await
0310: 0a 09 09 09 7d 29 3f 2c 0a 09 09 09 6f 77 6e 65 ....})?,....owne
0320: 72 5f 63 68 61 74 3a 20 74 65 6c 65 67 72 61 6d r_chat: telegram
0330: 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 _bot::UserId::ne
0340: 77 28 6f 77 6e 65 72 29 2c 0a 09 09 09 70 6f 6f w(owner),....poo
0350: 6c 3a 20 50 67 50 6f 6f 6c 4f 70 74 69 6f 6e 73 l: PgPoolOptions
0360: 3a 3a 6e 65 77 28 29 0a 09 09 09 09 2e 6d 61 78 ::new()......max
0370: 5f 63 6f 6e 6e 65 63 74 69 6f 6e 73 28 35 29 0a _connections(5).
0380: 09 09 09 09 2e 61 63 71 75 69 72 65 5f 74 69 6d .....acquire_tim
0390: 65 6f 75 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a eout(std::time::
03a0: 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 33 30 Duration::new(30
03b0: 30 2c 20 30 29 29 0a 09 09 09 09 2e 69 64 6c 65 0, 0))......idle
03c0: 5f 74 69 6d 65 6f 75 74 28 73 74 64 3a 3a 74 69 _timeout(std::ti
03d0: 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 me::Duration::ne
03e0: 77 28 36 30 2c 20 30 29 29 0a 09 09 09 09 2e 63 w(60, 0))......c
03f0: 6f 6e 6e 65 63 74 5f 6c 61 7a 79 28 26 73 65 74 onnect_lazy(&set
0400: 74 69 6e 67 73 2e 67 65 74 5f 73 74 72 69 6e 67 tings.get_string
0410: 28 22 70 67 22 29 3f 29 3f 2c 0a 09 09 09 73 6f ("pg")?)?,....so
0420: 75 72 63 65 73 3a 20 41 72 63 3a 3a 6e 65 77 28 urces: Arc::new(
0430: 4d 75 74 65 78 3a 3a 6e 65 77 28 48 61 73 68 53 Mutex::new(HashS
0440: 65 74 3a 3a 6e 65 77 28 29 29 29 2c 0a 09 09 7d et::new())),...}
0450: 29 3b 0a 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d );...let clone =
0460: 20 63 6f 72 65 2e 63 6c 6f 6e 65 28 29 3b 0a 09 core.clone();..
0470: 09 74 61 73 6b 3a 3a 73 70 61 77 6e 28 61 73 79 .task::spawn(asy
0480: 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 6c 6f 6f nc move {....loo
0490: 70 20 7b 0a 09 09 09 09 6c 65 74 20 64 65 6c 61 p {.....let dela
04a0: 79 20 3d 20 6d 61 74 63 68 20 26 63 6c 6f 6e 65 y = match &clone
04b0: 2e 61 75 74 6f 66 65 74 63 68 28 29 2e 61 77 61 .autofetch().awa
04c0: 69 74 20 7b 0a 09 09 09 09 09 45 72 72 28 65 72 it {......Err(er
04d0: 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 69 66 r) => {.......if
04e0: 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 let Err(err) =
04f0: 63 6c 6f 6e 65 2e 73 65 6e 64 28 66 6f 72 6d 61 clone.send(forma
0500: 74 21 28 22 f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 t!("š {:?}",
0510: 65 72 72 29 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 err), None, None
0520: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 ).await {.......
0530: 09 65 70 72 69 6e 74 6c 6e 21 28 22 41 75 74 6f .eprintln!("Auto
0540: 66 65 74 63 68 20 65 72 72 6f 72 3a 20 7b 7d 22 fetch error: {}"
0550: 2c 20 65 72 72 29 3b 0a 09 09 09 09 09 09 7d 3b , err);.......};
0560: 0a 09 09 09 09 09 09 73 74 64 3a 3a 74 69 6d 65 .......std::time
0570: 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 66 72 6f 6d ::Duration::from
0580: 5f 73 65 63 73 28 36 30 29 0a 09 09 09 09 09 7d _secs(60)......}
0590: 2c 0a 09 09 09 09 09 4f 6b 28 74 69 6d 65 29 20 ,......Ok(time)
05a0: 3d 3e 20 2a 74 69 6d 65 2c 0a 09 09 09 09 7d 3b => *time,.....};
05b0: 0a 09 09 09 09 74 61 73 6b 3a 3a 73 6c 65 65 70 .....task::sleep
05c0: 28 64 65 6c 61 79 29 2e 61 77 61 69 74 3b 0a 09 (delay).await;..
05d0: 09 09 7d 0a 09 09 7d 29 3b 0a 09 09 4f 6b 28 63 ..}...});...Ok(c
05e0: 6f 72 65 29 0a 09 7d 0a 0a 09 70 75 62 20 66 6e ore)..}...pub fn
05f0: 20 73 74 72 65 61 6d 28 26 73 65 6c 66 29 20 2d stream(&self) -
0600: 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a > telegram_bot::
0610: 55 70 64 61 74 65 73 53 74 72 65 61 6d 20 7b 0a UpdatesStream {.
0620: 09 09 73 65 6c 66 2e 74 67 2e 73 74 72 65 61 6d ..self.tg.stream
0630: 28 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e ()..}...pub asyn
0640: 63 20 66 6e 20 73 65 6e 64 3c 27 61 2c 20 53 3e c fn send<'a, S>
0650: 28 26 73 65 6c 66 2c 20 6d 73 67 3a 20 53 2c 20 (&self, msg: S,
0660: 74 61 72 67 65 74 3a 20 4f 70 74 69 6f 6e 3c 74 target: Option<t
0670: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 elegram_bot::Use
0680: 72 49 64 3e 2c 20 6d 6f 64 65 3a 20 4f 70 74 69 rId>, mode: Opti
0690: 6f 6e 3c 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a on<telegram_bot:
06a0: 3a 74 79 70 65 73 3a 3a 50 61 72 73 65 4d 6f 64 :types::ParseMod
06b0: 65 3e 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 e>) -> Result<()
06c0: 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f >..where S: Into
06d0: 3c 43 6f 77 3c 27 61 2c 20 73 74 72 3e 3e 20 7b <Cow<'a, str>> {
06e0: 0a 09 09 6c 65 74 20 6d 6f 64 65 20 3d 20 6d 6f ...let mode = mo
06f0: 64 65 2e 75 6e 77 72 61 70 5f 6f 72 28 74 65 6c de.unwrap_or(tel
0700: 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 egram_bot::types
0710: 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d ::ParseMode::Htm
0720: 6c 29 3b 0a 09 09 6c 65 74 20 74 61 72 67 65 74 l);...let target
0730: 20 3d 20 74 61 72 67 65 74 2e 75 6e 77 72 61 70 = target.unwrap
0740: 5f 6f 72 28 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 _or(self.owner_c
0750: 68 61 74 29 3b 0a 09 09 73 65 6c 66 2e 74 67 2e hat);...self.tg.
0760: 73 65 6e 64 28 74 65 6c 65 67 72 61 6d 5f 62 6f send(telegram_bo
0770: 74 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a t::SendMessage::
0780: 6e 65 77 28 74 61 72 67 65 74 2c 20 6d 73 67 29 new(target, msg)
0790: 2e 70 61 72 73 65 5f 6d 6f 64 65 28 6d 6f 64 65 .parse_mode(mode
07a0: 29 29 2e 61 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 )).await?;...Ok(
07b0: 28 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ())..}...pub asy
07c0: 6e 63 20 66 6e 20 63 68 65 63 6b 3c 53 3e 28 26 nc fn check<S>(&
07d0: 73 65 6c 66 2c 20 69 64 3a 20 26 69 33 32 2c 20 self, id: &i32,
07e0: 6f 77 6e 65 72 3a 20 53 2c 20 72 65 61 6c 3a 20 owner: S, real:
07f0: 62 6f 6f 6c 29 20 2d 3e 20 52 65 73 75 6c 74 3c bool) -> Result<
0800: 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 Cow<'_, str>>..w
0810: 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 here S: Into<i64
0820: 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 > {...let owner
0830: 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a = owner.into();.
0840: 0a 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 65 ...let mut poste
0850: 64 3a 20 69 33 32 20 3d 20 30 3b 0a 09 09 6c 65 d: i32 = 0;...le
0860: 74 20 69 64 20 3d 20 7b 0a 09 09 09 6c 65 74 20 t id = {....let
0870: 6d 75 74 20 73 65 74 20 3d 20 73 65 6c 66 2e 73 mut set = self.s
0880: 6f 75 72 63 65 73 2e 6c 6f 63 6b 28 29 2e 75 6e ources.lock().un
0890: 77 72 61 70 28 29 3b 0a 09 09 09 6d 61 74 63 68 wrap();....match
08a0: 20 73 65 74 2e 67 65 74 28 69 64 29 20 7b 0a 09 set.get(id) {..
08b0: 09 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 69 ...Some(id) => i
08c0: 64 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 09 4e d.clone(),.....N
08d0: 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 6c 65 one => {......le
08e0: 74 20 69 64 20 3d 20 41 72 63 3a 3a 6e 65 77 28 t id = Arc::new(
08f0: 2a 69 64 29 3b 0a 09 09 09 09 09 73 65 74 2e 69 *id);......set.i
0900: 6e 73 65 72 74 28 69 64 2e 63 6c 6f 6e 65 28 29 nsert(id.clone()
0910: 29 3b 0a 09 09 09 09 09 69 64 2e 63 6c 6f 6e 65 );......id.clone
0920: 28 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 ().....},....}..
0930: 09 7d 3b 0a 09 09 6c 65 74 20 63 6f 75 6e 74 20 .};...let count
0940: 3d 20 41 72 63 3a 3a 73 74 72 6f 6e 67 5f 63 6f = Arc::strong_co
0950: 75 6e 74 28 26 69 64 29 3b 0a 09 09 69 66 20 63 unt(&id);...if c
0960: 6f 75 6e 74 20 3d 3d 20 32 20 7b 0a 09 09 09 6c ount == 2 {....l
0970: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
0980: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
0990: 29 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 74 ).await......wit
09a0: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
09b0: 6d 61 74 21 28 22 51 75 65 72 79 20 71 75 65 75 mat!("Query queu
09c0: 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b e fetch conn:\n{
09d0: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
09e0: 29 29 3f 3b 0a 09 09 09 6c 65 74 20 72 6f 77 20 ))?;....let row
09f0: 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 = sqlx::query("s
0a00: 65 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 64 2c elect source_id,
0a10: 20 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c channel_id, url
0a20: 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 , iv_hash, owner
0a30: 2c 20 75 72 6c 5f 72 65 20 66 72 6f 6d 20 72 73 , url_re from rs
0a40: 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 stg_source where
0a50: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 source_id = $1
0a60: 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 and owner = $2")
0a70: 0a 09 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a ......bind(*id).
0a80: 09 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 .....bind(owner)
0a90: 0a 09 09 09 09 2e 66 65 74 63 68 5f 6f 6e 65 28 ......fetch_one(
0aa0: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
0ab0: 0a 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 ......with_conte
0ac0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 51 xt(|| format!("Q
0ad0: 75 65 72 79 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a uery source:\n{:
0ae0: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
0af0: 29 3f 3b 0a 09 09 09 64 72 6f 70 28 63 6f 6e 6e )?;....drop(conn
0b00: 29 3b 0a 09 09 09 6c 65 74 20 63 68 61 6e 6e 65 );....let channe
0b10: 6c 5f 69 64 3a 20 69 36 34 20 3d 20 72 6f 77 2e l_id: i64 = row.
0b20: 74 72 79 5f 67 65 74 28 22 63 68 61 6e 6e 65 6c try_get("channel
0b30: 5f 69 64 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 _id")?;....let u
0b40: 72 6c 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 rl: &str = row.t
0b50: 72 79 5f 67 65 74 28 22 75 72 6c 22 29 3f 3b 0a ry_get("url")?;.
0b60: 09 09 09 6c 65 74 20 69 76 5f 68 61 73 68 3a 20 ...let iv_hash:
0b70: 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 72 Option<&str> = r
0b80: 6f 77 2e 74 72 79 5f 67 65 74 28 22 69 76 5f 68 ow.try_get("iv_h
0b90: 61 73 68 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 ash")?;....let u
0ba0: 72 6c 5f 72 65 20 3d 20 6d 61 74 63 68 20 72 6f rl_re = match ro
0bb0: 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 5f 72 w.try_get("url_r
0bc0: 65 22 29 3f 20 7b 0a 09 09 09 09 53 6f 6d 65 28 e")? {.....Some(
0bd0: 78 29 20 3d 3e 20 53 6f 6d 65 28 73 65 64 72 65 x) => Some(sedre
0be0: 67 65 78 3a 3a 52 65 70 6c 61 63 65 43 6f 6d 6d gex::ReplaceComm
0bf0: 61 6e 64 3a 3a 6e 65 77 28 78 29 3f 29 2c 0a 09 and::new(x)?),..
0c00: 09 09 09 4e 6f 6e 65 20 3d 3e 20 4e 6f 6e 65 2c ...None => None,
0c10: 0a 09 09 09 7d 3b 0a 09 09 09 6c 65 74 20 64 65 ....};....let de
0c20: 73 74 69 6e 61 74 69 6f 6e 20 3d 20 6d 61 74 63 stination = matc
0c30: 68 20 72 65 61 6c 20 7b 0a 09 09 09 09 74 72 75 h real {.....tru
0c40: 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f e => telegram_bo
0c50: 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 63 t::UserId::new(c
0c60: 68 61 6e 6e 65 6c 5f 69 64 29 2c 0a 09 09 09 09 hannel_id),.....
0c70: 66 61 6c 73 65 20 3d 3e 20 74 65 6c 65 67 72 61 false => telegra
0c80: 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e m_bot::UserId::n
0c90: 65 77 28 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 ew(row.try_get("
0ca0: 6f 77 6e 65 72 22 29 3f 29 2c 0a 09 09 09 7d 3b owner")?),....};
0cb0: 0a 09 09 09 6c 65 74 20 6d 75 74 20 74 68 69 73 ....let mut this
0cc0: 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e 3c 44 _fetch: Option<D
0cd0: 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a ateTime<chrono::
0ce0: 46 69 78 65 64 4f 66 66 73 65 74 3e 3e 20 3d 20 FixedOffset>> =
0cf0: 4e 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 6d 75 74 None;....let mut
0d00: 20 70 6f 73 74 73 3a 20 42 54 72 65 65 4d 61 70 posts: BTreeMap
0d10: 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f <DateTime<chrono
0d20: 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 2c 20 ::FixedOffset>,
0d30: 53 74 72 69 6e 67 3e 20 3d 20 42 54 72 65 65 4d String> = BTreeM
0d40: 61 70 3a 3a 6e 65 77 28 29 3b 0a 09 09 09 6c 65 ap::new();....le
0d50: 74 20 72 65 73 70 6f 6e 73 65 20 3d 20 72 65 71 t response = req
0d60: 77 65 73 74 3a 3a 67 65 74 28 75 72 6c 29 2e 61 west::get(url).a
0d70: 77 61 69 74 3f 3b 0a 09 09 09 6c 65 74 20 73 74 wait?;....let st
0d80: 61 74 75 73 20 3d 20 72 65 73 70 6f 6e 73 65 2e atus = response.
0d90: 73 74 61 74 75 73 28 29 3b 0a 09 09 09 6c 65 74 status();....let
0da0: 20 63 6f 6e 74 65 6e 74 20 3d 20 72 65 73 70 6f content = respo
0db0: 6e 73 65 2e 62 79 74 65 73 28 29 2e 61 77 61 69 nse.bytes().awai
0dc0: 74 3f 3b 0a 09 09 09 6d 61 74 63 68 20 72 73 73 t?;....match rss
0dd0: 3a 3a 43 68 61 6e 6e 65 6c 3a 3a 72 65 61 64 5f ::Channel::read_
0de0: 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e from(&content[..
0df0: 5d 29 20 7b 0a 09 09 09 09 4f 6b 28 66 65 65 64 ]) {.....Ok(feed
0e00: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 66 6f 72 20 ) => {......for
0e10: 69 74 65 6d 20 69 6e 20 66 65 65 64 2e 69 74 65 item in feed.ite
0e20: 6d 73 28 29 20 7b 0a 09 09 09 09 09 09 69 66 20 ms() {.......if
0e30: 6c 65 74 20 53 6f 6d 65 28 6c 69 6e 6b 29 20 3d let Some(link) =
0e40: 20 69 74 65 6d 2e 6c 69 6e 6b 28 29 20 7b 0a 09 item.link() {..
0e50: 09 09 09 09 09 09 6c 65 74 20 64 61 74 65 20 3d ......let date =
0e60: 20 6d 61 74 63 68 20 69 74 65 6d 2e 70 75 62 5f match item.pub_
0e70: 64 61 74 65 28 29 20 7b 0a 09 09 09 09 09 09 09 date() {........
0e80: 09 53 6f 6d 65 28 66 65 65 64 5f 64 61 74 65 29 .Some(feed_date)
0e90: 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 => DateTime::pa
0ea0: 72 73 65 5f 66 72 6f 6d 5f 72 66 63 32 38 32 32 rse_from_rfc2822
0eb0: 28 66 65 65 64 5f 64 61 74 65 29 2c 0a 09 09 09 (feed_date),....
0ec0: 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 44 61 74 .....None => Dat
0ed0: 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f eTime::parse_fro
0ee0: 6d 5f 72 66 63 33 33 33 39 28 26 69 74 65 6d 2e m_rfc3339(&item.
0ef0: 64 75 62 6c 69 6e 5f 63 6f 72 65 5f 65 78 74 28 dublin_core_ext(
0f00: 29 2e 75 6e 77 72 61 70 28 29 2e 64 61 74 65 73 ).unwrap().dates
0f10: 28 29 5b 30 5d 29 2c 0a 09 09 09 09 09 09 09 7d ()[0]),........}
0f20: 3f 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 75 72 ?;........let ur
0f30: 6c 20 3d 20 6c 69 6e 6b 3b 0a 09 09 09 09 09 09 l = link;.......
0f40: 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 64 61 .posts.insert(da
0f50: 74 65 2c 20 75 72 6c 2e 74 6f 5f 73 74 72 69 6e te, url.to_strin
0f60: 67 28 29 29 3b 0a 09 09 09 09 09 09 7d 0a 09 09 g());.......}...
0f70: 09 09 09 7d 3b 0a 09 09 09 09 7d 2c 0a 09 09 09 ...};.....},....
0f80: 09 45 72 72 28 65 72 72 29 20 3d 3e 20 6d 61 74 .Err(err) => mat
0f90: 63 68 20 65 72 72 20 7b 0a 09 09 09 09 09 72 73 ch err {......rs
0fa0: 73 3a 3a 45 72 72 6f 72 3a 3a 49 6e 76 61 6c 69 s::Error::Invali
0fb0: 64 53 74 61 72 74 54 61 67 20 3d 3e 20 7b 0a 09 dStartTag => {..
0fc0: 09 09 09 09 09 6c 65 74 20 66 65 65 64 20 3d 20 .....let feed =
0fd0: 61 74 6f 6d 5f 73 79 6e 64 69 63 61 74 69 6f 6e atom_syndication
0fe0: 3a 3a 46 65 65 64 3a 3a 72 65 61 64 5f 66 72 6f ::Feed::read_fro
0ff0: 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 0a m(&content[..]).
1000: 09 09 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e ........with_con
1010: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1020: 22 50 72 6f 62 6c 65 6d 20 6f 70 65 6e 69 6e 67 "Problem opening
1030: 20 66 65 65 64 20 75 72 6c 3a 5c 6e 7b 7d 5c 6e feed url:\n{}\n
1040: 7b 7d 22 2c 20 26 75 72 6c 2c 20 73 74 61 74 75 {}", &url, statu
1050: 73 29 29 3f 3b 0a 09 09 09 09 09 09 66 6f 72 20 s))?;.......for
1060: 69 74 65 6d 20 69 6e 20 66 65 65 64 2e 65 6e 74 item in feed.ent
1070: 72 69 65 73 28 29 20 7b 0a 09 09 09 09 09 09 09 ries() {........
1080: 6c 65 74 20 64 61 74 65 20 3d 20 69 74 65 6d 2e let date = item.
1090: 70 75 62 6c 69 73 68 65 64 28 29 2e 75 6e 77 72 published().unwr
10a0: 61 70 28 29 3b 0a 09 09 09 09 09 09 09 6c 65 74 ap();........let
10b0: 20 75 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b url = item.link
10c0: 73 28 29 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 s()[0].href();..
10d0: 09 09 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 ......posts.inse
10e0: 72 74 28 2a 64 61 74 65 2c 20 75 72 6c 2e 74 6f rt(*date, url.to
10f0: 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 _string());.....
1100: 09 09 7d 3b 0a 09 09 09 09 09 7d 2c 0a 09 09 09 ..};......},....
1110: 09 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 6f ..rss::Error::Eo
1120: 66 20 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f 20 f => (),......_
1130: 3d 3e 20 62 61 69 6c 21 28 22 55 6e 73 75 70 70 => bail!("Unsupp
1140: 6f 72 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 64 orted or mangled
1150: 20 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c content:\n{:?}\
1160: 6e 7b 3a 23 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e 22 n{:#?}\n{:#?}\n"
1170: 2c 20 26 75 72 6c 2c 20 65 72 72 2c 20 73 74 61 , &url, err, sta
1180: 74 75 73 29 0a 09 09 09 09 7d 0a 09 09 09 7d 3b tus).....}....};
1190: 0a 09 09 09 66 6f 72 20 28 64 61 74 65 2c 20 75 ....for (date, u
11a0: 72 6c 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 65 rl) in posts.ite
11b0: 72 28 29 20 7b 0a 09 09 09 09 6c 65 74 20 6d 75 r() {.....let mu
11c0: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
11d0: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
11e0: 69 74 0a 09 09 09 09 09 2e 77 69 74 68 5f 63 6f it.......with_co
11f0: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
1200: 28 22 43 68 65 63 6b 20 70 6f 73 74 20 66 65 74 ("Check post fet
1210: 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c ch conn:\n{:?}",
1220: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a &self.pool))?;.
1230: 09 09 09 09 6c 65 74 20 70 6f 73 74 5f 75 72 6c ....let post_url
1240: 3a 20 43 6f 77 3c 73 74 72 3e 20 3d 20 6d 61 74 : Cow<str> = mat
1250: 63 68 20 75 72 6c 5f 72 65 20 7b 0a 09 09 09 09 ch url_re {.....
1260: 09 53 6f 6d 65 28 72 65 66 20 78 29 20 3d 3e 20 .Some(ref x) =>
1270: 78 2e 65 78 65 63 75 74 65 28 75 72 6c 29 2c 0a x.execute(url),.
1280: 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 75 72 6c .....None => url
1290: 2e 69 6e 74 6f 28 29 2c 0a 09 09 09 09 7d 3b 0a .into(),.....};.
12a0: 09 09 09 09 6c 65 74 20 72 6f 77 20 3d 20 73 71 ....let row = sq
12b0: 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 lx::query("selec
12c0: 74 20 65 78 69 73 74 73 28 73 65 6c 65 63 74 20 t exists(select
12d0: 74 72 75 65 20 66 72 6f 6d 20 72 73 73 74 67 5f true from rsstg_
12e0: 70 6f 73 74 20 77 68 65 72 65 20 75 72 6c 20 3d post where url =
12f0: 20 24 31 20 61 6e 64 20 73 6f 75 72 63 65 5f 69 $1 and source_i
1300: 64 20 3d 20 24 32 29 20 61 73 20 65 78 69 73 74 d = $2) as exist
1310: 73 3b 22 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 s;").......bind(
1320: 26 2a 70 6f 73 74 5f 75 72 6c 29 0a 09 09 09 09 &*post_url).....
1330: 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 ..bind(*id).....
1340: 09 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 ..fetch_one(&mut
1350: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 conn).await....
1360: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1370: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 68 65 63 || format!("Chec
1380: 6b 20 70 6f 73 74 3a 5c 6e 7b 3a 3f 7d 22 2c 20 k post:\n{:?}",
1390: 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 09 09 6c 65 &conn))?;.....le
13a0: 74 20 65 78 69 73 74 73 3a 20 62 6f 6f 6c 20 3d t exists: bool =
13b0: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 65 78 row.try_get("ex
13c0: 69 73 74 73 22 29 3f 3b 0a 09 09 09 09 69 66 20 ists")?;.....if
13d0: 21 20 65 78 69 73 74 73 20 7b 0a 09 09 09 09 09 ! exists {......
13e0: 69 66 20 74 68 69 73 5f 66 65 74 63 68 20 3d 3d if this_fetch ==
13f0: 20 4e 6f 6e 65 20 7c 7c 20 2a 64 61 74 65 20 3e None || *date >
1400: 20 74 68 69 73 5f 66 65 74 63 68 2e 75 6e 77 72 this_fetch.unwr
1410: 61 70 28 29 20 7b 0a 09 09 09 09 09 09 74 68 69 ap() {.......thi
1420: 73 5f 66 65 74 63 68 20 3d 20 53 6f 6d 65 28 2a s_fetch = Some(*
1430: 64 61 74 65 29 3b 0a 09 09 09 09 09 7d 3b 0a 09 date);......};..
1440: 09 09 09 09 73 65 6c 66 2e 74 67 2e 73 65 6e 64 ....self.tg.send
1450: 28 20 6d 61 74 63 68 20 69 76 5f 68 61 73 68 20 ( match iv_hash
1460: 7b 0a 09 09 09 09 09 09 09 53 6f 6d 65 28 68 61 {........Some(ha
1470: 73 68 29 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f sh) => telegram_
1480: 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 bot::SendMessage
1490: 3a 3a 6e 65 77 28 64 65 73 74 69 6e 61 74 69 6f ::new(destinatio
14a0: 6e 2c 20 66 6f 72 6d 61 74 21 28 22 3c 61 20 68 n, format!("<a h
14b0: 72 65 66 3d 5c 22 68 74 74 70 73 3a 2f 2f 74 2e ref=\"https://t.
14c0: 6d 65 2f 69 76 3f 75 72 6c 3d 7b 7d 26 72 68 61 me/iv?url={}&rha
14d0: 73 68 3d 7b 7d 5c 22 3e 20 3c 2f 61 3e 7b 30 7d sh={}\"> </a>{0}
14e0: 22 2c 20 26 70 6f 73 74 5f 75 72 6c 2c 20 68 61 ", &post_url, ha
14f0: 73 68 29 29 2c 0a 09 09 09 09 09 09 09 4e 6f 6e sh)),........Non
1500: 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f e => telegram_bo
1510: 74 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a t::SendMessage::
1520: 6e 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c new(destination,
1530: 20 66 6f 72 6d 61 74 21 28 22 7b 7d 22 2c 20 70 format!("{}", p
1540: 6f 73 74 5f 75 72 6c 29 29 2c 0a 09 09 09 09 09 ost_url)),......
1550: 09 7d 2e 70 61 72 73 65 5f 6d 6f 64 65 28 74 65 .}.parse_mode(te
1560: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 legram_bot::type
1570: 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 s::ParseMode::Ht
1580: 6d 6c 29 29 2e 61 77 61 69 74 0a 09 09 09 09 09 ml)).await......
1590: 09 2e 63 6f 6e 74 65 78 74 28 22 43 61 6e 27 74 ..context("Can't
15a0: 20 70 6f 73 74 20 6d 65 73 73 61 67 65 3a 22 29 post message:")
15b0: 3f 3b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 ?;......sqlx::qu
15c0: 65 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 6f ery("insert into
15d0: 20 72 73 73 74 67 5f 70 6f 73 74 20 28 73 6f 75 rsstg_post (sou
15e0: 72 63 65 5f 69 64 2c 20 70 6f 73 74 65 64 2c 20 rce_id, posted,
15f0: 75 72 6c 29 20 76 61 6c 75 65 73 20 28 24 31 2c url) values ($1,
1600: 20 24 32 2c 20 24 33 29 3b 22 29 0a 09 09 09 09 $2, $3);").....
1610: 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 ...bind(*id)....
1620: 09 09 09 2e 62 69 6e 64 28 64 61 74 65 29 0a 09 ....bind(date)..
1630: 09 09 09 09 09 2e 62 69 6e 64 28 26 2a 70 6f 73 ......bind(&*pos
1640: 74 5f 75 72 6c 29 0a 09 09 09 09 09 09 2e 65 78 t_url)........ex
1650: 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 ecute(&mut conn)
1660: 2e 61 77 61 69 74 0a 09 09 09 09 09 09 2e 77 69 .await........wi
1670: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
1680: 72 6d 61 74 21 28 22 52 65 63 6f 72 64 20 70 6f rmat!("Record po
1690: 73 74 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e st:\n{:?}", &con
16a0: 6e 29 29 3f 3b 0a 09 09 09 09 09 64 72 6f 70 28 n))?;......drop(
16b0: 63 6f 6e 6e 29 3b 0a 09 09 09 09 09 74 61 73 6b conn);......task
16c0: 3a 3a 73 6c 65 65 70 28 73 74 64 3a 3a 74 69 6d ::sleep(std::tim
16d0: 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 e::Duration::new
16e0: 28 34 2c 20 30 29 29 2e 61 77 61 69 74 3b 0a 09 (4, 0)).await;..
16f0: 09 09 09 7d 3b 0a 09 09 09 09 70 6f 73 74 65 64 ...};.....posted
1700: 20 2b 3d 20 31 3b 0a 09 09 09 7d 3b 0a 09 09 09 += 1;....};....
1710: 70 6f 73 74 73 2e 63 6c 65 61 72 28 29 3b 0a 09 posts.clear();..
1720: 09 7d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f .};...let mut co
1730: 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 nn = self.pool.a
1740: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 cquire().await..
1750: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1760: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 || format!("Upda
1770: 74 65 20 73 63 72 61 70 65 20 66 65 74 63 68 20 te scrape fetch
1780: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
1790: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 73 elf.pool))?;...s
17a0: 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 qlx::query("upda
17b0: 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 te rsstg_source
17c0: 73 65 74 20 6c 61 73 74 5f 73 63 72 61 70 65 20 set last_scrape
17d0: 3d 20 6e 6f 77 28 29 20 77 68 65 72 65 20 73 6f = now() where so
17e0: 75 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a urce_id = $1;").
17f0: 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 ....bind(*id)...
1800: 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 ..execute(&mut c
1810: 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 onn).await.....w
1820: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
1830: 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 73 ormat!("Update s
1840: 63 72 61 70 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 crape:\n{:?}", &
1850: 63 6f 6e 6e 29 29 3f 3b 0a 09 09 4f 6b 28 66 6f conn))?;...Ok(fo
1860: 72 6d 61 74 21 28 22 50 6f 73 74 65 64 3a 20 7b rmat!("Posted: {
1870: 7d 22 2c 20 26 70 6f 73 74 65 64 29 2e 69 6e 74 }", &posted).int
1880: 6f 28 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 o())..}...pub as
1890: 79 6e 63 20 66 6e 20 64 65 6c 65 74 65 3c 53 3e ync fn delete<S>
18a0: 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 (&self, source_i
18b0: 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 d: &i32, owner:
18c0: 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 S) -> Result<Cow
18d0: 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 <'_, str>>..wher
18e0: 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e S: Into<i64> {
18f0: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f ...let owner = o
1900: 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 wner.into();....
1910: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
1920: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
1930: 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 ().await.....wit
1940: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
1950: 6d 61 74 21 28 22 44 65 6c 65 74 65 20 66 65 74 mat!("Delete fet
1960: 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c ch conn:\n{:?}",
1970: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a &self.pool))?;.
1980: 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 ..match sqlx::qu
1990: 65 72 79 28 22 64 65 6c 65 74 65 20 66 72 6f 6d ery("delete from
19a0: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 rsstg_source wh
19b0: 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 ere source_id =
19c0: 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 $1 and owner = $
19d0: 32 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 2;").....bind(so
19e0: 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e urce_id).....bin
19f0: 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 d(owner).....exe
1a00: 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e cute(&mut conn).
1a10: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
1a20: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1a30: 21 28 22 44 65 6c 65 74 65 20 73 6f 75 72 63 65 !("Delete source
1a40: 20 72 75 6c 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 rule:\n{:?}", &
1a50: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 self.pool))?....
1a60: 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 .rows_affected()
1a70: 20 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 {....0 => { Ok(
1a80: 22 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 "No data found f
1a90: 6f 75 6e 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d ound.".into()) }
1aa0: 2c 0a 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 ,....x => { Ok(f
1ab0: 6f 72 6d 61 74 21 28 22 7b 7d 20 73 6f 75 72 63 ormat!("{} sourc
1ac0: 65 73 20 72 65 6d 6f 76 65 64 2e 22 2c 20 78 29 es removed.", x)
1ad0: 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a .into()) },...}.
1ae0: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
1af0: 6e 20 63 6c 65 61 6e 3c 53 3e 28 26 73 65 6c 66 n clean<S>(&self
1b00: 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 , source_id: &i3
1b10: 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 2, owner: S) ->
1b20: 52 65 73 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 Result<Cow<'_, s
1b30: 74 72 3e 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 tr>>..where S: I
1b40: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
1b50: 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 owner = owner.i
1b60: 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 nto();....let mu
1b70: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
1b80: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
1b90: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
1ba0: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1bb0: 43 6c 65 61 6e 20 66 65 74 63 68 20 63 6f 6e 6e Clean fetch conn
1bc0: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1bd0: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 pool))?;...match
1be0: 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 sqlx::query("de
1bf0: 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 74 67 5f lete from rsstg_
1c00: 70 6f 73 74 20 70 20 75 73 69 6e 67 20 72 73 73 post p using rss
1c10: 74 67 5f 73 6f 75 72 63 65 20 73 20 77 68 65 72 tg_source s wher
1c20: 65 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 e p.source_id =
1c30: 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 $1 and owner = $
1c40: 32 20 61 6e 64 20 70 2e 73 6f 75 72 63 65 5f 69 2 and p.source_i
1c50: 64 20 3d 20 73 2e 73 6f 75 72 63 65 5f 69 64 3b d = s.source_id;
1c60: 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 ").....bind(sour
1c70: 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 ce_id).....bind(
1c80: 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 owner).....execu
1c90: 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 te(&mut conn).aw
1ca0: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
1cb0: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1cc0: 22 43 6c 65 61 6e 20 73 65 65 6e 20 70 6f 73 74 "Clean seen post
1cd0: 73 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 s:\n{:?}", &self
1ce0: 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 .pool))?.....row
1cf0: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
1d00: 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 ..0 => { Ok("No
1d10: 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 data found found
1d20: 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 .".into()) },...
1d30: 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 .x => { Ok(forma
1d40: 74 21 28 22 7b 7d 20 70 6f 73 74 73 20 70 75 72 t!("{} posts pur
1d50: 67 65 64 2e 22 2c 20 78 29 2e 69 6e 74 6f 28 29 ged.", x).into()
1d60: 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 ) },...}..}...pu
1d70: 62 20 61 73 79 6e 63 20 66 6e 20 65 6e 61 62 6c b async fn enabl
1d80: 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 e<S>(&self, sour
1d90: 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e ce_id: &i32, own
1da0: 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 er: S) -> Result
1db0: 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a <&str>..where S:
1dc0: 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c Into<i64> {...l
1dd0: 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 et owner = owner
1de0: 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 .into();....let
1df0: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
1e00: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
1e10: 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f wait.....with_co
1e20: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
1e30: 28 22 45 6e 61 62 6c 65 20 66 65 74 63 68 20 63 ("Enable fetch c
1e40: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
1e50: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 lf.pool))?;...ma
1e60: 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 tch sqlx::query(
1e70: 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f "update rsstg_so
1e80: 75 72 63 65 20 73 65 74 20 65 6e 61 62 6c 65 64 urce set enabled
1e90: 20 3d 20 74 72 75 65 20 77 68 65 72 65 20 73 6f = true where so
1ea0: 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 urce_id = $1 and
1eb0: 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 owner = $2")...
1ec0: 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 ..bind(source_id
1ed0: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
1ee0: 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d ).....execute(&m
1ef0: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 ut conn).await..
1f00: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1f10: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 45 6e 61 62 || format!("Enab
1f20: 6c 65 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d le source:\n{:?}
1f30: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
1f40: 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 .....rows_affect
1f50: 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b ed() {....1 => {
1f60: 20 4f 6b 28 22 53 6f 75 72 63 65 20 65 6e 61 62 Ok("Source enab
1f70: 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d led.") },....0 =
1f80: 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e > { Ok("Source n
1f90: 6f 74 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 ot found.") },..
1fa0: 09 09 5f 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 .._ => { Err(any
1fb0: 68 6f 77 21 28 22 44 61 74 61 62 61 73 65 20 65 how!("Database e
1fc0: 72 72 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a rror.")) },...}.
1fd0: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
1fe0: 6e 20 64 69 73 61 62 6c 65 3c 53 3e 28 26 73 65 n disable<S>(&se
1ff0: 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 lf, source_id: &
2000: 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d i32, owner: S) -
2010: 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 > Result<&str>..
2020: 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 where S: Into<i6
2030: 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 4> {...let owner
2040: 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b = owner.into();
2050: 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ....let mut conn
2060: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
2070: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
2080: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
2090: 20 66 6f 72 6d 61 74 21 28 22 44 69 73 61 62 6c format!("Disabl
20a0: 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b e fetch conn:\n{
20b0: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
20c0: 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c ))?;...match sql
20d0: 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 x::query("update
20e0: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 rsstg_source se
20f0: 74 20 65 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 t enabled = fals
2100: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
2110: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
2120: 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 = $2").....bind
2130: 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e (source_id).....
2140: 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e bind(owner).....
2150: 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e execute(&mut con
2160: 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 n).await.....wit
2170: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
2180: 6d 61 74 21 28 22 44 69 73 61 62 6c 65 20 73 6f mat!("Disable so
2190: 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 urce:\n{:?}", &s
21a0: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e elf.pool))?.....
21b0: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 rows_affected()
21c0: 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 {....1 => { Ok("
21d0: 53 6f 75 72 63 65 20 64 69 73 61 62 6c 65 64 2e Source disabled.
21e0: 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 ") },....0 => {
21f0: 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 Ok("Source not f
2200: 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 ound.") },...._
2210: 3d 3e 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 => { Err(anyhow!
2220: 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 ("Database error
2230: 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a .")) },...}..}..
2240: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 75 70 .pub async fn up
2250: 64 61 74 65 3c 53 3e 28 26 73 65 6c 66 2c 20 75 date<S>(&self, u
2260: 70 64 61 74 65 3a 20 4f 70 74 69 6f 6e 3c 69 33 pdate: Option<i3
2270: 32 3e 2c 20 63 68 61 6e 6e 65 6c 3a 20 26 73 74 2>, channel: &st
2280: 72 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 r, channel_id: i
2290: 36 34 2c 20 75 72 6c 3a 20 26 73 74 72 2c 20 69 64, url: &str, i
22a0: 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 v_hash: Option<&
22b0: 73 74 72 3e 2c 20 75 72 6c 5f 72 65 3a 20 4f 70 str>, url_re: Op
22c0: 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 tion<&str>, owne
22d0: 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c r: S) -> Result<
22e0: 26 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 &str>..where S:
22f0: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
2300: 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e t owner = owner.
2310: 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d into();....let m
2320: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 ut conn = self.p
2330: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
2340: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
2350: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
2360: 22 55 70 64 61 74 65 20 66 65 74 63 68 20 63 6f "Update fetch co
2370: 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c nn:\n{:?}", &sel
2380: 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 0a 09 09 6d 61 f.pool))?;....ma
2390: 74 63 68 20 6d 61 74 63 68 20 75 70 64 61 74 65 tch match update
23a0: 20 7b 0a 09 09 09 09 53 6f 6d 65 28 69 64 29 20 {.....Some(id)
23b0: 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a => {......sqlx::
23c0: 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 73 query("update rs
23d0: 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 63 stg_source set c
23e0: 68 61 6e 6e 65 6c 5f 69 64 20 3d 20 24 32 2c 20 hannel_id = $2,
23f0: 75 72 6c 20 3d 20 24 33 2c 20 69 76 5f 68 61 73 url = $3, iv_has
2400: 68 20 3d 20 24 34 2c 20 6f 77 6e 65 72 20 3d 20 h = $4, owner =
2410: 24 35 2c 20 63 68 61 6e 6e 65 6c 20 3d 20 24 36 $5, channel = $6
2420: 2c 20 75 72 6c 5f 72 65 20 3d 20 24 37 20 77 68 , url_re = $7 wh
2430: 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 ere source_id =
2440: 24 31 22 29 2e 62 69 6e 64 28 69 64 29 0a 09 09 $1").bind(id)...
2450: 09 09 7d 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e ..},.....None =>
2460: 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 {......sqlx::qu
2470: 65 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 6f ery("insert into
2480: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 28 63 rsstg_source (c
2490: 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 hannel_id, url,
24a0: 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 iv_hash, owner,
24b0: 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 channel, url_re)
24c0: 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 2c values ($1, $2,
24d0: 20 24 33 2c 20 24 34 2c 20 24 35 2c 20 24 36 29 $3, $4, $5, $6)
24e0: 22 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 ").....},....}..
24f0: 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c 5f ...bind(channel_
2500: 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c id).....bind(url
2510: 29 0a 09 09 09 2e 62 69 6e 64 28 69 76 5f 68 61 ).....bind(iv_ha
2520: 73 68 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e sh).....bind(own
2530: 65 72 29 0a 09 09 09 2e 62 69 6e 64 28 63 68 61 er).....bind(cha
2540: 6e 6e 65 6c 29 0a 09 09 09 2e 62 69 6e 64 28 75 nnel).....bind(u
2550: 72 6c 5f 72 65 29 0a 09 09 09 2e 65 78 65 63 75 rl_re).....execu
2560: 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 te(&mut conn).aw
2570: 61 69 74 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d ait {....Ok(_) =
2580: 3e 20 4f 6b 28 6d 61 74 63 68 20 75 70 64 61 74 > Ok(match updat
2590: 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 5f 29 20 e {.....Some(_)
25a0: 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 75 70 64 61 => "Channel upda
25b0: 74 65 64 2e 22 2c 0a 09 09 09 09 4e 6f 6e 65 20 ted.",.....None
25c0: 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 61 64 64 65 => "Channel adde
25d0: 64 2e 22 2c 0a 09 09 09 7d 29 2c 0a 09 09 09 45 d.",....}),....E
25e0: 72 72 28 73 71 6c 78 3a 3a 45 72 72 6f 72 3a 3a rr(sqlx::Error::
25f0: 44 61 74 61 62 61 73 65 28 65 72 72 29 29 20 3d Database(err)) =
2600: 3e 20 7b 0a 09 09 09 09 6d 61 74 63 68 20 65 72 > {.....match er
2610: 72 2e 64 6f 77 6e 63 61 73 74 3a 3a 3c 73 71 6c r.downcast::<sql
2620: 78 3a 3a 70 6f 73 74 67 72 65 73 3a 3a 50 67 44 x::postgres::PgD
2630: 61 74 61 62 61 73 65 45 72 72 6f 72 3e 28 29 2e atabaseError>().
2640: 72 6f 75 74 69 6e 65 28 29 20 7b 0a 09 09 09 09 routine() {.....
2650: 09 53 6f 6d 65 28 22 5f 62 74 5f 63 68 65 63 6b .Some("_bt_check
2660: 5f 75 6e 69 71 75 65 22 2c 20 29 20 3d 3e 20 7b _unique", ) => {
2670: 0a 09 09 09 09 09 09 4f 6b 28 22 44 75 70 6c 69 .......Ok("Dupli
2680: 63 61 74 65 20 6b 65 79 2e 22 29 0a 09 09 09 09 cate key.").....
2690: 09 7d 2c 0a 09 09 09 09 09 53 6f 6d 65 28 5f 29 .},......Some(_)
26a0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 => {.......Ok("
26b0: 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 Database error."
26c0: 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 4e )......},......N
26d0: 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f one => {.......O
26e0: 6b 28 22 4e 6f 20 64 61 74 61 62 61 73 65 20 65 k("No database e
26f0: 72 72 6f 72 20 65 78 74 72 61 63 74 65 64 2e 22 rror extracted."
2700: 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 7d 0a )......},.....}.
2710: 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 65 72 72 ...},....Err(err
2720: 29 20 3d 3e 20 7b 0a 09 09 09 09 62 61 69 6c 21 ) => {.....bail!
2730: 28 22 53 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e ("Sorry, unknown
2740: 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 3f 7d 5c 6e error:\n{:#?}\n
2750: 22 2c 20 65 72 72 29 3b 0a 09 09 09 7d 2c 0a 09 ", err);....},..
2760: 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 66 6e .}..}...async fn
2770: 20 61 75 74 6f 66 65 74 63 68 28 26 73 65 6c 66 autofetch(&self
2780: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 73 74 64 3a ) -> Result<std:
2790: 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3e :time::Duration>
27a0: 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 64 65 6c {...let mut del
27b0: 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 72 ay = chrono::Dur
27c0: 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 31 ation::minutes(1
27d0: 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e );...let mut con
27e0: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
27f0: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
2800: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
2810: 7c 20 66 6f 72 6d 61 74 21 28 22 41 75 74 6f 66 | format!("Autof
2820: 65 74 63 68 20 66 65 74 63 68 20 63 6f 6e 6e 3a etch fetch conn:
2830: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
2840: 6f 6f 6c 29 29 3f 3b 0a 09 09 6c 65 74 20 6e 6f ool))?;...let no
2850: 77 20 3d 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 w = chrono::Loca
2860: 6c 3a 3a 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 20 l::now();...let
2870: 6d 75 74 20 71 75 65 75 65 20 3d 20 73 71 6c 78 mut queue = sqlx
2880: 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 ::query("select
2890: 73 6f 75 72 63 65 5f 69 64 2c 20 6e 65 78 74 5f source_id, next_
28a0: 66 65 74 63 68 2c 20 6f 77 6e 65 72 20 66 72 6f fetch, owner fro
28b0: 6d 20 72 73 73 74 67 5f 6f 72 64 65 72 20 6e 61 m rsstg_order na
28c0: 74 75 72 61 6c 20 6c 65 66 74 20 6a 6f 69 6e 20 tural left join
28d0: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 rsstg_source whe
28e0: 72 65 20 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 re next_fetch <
28f0: 6e 6f 77 28 29 20 2b 20 69 6e 74 65 72 76 61 6c now() + interval
2900: 20 27 31 20 6d 69 6e 75 74 65 27 3b 22 29 0a 09 '1 minute';")..
2910: 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 ...fetch_all(&mu
2920: 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a t conn).await?;.
2930: 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 71 75 65 ..for row in que
2940: 75 65 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 6c ue.iter() {....l
2950: 65 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 et source_id: i3
2960: 32 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 2 = row.try_get(
2970: 22 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 "source_id")?;..
2980: 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 ..let owner: i64
2990: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
29a0: 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 09 6c 65 74 owner")?;....let
29b0: 20 6e 65 78 74 5f 66 65 74 63 68 3a 20 44 61 74 next_fetch: Dat
29c0: 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 4c 6f eTime<chrono::Lo
29d0: 63 61 6c 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 cal> = row.try_g
29e0: 65 74 28 22 6e 65 78 74 5f 66 65 74 63 68 22 29 et("next_fetch")
29f0: 3f 3b 0a 09 09 09 69 66 20 6e 65 78 74 5f 66 65 ?;....if next_fe
2a00: 74 63 68 20 3c 20 6e 6f 77 20 7b 0a 09 09 09 09 tch < now {.....
2a10: 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 65 let clone = Core
2a20: 20 7b 0a 09 09 09 09 09 6f 77 6e 65 72 5f 63 68 {......owner_ch
2a30: 61 74 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 at: telegram_bot
2a40: 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 ::UserId::new(ow
2a50: 6e 65 72 29 2c 0a 09 09 09 09 09 2e 2e 73 65 6c ner),........sel
2a60: 66 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 7d 3b f.clone().....};
2a70: 0a 09 09 09 09 74 61 73 6b 3a 3a 73 70 61 77 6e .....task::spawn
2a80: 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 (async move {...
2a90: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
2aa0: 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b r) = clone.check
2ab0: 28 26 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e (&source_id, own
2ac0: 65 72 2c 20 74 72 75 65 29 2e 61 77 61 69 74 20 er, true).await
2ad0: 7b 0a 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 {.......if let E
2ae0: 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e rr(err) = clone.
2af0: 73 65 6e 64 28 26 66 6f 72 6d 61 74 21 28 22 f0 send(&format!("š
2b00: 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c {:?}", err),
2b10: 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 61 None, None).awa
2b20: 69 74 20 7b 0a 09 09 09 09 09 09 09 65 70 72 69 it {........epri
2b30: 6e 74 6c 6e 21 28 22 43 68 65 63 6b 20 65 72 72 ntln!("Check err
2b40: 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 or: {}", err);..
2b50: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 7d 3b 0a .....};......};.
2b60: 09 09 09 09 7d 29 3b 0a 09 09 09 7d 20 65 6c 73 ....});....} els
2b70: 65 20 69 66 20 6e 65 78 74 5f 66 65 74 63 68 20 e if next_fetch
2b80: 2d 20 6e 6f 77 20 3c 20 64 65 6c 61 79 20 7b 0a - now < delay {.
2b90: 09 09 09 09 64 65 6c 61 79 20 3d 20 6e 65 78 74 ....delay = next
2ba0: 5f 66 65 74 63 68 20 2d 20 6e 6f 77 3b 0a 09 09 _fetch - now;...
2bb0: 09 7d 0a 09 09 7d 3b 0a 09 09 71 75 65 75 65 2e .}...};...queue.
2bc0: 63 6c 65 61 72 28 29 3b 0a 09 09 4f 6b 28 64 65 clear();...Ok(de
2bd0: 6c 61 79 2e 74 6f 5f 73 74 64 28 29 3f 29 0a 09 lay.to_std()?)..
2be0: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
2bf0: 20 6c 69 73 74 3c 53 3e 28 26 73 65 6c 66 2c 20 list<S>(&self,
2c00: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
2c10: 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 ult<String>..whe
2c20: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
2c30: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
2c40: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 owner.into();...
2c50: 09 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 3a 20 .let mut reply:
2c60: 56 65 63 3c 43 6f 77 3c 73 74 72 3e 3e 20 3d 20 Vec<Cow<str>> =
2c70: 76 65 63 21 5b 5d 3b 0a 09 09 6c 65 74 20 6d 75 vec![];...let mu
2c80: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
2c90: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
2ca0: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
2cb0: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
2cc0: 4c 69 73 74 20 66 65 74 63 68 20 63 6f 6e 6e 3a List fetch conn:
2cd0: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
2ce0: 6f 6f 6c 29 29 3f 3b 0a 09 09 72 65 70 6c 79 2e ool))?;...reply.
2cf0: 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 push("Channels:"
2d00: 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 6c 65 74 20 .into());...let
2d10: 72 6f 77 73 20 3d 20 73 71 6c 78 3a 3a 71 75 65 rows = sqlx::que
2d20: 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 ry("select sourc
2d30: 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 65 e_id, channel, e
2d40: 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 76 5f nabled, url, iv_
2d50: 68 61 73 68 2c 20 75 72 6c 5f 72 65 20 66 72 6f hash, url_re fro
2d60: 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 m rsstg_source w
2d70: 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 31 20 here owner = $1
2d80: 6f 72 64 65 72 20 62 79 20 73 6f 75 72 63 65 5f order by source_
2d90: 69 64 22 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 id").....bind(ow
2da0: 6e 65 72 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 ner).....fetch_a
2db0: 6c 6c 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 ll(&mut conn).aw
2dc0: 61 69 74 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 ait?;...for row
2dd0: 69 6e 20 72 6f 77 73 2e 69 74 65 72 28 29 20 7b in rows.iter() {
2de0: 0a 09 09 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 ....let source_i
2df0: 64 3a 20 69 33 32 20 3d 20 72 6f 77 2e 74 72 79 d: i32 = row.try
2e00: 5f 67 65 74 28 22 73 6f 75 72 63 65 5f 69 64 22 _get("source_id"
2e10: 29 3f 3b 0a 09 09 09 6c 65 74 20 75 73 65 72 6e )?;....let usern
2e20: 61 6d 65 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e ame: &str = row.
2e30: 74 72 79 5f 67 65 74 28 22 63 68 61 6e 6e 65 6c try_get("channel
2e40: 22 29 3f 3b 0a 09 09 09 6c 65 74 20 65 6e 61 62 ")?;....let enab
2e50: 6c 65 64 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e led: bool = row.
2e60: 74 72 79 5f 67 65 74 28 22 65 6e 61 62 6c 65 64 try_get("enabled
2e70: 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c 3a ")?;....let url:
2e80: 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f &str = row.try_
2e90: 67 65 74 28 22 75 72 6c 22 29 3f 3b 0a 09 09 09 get("url")?;....
2ea0: 6c 65 74 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 let iv_hash: Opt
2eb0: 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e ion<&str> = row.
2ec0: 74 72 79 5f 67 65 74 28 22 69 76 5f 68 61 73 68 try_get("iv_hash
2ed0: 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c 5f ")?;....let url_
2ee0: 72 65 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e re: Option<&str>
2ef0: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
2f00: 75 72 6c 5f 72 65 22 29 3f 3b 0a 09 09 09 72 65 url_re")?;....re
2f10: 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 ply.push(format!
2f20: 28 22 5c 6e 5c 5c 23 ef b8 8f e2 83 a3 20 7b 7d ("\n\\#ļøā£ {}
2f30: 20 5c 5c 2a ef b8 8f e2 83 a3 20 60 7b 7d 60 20 \\*ļøā£ `{}`
2f40: 7b 7d 5c 6e f0 9f 94 97 20 60 7b 7d 60 22 2c 20 {}\nš `{}`",
2f50: 73 6f 75 72 63 65 5f 69 64 2c 20 75 73 65 72 6e source_id, usern
2f60: 61 6d 65 2c 20 20 0a 09 09 09 09 6d 61 74 63 68 ame, .....match
2f70: 20 65 6e 61 62 6c 65 64 20 7b 0a 09 09 09 09 09 enabled {......
2f80: 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 20 65 true => "š e
2f90: 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 66 61 nabled",......fa
2fa0: 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 73 61 lse => "ā disa
2fb0: 62 6c 65 64 22 2c 0a 09 09 09 09 7d 2c 20 75 72 bled",.....}, ur
2fc0: 6c 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 69 l).into());....i
2fd0: 66 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 68 29 f let Some(hash)
2fe0: 20 3d 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 = iv_hash {....
2ff0: 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d .reply.push(form
3000: 61 74 21 28 22 49 56 3a 20 60 7b 7d 60 22 2c 20 at!("IV: `{}`",
3010: 68 61 73 68 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 hash).into());..
3020: 09 09 7d 0a 09 09 09 69 66 20 6c 65 74 20 53 6f ..}....if let So
3030: 6d 65 28 72 65 29 20 3d 20 75 72 6c 5f 72 65 20 me(re) = url_re
3040: 7b 0a 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 {.....reply.push
3050: 28 66 6f 72 6d 61 74 21 28 22 52 45 3a 20 60 7b (format!("RE: `{
3060: 7d 60 22 2c 20 72 65 29 2e 69 6e 74 6f 28 29 29 }`", re).into())
3070: 3b 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 4f 6b ;....}...};...Ok
3080: 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 (reply.join("\n"
3090: 29 29 0a 09 7d 0a 7d 0a ))..}.}.