0000: 75 73 65 20 61 6e 79 68 6f 77 3a 3a 7b 61 6e 79 use anyhow::{any
0010: 68 6f 77 2c 20 62 61 69 6c 2c 20 43 6f 6e 74 65 how, bail, Conte
0020: 78 74 2c 20 52 65 73 75 6c 74 7d 3b 0a 75 73 65 xt, Result};.use
0030: 20 61 74 6f 6d 5f 73 79 6e 64 69 63 61 74 69 6f atom_syndicatio
0040: 6e 3b 0a 75 73 65 20 63 68 72 6f 6e 6f 3a 3a 44 n;.use chrono::D
0050: 61 74 65 54 69 6d 65 3b 0a 75 73 65 20 63 6f 6e ateTime;.use con
0060: 66 69 67 3b 0a 75 73 65 20 72 65 71 77 65 73 74 fig;.use reqwest
0070: 3b 0a 75 73 65 20 73 71 6c 78 3a 3a 7b 0a 09 70 ;.use sqlx::{..p
0080: 6f 73 74 67 72 65 73 3a 3a 50 67 50 6f 6f 6c 4f ostgres::PgPoolO
0090: 70 74 69 6f 6e 73 2c 0a 09 52 6f 77 2c 0a 7d 3b ptions,..Row,.};
00a0: 0a 75 73 65 20 72 73 73 3b 0a 75 73 65 20 73 74 .use rss;.use st
00b0: 64 3a 3a 7b 0a 09 63 6f 6c 6c 65 63 74 69 6f 6e d::{..collection
00c0: 73 3a 3a 7b 0a 09 09 42 54 72 65 65 4d 61 70 2c s::{...BTreeMap,
00d0: 0a 09 09 48 61 73 68 53 65 74 2c 0a 09 7d 2c 0a ...HashSet,..},.
00e0: 09 73 79 6e 63 3a 3a 7b 41 72 63 2c 20 4d 75 74 .sync::{Arc, Mut
00f0: 65 78 7d 2c 0a 7d 3b 0a 75 73 65 20 74 65 6c 65 ex},.};.use tele
0100: 67 72 61 6d 5f 62 6f 74 3b 0a 0a 23 5b 64 65 72 gram_bot;..#[der
0110: 69 76 65 28 43 6c 6f 6e 65 29 5d 0a 70 75 62 20 ive(Clone)].pub
0120: 73 74 72 75 63 74 20 43 6f 72 65 20 7b 0a 09 6f struct Core {..o
0130: 77 6e 65 72 3a 20 69 36 34 2c 0a 09 61 70 69 5f wner: i64,..api_
0140: 6b 65 79 3a 20 53 74 72 69 6e 67 2c 0a 09 6f 77 key: String,..ow
0150: 6e 65 72 5f 63 68 61 74 3a 20 74 65 6c 65 67 72 ner_chat: telegr
0160: 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 2c 0a am_bot::UserId,.
0170: 09 70 75 62 20 74 67 3a 20 74 65 6c 65 67 72 61 .pub tg: telegra
0180: 6d 5f 62 6f 74 3a 3a 41 70 69 2c 0a 09 70 75 62 m_bot::Api,..pub
0190: 20 6d 79 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f my: telegram_bo
01a0: 74 3a 3a 55 73 65 72 2c 0a 09 70 6f 6f 6c 3a 20 t::User,..pool:
01b0: 73 71 6c 78 3a 3a 50 6f 6f 6c 3c 73 71 6c 78 3a sqlx::Pool<sqlx:
01c0: 3a 50 6f 73 74 67 72 65 73 3e 2c 0a 09 73 6f 75 :Postgres>,..sou
01d0: 72 63 65 73 3a 20 41 72 63 3c 4d 75 74 65 78 3c rces: Arc<Mutex<
01e0: 48 61 73 68 53 65 74 3c 41 72 63 3c 69 33 32 3e HashSet<Arc<i32>
01f0: 3e 3e 3e 2c 0a 7d 0a 0a 69 6d 70 6c 20 43 6f 72 >>>,.}..impl Cor
0200: 65 20 7b 0a 09 70 75 62 20 61 73 79 6e 63 20 66 e {..pub async f
0210: 6e 20 6e 65 77 28 73 65 74 74 69 6e 67 73 3a 20 n new(settings:
0220: 63 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 29 20 config::Config)
0230: 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 72 65 3e 20 -> Result<Core>
0240: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
0250: 73 65 74 74 69 6e 67 73 2e 67 65 74 5f 69 6e 74 settings.get_int
0260: 28 22 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 6c 65 ("owner")?;...le
0270: 74 20 61 70 69 5f 6b 65 79 20 3d 20 73 65 74 74 t api_key = sett
0280: 69 6e 67 73 2e 67 65 74 5f 73 74 72 28 22 61 70 ings.get_str("ap
0290: 69 5f 6b 65 79 22 29 3f 3b 0a 09 09 6c 65 74 20 i_key")?;...let
02a0: 74 67 20 3d 20 74 65 6c 65 67 72 61 6d 5f 62 6f tg = telegram_bo
02b0: 74 3a 3a 41 70 69 3a 3a 6e 65 77 28 26 61 70 69 t::Api::new(&api
02c0: 5f 6b 65 79 29 3b 0a 09 09 6c 65 74 20 63 6f 72 _key);...let cor
02d0: 65 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 6f 77 e = Core {....ow
02e0: 6e 65 72 3a 20 6f 77 6e 65 72 2c 0a 09 09 09 61 ner: owner,....a
02f0: 70 69 5f 6b 65 79 3a 20 61 70 69 5f 6b 65 79 2e pi_key: api_key.
0300: 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 6d 79 3a 20 clone(),....my:
0310: 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 72 61 6d tg.send(telegram
0320: 5f 62 6f 74 3a 3a 47 65 74 4d 65 29 2e 61 77 61 _bot::GetMe).awa
0330: 69 74 3f 2c 0a 09 09 09 74 67 3a 20 74 67 2c 0a it?,....tg: tg,.
0340: 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 74 ...owner_chat: t
0350: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 elegram_bot::Use
0360: 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 2c rId::new(owner),
0370: 0a 09 09 09 70 6f 6f 6c 3a 20 50 67 50 6f 6f 6c ....pool: PgPool
0380: 4f 70 74 69 6f 6e 73 3a 3a 6e 65 77 28 29 0a 09 Options::new()..
0390: 09 09 09 2e 6d 61 78 5f 63 6f 6e 6e 65 63 74 69 ....max_connecti
03a0: 6f 6e 73 28 35 29 0a 09 09 09 09 2e 63 6f 6e 6e ons(5)......conn
03b0: 65 63 74 5f 74 69 6d 65 6f 75 74 28 73 74 64 3a ect_timeout(std:
03c0: 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a :time::Duration:
03d0: 3a 6e 65 77 28 33 30 30 2c 20 30 29 29 0a 09 09 :new(300, 0))...
03e0: 09 09 2e 69 64 6c 65 5f 74 69 6d 65 6f 75 74 28 ...idle_timeout(
03f0: 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 std::time::Durat
0400: 69 6f 6e 3a 3a 6e 65 77 28 36 30 2c 20 30 29 29 ion::new(60, 0))
0410: 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 74 5f 6c 61 ......connect_la
0420: 7a 79 28 26 73 65 74 74 69 6e 67 73 2e 67 65 74 zy(&settings.get
0430: 5f 73 74 72 28 22 70 67 22 29 3f 29 3f 2c 0a 09 _str("pg")?)?,..
0440: 09 09 73 6f 75 72 63 65 73 3a 20 41 72 63 3a 3a ..sources: Arc::
0450: 6e 65 77 28 4d 75 74 65 78 3a 3a 6e 65 77 28 48 new(Mutex::new(H
0460: 61 73 68 53 65 74 3a 3a 6e 65 77 28 29 29 29 2c ashSet::new())),
0470: 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 63 6c 6f 6e ...};...let clon
0480: 65 20 3d 20 63 6f 72 65 2e 63 6c 6f 6e 65 28 29 e = core.clone()
0490: 3b 0a 09 09 74 6f 6b 69 6f 3a 3a 73 70 61 77 6e ;...tokio::spawn
04a0: 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 (async move {...
04b0: 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 .if let Err(err)
04c0: 20 3d 20 26 63 6c 6f 6e 65 2e 61 75 74 6f 66 65 = &clone.autofe
04d0: 74 63 68 28 29 2e 61 77 61 69 74 20 7b 0a 09 09 tch().await {...
04e0: 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 ..if let Err(err
04f0: 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 26 ) = clone.send(&
0500: 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a format!("š {:
0510: 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 2c ?}", err), None,
0520: 20 4e 6f 6e 65 29 20 7b 0a 09 09 09 09 09 65 70 None) {......ep
0530: 72 69 6e 74 6c 6e 21 28 22 41 75 74 6f 66 65 74 rintln!("Autofet
0540: 63 68 20 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 ch error: {}", e
0550: 72 72 29 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 7d rr);.....};....}
0560: 0a 09 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 ...});...Ok(core
0570: 29 0a 09 7d 0a 0a 09 70 75 62 20 66 6e 20 73 74 )..}...pub fn st
0580: 72 65 61 6d 28 26 73 65 6c 66 29 20 2d 3e 20 74 ream(&self) -> t
0590: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 70 64 elegram_bot::Upd
05a0: 61 74 65 73 53 74 72 65 61 6d 20 7b 0a 09 09 73 atesStream {...s
05b0: 65 6c 66 2e 74 67 2e 73 74 72 65 61 6d 28 29 0a elf.tg.stream().
05c0: 09 7d 0a 0a 09 70 75 62 20 66 6e 20 73 65 6e 64 .}...pub fn send
05d0: 3c 53 3e 28 26 73 65 6c 66 2c 20 6d 73 67 3a 20 <S>(&self, msg:
05e0: 53 2c 20 74 61 72 67 65 74 3a 20 4f 70 74 69 6f S, target: Optio
05f0: 6e 3c 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a n<telegram_bot::
0600: 55 73 65 72 49 64 3e 2c 20 70 61 72 73 65 5f 6d UserId>, parse_m
0610: 6f 64 65 3a 20 4f 70 74 69 6f 6e 3c 74 65 6c 65 ode: Option<tele
0620: 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a gram_bot::types:
0630: 3a 50 61 72 73 65 4d 6f 64 65 3e 29 20 2d 3e 20 :ParseMode>) ->
0640: 52 65 73 75 6c 74 3c 28 29 3e 0a 09 77 68 65 72 Result<()>..wher
0650: 65 20 53 3a 20 49 6e 74 6f 3c 53 74 72 69 6e 67 e S: Into<String
0660: 3e 20 7b 0a 09 09 6c 65 74 20 6d 73 67 3a 20 53 > {...let msg: S
0670: 74 72 69 6e 67 20 3d 20 6d 73 67 2e 69 6e 74 6f tring = msg.into
0680: 28 29 3b 0a 09 09 6c 65 74 20 70 61 72 73 65 5f ();...let parse_
0690: 6d 6f 64 65 20 3d 20 6d 61 74 63 68 20 70 61 72 mode = match par
06a0: 73 65 5f 6d 6f 64 65 20 7b 0a 09 09 09 53 6f 6d se_mode {....Som
06b0: 65 28 6d 6f 64 65 29 20 3d 3e 20 6d 6f 64 65 2c e(mode) => mode,
06c0: 0a 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 65 6c 65 ....None => tele
06d0: 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a gram_bot::types:
06e0: 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c :ParseMode::Html
06f0: 2c 0a 09 09 7d 3b 0a 09 09 73 65 6c 66 2e 74 67 ,...};...self.tg
0700: 2e 73 70 61 77 6e 28 74 65 6c 65 67 72 61 6d 5f .spawn(telegram_
0710: 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 bot::SendMessage
0720: 3a 3a 6e 65 77 28 6d 61 74 63 68 20 74 61 72 67 ::new(match targ
0730: 65 74 20 7b 0a 09 09 09 53 6f 6d 65 28 75 73 65 et {....Some(use
0740: 72 29 20 3d 3e 20 75 73 65 72 2c 0a 09 09 09 4e r) => user,....N
0750: 6f 6e 65 20 3d 3e 20 73 65 6c 66 2e 6f 77 6e 65 one => self.owne
0760: 72 5f 63 68 61 74 2c 0a 09 09 7d 2c 20 6d 73 67 r_chat,...}, msg
0770: 2e 74 6f 5f 6f 77 6e 65 64 28 29 29 2e 70 61 72 .to_owned()).par
0780: 73 65 5f 6d 6f 64 65 28 70 61 72 73 65 5f 6d 6f se_mode(parse_mo
0790: 64 65 29 29 3b 0a 09 09 4f 6b 28 28 29 29 0a 09 de));...Ok(())..
07a0: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
07b0: 20 63 68 65 63 6b 3c 53 3e 28 26 73 65 6c 66 2c check<S>(&self,
07c0: 20 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 id: &i32, owner
07d0: 3a 20 53 2c 20 72 65 61 6c 3a 20 62 6f 6f 6c 29 : S, real: bool)
07e0: 20 2d 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e -> Result<Strin
07f0: 67 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 g>..where S: Int
0800: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6d o<i64> {...let m
0810: 75 74 20 70 6f 73 74 65 64 3a 20 69 33 32 20 3d ut posted: i32 =
0820: 20 30 3b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a 0;...let owner:
0830: 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 i64 = owner.int
0840: 6f 28 29 3b 0a 09 09 6c 65 74 20 69 64 20 3d 20 o();...let id =
0850: 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 73 65 74 {....let mut set
0860: 20 3d 20 73 65 6c 66 2e 73 6f 75 72 63 65 73 2e = self.sources.
0870: 6c 6f 63 6b 28 29 2e 75 6e 77 72 61 70 28 29 3b lock().unwrap();
0880: 0a 09 09 09 6d 61 74 63 68 20 73 65 74 2e 67 65 ....match set.ge
0890: 74 28 69 64 29 20 7b 0a 09 09 09 09 53 6f 6d 65 t(id) {.....Some
08a0: 28 69 64 29 20 3d 3e 20 69 64 2e 63 6c 6f 6e 65 (id) => id.clone
08b0: 28 29 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 (),.....None =>
08c0: 7b 0a 09 09 09 09 09 6c 65 74 20 69 64 20 3d 20 {......let id =
08d0: 41 72 63 3a 3a 6e 65 77 28 2a 69 64 29 3b 0a 09 Arc::new(*id);..
08e0: 09 09 09 09 73 65 74 2e 69 6e 73 65 72 74 28 69 ....set.insert(i
08f0: 64 2e 63 6c 6f 6e 65 28 29 29 3b 0a 09 09 09 09 d.clone());.....
0900: 09 69 64 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 .id.clone().....
0910: 7d 2c 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 6c },....}...};...l
0920: 65 74 20 63 6f 75 6e 74 20 3d 20 41 72 63 3a 3a et count = Arc::
0930: 73 74 72 6f 6e 67 5f 63 6f 75 6e 74 28 26 69 64 strong_count(&id
0940: 29 3b 0a 09 09 69 66 20 63 6f 75 6e 74 20 3d 3d );...if count ==
0950: 20 32 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 2 {....let mut
0960: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
0970: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
0980: 0a 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 ......with_conte
0990: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 51 xt(|| format!("Q
09a0: 75 65 72 79 20 71 75 65 75 65 20 66 65 74 63 68 uery queue fetch
09b0: 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 conn:\n{:?}", &
09c0: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 self.pool))?;...
09d0: 09 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c 78 3a .let row = sqlx:
09e0: 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 73 :query("select s
09f0: 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 ource_id, channe
0a00: 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 l_id, url, iv_ha
0a10: 73 68 2c 20 6f 77 6e 65 72 2c 20 75 72 6c 5f 72 sh, owner, url_r
0a20: 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 e from rsstg_sou
0a30: 72 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 rce where source
0a40: 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e _id = $1 and own
0a50: 65 72 20 3d 20 24 32 22 29 0a 09 09 09 09 2e 62 er = $2")......b
0a60: 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 2e 62 69 ind(*id)......bi
0a70: 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 09 2e 66 nd(owner)......f
0a80: 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 63 6f etch_one(&mut co
0a90: 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 nn).await......w
0aa0: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
0ab0: 6f 72 6d 61 74 21 28 22 51 75 65 72 79 20 73 6f ormat!("Query so
0ac0: 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 urce:\n{:?}", &s
0ad0: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 elf.pool))?;....
0ae0: 64 72 6f 70 28 63 6f 6e 6e 29 3b 0a 09 09 09 6c drop(conn);....l
0af0: 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 et channel_id: i
0b00: 36 34 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 64 = row.try_get
0b10: 28 22 63 68 61 6e 6e 65 6c 5f 69 64 22 29 3f 3b ("channel_id")?;
0b20: 0a 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 73 74 ....let url: &st
0b30: 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 r = row.try_get(
0b40: 22 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 "url")?;....let
0b50: 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c iv_hash: Option<
0b60: 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f &str> = row.try_
0b70: 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f 3b get("iv_hash")?;
0b80: 0a 09 09 09 6c 65 74 20 75 72 6c 5f 72 65 20 3d ....let url_re =
0b90: 20 6d 61 74 63 68 20 72 6f 77 2e 74 72 79 5f 67 match row.try_g
0ba0: 65 74 28 22 75 72 6c 5f 72 65 22 29 3f 20 7b 0a et("url_re")? {.
0bb0: 09 09 09 09 53 6f 6d 65 28 78 29 20 3d 3e 20 53 ....Some(x) => S
0bc0: 6f 6d 65 28 73 65 64 72 65 67 65 78 3a 3a 52 65 ome(sedregex::Re
0bd0: 70 6c 61 63 65 43 6f 6d 6d 61 6e 64 3a 3a 6e 65 placeCommand::ne
0be0: 77 28 78 29 3f 29 2c 0a 09 09 09 09 4e 6f 6e 65 w(x)?),.....None
0bf0: 20 3d 3e 20 4e 6f 6e 65 2c 0a 09 09 09 7d 3b 0a => None,....};.
0c00: 09 09 09 6c 65 74 20 64 65 73 74 69 6e 61 74 69 ...let destinati
0c10: 6f 6e 20 3d 20 6d 61 74 63 68 20 72 65 61 6c 20 on = match real
0c20: 7b 0a 09 09 09 09 74 72 75 65 20 3d 3e 20 74 65 {.....true => te
0c30: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 legram_bot::User
0c40: 49 64 3a 3a 6e 65 77 28 63 68 61 6e 6e 65 6c 5f Id::new(channel_
0c50: 69 64 29 2c 0a 09 09 09 09 66 61 6c 73 65 20 3d id),.....false =
0c60: 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a > telegram_bot::
0c70: 55 73 65 72 49 64 3a 3a 6e 65 77 28 72 6f 77 2e UserId::new(row.
0c80: 74 72 79 5f 67 65 74 28 22 6f 77 6e 65 72 22 29 try_get("owner")
0c90: 3f 29 2c 0a 09 09 09 7d 3b 0a 09 09 09 6c 65 74 ?),....};....let
0ca0: 20 6d 75 74 20 74 68 69 73 5f 66 65 74 63 68 3a mut this_fetch:
0cb0: 20 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 6d 65 Option<DateTime
0cc0: 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 <chrono::FixedOf
0cd0: 66 73 65 74 3e 3e 20 3d 20 4e 6f 6e 65 3b 0a 09 fset>> = None;..
0ce0: 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 73 3a ..let mut posts:
0cf0: 20 42 54 72 65 65 4d 61 70 3c 44 61 74 65 54 69 BTreeMap<DateTi
0d00: 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 me<chrono::Fixed
0d10: 4f 66 66 73 65 74 3e 2c 20 53 74 72 69 6e 67 3e Offset>, String>
0d20: 20 3d 20 42 54 72 65 65 4d 61 70 3a 3a 6e 65 77 = BTreeMap::new
0d30: 28 29 3b 0a 09 09 09 6c 65 74 20 63 6f 6e 74 65 ();....let conte
0d40: 6e 74 20 3d 20 72 65 71 77 65 73 74 3a 3a 67 65 nt = reqwest::ge
0d50: 74 28 75 72 6c 29 2e 61 77 61 69 74 3f 2e 62 79 t(url).await?.by
0d60: 74 65 73 28 29 2e 61 77 61 69 74 3f 3b 0a 09 09 tes().await?;...
0d70: 09 6d 61 74 63 68 20 72 73 73 3a 3a 43 68 61 6e .match rss::Chan
0d80: 6e 65 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 nel::read_from(&
0d90: 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 content[..]) {..
0da0: 09 09 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 7b ...Ok(feed) => {
0db0: 0a 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 69 ......for item i
0dc0: 6e 20 66 65 65 64 2e 69 74 65 6d 73 28 29 20 7b n feed.items() {
0dd0: 0a 09 09 09 09 09 09 6d 61 74 63 68 20 69 74 65 .......match ite
0de0: 6d 2e 6c 69 6e 6b 28 29 20 7b 0a 09 09 09 09 09 m.link() {......
0df0: 09 09 53 6f 6d 65 28 6c 69 6e 6b 29 20 3d 3e 20 ..Some(link) =>
0e00: 7b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 64 61 {.........let da
0e10: 74 65 20 3d 20 6d 61 74 63 68 20 69 74 65 6d 2e te = match item.
0e20: 70 75 62 5f 64 61 74 65 28 29 20 7b 0a 09 09 09 pub_date() {....
0e30: 09 09 09 09 09 09 53 6f 6d 65 28 66 65 65 64 5f ......Some(feed_
0e40: 64 61 74 65 29 20 3d 3e 20 44 61 74 65 54 69 6d date) => DateTim
0e50: 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 e::parse_from_rf
0e60: 63 32 38 32 32 28 66 65 65 64 5f 64 61 74 65 29 c2822(feed_date)
0e70: 2c 0a 09 09 09 09 09 09 09 09 09 4e 6f 6e 65 20 ,..........None
0e80: 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 => DateTime::par
0e90: 73 65 5f 66 72 6f 6d 5f 72 66 63 33 33 33 39 28 se_from_rfc3339(
0ea0: 26 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 63 6f 72 &item.dublin_cor
0eb0: 65 5f 65 78 74 28 29 2e 75 6e 77 72 61 70 28 29 e_ext().unwrap()
0ec0: 2e 64 61 74 65 73 28 29 5b 30 5d 29 2c 0a 09 09 .dates()[0]),...
0ed0: 09 09 09 09 09 09 7d 3f 3b 0a 09 09 09 09 09 09 ......}?;.......
0ee0: 09 09 6c 65 74 20 75 72 6c 20 3d 20 6c 69 6e 6b ..let url = link
0ef0: 3b 0a 09 09 09 09 09 09 09 09 70 6f 73 74 73 2e ;.........posts.
0f00: 69 6e 73 65 72 74 28 64 61 74 65 2e 63 6c 6f 6e insert(date.clon
0f10: 65 28 29 2c 20 75 72 6c 2e 69 6e 74 6f 28 29 29 e(), url.into())
0f20: 3b 0a 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 ;........},.....
0f30: 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 7d 0a 09 09 ...None => {}...
0f40: 09 09 09 09 7d 0a 09 09 09 09 09 7d 3b 0a 09 09 ....}......};...
0f50: 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 65 72 72 ..},.....Err(err
0f60: 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 72 20 7b ) => match err {
0f70: 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 6f 72 ......rss::Error
0f80: 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 74 54 61 ::InvalidStartTa
0f90: 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 6c 65 74 g => {.......let
0fa0: 20 66 65 65 64 20 3d 20 61 74 6f 6d 5f 73 79 6e feed = atom_syn
0fb0: 64 69 63 61 74 69 6f 6e 3a 3a 46 65 65 64 3a 3a dication::Feed::
0fc0: 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 read_from(&conte
0fd0: 6e 74 5b 2e 2e 5d 29 0a 09 09 09 09 09 09 09 2e nt[..]).........
0fe0: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
0ff0: 66 6f 72 6d 61 74 21 28 22 50 72 6f 62 6c 65 6d format!("Problem
1000: 20 6f 70 65 6e 69 6e 67 20 66 65 65 64 20 75 72 opening feed ur
1010: 6c 3a 5c 6e 7b 7d 22 2c 20 26 75 72 6c 29 29 3f l:\n{}", &url))?
1020: 3b 0a 09 09 09 09 09 09 66 6f 72 20 69 74 65 6d ;.......for item
1030: 20 69 6e 20 66 65 65 64 2e 65 6e 74 72 69 65 73 in feed.entries
1040: 28 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 () {........let
1050: 64 61 74 65 20 3d 20 69 74 65 6d 2e 70 75 62 6c date = item.publ
1060: 69 73 68 65 64 28 29 2e 75 6e 77 72 61 70 28 29 ished().unwrap()
1070: 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c ;........let url
1080: 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 29 5b = item.links()[
1090: 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 09 09 09 0].href();......
10a0: 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 64 ..posts.insert(d
10b0: 61 74 65 2e 63 6c 6f 6e 65 28 29 2c 20 75 72 6c ate.clone(), url
10c0: 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 09 09 09 .into());.......
10d0: 7d 3b 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 };......},......
10e0: 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 6f 66 20 rss::Error::Eof
10f0: 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f 20 3d 3e => (),......_ =>
1100: 20 62 61 69 6c 21 28 22 55 6e 73 75 70 70 6f 72 bail!("Unsuppor
1110: 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 64 20 63 ted or mangled c
1120: 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b ontent:\n{:?}\n{
1130: 3a 23 3f 7d 5c 6e 22 2c 20 26 75 72 6c 2c 20 65 :#?}\n", &url, e
1140: 72 72 29 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a rr).....}....};.
1150: 09 09 09 66 6f 72 20 28 64 61 74 65 2c 20 75 72 ...for (date, ur
1160: 6c 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 65 72 l) in posts.iter
1170: 28 29 20 7b 0a 09 09 09 09 6c 65 74 20 6d 75 74 () {.....let mut
1180: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
1190: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
11a0: 74 0a 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e t.......with_con
11b0: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
11c0: 22 43 68 65 63 6b 20 70 6f 73 74 20 66 65 74 63 "Check post fetc
11d0: 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 h conn:\n{:?}",
11e0: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 &self.pool))?;..
11f0: 09 09 09 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c ...let row = sql
1200: 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 x::query("select
1210: 20 65 78 69 73 74 73 28 73 65 6c 65 63 74 20 74 exists(select t
1220: 72 75 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 rue from rsstg_p
1230: 6f 73 74 20 77 68 65 72 65 20 75 72 6c 20 3d 20 ost where url =
1240: 24 31 20 61 6e 64 20 73 6f 75 72 63 65 5f 69 64 $1 and source_id
1250: 20 3d 20 24 32 29 20 61 73 20 65 78 69 73 74 73 = $2) as exists
1260: 3b 22 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 75 ;").......bind(u
1270: 72 6c 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 2a rl).......bind(*
1280: 69 64 29 0a 09 09 09 09 09 2e 66 65 74 63 68 5f id).......fetch_
1290: 6f 6e 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 one(&mut conn).a
12a0: 77 61 69 74 0a 09 09 09 09 09 2e 77 69 74 68 5f wait.......with_
12b0: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
12c0: 74 21 28 22 43 68 65 63 6b 20 70 6f 73 74 3a 5c t!("Check post:\
12d0: 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f n{:?}", &conn))?
12e0: 3b 0a 09 09 09 09 6c 65 74 20 65 78 69 73 74 73 ;.....let exists
12f0: 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 : bool = row.try
1300: 5f 67 65 74 28 22 65 78 69 73 74 73 22 29 3f 3b _get("exists")?;
1310: 0a 09 09 09 09 69 66 20 21 20 65 78 69 73 74 73 .....if ! exists
1320: 20 7b 0a 09 09 09 09 09 69 66 20 74 68 69 73 5f {......if this_
1330: 66 65 74 63 68 20 3d 3d 20 4e 6f 6e 65 20 7c 7c fetch == None ||
1340: 20 2a 64 61 74 65 20 3e 20 74 68 69 73 5f 66 65 *date > this_fe
1350: 74 63 68 2e 75 6e 77 72 61 70 28 29 20 7b 0a 09 tch.unwrap() {..
1360: 09 09 09 09 09 74 68 69 73 5f 66 65 74 63 68 20 .....this_fetch
1370: 3d 20 53 6f 6d 65 28 2a 64 61 74 65 29 3b 0a 09 = Some(*date);..
1380: 09 09 09 09 7d 3b 0a 09 09 09 09 09 73 65 6c 66 ....};......self
1390: 2e 74 67 2e 73 65 6e 64 28 20 6d 61 74 63 68 20 .tg.send( match
13a0: 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 iv_hash {.......
13b0: 09 53 6f 6d 65 28 68 61 73 68 29 20 3d 3e 20 74 .Some(hash) => t
13c0: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e elegram_bot::Sen
13d0: 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 dMessage::new(de
13e0: 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 stination, forma
13f0: 74 21 28 22 3c 61 20 68 72 65 66 3d 5c 22 68 74 t!("<a href=\"ht
1400: 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 72 tps://t.me/iv?ur
1410: 6c 3d 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c 22 3e l={}&rhash={}\">
1420: 20 3c 2f 61 3e 7b 30 7d 22 2c 20 6d 61 74 63 68 </a>{0}", match
1430: 20 75 72 6c 5f 72 65 20 7b 0a 09 09 09 09 09 09 url_re {.......
1440: 09 09 53 6f 6d 65 28 72 65 66 20 78 29 20 3d 3e ..Some(ref x) =>
1450: 20 78 2e 65 78 65 63 75 74 65 28 75 72 6c 29 2e x.execute(url).
1460: 74 6f 5f 73 74 72 69 6e 67 28 29 2c 0a 09 09 09 to_string(),....
1470: 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 75 72 6c .....None => url
1480: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 2c 0a 09 09 .to_string(),...
1490: 09 09 09 09 09 7d 2c 20 68 61 73 68 29 29 2c 0a .....}, hash)),.
14a0: 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 .......None => t
14b0: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e elegram_bot::Sen
14c0: 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 dMessage::new(de
14d0: 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 stination, forma
14e0: 74 21 28 22 7b 7d 22 2c 20 75 72 6c 29 29 2c 0a t!("{}", url)),.
14f0: 09 09 09 09 09 09 7d 2e 70 61 72 73 65 5f 6d 6f ......}.parse_mo
1500: 64 65 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a de(telegram_bot:
1510: 3a 74 79 70 65 73 3a 3a 50 61 72 73 65 4d 6f 64 :types::ParseMod
1520: 65 3a 3a 48 74 6d 6c 29 29 2e 61 77 61 69 74 0a e::Html)).await.
1530: 09 09 09 09 09 09 2e 63 6f 6e 74 65 78 74 28 22 .......context("
1540: 43 61 6e 27 74 20 70 6f 73 74 20 6d 65 73 73 61 Can't post messa
1550: 67 65 3a 22 29 3f 3b 0a 09 09 09 09 09 73 71 6c ge:")?;......sql
1560: 78 3a 3a 71 75 65 72 79 28 22 69 6e 73 65 72 74 x::query("insert
1570: 20 69 6e 74 6f 20 72 73 73 74 67 5f 70 6f 73 74 into rsstg_post
1580: 20 28 73 6f 75 72 63 65 5f 69 64 2c 20 70 6f 73 (source_id, pos
1590: 74 65 64 2c 20 75 72 6c 29 20 76 61 6c 75 65 73 ted, url) values
15a0: 20 28 24 31 2c 20 24 32 2c 20 24 33 29 3b 22 29 ($1, $2, $3);")
15b0: 0a 09 09 09 09 09 09 2e 62 69 6e 64 28 2a 69 64 ........bind(*id
15c0: 29 0a 09 09 09 09 09 09 2e 62 69 6e 64 28 64 61 )........bind(da
15d0: 74 65 29 0a 09 09 09 09 09 09 2e 62 69 6e 64 28 te)........bind(
15e0: 75 72 6c 29 0a 09 09 09 09 09 09 2e 65 78 65 63 url)........exec
15f0: 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 ute(&mut conn).a
1600: 77 61 69 74 0a 09 09 09 09 09 09 2e 77 69 74 68 wait........with
1610: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
1620: 61 74 21 28 22 52 65 63 6f 72 64 20 70 6f 73 74 at!("Record post
1630: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 :\n{:?}", &conn)
1640: 29 3f 3b 0a 09 09 09 09 09 64 72 6f 70 28 63 6f )?;......drop(co
1650: 6e 6e 29 3b 0a 09 09 09 09 09 74 6f 6b 69 6f 3a nn);......tokio:
1660: 3a 74 69 6d 65 3a 3a 73 6c 65 65 70 28 73 74 64 :time::sleep(std
1670: 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e ::time::Duration
1680: 3a 3a 6e 65 77 28 34 2c 20 30 29 29 2e 61 77 61 ::new(4, 0)).awa
1690: 69 74 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 09 70 it;.....};.....p
16a0: 6f 73 74 65 64 20 2b 3d 20 31 3b 0a 09 09 09 7d osted += 1;....}
16b0: 3b 0a 09 09 09 70 6f 73 74 73 2e 63 6c 65 61 72 ;....posts.clear
16c0: 28 29 3b 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 6d ();...};...let m
16d0: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 ut conn = self.p
16e0: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
16f0: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
1700: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1710: 22 55 70 64 61 74 65 20 73 63 72 61 70 65 20 66 "Update scrape f
1720: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
1730: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
1740: 3b 0a 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 ;...sqlx::query(
1750: 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f "update rsstg_so
1760: 75 72 63 65 20 73 65 74 20 6c 61 73 74 5f 73 63 urce set last_sc
1770: 72 61 70 65 20 3d 20 6e 6f 77 28 29 20 77 68 65 rape = now() whe
1780: 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 re source_id = $
1790: 31 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 2a 69 1;").....bind(*i
17a0: 64 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 d).....execute(&
17b0: 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a mut conn).await.
17c0: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
17d0: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 (|| format!("Upd
17e0: 61 74 65 20 73 63 72 61 70 65 3a 5c 6e 7b 3a 3f ate scrape:\n{:?
17f0: 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 }", &conn))?;...
1800: 4f 6b 28 66 6f 72 6d 61 74 21 28 22 50 6f 73 74 Ok(format!("Post
1810: 65 64 3a 20 7b 7d 22 2c 20 26 70 6f 73 74 65 64 ed: {}", &posted
1820: 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e ))..}...pub asyn
1830: 63 20 66 6e 20 64 65 6c 65 74 65 3c 53 3e 28 26 c fn delete<S>(&
1840: 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a self, source_id:
1850: 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 &i32, owner: S)
1860: 20 2d 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e -> Result<Strin
1870: 67 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 g>..where S: Int
1880: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f o<i64> {...let o
1890: 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 wner: i64 = owne
18a0: 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 r.into();...let
18b0: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
18c0: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
18d0: 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f wait.....with_co
18e0: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
18f0: 28 22 44 65 6c 65 74 65 20 66 65 74 63 68 20 63 ("Delete fetch c
1900: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
1910: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 lf.pool))?;...ma
1920: 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 tch sqlx::query(
1930: 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 "delete from rss
1940: 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 tg_source where
1950: 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 source_id = $1 a
1960: 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 3b 22 29 nd owner = $2;")
1970: 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 .....bind(source
1980: 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 _id).....bind(ow
1990: 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 65 ner).....execute
19a0: 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 (&mut conn).awai
19b0: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
19c0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 xt(|| format!("D
19d0: 65 6c 65 74 65 20 73 6f 75 72 63 65 20 72 75 6c elete source rul
19e0: 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 e:\n{:?}", &self
19f0: 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 .pool))?.....row
1a00: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
1a10: 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 ..0 => { Ok("No
1a20: 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 data found found
1a30: 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 20 .".to_string())
1a40: 7d 2c 0a 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 },....x => { Ok(
1a50: 66 6f 72 6d 61 74 21 28 22 7b 7d 20 73 6f 75 72 format!("{} sour
1a60: 63 65 73 20 72 65 6d 6f 76 65 64 2e 22 2c 20 78 ces removed.", x
1a70: 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 )) },...}..}...p
1a80: 75 62 20 61 73 79 6e 63 20 66 6e 20 63 6c 65 61 ub async fn clea
1a90: 6e 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 n<S>(&self, sour
1aa0: 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e ce_id: &i32, own
1ab0: 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 er: S) -> Result
1ac0: 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 72 65 20 <String>..where
1ad0: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
1ae0: 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 .let owner: i64
1af0: 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a = owner.into();.
1b00: 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d ..let mut conn =
1b10: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
1b20: 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 re().await.....w
1b30: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
1b40: 6f 72 6d 61 74 21 28 22 43 6c 65 61 6e 20 66 65 ormat!("Clean fe
1b50: 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 tch conn:\n{:?}"
1b60: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b , &self.pool))?;
1b70: 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 ...match sqlx::q
1b80: 75 65 72 79 28 22 64 65 6c 65 74 65 20 66 72 6f uery("delete fro
1b90: 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 70 20 75 m rsstg_post p u
1ba0: 73 69 6e 67 20 72 73 73 74 67 5f 73 6f 75 72 63 sing rsstg_sourc
1bb0: 65 20 73 20 77 68 65 72 65 20 70 2e 73 6f 75 72 e s where p.sour
1bc0: 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f ce_id = $1 and o
1bd0: 77 6e 65 72 20 3d 20 24 32 20 61 6e 64 20 70 2e wner = $2 and p.
1be0: 73 6f 75 72 63 65 5f 69 64 20 3d 20 73 2e 73 6f source_id = s.so
1bf0: 75 72 63 65 5f 69 64 3b 22 29 0a 09 09 09 2e 62 urce_id;").....b
1c00: 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 ind(source_id)..
1c10: 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 ...bind(owner)..
1c20: 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 ...execute(&mut
1c30: 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e conn).await.....
1c40: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
1c50: 66 6f 72 6d 61 74 21 28 22 43 6c 65 61 6e 20 73 format!("Clean s
1c60: 65 65 6e 20 70 6f 73 74 73 3a 5c 6e 7b 3a 3f 7d een posts:\n{:?}
1c70: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
1c80: 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 .....rows_affect
1c90: 65 64 28 29 20 7b 0a 09 09 09 30 20 3d 3e 20 7b ed() {....0 => {
1ca0: 20 4f 6b 28 22 4e 6f 20 64 61 74 61 20 66 6f 75 Ok("No data fou
1cb0: 6e 64 20 66 6f 75 6e 64 2e 22 2e 74 6f 5f 73 74 nd found.".to_st
1cc0: 72 69 6e 67 28 29 29 20 7d 2c 0a 09 09 09 78 20 ring()) },....x
1cd0: 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 => { Ok(format!(
1ce0: 22 7b 7d 20 70 6f 73 74 73 20 70 75 72 67 65 64 "{} posts purged
1cf0: 2e 22 2c 20 78 29 29 20 7d 2c 0a 09 09 7d 0a 09 .", x)) },...}..
1d00: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
1d10: 20 65 6e 61 62 6c 65 3c 53 3e 28 26 73 65 6c 66 enable<S>(&self
1d20: 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 , source_id: &i3
1d30: 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 2, owner: S) ->
1d40: 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 Result<&str>..wh
1d50: 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e ere S: Into<i64>
1d60: 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 {...let owner:
1d70: 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f i64 = owner.into
1d80: 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f ();...let mut co
1d90: 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 nn = self.pool.a
1da0: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 cquire().await..
1db0: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1dc0: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 45 6e 61 62 || format!("Enab
1dd0: 6c 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e le fetch conn:\n
1de0: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
1df0: 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 l))?;...match sq
1e00: 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 lx::query("updat
1e10: 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 e rsstg_source s
1e20: 65 74 20 65 6e 61 62 6c 65 64 20 3d 20 74 72 75 et enabled = tru
1e30: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
1e40: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
1e50: 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 = $2").....bind
1e60: 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e (source_id).....
1e70: 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e bind(owner).....
1e80: 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e execute(&mut con
1e90: 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 n).await.....wit
1ea0: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
1eb0: 6d 61 74 21 28 22 45 6e 61 62 6c 65 20 73 6f 75 mat!("Enable sou
1ec0: 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 rce:\n{:?}", &se
1ed0: 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 lf.pool))?.....r
1ee0: 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b ows_affected() {
1ef0: 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 ....1 => { Ok("S
1f00: 6f 75 72 63 65 20 65 6e 61 62 6c 65 64 2e 22 29 ource enabled.")
1f10: 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b },....0 => { Ok
1f20: 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 ("Source not fou
1f30: 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e nd.") },...._ =>
1f40: 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 22 { Err(anyhow!("
1f50: 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 Database error."
1f60: 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 )) },...}..}...p
1f70: 75 62 20 61 73 79 6e 63 20 66 6e 20 64 69 73 61 ub async fn disa
1f80: 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f ble<S>(&self, so
1f90: 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f urce_id: &i32, o
1fa0: 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 wner: S) -> Resu
1fb0: 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 20 lt<&str>..where
1fc0: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
1fd0: 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 .let owner: i64
1fe0: 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a = owner.into();.
1ff0: 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d ..let mut conn =
2000: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
2010: 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 re().await.....w
2020: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
2030: 6f 72 6d 61 74 21 28 22 44 69 73 61 62 6c 65 20 ormat!("Disable
2040: 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f fetch conn:\n{:?
2050: 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 }", &self.pool))
2060: 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a ?;...match sqlx:
2070: 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 :query("update r
2080: 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 sstg_source set
2090: 65 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 65 20 enabled = false
20a0: 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 where source_id
20b0: 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d = $1 and owner =
20c0: 20 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 $2").....bind(s
20d0: 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 ource_id).....bi
20e0: 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 nd(owner).....ex
20f0: 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 ecute(&mut conn)
2100: 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f .await.....with_
2110: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
2120: 74 21 28 22 44 69 73 61 62 6c 65 20 73 6f 75 72 t!("Disable sour
2130: 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c ce:\n{:?}", &sel
2140: 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f f.pool))?.....ro
2150: 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a ws_affected() {.
2160: 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f ...1 => { Ok("So
2170: 75 72 63 65 20 64 69 73 61 62 6c 65 64 2e 22 29 urce disabled.")
2180: 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b },....0 => { Ok
2190: 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 ("Source not fou
21a0: 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e nd.") },...._ =>
21b0: 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 22 { Err(anyhow!("
21c0: 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 Database error."
21d0: 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 )) },...}..}...p
21e0: 75 62 20 61 73 79 6e 63 20 66 6e 20 75 70 64 61 ub async fn upda
21f0: 74 65 3c 53 3e 28 26 73 65 6c 66 2c 20 75 70 64 te<S>(&self, upd
2200: 61 74 65 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e ate: Option<i32>
2210: 2c 20 63 68 61 6e 6e 65 6c 3a 20 26 73 74 72 2c , channel: &str,
2220: 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 channel_id: i64
2230: 2c 20 75 72 6c 3a 20 26 73 74 72 2c 20 69 76 5f , url: &str, iv_
2240: 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 hash: Option<&st
2250: 72 3e 2c 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 r>, url_re: Opti
2260: 6f 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 72 3a on<&str>, owner:
2270: 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 74 S) -> Result<St
2280: 72 69 6e 67 3e 0a 09 77 68 65 72 65 20 53 3a 20 ring>..where S:
2290: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
22a0: 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f t owner: i64 = o
22b0: 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c wner.into();...l
22c0: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
22d0: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
22e0: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
22f0: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
2300: 61 74 21 28 22 55 70 64 61 74 65 20 66 65 74 63 at!("Update fetc
2310: 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 h conn:\n{:?}",
2320: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 0a &self.pool))?;..
2330: 09 09 6d 61 74 63 68 20 6d 61 74 63 68 20 75 70 ..match match up
2340: 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 date {.....Some(
2350: 69 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 id) => {......sq
2360: 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 lx::query("updat
2370: 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 e rsstg_source s
2380: 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 20 3d 20 et channel_id =
2390: 24 32 2c 20 75 72 6c 20 3d 20 24 33 2c 20 69 76 $2, url = $3, iv
23a0: 5f 68 61 73 68 20 3d 20 24 34 2c 20 6f 77 6e 65 _hash = $4, owne
23b0: 72 20 3d 20 24 35 2c 20 63 68 61 6e 6e 65 6c 20 r = $5, channel
23c0: 3d 20 24 36 2c 20 75 72 6c 5f 72 65 20 3d 20 24 = $6, url_re = $
23d0: 37 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 7 where source_i
23e0: 64 20 3d 20 24 31 22 29 2e 62 69 6e 64 28 69 64 d = $1").bind(id
23f0: 29 0a 09 09 09 09 7d 2c 0a 09 09 09 09 4e 6f 6e ).....},.....Non
2400: 65 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 e => {......sqlx
2410: 3a 3a 71 75 65 72 79 28 22 69 6e 73 65 72 74 20 ::query("insert
2420: 69 6e 74 6f 20 72 73 73 74 67 5f 73 6f 75 72 63 into rsstg_sourc
2430: 65 20 28 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 e (channel_id, u
2440: 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e rl, iv_hash, own
2450: 65 72 2c 20 63 68 61 6e 6e 65 6c 2c 20 75 72 6c er, channel, url
2460: 5f 72 65 29 20 76 61 6c 75 65 73 20 28 24 31 2c _re) values ($1,
2470: 20 24 32 2c 20 24 33 2c 20 24 34 2c 20 24 35 2c $2, $3, $4, $5,
2480: 20 24 36 29 22 29 0a 09 09 09 09 7d 2c 0a 09 09 $6)").....},...
2490: 09 7d 0a 09 09 09 2e 62 69 6e 64 28 63 68 61 6e .}.....bind(chan
24a0: 6e 65 6c 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 nel_id).....bind
24b0: 28 75 72 6c 29 0a 09 09 09 2e 62 69 6e 64 28 69 (url).....bind(i
24c0: 76 5f 68 61 73 68 29 0a 09 09 09 2e 62 69 6e 64 v_hash).....bind
24d0: 28 6f 77 6e 65 72 29 0a 09 09 09 2e 62 69 6e 64 (owner).....bind
24e0: 28 63 68 61 6e 6e 65 6c 29 0a 09 09 09 2e 62 69 (channel).....bi
24f0: 6e 64 28 75 72 6c 5f 72 65 29 0a 09 09 09 2e 65 nd(url_re).....e
2500: 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e xecute(&mut conn
2510: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 4f 6b 28 ).await {....Ok(
2520: 5f 29 20 3d 3e 20 72 65 74 75 72 6e 20 4f 6b 28 _) => return Ok(
2530: 53 74 72 69 6e 67 3a 3a 66 72 6f 6d 28 6d 61 74 String::from(mat
2540: 63 68 20 75 70 64 61 74 65 20 7b 0a 09 09 09 09 ch update {.....
2550: 53 6f 6d 65 28 5f 29 20 3d 3e 20 22 43 68 61 6e Some(_) => "Chan
2560: 6e 65 6c 20 75 70 64 61 74 65 64 2e 22 2c 0a 09 nel updated.",..
2570: 09 09 09 4e 6f 6e 65 20 3d 3e 20 22 43 68 61 6e ...None => "Chan
2580: 6e 65 6c 20 61 64 64 65 64 2e 22 2c 0a 09 09 09 nel added.",....
2590: 7d 29 29 2c 0a 09 09 09 45 72 72 28 73 71 6c 78 })),....Err(sqlx
25a0: 3a 3a 45 72 72 6f 72 3a 3a 44 61 74 61 62 61 73 ::Error::Databas
25b0: 65 28 65 72 72 29 29 20 3d 3e 20 7b 0a 09 09 09 e(err)) => {....
25c0: 09 6d 61 74 63 68 20 65 72 72 2e 64 6f 77 6e 63 .match err.downc
25d0: 61 73 74 3a 3a 3c 73 71 6c 78 3a 3a 70 6f 73 74 ast::<sqlx::post
25e0: 67 72 65 73 3a 3a 50 67 44 61 74 61 62 61 73 65 gres::PgDatabase
25f0: 45 72 72 6f 72 3e 28 29 2e 72 6f 75 74 69 6e 65 Error>().routine
2600: 28 29 20 7b 0a 09 09 09 09 09 53 6f 6d 65 28 22 () {......Some("
2610: 5f 62 74 5f 63 68 65 63 6b 5f 75 6e 69 71 75 65 _bt_check_unique
2620: 22 2c 20 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 ", ) => {.......
2630: 72 65 74 75 72 6e 20 4f 6b 28 22 44 75 70 6c 69 return Ok("Dupli
2640: 63 61 74 65 20 6b 65 79 2e 22 2e 74 6f 5f 73 74 cate key.".to_st
2650: 72 69 6e 67 28 29 29 0a 09 09 09 09 09 7d 2c 0a ring())......},.
2660: 09 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 .....Some(_) =>
2670: 7b 0a 09 09 09 09 09 09 72 65 74 75 72 6e 20 4f {.......return O
2680: 6b 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f k("Database erro
2690: 72 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 r.".to_string())
26a0: 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 4e 6f ......},......No
26b0: 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 09 72 65 ne => {.......re
26c0: 74 75 72 6e 20 4f 6b 28 22 4e 6f 20 64 61 74 61 turn Ok("No data
26d0: 62 61 73 65 20 65 72 72 6f 72 20 65 78 74 72 61 base error extra
26e0: 63 74 65 64 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 cted.".to_string
26f0: 28 29 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 ())......},.....
2700: 7d 3b 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 };....},....Err(
2710: 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 62 61 err) => {.....ba
2720: 69 6c 21 28 22 53 6f 72 72 79 2c 20 75 6e 6b 6e il!("Sorry, unkn
2730: 6f 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 3f own error:\n{:#?
2740: 7d 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 09 09 7d }\n", err);....}
2750: 2c 0a 09 09 7d 3b 0a 09 7d 0a 0a 09 61 73 79 6e ,...};..}...asyn
2760: 63 20 66 6e 20 61 75 74 6f 66 65 74 63 68 28 26 c fn autofetch(&
2770: 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c self) -> Result<
2780: 28 29 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 ()> {...let mut
2790: 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a delay = chrono::
27a0: 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 Duration::minute
27b0: 73 28 31 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 s(1);...let mut
27c0: 6e 6f 77 3b 0a 09 09 6c 6f 6f 70 20 7b 0a 09 09 now;...loop {...
27d0: 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 .let mut conn =
27e0: 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 self.pool.acquir
27f0: 65 28 29 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 e().await......w
2800: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
2810: 6f 72 6d 61 74 21 28 22 41 75 74 6f 66 65 74 63 ormat!("Autofetc
2820: 68 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b h fetch conn:\n{
2830: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
2840: 29 29 3f 3b 0a 09 09 09 6e 6f 77 20 3d 20 63 68 ))?;....now = ch
2850: 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f 77 rono::Local::now
2860: 28 29 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 71 ();....let mut q
2870: 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a 71 75 65 ueue = sqlx::que
2880: 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 ry("select sourc
2890: 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 63 68 e_id, next_fetch
28a0: 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 73 , owner from rss
28b0: 74 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 6c tg_order natural
28c0: 20 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 67 left join rsstg
28d0: 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 6e 65 _source where ne
28e0: 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 29 xt_fetch < now()
28f0: 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 20 6d + interval '1 m
2900: 69 6e 75 74 65 27 3b 22 29 0a 09 09 09 09 2e 66 inute';")......f
2910: 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 6f etch_all(&mut co
2920: 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 66 nn).await?;....f
2930: 6f 72 20 72 6f 77 20 69 6e 20 71 75 65 75 65 2e or row in queue.
2940: 69 74 65 72 28 29 20 7b 0a 09 09 09 09 6c 65 74 iter() {.....let
2950: 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 20 source_id: i32
2960: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 = row.try_get("s
2970: 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 ource_id")?;....
2980: 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 .let owner: i64
2990: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 6f = row.try_get("o
29a0: 77 6e 65 72 22 29 3f 3b 0a 09 09 09 09 6c 65 74 wner")?;.....let
29b0: 20 6e 65 78 74 5f 66 65 74 63 68 3a 20 44 61 74 next_fetch: Dat
29c0: 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 4c 6f eTime<chrono::Lo
29d0: 63 61 6c 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 cal> = row.try_g
29e0: 65 74 28 22 6e 65 78 74 5f 66 65 74 63 68 22 29 et("next_fetch")
29f0: 3f 3b 0a 09 09 09 09 69 66 20 6e 65 78 74 5f 66 ?;.....if next_f
2a00: 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a 09 09 09 etch < now {....
2a10: 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 6f ..let clone = Co
2a20: 72 65 20 7b 0a 09 09 09 09 09 09 6f 77 6e 65 72 re {.......owner
2a30: 5f 63 68 61 74 3a 20 74 65 6c 65 67 72 61 6d 5f _chat: telegram_
2a40: 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 bot::UserId::new
2a50: 28 6f 77 6e 65 72 29 2c 0a 09 09 09 09 09 09 2e (owner),........
2a60: 2e 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 0a 09 09 .self.clone()...
2a70: 09 09 09 7d 3b 0a 09 09 09 09 09 74 6f 6b 69 6f ...};......tokio
2a80: 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 6d 6f ::spawn(async mo
2a90: 76 65 20 7b 0a 09 09 09 09 09 09 69 66 20 6c 65 ve {.......if le
2aa0: 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c 6f t Err(err) = clo
2ab0: 6e 65 2e 63 68 65 63 6b 28 26 73 6f 75 72 63 65 ne.check(&source
2ac0: 5f 69 64 2c 20 6f 77 6e 65 72 2c 20 74 72 75 65 _id, owner, true
2ad0: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 ).await {.......
2ae0: 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 .if let Err(err)
2af0: 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 26 66 = clone.send(&f
2b00: 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a 3f ormat!("š {:?
2b10: 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 2c 20 }", err), None,
2b20: 4e 6f 6e 65 29 20 7b 0a 09 09 09 09 09 09 09 09 None) {.........
2b30: 65 70 72 69 6e 74 6c 6e 21 28 22 43 68 65 63 6b eprintln!("Check
2b40: 20 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 error: {}", err
2b50: 29 3b 0a 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 );........};....
2b60: 09 09 09 7d 3b 0a 09 09 09 09 09 7d 29 3b 0a 09 ...};......});..
2b70: 09 09 09 7d 20 65 6c 73 65 20 7b 0a 09 09 09 09 ...} else {.....
2b80: 09 69 66 20 6e 65 78 74 5f 66 65 74 63 68 20 2d .if next_fetch -
2b90: 20 6e 6f 77 20 3c 20 64 65 6c 61 79 20 7b 0a 09 now < delay {..
2ba0: 09 09 09 09 09 64 65 6c 61 79 20 3d 20 6e 65 78 .....delay = nex
2bb0: 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 3b 0a 09 t_fetch - now;..
2bc0: 09 09 09 09 7d 0a 09 09 09 09 7d 0a 09 09 09 7d ....}.....}....}
2bd0: 3b 0a 09 09 09 71 75 65 75 65 2e 63 6c 65 61 72 ;....queue.clear
2be0: 28 29 3b 0a 09 09 09 74 6f 6b 69 6f 3a 3a 74 69 ();....tokio::ti
2bf0: 6d 65 3a 3a 73 6c 65 65 70 28 64 65 6c 61 79 2e me::sleep(delay.
2c00: 74 6f 5f 73 74 64 28 29 3f 29 2e 61 77 61 69 74 to_std()?).await
2c10: 3b 0a 09 09 09 64 65 6c 61 79 20 3d 20 63 68 72 ;....delay = chr
2c20: 6f 6e 6f 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d ono::Duration::m
2c30: 69 6e 75 74 65 73 28 31 29 3b 0a 09 09 7d 0a 09 inutes(1);...}..
2c40: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
2c50: 20 6c 69 73 74 3c 53 3e 28 26 73 65 6c 66 2c 20 list<S>(&self,
2c60: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
2c70: 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 ult<String>..whe
2c80: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
2c90: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
2ca0: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 owner.into();...
2cb0: 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 20 3d 20 let mut reply =
2cc0: 76 65 63 21 5b 5d 3b 0a 09 09 6c 65 74 20 6d 75 vec![];...let mu
2cd0: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
2ce0: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
2cf0: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
2d00: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
2d10: 4c 69 73 74 20 66 65 74 63 68 20 63 6f 6e 6e 3a List fetch conn:
2d20: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
2d30: 6f 6f 6c 29 29 3f 3b 0a 09 09 72 65 70 6c 79 2e ool))?;...reply.
2d40: 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 push("Channels:"
2d50: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 .to_string());..
2d60: 09 6c 65 74 20 72 6f 77 73 20 3d 20 73 71 6c 78 .let rows = sqlx
2d70: 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 ::query("select
2d80: 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e source_id, chann
2d90: 65 6c 2c 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c el, enabled, url
2da0: 2c 20 69 76 5f 68 61 73 68 20 66 72 6f 6d 20 72 , iv_hash from r
2db0: 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 sstg_source wher
2dc0: 65 20 6f 77 6e 65 72 20 3d 20 24 31 20 6f 72 64 e owner = $1 ord
2dd0: 65 72 20 62 79 20 73 6f 75 72 63 65 5f 69 64 22 er by source_id"
2de0: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
2df0: 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 ).....fetch_all(
2e00: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
2e10: 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 ?;...for row in
2e20: 72 6f 77 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 rows.iter() {...
2e30: 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 .let source_id:
2e40: 69 33 32 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 i32 = row.try_ge
2e50: 74 28 22 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b t("source_id")?;
2e60: 0a 09 09 09 6c 65 74 20 75 73 65 72 6e 61 6d 65 ....let username
2e70: 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 : &str = row.try
2e80: 5f 67 65 74 28 22 63 68 61 6e 6e 65 6c 22 29 3f _get("channel")?
2e90: 3b 0a 09 09 09 6c 65 74 20 65 6e 61 62 6c 65 64 ;....let enabled
2ea0: 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 : bool = row.try
2eb0: 5f 67 65 74 28 22 65 6e 61 62 6c 65 64 22 29 3f _get("enabled")?
2ec0: 3b 0a 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 73 ;....let url: &s
2ed0: 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 tr = row.try_get
2ee0: 28 22 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 ("url")?;....let
2ef0: 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e iv_hash: Option
2f00: 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 <&str> = row.try
2f10: 5f 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f _get("iv_hash")?
2f20: 3b 0a 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 ;....reply.push(
2f30: 66 6f 72 6d 61 74 21 28 22 5c 6e 5c 5c 23 ef b8 format!("\n\\#ļø
2f40: 8f e2 83 a3 20 7b 7d 20 5c 5c 2a ef b8 8f e2 83 ā£ {} \\*ļøā
2f50: a3 20 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 97 20 £ `{}` {}\nš
2f60: 60 7b 7d 60 22 2c 20 73 6f 75 72 63 65 5f 69 64 `{}`", source_id
2f70: 2c 20 75 73 65 72 6e 61 6d 65 2c 20 20 0a 09 09 , username, ...
2f80: 09 09 6d 61 74 63 68 20 65 6e 61 62 6c 65 64 20 ..match enabled
2f90: 7b 0a 09 09 09 09 09 74 72 75 65 20 20 3d 3e 20 {......true =>
2fa0: 22 f0 9f 94 84 20 65 6e 61 62 6c 65 64 22 2c 0a "š enabled",.
2fb0: 09 09 09 09 09 66 61 6c 73 65 20 3d 3e 20 22 e2 .....false => "ā
2fc0: 9b 94 20 64 69 73 61 62 6c 65 64 22 2c 0a 09 09 disabled",...
2fd0: 09 09 7d 2c 20 75 72 6c 29 29 3b 0a 09 09 09 69 ..}, url));....i
2fe0: 66 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 68 29 f let Some(hash)
2ff0: 20 3d 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 = iv_hash {....
3000: 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d .reply.push(form
3010: 61 74 21 28 22 49 56 20 60 7b 7d 60 22 2c 20 68 at!("IV `{}`", h
3020: 61 73 68 29 29 3b 0a 09 09 09 7d 0a 09 09 7d 3b ash));....}...};
3030: 0a 09 09 4f 6b 28 72 65 70 6c 79 2e 6a 6f 69 6e ...Ok(reply.join
3040: 28 22 5c 6e 22 29 29 0a 09 7d 0a 7d 0a ("\n"))..}.}.