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 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 .if let Err(err)
0490: 20 3d 20 26 63 6c 6f 6e 65 2e 61 75 74 6f 66 65 = &clone.autofe
04a0: 74 63 68 28 29 2e 61 77 61 69 74 20 7b 0a 09 09 tch().await {...
04b0: 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 ..if let Err(err
04c0: 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 26 ) = clone.send(&
04d0: 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a format!("š {:
04e0: 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 2c ?}", err), None,
04f0: 20 4e 6f 6e 65 29 20 7b 0a 09 09 09 09 09 65 70 None) {......ep
0500: 72 69 6e 74 6c 6e 21 28 22 41 75 74 6f 66 65 74 rintln!("Autofet
0510: 63 68 20 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 ch error: {}", e
0520: 72 72 29 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 7d rr);.....};....}
0530: 0a 09 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 ...});...Ok(core
0540: 29 0a 09 7d 0a 0a 09 70 75 62 20 66 6e 20 73 74 )..}...pub fn st
0550: 72 65 61 6d 28 26 73 65 6c 66 29 20 2d 3e 20 74 ream(&self) -> t
0560: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 70 64 elegram_bot::Upd
0570: 61 74 65 73 53 74 72 65 61 6d 20 7b 0a 09 09 73 atesStream {...s
0580: 65 6c 66 2e 74 67 2e 73 74 72 65 61 6d 28 29 0a elf.tg.stream().
0590: 09 7d 0a 0a 09 70 75 62 20 66 6e 20 73 65 6e 64 .}...pub fn send
05a0: 3c 27 61 2c 20 53 3e 28 26 73 65 6c 66 2c 20 6d <'a, S>(&self, m
05b0: 73 67 3a 20 53 2c 20 74 61 72 67 65 74 3a 20 4f sg: S, target: O
05c0: 70 74 69 6f 6e 3c 74 65 6c 65 67 72 61 6d 5f 62 ption<telegram_b
05d0: 6f 74 3a 3a 55 73 65 72 49 64 3e 2c 20 70 61 72 ot::UserId>, par
05e0: 73 65 5f 6d 6f 64 65 3a 20 4f 70 74 69 6f 6e 3c se_mode: Option<
05f0: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 telegram_bot::ty
0600: 70 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3e 29 pes::ParseMode>)
0610: 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 0a 09 -> Result<()>..
0620: 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 43 6f where S: Into<Co
0630: 77 3c 27 61 2c 20 73 74 72 3e 3e 20 7b 0a 09 09 w<'a, str>> {...
0640: 6c 65 74 20 6d 73 67 20 3d 20 6d 73 67 2e 69 6e let msg = msg.in
0650: 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 70 61 72 to();....let par
0660: 73 65 5f 6d 6f 64 65 20 3d 20 6d 61 74 63 68 20 se_mode = match
0670: 70 61 72 73 65 5f 6d 6f 64 65 20 7b 0a 09 09 09 parse_mode {....
0680: 53 6f 6d 65 28 6d 6f 64 65 29 20 3d 3e 20 6d 6f Some(mode) => mo
0690: 64 65 2c 0a 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 de,....None => t
06a0: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 elegram_bot::typ
06b0: 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 es::ParseMode::H
06c0: 74 6d 6c 2c 0a 09 09 7d 3b 0a 09 09 73 65 6c 66 tml,...};...self
06d0: 2e 74 67 2e 73 70 61 77 6e 28 74 65 6c 65 67 72 .tg.spawn(telegr
06e0: 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 73 am_bot::SendMess
06f0: 61 67 65 3a 3a 6e 65 77 28 6d 61 74 63 68 20 74 age::new(match t
0700: 61 72 67 65 74 20 7b 0a 09 09 09 53 6f 6d 65 28 arget {....Some(
0710: 75 73 65 72 29 20 3d 3e 20 75 73 65 72 2c 0a 09 user) => user,..
0720: 09 09 4e 6f 6e 65 20 3d 3e 20 73 65 6c 66 2e 6f ..None => self.o
0730: 77 6e 65 72 5f 63 68 61 74 2c 0a 09 09 7d 2c 20 wner_chat,...},
0740: 6d 73 67 29 2e 70 61 72 73 65 5f 6d 6f 64 65 28 msg).parse_mode(
0750: 70 61 72 73 65 5f 6d 6f 64 65 29 29 3b 0a 09 09 parse_mode));...
0760: 4f 6b 28 28 29 29 0a 09 7d 0a 0a 09 70 75 62 20 Ok(())..}...pub
0770: 61 73 79 6e 63 20 66 6e 20 63 68 65 63 6b 3c 53 async fn check<S
0780: 3e 28 26 73 65 6c 66 2c 20 69 64 3a 20 26 69 33 >(&self, id: &i3
0790: 32 2c 20 6f 77 6e 65 72 3a 20 53 2c 20 72 65 61 2, owner: S, rea
07a0: 6c 3a 20 62 6f 6f 6c 29 20 2d 3e 20 52 65 73 75 l: bool) -> Resu
07b0: 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e lt<Cow<'_, str>>
07c0: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
07d0: 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e i64> {...let own
07e0: 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 er = owner.into(
07f0: 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 70 6f );....let mut po
0800: 73 74 65 64 3a 20 69 33 32 20 3d 20 30 3b 0a 09 sted: i32 = 0;..
0810: 09 6c 65 74 20 69 64 20 3d 20 7b 0a 09 09 09 6c .let id = {....l
0820: 65 74 20 6d 75 74 20 73 65 74 20 3d 20 73 65 6c et mut set = sel
0830: 66 2e 73 6f 75 72 63 65 73 2e 6c 6f 63 6b 28 29 f.sources.lock()
0840: 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 6d 61 .unwrap();....ma
0850: 74 63 68 20 73 65 74 2e 67 65 74 28 69 64 29 20 tch set.get(id)
0860: 7b 0a 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d {.....Some(id) =
0870: 3e 20 69 64 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 > id.clone(),...
0880: 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 ..None => {.....
0890: 09 6c 65 74 20 69 64 20 3d 20 41 72 63 3a 3a 6e .let id = Arc::n
08a0: 65 77 28 2a 69 64 29 3b 0a 09 09 09 09 09 73 65 ew(*id);......se
08b0: 74 2e 69 6e 73 65 72 74 28 69 64 2e 63 6c 6f 6e t.insert(id.clon
08c0: 65 28 29 29 3b 0a 09 09 09 09 09 69 64 2e 63 6c e());......id.cl
08d0: 6f 6e 65 28 29 0a 09 09 09 09 7d 2c 0a 09 09 09 one().....},....
08e0: 7d 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 63 6f 75 }...};...let cou
08f0: 6e 74 20 3d 20 41 72 63 3a 3a 73 74 72 6f 6e 67 nt = Arc::strong
0900: 5f 63 6f 75 6e 74 28 26 69 64 29 3b 0a 09 09 69 _count(&id);...i
0910: 66 20 63 6f 75 6e 74 20 3d 3d 20 32 20 7b 0a 09 f count == 2 {..
0920: 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d ..let mut conn =
0930: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
0940: 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 09 2e re().await......
0950: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
0960: 66 6f 72 6d 61 74 21 28 22 51 75 65 72 79 20 71 format!("Query q
0970: 75 65 75 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a ueue fetch conn:
0980: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
0990: 6f 6f 6c 29 29 3f 3b 0a 09 09 09 6c 65 74 20 72 ool))?;....let r
09a0: 6f 77 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 ow = sqlx::query
09b0: 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 65 5f ("select source_
09c0: 69 64 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 id, channel_id,
09d0: 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 url, iv_hash, ow
09e0: 6e 65 72 2c 20 75 72 6c 5f 72 65 20 66 72 6f 6d ner, url_re from
09f0: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 rsstg_source wh
0a00: 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 ere source_id =
0a10: 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 $1 and owner = $
0a20: 32 22 29 0a 09 09 09 09 2e 62 69 6e 64 28 2a 69 2")......bind(*i
0a30: 64 29 0a 09 09 09 09 2e 62 69 6e 64 28 6f 77 6e d)......bind(own
0a40: 65 72 29 0a 09 09 09 09 2e 66 65 74 63 68 5f 6f er)......fetch_o
0a50: 6e 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 ne(&mut conn).aw
0a60: 61 69 74 0a 09 09 09 09 2e 77 69 74 68 5f 63 6f ait......with_co
0a70: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
0a80: 28 22 51 75 65 72 79 20 73 6f 75 72 63 65 3a 5c ("Query source:\
0a90: 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f n{:?}", &self.po
0aa0: 6f 6c 29 29 3f 3b 0a 09 09 09 64 72 6f 70 28 63 ol))?;....drop(c
0ab0: 6f 6e 6e 29 3b 0a 09 09 09 6c 65 74 20 63 68 61 onn);....let cha
0ac0: 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 20 3d 20 72 nnel_id: i64 = r
0ad0: 6f 77 2e 74 72 79 5f 67 65 74 28 22 63 68 61 6e ow.try_get("chan
0ae0: 6e 65 6c 5f 69 64 22 29 3f 3b 0a 09 09 09 6c 65 nel_id")?;....le
0af0: 74 20 75 72 6c 3a 20 26 73 74 72 20 3d 20 72 6f t url: &str = ro
0b00: 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 22 29 w.try_get("url")
0b10: 3f 3b 0a 09 09 09 6c 65 74 20 69 76 5f 68 61 73 ?;....let iv_has
0b20: 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 20 h: Option<&str>
0b30: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 69 = row.try_get("i
0b40: 76 5f 68 61 73 68 22 29 3f 3b 0a 09 09 09 6c 65 v_hash")?;....le
0b50: 74 20 75 72 6c 5f 72 65 20 3d 20 6d 61 74 63 68 t url_re = match
0b60: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 72 row.try_get("ur
0b70: 6c 5f 72 65 22 29 3f 20 7b 0a 09 09 09 09 53 6f l_re")? {.....So
0b80: 6d 65 28 78 29 20 3d 3e 20 53 6f 6d 65 28 73 65 me(x) => Some(se
0b90: 64 72 65 67 65 78 3a 3a 52 65 70 6c 61 63 65 43 dregex::ReplaceC
0ba0: 6f 6d 6d 61 6e 64 3a 3a 6e 65 77 28 78 29 3f 29 ommand::new(x)?)
0bb0: 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 4e 6f ,.....None => No
0bc0: 6e 65 2c 0a 09 09 09 7d 3b 0a 09 09 09 6c 65 74 ne,....};....let
0bd0: 20 64 65 73 74 69 6e 61 74 69 6f 6e 20 3d 20 6d destination = m
0be0: 61 74 63 68 20 72 65 61 6c 20 7b 0a 09 09 09 09 atch real {.....
0bf0: 74 72 75 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d true => telegram
0c00: 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 _bot::UserId::ne
0c10: 77 28 63 68 61 6e 6e 65 6c 5f 69 64 29 2c 0a 09 w(channel_id),..
0c20: 09 09 09 66 61 6c 73 65 20 3d 3e 20 74 65 6c 65 ...false => tele
0c30: 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 gram_bot::UserId
0c40: 3a 3a 6e 65 77 28 72 6f 77 2e 74 72 79 5f 67 65 ::new(row.try_ge
0c50: 74 28 22 6f 77 6e 65 72 22 29 3f 29 2c 0a 09 09 t("owner")?),...
0c60: 09 7d 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 74 .};....let mut t
0c70: 68 69 73 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f his_fetch: Optio
0c80: 6e 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e n<DateTime<chron
0c90: 6f 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 3e o::FixedOffset>>
0ca0: 20 3d 20 4e 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 = None;....let
0cb0: 6d 75 74 20 70 6f 73 74 73 3a 20 42 54 72 65 65 mut posts: BTree
0cc0: 4d 61 70 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 Map<DateTime<chr
0cd0: 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 ono::FixedOffset
0ce0: 3e 2c 20 53 74 72 69 6e 67 3e 20 3d 20 42 54 72 >, String> = BTr
0cf0: 65 65 4d 61 70 3a 3a 6e 65 77 28 29 3b 0a 09 09 eeMap::new();...
0d00: 09 6c 65 74 20 72 65 73 70 6f 6e 73 65 20 3d 20 .let response =
0d10: 72 65 71 77 65 73 74 3a 3a 67 65 74 28 75 72 6c reqwest::get(url
0d20: 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 6c 65 74 ).await?;....let
0d30: 20 73 74 61 74 75 73 20 3d 20 72 65 73 70 6f 6e status = respon
0d40: 73 65 2e 73 74 61 74 75 73 28 29 3b 0a 09 09 09 se.status();....
0d50: 6c 65 74 20 63 6f 6e 74 65 6e 74 20 3d 20 72 65 let content = re
0d60: 73 70 6f 6e 73 65 2e 62 79 74 65 73 28 29 2e 61 sponse.bytes().a
0d70: 77 61 69 74 3f 3b 0a 09 09 09 6d 61 74 63 68 20 wait?;....match
0d80: 72 73 73 3a 3a 43 68 61 6e 6e 65 6c 3a 3a 72 65 rss::Channel::re
0d90: 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 ad_from(&content
0da0: 5b 2e 2e 5d 29 20 7b 0a 09 09 09 09 4f 6b 28 66 [..]) {.....Ok(f
0db0: 65 65 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 66 eed) => {......f
0dc0: 6f 72 20 69 74 65 6d 20 69 6e 20 66 65 65 64 2e or item in feed.
0dd0: 69 74 65 6d 73 28 29 20 7b 0a 09 09 09 09 09 09 items() {.......
0de0: 69 66 20 6c 65 74 20 53 6f 6d 65 28 6c 69 6e 6b if let Some(link
0df0: 29 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 28 29 20 ) = item.link()
0e00: 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 64 61 74 {........let dat
0e10: 65 20 3d 20 6d 61 74 63 68 20 69 74 65 6d 2e 70 e = match item.p
0e20: 75 62 5f 64 61 74 65 28 29 20 7b 0a 09 09 09 09 ub_date() {.....
0e30: 09 09 09 09 53 6f 6d 65 28 66 65 65 64 5f 64 61 ....Some(feed_da
0e40: 74 65 29 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a te) => DateTime:
0e50: 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 32 :parse_from_rfc2
0e60: 38 32 32 28 66 65 65 64 5f 64 61 74 65 29 2c 0a 822(feed_date),.
0e70: 09 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 ........None =>
0e80: 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f DateTime::parse_
0e90: 66 72 6f 6d 5f 72 66 63 33 33 33 39 28 26 69 74 from_rfc3339(&it
0ea0: 65 6d 2e 64 75 62 6c 69 6e 5f 63 6f 72 65 5f 65 em.dublin_core_e
0eb0: 78 74 28 29 2e 75 6e 77 72 61 70 28 29 2e 64 61 xt().unwrap().da
0ec0: 74 65 73 28 29 5b 30 5d 29 2c 0a 09 09 09 09 09 tes()[0]),......
0ed0: 09 09 7d 3f 3b 0a 09 09 09 09 09 09 09 6c 65 74 ..}?;........let
0ee0: 20 75 72 6c 20 3d 20 6c 69 6e 6b 3b 0a 09 09 09 url = link;....
0ef0: 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 ....posts.insert
0f00: 28 64 61 74 65 2c 20 75 72 6c 2e 69 6e 74 6f 28 (date, url.into(
0f10: 29 29 3b 0a 09 09 09 09 09 09 7d 0a 09 09 09 09 ));.......}.....
0f20: 09 7d 3b 0a 09 09 09 09 7d 2c 0a 09 09 09 09 45 .};.....},.....E
0f30: 72 72 28 65 72 72 29 20 3d 3e 20 6d 61 74 63 68 rr(err) => match
0f40: 20 65 72 72 20 7b 0a 09 09 09 09 09 72 73 73 3a err {......rss:
0f50: 3a 45 72 72 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 :Error::InvalidS
0f60: 74 61 72 74 54 61 67 20 3d 3e 20 7b 0a 09 09 09 tartTag => {....
0f70: 09 09 09 6c 65 74 20 66 65 65 64 20 3d 20 61 74 ...let feed = at
0f80: 6f 6d 5f 73 79 6e 64 69 63 61 74 69 6f 6e 3a 3a om_syndication::
0f90: 46 65 65 64 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 Feed::read_from(
0fa0: 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 0a 09 09 &content[..])...
0fb0: 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 ......with_conte
0fc0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 50 xt(|| format!("P
0fd0: 72 6f 62 6c 65 6d 20 6f 70 65 6e 69 6e 67 20 66 roblem opening f
0fe0: 65 65 64 20 75 72 6c 3a 5c 6e 7b 7d 5c 6e 7b 7d eed url:\n{}\n{}
0ff0: 22 2c 20 26 75 72 6c 2c 20 73 74 61 74 75 73 29 ", &url, status)
1000: 29 3f 3b 0a 09 09 09 09 09 09 66 6f 72 20 69 74 )?;.......for it
1010: 65 6d 20 69 6e 20 66 65 65 64 2e 65 6e 74 72 69 em in feed.entri
1020: 65 73 28 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 es() {........le
1030: 74 20 64 61 74 65 20 3d 20 69 74 65 6d 2e 70 75 t date = item.pu
1040: 62 6c 69 73 68 65 64 28 29 2e 75 6e 77 72 61 70 blished().unwrap
1050: 28 29 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 75 ();........let u
1060: 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 rl = item.links(
1070: 29 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 09 )[0].href();....
1080: 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 ....posts.insert
1090: 28 2a 64 61 74 65 2c 20 75 72 6c 2e 69 6e 74 6f (*date, url.into
10a0: 28 29 29 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 ());.......};...
10b0: 09 09 09 7d 2c 0a 09 09 09 09 09 72 73 73 3a 3a ...},......rss::
10c0: 45 72 72 6f 72 3a 3a 45 6f 66 20 3d 3e 20 28 29 Error::Eof => ()
10d0: 2c 0a 09 09 09 09 09 5f 20 3d 3e 20 62 61 69 6c ,......_ => bail
10e0: 21 28 22 55 6e 73 75 70 70 6f 72 74 65 64 20 6f !("Unsupported o
10f0: 72 20 6d 61 6e 67 6c 65 64 20 63 6f 6e 74 65 6e r mangled conten
1100: 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b 3a 23 3f 7d 5c t:\n{:?}\n{:#?}\
1110: 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 26 75 72 6c 2c n{:#?}\n", &url,
1120: 20 65 72 72 2c 20 73 74 61 74 75 73 29 0a 09 09 err, status)...
1130: 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 66 6f 72 ..}....};....for
1140: 20 28 64 61 74 65 2c 20 75 72 6c 29 20 69 6e 20 (date, url) in
1150: 70 6f 73 74 73 2e 69 74 65 72 28 29 20 7b 0a 09 posts.iter() {..
1160: 09 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
1170: 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 = self.pool.acqu
1180: 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 09 ire().await.....
1190: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
11a0: 7c 20 66 6f 72 6d 61 74 21 28 22 43 68 65 63 6b | format!("Check
11b0: 20 70 6f 73 74 20 66 65 74 63 68 20 63 6f 6e 6e post fetch conn
11c0: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
11d0: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 09 6c 65 74 pool))?;.....let
11e0: 20 70 6f 73 74 5f 75 72 6c 3a 20 43 6f 77 3c 73 post_url: Cow<s
11f0: 74 72 3e 20 3d 20 6d 61 74 63 68 20 75 72 6c 5f tr> = match url_
1200: 72 65 20 7b 0a 09 09 09 09 09 53 6f 6d 65 28 72 re {......Some(r
1210: 65 66 20 78 29 20 3d 3e 20 78 2e 65 78 65 63 75 ef x) => x.execu
1220: 74 65 28 75 72 6c 29 2c 0a 09 09 09 09 09 4e 6f te(url),......No
1230: 6e 65 20 3d 3e 20 75 72 6c 2e 69 6e 74 6f 28 29 ne => url.into()
1240: 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 6c 65 74 ,.....};.....let
1250: 20 72 6f 77 20 3d 20 73 71 6c 78 3a 3a 71 75 65 row = sqlx::que
1260: 72 79 28 22 73 65 6c 65 63 74 20 65 78 69 73 74 ry("select exist
1270: 73 28 73 65 6c 65 63 74 20 74 72 75 65 20 66 72 s(select true fr
1280: 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 77 68 om rsstg_post wh
1290: 65 72 65 20 75 72 6c 20 3d 20 24 31 20 61 6e 64 ere url = $1 and
12a0: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 32 29 source_id = $2)
12b0: 20 61 73 20 65 78 69 73 74 73 3b 22 29 0a 09 09 as exists;")...
12c0: 09 09 09 2e 62 69 6e 64 28 26 2a 70 6f 73 74 5f ....bind(&*post_
12d0: 75 72 6c 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 url).......bind(
12e0: 2a 69 64 29 0a 09 09 09 09 09 2e 66 65 74 63 68 *id).......fetch
12f0: 5f 6f 6e 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e _one(&mut conn).
1300: 61 77 61 69 74 0a 09 09 09 09 09 2e 77 69 74 68 await.......with
1310: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
1320: 61 74 21 28 22 43 68 65 63 6b 20 70 6f 73 74 3a at!("Check post:
1330: 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 \n{:?}", &conn))
1340: 3f 3b 0a 09 09 09 09 6c 65 74 20 65 78 69 73 74 ?;.....let exist
1350: 73 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 s: bool = row.tr
1360: 79 5f 67 65 74 28 22 65 78 69 73 74 73 22 29 3f y_get("exists")?
1370: 3b 0a 09 09 09 09 69 66 20 21 20 65 78 69 73 74 ;.....if ! exist
1380: 73 20 7b 0a 09 09 09 09 09 69 66 20 74 68 69 73 s {......if this
1390: 5f 66 65 74 63 68 20 3d 3d 20 4e 6f 6e 65 20 7c _fetch == None |
13a0: 7c 20 2a 64 61 74 65 20 3e 20 74 68 69 73 5f 66 | *date > this_f
13b0: 65 74 63 68 2e 75 6e 77 72 61 70 28 29 20 7b 0a etch.unwrap() {.
13c0: 09 09 09 09 09 09 74 68 69 73 5f 66 65 74 63 68 ......this_fetch
13d0: 20 3d 20 53 6f 6d 65 28 2a 64 61 74 65 29 3b 0a = Some(*date);.
13e0: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 73 65 6c .....};......sel
13f0: 66 2e 74 67 2e 73 65 6e 64 28 20 6d 61 74 63 68 f.tg.send( match
1400: 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 iv_hash {......
1410: 09 09 53 6f 6d 65 28 68 61 73 68 29 20 3d 3e 20 ..Some(hash) =>
1420: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 telegram_bot::Se
1430: 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 ndMessage::new(d
1440: 65 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d estination, form
1450: 61 74 21 28 22 3c 61 20 68 72 65 66 3d 5c 22 68 at!("<a href=\"h
1460: 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 ttps://t.me/iv?u
1470: 72 6c 3d 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c 22 rl={}&rhash={}\"
1480: 3e 20 3c 2f 61 3e 7b 30 7d 22 2c 20 26 70 6f 73 > </a>{0}", &pos
1490: 74 5f 75 72 6c 2c 20 68 61 73 68 29 29 2c 0a 09 t_url, hash)),..
14a0: 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 65 ......None => te
14b0: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 legram_bot::Send
14c0: 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 73 Message::new(des
14d0: 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 tination, format
14e0: 21 28 22 7b 7d 22 2c 20 70 6f 73 74 5f 75 72 6c !("{}", post_url
14f0: 29 29 2c 0a 09 09 09 09 09 09 7d 2e 70 61 72 73 )),.......}.pars
1500: 65 5f 6d 6f 64 65 28 74 65 6c 65 67 72 61 6d 5f e_mode(telegram_
1510: 62 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 61 72 73 bot::types::Pars
1520: 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 2e 61 77 eMode::Html)).aw
1530: 61 69 74 0a 09 09 09 09 09 09 2e 63 6f 6e 74 65 ait........conte
1540: 78 74 28 22 43 61 6e 27 74 20 70 6f 73 74 20 6d xt("Can't post m
1550: 65 73 73 61 67 65 3a 22 29 3f 3b 0a 09 09 09 09 essage:")?;.....
1560: 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 69 6e .sqlx::query("in
1570: 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 67 5f sert into rsstg_
1580: 70 6f 73 74 20 28 73 6f 75 72 63 65 5f 69 64 2c post (source_id,
1590: 20 70 6f 73 74 65 64 2c 20 75 72 6c 29 20 76 61 posted, url) va
15a0: 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 24 33 lues ($1, $2, $3
15b0: 29 3b 22 29 0a 09 09 09 09 09 09 2e 62 69 6e 64 );")........bind
15c0: 28 2a 69 64 29 0a 09 09 09 09 09 09 2e 62 69 6e (*id)........bin
15d0: 64 28 64 61 74 65 29 0a 09 09 09 09 09 09 2e 62 d(date)........b
15e0: 69 6e 64 28 26 2a 70 6f 73 74 5f 75 72 6c 29 0a ind(&*post_url).
15f0: 09 09 09 09 09 09 2e 65 78 65 63 75 74 65 28 26 .......execute(&
1600: 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a mut conn).await.
1610: 09 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 .......with_cont
1620: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1630: 52 65 63 6f 72 64 20 70 6f 73 74 3a 5c 6e 7b 3a Record post:\n{:
1640: 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f 3b 0a 09 ?}", &conn))?;..
1650: 09 09 09 09 64 72 6f 70 28 63 6f 6e 6e 29 3b 0a ....drop(conn);.
1660: 09 09 09 09 09 74 6f 6b 69 6f 3a 3a 74 69 6d 65 .....tokio::time
1670: 3a 3a 73 6c 65 65 70 28 73 74 64 3a 3a 74 69 6d ::sleep(std::tim
1680: 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 e::Duration::new
1690: 28 34 2c 20 30 29 29 2e 61 77 61 69 74 3b 0a 09 (4, 0)).await;..
16a0: 09 09 09 7d 3b 0a 09 09 09 09 70 6f 73 74 65 64 ...};.....posted
16b0: 20 2b 3d 20 31 3b 0a 09 09 09 7d 3b 0a 09 09 09 += 1;....};....
16c0: 70 6f 73 74 73 2e 63 6c 65 61 72 28 29 3b 0a 09 posts.clear();..
16d0: 09 7d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f .};...let mut co
16e0: 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 nn = self.pool.a
16f0: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 cquire().await..
1700: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1710: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 || format!("Upda
1720: 74 65 20 73 63 72 61 70 65 20 66 65 74 63 68 20 te scrape fetch
1730: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
1740: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 73 elf.pool))?;...s
1750: 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 qlx::query("upda
1760: 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 te rsstg_source
1770: 73 65 74 20 6c 61 73 74 5f 73 63 72 61 70 65 20 set last_scrape
1780: 3d 20 6e 6f 77 28 29 20 77 68 65 72 65 20 73 6f = now() where so
1790: 75 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a urce_id = $1;").
17a0: 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 ....bind(*id)...
17b0: 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 ..execute(&mut c
17c0: 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 onn).await.....w
17d0: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
17e0: 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 73 ormat!("Update s
17f0: 63 72 61 70 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 crape:\n{:?}", &
1800: 63 6f 6e 6e 29 29 3f 3b 0a 09 09 4f 6b 28 66 6f conn))?;...Ok(fo
1810: 72 6d 61 74 21 28 22 50 6f 73 74 65 64 3a 20 7b rmat!("Posted: {
1820: 7d 22 2c 20 26 70 6f 73 74 65 64 29 2e 69 6e 74 }", &posted).int
1830: 6f 28 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 o())..}...pub as
1840: 79 6e 63 20 66 6e 20 64 65 6c 65 74 65 3c 53 3e ync fn delete<S>
1850: 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 (&self, source_i
1860: 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 d: &i32, owner:
1870: 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 S) -> Result<Cow
1880: 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 <'_, str>>..wher
1890: 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e S: Into<i64> {
18a0: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f ...let owner = o
18b0: 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 wner.into();....
18c0: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
18d0: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
18e0: 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 ().await.....wit
18f0: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
1900: 6d 61 74 21 28 22 44 65 6c 65 74 65 20 66 65 74 mat!("Delete fet
1910: 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c ch conn:\n{:?}",
1920: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a &self.pool))?;.
1930: 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 ..match sqlx::qu
1940: 65 72 79 28 22 64 65 6c 65 74 65 20 66 72 6f 6d ery("delete from
1950: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 rsstg_source wh
1960: 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 ere source_id =
1970: 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 $1 and owner = $
1980: 32 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 2;").....bind(so
1990: 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e urce_id).....bin
19a0: 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 d(owner).....exe
19b0: 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e cute(&mut conn).
19c0: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
19d0: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
19e0: 21 28 22 44 65 6c 65 74 65 20 73 6f 75 72 63 65 !("Delete source
19f0: 20 72 75 6c 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 rule:\n{:?}", &
1a00: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 self.pool))?....
1a10: 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 .rows_affected()
1a20: 20 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 {....0 => { Ok(
1a30: 22 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 "No data found f
1a40: 6f 75 6e 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d ound.".into()) }
1a50: 2c 0a 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 ,....x => { Ok(f
1a60: 6f 72 6d 61 74 21 28 22 7b 7d 20 73 6f 75 72 63 ormat!("{} sourc
1a70: 65 73 20 72 65 6d 6f 76 65 64 2e 22 2c 20 78 29 es removed.", x)
1a80: 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a .into()) },...}.
1a90: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
1aa0: 6e 20 63 6c 65 61 6e 3c 53 3e 28 26 73 65 6c 66 n clean<S>(&self
1ab0: 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 , source_id: &i3
1ac0: 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 2, owner: S) ->
1ad0: 52 65 73 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 Result<Cow<'_, s
1ae0: 74 72 3e 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 tr>>..where S: I
1af0: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
1b00: 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 owner = owner.i
1b10: 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 nto();....let mu
1b20: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
1b30: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
1b40: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
1b50: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1b60: 43 6c 65 61 6e 20 66 65 74 63 68 20 63 6f 6e 6e Clean fetch conn
1b70: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1b80: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 pool))?;...match
1b90: 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 sqlx::query("de
1ba0: 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 74 67 5f lete from rsstg_
1bb0: 70 6f 73 74 20 70 20 75 73 69 6e 67 20 72 73 73 post p using rss
1bc0: 74 67 5f 73 6f 75 72 63 65 20 73 20 77 68 65 72 tg_source s wher
1bd0: 65 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 e p.source_id =
1be0: 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 $1 and owner = $
1bf0: 32 20 61 6e 64 20 70 2e 73 6f 75 72 63 65 5f 69 2 and p.source_i
1c00: 64 20 3d 20 73 2e 73 6f 75 72 63 65 5f 69 64 3b d = s.source_id;
1c10: 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 ").....bind(sour
1c20: 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 ce_id).....bind(
1c30: 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 owner).....execu
1c40: 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 te(&mut conn).aw
1c50: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
1c60: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1c70: 22 43 6c 65 61 6e 20 73 65 65 6e 20 70 6f 73 74 "Clean seen post
1c80: 73 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 s:\n{:?}", &self
1c90: 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 .pool))?.....row
1ca0: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
1cb0: 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 ..0 => { Ok("No
1cc0: 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 data found found
1cd0: 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 .".into()) },...
1ce0: 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 .x => { Ok(forma
1cf0: 74 21 28 22 7b 7d 20 70 6f 73 74 73 20 70 75 72 t!("{} posts pur
1d00: 67 65 64 2e 22 2c 20 78 29 2e 69 6e 74 6f 28 29 ged.", x).into()
1d10: 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 ) },...}..}...pu
1d20: 62 20 61 73 79 6e 63 20 66 6e 20 65 6e 61 62 6c b async fn enabl
1d30: 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 e<S>(&self, sour
1d40: 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e ce_id: &i32, own
1d50: 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 er: S) -> Result
1d60: 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a <&str>..where S:
1d70: 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c Into<i64> {...l
1d80: 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 et owner = owner
1d90: 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 .into();....let
1da0: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
1db0: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
1dc0: 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f wait.....with_co
1dd0: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
1de0: 28 22 45 6e 61 62 6c 65 20 66 65 74 63 68 20 63 ("Enable fetch c
1df0: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
1e00: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 lf.pool))?;...ma
1e10: 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 tch sqlx::query(
1e20: 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f "update rsstg_so
1e30: 75 72 63 65 20 73 65 74 20 65 6e 61 62 6c 65 64 urce set enabled
1e40: 20 3d 20 74 72 75 65 20 77 68 65 72 65 20 73 6f = true where so
1e50: 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 urce_id = $1 and
1e60: 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 owner = $2")...
1e70: 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 ..bind(source_id
1e80: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
1e90: 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d ).....execute(&m
1ea0: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 ut conn).await..
1eb0: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1ec0: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 45 6e 61 62 || format!("Enab
1ed0: 6c 65 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d le source:\n{:?}
1ee0: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
1ef0: 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 .....rows_affect
1f00: 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b ed() {....1 => {
1f10: 20 4f 6b 28 22 53 6f 75 72 63 65 20 65 6e 61 62 Ok("Source enab
1f20: 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d led.") },....0 =
1f30: 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e > { Ok("Source n
1f40: 6f 74 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 ot found.") },..
1f50: 09 09 5f 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 .._ => { Err(any
1f60: 68 6f 77 21 28 22 44 61 74 61 62 61 73 65 20 65 how!("Database e
1f70: 72 72 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a rror.")) },...}.
1f80: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
1f90: 6e 20 64 69 73 61 62 6c 65 3c 53 3e 28 26 73 65 n disable<S>(&se
1fa0: 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 lf, source_id: &
1fb0: 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d i32, owner: S) -
1fc0: 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 > Result<&str>..
1fd0: 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 where S: Into<i6
1fe0: 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 4> {...let owner
1ff0: 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b = owner.into();
2000: 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ....let mut conn
2010: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
2020: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
2030: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
2040: 20 66 6f 72 6d 61 74 21 28 22 44 69 73 61 62 6c format!("Disabl
2050: 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b e fetch conn:\n{
2060: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
2070: 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c ))?;...match sql
2080: 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 x::query("update
2090: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 rsstg_source se
20a0: 74 20 65 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 t enabled = fals
20b0: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
20c0: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
20d0: 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 = $2").....bind
20e0: 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e (source_id).....
20f0: 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e bind(owner).....
2100: 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e execute(&mut con
2110: 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 n).await.....wit
2120: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
2130: 6d 61 74 21 28 22 44 69 73 61 62 6c 65 20 73 6f mat!("Disable so
2140: 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 urce:\n{:?}", &s
2150: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e elf.pool))?.....
2160: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 rows_affected()
2170: 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 {....1 => { Ok("
2180: 53 6f 75 72 63 65 20 64 69 73 61 62 6c 65 64 2e Source disabled.
2190: 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 ") },....0 => {
21a0: 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 Ok("Source not f
21b0: 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 ound.") },...._
21c0: 3d 3e 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 => { Err(anyhow!
21d0: 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 ("Database error
21e0: 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a .")) },...}..}..
21f0: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 75 70 .pub async fn up
2200: 64 61 74 65 28 26 73 65 6c 66 2c 20 75 70 64 61 date(&self, upda
2210: 74 65 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c te: Option<i32>,
2220: 20 63 68 61 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 channel: &str,
2230: 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c channel_id: i64,
2240: 20 75 72 6c 3a 20 26 73 74 72 2c 20 69 76 5f 68 url: &str, iv_h
2250: 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 ash: Option<&str
2260: 3e 2c 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f >, url_re: Optio
2270: 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 72 3a 20 n<&str>, owner:
2280: 69 36 34 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 i64) -> Result<&
2290: 73 74 72 3e 20 7b 0a 09 2f 2f 77 68 65 72 65 20 str> {..//where
22a0: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
22b0: 09 2f 2f 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f .//let owner = o
22c0: 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 wner.into();....
22d0: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
22e0: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
22f0: 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 ().await.....wit
2300: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
2310: 6d 61 74 21 28 22 55 70 64 61 74 65 20 66 65 74 mat!("Update fet
2320: 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c ch conn:\n{:?}",
2330: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a &self.pool))?;.
2340: 0a 09 09 6d 61 74 63 68 20 6d 61 74 63 68 20 75 ...match match u
2350: 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 pdate {.....Some
2360: 28 69 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 73 (id) => {......s
2370: 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 qlx::query("upda
2380: 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 te rsstg_source
2390: 73 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 20 3d set channel_id =
23a0: 20 24 32 2c 20 75 72 6c 20 3d 20 24 33 2c 20 69 $2, url = $3, i
23b0: 76 5f 68 61 73 68 20 3d 20 24 34 2c 20 6f 77 6e v_hash = $4, own
23c0: 65 72 20 3d 20 24 35 2c 20 63 68 61 6e 6e 65 6c er = $5, channel
23d0: 20 3d 20 24 36 2c 20 75 72 6c 5f 72 65 20 3d 20 = $6, url_re =
23e0: 24 37 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f $7 where source_
23f0: 69 64 20 3d 20 24 31 22 29 2e 62 69 6e 64 28 69 id = $1").bind(i
2400: 64 29 0a 09 09 09 09 7d 2c 0a 09 09 09 09 4e 6f d).....},.....No
2410: 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c ne => {......sql
2420: 78 3a 3a 71 75 65 72 79 28 22 69 6e 73 65 72 74 x::query("insert
2430: 20 69 6e 74 6f 20 72 73 73 74 67 5f 73 6f 75 72 into rsstg_sour
2440: 63 65 20 28 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 ce (channel_id,
2450: 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 url, iv_hash, ow
2460: 6e 65 72 2c 20 63 68 61 6e 6e 65 6c 2c 20 75 72 ner, channel, ur
2470: 6c 5f 72 65 29 20 76 61 6c 75 65 73 20 28 24 31 l_re) values ($1
2480: 2c 20 24 32 2c 20 24 33 2c 20 24 34 2c 20 24 35 , $2, $3, $4, $5
2490: 2c 20 24 36 29 22 29 0a 09 09 09 09 7d 2c 0a 09 , $6)").....},..
24a0: 09 09 7d 0a 09 09 09 2e 62 69 6e 64 28 63 68 61 ..}.....bind(cha
24b0: 6e 6e 65 6c 5f 69 64 29 0a 09 09 09 2e 62 69 6e nnel_id).....bin
24c0: 64 28 75 72 6c 29 0a 09 09 09 2e 62 69 6e 64 28 d(url).....bind(
24d0: 69 76 5f 68 61 73 68 29 0a 09 09 09 2e 62 69 6e iv_hash).....bin
24e0: 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 62 69 6e d(owner).....bin
24f0: 64 28 63 68 61 6e 6e 65 6c 29 0a 09 09 09 2e 62 d(channel).....b
2500: 69 6e 64 28 75 72 6c 5f 72 65 29 0a 09 09 09 2e ind(url_re).....
2510: 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e execute(&mut con
2520: 6e 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 4f 6b n).await {....Ok
2530: 28 5f 29 20 3d 3e 20 4f 6b 28 6d 61 74 63 68 20 (_) => Ok(match
2540: 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d update {.....Som
2550: 65 28 5f 29 20 3d 3e 20 22 43 68 61 6e 6e 65 6c e(_) => "Channel
2560: 20 75 70 64 61 74 65 64 2e 22 2c 0a 09 09 09 09 updated.",.....
2570: 4e 6f 6e 65 20 3d 3e 20 22 43 68 61 6e 6e 65 6c None => "Channel
2580: 20 61 64 64 65 64 2e 22 2c 0a 09 09 09 7d 29 2c added.",....}),
2590: 0a 09 09 09 45 72 72 28 73 71 6c 78 3a 3a 45 72 ....Err(sqlx::Er
25a0: 72 6f 72 3a 3a 44 61 74 61 62 61 73 65 28 65 72 ror::Database(er
25b0: 72 29 29 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 74 r)) => {.....mat
25c0: 63 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a ch err.downcast:
25d0: 3a 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 :<sqlx::postgres
25e0: 3a 3a 50 67 44 61 74 61 62 61 73 65 45 72 72 6f ::PgDatabaseErro
25f0: 72 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b r>().routine() {
2600: 0a 09 09 09 09 09 53 6f 6d 65 28 22 5f 62 74 5f ......Some("_bt_
2610: 63 68 65 63 6b 5f 75 6e 69 71 75 65 22 2c 20 29 check_unique", )
2620: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 => {.......Ok("
2630: 44 75 70 6c 69 63 61 74 65 20 6b 65 79 2e 22 29 Duplicate key.")
2640: 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 53 6f ......},......So
2650: 6d 65 28 5f 29 20 3d 3e 20 7b 0a 09 09 09 09 09 me(_) => {......
2660: 09 4f 6b 28 22 44 61 74 61 62 61 73 65 20 65 72 .Ok("Database er
2670: 72 6f 72 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 ror.")......},..
2680: 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 ....None => {...
2690: 09 09 09 09 4f 6b 28 22 4e 6f 20 64 61 74 61 62 ....Ok("No datab
26a0: 61 73 65 20 65 72 72 6f 72 20 65 78 74 72 61 63 ase error extrac
26b0: 74 65 64 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 ted.")......},..
26c0: 09 09 09 7d 0a 09 09 09 7d 2c 0a 09 09 09 45 72 ...}....},....Er
26d0: 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 r(err) => {.....
26e0: 62 61 69 6c 21 28 22 53 6f 72 72 79 2c 20 75 6e bail!("Sorry, un
26f0: 6b 6e 6f 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a known error:\n{:
2700: 23 3f 7d 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 09 #?}\n", err);...
2710: 09 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 .},...}..}...asy
2720: 6e 63 20 66 6e 20 61 75 74 6f 66 65 74 63 68 28 nc fn autofetch(
2730: 26 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 &self) -> Result
2740: 3c 28 29 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 <()> {...let mut
2750: 20 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a delay = chrono:
2760: 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 :Duration::minut
2770: 65 73 28 31 29 3b 0a 09 09 6c 65 74 20 6d 75 74 es(1);...let mut
2780: 20 6e 6f 77 3b 0a 09 09 6c 6f 6f 70 20 7b 0a 09 now;...loop {..
2790: 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d ..let mut conn =
27a0: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
27b0: 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 09 2e re().await......
27c0: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
27d0: 66 6f 72 6d 61 74 21 28 22 41 75 74 6f 66 65 74 format!("Autofet
27e0: 63 68 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e ch fetch conn:\n
27f0: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
2800: 6c 29 29 3f 3b 0a 09 09 09 6e 6f 77 20 3d 20 63 l))?;....now = c
2810: 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f hrono::Local::no
2820: 77 28 29 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 w();....let mut
2830: 71 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a 71 75 queue = sqlx::qu
2840: 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 ery("select sour
2850: 63 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 63 ce_id, next_fetc
2860: 68 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 h, owner from rs
2870: 73 74 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 stg_order natura
2880: 6c 20 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 l left join rsst
2890: 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 6e g_source where n
28a0: 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 ext_fetch < now(
28b0: 29 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 20 ) + interval '1
28c0: 6d 69 6e 75 74 65 27 3b 22 29 0a 09 09 09 09 2e minute';")......
28d0: 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 fetch_all(&mut c
28e0: 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 onn).await?;....
28f0: 66 6f 72 20 72 6f 77 20 69 6e 20 71 75 65 75 65 for row in queue
2900: 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 09 6c 65 .iter() {.....le
2910: 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 t source_id: i32
2920: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
2930: 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 source_id")?;...
2940: 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 ..let owner: i64
2950: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
2960: 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 09 09 6c 65 owner")?;.....le
2970: 74 20 6e 65 78 74 5f 66 65 74 63 68 3a 20 44 61 t next_fetch: Da
2980: 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 4c teTime<chrono::L
2990: 6f 63 61 6c 3e 20 3d 20 72 6f 77 2e 74 72 79 5f ocal> = row.try_
29a0: 67 65 74 28 22 6e 65 78 74 5f 66 65 74 63 68 22 get("next_fetch"
29b0: 29 3f 3b 0a 09 09 09 09 69 66 20 6e 65 78 74 5f )?;.....if next_
29c0: 66 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a 09 09 fetch < now {...
29d0: 09 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 ...let clone = C
29e0: 6f 72 65 20 7b 0a 09 09 09 09 09 09 6f 77 6e 65 ore {.......owne
29f0: 72 5f 63 68 61 74 3a 20 74 65 6c 65 67 72 61 6d r_chat: telegram
2a00: 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 _bot::UserId::ne
2a10: 77 28 6f 77 6e 65 72 29 2c 0a 09 09 09 09 09 09 w(owner),.......
2a20: 2e 2e 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 0a 09 ..self.clone()..
2a30: 09 09 09 09 7d 3b 0a 09 09 09 09 09 74 6f 6b 69 ....};......toki
2a40: 6f 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 6d o::spawn(async m
2a50: 6f 76 65 20 7b 0a 09 09 09 09 09 09 69 66 20 6c ove {.......if l
2a60: 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c et Err(err) = cl
2a70: 6f 6e 65 2e 63 68 65 63 6b 28 26 73 6f 75 72 63 one.check(&sourc
2a80: 65 5f 69 64 2c 20 6f 77 6e 65 72 2c 20 74 72 75 e_id, owner, tru
2a90: 65 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 e).await {......
2aa0: 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 ..if let Err(err
2ab0: 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 26 ) = clone.send(&
2ac0: 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a format!("š {:
2ad0: 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 2c ?}", err), None,
2ae0: 20 4e 6f 6e 65 29 20 7b 0a 09 09 09 09 09 09 09 None) {........
2af0: 09 65 70 72 69 6e 74 6c 6e 21 28 22 43 68 65 63 .eprintln!("Chec
2b00: 6b 20 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 k error: {}", er
2b10: 72 29 3b 0a 09 09 09 09 09 09 09 7d 3b 0a 09 09 r);........};...
2b20: 09 09 09 09 7d 3b 0a 09 09 09 09 09 7d 29 3b 0a ....};......});.
2b30: 09 09 09 09 7d 20 65 6c 73 65 20 69 66 20 6e 65 ....} else if ne
2b40: 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 20 3c xt_fetch - now <
2b50: 20 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 64 65 delay {......de
2b60: 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 63 68 lay = next_fetch
2b70: 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 7d 0a 09 09 - now;.....}...
2b80: 09 7d 3b 0a 09 09 09 71 75 65 75 65 2e 63 6c 65 .};....queue.cle
2b90: 61 72 28 29 3b 0a 09 09 09 74 6f 6b 69 6f 3a 3a ar();....tokio::
2ba0: 74 69 6d 65 3a 3a 73 6c 65 65 70 28 64 65 6c 61 time::sleep(dela
2bb0: 79 2e 74 6f 5f 73 74 64 28 29 3f 29 2e 61 77 61 y.to_std()?).awa
2bc0: 69 74 3b 0a 09 09 09 64 65 6c 61 79 20 3d 20 63 it;....delay = c
2bd0: 68 72 6f 6e 6f 3a 3a 44 75 72 61 74 69 6f 6e 3a hrono::Duration:
2be0: 3a 6d 69 6e 75 74 65 73 28 31 29 3b 0a 09 09 7d :minutes(1);...}
2bf0: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
2c00: 66 6e 20 6c 69 73 74 3c 53 3e 28 26 73 65 6c 66 fn list<S>(&self
2c10: 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 , owner: S) -> R
2c20: 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 esult<String>..w
2c30: 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 here S: Into<i64
2c40: 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 > {...let owner
2c50: 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a = owner.into();.
2c60: 0a 09 09 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 ...let mut reply
2c70: 3a 20 56 65 63 3c 43 6f 77 3c 73 74 72 3e 3e 20 : Vec<Cow<str>>
2c80: 3d 20 76 65 63 21 5b 5d 3b 0a 09 09 6c 65 74 20 = vec![];...let
2c90: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
2ca0: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
2cb0: 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f wait.....with_co
2cc0: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
2cd0: 28 22 4c 69 73 74 20 66 65 74 63 68 20 63 6f 6e ("List fetch con
2ce0: 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 n:\n{:?}", &self
2cf0: 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 72 65 70 6c .pool))?;...repl
2d00: 79 2e 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 y.push("Channels
2d10: 3a 22 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 6c 65 :".into());...le
2d20: 74 20 72 6f 77 73 20 3d 20 73 71 6c 78 3a 3a 71 t rows = sqlx::q
2d30: 75 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 uery("select sou
2d40: 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c rce_id, channel,
2d50: 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 enabled, url, i
2d60: 76 5f 68 61 73 68 2c 20 75 72 6c 5f 72 65 20 66 v_hash, url_re f
2d70: 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 rom rsstg_source
2d80: 20 77 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 where owner = $
2d90: 31 20 6f 72 64 65 72 20 62 79 20 73 6f 75 72 63 1 order by sourc
2da0: 65 5f 69 64 22 29 0a 09 09 09 2e 62 69 6e 64 28 e_id").....bind(
2db0: 6f 77 6e 65 72 29 0a 09 09 09 2e 66 65 74 63 68 owner).....fetch
2dc0: 5f 61 6c 6c 28 26 6d 75 74 20 63 6f 6e 6e 29 2e _all(&mut conn).
2dd0: 61 77 61 69 74 3f 3b 0a 09 09 66 6f 72 20 72 6f await?;...for ro
2de0: 77 20 69 6e 20 72 6f 77 73 2e 69 74 65 72 28 29 w in rows.iter()
2df0: 20 7b 0a 09 09 09 6c 65 74 20 73 6f 75 72 63 65 {....let source
2e00: 5f 69 64 3a 20 69 33 32 20 3d 20 72 6f 77 2e 74 _id: i32 = row.t
2e10: 72 79 5f 67 65 74 28 22 73 6f 75 72 63 65 5f 69 ry_get("source_i
2e20: 64 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 73 65 d")?;....let use
2e30: 72 6e 61 6d 65 3a 20 26 73 74 72 20 3d 20 72 6f rname: &str = ro
2e40: 77 2e 74 72 79 5f 67 65 74 28 22 63 68 61 6e 6e w.try_get("chann
2e50: 65 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 65 6e el")?;....let en
2e60: 61 62 6c 65 64 3a 20 62 6f 6f 6c 20 3d 20 72 6f abled: bool = ro
2e70: 77 2e 74 72 79 5f 67 65 74 28 22 65 6e 61 62 6c w.try_get("enabl
2e80: 65 64 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 ed")?;....let ur
2e90: 6c 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 l: &str = row.tr
2ea0: 79 5f 67 65 74 28 22 75 72 6c 22 29 3f 3b 0a 09 y_get("url")?;..
2eb0: 09 09 6c 65 74 20 69 76 5f 68 61 73 68 3a 20 4f ..let iv_hash: O
2ec0: 70 74 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 72 6f ption<&str> = ro
2ed0: 77 2e 74 72 79 5f 67 65 74 28 22 69 76 5f 68 61 w.try_get("iv_ha
2ee0: 73 68 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 sh")?;....let ur
2ef0: 6c 5f 72 65 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 l_re: Option<&st
2f00: 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 r> = row.try_get
2f10: 28 22 75 72 6c 5f 72 65 22 29 3f 3b 0a 09 09 09 ("url_re")?;....
2f20: 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 reply.push(forma
2f30: 74 21 28 22 5c 6e 5c 5c 23 ef b8 8f e2 83 a3 20 t!("\n\\#ļøā£
2f40: 7b 7d 20 5c 5c 2a ef b8 8f e2 83 a3 20 60 7b 7d {} \\*ļøā£ `{}
2f50: 60 20 7b 7d 5c 6e f0 9f 94 97 20 60 7b 7d 60 22 ` {}\nš `{}`"
2f60: 2c 20 73 6f 75 72 63 65 5f 69 64 2c 20 75 73 65 , source_id, use
2f70: 72 6e 61 6d 65 2c 20 20 0a 09 09 09 09 6d 61 74 rname, .....mat
2f80: 63 68 20 65 6e 61 62 6c 65 64 20 7b 0a 09 09 09 ch enabled {....
2f90: 09 09 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 ..true => "š
2fa0: 20 65 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 enabled",......
2fb0: 66 61 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 false => "ā di
2fc0: 73 61 62 6c 65 64 22 2c 0a 09 09 09 09 7d 2c 20 sabled",.....},
2fd0: 75 72 6c 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 url).into());...
2fe0: 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 .if let Some(has
2ff0: 68 29 20 3d 20 69 76 5f 68 61 73 68 20 7b 0a 09 h) = iv_hash {..
3000: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f ...reply.push(fo
3010: 72 6d 61 74 21 28 22 49 56 3a 20 60 7b 7d 60 22 rmat!("IV: `{}`"
3020: 2c 20 68 61 73 68 29 2e 69 6e 74 6f 28 29 29 3b , hash).into());
3030: 0a 09 09 09 7d 0a 09 09 09 69 66 20 6c 65 74 20 ....}....if let
3040: 53 6f 6d 65 28 72 65 29 20 3d 20 75 72 6c 5f 72 Some(re) = url_r
3050: 65 20 7b 0a 09 09 09 09 72 65 70 6c 79 2e 70 75 e {.....reply.pu
3060: 73 68 28 66 6f 72 6d 61 74 21 28 22 52 45 3a 20 sh(format!("RE:
3070: 60 7b 7d 60 22 2c 20 72 65 29 2e 69 6e 74 6f 28 `{}`", re).into(
3080: 29 29 3b 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 ));....}...};...
3090: 4f 6b 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c Ok(reply.join("\
30a0: 6e 22 29 29 0a 09 7d 0a 7d 0a n"))..}.}.