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 71 77 65 73 74 fig;.use reqwest
0070: 3b 0a 75 73 65 20 73 71 6c 78 3a 3a 7b 0a 09 70 ;.use sqlx::{..p
0080: 6f 73 74 67 72 65 73 3a 3a 50 67 50 6f 6f 6c 4f ostgres::PgPoolO
0090: 70 74 69 6f 6e 73 2c 0a 09 52 6f 77 2c 0a 7d 3b ptions,..Row,.};
00a0: 0a 75 73 65 20 72 73 73 3b 0a 75 73 65 20 73 74 .use rss;.use st
00b0: 64 3a 3a 7b 0a 09 62 6f 72 72 6f 77 3a 3a 43 6f d::{..borrow::Co
00c0: 77 2c 0a 09 63 6f 6c 6c 65 63 74 69 6f 6e 73 3a w,..collections:
00d0: 3a 7b 0a 09 09 42 54 72 65 65 4d 61 70 2c 0a 09 :{...BTreeMap,..
00e0: 09 48 61 73 68 53 65 74 2c 0a 09 7d 2c 0a 09 73 .HashSet,..},..s
00f0: 79 6e 63 3a 3a 7b 41 72 63 2c 20 4d 75 74 65 78 ync::{Arc, Mutex
0100: 7d 2c 0a 7d 3b 0a 75 73 65 20 74 65 6c 65 67 72 },.};.use telegr
0110: 61 6d 5f 62 6f 74 3b 0a 0a 23 5b 64 65 72 69 76 am_bot;..#[deriv
0120: 65 28 43 6c 6f 6e 65 29 5d 0a 70 75 62 20 73 74 e(Clone)].pub st
0130: 72 75 63 74 20 43 6f 72 65 20 7b 0a 09 6f 77 6e ruct Core {..own
0140: 65 72 3a 20 69 36 34 2c 0a 09 61 70 69 5f 6b 65 er: i64,..api_ke
0150: 79 3a 20 53 74 72 69 6e 67 2c 0a 09 6f 77 6e 65 y: String,..owne
0160: 72 5f 63 68 61 74 3a 20 74 65 6c 65 67 72 61 6d r_chat: telegram
0170: 5f 62 6f 74 3a 3a 55 73 65 72 49 64 2c 0a 09 70 _bot::UserId,..p
0180: 75 62 20 74 67 3a 20 74 65 6c 65 67 72 61 6d 5f ub tg: telegram_
0190: 62 6f 74 3a 3a 41 70 69 2c 0a 09 70 75 62 20 6d bot::Api,..pub m
01a0: 79 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a y: telegram_bot:
01b0: 3a 55 73 65 72 2c 0a 09 70 6f 6f 6c 3a 20 73 71 :User,..pool: sq
01c0: 6c 78 3a 3a 50 6f 6f 6c 3c 73 71 6c 78 3a 3a 50 lx::Pool<sqlx::P
01d0: 6f 73 74 67 72 65 73 3e 2c 0a 09 73 6f 75 72 63 ostgres>,..sourc
01e0: 65 73 3a 20 41 72 63 3c 4d 75 74 65 78 3c 48 61 es: Arc<Mutex<Ha
01f0: 73 68 53 65 74 3c 41 72 63 3c 69 33 32 3e 3e 3e shSet<Arc<i32>>>
0200: 3e 2c 0a 7d 0a 0a 69 6d 70 6c 20 43 6f 72 65 20 >,.}..impl Core
0210: 7b 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 {..pub async fn
0220: 6e 65 77 28 73 65 74 74 69 6e 67 73 3a 20 63 6f new(settings: co
0230: 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 29 20 2d 3e nfig::Config) ->
0240: 20 52 65 73 75 6c 74 3c 43 6f 72 65 3e 20 7b 0a Result<Core> {.
0250: 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 73 65 ..let owner = se
0260: 74 74 69 6e 67 73 2e 67 65 74 5f 69 6e 74 28 22 ttings.get_int("
0270: 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 6c 65 74 20 owner")?;...let
0280: 61 70 69 5f 6b 65 79 20 3d 20 73 65 74 74 69 6e api_key = settin
0290: 67 73 2e 67 65 74 5f 73 74 72 28 22 61 70 69 5f gs.get_str("api_
02a0: 6b 65 79 22 29 3f 3b 0a 09 09 6c 65 74 20 74 67 key")?;...let tg
02b0: 20 3d 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a = telegram_bot:
02c0: 3a 41 70 69 3a 3a 6e 65 77 28 26 61 70 69 5f 6b :Api::new(&api_k
02d0: 65 79 29 3b 0a 09 09 6c 65 74 20 63 6f 72 65 20 ey);...let core
02e0: 3d 20 43 6f 72 65 20 7b 0a 09 09 09 6f 77 6e 65 = Core {....owne
02f0: 72 3a 20 6f 77 6e 65 72 2c 0a 09 09 09 61 70 69 r: owner,....api
0300: 5f 6b 65 79 3a 20 61 70 69 5f 6b 65 79 2e 63 6c _key: api_key.cl
0310: 6f 6e 65 28 29 2c 0a 09 09 09 6d 79 3a 20 74 67 one(),....my: tg
0320: 2e 73 65 6e 64 28 74 65 6c 65 67 72 61 6d 5f 62 .send(telegram_b
0330: 6f 74 3a 3a 47 65 74 4d 65 29 2e 61 77 61 69 74 ot::GetMe).await
0340: 3f 2c 0a 09 09 09 74 67 3a 20 74 67 2c 0a 09 09 ?,....tg: tg,...
0350: 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 74 65 6c .owner_chat: tel
0360: 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 egram_bot::UserI
0370: 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 2c 0a 09 d::new(owner),..
0380: 09 09 70 6f 6f 6c 3a 20 50 67 50 6f 6f 6c 4f 70 ..pool: PgPoolOp
0390: 74 69 6f 6e 73 3a 3a 6e 65 77 28 29 0a 09 09 09 tions::new()....
03a0: 09 2e 6d 61 78 5f 63 6f 6e 6e 65 63 74 69 6f 6e ..max_connection
03b0: 73 28 35 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 s(5)......connec
03c0: 74 5f 74 69 6d 65 6f 75 74 28 73 74 64 3a 3a 74 t_timeout(std::t
03d0: 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e ime::Duration::n
03e0: 65 77 28 33 30 30 2c 20 30 29 29 0a 09 09 09 09 ew(300, 0)).....
03f0: 2e 69 64 6c 65 5f 74 69 6d 65 6f 75 74 28 73 74 .idle_timeout(st
0400: 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f d::time::Duratio
0410: 6e 3a 3a 6e 65 77 28 36 30 2c 20 30 29 29 0a 09 n::new(60, 0))..
0420: 09 09 09 2e 63 6f 6e 6e 65 63 74 5f 6c 61 7a 79 ....connect_lazy
0430: 28 26 73 65 74 74 69 6e 67 73 2e 67 65 74 5f 73 (&settings.get_s
0440: 74 72 28 22 70 67 22 29 3f 29 3f 2c 0a 09 09 09 tr("pg")?)?,....
0450: 73 6f 75 72 63 65 73 3a 20 41 72 63 3a 3a 6e 65 sources: Arc::ne
0460: 77 28 4d 75 74 65 78 3a 3a 6e 65 77 28 48 61 73 w(Mutex::new(Has
0470: 68 53 65 74 3a 3a 6e 65 77 28 29 29 29 2c 0a 09 hSet::new())),..
0480: 09 7d 3b 0a 09 09 6c 65 74 20 63 6c 6f 6e 65 20 .};...let clone
0490: 3d 20 63 6f 72 65 2e 63 6c 6f 6e 65 28 29 3b 0a = core.clone();.
04a0: 09 09 74 6f 6b 69 6f 3a 3a 73 70 61 77 6e 28 61 ..tokio::spawn(a
04b0: 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 69 sync move {....i
04c0: 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 3d f let Err(err) =
04d0: 20 26 63 6c 6f 6e 65 2e 61 75 74 6f 66 65 74 63 &clone.autofetc
04e0: 68 28 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 h().await {.....
04f0: 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 if let Err(err)
0500: 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 26 66 6f = clone.send(&fo
0510: 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a 3f 7d rmat!("š {:?}
0520: 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 2c 20 4e ", err), None, N
0530: 6f 6e 65 29 20 7b 0a 09 09 09 09 09 65 70 72 69 one) {......epri
0540: 6e 74 6c 6e 21 28 22 41 75 74 6f 66 65 74 63 68 ntln!("Autofetch
0550: 20 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 error: {}", err
0560: 29 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 7d 0a 09 );.....};....}..
0570: 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 0a .});...Ok(core).
0580: 09 7d 0a 0a 09 70 75 62 20 66 6e 20 73 74 72 65 .}...pub fn stre
0590: 61 6d 28 26 73 65 6c 66 29 20 2d 3e 20 74 65 6c am(&self) -> tel
05a0: 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 70 64 61 74 egram_bot::Updat
05b0: 65 73 53 74 72 65 61 6d 20 7b 0a 09 09 73 65 6c esStream {...sel
05c0: 66 2e 74 67 2e 73 74 72 65 61 6d 28 29 0a 09 7d f.tg.stream()..}
05d0: 0a 0a 09 70 75 62 20 66 6e 20 73 65 6e 64 3c 27 ...pub fn send<'
05e0: 61 2c 20 53 3e 28 26 73 65 6c 66 2c 20 6d 73 67 a, S>(&self, msg
05f0: 3a 20 53 2c 20 74 61 72 67 65 74 3a 20 4f 70 74 : S, target: Opt
0600: 69 6f 6e 3c 74 65 6c 65 67 72 61 6d 5f 62 6f 74 ion<telegram_bot
0610: 3a 3a 55 73 65 72 49 64 3e 2c 20 70 61 72 73 65 ::UserId>, parse
0620: 5f 6d 6f 64 65 3a 20 4f 70 74 69 6f 6e 3c 74 65 _mode: Option<te
0630: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 legram_bot::type
0640: 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3e 29 20 2d s::ParseMode>) -
0650: 3e 20 52 65 73 75 6c 74 3c 28 29 3e 0a 09 77 68 > Result<()>..wh
0660: 65 72 65 20 53 3a 20 49 6e 74 6f 3c 43 6f 77 3c ere S: Into<Cow<
0670: 27 61 2c 20 73 74 72 3e 3e 20 7b 0a 09 09 6c 65 'a, str>> {...le
0680: 74 20 6d 73 67 20 3d 20 6d 73 67 2e 69 6e 74 6f t msg = msg.into
0690: 28 29 3b 0a 0a 09 09 6c 65 74 20 70 61 72 73 65 ();....let parse
06a0: 5f 6d 6f 64 65 20 3d 20 6d 61 74 63 68 20 70 61 _mode = match pa
06b0: 72 73 65 5f 6d 6f 64 65 20 7b 0a 09 09 09 53 6f rse_mode {....So
06c0: 6d 65 28 6d 6f 64 65 29 20 3d 3e 20 6d 6f 64 65 me(mode) => mode
06d0: 2c 0a 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 65 6c ,....None => tel
06e0: 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 egram_bot::types
06f0: 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d ::ParseMode::Htm
0700: 6c 2c 0a 09 09 7d 3b 0a 09 09 73 65 6c 66 2e 74 l,...};...self.t
0710: 67 2e 73 70 61 77 6e 28 74 65 6c 65 67 72 61 6d g.spawn(telegram
0720: 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 73 61 67 _bot::SendMessag
0730: 65 3a 3a 6e 65 77 28 6d 61 74 63 68 20 74 61 72 e::new(match tar
0740: 67 65 74 20 7b 0a 09 09 09 53 6f 6d 65 28 75 73 get {....Some(us
0750: 65 72 29 20 3d 3e 20 75 73 65 72 2c 0a 09 09 09 er) => user,....
0760: 4e 6f 6e 65 20 3d 3e 20 73 65 6c 66 2e 6f 77 6e None => self.own
0770: 65 72 5f 63 68 61 74 2c 0a 09 09 7d 2c 20 6d 73 er_chat,...}, ms
0780: 67 29 2e 70 61 72 73 65 5f 6d 6f 64 65 28 70 61 g).parse_mode(pa
0790: 72 73 65 5f 6d 6f 64 65 29 29 3b 0a 09 09 4f 6b rse_mode));...Ok
07a0: 28 28 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 (())..}...pub as
07b0: 79 6e 63 20 66 6e 20 63 68 65 63 6b 3c 53 3e 28 ync fn check<S>(
07c0: 26 73 65 6c 66 2c 20 69 64 3a 20 26 69 33 32 2c &self, id: &i32,
07d0: 20 6f 77 6e 65 72 3a 20 53 2c 20 72 65 61 6c 3a owner: S, real:
07e0: 20 62 6f 6f 6c 29 20 2d 3e 20 52 65 73 75 6c 74 bool) -> Result
07f0: 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 <Cow<'_, str>>..
0800: 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 where S: Into<i6
0810: 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 4> {...let owner
0820: 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b = owner.into();
0830: 0a 0a 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 ....let mut post
0840: 65 64 3a 20 69 33 32 20 3d 20 30 3b 0a 09 09 6c ed: i32 = 0;...l
0850: 65 74 20 69 64 20 3d 20 7b 0a 09 09 09 6c 65 74 et id = {....let
0860: 20 6d 75 74 20 73 65 74 20 3d 20 73 65 6c 66 2e mut set = self.
0870: 73 6f 75 72 63 65 73 2e 6c 6f 63 6b 28 29 2e 75 sources.lock().u
0880: 6e 77 72 61 70 28 29 3b 0a 09 09 09 6d 61 74 63 nwrap();....matc
0890: 68 20 73 65 74 2e 67 65 74 28 69 64 29 20 7b 0a h set.get(id) {.
08a0: 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 ....Some(id) =>
08b0: 69 64 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 09 id.clone(),.....
08c0: 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 6c None => {......l
08d0: 65 74 20 69 64 20 3d 20 41 72 63 3a 3a 6e 65 77 et id = Arc::new
08e0: 28 2a 69 64 29 3b 0a 09 09 09 09 09 73 65 74 2e (*id);......set.
08f0: 69 6e 73 65 72 74 28 69 64 2e 63 6c 6f 6e 65 28 insert(id.clone(
0900: 29 29 3b 0a 09 09 09 09 09 69 64 2e 63 6c 6f 6e ));......id.clon
0910: 65 28 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a e().....},....}.
0920: 09 09 7d 3b 0a 09 09 6c 65 74 20 63 6f 75 6e 74 ..};...let count
0930: 20 3d 20 41 72 63 3a 3a 73 74 72 6f 6e 67 5f 63 = Arc::strong_c
0940: 6f 75 6e 74 28 26 69 64 29 3b 0a 09 09 69 66 20 ount(&id);...if
0950: 63 6f 75 6e 74 20 3d 3d 20 32 20 7b 0a 09 09 09 count == 2 {....
0960: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
0970: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
0980: 28 29 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 ().await......wi
0990: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
09a0: 72 6d 61 74 21 28 22 51 75 65 72 79 20 71 75 65 rmat!("Query que
09b0: 75 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e ue fetch conn:\n
09c0: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
09d0: 6c 29 29 3f 3b 0a 09 09 09 6c 65 74 20 72 6f 77 l))?;....let row
09e0: 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 = sqlx::query("
09f0: 73 65 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 64 select source_id
0a00: 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 , channel_id, ur
0a10: 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 l, iv_hash, owne
0a20: 72 2c 20 75 72 6c 5f 72 65 20 66 72 6f 6d 20 72 r, url_re from r
0a30: 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 sstg_source wher
0a40: 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 e source_id = $1
0a50: 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 and owner = $2"
0a60: 29 0a 09 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 )......bind(*id)
0a70: 0a 09 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ......bind(owner
0a80: 29 0a 09 09 09 09 2e 66 65 74 63 68 5f 6f 6e 65 )......fetch_one
0a90: 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 (&mut conn).awai
0aa0: 74 0a 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 t......with_cont
0ab0: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
0ac0: 51 75 65 72 79 20 73 6f 75 72 63 65 3a 5c 6e 7b Query source:\n{
0ad0: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
0ae0: 29 29 3f 3b 0a 09 09 09 64 72 6f 70 28 63 6f 6e ))?;....drop(con
0af0: 6e 29 3b 0a 09 09 09 6c 65 74 20 63 68 61 6e 6e n);....let chann
0b00: 65 6c 5f 69 64 3a 20 69 36 34 20 3d 20 72 6f 77 el_id: i64 = row
0b10: 2e 74 72 79 5f 67 65 74 28 22 63 68 61 6e 6e 65 .try_get("channe
0b20: 6c 5f 69 64 22 29 3f 3b 0a 09 09 09 6c 65 74 20 l_id")?;....let
0b30: 75 72 6c 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e url: &str = row.
0b40: 74 72 79 5f 67 65 74 28 22 75 72 6c 22 29 3f 3b try_get("url")?;
0b50: 0a 09 09 09 6c 65 74 20 69 76 5f 68 61 73 68 3a ....let iv_hash:
0b60: 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 Option<&str> =
0b70: 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 69 76 5f row.try_get("iv_
0b80: 68 61 73 68 22 29 3f 3b 0a 09 09 09 6c 65 74 20 hash")?;....let
0b90: 75 72 6c 5f 72 65 20 3d 20 6d 61 74 63 68 20 72 url_re = match r
0ba0: 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 5f ow.try_get("url_
0bb0: 72 65 22 29 3f 20 7b 0a 09 09 09 09 53 6f 6d 65 re")? {.....Some
0bc0: 28 78 29 20 3d 3e 20 53 6f 6d 65 28 73 65 64 72 (x) => Some(sedr
0bd0: 65 67 65 78 3a 3a 52 65 70 6c 61 63 65 43 6f 6d egex::ReplaceCom
0be0: 6d 61 6e 64 3a 3a 6e 65 77 28 78 29 3f 29 2c 0a mand::new(x)?),.
0bf0: 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 4e 6f 6e 65 ....None => None
0c00: 2c 0a 09 09 09 7d 3b 0a 09 09 09 6c 65 74 20 64 ,....};....let d
0c10: 65 73 74 69 6e 61 74 69 6f 6e 20 3d 20 6d 61 74 estination = mat
0c20: 63 68 20 72 65 61 6c 20 7b 0a 09 09 09 09 74 72 ch real {.....tr
0c30: 75 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 ue => telegram_b
0c40: 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 ot::UserId::new(
0c50: 63 68 61 6e 6e 65 6c 5f 69 64 29 2c 0a 09 09 09 channel_id),....
0c60: 09 66 61 6c 73 65 20 3d 3e 20 74 65 6c 65 67 72 .false => telegr
0c70: 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a am_bot::UserId::
0c80: 6e 65 77 28 72 6f 77 2e 74 72 79 5f 67 65 74 28 new(row.try_get(
0c90: 22 6f 77 6e 65 72 22 29 3f 29 2c 0a 09 09 09 7d "owner")?),....}
0ca0: 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 74 68 69 ;....let mut thi
0cb0: 73 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e 3c s_fetch: Option<
0cc0: 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a DateTime<chrono:
0cd0: 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 3e 20 3d :FixedOffset>> =
0ce0: 20 4e 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 6d 75 None;....let mu
0cf0: 74 20 70 6f 73 74 73 3a 20 42 54 72 65 65 4d 61 t posts: BTreeMa
0d00: 70 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e p<DateTime<chron
0d10: 6f 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 2c o::FixedOffset>,
0d20: 20 53 74 72 69 6e 67 3e 20 3d 20 42 54 72 65 65 String> = BTree
0d30: 4d 61 70 3a 3a 6e 65 77 28 29 3b 0a 09 09 09 6c Map::new();....l
0d40: 65 74 20 72 65 73 70 6f 6e 73 65 20 3d 20 72 65 et response = re
0d50: 71 77 65 73 74 3a 3a 67 65 74 28 75 72 6c 29 2e qwest::get(url).
0d60: 61 77 61 69 74 3f 3b 0a 09 09 09 6c 65 74 20 73 await?;....let s
0d70: 74 61 74 75 73 20 3d 20 72 65 73 70 6f 6e 73 65 tatus = response
0d80: 2e 73 74 61 74 75 73 28 29 3b 0a 09 09 09 6c 65 .status();....le
0d90: 74 20 63 6f 6e 74 65 6e 74 20 3d 20 72 65 73 70 t content = resp
0da0: 6f 6e 73 65 2e 62 79 74 65 73 28 29 2e 61 77 61 onse.bytes().awa
0db0: 69 74 3f 3b 0a 09 09 09 6d 61 74 63 68 20 72 73 it?;....match rs
0dc0: 73 3a 3a 43 68 61 6e 6e 65 6c 3a 3a 72 65 61 64 s::Channel::read
0dd0: 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e _from(&content[.
0de0: 2e 5d 29 20 7b 0a 09 09 09 09 4f 6b 28 66 65 65 .]) {.....Ok(fee
0df0: 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 66 6f 72 d) => {......for
0e00: 20 69 74 65 6d 20 69 6e 20 66 65 65 64 2e 69 74 item in feed.it
0e10: 65 6d 73 28 29 20 7b 0a 09 09 09 09 09 09 6d 61 ems() {.......ma
0e20: 74 63 68 20 69 74 65 6d 2e 6c 69 6e 6b 28 29 20 tch item.link()
0e30: 7b 0a 09 09 09 09 09 09 09 53 6f 6d 65 28 6c 69 {........Some(li
0e40: 6e 6b 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 nk) => {........
0e50: 09 6c 65 74 20 64 61 74 65 20 3d 20 6d 61 74 63 .let date = matc
0e60: 68 20 69 74 65 6d 2e 70 75 62 5f 64 61 74 65 28 h item.pub_date(
0e70: 29 20 7b 0a 09 09 09 09 09 09 09 09 09 53 6f 6d ) {..........Som
0e80: 65 28 66 65 65 64 5f 64 61 74 65 29 20 3d 3e 20 e(feed_date) =>
0e90: 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f DateTime::parse_
0ea0: 66 72 6f 6d 5f 72 66 63 32 38 32 32 28 66 65 65 from_rfc2822(fee
0eb0: 64 5f 64 61 74 65 29 2c 0a 09 09 09 09 09 09 09 d_date),........
0ec0: 09 09 4e 6f 6e 65 20 3d 3e 20 44 61 74 65 54 69 ..None => DateTi
0ed0: 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 me::parse_from_r
0ee0: 66 63 33 33 33 39 28 26 69 74 65 6d 2e 64 75 62 fc3339(&item.dub
0ef0: 6c 69 6e 5f 63 6f 72 65 5f 65 78 74 28 29 2e 75 lin_core_ext().u
0f00: 6e 77 72 61 70 28 29 2e 64 61 74 65 73 28 29 5b nwrap().dates()[
0f10: 30 5d 29 2c 0a 09 09 09 09 09 09 09 09 7d 3f 3b 0]),.........}?;
0f20: 0a 09 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c .........let url
0f30: 20 3d 20 6c 69 6e 6b 3b 0a 09 09 09 09 09 09 09 = link;........
0f40: 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 64 61 .posts.insert(da
0f50: 74 65 2e 63 6c 6f 6e 65 28 29 2c 20 75 72 6c 2e te.clone(), url.
0f60: 69 6e 74 6f 28 29 29 3b 0a 09 09 09 09 09 09 09 into());........
0f70: 7d 2c 0a 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d },........None =
0f80: 3e 20 7b 7d 0a 09 09 09 09 09 09 7d 0a 09 09 09 > {}.......}....
0f90: 09 09 7d 3b 0a 09 09 09 09 7d 2c 0a 09 09 09 09 ..};.....},.....
0fa0: 45 72 72 28 65 72 72 29 20 3d 3e 20 6d 61 74 63 Err(err) => matc
0fb0: 68 20 65 72 72 20 7b 0a 09 09 09 09 09 72 73 73 h err {......rss
0fc0: 3a 3a 45 72 72 6f 72 3a 3a 49 6e 76 61 6c 69 64 ::Error::Invalid
0fd0: 53 74 61 72 74 54 61 67 20 3d 3e 20 7b 0a 09 09 StartTag => {...
0fe0: 09 09 09 09 6c 65 74 20 66 65 65 64 20 3d 20 61 ....let feed = a
0ff0: 74 6f 6d 5f 73 79 6e 64 69 63 61 74 69 6f 6e 3a tom_syndication:
1000: 3a 46 65 65 64 3a 3a 72 65 61 64 5f 66 72 6f 6d :Feed::read_from
1010: 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 0a 09 (&content[..])..
1020: 09 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 .......with_cont
1030: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1040: 50 72 6f 62 6c 65 6d 20 6f 70 65 6e 69 6e 67 20 Problem opening
1050: 66 65 65 64 20 75 72 6c 3a 5c 6e 7b 7d 5c 6e 7b feed url:\n{}\n{
1060: 7d 22 2c 20 26 75 72 6c 2c 20 73 74 61 74 75 73 }", &url, status
1070: 29 29 3f 3b 0a 09 09 09 09 09 09 66 6f 72 20 69 ))?;.......for i
1080: 74 65 6d 20 69 6e 20 66 65 65 64 2e 65 6e 74 72 tem in feed.entr
1090: 69 65 73 28 29 20 7b 0a 09 09 09 09 09 09 09 6c ies() {........l
10a0: 65 74 20 64 61 74 65 20 3d 20 69 74 65 6d 2e 70 et date = item.p
10b0: 75 62 6c 69 73 68 65 64 28 29 2e 75 6e 77 72 61 ublished().unwra
10c0: 70 28 29 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 p();........let
10d0: 75 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 url = item.links
10e0: 28 29 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 ()[0].href();...
10f0: 09 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 .....posts.inser
1100: 74 28 64 61 74 65 2e 63 6c 6f 6e 65 28 29 2c 20 t(date.clone(),
1110: 75 72 6c 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 url.into());....
1120: 09 09 09 7d 3b 0a 09 09 09 09 09 7d 2c 0a 09 09 ...};......},...
1130: 09 09 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 ...rss::Error::E
1140: 6f 66 20 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f of => (),......_
1150: 20 3d 3e 20 62 61 69 6c 21 28 22 55 6e 73 75 70 => bail!("Unsup
1160: 70 6f 72 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 ported or mangle
1170: 64 20 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d d content:\n{:?}
1180: 5c 6e 7b 3a 23 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e \n{:#?}\n{:#?}\n
1190: 22 2c 20 26 75 72 6c 2c 20 65 72 72 2c 20 73 74 ", &url, err, st
11a0: 61 74 75 73 29 0a 09 09 09 09 7d 0a 09 09 09 7d atus).....}....}
11b0: 3b 0a 09 09 09 66 6f 72 20 28 64 61 74 65 2c 20 ;....for (date,
11c0: 75 72 6c 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 url) in posts.it
11d0: 65 72 28 29 20 7b 0a 09 09 09 09 6c 65 74 20 6d er() {.....let m
11e0: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 ut conn = self.p
11f0: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
1200: 61 69 74 0a 09 09 09 09 09 2e 77 69 74 68 5f 63 ait.......with_c
1210: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1220: 21 28 22 43 68 65 63 6b 20 70 6f 73 74 20 66 65 !("Check post fe
1230: 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 tch conn:\n{:?}"
1240: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b , &self.pool))?;
1250: 0a 09 09 09 09 6c 65 74 20 70 6f 73 74 5f 75 72 .....let post_ur
1260: 6c 3a 20 43 6f 77 3c 73 74 72 3e 20 3d 20 6d 61 l: Cow<str> = ma
1270: 74 63 68 20 75 72 6c 5f 72 65 20 7b 0a 09 09 09 tch url_re {....
1280: 09 09 53 6f 6d 65 28 72 65 66 20 78 29 20 3d 3e ..Some(ref x) =>
1290: 20 78 2e 65 78 65 63 75 74 65 28 75 72 6c 29 2e x.execute(url).
12a0: 69 6e 74 6f 28 29 2c 0a 09 09 09 09 09 4e 6f 6e into(),......Non
12b0: 65 20 3d 3e 20 75 72 6c 2e 69 6e 74 6f 28 29 2c e => url.into(),
12c0: 0a 09 09 09 09 7d 3b 0a 09 09 09 09 6c 65 74 20 .....};.....let
12d0: 72 6f 77 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 row = sqlx::quer
12e0: 79 28 22 73 65 6c 65 63 74 20 65 78 69 73 74 73 y("select exists
12f0: 28 73 65 6c 65 63 74 20 74 72 75 65 20 66 72 6f (select true fro
1300: 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 77 68 65 m rsstg_post whe
1310: 72 65 20 75 72 6c 20 3d 20 24 31 20 61 6e 64 20 re url = $1 and
1320: 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 32 29 20 source_id = $2)
1330: 61 73 20 65 78 69 73 74 73 3b 22 29 0a 09 09 09 as exists;")....
1340: 09 09 2e 62 69 6e 64 28 26 2a 70 6f 73 74 5f 75 ...bind(&*post_u
1350: 72 6c 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 2a rl).......bind(*
1360: 69 64 29 0a 09 09 09 09 09 2e 66 65 74 63 68 5f id).......fetch_
1370: 6f 6e 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 one(&mut conn).a
1380: 77 61 69 74 0a 09 09 09 09 09 2e 77 69 74 68 5f wait.......with_
1390: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
13a0: 74 21 28 22 43 68 65 63 6b 20 70 6f 73 74 3a 5c t!("Check post:\
13b0: 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f n{:?}", &conn))?
13c0: 3b 0a 09 09 09 09 6c 65 74 20 65 78 69 73 74 73 ;.....let exists
13d0: 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 : bool = row.try
13e0: 5f 67 65 74 28 22 65 78 69 73 74 73 22 29 3f 3b _get("exists")?;
13f0: 0a 09 09 09 09 69 66 20 21 20 65 78 69 73 74 73 .....if ! exists
1400: 20 7b 0a 09 09 09 09 09 69 66 20 74 68 69 73 5f {......if this_
1410: 66 65 74 63 68 20 3d 3d 20 4e 6f 6e 65 20 7c 7c fetch == None ||
1420: 20 2a 64 61 74 65 20 3e 20 74 68 69 73 5f 66 65 *date > this_fe
1430: 74 63 68 2e 75 6e 77 72 61 70 28 29 20 7b 0a 09 tch.unwrap() {..
1440: 09 09 09 09 09 74 68 69 73 5f 66 65 74 63 68 20 .....this_fetch
1450: 3d 20 53 6f 6d 65 28 2a 64 61 74 65 29 3b 0a 09 = Some(*date);..
1460: 09 09 09 09 7d 3b 0a 09 09 09 09 09 73 65 6c 66 ....};......self
1470: 2e 74 67 2e 73 65 6e 64 28 20 6d 61 74 63 68 20 .tg.send( match
1480: 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 iv_hash {.......
1490: 09 53 6f 6d 65 28 68 61 73 68 29 20 3d 3e 20 74 .Some(hash) => t
14a0: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e elegram_bot::Sen
14b0: 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 dMessage::new(de
14c0: 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 stination, forma
14d0: 74 21 28 22 3c 61 20 68 72 65 66 3d 5c 22 68 74 t!("<a href=\"ht
14e0: 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 72 tps://t.me/iv?ur
14f0: 6c 3d 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c 22 3e l={}&rhash={}\">
1500: 20 3c 2f 61 3e 7b 30 7d 22 2c 20 26 70 6f 73 74 </a>{0}", &post
1510: 5f 75 72 6c 2c 20 68 61 73 68 29 29 2c 0a 09 09 _url, hash)),...
1520: 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 65 6c .....None => tel
1530: 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d egram_bot::SendM
1540: 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 73 74 essage::new(dest
1550: 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 21 ination, format!
1560: 28 22 7b 7d 22 2c 20 70 6f 73 74 5f 75 72 6c 29 ("{}", post_url)
1570: 29 2c 0a 09 09 09 09 09 09 7d 2e 70 61 72 73 65 ),.......}.parse
1580: 5f 6d 6f 64 65 28 74 65 6c 65 67 72 61 6d 5f 62 _mode(telegram_b
1590: 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 61 72 73 65 ot::types::Parse
15a0: 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 2e 61 77 61 Mode::Html)).awa
15b0: 69 74 0a 09 09 09 09 09 09 2e 63 6f 6e 74 65 78 it........contex
15c0: 74 28 22 43 61 6e 27 74 20 70 6f 73 74 20 6d 65 t("Can't post me
15d0: 73 73 61 67 65 3a 22 29 3f 3b 0a 09 09 09 09 09 ssage:")?;......
15e0: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 69 6e 73 sqlx::query("ins
15f0: 65 72 74 20 69 6e 74 6f 20 72 73 73 74 67 5f 70 ert into rsstg_p
1600: 6f 73 74 20 28 73 6f 75 72 63 65 5f 69 64 2c 20 ost (source_id,
1610: 70 6f 73 74 65 64 2c 20 75 72 6c 29 20 76 61 6c posted, url) val
1620: 75 65 73 20 28 24 31 2c 20 24 32 2c 20 24 33 29 ues ($1, $2, $3)
1630: 3b 22 29 0a 09 09 09 09 09 09 2e 62 69 6e 64 28 ;")........bind(
1640: 2a 69 64 29 0a 09 09 09 09 09 09 2e 62 69 6e 64 *id)........bind
1650: 28 64 61 74 65 29 0a 09 09 09 09 09 09 2e 62 69 (date)........bi
1660: 6e 64 28 26 2a 70 6f 73 74 5f 75 72 6c 29 0a 09 nd(&*post_url)..
1670: 09 09 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d ......execute(&m
1680: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 ut conn).await..
1690: 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 ......with_conte
16a0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 52 xt(|| format!("R
16b0: 65 63 6f 72 64 20 70 6f 73 74 3a 5c 6e 7b 3a 3f ecord post:\n{:?
16c0: 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 }", &conn))?;...
16d0: 09 09 09 64 72 6f 70 28 63 6f 6e 6e 29 3b 0a 09 ...drop(conn);..
16e0: 09 09 09 09 74 6f 6b 69 6f 3a 3a 74 69 6d 65 3a ....tokio::time:
16f0: 3a 73 6c 65 65 70 28 73 74 64 3a 3a 74 69 6d 65 :sleep(std::time
1700: 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 ::Duration::new(
1710: 34 2c 20 30 29 29 2e 61 77 61 69 74 3b 0a 09 09 4, 0)).await;...
1720: 09 09 7d 3b 0a 09 09 09 09 70 6f 73 74 65 64 20 ..};.....posted
1730: 2b 3d 20 31 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 += 1;....};....p
1740: 6f 73 74 73 2e 63 6c 65 61 72 28 29 3b 0a 09 09 osts.clear();...
1750: 7d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e };...let mut con
1760: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
1770: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
1780: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
1790: 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 74 | format!("Updat
17a0: 65 20 73 63 72 61 70 65 20 66 65 74 63 68 20 63 e scrape fetch c
17b0: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
17c0: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 73 71 lf.pool))?;...sq
17d0: 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 lx::query("updat
17e0: 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 e rsstg_source s
17f0: 65 74 20 6c 61 73 74 5f 73 63 72 61 70 65 20 3d et last_scrape =
1800: 20 6e 6f 77 28 29 20 77 68 65 72 65 20 73 6f 75 now() where sou
1810: 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a 09 rce_id = $1;")..
1820: 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 ...bind(*id)....
1830: 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f .execute(&mut co
1840: 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 nn).await.....wi
1850: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
1860: 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 73 63 rmat!("Update sc
1870: 72 61 70 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 rape:\n{:?}", &c
1880: 6f 6e 6e 29 29 3f 3b 0a 09 09 4f 6b 28 66 6f 72 onn))?;...Ok(for
1890: 6d 61 74 21 28 22 50 6f 73 74 65 64 3a 20 7b 7d mat!("Posted: {}
18a0: 22 2c 20 26 70 6f 73 74 65 64 29 2e 69 6e 74 6f ", &posted).into
18b0: 28 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ())..}...pub asy
18c0: 6e 63 20 66 6e 20 64 65 6c 65 74 65 3c 53 3e 28 nc fn delete<S>(
18d0: 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 &self, source_id
18e0: 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 : &i32, owner: S
18f0: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 3c ) -> Result<Cow<
1900: 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 65 '_, str>>..where
1910: 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a S: Into<i64> {.
1920: 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 ..let owner = ow
1930: 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c ner.into();....l
1940: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
1950: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
1960: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
1970: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
1980: 61 74 21 28 22 44 65 6c 65 74 65 20 66 65 74 63 at!("Delete fetc
1990: 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 h conn:\n{:?}",
19a0: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 &self.pool))?;..
19b0: 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 .match sqlx::que
19c0: 72 79 28 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 ry("delete from
19d0: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 rsstg_source whe
19e0: 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 re source_id = $
19f0: 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 1 and owner = $2
1a00: 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 ;").....bind(sou
1a10: 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 rce_id).....bind
1a20: 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 (owner).....exec
1a30: 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 ute(&mut conn).a
1a40: 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f wait.....with_co
1a50: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
1a60: 28 22 44 65 6c 65 74 65 20 73 6f 75 72 63 65 20 ("Delete source
1a70: 72 75 6c 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 rule:\n{:?}", &s
1a80: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e elf.pool))?.....
1a90: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 rows_affected()
1aa0: 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 {....0 => { Ok("
1ab0: 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f No data found fo
1ac0: 75 6e 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c und.".into()) },
1ad0: 0a 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f ....x => { Ok(fo
1ae0: 72 6d 61 74 21 28 22 7b 7d 20 73 6f 75 72 63 65 rmat!("{} source
1af0: 73 20 72 65 6d 6f 76 65 64 2e 22 2c 20 78 29 2e s removed.", x).
1b00: 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 into()) },...}..
1b10: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
1b20: 20 63 6c 65 61 6e 3c 53 3e 28 26 73 65 6c 66 2c clean<S>(&self,
1b30: 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 source_id: &i32
1b40: 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 , owner: S) -> R
1b50: 65 73 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 esult<Cow<'_, st
1b60: 72 3e 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e r>>..where S: In
1b70: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 to<i64> {...let
1b80: 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e owner = owner.in
1b90: 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 to();....let mut
1ba0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
1bb0: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
1bc0: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
1bd0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 xt(|| format!("C
1be0: 6c 65 61 6e 20 66 65 74 63 68 20 63 6f 6e 6e 3a lean fetch conn:
1bf0: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
1c00: 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 ool))?;...match
1c10: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 6c sqlx::query("del
1c20: 65 74 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 ete from rsstg_p
1c30: 6f 73 74 20 70 20 75 73 69 6e 67 20 72 73 73 74 ost p using rsst
1c40: 67 5f 73 6f 75 72 63 65 20 73 20 77 68 65 72 65 g_source s where
1c50: 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 p.source_id = $
1c60: 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 1 and owner = $2
1c70: 20 61 6e 64 20 70 2e 73 6f 75 72 63 65 5f 69 64 and p.source_id
1c80: 20 3d 20 73 2e 73 6f 75 72 63 65 5f 69 64 3b 22 = s.source_id;"
1c90: 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 ).....bind(sourc
1ca0: 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f e_id).....bind(o
1cb0: 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 wner).....execut
1cc0: 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 e(&mut conn).awa
1cd0: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
1ce0: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1cf0: 43 6c 65 61 6e 20 73 65 65 6e 20 70 6f 73 74 73 Clean seen posts
1d00: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1d10: 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 pool))?.....rows
1d20: 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 _affected() {...
1d30: 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 .0 => { Ok("No d
1d40: 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 2e ata found found.
1d50: 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 09 ".into()) },....
1d60: 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 x => { Ok(format
1d70: 21 28 22 7b 7d 20 70 6f 73 74 73 20 70 75 72 67 !("{} posts purg
1d80: 65 64 2e 22 2c 20 78 29 2e 69 6e 74 6f 28 29 29 ed.", x).into())
1d90: 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 },...}..}...pub
1da0: 20 61 73 79 6e 63 20 66 6e 20 65 6e 61 62 6c 65 async fn enable
1db0: 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 <S>(&self, sourc
1dc0: 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 e_id: &i32, owne
1dd0: 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c r: S) -> Result<
1de0: 26 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 &str>..where S:
1df0: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
1e00: 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e t owner = owner.
1e10: 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d into();....let m
1e20: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 ut conn = self.p
1e30: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
1e40: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
1e50: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1e60: 22 45 6e 61 62 6c 65 20 66 65 74 63 68 20 63 6f "Enable fetch co
1e70: 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c nn:\n{:?}", &sel
1e80: 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 f.pool))?;...mat
1e90: 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ch sqlx::query("
1ea0: 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 update rsstg_sou
1eb0: 72 63 65 20 73 65 74 20 65 6e 61 62 6c 65 64 20 rce set enabled
1ec0: 3d 20 74 72 75 65 20 77 68 65 72 65 20 73 6f 75 = true where sou
1ed0: 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 rce_id = $1 and
1ee0: 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 09 owner = $2")....
1ef0: 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 .bind(source_id)
1f00: 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 .....bind(owner)
1f10: 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 .....execute(&mu
1f20: 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 t conn).await...
1f30: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
1f40: 7c 20 66 6f 72 6d 61 74 21 28 22 45 6e 61 62 6c | format!("Enabl
1f50: 65 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 e source:\n{:?}"
1f60: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a , &self.pool))?.
1f70: 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 ....rows_affecte
1f80: 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 d() {....1 => {
1f90: 4f 6b 28 22 53 6f 75 72 63 65 20 65 6e 61 62 6c Ok("Source enabl
1fa0: 65 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e ed.") },....0 =>
1fb0: 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f { Ok("Source no
1fc0: 74 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 t found.") },...
1fd0: 09 5f 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 68 ._ => { Err(anyh
1fe0: 6f 77 21 28 22 44 61 74 61 62 61 73 65 20 65 72 ow!("Database er
1ff0: 72 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 ror.")) },...}..
2000: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
2010: 20 64 69 73 61 62 6c 65 3c 53 3e 28 26 73 65 6c disable<S>(&sel
2020: 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 f, source_id: &i
2030: 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 32, owner: S) ->
2040: 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 77 Result<&str>..w
2050: 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 here S: Into<i64
2060: 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 > {...let owner
2070: 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a = owner.into();.
2080: 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
2090: 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 = self.pool.acqu
20a0: 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e ire().await.....
20b0: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
20c0: 66 6f 72 6d 61 74 21 28 22 44 69 73 61 62 6c 65 format!("Disable
20d0: 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a fetch conn:\n{:
20e0: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
20f0: 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 )?;...match sqlx
2100: 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 20 ::query("update
2110: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 rsstg_source set
2120: 20 65 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 65 enabled = false
2130: 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 where source_id
2140: 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 = $1 and owner
2150: 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 28 = $2").....bind(
2160: 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 source_id).....b
2170: 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 ind(owner).....e
2180: 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e xecute(&mut conn
2190: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
21a0: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
21b0: 61 74 21 28 22 44 69 73 61 62 6c 65 20 73 6f 75 at!("Disable sou
21c0: 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 rce:\n{:?}", &se
21d0: 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 lf.pool))?.....r
21e0: 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b ows_affected() {
21f0: 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 ....1 => { Ok("S
2200: 6f 75 72 63 65 20 64 69 73 61 62 6c 65 64 2e 22 ource disabled."
2210: 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f ) },....0 => { O
2220: 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f k("Source not fo
2230: 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d und.") },...._ =
2240: 3e 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 > { Err(anyhow!(
2250: 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e "Database error.
2260: 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 ")) },...}..}...
2270: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 75 70 64 pub async fn upd
2280: 61 74 65 3c 53 3e 28 26 73 65 6c 66 2c 20 75 70 ate<S>(&self, up
2290: 64 61 74 65 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 date: Option<i32
22a0: 3e 2c 20 63 68 61 6e 6e 65 6c 3a 20 26 73 74 72 >, channel: &str
22b0: 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 , channel_id: i6
22c0: 34 2c 20 75 72 6c 3a 20 26 73 74 72 2c 20 69 76 4, url: &str, iv
22d0: 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 _hash: Option<&s
22e0: 74 72 3e 2c 20 75 72 6c 5f 72 65 3a 20 4f 70 74 tr>, url_re: Opt
22f0: 69 6f 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 72 ion<&str>, owner
2300: 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 : S) -> Result<&
2310: 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 str>..where S: I
2320: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
2330: 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 owner = owner.i
2340: 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 nto();....let mu
2350: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
2360: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
2370: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
2380: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
2390: 55 70 64 61 74 65 20 66 65 74 63 68 20 63 6f 6e Update fetch con
23a0: 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 n:\n{:?}", &self
23b0: 2e 70 6f 6f 6c 29 29 3f 3b 0a 0a 09 09 6d 61 74 .pool))?;....mat
23c0: 63 68 20 6d 61 74 63 68 20 75 70 64 61 74 65 20 ch match update
23d0: 7b 0a 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d {.....Some(id) =
23e0: 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 > {......sqlx::q
23f0: 75 65 72 79 28 22 75 70 64 61 74 65 20 72 73 73 uery("update rss
2400: 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 63 68 tg_source set ch
2410: 61 6e 6e 65 6c 5f 69 64 20 3d 20 24 32 2c 20 75 annel_id = $2, u
2420: 72 6c 20 3d 20 24 33 2c 20 69 76 5f 68 61 73 68 rl = $3, iv_hash
2430: 20 3d 20 24 34 2c 20 6f 77 6e 65 72 20 3d 20 24 = $4, owner = $
2440: 35 2c 20 63 68 61 6e 6e 65 6c 20 3d 20 24 36 2c 5, channel = $6,
2450: 20 75 72 6c 5f 72 65 20 3d 20 24 37 20 77 68 65 url_re = $7 whe
2460: 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 re source_id = $
2470: 31 22 29 2e 62 69 6e 64 28 69 64 29 0a 09 09 09 1").bind(id)....
2480: 09 7d 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 .},.....None =>
2490: 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 {......sqlx::que
24a0: 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 ry("insert into
24b0: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 28 63 68 rsstg_source (ch
24c0: 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 annel_id, url, i
24d0: 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 63 v_hash, owner, c
24e0: 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 20 hannel, url_re)
24f0: 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 values ($1, $2,
2500: 24 33 2c 20 24 34 2c 20 24 35 2c 20 24 36 29 22 $3, $4, $5, $6)"
2510: 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 ).....},....}...
2520: 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c 5f 69 ..bind(channel_i
2530: 64 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c 29 d).....bind(url)
2540: 0a 09 09 09 2e 62 69 6e 64 28 69 76 5f 68 61 73 .....bind(iv_has
2550: 68 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 h).....bind(owne
2560: 72 29 0a 09 09 09 2e 62 69 6e 64 28 63 68 61 6e r).....bind(chan
2570: 6e 65 6c 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 nel).....bind(ur
2580: 6c 5f 72 65 29 0a 09 09 09 2e 65 78 65 63 75 74 l_re).....execut
2590: 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 e(&mut conn).awa
25a0: 69 74 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e it {....Ok(_) =>
25b0: 20 72 65 74 75 72 6e 20 4f 6b 28 6d 61 74 63 68 return Ok(match
25c0: 20 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f update {.....So
25d0: 6d 65 28 5f 29 20 3d 3e 20 22 43 68 61 6e 6e 65 me(_) => "Channe
25e0: 6c 20 75 70 64 61 74 65 64 2e 22 2c 0a 09 09 09 l updated.",....
25f0: 09 4e 6f 6e 65 20 3d 3e 20 22 43 68 61 6e 6e 65 .None => "Channe
2600: 6c 20 61 64 64 65 64 2e 22 2c 0a 09 09 09 7d 29 l added.",....})
2610: 2c 0a 09 09 09 45 72 72 28 73 71 6c 78 3a 3a 45 ,....Err(sqlx::E
2620: 72 72 6f 72 3a 3a 44 61 74 61 62 61 73 65 28 65 rror::Database(e
2630: 72 72 29 29 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 rr)) => {.....ma
2640: 74 63 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 tch err.downcast
2650: 3a 3a 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 ::<sqlx::postgre
2660: 73 3a 3a 50 67 44 61 74 61 62 61 73 65 45 72 72 s::PgDatabaseErr
2670: 6f 72 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 or>().routine()
2680: 7b 0a 09 09 09 09 09 53 6f 6d 65 28 22 5f 62 74 {......Some("_bt
2690: 5f 63 68 65 63 6b 5f 75 6e 69 71 75 65 22 2c 20 _check_unique",
26a0: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 72 65 74 ) => {.......ret
26b0: 75 72 6e 20 4f 6b 28 22 44 75 70 6c 69 63 61 74 urn Ok("Duplicat
26c0: 65 20 6b 65 79 2e 22 29 0a 09 09 09 09 09 7d 2c e key.")......},
26d0: 0a 09 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e ......Some(_) =>
26e0: 20 7b 0a 09 09 09 09 09 09 72 65 74 75 72 6e 20 {.......return
26f0: 4f 6b 28 22 44 61 74 61 62 61 73 65 20 65 72 72 Ok("Database err
2700: 6f 72 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 or.")......},...
2710: 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 ...None => {....
2720: 09 09 09 72 65 74 75 72 6e 20 4f 6b 28 22 4e 6f ...return Ok("No
2730: 20 64 61 74 61 62 61 73 65 20 65 72 72 6f 72 20 database error
2740: 65 78 74 72 61 63 74 65 64 2e 22 29 0a 09 09 09 extracted.")....
2750: 09 09 7d 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 7d ..},.....};....}
2760: 2c 0a 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e ,....Err(err) =>
2770: 20 7b 0a 09 09 09 09 62 61 69 6c 21 28 22 53 6f {.....bail!("So
2780: 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 72 72 rry, unknown err
2790: 6f 72 3a 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 65 or:\n{:#?}\n", e
27a0: 72 72 29 3b 0a 09 09 09 7d 2c 0a 09 09 7d 3b 0a rr);....},...};.
27b0: 09 7d 0a 0a 09 61 73 79 6e 63 20 66 6e 20 61 75 .}...async fn au
27c0: 74 6f 66 65 74 63 68 28 26 73 65 6c 66 29 20 2d tofetch(&self) -
27d0: 3e 20 52 65 73 75 6c 74 3c 28 29 3e 20 7b 0a 09 > Result<()> {..
27e0: 09 6c 65 74 20 6d 75 74 20 64 65 6c 61 79 20 3d .let mut delay =
27f0: 20 63 68 72 6f 6e 6f 3a 3a 44 75 72 61 74 69 6f chrono::Duratio
2800: 6e 3a 3a 6d 69 6e 75 74 65 73 28 31 29 3b 0a 09 n::minutes(1);..
2810: 09 6c 65 74 20 6d 75 74 20 6e 6f 77 3b 0a 09 09 .let mut now;...
2820: 6c 6f 6f 70 20 7b 0a 09 09 09 6c 65 74 20 6d 75 loop {....let mu
2830: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
2840: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
2850: 69 74 0a 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e it......with_con
2860: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
2870: 22 41 75 74 6f 66 65 74 63 68 20 66 65 74 63 68 "Autofetch fetch
2880: 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 conn:\n{:?}", &
2890: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 self.pool))?;...
28a0: 09 6e 6f 77 20 3d 20 63 68 72 6f 6e 6f 3a 3a 4c .now = chrono::L
28b0: 6f 63 61 6c 3a 3a 6e 6f 77 28 29 3b 0a 09 09 09 ocal::now();....
28c0: 6c 65 74 20 6d 75 74 20 71 75 65 75 65 20 3d 20 let mut queue =
28d0: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c sqlx::query("sel
28e0: 65 63 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 6e ect source_id, n
28f0: 65 78 74 5f 66 65 74 63 68 2c 20 6f 77 6e 65 72 ext_fetch, owner
2900: 20 66 72 6f 6d 20 72 73 73 74 67 5f 6f 72 64 65 from rsstg_orde
2910: 72 20 6e 61 74 75 72 61 6c 20 6c 65 66 74 20 6a r natural left j
2920: 6f 69 6e 20 72 73 73 74 67 5f 73 6f 75 72 63 65 oin rsstg_source
2930: 20 77 68 65 72 65 20 6e 65 78 74 5f 66 65 74 63 where next_fetc
2940: 68 20 3c 20 6e 6f 77 28 29 20 2b 20 69 6e 74 65 h < now() + inte
2950: 72 76 61 6c 20 27 31 20 6d 69 6e 75 74 65 27 3b rval '1 minute';
2960: 22 29 0a 09 09 09 09 2e 66 65 74 63 68 5f 61 6c ")......fetch_al
2970: 6c 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 l(&mut conn).awa
2980: 69 74 3f 3b 0a 09 09 09 66 6f 72 20 72 6f 77 20 it?;....for row
2990: 69 6e 20 71 75 65 75 65 2e 69 74 65 72 28 29 20 in queue.iter()
29a0: 7b 0a 09 09 09 09 6c 65 74 20 73 6f 75 72 63 65 {.....let source
29b0: 5f 69 64 3a 20 69 33 32 20 3d 20 72 6f 77 2e 74 _id: i32 = row.t
29c0: 72 79 5f 67 65 74 28 22 73 6f 75 72 63 65 5f 69 ry_get("source_i
29d0: 64 22 29 3f 3b 0a 09 09 09 09 6c 65 74 20 6f 77 d")?;.....let ow
29e0: 6e 65 72 3a 20 69 36 34 20 3d 20 72 6f 77 2e 74 ner: i64 = row.t
29f0: 72 79 5f 67 65 74 28 22 6f 77 6e 65 72 22 29 3f ry_get("owner")?
2a00: 3b 0a 09 09 09 09 6c 65 74 20 6e 65 78 74 5f 66 ;.....let next_f
2a10: 65 74 63 68 3a 20 44 61 74 65 54 69 6d 65 3c 63 etch: DateTime<c
2a20: 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3e 20 3d 20 hrono::Local> =
2a30: 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 6e 65 78 row.try_get("nex
2a40: 74 5f 66 65 74 63 68 22 29 3f 3b 0a 09 09 09 09 t_fetch")?;.....
2a50: 69 66 20 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 if next_fetch <
2a60: 6e 6f 77 20 7b 0a 09 09 09 09 09 6c 65 74 20 63 now {......let c
2a70: 6c 6f 6e 65 20 3d 20 43 6f 72 65 20 7b 0a 09 09 lone = Core {...
2a80: 09 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 ....owner_chat:
2a90: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 telegram_bot::Us
2aa0: 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 erId::new(owner)
2ab0: 2c 0a 09 09 09 09 09 09 2e 2e 73 65 6c 66 2e 63 ,.........self.c
2ac0: 6c 6f 6e 65 28 29 0a 09 09 09 09 09 7d 3b 0a 09 lone()......};..
2ad0: 09 09 09 09 74 6f 6b 69 6f 3a 3a 73 70 61 77 6e ....tokio::spawn
2ae0: 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 (async move {...
2af0: 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 ....if let Err(e
2b00: 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 rr) = clone.chec
2b10: 6b 28 26 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 k(&source_id, ow
2b20: 6e 65 72 2c 20 74 72 75 65 29 2e 61 77 61 69 74 ner, true).await
2b30: 20 7b 0a 09 09 09 09 09 09 09 69 66 20 6c 65 74 {........if let
2b40: 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e Err(err) = clon
2b50: 65 2e 73 65 6e 64 28 26 66 6f 72 6d 61 74 21 28 e.send(&format!(
2b60: 22 f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 "š {:?}", err
2b70: 29 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 20 7b ), None, None) {
2b80: 0a 09 09 09 09 09 09 09 09 65 70 72 69 6e 74 6c .........eprintl
2b90: 6e 21 28 22 43 68 65 63 6b 20 65 72 72 6f 72 3a n!("Check error:
2ba0: 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 09 {}", err);.....
2bb0: 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 ...};.......};..
2bc0: 09 09 09 09 7d 29 3b 0a 09 09 09 09 7d 20 65 6c ....});.....} el
2bd0: 73 65 20 7b 0a 09 09 09 09 09 69 66 20 6e 65 78 se {......if nex
2be0: 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 20 3c 20 t_fetch - now <
2bf0: 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 09 64 65 delay {.......de
2c00: 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 63 68 lay = next_fetch
2c10: 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 09 7d 0a 09 - now;......}..
2c20: 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 71 75 ...}....};....qu
2c30: 65 75 65 2e 63 6c 65 61 72 28 29 3b 0a 09 09 09 eue.clear();....
2c40: 74 6f 6b 69 6f 3a 3a 74 69 6d 65 3a 3a 73 6c 65 tokio::time::sle
2c50: 65 70 28 64 65 6c 61 79 2e 74 6f 5f 73 74 64 28 ep(delay.to_std(
2c60: 29 3f 29 2e 61 77 61 69 74 3b 0a 09 09 09 64 65 )?).await;....de
2c70: 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 lay = chrono::Du
2c80: 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 ration::minutes(
2c90: 31 29 3b 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 1);...}..}...pub
2ca0: 20 61 73 79 6e 63 20 66 6e 20 6c 69 73 74 3c 53 async fn list<S
2cb0: 3e 28 26 73 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 >(&self, owner:
2cc0: 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 74 72 S) -> Result<Str
2cd0: 69 6e 67 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 ing>..where S: I
2ce0: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
2cf0: 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 owner = owner.i
2d00: 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 nto();....let mu
2d10: 74 20 72 65 70 6c 79 3a 20 56 65 63 3c 43 6f 77 t reply: Vec<Cow
2d20: 3c 73 74 72 3e 3e 20 3d 20 76 65 63 21 5b 5d 3b <str>> = vec![];
2d30: 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
2d40: 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 = self.pool.acqu
2d50: 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e ire().await.....
2d60: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
2d70: 66 6f 72 6d 61 74 21 28 22 4c 69 73 74 20 66 65 format!("List fe
2d80: 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 tch conn:\n{:?}"
2d90: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b , &self.pool))?;
2da0: 0a 09 09 72 65 70 6c 79 2e 70 75 73 68 28 22 43 ...reply.push("C
2db0: 68 61 6e 6e 65 6c 73 3a 22 2e 69 6e 74 6f 28 29 hannels:".into()
2dc0: 29 3b 0a 09 09 6c 65 74 20 72 6f 77 73 20 3d 20 );...let rows =
2dd0: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c sqlx::query("sel
2de0: 65 63 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 63 ect source_id, c
2df0: 68 61 6e 6e 65 6c 2c 20 65 6e 61 62 6c 65 64 2c hannel, enabled,
2e00: 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 75 url, iv_hash, u
2e10: 72 6c 5f 72 65 20 66 72 6f 6d 20 72 73 73 74 67 rl_re from rsstg
2e20: 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 6f 77 _source where ow
2e30: 6e 65 72 20 3d 20 24 31 20 6f 72 64 65 72 20 62 ner = $1 order b
2e40: 79 20 73 6f 75 72 63 65 5f 69 64 22 29 0a 09 09 y source_id")...
2e50: 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 ..bind(owner)...
2e60: 09 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 ..fetch_all(&mut
2e70: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 conn).await?;..
2e80: 09 66 6f 72 20 72 6f 77 20 69 6e 20 72 6f 77 73 .for row in rows
2e90: 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 6c 65 74 .iter() {....let
2ea0: 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 20 source_id: i32
2eb0: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 = row.try_get("s
2ec0: 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 ource_id")?;....
2ed0: 6c 65 74 20 75 73 65 72 6e 61 6d 65 3a 20 26 73 let username: &s
2ee0: 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 tr = row.try_get
2ef0: 28 22 63 68 61 6e 6e 65 6c 22 29 3f 3b 0a 09 09 ("channel")?;...
2f00: 09 6c 65 74 20 65 6e 61 62 6c 65 64 3a 20 62 6f .let enabled: bo
2f10: 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 ol = row.try_get
2f20: 28 22 65 6e 61 62 6c 65 64 22 29 3f 3b 0a 09 09 ("enabled")?;...
2f30: 09 6c 65 74 20 75 72 6c 3a 20 26 73 74 72 20 3d .let url: &str =
2f40: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 72 row.try_get("ur
2f50: 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 69 76 5f l")?;....let iv_
2f60: 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 hash: Option<&st
2f70: 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 r> = row.try_get
2f80: 28 22 69 76 5f 68 61 73 68 22 29 3f 3b 0a 09 09 ("iv_hash")?;...
2f90: 09 6c 65 74 20 75 72 6c 5f 72 65 3a 20 4f 70 74 .let url_re: Opt
2fa0: 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e ion<&str> = row.
2fb0: 74 72 79 5f 67 65 74 28 22 75 72 6c 5f 72 65 22 try_get("url_re"
2fc0: 29 3f 3b 0a 09 09 09 72 65 70 6c 79 2e 70 75 73 )?;....reply.pus
2fd0: 68 28 66 6f 72 6d 61 74 21 28 22 5c 6e 5c 5c 23 h(format!("\n\\#
2fe0: ef b8 8f e2 83 a3 20 7b 7d 20 5c 5c 2a ef b8 8f ļøā£ {} \\*ļø
2ff0: e2 83 a3 20 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 ā£ `{}` {}\nš
3000: 97 20 60 7b 7d 60 22 2c 20 73 6f 75 72 63 65 5f `{}`", source_
3010: 69 64 2c 20 75 73 65 72 6e 61 6d 65 2c 20 20 0a id, username, .
3020: 09 09 09 09 6d 61 74 63 68 20 65 6e 61 62 6c 65 ....match enable
3030: 64 20 7b 0a 09 09 09 09 09 74 72 75 65 20 20 3d d {......true =
3040: 3e 20 22 f0 9f 94 84 20 65 6e 61 62 6c 65 64 22 > "š enabled"
3050: 2c 0a 09 09 09 09 09 66 61 6c 73 65 20 3d 3e 20 ,......false =>
3060: 22 e2 9b 94 20 64 69 73 61 62 6c 65 64 22 2c 0a "ā disabled",.
3070: 09 09 09 09 7d 2c 20 75 72 6c 29 2e 69 6e 74 6f ....}, url).into
3080: 28 29 29 3b 0a 09 09 09 69 66 20 6c 65 74 20 53 ());....if let S
3090: 6f 6d 65 28 68 61 73 68 29 20 3d 20 69 76 5f 68 ome(hash) = iv_h
30a0: 61 73 68 20 7b 0a 09 09 09 09 72 65 70 6c 79 2e ash {.....reply.
30b0: 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 49 56 push(format!("IV
30c0: 3a 20 60 7b 7d 60 22 2c 20 68 61 73 68 29 2e 69 : `{}`", hash).i
30d0: 6e 74 6f 28 29 29 3b 0a 09 09 09 7d 0a 09 09 09 nto());....}....
30e0: 69 66 20 6c 65 74 20 53 6f 6d 65 28 72 65 29 20 if let Some(re)
30f0: 3d 20 75 72 6c 5f 72 65 20 7b 0a 09 09 09 09 72 = url_re {.....r
3100: 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 eply.push(format
3110: 21 28 22 52 45 3a 20 60 7b 7d 60 22 2c 20 72 65 !("RE: `{}`", re
3120: 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 7d 0a ).into());....}.
3130: 09 09 7d 3b 0a 09 09 4f 6b 28 72 65 70 6c 79 2e ..};...Ok(reply.
3140: 6a 6f 69 6e 28 22 5c 6e 22 29 29 0a 09 7d 0a 7d join("\n"))..}.}
3150: 0a .