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 63 68 72 6f 6e 6f 3a 3a 44 61 74 65 54 69 6d chrono::DateTim
0040: 65 3b 0a 75 73 65 20 73 71 6c 78 3a 3a 7b 0a 09 e;.use sqlx::{..
0050: 70 6f 73 74 67 72 65 73 3a 3a 50 67 50 6f 6f 6c postgres::PgPool
0060: 4f 70 74 69 6f 6e 73 2c 0a 09 52 6f 77 2c 0a 7d Options,..Row,.}
0070: 3b 0a 75 73 65 20 73 74 64 3a 3a 7b 0a 09 62 6f ;.use std::{..bo
0080: 72 72 6f 77 3a 3a 43 6f 77 2c 0a 09 63 6f 6c 6c rrow::Cow,..coll
0090: 65 63 74 69 6f 6e 73 3a 3a 7b 0a 09 09 42 54 72 ections::{...BTr
00a0: 65 65 4d 61 70 2c 0a 09 09 48 61 73 68 53 65 74 eeMap,...HashSet
00b0: 2c 0a 09 7d 2c 0a 09 73 79 6e 63 3a 3a 7b 41 72 ,..},..sync::{Ar
00c0: 63 2c 20 4d 75 74 65 78 7d 2c 0a 7d 3b 0a 0a 23 c, Mutex},.};..#
00d0: 5b 64 65 72 69 76 65 28 43 6c 6f 6e 65 29 5d 0a [derive(Clone)].
00e0: 70 75 62 20 73 74 72 75 63 74 20 43 6f 72 65 20 pub struct Core
00f0: 7b 0a 09 2f 2f 6f 77 6e 65 72 3a 20 69 36 34 2c {..//owner: i64,
0100: 0a 09 2f 2f 61 70 69 5f 6b 65 79 3a 20 53 74 72 ..//api_key: Str
0110: 69 6e 67 2c 0a 09 6f 77 6e 65 72 5f 63 68 61 74 ing,..owner_chat
0120: 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a : telegram_bot::
0130: 55 73 65 72 49 64 2c 0a 09 70 75 62 20 74 67 3a UserId,..pub tg:
0140: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 41 telegram_bot::A
0150: 70 69 2c 0a 09 70 75 62 20 6d 79 3a 20 74 65 6c pi,..pub my: tel
0160: 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 2c egram_bot::User,
0170: 0a 09 70 6f 6f 6c 3a 20 73 71 6c 78 3a 3a 50 6f ..pool: sqlx::Po
0180: 6f 6c 3c 73 71 6c 78 3a 3a 50 6f 73 74 67 72 65 ol<sqlx::Postgre
0190: 73 3e 2c 0a 09 73 6f 75 72 63 65 73 3a 20 41 72 s>,..sources: Ar
01a0: 63 3c 4d 75 74 65 78 3c 48 61 73 68 53 65 74 3c c<Mutex<HashSet<
01b0: 41 72 63 3c 69 33 32 3e 3e 3e 3e 2c 0a 7d 0a 0a Arc<i32>>>>,.}..
01c0: 69 6d 70 6c 20 43 6f 72 65 20 7b 0a 09 70 75 62 impl Core {..pub
01d0: 20 61 73 79 6e 63 20 66 6e 20 6e 65 77 28 73 65 async fn new(se
01e0: 74 74 69 6e 67 73 3a 20 63 6f 6e 66 69 67 3a 3a ttings: config::
01f0: 43 6f 6e 66 69 67 29 20 2d 3e 20 52 65 73 75 6c Config) -> Resul
0200: 74 3c 41 72 63 3c 43 6f 72 65 3e 3e 20 7b 0a 09 t<Arc<Core>> {..
0210: 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 73 65 74 .let owner = set
0220: 74 69 6e 67 73 2e 67 65 74 5f 69 6e 74 28 22 6f tings.get_int("o
0230: 77 6e 65 72 22 29 3f 3b 0a 09 09 6c 65 74 20 61 wner")?;...let a
0240: 70 69 5f 6b 65 79 20 3d 20 73 65 74 74 69 6e 67 pi_key = setting
0250: 73 2e 67 65 74 5f 73 74 72 28 22 61 70 69 5f 6b s.get_str("api_k
0260: 65 79 22 29 3f 3b 0a 09 09 6c 65 74 20 74 67 20 ey")?;...let tg
0270: 3d 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a = telegram_bot::
0280: 41 70 69 3a 3a 6e 65 77 28 26 61 70 69 5f 6b 65 Api::new(&api_ke
0290: 79 29 3b 0a 09 09 6c 65 74 20 63 6f 72 65 20 3d y);...let core =
02a0: 20 41 72 63 3a 3a 6e 65 77 28 43 6f 72 65 20 7b Arc::new(Core {
02b0: 0a 09 09 09 2f 2f 6f 77 6e 65 72 2c 0a 09 09 09 ....//owner,....
02c0: 2f 2f 61 70 69 5f 6b 65 79 3a 20 61 70 69 5f 6b //api_key: api_k
02d0: 65 79 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 6d ey.clone(),....m
02e0: 79 3a 20 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 y: tg.send(teleg
02f0: 72 61 6d 5f 62 6f 74 3a 3a 47 65 74 4d 65 29 2e ram_bot::GetMe).
0300: 61 77 61 69 74 3f 2c 0a 09 09 09 74 67 2c 0a 09 await?,....tg,..
0310: 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 74 65 ..owner_chat: te
0320: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 legram_bot::User
0330: 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 2c 0a Id::new(owner),.
0340: 09 09 09 70 6f 6f 6c 3a 20 50 67 50 6f 6f 6c 4f ...pool: PgPoolO
0350: 70 74 69 6f 6e 73 3a 3a 6e 65 77 28 29 0a 09 09 ptions::new()...
0360: 09 09 2e 6d 61 78 5f 63 6f 6e 6e 65 63 74 69 6f ...max_connectio
0370: 6e 73 28 35 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 ns(5)......conne
0380: 63 74 5f 74 69 6d 65 6f 75 74 28 73 74 64 3a 3a ct_timeout(std::
0390: 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a time::Duration::
03a0: 6e 65 77 28 33 30 30 2c 20 30 29 29 0a 09 09 09 new(300, 0))....
03b0: 09 2e 69 64 6c 65 5f 74 69 6d 65 6f 75 74 28 73 ..idle_timeout(s
03c0: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
03d0: 6f 6e 3a 3a 6e 65 77 28 36 30 2c 20 30 29 29 0a on::new(60, 0)).
03e0: 09 09 09 09 2e 63 6f 6e 6e 65 63 74 5f 6c 61 7a .....connect_laz
03f0: 79 28 26 73 65 74 74 69 6e 67 73 2e 67 65 74 5f y(&settings.get_
0400: 73 74 72 28 22 70 67 22 29 3f 29 3f 2c 0a 09 09 str("pg")?)?,...
0410: 09 73 6f 75 72 63 65 73 3a 20 41 72 63 3a 3a 6e .sources: Arc::n
0420: 65 77 28 4d 75 74 65 78 3a 3a 6e 65 77 28 48 61 ew(Mutex::new(Ha
0430: 73 68 53 65 74 3a 3a 6e 65 77 28 29 29 29 2c 0a shSet::new())),.
0440: 09 09 7d 29 3b 0a 09 09 6c 65 74 20 63 6c 6f 6e ..});...let clon
0450: 65 20 3d 20 63 6f 72 65 2e 63 6c 6f 6e 65 28 29 e = core.clone()
0460: 3b 0a 09 09 74 6f 6b 69 6f 3a 3a 73 70 61 77 6e ;...tokio::spawn
0470: 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 (async move {...
0480: 09 6c 6f 6f 70 20 7b 0a 09 09 09 09 6c 65 74 20 .loop {.....let
0490: 64 65 6c 61 79 20 3d 20 6d 61 74 63 68 20 26 63 delay = match &c
04a0: 6c 6f 6e 65 2e 61 75 74 6f 66 65 74 63 68 28 29 lone.autofetch()
04b0: 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 45 72 .await {......Er
04c0: 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 r(err) => {.....
04d0: 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 ..if let Err(err
04e0: 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 66 ) = clone.send(f
04f0: 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a 3f ormat!("š {:?
0500: 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 2c 20 }", err), None,
0510: 4e 6f 6e 65 29 2e 61 77 61 69 74 20 7b 0a 09 09 None).await {...
0520: 09 09 09 09 09 65 70 72 69 6e 74 6c 6e 21 28 22 .....eprintln!("
0530: 41 75 74 6f 66 65 74 63 68 20 65 72 72 6f 72 3a Autofetch error:
0540: 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 09 {}", err);.....
0550: 09 09 7d 3b 0a 09 09 09 09 09 09 74 6f 6b 69 6f ..};.......tokio
0560: 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e ::time::Duration
0570: 3a 3a 66 72 6f 6d 5f 73 65 63 73 28 36 30 29 0a ::from_secs(60).
0580: 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 4f 6b 28 .....},......Ok(
0590: 74 69 6d 65 29 20 3d 3e 20 2a 74 69 6d 65 2c 0a time) => *time,.
05a0: 09 09 09 09 7d 3b 0a 09 09 09 09 74 6f 6b 69 6f ....};.....tokio
05b0: 3a 3a 74 69 6d 65 3a 3a 73 6c 65 65 70 28 64 65 ::time::sleep(de
05c0: 6c 61 79 29 2e 61 77 61 69 74 3b 0a 09 09 09 7d lay).await;....}
05d0: 0a 09 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 ...});...Ok(core
05e0: 29 0a 09 7d 0a 0a 09 70 75 62 20 66 6e 20 73 74 )..}...pub fn st
05f0: 72 65 61 6d 28 26 73 65 6c 66 29 20 2d 3e 20 74 ream(&self) -> t
0600: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 70 64 elegram_bot::Upd
0610: 61 74 65 73 53 74 72 65 61 6d 20 7b 0a 09 09 73 atesStream {...s
0620: 65 6c 66 2e 74 67 2e 73 74 72 65 61 6d 28 29 0a elf.tg.stream().
0630: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
0640: 6e 20 73 65 6e 64 3c 27 61 2c 20 53 3e 28 26 73 n send<'a, S>(&s
0650: 65 6c 66 2c 20 6d 73 67 3a 20 53 2c 20 74 61 72 elf, msg: S, tar
0660: 67 65 74 3a 20 4f 70 74 69 6f 6e 3c 74 65 6c 65 get: Option<tele
0670: 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 gram_bot::UserId
0680: 3e 2c 20 6d 6f 64 65 3a 20 4f 70 74 69 6f 6e 3c >, mode: Option<
0690: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 telegram_bot::ty
06a0: 70 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3e 29 pes::ParseMode>)
06b0: 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 0a 09 -> Result<()>..
06c0: 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 43 6f where S: Into<Co
06d0: 77 3c 27 61 2c 20 73 74 72 3e 3e 20 7b 0a 09 09 w<'a, str>> {...
06e0: 6c 65 74 20 6d 6f 64 65 20 3d 20 6d 6f 64 65 2e let mode = mode.
06f0: 75 6e 77 72 61 70 5f 6f 72 28 74 65 6c 65 67 72 unwrap_or(telegr
0700: 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 am_bot::types::P
0710: 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 3b arseMode::Html);
0720: 0a 09 09 6c 65 74 20 74 61 72 67 65 74 20 3d 20 ...let target =
0730: 74 61 72 67 65 74 2e 75 6e 77 72 61 70 5f 6f 72 target.unwrap_or
0740: 28 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 61 74 (self.owner_chat
0750: 29 3b 0a 09 09 73 65 6c 66 2e 74 67 2e 73 65 6e );...self.tg.sen
0760: 64 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a d(telegram_bot::
0770: 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 SendMessage::new
0780: 28 74 61 72 67 65 74 2c 20 6d 73 67 29 2e 70 61 (target, msg).pa
0790: 72 73 65 5f 6d 6f 64 65 28 6d 6f 64 65 29 29 2e rse_mode(mode)).
07a0: 61 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 28 29 29 await?;...Ok(())
07b0: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
07c0: 66 6e 20 63 68 65 63 6b 3c 53 3e 28 26 73 65 6c fn check<S>(&sel
07d0: 66 2c 20 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e f, id: &i32, own
07e0: 65 72 3a 20 53 2c 20 72 65 61 6c 3a 20 62 6f 6f er: S, real: boo
07f0: 6c 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 l) -> Result<Cow
0800: 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 <'_, str>>..wher
0810: 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e S: Into<i64> {
0820: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f ...let owner = o
0830: 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 wner.into();....
0840: 6c 65 74 20 6d 75 74 20 70 6f 73 74 65 64 3a 20 let mut posted:
0850: 69 33 32 20 3d 20 30 3b 0a 09 09 6c 65 74 20 69 i32 = 0;...let i
0860: 64 20 3d 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 d = {....let mut
0870: 20 73 65 74 20 3d 20 73 65 6c 66 2e 73 6f 75 72 set = self.sour
0880: 63 65 73 2e 6c 6f 63 6b 28 29 2e 75 6e 77 72 61 ces.lock().unwra
0890: 70 28 29 3b 0a 09 09 09 6d 61 74 63 68 20 73 65 p();....match se
08a0: 74 2e 67 65 74 28 69 64 29 20 7b 0a 09 09 09 09 t.get(id) {.....
08b0: 53 6f 6d 65 28 69 64 29 20 3d 3e 20 69 64 2e 63 Some(id) => id.c
08c0: 6c 6f 6e 65 28 29 2c 0a 09 09 09 09 4e 6f 6e 65 lone(),.....None
08d0: 20 3d 3e 20 7b 0a 09 09 09 09 09 6c 65 74 20 69 => {......let i
08e0: 64 20 3d 20 41 72 63 3a 3a 6e 65 77 28 2a 69 64 d = Arc::new(*id
08f0: 29 3b 0a 09 09 09 09 09 73 65 74 2e 69 6e 73 65 );......set.inse
0900: 72 74 28 69 64 2e 63 6c 6f 6e 65 28 29 29 3b 0a rt(id.clone());.
0910: 09 09 09 09 09 69 64 2e 63 6c 6f 6e 65 28 29 0a .....id.clone().
0920: 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 7d 3b ....},....}...};
0930: 0a 09 09 6c 65 74 20 63 6f 75 6e 74 20 3d 20 41 ...let count = A
0940: 72 63 3a 3a 73 74 72 6f 6e 67 5f 63 6f 75 6e 74 rc::strong_count
0950: 28 26 69 64 29 3b 0a 09 09 69 66 20 63 6f 75 6e (&id);...if coun
0960: 74 20 3d 3d 20 32 20 7b 0a 09 09 09 6c 65 74 20 t == 2 {....let
0970: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
0980: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
0990: 77 61 69 74 0a 09 09 09 09 2e 77 69 74 68 5f 63 wait......with_c
09a0: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
09b0: 21 28 22 51 75 65 72 79 20 71 75 65 75 65 20 66 !("Query queue f
09c0: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
09d0: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
09e0: 3b 0a 09 09 09 6c 65 74 20 72 6f 77 20 3d 20 73 ;....let row = s
09f0: 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 qlx::query("sele
0a00: 63 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 ct source_id, ch
0a10: 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 annel_id, url, i
0a20: 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 75 v_hash, owner, u
0a30: 72 6c 5f 72 65 20 66 72 6f 6d 20 72 73 73 74 67 rl_re from rsstg
0a40: 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 73 6f _source where so
0a50: 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 urce_id = $1 and
0a60: 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 owner = $2")...
0a70: 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 ...bind(*id)....
0a80: 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 ..bind(owner)...
0a90: 09 09 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d 75 ...fetch_one(&mu
0aa0: 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 t conn).await...
0ab0: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
0ac0: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 51 75 65 72 || format!("Quer
0ad0: 79 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 y source:\n{:?}"
0ae0: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b , &self.pool))?;
0af0: 0a 09 09 09 64 72 6f 70 28 63 6f 6e 6e 29 3b 0a ....drop(conn);.
0b00: 09 09 09 6c 65 74 20 63 68 61 6e 6e 65 6c 5f 69 ...let channel_i
0b10: 64 3a 20 69 36 34 20 3d 20 72 6f 77 2e 74 72 79 d: i64 = row.try
0b20: 5f 67 65 74 28 22 63 68 61 6e 6e 65 6c 5f 69 64 _get("channel_id
0b30: 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c 3a ")?;....let url:
0b40: 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f &str = row.try_
0b50: 67 65 74 28 22 75 72 6c 22 29 3f 3b 0a 09 09 09 get("url")?;....
0b60: 6c 65 74 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 let iv_hash: Opt
0b70: 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e ion<&str> = row.
0b80: 74 72 79 5f 67 65 74 28 22 69 76 5f 68 61 73 68 try_get("iv_hash
0b90: 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c 5f ")?;....let url_
0ba0: 72 65 20 3d 20 6d 61 74 63 68 20 72 6f 77 2e 74 re = match row.t
0bb0: 72 79 5f 67 65 74 28 22 75 72 6c 5f 72 65 22 29 ry_get("url_re")
0bc0: 3f 20 7b 0a 09 09 09 09 53 6f 6d 65 28 78 29 20 ? {.....Some(x)
0bd0: 3d 3e 20 53 6f 6d 65 28 73 65 64 72 65 67 65 78 => Some(sedregex
0be0: 3a 3a 52 65 70 6c 61 63 65 43 6f 6d 6d 61 6e 64 ::ReplaceCommand
0bf0: 3a 3a 6e 65 77 28 78 29 3f 29 2c 0a 09 09 09 09 ::new(x)?),.....
0c00: 4e 6f 6e 65 20 3d 3e 20 4e 6f 6e 65 2c 0a 09 09 None => None,...
0c10: 09 7d 3b 0a 09 09 09 6c 65 74 20 64 65 73 74 69 .};....let desti
0c20: 6e 61 74 69 6f 6e 20 3d 20 6d 61 74 63 68 20 72 nation = match r
0c30: 65 61 6c 20 7b 0a 09 09 09 09 74 72 75 65 20 3d eal {.....true =
0c40: 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a > telegram_bot::
0c50: 55 73 65 72 49 64 3a 3a 6e 65 77 28 63 68 61 6e UserId::new(chan
0c60: 6e 65 6c 5f 69 64 29 2c 0a 09 09 09 09 66 61 6c nel_id),.....fal
0c70: 73 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 se => telegram_b
0c80: 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 ot::UserId::new(
0c90: 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e row.try_get("own
0ca0: 65 72 22 29 3f 29 2c 0a 09 09 09 7d 3b 0a 09 09 er")?),....};...
0cb0: 09 6c 65 74 20 6d 75 74 20 74 68 69 73 5f 66 65 .let mut this_fe
0cc0: 74 63 68 3a 20 4f 70 74 69 6f 6e 3c 44 61 74 65 tch: Option<Date
0cd0: 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 Time<chrono::Fix
0ce0: 65 64 4f 66 66 73 65 74 3e 3e 20 3d 20 4e 6f 6e edOffset>> = Non
0cf0: 65 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 70 6f e;....let mut po
0d00: 73 74 73 3a 20 42 54 72 65 65 4d 61 70 3c 44 61 sts: BTreeMap<Da
0d10: 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 teTime<chrono::F
0d20: 69 78 65 64 4f 66 66 73 65 74 3e 2c 20 53 74 72 ixedOffset>, Str
0d30: 69 6e 67 3e 20 3d 20 42 54 72 65 65 4d 61 70 3a ing> = BTreeMap:
0d40: 3a 6e 65 77 28 29 3b 0a 09 09 09 6c 65 74 20 72 :new();....let r
0d50: 65 73 70 6f 6e 73 65 20 3d 20 72 65 71 77 65 73 esponse = reqwes
0d60: 74 3a 3a 67 65 74 28 75 72 6c 29 2e 61 77 61 69 t::get(url).awai
0d70: 74 3f 3b 0a 09 09 09 6c 65 74 20 73 74 61 74 75 t?;....let statu
0d80: 73 20 3d 20 72 65 73 70 6f 6e 73 65 2e 73 74 61 s = response.sta
0d90: 74 75 73 28 29 3b 0a 09 09 09 6c 65 74 20 63 6f tus();....let co
0da0: 6e 74 65 6e 74 20 3d 20 72 65 73 70 6f 6e 73 65 ntent = response
0db0: 2e 62 79 74 65 73 28 29 2e 61 77 61 69 74 3f 3b .bytes().await?;
0dc0: 0a 09 09 09 6d 61 74 63 68 20 72 73 73 3a 3a 43 ....match rss::C
0dd0: 68 61 6e 6e 65 6c 3a 3a 72 65 61 64 5f 66 72 6f hannel::read_fro
0de0: 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 m(&content[..])
0df0: 7b 0a 09 09 09 09 4f 6b 28 66 65 65 64 29 20 3d {.....Ok(feed) =
0e00: 3e 20 7b 0a 09 09 09 09 09 66 6f 72 20 69 74 65 > {......for ite
0e10: 6d 20 69 6e 20 66 65 65 64 2e 69 74 65 6d 73 28 m in feed.items(
0e20: 29 20 7b 0a 09 09 09 09 09 09 69 66 20 6c 65 74 ) {.......if let
0e30: 20 53 6f 6d 65 28 6c 69 6e 6b 29 20 3d 20 69 74 Some(link) = it
0e40: 65 6d 2e 6c 69 6e 6b 28 29 20 7b 0a 09 09 09 09 em.link() {.....
0e50: 09 09 09 6c 65 74 20 64 61 74 65 20 3d 20 6d 61 ...let date = ma
0e60: 74 63 68 20 69 74 65 6d 2e 70 75 62 5f 64 61 74 tch item.pub_dat
0e70: 65 28 29 20 7b 0a 09 09 09 09 09 09 09 09 53 6f e() {.........So
0e80: 6d 65 28 66 65 65 64 5f 64 61 74 65 29 20 3d 3e me(feed_date) =>
0e90: 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 DateTime::parse
0ea0: 5f 66 72 6f 6d 5f 72 66 63 32 38 32 32 28 66 65 _from_rfc2822(fe
0eb0: 65 64 5f 64 61 74 65 29 2c 0a 09 09 09 09 09 09 ed_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 7d 3f 3b 0a 0]),........}?;.
0f20: 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c 20 3d .......let url =
0f30: 20 6c 69 6e 6b 3b 0a 09 09 09 09 09 09 09 70 6f link;........po
0f40: 73 74 73 2e 69 6e 73 65 72 74 28 64 61 74 65 2c sts.insert(date,
0f50: 20 75 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 url.to_string()
0f60: 29 3b 0a 09 09 09 09 09 09 7d 0a 09 09 09 09 09 );.......}......
0f70: 7d 3b 0a 09 09 09 09 7d 2c 0a 09 09 09 09 45 72 };.....},.....Er
0f80: 72 28 65 72 72 29 20 3d 3e 20 6d 61 74 63 68 20 r(err) => match
0f90: 65 72 72 20 7b 0a 09 09 09 09 09 72 73 73 3a 3a err {......rss::
0fa0: 45 72 72 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 74 Error::InvalidSt
0fb0: 61 72 74 54 61 67 20 3d 3e 20 7b 0a 09 09 09 09 artTag => {.....
0fc0: 09 09 6c 65 74 20 66 65 65 64 20 3d 20 61 74 6f ..let feed = ato
0fd0: 6d 5f 73 79 6e 64 69 63 61 74 69 6f 6e 3a 3a 46 m_syndication::F
0fe0: 65 65 64 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 eed::read_from(&
0ff0: 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 0a 09 09 09 content[..])....
1000: 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
1010: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 50 72 t(|| format!("Pr
1020: 6f 62 6c 65 6d 20 6f 70 65 6e 69 6e 67 20 66 65 oblem opening fe
1030: 65 64 20 75 72 6c 3a 5c 6e 7b 7d 5c 6e 7b 7d 22 ed url:\n{}\n{}"
1040: 2c 20 26 75 72 6c 2c 20 73 74 61 74 75 73 29 29 , &url, status))
1050: 3f 3b 0a 09 09 09 09 09 09 66 6f 72 20 69 74 65 ?;.......for ite
1060: 6d 20 69 6e 20 66 65 65 64 2e 65 6e 74 72 69 65 m in feed.entrie
1070: 73 28 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 s() {........let
1080: 20 64 61 74 65 20 3d 20 69 74 65 6d 2e 70 75 62 date = item.pub
1090: 6c 69 73 68 65 64 28 29 2e 75 6e 77 72 61 70 28 lished().unwrap(
10a0: 29 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 75 72 );........let ur
10b0: 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 29 l = item.links()
10c0: 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 09 09 [0].href();.....
10d0: 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 ...posts.insert(
10e0: 2a 64 61 74 65 2c 20 75 72 6c 2e 74 6f 5f 73 74 *date, url.to_st
10f0: 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 7d ring());.......}
1100: 3b 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 72 ;......},......r
1110: 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 6f 66 20 3d ss::Error::Eof =
1120: 3e 20 28 29 2c 0a 09 09 09 09 09 5f 20 3d 3e 20 > (),......_ =>
1130: 62 61 69 6c 21 28 22 55 6e 73 75 70 70 6f 72 74 bail!("Unsupport
1140: 65 64 20 6f 72 20 6d 61 6e 67 6c 65 64 20 63 6f ed or mangled co
1150: 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b 3a ntent:\n{:?}\n{:
1160: 23 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 26 #?}\n{:#?}\n", &
1170: 75 72 6c 2c 20 65 72 72 2c 20 73 74 61 74 75 73 url, err, status
1180: 29 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 ).....}....};...
1190: 09 66 6f 72 20 28 64 61 74 65 2c 20 75 72 6c 29 .for (date, url)
11a0: 20 69 6e 20 70 6f 73 74 73 2e 69 74 65 72 28 29 in posts.iter()
11b0: 20 7b 0a 09 09 09 09 6c 65 74 20 6d 75 74 20 63 {.....let mut c
11c0: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e onn = self.pool.
11d0: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a acquire().await.
11e0: 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 ......with_conte
11f0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 xt(|| format!("C
1200: 68 65 63 6b 20 70 6f 73 74 20 66 65 74 63 68 20 heck post fetch
1210: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
1220: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 elf.pool))?;....
1230: 09 6c 65 74 20 70 6f 73 74 5f 75 72 6c 3a 20 43 .let post_url: C
1240: 6f 77 3c 73 74 72 3e 20 3d 20 6d 61 74 63 68 20 ow<str> = match
1250: 75 72 6c 5f 72 65 20 7b 0a 09 09 09 09 09 53 6f url_re {......So
1260: 6d 65 28 72 65 66 20 78 29 20 3d 3e 20 78 2e 65 me(ref x) => x.e
1270: 78 65 63 75 74 65 28 75 72 6c 29 2c 0a 09 09 09 xecute(url),....
1280: 09 09 4e 6f 6e 65 20 3d 3e 20 75 72 6c 2e 69 6e ..None => url.in
1290: 74 6f 28 29 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 to(),.....};....
12a0: 09 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c 78 3a .let row = sqlx:
12b0: 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 65 :query("select e
12c0: 78 69 73 74 73 28 73 65 6c 65 63 74 20 74 72 75 xists(select tru
12d0: 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 6f 73 e from rsstg_pos
12e0: 74 20 77 68 65 72 65 20 75 72 6c 20 3d 20 24 31 t where url = $1
12f0: 20 61 6e 64 20 73 6f 75 72 63 65 5f 69 64 20 3d and source_id =
1300: 20 24 32 29 20 61 73 20 65 78 69 73 74 73 3b 22 $2) as exists;"
1310: 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 26 2a 70 ).......bind(&*p
1320: 6f 73 74 5f 75 72 6c 29 0a 09 09 09 09 09 2e 62 ost_url).......b
1330: 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 09 2e 66 ind(*id).......f
1340: 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 63 6f etch_one(&mut co
1350: 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 09 2e nn).await.......
1360: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
1370: 66 6f 72 6d 61 74 21 28 22 43 68 65 63 6b 20 70 format!("Check p
1380: 6f 73 74 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f ost:\n{:?}", &co
1390: 6e 6e 29 29 3f 3b 0a 09 09 09 09 6c 65 74 20 65 nn))?;.....let e
13a0: 78 69 73 74 73 3a 20 62 6f 6f 6c 20 3d 20 72 6f xists: bool = ro
13b0: 77 2e 74 72 79 5f 67 65 74 28 22 65 78 69 73 74 w.try_get("exist
13c0: 73 22 29 3f 3b 0a 09 09 09 09 69 66 20 21 20 65 s")?;.....if ! e
13d0: 78 69 73 74 73 20 7b 0a 09 09 09 09 09 69 66 20 xists {......if
13e0: 74 68 69 73 5f 66 65 74 63 68 20 3d 3d 20 4e 6f this_fetch == No
13f0: 6e 65 20 7c 7c 20 2a 64 61 74 65 20 3e 20 74 68 ne || *date > th
1400: 69 73 5f 66 65 74 63 68 2e 75 6e 77 72 61 70 28 is_fetch.unwrap(
1410: 29 20 7b 0a 09 09 09 09 09 09 74 68 69 73 5f 66 ) {.......this_f
1420: 65 74 63 68 20 3d 20 53 6f 6d 65 28 2a 64 61 74 etch = Some(*dat
1430: 65 29 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 e);......};.....
1440: 09 73 65 6c 66 2e 74 67 2e 73 65 6e 64 28 20 6d .self.tg.send( m
1450: 61 74 63 68 20 69 76 5f 68 61 73 68 20 7b 0a 09 atch iv_hash {..
1460: 09 09 09 09 09 09 53 6f 6d 65 28 68 61 73 68 29 ......Some(hash)
1470: 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 => telegram_bot
1480: 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e ::SendMessage::n
1490: 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c 20 ew(destination,
14a0: 66 6f 72 6d 61 74 21 28 22 3c 61 20 68 72 65 66 format!("<a href
14b0: 3d 5c 22 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f =\"https://t.me/
14c0: 69 76 3f 75 72 6c 3d 7b 7d 26 72 68 61 73 68 3d iv?url={}&rhash=
14d0: 7b 7d 5c 22 3e 20 3c 2f 61 3e 7b 30 7d 22 2c 20 {}\"> </a>{0}",
14e0: 26 70 6f 73 74 5f 75 72 6c 2c 20 68 61 73 68 29 &post_url, hash)
14f0: 29 2c 0a 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d ),........None =
1500: 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a > telegram_bot::
1510: 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 SendMessage::new
1520: 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f (destination, fo
1530: 72 6d 61 74 21 28 22 7b 7d 22 2c 20 70 6f 73 74 rmat!("{}", post
1540: 5f 75 72 6c 29 29 2c 0a 09 09 09 09 09 09 7d 2e _url)),.......}.
1550: 70 61 72 73 65 5f 6d 6f 64 65 28 74 65 6c 65 67 parse_mode(teleg
1560: 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a ram_bot::types::
1570: 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 ParseMode::Html)
1580: 29 2e 61 77 61 69 74 0a 09 09 09 09 09 09 2e 63 ).await........c
1590: 6f 6e 74 65 78 74 28 22 43 61 6e 27 74 20 70 6f ontext("Can't po
15a0: 73 74 20 6d 65 73 73 61 67 65 3a 22 29 3f 3b 0a st message:")?;.
15b0: 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 .....sqlx::query
15c0: 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 ("insert into rs
15d0: 73 74 67 5f 70 6f 73 74 20 28 73 6f 75 72 63 65 stg_post (source
15e0: 5f 69 64 2c 20 70 6f 73 74 65 64 2c 20 75 72 6c _id, posted, url
15f0: 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 ) values ($1, $2
1600: 2c 20 24 33 29 3b 22 29 0a 09 09 09 09 09 09 2e , $3);")........
1610: 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 09 09 bind(*id).......
1620: 2e 62 69 6e 64 28 64 61 74 65 29 0a 09 09 09 09 .bind(date).....
1630: 09 09 2e 62 69 6e 64 28 26 2a 70 6f 73 74 5f 75 ...bind(&*post_u
1640: 72 6c 29 0a 09 09 09 09 09 09 2e 65 78 65 63 75 rl)........execu
1650: 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 te(&mut conn).aw
1660: 61 69 74 0a 09 09 09 09 09 09 2e 77 69 74 68 5f ait........with_
1670: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
1680: 74 21 28 22 52 65 63 6f 72 64 20 70 6f 73 74 3a t!("Record post:
1690: 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 \n{:?}", &conn))
16a0: 3f 3b 0a 09 09 09 09 09 64 72 6f 70 28 63 6f 6e ?;......drop(con
16b0: 6e 29 3b 0a 09 09 09 09 09 74 6f 6b 69 6f 3a 3a n);......tokio::
16c0: 74 69 6d 65 3a 3a 73 6c 65 65 70 28 73 74 64 3a time::sleep(std:
16d0: 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a :time::Duration:
16e0: 3a 6e 65 77 28 34 2c 20 30 29 29 2e 61 77 61 69 :new(4, 0)).awai
16f0: 74 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 09 70 6f t;.....};.....po
1700: 73 74 65 64 20 2b 3d 20 31 3b 0a 09 09 09 7d 3b sted += 1;....};
1710: 0a 09 09 09 70 6f 73 74 73 2e 63 6c 65 61 72 28 ....posts.clear(
1720: 29 3b 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 6d 75 );...};...let mu
1730: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
1740: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
1750: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
1760: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1770: 55 70 64 61 74 65 20 73 63 72 61 70 65 20 66 65 Update scrape fe
1780: 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 tch conn:\n{:?}"
1790: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b , &self.pool))?;
17a0: 0a 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ...sqlx::query("
17b0: 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 update rsstg_sou
17c0: 72 63 65 20 73 65 74 20 6c 61 73 74 5f 73 63 72 rce set last_scr
17d0: 61 70 65 20 3d 20 6e 6f 77 28 29 20 77 68 65 72 ape = now() wher
17e0: 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 e source_id = $1
17f0: 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 2a 69 64 ;").....bind(*id
1800: 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d ).....execute(&m
1810: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 ut conn).await..
1820: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1830: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 || format!("Upda
1840: 74 65 20 73 63 72 61 70 65 3a 5c 6e 7b 3a 3f 7d te scrape:\n{:?}
1850: 22 2c 20 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 4f ", &conn))?;...O
1860: 6b 28 66 6f 72 6d 61 74 21 28 22 50 6f 73 74 65 k(format!("Poste
1870: 64 3a 20 7b 7d 22 2c 20 26 70 6f 73 74 65 64 29 d: {}", &posted)
1880: 2e 69 6e 74 6f 28 29 29 0a 09 7d 0a 0a 09 70 75 .into())..}...pu
1890: 62 20 61 73 79 6e 63 20 66 6e 20 64 65 6c 65 74 b async fn delet
18a0: 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 e<S>(&self, sour
18b0: 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e ce_id: &i32, own
18c0: 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 er: S) -> Result
18d0: 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 <Cow<'_, str>>..
18e0: 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 where S: Into<i6
18f0: 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 4> {...let owner
1900: 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b = owner.into();
1910: 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ....let mut conn
1920: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
1930: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
1940: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
1950: 20 66 6f 72 6d 61 74 21 28 22 44 65 6c 65 74 65 format!("Delete
1960: 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a fetch conn:\n{:
1970: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
1980: 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 )?;...match sqlx
1990: 3a 3a 71 75 65 72 79 28 22 64 65 6c 65 74 65 20 ::query("delete
19a0: 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 from rsstg_sourc
19b0: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
19c0: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
19d0: 20 3d 20 24 32 3b 22 29 0a 09 09 09 2e 62 69 6e = $2;").....bin
19e0: 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 d(source_id)....
19f0: 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 .bind(owner)....
1a00: 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f .execute(&mut co
1a10: 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 nn).await.....wi
1a20: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
1a30: 72 6d 61 74 21 28 22 44 65 6c 65 74 65 20 73 6f rmat!("Delete so
1a40: 75 72 63 65 20 72 75 6c 65 3a 5c 6e 7b 3a 3f 7d urce rule:\n{:?}
1a50: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
1a60: 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 .....rows_affect
1a70: 65 64 28 29 20 7b 0a 09 09 09 30 20 3d 3e 20 7b ed() {....0 => {
1a80: 20 4f 6b 28 22 4e 6f 20 64 61 74 61 20 66 6f 75 Ok("No data fou
1a90: 6e 64 20 66 6f 75 6e 64 2e 22 2e 69 6e 74 6f 28 nd found.".into(
1aa0: 29 29 20 7d 2c 0a 09 09 09 78 20 3d 3e 20 7b 20 )) },....x => {
1ab0: 4f 6b 28 66 6f 72 6d 61 74 21 28 22 7b 7d 20 73 Ok(format!("{} s
1ac0: 6f 75 72 63 65 73 20 72 65 6d 6f 76 65 64 2e 22 ources removed."
1ad0: 2c 20 78 29 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a , x).into()) },.
1ae0: 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ..}..}...pub asy
1af0: 6e 63 20 66 6e 20 63 6c 65 61 6e 3c 53 3e 28 26 nc fn clean<S>(&
1b00: 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a self, source_id:
1b10: 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 &i32, owner: S)
1b20: 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 3c 27 -> Result<Cow<'
1b30: 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 65 20 _, str>>..where
1b40: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
1b50: 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e .let owner = own
1b60: 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 er.into();....le
1b70: 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c t mut conn = sel
1b80: 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 f.pool.acquire()
1b90: 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f .await.....with_
1ba0: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
1bb0: 74 21 28 22 43 6c 65 61 6e 20 66 65 74 63 68 20 t!("Clean fetch
1bc0: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
1bd0: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d elf.pool))?;...m
1be0: 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 atch sqlx::query
1bf0: 28 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 ("delete from rs
1c00: 73 74 67 5f 70 6f 73 74 20 70 20 75 73 69 6e 67 stg_post p using
1c10: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 20 rsstg_source s
1c20: 77 68 65 72 65 20 70 2e 73 6f 75 72 63 65 5f 69 where p.source_i
1c30: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
1c40: 20 3d 20 24 32 20 61 6e 64 20 70 2e 73 6f 75 72 = $2 and p.sour
1c50: 63 65 5f 69 64 20 3d 20 73 2e 73 6f 75 72 63 65 ce_id = s.source
1c60: 5f 69 64 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 _id;").....bind(
1c70: 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 source_id).....b
1c80: 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 ind(owner).....e
1c90: 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e xecute(&mut conn
1ca0: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
1cb0: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
1cc0: 61 74 21 28 22 43 6c 65 61 6e 20 73 65 65 6e 20 at!("Clean seen
1cd0: 70 6f 73 74 73 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 posts:\n{:?}", &
1ce0: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 self.pool))?....
1cf0: 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 .rows_affected()
1d00: 20 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 {....0 => { Ok(
1d10: 22 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 "No data found f
1d20: 6f 75 6e 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d ound.".into()) }
1d30: 2c 0a 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 ,....x => { Ok(f
1d40: 6f 72 6d 61 74 21 28 22 7b 7d 20 70 6f 73 74 73 ormat!("{} posts
1d50: 20 70 75 72 67 65 64 2e 22 2c 20 78 29 2e 69 6e purged.", x).in
1d60: 74 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a to()) },...}..}.
1d70: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 65 ..pub async fn e
1d80: 6e 61 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 nable<S>(&self,
1d90: 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c source_id: &i32,
1da0: 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 owner: S) -> Re
1db0: 73 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 sult<&str>..wher
1dc0: 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e S: Into<i64> {
1dd0: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f ...let owner = o
1de0: 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 wner.into();....
1df0: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
1e00: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
1e10: 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 ().await.....wit
1e20: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
1e30: 6d 61 74 21 28 22 45 6e 61 62 6c 65 20 66 65 74 mat!("Enable fet
1e40: 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c ch conn:\n{:?}",
1e50: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a &self.pool))?;.
1e60: 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 ..match sqlx::qu
1e70: 65 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 ery("update rsst
1e80: 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 6e 61 g_source set ena
1e90: 62 6c 65 64 20 3d 20 74 72 75 65 20 77 68 65 72 bled = true wher
1ea0: 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 e source_id = $1
1eb0: 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 and owner = $2"
1ec0: 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 ).....bind(sourc
1ed0: 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f e_id).....bind(o
1ee0: 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 wner).....execut
1ef0: 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 e(&mut conn).awa
1f00: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
1f10: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1f20: 45 6e 61 62 6c 65 20 73 6f 75 72 63 65 3a 5c 6e Enable source:\n
1f30: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
1f40: 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 66 l))?.....rows_af
1f50: 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 31 20 fected() {....1
1f60: 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 => { Ok("Source
1f70: 65 6e 61 62 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 enabled.") },...
1f80: 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 .0 => { Ok("Sour
1f90: 63 65 20 6e 6f 74 20 66 6f 75 6e 64 2e 22 29 20 ce not found.")
1fa0: 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 45 72 72 },...._ => { Err
1fb0: 28 61 6e 79 68 6f 77 21 28 22 44 61 74 61 62 61 (anyhow!("Databa
1fc0: 73 65 20 65 72 72 6f 72 2e 22 29 29 20 7d 2c 0a se error.")) },.
1fd0: 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ..}..}...pub asy
1fe0: 6e 63 20 66 6e 20 64 69 73 61 62 6c 65 3c 53 3e nc fn disable<S>
1ff0: 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 (&self, source_i
2000: 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 d: &i32, owner:
2010: 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 S) -> Result<&st
2020: 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 r>..where S: Int
2030: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f o<i64> {...let o
2040: 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 wner = owner.int
2050: 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 o();....let mut
2060: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
2070: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
2080: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
2090: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 69 t(|| format!("Di
20a0: 73 61 62 6c 65 20 66 65 74 63 68 20 63 6f 6e 6e sable fetch conn
20b0: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
20c0: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 pool))?;...match
20d0: 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 sqlx::query("up
20e0: 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 date rsstg_sourc
20f0: 65 20 73 65 74 20 65 6e 61 62 6c 65 64 20 3d 20 e set enabled =
2100: 66 61 6c 73 65 20 77 68 65 72 65 20 73 6f 75 72 false where sour
2110: 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f ce_id = $1 and o
2120: 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 09 2e wner = $2").....
2130: 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a bind(source_id).
2140: 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a ....bind(owner).
2150: 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 ....execute(&mut
2160: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 conn).await....
2170: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
2180: 20 66 6f 72 6d 61 74 21 28 22 44 69 73 61 62 6c format!("Disabl
2190: 65 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 e source:\n{:?}"
21a0: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a , &self.pool))?.
21b0: 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 ....rows_affecte
21c0: 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 d() {....1 => {
21d0: 4f 6b 28 22 53 6f 75 72 63 65 20 64 69 73 61 62 Ok("Source disab
21e0: 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d led.") },....0 =
21f0: 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e > { Ok("Source n
2200: 6f 74 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 ot found.") },..
2210: 09 09 5f 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 .._ => { Err(any
2220: 68 6f 77 21 28 22 44 61 74 61 62 61 73 65 20 65 how!("Database e
2230: 72 72 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a rror.")) },...}.
2240: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
2250: 6e 20 75 70 64 61 74 65 3c 53 3e 28 26 73 65 6c n update<S>(&sel
2260: 66 2c 20 75 70 64 61 74 65 3a 20 4f 70 74 69 6f f, update: Optio
2270: 6e 3c 69 33 32 3e 2c 20 63 68 61 6e 6e 65 6c 3a n<i32>, channel:
2280: 20 26 73 74 72 2c 20 63 68 61 6e 6e 65 6c 5f 69 &str, channel_i
2290: 64 3a 20 69 36 34 2c 20 75 72 6c 3a 20 26 73 74 d: i64, url: &st
22a0: 72 2c 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 r, iv_hash: Opti
22b0: 6f 6e 3c 26 73 74 72 3e 2c 20 75 72 6c 5f 72 65 on<&str>, url_re
22c0: 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 : Option<&str>,
22d0: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
22e0: 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 ult<&str>..where
22f0: 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a S: Into<i64> {.
2300: 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 ..let owner = ow
2310: 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c ner.into();....l
2320: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
2330: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
2340: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
2350: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
2360: 61 74 21 28 22 55 70 64 61 74 65 20 66 65 74 63 at!("Update fetc
2370: 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 h conn:\n{:?}",
2380: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 0a &self.pool))?;..
2390: 09 09 6d 61 74 63 68 20 6d 61 74 63 68 20 75 70 ..match match up
23a0: 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 date {.....Some(
23b0: 69 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 id) => {......sq
23c0: 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 lx::query("updat
23d0: 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 e rsstg_source s
23e0: 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 20 3d 20 et channel_id =
23f0: 24 32 2c 20 75 72 6c 20 3d 20 24 33 2c 20 69 76 $2, url = $3, iv
2400: 5f 68 61 73 68 20 3d 20 24 34 2c 20 6f 77 6e 65 _hash = $4, owne
2410: 72 20 3d 20 24 35 2c 20 63 68 61 6e 6e 65 6c 20 r = $5, channel
2420: 3d 20 24 36 2c 20 75 72 6c 5f 72 65 20 3d 20 24 = $6, url_re = $
2430: 37 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 7 where source_i
2440: 64 20 3d 20 24 31 22 29 2e 62 69 6e 64 28 69 64 d = $1").bind(id
2450: 29 0a 09 09 09 09 7d 2c 0a 09 09 09 09 4e 6f 6e ).....},.....Non
2460: 65 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 e => {......sqlx
2470: 3a 3a 71 75 65 72 79 28 22 69 6e 73 65 72 74 20 ::query("insert
2480: 69 6e 74 6f 20 72 73 73 74 67 5f 73 6f 75 72 63 into rsstg_sourc
2490: 65 20 28 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 e (channel_id, u
24a0: 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e rl, iv_hash, own
24b0: 65 72 2c 20 63 68 61 6e 6e 65 6c 2c 20 75 72 6c er, channel, url
24c0: 5f 72 65 29 20 76 61 6c 75 65 73 20 28 24 31 2c _re) values ($1,
24d0: 20 24 32 2c 20 24 33 2c 20 24 34 2c 20 24 35 2c $2, $3, $4, $5,
24e0: 20 24 36 29 22 29 0a 09 09 09 09 7d 2c 0a 09 09 $6)").....},...
24f0: 09 7d 0a 09 09 09 2e 62 69 6e 64 28 63 68 61 6e .}.....bind(chan
2500: 6e 65 6c 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 nel_id).....bind
2510: 28 75 72 6c 29 0a 09 09 09 2e 62 69 6e 64 28 69 (url).....bind(i
2520: 76 5f 68 61 73 68 29 0a 09 09 09 2e 62 69 6e 64 v_hash).....bind
2530: 28 6f 77 6e 65 72 29 0a 09 09 09 2e 62 69 6e 64 (owner).....bind
2540: 28 63 68 61 6e 6e 65 6c 29 0a 09 09 09 2e 62 69 (channel).....bi
2550: 6e 64 28 75 72 6c 5f 72 65 29 0a 09 09 09 2e 65 nd(url_re).....e
2560: 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e xecute(&mut conn
2570: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 4f 6b 28 ).await {....Ok(
2580: 5f 29 20 3d 3e 20 4f 6b 28 6d 61 74 63 68 20 75 _) => Ok(match u
2590: 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 pdate {.....Some
25a0: 28 5f 29 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 (_) => "Channel
25b0: 75 70 64 61 74 65 64 2e 22 2c 0a 09 09 09 09 4e updated.",.....N
25c0: 6f 6e 65 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 one => "Channel
25d0: 61 64 64 65 64 2e 22 2c 0a 09 09 09 7d 29 2c 0a added.",....}),.
25e0: 09 09 09 45 72 72 28 73 71 6c 78 3a 3a 45 72 72 ...Err(sqlx::Err
25f0: 6f 72 3a 3a 44 61 74 61 62 61 73 65 28 65 72 72 or::Database(err
2600: 29 29 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 74 63 )) => {.....matc
2610: 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a 3a h err.downcast::
2620: 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a <sqlx::postgres:
2630: 3a 50 67 44 61 74 61 62 61 73 65 45 72 72 6f 72 :PgDatabaseError
2640: 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b 0a >().routine() {.
2650: 09 09 09 09 09 53 6f 6d 65 28 22 5f 62 74 5f 63 .....Some("_bt_c
2660: 68 65 63 6b 5f 75 6e 69 71 75 65 22 2c 20 29 20 heck_unique", )
2670: 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 44 => {.......Ok("D
2680: 75 70 6c 69 63 61 74 65 20 6b 65 79 2e 22 29 0a uplicate key.").
2690: 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 53 6f 6d .....},......Som
26a0: 65 28 5f 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 e(_) => {.......
26b0: 4f 6b 28 22 44 61 74 61 62 61 73 65 20 65 72 72 Ok("Database err
26c0: 6f 72 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 or.")......},...
26d0: 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 ...None => {....
26e0: 09 09 09 4f 6b 28 22 4e 6f 20 64 61 74 61 62 61 ...Ok("No databa
26f0: 73 65 20 65 72 72 6f 72 20 65 78 74 72 61 63 74 se error extract
2700: 65 64 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 ed.")......},...
2710: 09 09 7d 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 ..}....},....Err
2720: 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 62 (err) => {.....b
2730: 61 69 6c 21 28 22 53 6f 72 72 79 2c 20 75 6e 6b ail!("Sorry, unk
2740: 6e 6f 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 nown error:\n{:#
2750: 3f 7d 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 09 09 ?}\n", err);....
2760: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e },...}..}...asyn
2770: 63 20 66 6e 20 61 75 74 6f 66 65 74 63 68 28 26 c fn autofetch(&
2780: 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c self) -> Result<
2790: 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 std::time::Durat
27a0: 69 6f 6e 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 ion> {...let mut
27b0: 20 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a delay = chrono:
27c0: 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 :Duration::minut
27d0: 65 73 28 31 29 3b 0a 09 09 6c 65 74 20 6d 75 74 es(1);...let mut
27e0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
27f0: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
2800: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
2810: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 41 xt(|| format!("A
2820: 75 74 6f 66 65 74 63 68 20 66 65 74 63 68 20 63 utofetch fetch c
2830: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
2840: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6c 65 lf.pool))?;...le
2850: 74 20 6e 6f 77 20 3d 20 63 68 72 6f 6e 6f 3a 3a t now = chrono::
2860: 4c 6f 63 61 6c 3a 3a 6e 6f 77 28 29 3b 0a 09 09 Local::now();...
2870: 6c 65 74 20 6d 75 74 20 71 75 65 75 65 20 3d 20 let mut queue =
2880: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c sqlx::query("sel
2890: 65 63 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 6e ect source_id, n
28a0: 65 78 74 5f 66 65 74 63 68 2c 20 6f 77 6e 65 72 ext_fetch, owner
28b0: 20 66 72 6f 6d 20 72 73 73 74 67 5f 6f 72 64 65 from rsstg_orde
28c0: 72 20 6e 61 74 75 72 61 6c 20 6c 65 66 74 20 6a r natural left j
28d0: 6f 69 6e 20 72 73 73 74 67 5f 73 6f 75 72 63 65 oin rsstg_source
28e0: 20 77 68 65 72 65 20 6e 65 78 74 5f 66 65 74 63 where next_fetc
28f0: 68 20 3c 20 6e 6f 77 28 29 20 2b 20 69 6e 74 65 h < now() + inte
2900: 72 76 61 6c 20 27 31 20 6d 69 6e 75 74 65 27 3b rval '1 minute';
2910: 22 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c ").....fetch_all
2920: 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 (&mut conn).awai
2930: 74 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e t?;...for row in
2940: 20 71 75 65 75 65 2e 69 74 65 72 28 29 20 7b 0a queue.iter() {.
2950: 09 09 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 64 ...let source_id
2960: 3a 20 69 33 32 20 3d 20 72 6f 77 2e 74 72 79 5f : i32 = row.try_
2970: 67 65 74 28 22 73 6f 75 72 63 65 5f 69 64 22 29 get("source_id")
2980: 3f 3b 0a 09 09 09 6c 65 74 20 6f 77 6e 65 72 3a ?;....let owner:
2990: 20 69 36 34 20 3d 20 72 6f 77 2e 74 72 79 5f 67 i64 = row.try_g
29a0: 65 74 28 22 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 et("owner")?;...
29b0: 09 6c 65 74 20 6e 65 78 74 5f 66 65 74 63 68 3a .let next_fetch:
29c0: 20 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f DateTime<chrono
29d0: 3a 3a 4c 6f 63 61 6c 3e 20 3d 20 72 6f 77 2e 74 ::Local> = row.t
29e0: 72 79 5f 67 65 74 28 22 6e 65 78 74 5f 66 65 74 ry_get("next_fet
29f0: 63 68 22 29 3f 3b 0a 09 09 09 69 66 20 6e 65 78 ch")?;....if nex
2a00: 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a t_fetch < now {.
2a10: 09 09 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 ....let clone =
2a20: 43 6f 72 65 20 7b 0a 09 09 09 09 09 6f 77 6e 65 Core {......owne
2a30: 72 5f 63 68 61 74 3a 20 74 65 6c 65 67 72 61 6d r_chat: telegram
2a40: 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 _bot::UserId::ne
2a50: 77 28 6f 77 6e 65 72 29 2c 0a 09 09 09 09 09 2e w(owner),.......
2a60: 2e 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 0a 09 09 .self.clone()...
2a70: 09 09 7d 3b 0a 09 09 09 09 74 6f 6b 69 6f 3a 3a ..};.....tokio::
2a80: 73 70 61 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 spawn(async move
2a90: 20 7b 0a 09 09 09 09 09 69 66 20 6c 65 74 20 45 {......if let E
2aa0: 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e rr(err) = clone.
2ab0: 63 68 65 63 6b 28 26 73 6f 75 72 63 65 5f 69 64 check(&source_id
2ac0: 2c 20 6f 77 6e 65 72 2c 20 74 72 75 65 29 2e 61 , owner, true).a
2ad0: 77 61 69 74 20 7b 0a 09 09 09 09 09 09 69 66 20 wait {.......if
2ae0: 6c 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 let Err(err) = c
2af0: 6c 6f 6e 65 2e 73 65 6e 64 28 26 66 6f 72 6d 61 lone.send(&forma
2b00: 74 21 28 22 f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 t!("š {:?}",
2b10: 65 72 72 29 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 err), None, None
2b20: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 ).await {.......
2b30: 09 65 70 72 69 6e 74 6c 6e 21 28 22 43 68 65 63 .eprintln!("Chec
2b40: 6b 20 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 k error: {}", er
2b50: 72 29 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 r);.......};....
2b60: 09 09 7d 3b 0a 09 09 09 09 7d 29 3b 0a 09 09 09 ..};.....});....
2b70: 7d 20 65 6c 73 65 20 69 66 20 6e 65 78 74 5f 66 } else if next_f
2b80: 65 74 63 68 20 2d 20 6e 6f 77 20 3c 20 64 65 6c etch - now < del
2b90: 61 79 20 7b 0a 09 09 09 09 64 65 6c 61 79 20 3d ay {.....delay =
2ba0: 20 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f next_fetch - no
2bb0: 77 3b 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 71 w;....}...};...q
2bc0: 75 65 75 65 2e 63 6c 65 61 72 28 29 3b 0a 09 09 ueue.clear();...
2bd0: 4f 6b 28 64 65 6c 61 79 2e 74 6f 5f 73 74 64 28 Ok(delay.to_std(
2be0: 29 3f 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 )?)..}...pub asy
2bf0: 6e 63 20 66 6e 20 6c 69 73 74 3c 53 3e 28 26 73 nc fn list<S>(&s
2c00: 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d elf, owner: S) -
2c10: 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e > Result<String>
2c20: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
2c30: 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e i64> {...let own
2c40: 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 er = owner.into(
2c50: 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 72 65 );....let mut re
2c60: 70 6c 79 3a 20 56 65 63 3c 43 6f 77 3c 73 74 72 ply: Vec<Cow<str
2c70: 3e 3e 20 3d 20 76 65 63 21 5b 5d 3b 0a 09 09 6c >> = vec![];...l
2c80: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
2c90: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
2ca0: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
2cb0: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
2cc0: 61 74 21 28 22 4c 69 73 74 20 66 65 74 63 68 20 at!("List fetch
2cd0: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
2ce0: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 72 elf.pool))?;...r
2cf0: 65 70 6c 79 2e 70 75 73 68 28 22 43 68 61 6e 6e eply.push("Chann
2d00: 65 6c 73 3a 22 2e 69 6e 74 6f 28 29 29 3b 0a 09 els:".into());..
2d10: 09 6c 65 74 20 72 6f 77 73 20 3d 20 73 71 6c 78 .let rows = sqlx
2d20: 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 ::query("select
2d30: 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e source_id, chann
2d40: 65 6c 2c 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c el, enabled, url
2d50: 2c 20 69 76 5f 68 61 73 68 2c 20 75 72 6c 5f 72 , iv_hash, url_r
2d60: 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 e from rsstg_sou
2d70: 72 63 65 20 77 68 65 72 65 20 6f 77 6e 65 72 20 rce where owner
2d80: 3d 20 24 31 20 6f 72 64 65 72 20 62 79 20 73 6f = $1 order by so
2d90: 75 72 63 65 5f 69 64 22 29 0a 09 09 09 2e 62 69 urce_id").....bi
2da0: 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 66 65 nd(owner).....fe
2db0: 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 6f 6e tch_all(&mut con
2dc0: 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 66 6f 72 n).await?;...for
2dd0: 20 72 6f 77 20 69 6e 20 72 6f 77 73 2e 69 74 65 row in rows.ite
2de0: 72 28 29 20 7b 0a 09 09 09 6c 65 74 20 73 6f 75 r() {....let sou
2df0: 72 63 65 5f 69 64 3a 20 69 33 32 20 3d 20 72 6f rce_id: i32 = ro
2e00: 77 2e 74 72 79 5f 67 65 74 28 22 73 6f 75 72 63 w.try_get("sourc
2e10: 65 5f 69 64 22 29 3f 3b 0a 09 09 09 6c 65 74 20 e_id")?;....let
2e20: 75 73 65 72 6e 61 6d 65 3a 20 26 73 74 72 20 3d username: &str =
2e30: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 63 68 row.try_get("ch
2e40: 61 6e 6e 65 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 annel")?;....let
2e50: 20 65 6e 61 62 6c 65 64 3a 20 62 6f 6f 6c 20 3d enabled: bool =
2e60: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 65 6e row.try_get("en
2e70: 61 62 6c 65 64 22 29 3f 3b 0a 09 09 09 6c 65 74 abled")?;....let
2e80: 20 75 72 6c 3a 20 26 73 74 72 20 3d 20 72 6f 77 url: &str = row
2e90: 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 22 29 3f .try_get("url")?
2ea0: 3b 0a 09 09 09 6c 65 74 20 69 76 5f 68 61 73 68 ;....let iv_hash
2eb0: 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 20 3d : Option<&str> =
2ec0: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 69 76 row.try_get("iv
2ed0: 5f 68 61 73 68 22 29 3f 3b 0a 09 09 09 6c 65 74 _hash")?;....let
2ee0: 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f 6e 3c url_re: Option<
2ef0: 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f &str> = row.try_
2f00: 67 65 74 28 22 75 72 6c 5f 72 65 22 29 3f 3b 0a get("url_re")?;.
2f10: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f ...reply.push(fo
2f20: 72 6d 61 74 21 28 22 5c 6e 5c 5c 23 ef b8 8f e2 rmat!("\n\\#ļøā
2f30: 83 a3 20 7b 7d 20 5c 5c 2a ef b8 8f e2 83 a3 20 £ {} \\*ļøā£
2f40: 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 97 20 60 7b `{}` {}\nš `{
2f50: 7d 60 22 2c 20 73 6f 75 72 63 65 5f 69 64 2c 20 }`", source_id,
2f60: 75 73 65 72 6e 61 6d 65 2c 20 20 0a 09 09 09 09 username, .....
2f70: 6d 61 74 63 68 20 65 6e 61 62 6c 65 64 20 7b 0a match enabled {.
2f80: 09 09 09 09 09 74 72 75 65 20 20 3d 3e 20 22 f0 .....true => "š
2f90: 9f 94 84 20 65 6e 61 62 6c 65 64 22 2c 0a 09 09 enabled",...
2fa0: 09 09 09 66 61 6c 73 65 20 3d 3e 20 22 e2 9b 94 ...false => "ā
2fb0: 20 64 69 73 61 62 6c 65 64 22 2c 0a 09 09 09 09 disabled",.....
2fc0: 7d 2c 20 75 72 6c 29 2e 69 6e 74 6f 28 29 29 3b }, url).into());
2fd0: 0a 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 ....if let Some(
2fe0: 68 61 73 68 29 20 3d 20 69 76 5f 68 61 73 68 20 hash) = iv_hash
2ff0: 7b 0a 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 {.....reply.push
3000: 28 66 6f 72 6d 61 74 21 28 22 49 56 3a 20 60 7b (format!("IV: `{
3010: 7d 60 22 2c 20 68 61 73 68 29 2e 69 6e 74 6f 28 }`", hash).into(
3020: 29 29 3b 0a 09 09 09 7d 0a 09 09 09 69 66 20 6c ));....}....if l
3030: 65 74 20 53 6f 6d 65 28 72 65 29 20 3d 20 75 72 et Some(re) = ur
3040: 6c 5f 72 65 20 7b 0a 09 09 09 09 72 65 70 6c 79 l_re {.....reply
3050: 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 52 .push(format!("R
3060: 45 3a 20 60 7b 7d 60 22 2c 20 72 65 29 2e 69 6e E: `{}`", re).in
3070: 74 6f 28 29 29 3b 0a 09 09 09 7d 0a 09 09 7d 3b to());....}...};
3080: 0a 09 09 4f 6b 28 72 65 70 6c 79 2e 6a 6f 69 6e ...Ok(reply.join
3090: 28 22 5c 6e 22 29 29 0a 09 7d 0a 7d 0a ("\n"))..}.}.