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 73 67 20 3d 20 6d 73 67 2e 69 6e let msg = msg.in
06f0: 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 6f 64 to();....let mod
0700: 65 20 3d 20 6d 6f 64 65 2e 75 6e 77 72 61 70 5f e = mode.unwrap_
0710: 6f 72 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a or(telegram_bot:
0720: 3a 74 79 70 65 73 3a 3a 50 61 72 73 65 4d 6f 64 :types::ParseMod
0730: 65 3a 3a 48 74 6d 6c 29 3b 0a 09 09 6c 65 74 20 e::Html);...let
0740: 74 61 72 67 65 74 20 3d 20 74 61 72 67 65 74 2e target = target.
0750: 75 6e 77 72 61 70 5f 6f 72 28 73 65 6c 66 2e 6f unwrap_or(self.o
0760: 77 6e 65 72 5f 63 68 61 74 29 3b 0a 09 09 73 65 wner_chat);...se
0770: 6c 66 2e 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 lf.tg.send(teleg
0780: 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 ram_bot::SendMes
0790: 73 61 67 65 3a 3a 6e 65 77 28 74 61 72 67 65 74 sage::new(target
07a0: 2c 20 6d 73 67 29 2e 70 61 72 73 65 5f 6d 6f 64 , msg).parse_mod
07b0: 65 28 6d 6f 64 65 29 29 2e 61 77 61 69 74 3f 3b e(mode)).await?;
07c0: 0a 09 09 4f 6b 28 28 29 29 0a 09 7d 0a 0a 09 70 ...Ok(())..}...p
07d0: 75 62 20 61 73 79 6e 63 20 66 6e 20 63 68 65 63 ub async fn chec
07e0: 6b 3c 53 3e 28 26 73 65 6c 66 2c 20 69 64 3a 20 k<S>(&self, id:
07f0: 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 2c 20 &i32, owner: S,
0800: 72 65 61 6c 3a 20 62 6f 6f 6c 29 20 2d 3e 20 52 real: bool) -> R
0810: 65 73 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 esult<Cow<'_, st
0820: 72 3e 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e r>>..where S: In
0830: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 to<i64> {...let
0840: 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e owner = owner.in
0850: 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 to();....let mut
0860: 20 70 6f 73 74 65 64 3a 20 69 33 32 20 3d 20 30 posted: i32 = 0
0870: 3b 0a 09 09 6c 65 74 20 69 64 20 3d 20 7b 0a 09 ;...let id = {..
0880: 09 09 6c 65 74 20 6d 75 74 20 73 65 74 20 3d 20 ..let mut set =
0890: 73 65 6c 66 2e 73 6f 75 72 63 65 73 2e 6c 6f 63 self.sources.loc
08a0: 6b 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 k().unwrap();...
08b0: 09 6d 61 74 63 68 20 73 65 74 2e 67 65 74 28 69 .match set.get(i
08c0: 64 29 20 7b 0a 09 09 09 09 53 6f 6d 65 28 69 64 d) {.....Some(id
08d0: 29 20 3d 3e 20 69 64 2e 63 6c 6f 6e 65 28 29 2c ) => id.clone(),
08e0: 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 .....None => {..
08f0: 09 09 09 09 6c 65 74 20 69 64 20 3d 20 41 72 63 ....let id = Arc
0900: 3a 3a 6e 65 77 28 2a 69 64 29 3b 0a 09 09 09 09 ::new(*id);.....
0910: 09 73 65 74 2e 69 6e 73 65 72 74 28 69 64 2e 63 .set.insert(id.c
0920: 6c 6f 6e 65 28 29 29 3b 0a 09 09 09 09 09 69 64 lone());......id
0930: 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 7d 2c 0a .clone().....},.
0940: 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 ...}...};...let
0950: 63 6f 75 6e 74 20 3d 20 41 72 63 3a 3a 73 74 72 count = Arc::str
0960: 6f 6e 67 5f 63 6f 75 6e 74 28 26 69 64 29 3b 0a ong_count(&id);.
0970: 09 09 69 66 20 63 6f 75 6e 74 20 3d 3d 20 32 20 ..if count == 2
0980: 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e {....let mut con
0990: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
09a0: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
09b0: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
09c0: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 51 75 65 72 || format!("Quer
09d0: 79 20 71 75 65 75 65 20 66 65 74 63 68 20 63 6f y queue fetch co
09e0: 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c nn:\n{:?}", &sel
09f0: 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 6c 65 f.pool))?;....le
0a00: 74 20 72 6f 77 20 3d 20 73 71 6c 78 3a 3a 71 75 t row = sqlx::qu
0a10: 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 ery("select sour
0a20: 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 5f 69 ce_id, channel_i
0a30: 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c d, url, iv_hash,
0a40: 20 6f 77 6e 65 72 2c 20 75 72 6c 5f 72 65 20 66 owner, url_re f
0a50: 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 rom rsstg_source
0a60: 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 where source_id
0a70: 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 = $1 and owner
0a80: 3d 20 24 32 22 29 0a 09 09 09 09 2e 62 69 6e 64 = $2")......bind
0a90: 28 2a 69 64 29 0a 09 09 09 09 2e 62 69 6e 64 28 (*id)......bind(
0aa0: 6f 77 6e 65 72 29 0a 09 09 09 09 2e 66 65 74 63 owner)......fetc
0ab0: 68 5f 6f 6e 65 28 26 6d 75 74 20 63 6f 6e 6e 29 h_one(&mut conn)
0ac0: 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 74 68 .await......with
0ad0: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
0ae0: 61 74 21 28 22 51 75 65 72 79 20 73 6f 75 72 63 at!("Query sourc
0af0: 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 e:\n{:?}", &self
0b00: 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 64 72 6f .pool))?;....dro
0b10: 70 28 63 6f 6e 6e 29 3b 0a 09 09 09 6c 65 74 20 p(conn);....let
0b20: 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 20 channel_id: i64
0b30: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 63 = row.try_get("c
0b40: 68 61 6e 6e 65 6c 5f 69 64 22 29 3f 3b 0a 09 09 hannel_id")?;...
0b50: 09 6c 65 74 20 75 72 6c 3a 20 26 73 74 72 20 3d .let url: &str =
0b60: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 72 row.try_get("ur
0b70: 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 69 76 5f l")?;....let iv_
0b80: 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 hash: Option<&st
0b90: 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 r> = row.try_get
0ba0: 28 22 69 76 5f 68 61 73 68 22 29 3f 3b 0a 09 09 ("iv_hash")?;...
0bb0: 09 6c 65 74 20 75 72 6c 5f 72 65 20 3d 20 6d 61 .let url_re = ma
0bc0: 74 63 68 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 tch row.try_get(
0bd0: 22 75 72 6c 5f 72 65 22 29 3f 20 7b 0a 09 09 09 "url_re")? {....
0be0: 09 53 6f 6d 65 28 78 29 20 3d 3e 20 53 6f 6d 65 .Some(x) => Some
0bf0: 28 73 65 64 72 65 67 65 78 3a 3a 52 65 70 6c 61 (sedregex::Repla
0c00: 63 65 43 6f 6d 6d 61 6e 64 3a 3a 6e 65 77 28 78 ceCommand::new(x
0c10: 29 3f 29 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e )?),.....None =>
0c20: 20 4e 6f 6e 65 2c 0a 09 09 09 7d 3b 0a 09 09 09 None,....};....
0c30: 6c 65 74 20 64 65 73 74 69 6e 61 74 69 6f 6e 20 let destination
0c40: 3d 20 6d 61 74 63 68 20 72 65 61 6c 20 7b 0a 09 = match real {..
0c50: 09 09 09 74 72 75 65 20 3d 3e 20 74 65 6c 65 67 ...true => teleg
0c60: 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a ram_bot::UserId:
0c70: 3a 6e 65 77 28 63 68 61 6e 6e 65 6c 5f 69 64 29 :new(channel_id)
0c80: 2c 0a 09 09 09 09 66 61 6c 73 65 20 3d 3e 20 74 ,.....false => t
0c90: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 elegram_bot::Use
0ca0: 72 49 64 3a 3a 6e 65 77 28 72 6f 77 2e 74 72 79 rId::new(row.try
0cb0: 5f 67 65 74 28 22 6f 77 6e 65 72 22 29 3f 29 2c _get("owner")?),
0cc0: 0a 09 09 09 7d 3b 0a 09 09 09 6c 65 74 20 6d 75 ....};....let mu
0cd0: 74 20 74 68 69 73 5f 66 65 74 63 68 3a 20 4f 70 t this_fetch: Op
0ce0: 74 69 6f 6e 3c 44 61 74 65 54 69 6d 65 3c 63 68 tion<DateTime<ch
0cf0: 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 66 73 65 rono::FixedOffse
0d00: 74 3e 3e 20 3d 20 4e 6f 6e 65 3b 0a 09 09 09 6c t>> = None;....l
0d10: 65 74 20 6d 75 74 20 70 6f 73 74 73 3a 20 42 54 et mut posts: BT
0d20: 72 65 65 4d 61 70 3c 44 61 74 65 54 69 6d 65 3c reeMap<DateTime<
0d30: 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 66 chrono::FixedOff
0d40: 73 65 74 3e 2c 20 53 74 72 69 6e 67 3e 20 3d 20 set>, String> =
0d50: 42 54 72 65 65 4d 61 70 3a 3a 6e 65 77 28 29 3b BTreeMap::new();
0d60: 0a 09 09 09 6c 65 74 20 72 65 73 70 6f 6e 73 65 ....let response
0d70: 20 3d 20 72 65 71 77 65 73 74 3a 3a 67 65 74 28 = reqwest::get(
0d80: 75 72 6c 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 url).await?;....
0d90: 6c 65 74 20 73 74 61 74 75 73 20 3d 20 72 65 73 let status = res
0da0: 70 6f 6e 73 65 2e 73 74 61 74 75 73 28 29 3b 0a ponse.status();.
0db0: 09 09 09 6c 65 74 20 63 6f 6e 74 65 6e 74 20 3d ...let content =
0dc0: 20 72 65 73 70 6f 6e 73 65 2e 62 79 74 65 73 28 response.bytes(
0dd0: 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 6d 61 74 ).await?;....mat
0de0: 63 68 20 72 73 73 3a 3a 43 68 61 6e 6e 65 6c 3a ch rss::Channel:
0df0: 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 :read_from(&cont
0e00: 65 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 09 09 09 4f ent[..]) {.....O
0e10: 6b 28 66 65 65 64 29 20 3d 3e 20 7b 0a 09 09 09 k(feed) => {....
0e20: 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 65 ..for item in fe
0e30: 65 64 2e 69 74 65 6d 73 28 29 20 7b 0a 09 09 09 ed.items() {....
0e40: 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 6c ...if let Some(l
0e50: 69 6e 6b 29 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b ink) = item.link
0e60: 28 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 () {........let
0e70: 64 61 74 65 20 3d 20 6d 61 74 63 68 20 69 74 65 date = match ite
0e80: 6d 2e 70 75 62 5f 64 61 74 65 28 29 20 7b 0a 09 m.pub_date() {..
0e90: 09 09 09 09 09 09 09 53 6f 6d 65 28 66 65 65 64 .......Some(feed
0ea0: 5f 64 61 74 65 29 20 3d 3e 20 44 61 74 65 54 69 _date) => DateTi
0eb0: 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 me::parse_from_r
0ec0: 66 63 32 38 32 32 28 66 65 65 64 5f 64 61 74 65 fc2822(feed_date
0ed0: 29 2c 0a 09 09 09 09 09 09 09 09 4e 6f 6e 65 20 ),.........None
0ee0: 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 => DateTime::par
0ef0: 73 65 5f 66 72 6f 6d 5f 72 66 63 33 33 33 39 28 se_from_rfc3339(
0f00: 26 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 63 6f 72 &item.dublin_cor
0f10: 65 5f 65 78 74 28 29 2e 75 6e 77 72 61 70 28 29 e_ext().unwrap()
0f20: 2e 64 61 74 65 73 28 29 5b 30 5d 29 2c 0a 09 09 .dates()[0]),...
0f30: 09 09 09 09 09 7d 3f 3b 0a 09 09 09 09 09 09 09 .....}?;........
0f40: 6c 65 74 20 75 72 6c 20 3d 20 6c 69 6e 6b 3b 0a let url = link;.
0f50: 09 09 09 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 .......posts.ins
0f60: 65 72 74 28 64 61 74 65 2c 20 75 72 6c 2e 74 6f ert(date, url.to
0f70: 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 _string());.....
0f80: 09 09 7d 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 ..}......};.....
0f90: 7d 2c 0a 09 09 09 09 45 72 72 28 65 72 72 29 20 },.....Err(err)
0fa0: 3d 3e 20 6d 61 74 63 68 20 65 72 72 20 7b 0a 09 => match err {..
0fb0: 09 09 09 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a ....rss::Error::
0fc0: 49 6e 76 61 6c 69 64 53 74 61 72 74 54 61 67 20 InvalidStartTag
0fd0: 3d 3e 20 7b 0a 09 09 09 09 09 09 6c 65 74 20 66 => {.......let f
0fe0: 65 65 64 20 3d 20 61 74 6f 6d 5f 73 79 6e 64 69 eed = atom_syndi
0ff0: 63 61 74 69 6f 6e 3a 3a 46 65 65 64 3a 3a 72 65 cation::Feed::re
1000: 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 ad_from(&content
1010: 5b 2e 2e 5d 29 0a 09 09 09 09 09 09 09 2e 77 69 [..]).........wi
1020: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
1030: 72 6d 61 74 21 28 22 50 72 6f 62 6c 65 6d 20 6f rmat!("Problem o
1040: 70 65 6e 69 6e 67 20 66 65 65 64 20 75 72 6c 3a pening feed url:
1050: 5c 6e 7b 7d 5c 6e 7b 7d 22 2c 20 26 75 72 6c 2c \n{}\n{}", &url,
1060: 20 73 74 61 74 75 73 29 29 3f 3b 0a 09 09 09 09 status))?;.....
1070: 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 65 ..for item in fe
1080: 65 64 2e 65 6e 74 72 69 65 73 28 29 20 7b 0a 09 ed.entries() {..
1090: 09 09 09 09 09 09 6c 65 74 20 64 61 74 65 20 3d ......let date =
10a0: 20 69 74 65 6d 2e 70 75 62 6c 69 73 68 65 64 28 item.published(
10b0: 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 ).unwrap();.....
10c0: 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 69 74 65 ...let url = ite
10d0: 6d 2e 6c 69 6e 6b 73 28 29 5b 30 5d 2e 68 72 65 m.links()[0].hre
10e0: 66 28 29 3b 0a 09 09 09 09 09 09 09 70 6f 73 74 f();........post
10f0: 73 2e 69 6e 73 65 72 74 28 2a 64 61 74 65 2c 20 s.insert(*date,
1100: 75 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 url.to_string())
1110: 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ;.......};......
1120: 7d 2c 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 },......rss::Err
1130: 6f 72 3a 3a 45 6f 66 20 3d 3e 20 28 29 2c 0a 09 or::Eof => (),..
1140: 09 09 09 09 5f 20 3d 3e 20 62 61 69 6c 21 28 22 ...._ => bail!("
1150: 55 6e 73 75 70 70 6f 72 74 65 64 20 6f 72 20 6d Unsupported or m
1160: 61 6e 67 6c 65 64 20 63 6f 6e 74 65 6e 74 3a 5c angled content:\
1170: 6e 7b 3a 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e 7b 3a n{:?}\n{:#?}\n{:
1180: 23 3f 7d 5c 6e 22 2c 20 26 75 72 6c 2c 20 65 72 #?}\n", &url, er
1190: 72 2c 20 73 74 61 74 75 73 29 0a 09 09 09 09 7d r, status).....}
11a0: 0a 09 09 09 7d 3b 0a 09 09 09 66 6f 72 20 28 64 ....};....for (d
11b0: 61 74 65 2c 20 75 72 6c 29 20 69 6e 20 70 6f 73 ate, url) in pos
11c0: 74 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 09 ts.iter() {.....
11d0: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
11e0: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
11f0: 28 29 2e 61 77 61 69 74 0a 09 09 09 09 09 2e 77 ().await.......w
1200: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
1210: 6f 72 6d 61 74 21 28 22 43 68 65 63 6b 20 70 6f ormat!("Check po
1220: 73 74 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e st fetch conn:\n
1230: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
1240: 6c 29 29 3f 3b 0a 09 09 09 09 6c 65 74 20 70 6f l))?;.....let po
1250: 73 74 5f 75 72 6c 3a 20 43 6f 77 3c 73 74 72 3e st_url: Cow<str>
1260: 20 3d 20 6d 61 74 63 68 20 75 72 6c 5f 72 65 20 = match url_re
1270: 7b 0a 09 09 09 09 09 53 6f 6d 65 28 72 65 66 20 {......Some(ref
1280: 78 29 20 3d 3e 20 78 2e 65 78 65 63 75 74 65 28 x) => x.execute(
1290: 75 72 6c 29 2c 0a 09 09 09 09 09 4e 6f 6e 65 20 url),......None
12a0: 3d 3e 20 75 72 6c 2e 69 6e 74 6f 28 29 2c 0a 09 => url.into(),..
12b0: 09 09 09 7d 3b 0a 09 09 09 09 6c 65 74 20 72 6f ...};.....let ro
12c0: 77 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 w = sqlx::query(
12d0: 22 73 65 6c 65 63 74 20 65 78 69 73 74 73 28 73 "select exists(s
12e0: 65 6c 65 63 74 20 74 72 75 65 20 66 72 6f 6d 20 elect true from
12f0: 72 73 73 74 67 5f 70 6f 73 74 20 77 68 65 72 65 rsstg_post where
1300: 20 75 72 6c 20 3d 20 24 31 20 61 6e 64 20 73 6f url = $1 and so
1310: 75 72 63 65 5f 69 64 20 3d 20 24 32 29 20 61 73 urce_id = $2) as
1320: 20 65 78 69 73 74 73 3b 22 29 0a 09 09 09 09 09 exists;")......
1330: 2e 62 69 6e 64 28 26 2a 70 6f 73 74 5f 75 72 6c .bind(&*post_url
1340: 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 2a 69 64 ).......bind(*id
1350: 29 0a 09 09 09 09 09 2e 66 65 74 63 68 5f 6f 6e ).......fetch_on
1360: 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 e(&mut conn).awa
1370: 69 74 0a 09 09 09 09 09 2e 77 69 74 68 5f 63 6f it.......with_co
1380: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
1390: 28 22 43 68 65 63 6b 20 70 6f 73 74 3a 5c 6e 7b ("Check post:\n{
13a0: 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f 3b 0a :?}", &conn))?;.
13b0: 09 09 09 09 6c 65 74 20 65 78 69 73 74 73 3a 20 ....let exists:
13c0: 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 5f 67 bool = row.try_g
13d0: 65 74 28 22 65 78 69 73 74 73 22 29 3f 3b 0a 09 et("exists")?;..
13e0: 09 09 09 69 66 20 21 20 65 78 69 73 74 73 20 7b ...if ! exists {
13f0: 0a 09 09 09 09 09 69 66 20 74 68 69 73 5f 66 65 ......if this_fe
1400: 74 63 68 20 3d 3d 20 4e 6f 6e 65 20 7c 7c 20 2a tch == None || *
1410: 64 61 74 65 20 3e 20 74 68 69 73 5f 66 65 74 63 date > this_fetc
1420: 68 2e 75 6e 77 72 61 70 28 29 20 7b 0a 09 09 09 h.unwrap() {....
1430: 09 09 09 74 68 69 73 5f 66 65 74 63 68 20 3d 20 ...this_fetch =
1440: 53 6f 6d 65 28 2a 64 61 74 65 29 3b 0a 09 09 09 Some(*date);....
1450: 09 09 7d 3b 0a 09 09 09 09 09 73 65 6c 66 2e 74 ..};......self.t
1460: 67 2e 73 65 6e 64 28 20 6d 61 74 63 68 20 69 76 g.send( match iv
1470: 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 09 53 _hash {........S
1480: 6f 6d 65 28 68 61 73 68 29 20 3d 3e 20 74 65 6c ome(hash) => tel
1490: 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d egram_bot::SendM
14a0: 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 73 74 essage::new(dest
14b0: 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 21 ination, format!
14c0: 28 22 3c 61 20 68 72 65 66 3d 5c 22 68 74 74 70 ("<a href=\"http
14d0: 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 72 6c 3d s://t.me/iv?url=
14e0: 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c 22 3e 20 3c {}&rhash={}\"> <
14f0: 2f 61 3e 7b 30 7d 22 2c 20 26 70 6f 73 74 5f 75 /a>{0}", &post_u
1500: 72 6c 2c 20 68 61 73 68 29 29 2c 0a 09 09 09 09 rl, hash)),.....
1510: 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 65 6c 65 67 ...None => teleg
1520: 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 ram_bot::SendMes
1530: 73 61 67 65 3a 3a 6e 65 77 28 64 65 73 74 69 6e sage::new(destin
1540: 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 21 28 22 ation, format!("
1550: 7b 7d 22 2c 20 70 6f 73 74 5f 75 72 6c 29 29 2c {}", post_url)),
1560: 0a 09 09 09 09 09 09 7d 2e 70 61 72 73 65 5f 6d .......}.parse_m
1570: 6f 64 65 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 ode(telegram_bot
1580: 3a 3a 74 79 70 65 73 3a 3a 50 61 72 73 65 4d 6f ::types::ParseMo
1590: 64 65 3a 3a 48 74 6d 6c 29 29 2e 61 77 61 69 74 de::Html)).await
15a0: 0a 09 09 09 09 09 09 2e 63 6f 6e 74 65 78 74 28 ........context(
15b0: 22 43 61 6e 27 74 20 70 6f 73 74 20 6d 65 73 73 "Can't post mess
15c0: 61 67 65 3a 22 29 3f 3b 0a 09 09 09 09 09 73 71 age:")?;......sq
15d0: 6c 78 3a 3a 71 75 65 72 79 28 22 69 6e 73 65 72 lx::query("inser
15e0: 74 20 69 6e 74 6f 20 72 73 73 74 67 5f 70 6f 73 t into rsstg_pos
15f0: 74 20 28 73 6f 75 72 63 65 5f 69 64 2c 20 70 6f t (source_id, po
1600: 73 74 65 64 2c 20 75 72 6c 29 20 76 61 6c 75 65 sted, url) value
1610: 73 20 28 24 31 2c 20 24 32 2c 20 24 33 29 3b 22 s ($1, $2, $3);"
1620: 29 0a 09 09 09 09 09 09 2e 62 69 6e 64 28 2a 69 )........bind(*i
1630: 64 29 0a 09 09 09 09 09 09 2e 62 69 6e 64 28 64 d)........bind(d
1640: 61 74 65 29 0a 09 09 09 09 09 09 2e 62 69 6e 64 ate)........bind
1650: 28 26 2a 70 6f 73 74 5f 75 72 6c 29 0a 09 09 09 (&*post_url)....
1660: 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 ....execute(&mut
1670: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 conn).await....
1680: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
1690: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 52 65 63 (|| format!("Rec
16a0: 6f 72 64 20 70 6f 73 74 3a 5c 6e 7b 3a 3f 7d 22 ord post:\n{:?}"
16b0: 2c 20 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 09 09 , &conn))?;.....
16c0: 09 64 72 6f 70 28 63 6f 6e 6e 29 3b 0a 09 09 09 .drop(conn);....
16d0: 09 09 74 6f 6b 69 6f 3a 3a 74 69 6d 65 3a 3a 73 ..tokio::time::s
16e0: 6c 65 65 70 28 73 74 64 3a 3a 74 69 6d 65 3a 3a leep(std::time::
16f0: 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 34 2c Duration::new(4,
1700: 20 30 29 29 2e 61 77 61 69 74 3b 0a 09 09 09 09 0)).await;.....
1710: 7d 3b 0a 09 09 09 09 70 6f 73 74 65 64 20 2b 3d };.....posted +=
1720: 20 31 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 6f 73 1;....};....pos
1730: 74 73 2e 63 6c 65 61 72 28 29 3b 0a 09 09 7d 3b ts.clear();...};
1740: 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
1750: 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 = self.pool.acqu
1760: 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e ire().await.....
1770: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
1780: 66 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 format!("Update
1790: 73 63 72 61 70 65 20 66 65 74 63 68 20 63 6f 6e scrape fetch con
17a0: 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 n:\n{:?}", &self
17b0: 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 73 71 6c 78 .pool))?;...sqlx
17c0: 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 20 ::query("update
17d0: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 rsstg_source set
17e0: 20 6c 61 73 74 5f 73 63 72 61 70 65 20 3d 20 6e last_scrape = n
17f0: 6f 77 28 29 20 77 68 65 72 65 20 73 6f 75 72 63 ow() where sourc
1800: 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a 09 09 09 e_id = $1;")....
1810: 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 2e 65 .bind(*id).....e
1820: 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e xecute(&mut conn
1830: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
1840: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
1850: 61 74 21 28 22 55 70 64 61 74 65 20 73 63 72 61 at!("Update scra
1860: 70 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e pe:\n{:?}", &con
1870: 6e 29 29 3f 3b 0a 09 09 4f 6b 28 66 6f 72 6d 61 n))?;...Ok(forma
1880: 74 21 28 22 50 6f 73 74 65 64 3a 20 7b 7d 22 2c t!("Posted: {}",
1890: 20 26 70 6f 73 74 65 64 29 2e 69 6e 74 6f 28 29 &posted).into()
18a0: 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 )..}...pub async
18b0: 20 66 6e 20 64 65 6c 65 74 65 3c 53 3e 28 26 73 fn delete<S>(&s
18c0: 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 elf, source_id:
18d0: 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 &i32, owner: S)
18e0: 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 3c 27 5f -> Result<Cow<'_
18f0: 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 65 20 53 , str>>..where S
1900: 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 : Into<i64> {...
1910: 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 let owner = owne
1920: 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 r.into();....let
1930: 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 mut conn = self
1940: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
1950: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
1960: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1970: 21 28 22 44 65 6c 65 74 65 20 66 65 74 63 68 20 !("Delete fetch
1980: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
1990: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d elf.pool))?;...m
19a0: 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 atch sqlx::query
19b0: 28 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 ("delete from rs
19c0: 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 stg_source where
19d0: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 source_id = $1
19e0: 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 3b 22 and owner = $2;"
19f0: 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 ).....bind(sourc
1a00: 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f e_id).....bind(o
1a10: 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 wner).....execut
1a20: 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 e(&mut conn).awa
1a30: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
1a40: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1a50: 44 65 6c 65 74 65 20 73 6f 75 72 63 65 20 72 75 Delete source ru
1a60: 6c 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c le:\n{:?}", &sel
1a70: 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f f.pool))?.....ro
1a80: 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a ws_affected() {.
1a90: 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f ...0 => { Ok("No
1aa0: 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e data found foun
1ab0: 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 d.".into()) },..
1ac0: 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d ..x => { Ok(form
1ad0: 61 74 21 28 22 7b 7d 20 73 6f 75 72 63 65 73 20 at!("{} sources
1ae0: 72 65 6d 6f 76 65 64 2e 22 2c 20 78 29 2e 69 6e removed.", x).in
1af0: 74 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a to()) },...}..}.
1b00: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 63 ..pub async fn c
1b10: 6c 65 61 6e 3c 53 3e 28 26 73 65 6c 66 2c 20 73 lean<S>(&self, s
1b20: 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 ource_id: &i32,
1b30: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
1b40: 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e ult<Cow<'_, str>
1b50: 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f >..where S: Into
1b60: 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 <i64> {...let ow
1b70: 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f ner = owner.into
1b80: 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 ();....let mut c
1b90: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e onn = self.pool.
1ba0: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a acquire().await.
1bb0: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
1bc0: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 6c 65 (|| format!("Cle
1bd0: 61 6e 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e an fetch conn:\n
1be0: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
1bf0: 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 l))?;...match sq
1c00: 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 6c 65 74 lx::query("delet
1c10: 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 6f 73 e from rsstg_pos
1c20: 74 20 70 20 75 73 69 6e 67 20 72 73 73 74 67 5f t p using rsstg_
1c30: 73 6f 75 72 63 65 20 73 20 77 68 65 72 65 20 70 source s where p
1c40: 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 .source_id = $1
1c50: 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 20 61 and owner = $2 a
1c60: 6e 64 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d nd p.source_id =
1c70: 20 73 2e 73 6f 75 72 63 65 5f 69 64 3b 22 29 0a s.source_id;").
1c80: 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f ....bind(source_
1c90: 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e id).....bind(own
1ca0: 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 er).....execute(
1cb0: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
1cc0: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
1cd0: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 6c t(|| format!("Cl
1ce0: 65 61 6e 20 73 65 65 6e 20 70 6f 73 74 73 3a 5c ean seen posts:\
1cf0: 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f n{:?}", &self.po
1d00: 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 ol))?.....rows_a
1d10: 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 30 ffected() {....0
1d20: 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 61 74 => { Ok("No dat
1d30: 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 2e 22 2e a found found.".
1d40: 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 09 78 20 into()) },....x
1d50: 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 => { Ok(format!(
1d60: 22 7b 7d 20 70 6f 73 74 73 20 70 75 72 67 65 64 "{} posts purged
1d70: 2e 22 2c 20 78 29 2e 69 6e 74 6f 28 29 29 20 7d .", x).into()) }
1d80: 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 ,...}..}...pub a
1d90: 73 79 6e 63 20 66 6e 20 65 6e 61 62 6c 65 3c 53 sync fn enable<S
1da0: 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f >(&self, source_
1db0: 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a id: &i32, owner:
1dc0: 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 S) -> Result<&s
1dd0: 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e tr>..where S: In
1de0: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 to<i64> {...let
1df0: 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e owner = owner.in
1e00: 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 to();....let mut
1e10: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
1e20: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
1e30: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
1e40: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 45 xt(|| format!("E
1e50: 6e 61 62 6c 65 20 66 65 74 63 68 20 63 6f 6e 6e nable fetch conn
1e60: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1e70: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 pool))?;...match
1e80: 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 sqlx::query("up
1e90: 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 date rsstg_sourc
1ea0: 65 20 73 65 74 20 65 6e 61 62 6c 65 64 20 3d 20 e set enabled =
1eb0: 74 72 75 65 20 77 68 65 72 65 20 73 6f 75 72 63 true where sourc
1ec0: 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 e_id = $1 and ow
1ed0: 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 ner = $2").....b
1ee0: 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 ind(source_id)..
1ef0: 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 ...bind(owner)..
1f00: 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 ...execute(&mut
1f10: 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e conn).await.....
1f20: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
1f30: 66 6f 72 6d 61 74 21 28 22 45 6e 61 62 6c 65 20 format!("Enable
1f40: 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 source:\n{:?}",
1f50: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 &self.pool))?...
1f60: 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 ..rows_affected(
1f70: 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b ) {....1 => { Ok
1f80: 28 22 53 6f 75 72 63 65 20 65 6e 61 62 6c 65 64 ("Source enabled
1f90: 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b .") },....0 => {
1fa0: 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 Ok("Source not
1fb0: 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f found.") },...._
1fc0: 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 => { Err(anyhow
1fd0: 21 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f !("Database erro
1fe0: 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a r.")) },...}..}.
1ff0: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 64 ..pub async fn d
2000: 69 73 61 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c isable<S>(&self,
2010: 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 source_id: &i32
2020: 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 , owner: S) -> R
2030: 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 esult<&str>..whe
2040: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
2050: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
2060: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 owner.into();...
2070: 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 .let mut conn =
2080: 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 self.pool.acquir
2090: 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 e().await.....wi
20a0: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
20b0: 72 6d 61 74 21 28 22 44 69 73 61 62 6c 65 20 66 rmat!("Disable f
20c0: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
20d0: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
20e0: 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a ;...match sqlx::
20f0: 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 73 query("update rs
2100: 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 stg_source set e
2110: 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 65 20 77 nabled = false w
2120: 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d here source_id =
2130: 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 $1 and owner =
2140: 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f $2").....bind(so
2150: 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e urce_id).....bin
2160: 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 d(owner).....exe
2170: 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e cute(&mut conn).
2180: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
2190: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
21a0: 21 28 22 44 69 73 61 62 6c 65 20 73 6f 75 72 63 !("Disable sourc
21b0: 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 e:\n{:?}", &self
21c0: 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 .pool))?.....row
21d0: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
21e0: 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 ..1 => { Ok("Sou
21f0: 72 63 65 20 64 69 73 61 62 6c 65 64 2e 22 29 20 rce disabled.")
2200: 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 },....0 => { Ok(
2210: 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e "Source not foun
2220: 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 d.") },...._ =>
2230: 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 { Err(anyhow!("D
2240: 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 atabase error.")
2250: 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 ) },...}..}...pu
2260: 62 20 61 73 79 6e 63 20 66 6e 20 75 70 64 61 74 b async fn updat
2270: 65 3c 53 3e 28 26 73 65 6c 66 2c 20 75 70 64 61 e<S>(&self, upda
2280: 74 65 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c te: Option<i32>,
2290: 20 63 68 61 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 channel: &str,
22a0: 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c channel_id: i64,
22b0: 20 75 72 6c 3a 20 26 73 74 72 2c 20 69 76 5f 68 url: &str, iv_h
22c0: 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 ash: Option<&str
22d0: 3e 2c 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f >, url_re: Optio
22e0: 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 72 3a 20 n<&str>, owner:
22f0: 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 S) -> Result<&st
2300: 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 r>..where S: Int
2310: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f o<i64> {...let o
2320: 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 wner = owner.int
2330: 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 o();....let mut
2340: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
2350: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
2360: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
2370: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 t(|| format!("Up
2380: 64 61 74 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a date fetch conn:
2390: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
23a0: 6f 6f 6c 29 29 3f 3b 0a 0a 09 09 6d 61 74 63 68 ool))?;....match
23b0: 20 6d 61 74 63 68 20 75 70 64 61 74 65 20 7b 0a match update {.
23c0: 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 ....Some(id) =>
23d0: 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 {......sqlx::que
23e0: 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 67 ry("update rsstg
23f0: 5f 73 6f 75 72 63 65 20 73 65 74 20 63 68 61 6e _source set chan
2400: 6e 65 6c 5f 69 64 20 3d 20 24 32 2c 20 75 72 6c nel_id = $2, url
2410: 20 3d 20 24 33 2c 20 69 76 5f 68 61 73 68 20 3d = $3, iv_hash =
2420: 20 24 34 2c 20 6f 77 6e 65 72 20 3d 20 24 35 2c $4, owner = $5,
2430: 20 63 68 61 6e 6e 65 6c 20 3d 20 24 36 2c 20 75 channel = $6, u
2440: 72 6c 5f 72 65 20 3d 20 24 37 20 77 68 65 72 65 rl_re = $7 where
2450: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 22 source_id = $1"
2460: 29 2e 62 69 6e 64 28 69 64 29 0a 09 09 09 09 7d ).bind(id).....}
2470: 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a ,.....None => {.
2480: 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 .....sqlx::query
2490: 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 ("insert into rs
24a0: 73 74 67 5f 73 6f 75 72 63 65 20 28 63 68 61 6e stg_source (chan
24b0: 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f nel_id, url, iv_
24c0: 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 hash, owner, cha
24d0: 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 20 76 61 nnel, url_re) va
24e0: 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 24 33 lues ($1, $2, $3
24f0: 2c 20 24 34 2c 20 24 35 2c 20 24 36 29 22 29 0a , $4, $5, $6)").
2500: 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 09 2e ....},....}.....
2510: 62 69 6e 64 28 63 68 61 6e 6e 65 6c 5f 69 64 29 bind(channel_id)
2520: 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c 29 0a 09 .....bind(url)..
2530: 09 09 2e 62 69 6e 64 28 69 76 5f 68 61 73 68 29 ...bind(iv_hash)
2540: 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 .....bind(owner)
2550: 0a 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 .....bind(channe
2560: 6c 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c 5f l).....bind(url_
2570: 72 65 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 re).....execute(
2580: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
2590: 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 4f {....Ok(_) => O
25a0: 6b 28 6d 61 74 63 68 20 75 70 64 61 74 65 20 7b k(match update {
25b0: 0a 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 .....Some(_) =>
25c0: 22 43 68 61 6e 6e 65 6c 20 75 70 64 61 74 65 64 "Channel updated
25d0: 2e 22 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 .",.....None =>
25e0: 22 43 68 61 6e 6e 65 6c 20 61 64 64 65 64 2e 22 "Channel added."
25f0: 2c 0a 09 09 09 7d 29 2c 0a 09 09 09 45 72 72 28 ,....}),....Err(
2600: 73 71 6c 78 3a 3a 45 72 72 6f 72 3a 3a 44 61 74 sqlx::Error::Dat
2610: 61 62 61 73 65 28 65 72 72 29 29 20 3d 3e 20 7b abase(err)) => {
2620: 0a 09 09 09 09 6d 61 74 63 68 20 65 72 72 2e 64 .....match err.d
2630: 6f 77 6e 63 61 73 74 3a 3a 3c 73 71 6c 78 3a 3a owncast::<sqlx::
2640: 70 6f 73 74 67 72 65 73 3a 3a 50 67 44 61 74 61 postgres::PgData
2650: 62 61 73 65 45 72 72 6f 72 3e 28 29 2e 72 6f 75 baseError>().rou
2660: 74 69 6e 65 28 29 20 7b 0a 09 09 09 09 09 53 6f tine() {......So
2670: 6d 65 28 22 5f 62 74 5f 63 68 65 63 6b 5f 75 6e me("_bt_check_un
2680: 69 71 75 65 22 2c 20 29 20 3d 3e 20 7b 0a 09 09 ique", ) => {...
2690: 09 09 09 09 4f 6b 28 22 44 75 70 6c 69 63 61 74 ....Ok("Duplicat
26a0: 65 20 6b 65 79 2e 22 29 0a 09 09 09 09 09 7d 2c e key.")......},
26b0: 0a 09 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e ......Some(_) =>
26c0: 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 44 61 74 {.......Ok("Dat
26d0: 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 0a 09 abase error.")..
26e0: 09 09 09 09 7d 2c 0a 09 09 09 09 09 4e 6f 6e 65 ....},......None
26f0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 => {.......Ok("
2700: 4e 6f 20 64 61 74 61 62 61 73 65 20 65 72 72 6f No database erro
2710: 72 20 65 78 74 72 61 63 74 65 64 2e 22 29 0a 09 r extracted.")..
2720: 09 09 09 09 7d 2c 0a 09 09 09 09 7d 0a 09 09 09 ....},.....}....
2730: 7d 2c 0a 09 09 09 45 72 72 28 65 72 72 29 20 3d },....Err(err) =
2740: 3e 20 7b 0a 09 09 09 09 62 61 69 6c 21 28 22 53 > {.....bail!("S
2750: 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 72 orry, unknown er
2760: 72 6f 72 3a 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 ror:\n{:#?}\n",
2770: 65 72 72 29 3b 0a 09 09 09 7d 2c 0a 09 09 7d 0a err);....},...}.
2780: 09 7d 0a 0a 09 61 73 79 6e 63 20 66 6e 20 61 75 .}...async fn au
2790: 74 6f 66 65 74 63 68 28 26 73 65 6c 66 29 20 2d tofetch(&self) -
27a0: 3e 20 52 65 73 75 6c 74 3c 73 74 64 3a 3a 74 69 > Result<std::ti
27b0: 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3e 20 7b 0a me::Duration> {.
27c0: 09 09 6c 65 74 20 6d 75 74 20 64 65 6c 61 79 20 ..let mut delay
27d0: 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 72 61 74 69 = chrono::Durati
27e0: 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 31 29 3b 0a on::minutes(1);.
27f0: 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d ..let mut conn =
2800: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
2810: 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 re().await.....w
2820: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
2830: 6f 72 6d 61 74 21 28 22 41 75 74 6f 66 65 74 63 ormat!("Autofetc
2840: 68 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b h fetch conn:\n{
2850: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
2860: 29 29 3f 3b 0a 09 09 6c 65 74 20 6e 6f 77 20 3d ))?;...let now =
2870: 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a chrono::Local::
2880: 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 now();...let mut
2890: 20 71 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a 71 queue = sqlx::q
28a0: 75 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 uery("select sou
28b0: 72 63 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 rce_id, next_fet
28c0: 63 68 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 ch, owner from r
28d0: 73 73 74 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 sstg_order natur
28e0: 61 6c 20 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 al left join rss
28f0: 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 tg_source where
2900: 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 next_fetch < now
2910: 28 29 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 () + interval '1
2920: 20 6d 69 6e 75 74 65 27 3b 22 29 0a 09 09 09 2e minute';").....
2930: 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 fetch_all(&mut c
2940: 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 66 onn).await?;...f
2950: 6f 72 20 72 6f 77 20 69 6e 20 71 75 65 75 65 2e or row in queue.
2960: 69 74 65 72 28 29 20 7b 0a 09 09 09 6c 65 74 20 iter() {....let
2970: 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 20 3d source_id: i32 =
2980: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 6f row.try_get("so
2990: 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 6c urce_id")?;....l
29a0: 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 et owner: i64 =
29b0: 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e row.try_get("own
29c0: 65 72 22 29 3f 3b 0a 09 09 09 6c 65 74 20 6e 65 er")?;....let ne
29d0: 78 74 5f 66 65 74 63 68 3a 20 44 61 74 65 54 69 xt_fetch: DateTi
29e0: 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c me<chrono::Local
29f0: 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 > = row.try_get(
2a00: 22 6e 65 78 74 5f 66 65 74 63 68 22 29 3f 3b 0a "next_fetch")?;.
2a10: 09 09 09 69 66 20 6e 65 78 74 5f 66 65 74 63 68 ...if next_fetch
2a20: 20 3c 20 6e 6f 77 20 7b 0a 09 09 09 09 6c 65 74 < now {.....let
2a30: 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 65 20 7b 0a clone = Core {.
2a40: 09 09 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a .....owner_chat:
2a50: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 telegram_bot::U
2a60: 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 serId::new(owner
2a70: 29 2c 0a 09 09 09 09 09 2e 2e 73 65 6c 66 2e 63 ),........self.c
2a80: 6c 6f 6e 65 28 29 0a 09 09 09 09 7d 3b 0a 09 09 lone().....};...
2a90: 09 09 74 6f 6b 69 6f 3a 3a 73 70 61 77 6e 28 61 ..tokio::spawn(a
2aa0: 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 sync move {.....
2ab0: 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 .if let Err(err)
2ac0: 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b 28 26 = clone.check(&
2ad0: 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 source_id, owner
2ae0: 2c 20 74 72 75 65 29 2e 61 77 61 69 74 20 7b 0a , true).await {.
2af0: 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 ......if let Err
2b00: 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 (err) = clone.se
2b10: 6e 64 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b nd(&format!("š
2b20: 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 4e {:?}", err), N
2b30: 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 74 one, None).await
2b40: 20 7b 0a 09 09 09 09 09 09 09 65 70 72 69 6e 74 {........eprint
2b50: 6c 6e 21 28 22 43 68 65 63 6b 20 65 72 72 6f 72 ln!("Check error
2b60: 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 : {}", err);....
2b70: 09 09 09 7d 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 ...};......};...
2b80: 09 09 7d 29 3b 0a 09 09 09 7d 20 65 6c 73 65 20 ..});....} else
2b90: 69 66 20 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 if next_fetch -
2ba0: 6e 6f 77 20 3c 20 64 65 6c 61 79 20 7b 0a 09 09 now < delay {...
2bb0: 09 09 64 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 ..delay = next_f
2bc0: 65 74 63 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 7d etch - now;....}
2bd0: 0a 09 09 7d 3b 0a 09 09 71 75 65 75 65 2e 63 6c ...};...queue.cl
2be0: 65 61 72 28 29 3b 0a 09 09 4f 6b 28 64 65 6c 61 ear();...Ok(dela
2bf0: 79 2e 74 6f 5f 73 74 64 28 29 3f 29 0a 09 7d 0a y.to_std()?)..}.
2c00: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 6c ..pub async fn l
2c10: 69 73 74 3c 53 3e 28 26 73 65 6c 66 2c 20 6f 77 ist<S>(&self, ow
2c20: 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c ner: S) -> Resul
2c30: 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 72 65 t<String>..where
2c40: 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a S: Into<i64> {.
2c50: 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 ..let owner = ow
2c60: 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c ner.into();....l
2c70: 65 74 20 6d 75 74 20 72 65 70 6c 79 3a 20 56 65 et mut reply: Ve
2c80: 63 3c 43 6f 77 3c 73 74 72 3e 3e 20 3d 20 76 65 c<Cow<str>> = ve
2c90: 63 21 5b 5d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 c![];...let mut
2ca0: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
2cb0: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
2cc0: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
2cd0: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 4c 69 t(|| format!("Li
2ce0: 73 74 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e st fetch conn:\n
2cf0: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
2d00: 6c 29 29 3f 3b 0a 09 09 72 65 70 6c 79 2e 70 75 l))?;...reply.pu
2d10: 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 2e 69 sh("Channels:".i
2d20: 6e 74 6f 28 29 29 3b 0a 09 09 6c 65 74 20 72 6f nto());...let ro
2d30: 77 73 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 ws = sqlx::query
2d40: 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 65 5f ("select source_
2d50: 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 65 6e 61 id, channel, ena
2d60: 62 6c 65 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 bled, url, iv_ha
2d70: 73 68 2c 20 75 72 6c 5f 72 65 20 66 72 6f 6d 20 sh, url_re from
2d80: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 rsstg_source whe
2d90: 72 65 20 6f 77 6e 65 72 20 3d 20 24 31 20 6f 72 re owner = $1 or
2da0: 64 65 72 20 62 79 20 73 6f 75 72 63 65 5f 69 64 der by source_id
2db0: 22 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 ").....bind(owne
2dc0: 72 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c r).....fetch_all
2dd0: 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 (&mut conn).awai
2de0: 74 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e t?;...for row in
2df0: 20 72 6f 77 73 2e 69 74 65 72 28 29 20 7b 0a 09 rows.iter() {..
2e00: 09 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 64 3a ..let source_id:
2e10: 20 69 33 32 20 3d 20 72 6f 77 2e 74 72 79 5f 67 i32 = row.try_g
2e20: 65 74 28 22 73 6f 75 72 63 65 5f 69 64 22 29 3f et("source_id")?
2e30: 3b 0a 09 09 09 6c 65 74 20 75 73 65 72 6e 61 6d ;....let usernam
2e40: 65 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 e: &str = row.tr
2e50: 79 5f 67 65 74 28 22 63 68 61 6e 6e 65 6c 22 29 y_get("channel")
2e60: 3f 3b 0a 09 09 09 6c 65 74 20 65 6e 61 62 6c 65 ?;....let enable
2e70: 64 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 d: bool = row.tr
2e80: 79 5f 67 65 74 28 22 65 6e 61 62 6c 65 64 22 29 y_get("enabled")
2e90: 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 ?;....let url: &
2ea0: 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 str = row.try_ge
2eb0: 74 28 22 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 t("url")?;....le
2ec0: 74 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f t iv_hash: Optio
2ed0: 6e 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 n<&str> = row.tr
2ee0: 79 5f 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 y_get("iv_hash")
2ef0: 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c 5f 72 65 ?;....let url_re
2f00: 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 20 3d : Option<&str> =
2f10: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 72 row.try_get("ur
2f20: 6c 5f 72 65 22 29 3f 3b 0a 09 09 09 72 65 70 6c l_re")?;....repl
2f30: 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 y.push(format!("
2f40: 5c 6e 5c 5c 23 ef b8 8f e2 83 a3 20 7b 7d 20 5c \n\\#ļøā£ {} \
2f50: 5c 2a ef b8 8f e2 83 a3 20 60 7b 7d 60 20 7b 7d \*ļøā£ `{}` {}
2f60: 5c 6e f0 9f 94 97 20 60 7b 7d 60 22 2c 20 73 6f \nš `{}`", so
2f70: 75 72 63 65 5f 69 64 2c 20 75 73 65 72 6e 61 6d urce_id, usernam
2f80: 65 2c 20 20 0a 09 09 09 09 6d 61 74 63 68 20 65 e, .....match e
2f90: 6e 61 62 6c 65 64 20 7b 0a 09 09 09 09 09 74 72 nabled {......tr
2fa0: 75 65 20 20 3d 3e 20 22 f0 9f 94 84 20 65 6e 61 ue => "š ena
2fb0: 62 6c 65 64 22 2c 0a 09 09 09 09 09 66 61 6c 73 bled",......fals
2fc0: 65 20 3d 3e 20 22 e2 9b 94 20 64 69 73 61 62 6c e => "ā disabl
2fd0: 65 64 22 2c 0a 09 09 09 09 7d 2c 20 75 72 6c 29 ed",.....}, url)
2fe0: 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 69 66 20 .into());....if
2ff0: 6c 65 74 20 53 6f 6d 65 28 68 61 73 68 29 20 3d let Some(hash) =
3000: 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 72 iv_hash {.....r
3010: 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 eply.push(format
3020: 21 28 22 49 56 3a 20 60 7b 7d 60 22 2c 20 68 61 !("IV: `{}`", ha
3030: 73 68 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 sh).into());....
3040: 7d 0a 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 }....if let Some
3050: 28 72 65 29 20 3d 20 75 72 6c 5f 72 65 20 7b 0a (re) = url_re {.
3060: 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 ....reply.push(f
3070: 6f 72 6d 61 74 21 28 22 52 45 3a 20 60 7b 7d 60 ormat!("RE: `{}`
3080: 22 2c 20 72 65 29 2e 69 6e 74 6f 28 29 29 3b 0a ", re).into());.
3090: 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 ...}...};...Ok(r
30a0: 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 eply.join("\n"))
30b0: 0a 09 7d 0a 7d 0a ..}.}.