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 2f 2a 6c 65 "url")?;..../*le
0b50: 74 20 75 72 6c 3a 20 43 6f 77 3c 27 61 2c 20 73 t url: Cow<'a, s
0b60: 74 72 3e 20 3d 20 6d 61 74 63 68 20 72 6f 77 2e tr> = match row.
0b70: 74 72 79 5f 67 65 74 28 22 75 72 6c 22 29 20 7b try_get("url") {
0b80: 0a 09 09 09 09 4f 6b 28 78 29 20 3d 3e 20 53 74 .....Ok(x) => St
0b90: 72 69 6e 67 3a 3a 66 72 6f 6d 28 78 29 2e 74 6f ring::from(x).to
0ba0: 5f 6f 77 6e 65 64 28 29 2e 69 6e 74 6f 28 29 2c _owned().into(),
0bb0: 0a 09 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e .....Err(err) =>
0bc0: 20 62 61 69 6c 21 28 22 54 65 73 74 22 29 2c 0a bail!("Test"),.
0bd0: 09 09 09 7d 3b 2a 2f 0a 09 09 09 6c 65 74 20 69 ...};*/....let i
0be0: 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 v_hash: Option<&
0bf0: 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 str> = row.try_g
0c00: 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f 3b 0a et("iv_hash")?;.
0c10: 09 09 09 2f 2f 6c 65 74 20 75 72 6c 5f 72 65 3a ...//let url_re:
0c20: 20 4f 70 74 69 6f 6e 3c 73 65 64 72 65 67 65 78 Option<sedregex
0c30: 3a 3a 52 65 70 6c 61 63 65 43 6f 6d 6d 61 6e 64 ::ReplaceCommand
0c40: 3e 20 3d 20 6d 61 74 63 68 20 72 6f 77 2e 74 72 > = match row.tr
0c50: 79 5f 67 65 74 28 22 75 72 6c 5f 72 65 22 29 3f y_get("url_re")?
0c60: 20 7b 0a 09 09 09 6c 65 74 20 75 72 6c 5f 72 65 {....let url_re
0c70: 20 3d 20 6d 61 74 63 68 20 72 6f 77 2e 74 72 79 = match row.try
0c80: 5f 67 65 74 28 22 75 72 6c 5f 72 65 22 29 3f 20 _get("url_re")?
0c90: 7b 0a 09 09 09 09 53 6f 6d 65 28 78 29 20 3d 3e {.....Some(x) =>
0ca0: 20 53 6f 6d 65 28 73 65 64 72 65 67 65 78 3a 3a Some(sedregex::
0cb0: 52 65 70 6c 61 63 65 43 6f 6d 6d 61 6e 64 3a 3a ReplaceCommand::
0cc0: 6e 65 77 28 78 29 3f 29 2c 0a 09 09 09 09 4e 6f new(x)?),.....No
0cd0: 6e 65 20 3d 3e 20 4e 6f 6e 65 2c 0a 09 09 09 7d ne => None,....}
0ce0: 3b 0a 09 09 09 6c 65 74 20 64 65 73 74 69 6e 61 ;....let destina
0cf0: 74 69 6f 6e 20 3d 20 6d 61 74 63 68 20 72 65 61 tion = match rea
0d00: 6c 20 7b 0a 09 09 09 09 74 72 75 65 20 3d 3e 20 l {.....true =>
0d10: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 telegram_bot::Us
0d20: 65 72 49 64 3a 3a 6e 65 77 28 63 68 61 6e 6e 65 erId::new(channe
0d30: 6c 5f 69 64 29 2c 0a 09 09 09 09 66 61 6c 73 65 l_id),.....false
0d40: 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 => telegram_bot
0d50: 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 72 6f ::UserId::new(ro
0d60: 77 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e 65 72 w.try_get("owner
0d70: 22 29 3f 29 2c 0a 09 09 09 7d 3b 0a 09 09 09 6c ")?),....};....l
0d80: 65 74 20 6d 75 74 20 74 68 69 73 5f 66 65 74 63 et mut this_fetc
0d90: 68 3a 20 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 h: Option<DateTi
0da0: 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 me<chrono::Fixed
0db0: 4f 66 66 73 65 74 3e 3e 20 3d 20 4e 6f 6e 65 3b Offset>> = None;
0dc0: 0a 09 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 ....let mut post
0dd0: 73 3a 20 42 54 72 65 65 4d 61 70 3c 44 61 74 65 s: BTreeMap<Date
0de0: 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 Time<chrono::Fix
0df0: 65 64 4f 66 66 73 65 74 3e 2c 20 53 74 72 69 6e edOffset>, Strin
0e00: 67 3e 20 3d 20 42 54 72 65 65 4d 61 70 3a 3a 6e g> = BTreeMap::n
0e10: 65 77 28 29 3b 0a 09 09 09 6c 65 74 20 63 6f 6e ew();....let con
0e20: 74 65 6e 74 20 3d 20 72 65 71 77 65 73 74 3a 3a tent = reqwest::
0e30: 67 65 74 28 75 72 6c 29 2e 61 77 61 69 74 3f 2e get(url).await?.
0e40: 62 79 74 65 73 28 29 2e 61 77 61 69 74 3f 3b 0a bytes().await?;.
0e50: 09 09 09 6d 61 74 63 68 20 72 73 73 3a 3a 43 68 ...match rss::Ch
0e60: 61 6e 6e 65 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d annel::read_from
0e70: 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b (&content[..]) {
0e80: 0a 09 09 09 09 4f 6b 28 66 65 65 64 29 20 3d 3e .....Ok(feed) =>
0e90: 20 7b 0a 09 09 09 09 09 66 6f 72 20 69 74 65 6d {......for item
0ea0: 20 69 6e 20 66 65 65 64 2e 69 74 65 6d 73 28 29 in feed.items()
0eb0: 20 7b 0a 09 09 09 09 09 09 6d 61 74 63 68 20 69 {.......match i
0ec0: 74 65 6d 2e 6c 69 6e 6b 28 29 20 7b 0a 09 09 09 tem.link() {....
0ed0: 09 09 09 09 53 6f 6d 65 28 6c 69 6e 6b 29 20 3d ....Some(link) =
0ee0: 3e 20 7b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 > {.........let
0ef0: 64 61 74 65 20 3d 20 6d 61 74 63 68 20 69 74 65 date = match ite
0f00: 6d 2e 70 75 62 5f 64 61 74 65 28 29 20 7b 0a 09 m.pub_date() {..
0f10: 09 09 09 09 09 09 09 09 53 6f 6d 65 28 66 65 65 ........Some(fee
0f20: 64 5f 64 61 74 65 29 20 3d 3e 20 44 61 74 65 54 d_date) => DateT
0f30: 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f ime::parse_from_
0f40: 72 66 63 32 38 32 32 28 66 65 65 64 5f 64 61 74 rfc2822(feed_dat
0f50: 65 29 2c 0a 09 09 09 09 09 09 09 09 09 4e 6f 6e e),..........Non
0f60: 65 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 e => DateTime::p
0f70: 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 33 33 33 arse_from_rfc333
0f80: 39 28 26 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 63 9(&item.dublin_c
0f90: 6f 72 65 5f 65 78 74 28 29 2e 75 6e 77 72 61 70 ore_ext().unwrap
0fa0: 28 29 2e 64 61 74 65 73 28 29 5b 30 5d 29 2c 0a ().dates()[0]),.
0fb0: 09 09 09 09 09 09 09 09 7d 3f 3b 0a 09 09 09 09 ........}?;.....
0fc0: 09 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 6c 69 ....let url = li
0fd0: 6e 6b 3b 0a 09 09 09 09 09 09 09 09 70 6f 73 74 nk;.........post
0fe0: 73 2e 69 6e 73 65 72 74 28 64 61 74 65 2e 63 6c s.insert(date.cl
0ff0: 6f 6e 65 28 29 2c 20 75 72 6c 2e 69 6e 74 6f 28 one(), url.into(
1000: 29 29 3b 0a 09 09 09 09 09 09 09 7d 2c 0a 09 09 ));........},...
1010: 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 7d 0a .....None => {}.
1020: 09 09 09 09 09 09 7d 0a 09 09 09 09 09 7d 3b 0a ......}......};.
1030: 09 09 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 65 ....},.....Err(e
1040: 72 72 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 72 rr) => match err
1050: 20 7b 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 {......rss::Err
1060: 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 74 or::InvalidStart
1070: 54 61 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 6c Tag => {.......l
1080: 65 74 20 66 65 65 64 20 3d 20 61 74 6f 6d 5f 73 et feed = atom_s
1090: 79 6e 64 69 63 61 74 69 6f 6e 3a 3a 46 65 65 64 yndication::Feed
10a0: 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f 6e ::read_from(&con
10b0: 74 65 6e 74 5b 2e 2e 5d 29 0a 09 09 09 09 09 09 tent[..]).......
10c0: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
10d0: 7c 20 66 6f 72 6d 61 74 21 28 22 50 72 6f 62 6c | format!("Probl
10e0: 65 6d 20 6f 70 65 6e 69 6e 67 20 66 65 65 64 20 em opening feed
10f0: 75 72 6c 3a 5c 6e 7b 7d 22 2c 20 26 75 72 6c 29 url:\n{}", &url)
1100: 29 3f 3b 0a 09 09 09 09 09 09 66 6f 72 20 69 74 )?;.......for it
1110: 65 6d 20 69 6e 20 66 65 65 64 2e 65 6e 74 72 69 em in feed.entri
1120: 65 73 28 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 es() {........le
1130: 74 20 64 61 74 65 20 3d 20 69 74 65 6d 2e 70 75 t date = item.pu
1140: 62 6c 69 73 68 65 64 28 29 2e 75 6e 77 72 61 70 blished().unwrap
1150: 28 29 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 75 ();........let u
1160: 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 rl = item.links(
1170: 29 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 09 )[0].href();....
1180: 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 ....posts.insert
1190: 28 64 61 74 65 2e 63 6c 6f 6e 65 28 29 2c 20 75 (date.clone(), u
11a0: 72 6c 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 09 rl.into());.....
11b0: 09 09 7d 3b 0a 09 09 09 09 09 7d 2c 0a 09 09 09 ..};......},....
11c0: 09 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 6f ..rss::Error::Eo
11d0: 66 20 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f 20 f => (),......_
11e0: 3d 3e 20 62 61 69 6c 21 28 22 55 6e 73 75 70 70 => bail!("Unsupp
11f0: 6f 72 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 64 orted or mangled
1200: 20 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c content:\n{:?}\
1210: 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 26 75 72 6c 2c n{:#?}\n", &url,
1220: 20 65 72 72 29 0a 09 09 09 09 7d 0a 09 09 09 7d err).....}....}
1230: 3b 0a 09 09 09 66 6f 72 20 28 64 61 74 65 2c 20 ;....for (date,
1240: 75 72 6c 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 url) in posts.it
1250: 65 72 28 29 20 7b 0a 09 09 09 09 6c 65 74 20 6d er() {.....let m
1260: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 ut conn = self.p
1270: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
1280: 61 69 74 0a 09 09 09 09 09 2e 77 69 74 68 5f 63 ait.......with_c
1290: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
12a0: 21 28 22 43 68 65 63 6b 20 70 6f 73 74 20 66 65 !("Check post fe
12b0: 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 tch conn:\n{:?}"
12c0: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b , &self.pool))?;
12d0: 0a 09 09 09 09 6c 65 74 20 72 6f 77 20 3d 20 73 .....let row = s
12e0: 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 qlx::query("sele
12f0: 63 74 20 65 78 69 73 74 73 28 73 65 6c 65 63 74 ct exists(select
1300: 20 74 72 75 65 20 66 72 6f 6d 20 72 73 73 74 67 true from rsstg
1310: 5f 70 6f 73 74 20 77 68 65 72 65 20 75 72 6c 20 _post where url
1320: 3d 20 24 31 20 61 6e 64 20 73 6f 75 72 63 65 5f = $1 and source_
1330: 69 64 20 3d 20 24 32 29 20 61 73 20 65 78 69 73 id = $2) as exis
1340: 74 73 3b 22 29 0a 09 09 09 09 09 2e 62 69 6e 64 ts;").......bind
1350: 28 75 72 6c 29 0a 09 09 09 09 09 2e 62 69 6e 64 (url).......bind
1360: 28 2a 69 64 29 0a 09 09 09 09 09 2e 66 65 74 63 (*id).......fetc
1370: 68 5f 6f 6e 65 28 26 6d 75 74 20 63 6f 6e 6e 29 h_one(&mut conn)
1380: 2e 61 77 61 69 74 0a 09 09 09 09 09 2e 77 69 74 .await.......wit
1390: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
13a0: 6d 61 74 21 28 22 43 68 65 63 6b 20 70 6f 73 74 mat!("Check post
13b0: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 :\n{:?}", &conn)
13c0: 29 3f 3b 0a 09 09 09 09 6c 65 74 20 65 78 69 73 )?;.....let exis
13d0: 74 73 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 ts: bool = row.t
13e0: 72 79 5f 67 65 74 28 22 65 78 69 73 74 73 22 29 ry_get("exists")
13f0: 3f 3b 0a 09 09 09 09 69 66 20 21 20 65 78 69 73 ?;.....if ! exis
1400: 74 73 20 7b 0a 09 09 09 09 09 69 66 20 74 68 69 ts {......if thi
1410: 73 5f 66 65 74 63 68 20 3d 3d 20 4e 6f 6e 65 20 s_fetch == None
1420: 7c 7c 20 2a 64 61 74 65 20 3e 20 74 68 69 73 5f || *date > this_
1430: 66 65 74 63 68 2e 75 6e 77 72 61 70 28 29 20 7b fetch.unwrap() {
1440: 0a 09 09 09 09 09 09 74 68 69 73 5f 66 65 74 63 .......this_fetc
1450: 68 20 3d 20 53 6f 6d 65 28 2a 64 61 74 65 29 3b h = Some(*date);
1460: 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 73 65 ......};......se
1470: 6c 66 2e 74 67 2e 73 65 6e 64 28 20 6d 61 74 63 lf.tg.send( matc
1480: 68 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 h iv_hash {.....
1490: 09 09 09 53 6f 6d 65 28 68 61 73 68 29 20 3d 3e ...Some(hash) =>
14a0: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 telegram_bot::S
14b0: 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 endMessage::new(
14c0: 64 65 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 destination, for
14d0: 6d 61 74 21 28 22 3c 61 20 68 72 65 66 3d 5c 22 mat!("<a href=\"
14e0: 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f https://t.me/iv?
14f0: 75 72 6c 3d 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c url={}&rhash={}\
1500: 22 3e 20 3c 2f 61 3e 7b 30 7d 22 2c 20 6d 61 74 "> </a>{0}", mat
1510: 63 68 20 75 72 6c 5f 72 65 20 7b 0a 09 09 09 09 ch url_re {.....
1520: 09 09 09 09 53 6f 6d 65 28 78 29 20 3d 3e 20 7b ....Some(x) => {
1530: 0a 09 09 09 09 09 09 09 09 09 62 61 69 6c 21 28 ..........bail!(
1540: 22 52 65 67 65 78 20 68 69 74 2c 20 72 65 73 75 "Regex hit, resu
1550: 6c 74 3a 5c 6e 7b 3a 23 3f 7d 22 2c 20 78 2e 65 lt:\n{:#?}", x.e
1560: 78 65 63 75 74 65 28 75 72 6c 29 29 3b 0a 09 09 xecute(url));...
1570: 09 09 09 09 09 09 09 75 72 6c 0a 09 09 09 09 09 .......url......
1580: 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 4e 6f ...},.........No
1590: 6e 65 20 3d 3e 20 75 72 6c 2c 0a 09 09 09 09 09 ne => url,......
15a0: 09 09 7d 2c 20 68 61 73 68 29 29 2c 0a 09 09 09 ..}, hash)),....
15b0: 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 65 6c 65 ....None => tele
15c0: 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 gram_bot::SendMe
15d0: 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 73 74 69 ssage::new(desti
15e0: 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 21 28 nation, format!(
15f0: 22 7b 7d 22 2c 20 75 72 6c 29 29 2c 0a 09 09 09 "{}", url)),....
1600: 09 09 09 7d 2e 70 61 72 73 65 5f 6d 6f 64 65 28 ...}.parse_mode(
1610: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 telegram_bot::ty
1620: 70 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a pes::ParseMode::
1630: 48 74 6d 6c 29 29 2e 61 77 61 69 74 0a 09 09 09 Html)).await....
1640: 09 09 09 2e 63 6f 6e 74 65 78 74 28 22 43 61 6e ....context("Can
1650: 27 74 20 70 6f 73 74 20 6d 65 73 73 61 67 65 3a 't post message:
1660: 22 29 3f 3b 0a 09 09 09 09 09 73 71 6c 78 3a 3a ")?;......sqlx::
1670: 71 75 65 72 79 28 22 69 6e 73 65 72 74 20 69 6e query("insert in
1680: 74 6f 20 72 73 73 74 67 5f 70 6f 73 74 20 28 73 to rsstg_post (s
1690: 6f 75 72 63 65 5f 69 64 2c 20 70 6f 73 74 65 64 ource_id, posted
16a0: 2c 20 75 72 6c 29 20 76 61 6c 75 65 73 20 28 24 , url) values ($
16b0: 31 2c 20 24 32 2c 20 24 33 29 3b 22 29 0a 09 09 1, $2, $3);")...
16c0: 09 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 .....bind(*id)..
16d0: 09 09 09 09 09 2e 62 69 6e 64 28 64 61 74 65 29 ......bind(date)
16e0: 0a 09 09 09 09 09 09 2e 62 69 6e 64 28 75 72 6c ........bind(url
16f0: 29 0a 09 09 09 09 09 09 2e 65 78 65 63 75 74 65 )........execute
1700: 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 (&mut conn).awai
1710: 74 0a 09 09 09 09 09 09 2e 77 69 74 68 5f 63 6f t........with_co
1720: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
1730: 28 22 52 65 63 6f 72 64 20 70 6f 73 74 3a 5c 6e ("Record post:\n
1740: 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f 3b {:?}", &conn))?;
1750: 0a 09 09 09 09 09 64 72 6f 70 28 63 6f 6e 6e 29 ......drop(conn)
1760: 3b 0a 09 09 09 09 09 74 6f 6b 69 6f 3a 3a 74 69 ;......tokio::ti
1770: 6d 65 3a 3a 73 6c 65 65 70 28 73 74 64 3a 3a 74 me::sleep(std::t
1780: 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e ime::Duration::n
1790: 65 77 28 34 2c 20 30 29 29 2e 61 77 61 69 74 3b ew(4, 0)).await;
17a0: 0a 09 09 09 09 7d 3b 0a 09 09 09 09 70 6f 73 74 .....};.....post
17b0: 65 64 20 2b 3d 20 31 3b 0a 09 09 09 7d 3b 0a 09 ed += 1;....};..
17c0: 09 09 70 6f 73 74 73 2e 63 6c 65 61 72 28 29 3b ..posts.clear();
17d0: 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 ...};...let mut
17e0: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
17f0: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
1800: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
1810: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 t(|| format!("Up
1820: 64 61 74 65 20 73 63 72 61 70 65 20 66 65 74 63 date scrape fetc
1830: 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 h conn:\n{:?}",
1840: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 &self.pool))?;..
1850: 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 .sqlx::query("up
1860: 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 date rsstg_sourc
1870: 65 20 73 65 74 20 6c 61 73 74 5f 73 63 72 61 70 e set last_scrap
1880: 65 20 3d 20 6e 6f 77 28 29 20 77 68 65 72 65 20 e = now() where
1890: 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 source_id = $1;"
18a0: 29 0a 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a ).....bind(*id).
18b0: 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 ....execute(&mut
18c0: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 conn).await....
18d0: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
18e0: 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 format!("Update
18f0: 20 73 63 72 61 70 65 3a 5c 6e 7b 3a 3f 7d 22 2c scrape:\n{:?}",
1900: 20 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 4f 6b 28 &conn))?;...Ok(
1910: 66 6f 72 6d 61 74 21 28 22 50 6f 73 74 65 64 3a format!("Posted:
1920: 20 7b 7d 22 2c 20 26 70 6f 73 74 65 64 29 29 0a {}", &posted)).
1930: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
1940: 6e 20 64 65 6c 65 74 65 3c 53 3e 28 26 73 65 6c n delete<S>(&sel
1950: 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 f, source_id: &i
1960: 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 32, owner: S) ->
1970: 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e 0a Result<String>.
1980: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
1990: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
19a0: 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 r: i64 = owner.i
19b0: 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 nto();...let mut
19c0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
19d0: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
19e0: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
19f0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 xt(|| format!("D
1a00: 65 6c 65 74 65 20 66 65 74 63 68 20 63 6f 6e 6e elete fetch conn
1a10: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1a20: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 pool))?;...match
1a30: 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 sqlx::query("de
1a40: 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 74 67 5f lete from rsstg_
1a50: 73 6f 75 72 63 65 20 77 68 65 72 65 20 73 6f 75 source where sou
1a60: 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 rce_id = $1 and
1a70: 6f 77 6e 65 72 20 3d 20 24 32 3b 22 29 0a 09 09 owner = $2;")...
1a80: 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 ..bind(source_id
1a90: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
1aa0: 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d ).....execute(&m
1ab0: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 ut conn).await..
1ac0: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1ad0: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 65 6c 65 || format!("Dele
1ae0: 74 65 20 73 6f 75 72 63 65 20 72 75 6c 65 3a 5c te source rule:\
1af0: 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f n{:?}", &self.po
1b00: 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 ol))?.....rows_a
1b10: 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 30 ffected() {....0
1b20: 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 61 74 => { Ok("No dat
1b30: 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 2e 22 2e a found found.".
1b40: 74 6f 5f 73 74 72 69 6e 67 28 29 29 20 7d 2c 0a to_string()) },.
1b50: 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 ...x => { Ok(for
1b60: 6d 61 74 21 28 22 7b 7d 20 73 6f 75 72 63 65 73 mat!("{} sources
1b70: 20 72 65 6d 6f 76 65 64 2e 22 2c 20 78 29 29 20 removed.", x))
1b80: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 },...}..}...pub
1b90: 61 73 79 6e 63 20 66 6e 20 63 6c 65 61 6e 3c 53 async fn clean<S
1ba0: 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f >(&self, source_
1bb0: 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a id: &i32, owner:
1bc0: 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 74 S) -> Result<St
1bd0: 72 69 6e 67 3e 0a 09 77 68 65 72 65 20 53 3a 20 ring>..where S:
1be0: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
1bf0: 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f t owner: i64 = o
1c00: 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c wner.into();...l
1c10: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
1c20: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
1c30: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
1c40: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
1c50: 61 74 21 28 22 43 6c 65 61 6e 20 66 65 74 63 68 at!("Clean fetch
1c60: 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 conn:\n{:?}", &
1c70: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 self.pool))?;...
1c80: 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 match sqlx::quer
1c90: 79 28 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 y("delete from r
1ca0: 73 73 74 67 5f 70 6f 73 74 20 70 20 75 73 69 6e sstg_post p usin
1cb0: 67 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 g rsstg_source s
1cc0: 20 77 68 65 72 65 20 70 2e 73 6f 75 72 63 65 5f where p.source_
1cd0: 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 id = $1 and owne
1ce0: 72 20 3d 20 24 32 20 61 6e 64 20 70 2e 73 6f 75 r = $2 and p.sou
1cf0: 72 63 65 5f 69 64 20 3d 20 73 2e 73 6f 75 72 63 rce_id = s.sourc
1d00: 65 5f 69 64 3b 22 29 0a 09 09 09 2e 62 69 6e 64 e_id;").....bind
1d10: 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e (source_id).....
1d20: 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e bind(owner).....
1d30: 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e execute(&mut con
1d40: 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 n).await.....wit
1d50: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
1d60: 6d 61 74 21 28 22 43 6c 65 61 6e 20 73 65 65 6e mat!("Clean seen
1d70: 20 70 6f 73 74 73 3a 5c 6e 7b 3a 3f 7d 22 2c 20 posts:\n{:?}",
1d80: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 &self.pool))?...
1d90: 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 ..rows_affected(
1da0: 29 20 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b ) {....0 => { Ok
1db0: 28 22 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 ("No data found
1dc0: 66 6f 75 6e 64 2e 22 2e 74 6f 5f 73 74 72 69 6e found.".to_strin
1dd0: 67 28 29 29 20 7d 2c 0a 09 09 09 78 20 3d 3e 20 g()) },....x =>
1de0: 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 22 7b 7d { Ok(format!("{}
1df0: 20 70 6f 73 74 73 20 70 75 72 67 65 64 2e 22 2c posts purged.",
1e00: 20 78 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a x)) },...}..}..
1e10: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 65 6e .pub async fn en
1e20: 61 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 able<S>(&self, s
1e30: 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 ource_id: &i32,
1e40: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
1e50: 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 ult<&str>..where
1e60: 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a S: Into<i64> {.
1e70: 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 ..let owner: i64
1e80: 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b = owner.into();
1e90: 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
1ea0: 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 = self.pool.acqu
1eb0: 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e ire().await.....
1ec0: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
1ed0: 66 6f 72 6d 61 74 21 28 22 45 6e 61 62 6c 65 20 format!("Enable
1ee0: 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f fetch conn:\n{:?
1ef0: 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 }", &self.pool))
1f00: 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a ?;...match sqlx:
1f10: 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 :query("update r
1f20: 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 sstg_source set
1f30: 65 6e 61 62 6c 65 64 20 3d 20 74 72 75 65 20 77 enabled = true w
1f40: 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d here source_id =
1f50: 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 $1 and owner =
1f60: 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f $2").....bind(so
1f70: 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e urce_id).....bin
1f80: 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 d(owner).....exe
1f90: 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e cute(&mut conn).
1fa0: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
1fb0: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1fc0: 21 28 22 45 6e 61 62 6c 65 20 73 6f 75 72 63 65 !("Enable source
1fd0: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1fe0: 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 pool))?.....rows
1ff0: 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 _affected() {...
2000: 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 .1 => { Ok("Sour
2010: 63 65 20 65 6e 61 62 6c 65 64 2e 22 29 20 7d 2c ce enabled.") },
2020: 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 ....0 => { Ok("S
2030: 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 2e ource not found.
2040: 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 ") },...._ => {
2050: 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 61 74 Err(anyhow!("Dat
2060: 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 29 20 abase error."))
2070: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 },...}..}...pub
2080: 61 73 79 6e 63 20 66 6e 20 64 69 73 61 62 6c 65 async fn disable
2090: 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 <S>(&self, sourc
20a0: 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 e_id: &i32, owne
20b0: 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c r: S) -> Result<
20c0: 26 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 &str>..where S:
20d0: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
20e0: 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f t owner: i64 = o
20f0: 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c wner.into();...l
2100: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
2110: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
2120: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
2130: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
2140: 61 74 21 28 22 44 69 73 61 62 6c 65 20 66 65 74 at!("Disable fet
2150: 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c ch conn:\n{:?}",
2160: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a &self.pool))?;.
2170: 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 ..match sqlx::qu
2180: 65 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 ery("update rsst
2190: 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 6e 61 g_source set ena
21a0: 62 6c 65 64 20 3d 20 66 61 6c 73 65 20 77 68 65 bled = false whe
21b0: 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 re source_id = $
21c0: 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 1 and owner = $2
21d0: 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 ").....bind(sour
21e0: 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 ce_id).....bind(
21f0: 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 owner).....execu
2200: 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 te(&mut conn).aw
2210: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
2220: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
2230: 22 44 69 73 61 62 6c 65 20 73 6f 75 72 63 65 3a "Disable source:
2240: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
2250: 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f ool))?.....rows_
2260: 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 affected() {....
2270: 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 1 => { Ok("Sourc
2280: 65 20 64 69 73 61 62 6c 65 64 2e 22 29 20 7d 2c e disabled.") },
2290: 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 ....0 => { Ok("S
22a0: 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 2e ource not found.
22b0: 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 ") },...._ => {
22c0: 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 61 74 Err(anyhow!("Dat
22d0: 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 29 20 abase error."))
22e0: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 },...}..}...pub
22f0: 61 73 79 6e 63 20 66 6e 20 75 70 64 61 74 65 3c async fn update<
2300: 53 3e 28 26 73 65 6c 66 2c 20 75 70 64 61 74 65 S>(&self, update
2310: 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c 20 63 : Option<i32>, c
2320: 68 61 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 63 68 hannel: &str, ch
2330: 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c 20 75 annel_id: i64, u
2340: 72 6c 3a 20 26 73 74 72 2c 20 69 76 5f 68 61 73 rl: &str, iv_has
2350: 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c h: Option<&str>,
2360: 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f 6e 3c url_re: Option<
2370: 26 73 74 72 3e 2c 20 6f 77 6e 65 72 3a 20 53 29 &str>, owner: S)
2380: 20 2d 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e -> Result<Strin
2390: 67 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 g>..where S: Int
23a0: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f o<i64> {...let o
23b0: 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 wner: i64 = owne
23c0: 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 r.into();...let
23d0: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
23e0: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
23f0: 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f wait.....with_co
2400: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
2410: 28 22 55 70 64 61 74 65 20 66 65 74 63 68 20 63 ("Update fetch c
2420: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
2430: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 0a 09 09 6d lf.pool))?;....m
2440: 61 74 63 68 20 6d 61 74 63 68 20 75 70 64 61 74 atch match updat
2450: 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 69 64 29 e {.....Some(id)
2460: 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a => {......sqlx:
2470: 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 :query("update r
2480: 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 sstg_source set
2490: 63 68 61 6e 6e 65 6c 5f 69 64 20 3d 20 24 32 2c channel_id = $2,
24a0: 20 75 72 6c 20 3d 20 24 33 2c 20 69 76 5f 68 61 url = $3, iv_ha
24b0: 73 68 20 3d 20 24 34 2c 20 6f 77 6e 65 72 20 3d sh = $4, owner =
24c0: 20 24 35 2c 20 63 68 61 6e 6e 65 6c 20 3d 20 24 $5, channel = $
24d0: 36 2c 20 75 72 6c 5f 72 65 20 3d 20 24 37 20 77 6, url_re = $7 w
24e0: 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d here source_id =
24f0: 20 24 31 22 29 2e 62 69 6e 64 28 69 64 29 0a 09 $1").bind(id)..
2500: 09 09 09 7d 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d ...},.....None =
2510: 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 > {......sqlx::q
2520: 75 65 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 uery("insert int
2530: 6f 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 28 o rsstg_source (
2540: 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c channel_id, url,
2550: 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c iv_hash, owner,
2560: 20 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 channel, url_re
2570: 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 ) values ($1, $2
2580: 2c 20 24 33 2c 20 24 34 2c 20 24 35 2c 20 24 36 , $3, $4, $5, $6
2590: 29 22 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a )").....},....}.
25a0: 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c ....bind(channel
25b0: 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 _id).....bind(ur
25c0: 6c 29 0a 09 09 09 2e 62 69 6e 64 28 69 76 5f 68 l).....bind(iv_h
25d0: 61 73 68 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 ash).....bind(ow
25e0: 6e 65 72 29 0a 09 09 09 2e 62 69 6e 64 28 63 68 ner).....bind(ch
25f0: 61 6e 6e 65 6c 29 0a 09 09 09 2e 62 69 6e 64 28 annel).....bind(
2600: 75 72 6c 5f 72 65 29 0a 09 09 09 2e 65 78 65 63 url_re).....exec
2610: 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 ute(&mut conn).a
2620: 77 61 69 74 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 wait {....Ok(_)
2630: 3d 3e 20 72 65 74 75 72 6e 20 4f 6b 28 53 74 72 => return Ok(Str
2640: 69 6e 67 3a 3a 66 72 6f 6d 28 6d 61 74 63 68 20 ing::from(match
2650: 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d update {.....Som
2660: 65 28 5f 29 20 3d 3e 20 22 43 68 61 6e 6e 65 6c e(_) => "Channel
2670: 20 75 70 64 61 74 65 64 2e 22 2c 0a 09 09 09 09 updated.",.....
2680: 4e 6f 6e 65 20 3d 3e 20 22 43 68 61 6e 6e 65 6c None => "Channel
2690: 20 61 64 64 65 64 2e 22 2c 0a 09 09 09 7d 29 29 added.",....}))
26a0: 2c 0a 09 09 09 45 72 72 28 73 71 6c 78 3a 3a 45 ,....Err(sqlx::E
26b0: 72 72 6f 72 3a 3a 44 61 74 61 62 61 73 65 28 65 rror::Database(e
26c0: 72 72 29 29 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 rr)) => {.....ma
26d0: 74 63 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 tch err.downcast
26e0: 3a 3a 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 ::<sqlx::postgre
26f0: 73 3a 3a 50 67 44 61 74 61 62 61 73 65 45 72 72 s::PgDatabaseErr
2700: 6f 72 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 or>().routine()
2710: 7b 0a 09 09 09 09 09 53 6f 6d 65 28 22 5f 62 74 {......Some("_bt
2720: 5f 63 68 65 63 6b 5f 75 6e 69 71 75 65 22 2c 20 _check_unique",
2730: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 72 65 74 ) => {.......ret
2740: 75 72 6e 20 4f 6b 28 22 44 75 70 6c 69 63 61 74 urn Ok("Duplicat
2750: 65 20 6b 65 79 2e 22 2e 74 6f 5f 73 74 72 69 6e e key.".to_strin
2760: 67 28 29 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 g())......},....
2770: 09 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 7b 0a 09 ..Some(_) => {..
2780: 09 09 09 09 09 72 65 74 75 72 6e 20 4f 6b 28 22 .....return Ok("
2790: 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 Database error."
27a0: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 0a 09 09 .to_string())...
27b0: 09 09 09 7d 2c 0a 09 09 09 09 09 4e 6f 6e 65 20 ...},......None
27c0: 3d 3e 20 7b 0a 09 09 09 09 09 09 72 65 74 75 72 => {.......retur
27d0: 6e 20 4f 6b 28 22 4e 6f 20 64 61 74 61 62 61 73 n Ok("No databas
27e0: 65 20 65 72 72 6f 72 20 65 78 74 72 61 63 74 65 e error extracte
27f0: 64 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 d.".to_string())
2800: 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 7d 3b 0a ......},.....};.
2810: 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 65 72 72 ...},....Err(err
2820: 29 20 3d 3e 20 7b 0a 09 09 09 09 62 61 69 6c 21 ) => {.....bail!
2830: 28 22 53 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e ("Sorry, unknown
2840: 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 3f 7d 5c 6e error:\n{:#?}\n
2850: 22 2c 20 65 72 72 29 3b 0a 09 09 09 7d 2c 0a 09 ", err);....},..
2860: 09 7d 3b 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 66 .};..}...async f
2870: 6e 20 61 75 74 6f 66 65 74 63 68 28 26 73 65 6c n autofetch(&sel
2880: 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e f) -> Result<()>
2890: 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 64 65 6c {...let mut del
28a0: 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 72 ay = chrono::Dur
28b0: 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 31 ation::minutes(1
28c0: 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 6e 6f 77 );...let mut now
28d0: 3b 0a 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 6c 65 ;...loop {....le
28e0: 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c t mut conn = sel
28f0: 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 f.pool.acquire()
2900: 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 74 68 .await......with
2910: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
2920: 61 74 21 28 22 41 75 74 6f 66 65 74 63 68 20 66 at!("Autofetch f
2930: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
2940: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
2950: 3b 0a 09 09 09 6e 6f 77 20 3d 20 63 68 72 6f 6e ;....now = chron
2960: 6f 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f 77 28 29 3b o::Local::now();
2970: 0a 09 09 09 6c 65 74 20 6d 75 74 20 71 75 65 75 ....let mut queu
2980: 65 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 e = sqlx::query(
2990: 22 73 65 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 "select source_i
29a0: 64 2c 20 6e 65 78 74 5f 66 65 74 63 68 2c 20 6f d, next_fetch, o
29b0: 77 6e 65 72 20 66 72 6f 6d 20 72 73 73 74 67 5f wner from rsstg_
29c0: 6f 72 64 65 72 20 6e 61 74 75 72 61 6c 20 6c 65 order natural le
29d0: 66 74 20 6a 6f 69 6e 20 72 73 73 74 67 5f 73 6f ft join rsstg_so
29e0: 75 72 63 65 20 77 68 65 72 65 20 6e 65 78 74 5f urce where next_
29f0: 66 65 74 63 68 20 3c 20 6e 6f 77 28 29 20 2b 20 fetch < now() +
2a00: 69 6e 74 65 72 76 61 6c 20 27 31 20 6d 69 6e 75 interval '1 minu
2a10: 74 65 27 3b 22 29 0a 09 09 09 09 2e 66 65 74 63 te';")......fetc
2a20: 68 5f 61 6c 6c 28 26 6d 75 74 20 63 6f 6e 6e 29 h_all(&mut conn)
2a30: 2e 61 77 61 69 74 3f 3b 0a 09 09 09 66 6f 72 20 .await?;....for
2a40: 72 6f 77 20 69 6e 20 71 75 65 75 65 2e 69 74 65 row in queue.ite
2a50: 72 28 29 20 7b 0a 09 09 09 09 6c 65 74 20 73 6f r() {.....let so
2a60: 75 72 63 65 5f 69 64 3a 20 69 33 32 20 3d 20 72 urce_id: i32 = r
2a70: 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 6f 75 72 ow.try_get("sour
2a80: 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 09 6c 65 ce_id")?;.....le
2a90: 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 72 t owner: i64 = r
2aa0: 6f 77 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e 65 ow.try_get("owne
2ab0: 72 22 29 3f 3b 0a 09 09 09 09 6c 65 74 20 6e 65 r")?;.....let ne
2ac0: 78 74 5f 66 65 74 63 68 3a 20 44 61 74 65 54 69 xt_fetch: DateTi
2ad0: 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c me<chrono::Local
2ae0: 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 > = row.try_get(
2af0: 22 6e 65 78 74 5f 66 65 74 63 68 22 29 3f 3b 0a "next_fetch")?;.
2b00: 09 09 09 09 69 66 20 6e 65 78 74 5f 66 65 74 63 ....if next_fetc
2b10: 68 20 3c 20 6e 6f 77 20 7b 0a 09 09 09 09 09 2f h < now {....../
2b20: 2f 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 73 65 6c /let clone = sel
2b30: 66 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 09 09 09 f.clone();......
2b40: 2f 2f 63 6c 6f 6e 65 2e 6f 77 6e 65 72 5f 63 68 //clone.owner_ch
2b50: 61 74 28 55 73 65 72 49 64 3a 3a 6e 65 77 28 6f at(UserId::new(o
2b60: 77 6e 65 72 29 29 3b 0a 09 09 09 09 09 6c 65 74 wner));......let
2b70: 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 65 20 7b 0a clone = Core {.
2b80: 09 09 09 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 ......owner_chat
2b90: 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a : telegram_bot::
2ba0: 55 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 UserId::new(owne
2bb0: 72 29 2c 0a 09 09 09 09 09 09 2e 2e 73 65 6c 66 r),.........self
2bc0: 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 09 7d 3b .clone()......};
2bd0: 0a 09 09 09 09 09 74 6f 6b 69 6f 3a 3a 73 70 61 ......tokio::spa
2be0: 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a wn(async move {.
2bf0: 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 ......if let Err
2c00: 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 68 (err) = clone.ch
2c10: 65 63 6b 28 26 73 6f 75 72 63 65 5f 69 64 2c 20 eck(&source_id,
2c20: 6f 77 6e 65 72 2c 20 74 72 75 65 29 2e 61 77 61 owner, true).awa
2c30: 69 74 20 7b 0a 09 09 09 09 09 09 09 69 66 20 6c it {........if l
2c40: 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c et Err(err) = cl
2c50: 6f 6e 65 2e 73 65 6e 64 28 26 66 6f 72 6d 61 74 one.send(&format
2c60: 21 28 22 f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 65 !("š {:?}", e
2c70: 72 72 29 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 rr), None, None)
2c80: 20 7b 0a 09 09 09 09 09 09 09 09 65 70 72 69 6e {.........eprin
2c90: 74 6c 6e 21 28 22 43 68 65 63 6b 20 65 72 72 6f tln!("Check erro
2ca0: 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 r: {}", err);...
2cb0: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 3b .....};.......};
2cc0: 0a 09 09 09 09 09 7d 29 3b 0a 09 09 09 09 7d 20 ......});.....}
2cd0: 65 6c 73 65 20 7b 0a 09 09 09 09 09 69 66 20 6e else {......if n
2ce0: 65 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 20 ext_fetch - now
2cf0: 3c 20 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 09 < delay {.......
2d00: 64 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 delay = next_fet
2d10: 63 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 09 7d ch - now;......}
2d20: 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 .....}....};....
2d30: 71 75 65 75 65 2e 63 6c 65 61 72 28 29 3b 0a 09 queue.clear();..
2d40: 09 09 74 6f 6b 69 6f 3a 3a 74 69 6d 65 3a 3a 73 ..tokio::time::s
2d50: 6c 65 65 70 28 64 65 6c 61 79 2e 74 6f 5f 73 74 leep(delay.to_st
2d60: 64 28 29 3f 29 2e 61 77 61 69 74 3b 0a 09 09 09 d()?).await;....
2d70: 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a delay = chrono::
2d80: 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 Duration::minute
2d90: 73 28 31 29 3b 0a 09 09 7d 0a 09 7d 0a 0a 09 70 s(1);...}..}...p
2da0: 75 62 20 61 73 79 6e 63 20 66 6e 20 6c 69 73 74 ub async fn list
2db0: 3c 53 3e 28 26 73 65 6c 66 2c 20 6f 77 6e 65 72 <S>(&self, owner
2dc0: 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 : S) -> Result<S
2dd0: 74 72 69 6e 67 3e 0a 09 77 68 65 72 65 20 53 3a tring>..where S:
2de0: 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c Into<i64> {...l
2df0: 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 et owner = owner
2e00: 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d .into();...let m
2e10: 75 74 20 72 65 70 6c 79 20 3d 20 76 65 63 21 5b ut reply = vec![
2e20: 5d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e ];...let mut con
2e30: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
2e40: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
2e50: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
2e60: 7c 20 66 6f 72 6d 61 74 21 28 22 4c 69 73 74 20 | format!("List
2e70: 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f fetch conn:\n{:?
2e80: 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 }", &self.pool))
2e90: 3f 3b 0a 09 09 72 65 70 6c 79 2e 70 75 73 68 28 ?;...reply.push(
2ea0: 22 43 68 61 6e 6e 65 6c 73 3a 22 2e 74 6f 5f 73 "Channels:".to_s
2eb0: 74 72 69 6e 67 28 29 29 3b 0a 09 09 6c 65 74 20 tring());...let
2ec0: 72 6f 77 73 20 3d 20 73 71 6c 78 3a 3a 71 75 65 rows = sqlx::que
2ed0: 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 ry("select sourc
2ee0: 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 65 e_id, channel, e
2ef0: 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 76 5f nabled, url, iv_
2f00: 68 61 73 68 20 66 72 6f 6d 20 72 73 73 74 67 5f hash from rsstg_
2f10: 73 6f 75 72 63 65 20 77 68 65 72 65 20 6f 77 6e source where own
2f20: 65 72 20 3d 20 24 31 20 6f 72 64 65 72 20 62 79 er = $1 order by
2f30: 20 73 6f 75 72 63 65 5f 69 64 22 29 0a 09 09 09 source_id")....
2f40: 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 .bind(owner)....
2f50: 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 .fetch_all(&mut
2f60: 63 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 conn).await?;...
2f70: 66 6f 72 20 72 6f 77 20 69 6e 20 72 6f 77 73 2e for row in rows.
2f80: 69 74 65 72 28 29 20 7b 0a 09 09 09 6c 65 74 20 iter() {....let
2f90: 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 20 3d source_id: i32 =
2fa0: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 6f row.try_get("so
2fb0: 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 6c urce_id")?;....l
2fc0: 65 74 20 75 73 65 72 6e 61 6d 65 3a 20 26 73 74 et username: &st
2fd0: 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 r = row.try_get(
2fe0: 22 63 68 61 6e 6e 65 6c 22 29 3f 3b 0a 09 09 09 "channel")?;....
2ff0: 6c 65 74 20 65 6e 61 62 6c 65 64 3a 20 62 6f 6f let enabled: boo
3000: 6c 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 l = row.try_get(
3010: 22 65 6e 61 62 6c 65 64 22 29 3f 3b 0a 09 09 09 "enabled")?;....
3020: 6c 65 74 20 75 72 6c 3a 20 26 73 74 72 20 3d 20 let url: &str =
3030: 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c row.try_get("url
3040: 22 29 3f 3b 0a 09 09 09 6c 65 74 20 69 76 5f 68 ")?;....let iv_h
3050: 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 ash: Option<&str
3060: 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 > = row.try_get(
3070: 22 69 76 5f 68 61 73 68 22 29 3f 3b 0a 09 09 09 "iv_hash")?;....
3080: 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 reply.push(forma
3090: 74 21 28 22 5c 6e 5c 5c 23 ef b8 8f e2 83 a3 20 t!("\n\\#ļøā£
30a0: 7b 7d 20 5c 5c 2a ef b8 8f e2 83 a3 20 60 7b 7d {} \\*ļøā£ `{}
30b0: 60 20 7b 7d 5c 6e f0 9f 94 97 20 60 7b 7d 60 22 ` {}\nš `{}`"
30c0: 2c 20 73 6f 75 72 63 65 5f 69 64 2c 20 75 73 65 , source_id, use
30d0: 72 6e 61 6d 65 2c 20 20 0a 09 09 09 09 6d 61 74 rname, .....mat
30e0: 63 68 20 65 6e 61 62 6c 65 64 20 7b 0a 09 09 09 ch enabled {....
30f0: 09 09 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 ..true => "š
3100: 20 65 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 enabled",......
3110: 66 61 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 false => "ā di
3120: 73 61 62 6c 65 64 22 2c 0a 09 09 09 09 7d 2c 20 sabled",.....},
3130: 75 72 6c 29 29 3b 0a 09 09 09 69 66 20 6c 65 74 url));....if let
3140: 20 53 6f 6d 65 28 68 61 73 68 29 20 3d 20 69 76 Some(hash) = iv
3150: 5f 68 61 73 68 20 7b 0a 09 09 09 09 72 65 70 6c _hash {.....repl
3160: 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 y.push(format!("
3170: 49 56 20 60 7b 7d 60 22 2c 20 68 61 73 68 29 29 IV `{}`", hash))
3180: 3b 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 4f 6b ;....}...};...Ok
3190: 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 (reply.join("\n"
31a0: 29 29 0a 09 7d 0a 7d 0a ))..}.}.