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 70 72 6f 78 79 20 3d 20 73 65 ..let proxy = se
02d0: 74 74 69 6e 67 73 2e 67 65 74 5f 73 74 72 69 6e ttings.get_strin
02e0: 67 28 22 70 72 6f 78 79 22 29 3f 3b 0a 09 09 6c g("proxy")?;...l
02f0: 65 74 20 6d 75 74 20 63 6c 69 65 6e 74 20 3d 20 et mut client =
0300: 72 65 71 77 65 73 74 3a 3a 43 6c 69 65 6e 74 3a reqwest::Client:
0310: 3a 62 75 69 6c 64 65 72 28 29 3b 0a 09 09 69 66 :builder();...if
0320: 20 21 70 72 6f 78 79 2e 69 73 5f 65 6d 70 74 79 !proxy.is_empty
0330: 28 29 20 7b 0a 09 09 09 6c 65 74 20 70 72 6f 78 () {....let prox
0340: 79 20 3d 20 72 65 71 77 65 73 74 3a 3a 50 72 6f y = reqwest::Pro
0350: 78 79 3a 3a 61 6c 6c 28 70 72 6f 78 79 29 3f 3b xy::all(proxy)?;
0360: 0a 09 09 09 63 6c 69 65 6e 74 20 3d 20 63 6c 69 ....client = cli
0370: 65 6e 74 2e 70 72 6f 78 79 28 70 72 6f 78 79 29 ent.proxy(proxy)
0380: 3b 0a 09 09 7d 0a 09 09 6c 65 74 20 68 74 74 70 ;...}...let http
0390: 5f 63 6c 69 65 6e 74 20 3d 20 63 6c 69 65 6e 74 _client = client
03a0: 2e 62 75 69 6c 64 28 29 3f 3b 0a 09 09 6c 65 74 .build()?;...let
03b0: 20 63 6f 72 65 20 3d 20 41 72 63 3a 3a 6e 65 77 core = Arc::new
03c0: 28 43 6f 72 65 20 7b 0a 09 09 09 74 67 2c 0a 09 (Core {....tg,..
03d0: 09 09 6d 79 3a 20 74 61 73 6b 3a 3a 62 6c 6f 63 ..my: task::bloc
03e0: 6b 5f 6f 6e 28 61 73 79 6e 63 20 7b 0a 09 09 09 k_on(async {....
03f0: 09 74 67 5f 63 6c 6f 6e 65 64 2e 73 65 6e 64 28 .tg_cloned.send(
0400: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 47 65 telegram_bot::Ge
0410: 74 4d 65 29 2e 61 77 61 69 74 0a 09 09 09 7d 29 tMe).await....})
0420: 3f 2c 0a 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 ?,....owner_chat
0430: 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a : telegram_bot::
0440: 55 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 UserId::new(owne
0450: 72 29 2c 0a 09 09 09 70 6f 6f 6c 3a 20 50 67 50 r),....pool: PgP
0460: 6f 6f 6c 4f 70 74 69 6f 6e 73 3a 3a 6e 65 77 28 oolOptions::new(
0470: 29 0a 09 09 09 09 2e 6d 61 78 5f 63 6f 6e 6e 65 )......max_conne
0480: 63 74 69 6f 6e 73 28 35 29 0a 09 09 09 09 2e 61 ctions(5)......a
0490: 63 71 75 69 72 65 5f 74 69 6d 65 6f 75 74 28 73 cquire_timeout(s
04a0: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
04b0: 6f 6e 3a 3a 6e 65 77 28 33 30 30 2c 20 30 29 29 on::new(300, 0))
04c0: 0a 09 09 09 09 2e 69 64 6c 65 5f 74 69 6d 65 6f ......idle_timeo
04d0: 75 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 ut(std::time::Du
04e0: 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 36 30 2c 20 ration::new(60,
04f0: 30 29 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 74 0))......connect
0500: 5f 6c 61 7a 79 28 26 73 65 74 74 69 6e 67 73 2e _lazy(&settings.
0510: 67 65 74 5f 73 74 72 69 6e 67 28 22 70 67 22 29 get_string("pg")
0520: 3f 29 3f 2c 0a 09 09 09 73 6f 75 72 63 65 73 3a ?)?,....sources:
0530: 20 41 72 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a Arc::new(Mutex:
0540: 3a 6e 65 77 28 48 61 73 68 53 65 74 3a 3a 6e 65 :new(HashSet::ne
0550: 77 28 29 29 29 2c 0a 09 09 09 68 74 74 70 5f 63 w())),....http_c
0560: 6c 69 65 6e 74 2c 0a 09 09 7d 29 3b 0a 09 09 6c lient,...});...l
0570: 65 74 20 63 6c 6f 6e 65 20 3d 20 63 6f 72 65 2e et clone = core.
0580: 63 6c 6f 6e 65 28 29 3b 0a 09 09 74 61 73 6b 3a clone();...task:
0590: 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 6d 6f 76 :spawn(async mov
05a0: 65 20 7b 0a 09 09 09 6c 6f 6f 70 20 7b 0a 09 09 e {....loop {...
05b0: 09 09 6c 65 74 20 64 65 6c 61 79 20 3d 20 6d 61 ..let delay = ma
05c0: 74 63 68 20 26 63 6c 6f 6e 65 2e 61 75 74 6f 66 tch &clone.autof
05d0: 65 74 63 68 28 29 2e 61 77 61 69 74 20 7b 0a 09 etch().await {..
05e0: 09 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 ....Err(err) =>
05f0: 7b 0a 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 {.......if let E
0600: 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e rr(err) = clone.
0610: 73 65 6e 64 28 66 6f 72 6d 61 74 21 28 22 f0 9f send(format!("š
0620: 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 {:?}", err),
0630: 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 None, None).awai
0640: 74 20 7b 0a 09 09 09 09 09 09 09 65 70 72 69 6e t {........eprin
0650: 74 6c 6e 21 28 22 41 75 74 6f 66 65 74 63 68 20 tln!("Autofetch
0660: 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 29 error: {}", err)
0670: 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ;.......};......
0680: 09 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 .std::time::Dura
0690: 74 69 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 63 73 28 tion::from_secs(
06a0: 36 30 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 60)......},.....
06b0: 09 4f 6b 28 74 69 6d 65 29 20 3d 3e 20 2a 74 69 .Ok(time) => *ti
06c0: 6d 65 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 74 me,.....};.....t
06d0: 61 73 6b 3a 3a 73 6c 65 65 70 28 64 65 6c 61 79 ask::sleep(delay
06e0: 29 2e 61 77 61 69 74 3b 0a 09 09 09 7d 0a 09 09 ).await;....}...
06f0: 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 0a 09 });...Ok(core)..
0700: 7d 0a 0a 09 70 75 62 20 66 6e 20 73 74 72 65 61 }...pub fn strea
0710: 6d 28 26 73 65 6c 66 29 20 2d 3e 20 74 65 6c 65 m(&self) -> tele
0720: 67 72 61 6d 5f 62 6f 74 3a 3a 55 70 64 61 74 65 gram_bot::Update
0730: 73 53 74 72 65 61 6d 20 7b 0a 09 09 73 65 6c 66 sStream {...self
0740: 2e 74 67 2e 73 74 72 65 61 6d 28 29 0a 09 7d 0a .tg.stream()..}.
0750: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 73 ..pub async fn s
0760: 65 6e 64 3c 27 61 2c 20 53 3e 28 26 73 65 6c 66 end<'a, S>(&self
0770: 2c 20 6d 73 67 3a 20 53 2c 20 74 61 72 67 65 74 , msg: S, target
0780: 3a 20 4f 70 74 69 6f 6e 3c 74 65 6c 65 67 72 61 : Option<telegra
0790: 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3e 2c 20 m_bot::UserId>,
07a0: 6d 6f 64 65 3a 20 4f 70 74 69 6f 6e 3c 74 65 6c mode: Option<tel
07b0: 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 egram_bot::types
07c0: 3a 3a 50 61 72 73 65 4d 6f 64 65 3e 29 20 2d 3e ::ParseMode>) ->
07d0: 20 52 65 73 75 6c 74 3c 28 29 3e 0a 09 77 68 65 Result<()>..whe
07e0: 72 65 20 53 3a 20 49 6e 74 6f 3c 43 6f 77 3c 27 re S: Into<Cow<'
07f0: 61 2c 20 73 74 72 3e 3e 20 7b 0a 09 09 6c 65 74 a, str>> {...let
0800: 20 6d 6f 64 65 20 3d 20 6d 6f 64 65 2e 75 6e 77 mode = mode.unw
0810: 72 61 70 5f 6f 72 28 74 65 6c 65 67 72 61 6d 5f rap_or(telegram_
0820: 62 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 61 72 73 bot::types::Pars
0830: 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 3b 0a 09 09 eMode::Html);...
0840: 6c 65 74 20 74 61 72 67 65 74 20 3d 20 74 61 72 let target = tar
0850: 67 65 74 2e 75 6e 77 72 61 70 5f 6f 72 28 73 65 get.unwrap_or(se
0860: 6c 66 2e 6f 77 6e 65 72 5f 63 68 61 74 29 3b 0a lf.owner_chat);.
0870: 09 09 73 65 6c 66 2e 74 67 2e 73 65 6e 64 28 74 ..self.tg.send(t
0880: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e elegram_bot::Sen
0890: 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 74 61 dMessage::new(ta
08a0: 72 67 65 74 2c 20 6d 73 67 29 2e 70 61 72 73 65 rget, msg).parse
08b0: 5f 6d 6f 64 65 28 6d 6f 64 65 29 29 2e 61 77 61 _mode(mode)).awa
08c0: 69 74 3f 3b 0a 09 09 4f 6b 28 28 29 29 0a 09 7d it?;...Ok(())..}
08d0: 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 ...pub async fn
08e0: 63 68 65 63 6b 3c 53 3e 28 26 73 65 6c 66 2c 20 check<S>(&self,
08f0: 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a id: &i32, owner:
0900: 20 53 2c 20 72 65 61 6c 3a 20 62 6f 6f 6c 29 20 S, real: bool)
0910: 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 3c 27 5f -> Result<Cow<'_
0920: 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 65 20 53 , str>>..where S
0930: 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 : Into<i64> {...
0940: 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 let owner = owne
0950: 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 r.into();....let
0960: 20 6d 75 74 20 70 6f 73 74 65 64 3a 20 69 33 32 mut posted: i32
0970: 20 3d 20 30 3b 0a 09 09 6c 65 74 20 69 64 20 3d = 0;...let id =
0980: 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 73 65 {....let mut se
0990: 74 20 3d 20 73 65 6c 66 2e 73 6f 75 72 63 65 73 t = self.sources
09a0: 2e 6c 6f 63 6b 28 29 2e 75 6e 77 72 61 70 28 29 .lock().unwrap()
09b0: 3b 0a 09 09 09 6d 61 74 63 68 20 73 65 74 2e 67 ;....match set.g
09c0: 65 74 28 69 64 29 20 7b 0a 09 09 09 09 53 6f 6d et(id) {.....Som
09d0: 65 28 69 64 29 20 3d 3e 20 69 64 2e 63 6c 6f 6e e(id) => id.clon
09e0: 65 28 29 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e e(),.....None =>
09f0: 20 7b 0a 09 09 09 09 09 6c 65 74 20 69 64 20 3d {......let id =
0a00: 20 41 72 63 3a 3a 6e 65 77 28 2a 69 64 29 3b 0a Arc::new(*id);.
0a10: 09 09 09 09 09 73 65 74 2e 69 6e 73 65 72 74 28 .....set.insert(
0a20: 69 64 2e 63 6c 6f 6e 65 28 29 29 3b 0a 09 09 09 id.clone());....
0a30: 09 09 69 64 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 ..id.clone()....
0a40: 09 7d 2c 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 .},....}...};...
0a50: 6c 65 74 20 63 6f 75 6e 74 20 3d 20 41 72 63 3a let count = Arc:
0a60: 3a 73 74 72 6f 6e 67 5f 63 6f 75 6e 74 28 26 69 :strong_count(&i
0a70: 64 29 3b 0a 09 09 69 66 20 63 6f 75 6e 74 20 3d d);...if count =
0a80: 3d 20 32 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 = 2 {....let mut
0a90: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
0aa0: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
0ab0: 74 0a 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 t......with_cont
0ac0: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
0ad0: 51 75 65 72 79 20 71 75 65 75 65 20 66 65 74 63 Query queue fetc
0ae0: 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 h conn:\n{:?}",
0af0: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 &self.pool))?;..
0b00: 09 09 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c 78 ..let row = sqlx
0b10: 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 ::query("select
0b20: 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e source_id, chann
0b30: 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 el_id, url, iv_h
0b40: 61 73 68 2c 20 6f 77 6e 65 72 2c 20 75 72 6c 5f ash, owner, url_
0b50: 72 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f re from rsstg_so
0b60: 75 72 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 urce where sourc
0b70: 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 e_id = $1 and ow
0b80: 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 09 09 2e ner = $2")......
0b90: 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 2e 62 bind(*id)......b
0ba0: 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 09 2e ind(owner)......
0bb0: 66 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 63 fetch_one(&mut c
0bc0: 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 2e onn).await......
0bd0: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
0be0: 66 6f 72 6d 61 74 21 28 22 51 75 65 72 79 20 73 format!("Query s
0bf0: 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 ource:\n{:?}", &
0c00: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 self.pool))?;...
0c10: 09 64 72 6f 70 28 63 6f 6e 6e 29 3b 0a 0a 09 09 .drop(conn);....
0c20: 09 6c 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 3a .let channel_id:
0c30: 20 69 36 34 20 3d 20 72 6f 77 2e 74 72 79 5f 67 i64 = row.try_g
0c40: 65 74 28 22 63 68 61 6e 6e 65 6c 5f 69 64 22 29 et("channel_id")
0c50: 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 ?;....let url: &
0c60: 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 str = row.try_ge
0c70: 74 28 22 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 t("url")?;....le
0c80: 74 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f t iv_hash: Optio
0c90: 6e 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 n<&str> = row.tr
0ca0: 79 5f 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 y_get("iv_hash")
0cb0: 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c 5f 72 65 ?;....let url_re
0cc0: 20 3d 20 6d 61 74 63 68 20 72 6f 77 2e 74 72 79 = match row.try
0cd0: 5f 67 65 74 28 22 75 72 6c 5f 72 65 22 29 3f 20 _get("url_re")?
0ce0: 7b 0a 09 09 09 09 53 6f 6d 65 28 78 29 20 3d 3e {.....Some(x) =>
0cf0: 20 53 6f 6d 65 28 73 65 64 72 65 67 65 78 3a 3a Some(sedregex::
0d00: 52 65 70 6c 61 63 65 43 6f 6d 6d 61 6e 64 3a 3a ReplaceCommand::
0d10: 6e 65 77 28 78 29 3f 29 2c 0a 09 09 09 09 4e 6f new(x)?),.....No
0d20: 6e 65 20 3d 3e 20 4e 6f 6e 65 2c 0a 09 09 09 7d ne => None,....}
0d30: 3b 0a 09 09 09 6c 65 74 20 64 65 73 74 69 6e 61 ;....let destina
0d40: 74 69 6f 6e 20 3d 20 6d 61 74 63 68 20 72 65 61 tion = match rea
0d50: 6c 20 7b 0a 09 09 09 09 74 72 75 65 20 3d 3e 20 l {.....true =>
0d60: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 telegram_bot::Us
0d70: 65 72 49 64 3a 3a 6e 65 77 28 63 68 61 6e 6e 65 erId::new(channe
0d80: 6c 5f 69 64 29 2c 0a 09 09 09 09 66 61 6c 73 65 l_id),.....false
0d90: 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 => telegram_bot
0da0: 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 72 6f ::UserId::new(ro
0db0: 77 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e 65 72 w.try_get("owner
0dc0: 22 29 3f 29 2c 0a 09 09 09 7d 3b 0a 09 09 09 6c ")?),....};....l
0dd0: 65 74 20 6d 75 74 20 74 68 69 73 5f 66 65 74 63 et mut this_fetc
0de0: 68 3a 20 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 h: Option<DateTi
0df0: 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 me<chrono::Fixed
0e00: 4f 66 66 73 65 74 3e 3e 20 3d 20 4e 6f 6e 65 3b Offset>> = None;
0e10: 0a 09 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 ....let mut post
0e20: 73 3a 20 42 54 72 65 65 4d 61 70 3c 44 61 74 65 s: BTreeMap<Date
0e30: 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 Time<chrono::Fix
0e40: 65 64 4f 66 66 73 65 74 3e 2c 20 53 74 72 69 6e edOffset>, Strin
0e50: 67 3e 20 3d 20 42 54 72 65 65 4d 61 70 3a 3a 6e g> = BTreeMap::n
0e60: 65 77 28 29 3b 0a 0a 09 09 09 6c 65 74 20 72 65 ew();.....let re
0e70: 73 70 6f 6e 73 65 20 3d 20 73 65 6c 66 2e 68 74 sponse = self.ht
0e80: 74 70 5f 63 6c 69 65 6e 74 2e 67 65 74 28 75 72 tp_client.get(ur
0e90: 6c 29 2e 73 65 6e 64 28 29 2e 61 77 61 69 74 3f l).send().await?
0ea0: 3b 0a 09 09 09 6c 65 74 20 73 74 61 74 75 73 20 ;....let status
0eb0: 3d 20 72 65 73 70 6f 6e 73 65 2e 73 74 61 74 75 = response.statu
0ec0: 73 28 29 3b 0a 09 09 09 6c 65 74 20 63 6f 6e 74 s();....let cont
0ed0: 65 6e 74 20 3d 20 72 65 73 70 6f 6e 73 65 2e 62 ent = response.b
0ee0: 79 74 65 73 28 29 2e 61 77 61 69 74 3f 3b 0a 09 ytes().await?;..
0ef0: 09 09 6d 61 74 63 68 20 72 73 73 3a 3a 43 68 61 ..match rss::Cha
0f00: 6e 6e 65 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 nnel::read_from(
0f10: 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a &content[..]) {.
0f20: 09 09 09 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 ....Ok(feed) =>
0f30: 7b 0a 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 {......for item
0f40: 69 6e 20 66 65 65 64 2e 69 74 65 6d 73 28 29 20 in feed.items()
0f50: 7b 0a 09 09 09 09 09 09 69 66 20 6c 65 74 20 53 {.......if let S
0f60: 6f 6d 65 28 6c 69 6e 6b 29 20 3d 20 69 74 65 6d ome(link) = item
0f70: 2e 6c 69 6e 6b 28 29 20 7b 0a 09 09 09 09 09 09 .link() {.......
0f80: 09 6c 65 74 20 64 61 74 65 20 3d 20 6d 61 74 63 .let date = matc
0f90: 68 20 69 74 65 6d 2e 70 75 62 5f 64 61 74 65 28 h item.pub_date(
0fa0: 29 20 7b 0a 09 09 09 09 09 09 09 09 53 6f 6d 65 ) {.........Some
0fb0: 28 66 65 65 64 5f 64 61 74 65 29 20 3d 3e 20 44 (feed_date) => D
0fc0: 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 ateTime::parse_f
0fd0: 72 6f 6d 5f 72 66 63 32 38 32 32 28 66 65 65 64 rom_rfc2822(feed
0fe0: 5f 64 61 74 65 29 2c 0a 09 09 09 09 09 09 09 09 _date),.........
0ff0: 4e 6f 6e 65 20 3d 3e 20 44 61 74 65 54 69 6d 65 None => DateTime
1000: 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 ::parse_from_rfc
1010: 33 33 33 39 28 26 69 74 65 6d 2e 64 75 62 6c 69 3339(&item.dubli
1020: 6e 5f 63 6f 72 65 5f 65 78 74 28 29 2e 75 6e 77 n_core_ext().unw
1030: 72 61 70 28 29 2e 64 61 74 65 73 28 29 5b 30 5d rap().dates()[0]
1040: 29 2c 0a 09 09 09 09 09 09 09 7d 3f 3b 0a 09 09 ),........}?;...
1050: 09 09 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 6c .....let url = l
1060: 69 6e 6b 3b 0a 09 09 09 09 09 09 09 70 6f 73 74 ink;........post
1070: 73 2e 69 6e 73 65 72 74 28 64 61 74 65 2c 20 75 s.insert(date, u
1080: 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b rl.to_string());
1090: 0a 09 09 09 09 09 09 7d 0a 09 09 09 09 09 7d 3b .......}......};
10a0: 0a 09 09 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 .....},.....Err(
10b0: 65 72 72 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 err) => match er
10c0: 72 20 7b 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 r {......rss::Er
10d0: 72 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 ror::InvalidStar
10e0: 74 54 61 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 tTag => {.......
10f0: 6c 65 74 20 66 65 65 64 20 3d 20 61 74 6f 6d 5f let feed = atom_
1100: 73 79 6e 64 69 63 61 74 69 6f 6e 3a 3a 46 65 65 syndication::Fee
1110: 64 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f d::read_from(&co
1120: 6e 74 65 6e 74 5b 2e 2e 5d 29 0a 09 09 09 09 09 ntent[..])......
1130: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1140: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 50 72 6f 62 || format!("Prob
1150: 6c 65 6d 20 6f 70 65 6e 69 6e 67 20 66 65 65 64 lem opening feed
1160: 20 75 72 6c 3a 5c 6e 7b 7d 5c 6e 7b 7d 22 2c 20 url:\n{}\n{}",
1170: 26 75 72 6c 2c 20 73 74 61 74 75 73 29 29 3f 3b &url, status))?;
1180: 0a 09 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 .......for item
1190: 69 6e 20 66 65 65 64 2e 65 6e 74 72 69 65 73 28 in feed.entries(
11a0: 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 64 ) {........let d
11b0: 61 74 65 20 3d 20 69 74 65 6d 2e 70 75 62 6c 69 ate = item.publi
11c0: 73 68 65 64 28 29 2e 75 6e 77 72 61 70 28 29 3b shed().unwrap();
11d0: 0a 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c 20 ........let url
11e0: 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 29 5b 30 = item.links()[0
11f0: 5d 2e 68 72 65 66 28 29 3b 0a 09 09 09 09 09 09 ].href();.......
1200: 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 2a 64 .posts.insert(*d
1210: 61 74 65 2c 20 75 72 6c 2e 74 6f 5f 73 74 72 69 ate, url.to_stri
1220: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 7d 3b 0a ng());.......};.
1230: 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 72 73 73 .....},......rss
1240: 3a 3a 45 72 72 6f 72 3a 3a 45 6f 66 20 3d 3e 20 ::Error::Eof =>
1250: 28 29 2c 0a 09 09 09 09 09 5f 20 3d 3e 20 62 61 (),......_ => ba
1260: 69 6c 21 28 22 55 6e 73 75 70 70 6f 72 74 65 64 il!("Unsupported
1270: 20 6f 72 20 6d 61 6e 67 6c 65 64 20 63 6f 6e 74 or mangled cont
1280: 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b 3a 23 3f ent:\n{:?}\n{:#?
1290: 7d 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 26 75 72 }\n{:#?}\n", &ur
12a0: 6c 2c 20 65 72 72 2c 20 73 74 61 74 75 73 29 0a l, err, status).
12b0: 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 66 ....}....};....f
12c0: 6f 72 20 28 64 61 74 65 2c 20 75 72 6c 29 20 69 or (date, url) i
12d0: 6e 20 70 6f 73 74 73 2e 69 74 65 72 28 29 20 7b n posts.iter() {
12e0: 0a 09 09 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e .....let mut con
12f0: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
1300: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
1310: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
1320: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 68 65 (|| format!("Che
1330: 63 6b 20 70 6f 73 74 20 66 65 74 63 68 20 63 6f ck post fetch co
1340: 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c nn:\n{:?}", &sel
1350: 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 09 6c f.pool))?;.....l
1360: 65 74 20 70 6f 73 74 5f 75 72 6c 3a 20 43 6f 77 et post_url: Cow
1370: 3c 73 74 72 3e 20 3d 20 6d 61 74 63 68 20 75 72 <str> = match ur
1380: 6c 5f 72 65 20 7b 0a 09 09 09 09 09 53 6f 6d 65 l_re {......Some
1390: 28 72 65 66 20 78 29 20 3d 3e 20 78 2e 65 78 65 (ref x) => x.exe
13a0: 63 75 74 65 28 75 72 6c 29 2c 0a 09 09 09 09 09 cute(url),......
13b0: 4e 6f 6e 65 20 3d 3e 20 75 72 6c 2e 69 6e 74 6f None => url.into
13c0: 28 29 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 6c (),.....};.....l
13d0: 65 74 20 72 6f 77 20 3d 20 73 71 6c 78 3a 3a 71 et row = sqlx::q
13e0: 75 65 72 79 28 22 73 65 6c 65 63 74 20 65 78 69 uery("select exi
13f0: 73 74 73 28 73 65 6c 65 63 74 20 74 72 75 65 20 sts(select true
1400: 66 72 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 from rsstg_post
1410: 77 68 65 72 65 20 75 72 6c 20 3d 20 24 31 20 61 where url = $1 a
1420: 6e 64 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 nd source_id = $
1430: 32 29 20 61 73 20 65 78 69 73 74 73 3b 22 29 0a 2) as exists;").
1440: 09 09 09 09 09 2e 62 69 6e 64 28 26 2a 70 6f 73 ......bind(&*pos
1450: 74 5f 75 72 6c 29 0a 09 09 09 09 09 2e 62 69 6e t_url).......bin
1460: 64 28 2a 69 64 29 0a 09 09 09 09 09 2e 66 65 74 d(*id).......fet
1470: 63 68 5f 6f 6e 65 28 26 6d 75 74 20 63 6f 6e 6e ch_one(&mut conn
1480: 29 2e 61 77 61 69 74 0a 09 09 09 09 09 2e 77 69 ).await.......wi
1490: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
14a0: 72 6d 61 74 21 28 22 43 68 65 63 6b 20 70 6f 73 rmat!("Check pos
14b0: 74 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e t:\n{:?}", &conn
14c0: 29 29 3f 3b 0a 09 09 09 09 6c 65 74 20 65 78 69 ))?;.....let exi
14d0: 73 74 73 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e sts: bool = row.
14e0: 74 72 79 5f 67 65 74 28 22 65 78 69 73 74 73 22 try_get("exists"
14f0: 29 3f 3b 0a 09 09 09 09 69 66 20 21 20 65 78 69 )?;.....if ! exi
1500: 73 74 73 20 7b 0a 09 09 09 09 09 69 66 20 74 68 sts {......if th
1510: 69 73 5f 66 65 74 63 68 2e 69 73 5f 6e 6f 6e 65 is_fetch.is_none
1520: 28 29 20 7c 7c 20 2a 64 61 74 65 20 3e 20 74 68 () || *date > th
1530: 69 73 5f 66 65 74 63 68 2e 75 6e 77 72 61 70 28 is_fetch.unwrap(
1540: 29 20 7b 0a 09 09 09 09 09 09 74 68 69 73 5f 66 ) {.......this_f
1550: 65 74 63 68 20 3d 20 53 6f 6d 65 28 2a 64 61 74 etch = Some(*dat
1560: 65 29 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 e);......};.....
1570: 09 73 65 6c 66 2e 74 67 2e 73 65 6e 64 28 20 6d .self.tg.send( m
1580: 61 74 63 68 20 69 76 5f 68 61 73 68 20 7b 0a 09 atch iv_hash {..
1590: 09 09 09 09 09 09 53 6f 6d 65 28 68 61 73 68 29 ......Some(hash)
15a0: 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 => telegram_bot
15b0: 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e ::SendMessage::n
15c0: 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c 20 ew(destination,
15d0: 66 6f 72 6d 61 74 21 28 22 3c 61 20 68 72 65 66 format!("<a href
15e0: 3d 5c 22 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f =\"https://t.me/
15f0: 69 76 3f 75 72 6c 3d 7b 7d 26 72 68 61 73 68 3d iv?url={}&rhash=
1600: 7b 7d 5c 22 3e 20 3c 2f 61 3e 7b 30 7d 22 2c 20 {}\"> </a>{0}",
1610: 26 70 6f 73 74 5f 75 72 6c 2c 20 68 61 73 68 29 &post_url, hash)
1620: 29 2c 0a 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d ),........None =
1630: 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a > telegram_bot::
1640: 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 SendMessage::new
1650: 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f (destination, fo
1660: 72 6d 61 74 21 28 22 7b 7d 22 2c 20 70 6f 73 74 rmat!("{}", post
1670: 5f 75 72 6c 29 29 2c 0a 09 09 09 09 09 09 7d 2e _url)),.......}.
1680: 70 61 72 73 65 5f 6d 6f 64 65 28 74 65 6c 65 67 parse_mode(teleg
1690: 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a ram_bot::types::
16a0: 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 ParseMode::Html)
16b0: 29 2e 61 77 61 69 74 0a 09 09 09 09 09 09 2e 63 ).await........c
16c0: 6f 6e 74 65 78 74 28 22 43 61 6e 27 74 20 70 6f ontext("Can't po
16d0: 73 74 20 6d 65 73 73 61 67 65 3a 22 29 3f 3b 0a st message:")?;.
16e0: 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 .....sqlx::query
16f0: 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 ("insert into rs
1700: 73 74 67 5f 70 6f 73 74 20 28 73 6f 75 72 63 65 stg_post (source
1710: 5f 69 64 2c 20 70 6f 73 74 65 64 2c 20 75 72 6c _id, posted, url
1720: 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 ) values ($1, $2
1730: 2c 20 24 33 29 3b 22 29 0a 09 09 09 09 09 09 2e , $3);")........
1740: 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 09 09 bind(*id).......
1750: 2e 62 69 6e 64 28 64 61 74 65 29 0a 09 09 09 09 .bind(date).....
1760: 09 09 2e 62 69 6e 64 28 26 2a 70 6f 73 74 5f 75 ...bind(&*post_u
1770: 72 6c 29 0a 09 09 09 09 09 09 2e 65 78 65 63 75 rl)........execu
1780: 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 te(&mut conn).aw
1790: 61 69 74 0a 09 09 09 09 09 09 2e 77 69 74 68 5f ait........with_
17a0: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
17b0: 74 21 28 22 52 65 63 6f 72 64 20 70 6f 73 74 3a t!("Record post:
17c0: 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 \n{:?}", &conn))
17d0: 3f 3b 0a 09 09 09 09 09 64 72 6f 70 28 63 6f 6e ?;......drop(con
17e0: 6e 29 3b 0a 09 09 09 09 09 74 61 73 6b 3a 3a 73 n);......task::s
17f0: 6c 65 65 70 28 73 74 64 3a 3a 74 69 6d 65 3a 3a leep(std::time::
1800: 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 34 2c Duration::new(4,
1810: 20 30 29 29 2e 61 77 61 69 74 3b 0a 09 09 09 09 0)).await;.....
1820: 7d 3b 0a 09 09 09 09 70 6f 73 74 65 64 20 2b 3d };.....posted +=
1830: 20 31 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 6f 73 1;....};....pos
1840: 74 73 2e 63 6c 65 61 72 28 29 3b 0a 09 09 7d 3b ts.clear();...};
1850: 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
1860: 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 = self.pool.acqu
1870: 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e ire().await.....
1880: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
1890: 66 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 format!("Update
18a0: 73 63 72 61 70 65 20 66 65 74 63 68 20 63 6f 6e scrape fetch con
18b0: 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 n:\n{:?}", &self
18c0: 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 73 71 6c 78 .pool))?;...sqlx
18d0: 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 20 ::query("update
18e0: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 rsstg_source set
18f0: 20 6c 61 73 74 5f 73 63 72 61 70 65 20 3d 20 6e last_scrape = n
1900: 6f 77 28 29 20 77 68 65 72 65 20 73 6f 75 72 63 ow() where sourc
1910: 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a 09 09 09 e_id = $1;")....
1920: 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 2e 65 .bind(*id).....e
1930: 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e xecute(&mut conn
1940: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
1950: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
1960: 61 74 21 28 22 55 70 64 61 74 65 20 73 63 72 61 at!("Update scra
1970: 70 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e pe:\n{:?}", &con
1980: 6e 29 29 3f 3b 0a 09 09 4f 6b 28 66 6f 72 6d 61 n))?;...Ok(forma
1990: 74 21 28 22 50 6f 73 74 65 64 3a 20 7b 7d 22 2c t!("Posted: {}",
19a0: 20 26 70 6f 73 74 65 64 29 2e 69 6e 74 6f 28 29 &posted).into()
19b0: 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 )..}...pub async
19c0: 20 66 6e 20 64 65 6c 65 74 65 3c 53 3e 28 26 73 fn delete<S>(&s
19d0: 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 elf, source_id:
19e0: 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 &i32, owner: S)
19f0: 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 3c 27 5f -> Result<Cow<'_
1a00: 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 65 20 53 , str>>..where S
1a10: 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 : Into<i64> {...
1a20: 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 let owner = owne
1a30: 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 r.into();....let
1a40: 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 mut conn = self
1a50: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
1a60: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
1a70: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1a80: 21 28 22 44 65 6c 65 74 65 20 66 65 74 63 68 20 !("Delete fetch
1a90: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
1aa0: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d elf.pool))?;...m
1ab0: 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 atch sqlx::query
1ac0: 28 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 ("delete from rs
1ad0: 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 stg_source where
1ae0: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 source_id = $1
1af0: 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 3b 22 and owner = $2;"
1b00: 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 ).....bind(sourc
1b10: 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f e_id).....bind(o
1b20: 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 wner).....execut
1b30: 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 e(&mut conn).awa
1b40: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
1b50: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1b60: 44 65 6c 65 74 65 20 73 6f 75 72 63 65 20 72 75 Delete source ru
1b70: 6c 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c le:\n{:?}", &sel
1b80: 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f f.pool))?.....ro
1b90: 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a ws_affected() {.
1ba0: 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f ...0 => { Ok("No
1bb0: 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e data found foun
1bc0: 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 d.".into()) },..
1bd0: 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d ..x => { Ok(form
1be0: 61 74 21 28 22 7b 7d 20 73 6f 75 72 63 65 73 20 at!("{} sources
1bf0: 72 65 6d 6f 76 65 64 2e 22 2c 20 78 29 2e 69 6e removed.", x).in
1c00: 74 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a to()) },...}..}.
1c10: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 63 ..pub async fn c
1c20: 6c 65 61 6e 3c 53 3e 28 26 73 65 6c 66 2c 20 73 lean<S>(&self, s
1c30: 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 ource_id: &i32,
1c40: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
1c50: 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e ult<Cow<'_, str>
1c60: 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f >..where S: Into
1c70: 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 <i64> {...let ow
1c80: 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f ner = owner.into
1c90: 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 ();....let mut c
1ca0: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e onn = self.pool.
1cb0: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a acquire().await.
1cc0: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
1cd0: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 6c 65 (|| format!("Cle
1ce0: 61 6e 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e an fetch conn:\n
1cf0: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
1d00: 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 l))?;...match sq
1d10: 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 6c 65 74 lx::query("delet
1d20: 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 6f 73 e from rsstg_pos
1d30: 74 20 70 20 75 73 69 6e 67 20 72 73 73 74 67 5f t p using rsstg_
1d40: 73 6f 75 72 63 65 20 73 20 77 68 65 72 65 20 70 source s where p
1d50: 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 .source_id = $1
1d60: 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 20 61 and owner = $2 a
1d70: 6e 64 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d nd p.source_id =
1d80: 20 73 2e 73 6f 75 72 63 65 5f 69 64 3b 22 29 0a s.source_id;").
1d90: 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f ....bind(source_
1da0: 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e id).....bind(own
1db0: 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 er).....execute(
1dc0: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
1dd0: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
1de0: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 6c t(|| format!("Cl
1df0: 65 61 6e 20 73 65 65 6e 20 70 6f 73 74 73 3a 5c ean seen posts:\
1e00: 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f n{:?}", &self.po
1e10: 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 ol))?.....rows_a
1e20: 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 30 ffected() {....0
1e30: 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 61 74 => { Ok("No dat
1e40: 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 2e 22 2e a found found.".
1e50: 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 09 78 20 into()) },....x
1e60: 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 => { Ok(format!(
1e70: 22 7b 7d 20 70 6f 73 74 73 20 70 75 72 67 65 64 "{} posts purged
1e80: 2e 22 2c 20 78 29 2e 69 6e 74 6f 28 29 29 20 7d .", x).into()) }
1e90: 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 ,...}..}...pub a
1ea0: 73 79 6e 63 20 66 6e 20 65 6e 61 62 6c 65 3c 53 sync fn enable<S
1eb0: 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f >(&self, source_
1ec0: 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a id: &i32, owner:
1ed0: 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 S) -> Result<&s
1ee0: 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e tr>..where S: In
1ef0: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 to<i64> {...let
1f00: 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e owner = owner.in
1f10: 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 to();....let mut
1f20: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
1f30: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
1f40: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
1f50: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 45 xt(|| format!("E
1f60: 6e 61 62 6c 65 20 66 65 74 63 68 20 63 6f 6e 6e nable fetch conn
1f70: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1f80: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 pool))?;...match
1f90: 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 sqlx::query("up
1fa0: 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 date rsstg_sourc
1fb0: 65 20 73 65 74 20 65 6e 61 62 6c 65 64 20 3d 20 e set enabled =
1fc0: 74 72 75 65 20 77 68 65 72 65 20 73 6f 75 72 63 true where sourc
1fd0: 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 e_id = $1 and ow
1fe0: 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 ner = $2").....b
1ff0: 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 ind(source_id)..
2000: 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 ...bind(owner)..
2010: 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 ...execute(&mut
2020: 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e conn).await.....
2030: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
2040: 66 6f 72 6d 61 74 21 28 22 45 6e 61 62 6c 65 20 format!("Enable
2050: 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 source:\n{:?}",
2060: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 &self.pool))?...
2070: 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 ..rows_affected(
2080: 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b ) {....1 => { Ok
2090: 28 22 53 6f 75 72 63 65 20 65 6e 61 62 6c 65 64 ("Source enabled
20a0: 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b .") },....0 => {
20b0: 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 Ok("Source not
20c0: 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f found.") },...._
20d0: 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 => { Err(anyhow
20e0: 21 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f !("Database erro
20f0: 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a r.")) },...}..}.
2100: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 64 ..pub async fn d
2110: 69 73 61 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c isable<S>(&self,
2120: 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 source_id: &i32
2130: 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 , owner: S) -> R
2140: 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 esult<&str>..whe
2150: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
2160: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
2170: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 owner.into();...
2180: 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 .let mut conn =
2190: 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 self.pool.acquir
21a0: 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 e().await.....wi
21b0: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
21c0: 72 6d 61 74 21 28 22 44 69 73 61 62 6c 65 20 66 rmat!("Disable f
21d0: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
21e0: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
21f0: 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a ;...match sqlx::
2200: 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 73 query("update rs
2210: 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 stg_source set e
2220: 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 65 20 77 nabled = false w
2230: 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d here source_id =
2240: 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 $1 and owner =
2250: 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f $2").....bind(so
2260: 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e urce_id).....bin
2270: 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 d(owner).....exe
2280: 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e cute(&mut conn).
2290: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
22a0: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
22b0: 21 28 22 44 69 73 61 62 6c 65 20 73 6f 75 72 63 !("Disable sourc
22c0: 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 e:\n{:?}", &self
22d0: 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 .pool))?.....row
22e0: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
22f0: 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 ..1 => { Ok("Sou
2300: 72 63 65 20 64 69 73 61 62 6c 65 64 2e 22 29 20 rce disabled.")
2310: 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 },....0 => { Ok(
2320: 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e "Source not foun
2330: 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 d.") },...._ =>
2340: 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 { Err(anyhow!("D
2350: 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 atabase error.")
2360: 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 ) },...}..}...pu
2370: 62 20 61 73 79 6e 63 20 66 6e 20 75 70 64 61 74 b async fn updat
2380: 65 3c 53 3e 28 26 73 65 6c 66 2c 20 75 70 64 61 e<S>(&self, upda
2390: 74 65 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c te: Option<i32>,
23a0: 20 63 68 61 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 channel: &str,
23b0: 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c channel_id: i64,
23c0: 20 75 72 6c 3a 20 26 73 74 72 2c 20 69 76 5f 68 url: &str, iv_h
23d0: 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 ash: Option<&str
23e0: 3e 2c 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f >, url_re: Optio
23f0: 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 72 3a 20 n<&str>, owner:
2400: 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 S) -> Result<&st
2410: 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 r>..where S: Int
2420: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f o<i64> {...let o
2430: 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 wner = owner.int
2440: 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 o();....let mut
2450: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
2460: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
2470: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
2480: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 t(|| format!("Up
2490: 64 61 74 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a date fetch conn:
24a0: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
24b0: 6f 6f 6c 29 29 3f 3b 0a 0a 09 09 6d 61 74 63 68 ool))?;....match
24c0: 20 6d 61 74 63 68 20 75 70 64 61 74 65 20 7b 0a match update {.
24d0: 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 ....Some(id) =>
24e0: 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 {......sqlx::que
24f0: 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 67 ry("update rsstg
2500: 5f 73 6f 75 72 63 65 20 73 65 74 20 63 68 61 6e _source set chan
2510: 6e 65 6c 5f 69 64 20 3d 20 24 32 2c 20 75 72 6c nel_id = $2, url
2520: 20 3d 20 24 33 2c 20 69 76 5f 68 61 73 68 20 3d = $3, iv_hash =
2530: 20 24 34 2c 20 6f 77 6e 65 72 20 3d 20 24 35 2c $4, owner = $5,
2540: 20 63 68 61 6e 6e 65 6c 20 3d 20 24 36 2c 20 75 channel = $6, u
2550: 72 6c 5f 72 65 20 3d 20 24 37 20 77 68 65 72 65 rl_re = $7 where
2560: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 22 source_id = $1"
2570: 29 2e 62 69 6e 64 28 69 64 29 0a 09 09 09 09 7d ).bind(id).....}
2580: 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a ,.....None => {.
2590: 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 .....sqlx::query
25a0: 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 ("insert into rs
25b0: 73 74 67 5f 73 6f 75 72 63 65 20 28 63 68 61 6e stg_source (chan
25c0: 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f nel_id, url, iv_
25d0: 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 hash, owner, cha
25e0: 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 20 76 61 nnel, url_re) va
25f0: 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 24 33 lues ($1, $2, $3
2600: 2c 20 24 34 2c 20 24 35 2c 20 24 36 29 22 29 0a , $4, $5, $6)").
2610: 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 09 2e ....},....}.....
2620: 62 69 6e 64 28 63 68 61 6e 6e 65 6c 5f 69 64 29 bind(channel_id)
2630: 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c 29 0a 09 .....bind(url)..
2640: 09 09 2e 62 69 6e 64 28 69 76 5f 68 61 73 68 29 ...bind(iv_hash)
2650: 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 .....bind(owner)
2660: 0a 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 .....bind(channe
2670: 6c 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c 5f l).....bind(url_
2680: 72 65 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 re).....execute(
2690: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
26a0: 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 4f {....Ok(_) => O
26b0: 6b 28 6d 61 74 63 68 20 75 70 64 61 74 65 20 7b k(match update {
26c0: 0a 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 .....Some(_) =>
26d0: 22 43 68 61 6e 6e 65 6c 20 75 70 64 61 74 65 64 "Channel updated
26e0: 2e 22 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 .",.....None =>
26f0: 22 43 68 61 6e 6e 65 6c 20 61 64 64 65 64 2e 22 "Channel added."
2700: 2c 0a 09 09 09 7d 29 2c 0a 09 09 09 45 72 72 28 ,....}),....Err(
2710: 73 71 6c 78 3a 3a 45 72 72 6f 72 3a 3a 44 61 74 sqlx::Error::Dat
2720: 61 62 61 73 65 28 65 72 72 29 29 20 3d 3e 20 7b abase(err)) => {
2730: 0a 09 09 09 09 6d 61 74 63 68 20 65 72 72 2e 64 .....match err.d
2740: 6f 77 6e 63 61 73 74 3a 3a 3c 73 71 6c 78 3a 3a owncast::<sqlx::
2750: 70 6f 73 74 67 72 65 73 3a 3a 50 67 44 61 74 61 postgres::PgData
2760: 62 61 73 65 45 72 72 6f 72 3e 28 29 2e 72 6f 75 baseError>().rou
2770: 74 69 6e 65 28 29 20 7b 0a 09 09 09 09 09 53 6f tine() {......So
2780: 6d 65 28 22 5f 62 74 5f 63 68 65 63 6b 5f 75 6e me("_bt_check_un
2790: 69 71 75 65 22 2c 20 29 20 3d 3e 20 7b 0a 09 09 ique", ) => {...
27a0: 09 09 09 09 4f 6b 28 22 44 75 70 6c 69 63 61 74 ....Ok("Duplicat
27b0: 65 20 6b 65 79 2e 22 29 0a 09 09 09 09 09 7d 2c e key.")......},
27c0: 0a 09 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e ......Some(_) =>
27d0: 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 44 61 74 {.......Ok("Dat
27e0: 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 0a 09 abase error.")..
27f0: 09 09 09 09 7d 2c 0a 09 09 09 09 09 4e 6f 6e 65 ....},......None
2800: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 => {.......Ok("
2810: 4e 6f 20 64 61 74 61 62 61 73 65 20 65 72 72 6f No database erro
2820: 72 20 65 78 74 72 61 63 74 65 64 2e 22 29 0a 09 r extracted.")..
2830: 09 09 09 09 7d 2c 0a 09 09 09 09 7d 0a 09 09 09 ....},.....}....
2840: 7d 2c 0a 09 09 09 45 72 72 28 65 72 72 29 20 3d },....Err(err) =
2850: 3e 20 7b 0a 09 09 09 09 62 61 69 6c 21 28 22 53 > {.....bail!("S
2860: 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 72 orry, unknown er
2870: 72 6f 72 3a 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 ror:\n{:#?}\n",
2880: 65 72 72 29 3b 0a 09 09 09 7d 2c 0a 09 09 7d 0a err);....},...}.
2890: 09 7d 0a 0a 09 61 73 79 6e 63 20 66 6e 20 61 75 .}...async fn au
28a0: 74 6f 66 65 74 63 68 28 26 73 65 6c 66 29 20 2d tofetch(&self) -
28b0: 3e 20 52 65 73 75 6c 74 3c 73 74 64 3a 3a 74 69 > Result<std::ti
28c0: 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3e 20 7b 0a me::Duration> {.
28d0: 09 09 6c 65 74 20 6d 75 74 20 64 65 6c 61 79 20 ..let mut delay
28e0: 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 72 61 74 69 = chrono::Durati
28f0: 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 31 29 3b 0a on::minutes(1);.
2900: 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d ..let mut conn =
2910: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
2920: 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 re().await.....w
2930: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
2940: 6f 72 6d 61 74 21 28 22 41 75 74 6f 66 65 74 63 ormat!("Autofetc
2950: 68 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b h fetch conn:\n{
2960: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
2970: 29 29 3f 3b 0a 09 09 6c 65 74 20 6e 6f 77 20 3d ))?;...let now =
2980: 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a chrono::Local::
2990: 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 now();...let mut
29a0: 20 71 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a 71 queue = sqlx::q
29b0: 75 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 uery("select sou
29c0: 72 63 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 rce_id, next_fet
29d0: 63 68 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 ch, owner from r
29e0: 73 73 74 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 sstg_order natur
29f0: 61 6c 20 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 al left join rss
2a00: 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 tg_source where
2a10: 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 next_fetch < now
2a20: 28 29 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 () + interval '1
2a30: 20 6d 69 6e 75 74 65 27 3b 22 29 0a 09 09 09 2e minute';").....
2a40: 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 fetch_all(&mut c
2a50: 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 66 onn).await?;...f
2a60: 6f 72 20 72 6f 77 20 69 6e 20 71 75 65 75 65 2e or row in queue.
2a70: 69 74 65 72 28 29 20 7b 0a 09 09 09 6c 65 74 20 iter() {....let
2a80: 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 20 3d source_id: i32 =
2a90: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 6f row.try_get("so
2aa0: 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 6c urce_id")?;....l
2ab0: 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 et owner: i64 =
2ac0: 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e row.try_get("own
2ad0: 65 72 22 29 3f 3b 0a 09 09 09 6c 65 74 20 6e 65 er")?;....let ne
2ae0: 78 74 5f 66 65 74 63 68 3a 20 44 61 74 65 54 69 xt_fetch: DateTi
2af0: 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c me<chrono::Local
2b00: 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 > = row.try_get(
2b10: 22 6e 65 78 74 5f 66 65 74 63 68 22 29 3f 3b 0a "next_fetch")?;.
2b20: 09 09 09 69 66 20 6e 65 78 74 5f 66 65 74 63 68 ...if next_fetch
2b30: 20 3c 20 6e 6f 77 20 7b 0a 09 09 09 09 6c 65 74 < now {.....let
2b40: 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 65 20 7b 0a clone = Core {.
2b50: 09 09 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a .....owner_chat:
2b60: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 telegram_bot::U
2b70: 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 serId::new(owner
2b80: 29 2c 0a 09 09 09 09 09 2e 2e 73 65 6c 66 2e 63 ),........self.c
2b90: 6c 6f 6e 65 28 29 0a 09 09 09 09 7d 3b 0a 09 09 lone().....};...
2ba0: 09 09 74 61 73 6b 3a 3a 73 70 61 77 6e 28 61 73 ..task::spawn(as
2bb0: 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 09 ync move {......
2bc0: 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 if let Err(err)
2bd0: 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b 28 26 73 = clone.check(&s
2be0: 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 2c ource_id, owner,
2bf0: 20 74 72 75 65 29 2e 61 77 61 69 74 20 7b 0a 09 true).await {..
2c00: 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 .....if let Err(
2c10: 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e err) = clone.sen
2c20: 64 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 d(&format!("š
2c30: 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f {:?}", err), No
2c40: 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 74 20 ne, None).await
2c50: 7b 0a 09 09 09 09 09 09 09 65 70 72 69 6e 74 6c {........eprintl
2c60: 6e 21 28 22 43 68 65 63 6b 20 65 72 72 6f 72 3a n!("Check error:
2c70: 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 09 {}", err);.....
2c80: 09 09 7d 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 ..};......};....
2c90: 09 7d 29 3b 0a 09 09 09 7d 20 65 6c 73 65 20 69 .});....} else i
2ca0: 66 20 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 6e f next_fetch - n
2cb0: 6f 77 20 3c 20 64 65 6c 61 79 20 7b 0a 09 09 09 ow < delay {....
2cc0: 09 64 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 .delay = next_fe
2cd0: 74 63 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 7d 0a tch - now;....}.
2ce0: 09 09 7d 3b 0a 09 09 71 75 65 75 65 2e 63 6c 65 ..};...queue.cle
2cf0: 61 72 28 29 3b 0a 09 09 4f 6b 28 64 65 6c 61 79 ar();...Ok(delay
2d00: 2e 74 6f 5f 73 74 64 28 29 3f 29 0a 09 7d 0a 0a .to_std()?)..}..
2d10: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 6c 69 .pub async fn li
2d20: 73 74 3c 53 3e 28 26 73 65 6c 66 2c 20 6f 77 6e st<S>(&self, own
2d30: 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 er: S) -> Result
2d40: 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 72 65 20 <String>..where
2d50: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
2d60: 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e .let owner = own
2d70: 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 er.into();....le
2d80: 74 20 6d 75 74 20 72 65 70 6c 79 3a 20 56 65 63 t mut reply: Vec
2d90: 3c 43 6f 77 3c 73 74 72 3e 3e 20 3d 20 76 65 63 <Cow<str>> = vec
2da0: 21 5b 5d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 ![];...let mut c
2db0: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e onn = self.pool.
2dc0: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a acquire().await.
2dd0: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
2de0: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 4c 69 73 (|| format!("Lis
2df0: 74 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b t fetch conn:\n{
2e00: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
2e10: 29 29 3f 3b 0a 09 09 72 65 70 6c 79 2e 70 75 73 ))?;...reply.pus
2e20: 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 2e 69 6e h("Channels:".in
2e30: 74 6f 28 29 29 3b 0a 09 09 6c 65 74 20 72 6f 77 to());...let row
2e40: 73 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 s = sqlx::query(
2e50: 22 73 65 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 "select source_i
2e60: 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 65 6e 61 62 d, channel, enab
2e70: 6c 65 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 led, url, iv_has
2e80: 68 2c 20 75 72 6c 5f 72 65 20 66 72 6f 6d 20 72 h, url_re from r
2e90: 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 sstg_source wher
2ea0: 65 20 6f 77 6e 65 72 20 3d 20 24 31 20 6f 72 64 e owner = $1 ord
2eb0: 65 72 20 62 79 20 73 6f 75 72 63 65 5f 69 64 22 er by source_id"
2ec0: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
2ed0: 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 ).....fetch_all(
2ee0: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
2ef0: 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 ?;...for row in
2f00: 72 6f 77 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 rows.iter() {...
2f10: 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 .let source_id:
2f20: 69 33 32 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 i32 = row.try_ge
2f30: 74 28 22 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b t("source_id")?;
2f40: 0a 09 09 09 6c 65 74 20 75 73 65 72 6e 61 6d 65 ....let username
2f50: 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 : &str = row.try
2f60: 5f 67 65 74 28 22 63 68 61 6e 6e 65 6c 22 29 3f _get("channel")?
2f70: 3b 0a 09 09 09 6c 65 74 20 65 6e 61 62 6c 65 64 ;....let enabled
2f80: 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 : bool = row.try
2f90: 5f 67 65 74 28 22 65 6e 61 62 6c 65 64 22 29 3f _get("enabled")?
2fa0: 3b 0a 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 73 ;....let url: &s
2fb0: 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 tr = row.try_get
2fc0: 28 22 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 ("url")?;....let
2fd0: 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e iv_hash: Option
2fe0: 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 <&str> = row.try
2ff0: 5f 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f _get("iv_hash")?
3000: 3b 0a 09 09 09 6c 65 74 20 75 72 6c 5f 72 65 3a ;....let url_re:
3010: 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 Option<&str> =
3020: 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c row.try_get("url
3030: 5f 72 65 22 29 3f 3b 0a 09 09 09 72 65 70 6c 79 _re")?;....reply
3040: 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 5c .push(format!("\
3050: 6e 5c 5c 23 ef b8 8f e2 83 a3 20 7b 7d 20 5c 5c n\\#ļøā£ {} \\
3060: 2a ef b8 8f e2 83 a3 20 60 7b 7d 60 20 7b 7d 5c *ļøā£ `{}` {}\
3070: 6e f0 9f 94 97 20 60 7b 7d 60 22 2c 20 73 6f 75 nš `{}`", sou
3080: 72 63 65 5f 69 64 2c 20 75 73 65 72 6e 61 6d 65 rce_id, username
3090: 2c 20 20 0a 09 09 09 09 6d 61 74 63 68 20 65 6e , .....match en
30a0: 61 62 6c 65 64 20 7b 0a 09 09 09 09 09 74 72 75 abled {......tru
30b0: 65 20 20 3d 3e 20 22 f0 9f 94 84 20 65 6e 61 62 e => "š enab
30c0: 6c 65 64 22 2c 0a 09 09 09 09 09 66 61 6c 73 65 led",......false
30d0: 20 3d 3e 20 22 e2 9b 94 20 64 69 73 61 62 6c 65 => "ā disable
30e0: 64 22 2c 0a 09 09 09 09 7d 2c 20 75 72 6c 29 2e d",.....}, url).
30f0: 69 6e 74 6f 28 29 29 3b 0a 09 09 09 69 66 20 6c into());....if l
3100: 65 74 20 53 6f 6d 65 28 68 61 73 68 29 20 3d 20 et Some(hash) =
3110: 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 72 65 iv_hash {.....re
3120: 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 ply.push(format!
3130: 28 22 49 56 3a 20 60 7b 7d 60 22 2c 20 68 61 73 ("IV: `{}`", has
3140: 68 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 7d h).into());....}
3150: 0a 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 ....if let Some(
3160: 72 65 29 20 3d 20 75 72 6c 5f 72 65 20 7b 0a 09 re) = url_re {..
3170: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f ...reply.push(fo
3180: 72 6d 61 74 21 28 22 52 45 3a 20 60 7b 7d 60 22 rmat!("RE: `{}`"
3190: 2c 20 72 65 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 , re).into());..
31a0: 09 09 7d 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 65 ..}...};...Ok(re
31b0: 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 0a ply.join("\n")).
31c0: 09 7d 0a 7d 0a .}.}.