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 74 6f 6d 5f 73 79 6e 64 69 63 61 74 69 6f atom_syndicatio
0040: 6e 3b 0a 75 73 65 20 63 68 72 6f 6e 6f 3a 3a 44 n;.use chrono::D
0050: 61 74 65 54 69 6d 65 3b 0a 75 73 65 20 63 6f 6e ateTime;.use con
0060: 66 69 67 3b 0a 75 73 65 20 72 65 67 65 78 3a 3a fig;.use regex::
0070: 52 65 67 65 78 3b 0a 75 73 65 20 72 65 71 77 65 Regex;.use reqwe
0080: 73 74 3b 0a 75 73 65 20 73 71 6c 78 3a 3a 7b 0a st;.use sqlx::{.
0090: 09 70 6f 73 74 67 72 65 73 3a 3a 50 67 50 6f 6f .postgres::PgPoo
00a0: 6c 4f 70 74 69 6f 6e 73 2c 0a 09 52 6f 77 2c 0a lOptions,..Row,.
00b0: 7d 3b 0a 75 73 65 20 72 73 73 3b 0a 75 73 65 20 };.use rss;.use
00c0: 73 74 64 3a 3a 7b 0a 09 63 6f 6c 6c 65 63 74 69 std::{..collecti
00d0: 6f 6e 73 3a 3a 7b 0a 09 09 42 54 72 65 65 4d 61 ons::{...BTreeMa
00e0: 70 2c 0a 09 09 48 61 73 68 53 65 74 2c 0a 09 7d p,...HashSet,..}
00f0: 2c 0a 09 73 79 6e 63 3a 3a 7b 41 72 63 2c 20 4d ,..sync::{Arc, M
0100: 75 74 65 78 7d 2c 0a 7d 3b 0a 75 73 65 20 74 65 utex},.};.use te
0110: 6c 65 67 72 61 6d 5f 62 6f 74 3b 0a 0a 23 5b 64 legram_bot;..#[d
0120: 65 72 69 76 65 28 43 6c 6f 6e 65 29 5d 0a 70 75 erive(Clone)].pu
0130: 62 20 73 74 72 75 63 74 20 43 6f 72 65 20 7b 0a b struct Core {.
0140: 09 6f 77 6e 65 72 3a 20 69 36 34 2c 0a 09 61 70 .owner: i64,..ap
0150: 69 5f 6b 65 79 3a 20 53 74 72 69 6e 67 2c 0a 09 i_key: String,..
0160: 6f 77 6e 65 72 5f 63 68 61 74 3a 20 74 65 6c 65 owner_chat: tele
0170: 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 gram_bot::UserId
0180: 2c 0a 09 70 75 62 20 74 67 3a 20 74 65 6c 65 67 ,..pub tg: teleg
0190: 72 61 6d 5f 62 6f 74 3a 3a 41 70 69 2c 0a 09 70 ram_bot::Api,..p
01a0: 75 62 20 6d 79 3a 20 74 65 6c 65 67 72 61 6d 5f ub my: telegram_
01b0: 62 6f 74 3a 3a 55 73 65 72 2c 0a 09 70 6f 6f 6c bot::User,..pool
01c0: 3a 20 73 71 6c 78 3a 3a 50 6f 6f 6c 3c 73 71 6c : sqlx::Pool<sql
01d0: 78 3a 3a 50 6f 73 74 67 72 65 73 3e 2c 0a 09 73 x::Postgres>,..s
01e0: 6f 75 72 63 65 73 3a 20 41 72 63 3c 4d 75 74 65 ources: Arc<Mute
01f0: 78 3c 48 61 73 68 53 65 74 3c 41 72 63 3c 69 33 x<HashSet<Arc<i3
0200: 32 3e 3e 3e 3e 2c 0a 7d 0a 0a 69 6d 70 6c 20 43 2>>>>,.}..impl C
0210: 6f 72 65 20 7b 0a 09 70 75 62 20 61 73 79 6e 63 ore {..pub async
0220: 20 66 6e 20 6e 65 77 28 73 65 74 74 69 6e 67 73 fn new(settings
0230: 3a 20 63 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 : config::Config
0240: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 72 65 ) -> Result<Core
0250: 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 > {...let owner
0260: 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 5f 69 = settings.get_i
0270: 6e 74 28 22 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 nt("owner")?;...
0280: 6c 65 74 20 61 70 69 5f 6b 65 79 20 3d 20 73 65 let api_key = se
0290: 74 74 69 6e 67 73 2e 67 65 74 5f 73 74 72 28 22 ttings.get_str("
02a0: 61 70 69 5f 6b 65 79 22 29 3f 3b 0a 09 09 6c 65 api_key")?;...le
02b0: 74 20 74 67 20 3d 20 74 65 6c 65 67 72 61 6d 5f t tg = telegram_
02c0: 62 6f 74 3a 3a 41 70 69 3a 3a 6e 65 77 28 26 61 bot::Api::new(&a
02d0: 70 69 5f 6b 65 79 29 3b 0a 09 09 6c 65 74 20 63 pi_key);...let c
02e0: 6f 72 65 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 ore = Core {....
02f0: 6f 77 6e 65 72 3a 20 6f 77 6e 65 72 2c 0a 09 09 owner: owner,...
0300: 09 61 70 69 5f 6b 65 79 3a 20 61 70 69 5f 6b 65 .api_key: api_ke
0310: 79 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 6d 79 y.clone(),....my
0320: 3a 20 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 72 : tg.send(telegr
0330: 61 6d 5f 62 6f 74 3a 3a 47 65 74 4d 65 29 2e 61 am_bot::GetMe).a
0340: 77 61 69 74 3f 2c 0a 09 09 09 74 67 3a 20 74 67 wait?,....tg: tg
0350: 2c 0a 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a ,....owner_chat:
0360: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 telegram_bot::U
0370: 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 serId::new(owner
0380: 29 2c 0a 09 09 09 70 6f 6f 6c 3a 20 50 67 50 6f ),....pool: PgPo
0390: 6f 6c 4f 70 74 69 6f 6e 73 3a 3a 6e 65 77 28 29 olOptions::new()
03a0: 0a 09 09 09 09 2e 6d 61 78 5f 63 6f 6e 6e 65 63 ......max_connec
03b0: 74 69 6f 6e 73 28 35 29 0a 09 09 09 09 2e 63 6f tions(5)......co
03c0: 6e 6e 65 63 74 5f 74 69 6d 65 6f 75 74 28 73 74 nnect_timeout(st
03d0: 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f d::time::Duratio
03e0: 6e 3a 3a 6e 65 77 28 33 30 30 2c 20 30 29 29 0a n::new(300, 0)).
03f0: 09 09 09 09 2e 69 64 6c 65 5f 74 69 6d 65 6f 75 .....idle_timeou
0400: 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 t(std::time::Dur
0410: 61 74 69 6f 6e 3a 3a 6e 65 77 28 36 30 2c 20 30 ation::new(60, 0
0420: 29 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 74 5f ))......connect_
0430: 6c 61 7a 79 28 26 73 65 74 74 69 6e 67 73 2e 67 lazy(&settings.g
0440: 65 74 5f 73 74 72 28 22 70 67 22 29 3f 29 3f 2c et_str("pg")?)?,
0450: 0a 09 09 09 73 6f 75 72 63 65 73 3a 20 41 72 63 ....sources: Arc
0460: 3a 3a 6e 65 77 28 4d 75 74 65 78 3a 3a 6e 65 77 ::new(Mutex::new
0470: 28 48 61 73 68 53 65 74 3a 3a 6e 65 77 28 29 29 (HashSet::new())
0480: 29 2c 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 63 6c ),...};...let cl
0490: 6f 6e 65 20 3d 20 63 6f 72 65 2e 63 6c 6f 6e 65 one = core.clone
04a0: 28 29 3b 0a 09 09 74 6f 6b 69 6f 3a 3a 73 70 61 ();...tokio::spa
04b0: 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a wn(async move {.
04c0: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
04d0: 72 29 20 3d 20 26 63 6c 6f 6e 65 2e 61 75 74 6f r) = &clone.auto
04e0: 66 65 74 63 68 28 29 2e 61 77 61 69 74 20 7b 0a fetch().await {.
04f0: 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 ....if let Err(e
0500: 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 rr) = clone.send
0510: 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 (&format!("š
0520: 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e {:?}", err), Non
0530: 65 29 20 7b 0a 09 09 09 09 09 65 70 72 69 6e 74 e) {......eprint
0540: 6c 6e 21 28 22 41 75 74 6f 66 65 74 63 68 20 65 ln!("Autofetch e
0550: 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b rror: {}", err);
0560: 0a 09 09 09 09 7d 3b 0a 09 09 09 7d 0a 09 09 7d .....};....}...}
0570: 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 0a 09 7d );...Ok(core)..}
0580: 0a 0a 09 70 75 62 20 66 6e 20 73 74 72 65 61 6d ...pub fn stream
0590: 28 26 73 65 6c 66 29 20 2d 3e 20 74 65 6c 65 67 (&self) -> teleg
05a0: 72 61 6d 5f 62 6f 74 3a 3a 55 70 64 61 74 65 73 ram_bot::Updates
05b0: 53 74 72 65 61 6d 20 7b 0a 09 09 73 65 6c 66 2e Stream {...self.
05c0: 74 67 2e 73 74 72 65 61 6d 28 29 0a 09 7d 0a 0a tg.stream()..}..
05d0: 09 70 75 62 20 66 6e 20 73 65 6e 64 3c 53 3e 28 .pub fn send<S>(
05e0: 26 73 65 6c 66 2c 20 6d 73 67 3a 20 53 2c 20 74 &self, msg: S, t
05f0: 61 72 67 65 74 3a 20 4f 70 74 69 6f 6e 3c 74 65 arget: Option<te
0600: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 legram_bot::User
0610: 49 64 3e 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 Id>) -> Result<(
0620: 29 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 )>..where S: Int
0630: 6f 3c 53 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 o<String> {...le
0640: 74 20 6d 73 67 3a 20 53 74 72 69 6e 67 20 3d 20 t msg: String =
0650: 6d 73 67 2e 69 6e 74 6f 28 29 3b 0a 09 09 73 65 msg.into();...se
0660: 6c 66 2e 74 67 2e 73 70 61 77 6e 28 74 65 6c 65 lf.tg.spawn(tele
0670: 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 gram_bot::SendMe
0680: 73 73 61 67 65 3a 3a 6e 65 77 28 6d 61 74 63 68 ssage::new(match
0690: 20 74 61 72 67 65 74 20 7b 0a 09 09 09 53 6f 6d target {....Som
06a0: 65 28 75 73 65 72 29 20 3d 3e 20 75 73 65 72 2c e(user) => user,
06b0: 0a 09 09 09 4e 6f 6e 65 20 3d 3e 20 73 65 6c 66 ....None => self
06c0: 2e 6f 77 6e 65 72 5f 63 68 61 74 2c 0a 09 09 7d .owner_chat,...}
06d0: 2c 20 6d 73 67 2e 74 6f 5f 6f 77 6e 65 64 28 29 , msg.to_owned()
06e0: 29 29 3b 0a 09 09 4f 6b 28 28 29 29 0a 09 7d 0a ));...Ok(())..}.
06f0: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 63 ..pub async fn c
0700: 68 65 63 6b 3c 53 3e 28 26 73 65 6c 66 2c 20 69 heck<S>(&self, i
0710: 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 d: &i32, owner:
0720: 53 2c 20 72 65 61 6c 3a 20 62 6f 6f 6c 29 20 2d S, real: bool) -
0730: 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e > Result<String>
0740: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
0750: 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 i64> {...let mut
0760: 20 70 6f 73 74 65 64 3a 20 69 33 32 20 3d 20 30 posted: i32 = 0
0770: 3b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 ;...let owner: i
0780: 36 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 64 = owner.into(
0790: 29 3b 0a 09 09 6c 65 74 20 69 64 20 3d 20 7b 0a );...let id = {.
07a0: 09 09 09 6c 65 74 20 6d 75 74 20 73 65 74 20 3d ...let mut set =
07b0: 20 73 65 6c 66 2e 73 6f 75 72 63 65 73 2e 6c 6f self.sources.lo
07c0: 63 6b 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 ck().unwrap();..
07d0: 09 09 6d 61 74 63 68 20 73 65 74 2e 67 65 74 28 ..match set.get(
07e0: 69 64 29 20 7b 0a 09 09 09 09 53 6f 6d 65 28 69 id) {.....Some(i
07f0: 64 29 20 3d 3e 20 69 64 2e 63 6c 6f 6e 65 28 29 d) => id.clone()
0800: 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a ,.....None => {.
0810: 09 09 09 09 09 6c 65 74 20 69 64 20 3d 20 41 72 .....let id = Ar
0820: 63 3a 3a 6e 65 77 28 2a 69 64 29 3b 0a 09 09 09 c::new(*id);....
0830: 09 09 73 65 74 2e 69 6e 73 65 72 74 28 69 64 2e ..set.insert(id.
0840: 63 6c 6f 6e 65 28 29 29 3b 0a 09 09 09 09 09 69 clone());......i
0850: 64 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 7d 2c d.clone().....},
0860: 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 6c 65 74 ....}...};...let
0870: 20 63 6f 75 6e 74 20 3d 20 41 72 63 3a 3a 73 74 count = Arc::st
0880: 72 6f 6e 67 5f 63 6f 75 6e 74 28 26 69 64 29 3b rong_count(&id);
0890: 0a 09 09 69 66 20 63 6f 75 6e 74 20 3d 3d 20 32 ...if count == 2
08a0: 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 63 6f {....let mut co
08b0: 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 nn = self.pool.a
08c0: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 cquire().await..
08d0: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
08e0: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 51 75 65 (|| format!("Que
08f0: 72 79 20 71 75 65 75 65 20 66 65 74 63 68 20 63 ry queue fetch c
0900: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
0910: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 6c lf.pool))?;....l
0920: 65 74 20 72 6f 77 20 3d 20 73 71 6c 78 3a 3a 71 et row = sqlx::q
0930: 75 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 uery("select sou
0940: 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 5f rce_id, channel_
0950: 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 id, url, iv_hash
0960: 2c 20 6f 77 6e 65 72 2c 20 75 72 6c 5f 72 65 20 , owner, url_re
0970: 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 from rsstg_sourc
0980: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
0990: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
09a0: 20 3d 20 24 32 22 29 0a 09 09 09 09 2e 62 69 6e = $2")......bin
09b0: 64 28 2a 69 64 29 0a 09 09 09 09 2e 62 69 6e 64 d(*id)......bind
09c0: 28 6f 77 6e 65 72 29 0a 09 09 09 09 2e 66 65 74 (owner)......fet
09d0: 63 68 5f 6f 6e 65 28 26 6d 75 74 20 63 6f 6e 6e ch_one(&mut conn
09e0: 29 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 74 ).await......wit
09f0: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
0a00: 6d 61 74 21 28 22 51 75 65 72 79 20 73 6f 75 72 mat!("Query sour
0a10: 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c ce:\n{:?}", &sel
0a20: 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 64 72 f.pool))?;....dr
0a30: 6f 70 28 63 6f 6e 6e 29 3b 0a 09 09 09 6c 65 74 op(conn);....let
0a40: 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 channel_id: i64
0a50: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
0a60: 63 68 61 6e 6e 65 6c 5f 69 64 22 29 3f 3b 0a 09 channel_id")?;..
0a70: 09 09 6c 65 74 20 75 72 6c 3a 20 26 73 74 72 20 ..let url: &str
0a80: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 = row.try_get("u
0a90: 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 69 76 rl")?;....let iv
0aa0: 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 _hash: Option<&s
0ab0: 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 tr> = row.try_ge
0ac0: 74 28 22 69 76 5f 68 61 73 68 22 29 3f 3b 0a 09 t("iv_hash")?;..
0ad0: 09 09 6c 65 74 20 75 72 6c 5f 72 65 3a 20 4f 70 ..let url_re: Op
0ae0: 74 69 6f 6e 3c 72 65 67 65 78 3a 3a 52 65 67 65 tion<regex::Rege
0af0: 78 3e 20 3d 20 6d 61 74 63 68 20 72 6f 77 2e 74 x> = match row.t
0b00: 72 79 5f 67 65 74 28 22 75 72 6c 5f 72 65 22 29 ry_get("url_re")
0b10: 3f 20 7b 0a 09 09 09 09 53 6f 6d 65 28 78 29 20 ? {.....Some(x)
0b20: 3d 3e 20 53 6f 6d 65 28 52 65 67 65 78 3a 3a 6e => Some(Regex::n
0b30: 65 77 28 78 29 3f 29 2c 0a 09 09 09 09 4e 6f 6e ew(x)?),.....Non
0b40: 65 20 3d 3e 20 4e 6f 6e 65 2c 0a 09 09 09 7d 3b e => None,....};
0b50: 0a 09 09 09 6c 65 74 20 64 65 73 74 69 6e 61 74 ....let destinat
0b60: 69 6f 6e 20 3d 20 6d 61 74 63 68 20 72 65 61 6c ion = match real
0b70: 20 7b 0a 09 09 09 09 74 72 75 65 20 3d 3e 20 74 {.....true => t
0b80: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 elegram_bot::Use
0b90: 72 49 64 3a 3a 6e 65 77 28 63 68 61 6e 6e 65 6c rId::new(channel
0ba0: 5f 69 64 29 2c 0a 09 09 09 09 66 61 6c 73 65 20 _id),.....false
0bb0: 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a => telegram_bot:
0bc0: 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 72 6f 77 :UserId::new(row
0bd0: 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e 65 72 22 .try_get("owner"
0be0: 29 3f 29 2c 0a 09 09 09 7d 3b 0a 09 09 09 6c 65 )?),....};....le
0bf0: 74 20 6d 75 74 20 74 68 69 73 5f 66 65 74 63 68 t mut this_fetch
0c00: 3a 20 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 6d : Option<DateTim
0c10: 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f e<chrono::FixedO
0c20: 66 66 73 65 74 3e 3e 20 3d 20 4e 6f 6e 65 3b 0a ffset>> = None;.
0c30: 09 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 73 ...let mut posts
0c40: 3a 20 42 54 72 65 65 4d 61 70 3c 44 61 74 65 54 : BTreeMap<DateT
0c50: 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 ime<chrono::Fixe
0c60: 64 4f 66 66 73 65 74 3e 2c 20 53 74 72 69 6e 67 dOffset>, String
0c70: 3e 20 3d 20 42 54 72 65 65 4d 61 70 3a 3a 6e 65 > = BTreeMap::ne
0c80: 77 28 29 3b 0a 09 09 09 6c 65 74 20 63 6f 6e 74 w();....let cont
0c90: 65 6e 74 20 3d 20 72 65 71 77 65 73 74 3a 3a 67 ent = reqwest::g
0ca0: 65 74 28 75 72 6c 29 2e 61 77 61 69 74 3f 2e 62 et(url).await?.b
0cb0: 79 74 65 73 28 29 2e 61 77 61 69 74 3f 3b 0a 09 ytes().await?;..
0cc0: 09 09 6d 61 74 63 68 20 72 73 73 3a 3a 43 68 61 ..match rss::Cha
0cd0: 6e 6e 65 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 nnel::read_from(
0ce0: 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a &content[..]) {.
0cf0: 09 09 09 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 ....Ok(feed) =>
0d00: 7b 0a 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 {......for item
0d10: 69 6e 20 66 65 65 64 2e 69 74 65 6d 73 28 29 20 in feed.items()
0d20: 7b 0a 09 09 09 09 09 09 6d 61 74 63 68 20 69 74 {.......match it
0d30: 65 6d 2e 6c 69 6e 6b 28 29 20 7b 0a 09 09 09 09 em.link() {.....
0d40: 09 09 09 53 6f 6d 65 28 6c 69 6e 6b 29 20 3d 3e ...Some(link) =>
0d50: 20 7b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 64 {.........let d
0d60: 61 74 65 20 3d 20 6d 61 74 63 68 20 69 74 65 6d ate = match item
0d70: 2e 70 75 62 5f 64 61 74 65 28 29 20 7b 0a 09 09 .pub_date() {...
0d80: 09 09 09 09 09 09 09 53 6f 6d 65 28 66 65 65 64 .......Some(feed
0d90: 5f 64 61 74 65 29 20 3d 3e 20 44 61 74 65 54 69 _date) => DateTi
0da0: 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 me::parse_from_r
0db0: 66 63 32 38 32 32 28 66 65 65 64 5f 64 61 74 65 fc2822(feed_date
0dc0: 29 2c 0a 09 09 09 09 09 09 09 09 09 4e 6f 6e 65 ),..........None
0dd0: 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 => DateTime::pa
0de0: 72 73 65 5f 66 72 6f 6d 5f 72 66 63 33 33 33 39 rse_from_rfc3339
0df0: 28 26 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 63 6f (&item.dublin_co
0e00: 72 65 5f 65 78 74 28 29 2e 75 6e 77 72 61 70 28 re_ext().unwrap(
0e10: 29 2e 64 61 74 65 73 28 29 5b 30 5d 29 2c 0a 09 ).dates()[0]),..
0e20: 09 09 09 09 09 09 09 7d 3f 3b 0a 09 09 09 09 09 .......}?;......
0e30: 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 6c 69 6e ...let url = lin
0e40: 6b 2e 74 6f 5f 73 74 72 69 6e 67 28 29 3b 0a 09 k.to_string();..
0e50: 09 09 09 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 .......posts.ins
0e60: 65 72 74 28 64 61 74 65 2e 63 6c 6f 6e 65 28 29 ert(date.clone()
0e70: 2c 20 75 72 6c 2e 63 6c 6f 6e 65 28 29 29 3b 0a , url.clone());.
0e80: 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 .......},.......
0e90: 09 4e 6f 6e 65 20 3d 3e 20 7b 7d 0a 09 09 09 09 .None => {}.....
0ea0: 09 09 7d 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 ..}......};.....
0eb0: 7d 2c 0a 09 09 09 09 45 72 72 28 65 72 72 29 20 },.....Err(err)
0ec0: 3d 3e 20 6d 61 74 63 68 20 65 72 72 20 7b 0a 09 => match err {..
0ed0: 09 09 09 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a ....rss::Error::
0ee0: 49 6e 76 61 6c 69 64 53 74 61 72 74 54 61 67 20 InvalidStartTag
0ef0: 3d 3e 20 7b 0a 09 09 09 09 09 09 6c 65 74 20 66 => {.......let f
0f00: 65 65 64 20 3d 20 61 74 6f 6d 5f 73 79 6e 64 69 eed = atom_syndi
0f10: 63 61 74 69 6f 6e 3a 3a 46 65 65 64 3a 3a 72 65 cation::Feed::re
0f20: 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 ad_from(&content
0f30: 5b 2e 2e 5d 29 0a 09 09 09 09 09 09 09 2e 77 69 [..]).........wi
0f40: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
0f50: 72 6d 61 74 21 28 22 50 72 6f 62 6c 65 6d 20 6f rmat!("Problem o
0f60: 70 65 6e 69 6e 67 20 66 65 65 64 20 75 72 6c 3a pening feed url:
0f70: 5c 6e 7b 7d 22 2c 20 26 75 72 6c 29 29 3f 3b 0a \n{}", &url))?;.
0f80: 09 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 69 ......for item i
0f90: 6e 20 66 65 65 64 2e 65 6e 74 72 69 65 73 28 29 n feed.entries()
0fa0: 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 64 61 {........let da
0fb0: 74 65 20 3d 20 69 74 65 6d 2e 70 75 62 6c 69 73 te = item.publis
0fc0: 68 65 64 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a hed().unwrap();.
0fd0: 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c 20 3d .......let url =
0fe0: 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 29 5b 30 5d item.links()[0]
0ff0: 2e 68 72 65 66 28 29 3b 0a 09 09 09 09 09 09 09 .href();........
1000: 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 64 61 74 posts.insert(dat
1010: 65 2e 63 6c 6f 6e 65 28 29 2c 20 75 72 6c 2e 74 e.clone(), url.t
1020: 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 o_string());....
1030: 09 09 09 7d 3b 0a 09 09 09 09 09 7d 2c 0a 09 09 ...};......},...
1040: 09 09 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 ...rss::Error::E
1050: 6f 66 20 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f of => (),......_
1060: 20 3d 3e 20 62 61 69 6c 21 28 22 55 6e 73 75 70 => bail!("Unsup
1070: 70 6f 72 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 ported or mangle
1080: 64 20 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d d content:\n{:?}
1090: 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 26 75 72 6c \n{:#?}\n", &url
10a0: 2c 20 65 72 72 29 0a 09 09 09 09 7d 0a 09 09 09 , err).....}....
10b0: 7d 3b 0a 09 09 09 66 6f 72 20 28 64 61 74 65 2c };....for (date,
10c0: 20 75 72 6c 29 20 69 6e 20 70 6f 73 74 73 2e 69 url) in posts.i
10d0: 74 65 72 28 29 20 7b 0a 09 09 09 09 6c 65 74 20 ter() {.....let
10e0: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
10f0: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
1100: 77 61 69 74 0a 09 09 09 09 09 2e 77 69 74 68 5f wait.......with_
1110: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
1120: 74 21 28 22 43 68 65 63 6b 20 70 6f 73 74 20 66 t!("Check post f
1130: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
1140: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
1150: 3b 0a 09 09 09 09 6c 65 74 20 72 6f 77 20 3d 20 ;.....let row =
1160: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c sqlx::query("sel
1170: 65 63 74 20 65 78 69 73 74 73 28 73 65 6c 65 63 ect exists(selec
1180: 74 20 74 72 75 65 20 66 72 6f 6d 20 72 73 73 74 t true from rsst
1190: 67 5f 70 6f 73 74 20 77 68 65 72 65 20 75 72 6c g_post where url
11a0: 20 3d 20 24 31 20 61 6e 64 20 73 6f 75 72 63 65 = $1 and source
11b0: 5f 69 64 20 3d 20 24 32 29 20 61 73 20 65 78 69 _id = $2) as exi
11c0: 73 74 73 3b 22 29 0a 09 09 09 09 09 2e 62 69 6e sts;").......bin
11d0: 64 28 26 75 72 6c 29 0a 09 09 09 09 09 2e 62 69 d(&url).......bi
11e0: 6e 64 28 2a 69 64 29 0a 09 09 09 09 09 2e 66 65 nd(*id).......fe
11f0: 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 63 6f 6e tch_one(&mut con
1200: 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 09 2e 77 n).await.......w
1210: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
1220: 6f 72 6d 61 74 21 28 22 43 68 65 63 6b 20 70 6f ormat!("Check po
1230: 73 74 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e st:\n{:?}", &con
1240: 6e 29 29 3f 3b 0a 09 09 09 09 6c 65 74 20 65 78 n))?;.....let ex
1250: 69 73 74 73 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 ists: bool = row
1260: 2e 74 72 79 5f 67 65 74 28 22 65 78 69 73 74 73 .try_get("exists
1270: 22 29 3f 3b 0a 09 09 09 09 69 66 20 21 20 65 78 ")?;.....if ! ex
1280: 69 73 74 73 20 7b 0a 09 09 09 09 09 69 66 20 74 ists {......if t
1290: 68 69 73 5f 66 65 74 63 68 20 3d 3d 20 4e 6f 6e his_fetch == Non
12a0: 65 20 7c 7c 20 2a 64 61 74 65 20 3e 20 74 68 69 e || *date > thi
12b0: 73 5f 66 65 74 63 68 2e 75 6e 77 72 61 70 28 29 s_fetch.unwrap()
12c0: 20 7b 0a 09 09 09 09 09 09 74 68 69 73 5f 66 65 {.......this_fe
12d0: 74 63 68 20 3d 20 53 6f 6d 65 28 2a 64 61 74 65 tch = Some(*date
12e0: 29 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 );......};......
12f0: 73 65 6c 66 2e 74 67 2e 73 65 6e 64 28 20 6d 61 self.tg.send( ma
1300: 74 63 68 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 tch iv_hash {...
1310: 09 09 09 09 09 53 6f 6d 65 28 78 29 20 3d 3e 20 .....Some(x) =>
1320: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 telegram_bot::Se
1330: 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 ndMessage::new(d
1340: 65 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d estination, form
1350: 61 74 21 28 22 3c 61 20 68 72 65 66 3d 5c 22 68 at!("<a href=\"h
1360: 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 ttps://t.me/iv?u
1370: 72 6c 3d 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c 22 rl={}&rhash={}\"
1380: 3e 20 3c 2f 61 3e 7b 30 7d 22 2c 20 6d 61 74 63 > </a>{0}", matc
1390: 68 20 26 75 72 6c 5f 72 65 20 7b 0a 09 09 09 09 h &url_re {.....
13a0: 09 09 09 09 53 6f 6d 65 28 78 29 20 3d 3e 20 6d ....Some(x) => m
13b0: 61 74 63 68 20 78 2e 63 61 70 74 75 72 65 73 28 atch x.captures(
13c0: 26 75 72 6c 29 20 7b 0a 09 09 09 09 09 09 09 09 &url) {.........
13d0: 09 53 6f 6d 65 28 78 29 20 3d 3e 20 7b 0a 09 09 .Some(x) => {...
13e0: 09 09 09 09 09 09 09 09 62 61 69 6c 21 28 22 52 ........bail!("R
13f0: 65 67 65 78 20 68 69 74 2c 20 72 65 73 75 6c 74 egex hit, result
1400: 3a 5c 6e 7b 3a 23 3f 7d 22 2c 20 26 78 5b 30 5d :\n{:#?}", &x[0]
1410: 29 3b 0a 09 09 09 09 09 09 09 09 09 09 26 78 5b );...........&x[
1420: 30 5d 0a 09 09 09 09 09 09 09 09 09 7d 2c 0a 09 0]..........},..
1430: 09 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 ........None =>
1440: 26 75 72 6c 2c 0a 09 09 09 09 09 09 09 09 7d 2c &url,.........},
1450: 0a 09 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e .........None =>
1460: 20 26 75 72 6c 2c 0a 09 09 09 09 09 09 09 7d 2c &url,........},
1470: 20 78 29 29 2c 0a 09 09 09 09 09 09 09 4e 6f 6e x)),........Non
1480: 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f e => telegram_bo
1490: 74 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a t::SendMessage::
14a0: 6e 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c new(destination,
14b0: 20 66 6f 72 6d 61 74 21 28 22 7b 7d 22 2c 20 75 format!("{}", u
14c0: 72 6c 29 29 2c 0a 09 09 09 09 09 09 7d 2e 70 61 rl)),.......}.pa
14d0: 72 73 65 5f 6d 6f 64 65 28 74 65 6c 65 67 72 61 rse_mode(telegra
14e0: 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 61 m_bot::types::Pa
14f0: 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 2e rseMode::Html)).
1500: 61 77 61 69 74 0a 09 09 09 09 09 09 2e 63 6f 6e await........con
1510: 74 65 78 74 28 22 43 61 6e 27 74 20 70 6f 73 74 text("Can't post
1520: 20 6d 65 73 73 61 67 65 3a 22 29 3f 3b 0a 09 09 message:")?;...
1530: 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ...sqlx::query("
1540: 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 insert into rsst
1550: 67 5f 70 6f 73 74 20 28 73 6f 75 72 63 65 5f 69 g_post (source_i
1560: 64 2c 20 70 6f 73 74 65 64 2c 20 75 72 6c 29 20 d, posted, url)
1570: 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 values ($1, $2,
1580: 24 33 29 3b 22 29 0a 09 09 09 09 09 09 2e 62 69 $3);")........bi
1590: 6e 64 28 2a 69 64 29 0a 09 09 09 09 09 09 2e 62 nd(*id)........b
15a0: 69 6e 64 28 64 61 74 65 29 0a 09 09 09 09 09 09 ind(date).......
15b0: 2e 62 69 6e 64 28 75 72 6c 29 0a 09 09 09 09 09 .bind(url)......
15c0: 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 ..execute(&mut c
15d0: 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 09 onn).await......
15e0: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
15f0: 7c 20 66 6f 72 6d 61 74 21 28 22 52 65 63 6f 72 | format!("Recor
1600: 64 20 70 6f 73 74 3a 5c 6e 7b 3a 3f 7d 22 2c 20 d post:\n{:?}",
1610: 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 09 09 09 64 &conn))?;......d
1620: 72 6f 70 28 63 6f 6e 6e 29 3b 0a 09 09 09 09 09 rop(conn);......
1630: 74 6f 6b 69 6f 3a 3a 74 69 6d 65 3a 3a 73 6c 65 tokio::time::sle
1640: 65 70 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 ep(std::time::Du
1650: 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 34 2c 20 30 ration::new(4, 0
1660: 29 29 2e 61 77 61 69 74 3b 0a 09 09 09 09 7d 3b )).await;.....};
1670: 0a 09 09 09 09 70 6f 73 74 65 64 20 2b 3d 20 31 .....posted += 1
1680: 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 6f 73 74 73 ;....};....posts
1690: 2e 63 6c 65 61 72 28 29 3b 0a 09 09 7d 3b 0a 09 .clear();...};..
16a0: 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 .let mut conn =
16b0: 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 self.pool.acquir
16c0: 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 e().await.....wi
16d0: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
16e0: 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 73 63 rmat!("Update sc
16f0: 72 61 70 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a rape fetch conn:
1700: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
1710: 6f 6f 6c 29 29 3f 3b 0a 09 09 73 71 6c 78 3a 3a ool))?;...sqlx::
1720: 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 73 query("update rs
1730: 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 6c stg_source set l
1740: 61 73 74 5f 73 63 72 61 70 65 20 3d 20 6e 6f 77 ast_scrape = now
1750: 28 29 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f () where source_
1760: 69 64 20 3d 20 24 31 3b 22 29 0a 09 09 09 2e 62 id = $1;").....b
1770: 69 6e 64 28 2a 69 64 29 0a 09 09 09 2e 65 78 65 ind(*id).....exe
1780: 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e cute(&mut conn).
1790: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
17a0: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
17b0: 21 28 22 55 70 64 61 74 65 20 73 63 72 61 70 65 !("Update scrape
17c0: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 :\n{:?}", &conn)
17d0: 29 3f 3b 0a 09 09 4f 6b 28 66 6f 72 6d 61 74 21 )?;...Ok(format!
17e0: 28 22 50 6f 73 74 65 64 3a 20 7b 7d 22 2c 20 26 ("Posted: {}", &
17f0: 70 6f 73 74 65 64 29 29 0a 09 7d 0a 0a 09 70 75 posted))..}...pu
1800: 62 20 61 73 79 6e 63 20 66 6e 20 64 65 6c 65 74 b async fn delet
1810: 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 e<S>(&self, sour
1820: 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e ce_id: &i32, own
1830: 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 er: S) -> Result
1840: 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 72 65 20 <String>..where
1850: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
1860: 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 .let owner: i64
1870: 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a = owner.into();.
1880: 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d ..let mut conn =
1890: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
18a0: 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 re().await.....w
18b0: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
18c0: 6f 72 6d 61 74 21 28 22 44 65 6c 65 74 65 20 66 ormat!("Delete f
18d0: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
18e0: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
18f0: 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a ;...match sqlx::
1900: 71 75 65 72 79 28 22 64 65 6c 65 74 65 20 66 72 query("delete fr
1910: 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 om rsstg_source
1920: 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 where source_id
1930: 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d = $1 and owner =
1940: 20 24 32 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 $2;").....bind(
1950: 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 source_id).....b
1960: 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 ind(owner).....e
1970: 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e xecute(&mut conn
1980: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
1990: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
19a0: 61 74 21 28 22 44 65 6c 65 74 65 20 73 6f 75 72 at!("Delete sour
19b0: 63 65 20 72 75 6c 65 3a 5c 6e 7b 3a 3f 7d 22 2c ce rule:\n{:?}",
19c0: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 &self.pool))?..
19d0: 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 ...rows_affected
19e0: 28 29 20 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f () {....0 => { O
19f0: 6b 28 22 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 k("No data found
1a00: 20 66 6f 75 6e 64 5c 5c 2e 22 2e 74 6f 5f 73 74 found\\.".to_st
1a10: 72 69 6e 67 28 29 29 20 7d 2c 0a 09 09 09 78 20 ring()) },....x
1a20: 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 => { Ok(format!(
1a30: 22 7b 7d 20 73 6f 75 72 63 65 73 20 72 65 6d 6f "{} sources remo
1a40: 76 65 64 5c 5c 2e 22 2c 20 78 29 29 20 7d 2c 0a ved\\.", x)) },.
1a50: 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ..}..}...pub asy
1a60: 6e 63 20 66 6e 20 63 6c 65 61 6e 3c 53 3e 28 26 nc fn clean<S>(&
1a70: 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a self, source_id:
1a80: 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 &i32, owner: S)
1a90: 20 2d 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e -> Result<Strin
1aa0: 67 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 g>..where S: Int
1ab0: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f o<i64> {...let o
1ac0: 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 wner: i64 = owne
1ad0: 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 r.into();...let
1ae0: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
1af0: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
1b00: 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f wait.....with_co
1b10: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
1b20: 28 22 43 6c 65 61 6e 20 66 65 74 63 68 20 63 6f ("Clean fetch co
1b30: 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c nn:\n{:?}", &sel
1b40: 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 f.pool))?;...mat
1b50: 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ch sqlx::query("
1b60: 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 74 delete from rsst
1b70: 67 5f 70 6f 73 74 20 70 20 75 73 69 6e 67 20 72 g_post p using r
1b80: 73 73 74 67 5f 73 6f 75 72 63 65 20 73 20 77 68 sstg_source s wh
1b90: 65 72 65 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 ere p.source_id
1ba0: 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d = $1 and owner =
1bb0: 20 24 32 20 61 6e 64 20 70 2e 73 6f 75 72 63 65 $2 and p.source
1bc0: 5f 69 64 20 3d 20 73 2e 73 6f 75 72 63 65 5f 69 _id = s.source_i
1bd0: 64 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f d;").....bind(so
1be0: 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e urce_id).....bin
1bf0: 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 d(owner).....exe
1c00: 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e cute(&mut conn).
1c10: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
1c20: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1c30: 21 28 22 43 6c 65 61 6e 20 73 65 65 6e 20 70 6f !("Clean seen po
1c40: 73 74 73 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 sts:\n{:?}", &se
1c50: 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 lf.pool))?.....r
1c60: 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b ows_affected() {
1c70: 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e ....0 => { Ok("N
1c80: 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 o data found fou
1c90: 6e 64 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 nd\\.".to_string
1ca0: 28 29 29 20 7d 2c 0a 09 09 09 78 20 3d 3e 20 7b ()) },....x => {
1cb0: 20 4f 6b 28 66 6f 72 6d 61 74 21 28 22 7b 7d 20 Ok(format!("{}
1cc0: 70 6f 73 74 73 20 70 75 72 67 65 64 5c 5c 2e 22 posts purged\\."
1cd0: 2c 20 78 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a , x)) },...}..}.
1ce0: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 65 ..pub async fn e
1cf0: 6e 61 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 nable<S>(&self,
1d00: 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c source_id: &i32,
1d10: 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 owner: S) -> Re
1d20: 73 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 sult<&str>..wher
1d30: 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e S: Into<i64> {
1d40: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 ...let owner: i6
1d50: 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 4 = owner.into()
1d60: 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ;...let mut conn
1d70: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
1d80: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
1d90: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
1da0: 20 66 6f 72 6d 61 74 21 28 22 45 6e 61 62 6c 65 format!("Enable
1db0: 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a fetch conn:\n{:
1dc0: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
1dd0: 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 )?;...match sqlx
1de0: 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 20 ::query("update
1df0: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 rsstg_source set
1e00: 20 65 6e 61 62 6c 65 64 20 3d 20 74 72 75 65 20 enabled = true
1e10: 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 where source_id
1e20: 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d = $1 and owner =
1e30: 20 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 $2").....bind(s
1e40: 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 ource_id).....bi
1e50: 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 nd(owner).....ex
1e60: 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 ecute(&mut conn)
1e70: 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f .await.....with_
1e80: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
1e90: 74 21 28 22 45 6e 61 62 6c 65 20 73 6f 75 72 63 t!("Enable sourc
1ea0: 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 e:\n{:?}", &self
1eb0: 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 .pool))?.....row
1ec0: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
1ed0: 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 ..1 => { Ok("Sou
1ee0: 72 63 65 20 65 6e 61 62 6c 65 64 5c 5c 2e 22 29 rce enabled\\.")
1ef0: 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b },....0 => { Ok
1f00: 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 ("Source not fou
1f10: 6e 64 5c 5c 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 nd\\.") },...._
1f20: 3d 3e 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 => { Err(anyhow!
1f30: 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 ("Database error
1f40: 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a .")) },...}..}..
1f50: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 64 69 .pub async fn di
1f60: 73 61 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 sable<S>(&self,
1f70: 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c source_id: &i32,
1f80: 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 owner: S) -> Re
1f90: 73 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 sult<&str>..wher
1fa0: 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e S: Into<i64> {
1fb0: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 ...let owner: i6
1fc0: 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 4 = owner.into()
1fd0: 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ;...let mut conn
1fe0: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
1ff0: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
2000: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
2010: 20 66 6f 72 6d 61 74 21 28 22 44 69 73 61 62 6c format!("Disabl
2020: 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b e fetch conn:\n{
2030: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
2040: 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c ))?;...match sql
2050: 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 x::query("update
2060: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 rsstg_source se
2070: 74 20 65 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 t enabled = fals
2080: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
2090: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
20a0: 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 = $2").....bind
20b0: 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e (source_id).....
20c0: 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e bind(owner).....
20d0: 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e execute(&mut con
20e0: 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 n).await.....wit
20f0: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
2100: 6d 61 74 21 28 22 44 69 73 61 62 6c 65 20 73 6f mat!("Disable so
2110: 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 urce:\n{:?}", &s
2120: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e elf.pool))?.....
2130: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 rows_affected()
2140: 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 {....1 => { Ok("
2150: 53 6f 75 72 63 65 20 64 69 73 61 62 6c 65 64 5c Source disabled\
2160: 5c 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 \.") },....0 =>
2170: 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 { Ok("Source not
2180: 20 66 6f 75 6e 64 5c 5c 2e 22 29 20 7d 2c 0a 09 found\\.") },..
2190: 09 09 5f 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 .._ => { Err(any
21a0: 68 6f 77 21 28 22 44 61 74 61 62 61 73 65 20 65 how!("Database e
21b0: 72 72 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a rror.")) },...}.
21c0: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
21d0: 6e 20 75 70 64 61 74 65 3c 53 3e 28 26 73 65 6c n update<S>(&sel
21e0: 66 2c 20 75 70 64 61 74 65 3a 20 4f 70 74 69 6f f, update: Optio
21f0: 6e 3c 69 33 32 3e 2c 20 63 68 61 6e 6e 65 6c 3a n<i32>, channel:
2200: 20 26 73 74 72 2c 20 63 68 61 6e 6e 65 6c 5f 69 &str, channel_i
2210: 64 3a 20 69 36 34 2c 20 75 72 6c 3a 20 26 73 74 d: i64, url: &st
2220: 72 2c 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 r, iv_hash: Opti
2230: 6f 6e 3c 26 73 74 72 3e 2c 20 75 72 6c 5f 72 65 on<&str>, url_re
2240: 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 : Option<&str>,
2250: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
2260: 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 ult<String>..whe
2270: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
2280: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 {...let owner: i
2290: 36 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 64 = owner.into(
22a0: 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e );...let mut con
22b0: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
22c0: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
22d0: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
22e0: 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 74 | format!("Updat
22f0: 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b e fetch conn:\n{
2300: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
2310: 29 29 3f 3b 0a 0a 09 09 6d 61 74 63 68 20 6d 61 ))?;....match ma
2320: 74 63 68 20 75 70 64 61 74 65 20 7b 0a 09 09 09 tch update {....
2330: 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 7b 0a 09 .Some(id) => {..
2340: 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 ....sqlx::query(
2350: 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f "update rsstg_so
2360: 75 72 63 65 20 73 65 74 20 63 68 61 6e 6e 65 6c urce set channel
2370: 5f 69 64 20 3d 20 24 32 2c 20 75 72 6c 20 3d 20 _id = $2, url =
2380: 24 33 2c 20 69 76 5f 68 61 73 68 20 3d 20 24 34 $3, iv_hash = $4
2390: 2c 20 6f 77 6e 65 72 20 3d 20 24 35 2c 20 63 68 , owner = $5, ch
23a0: 61 6e 6e 65 6c 20 3d 20 24 36 20 77 68 65 72 65 annel = $6 where
23b0: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 22 source_id = $1"
23c0: 29 2e 62 69 6e 64 28 69 64 29 0a 09 09 09 09 7d ).bind(id).....}
23d0: 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a ,.....None => {.
23e0: 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 .....sqlx::query
23f0: 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 ("insert into rs
2400: 73 74 67 5f 73 6f 75 72 63 65 20 28 63 68 61 6e stg_source (chan
2410: 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f nel_id, url, iv_
2420: 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 hash, owner, cha
2430: 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 20 76 61 nnel, url_re) va
2440: 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 24 33 lues ($1, $2, $3
2450: 2c 20 24 34 2c 20 24 35 2c 20 24 36 29 22 29 0a , $4, $5, $6)").
2460: 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 09 2e ....},....}.....
2470: 62 69 6e 64 28 63 68 61 6e 6e 65 6c 5f 69 64 29 bind(channel_id)
2480: 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c 29 0a 09 .....bind(url)..
2490: 09 09 2e 62 69 6e 64 28 69 76 5f 68 61 73 68 29 ...bind(iv_hash)
24a0: 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 .....bind(owner)
24b0: 0a 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 .....bind(channe
24c0: 6c 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c 5f l).....bind(url_
24d0: 72 65 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 re).....execute(
24e0: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
24f0: 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 72 {....Ok(_) => r
2500: 65 74 75 72 6e 20 4f 6b 28 53 74 72 69 6e 67 3a eturn Ok(String:
2510: 3a 66 72 6f 6d 28 6d 61 74 63 68 20 75 70 64 61 :from(match upda
2520: 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 5f 29 te {.....Some(_)
2530: 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 75 70 64 => "Channel upd
2540: 61 74 65 64 5c 5c 2e 22 2c 0a 09 09 09 09 4e 6f ated\\.",.....No
2550: 6e 65 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 61 ne => "Channel a
2560: 64 64 65 64 5c 5c 2e 22 2c 0a 09 09 09 7d 29 29 dded\\.",....}))
2570: 2c 0a 09 09 09 45 72 72 28 73 71 6c 78 3a 3a 45 ,....Err(sqlx::E
2580: 72 72 6f 72 3a 3a 44 61 74 61 62 61 73 65 28 65 rror::Database(e
2590: 72 72 29 29 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 rr)) => {.....ma
25a0: 74 63 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 tch err.downcast
25b0: 3a 3a 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 ::<sqlx::postgre
25c0: 73 3a 3a 50 67 44 61 74 61 62 61 73 65 45 72 72 s::PgDatabaseErr
25d0: 6f 72 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 or>().routine()
25e0: 7b 0a 09 09 09 09 09 53 6f 6d 65 28 22 5f 62 74 {......Some("_bt
25f0: 5f 63 68 65 63 6b 5f 75 6e 69 71 75 65 22 2c 20 _check_unique",
2600: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 72 65 74 ) => {.......ret
2610: 75 72 6e 20 4f 6b 28 22 44 75 70 6c 69 63 61 74 urn Ok("Duplicat
2620: 65 20 6b 65 79 5c 5c 2e 22 2e 74 6f 5f 73 74 72 e key\\.".to_str
2630: 69 6e 67 28 29 29 0a 09 09 09 09 09 7d 2c 0a 09 ing())......},..
2640: 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 7b ....Some(_) => {
2650: 0a 09 09 09 09 09 09 72 65 74 75 72 6e 20 4f 6b .......return Ok
2660: 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 ("Database error
2670: 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 \\.".to_string()
2680: 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 4e )......},......N
2690: 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 09 72 one => {.......r
26a0: 65 74 75 72 6e 20 4f 6b 28 22 4e 6f 20 64 61 74 eturn Ok("No dat
26b0: 61 62 61 73 65 20 65 72 72 6f 72 20 65 78 74 72 abase error extr
26c0: 61 63 74 65 64 5c 5c 2e 22 2e 74 6f 5f 73 74 72 acted\\.".to_str
26d0: 69 6e 67 28 29 29 0a 09 09 09 09 09 7d 2c 0a 09 ing())......},..
26e0: 09 09 09 7d 3b 0a 09 09 09 7d 2c 0a 09 09 09 45 ...};....},....E
26f0: 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 rr(err) => {....
2700: 09 62 61 69 6c 21 28 22 53 6f 72 72 79 2c 20 75 .bail!("Sorry, u
2710: 6e 6b 6e 6f 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b nknown error:\n{
2720: 3a 23 3f 7d 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 :#?}\n", err);..
2730: 09 09 7d 2c 0a 09 09 7d 3b 0a 09 7d 0a 0a 09 61 ..},...};..}...a
2740: 73 79 6e 63 20 66 6e 20 61 75 74 6f 66 65 74 63 sync fn autofetc
2750: 68 28 26 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 h(&self) -> Resu
2760: 6c 74 3c 28 29 3e 20 7b 0a 09 09 6c 65 74 20 6d lt<()> {...let m
2770: 75 74 20 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e ut delay = chron
2780: 6f 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e o::Duration::min
2790: 75 74 65 73 28 31 29 3b 0a 09 09 6c 65 74 20 6d utes(1);...let m
27a0: 75 74 20 6e 6f 77 3b 0a 09 09 6c 6f 6f 70 20 7b ut now;...loop {
27b0: 0a 09 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ....let mut conn
27c0: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
27d0: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
27e0: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
27f0: 7c 20 66 6f 72 6d 61 74 21 28 22 41 75 74 6f 66 | format!("Autof
2800: 65 74 63 68 20 66 65 74 63 68 20 63 6f 6e 6e 3a etch fetch conn:
2810: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
2820: 6f 6f 6c 29 29 3f 3b 0a 09 09 09 6e 6f 77 20 3d ool))?;....now =
2830: 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a chrono::Local::
2840: 6e 6f 77 28 29 3b 0a 09 09 09 6c 65 74 20 6d 75 now();....let mu
2850: 74 20 71 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a t queue = sqlx::
2860: 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f query("select so
2870: 75 72 63 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 urce_id, next_fe
2880: 74 63 68 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 tch, owner from
2890: 72 73 73 74 67 5f 6f 72 64 65 72 20 6e 61 74 75 rsstg_order natu
28a0: 72 61 6c 20 6c 65 66 74 20 6a 6f 69 6e 20 72 73 ral left join rs
28b0: 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 stg_source where
28c0: 20 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f next_fetch < no
28d0: 77 28 29 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 w() + interval '
28e0: 31 20 6d 69 6e 75 74 65 27 3b 22 29 0a 09 09 09 1 minute';")....
28f0: 09 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 ..fetch_all(&mut
2900: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 conn).await?;..
2910: 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 71 75 65 ..for row in que
2920: 75 65 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 09 ue.iter() {.....
2930: 6c 65 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 let source_id: i
2940: 33 32 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 32 = row.try_get
2950: 28 22 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a ("source_id")?;.
2960: 09 09 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 ....let owner: i
2970: 36 34 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 64 = row.try_get
2980: 28 22 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 09 09 ("owner")?;.....
2990: 6c 65 74 20 6e 65 78 74 5f 66 65 74 63 68 3a 20 let next_fetch:
29a0: 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a DateTime<chrono:
29b0: 3a 4c 6f 63 61 6c 3e 20 3d 20 72 6f 77 2e 74 72 :Local> = row.tr
29c0: 79 5f 67 65 74 28 22 6e 65 78 74 5f 66 65 74 63 y_get("next_fetc
29d0: 68 22 29 3f 3b 0a 09 09 09 09 69 66 20 6e 65 78 h")?;.....if nex
29e0: 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a t_fetch < now {.
29f0: 09 09 09 09 09 2f 2f 6c 65 74 20 63 6c 6f 6e 65 .....//let clone
2a00: 20 3d 20 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 3b = self.clone();
2a10: 0a 09 09 09 09 09 2f 2f 63 6c 6f 6e 65 2e 6f 77 ......//clone.ow
2a20: 6e 65 72 5f 63 68 61 74 28 55 73 65 72 49 64 3a ner_chat(UserId:
2a30: 3a 6e 65 77 28 6f 77 6e 65 72 29 29 3b 0a 09 09 :new(owner));...
2a40: 09 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 ...let clone = C
2a50: 6f 72 65 20 7b 0a 09 09 09 09 09 09 6f 77 6e 65 ore {.......owne
2a60: 72 5f 63 68 61 74 3a 20 74 65 6c 65 67 72 61 6d r_chat: telegram
2a70: 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 _bot::UserId::ne
2a80: 77 28 6f 77 6e 65 72 29 2c 0a 09 09 09 09 09 09 w(owner),.......
2a90: 2e 2e 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 0a 09 ..self.clone()..
2aa0: 09 09 09 09 7d 3b 0a 09 09 09 09 09 74 6f 6b 69 ....};......toki
2ab0: 6f 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 6d o::spawn(async m
2ac0: 6f 76 65 20 7b 0a 09 09 09 09 09 09 69 66 20 6c ove {.......if l
2ad0: 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c et Err(err) = cl
2ae0: 6f 6e 65 2e 63 68 65 63 6b 28 26 73 6f 75 72 63 one.check(&sourc
2af0: 65 5f 69 64 2c 20 6f 77 6e 65 72 2c 20 74 72 75 e_id, owner, tru
2b00: 65 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 e).await {......
2b10: 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 ..if let Err(err
2b20: 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 26 ) = clone.send(&
2b30: 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a format!("š {:
2b40: 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 29 ?}", err), None)
2b50: 20 7b 0a 09 09 09 09 09 09 09 09 65 70 72 69 6e {.........eprin
2b60: 74 6c 6e 21 28 22 43 68 65 63 6b 20 65 72 72 6f tln!("Check erro
2b70: 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 r: {}", err);...
2b80: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 3b .....};.......};
2b90: 0a 09 09 09 09 09 7d 29 3b 0a 09 09 09 09 7d 20 ......});.....}
2ba0: 65 6c 73 65 20 7b 0a 09 09 09 09 09 69 66 20 6e else {......if n
2bb0: 65 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 20 ext_fetch - now
2bc0: 3c 20 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 09 < delay {.......
2bd0: 64 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 delay = next_fet
2be0: 63 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 09 7d ch - now;......}
2bf0: 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 .....}....};....
2c00: 71 75 65 75 65 2e 63 6c 65 61 72 28 29 3b 0a 09 queue.clear();..
2c10: 09 09 74 6f 6b 69 6f 3a 3a 74 69 6d 65 3a 3a 73 ..tokio::time::s
2c20: 6c 65 65 70 28 64 65 6c 61 79 2e 74 6f 5f 73 74 leep(delay.to_st
2c30: 64 28 29 3f 29 2e 61 77 61 69 74 3b 0a 09 09 09 d()?).await;....
2c40: 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a delay = chrono::
2c50: 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 Duration::minute
2c60: 73 28 31 29 3b 0a 09 09 7d 0a 09 7d 0a 0a 09 70 s(1);...}..}...p
2c70: 75 62 20 61 73 79 6e 63 20 66 6e 20 6c 69 73 74 ub async fn list
2c80: 3c 53 3e 28 26 73 65 6c 66 2c 20 6f 77 6e 65 72 <S>(&self, owner
2c90: 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 56 : S) -> Result<V
2ca0: 65 63 3c 53 74 72 69 6e 67 3e 3e 0a 09 77 68 65 ec<String>>..whe
2cb0: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
2cc0: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
2cd0: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 owner.into();...
2ce0: 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 20 3d 20 let mut reply =
2cf0: 76 65 63 21 5b 5d 3b 0a 09 09 6c 65 74 20 6d 75 vec![];...let mu
2d00: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
2d10: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
2d20: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
2d30: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
2d40: 4c 69 73 74 20 66 65 74 63 68 20 63 6f 6e 6e 3a List fetch conn:
2d50: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
2d60: 6f 6f 6c 29 29 3f 3b 0a 09 09 72 65 70 6c 79 2e ool))?;...reply.
2d70: 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 push("Channels:"
2d80: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 .to_string());..
2d90: 09 6c 65 74 20 72 6f 77 73 20 3d 20 73 71 6c 78 .let rows = sqlx
2da0: 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 ::query("select
2db0: 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e source_id, chann
2dc0: 65 6c 2c 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c el, enabled, url
2dd0: 2c 20 69 76 5f 68 61 73 68 20 66 72 6f 6d 20 72 , iv_hash from r
2de0: 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 sstg_source wher
2df0: 65 20 6f 77 6e 65 72 20 3d 20 24 31 20 6f 72 64 e owner = $1 ord
2e00: 65 72 20 62 79 20 73 6f 75 72 63 65 5f 69 64 22 er by source_id"
2e10: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
2e20: 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 ).....fetch_all(
2e30: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
2e40: 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 ?;...for row in
2e50: 72 6f 77 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 rows.iter() {...
2e60: 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 .let source_id:
2e70: 69 33 32 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 i32 = row.try_ge
2e80: 74 28 22 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b t("source_id")?;
2e90: 0a 09 09 09 6c 65 74 20 75 73 65 72 6e 61 6d 65 ....let username
2ea0: 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 : &str = row.try
2eb0: 5f 67 65 74 28 22 63 68 61 6e 6e 65 6c 22 29 3f _get("channel")?
2ec0: 3b 0a 09 09 09 6c 65 74 20 65 6e 61 62 6c 65 64 ;....let enabled
2ed0: 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 : bool = row.try
2ee0: 5f 67 65 74 28 22 65 6e 61 62 6c 65 64 22 29 3f _get("enabled")?
2ef0: 3b 0a 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 73 ;....let url: &s
2f00: 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 tr = row.try_get
2f10: 28 22 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 ("url")?;....let
2f20: 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e iv_hash: Option
2f30: 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 <&str> = row.try
2f40: 5f 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f _get("iv_hash")?
2f50: 3b 0a 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 ;....reply.push(
2f60: 66 6f 72 6d 61 74 21 28 22 5c 6e 5c 5c 23 ef b8 format!("\n\\#ļø
2f70: 8f e2 83 a3 20 7b 7d 20 5c 5c 2a ef b8 8f e2 83 ā£ {} \\*ļøā
2f80: a3 20 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 97 20 £ `{}` {}\nš
2f90: 60 7b 7d 60 22 2c 20 73 6f 75 72 63 65 5f 69 64 `{}`", source_id
2fa0: 2c 20 75 73 65 72 6e 61 6d 65 2c 20 20 0a 09 09 , username, ...
2fb0: 09 09 6d 61 74 63 68 20 65 6e 61 62 6c 65 64 20 ..match enabled
2fc0: 7b 0a 09 09 09 09 09 74 72 75 65 20 20 3d 3e 20 {......true =>
2fd0: 22 f0 9f 94 84 20 65 6e 61 62 6c 65 64 22 2c 0a "š enabled",.
2fe0: 09 09 09 09 09 66 61 6c 73 65 20 3d 3e 20 22 e2 .....false => "ā
2ff0: 9b 94 20 64 69 73 61 62 6c 65 64 22 2c 0a 09 09 disabled",...
3000: 09 09 7d 2c 20 75 72 6c 29 29 3b 0a 09 09 09 69 ..}, url));....i
3010: 66 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 68 29 f let Some(hash)
3020: 20 3d 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 = iv_hash {....
3030: 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d .reply.push(form
3040: 61 74 21 28 22 49 56 20 60 7b 7d 60 22 2c 20 68 at!("IV `{}`", h
3050: 61 73 68 29 29 3b 0a 09 09 09 7d 0a 09 09 7d 3b ash));....}...};
3060: 0a 09 09 4f 6b 28 72 65 70 6c 79 29 0a 09 7d 0a ...Ok(reply)..}.
3070: 7d 0a }.