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 67 65 78 3a 3a fig;.use regex::
0070: 52 65 67 65 78 3b 0a 75 73 65 20 72 65 71 77 65 Regex;.use reqwe
0080: 73 74 3b 0a 75 73 65 20 73 71 6c 78 3a 3a 7b 0a st;.use sqlx::{.
0090: 09 70 6f 73 74 67 72 65 73 3a 3a 50 67 50 6f 6f .postgres::PgPoo
00a0: 6c 4f 70 74 69 6f 6e 73 2c 0a 09 52 6f 77 2c 0a lOptions,..Row,.
00b0: 7d 3b 0a 75 73 65 20 72 73 73 3b 0a 75 73 65 20 };.use rss;.use
00c0: 73 74 64 3a 3a 7b 0a 09 63 6f 6c 6c 65 63 74 69 std::{..collecti
00d0: 6f 6e 73 3a 3a 7b 0a 09 09 42 54 72 65 65 4d 61 ons::{...BTreeMa
00e0: 70 2c 0a 09 09 48 61 73 68 53 65 74 2c 0a 09 7d p,...HashSet,..}
00f0: 2c 0a 09 73 79 6e 63 3a 3a 7b 41 72 63 2c 20 4d ,..sync::{Arc, M
0100: 75 74 65 78 7d 2c 0a 7d 3b 0a 75 73 65 20 74 65 utex},.};.use te
0110: 6c 65 67 72 61 6d 5f 62 6f 74 3b 0a 0a 23 5b 64 legram_bot;..#[d
0120: 65 72 69 76 65 28 43 6c 6f 6e 65 29 5d 0a 70 75 erive(Clone)].pu
0130: 62 20 73 74 72 75 63 74 20 43 6f 72 65 20 7b 0a b struct Core {.
0140: 09 6f 77 6e 65 72 3a 20 69 36 34 2c 0a 09 61 70 .owner: i64,..ap
0150: 69 5f 6b 65 79 3a 20 53 74 72 69 6e 67 2c 0a 09 i_key: String,..
0160: 6f 77 6e 65 72 5f 63 68 61 74 3a 20 74 65 6c 65 owner_chat: tele
0170: 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 gram_bot::UserId
0180: 2c 0a 09 70 75 62 20 74 67 3a 20 74 65 6c 65 67 ,..pub tg: teleg
0190: 72 61 6d 5f 62 6f 74 3a 3a 41 70 69 2c 0a 09 70 ram_bot::Api,..p
01a0: 75 62 20 6d 79 3a 20 74 65 6c 65 67 72 61 6d 5f ub my: telegram_
01b0: 62 6f 74 3a 3a 55 73 65 72 2c 0a 09 70 6f 6f 6c bot::User,..pool
01c0: 3a 20 73 71 6c 78 3a 3a 50 6f 6f 6c 3c 73 71 6c : sqlx::Pool<sql
01d0: 78 3a 3a 50 6f 73 74 67 72 65 73 3e 2c 0a 09 73 x::Postgres>,..s
01e0: 6f 75 72 63 65 73 3a 20 41 72 63 3c 4d 75 74 65 ources: Arc<Mute
01f0: 78 3c 48 61 73 68 53 65 74 3c 41 72 63 3c 69 33 x<HashSet<Arc<i3
0200: 32 3e 3e 3e 3e 2c 0a 7d 0a 0a 69 6d 70 6c 20 43 2>>>>,.}..impl C
0210: 6f 72 65 20 7b 0a 09 70 75 62 20 61 73 79 6e 63 ore {..pub async
0220: 20 66 6e 20 6e 65 77 28 73 65 74 74 69 6e 67 73 fn new(settings
0230: 3a 20 63 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 : config::Config
0240: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 72 65 ) -> Result<Core
0250: 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 > {...let owner
0260: 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 5f 69 = settings.get_i
0270: 6e 74 28 22 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 nt("owner")?;...
0280: 6c 65 74 20 61 70 69 5f 6b 65 79 20 3d 20 73 65 let api_key = se
0290: 74 74 69 6e 67 73 2e 67 65 74 5f 73 74 72 28 22 ttings.get_str("
02a0: 61 70 69 5f 6b 65 79 22 29 3f 3b 0a 09 09 6c 65 api_key")?;...le
02b0: 74 20 74 67 20 3d 20 74 65 6c 65 67 72 61 6d 5f t tg = telegram_
02c0: 62 6f 74 3a 3a 41 70 69 3a 3a 6e 65 77 28 26 61 bot::Api::new(&a
02d0: 70 69 5f 6b 65 79 29 3b 0a 09 09 6c 65 74 20 63 pi_key);...let c
02e0: 6f 72 65 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 ore = Core {....
02f0: 6f 77 6e 65 72 3a 20 6f 77 6e 65 72 2c 0a 09 09 owner: owner,...
0300: 09 61 70 69 5f 6b 65 79 3a 20 61 70 69 5f 6b 65 .api_key: api_ke
0310: 79 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 6d 79 y.clone(),....my
0320: 3a 20 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 72 : tg.send(telegr
0330: 61 6d 5f 62 6f 74 3a 3a 47 65 74 4d 65 29 2e 61 am_bot::GetMe).a
0340: 77 61 69 74 3f 2c 0a 09 09 09 74 67 3a 20 74 67 wait?,....tg: tg
0350: 2c 0a 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a ,....owner_chat:
0360: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 telegram_bot::U
0370: 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 serId::new(owner
0380: 29 2c 0a 09 09 09 70 6f 6f 6c 3a 20 50 67 50 6f ),....pool: PgPo
0390: 6f 6c 4f 70 74 69 6f 6e 73 3a 3a 6e 65 77 28 29 olOptions::new()
03a0: 0a 09 09 09 09 2e 6d 61 78 5f 63 6f 6e 6e 65 63 ......max_connec
03b0: 74 69 6f 6e 73 28 35 29 0a 09 09 09 09 2e 63 6f tions(5)......co
03c0: 6e 6e 65 63 74 5f 74 69 6d 65 6f 75 74 28 73 74 nnect_timeout(st
03d0: 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f d::time::Duratio
03e0: 6e 3a 3a 6e 65 77 28 33 30 30 2c 20 30 29 29 0a n::new(300, 0)).
03f0: 09 09 09 09 2e 69 64 6c 65 5f 74 69 6d 65 6f 75 .....idle_timeou
0400: 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 t(std::time::Dur
0410: 61 74 69 6f 6e 3a 3a 6e 65 77 28 36 30 2c 20 30 ation::new(60, 0
0420: 29 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 74 5f ))......connect_
0430: 6c 61 7a 79 28 26 73 65 74 74 69 6e 67 73 2e 67 lazy(&settings.g
0440: 65 74 5f 73 74 72 28 22 70 67 22 29 3f 29 3f 2c et_str("pg")?)?,
0450: 0a 09 09 09 73 6f 75 72 63 65 73 3a 20 41 72 63 ....sources: Arc
0460: 3a 3a 6e 65 77 28 4d 75 74 65 78 3a 3a 6e 65 77 ::new(Mutex::new
0470: 28 48 61 73 68 53 65 74 3a 3a 6e 65 77 28 29 29 (HashSet::new())
0480: 29 2c 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 63 6c ),...};...let cl
0490: 6f 6e 65 20 3d 20 63 6f 72 65 2e 63 6c 6f 6e 65 one = core.clone
04a0: 28 29 3b 0a 09 09 74 6f 6b 69 6f 3a 3a 73 70 61 ();...tokio::spa
04b0: 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a wn(async move {.
04c0: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
04d0: 72 29 20 3d 20 26 63 6c 6f 6e 65 2e 61 75 74 6f r) = &clone.auto
04e0: 66 65 74 63 68 28 29 2e 61 77 61 69 74 20 7b 0a fetch().await {.
04f0: 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 ....if let Err(e
0500: 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 rr) = clone.send
0510: 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 (&format!("🛑
0520: 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e {:?}", err), Non
0530: 65 2c 20 4e 6f 6e 65 29 20 7b 0a 09 09 09 09 09 e, None) {......
0540: 65 70 72 69 6e 74 6c 6e 21 28 22 41 75 74 6f 66 eprintln!("Autof
0550: 65 74 63 68 20 65 72 72 6f 72 3a 20 7b 7d 22 2c etch error: {}",
0560: 20 65 72 72 29 3b 0a 09 09 09 09 7d 3b 0a 09 09 err);.....};...
0570: 09 7d 0a 09 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f .}...});...Ok(co
0580: 72 65 29 0a 09 7d 0a 0a 09 70 75 62 20 66 6e 20 re)..}...pub fn
0590: 73 74 72 65 61 6d 28 26 73 65 6c 66 29 20 2d 3e stream(&self) ->
05a0: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 telegram_bot::U
05b0: 70 64 61 74 65 73 53 74 72 65 61 6d 20 7b 0a 09 pdatesStream {..
05c0: 09 73 65 6c 66 2e 74 67 2e 73 74 72 65 61 6d 28 .self.tg.stream(
05d0: 29 0a 09 7d 0a 0a 09 70 75 62 20 66 6e 20 73 65 )..}...pub fn se
05e0: 6e 64 3c 53 3e 28 26 73 65 6c 66 2c 20 6d 73 67 nd<S>(&self, msg
05f0: 3a 20 53 2c 20 74 61 72 67 65 74 3a 20 4f 70 74 : S, target: Opt
0600: 69 6f 6e 3c 74 65 6c 65 67 72 61 6d 5f 62 6f 74 ion<telegram_bot
0610: 3a 3a 55 73 65 72 49 64 3e 2c 20 70 61 72 73 65 ::UserId>, parse
0620: 5f 6d 6f 64 65 3a 20 4f 70 74 69 6f 6e 3c 74 65 _mode: Option<te
0630: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 legram_bot::type
0640: 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3e 29 20 2d s::ParseMode>) -
0650: 3e 20 52 65 73 75 6c 74 3c 28 29 3e 0a 09 77 68 > Result<()>..wh
0660: 65 72 65 20 53 3a 20 49 6e 74 6f 3c 53 74 72 69 ere S: Into<Stri
0670: 6e 67 3e 20 7b 0a 09 09 6c 65 74 20 6d 73 67 3a ng> {...let msg:
0680: 20 53 74 72 69 6e 67 20 3d 20 6d 73 67 2e 69 6e String = msg.in
0690: 74 6f 28 29 3b 0a 09 09 6c 65 74 20 70 61 72 73 to();...let pars
06a0: 65 5f 6d 6f 64 65 20 3d 20 6d 61 74 63 68 20 70 e_mode = match p
06b0: 61 72 73 65 5f 6d 6f 64 65 20 7b 0a 09 09 09 53 arse_mode {....S
06c0: 6f 6d 65 28 6d 6f 64 65 29 20 3d 3e 20 6d 6f 64 ome(mode) => mod
06d0: 65 2c 0a 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 65 e,....None => te
06e0: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 legram_bot::type
06f0: 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 s::ParseMode::Ht
0700: 6d 6c 2c 0a 09 09 7d 3b 0a 09 09 73 65 6c 66 2e ml,...};...self.
0710: 74 67 2e 73 70 61 77 6e 28 74 65 6c 65 67 72 61 tg.spawn(telegra
0720: 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 73 61 m_bot::SendMessa
0730: 67 65 3a 3a 6e 65 77 28 6d 61 74 63 68 20 74 61 ge::new(match ta
0740: 72 67 65 74 20 7b 0a 09 09 09 53 6f 6d 65 28 75 rget {....Some(u
0750: 73 65 72 29 20 3d 3e 20 75 73 65 72 2c 0a 09 09 ser) => user,...
0760: 09 4e 6f 6e 65 20 3d 3e 20 73 65 6c 66 2e 6f 77 .None => self.ow
0770: 6e 65 72 5f 63 68 61 74 2c 0a 09 09 7d 2c 20 6d ner_chat,...}, m
0780: 73 67 2e 74 6f 5f 6f 77 6e 65 64 28 29 29 2e 70 sg.to_owned()).p
0790: 61 72 73 65 5f 6d 6f 64 65 28 70 61 72 73 65 5f arse_mode(parse_
07a0: 6d 6f 64 65 29 29 3b 0a 09 09 4f 6b 28 28 29 29 mode));...Ok(())
07b0: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
07c0: 66 6e 20 63 68 65 63 6b 3c 53 3e 28 26 73 65 6c fn check<S>(&sel
07d0: 66 2c 20 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e f, id: &i32, own
07e0: 65 72 3a 20 53 2c 20 72 65 61 6c 3a 20 62 6f 6f er: S, real: boo
07f0: 6c 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 74 72 l) -> Result<Str
0800: 69 6e 67 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 ing>..where S: I
0810: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
0820: 20 6d 75 74 20 70 6f 73 74 65 64 3a 20 69 33 32 mut posted: i32
0830: 20 3d 20 30 3b 0a 09 09 6c 65 74 20 6f 77 6e 65 = 0;...let owne
0840: 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 r: i64 = owner.i
0850: 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 69 64 20 nto();...let id
0860: 3d 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 73 = {....let mut s
0870: 65 74 20 3d 20 73 65 6c 66 2e 73 6f 75 72 63 65 et = self.source
0880: 73 2e 6c 6f 63 6b 28 29 2e 75 6e 77 72 61 70 28 s.lock().unwrap(
0890: 29 3b 0a 09 09 09 6d 61 74 63 68 20 73 65 74 2e );....match set.
08a0: 67 65 74 28 69 64 29 20 7b 0a 09 09 09 09 53 6f get(id) {.....So
08b0: 6d 65 28 69 64 29 20 3d 3e 20 69 64 2e 63 6c 6f me(id) => id.clo
08c0: 6e 65 28 29 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d ne(),.....None =
08d0: 3e 20 7b 0a 09 09 09 09 09 6c 65 74 20 69 64 20 > {......let id
08e0: 3d 20 41 72 63 3a 3a 6e 65 77 28 2a 69 64 29 3b = Arc::new(*id);
08f0: 0a 09 09 09 09 09 73 65 74 2e 69 6e 73 65 72 74 ......set.insert
0900: 28 69 64 2e 63 6c 6f 6e 65 28 29 29 3b 0a 09 09 (id.clone());...
0910: 09 09 09 69 64 2e 63 6c 6f 6e 65 28 29 0a 09 09 ...id.clone()...
0920: 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 ..},....}...};..
0930: 09 6c 65 74 20 63 6f 75 6e 74 20 3d 20 41 72 63 .let count = Arc
0940: 3a 3a 73 74 72 6f 6e 67 5f 63 6f 75 6e 74 28 26 ::strong_count(&
0950: 69 64 29 3b 0a 09 09 69 66 20 63 6f 75 6e 74 20 id);...if count
0960: 3d 3d 20 32 20 7b 0a 09 09 09 6c 65 74 20 6d 75 == 2 {....let mu
0970: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
0980: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
0990: 69 74 0a 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e it......with_con
09a0: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
09b0: 22 51 75 65 72 79 20 71 75 65 75 65 20 66 65 74 "Query queue fet
09c0: 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c ch conn:\n{:?}",
09d0: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a &self.pool))?;.
09e0: 09 09 09 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c ...let row = sql
09f0: 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 x::query("select
0a00: 20 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e source_id, chan
0a10: 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f nel_id, url, iv_
0a20: 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 75 72 6c hash, owner, url
0a30: 5f 72 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 _re from rsstg_s
0a40: 6f 75 72 63 65 20 77 68 65 72 65 20 73 6f 75 72 ource where sour
0a50: 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f ce_id = $1 and o
0a60: 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 09 09 wner = $2").....
0a70: 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 2e .bind(*id)......
0a80: 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 09 bind(owner).....
0a90: 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 .fetch_one(&mut
0aa0: 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 conn).await.....
0ab0: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
0ac0: 20 66 6f 72 6d 61 74 21 28 22 51 75 65 72 79 20 format!("Query
0ad0: 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 source:\n{:?}",
0ae0: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 &self.pool))?;..
0af0: 09 09 64 72 6f 70 28 63 6f 6e 6e 29 3b 0a 09 09 ..drop(conn);...
0b00: 09 6c 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 3a .let channel_id:
0b10: 20 69 36 34 20 3d 20 72 6f 77 2e 74 72 79 5f 67 i64 = row.try_g
0b20: 65 74 28 22 63 68 61 6e 6e 65 6c 5f 69 64 22 29 et("channel_id")
0b30: 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 ?;....let url: &
0b40: 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 str = row.try_ge
0b50: 74 28 22 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 t("url")?;....le
0b60: 74 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f t iv_hash: Optio
0b70: 6e 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 n<&str> = row.tr
0b80: 79 5f 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 y_get("iv_hash")
0b90: 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c 5f 72 65 ?;....let url_re
0ba0: 3a 20 4f 70 74 69 6f 6e 3c 72 65 67 65 78 3a 3a : Option<regex::
0bb0: 52 65 67 65 78 3e 20 3d 20 6d 61 74 63 68 20 72 Regex> = match r
0bc0: 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 5f ow.try_get("url_
0bd0: 72 65 22 29 3f 20 7b 0a 09 09 09 09 53 6f 6d 65 re")? {.....Some
0be0: 28 78 29 20 3d 3e 20 53 6f 6d 65 28 52 65 67 65 (x) => Some(Rege
0bf0: 78 3a 3a 6e 65 77 28 78 29 3f 29 2c 0a 09 09 09 x::new(x)?),....
0c00: 09 4e 6f 6e 65 20 3d 3e 20 4e 6f 6e 65 2c 0a 09 .None => None,..
0c10: 09 09 7d 3b 0a 09 09 09 6c 65 74 20 64 65 73 74 ..};....let dest
0c20: 69 6e 61 74 69 6f 6e 20 3d 20 6d 61 74 63 68 20 ination = match
0c30: 72 65 61 6c 20 7b 0a 09 09 09 09 74 72 75 65 20 real {.....true
0c40: 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a => telegram_bot:
0c50: 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 63 68 61 :UserId::new(cha
0c60: 6e 6e 65 6c 5f 69 64 29 2c 0a 09 09 09 09 66 61 nnel_id),.....fa
0c70: 6c 73 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f lse => telegram_
0c80: 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 bot::UserId::new
0c90: 28 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 6f 77 (row.try_get("ow
0ca0: 6e 65 72 22 29 3f 29 2c 0a 09 09 09 7d 3b 0a 09 ner")?),....};..
0cb0: 09 09 6c 65 74 20 6d 75 74 20 74 68 69 73 5f 66 ..let mut this_f
0cc0: 65 74 63 68 3a 20 4f 70 74 69 6f 6e 3c 44 61 74 etch: Option<Dat
0cd0: 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 eTime<chrono::Fi
0ce0: 78 65 64 4f 66 66 73 65 74 3e 3e 20 3d 20 4e 6f xedOffset>> = No
0cf0: 6e 65 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 70 ne;....let mut p
0d00: 6f 73 74 73 3a 20 42 54 72 65 65 4d 61 70 3c 44 osts: BTreeMap<D
0d10: 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a ateTime<chrono::
0d20: 46 69 78 65 64 4f 66 66 73 65 74 3e 2c 20 53 74 FixedOffset>, St
0d30: 72 69 6e 67 3e 20 3d 20 42 54 72 65 65 4d 61 70 ring> = BTreeMap
0d40: 3a 3a 6e 65 77 28 29 3b 0a 09 09 09 6c 65 74 20 ::new();....let
0d50: 63 6f 6e 74 65 6e 74 20 3d 20 72 65 71 77 65 73 content = reqwes
0d60: 74 3a 3a 67 65 74 28 75 72 6c 29 2e 61 77 61 69 t::get(url).awai
0d70: 74 3f 2e 62 79 74 65 73 28 29 2e 61 77 61 69 74 t?.bytes().await
0d80: 3f 3b 0a 09 09 09 6d 61 74 63 68 20 72 73 73 3a ?;....match rss:
0d90: 3a 43 68 61 6e 6e 65 6c 3a 3a 72 65 61 64 5f 66 :Channel::read_f
0da0: 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d rom(&content[..]
0db0: 29 20 7b 0a 09 09 09 09 4f 6b 28 66 65 65 64 29 ) {.....Ok(feed)
0dc0: 20 3d 3e 20 7b 0a 09 09 09 09 09 66 6f 72 20 69 => {......for i
0dd0: 74 65 6d 20 69 6e 20 66 65 65 64 2e 69 74 65 6d tem in feed.item
0de0: 73 28 29 20 7b 0a 09 09 09 09 09 09 6d 61 74 63 s() {.......matc
0df0: 68 20 69 74 65 6d 2e 6c 69 6e 6b 28 29 20 7b 0a h item.link() {.
0e00: 09 09 09 09 09 09 09 53 6f 6d 65 28 6c 69 6e 6b .......Some(link
0e10: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 6c ) => {.........l
0e20: 65 74 20 64 61 74 65 20 3d 20 6d 61 74 63 68 20 et date = match
0e30: 69 74 65 6d 2e 70 75 62 5f 64 61 74 65 28 29 20 item.pub_date()
0e40: 7b 0a 09 09 09 09 09 09 09 09 09 53 6f 6d 65 28 {..........Some(
0e50: 66 65 65 64 5f 64 61 74 65 29 20 3d 3e 20 44 61 feed_date) => Da
0e60: 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 teTime::parse_fr
0e70: 6f 6d 5f 72 66 63 32 38 32 32 28 66 65 65 64 5f om_rfc2822(feed_
0e80: 64 61 74 65 29 2c 0a 09 09 09 09 09 09 09 09 09 date),..........
0e90: 4e 6f 6e 65 20 3d 3e 20 44 61 74 65 54 69 6d 65 None => DateTime
0ea0: 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 ::parse_from_rfc
0eb0: 33 33 33 39 28 26 69 74 65 6d 2e 64 75 62 6c 69 3339(&item.dubli
0ec0: 6e 5f 63 6f 72 65 5f 65 78 74 28 29 2e 75 6e 77 n_core_ext().unw
0ed0: 72 61 70 28 29 2e 64 61 74 65 73 28 29 5b 30 5d rap().dates()[0]
0ee0: 29 2c 0a 09 09 09 09 09 09 09 09 7d 3f 3b 0a 09 ),.........}?;..
0ef0: 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c 20 3d .......let url =
0f00: 20 6c 69 6e 6b 2e 74 6f 5f 73 74 72 69 6e 67 28 link.to_string(
0f10: 29 3b 0a 09 09 09 09 09 09 09 09 70 6f 73 74 73 );.........posts
0f20: 2e 69 6e 73 65 72 74 28 64 61 74 65 2e 63 6c 6f .insert(date.clo
0f30: 6e 65 28 29 2c 20 75 72 6c 2e 63 6c 6f 6e 65 28 ne(), url.clone(
0f40: 29 29 3b 0a 09 09 09 09 09 09 09 7d 2c 0a 09 09 ));........},...
0f50: 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 7d 0a .....None => {}.
0f60: 09 09 09 09 09 09 7d 0a 09 09 09 09 09 7d 3b 0a ......}......};.
0f70: 09 09 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 65 ....},.....Err(e
0f80: 72 72 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 72 rr) => match err
0f90: 20 7b 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 {......rss::Err
0fa0: 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 74 or::InvalidStart
0fb0: 54 61 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 6c Tag => {.......l
0fc0: 65 74 20 66 65 65 64 20 3d 20 61 74 6f 6d 5f 73 et feed = atom_s
0fd0: 79 6e 64 69 63 61 74 69 6f 6e 3a 3a 46 65 65 64 yndication::Feed
0fe0: 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f 6e ::read_from(&con
0ff0: 74 65 6e 74 5b 2e 2e 5d 29 0a 09 09 09 09 09 09 tent[..]).......
1000: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
1010: 7c 20 66 6f 72 6d 61 74 21 28 22 50 72 6f 62 6c | format!("Probl
1020: 65 6d 20 6f 70 65 6e 69 6e 67 20 66 65 65 64 20 em opening feed
1030: 75 72 6c 3a 5c 6e 7b 7d 22 2c 20 26 75 72 6c 29 url:\n{}", &url)
1040: 29 3f 3b 0a 09 09 09 09 09 09 66 6f 72 20 69 74 )?;.......for it
1050: 65 6d 20 69 6e 20 66 65 65 64 2e 65 6e 74 72 69 em in feed.entri
1060: 65 73 28 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 es() {........le
1070: 74 20 64 61 74 65 20 3d 20 69 74 65 6d 2e 70 75 t date = item.pu
1080: 62 6c 69 73 68 65 64 28 29 2e 75 6e 77 72 61 70 blished().unwrap
1090: 28 29 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 75 ();........let u
10a0: 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 rl = item.links(
10b0: 29 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 09 )[0].href();....
10c0: 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 ....posts.insert
10d0: 28 64 61 74 65 2e 63 6c 6f 6e 65 28 29 2c 20 75 (date.clone(), u
10e0: 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b rl.to_string());
10f0: 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 7d .......};......}
1100: 2c 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 6f ,......rss::Erro
1110: 72 3a 3a 45 6f 66 20 3d 3e 20 28 29 2c 0a 09 09 r::Eof => (),...
1120: 09 09 09 5f 20 3d 3e 20 62 61 69 6c 21 28 22 55 ..._ => bail!("U
1130: 6e 73 75 70 70 6f 72 74 65 64 20 6f 72 20 6d 61 nsupported or ma
1140: 6e 67 6c 65 64 20 63 6f 6e 74 65 6e 74 3a 5c 6e ngled content:\n
1150: 7b 3a 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 {:?}\n{:#?}\n",
1160: 26 75 72 6c 2c 20 65 72 72 29 0a 09 09 09 09 7d &url, err).....}
1170: 0a 09 09 09 7d 3b 0a 09 09 09 66 6f 72 20 28 64 ....};....for (d
1180: 61 74 65 2c 20 75 72 6c 29 20 69 6e 20 70 6f 73 ate, url) in pos
1190: 74 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 09 ts.iter() {.....
11a0: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
11b0: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
11c0: 28 29 2e 61 77 61 69 74 0a 09 09 09 09 09 2e 77 ().await.......w
11d0: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
11e0: 6f 72 6d 61 74 21 28 22 43 68 65 63 6b 20 70 6f ormat!("Check po
11f0: 73 74 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e st fetch conn:\n
1200: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
1210: 6c 29 29 3f 3b 0a 09 09 09 09 6c 65 74 20 72 6f l))?;.....let ro
1220: 77 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 w = sqlx::query(
1230: 22 73 65 6c 65 63 74 20 65 78 69 73 74 73 28 73 "select exists(s
1240: 65 6c 65 63 74 20 74 72 75 65 20 66 72 6f 6d 20 elect true from
1250: 72 73 73 74 67 5f 70 6f 73 74 20 77 68 65 72 65 rsstg_post where
1260: 20 75 72 6c 20 3d 20 24 31 20 61 6e 64 20 73 6f url = $1 and so
1270: 75 72 63 65 5f 69 64 20 3d 20 24 32 29 20 61 73 urce_id = $2) as
1280: 20 65 78 69 73 74 73 3b 22 29 0a 09 09 09 09 09 exists;")......
1290: 2e 62 69 6e 64 28 26 75 72 6c 29 0a 09 09 09 09 .bind(&url).....
12a0: 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 ..bind(*id).....
12b0: 09 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 ..fetch_one(&mut
12c0: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 conn).await....
12d0: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
12e0: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 68 65 63 || format!("Chec
12f0: 6b 20 70 6f 73 74 3a 5c 6e 7b 3a 3f 7d 22 2c 20 k post:\n{:?}",
1300: 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 09 09 6c 65 &conn))?;.....le
1310: 74 20 65 78 69 73 74 73 3a 20 62 6f 6f 6c 20 3d t exists: bool =
1320: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 65 78 row.try_get("ex
1330: 69 73 74 73 22 29 3f 3b 0a 09 09 09 09 69 66 20 ists")?;.....if
1340: 21 20 65 78 69 73 74 73 20 7b 0a 09 09 09 09 09 ! exists {......
1350: 69 66 20 74 68 69 73 5f 66 65 74 63 68 20 3d 3d if this_fetch ==
1360: 20 4e 6f 6e 65 20 7c 7c 20 2a 64 61 74 65 20 3e None || *date >
1370: 20 74 68 69 73 5f 66 65 74 63 68 2e 75 6e 77 72 this_fetch.unwr
1380: 61 70 28 29 20 7b 0a 09 09 09 09 09 09 74 68 69 ap() {.......thi
1390: 73 5f 66 65 74 63 68 20 3d 20 53 6f 6d 65 28 2a s_fetch = Some(*
13a0: 64 61 74 65 29 3b 0a 09 09 09 09 09 7d 3b 0a 09 date);......};..
13b0: 09 09 09 09 73 65 6c 66 2e 74 67 2e 73 65 6e 64 ....self.tg.send
13c0: 28 20 6d 61 74 63 68 20 69 76 5f 68 61 73 68 20 ( match iv_hash
13d0: 7b 0a 09 09 09 09 09 09 09 53 6f 6d 65 28 78 29 {........Some(x)
13e0: 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 => telegram_bot
13f0: 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e ::SendMessage::n
1400: 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c 20 ew(destination,
1410: 66 6f 72 6d 61 74 21 28 22 3c 61 20 68 72 65 66 format!("<a href
1420: 3d 5c 22 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f =\"https://t.me/
1430: 69 76 3f 75 72 6c 3d 7b 7d 26 72 68 61 73 68 3d iv?url={}&rhash=
1440: 7b 7d 5c 22 3e 20 3c 2f 61 3e 7b 30 7d 22 2c 20 {}\"> </a>{0}",
1450: 6d 61 74 63 68 20 26 75 72 6c 5f 72 65 20 7b 0a match &url_re {.
1460: 09 09 09 09 09 09 09 09 53 6f 6d 65 28 78 29 20 ........Some(x)
1470: 3d 3e 20 6d 61 74 63 68 20 78 2e 63 61 70 74 75 => match x.captu
1480: 72 65 73 28 26 75 72 6c 29 20 7b 0a 09 09 09 09 res(&url) {.....
1490: 09 09 09 09 09 53 6f 6d 65 28 78 29 20 3d 3e 20 .....Some(x) =>
14a0: 7b 0a 09 09 09 09 09 09 09 09 09 09 62 61 69 6c {...........bail
14b0: 21 28 22 52 65 67 65 78 20 68 69 74 2c 20 72 65 !("Regex hit, re
14c0: 73 75 6c 74 3a 5c 6e 7b 3a 23 3f 7d 22 2c 20 26 sult:\n{:#?}", &
14d0: 78 5b 30 5d 29 3b 0a 09 09 09 09 09 09 09 09 09 x[0]);..........
14e0: 09 26 78 5b 30 5d 0a 09 09 09 09 09 09 09 09 09 .&x[0]..........
14f0: 7d 2c 0a 09 09 09 09 09 09 09 09 09 4e 6f 6e 65 },..........None
1500: 20 3d 3e 20 26 75 72 6c 2c 0a 09 09 09 09 09 09 => &url,.......
1510: 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 4e 6f 6e ..},.........Non
1520: 65 20 3d 3e 20 26 75 72 6c 2c 0a 09 09 09 09 09 e => &url,......
1530: 09 09 7d 2c 20 78 29 29 2c 0a 09 09 09 09 09 09 ..}, x)),.......
1540: 09 4e 6f 6e 65 20 3d 3e 20 74 65 6c 65 67 72 61 .None => telegra
1550: 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 73 61 m_bot::SendMessa
1560: 67 65 3a 3a 6e 65 77 28 64 65 73 74 69 6e 61 74 ge::new(destinat
1570: 69 6f 6e 2c 20 66 6f 72 6d 61 74 21 28 22 7b 7d ion, format!("{}
1580: 22 2c 20 75 72 6c 29 29 2c 0a 09 09 09 09 09 09 ", url)),.......
1590: 7d 2e 70 61 72 73 65 5f 6d 6f 64 65 28 74 65 6c }.parse_mode(tel
15a0: 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 egram_bot::types
15b0: 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d ::ParseMode::Htm
15c0: 6c 29 29 2e 61 77 61 69 74 0a 09 09 09 09 09 09 l)).await.......
15d0: 2e 63 6f 6e 74 65 78 74 28 22 43 61 6e 27 74 20 .context("Can't
15e0: 70 6f 73 74 20 6d 65 73 73 61 67 65 3a 22 29 3f post message:")?
15f0: 3b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 ;......sqlx::que
1600: 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 ry("insert into
1610: 72 73 73 74 67 5f 70 6f 73 74 20 28 73 6f 75 72 rsstg_post (sour
1620: 63 65 5f 69 64 2c 20 70 6f 73 74 65 64 2c 20 75 ce_id, posted, u
1630: 72 6c 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 rl) values ($1,
1640: 24 32 2c 20 24 33 29 3b 22 29 0a 09 09 09 09 09 $2, $3);")......
1650: 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 ..bind(*id).....
1660: 09 09 2e 62 69 6e 64 28 64 61 74 65 29 0a 09 09 ...bind(date)...
1670: 09 09 09 09 2e 62 69 6e 64 28 75 72 6c 29 0a 09 .....bind(url)..
1680: 09 09 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d ......execute(&m
1690: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 ut conn).await..
16a0: 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 ......with_conte
16b0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 52 xt(|| format!("R
16c0: 65 63 6f 72 64 20 70 6f 73 74 3a 5c 6e 7b 3a 3f ecord post:\n{:?
16d0: 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 }", &conn))?;...
16e0: 09 09 09 64 72 6f 70 28 63 6f 6e 6e 29 3b 0a 09 ...drop(conn);..
16f0: 09 09 09 09 74 6f 6b 69 6f 3a 3a 74 69 6d 65 3a ....tokio::time:
1700: 3a 73 6c 65 65 70 28 73 74 64 3a 3a 74 69 6d 65 :sleep(std::time
1710: 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 ::Duration::new(
1720: 34 2c 20 30 29 29 2e 61 77 61 69 74 3b 0a 09 09 4, 0)).await;...
1730: 09 09 7d 3b 0a 09 09 09 09 70 6f 73 74 65 64 20 ..};.....posted
1740: 2b 3d 20 31 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 += 1;....};....p
1750: 6f 73 74 73 2e 63 6c 65 61 72 28 29 3b 0a 09 09 osts.clear();...
1760: 7d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e };...let mut con
1770: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
1780: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
1790: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
17a0: 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 74 | format!("Updat
17b0: 65 20 73 63 72 61 70 65 20 66 65 74 63 68 20 63 e scrape fetch c
17c0: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
17d0: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 73 71 lf.pool))?;...sq
17e0: 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 lx::query("updat
17f0: 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 e rsstg_source s
1800: 65 74 20 6c 61 73 74 5f 73 63 72 61 70 65 20 3d et last_scrape =
1810: 20 6e 6f 77 28 29 20 77 68 65 72 65 20 73 6f 75 now() where sou
1820: 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a 09 rce_id = $1;")..
1830: 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 ...bind(*id)....
1840: 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f .execute(&mut co
1850: 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 nn).await.....wi
1860: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
1870: 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 73 63 rmat!("Update sc
1880: 72 61 70 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 rape:\n{:?}", &c
1890: 6f 6e 6e 29 29 3f 3b 0a 09 09 4f 6b 28 66 6f 72 onn))?;...Ok(for
18a0: 6d 61 74 21 28 22 50 6f 73 74 65 64 3a 20 7b 7d mat!("Posted: {}
18b0: 22 2c 20 26 70 6f 73 74 65 64 29 29 0a 09 7d 0a ", &posted))..}.
18c0: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 64 ..pub async fn d
18d0: 65 6c 65 74 65 3c 53 3e 28 26 73 65 6c 66 2c 20 elete<S>(&self,
18e0: 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c source_id: &i32,
18f0: 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 owner: S) -> Re
1900: 73 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 sult<String>..wh
1910: 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e ere S: Into<i64>
1920: 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 {...let owner:
1930: 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f i64 = owner.into
1940: 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f ();...let mut co
1950: 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 nn = self.pool.a
1960: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 cquire().await..
1970: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1980: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 65 6c 65 || format!("Dele
1990: 74 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e te fetch conn:\n
19a0: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
19b0: 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 l))?;...match sq
19c0: 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 6c 65 74 lx::query("delet
19d0: 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 e from rsstg_sou
19e0: 72 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 rce where source
19f0: 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e _id = $1 and own
1a00: 65 72 20 3d 20 24 32 3b 22 29 0a 09 09 09 2e 62 er = $2;").....b
1a10: 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 ind(source_id)..
1a20: 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 ...bind(owner)..
1a30: 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 ...execute(&mut
1a40: 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e conn).await.....
1a50: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
1a60: 66 6f 72 6d 61 74 21 28 22 44 65 6c 65 74 65 20 format!("Delete
1a70: 73 6f 75 72 63 65 20 72 75 6c 65 3a 5c 6e 7b 3a source rule:\n{:
1a80: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
1a90: 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 )?.....rows_affe
1aa0: 63 74 65 64 28 29 20 7b 0a 09 09 09 30 20 3d 3e cted() {....0 =>
1ab0: 20 7b 20 4f 6b 28 22 4e 6f 20 64 61 74 61 20 66 { Ok("No data f
1ac0: 6f 75 6e 64 20 66 6f 75 6e 64 5c 5c 2e 22 2e 74 ound found\\.".t
1ad0: 6f 5f 73 74 72 69 6e 67 28 29 29 20 7d 2c 0a 09 o_string()) },..
1ae0: 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d ..x => { Ok(form
1af0: 61 74 21 28 22 7b 7d 20 73 6f 75 72 63 65 73 20 at!("{} sources
1b00: 72 65 6d 6f 76 65 64 5c 5c 2e 22 2c 20 78 29 29 removed\\.", x))
1b10: 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 },...}..}...pub
1b20: 20 61 73 79 6e 63 20 66 6e 20 63 6c 65 61 6e 3c async fn clean<
1b30: 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 S>(&self, source
1b40: 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 _id: &i32, owner
1b50: 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 : S) -> Result<S
1b60: 74 72 69 6e 67 3e 0a 09 77 68 65 72 65 20 53 3a tring>..where S:
1b70: 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c Into<i64> {...l
1b80: 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 et owner: i64 =
1b90: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 owner.into();...
1ba0: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
1bb0: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
1bc0: 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 ().await.....wit
1bd0: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
1be0: 6d 61 74 21 28 22 43 6c 65 61 6e 20 66 65 74 63 mat!("Clean fetc
1bf0: 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 h conn:\n{:?}",
1c00: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 &self.pool))?;..
1c10: 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 .match sqlx::que
1c20: 72 79 28 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 ry("delete from
1c30: 72 73 73 74 67 5f 70 6f 73 74 20 70 20 75 73 69 rsstg_post p usi
1c40: 6e 67 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 ng rsstg_source
1c50: 73 20 77 68 65 72 65 20 70 2e 73 6f 75 72 63 65 s where p.source
1c60: 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e _id = $1 and own
1c70: 65 72 20 3d 20 24 32 20 61 6e 64 20 70 2e 73 6f er = $2 and p.so
1c80: 75 72 63 65 5f 69 64 20 3d 20 73 2e 73 6f 75 72 urce_id = s.sour
1c90: 63 65 5f 69 64 3b 22 29 0a 09 09 09 2e 62 69 6e ce_id;").....bin
1ca0: 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 d(source_id)....
1cb0: 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 .bind(owner)....
1cc0: 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f .execute(&mut co
1cd0: 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 nn).await.....wi
1ce0: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
1cf0: 72 6d 61 74 21 28 22 43 6c 65 61 6e 20 73 65 65 rmat!("Clean see
1d00: 6e 20 70 6f 73 74 73 3a 5c 6e 7b 3a 3f 7d 22 2c n posts:\n{:?}",
1d10: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 &self.pool))?..
1d20: 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 ...rows_affected
1d30: 28 29 20 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f () {....0 => { O
1d40: 6b 28 22 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 k("No data found
1d50: 20 66 6f 75 6e 64 5c 5c 2e 22 2e 74 6f 5f 73 74 found\\.".to_st
1d60: 72 69 6e 67 28 29 29 20 7d 2c 0a 09 09 09 78 20 ring()) },....x
1d70: 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 => { Ok(format!(
1d80: 22 7b 7d 20 70 6f 73 74 73 20 70 75 72 67 65 64 "{} posts purged
1d90: 5c 5c 2e 22 2c 20 78 29 29 20 7d 2c 0a 09 09 7d \\.", x)) },...}
1da0: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
1db0: 66 6e 20 65 6e 61 62 6c 65 3c 53 3e 28 26 73 65 fn enable<S>(&se
1dc0: 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 lf, source_id: &
1dd0: 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d i32, owner: S) -
1de0: 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 > Result<&str>..
1df0: 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 where S: Into<i6
1e00: 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 4> {...let owner
1e10: 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 6e : i64 = owner.in
1e20: 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 to();...let mut
1e30: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
1e40: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
1e50: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
1e60: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 45 6e t(|| format!("En
1e70: 61 62 6c 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a able fetch conn:
1e80: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
1e90: 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 ool))?;...match
1ea0: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 sqlx::query("upd
1eb0: 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 ate rsstg_source
1ec0: 20 73 65 74 20 65 6e 61 62 6c 65 64 20 3d 20 74 set enabled = t
1ed0: 72 75 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 rue where source
1ee0: 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e _id = $1 and own
1ef0: 65 72 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 er = $2").....bi
1f00: 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 nd(source_id)...
1f10: 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 ..bind(owner)...
1f20: 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 ..execute(&mut c
1f30: 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 onn).await.....w
1f40: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
1f50: 6f 72 6d 61 74 21 28 22 45 6e 61 62 6c 65 20 73 ormat!("Enable s
1f60: 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 ource:\n{:?}", &
1f70: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 self.pool))?....
1f80: 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 .rows_affected()
1f90: 20 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 {....1 => { Ok(
1fa0: 22 53 6f 75 72 63 65 20 65 6e 61 62 6c 65 64 5c "Source enabled\
1fb0: 5c 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 \.") },....0 =>
1fc0: 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 { Ok("Source not
1fd0: 20 66 6f 75 6e 64 5c 5c 2e 22 29 20 7d 2c 0a 09 found\\.") },..
1fe0: 09 09 5f 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 .._ => { Err(any
1ff0: 68 6f 77 21 28 22 44 61 74 61 62 61 73 65 20 65 how!("Database e
2000: 72 72 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a rror.")) },...}.
2010: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
2020: 6e 20 64 69 73 61 62 6c 65 3c 53 3e 28 26 73 65 n disable<S>(&se
2030: 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 lf, source_id: &
2040: 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d i32, owner: S) -
2050: 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 > Result<&str>..
2060: 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 where S: Into<i6
2070: 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 4> {...let owner
2080: 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 6e : i64 = owner.in
2090: 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 to();...let mut
20a0: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
20b0: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
20c0: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
20d0: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 69 t(|| format!("Di
20e0: 73 61 62 6c 65 20 66 65 74 63 68 20 63 6f 6e 6e sable fetch conn
20f0: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
2100: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 pool))?;...match
2110: 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 sqlx::query("up
2120: 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 date rsstg_sourc
2130: 65 20 73 65 74 20 65 6e 61 62 6c 65 64 20 3d 20 e set enabled =
2140: 66 61 6c 73 65 20 77 68 65 72 65 20 73 6f 75 72 false where sour
2150: 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f ce_id = $1 and o
2160: 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 09 2e wner = $2").....
2170: 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a bind(source_id).
2180: 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a ....bind(owner).
2190: 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 ....execute(&mut
21a0: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 conn).await....
21b0: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
21c0: 20 66 6f 72 6d 61 74 21 28 22 44 69 73 61 62 6c format!("Disabl
21d0: 65 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 e source:\n{:?}"
21e0: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a , &self.pool))?.
21f0: 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 ....rows_affecte
2200: 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 d() {....1 => {
2210: 4f 6b 28 22 53 6f 75 72 63 65 20 64 69 73 61 62 Ok("Source disab
2220: 6c 65 64 5c 5c 2e 22 29 20 7d 2c 0a 09 09 09 30 led\\.") },....0
2230: 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 => { Ok("Source
2240: 20 6e 6f 74 20 66 6f 75 6e 64 5c 5c 2e 22 29 20 not found\\.")
2250: 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 45 72 72 },...._ => { Err
2260: 28 61 6e 79 68 6f 77 21 28 22 44 61 74 61 62 61 (anyhow!("Databa
2270: 73 65 20 65 72 72 6f 72 2e 22 29 29 20 7d 2c 0a se error.")) },.
2280: 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ..}..}...pub asy
2290: 6e 63 20 66 6e 20 75 70 64 61 74 65 3c 53 3e 28 nc fn update<S>(
22a0: 26 73 65 6c 66 2c 20 75 70 64 61 74 65 3a 20 4f &self, update: O
22b0: 70 74 69 6f 6e 3c 69 33 32 3e 2c 20 63 68 61 6e ption<i32>, chan
22c0: 6e 65 6c 3a 20 26 73 74 72 2c 20 63 68 61 6e 6e nel: &str, chann
22d0: 65 6c 5f 69 64 3a 20 69 36 34 2c 20 75 72 6c 3a el_id: i64, url:
22e0: 20 26 73 74 72 2c 20 69 76 5f 68 61 73 68 3a 20 &str, iv_hash:
22f0: 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 75 72 Option<&str>, ur
2300: 6c 5f 72 65 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 l_re: Option<&st
2310: 72 3e 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e r>, owner: S) ->
2320: 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e 0a Result<String>.
2330: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
2340: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
2350: 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 r: i64 = owner.i
2360: 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 nto();...let mut
2370: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
2380: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
2390: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
23a0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 xt(|| format!("U
23b0: 70 64 61 74 65 20 66 65 74 63 68 20 63 6f 6e 6e pdate fetch conn
23c0: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
23d0: 70 6f 6f 6c 29 29 3f 3b 0a 0a 09 09 6d 61 74 63 pool))?;....matc
23e0: 68 20 6d 61 74 63 68 20 75 70 64 61 74 65 20 7b h match update {
23f0: 0a 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e .....Some(id) =>
2400: 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 {......sqlx::qu
2410: 65 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 ery("update rsst
2420: 67 5f 73 6f 75 72 63 65 20 73 65 74 20 63 68 61 g_source set cha
2430: 6e 6e 65 6c 5f 69 64 20 3d 20 24 32 2c 20 75 72 nnel_id = $2, ur
2440: 6c 20 3d 20 24 33 2c 20 69 76 5f 68 61 73 68 20 l = $3, iv_hash
2450: 3d 20 24 34 2c 20 6f 77 6e 65 72 20 3d 20 24 35 = $4, owner = $5
2460: 2c 20 63 68 61 6e 6e 65 6c 20 3d 20 24 36 20 77 , channel = $6 w
2470: 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d here source_id =
2480: 20 24 31 22 29 2e 62 69 6e 64 28 69 64 29 0a 09 $1").bind(id)..
2490: 09 09 09 7d 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d ...},.....None =
24a0: 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 > {......sqlx::q
24b0: 75 65 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 uery("insert int
24c0: 6f 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 28 o rsstg_source (
24d0: 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c channel_id, url,
24e0: 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c iv_hash, owner,
24f0: 20 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 channel, url_re
2500: 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 ) values ($1, $2
2510: 2c 20 24 33 2c 20 24 34 2c 20 24 35 2c 20 24 36 , $3, $4, $5, $6
2520: 29 22 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a )").....},....}.
2530: 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c ....bind(channel
2540: 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 _id).....bind(ur
2550: 6c 29 0a 09 09 09 2e 62 69 6e 64 28 69 76 5f 68 l).....bind(iv_h
2560: 61 73 68 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 ash).....bind(ow
2570: 6e 65 72 29 0a 09 09 09 2e 62 69 6e 64 28 63 68 ner).....bind(ch
2580: 61 6e 6e 65 6c 29 0a 09 09 09 2e 62 69 6e 64 28 annel).....bind(
2590: 75 72 6c 5f 72 65 29 0a 09 09 09 2e 65 78 65 63 url_re).....exec
25a0: 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 ute(&mut conn).a
25b0: 77 61 69 74 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 wait {....Ok(_)
25c0: 3d 3e 20 72 65 74 75 72 6e 20 4f 6b 28 53 74 72 => return Ok(Str
25d0: 69 6e 67 3a 3a 66 72 6f 6d 28 6d 61 74 63 68 20 ing::from(match
25e0: 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d update {.....Som
25f0: 65 28 5f 29 20 3d 3e 20 22 43 68 61 6e 6e 65 6c e(_) => "Channel
2600: 20 75 70 64 61 74 65 64 5c 5c 2e 22 2c 0a 09 09 updated\\.",...
2610: 09 09 4e 6f 6e 65 20 3d 3e 20 22 43 68 61 6e 6e ..None => "Chann
2620: 65 6c 20 61 64 64 65 64 5c 5c 2e 22 2c 0a 09 09 el added\\.",...
2630: 09 7d 29 29 2c 0a 09 09 09 45 72 72 28 73 71 6c .})),....Err(sql
2640: 78 3a 3a 45 72 72 6f 72 3a 3a 44 61 74 61 62 61 x::Error::Databa
2650: 73 65 28 65 72 72 29 29 20 3d 3e 20 7b 0a 09 09 se(err)) => {...
2660: 09 09 6d 61 74 63 68 20 65 72 72 2e 64 6f 77 6e ..match err.down
2670: 63 61 73 74 3a 3a 3c 73 71 6c 78 3a 3a 70 6f 73 cast::<sqlx::pos
2680: 74 67 72 65 73 3a 3a 50 67 44 61 74 61 62 61 73 tgres::PgDatabas
2690: 65 45 72 72 6f 72 3e 28 29 2e 72 6f 75 74 69 6e eError>().routin
26a0: 65 28 29 20 7b 0a 09 09 09 09 09 53 6f 6d 65 28 e() {......Some(
26b0: 22 5f 62 74 5f 63 68 65 63 6b 5f 75 6e 69 71 75 "_bt_check_uniqu
26c0: 65 22 2c 20 29 20 3d 3e 20 7b 0a 09 09 09 09 09 e", ) => {......
26d0: 09 72 65 74 75 72 6e 20 4f 6b 28 22 44 75 70 6c .return Ok("Dupl
26e0: 69 63 61 74 65 20 6b 65 79 5c 5c 2e 22 2e 74 6f icate key\\.".to
26f0: 5f 73 74 72 69 6e 67 28 29 29 0a 09 09 09 09 09 _string())......
2700: 7d 2c 0a 09 09 09 09 09 53 6f 6d 65 28 5f 29 20 },......Some(_)
2710: 3d 3e 20 7b 0a 09 09 09 09 09 09 72 65 74 75 72 => {.......retur
2720: 6e 20 4f 6b 28 22 44 61 74 61 62 61 73 65 20 65 n Ok("Database e
2730: 72 72 6f 72 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 rror\\.".to_stri
2740: 6e 67 28 29 29 0a 09 09 09 09 09 7d 2c 0a 09 09 ng())......},...
2750: 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 ...None => {....
2760: 09 09 09 72 65 74 75 72 6e 20 4f 6b 28 22 4e 6f ...return Ok("No
2770: 20 64 61 74 61 62 61 73 65 20 65 72 72 6f 72 20 database error
2780: 65 78 74 72 61 63 74 65 64 5c 5c 2e 22 2e 74 6f extracted\\.".to
2790: 5f 73 74 72 69 6e 67 28 29 29 0a 09 09 09 09 09 _string())......
27a0: 7d 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 7d 2c 0a },.....};....},.
27b0: 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 7b ...Err(err) => {
27c0: 0a 09 09 09 09 62 61 69 6c 21 28 22 53 6f 72 72 .....bail!("Sorr
27d0: 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 72 72 6f 72 y, unknown error
27e0: 3a 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 65 72 72 :\n{:#?}\n", err
27f0: 29 3b 0a 09 09 09 7d 2c 0a 09 09 7d 3b 0a 09 7d );....},...};..}
2800: 0a 0a 09 61 73 79 6e 63 20 66 6e 20 61 75 74 6f ...async fn auto
2810: 66 65 74 63 68 28 26 73 65 6c 66 29 20 2d 3e 20 fetch(&self) ->
2820: 52 65 73 75 6c 74 3c 28 29 3e 20 7b 0a 09 09 6c Result<()> {...l
2830: 65 74 20 6d 75 74 20 64 65 6c 61 79 20 3d 20 63 et mut delay = c
2840: 68 72 6f 6e 6f 3a 3a 44 75 72 61 74 69 6f 6e 3a hrono::Duration:
2850: 3a 6d 69 6e 75 74 65 73 28 31 29 3b 0a 09 09 6c :minutes(1);...l
2860: 65 74 20 6d 75 74 20 6e 6f 77 3b 0a 09 09 6c 6f et mut now;...lo
2870: 6f 70 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 op {....let mut
2880: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
2890: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
28a0: 0a 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 ......with_conte
28b0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 41 xt(|| format!("A
28c0: 75 74 6f 66 65 74 63 68 20 66 65 74 63 68 20 63 utofetch fetch c
28d0: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
28e0: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 6e lf.pool))?;....n
28f0: 6f 77 20 3d 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 ow = chrono::Loc
2900: 61 6c 3a 3a 6e 6f 77 28 29 3b 0a 09 09 09 6c 65 al::now();....le
2910: 74 20 6d 75 74 20 71 75 65 75 65 20 3d 20 73 71 t mut queue = sq
2920: 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 lx::query("selec
2930: 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 6e 65 78 t source_id, nex
2940: 74 5f 66 65 74 63 68 2c 20 6f 77 6e 65 72 20 66 t_fetch, owner f
2950: 72 6f 6d 20 72 73 73 74 67 5f 6f 72 64 65 72 20 rom rsstg_order
2960: 6e 61 74 75 72 61 6c 20 6c 65 66 74 20 6a 6f 69 natural left joi
2970: 6e 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 n rsstg_source w
2980: 68 65 72 65 20 6e 65 78 74 5f 66 65 74 63 68 20 here next_fetch
2990: 3c 20 6e 6f 77 28 29 20 2b 20 69 6e 74 65 72 76 < now() + interv
29a0: 61 6c 20 27 31 20 6d 69 6e 75 74 65 27 3b 22 29 al '1 minute';")
29b0: 0a 09 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 ......fetch_all(
29c0: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
29d0: 3f 3b 0a 09 09 09 66 6f 72 20 72 6f 77 20 69 6e ?;....for row in
29e0: 20 71 75 65 75 65 2e 69 74 65 72 28 29 20 7b 0a queue.iter() {.
29f0: 09 09 09 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 ....let source_i
2a00: 64 3a 20 69 33 32 20 3d 20 72 6f 77 2e 74 72 79 d: i32 = row.try
2a10: 5f 67 65 74 28 22 73 6f 75 72 63 65 5f 69 64 22 _get("source_id"
2a20: 29 3f 3b 0a 09 09 09 09 6c 65 74 20 6f 77 6e 65 )?;.....let owne
2a30: 72 3a 20 69 36 34 20 3d 20 72 6f 77 2e 74 72 79 r: i64 = row.try
2a40: 5f 67 65 74 28 22 6f 77 6e 65 72 22 29 3f 3b 0a _get("owner")?;.
2a50: 09 09 09 09 6c 65 74 20 6e 65 78 74 5f 66 65 74 ....let next_fet
2a60: 63 68 3a 20 44 61 74 65 54 69 6d 65 3c 63 68 72 ch: DateTime<chr
2a70: 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3e 20 3d 20 72 6f ono::Local> = ro
2a80: 77 2e 74 72 79 5f 67 65 74 28 22 6e 65 78 74 5f w.try_get("next_
2a90: 66 65 74 63 68 22 29 3f 3b 0a 09 09 09 09 69 66 fetch")?;.....if
2aa0: 20 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f next_fetch < no
2ab0: 77 20 7b 0a 09 09 09 09 09 2f 2f 6c 65 74 20 63 w {......//let c
2ac0: 6c 6f 6e 65 20 3d 20 73 65 6c 66 2e 63 6c 6f 6e lone = self.clon
2ad0: 65 28 29 3b 0a 09 09 09 09 09 2f 2f 63 6c 6f 6e e();......//clon
2ae0: 65 2e 6f 77 6e 65 72 5f 63 68 61 74 28 55 73 65 e.owner_chat(Use
2af0: 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 29 rId::new(owner))
2b00: 3b 0a 09 09 09 09 09 6c 65 74 20 63 6c 6f 6e 65 ;......let clone
2b10: 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 09 09 09 = Core {.......
2b20: 6f 77 6e 65 72 5f 63 68 61 74 3a 20 74 65 6c 65 owner_chat: tele
2b30: 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 gram_bot::UserId
2b40: 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 2c 0a 09 09 ::new(owner),...
2b50: 09 09 09 09 2e 2e 73 65 6c 66 2e 63 6c 6f 6e 65 ......self.clone
2b60: 28 29 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ()......};......
2b70: 74 6f 6b 69 6f 3a 3a 73 70 61 77 6e 28 61 73 79 tokio::spawn(asy
2b80: 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 09 09 nc move {.......
2b90: 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 if let Err(err)
2ba0: 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b 28 26 73 = clone.check(&s
2bb0: 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 2c ource_id, owner,
2bc0: 20 74 72 75 65 29 2e 61 77 61 69 74 20 7b 0a 09 true).await {..
2bd0: 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 ......if let Err
2be0: 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 (err) = clone.se
2bf0: 6e 64 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b nd(&format!("
2c00: 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 4e {:?}", err), N
2c10: 6f 6e 65 2c 20 4e 6f 6e 65 29 20 7b 0a 09 09 09 one, None) {....
2c20: 09 09 09 09 09 65 70 72 69 6e 74 6c 6e 21 28 22 .....eprintln!("
2c30: 43 68 65 63 6b 20 65 72 72 6f 72 3a 20 7b 7d 22 Check error: {}"
2c40: 2c 20 65 72 72 29 3b 0a 09 09 09 09 09 09 09 7d , err);........}
2c50: 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ;.......};......
2c60: 7d 29 3b 0a 09 09 09 09 7d 20 65 6c 73 65 20 7b });.....} else {
2c70: 0a 09 09 09 09 09 69 66 20 6e 65 78 74 5f 66 65 ......if next_fe
2c80: 74 63 68 20 2d 20 6e 6f 77 20 3c 20 64 65 6c 61 tch - now < dela
2c90: 79 20 7b 0a 09 09 09 09 09 09 64 65 6c 61 79 20 y {.......delay
2ca0: 3d 20 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 6e = next_fetch - n
2cb0: 6f 77 3b 0a 09 09 09 09 09 7d 0a 09 09 09 09 7d ow;......}.....}
2cc0: 0a 09 09 09 7d 3b 0a 09 09 09 71 75 65 75 65 2e ....};....queue.
2cd0: 63 6c 65 61 72 28 29 3b 0a 09 09 09 74 6f 6b 69 clear();....toki
2ce0: 6f 3a 3a 74 69 6d 65 3a 3a 73 6c 65 65 70 28 64 o::time::sleep(d
2cf0: 65 6c 61 79 2e 74 6f 5f 73 74 64 28 29 3f 29 2e elay.to_std()?).
2d00: 61 77 61 69 74 3b 0a 09 09 09 64 65 6c 61 79 20 await;....delay
2d10: 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 72 61 74 69 = chrono::Durati
2d20: 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 31 29 3b 0a on::minutes(1);.
2d30: 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ..}..}...pub asy
2d40: 6e 63 20 66 6e 20 6c 69 73 74 3c 53 3e 28 26 73 nc fn list<S>(&s
2d50: 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d elf, owner: S) -
2d60: 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e > Result<String>
2d70: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
2d80: 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e i64> {...let own
2d90: 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 er = owner.into(
2da0: 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 72 65 70 );...let mut rep
2db0: 6c 79 20 3d 20 76 65 63 21 5b 5d 3b 0a 09 09 6c ly = vec![];...l
2dc0: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
2dd0: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
2de0: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
2df0: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
2e00: 61 74 21 28 22 4c 69 73 74 20 66 65 74 63 68 20 at!("List fetch
2e10: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
2e20: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 72 elf.pool))?;...r
2e30: 65 70 6c 79 2e 70 75 73 68 28 22 43 68 61 6e 6e eply.push("Chann
2e40: 65 6c 73 3a 22 2e 74 6f 5f 73 74 72 69 6e 67 28 els:".to_string(
2e50: 29 29 3b 0a 09 09 6c 65 74 20 72 6f 77 73 20 3d ));...let rows =
2e60: 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 sqlx::query("se
2e70: 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 lect source_id,
2e80: 63 68 61 6e 6e 65 6c 2c 20 65 6e 61 62 6c 65 64 channel, enabled
2e90: 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 20 66 , url, iv_hash f
2ea0: 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 rom rsstg_source
2eb0: 20 77 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 where owner = $
2ec0: 31 20 6f 72 64 65 72 20 62 79 20 73 6f 75 72 63 1 order by sourc
2ed0: 65 5f 69 64 22 29 0a 09 09 09 2e 62 69 6e 64 28 e_id").....bind(
2ee0: 6f 77 6e 65 72 29 0a 09 09 09 2e 66 65 74 63 68 owner).....fetch
2ef0: 5f 61 6c 6c 28 26 6d 75 74 20 63 6f 6e 6e 29 2e _all(&mut conn).
2f00: 61 77 61 69 74 3f 3b 0a 09 09 66 6f 72 20 72 6f await?;...for ro
2f10: 77 20 69 6e 20 72 6f 77 73 2e 69 74 65 72 28 29 w in rows.iter()
2f20: 20 7b 0a 09 09 09 6c 65 74 20 73 6f 75 72 63 65 {....let source
2f30: 5f 69 64 3a 20 69 33 32 20 3d 20 72 6f 77 2e 74 _id: i32 = row.t
2f40: 72 79 5f 67 65 74 28 22 73 6f 75 72 63 65 5f 69 ry_get("source_i
2f50: 64 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 73 65 d")?;....let use
2f60: 72 6e 61 6d 65 3a 20 26 73 74 72 20 3d 20 72 6f rname: &str = ro
2f70: 77 2e 74 72 79 5f 67 65 74 28 22 63 68 61 6e 6e w.try_get("chann
2f80: 65 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 65 6e el")?;....let en
2f90: 61 62 6c 65 64 3a 20 62 6f 6f 6c 20 3d 20 72 6f abled: bool = ro
2fa0: 77 2e 74 72 79 5f 67 65 74 28 22 65 6e 61 62 6c w.try_get("enabl
2fb0: 65 64 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 ed")?;....let ur
2fc0: 6c 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 l: &str = row.tr
2fd0: 79 5f 67 65 74 28 22 75 72 6c 22 29 3f 3b 0a 09 y_get("url")?;..
2fe0: 09 09 6c 65 74 20 69 76 5f 68 61 73 68 3a 20 4f ..let iv_hash: O
2ff0: 70 74 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 72 6f ption<&str> = ro
3000: 77 2e 74 72 79 5f 67 65 74 28 22 69 76 5f 68 61 w.try_get("iv_ha
3010: 73 68 22 29 3f 3b 0a 09 09 09 72 65 70 6c 79 2e sh")?;....reply.
3020: 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 5c 6e push(format!("\n
3030: 5c 5c 23 ef b8 8f e2 83 a3 20 7b 7d 20 5c 5c 2a \\#️⃣ {} \\*
3040: ef b8 8f e2 83 a3 20 60 7b 7d 60 20 7b 7d 5c 6e ️⃣ `{}` {}\n
3050: f0 9f 94 97 20 60 7b 7d 60 22 2c 20 73 6f 75 72 🔗 `{}`", sour
3060: 63 65 5f 69 64 2c 20 75 73 65 72 6e 61 6d 65 2c ce_id, username,
3070: 20 20 0a 09 09 09 09 6d 61 74 63 68 20 65 6e 61 .....match ena
3080: 62 6c 65 64 20 7b 0a 09 09 09 09 09 74 72 75 65 bled {......true
3090: 20 20 3d 3e 20 22 f0 9f 94 84 20 65 6e 61 62 6c => "🔄 enabl
30a0: 65 64 22 2c 0a 09 09 09 09 09 66 61 6c 73 65 20 ed",......false
30b0: 3d 3e 20 22 e2 9b 94 20 64 69 73 61 62 6c 65 64 => "⛔ disabled
30c0: 22 2c 0a 09 09 09 09 7d 2c 20 75 72 6c 29 29 3b ",.....}, url));
30d0: 0a 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 ....if let Some(
30e0: 68 61 73 68 29 20 3d 20 69 76 5f 68 61 73 68 20 hash) = iv_hash
30f0: 7b 0a 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 {.....reply.push
3100: 28 66 6f 72 6d 61 74 21 28 22 49 56 20 60 7b 7d (format!("IV `{}
3110: 60 22 2c 20 68 61 73 68 29 29 3b 0a 09 09 09 7d `", hash));....}
3120: 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 65 70 6c 79 ...};...Ok(reply
3130: 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 0a 09 7d 0a .join("\n"))..}.
3140: 7d 0a }.