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 0a 75 73 65 20 72 73 73 3b 0a 0a 75 73 65 20 ..use rss;..use
0070: 63 68 72 6f 6e 6f 3a 3a 44 61 74 65 54 69 6d 65 chrono::DateTime
0080: 3b 0a 0a 75 73 65 20 72 65 67 65 78 3a 3a 52 65 ;..use regex::Re
0090: 67 65 78 3b 0a 0a 75 73 65 20 74 65 6c 65 67 72 gex;..use telegr
00a0: 61 6d 5f 62 6f 74 3a 3a 2a 3b 0a 75 73 65 20 74 am_bot::*;.use t
00b0: 6f 6b 69 6f 3a 3a 73 74 72 65 61 6d 3a 3a 53 74 okio::stream::St
00c0: 72 65 61 6d 45 78 74 3b 0a 0a 75 73 65 20 73 71 reamExt;..use sq
00d0: 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a 3a 50 67 lx::postgres::Pg
00e0: 50 6f 6f 6c 4f 70 74 69 6f 6e 73 3b 0a 75 73 65 PoolOptions;.use
00f0: 20 73 71 6c 78 3a 3a 52 6f 77 3b 0a 75 73 65 20 sqlx::Row;.use
0100: 73 71 6c 78 3a 3a 44 6f 6e 65 3b 20 2f 2f 20 2e sqlx::Done; // .
0110: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 0a rows_affected().
0120: 0a 23 5b 6d 61 63 72 6f 5f 75 73 65 5d 0a 65 78 .#[macro_use].ex
0130: 74 65 72 6e 20 63 72 61 74 65 20 6c 61 7a 79 5f tern crate lazy_
0140: 73 74 61 74 69 63 3b 0a 0a 75 73 65 20 61 6e 79 static;..use any
0150: 68 6f 77 3a 3a 7b 61 6e 79 68 6f 77 2c 20 62 61 how::{anyhow, ba
0160: 69 6c 2c 20 43 6f 6e 74 65 78 74 2c 20 52 65 73 il, Context, Res
0170: 75 6c 74 7d 3b 0a 0a 23 5b 64 65 72 69 76 65 28 ult};..#[derive(
0180: 43 6c 6f 6e 65 29 5d 0a 73 74 72 75 63 74 20 43 Clone)].struct C
0190: 6f 72 65 20 7b 0a 09 6f 77 6e 65 72 3a 20 69 36 ore {..owner: i6
01a0: 34 2c 0a 09 61 70 69 5f 6b 65 79 3a 20 53 74 72 4,..api_key: Str
01b0: 69 6e 67 2c 0a 09 6f 77 6e 65 72 5f 63 68 61 74 ing,..owner_chat
01c0: 3a 20 55 73 65 72 49 64 2c 0a 09 74 67 3a 20 74 : UserId,..tg: t
01d0: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 41 70 69 elegram_bot::Api
01e0: 2c 0a 09 6d 79 3a 20 55 73 65 72 2c 0a 09 70 6f ,..my: User,..po
01f0: 6f 6c 3a 20 73 71 6c 78 3a 3a 50 6f 6f 6c 3c 73 ol: sqlx::Pool<s
0200: 71 6c 78 3a 3a 50 6f 73 74 67 72 65 73 3e 2c 0a qlx::Postgres>,.
0210: 09 73 6f 75 72 63 65 73 3a 20 41 72 63 3c 4d 75 .sources: Arc<Mu
0220: 74 65 78 3c 48 61 73 68 53 65 74 3c 41 72 63 3c tex<HashSet<Arc<
0230: 69 33 32 3e 3e 3e 3e 2c 0a 7d 0a 0a 69 6d 70 6c i32>>>>,.}..impl
0240: 20 43 6f 72 65 20 7b 0a 09 61 73 79 6e 63 20 66 Core {..async f
0250: 6e 20 6e 65 77 28 73 65 74 74 69 6e 67 73 3a 20 n new(settings:
0260: 63 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 29 20 config::Config)
0270: 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 72 65 3e 20 -> Result<Core>
0280: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
0290: 73 65 74 74 69 6e 67 73 2e 67 65 74 5f 69 6e 74 settings.get_int
02a0: 28 22 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 6c 65 ("owner")?;...le
02b0: 74 20 61 70 69 5f 6b 65 79 20 3d 20 73 65 74 74 t api_key = sett
02c0: 69 6e 67 73 2e 67 65 74 5f 73 74 72 28 22 61 70 ings.get_str("ap
02d0: 69 5f 6b 65 79 22 29 3f 3b 0a 09 09 6c 65 74 20 i_key")?;...let
02e0: 74 67 20 3d 20 41 70 69 3a 3a 6e 65 77 28 26 61 tg = Api::new(&a
02f0: 70 69 5f 6b 65 79 29 3b 0a 09 09 6c 65 74 20 63 pi_key);...let c
0300: 6f 72 65 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 ore = Core {....
0310: 6f 77 6e 65 72 3a 20 6f 77 6e 65 72 2c 0a 09 09 owner: owner,...
0320: 09 61 70 69 5f 6b 65 79 3a 20 61 70 69 5f 6b 65 .api_key: api_ke
0330: 79 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 6d 79 y.clone(),....my
0340: 3a 20 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 72 : tg.send(telegr
0350: 61 6d 5f 62 6f 74 3a 3a 47 65 74 4d 65 29 2e 61 am_bot::GetMe).a
0360: 77 61 69 74 3f 2c 0a 09 09 09 74 67 3a 20 74 67 wait?,....tg: tg
0370: 2c 0a 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a ,....owner_chat:
0380: 20 55 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e UserId::new(own
0390: 65 72 29 2c 0a 09 09 09 70 6f 6f 6c 3a 20 50 67 er),....pool: Pg
03a0: 50 6f 6f 6c 4f 70 74 69 6f 6e 73 3a 3a 6e 65 77 PoolOptions::new
03b0: 28 29 0a 09 09 09 09 2e 6d 61 78 5f 63 6f 6e 6e ()......max_conn
03c0: 65 63 74 69 6f 6e 73 28 35 29 0a 09 09 09 09 2e ections(5)......
03d0: 63 6f 6e 6e 65 63 74 5f 74 69 6d 65 6f 75 74 28 connect_timeout(
03e0: 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 std::time::Durat
03f0: 69 6f 6e 3a 3a 6e 65 77 28 33 30 30 2c 20 30 29 ion::new(300, 0)
0400: 29 0a 09 09 09 09 2e 69 64 6c 65 5f 74 69 6d 65 )......idle_time
0410: 6f 75 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 out(std::time::D
0420: 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 36 30 2c uration::new(60,
0430: 20 30 29 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 0))......connec
0440: 74 5f 6c 61 7a 79 28 26 73 65 74 74 69 6e 67 73 t_lazy(&settings
0450: 2e 67 65 74 5f 73 74 72 28 22 70 67 22 29 3f 29 .get_str("pg")?)
0460: 3f 2c 0a 09 09 09 73 6f 75 72 63 65 73 3a 20 41 ?,....sources: A
0470: 72 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a 3a 6e rc::new(Mutex::n
0480: 65 77 28 48 61 73 68 53 65 74 3a 3a 6e 65 77 28 ew(HashSet::new(
0490: 29 29 29 2c 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 ))),...};...let
04a0: 63 6c 6f 6e 65 20 3d 20 63 6f 72 65 2e 63 6c 6f clone = core.clo
04b0: 6e 65 28 29 3b 0a 09 09 74 6f 6b 69 6f 3a 3a 73 ne();...tokio::s
04c0: 70 61 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 pawn(async move
04d0: 7b 0a 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 {....if let Err(
04e0: 65 72 72 29 20 3d 20 26 63 6c 6f 6e 65 2e 61 75 err) = &clone.au
04f0: 74 6f 66 65 74 63 68 28 29 2e 61 77 61 69 74 20 tofetch().await
0500: 7b 0a 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 {.....if let Err
0510: 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 64 65 (err) = clone.de
0520: 62 75 67 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f bug(&format!("š
0530: 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 29 20 {:?}", err))
0540: 7b 0a 09 09 09 09 09 65 70 72 69 6e 74 6c 6e 21 {......eprintln!
0550: 28 22 41 75 74 6f 66 65 74 63 68 20 65 72 72 6f ("Autofetch erro
0560: 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 r: {}", err);...
0570: 09 09 7d 3b 0a 09 09 09 7d 0a 09 09 7d 29 3b 0a ..};....}...});.
0580: 09 09 4f 6b 28 63 6f 72 65 29 0a 09 7d 0a 0a 09 ..Ok(core)..}...
0590: 66 6e 20 73 74 72 65 61 6d 28 26 73 65 6c 66 29 fn stream(&self)
05a0: 20 2d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 -> telegram_bot
05b0: 3a 3a 55 70 64 61 74 65 73 53 74 72 65 61 6d 20 ::UpdatesStream
05c0: 7b 0a 09 09 73 65 6c 66 2e 74 67 2e 73 74 72 65 {...self.tg.stre
05d0: 61 6d 28 29 0a 09 7d 0a 0a 09 66 6e 20 64 65 62 am()..}...fn deb
05e0: 75 67 28 26 73 65 6c 66 2c 20 6d 73 67 3a 20 26 ug(&self, msg: &
05f0: 73 74 72 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 str) -> Result<(
0600: 29 3e 20 7b 0a 09 09 73 65 6c 66 2e 74 67 2e 73 )> {...self.tg.s
0610: 70 61 77 6e 28 53 65 6e 64 4d 65 73 73 61 67 65 pawn(SendMessage
0620: 3a 3a 6e 65 77 28 73 65 6c 66 2e 6f 77 6e 65 72 ::new(self.owner
0630: 5f 63 68 61 74 2c 20 6d 73 67 29 29 3b 0a 09 09 _chat, msg));...
0640: 4f 6b 28 28 29 29 0a 09 7d 0a 0a 09 61 73 79 6e Ok(())..}...asyn
0650: 63 20 66 6e 20 63 68 65 63 6b 3c 53 3e 28 26 73 c fn check<S>(&s
0660: 65 6c 66 2c 20 69 64 3a 20 69 33 32 2c 20 6f 77 elf, id: i32, ow
0670: 6e 65 72 3a 20 53 2c 20 72 65 61 6c 3a 20 62 6f ner: S, real: bo
0680: 6f 6c 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 ol) -> Result<()
0690: 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f >..where S: Into
06a0: 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 <i64> {...let ow
06b0: 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 ner: i64 = owner
06c0: 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 69 .into();...let i
06d0: 64 20 3d 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 d = {....let mut
06e0: 20 73 65 74 20 3d 20 73 65 6c 66 2e 73 6f 75 72 set = self.sour
06f0: 63 65 73 2e 6c 6f 63 6b 28 29 2e 75 6e 77 72 61 ces.lock().unwra
0700: 70 28 29 3b 0a 09 09 09 6d 61 74 63 68 20 73 65 p();....match se
0710: 74 2e 67 65 74 28 26 69 64 29 20 7b 0a 09 09 09 t.get(&id) {....
0720: 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 69 64 2e .Some(id) => id.
0730: 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 09 4e 6f 6e clone(),.....Non
0740: 65 20 3d 3e 20 7b 0a 09 09 09 09 09 6c 65 74 20 e => {......let
0750: 69 64 20 3d 20 41 72 63 3a 3a 6e 65 77 28 69 64 id = Arc::new(id
0760: 29 3b 0a 09 09 09 09 09 73 65 74 2e 69 6e 73 65 );......set.inse
0770: 72 74 28 69 64 2e 63 6c 6f 6e 65 28 29 29 3b 0a rt(id.clone());.
0780: 09 09 09 09 09 69 64 2e 63 6c 6f 6e 65 28 29 0a .....id.clone().
0790: 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 7d 3b ....},....}...};
07a0: 0a 09 09 6c 65 74 20 63 6f 75 6e 74 20 3d 20 41 ...let count = A
07b0: 72 63 3a 3a 73 74 72 6f 6e 67 5f 63 6f 75 6e 74 rc::strong_count
07c0: 28 26 69 64 29 3b 0a 09 09 69 66 20 63 6f 75 6e (&id);...if coun
07d0: 74 20 3d 3d 20 32 20 7b 0a 09 09 09 6c 65 74 20 t == 2 {....let
07e0: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
07f0: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
0800: 77 61 69 74 0a 09 09 09 09 2e 77 69 74 68 5f 63 wait......with_c
0810: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
0820: 21 28 22 51 75 65 72 79 20 71 75 65 75 65 20 66 !("Query queue f
0830: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
0840: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
0850: 3b 0a 09 09 09 6c 65 74 20 72 6f 77 20 3d 20 73 ;....let row = s
0860: 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 qlx::query("sele
0870: 63 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 ct source_id, ch
0880: 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 annel_id, url, i
0890: 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 20 66 72 v_hash, owner fr
08a0: 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 om rsstg_source
08b0: 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 where source_id
08c0: 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d = $1 and owner =
08d0: 20 24 32 22 29 0a 09 09 09 09 2e 62 69 6e 64 28 $2")......bind(
08e0: 2a 69 64 29 0a 09 09 09 09 2e 62 69 6e 64 28 6f *id)......bind(o
08f0: 77 6e 65 72 29 0a 09 09 09 09 2e 66 65 74 63 68 wner)......fetch
0900: 5f 6f 6e 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e _one(&mut conn).
0910: 61 77 61 69 74 0a 09 09 09 09 2e 77 69 74 68 5f await......with_
0920: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
0930: 74 21 28 22 51 75 65 72 79 20 73 6f 75 72 63 65 t!("Query source
0940: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
0950: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 64 72 6f 70 pool))?;....drop
0960: 28 63 6f 6e 6e 29 3b 0a 09 09 09 6c 65 74 20 63 (conn);....let c
0970: 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 20 3d hannel_id: i64 =
0980: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 63 68 row.try_get("ch
0990: 61 6e 6e 65 6c 5f 69 64 22 29 3f 3b 0a 09 09 09 annel_id")?;....
09a0: 6c 65 74 20 64 65 73 74 69 6e 61 74 69 6f 6e 20 let destination
09b0: 3d 20 6d 61 74 63 68 20 72 65 61 6c 20 7b 0a 09 = match real {..
09c0: 09 09 09 74 72 75 65 20 3d 3e 20 55 73 65 72 49 ...true => UserI
09d0: 64 3a 3a 6e 65 77 28 63 68 61 6e 6e 65 6c 5f 69 d::new(channel_i
09e0: 64 29 2c 0a 09 09 09 09 66 61 6c 73 65 20 3d 3e d),.....false =>
09f0: 20 55 73 65 72 49 64 3a 3a 6e 65 77 28 72 6f 77 UserId::new(row
0a00: 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e 65 72 22 .try_get("owner"
0a10: 29 3f 29 2c 0a 09 09 09 7d 3b 0a 09 09 09 6c 65 )?),....};....le
0a20: 74 20 75 72 6c 3a 20 26 73 74 72 20 3d 20 72 6f t url: &str = ro
0a30: 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 22 29 w.try_get("url")
0a40: 3f 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 74 68 ?;....let mut th
0a50: 69 73 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e is_fetch: Option
0a60: 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f <DateTime<chrono
0a70: 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 3e 20 ::FixedOffset>>
0a80: 3d 20 4e 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 69 = None;....let i
0a90: 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 v_hash: Option<&
0aa0: 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 str> = row.try_g
0ab0: 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f 3b 0a et("iv_hash")?;.
0ac0: 09 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 73 ...let mut posts
0ad0: 3a 20 42 54 72 65 65 4d 61 70 3c 44 61 74 65 54 : BTreeMap<DateT
0ae0: 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 ime<chrono::Fixe
0af0: 64 4f 66 66 73 65 74 3e 2c 20 53 74 72 69 6e 67 dOffset>, String
0b00: 3e 20 3d 20 42 54 72 65 65 4d 61 70 3a 3a 6e 65 > = BTreeMap::ne
0b10: 77 28 29 3b 0a 09 09 09 6c 65 74 20 66 65 65 64 w();....let feed
0b20: 20 3d 20 72 73 73 3a 3a 43 68 61 6e 6e 65 6c 3a = rss::Channel:
0b30: 3a 66 72 6f 6d 5f 75 72 6c 28 75 72 6c 29 0a 09 :from_url(url)..
0b40: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
0b50: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 50 72 6f (|| format!("Pro
0b60: 62 6c 65 6d 20 6f 70 65 6e 69 6e 67 20 66 65 65 blem opening fee
0b70: 64 20 75 72 6c 3a 5c 6e 7b 7d 22 2c 20 26 75 72 d url:\n{}", &ur
0b80: 6c 29 29 3f 3b 0a 09 09 09 66 6f 72 20 69 74 65 l))?;....for ite
0b90: 6d 20 69 6e 20 66 65 65 64 2e 69 74 65 6d 73 28 m in feed.items(
0ba0: 29 20 7b 0a 09 09 09 09 6c 65 74 20 64 61 74 65 ) {.....let date
0bb0: 20 3d 20 6d 61 74 63 68 20 69 74 65 6d 2e 70 75 = match item.pu
0bc0: 62 5f 64 61 74 65 28 29 20 7b 0a 09 09 09 09 09 b_date() {......
0bd0: 53 6f 6d 65 28 66 65 65 64 5f 64 61 74 65 29 20 Some(feed_date)
0be0: 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 => DateTime::par
0bf0: 73 65 5f 66 72 6f 6d 5f 72 66 63 32 38 32 32 28 se_from_rfc2822(
0c00: 66 65 65 64 5f 64 61 74 65 29 2c 0a 09 09 09 09 feed_date),.....
0c10: 09 4e 6f 6e 65 20 3d 3e 20 44 61 74 65 54 69 6d .None => DateTim
0c20: 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 e::parse_from_rf
0c30: 63 33 33 33 39 28 26 69 74 65 6d 2e 64 75 62 6c c3339(&item.dubl
0c40: 69 6e 5f 63 6f 72 65 5f 65 78 74 28 29 2e 75 6e in_core_ext().un
0c50: 77 72 61 70 28 29 2e 64 61 74 65 73 28 29 5b 30 wrap().dates()[0
0c60: 5d 29 2c 0a 09 09 09 09 7d 3f 3b 0a 09 09 09 09 ]),.....}?;.....
0c70: 6c 65 74 20 75 72 6c 20 3d 20 69 74 65 6d 2e 6c let url = item.l
0c80: 69 6e 6b 28 29 2e 75 6e 77 72 61 70 28 29 2e 74 ink().unwrap().t
0c90: 6f 5f 73 74 72 69 6e 67 28 29 3b 0a 09 09 09 09 o_string();.....
0ca0: 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 64 61 74 posts.insert(dat
0cb0: 65 2e 63 6c 6f 6e 65 28 29 2c 20 75 72 6c 2e 63 e.clone(), url.c
0cc0: 6c 6f 6e 65 28 29 29 3b 0a 09 09 09 7d 3b 0a 09 lone());....};..
0cd0: 09 09 66 6f 72 20 28 64 61 74 65 2c 20 75 72 6c ..for (date, url
0ce0: 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 65 72 28 ) in posts.iter(
0cf0: 29 20 7b 0a 09 09 09 09 6c 65 74 20 6d 75 74 20 ) {.....let mut
0d00: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
0d10: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
0d20: 0a 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 .......with_cont
0d30: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
0d40: 43 68 65 63 6b 20 70 6f 73 74 20 66 65 74 63 68 Check post fetch
0d50: 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 conn:\n{:?}", &
0d60: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 self.pool))?;...
0d70: 09 09 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c 78 ..let row = sqlx
0d80: 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 ::query("select
0d90: 65 78 69 73 74 73 28 73 65 6c 65 63 74 20 74 72 exists(select tr
0da0: 75 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 6f ue from rsstg_po
0db0: 73 74 20 77 68 65 72 65 20 75 72 6c 20 3d 20 24 st where url = $
0dc0: 31 20 61 6e 64 20 73 6f 75 72 63 65 5f 69 64 20 1 and source_id
0dd0: 3d 20 24 32 29 20 61 73 20 65 78 69 73 74 73 3b = $2) as exists;
0de0: 22 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 26 75 ").......bind(&u
0df0: 72 6c 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 2a rl).......bind(*
0e00: 69 64 29 0a 09 09 09 09 09 2e 66 65 74 63 68 5f id).......fetch_
0e10: 6f 6e 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 one(&mut conn).a
0e20: 77 61 69 74 0a 09 09 09 09 09 2e 77 69 74 68 5f wait.......with_
0e30: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
0e40: 74 21 28 22 43 68 65 63 6b 20 70 6f 73 74 3a 5c t!("Check post:\
0e50: 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 29 3f n{:?}", &conn))?
0e60: 3b 0a 09 09 09 09 6c 65 74 20 65 78 69 73 74 73 ;.....let exists
0e70: 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 : bool = row.try
0e80: 5f 67 65 74 28 22 65 78 69 73 74 73 22 29 3f 3b _get("exists")?;
0e90: 0a 09 09 09 09 69 66 20 21 20 65 78 69 73 74 73 .....if ! exists
0ea0: 20 7b 0a 09 09 09 09 09 69 66 20 74 68 69 73 5f {......if this_
0eb0: 66 65 74 63 68 20 3d 3d 20 4e 6f 6e 65 20 7c 7c fetch == None ||
0ec0: 20 2a 64 61 74 65 20 3e 20 74 68 69 73 5f 66 65 *date > this_fe
0ed0: 74 63 68 2e 75 6e 77 72 61 70 28 29 20 7b 0a 09 tch.unwrap() {..
0ee0: 09 09 09 09 09 74 68 69 73 5f 66 65 74 63 68 20 .....this_fetch
0ef0: 3d 20 53 6f 6d 65 28 2a 64 61 74 65 29 3b 0a 09 = Some(*date);..
0f00: 09 09 09 09 7d 3b 0a 09 09 09 09 09 73 65 6c 66 ....};......self
0f10: 2e 74 67 2e 73 65 6e 64 28 20 6d 61 74 63 68 20 .tg.send( match
0f20: 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 iv_hash {.......
0f30: 09 53 6f 6d 65 28 78 29 20 3d 3e 20 53 65 6e 64 .Some(x) => Send
0f40: 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 73 Message::new(des
0f50: 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 tination, format
0f60: 21 28 22 3c 61 20 68 72 65 66 3d 5c 22 68 74 74 !("<a href=\"htt
0f70: 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 72 6c ps://t.me/iv?url
0f80: 3d 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c 22 3e 20 ={}&rhash={}\">
0f90: 3c 2f 61 3e 7b 30 7d 22 2c 20 75 72 6c 2c 20 78 </a>{0}", url, x
0fa0: 29 29 2c 0a 09 09 09 09 09 09 09 4e 6f 6e 65 20 )),........None
0fb0: 3d 3e 20 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a => SendMessage::
0fc0: 6e 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c new(destination,
0fd0: 20 66 6f 72 6d 61 74 21 28 22 7b 7d 22 2c 20 75 format!("{}", u
0fe0: 72 6c 29 29 2c 0a 09 09 09 09 09 09 7d 2e 70 61 rl)),.......}.pa
0ff0: 72 73 65 5f 6d 6f 64 65 28 74 79 70 65 73 3a 3a rse_mode(types::
1000: 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 ParseMode::Html)
1010: 29 2e 61 77 61 69 74 0a 09 09 09 09 09 09 2e 63 ).await........c
1020: 6f 6e 74 65 78 74 28 22 43 61 6e 27 74 20 70 6f ontext("Can't po
1030: 73 74 20 6d 65 73 73 61 67 65 3a 22 29 3f 3b 0a st message:")?;.
1040: 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 .....sqlx::query
1050: 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 ("insert into rs
1060: 73 74 67 5f 70 6f 73 74 20 28 73 6f 75 72 63 65 stg_post (source
1070: 5f 69 64 2c 20 70 6f 73 74 65 64 2c 20 75 72 6c _id, posted, url
1080: 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 ) values ($1, $2
1090: 2c 20 24 33 29 3b 22 29 0a 09 09 09 09 09 09 2e , $3);")........
10a0: 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 09 09 bind(*id).......
10b0: 2e 62 69 6e 64 28 64 61 74 65 29 0a 09 09 09 09 .bind(date).....
10c0: 09 09 2e 62 69 6e 64 28 75 72 6c 29 0a 09 09 09 ...bind(url)....
10d0: 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 ....execute(&mut
10e0: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 conn).await....
10f0: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
1100: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 52 65 63 (|| format!("Rec
1110: 6f 72 64 20 70 6f 73 74 3a 5c 6e 7b 3a 3f 7d 22 ord post:\n{:?}"
1120: 2c 20 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 09 09 , &conn))?;.....
1130: 09 64 72 6f 70 28 63 6f 6e 6e 29 3b 0a 09 09 09 .drop(conn);....
1140: 09 09 74 6f 6b 69 6f 3a 3a 74 69 6d 65 3a 3a 64 ..tokio::time::d
1150: 65 6c 61 79 5f 66 6f 72 28 73 74 64 3a 3a 74 69 elay_for(std::ti
1160: 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 me::Duration::ne
1170: 77 28 34 2c 20 30 29 29 2e 61 77 61 69 74 3b 0a w(4, 0)).await;.
1180: 09 09 09 09 7d 3b 0a 09 09 09 7d 3b 0a 09 09 09 ....};....};....
1190: 70 6f 73 74 73 2e 63 6c 65 61 72 28 29 3b 0a 09 posts.clear();..
11a0: 09 7d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f .};...let mut co
11b0: 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 nn = self.pool.a
11c0: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 cquire().await..
11d0: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
11e0: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 || format!("Upda
11f0: 74 65 20 73 63 72 61 70 65 20 66 65 74 63 68 20 te scrape fetch
1200: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
1210: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 73 elf.pool))?;...s
1220: 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 qlx::query("upda
1230: 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 te rsstg_source
1240: 73 65 74 20 6c 61 73 74 5f 73 63 72 61 70 65 20 set last_scrape
1250: 3d 20 6e 6f 77 28 29 20 77 68 65 72 65 20 73 6f = now() where so
1260: 75 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a urce_id = $1;").
1270: 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 ....bind(*id)...
1280: 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 ..execute(&mut c
1290: 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 onn).await.....w
12a0: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
12b0: 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 73 ormat!("Update s
12c0: 63 72 61 70 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 crape:\n{:?}", &
12d0: 63 6f 6e 6e 29 29 3f 3b 0a 09 09 4f 6b 28 28 29 conn))?;...Ok(()
12e0: 29 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 66 6e 20 )..}...async fn
12f0: 64 65 6c 65 74 65 3c 53 3e 28 26 73 65 6c 66 2c delete<S>(&self,
1300: 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 source_id: &i32
1310: 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 , owner: S) -> R
1320: 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 esult<String>..w
1330: 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 here S: Into<i64
1340: 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a > {...let owner:
1350: 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 i64 = owner.int
1360: 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 o();...let mut c
1370: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e onn = self.pool.
1380: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a acquire().await.
1390: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
13a0: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 65 6c (|| format!("Del
13b0: 65 74 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c ete fetch conn:\
13c0: 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f n{:?}", &self.po
13d0: 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 ol))?;...match s
13e0: 71 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 6c 65 qlx::query("dele
13f0: 74 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f te from rsstg_so
1400: 75 72 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 urce where sourc
1410: 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 e_id = $1 and ow
1420: 6e 65 72 20 3d 20 24 32 3b 22 29 0a 09 09 09 2e ner = $2;").....
1430: 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a bind(source_id).
1440: 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a ....bind(owner).
1450: 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 ....execute(&mut
1460: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 conn).await....
1470: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
1480: 20 66 6f 72 6d 61 74 21 28 22 44 65 6c 65 74 65 format!("Delete
1490: 20 73 6f 75 72 63 65 20 72 75 6c 65 3a 5c 6e 7b source rule:\n{
14a0: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
14b0: 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 ))?.....rows_aff
14c0: 65 63 74 65 64 28 29 20 7b 0a 09 09 09 30 20 3d ected() {....0 =
14d0: 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 61 74 61 20 > { Ok("No data
14e0: 66 6f 75 6e 64 20 66 6f 75 6e 64 5c 5c 2e 22 2e found found\\.".
14f0: 74 6f 5f 73 74 72 69 6e 67 28 29 29 20 7d 2c 0a to_string()) },.
1500: 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 ...x => { Ok(for
1510: 6d 61 74 21 28 22 7b 7d 20 73 6f 75 72 63 65 73 mat!("{} sources
1520: 20 72 65 6d 6f 76 65 64 5c 5c 2e 22 2c 20 78 29 removed\\.", x)
1530: 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 ) },...}..}...as
1540: 79 6e 63 20 66 6e 20 63 6c 65 61 6e 3c 53 3e 28 ync fn clean<S>(
1550: 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 &self, source_id
1560: 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 : &i32, owner: S
1570: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 ) -> Result<Stri
1580: 6e 67 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e ng>..where S: In
1590: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 to<i64> {...let
15a0: 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 6e owner: i64 = own
15b0: 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 er.into();...let
15c0: 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 mut conn = self
15d0: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
15e0: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
15f0: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1600: 21 28 22 43 6c 65 61 6e 20 66 65 74 63 68 20 63 !("Clean fetch c
1610: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
1620: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 lf.pool))?;...ma
1630: 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 tch sqlx::query(
1640: 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 "delete from rss
1650: 74 67 5f 70 6f 73 74 20 70 20 75 73 69 6e 67 20 tg_post p using
1660: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 20 77 rsstg_source s w
1670: 68 65 72 65 20 70 2e 73 6f 75 72 63 65 5f 69 64 here p.source_id
1680: 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 = $1 and owner
1690: 3d 20 24 32 20 61 6e 64 20 70 2e 73 6f 75 72 63 = $2 and p.sourc
16a0: 65 5f 69 64 20 3d 20 73 2e 73 6f 75 72 63 65 5f e_id = s.source_
16b0: 69 64 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 id;").....bind(s
16c0: 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 ource_id).....bi
16d0: 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 nd(owner).....ex
16e0: 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 ecute(&mut conn)
16f0: 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f .await.....with_
1700: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
1710: 74 21 28 22 43 6c 65 61 6e 20 73 65 65 6e 20 70 t!("Clean seen p
1720: 6f 73 74 73 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 osts:\n{:?}", &s
1730: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e elf.pool))?.....
1740: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 rows_affected()
1750: 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 {....0 => { Ok("
1760: 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f No data found fo
1770: 75 6e 64 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e und\\.".to_strin
1780: 67 28 29 29 20 7d 2c 0a 09 09 09 78 20 3d 3e 20 g()) },....x =>
1790: 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 22 7b 7d { Ok(format!("{}
17a0: 20 70 6f 73 74 73 20 70 75 72 67 65 64 5c 5c 2e posts purged\\.
17b0: 22 2c 20 78 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d ", x)) },...}..}
17c0: 0a 0a 09 61 73 79 6e 63 20 66 6e 20 65 6e 61 62 ...async fn enab
17d0: 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 le<S>(&self, sou
17e0: 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 rce_id: &i32, ow
17f0: 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c ner: S) -> Resul
1800: 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 20 53 t<&str>..where S
1810: 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 : Into<i64> {...
1820: 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d let owner: i64 =
1830: 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 owner.into();..
1840: 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 .let mut conn =
1850: 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 self.pool.acquir
1860: 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 e().await.....wi
1870: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
1880: 72 6d 61 74 21 28 22 45 6e 61 62 6c 65 20 66 65 rmat!("Enable fe
1890: 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 tch conn:\n{:?}"
18a0: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b , &self.pool))?;
18b0: 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 ...match sqlx::q
18c0: 75 65 72 79 28 22 75 70 64 61 74 65 20 72 73 73 uery("update rss
18d0: 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 6e tg_source set en
18e0: 61 62 6c 65 64 20 3d 20 74 72 75 65 20 77 68 65 abled = true whe
18f0: 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 re source_id = $
1900: 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 1 and owner = $2
1910: 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 ").....bind(sour
1920: 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 ce_id).....bind(
1930: 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 owner).....execu
1940: 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 te(&mut conn).aw
1950: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
1960: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1970: 22 45 6e 61 62 6c 65 20 73 6f 75 72 63 65 3a 5c "Enable source:\
1980: 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f n{:?}", &self.po
1990: 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 ol))?.....rows_a
19a0: 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 31 ffected() {....1
19b0: 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 => { Ok("Source
19c0: 20 64 69 73 61 62 6c 65 64 5c 5c 2e 22 29 20 7d disabled\\.") }
19d0: 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 ,....0 => { Ok("
19e0: 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 Source not found
19f0: 5c 5c 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e \\.") },...._ =>
1a00: 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 22 { Err(anyhow!("
1a10: 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 Database error."
1a20: 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 )) },...}..}...a
1a30: 73 79 6e 63 20 66 6e 20 64 69 73 61 62 6c 65 3c sync fn disable<
1a40: 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 S>(&self, source
1a50: 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 _id: &i32, owner
1a60: 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 : S) -> Result<&
1a70: 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 str>..where S: I
1a80: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
1a90: 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 owner: i64 = ow
1aa0: 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 ner.into();...le
1ab0: 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c t mut conn = sel
1ac0: 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 f.pool.acquire()
1ad0: 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f .await.....with_
1ae0: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
1af0: 74 21 28 22 44 69 73 61 62 6c 65 20 66 65 74 63 t!("Disable fetc
1b00: 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 h conn:\n{:?}",
1b10: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 &self.pool))?;..
1b20: 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 .match sqlx::que
1b30: 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 67 ry("update rsstg
1b40: 5f 73 6f 75 72 63 65 20 73 65 74 20 65 6e 61 62 _source set enab
1b50: 6c 65 64 20 3d 20 66 61 6c 73 65 20 77 68 65 72 led = false wher
1b60: 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 e source_id = $1
1b70: 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 and owner = $2"
1b80: 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 ).....bind(sourc
1b90: 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f e_id).....bind(o
1ba0: 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 wner).....execut
1bb0: 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 e(&mut conn).awa
1bc0: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
1bd0: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1be0: 44 69 73 61 62 6c 65 20 73 6f 75 72 63 65 3a 5c Disable source:\
1bf0: 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f n{:?}", &self.po
1c00: 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 ol))?.....rows_a
1c10: 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 31 ffected() {....1
1c20: 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 => { Ok("Source
1c30: 20 64 69 73 61 62 6c 65 64 5c 5c 2e 22 29 20 7d disabled\\.") }
1c40: 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 ,....0 => { Ok("
1c50: 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 Source not found
1c60: 5c 5c 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e \\.") },...._ =>
1c70: 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 22 { Err(anyhow!("
1c80: 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 Database error."
1c90: 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 )) },...}..}...a
1ca0: 73 79 6e 63 20 66 6e 20 75 70 64 61 74 65 3c 53 sync fn update<S
1cb0: 3e 28 26 73 65 6c 66 2c 20 75 70 64 61 74 65 3a >(&self, update:
1cc0: 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c 20 63 68 Option<i32>, ch
1cd0: 61 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 63 68 61 annel: &str, cha
1ce0: 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c 20 75 72 nnel_id: i64, ur
1cf0: 6c 3a 20 26 73 74 72 2c 20 69 76 5f 68 61 73 68 l: &str, iv_hash
1d00: 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 : Option<&str>,
1d10: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
1d20: 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 ult<String>..whe
1d30: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
1d40: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 {...let owner: i
1d50: 36 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 64 = owner.into(
1d60: 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e );...let mut con
1d70: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
1d80: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
1d90: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
1da0: 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 74 | format!("Updat
1db0: 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b e fetch conn:\n{
1dc0: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
1dd0: 29 29 3f 3b 0a 0a 09 09 6d 61 74 63 68 20 6d 61 ))?;....match ma
1de0: 74 63 68 20 75 70 64 61 74 65 20 7b 0a 09 09 09 tch update {....
1df0: 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 7b 0a 09 .Some(id) => {..
1e00: 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 ....sqlx::query(
1e10: 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f "update rsstg_so
1e20: 75 72 63 65 20 73 65 74 20 63 68 61 6e 6e 65 6c urce set channel
1e30: 5f 69 64 20 3d 20 24 32 2c 20 75 72 6c 20 3d 20 _id = $2, url =
1e40: 24 33 2c 20 69 76 5f 68 61 73 68 20 3d 20 24 34 $3, iv_hash = $4
1e50: 2c 20 6f 77 6e 65 72 20 3d 20 24 35 2c 20 63 68 , owner = $5, ch
1e60: 61 6e 6e 65 6c 20 3d 20 24 36 20 77 68 65 72 65 annel = $6 where
1e70: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 22 source_id = $1"
1e80: 29 2e 62 69 6e 64 28 69 64 29 0a 09 09 09 09 7d ).bind(id).....}
1e90: 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a ,.....None => {.
1ea0: 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 .....sqlx::query
1eb0: 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 ("insert into rs
1ec0: 73 74 67 5f 73 6f 75 72 63 65 20 28 63 68 61 6e stg_source (chan
1ed0: 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f nel_id, url, iv_
1ee0: 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 hash, owner, cha
1ef0: 6e 6e 65 6c 29 20 76 61 6c 75 65 73 20 28 24 31 nnel) values ($1
1f00: 2c 20 24 32 2c 20 24 33 2c 20 24 34 2c 20 24 35 , $2, $3, $4, $5
1f10: 29 22 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a )").....},....}.
1f20: 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c ....bind(channel
1f30: 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 _id).....bind(ur
1f40: 6c 29 0a 09 09 09 2e 62 69 6e 64 28 69 76 5f 68 l).....bind(iv_h
1f50: 61 73 68 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 ash).....bind(ow
1f60: 6e 65 72 29 0a 09 09 09 2e 62 69 6e 64 28 63 68 ner).....bind(ch
1f70: 61 6e 6e 65 6c 29 0a 09 09 09 2e 65 78 65 63 75 annel).....execu
1f80: 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 te(&mut conn).aw
1f90: 61 69 74 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d ait {....Ok(_) =
1fa0: 3e 20 72 65 74 75 72 6e 20 4f 6b 28 53 74 72 69 > return Ok(Stri
1fb0: 6e 67 3a 3a 66 72 6f 6d 28 22 43 68 61 6e 6e 65 ng::from("Channe
1fc0: 6c 20 61 64 64 65 64 5c 5c 2e 22 29 29 2c 0a 09 l added\\.")),..
1fd0: 09 09 45 72 72 28 73 71 6c 78 3a 3a 45 72 72 6f ..Err(sqlx::Erro
1fe0: 72 3a 3a 44 61 74 61 62 61 73 65 28 65 72 72 29 r::Database(err)
1ff0: 29 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 74 63 68 ) => {.....match
2000: 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a 3a 3c err.downcast::<
2010: 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a 3a sqlx::postgres::
2020: 50 67 44 61 74 61 62 61 73 65 45 72 72 6f 72 3e PgDatabaseError>
2030: 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b 0a 09 ().routine() {..
2040: 09 09 09 09 53 6f 6d 65 28 22 5f 62 74 5f 63 68 ....Some("_bt_ch
2050: 65 63 6b 5f 75 6e 69 71 75 65 22 2c 20 29 20 3d eck_unique", ) =
2060: 3e 20 7b 0a 09 09 09 09 09 09 72 65 74 75 72 6e > {.......return
2070: 20 4f 6b 28 22 44 75 70 6c 69 63 61 74 65 20 6b Ok("Duplicate k
2080: 65 79 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 ey\\.".to_string
2090: 28 29 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 ())......},.....
20a0: 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 7b 0a 09 09 .Some(_) => {...
20b0: 09 09 09 09 72 65 74 75 72 6e 20 4f 6b 28 22 44 ....return Ok("D
20c0: 61 74 61 62 61 73 65 20 65 72 72 6f 72 5c 5c 2e atabase error\\.
20d0: 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 0a 09 ".to_string())..
20e0: 09 09 09 09 7d 2c 0a 09 09 09 09 09 4e 6f 6e 65 ....},......None
20f0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 72 65 74 75 => {.......retu
2100: 72 6e 20 4f 6b 28 22 4e 6f 20 64 61 74 61 62 61 rn Ok("No databa
2110: 73 65 20 65 72 72 6f 72 20 65 78 74 72 61 63 74 se error extract
2120: 65 64 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 ed\\.".to_string
2130: 28 29 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 ())......},.....
2140: 7d 3b 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 };....},....Err(
2150: 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 62 61 err) => {.....ba
2160: 69 6c 21 28 22 53 6f 72 72 79 2c 20 75 6e 6b 6e il!("Sorry, unkn
2170: 6f 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 3f own error:\n{:#?
2180: 7d 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 09 09 7d }\n", err);....}
2190: 2c 0a 09 09 7d 3b 0a 09 7d 0a 0a 09 61 73 79 6e ,...};..}...asyn
21a0: 63 20 66 6e 20 61 75 74 6f 66 65 74 63 68 28 26 c fn autofetch(&
21b0: 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c self) -> Result<
21c0: 28 29 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 ()> {...let mut
21d0: 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a delay = chrono::
21e0: 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 Duration::minute
21f0: 73 28 35 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 s(5);...let mut
2200: 6e 6f 77 3b 0a 09 09 6c 6f 6f 70 20 7b 0a 09 09 now;...loop {...
2210: 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 .let mut conn =
2220: 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 self.pool.acquir
2230: 65 28 29 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 e().await......w
2240: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
2250: 6f 72 6d 61 74 21 28 22 41 75 74 6f 66 65 74 63 ormat!("Autofetc
2260: 68 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b h fetch conn:\n{
2270: 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c :?}", &self.pool
2280: 29 29 3f 3b 0a 09 09 09 6e 6f 77 20 3d 20 63 68 ))?;....now = ch
2290: 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f 77 rono::Local::now
22a0: 28 29 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 71 ();....let mut q
22b0: 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a 71 75 65 ueue = sqlx::que
22c0: 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 ry("select sourc
22d0: 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 63 68 e_id, next_fetch
22e0: 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 73 , owner from rss
22f0: 74 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 6c tg_order natural
2300: 20 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 67 left join rsstg
2310: 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 6e 65 _source where ne
2320: 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 29 xt_fetch < now()
2330: 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 35 20 6d + interval '5 m
2340: 69 6e 75 74 65 73 27 3b 22 29 0a 09 09 09 09 2e inutes';")......
2350: 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 fetch_all(&mut c
2360: 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 onn).await?;....
2370: 66 6f 72 20 72 6f 77 20 69 6e 20 71 75 65 75 65 for row in queue
2380: 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 09 6c 65 .iter() {.....le
2390: 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 t source_id: i32
23a0: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
23b0: 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 source_id")?;...
23c0: 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 ..let owner: i64
23d0: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
23e0: 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 09 09 6c 65 owner")?;.....le
23f0: 74 20 6e 65 78 74 5f 66 65 74 63 68 3a 20 44 61 t next_fetch: Da
2400: 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 4c teTime<chrono::L
2410: 6f 63 61 6c 3e 20 3d 20 72 6f 77 2e 74 72 79 5f ocal> = row.try_
2420: 67 65 74 28 22 6e 65 78 74 5f 66 65 74 63 68 22 get("next_fetch"
2430: 29 3f 3b 0a 09 09 09 09 69 66 20 6e 65 78 74 5f )?;.....if next_
2440: 66 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a 09 09 fetch < now {...
2450: 09 09 09 2f 2f 6c 65 74 20 63 6c 6f 6e 65 20 3d ...//let clone =
2460: 20 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 3b 0a 09 self.clone();..
2470: 09 09 09 09 2f 2f 63 6c 6f 6e 65 2e 6f 77 6e 65 ....//clone.owne
2480: 72 5f 63 68 61 74 28 55 73 65 72 49 64 3a 3a 6e r_chat(UserId::n
2490: 65 77 28 6f 77 6e 65 72 29 29 3b 0a 09 09 09 09 ew(owner));.....
24a0: 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 .let clone = Cor
24b0: 65 20 7b 0a 09 09 09 09 09 09 6f 77 6e 65 72 5f e {.......owner_
24c0: 63 68 61 74 3a 20 55 73 65 72 49 64 3a 3a 6e 65 chat: UserId::ne
24d0: 77 28 6f 77 6e 65 72 29 2c 0a 09 09 09 09 09 09 w(owner),.......
24e0: 2e 2e 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 0a 09 ..self.clone()..
24f0: 09 09 09 09 7d 3b 0a 09 09 09 09 09 74 6f 6b 69 ....};......toki
2500: 6f 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 6d o::spawn(async m
2510: 6f 76 65 20 7b 0a 09 09 09 09 09 09 69 66 20 6c ove {.......if l
2520: 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c et Err(err) = cl
2530: 6f 6e 65 2e 63 68 65 63 6b 28 73 6f 75 72 63 65 one.check(source
2540: 5f 69 64 2c 20 6f 77 6e 65 72 2c 20 74 72 75 65 _id, owner, true
2550: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 ).await {.......
2560: 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 .if let Err(err)
2570: 20 3d 20 63 6c 6f 6e 65 2e 64 65 62 75 67 28 26 = clone.debug(&
2580: 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a format!("š {:
2590: 3f 7d 22 2c 20 65 72 72 29 29 20 7b 0a 09 09 09 ?}", err)) {....
25a0: 09 09 09 09 09 65 70 72 69 6e 74 6c 6e 21 28 22 .....eprintln!("
25b0: 43 68 65 63 6b 20 65 72 72 6f 72 3a 20 7b 7d 22 Check error: {}"
25c0: 2c 20 65 72 72 29 3b 0a 09 09 09 09 09 09 09 7d , err);........}
25d0: 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ;.......};......
25e0: 7d 29 3b 0a 09 09 09 09 7d 20 65 6c 73 65 20 7b });.....} else {
25f0: 0a 09 09 09 09 09 69 66 20 6e 65 78 74 5f 66 65 ......if next_fe
2600: 74 63 68 20 2d 20 6e 6f 77 20 3c 20 64 65 6c 61 tch - now < dela
2610: 79 20 7b 0a 09 09 09 09 09 09 64 65 6c 61 79 20 y {.......delay
2620: 3d 20 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 6e = next_fetch - n
2630: 6f 77 3b 0a 09 09 09 09 09 7d 0a 09 09 09 09 7d ow;......}.....}
2640: 0a 09 09 09 7d 3b 0a 09 09 09 71 75 65 75 65 2e ....};....queue.
2650: 63 6c 65 61 72 28 29 3b 0a 09 09 09 74 6f 6b 69 clear();....toki
2660: 6f 3a 3a 74 69 6d 65 3a 3a 64 65 6c 61 79 5f 66 o::time::delay_f
2670: 6f 72 28 64 65 6c 61 79 2e 74 6f 5f 73 74 64 28 or(delay.to_std(
2680: 29 3f 29 2e 61 77 61 69 74 3b 0a 09 09 09 64 65 )?).await;....de
2690: 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 lay = chrono::Du
26a0: 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 ration::minutes(
26b0: 35 29 3b 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 5);...}..}...asy
26c0: 6e 63 20 66 6e 20 6c 69 73 74 3c 53 3e 28 26 73 nc fn list<S>(&s
26d0: 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d elf, owner: S) -
26e0: 3e 20 52 65 73 75 6c 74 3c 56 65 63 3c 53 74 72 > Result<Vec<Str
26f0: 69 6e 67 3e 3e 0a 09 77 68 65 72 65 20 53 3a 20 ing>>..where S:
2700: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
2710: 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e t owner = owner.
2720: 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 into();...let mu
2730: 74 20 72 65 70 6c 79 20 3d 20 76 65 63 21 5b 5d t reply = vec![]
2740: 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ;...let mut conn
2750: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
2760: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
2770: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
2780: 20 66 6f 72 6d 61 74 21 28 22 4c 69 73 74 20 66 format!("List f
2790: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
27a0: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
27b0: 3b 0a 09 09 72 65 70 6c 79 2e 70 75 73 68 28 22 ;...reply.push("
27c0: 43 68 61 6e 6e 65 6c 73 3a 22 2e 74 6f 5f 73 74 Channels:".to_st
27d0: 72 69 6e 67 28 29 29 3b 0a 09 09 6c 65 74 20 72 ring());...let r
27e0: 6f 77 73 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 ows = sqlx::quer
27f0: 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 65 y("select source
2800: 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 65 6e _id, channel, en
2810: 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 76 5f 68 abled, url, iv_h
2820: 61 73 68 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 ash from rsstg_s
2830: 6f 75 72 63 65 20 77 68 65 72 65 20 6f 77 6e 65 ource where owne
2840: 72 20 3d 20 24 31 20 6f 72 64 65 72 20 62 79 20 r = $1 order by
2850: 73 6f 75 72 63 65 5f 69 64 22 29 0a 09 09 09 2e source_id").....
2860: 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e bind(owner).....
2870: 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 fetch_all(&mut c
2880: 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 66 onn).await?;...f
2890: 6f 72 20 72 6f 77 20 69 6e 20 72 6f 77 73 2e 69 or row in rows.i
28a0: 74 65 72 28 29 20 7b 0a 09 09 09 6c 65 74 20 73 ter() {....let s
28b0: 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 20 3d 20 ource_id: i32 =
28c0: 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 6f 75 row.try_get("sou
28d0: 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 6c 65 rce_id")?;....le
28e0: 74 20 75 73 65 72 6e 61 6d 65 3a 20 26 73 74 72 t username: &str
28f0: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
2900: 63 68 61 6e 6e 65 6c 22 29 3f 3b 0a 09 09 09 6c channel")?;....l
2910: 65 74 20 65 6e 61 62 6c 65 64 3a 20 62 6f 6f 6c et enabled: bool
2920: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
2930: 65 6e 61 62 6c 65 64 22 29 3f 3b 0a 09 09 09 6c enabled")?;....l
2940: 65 74 20 75 72 6c 3a 20 26 73 74 72 20 3d 20 72 et url: &str = r
2950: 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 22 ow.try_get("url"
2960: 29 3f 3b 0a 09 09 09 6c 65 74 20 69 76 5f 68 61 )?;....let iv_ha
2970: 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e sh: Option<&str>
2980: 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 = row.try_get("
2990: 69 76 5f 68 61 73 68 22 29 3f 3b 0a 09 09 09 72 iv_hash")?;....r
29a0: 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 eply.push(format
29b0: 21 28 22 5c 6e 5c 5c 23 ef b8 8f e2 83 a3 20 7b !("\n\\#ļøā£ {
29c0: 7d 20 5c 5c 2a ef b8 8f e2 83 a3 20 60 7b 7d 60 } \\*ļøā£ `{}`
29d0: 20 7b 7d 5c 6e f0 9f 94 97 20 60 7b 7d 60 22 2c {}\nš `{}`",
29e0: 20 73 6f 75 72 63 65 5f 69 64 2c 20 75 73 65 72 source_id, user
29f0: 6e 61 6d 65 2c 20 20 0a 09 09 09 09 6d 61 74 63 name, .....matc
2a00: 68 20 65 6e 61 62 6c 65 64 20 7b 0a 09 09 09 09 h enabled {.....
2a10: 09 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 20 .true => "š
2a20: 65 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 66 enabled",......f
2a30: 61 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 73 alse => "ā dis
2a40: 61 62 6c 65 64 22 2c 0a 09 09 09 09 7d 2c 20 75 abled",.....}, u
2a50: 72 6c 29 29 3b 0a 09 09 09 69 66 20 6c 65 74 20 rl));....if let
2a60: 53 6f 6d 65 28 68 61 73 68 29 20 3d 20 69 76 5f Some(hash) = iv_
2a70: 68 61 73 68 20 7b 0a 09 09 09 09 72 65 70 6c 79 hash {.....reply
2a80: 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 49 .push(format!("I
2a90: 56 20 60 7b 7d 60 22 2c 20 68 61 73 68 29 29 3b V `{}`", hash));
2aa0: 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 4f 6b 28 ....}...};...Ok(
2ab0: 72 65 70 6c 79 29 0a 09 7d 0a 7d 0a 0a 23 5b 74 reply)..}.}..#[t
2ac0: 6f 6b 69 6f 3a 3a 6d 61 69 6e 5d 0a 61 73 79 6e okio::main].asyn
2ad0: 63 20 66 6e 20 6d 61 69 6e 28 29 20 2d 3e 20 52 c fn main() -> R
2ae0: 65 73 75 6c 74 3c 28 29 3e 20 7b 0a 09 6c 65 74 esult<()> {..let
2af0: 20 6d 75 74 20 73 65 74 74 69 6e 67 73 20 3d 20 mut settings =
2b00: 63 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 3a 3a config::Config::
2b10: 64 65 66 61 75 6c 74 28 29 3b 0a 09 73 65 74 74 default();..sett
2b20: 69 6e 67 73 2e 6d 65 72 67 65 28 63 6f 6e 66 69 ings.merge(confi
2b30: 67 3a 3a 46 69 6c 65 3a 3a 77 69 74 68 5f 6e 61 g::File::with_na
2b40: 6d 65 28 22 72 73 73 74 67 22 29 29 3f 3b 0a 0a me("rsstg"))?;..
2b50: 09 6c 65 74 20 63 6f 72 65 20 3d 20 43 6f 72 65 .let core = Core
2b60: 3a 3a 6e 65 77 28 73 65 74 74 69 6e 67 73 29 2e ::new(settings).
2b70: 61 77 61 69 74 3f 3b 0a 0a 09 6c 65 74 20 6d 75 await?;...let mu
2b80: 74 20 73 74 72 65 61 6d 20 3d 20 63 6f 72 65 2e t stream = core.
2b90: 73 74 72 65 61 6d 28 29 3b 0a 0a 09 77 68 69 6c stream();...whil
2ba0: 65 20 6c 65 74 20 53 6f 6d 65 28 75 70 64 61 74 e let Some(updat
2bb0: 65 29 20 3d 20 73 74 72 65 61 6d 2e 6e 65 78 74 e) = stream.next
2bc0: 28 29 2e 61 77 61 69 74 20 7b 0a 09 09 69 66 20 ().await {...if
2bd0: 6c 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 68 let Err(err) = h
2be0: 61 6e 64 6c 65 28 75 70 64 61 74 65 3f 2c 20 26 andle(update?, &
2bf0: 63 6f 72 65 29 2e 61 77 61 69 74 20 7b 0a 09 09 core).await {...
2c00: 09 63 6f 72 65 2e 64 65 62 75 67 28 26 66 6f 72 .core.debug(&for
2c10: 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a 3f 7d 22 mat!("š {:?}"
2c20: 2c 20 65 72 72 29 29 3f 3b 0a 09 09 7d 3b 0a 09 , err))?;...};..
2c30: 7d 0a 0a 09 4f 6b 28 28 29 29 0a 7d 0a 0a 61 73 }...Ok(()).}..as
2c40: 79 6e 63 20 66 6e 20 68 61 6e 64 6c 65 28 75 70 ync fn handle(up
2c50: 64 61 74 65 3a 20 74 65 6c 65 67 72 61 6d 5f 62 date: telegram_b
2c60: 6f 74 3a 3a 55 70 64 61 74 65 2c 20 63 6f 72 65 ot::Update, core
2c70: 3a 20 26 43 6f 72 65 29 20 2d 3e 20 52 65 73 75 : &Core) -> Resu
2c80: 6c 74 3c 28 29 3e 20 7b 0a 09 6c 61 7a 79 5f 73 lt<()> {..lazy_s
2c90: 74 61 74 69 63 21 20 7b 0a 09 09 73 74 61 74 69 tatic! {...stati
2ca0: 63 20 72 65 66 20 52 45 5f 55 53 45 52 4e 41 4d c ref RE_USERNAM
2cb0: 45 3a 20 52 65 67 65 78 20 3d 20 52 65 67 65 78 E: Regex = Regex
2cc0: 3a 3a 6e 65 77 28 72 22 5e 40 5b 61 2d 7a 41 2d ::new(r"^@[a-zA-
2cd0: 5a 5d 5b 61 2d 7a 41 2d 5a 30 2d 39 5f 5d 2b 24 Z][a-zA-Z0-9_]+$
2ce0: 22 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 73 ").unwrap();...s
2cf0: 74 61 74 69 63 20 72 65 66 20 52 45 5f 4c 49 4e tatic ref RE_LIN
2d00: 4b 3a 20 52 65 67 65 78 20 3d 20 52 65 67 65 78 K: Regex = Regex
2d10: 3a 3a 6e 65 77 28 72 22 5e 68 74 74 70 73 3f 3a ::new(r"^https?:
2d20: 2f 2f 5b 61 2d 7a 41 2d 5a 2e 30 2d 39 2d 5d 2b //[a-zA-Z.0-9-]+
2d30: 2f 5b 2d 5f 61 2d 7a 41 2d 5a 2e 30 2d 39 2f 3f /[-_a-zA-Z.0-9/?
2d40: 3d 5d 2b 24 22 29 2e 75 6e 77 72 61 70 28 29 3b =]+$").unwrap();
2d50: 0a 09 09 73 74 61 74 69 63 20 72 65 66 20 52 45 ...static ref RE
2d60: 5f 49 56 5f 48 41 53 48 3a 20 52 65 67 65 78 20 _IV_HASH: Regex
2d70: 3d 20 52 65 67 65 78 3a 3a 6e 65 77 28 72 22 5e = Regex::new(r"^
2d80: 5b 61 2d 66 30 2d 39 5d 7b 31 34 7d 24 22 29 2e [a-f0-9]{14}$").
2d90: 75 6e 77 72 61 70 28 29 3b 0a 09 7d 0a 0a 09 6d unwrap();..}...m
2da0: 61 74 63 68 20 75 70 64 61 74 65 2e 6b 69 6e 64 atch update.kind
2db0: 20 7b 0a 09 09 55 70 64 61 74 65 4b 69 6e 64 3a {...UpdateKind:
2dc0: 3a 4d 65 73 73 61 67 65 28 6d 65 73 73 61 67 65 :Message(message
2dd0: 29 20 3d 3e 20 7b 0a 09 09 09 6c 65 74 20 6d 75 ) => {....let mu
2de0: 74 20 72 65 70 6c 79 3a 20 56 65 63 3c 53 74 72 t reply: Vec<Str
2df0: 69 6e 67 3e 20 3d 20 76 65 63 21 5b 5d 3b 0a 09 ing> = vec![];..
2e00: 09 09 6d 61 74 63 68 20 6d 65 73 73 61 67 65 2e ..match message.
2e10: 6b 69 6e 64 20 7b 0a 09 09 09 09 4d 65 73 73 61 kind {.....Messa
2e20: 67 65 4b 69 6e 64 3a 3a 54 65 78 74 20 7b 20 72 geKind::Text { r
2e30: 65 66 20 64 61 74 61 2c 20 2e 2e 20 7d 20 3d 3e ef data, .. } =>
2e40: 20 7b 0a 09 09 09 09 09 6c 65 74 20 6d 75 74 20 {......let mut
2e50: 77 6f 72 64 73 20 3d 20 64 61 74 61 2e 73 70 6c words = data.spl
2e60: 69 74 5f 77 68 69 74 65 73 70 61 63 65 28 29 3b it_whitespace();
2e70: 0a 09 09 09 09 09 6c 65 74 20 63 6d 64 20 3d 20 ......let cmd =
2e80: 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 words.next().unw
2e90: 72 61 70 28 29 3b 0a 09 09 09 09 09 6d 61 74 63 rap();......matc
2ea0: 68 20 63 6d 64 20 7b 0a 0a 2f 2f 20 73 74 61 72 h cmd {..// star
2eb0: 74 0a 0a 09 09 09 09 09 09 22 2f 73 74 61 72 74 t........"/start
2ec0: 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 72 65 " => {........re
2ed0: 70 6c 79 2e 70 75 73 68 28 22 57 65 20 61 72 65 ply.push("We are
2ee0: 20 6f 70 65 6e 5c 5c 2e 20 50 72 6f 62 61 62 6c open\\. Probabl
2ef0: 79 5c 5c 2e 20 56 69 73 69 74 20 5b 63 68 61 6e y\\. Visit [chan
2f00: 6e 65 6c 5d 28 68 74 74 70 73 3a 2f 2f 74 2e 6d nel](https://t.m
2f10: 65 2f 72 73 73 74 67 5f 62 6f 74 5f 68 65 6c 70 e/rsstg_bot_help
2f20: 2f 33 29 20 66 6f 72 20 64 65 74 61 69 6c 73 5c /3) for details\
2f30: 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 \.".to_string())
2f40: 3b 0a 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 6c ;.......},..// l
2f50: 69 73 74 0a 0a 09 09 09 09 09 09 22 2f 6c 69 73 ist........"/lis
2f60: 74 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 72 t" => {........r
2f70: 65 70 6c 79 2e 61 70 70 65 6e 64 28 26 6d 75 74 eply.append(&mut
2f80: 20 63 6f 72 65 2e 6c 69 73 74 28 6d 65 73 73 61 core.list(messa
2f90: 67 65 2e 66 72 6f 6d 2e 69 64 29 2e 61 77 61 69 ge.from.id).awai
2fa0: 74 3f 29 3b 0a 09 09 09 09 09 09 7d 2c 0a 0a 2f t?);.......},../
2fb0: 2f 20 61 64 64 0a 0a 09 09 09 09 09 09 22 2f 61 / add........"/a
2fc0: 64 64 22 20 7c 20 22 2f 75 70 64 61 74 65 22 20 dd" | "/update"
2fd0: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 => {........let
2fe0: 6d 75 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 4f mut source_id: O
2ff0: 70 74 69 6f 6e 3c 69 33 32 3e 20 3d 20 4e 6f 6e ption<i32> = Non
3000: 65 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 61 74 e;........let at
3010: 5f 6c 65 61 73 74 20 3d 20 22 52 65 71 75 69 72 _least = "Requir
3020: 65 73 20 61 74 20 6c 65 61 73 74 20 33 20 70 61 es at least 3 pa
3030: 72 61 6d 65 74 65 72 73 2e 22 3b 0a 09 09 09 09 rameters.";.....
3040: 09 09 09 69 66 20 63 6d 64 20 3d 3d 20 22 2f 75 ...if cmd == "/u
3050: 70 64 61 74 65 22 20 7b 0a 09 09 09 09 09 09 09 pdate" {........
3060: 09 6c 65 74 20 66 69 72 73 74 5f 77 6f 72 64 20 .let first_word
3070: 3d 20 77 6f 72 64 73 2e 6e 65 78 74 28 29 0a 09 = words.next()..
3080: 09 09 09 09 09 09 09 09 2e 63 6f 6e 74 65 78 74 .........context
3090: 28 61 74 5f 6c 65 61 73 74 29 3f 3b 0a 09 09 09 (at_least)?;....
30a0: 09 09 09 09 09 73 6f 75 72 63 65 5f 69 64 20 3d .....source_id =
30b0: 20 53 6f 6d 65 28 66 69 72 73 74 5f 77 6f 72 64 Some(first_word
30c0: 2e 70 61 72 73 65 3a 3a 3c 69 33 32 3e 28 29 0a .parse::<i32>().
30d0: 09 09 09 09 09 09 09 09 09 2e 77 69 74 68 5f 63 ..........with_c
30e0: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
30f0: 21 28 22 49 20 6e 65 65 64 20 61 20 6e 75 6d 62 !("I need a numb
3100: 65 72 2c 20 62 75 74 20 67 6f 74 20 7b 7d 2e 22 er, but got {}."
3110: 2c 20 66 69 72 73 74 5f 77 6f 72 64 29 29 3f 29 , first_word))?)
3120: 3b 0a 09 09 09 09 09 09 09 7d 0a 09 09 09 09 09 ;........}......
3130: 09 09 6c 65 74 20 28 63 68 61 6e 6e 65 6c 2c 20 ..let (channel,
3140: 75 72 6c 2c 20 69 76 5f 68 61 73 68 29 20 3d 20 url, iv_hash) =
3150: 28 0a 09 09 09 09 09 09 09 09 77 6f 72 64 73 2e (.........words.
3160: 6e 65 78 74 28 29 2e 63 6f 6e 74 65 78 74 28 61 next().context(a
3170: 74 5f 6c 65 61 73 74 29 3f 2c 0a 09 09 09 09 09 t_least)?,......
3180: 09 09 09 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e ...words.next().
3190: 63 6f 6e 74 65 78 74 28 61 74 5f 6c 65 61 73 74 context(at_least
31a0: 29 3f 2c 0a 09 09 09 09 09 09 09 09 77 6f 72 64 )?,.........word
31b0: 73 2e 6e 65 78 74 28 29 29 3b 0a 09 09 09 09 09 s.next());......
31c0: 09 09 69 66 20 21 20 52 45 5f 55 53 45 52 4e 41 ..if ! RE_USERNA
31d0: 4d 45 2e 69 73 5f 6d 61 74 63 68 28 26 63 68 61 ME.is_match(&cha
31e0: 6e 6e 65 6c 29 20 7b 0a 09 09 09 09 09 09 09 09 nnel) {.........
31f0: 72 65 70 6c 79 2e 70 75 73 68 28 22 55 73 65 72 reply.push("User
3200: 6e 61 6d 65 73 20 73 68 6f 75 6c 64 20 62 65 20 names should be
3210: 73 6f 6d 65 74 68 69 6e 67 20 6c 69 6b 65 20 5c something like \
3220: 22 40 5c 5c 5b 61 5c 5c 2d 7a 41 5c 5c 2d 5a 5d "@\\[a\\-zA\\-Z]
3230: 5c 5c 5b 61 5c 5c 2d 7a 41 5c 5c 2d 5a 30 5c 5c \\[a\\-zA\\-Z0\\
3240: 2d 39 5c 5c 5f 5d 2b 5c 22 2c 20 61 72 65 6e 27 -9\\_]+\", aren'
3250: 74 20 74 68 65 79 3f 22 2e 74 6f 5f 73 74 72 69 t they?".to_stri
3260: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 62 ng());.........b
3270: 61 69 6c 21 28 22 57 72 6f 6e 67 20 75 73 65 72 ail!("Wrong user
3280: 6e 61 6d 65 20 7b 3a 3f 7d 2e 22 2c 20 26 63 68 name {:?}.", &ch
3290: 61 6e 6e 65 6c 29 3b 0a 09 09 09 09 09 09 09 7d annel);........}
32a0: 0a 09 09 09 09 09 09 09 69 66 20 21 20 52 45 5f ........if ! RE_
32b0: 4c 49 4e 4b 2e 69 73 5f 6d 61 74 63 68 28 26 75 LINK.is_match(&u
32c0: 72 6c 29 20 7b 0a 09 09 09 09 09 09 09 09 72 65 rl) {.........re
32d0: 70 6c 79 2e 70 75 73 68 28 22 4c 69 6e 6b 20 73 ply.push("Link s
32e0: 68 6f 75 6c 64 20 62 65 20 6c 69 6e 6b 20 74 6f hould be link to
32f0: 20 61 74 6f 6d 2f 72 73 73 20 66 65 65 64 2c 20 atom/rss feed,
3300: 73 6f 6d 65 74 68 69 6e 67 20 6c 69 6b 65 20 5c something like \
3310: 22 68 74 74 70 73 3a 2f 2f 64 6f 6d 61 69 6e 2f "https://domain/
3320: 70 61 74 68 5c 22 5c 5c 2e 22 2e 74 6f 5f 73 74 path\"\\.".to_st
3330: 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 ring());........
3340: 09 62 61 69 6c 21 28 22 55 72 6c 3a 20 7b 3a 3f .bail!("Url: {:?
3350: 7d 22 2c 20 26 75 72 6c 29 3b 0a 09 09 09 09 09 }", &url);......
3360: 09 09 7d 0a 09 09 09 09 09 09 09 69 66 20 6c 65 ..}........if le
3370: 74 20 53 6f 6d 65 28 68 61 73 68 29 20 3d 20 69 t Some(hash) = i
3380: 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 09 v_hash {........
3390: 09 69 66 20 21 20 52 45 5f 49 56 5f 48 41 53 48 .if ! RE_IV_HASH
33a0: 2e 69 73 5f 6d 61 74 63 68 28 26 68 61 73 68 29 .is_match(&hash)
33b0: 20 7b 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c {..........repl
33c0: 79 2e 70 75 73 68 28 22 49 56 20 68 61 73 68 20 y.push("IV hash
33d0: 73 68 6f 75 6c 64 20 62 65 20 31 34 20 68 65 78 should be 14 hex
33e0: 20 64 69 67 69 74 73 2e 22 2e 74 6f 5f 73 74 72 digits.".to_str
33f0: 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 ing());.........
3400: 09 62 61 69 6c 21 28 22 49 56 3a 20 7b 3a 3f 7d .bail!("IV: {:?}
3410: 22 2c 20 26 69 76 5f 68 61 73 68 29 3b 0a 09 09 ", &iv_hash);...
3420: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 ......};........
3430: 7d 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 63 68 };........let ch
3440: 61 6e 6e 65 6c 5f 69 64 20 3d 20 69 36 34 3a 3a annel_id = i64::
3450: 66 72 6f 6d 28 63 6f 72 65 2e 74 67 2e 73 65 6e from(core.tg.sen
3460: 64 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a d(telegram_bot::
3470: 47 65 74 43 68 61 74 3a 3a 6e 65 77 28 74 65 6c GetChat::new(tel
3480: 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 egram_bot::types
3490: 3a 3a 43 68 61 74 52 65 66 3a 3a 43 68 61 6e 6e ::ChatRef::Chann
34a0: 65 6c 55 73 65 72 6e 61 6d 65 28 63 68 61 6e 6e elUsername(chann
34b0: 65 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 29 el.to_string()))
34c0: 29 2e 61 77 61 69 74 3f 2e 69 64 28 29 29 3b 0a ).await?.id());.
34d0: 09 09 09 09 09 09 09 6c 65 74 20 63 68 61 6e 5f .......let chan_
34e0: 61 64 6d 20 3d 20 63 6f 72 65 2e 74 67 2e 73 65 adm = core.tg.se
34f0: 6e 64 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a nd(telegram_bot:
3500: 3a 47 65 74 43 68 61 74 41 64 6d 69 6e 69 73 74 :GetChatAdminist
3510: 72 61 74 6f 72 73 3a 3a 6e 65 77 28 74 65 6c 65 rators::new(tele
3520: 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a gram_bot::types:
3530: 3a 43 68 61 74 52 65 66 3a 3a 43 68 61 6e 6e 65 :ChatRef::Channe
3540: 6c 55 73 65 72 6e 61 6d 65 28 63 68 61 6e 6e 65 lUsername(channe
3550: 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 29 29 l.to_string())))
3560: 2e 61 77 61 69 74 0a 09 09 09 09 09 09 09 09 2e .await..........
3570: 63 6f 6e 74 65 78 74 28 22 53 6f 72 72 79 2c 20 context("Sorry,
3580: 49 20 68 61 76 65 20 6e 6f 20 61 63 63 65 73 73 I have no access
3590: 20 74 6f 20 74 68 61 74 20 63 68 61 74 5c 5c 2e to that chat\\.
35a0: 22 29 3f 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 ")?;........let
35b0: 28 6d 75 74 20 6d 65 2c 20 6d 75 74 20 75 73 65 (mut me, mut use
35c0: 72 29 20 3d 20 28 66 61 6c 73 65 2c 20 66 61 6c r) = (false, fal
35d0: 73 65 29 3b 0a 09 09 09 09 09 09 09 66 6f 72 20 se);........for
35e0: 61 64 6d 69 6e 20 69 6e 20 63 68 61 6e 5f 61 64 admin in chan_ad
35f0: 6d 20 7b 0a 09 09 09 09 09 09 09 09 69 66 20 61 m {.........if a
3600: 64 6d 69 6e 2e 75 73 65 72 2e 69 64 20 3d 3d 20 dmin.user.id ==
3610: 63 6f 72 65 2e 6d 79 2e 69 64 20 7b 0a 09 09 09 core.my.id {....
3620: 09 09 09 09 09 09 6d 65 20 3d 20 74 72 75 65 3b ......me = true;
3630: 0a 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 .........};.....
3640: 09 09 09 09 69 66 20 61 64 6d 69 6e 2e 75 73 65 ....if admin.use
3650: 72 2e 69 64 20 3d 3d 20 6d 65 73 73 61 67 65 2e r.id == message.
3660: 66 72 6f 6d 2e 69 64 20 7b 0a 09 09 09 09 09 09 from.id {.......
3670: 09 09 09 75 73 65 72 20 3d 20 74 72 75 65 3b 0a ...user = true;.
3680: 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ........};......
3690: 09 09 7d 3b 0a 09 09 09 09 09 09 09 69 66 20 21 ..};........if !
36a0: 20 6d 65 20 20 20 7b 20 62 61 69 6c 21 28 22 49 me { bail!("I
36b0: 20 6e 65 65 64 20 74 6f 20 62 65 20 61 64 6d 69 need to be admi
36c0: 6e 20 6f 6e 20 74 68 61 74 20 63 68 61 6e 6e 65 n on that channe
36d0: 6c 5c 5c 2e 22 29 3b 20 7d 3b 0a 09 09 09 09 09 l\\."); };......
36e0: 09 09 69 66 20 21 20 75 73 65 72 20 7b 20 62 61 ..if ! user { ba
36f0: 69 6c 21 28 22 59 6f 75 20 73 68 6f 75 6c 64 20 il!("You should
3700: 62 65 20 61 64 6d 69 6e 20 6f 6e 20 74 68 61 74 be admin on that
3710: 20 63 68 61 6e 6e 65 6c 5c 5c 2e 22 29 3b 20 7d channel\\."); }
3720: 3b 0a 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 ;........reply.p
3730: 75 73 68 28 63 6f 72 65 2e 75 70 64 61 74 65 28 ush(core.update(
3740: 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e source_id, chann
3750: 65 6c 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 el, channel_id,
3760: 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6d 65 url, iv_hash, me
3770: 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 29 2e 61 ssage.from.id).a
3780: 77 61 69 74 3f 29 3b 0a 09 09 09 09 09 09 7d 2c wait?);.......},
3790: 0a 0a 2f 2f 20 63 68 65 63 6b 0a 0a 09 09 09 09 ..// check......
37a0: 09 09 22 2f 63 68 65 63 6b 22 20 3d 3e 20 7b 0a .."/check" => {.
37b0: 09 09 09 09 09 09 09 6d 61 74 63 68 20 26 77 6f .......match &wo
37c0: 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 rds.next().unwra
37d0: 70 28 29 2e 70 61 72 73 65 3a 3a 3c 69 33 32 3e p().parse::<i32>
37e0: 28 29 20 7b 0a 09 09 09 09 09 09 09 09 45 72 72 () {.........Err
37f0: 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 (err) => {......
3800: 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 ....reply.push(f
3810: 6f 72 6d 61 74 21 28 22 49 20 6e 65 65 64 20 61 ormat!("I need a
3820: 20 6e 75 6d 62 65 72 5c 5c 2e 5c 6e 7b 7d 22 2c number\\.\n{}",
3830: 20 26 65 72 72 29 29 3b 0a 09 09 09 09 09 09 09 &err));........
3840: 09 7d 2c 0a 09 09 09 09 09 09 09 09 4f 6b 28 6e .},.........Ok(n
3850: 75 6d 62 65 72 29 20 3d 3e 20 7b 0a 09 09 09 09 umber) => {.....
3860: 09 09 09 09 09 63 6f 72 65 2e 63 68 65 63 6b 28 .....core.check(
3870: 2a 6e 75 6d 62 65 72 2c 20 6d 65 73 73 61 67 65 *number, message
3880: 2e 66 72 6f 6d 2e 69 64 2c 20 66 61 6c 73 65 29 .from.id, false)
3890: 2e 61 77 61 69 74 0a 09 09 09 09 09 09 09 09 09 .await..........
38a0: 09 2e 63 6f 6e 74 65 78 74 28 22 43 68 61 6e 6e ..context("Chann
38b0: 65 6c 20 63 68 65 63 6b 20 66 61 69 6c 65 64 2e el check failed.
38c0: 22 29 3f 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a ")?;.........},.
38d0: 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 .......};.......
38e0: 7d 2c 0a 0a 2f 2f 20 63 6c 65 61 6e 0a 0a 09 09 },..// clean....
38f0: 09 09 09 09 22 2f 63 6c 65 61 6e 22 20 3d 3e 20 ...."/clean" =>
3900: 7b 0a 09 09 09 09 09 09 09 6d 61 74 63 68 20 26 {........match &
3910: 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 words.next().unw
3920: 72 61 70 28 29 2e 70 61 72 73 65 3a 3a 3c 69 33 rap().parse::<i3
3930: 32 3e 28 29 20 7b 0a 09 09 09 09 09 09 09 09 45 2>() {.........E
3940: 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 rr(err) => {....
3950: 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 ......reply.push
3960: 28 66 6f 72 6d 61 74 21 28 22 49 20 6e 65 65 64 (format!("I need
3970: 20 61 20 6e 75 6d 62 65 72 5c 5c 2e 5c 6e 7b 7d a number\\.\n{}
3980: 22 2c 20 26 65 72 72 29 29 3b 0a 09 09 09 09 09 ", &err));......
3990: 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 4f 6b ...},.........Ok
39a0: 28 6e 75 6d 62 65 72 29 20 3d 3e 20 7b 0a 09 09 (number) => {...
39b0: 09 09 09 09 09 09 09 6c 65 74 20 72 65 73 75 6c .......let resul
39c0: 74 20 3d 20 63 6f 72 65 2e 63 6c 65 61 6e 28 26 t = core.clean(&
39d0: 6e 75 6d 62 65 72 2c 20 6d 65 73 73 61 67 65 2e number, message.
39e0: 66 72 6f 6d 2e 69 64 29 2e 61 77 61 69 74 3f 3b from.id).await?;
39f0: 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e ..........reply.
3a00: 70 75 73 68 28 72 65 73 75 6c 74 2e 74 6f 5f 73 push(result.to_s
3a10: 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 tring());.......
3a20: 09 09 7d 2c 0a 09 09 09 09 09 09 09 7d 3b 0a 09 ..},........};..
3a30: 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 65 6e 61 62 .....},..// enab
3a40: 6c 65 0a 0a 09 09 09 09 09 09 22 2f 65 6e 61 62 le........"/enab
3a50: 6c 65 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 le" => {........
3a60: 6d 61 74 63 68 20 26 77 6f 72 64 73 2e 6e 65 78 match &words.nex
3a70: 74 28 29 2e 75 6e 77 72 61 70 28 29 2e 70 61 72 t().unwrap().par
3a80: 73 65 3a 3a 3c 69 33 32 3e 28 29 20 7b 0a 09 09 se::<i32>() {...
3a90: 09 09 09 09 09 09 45 72 72 28 65 72 72 29 20 3d ......Err(err) =
3aa0: 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 72 65 70 > {..........rep
3ab0: 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 ly.push(format!(
3ac0: 22 49 20 6e 65 65 64 20 61 20 6e 75 6d 62 65 72 "I need a number
3ad0: 5c 5c 2e 5c 6e 7b 7d 22 2c 20 26 65 72 72 29 29 \\.\n{}", &err))
3ae0: 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 ;.........},....
3af0: 09 09 09 09 09 4f 6b 28 6e 75 6d 62 65 72 29 20 .....Ok(number)
3b00: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 6c 65 => {..........le
3b10: 74 20 72 65 73 75 6c 74 20 3d 20 63 6f 72 65 2e t result = core.
3b20: 65 6e 61 62 6c 65 28 26 6e 75 6d 62 65 72 2c 20 enable(&number,
3b30: 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 29 message.from.id)
3b40: 2e 61 77 61 69 74 3f 3b 0a 09 09 09 09 09 09 09 .await?;........
3b50: 09 09 72 65 70 6c 79 2e 70 75 73 68 28 72 65 73 ..reply.push(res
3b60: 75 6c 74 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 ult.to_string())
3b70: 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 ;.........},....
3b80: 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 2c 0a ....};.......},.
3b90: 0a 2f 2f 20 64 65 6c 65 74 65 0a 0a 09 09 09 09 .// delete......
3ba0: 09 09 22 2f 64 65 6c 65 74 65 22 20 3d 3e 20 7b .."/delete" => {
3bb0: 0a 09 09 09 09 09 09 09 6d 61 74 63 68 20 26 77 ........match &w
3bc0: 6f 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 ords.next().unwr
3bd0: 61 70 28 29 2e 70 61 72 73 65 3a 3a 3c 69 33 32 ap().parse::<i32
3be0: 3e 28 29 20 7b 0a 09 09 09 09 09 09 09 09 45 72 >() {.........Er
3bf0: 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 r(err) => {.....
3c00: 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 .....reply.push(
3c10: 66 6f 72 6d 61 74 21 28 22 49 20 6e 65 65 64 20 format!("I need
3c20: 61 20 6e 75 6d 62 65 72 5c 5c 2e 5c 6e 7b 7d 22 a number\\.\n{}"
3c30: 2c 20 26 65 72 72 29 29 3b 0a 09 09 09 09 09 09 , &err));.......
3c40: 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 4f 6b 28 ..},.........Ok(
3c50: 6e 75 6d 62 65 72 29 20 3d 3e 20 7b 0a 09 09 09 number) => {....
3c60: 09 09 09 09 09 09 6c 65 74 20 72 65 73 75 6c 74 ......let result
3c70: 20 3d 20 63 6f 72 65 2e 64 65 6c 65 74 65 28 26 = core.delete(&
3c80: 6e 75 6d 62 65 72 2c 20 6d 65 73 73 61 67 65 2e number, message.
3c90: 66 72 6f 6d 2e 69 64 29 2e 61 77 61 69 74 3f 3b from.id).await?;
3ca0: 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e ..........reply.
3cb0: 70 75 73 68 28 72 65 73 75 6c 74 2e 74 6f 5f 73 push(result.to_s
3cc0: 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 tring());.......
3cd0: 09 09 7d 2c 0a 09 09 09 09 09 09 09 7d 3b 0a 09 ..},........};..
3ce0: 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 64 69 73 61 .....},..// disa
3cf0: 62 6c 65 0a 0a 09 09 09 09 09 09 22 2f 64 69 73 ble........"/dis
3d00: 61 62 6c 65 22 20 3d 3e 20 7b 0a 09 09 09 09 09 able" => {......
3d10: 09 09 6d 61 74 63 68 20 26 77 6f 72 64 73 2e 6e ..match &words.n
3d20: 65 78 74 28 29 2e 75 6e 77 72 61 70 28 29 2e 70 ext().unwrap().p
3d30: 61 72 73 65 3a 3a 3c 69 33 32 3e 28 29 20 7b 0a arse::<i32>() {.
3d40: 09 09 09 09 09 09 09 09 45 72 72 28 65 72 72 29 ........Err(err)
3d50: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 72 => {..........r
3d60: 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 eply.push(format
3d70: 21 28 22 49 20 6e 65 65 64 20 61 20 6e 75 6d 62 !("I need a numb
3d80: 65 72 5c 5c 2e 5c 6e 7b 7d 22 2c 20 26 65 72 72 er\\.\n{}", &err
3d90: 29 29 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 ));.........},..
3da0: 09 09 09 09 09 09 09 4f 6b 28 6e 75 6d 62 65 72 .......Ok(number
3db0: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 ) => {..........
3dc0: 6c 65 74 20 72 65 73 75 6c 74 20 3d 20 63 6f 72 let result = cor
3dd0: 65 2e 64 69 73 61 62 6c 65 28 26 6e 75 6d 62 65 e.disable(&numbe
3de0: 72 2c 20 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e r, message.from.
3df0: 69 64 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 09 id).await?;.....
3e00: 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 .....reply.push(
3e10: 72 65 73 75 6c 74 2e 74 6f 5f 73 74 72 69 6e 67 result.to_string
3e20: 28 29 29 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a ());.........},.
3e30: 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 .......};.......
3e40: 7d 2c 0a 0a 09 09 09 09 09 09 5f 20 3d 3e 20 7b },........_ => {
3e50: 0a 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 7d .......},......}
3e60: 3b 0a 09 09 09 09 7d 2c 0a 09 09 09 09 5f 20 3d ;.....},....._ =
3e70: 3e 20 7b 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 3b > {.....},....};
3e80: 0a 0a 09 09 09 69 66 20 72 65 70 6c 79 2e 6c 65 .....if reply.le
3e90: 6e 28 29 20 3e 20 30 20 7b 0a 09 09 09 09 69 66 n() > 0 {.....if
3ea0: 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 let Err(err) =
3eb0: 63 6f 72 65 2e 74 67 2e 73 65 6e 64 28 6d 65 73 core.tg.send(mes
3ec0: 73 61 67 65 2e 74 65 78 74 5f 72 65 70 6c 79 28 sage.text_reply(
3ed0: 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 reply.join("\n")
3ee0: 29 2e 70 61 72 73 65 5f 6d 6f 64 65 28 74 79 70 ).parse_mode(typ
3ef0: 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 4d es::ParseMode::M
3f00: 61 72 6b 64 6f 77 6e 56 32 29 29 2e 61 77 61 69 arkdownV2)).awai
3f10: 74 20 7b 0a 09 09 09 09 09 64 62 67 21 28 72 65 t {......dbg!(re
3f20: 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 3b ply.join("\n"));
3f30: 0a 09 09 09 09 09 70 72 69 6e 74 6c 6e 21 28 22 ......println!("
3f40: 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 09 7d {}", err);.....}
3f50: 3b 0a 09 09 09 7d 3b 0a 09 09 7d 2c 0a 09 09 5f ;....};...},..._
3f60: 20 3d 3e 20 7b 7d 2c 0a 09 7d 3b 0a 0a 09 4f 6b => {},..};...Ok
3f70: 28 28 29 29 0a 7d 0a (()).}.