0000: 75 73 65 20 73 74 64 3a 3a 63 6f 6c 6c 65 63 74 use std::collect
0010: 69 6f 6e 73 3a 3a 7b 42 54 72 65 65 4d 61 70 2c ions::{BTreeMap,
0020: 20 48 61 73 68 53 65 74 7d 3b 0a 75 73 65 20 73 HashSet};.use s
0030: 74 64 3a 3a 73 79 6e 63 3a 3a 7b 41 72 63 2c 20 td::sync::{Arc,
0040: 4d 75 74 65 78 7d 3b 0a 0a 75 73 65 20 63 6f 6e Mutex};..use con
0050: 66 69 67 3b 0a 0a 75 73 65 20 74 6f 6b 69 6f 3b fig;..use tokio;
0060: 0a 75 73 65 20 72 65 71 77 65 73 74 3b 0a 0a 75 .use reqwest;..u
0070: 73 65 20 72 73 73 3b 0a 75 73 65 20 61 74 6f 6d se rss;.use atom
0080: 5f 73 79 6e 64 69 63 61 74 69 6f 6e 3b 0a 0a 75 _syndication;..u
0090: 73 65 20 63 68 72 6f 6e 6f 3a 3a 44 61 74 65 54 se chrono::DateT
00a0: 69 6d 65 3b 0a 0a 75 73 65 20 72 65 67 65 78 3a ime;..use regex:
00b0: 3a 52 65 67 65 78 3b 0a 0a 75 73 65 20 74 65 6c :Regex;..use tel
00c0: 65 67 72 61 6d 5f 62 6f 74 3a 3a 2a 3b 0a 75 73 egram_bot::*;.us
00d0: 65 20 74 6f 6b 69 6f 3a 3a 73 74 72 65 61 6d 3a e tokio::stream:
00e0: 3a 53 74 72 65 61 6d 45 78 74 3b 0a 0a 75 73 65 :StreamExt;..use
00f0: 20 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a sqlx::postgres:
0100: 3a 50 67 50 6f 6f 6c 4f 70 74 69 6f 6e 73 3b 0a :PgPoolOptions;.
0110: 75 73 65 20 73 71 6c 78 3a 3a 52 6f 77 3b 0a 0a use sqlx::Row;..
0120: 23 5b 6d 61 63 72 6f 5f 75 73 65 5d 0a 65 78 74 #[macro_use].ext
0130: 65 72 6e 20 63 72 61 74 65 20 6c 61 7a 79 5f 73 ern crate lazy_s
0140: 74 61 74 69 63 3b 0a 0a 75 73 65 20 61 6e 79 68 tatic;..use anyh
0150: 6f 77 3a 3a 7b 61 6e 79 68 6f 77 2c 20 62 61 69 ow::{anyhow, bai
0160: 6c 2c 20 43 6f 6e 74 65 78 74 2c 20 52 65 73 75 l, Context, Resu
0170: 6c 74 7d 3b 0a 0a 23 5b 64 65 72 69 76 65 28 43 lt};..#[derive(C
0180: 6c 6f 6e 65 29 5d 0a 73 74 72 75 63 74 20 43 6f lone)].struct Co
0190: 72 65 20 7b 0a 09 6f 77 6e 65 72 3a 20 69 36 34 re {..owner: i64
01a0: 2c 0a 09 61 70 69 5f 6b 65 79 3a 20 53 74 72 69 ,..api_key: Stri
01b0: 6e 67 2c 0a 09 6f 77 6e 65 72 5f 63 68 61 74 3a ng,..owner_chat:
01c0: 20 55 73 65 72 49 64 2c 0a 09 74 67 3a 20 74 65 UserId,..tg: te
01d0: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 41 70 69 2c legram_bot::Api,
01e0: 0a 09 6d 79 3a 20 55 73 65 72 2c 0a 09 70 6f 6f ..my: User,..poo
01f0: 6c 3a 20 73 71 6c 78 3a 3a 50 6f 6f 6c 3c 73 71 l: sqlx::Pool<sq
0200: 6c 78 3a 3a 50 6f 73 74 67 72 65 73 3e 2c 0a 09 lx::Postgres>,..
0210: 73 6f 75 72 63 65 73 3a 20 41 72 63 3c 4d 75 74 sources: Arc<Mut
0220: 65 78 3c 48 61 73 68 53 65 74 3c 41 72 63 3c 69 ex<HashSet<Arc<i
0230: 33 32 3e 3e 3e 3e 2c 0a 7d 0a 0a 69 6d 70 6c 20 32>>>>,.}..impl
0240: 43 6f 72 65 20 7b 0a 09 61 73 79 6e 63 20 66 6e Core {..async fn
0250: 20 6e 65 77 28 73 65 74 74 69 6e 67 73 3a 20 63 new(settings: c
0260: 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 29 20 2d onfig::Config) -
0270: 3e 20 52 65 73 75 6c 74 3c 43 6f 72 65 3e 20 7b > Result<Core> {
0280: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 73 ...let owner = s
0290: 65 74 74 69 6e 67 73 2e 67 65 74 5f 69 6e 74 28 ettings.get_int(
02a0: 22 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 6c 65 74 "owner")?;...let
02b0: 20 61 70 69 5f 6b 65 79 20 3d 20 73 65 74 74 69 api_key = setti
02c0: 6e 67 73 2e 67 65 74 5f 73 74 72 28 22 61 70 69 ngs.get_str("api
02d0: 5f 6b 65 79 22 29 3f 3b 0a 09 09 6c 65 74 20 74 _key")?;...let t
02e0: 67 20 3d 20 41 70 69 3a 3a 6e 65 77 28 26 61 70 g = Api::new(&ap
02f0: 69 5f 6b 65 79 29 3b 0a 09 09 6c 65 74 20 63 6f i_key);...let co
0300: 72 65 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 6f re = Core {....o
0310: 77 6e 65 72 3a 20 6f 77 6e 65 72 2c 0a 09 09 09 wner: owner,....
0320: 61 70 69 5f 6b 65 79 3a 20 61 70 69 5f 6b 65 79 api_key: api_key
0330: 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 6d 79 3a .clone(),....my:
0340: 20 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 72 61 tg.send(telegra
0350: 6d 5f 62 6f 74 3a 3a 47 65 74 4d 65 29 2e 61 77 m_bot::GetMe).aw
0360: 61 69 74 3f 2c 0a 09 09 09 74 67 3a 20 74 67 2c ait?,....tg: tg,
0370: 0a 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 ....owner_chat:
0380: 55 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 UserId::new(owne
0390: 72 29 2c 0a 09 09 09 70 6f 6f 6c 3a 20 50 67 50 r),....pool: PgP
03a0: 6f 6f 6c 4f 70 74 69 6f 6e 73 3a 3a 6e 65 77 28 oolOptions::new(
03b0: 29 0a 09 09 09 09 2e 6d 61 78 5f 63 6f 6e 6e 65 )......max_conne
03c0: 63 74 69 6f 6e 73 28 35 29 0a 09 09 09 09 2e 63 ctions(5)......c
03d0: 6f 6e 6e 65 63 74 5f 74 69 6d 65 6f 75 74 28 73 onnect_timeout(s
03e0: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
03f0: 6f 6e 3a 3a 6e 65 77 28 33 30 30 2c 20 30 29 29 on::new(300, 0))
0400: 0a 09 09 09 09 2e 69 64 6c 65 5f 74 69 6d 65 6f ......idle_timeo
0410: 75 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 ut(std::time::Du
0420: 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 36 30 2c 20 ration::new(60,
0430: 30 29 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 74 0))......connect
0440: 5f 6c 61 7a 79 28 26 73 65 74 74 69 6e 67 73 2e _lazy(&settings.
0450: 67 65 74 5f 73 74 72 28 22 70 67 22 29 3f 29 3f get_str("pg")?)?
0460: 2c 0a 09 09 09 73 6f 75 72 63 65 73 3a 20 41 72 ,....sources: Ar
0470: 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a 3a 6e 65 c::new(Mutex::ne
0480: 77 28 48 61 73 68 53 65 74 3a 3a 6e 65 77 28 29 w(HashSet::new()
0490: 29 29 2c 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 63 )),...};...let c
04a0: 6c 6f 6e 65 20 3d 20 63 6f 72 65 2e 63 6c 6f 6e lone = core.clon
04b0: 65 28 29 3b 0a 09 09 74 6f 6b 69 6f 3a 3a 73 70 e();...tokio::sp
04c0: 61 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b awn(async move {
04d0: 0a 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 ....if let Err(e
04e0: 72 72 29 20 3d 20 26 63 6c 6f 6e 65 2e 61 75 74 rr) = &clone.aut
04f0: 6f 66 65 74 63 68 28 29 2e 61 77 61 69 74 20 7b ofetch().await {
0500: 0a 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 .....if let Err(
0510: 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 64 65 62 err) = clone.deb
0520: 75 67 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b ug(&format!("š
0530: 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 4e {:?}", err), N
0540: 6f 6e 65 29 20 7b 0a 09 09 09 09 09 65 70 72 69 one) {......epri
0550: 6e 74 6c 6e 21 28 22 41 75 74 6f 66 65 74 63 68 ntln!("Autofetch
0560: 20 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 error: {}", err
0570: 29 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 7d 0a 09 );.....};....}..
0580: 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 0a .});...Ok(core).
0590: 09 7d 0a 0a 09 66 6e 20 73 74 72 65 61 6d 28 26 .}...fn stream(&
05a0: 73 65 6c 66 29 20 2d 3e 20 74 65 6c 65 67 72 61 self) -> telegra
05b0: 6d 5f 62 6f 74 3a 3a 55 70 64 61 74 65 73 53 74 m_bot::UpdatesSt
05c0: 72 65 61 6d 20 7b 0a 09 09 73 65 6c 66 2e 74 67 ream {...self.tg
05d0: 2e 73 74 72 65 61 6d 28 29 0a 09 7d 0a 0a 09 66 .stream()..}...f
05e0: 6e 20 64 65 62 75 67 28 26 73 65 6c 66 2c 20 6d n debug(&self, m
05f0: 73 67 3a 20 26 73 74 72 2c 20 74 61 72 67 65 74 sg: &str, target
0600: 3a 20 4f 70 74 69 6f 6e 3c 55 73 65 72 49 64 3e : Option<UserId>
0610: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 20 ) -> Result<()>
0620: 7b 0a 09 09 73 65 6c 66 2e 74 67 2e 73 70 61 77 {...self.tg.spaw
0630: 6e 28 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e n(SendMessage::n
0640: 65 77 28 6d 61 74 63 68 20 74 61 72 67 65 74 20 ew(match target
0650: 7b 0a 09 09 09 53 6f 6d 65 28 75 73 65 72 29 20 {....Some(user)
0660: 3d 3e 20 75 73 65 72 2c 0a 09 09 09 4e 6f 6e 65 => user,....None
0670: 20 3d 3e 20 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 => self.owner_c
0680: 68 61 74 2c 0a 09 09 7d 2c 20 6d 73 67 29 29 3b hat,...}, msg));
0690: 0a 09 09 4f 6b 28 28 29 29 0a 09 7d 0a 0a 09 61 ...Ok(())..}...a
06a0: 73 79 6e 63 20 66 6e 20 63 68 65 63 6b 3c 53 3e sync fn check<S>
06b0: 28 26 73 65 6c 66 2c 20 69 64 3a 20 69 33 32 2c (&self, id: i32,
06c0: 20 6f 77 6e 65 72 3a 20 53 2c 20 72 65 61 6c 3a owner: S, real:
06d0: 20 62 6f 6f 6c 29 20 2d 3e 20 52 65 73 75 6c 74 bool) -> Result
06e0: 3c 28 29 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 <()>..where S: I
06f0: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
0700: 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 owner: i64 = ow
0710: 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 ner.into();...le
0720: 74 20 69 64 20 3d 20 7b 0a 09 09 09 6c 65 74 20 t id = {....let
0730: 6d 75 74 20 73 65 74 20 3d 20 73 65 6c 66 2e 73 mut set = self.s
0740: 6f 75 72 63 65 73 2e 6c 6f 63 6b 28 29 2e 75 6e ources.lock().un
0750: 77 72 61 70 28 29 3b 0a 09 09 09 6d 61 74 63 68 wrap();....match
0760: 20 73 65 74 2e 67 65 74 28 26 69 64 29 20 7b 0a set.get(&id) {.
0770: 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 ....Some(id) =>
0780: 69 64 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 09 id.clone(),.....
0790: 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 6c None => {......l
07a0: 65 74 20 69 64 20 3d 20 41 72 63 3a 3a 6e 65 77 et id = Arc::new
07b0: 28 69 64 29 3b 0a 09 09 09 09 09 73 65 74 2e 69 (id);......set.i
07c0: 6e 73 65 72 74 28 69 64 2e 63 6c 6f 6e 65 28 29 nsert(id.clone()
07d0: 29 3b 0a 09 09 09 09 09 69 64 2e 63 6c 6f 6e 65 );......id.clone
07e0: 28 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 ().....},....}..
07f0: 09 7d 3b 0a 09 09 6c 65 74 20 63 6f 75 6e 74 20 .};...let count
0800: 3d 20 41 72 63 3a 3a 73 74 72 6f 6e 67 5f 63 6f = Arc::strong_co
0810: 75 6e 74 28 26 69 64 29 3b 0a 09 09 69 66 20 63 unt(&id);...if c
0820: 6f 75 6e 74 20 3d 3d 20 32 20 7b 0a 09 09 09 6c ount == 2 {....l
0830: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
0840: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
0850: 29 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 74 ).await......wit
0860: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
0870: 6d 61 74 21 28 22 51 75 65 72 79 20 71 75 65 75 mat!("Query queu
0880: 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b e fetch conn:\n{
0890: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
08a0: 29 29 3f 3b 0a 09 09 09 6c 65 74 20 72 6f 77 20 ))?;....let row
08b0: 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 = sqlx::query("s
08c0: 65 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 64 2c elect source_id,
08d0: 20 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c channel_id, url
08e0: 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 , iv_hash, owner
08f0: 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 from rsstg_sour
0900: 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f ce where source_
0910: 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 id = $1 and owne
0920: 72 20 3d 20 24 32 22 29 0a 09 09 09 09 2e 62 69 r = $2")......bi
0930: 6e 64 28 2a 69 64 29 0a 09 09 09 09 2e 62 69 6e nd(*id)......bin
0940: 64 28 6f 77 6e 65 72 29 0a 09 09 09 09 2e 66 65 d(owner)......fe
0950: 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 63 6f 6e tch_one(&mut con
0960: 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 n).await......wi
0970: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
0980: 72 6d 61 74 21 28 22 51 75 65 72 79 20 73 6f 75 rmat!("Query sou
0990: 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 rce:\n{:?}", &se
09a0: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 64 lf.pool))?;....d
09b0: 72 6f 70 28 63 6f 6e 6e 29 3b 0a 09 09 09 6c 65 rop(conn);....le
09c0: 74 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 t channel_id: i6
09d0: 34 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 4 = row.try_get(
09e0: 22 63 68 61 6e 6e 65 6c 5f 69 64 22 29 3f 3b 0a "channel_id")?;.
09f0: 09 09 09 6c 65 74 20 64 65 73 74 69 6e 61 74 69 ...let destinati
0a00: 6f 6e 20 3d 20 6d 61 74 63 68 20 72 65 61 6c 20 on = match real
0a10: 7b 0a 09 09 09 09 74 72 75 65 20 3d 3e 20 55 73 {.....true => Us
0a20: 65 72 49 64 3a 3a 6e 65 77 28 63 68 61 6e 6e 65 erId::new(channe
0a30: 6c 5f 69 64 29 2c 0a 09 09 09 09 66 61 6c 73 65 l_id),.....false
0a40: 20 3d 3e 20 55 73 65 72 49 64 3a 3a 6e 65 77 28 => UserId::new(
0a50: 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e row.try_get("own
0a60: 65 72 22 29 3f 29 2c 0a 09 09 09 7d 3b 0a 09 09 er")?),....};...
0a70: 09 6c 65 74 20 75 72 6c 3a 20 26 73 74 72 20 3d .let url: &str =
0a80: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 72 row.try_get("ur
0a90: 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 6d 75 74 l")?;....let mut
0aa0: 20 74 68 69 73 5f 66 65 74 63 68 3a 20 4f 70 74 this_fetch: Opt
0ab0: 69 6f 6e 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 ion<DateTime<chr
0ac0: 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 ono::FixedOffset
0ad0: 3e 3e 20 3d 20 4e 6f 6e 65 3b 0a 09 09 09 6c 65 >> = None;....le
0ae0: 74 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f t iv_hash: Optio
0af0: 6e 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 n<&str> = row.tr
0b00: 79 5f 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 y_get("iv_hash")
0b10: 3f 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 70 6f ?;....let mut po
0b20: 73 74 73 3a 20 42 54 72 65 65 4d 61 70 3c 44 61 sts: BTreeMap<Da
0b30: 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 teTime<chrono::F
0b40: 69 78 65 64 4f 66 66 73 65 74 3e 2c 20 53 74 72 ixedOffset>, Str
0b50: 69 6e 67 3e 20 3d 20 42 54 72 65 65 4d 61 70 3a ing> = BTreeMap:
0b60: 3a 6e 65 77 28 29 3b 0a 09 09 09 6c 65 74 20 63 :new();....let c
0b70: 6f 6e 74 65 6e 74 20 3d 20 72 65 71 77 65 73 74 ontent = reqwest
0b80: 3a 3a 67 65 74 28 75 72 6c 29 2e 61 77 61 69 74 ::get(url).await
0b90: 3f 2e 62 79 74 65 73 28 29 2e 61 77 61 69 74 3f ?.bytes().await?
0ba0: 3b 0a 09 09 09 2f 2f 6c 65 74 20 6d 75 74 20 63 ;....//let mut c
0bb0: 6f 6e 74 65 6e 74 5f 20 3d 20 73 75 72 66 3a 3a ontent_ = surf::
0bc0: 67 65 74 28 75 72 6c 29 2e 61 77 61 69 74 2e 6d get(url).await.m
0bd0: 61 70 5f 65 72 72 28 7c 65 72 72 7c 20 61 6e 79 ap_err(|err| any
0be0: 68 6f 77 21 28 65 72 72 29 29 3f 3b 0a 09 09 09 how!(err))?;....
0bf0: 2f 2f 65 70 72 69 6e 74 6c 6e 21 28 22 44 61 74 //eprintln!("Dat
0c00: 61 3a 20 7b 3a 23 3f 7d 22 2c 20 26 63 6f 6e 74 a: {:#?}", &cont
0c10: 65 6e 74 5f 29 3b 0a 09 09 09 2f 2f 6c 65 74 20 ent_);....//let
0c20: 63 6f 6e 74 65 6e 74 20 3d 20 63 6f 6e 74 65 6e content = conten
0c30: 74 5f 2e 62 6f 64 79 5f 62 79 74 65 73 28 29 2e t_.body_bytes().
0c40: 61 77 61 69 74 2e 6d 61 70 5f 65 72 72 28 7c 65 await.map_err(|e
0c50: 72 72 7c 20 61 6e 79 68 6f 77 21 28 65 72 72 29 rr| anyhow!(err)
0c60: 29 3f 3b 0a 09 09 09 2f 2a 0a 09 09 09 6c 65 74 )?;..../*....let
0c70: 20 66 65 65 64 20 3d 20 72 73 73 3a 3a 43 68 61 feed = rss::Cha
0c80: 6e 6e 65 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 nnel::read_from(
0c90: 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 0a 09 09 &content[..])...
0ca0: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
0cb0: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 50 72 6f 62 || format!("Prob
0cc0: 6c 65 6d 20 6f 70 65 6e 69 6e 67 20 66 65 65 64 lem opening feed
0cd0: 20 75 72 6c 3a 5c 6e 7b 7d 22 2c 20 26 75 72 6c url:\n{}", &url
0ce0: 29 29 3f 3b 0a 09 09 09 66 6f 72 20 69 74 65 6d ))?;....for item
0cf0: 20 69 6e 20 66 65 65 64 2e 69 74 65 6d 73 28 29 in feed.items()
0d00: 20 7b 0a 09 09 09 09 6c 65 74 20 64 61 74 65 20 {.....let date
0d10: 3d 20 6d 61 74 63 68 20 69 74 65 6d 2e 70 75 62 = match item.pub
0d20: 5f 64 61 74 65 28 29 20 7b 0a 09 09 09 09 09 53 _date() {......S
0d30: 6f 6d 65 28 66 65 65 64 5f 64 61 74 65 29 20 3d ome(feed_date) =
0d40: 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 > DateTime::pars
0d50: 65 5f 66 72 6f 6d 5f 72 66 63 32 38 32 32 28 66 e_from_rfc2822(f
0d60: 65 65 64 5f 64 61 74 65 29 2c 0a 09 09 09 09 09 eed_date),......
0d70: 4e 6f 6e 65 20 3d 3e 20 44 61 74 65 54 69 6d 65 None => DateTime
0d80: 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 ::parse_from_rfc
0d90: 33 33 33 39 28 26 69 74 65 6d 2e 64 75 62 6c 69 3339(&item.dubli
0da0: 6e 5f 63 6f 72 65 5f 65 78 74 28 29 2e 75 6e 77 n_core_ext().unw
0db0: 72 61 70 28 29 2e 64 61 74 65 73 28 29 5b 30 5d rap().dates()[0]
0dc0: 29 2c 0a 09 09 09 09 7d 3f 3b 0a 09 09 09 09 6c ),.....}?;.....l
0dd0: 65 74 20 75 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 et url = item.li
0de0: 6e 6b 28 29 2e 75 6e 77 72 61 70 28 29 2e 74 6f nk().unwrap().to
0df0: 5f 73 74 72 69 6e 67 28 29 3b 0a 09 09 09 09 70 _string();.....p
0e00: 6f 73 74 73 2e 69 6e 73 65 72 74 28 64 61 74 65 osts.insert(date
0e10: 2e 63 6c 6f 6e 65 28 29 2c 20 75 72 6c 2e 63 6c .clone(), url.cl
0e20: 6f 6e 65 28 29 29 3b 0a 09 09 09 7d 3b 0a 09 09 one());....};...
0e30: 09 2a 2f 0a 09 09 09 6d 61 74 63 68 20 72 73 73 .*/....match rss
0e40: 3a 3a 43 68 61 6e 6e 65 6c 3a 3a 72 65 61 64 5f ::Channel::read_
0e50: 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e from(&content[..
0e60: 5d 29 20 7b 0a 09 09 09 09 4f 6b 28 66 65 65 64 ]) {.....Ok(feed
0e70: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 66 6f 72 20 ) => {......for
0e80: 69 74 65 6d 20 69 6e 20 66 65 65 64 2e 69 74 65 item in feed.ite
0e90: 6d 73 28 29 20 7b 0a 09 09 09 09 09 09 6c 65 74 ms() {.......let
0ea0: 20 64 61 74 65 20 3d 20 6d 61 74 63 68 20 69 74 date = match it
0eb0: 65 6d 2e 70 75 62 5f 64 61 74 65 28 29 20 7b 0a em.pub_date() {.
0ec0: 09 09 09 09 09 09 09 53 6f 6d 65 28 66 65 65 64 .......Some(feed
0ed0: 5f 64 61 74 65 29 20 3d 3e 20 44 61 74 65 54 69 _date) => DateTi
0ee0: 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 me::parse_from_r
0ef0: 66 63 32 38 32 32 28 66 65 65 64 5f 64 61 74 65 fc2822(feed_date
0f00: 29 2c 0a 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d ),........None =
0f10: 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 > DateTime::pars
0f20: 65 5f 66 72 6f 6d 5f 72 66 63 33 33 33 39 28 26 e_from_rfc3339(&
0f30: 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 63 6f 72 65 item.dublin_core
0f40: 5f 65 78 74 28 29 2e 75 6e 77 72 61 70 28 29 2e _ext().unwrap().
0f50: 64 61 74 65 73 28 29 5b 30 5d 29 2c 0a 09 09 09 dates()[0]),....
0f60: 09 09 09 7d 3f 3b 0a 09 09 09 09 09 09 6c 65 74 ...}?;.......let
0f70: 20 75 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b url = item.link
0f80: 28 29 2e 75 6e 77 72 61 70 28 29 2e 74 6f 5f 73 ().unwrap().to_s
0f90: 74 72 69 6e 67 28 29 3b 0a 09 09 09 09 09 09 70 tring();.......p
0fa0: 6f 73 74 73 2e 69 6e 73 65 72 74 28 64 61 74 65 osts.insert(date
0fb0: 2e 63 6c 6f 6e 65 28 29 2c 20 75 72 6c 2e 63 6c .clone(), url.cl
0fc0: 6f 6e 65 28 29 29 3b 0a 09 09 09 09 09 7d 3b 0a one());......};.
0fd0: 09 09 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 65 ....},.....Err(e
0fe0: 72 72 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 72 rr) => match err
0ff0: 20 7b 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 {......rss::Err
1000: 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 74 or::InvalidStart
1010: 54 61 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 6c Tag => {.......l
1020: 65 74 20 66 65 65 64 20 3d 20 61 74 6f 6d 5f 73 et feed = atom_s
1030: 79 6e 64 69 63 61 74 69 6f 6e 3a 3a 46 65 65 64 yndication::Feed
1040: 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f 6e ::read_from(&con
1050: 74 65 6e 74 5b 2e 2e 5d 29 0a 09 09 09 09 09 09 tent[..]).......
1060: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
1070: 7c 20 66 6f 72 6d 61 74 21 28 22 50 72 6f 62 6c | format!("Probl
1080: 65 6d 20 6f 70 65 6e 69 6e 67 20 66 65 65 64 20 em opening feed
1090: 75 72 6c 3a 5c 6e 7b 7d 22 2c 20 26 75 72 6c 29 url:\n{}", &url)
10a0: 29 3f 3b 0a 09 09 09 09 09 09 66 6f 72 20 69 74 )?;.......for it
10b0: 65 6d 20 69 6e 20 66 65 65 64 2e 65 6e 74 72 69 em in feed.entri
10c0: 65 73 28 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 es() {........le
10d0: 74 20 64 61 74 65 20 3d 20 69 74 65 6d 2e 70 75 t date = item.pu
10e0: 62 6c 69 73 68 65 64 28 29 2e 75 6e 77 72 61 70 blished().unwrap
10f0: 28 29 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 75 ();........let u
1100: 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 rl = item.links(
1110: 29 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 09 )[0].href();....
1120: 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 ....posts.insert
1130: 28 64 61 74 65 2e 63 6c 6f 6e 65 28 29 2c 20 75 (date.clone(), u
1140: 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b rl.to_string());
1150: 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 7d .......};......}
1160: 2c 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 6f ,......rss::Erro
1170: 72 3a 3a 45 6f 66 20 3d 3e 20 28 29 2c 0a 09 09 r::Eof => (),...
1180: 09 09 09 5f 20 3d 3e 20 62 61 69 6c 21 28 22 55 ..._ => bail!("U
1190: 6e 73 75 70 70 6f 72 74 65 64 20 6f 72 20 6d 61 nsupported or ma
11a0: 6e 67 6c 65 64 20 63 6f 6e 74 65 6e 74 3a 5c 6e ngled content:\n
11b0: 7b 3a 23 3f 7d 5c 6e 22 2c 20 65 72 72 29 0a 09 {:#?}\n", err)..
11c0: 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 66 6f ...}....};....fo
11d0: 72 20 28 64 61 74 65 2c 20 75 72 6c 29 20 69 6e r (date, url) in
11e0: 20 70 6f 73 74 73 2e 69 74 65 72 28 29 20 7b 0a posts.iter() {.
11f0: 09 09 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ....let mut conn
1200: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
1210: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
1220: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
1230: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 68 65 63 || format!("Chec
1240: 6b 20 70 6f 73 74 20 66 65 74 63 68 20 63 6f 6e k post fetch con
1250: 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 n:\n{:?}", &self
1260: 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 09 6c 65 .pool))?;.....le
1270: 74 20 72 6f 77 20 3d 20 73 71 6c 78 3a 3a 71 75 t row = sqlx::qu
1280: 65 72 79 28 22 73 65 6c 65 63 74 20 65 78 69 73 ery("select exis
1290: 74 73 28 73 65 6c 65 63 74 20 74 72 75 65 20 66 ts(select true f
12a0: 72 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 77 rom rsstg_post w
12b0: 68 65 72 65 20 75 72 6c 20 3d 20 24 31 20 61 6e here url = $1 an
12c0: 64 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 32 d source_id = $2
12d0: 29 20 61 73 20 65 78 69 73 74 73 3b 22 29 0a 09 ) as exists;")..
12e0: 09 09 09 09 2e 62 69 6e 64 28 26 75 72 6c 29 0a .....bind(&url).
12f0: 09 09 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a ......bind(*id).
1300: 09 09 09 09 09 2e 66 65 74 63 68 5f 6f 6e 65 28 ......fetch_one(
1310: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
1320: 0a 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 .......with_cont
1330: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1340: 43 68 65 63 6b 20 70 6f 73 74 3a 5c 6e 7b 3a 3f Check post:\n{:?
1350: 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 }", &conn))?;...
1360: 09 09 6c 65 74 20 65 78 69 73 74 73 3a 20 62 6f ..let exists: bo
1370: 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 ol = row.try_get
1380: 28 22 65 78 69 73 74 73 22 29 3f 3b 0a 09 09 09 ("exists")?;....
1390: 09 69 66 20 21 20 65 78 69 73 74 73 20 7b 0a 09 .if ! exists {..
13a0: 09 09 09 09 69 66 20 74 68 69 73 5f 66 65 74 63 ....if this_fetc
13b0: 68 20 3d 3d 20 4e 6f 6e 65 20 7c 7c 20 2a 64 61 h == None || *da
13c0: 74 65 20 3e 20 74 68 69 73 5f 66 65 74 63 68 2e te > this_fetch.
13d0: 75 6e 77 72 61 70 28 29 20 7b 0a 09 09 09 09 09 unwrap() {......
13e0: 09 74 68 69 73 5f 66 65 74 63 68 20 3d 20 53 6f .this_fetch = So
13f0: 6d 65 28 2a 64 61 74 65 29 3b 0a 09 09 09 09 09 me(*date);......
1400: 7d 3b 0a 09 09 09 09 09 73 65 6c 66 2e 74 67 2e };......self.tg.
1410: 73 65 6e 64 28 20 6d 61 74 63 68 20 69 76 5f 68 send( match iv_h
1420: 61 73 68 20 7b 0a 09 09 09 09 09 09 09 53 6f 6d ash {........Som
1430: 65 28 78 29 20 3d 3e 20 53 65 6e 64 4d 65 73 73 e(x) => SendMess
1440: 61 67 65 3a 3a 6e 65 77 28 64 65 73 74 69 6e 61 age::new(destina
1450: 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 21 28 22 3c tion, format!("<
1460: 61 20 68 72 65 66 3d 5c 22 68 74 74 70 73 3a 2f a href=\"https:/
1470: 2f 74 2e 6d 65 2f 69 76 3f 75 72 6c 3d 7b 7d 26 /t.me/iv?url={}&
1480: 72 68 61 73 68 3d 7b 7d 5c 22 3e 20 3c 2f 61 3e rhash={}\"> </a>
1490: 7b 30 7d 22 2c 20 75 72 6c 2c 20 78 29 29 2c 0a {0}", url, x)),.
14a0: 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 53 .......None => 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 7b 7d 22 2c 20 75 72 6c 29 29 mat!("{}", url))
14e0: 2c 0a 09 09 09 09 09 09 7d 2e 70 61 72 73 65 5f ,.......}.parse_
14f0: 6d 6f 64 65 28 74 79 70 65 73 3a 3a 50 61 72 73 mode(types::Pars
1500: 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 2e 61 77 eMode::Html)).aw
1510: 61 69 74 0a 09 09 09 09 09 09 2e 63 6f 6e 74 65 ait........conte
1520: 78 74 28 22 43 61 6e 27 74 20 70 6f 73 74 20 6d xt("Can't post m
1530: 65 73 73 61 67 65 3a 22 29 3f 3b 0a 09 09 09 09 essage:")?;.....
1540: 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 69 6e .sqlx::query("in
1550: 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 67 5f sert into rsstg_
1560: 70 6f 73 74 20 28 73 6f 75 72 63 65 5f 69 64 2c post (source_id,
1570: 20 70 6f 73 74 65 64 2c 20 75 72 6c 29 20 76 61 posted, url) va
1580: 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 24 33 lues ($1, $2, $3
1590: 29 3b 22 29 0a 09 09 09 09 09 09 2e 62 69 6e 64 );")........bind
15a0: 28 2a 69 64 29 0a 09 09 09 09 09 09 2e 62 69 6e (*id)........bin
15b0: 64 28 64 61 74 65 29 0a 09 09 09 09 09 09 2e 62 d(date)........b
15c0: 69 6e 64 28 75 72 6c 29 0a 09 09 09 09 09 09 2e ind(url)........
15d0: 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e execute(&mut con
15e0: 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 09 09 2e n).await........
15f0: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
1600: 66 6f 72 6d 61 74 21 28 22 52 65 63 6f 72 64 20 format!("Record
1610: 70 6f 73 74 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 post:\n{:?}", &c
1620: 6f 6e 6e 29 29 3f 3b 0a 09 09 09 09 09 64 72 6f onn))?;......dro
1630: 70 28 63 6f 6e 6e 29 3b 0a 09 09 09 09 09 74 6f p(conn);......to
1640: 6b 69 6f 3a 3a 74 69 6d 65 3a 3a 64 65 6c 61 79 kio::time::delay
1650: 5f 66 6f 72 28 73 74 64 3a 3a 74 69 6d 65 3a 3a _for(std::time::
1660: 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 34 2c Duration::new(4,
1670: 20 30 29 29 2e 61 77 61 69 74 3b 0a 09 09 09 09 0)).await;.....
1680: 7d 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 6f 73 74 };....};....post
1690: 73 2e 63 6c 65 61 72 28 29 3b 0a 09 09 7d 3b 0a s.clear();...};.
16a0: 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d ..let mut conn =
16b0: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
16c0: 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 re().await.....w
16d0: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
16e0: 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 73 ormat!("Update s
16f0: 63 72 61 70 65 20 66 65 74 63 68 20 63 6f 6e 6e crape fetch conn
1700: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1710: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 73 71 6c 78 3a pool))?;...sqlx:
1720: 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 :query("update r
1730: 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 sstg_source set
1740: 6c 61 73 74 5f 73 63 72 61 70 65 20 3d 20 6e 6f last_scrape = no
1750: 77 28 29 20 77 68 65 72 65 20 73 6f 75 72 63 65 w() where source
1760: 5f 69 64 20 3d 20 24 31 3b 22 29 0a 09 09 09 2e _id = $1;").....
1770: 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 2e 65 78 bind(*id).....ex
1780: 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 ecute(&mut conn)
1790: 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f .await.....with_
17a0: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
17b0: 74 21 28 22 55 70 64 61 74 65 20 73 63 72 61 70 t!("Update scrap
17c0: 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e e:\n{:?}", &conn
17d0: 29 29 3f 3b 0a 09 09 4f 6b 28 28 29 29 0a 09 7d ))?;...Ok(())..}
17e0: 0a 0a 09 61 73 79 6e 63 20 66 6e 20 64 65 6c 65 ...async fn dele
17f0: 74 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 te<S>(&self, sou
1800: 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 rce_id: &i32, ow
1810: 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c ner: S) -> Resul
1820: 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 72 65 t<String>..where
1830: 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a S: Into<i64> {.
1840: 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 ..let owner: i64
1850: 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b = owner.into();
1860: 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
1870: 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 = self.pool.acqu
1880: 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e ire().await.....
1890: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
18a0: 66 6f 72 6d 61 74 21 28 22 44 65 6c 65 74 65 20 format!("Delete
18b0: 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f fetch conn:\n{:?
18c0: 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 }", &self.pool))
18d0: 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a ?;...match sqlx:
18e0: 3a 71 75 65 72 79 28 22 64 65 6c 65 74 65 20 66 :query("delete f
18f0: 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 rom rsstg_source
1900: 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 where source_id
1910: 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 = $1 and owner
1920: 3d 20 24 32 3b 22 29 0a 09 09 09 2e 62 69 6e 64 = $2;").....bind
1930: 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e (source_id).....
1940: 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e bind(owner).....
1950: 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e execute(&mut con
1960: 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 n).await.....wit
1970: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
1980: 6d 61 74 21 28 22 44 65 6c 65 74 65 20 73 6f 75 mat!("Delete sou
1990: 72 63 65 20 72 75 6c 65 3a 5c 6e 7b 3a 3f 7d 22 rce rule:\n{:?}"
19a0: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a , &self.pool))?.
19b0: 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 ....rows_affecte
19c0: 64 28 29 20 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 d() {....0 => {
19d0: 4f 6b 28 22 4e 6f 20 64 61 74 61 20 66 6f 75 6e Ok("No data foun
19e0: 64 20 66 6f 75 6e 64 5c 5c 2e 22 2e 74 6f 5f 73 d found\\.".to_s
19f0: 74 72 69 6e 67 28 29 29 20 7d 2c 0a 09 09 09 78 tring()) },....x
1a00: 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 => { Ok(format!
1a10: 28 22 7b 7d 20 73 6f 75 72 63 65 73 20 72 65 6d ("{} sources rem
1a20: 6f 76 65 64 5c 5c 2e 22 2c 20 78 29 29 20 7d 2c oved\\.", x)) },
1a30: 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 ...}..}...async
1a40: 66 6e 20 63 6c 65 61 6e 3c 53 3e 28 26 73 65 6c fn clean<S>(&sel
1a50: 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 f, source_id: &i
1a60: 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 32, owner: S) ->
1a70: 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e 0a Result<String>.
1a80: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
1a90: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
1aa0: 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 r: i64 = owner.i
1ab0: 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 nto();...let mut
1ac0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
1ad0: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
1ae0: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
1af0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 xt(|| format!("C
1b00: 6c 65 61 6e 20 66 65 74 63 68 20 63 6f 6e 6e 3a lean fetch conn:
1b10: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
1b20: 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 ool))?;...match
1b30: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 6c sqlx::query("del
1b40: 65 74 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 ete from rsstg_p
1b50: 6f 73 74 20 70 20 75 73 69 6e 67 20 72 73 73 74 ost p using rsst
1b60: 67 5f 73 6f 75 72 63 65 20 73 20 77 68 65 72 65 g_source s where
1b70: 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 p.source_id = $
1b80: 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 1 and owner = $2
1b90: 20 61 6e 64 20 70 2e 73 6f 75 72 63 65 5f 69 64 and p.source_id
1ba0: 20 3d 20 73 2e 73 6f 75 72 63 65 5f 69 64 3b 22 = s.source_id;"
1bb0: 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 ).....bind(sourc
1bc0: 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f e_id).....bind(o
1bd0: 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 wner).....execut
1be0: 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 e(&mut conn).awa
1bf0: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
1c00: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1c10: 43 6c 65 61 6e 20 73 65 65 6e 20 70 6f 73 74 73 Clean seen posts
1c20: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1c30: 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 pool))?.....rows
1c40: 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 _affected() {...
1c50: 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 .0 => { Ok("No d
1c60: 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 5c ata found found\
1c70: 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 \.".to_string())
1c80: 20 7d 2c 0a 09 09 09 78 20 3d 3e 20 7b 20 4f 6b },....x => { Ok
1c90: 28 66 6f 72 6d 61 74 21 28 22 7b 7d 20 70 6f 73 (format!("{} pos
1ca0: 74 73 20 70 75 72 67 65 64 5c 5c 2e 22 2c 20 78 ts purged\\.", x
1cb0: 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 )) },...}..}...a
1cc0: 73 79 6e 63 20 66 6e 20 65 6e 61 62 6c 65 3c 53 sync fn enable<S
1cd0: 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f >(&self, source_
1ce0: 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a id: &i32, owner:
1cf0: 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 S) -> Result<&s
1d00: 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e tr>..where S: In
1d10: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 to<i64> {...let
1d20: 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 6e owner: i64 = own
1d30: 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 er.into();...let
1d40: 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 mut conn = self
1d50: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
1d60: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
1d70: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1d80: 21 28 22 45 6e 61 62 6c 65 20 66 65 74 63 68 20 !("Enable fetch
1d90: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
1da0: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d elf.pool))?;...m
1db0: 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 atch sqlx::query
1dc0: 28 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 ("update rsstg_s
1dd0: 6f 75 72 63 65 20 73 65 74 20 65 6e 61 62 6c 65 ource set enable
1de0: 64 20 3d 20 74 72 75 65 20 77 68 65 72 65 20 73 d = true where s
1df0: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e ource_id = $1 an
1e00: 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 d owner = $2")..
1e10: 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 ...bind(source_i
1e20: 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 d).....bind(owne
1e30: 72 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 r).....execute(&
1e40: 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a mut conn).await.
1e50: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
1e60: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 45 6e 61 (|| format!("Ena
1e70: 62 6c 65 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f ble source:\n{:?
1e80: 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 }", &self.pool))
1e90: 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 ?.....rows_affec
1ea0: 74 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 ted() {....1 =>
1eb0: 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 65 6e 61 { Ok("Source ena
1ec0: 62 6c 65 64 5c 5c 2e 22 29 20 7d 2c 0a 09 09 09 bled\\.") },....
1ed0: 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 0 => { Ok("Sourc
1ee0: 65 20 6e 6f 74 20 66 6f 75 6e 64 5c 5c 2e 22 29 e not found\\.")
1ef0: 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 45 72 },...._ => { Er
1f00: 72 28 61 6e 79 68 6f 77 21 28 22 44 61 74 61 62 r(anyhow!("Datab
1f10: 61 73 65 20 65 72 72 6f 72 2e 22 29 29 20 7d 2c ase error.")) },
1f20: 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 ...}..}...async
1f30: 66 6e 20 64 69 73 61 62 6c 65 3c 53 3e 28 26 73 fn disable<S>(&s
1f40: 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 elf, source_id:
1f50: 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 &i32, owner: S)
1f60: 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a -> Result<&str>.
1f70: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
1f80: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
1f90: 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 r: i64 = owner.i
1fa0: 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 nto();...let mut
1fb0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
1fc0: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
1fd0: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
1fe0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 xt(|| format!("D
1ff0: 69 73 61 62 6c 65 20 66 65 74 63 68 20 63 6f 6e isable fetch con
2000: 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 n:\n{:?}", &self
2010: 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 .pool))?;...matc
2020: 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 h sqlx::query("u
2030: 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 pdate rsstg_sour
2040: 63 65 20 73 65 74 20 65 6e 61 62 6c 65 64 20 3d ce set enabled =
2050: 20 66 61 6c 73 65 20 77 68 65 72 65 20 73 6f 75 false where sou
2060: 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 rce_id = $1 and
2070: 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 09 owner = $2")....
2080: 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 .bind(source_id)
2090: 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 .....bind(owner)
20a0: 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 .....execute(&mu
20b0: 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 t conn).await...
20c0: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
20d0: 7c 20 66 6f 72 6d 61 74 21 28 22 44 69 73 61 62 | format!("Disab
20e0: 6c 65 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d le source:\n{:?}
20f0: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
2100: 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 .....rows_affect
2110: 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b ed() {....1 => {
2120: 20 4f 6b 28 22 53 6f 75 72 63 65 20 64 69 73 61 Ok("Source disa
2130: 62 6c 65 64 5c 5c 2e 22 29 20 7d 2c 0a 09 09 09 bled\\.") },....
2140: 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 0 => { Ok("Sourc
2150: 65 20 6e 6f 74 20 66 6f 75 6e 64 5c 5c 2e 22 29 e not found\\.")
2160: 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 45 72 },...._ => { Er
2170: 72 28 61 6e 79 68 6f 77 21 28 22 44 61 74 61 62 r(anyhow!("Datab
2180: 61 73 65 20 65 72 72 6f 72 2e 22 29 29 20 7d 2c ase error.")) },
2190: 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 ...}..}...async
21a0: 66 6e 20 75 70 64 61 74 65 3c 53 3e 28 26 73 65 fn update<S>(&se
21b0: 6c 66 2c 20 75 70 64 61 74 65 3a 20 4f 70 74 69 lf, update: Opti
21c0: 6f 6e 3c 69 33 32 3e 2c 20 63 68 61 6e 6e 65 6c on<i32>, channel
21d0: 3a 20 26 73 74 72 2c 20 63 68 61 6e 6e 65 6c 5f : &str, channel_
21e0: 69 64 3a 20 69 36 34 2c 20 75 72 6c 3a 20 26 73 id: i64, url: &s
21f0: 74 72 2c 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 tr, iv_hash: Opt
2200: 69 6f 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 72 ion<&str>, owner
2210: 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 : S) -> Result<S
2220: 74 72 69 6e 67 3e 0a 09 77 68 65 72 65 20 53 3a tring>..where S:
2230: 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c Into<i64> {...l
2240: 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 et owner: i64 =
2250: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 owner.into();...
2260: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
2270: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
2280: 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 ().await.....wit
2290: 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 h_context(|| for
22a0: 6d 61 74 21 28 22 55 70 64 61 74 65 20 66 65 74 mat!("Update fet
22b0: 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c ch conn:\n{:?}",
22c0: 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a &self.pool))?;.
22d0: 0a 09 09 6d 61 74 63 68 20 6d 61 74 63 68 20 75 ...match match u
22e0: 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 pdate {.....Some
22f0: 28 69 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 73 (id) => {......s
2300: 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 qlx::query("upda
2310: 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 te rsstg_source
2320: 73 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 20 3d set channel_id =
2330: 20 24 32 2c 20 75 72 6c 20 3d 20 24 33 2c 20 69 $2, url = $3, i
2340: 76 5f 68 61 73 68 20 3d 20 24 34 2c 20 6f 77 6e v_hash = $4, own
2350: 65 72 20 3d 20 24 35 2c 20 63 68 61 6e 6e 65 6c er = $5, channel
2360: 20 3d 20 24 36 20 77 68 65 72 65 20 73 6f 75 72 = $6 where sour
2370: 63 65 5f 69 64 20 3d 20 24 31 22 29 2e 62 69 6e ce_id = $1").bin
2380: 64 28 69 64 29 0a 09 09 09 09 7d 2c 0a 09 09 09 d(id).....},....
2390: 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 .None => {......
23a0: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 69 6e 73 sqlx::query("ins
23b0: 65 72 74 20 69 6e 74 6f 20 72 73 73 74 67 5f 73 ert into rsstg_s
23c0: 6f 75 72 63 65 20 28 63 68 61 6e 6e 65 6c 5f 69 ource (channel_i
23d0: 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c d, url, iv_hash,
23e0: 20 6f 77 6e 65 72 2c 20 63 68 61 6e 6e 65 6c 29 owner, channel)
23f0: 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 2c values ($1, $2,
2400: 20 24 33 2c 20 24 34 2c 20 24 35 29 22 29 0a 09 $3, $4, $5)")..
2410: 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 09 2e 62 ...},....}.....b
2420: 69 6e 64 28 63 68 61 6e 6e 65 6c 5f 69 64 29 0a ind(channel_id).
2430: 09 09 09 2e 62 69 6e 64 28 75 72 6c 29 0a 09 09 ....bind(url)...
2440: 09 2e 62 69 6e 64 28 69 76 5f 68 61 73 68 29 0a ..bind(iv_hash).
2450: 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a ....bind(owner).
2460: 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c ....bind(channel
2470: 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d ).....execute(&m
2480: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 20 7b ut conn).await {
2490: 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 72 65 74 ....Ok(_) => ret
24a0: 75 72 6e 20 4f 6b 28 53 74 72 69 6e 67 3a 3a 66 urn Ok(String::f
24b0: 72 6f 6d 28 6d 61 74 63 68 20 75 70 64 61 74 65 rom(match update
24c0: 20 7b 0a 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d {.....Some(_) =
24d0: 3e 20 22 43 68 61 6e 6e 65 6c 20 75 70 64 61 74 > "Channel updat
24e0: 65 64 5c 5c 2e 22 2c 0a 09 09 09 09 4e 6f 6e 65 ed\\.",.....None
24f0: 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 61 64 64 => "Channel add
2500: 65 64 5c 5c 2e 22 2c 0a 09 09 09 7d 29 29 2c 0a ed\\.",....})),.
2510: 09 09 09 45 72 72 28 73 71 6c 78 3a 3a 45 72 72 ...Err(sqlx::Err
2520: 6f 72 3a 3a 44 61 74 61 62 61 73 65 28 65 72 72 or::Database(err
2530: 29 29 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 74 63 )) => {.....matc
2540: 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a 3a h err.downcast::
2550: 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a <sqlx::postgres:
2560: 3a 50 67 44 61 74 61 62 61 73 65 45 72 72 6f 72 :PgDatabaseError
2570: 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b 0a >().routine() {.
2580: 09 09 09 09 09 53 6f 6d 65 28 22 5f 62 74 5f 63 .....Some("_bt_c
2590: 68 65 63 6b 5f 75 6e 69 71 75 65 22 2c 20 29 20 heck_unique", )
25a0: 3d 3e 20 7b 0a 09 09 09 09 09 09 72 65 74 75 72 => {.......retur
25b0: 6e 20 4f 6b 28 22 44 75 70 6c 69 63 61 74 65 20 n Ok("Duplicate
25c0: 6b 65 79 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e key\\.".to_strin
25d0: 67 28 29 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 g())......},....
25e0: 09 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 7b 0a 09 ..Some(_) => {..
25f0: 09 09 09 09 09 72 65 74 75 72 6e 20 4f 6b 28 22 .....return Ok("
2600: 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 5c 5c Database error\\
2610: 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 0a .".to_string()).
2620: 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 4e 6f 6e .....},......Non
2630: 65 20 3d 3e 20 7b 0a 09 09 09 09 09 09 72 65 74 e => {.......ret
2640: 75 72 6e 20 4f 6b 28 22 4e 6f 20 64 61 74 61 62 urn Ok("No datab
2650: 61 73 65 20 65 72 72 6f 72 20 65 78 74 72 61 63 ase error extrac
2660: 74 65 64 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e ted\\.".to_strin
2670: 67 28 29 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 g())......},....
2680: 09 7d 3b 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 .};....},....Err
2690: 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 62 (err) => {.....b
26a0: 61 69 6c 21 28 22 53 6f 72 72 79 2c 20 75 6e 6b ail!("Sorry, unk
26b0: 6e 6f 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 nown error:\n{:#
26c0: 3f 7d 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 09 09 ?}\n", err);....
26d0: 7d 2c 0a 09 09 7d 3b 0a 09 7d 0a 0a 09 61 73 79 },...};..}...asy
26e0: 6e 63 20 66 6e 20 61 75 74 6f 66 65 74 63 68 28 nc fn autofetch(
26f0: 26 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 &self) -> Result
2700: 3c 28 29 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 <()> {...let mut
2710: 20 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a delay = chrono:
2720: 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 :Duration::minut
2730: 65 73 28 31 29 3b 0a 09 09 6c 65 74 20 6d 75 74 es(1);...let mut
2740: 20 6e 6f 77 3b 0a 09 09 6c 6f 6f 70 20 7b 0a 09 now;...loop {..
2750: 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d ..let mut conn =
2760: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
2770: 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 09 2e re().await......
2780: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
2790: 66 6f 72 6d 61 74 21 28 22 41 75 74 6f 66 65 74 format!("Autofet
27a0: 63 68 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e ch fetch conn:\n
27b0: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
27c0: 6c 29 29 3f 3b 0a 09 09 09 6e 6f 77 20 3d 20 63 l))?;....now = c
27d0: 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f hrono::Local::no
27e0: 77 28 29 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 w();....let mut
27f0: 71 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a 71 75 queue = sqlx::qu
2800: 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 ery("select sour
2810: 63 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 63 ce_id, next_fetc
2820: 68 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 h, owner from rs
2830: 73 74 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 stg_order natura
2840: 6c 20 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 l left join rsst
2850: 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 6e g_source where n
2860: 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 ext_fetch < now(
2870: 29 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 20 ) + interval '1
2880: 6d 69 6e 75 74 65 27 3b 22 29 0a 09 09 09 09 2e minute';")......
2890: 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 fetch_all(&mut c
28a0: 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 onn).await?;....
28b0: 66 6f 72 20 72 6f 77 20 69 6e 20 71 75 65 75 65 for row in queue
28c0: 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 09 6c 65 .iter() {.....le
28d0: 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 t source_id: i32
28e0: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
28f0: 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 source_id")?;...
2900: 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 ..let owner: i64
2910: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
2920: 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 09 09 6c 65 owner")?;.....le
2930: 74 20 6e 65 78 74 5f 66 65 74 63 68 3a 20 44 61 t next_fetch: Da
2940: 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 4c teTime<chrono::L
2950: 6f 63 61 6c 3e 20 3d 20 72 6f 77 2e 74 72 79 5f ocal> = row.try_
2960: 67 65 74 28 22 6e 65 78 74 5f 66 65 74 63 68 22 get("next_fetch"
2970: 29 3f 3b 0a 09 09 09 09 69 66 20 6e 65 78 74 5f )?;.....if next_
2980: 66 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a 09 09 fetch < now {...
2990: 09 09 09 2f 2f 6c 65 74 20 63 6c 6f 6e 65 20 3d ...//let clone =
29a0: 20 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 3b 0a 09 self.clone();..
29b0: 09 09 09 09 2f 2f 63 6c 6f 6e 65 2e 6f 77 6e 65 ....//clone.owne
29c0: 72 5f 63 68 61 74 28 55 73 65 72 49 64 3a 3a 6e r_chat(UserId::n
29d0: 65 77 28 6f 77 6e 65 72 29 29 3b 0a 09 09 09 09 ew(owner));.....
29e0: 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 .let clone = Cor
29f0: 65 20 7b 0a 09 09 09 09 09 09 6f 77 6e 65 72 5f e {.......owner_
2a00: 63 68 61 74 3a 20 55 73 65 72 49 64 3a 3a 6e 65 chat: UserId::ne
2a10: 77 28 6f 77 6e 65 72 29 2c 0a 09 09 09 09 09 09 w(owner),.......
2a20: 2e 2e 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 0a 09 ..self.clone()..
2a30: 09 09 09 09 7d 3b 0a 09 09 09 09 09 74 6f 6b 69 ....};......toki
2a40: 6f 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 6d o::spawn(async m
2a50: 6f 76 65 20 7b 0a 09 09 09 09 09 09 69 66 20 6c ove {.......if l
2a60: 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c et Err(err) = cl
2a70: 6f 6e 65 2e 63 68 65 63 6b 28 73 6f 75 72 63 65 one.check(source
2a80: 5f 69 64 2c 20 6f 77 6e 65 72 2c 20 74 72 75 65 _id, owner, true
2a90: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 ).await {.......
2aa0: 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 .if let Err(err)
2ab0: 20 3d 20 63 6c 6f 6e 65 2e 64 65 62 75 67 28 26 = clone.debug(&
2ac0: 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a format!("š {:
2ad0: 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 29 ?}", err), None)
2ae0: 20 7b 0a 09 09 09 09 09 09 09 09 65 70 72 69 6e {.........eprin
2af0: 74 6c 6e 21 28 22 43 68 65 63 6b 20 65 72 72 6f tln!("Check erro
2b00: 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 r: {}", err);...
2b10: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 3b .....};.......};
2b20: 0a 09 09 09 09 09 7d 29 3b 0a 09 09 09 09 7d 20 ......});.....}
2b30: 65 6c 73 65 20 7b 0a 09 09 09 09 09 69 66 20 6e else {......if n
2b40: 65 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 20 ext_fetch - now
2b50: 3c 20 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 09 < delay {.......
2b60: 64 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 delay = next_fet
2b70: 63 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 09 7d ch - now;......}
2b80: 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 .....}....};....
2b90: 71 75 65 75 65 2e 63 6c 65 61 72 28 29 3b 0a 09 queue.clear();..
2ba0: 09 09 74 6f 6b 69 6f 3a 3a 74 69 6d 65 3a 3a 64 ..tokio::time::d
2bb0: 65 6c 61 79 5f 66 6f 72 28 64 65 6c 61 79 2e 74 elay_for(delay.t
2bc0: 6f 5f 73 74 64 28 29 3f 29 2e 61 77 61 69 74 3b o_std()?).await;
2bd0: 0a 09 09 09 64 65 6c 61 79 20 3d 20 63 68 72 6f ....delay = chro
2be0: 6e 6f 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 no::Duration::mi
2bf0: 6e 75 74 65 73 28 31 29 3b 0a 09 09 7d 0a 09 7d nutes(1);...}..}
2c00: 0a 0a 09 61 73 79 6e 63 20 66 6e 20 6c 69 73 74 ...async fn list
2c10: 3c 53 3e 28 26 73 65 6c 66 2c 20 6f 77 6e 65 72 <S>(&self, owner
2c20: 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 56 : S) -> Result<V
2c30: 65 63 3c 53 74 72 69 6e 67 3e 3e 0a 09 77 68 65 ec<String>>..whe
2c40: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
2c50: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
2c60: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 owner.into();...
2c70: 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 20 3d 20 let mut reply =
2c80: 76 65 63 21 5b 5d 3b 0a 09 09 6c 65 74 20 6d 75 vec![];...let mu
2c90: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
2ca0: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
2cb0: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
2cc0: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
2cd0: 4c 69 73 74 20 66 65 74 63 68 20 63 6f 6e 6e 3a List fetch conn:
2ce0: 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 \n{:?}", &self.p
2cf0: 6f 6f 6c 29 29 3f 3b 0a 09 09 72 65 70 6c 79 2e ool))?;...reply.
2d00: 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 push("Channels:"
2d10: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 .to_string());..
2d20: 09 6c 65 74 20 72 6f 77 73 20 3d 20 73 71 6c 78 .let rows = sqlx
2d30: 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 ::query("select
2d40: 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e source_id, chann
2d50: 65 6c 2c 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c el, enabled, url
2d60: 2c 20 69 76 5f 68 61 73 68 20 66 72 6f 6d 20 72 , iv_hash from r
2d70: 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 sstg_source wher
2d80: 65 20 6f 77 6e 65 72 20 3d 20 24 31 20 6f 72 64 e owner = $1 ord
2d90: 65 72 20 62 79 20 73 6f 75 72 63 65 5f 69 64 22 er by source_id"
2da0: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
2db0: 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 ).....fetch_all(
2dc0: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
2dd0: 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 ?;...for row in
2de0: 72 6f 77 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 rows.iter() {...
2df0: 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 .let source_id:
2e00: 69 33 32 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 i32 = row.try_ge
2e10: 74 28 22 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b t("source_id")?;
2e20: 0a 09 09 09 6c 65 74 20 75 73 65 72 6e 61 6d 65 ....let username
2e30: 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 : &str = row.try
2e40: 5f 67 65 74 28 22 63 68 61 6e 6e 65 6c 22 29 3f _get("channel")?
2e50: 3b 0a 09 09 09 6c 65 74 20 65 6e 61 62 6c 65 64 ;....let enabled
2e60: 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 : bool = row.try
2e70: 5f 67 65 74 28 22 65 6e 61 62 6c 65 64 22 29 3f _get("enabled")?
2e80: 3b 0a 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 73 ;....let url: &s
2e90: 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 tr = row.try_get
2ea0: 28 22 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 ("url")?;....let
2eb0: 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e iv_hash: Option
2ec0: 3c 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 <&str> = row.try
2ed0: 5f 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f _get("iv_hash")?
2ee0: 3b 0a 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 ;....reply.push(
2ef0: 66 6f 72 6d 61 74 21 28 22 5c 6e 5c 5c 23 ef b8 format!("\n\\#ļø
2f00: 8f e2 83 a3 20 7b 7d 20 5c 5c 2a ef b8 8f e2 83 ā£ {} \\*ļøā
2f10: a3 20 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 97 20 £ `{}` {}\nš
2f20: 60 7b 7d 60 22 2c 20 73 6f 75 72 63 65 5f 69 64 `{}`", source_id
2f30: 2c 20 75 73 65 72 6e 61 6d 65 2c 20 20 0a 09 09 , username, ...
2f40: 09 09 6d 61 74 63 68 20 65 6e 61 62 6c 65 64 20 ..match enabled
2f50: 7b 0a 09 09 09 09 09 74 72 75 65 20 20 3d 3e 20 {......true =>
2f60: 22 f0 9f 94 84 20 65 6e 61 62 6c 65 64 22 2c 0a "š enabled",.
2f70: 09 09 09 09 09 66 61 6c 73 65 20 3d 3e 20 22 e2 .....false => "ā
2f80: 9b 94 20 64 69 73 61 62 6c 65 64 22 2c 0a 09 09 disabled",...
2f90: 09 09 7d 2c 20 75 72 6c 29 29 3b 0a 09 09 09 69 ..}, url));....i
2fa0: 66 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 68 29 f let Some(hash)
2fb0: 20 3d 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 = iv_hash {....
2fc0: 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d .reply.push(form
2fd0: 61 74 21 28 22 49 56 20 60 7b 7d 60 22 2c 20 68 at!("IV `{}`", h
2fe0: 61 73 68 29 29 3b 0a 09 09 09 7d 0a 09 09 7d 3b ash));....}...};
2ff0: 0a 09 09 4f 6b 28 72 65 70 6c 79 29 0a 09 7d 0a ...Ok(reply)..}.
3000: 7d 0a 0a 23 5b 74 6f 6b 69 6f 3a 3a 6d 61 69 6e }..#[tokio::main
3010: 5d 0a 61 73 79 6e 63 20 66 6e 20 6d 61 69 6e 28 ].async fn main(
3020: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 20 ) -> Result<()>
3030: 7b 0a 09 6c 65 74 20 6d 75 74 20 73 65 74 74 69 {..let mut setti
3040: 6e 67 73 20 3d 20 63 6f 6e 66 69 67 3a 3a 43 6f ngs = config::Co
3050: 6e 66 69 67 3a 3a 64 65 66 61 75 6c 74 28 29 3b nfig::default();
3060: 0a 09 73 65 74 74 69 6e 67 73 2e 6d 65 72 67 65 ..settings.merge
3070: 28 63 6f 6e 66 69 67 3a 3a 46 69 6c 65 3a 3a 77 (config::File::w
3080: 69 74 68 5f 6e 61 6d 65 28 22 72 73 73 74 67 22 ith_name("rsstg"
3090: 29 29 3f 3b 0a 0a 09 6c 65 74 20 63 6f 72 65 20 ))?;...let core
30a0: 3d 20 43 6f 72 65 3a 3a 6e 65 77 28 73 65 74 74 = Core::new(sett
30b0: 69 6e 67 73 29 2e 61 77 61 69 74 3f 3b 0a 0a 09 ings).await?;...
30c0: 6c 65 74 20 6d 75 74 20 73 74 72 65 61 6d 20 3d let mut stream =
30d0: 20 63 6f 72 65 2e 73 74 72 65 61 6d 28 29 3b 0a core.stream();.
30e0: 09 73 74 72 65 61 6d 2e 61 6c 6c 6f 77 65 64 5f .stream.allowed_
30f0: 75 70 64 61 74 65 73 28 26 5b 41 6c 6c 6f 77 65 updates(&[Allowe
3100: 64 55 70 64 61 74 65 3a 3a 4d 65 73 73 61 67 65 dUpdate::Message
3110: 5d 29 3b 0a 09 6c 65 74 20 6d 75 74 20 72 65 70 ]);..let mut rep
3120: 6c 79 5f 74 6f 3a 20 4f 70 74 69 6f 6e 3c 55 73 ly_to: Option<Us
3130: 65 72 49 64 3e 3b 0a 0a 09 6c 6f 6f 70 20 7b 0a erId>;...loop {.
3140: 09 09 72 65 70 6c 79 5f 74 6f 20 3d 20 4e 6f 6e ..reply_to = Non
3150: 65 3b 0a 09 09 6d 61 74 63 68 20 73 74 72 65 61 e;...match strea
3160: 6d 2e 6e 65 78 74 28 29 2e 61 77 61 69 74 20 7b m.next().await {
3170: 0a 09 09 09 53 6f 6d 65 28 75 70 64 61 74 65 29 ....Some(update)
3180: 20 3d 3e 20 7b 0a 09 09 09 09 69 66 20 6c 65 74 => {.....if let
3190: 20 45 72 72 28 65 72 72 29 20 3d 20 68 61 6e 64 Err(err) = hand
31a0: 6c 65 28 75 70 64 61 74 65 3f 2c 20 26 63 6f 72 le(update?, &cor
31b0: 65 2c 20 26 6d 75 74 20 72 65 70 6c 79 5f 74 6f e, &mut reply_to
31c0: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 63 ).await {......c
31d0: 6f 72 65 2e 64 65 62 75 67 28 26 66 6f 72 6d 61 ore.debug(&forma
31e0: 74 21 28 22 f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 t!("š {:?}",
31f0: 65 72 72 29 2c 20 72 65 70 6c 79 5f 74 6f 29 3f err), reply_to)?
3200: 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 7d 2c 0a 09 ;.....};....},..
3210: 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 ..None => {.....
3220: 63 6f 72 65 2e 64 65 62 75 67 28 26 66 6f 72 6d core.debug(&form
3230: 61 74 21 28 22 f0 9f 9b 91 20 4e 6f 6e 65 20 65 at!("š None e
3240: 72 72 6f 72 2e 22 29 2c 20 4e 6f 6e 65 29 3f 3b rror."), None)?;
3250: 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 7d 0a 0a 09 ....}...};..}...
3260: 2f 2f 4f 6b 28 28 29 29 0a 7d 0a 0a 61 73 79 6e //Ok(()).}..asyn
3270: 63 20 66 6e 20 68 61 6e 64 6c 65 28 75 70 64 61 c fn handle(upda
3280: 74 65 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 te: telegram_bot
3290: 3a 3a 55 70 64 61 74 65 2c 20 63 6f 72 65 3a 20 ::Update, core:
32a0: 26 43 6f 72 65 2c 20 6d 75 74 20 5f 72 65 70 6c &Core, mut _repl
32b0: 79 5f 74 6f 3a 20 26 4f 70 74 69 6f 6e 3c 55 73 y_to: &Option<Us
32c0: 65 72 49 64 3e 29 20 2d 3e 20 52 65 73 75 6c 74 erId>) -> Result
32d0: 3c 28 29 3e 20 7b 0a 09 6c 61 7a 79 5f 73 74 61 <()> {..lazy_sta
32e0: 74 69 63 21 20 7b 0a 09 09 73 74 61 74 69 63 20 tic! {...static
32f0: 72 65 66 20 52 45 5f 55 53 45 52 4e 41 4d 45 3a ref RE_USERNAME:
3300: 20 52 65 67 65 78 20 3d 20 52 65 67 65 78 3a 3a Regex = Regex::
3310: 6e 65 77 28 72 22 5e 40 5b 61 2d 7a 41 2d 5a 5d new(r"^@[a-zA-Z]
3320: 5b 61 2d 7a 41 2d 5a 30 2d 39 5f 5d 2b 24 22 29 [a-zA-Z0-9_]+$")
3330: 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 73 74 61 .unwrap();...sta
3340: 74 69 63 20 72 65 66 20 52 45 5f 4c 49 4e 4b 3a tic ref RE_LINK:
3350: 20 52 65 67 65 78 20 3d 20 52 65 67 65 78 3a 3a Regex = Regex::
3360: 6e 65 77 28 72 22 5e 68 74 74 70 73 3f 3a 2f 2f new(r"^https?://
3370: 5b 61 2d 7a 41 2d 5a 2e 30 2d 39 2d 5d 2b 2f 5b [a-zA-Z.0-9-]+/[
3380: 2d 5f 61 2d 7a 41 2d 5a 2e 30 2d 39 2f 3f 3d 5d -_a-zA-Z.0-9/?=]
3390: 2b 24 22 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 +$").unwrap();..
33a0: 09 73 74 61 74 69 63 20 72 65 66 20 52 45 5f 49 .static ref RE_I
33b0: 56 5f 48 41 53 48 3a 20 52 65 67 65 78 20 3d 20 V_HASH: Regex =
33c0: 52 65 67 65 78 3a 3a 6e 65 77 28 72 22 5e 5b 61 Regex::new(r"^[a
33d0: 2d 66 30 2d 39 5d 7b 31 34 7d 24 22 29 2e 75 6e -f0-9]{14}$").un
33e0: 77 72 61 70 28 29 3b 0a 09 7d 0a 0a 09 6d 61 74 wrap();..}...mat
33f0: 63 68 20 75 70 64 61 74 65 2e 6b 69 6e 64 20 7b ch update.kind {
3400: 0a 09 09 55 70 64 61 74 65 4b 69 6e 64 3a 3a 4d ...UpdateKind::M
3410: 65 73 73 61 67 65 28 6d 65 73 73 61 67 65 29 20 essage(message)
3420: 3d 3e 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 => {....let mut
3430: 72 65 70 6c 79 3a 20 56 65 63 3c 53 74 72 69 6e reply: Vec<Strin
3440: 67 3e 20 3d 20 76 65 63 21 5b 5d 3b 0a 09 09 09 g> = vec![];....
3450: 6d 61 74 63 68 20 6d 65 73 73 61 67 65 2e 6b 69 match message.ki
3460: 6e 64 20 7b 0a 09 09 09 09 4d 65 73 73 61 67 65 nd {.....Message
3470: 4b 69 6e 64 3a 3a 54 65 78 74 20 7b 20 72 65 66 Kind::Text { ref
3480: 20 64 61 74 61 2c 20 2e 2e 20 7d 20 3d 3e 20 7b data, .. } => {
3490: 0a 09 09 09 09 09 6c 65 74 20 6d 75 74 20 77 6f ......let mut wo
34a0: 72 64 73 20 3d 20 64 61 74 61 2e 73 70 6c 69 74 rds = data.split
34b0: 5f 77 68 69 74 65 73 70 61 63 65 28 29 3b 0a 09 _whitespace();..
34c0: 09 09 09 09 6c 65 74 20 63 6d 64 20 3d 20 77 6f ....let cmd = wo
34d0: 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 rds.next().unwra
34e0: 70 28 29 3b 0a 09 09 09 09 09 6d 61 74 63 68 20 p();......match
34f0: 63 6d 64 20 7b 0a 0a 2f 2f 20 73 74 61 72 74 0a cmd {..// start.
3500: 0a 09 09 09 09 09 09 22 2f 73 74 61 72 74 22 20 ......."/start"
3510: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 72 65 70 6c => {........repl
3520: 79 2e 70 75 73 68 28 22 57 65 20 61 72 65 20 6f y.push("We are o
3530: 70 65 6e 5c 5c 2e 20 50 72 6f 62 61 62 6c 79 5c pen\\. Probably\
3540: 5c 2e 20 56 69 73 69 74 20 5b 63 68 61 6e 6e 65 \. Visit [channe
3550: 6c 5d 28 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f l](https://t.me/
3560: 72 73 73 74 67 5f 62 6f 74 5f 68 65 6c 70 2f 33 rsstg_bot_help/3
3570: 29 20 66 6f 72 20 64 65 74 61 69 6c 73 5c 5c 2e ) for details\\.
3580: 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a ".to_string());.
3590: 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 6c 69 73 ......},..// lis
35a0: 74 0a 0a 09 09 09 09 09 09 22 2f 6c 69 73 74 22 t........"/list"
35b0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 72 65 70 => {........rep
35c0: 6c 79 2e 61 70 70 65 6e 64 28 26 6d 75 74 20 63 ly.append(&mut c
35d0: 6f 72 65 2e 6c 69 73 74 28 6d 65 73 73 61 67 65 ore.list(message
35e0: 2e 66 72 6f 6d 2e 69 64 29 2e 61 77 61 69 74 3f .from.id).await?
35f0: 29 3b 0a 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 );.......},..//
3600: 61 64 64 0a 0a 09 09 09 09 09 09 22 2f 61 64 64 add........"/add
3610: 22 20 7c 20 22 2f 75 70 64 61 74 65 22 20 3d 3e " | "/update" =>
3620: 20 7b 0a 09 09 09 09 09 09 09 5f 72 65 70 6c 79 {........_reply
3630: 5f 74 6f 20 3d 20 26 53 6f 6d 65 28 6d 65 73 73 _to = &Some(mess
3640: 61 67 65 2e 66 72 6f 6d 2e 69 64 29 3b 0a 09 09 age.from.id);...
3650: 09 09 09 09 09 6c 65 74 20 6d 75 74 20 73 6f 75 .....let mut sou
3660: 72 63 65 5f 69 64 3a 20 4f 70 74 69 6f 6e 3c 69 rce_id: Option<i
3670: 33 32 3e 20 3d 20 4e 6f 6e 65 3b 0a 09 09 09 09 32> = None;.....
3680: 09 09 09 6c 65 74 20 61 74 5f 6c 65 61 73 74 20 ...let at_least
3690: 3d 20 22 52 65 71 75 69 72 65 73 20 61 74 20 6c = "Requires at l
36a0: 65 61 73 74 20 33 20 70 61 72 61 6d 65 74 65 72 east 3 parameter
36b0: 73 2e 22 3b 0a 09 09 09 09 09 09 09 69 66 20 63 s.";........if c
36c0: 6d 64 20 3d 3d 20 22 2f 75 70 64 61 74 65 22 20 md == "/update"
36d0: 7b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 66 69 {.........let fi
36e0: 72 73 74 5f 77 6f 72 64 20 3d 20 77 6f 72 64 73 rst_word = words
36f0: 2e 6e 65 78 74 28 29 0a 09 09 09 09 09 09 09 09 .next().........
3700: 09 2e 63 6f 6e 74 65 78 74 28 61 74 5f 6c 65 61 ..context(at_lea
3710: 73 74 29 3f 3b 0a 09 09 09 09 09 09 09 09 73 6f st)?;.........so
3720: 75 72 63 65 5f 69 64 20 3d 20 53 6f 6d 65 28 66 urce_id = Some(f
3730: 69 72 73 74 5f 77 6f 72 64 2e 70 61 72 73 65 3a irst_word.parse:
3740: 3a 3c 69 33 32 3e 28 29 0a 09 09 09 09 09 09 09 :<i32>()........
3750: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
3760: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 49 20 6e 65 || format!("I ne
3770: 65 64 20 61 20 6e 75 6d 62 65 72 2c 20 62 75 74 ed a number, but
3780: 20 67 6f 74 20 7b 7d 2e 22 2c 20 66 69 72 73 74 got {}.", first
3790: 5f 77 6f 72 64 29 29 3f 29 3b 0a 09 09 09 09 09 _word))?);......
37a0: 09 09 7d 0a 09 09 09 09 09 09 09 6c 65 74 20 28 ..}........let (
37b0: 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 2c 20 69 76 channel, url, iv
37c0: 5f 68 61 73 68 29 20 3d 20 28 0a 09 09 09 09 09 _hash) = (......
37d0: 09 09 09 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e ...words.next().
37e0: 63 6f 6e 74 65 78 74 28 61 74 5f 6c 65 61 73 74 context(at_least
37f0: 29 3f 2c 0a 09 09 09 09 09 09 09 09 77 6f 72 64 )?,.........word
3800: 73 2e 6e 65 78 74 28 29 2e 63 6f 6e 74 65 78 74 s.next().context
3810: 28 61 74 5f 6c 65 61 73 74 29 3f 2c 0a 09 09 09 (at_least)?,....
3820: 09 09 09 09 09 77 6f 72 64 73 2e 6e 65 78 74 28 .....words.next(
3830: 29 29 3b 0a 09 09 09 09 09 09 09 69 66 20 21 20 ));........if !
3840: 52 45 5f 55 53 45 52 4e 41 4d 45 2e 69 73 5f 6d RE_USERNAME.is_m
3850: 61 74 63 68 28 26 63 68 61 6e 6e 65 6c 29 20 7b atch(&channel) {
3860: 0a 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 .........reply.p
3870: 75 73 68 28 22 55 73 65 72 6e 61 6d 65 73 20 73 ush("Usernames s
3880: 68 6f 75 6c 64 20 62 65 20 73 6f 6d 65 74 68 69 hould be somethi
3890: 6e 67 20 6c 69 6b 65 20 5c 22 40 5c 5c 5b 61 5c ng like \"@\\[a\
38a0: 5c 2d 7a 41 5c 5c 2d 5a 5d 5c 5c 5b 61 5c 5c 2d \-zA\\-Z]\\[a\\-
38b0: 7a 41 5c 5c 2d 5a 30 5c 5c 2d 39 5c 5c 5f 5d 2b zA\\-Z0\\-9\\_]+
38c0: 5c 22 2c 20 61 72 65 6e 27 74 20 74 68 65 79 3f \", aren't they?
38d0: 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a ".to_string());.
38e0: 09 09 09 09 09 09 09 09 62 61 69 6c 21 28 22 57 ........bail!("W
38f0: 72 6f 6e 67 20 75 73 65 72 6e 61 6d 65 20 7b 3a rong username {:
3900: 3f 7d 2e 22 2c 20 26 63 68 61 6e 6e 65 6c 29 3b ?}.", &channel);
3910: 0a 09 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 ........}.......
3920: 09 69 66 20 21 20 52 45 5f 4c 49 4e 4b 2e 69 73 .if ! RE_LINK.is
3930: 5f 6d 61 74 63 68 28 26 75 72 6c 29 20 7b 0a 09 _match(&url) {..
3940: 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 .......reply.pus
3950: 68 28 22 4c 69 6e 6b 20 73 68 6f 75 6c 64 20 62 h("Link should b
3960: 65 20 6c 69 6e 6b 20 74 6f 20 61 74 6f 6d 2f 72 e link to atom/r
3970: 73 73 20 66 65 65 64 2c 20 73 6f 6d 65 74 68 69 ss feed, somethi
3980: 6e 67 20 6c 69 6b 65 20 5c 22 68 74 74 70 73 3a ng like \"https:
3990: 2f 2f 64 6f 6d 61 69 6e 2f 70 61 74 68 5c 22 5c //domain/path\"\
39a0: 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 \.".to_string())
39b0: 3b 0a 09 09 09 09 09 09 09 09 62 61 69 6c 21 28 ;.........bail!(
39c0: 22 55 72 6c 3a 20 7b 3a 3f 7d 22 2c 20 26 75 72 "Url: {:?}", &ur
39d0: 6c 29 3b 0a 09 09 09 09 09 09 09 7d 0a 09 09 09 l);........}....
39e0: 09 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 ....if let Some(
39f0: 68 61 73 68 29 20 3d 20 69 76 5f 68 61 73 68 20 hash) = iv_hash
3a00: 7b 0a 09 09 09 09 09 09 09 09 69 66 20 21 20 52 {.........if ! R
3a10: 45 5f 49 56 5f 48 41 53 48 2e 69 73 5f 6d 61 74 E_IV_HASH.is_mat
3a20: 63 68 28 26 68 61 73 68 29 20 7b 0a 09 09 09 09 ch(&hash) {.....
3a30: 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 .....reply.push(
3a40: 22 49 56 20 68 61 73 68 20 73 68 6f 75 6c 64 20 "IV hash should
3a50: 62 65 20 31 34 20 68 65 78 20 64 69 67 69 74 73 be 14 hex digits
3a60: 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b .".to_string());
3a70: 0a 09 09 09 09 09 09 09 09 09 62 61 69 6c 21 28 ..........bail!(
3a80: 22 49 56 3a 20 7b 3a 3f 7d 22 2c 20 26 69 76 5f "IV: {:?}", &iv_
3a90: 68 61 73 68 29 3b 0a 09 09 09 09 09 09 09 09 7d hash);.........}
3aa0: 3b 0a 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 ;........};.....
3ab0: 09 09 09 6c 65 74 20 63 68 61 6e 6e 65 6c 5f 69 ...let channel_i
3ac0: 64 20 3d 20 69 36 34 3a 3a 66 72 6f 6d 28 63 6f d = i64::from(co
3ad0: 72 65 2e 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 re.tg.send(teleg
3ae0: 72 61 6d 5f 62 6f 74 3a 3a 47 65 74 43 68 61 74 ram_bot::GetChat
3af0: 3a 3a 6e 65 77 28 74 65 6c 65 67 72 61 6d 5f 62 ::new(telegram_b
3b00: 6f 74 3a 3a 74 79 70 65 73 3a 3a 43 68 61 74 52 ot::types::ChatR
3b10: 65 66 3a 3a 43 68 61 6e 6e 65 6c 55 73 65 72 6e ef::ChannelUsern
3b20: 61 6d 65 28 63 68 61 6e 6e 65 6c 2e 74 6f 5f 73 ame(channel.to_s
3b30: 74 72 69 6e 67 28 29 29 29 29 2e 61 77 61 69 74 tring()))).await
3b40: 3f 2e 69 64 28 29 29 3b 0a 09 09 09 09 09 09 09 ?.id());........
3b50: 6c 65 74 20 63 68 61 6e 5f 61 64 6d 20 3d 20 63 let chan_adm = c
3b60: 6f 72 65 2e 74 67 2e 73 65 6e 64 28 74 65 6c 65 ore.tg.send(tele
3b70: 67 72 61 6d 5f 62 6f 74 3a 3a 47 65 74 43 68 61 gram_bot::GetCha
3b80: 74 41 64 6d 69 6e 69 73 74 72 61 74 6f 72 73 3a tAdministrators:
3b90: 3a 6e 65 77 28 74 65 6c 65 67 72 61 6d 5f 62 6f :new(telegram_bo
3ba0: 74 3a 3a 74 79 70 65 73 3a 3a 43 68 61 74 52 65 t::types::ChatRe
3bb0: 66 3a 3a 43 68 61 6e 6e 65 6c 55 73 65 72 6e 61 f::ChannelUserna
3bc0: 6d 65 28 63 68 61 6e 6e 65 6c 2e 74 6f 5f 73 74 me(channel.to_st
3bd0: 72 69 6e 67 28 29 29 29 29 2e 61 77 61 69 74 0a ring()))).await.
3be0: 09 09 09 09 09 09 09 09 2e 63 6f 6e 74 65 78 74 .........context
3bf0: 28 22 53 6f 72 72 79 2c 20 49 20 68 61 76 65 20 ("Sorry, I have
3c00: 6e 6f 20 61 63 63 65 73 73 20 74 6f 20 74 68 61 no access to tha
3c10: 74 20 63 68 61 74 5c 5c 2e 22 29 3f 3b 0a 09 09 t chat\\.")?;...
3c20: 09 09 09 09 09 6c 65 74 20 28 6d 75 74 20 6d 65 .....let (mut me
3c30: 2c 20 6d 75 74 20 75 73 65 72 29 20 3d 20 28 66 , mut user) = (f
3c40: 61 6c 73 65 2c 20 66 61 6c 73 65 29 3b 0a 09 09 alse, false);...
3c50: 09 09 09 09 09 66 6f 72 20 61 64 6d 69 6e 20 69 .....for admin i
3c60: 6e 20 63 68 61 6e 5f 61 64 6d 20 7b 0a 09 09 09 n chan_adm {....
3c70: 09 09 09 09 09 69 66 20 61 64 6d 69 6e 2e 75 73 .....if admin.us
3c80: 65 72 2e 69 64 20 3d 3d 20 63 6f 72 65 2e 6d 79 er.id == core.my
3c90: 2e 69 64 20 7b 0a 09 09 09 09 09 09 09 09 09 6d .id {..........m
3ca0: 65 20 3d 20 74 72 75 65 3b 0a 09 09 09 09 09 09 e = true;.......
3cb0: 09 09 7d 3b 0a 09 09 09 09 09 09 09 09 69 66 20 ..};.........if
3cc0: 61 64 6d 69 6e 2e 75 73 65 72 2e 69 64 20 3d 3d admin.user.id ==
3cd0: 20 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 message.from.id
3ce0: 20 7b 0a 09 09 09 09 09 09 09 09 09 75 73 65 72 {..........user
3cf0: 20 3d 20 74 72 75 65 3b 0a 09 09 09 09 09 09 09 = true;........
3d00: 09 7d 3b 0a 09 09 09 09 09 09 09 7d 3b 0a 09 09 .};........};...
3d10: 09 09 09 09 09 69 66 20 21 20 6d 65 20 20 20 7b .....if ! me {
3d20: 20 62 61 69 6c 21 28 22 49 20 6e 65 65 64 20 74 bail!("I need t
3d30: 6f 20 62 65 20 61 64 6d 69 6e 20 6f 6e 20 74 68 o be admin on th
3d40: 61 74 20 63 68 61 6e 6e 65 6c 5c 5c 2e 22 29 3b at channel\\.");
3d50: 20 7d 3b 0a 09 09 09 09 09 09 09 69 66 20 21 20 };........if !
3d60: 75 73 65 72 20 7b 20 62 61 69 6c 21 28 22 59 6f user { bail!("Yo
3d70: 75 20 73 68 6f 75 6c 64 20 62 65 20 61 64 6d 69 u should be admi
3d80: 6e 20 6f 6e 20 74 68 61 74 20 63 68 61 6e 6e 65 n on that channe
3d90: 6c 5c 5c 2e 22 29 3b 20 7d 3b 0a 09 09 09 09 09 l\\."); };......
3da0: 09 09 72 65 70 6c 79 2e 70 75 73 68 28 63 6f 72 ..reply.push(cor
3db0: 65 2e 75 70 64 61 74 65 28 73 6f 75 72 63 65 5f e.update(source_
3dc0: 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 63 68 61 id, channel, cha
3dd0: 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 nnel_id, url, iv
3de0: 5f 68 61 73 68 2c 20 6d 65 73 73 61 67 65 2e 66 _hash, message.f
3df0: 72 6f 6d 2e 69 64 29 2e 61 77 61 69 74 3f 29 3b rom.id).await?);
3e00: 0a 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 63 68 .......},..// ch
3e10: 65 63 6b 0a 0a 09 09 09 09 09 09 22 2f 63 68 65 eck........"/che
3e20: 63 6b 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 ck" => {........
3e30: 6d 61 74 63 68 20 26 77 6f 72 64 73 2e 6e 65 78 match &words.nex
3e40: 74 28 29 2e 75 6e 77 72 61 70 28 29 2e 70 61 72 t().unwrap().par
3e50: 73 65 3a 3a 3c 69 33 32 3e 28 29 20 7b 0a 09 09 se::<i32>() {...
3e60: 09 09 09 09 09 09 45 72 72 28 65 72 72 29 20 3d ......Err(err) =
3e70: 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 72 65 70 > {..........rep
3e80: 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 ly.push(format!(
3e90: 22 49 20 6e 65 65 64 20 61 20 6e 75 6d 62 65 72 "I need a number
3ea0: 5c 5c 2e 5c 6e 7b 7d 22 2c 20 26 65 72 72 29 29 \\.\n{}", &err))
3eb0: 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 ;.........},....
3ec0: 09 09 09 09 09 4f 6b 28 6e 75 6d 62 65 72 29 20 .....Ok(number)
3ed0: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 63 6f => {..........co
3ee0: 72 65 2e 63 68 65 63 6b 28 2a 6e 75 6d 62 65 72 re.check(*number
3ef0: 2c 20 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 , message.from.i
3f00: 64 2c 20 66 61 6c 73 65 29 2e 61 77 61 69 74 0a d, false).await.
3f10: 09 09 09 09 09 09 09 09 09 09 2e 63 6f 6e 74 65 ...........conte
3f20: 78 74 28 22 43 68 61 6e 6e 65 6c 20 63 68 65 63 xt("Channel chec
3f30: 6b 20 66 61 69 6c 65 64 2e 22 29 3f 3b 0a 09 09 k failed.")?;...
3f40: 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 ......},........
3f50: 7d 3b 0a 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 };.......},..//
3f60: 63 6c 65 61 6e 0a 0a 09 09 09 09 09 09 22 2f 63 clean........"/c
3f70: 6c 65 61 6e 22 20 3d 3e 20 7b 0a 09 09 09 09 09 lean" => {......
3f80: 09 09 6d 61 74 63 68 20 26 77 6f 72 64 73 2e 6e ..match &words.n
3f90: 65 78 74 28 29 2e 75 6e 77 72 61 70 28 29 2e 70 ext().unwrap().p
3fa0: 61 72 73 65 3a 3a 3c 69 33 32 3e 28 29 20 7b 0a arse::<i32>() {.
3fb0: 09 09 09 09 09 09 09 09 45 72 72 28 65 72 72 29 ........Err(err)
3fc0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 72 => {..........r
3fd0: 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 eply.push(format
3fe0: 21 28 22 49 20 6e 65 65 64 20 61 20 6e 75 6d 62 !("I need a numb
3ff0: 65 72 5c 5c 2e 5c 6e 7b 7d 22 2c 20 26 65 72 72 er\\.\n{}", &err
4000: 29 29 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 ));.........},..
4010: 09 09 09 09 09 09 09 4f 6b 28 6e 75 6d 62 65 72 .......Ok(number
4020: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 ) => {..........
4030: 6c 65 74 20 72 65 73 75 6c 74 20 3d 20 63 6f 72 let result = cor
4040: 65 2e 63 6c 65 61 6e 28 26 6e 75 6d 62 65 72 2c e.clean(&number,
4050: 20 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 message.from.id
4060: 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 09 09 09 ).await?;.......
4070: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 72 65 ...reply.push(re
4080: 73 75 6c 74 2e 74 6f 5f 73 74 72 69 6e 67 28 29 sult.to_string()
4090: 29 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 );.........},...
40a0: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 2c .....};.......},
40b0: 0a 0a 2f 2f 20 65 6e 61 62 6c 65 0a 0a 09 09 09 ..// enable.....
40c0: 09 09 09 22 2f 65 6e 61 62 6c 65 22 20 3d 3e 20 ..."/enable" =>
40d0: 7b 0a 09 09 09 09 09 09 09 6d 61 74 63 68 20 26 {........match &
40e0: 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 words.next().unw
40f0: 72 61 70 28 29 2e 70 61 72 73 65 3a 3a 3c 69 33 rap().parse::<i3
4100: 32 3e 28 29 20 7b 0a 09 09 09 09 09 09 09 09 45 2>() {.........E
4110: 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 rr(err) => {....
4120: 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 ......reply.push
4130: 28 66 6f 72 6d 61 74 21 28 22 49 20 6e 65 65 64 (format!("I need
4140: 20 61 20 6e 75 6d 62 65 72 5c 5c 2e 5c 6e 7b 7d a number\\.\n{}
4150: 22 2c 20 26 65 72 72 29 29 3b 0a 09 09 09 09 09 ", &err));......
4160: 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 4f 6b ...},.........Ok
4170: 28 6e 75 6d 62 65 72 29 20 3d 3e 20 7b 0a 09 09 (number) => {...
4180: 09 09 09 09 09 09 09 6c 65 74 20 72 65 73 75 6c .......let resul
4190: 74 20 3d 20 63 6f 72 65 2e 65 6e 61 62 6c 65 28 t = core.enable(
41a0: 26 6e 75 6d 62 65 72 2c 20 6d 65 73 73 61 67 65 &number, message
41b0: 2e 66 72 6f 6d 2e 69 64 29 2e 61 77 61 69 74 3f .from.id).await?
41c0: 3b 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 ;..........reply
41d0: 2e 70 75 73 68 28 72 65 73 75 6c 74 2e 74 6f 5f .push(result.to_
41e0: 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 string());......
41f0: 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 7d 3b 0a ...},........};.
4200: 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 64 65 6c ......},..// del
4210: 65 74 65 0a 0a 09 09 09 09 09 09 22 2f 64 65 6c ete........"/del
4220: 65 74 65 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 ete" => {.......
4230: 09 6d 61 74 63 68 20 26 77 6f 72 64 73 2e 6e 65 .match &words.ne
4240: 78 74 28 29 2e 75 6e 77 72 61 70 28 29 2e 70 61 xt().unwrap().pa
4250: 72 73 65 3a 3a 3c 69 33 32 3e 28 29 20 7b 0a 09 rse::<i32>() {..
4260: 09 09 09 09 09 09 09 45 72 72 28 65 72 72 29 20 .......Err(err)
4270: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 72 65 => {..........re
4280: 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 ply.push(format!
4290: 28 22 49 20 6e 65 65 64 20 61 20 6e 75 6d 62 65 ("I need a numbe
42a0: 72 5c 5c 2e 5c 6e 7b 7d 22 2c 20 26 65 72 72 29 r\\.\n{}", &err)
42b0: 29 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 );.........},...
42c0: 09 09 09 09 09 09 4f 6b 28 6e 75 6d 62 65 72 29 ......Ok(number)
42d0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 6c => {..........l
42e0: 65 74 20 72 65 73 75 6c 74 20 3d 20 63 6f 72 65 et result = core
42f0: 2e 64 65 6c 65 74 65 28 26 6e 75 6d 62 65 72 2c .delete(&number,
4300: 20 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 message.from.id
4310: 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 09 09 09 ).await?;.......
4320: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 72 65 ...reply.push(re
4330: 73 75 6c 74 2e 74 6f 5f 73 74 72 69 6e 67 28 29 sult.to_string()
4340: 29 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 );.........},...
4350: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 2c .....};.......},
4360: 0a 0a 2f 2f 20 64 69 73 61 62 6c 65 0a 0a 09 09 ..// disable....
4370: 09 09 09 09 22 2f 64 69 73 61 62 6c 65 22 20 3d ...."/disable" =
4380: 3e 20 7b 0a 09 09 09 09 09 09 09 6d 61 74 63 68 > {........match
4390: 20 26 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e 75 &words.next().u
43a0: 6e 77 72 61 70 28 29 2e 70 61 72 73 65 3a 3a 3c nwrap().parse::<
43b0: 69 33 32 3e 28 29 20 7b 0a 09 09 09 09 09 09 09 i32>() {........
43c0: 09 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 .Err(err) => {..
43d0: 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 ........reply.pu
43e0: 73 68 28 66 6f 72 6d 61 74 21 28 22 49 20 6e 65 sh(format!("I ne
43f0: 65 64 20 61 20 6e 75 6d 62 65 72 5c 5c 2e 5c 6e ed a number\\.\n
4400: 7b 7d 22 2c 20 26 65 72 72 29 29 3b 0a 09 09 09 {}", &err));....
4410: 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 .....},.........
4420: 4f 6b 28 6e 75 6d 62 65 72 29 20 3d 3e 20 7b 0a Ok(number) => {.
4430: 09 09 09 09 09 09 09 09 09 6c 65 74 20 72 65 73 .........let res
4440: 75 6c 74 20 3d 20 63 6f 72 65 2e 64 69 73 61 62 ult = core.disab
4450: 6c 65 28 26 6e 75 6d 62 65 72 2c 20 6d 65 73 73 le(&number, mess
4460: 61 67 65 2e 66 72 6f 6d 2e 69 64 29 2e 61 77 61 age.from.id).awa
4470: 69 74 3f 3b 0a 09 09 09 09 09 09 09 09 09 72 65 it?;..........re
4480: 70 6c 79 2e 70 75 73 68 28 72 65 73 75 6c 74 2e ply.push(result.
4490: 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 to_string());...
44a0: 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 ......},........
44b0: 7d 3b 0a 09 09 09 09 09 09 7d 2c 0a 0a 09 09 09 };.......},.....
44c0: 09 09 09 5f 20 3d 3e 20 7b 0a 09 09 09 09 09 09 ..._ => {.......
44d0: 7d 2c 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 7d },......};.....}
44e0: 2c 0a 09 09 09 09 5f 20 3d 3e 20 7b 0a 09 09 09 ,....._ => {....
44f0: 09 7d 2c 0a 09 09 09 7d 3b 0a 0a 09 09 09 69 66 .},....};.....if
4500: 20 72 65 70 6c 79 2e 6c 65 6e 28 29 20 3e 20 30 reply.len() > 0
4510: 20 7b 0a 09 09 09 09 69 66 20 6c 65 74 20 45 72 {.....if let Er
4520: 72 28 65 72 72 29 20 3d 20 63 6f 72 65 2e 74 67 r(err) = core.tg
4530: 2e 73 65 6e 64 28 6d 65 73 73 61 67 65 2e 74 65 .send(message.te
4540: 78 74 5f 72 65 70 6c 79 28 72 65 70 6c 79 2e 6a xt_reply(reply.j
4550: 6f 69 6e 28 22 5c 6e 22 29 29 2e 70 61 72 73 65 oin("\n")).parse
4560: 5f 6d 6f 64 65 28 74 79 70 65 73 3a 3a 50 61 72 _mode(types::Par
4570: 73 65 4d 6f 64 65 3a 3a 4d 61 72 6b 64 6f 77 6e seMode::Markdown
4580: 56 32 29 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 V2)).await {....
4590: 09 09 64 62 67 21 28 72 65 70 6c 79 2e 6a 6f 69 ..dbg!(reply.joi
45a0: 6e 28 22 5c 6e 22 29 29 3b 0a 09 09 09 09 09 70 n("\n"));......p
45b0: 72 69 6e 74 6c 6e 21 28 22 7b 7d 22 2c 20 65 72 rintln!("{}", er
45c0: 72 29 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 7d 3b r);.....};....};
45d0: 0a 09 09 7d 2c 0a 09 09 5f 20 3d 3e 20 7b 7d 2c ...},..._ => {},
45e0: 0a 09 7d 3b 0a 0a 09 4f 6b 28 28 29 29 0a 7d 0a ..};...Ok(()).}.