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 65 6e 61 62 6c 65 64 5c 5c 2e 22 29 20 7d 2c enabled\\.") },
19d0: 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 ....0 => { Ok("S
19e0: 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 5c ource not found\
19f0: 5c 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 \.") },...._ =>
1a00: 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 { Err(anyhow!("D
1a10: 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 atabase error.")
1a20: 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 ) },...}..}...as
1a30: 79 6e 63 20 66 6e 20 64 69 73 61 62 6c 65 3c 53 ync fn disable<S
1a40: 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f >(&self, source_
1a50: 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a id: &i32, owner:
1a60: 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 S) -> Result<&s
1a70: 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e tr>..where S: In
1a80: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 to<i64> {...let
1a90: 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 6e owner: i64 = own
1aa0: 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 er.into();...let
1ab0: 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 mut conn = self
1ac0: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
1ad0: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
1ae0: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1af0: 21 28 22 44 69 73 61 62 6c 65 20 66 65 74 63 68 !("Disable fetch
1b00: 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 conn:\n{:?}", &
1b10: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 self.pool))?;...
1b20: 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 match sqlx::quer
1b30: 79 28 22 75 70 64 61 74 65 20 72 73 73 74 67 5f y("update rsstg_
1b40: 73 6f 75 72 63 65 20 73 65 74 20 65 6e 61 62 6c source set enabl
1b50: 65 64 20 3d 20 66 61 6c 73 65 20 77 68 65 72 65 ed = false where
1b60: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 source_id = $1
1b70: 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 and owner = $2")
1b80: 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 .....bind(source
1b90: 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 _id).....bind(ow
1ba0: 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 65 ner).....execute
1bb0: 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 (&mut conn).awai
1bc0: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
1bd0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 xt(|| format!("D
1be0: 69 73 61 62 6c 65 20 73 6f 75 72 63 65 3a 5c 6e isable source:\n
1bf0: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
1c00: 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 66 l))?.....rows_af
1c10: 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 31 20 fected() {....1
1c20: 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 => { Ok("Source
1c30: 64 69 73 61 62 6c 65 64 5c 5c 2e 22 29 20 7d 2c disabled\\.") },
1c40: 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 ....0 => { Ok("S
1c50: 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 5c ource not found\
1c60: 5c 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 \.") },...._ =>
1c70: 7b 20 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 { Err(anyhow!("D
1c80: 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 atabase error.")
1c90: 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 ) },...}..}...as
1ca0: 79 6e 63 20 66 6e 20 75 70 64 61 74 65 3c 53 3e ync fn update<S>
1cb0: 28 26 73 65 6c 66 2c 20 75 70 64 61 74 65 3a 20 (&self, update:
1cc0: 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c 20 63 68 61 Option<i32>, cha
1cd0: 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 63 68 61 6e nnel: &str, chan
1ce0: 6e 65 6c 5f 69 64 3a 20 69 36 34 2c 20 75 72 6c nel_id: i64, url
1cf0: 3a 20 26 73 74 72 2c 20 69 76 5f 68 61 73 68 3a : &str, iv_hash:
1d00: 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 6f Option<&str>, o
1d10: 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 wner: S) -> Resu
1d20: 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 72 lt<String>..wher
1d30: 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e S: Into<i64> {
1d40: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 ...let owner: i6
1d50: 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 4 = owner.into()
1d60: 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ;...let mut conn
1d70: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
1d80: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
1d90: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
1da0: 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 format!("Update
1db0: 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a fetch conn:\n{:
1dc0: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
1dd0: 29 3f 3b 0a 0a 09 09 6d 61 74 63 68 20 6d 61 74 )?;....match mat
1de0: 63 68 20 75 70 64 61 74 65 20 7b 0a 09 09 09 09 ch update {.....
1df0: 53 6f 6d 65 28 69 64 29 20 3d 3e 20 7b 0a 09 09 Some(id) => {...
1e00: 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ...sqlx::query("
1e10: 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 update rsstg_sou
1e20: 72 63 65 20 73 65 74 20 63 68 61 6e 6e 65 6c 5f rce set channel_
1e30: 69 64 20 3d 20 24 32 2c 20 75 72 6c 20 3d 20 24 id = $2, url = $
1e40: 33 2c 20 69 76 5f 68 61 73 68 20 3d 20 24 34 2c 3, iv_hash = $4,
1e50: 20 6f 77 6e 65 72 20 3d 20 24 35 2c 20 63 68 61 owner = $5, cha
1e60: 6e 6e 65 6c 20 3d 20 24 36 20 77 68 65 72 65 20 nnel = $6 where
1e70: 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 22 29 source_id = $1")
1e80: 2e 62 69 6e 64 28 69 64 29 0a 09 09 09 09 7d 2c .bind(id).....},
1e90: 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 .....None => {..
1ea0: 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 ....sqlx::query(
1eb0: 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 73 "insert into rss
1ec0: 74 67 5f 73 6f 75 72 63 65 20 28 63 68 61 6e 6e tg_source (chann
1ed0: 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 el_id, url, iv_h
1ee0: 61 73 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 6e ash, owner, chan
1ef0: 6e 65 6c 29 20 76 61 6c 75 65 73 20 28 24 31 2c nel) values ($1,
1f00: 20 24 32 2c 20 24 33 2c 20 24 34 2c 20 24 35 29 $2, $3, $4, $5)
1f10: 22 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 ").....},....}..
1f20: 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c 5f ...bind(channel_
1f30: 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c id).....bind(url
1f40: 29 0a 09 09 09 2e 62 69 6e 64 28 69 76 5f 68 61 ).....bind(iv_ha
1f50: 73 68 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e sh).....bind(own
1f60: 65 72 29 0a 09 09 09 2e 62 69 6e 64 28 63 68 61 er).....bind(cha
1f70: 6e 6e 65 6c 29 0a 09 09 09 2e 65 78 65 63 75 74 nnel).....execut
1f80: 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 e(&mut conn).awa
1f90: 69 74 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e it {....Ok(_) =>
1fa0: 20 72 65 74 75 72 6e 20 4f 6b 28 53 74 72 69 6e return Ok(Strin
1fb0: 67 3a 3a 66 72 6f 6d 28 22 43 68 61 6e 6e 65 6c g::from("Channel
1fc0: 20 61 64 64 65 64 5c 5c 2e 22 29 29 2c 0a 09 09 added\\.")),...
1fd0: 09 45 72 72 28 73 71 6c 78 3a 3a 45 72 72 6f 72 .Err(sqlx::Error
1fe0: 3a 3a 44 61 74 61 62 61 73 65 28 65 72 72 29 29 ::Database(err))
1ff0: 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 74 63 68 20 => {.....match
2000: 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a 3a 3c 73 err.downcast::<s
2010: 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a 3a 50 qlx::postgres::P
2020: 67 44 61 74 61 62 61 73 65 45 72 72 6f 72 3e 28 gDatabaseError>(
2030: 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b 0a 09 09 ).routine() {...
2040: 09 09 09 53 6f 6d 65 28 22 5f 62 74 5f 63 68 65 ...Some("_bt_che
2050: 63 6b 5f 75 6e 69 71 75 65 22 2c 20 29 20 3d 3e ck_unique", ) =>
2060: 20 7b 0a 09 09 09 09 09 09 72 65 74 75 72 6e 20 {.......return
2070: 4f 6b 28 22 44 75 70 6c 69 63 61 74 65 20 6b 65 Ok("Duplicate ke
2080: 79 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 y\\.".to_string(
2090: 29 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ))......},......
20a0: 53 6f 6d 65 28 5f 29 20 3d 3e 20 7b 0a 09 09 09 Some(_) => {....
20b0: 09 09 09 72 65 74 75 72 6e 20 4f 6b 28 22 44 61 ...return Ok("Da
20c0: 74 61 62 61 73 65 20 65 72 72 6f 72 5c 5c 2e 22 tabase error\\."
20d0: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 0a 09 09 .to_string())...
20e0: 09 09 09 7d 2c 0a 09 09 09 09 09 4e 6f 6e 65 20 ...},......None
20f0: 3d 3e 20 7b 0a 09 09 09 09 09 09 72 65 74 75 72 => {.......retur
2100: 6e 20 4f 6b 28 22 4e 6f 20 64 61 74 61 62 61 73 n Ok("No databas
2110: 65 20 65 72 72 6f 72 20 65 78 74 72 61 63 74 65 e error extracte
2120: 64 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 d\\.".to_string(
2130: 29 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 7d ))......},.....}
2140: 3b 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 65 ;....},....Err(e
2150: 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 62 61 69 rr) => {.....bai
2160: 6c 21 28 22 53 6f 72 72 79 2c 20 75 6e 6b 6e 6f l!("Sorry, unkno
2170: 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 3f 7d wn error:\n{:#?}
2180: 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 09 09 7d 2c \n", err);....},
2190: 0a 09 09 7d 3b 0a 09 7d 0a 0a 09 61 73 79 6e 63 ...};..}...async
21a0: 20 66 6e 20 61 75 74 6f 66 65 74 63 68 28 26 73 fn autofetch(&s
21b0: 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 elf) -> Result<(
21c0: 29 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 64 )> {...let mut d
21d0: 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 elay = chrono::D
21e0: 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 uration::minutes
21f0: 28 35 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 6e (5);...let mut n
2200: 6f 77 3b 0a 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 ow;...loop {....
2210: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
2220: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
2230: 28 29 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 ().await......wi
2240: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
2250: 72 6d 61 74 21 28 22 41 75 74 6f 66 65 74 63 68 rmat!("Autofetch
2260: 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a fetch conn:\n{:
2270: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
2280: 29 3f 3b 0a 09 09 09 6e 6f 77 20 3d 20 63 68 72 )?;....now = chr
2290: 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f 77 28 ono::Local::now(
22a0: 29 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 71 75 );....let mut qu
22b0: 65 75 65 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 eue = sqlx::quer
22c0: 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 65 y("select source
22d0: 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 63 68 2c _id, next_fetch,
22e0: 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 73 74 owner from rsst
22f0: 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 6c 20 g_order natural
2300: 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 67 5f left join rsstg_
2310: 73 6f 75 72 63 65 20 77 68 65 72 65 20 6e 65 78 source where nex
2320: 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 29 20 t_fetch < now()
2330: 2b 20 69 6e 74 65 72 76 61 6c 20 27 35 20 6d 69 + interval '5 mi
2340: 6e 75 74 65 73 27 3b 22 29 0a 09 09 09 09 2e 66 nutes';")......f
2350: 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 6f etch_all(&mut co
2360: 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 66 nn).await?;....f
2370: 6f 72 20 72 6f 77 20 69 6e 20 71 75 65 75 65 2e or row in queue.
2380: 69 74 65 72 28 29 20 7b 0a 09 09 09 09 6c 65 74 iter() {.....let
2390: 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 20 source_id: i32
23a0: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 = row.try_get("s
23b0: 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 ource_id")?;....
23c0: 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 .let owner: i64
23d0: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 6f = row.try_get("o
23e0: 77 6e 65 72 22 29 3f 3b 0a 09 09 09 09 6c 65 74 wner")?;.....let
23f0: 20 6e 65 78 74 5f 66 65 74 63 68 3a 20 44 61 74 next_fetch: Dat
2400: 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 4c 6f eTime<chrono::Lo
2410: 63 61 6c 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 cal> = row.try_g
2420: 65 74 28 22 6e 65 78 74 5f 66 65 74 63 68 22 29 et("next_fetch")
2430: 3f 3b 0a 09 09 09 09 69 66 20 6e 65 78 74 5f 66 ?;.....if next_f
2440: 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a 09 09 09 etch < now {....
2450: 09 09 2f 2f 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 ..//let clone =
2460: 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 self.clone();...
2470: 09 09 09 2f 2f 63 6c 6f 6e 65 2e 6f 77 6e 65 72 ...//clone.owner
2480: 5f 63 68 61 74 28 55 73 65 72 49 64 3a 3a 6e 65 _chat(UserId::ne
2490: 77 28 6f 77 6e 65 72 29 29 3b 0a 09 09 09 09 09 w(owner));......
24a0: 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 65 let clone = Core
24b0: 20 7b 0a 09 09 09 09 09 09 6f 77 6e 65 72 5f 63 {.......owner_c
24c0: 68 61 74 3a 20 55 73 65 72 49 64 3a 3a 6e 65 77 hat: UserId::new
24d0: 28 6f 77 6e 65 72 29 2c 0a 09 09 09 09 09 09 2e (owner),........
24e0: 2e 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 0a 09 09 .self.clone()...
24f0: 09 09 09 7d 3b 0a 09 09 09 09 09 74 6f 6b 69 6f ...};......tokio
2500: 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 6d 6f ::spawn(async mo
2510: 76 65 20 7b 0a 09 09 09 09 09 09 69 66 20 6c 65 ve {.......if le
2520: 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c 6f t Err(err) = clo
2530: 6e 65 2e 63 68 65 63 6b 28 73 6f 75 72 63 65 5f ne.check(source_
2540: 69 64 2c 20 6f 77 6e 65 72 2c 20 74 72 75 65 29 id, owner, true)
2550: 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 09 .await {........
2560: 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 if let Err(err)
2570: 3d 20 63 6c 6f 6e 65 2e 64 65 62 75 67 28 26 66 = clone.debug(&f
2580: 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a 3f ormat!("š {:?
2590: 7d 22 2c 20 65 72 72 29 29 20 7b 0a 09 09 09 09 }", err)) {.....
25a0: 09 09 09 09 65 70 72 69 6e 74 6c 6e 21 28 22 43 ....eprintln!("C
25b0: 68 65 63 6b 20 65 72 72 6f 72 3a 20 7b 7d 22 2c heck error: {}",
25c0: 20 65 72 72 29 3b 0a 09 09 09 09 09 09 09 7d 3b err);........};
25d0: 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 7d .......};......}
25e0: 29 3b 0a 09 09 09 09 7d 20 65 6c 73 65 20 7b 0a );.....} else {.
25f0: 09 09 09 09 09 69 66 20 6e 65 78 74 5f 66 65 74 .....if next_fet
2600: 63 68 20 2d 20 6e 6f 77 20 3c 20 64 65 6c 61 79 ch - now < delay
2610: 20 7b 0a 09 09 09 09 09 09 64 65 6c 61 79 20 3d {.......delay =
2620: 20 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f next_fetch - no
2630: 77 3b 0a 09 09 09 09 09 7d 0a 09 09 09 09 7d 0a w;......}.....}.
2640: 09 09 09 7d 3b 0a 09 09 09 71 75 65 75 65 2e 63 ...};....queue.c
2650: 6c 65 61 72 28 29 3b 0a 09 09 09 74 6f 6b 69 6f lear();....tokio
2660: 3a 3a 74 69 6d 65 3a 3a 64 65 6c 61 79 5f 66 6f ::time::delay_fo
2670: 72 28 64 65 6c 61 79 2e 74 6f 5f 73 74 64 28 29 r(delay.to_std()
2680: 3f 29 2e 61 77 61 69 74 3b 0a 09 09 09 64 65 6c ?).await;....del
2690: 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 72 ay = chrono::Dur
26a0: 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 35 ation::minutes(5
26b0: 29 3b 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e );...}..}...asyn
26c0: 63 20 66 6e 20 6c 69 73 74 3c 53 3e 28 26 73 65 c fn list<S>(&se
26d0: 6c 66 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e lf, owner: S) ->
26e0: 20 52 65 73 75 6c 74 3c 56 65 63 3c 53 74 72 69 Result<Vec<Stri
26f0: 6e 67 3e 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 ng>>..where S: I
2700: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
2710: 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 owner = owner.i
2720: 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 nto();...let mut
2730: 20 72 65 70 6c 79 20 3d 20 76 65 63 21 5b 5d 3b reply = vec![];
2740: 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
2750: 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 = self.pool.acqu
2760: 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e ire().await.....
2770: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
2780: 66 6f 72 6d 61 74 21 28 22 4c 69 73 74 20 66 65 format!("List fe
2790: 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 tch conn:\n{:?}"
27a0: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b , &self.pool))?;
27b0: 0a 09 09 72 65 70 6c 79 2e 70 75 73 68 28 22 43 ...reply.push("C
27c0: 68 61 6e 6e 65 6c 73 3a 22 2e 74 6f 5f 73 74 72 hannels:".to_str
27d0: 69 6e 67 28 29 29 3b 0a 09 09 6c 65 74 20 72 6f ing());...let ro
27e0: 77 73 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 ws = sqlx::query
27f0: 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 65 5f ("select source_
2800: 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 65 6e 61 id, channel, ena
2810: 62 6c 65 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 bled, url, iv_ha
2820: 73 68 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f sh from rsstg_so
2830: 75 72 63 65 20 77 68 65 72 65 20 6f 77 6e 65 72 urce where owner
2840: 20 3d 20 24 31 20 6f 72 64 65 72 20 62 79 20 73 = $1 order by s
2850: 6f 75 72 63 65 5f 69 64 22 29 0a 09 09 09 2e 62 ource_id").....b
2860: 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 66 ind(owner).....f
2870: 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 63 6f etch_all(&mut co
2880: 6e 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 66 6f nn).await?;...fo
2890: 72 20 72 6f 77 20 69 6e 20 72 6f 77 73 2e 69 74 r row in rows.it
28a0: 65 72 28 29 20 7b 0a 09 09 09 6c 65 74 20 73 6f er() {....let so
28b0: 75 72 63 65 5f 69 64 3a 20 69 33 32 20 3d 20 72 urce_id: i32 = r
28c0: 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 6f 75 72 ow.try_get("sour
28d0: 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 6c 65 74 ce_id")?;....let
28e0: 20 75 73 65 72 6e 61 6d 65 3a 20 26 73 74 72 20 username: &str
28f0: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 63 = row.try_get("c
2900: 68 61 6e 6e 65 6c 22 29 3f 3b 0a 09 09 09 6c 65 hannel")?;....le
2910: 74 20 65 6e 61 62 6c 65 64 3a 20 62 6f 6f 6c 20 t enabled: bool
2920: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 65 = row.try_get("e
2930: 6e 61 62 6c 65 64 22 29 3f 3b 0a 09 09 09 6c 65 nabled")?;....le
2940: 74 20 75 72 6c 3a 20 26 73 74 72 20 3d 20 72 6f t url: &str = ro
2950: 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 22 29 w.try_get("url")
2960: 3f 3b 0a 09 09 09 6c 65 74 20 69 76 5f 68 61 73 ?;....let iv_has
2970: 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 20 h: Option<&str>
2980: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 69 = row.try_get("i
2990: 76 5f 68 61 73 68 22 29 3f 3b 0a 09 09 09 72 65 v_hash")?;....re
29a0: 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 ply.push(format!
29b0: 28 22 5c 6e 5c 5c 23 ef b8 8f e2 83 a3 20 7b 7d ("\n\\#ļøā£ {}
29c0: 20 5c 5c 2a ef b8 8f e2 83 a3 20 60 7b 7d 60 20 \\*ļøā£ `{}`
29d0: 7b 7d 5c 6e f0 9f 94 97 20 60 7b 7d 60 22 2c 20 {}\nš `{}`",
29e0: 73 6f 75 72 63 65 5f 69 64 2c 20 75 73 65 72 6e source_id, usern
29f0: 61 6d 65 2c 20 20 0a 09 09 09 09 6d 61 74 63 68 ame, .....match
2a00: 20 65 6e 61 62 6c 65 64 20 7b 0a 09 09 09 09 09 enabled {......
2a10: 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 20 65 true => "š e
2a20: 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 66 61 nabled",......fa
2a30: 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 73 61 lse => "ā disa
2a40: 62 6c 65 64 22 2c 0a 09 09 09 09 7d 2c 20 75 72 bled",.....}, ur
2a50: 6c 29 29 3b 0a 09 09 09 69 66 20 6c 65 74 20 53 l));....if let S
2a60: 6f 6d 65 28 68 61 73 68 29 20 3d 20 69 76 5f 68 ome(hash) = iv_h
2a70: 61 73 68 20 7b 0a 09 09 09 09 72 65 70 6c 79 2e ash {.....reply.
2a80: 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 49 56 push(format!("IV
2a90: 20 60 7b 7d 60 22 2c 20 68 61 73 68 29 29 3b 0a `{}`", hash));.
2aa0: 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 ...}...};...Ok(r
2ab0: 65 70 6c 79 29 0a 09 7d 0a 7d 0a 0a 23 5b 74 6f eply)..}.}..#[to
2ac0: 6b 69 6f 3a 3a 6d 61 69 6e 5d 0a 61 73 79 6e 63 kio::main].async
2ad0: 20 66 6e 20 6d 61 69 6e 28 29 20 2d 3e 20 52 65 fn main() -> Re
2ae0: 73 75 6c 74 3c 28 29 3e 20 7b 0a 09 6c 65 74 20 sult<()> {..let
2af0: 6d 75 74 20 73 65 74 74 69 6e 67 73 20 3d 20 63 mut settings = c
2b00: 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 3a 3a 64 onfig::Config::d
2b10: 65 66 61 75 6c 74 28 29 3b 0a 09 73 65 74 74 69 efault();..setti
2b20: 6e 67 73 2e 6d 65 72 67 65 28 63 6f 6e 66 69 67 ngs.merge(config
2b30: 3a 3a 46 69 6c 65 3a 3a 77 69 74 68 5f 6e 61 6d ::File::with_nam
2b40: 65 28 22 72 73 73 74 67 22 29 29 3f 3b 0a 0a 09 e("rsstg"))?;...
2b50: 6c 65 74 20 63 6f 72 65 20 3d 20 43 6f 72 65 3a let core = Core:
2b60: 3a 6e 65 77 28 73 65 74 74 69 6e 67 73 29 2e 61 :new(settings).a
2b70: 77 61 69 74 3f 3b 0a 0a 09 6c 65 74 20 6d 75 74 wait?;...let mut
2b80: 20 73 74 72 65 61 6d 20 3d 20 63 6f 72 65 2e 73 stream = core.s
2b90: 74 72 65 61 6d 28 29 3b 0a 0a 09 6c 6f 6f 70 20 tream();...loop
2ba0: 7b 0a 09 09 6d 61 74 63 68 20 73 74 72 65 61 6d {...match stream
2bb0: 2e 6e 65 78 74 28 29 2e 61 77 61 69 74 20 7b 0a .next().await {.
2bc0: 09 09 09 53 6f 6d 65 28 75 70 64 61 74 65 29 20 ...Some(update)
2bd0: 3d 3e 20 7b 0a 09 09 09 09 69 66 20 6c 65 74 20 => {.....if let
2be0: 45 72 72 28 65 72 72 29 20 3d 20 68 61 6e 64 6c Err(err) = handl
2bf0: 65 28 75 70 64 61 74 65 3f 2c 20 26 63 6f 72 65 e(update?, &core
2c00: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 63 ).await {......c
2c10: 6f 72 65 2e 64 65 62 75 67 28 26 66 6f 72 6d 61 ore.debug(&forma
2c20: 74 21 28 22 f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 t!("š {:?}",
2c30: 65 72 72 29 29 3f 3b 0a 09 09 09 09 7d 3b 0a 09 err))?;.....};..
2c40: 09 09 7d 2c 0a 09 09 09 4e 6f 6e 65 20 3d 3e 20 ..},....None =>
2c50: 7b 0a 09 09 09 09 63 6f 72 65 2e 64 65 62 75 67 {.....core.debug
2c60: 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 (&format!("š
2c70: 4e 6f 6e 65 20 65 72 72 6f 72 2e 22 29 29 3f 3b None error."))?;
2c80: 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 7d 0a 0a 09 ....}...};..}...
2c90: 2f 2f 4f 6b 28 28 29 29 0a 7d 0a 0a 61 73 79 6e //Ok(()).}..asyn
2ca0: 63 20 66 6e 20 68 61 6e 64 6c 65 28 75 70 64 61 c fn handle(upda
2cb0: 74 65 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 te: telegram_bot
2cc0: 3a 3a 55 70 64 61 74 65 2c 20 63 6f 72 65 3a 20 ::Update, core:
2cd0: 26 43 6f 72 65 29 20 2d 3e 20 52 65 73 75 6c 74 &Core) -> Result
2ce0: 3c 28 29 3e 20 7b 0a 09 6c 61 7a 79 5f 73 74 61 <()> {..lazy_sta
2cf0: 74 69 63 21 20 7b 0a 09 09 73 74 61 74 69 63 20 tic! {...static
2d00: 72 65 66 20 52 45 5f 55 53 45 52 4e 41 4d 45 3a ref RE_USERNAME:
2d10: 20 52 65 67 65 78 20 3d 20 52 65 67 65 78 3a 3a Regex = Regex::
2d20: 6e 65 77 28 72 22 5e 40 5b 61 2d 7a 41 2d 5a 5d new(r"^@[a-zA-Z]
2d30: 5b 61 2d 7a 41 2d 5a 30 2d 39 5f 5d 2b 24 22 29 [a-zA-Z0-9_]+$")
2d40: 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 73 74 61 .unwrap();...sta
2d50: 74 69 63 20 72 65 66 20 52 45 5f 4c 49 4e 4b 3a tic ref RE_LINK:
2d60: 20 52 65 67 65 78 20 3d 20 52 65 67 65 78 3a 3a Regex = Regex::
2d70: 6e 65 77 28 72 22 5e 68 74 74 70 73 3f 3a 2f 2f new(r"^https?://
2d80: 5b 61 2d 7a 41 2d 5a 2e 30 2d 39 2d 5d 2b 2f 5b [a-zA-Z.0-9-]+/[
2d90: 2d 5f 61 2d 7a 41 2d 5a 2e 30 2d 39 2f 3f 3d 5d -_a-zA-Z.0-9/?=]
2da0: 2b 24 22 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 +$").unwrap();..
2db0: 09 73 74 61 74 69 63 20 72 65 66 20 52 45 5f 49 .static ref RE_I
2dc0: 56 5f 48 41 53 48 3a 20 52 65 67 65 78 20 3d 20 V_HASH: Regex =
2dd0: 52 65 67 65 78 3a 3a 6e 65 77 28 72 22 5e 5b 61 Regex::new(r"^[a
2de0: 2d 66 30 2d 39 5d 7b 31 34 7d 24 22 29 2e 75 6e -f0-9]{14}$").un
2df0: 77 72 61 70 28 29 3b 0a 09 7d 0a 0a 09 6d 61 74 wrap();..}...mat
2e00: 63 68 20 75 70 64 61 74 65 2e 6b 69 6e 64 20 7b ch update.kind {
2e10: 0a 09 09 55 70 64 61 74 65 4b 69 6e 64 3a 3a 4d ...UpdateKind::M
2e20: 65 73 73 61 67 65 28 6d 65 73 73 61 67 65 29 20 essage(message)
2e30: 3d 3e 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 => {....let mut
2e40: 72 65 70 6c 79 3a 20 56 65 63 3c 53 74 72 69 6e reply: Vec<Strin
2e50: 67 3e 20 3d 20 76 65 63 21 5b 5d 3b 0a 09 09 09 g> = vec![];....
2e60: 6d 61 74 63 68 20 6d 65 73 73 61 67 65 2e 6b 69 match message.ki
2e70: 6e 64 20 7b 0a 09 09 09 09 4d 65 73 73 61 67 65 nd {.....Message
2e80: 4b 69 6e 64 3a 3a 54 65 78 74 20 7b 20 72 65 66 Kind::Text { ref
2e90: 20 64 61 74 61 2c 20 2e 2e 20 7d 20 3d 3e 20 7b data, .. } => {
2ea0: 0a 09 09 09 09 09 6c 65 74 20 6d 75 74 20 77 6f ......let mut wo
2eb0: 72 64 73 20 3d 20 64 61 74 61 2e 73 70 6c 69 74 rds = data.split
2ec0: 5f 77 68 69 74 65 73 70 61 63 65 28 29 3b 0a 09 _whitespace();..
2ed0: 09 09 09 09 6c 65 74 20 63 6d 64 20 3d 20 77 6f ....let cmd = wo
2ee0: 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 rds.next().unwra
2ef0: 70 28 29 3b 0a 09 09 09 09 09 6d 61 74 63 68 20 p();......match
2f00: 63 6d 64 20 7b 0a 0a 2f 2f 20 73 74 61 72 74 0a cmd {..// start.
2f10: 0a 09 09 09 09 09 09 22 2f 73 74 61 72 74 22 20 ......."/start"
2f20: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 72 65 70 6c => {........repl
2f30: 79 2e 70 75 73 68 28 22 57 65 20 61 72 65 20 6f y.push("We are o
2f40: 70 65 6e 5c 5c 2e 20 50 72 6f 62 61 62 6c 79 5c pen\\. Probably\
2f50: 5c 2e 20 56 69 73 69 74 20 5b 63 68 61 6e 6e 65 \. Visit [channe
2f60: 6c 5d 28 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f l](https://t.me/
2f70: 72 73 73 74 67 5f 62 6f 74 5f 68 65 6c 70 2f 33 rsstg_bot_help/3
2f80: 29 20 66 6f 72 20 64 65 74 61 69 6c 73 5c 5c 2e ) for details\\.
2f90: 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a ".to_string());.
2fa0: 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 6c 69 73 ......},..// lis
2fb0: 74 0a 0a 09 09 09 09 09 09 22 2f 6c 69 73 74 22 t........"/list"
2fc0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 72 65 70 => {........rep
2fd0: 6c 79 2e 61 70 70 65 6e 64 28 26 6d 75 74 20 63 ly.append(&mut c
2fe0: 6f 72 65 2e 6c 69 73 74 28 6d 65 73 73 61 67 65 ore.list(message
2ff0: 2e 66 72 6f 6d 2e 69 64 29 2e 61 77 61 69 74 3f .from.id).await?
3000: 29 3b 0a 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 );.......},..//
3010: 61 64 64 0a 0a 09 09 09 09 09 09 22 2f 61 64 64 add........"/add
3020: 22 20 7c 20 22 2f 75 70 64 61 74 65 22 20 3d 3e " | "/update" =>
3030: 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 6d 75 {........let mu
3040: 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 4f 70 74 t source_id: Opt
3050: 69 6f 6e 3c 69 33 32 3e 20 3d 20 4e 6f 6e 65 3b ion<i32> = None;
3060: 0a 09 09 09 09 09 09 09 6c 65 74 20 61 74 5f 6c ........let at_l
3070: 65 61 73 74 20 3d 20 22 52 65 71 75 69 72 65 73 east = "Requires
3080: 20 61 74 20 6c 65 61 73 74 20 33 20 70 61 72 61 at least 3 para
3090: 6d 65 74 65 72 73 2e 22 3b 0a 09 09 09 09 09 09 meters.";.......
30a0: 09 69 66 20 63 6d 64 20 3d 3d 20 22 2f 75 70 64 .if cmd == "/upd
30b0: 61 74 65 22 20 7b 0a 09 09 09 09 09 09 09 09 6c ate" {.........l
30c0: 65 74 20 66 69 72 73 74 5f 77 6f 72 64 20 3d 20 et first_word =
30d0: 77 6f 72 64 73 2e 6e 65 78 74 28 29 0a 09 09 09 words.next()....
30e0: 09 09 09 09 09 09 2e 63 6f 6e 74 65 78 74 28 61 .......context(a
30f0: 74 5f 6c 65 61 73 74 29 3f 3b 0a 09 09 09 09 09 t_least)?;......
3100: 09 09 09 73 6f 75 72 63 65 5f 69 64 20 3d 20 53 ...source_id = S
3110: 6f 6d 65 28 66 69 72 73 74 5f 77 6f 72 64 2e 70 ome(first_word.p
3120: 61 72 73 65 3a 3a 3c 69 33 32 3e 28 29 0a 09 09 arse::<i32>()...
3130: 09 09 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e ........with_con
3140: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
3150: 22 49 20 6e 65 65 64 20 61 20 6e 75 6d 62 65 72 "I need a number
3160: 2c 20 62 75 74 20 67 6f 74 20 7b 7d 2e 22 2c 20 , but got {}.",
3170: 66 69 72 73 74 5f 77 6f 72 64 29 29 3f 29 3b 0a first_word))?);.
3180: 09 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 09 .......}........
3190: 6c 65 74 20 28 63 68 61 6e 6e 65 6c 2c 20 75 72 let (channel, ur
31a0: 6c 2c 20 69 76 5f 68 61 73 68 29 20 3d 20 28 0a l, iv_hash) = (.
31b0: 09 09 09 09 09 09 09 09 77 6f 72 64 73 2e 6e 65 ........words.ne
31c0: 78 74 28 29 2e 63 6f 6e 74 65 78 74 28 61 74 5f xt().context(at_
31d0: 6c 65 61 73 74 29 3f 2c 0a 09 09 09 09 09 09 09 least)?,........
31e0: 09 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e 63 6f .words.next().co
31f0: 6e 74 65 78 74 28 61 74 5f 6c 65 61 73 74 29 3f ntext(at_least)?
3200: 2c 0a 09 09 09 09 09 09 09 09 77 6f 72 64 73 2e ,.........words.
3210: 6e 65 78 74 28 29 29 3b 0a 09 09 09 09 09 09 09 next());........
3220: 69 66 20 21 20 52 45 5f 55 53 45 52 4e 41 4d 45 if ! RE_USERNAME
3230: 2e 69 73 5f 6d 61 74 63 68 28 26 63 68 61 6e 6e .is_match(&chann
3240: 65 6c 29 20 7b 0a 09 09 09 09 09 09 09 09 72 65 el) {.........re
3250: 70 6c 79 2e 70 75 73 68 28 22 55 73 65 72 6e 61 ply.push("Userna
3260: 6d 65 73 20 73 68 6f 75 6c 64 20 62 65 20 73 6f mes should be so
3270: 6d 65 74 68 69 6e 67 20 6c 69 6b 65 20 5c 22 40 mething like \"@
3280: 5c 5c 5b 61 5c 5c 2d 7a 41 5c 5c 2d 5a 5d 5c 5c \\[a\\-zA\\-Z]\\
3290: 5b 61 5c 5c 2d 7a 41 5c 5c 2d 5a 30 5c 5c 2d 39 [a\\-zA\\-Z0\\-9
32a0: 5c 5c 5f 5d 2b 5c 22 2c 20 61 72 65 6e 27 74 20 \\_]+\", aren't
32b0: 74 68 65 79 3f 22 2e 74 6f 5f 73 74 72 69 6e 67 they?".to_string
32c0: 28 29 29 3b 0a 09 09 09 09 09 09 09 09 62 61 69 ());.........bai
32d0: 6c 21 28 22 57 72 6f 6e 67 20 75 73 65 72 6e 61 l!("Wrong userna
32e0: 6d 65 20 7b 3a 3f 7d 2e 22 2c 20 26 63 68 61 6e me {:?}.", &chan
32f0: 6e 65 6c 29 3b 0a 09 09 09 09 09 09 09 7d 0a 09 nel);........}..
3300: 09 09 09 09 09 09 69 66 20 21 20 52 45 5f 4c 49 ......if ! RE_LI
3310: 4e 4b 2e 69 73 5f 6d 61 74 63 68 28 26 75 72 6c NK.is_match(&url
3320: 29 20 7b 0a 09 09 09 09 09 09 09 09 72 65 70 6c ) {.........repl
3330: 79 2e 70 75 73 68 28 22 4c 69 6e 6b 20 73 68 6f y.push("Link sho
3340: 75 6c 64 20 62 65 20 6c 69 6e 6b 20 74 6f 20 61 uld be link to a
3350: 74 6f 6d 2f 72 73 73 20 66 65 65 64 2c 20 73 6f tom/rss feed, so
3360: 6d 65 74 68 69 6e 67 20 6c 69 6b 65 20 5c 22 68 mething like \"h
3370: 74 74 70 73 3a 2f 2f 64 6f 6d 61 69 6e 2f 70 61 ttps://domain/pa
3380: 74 68 5c 22 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 th\"\\.".to_stri
3390: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 62 ng());.........b
33a0: 61 69 6c 21 28 22 55 72 6c 3a 20 7b 3a 3f 7d 22 ail!("Url: {:?}"
33b0: 2c 20 26 75 72 6c 29 3b 0a 09 09 09 09 09 09 09 , &url);........
33c0: 7d 0a 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 }........if let
33d0: 53 6f 6d 65 28 68 61 73 68 29 20 3d 20 69 76 5f Some(hash) = iv_
33e0: 68 61 73 68 20 7b 0a 09 09 09 09 09 09 09 09 69 hash {.........i
33f0: 66 20 21 20 52 45 5f 49 56 5f 48 41 53 48 2e 69 f ! RE_IV_HASH.i
3400: 73 5f 6d 61 74 63 68 28 26 68 61 73 68 29 20 7b s_match(&hash) {
3410: 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e ..........reply.
3420: 70 75 73 68 28 22 49 56 20 68 61 73 68 20 73 68 push("IV hash sh
3430: 6f 75 6c 64 20 62 65 20 31 34 20 68 65 78 20 64 ould be 14 hex d
3440: 69 67 69 74 73 2e 22 2e 74 6f 5f 73 74 72 69 6e igits.".to_strin
3450: 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 09 62 g());..........b
3460: 61 69 6c 21 28 22 49 56 3a 20 7b 3a 3f 7d 22 2c ail!("IV: {:?}",
3470: 20 26 69 76 5f 68 61 73 68 29 3b 0a 09 09 09 09 &iv_hash);.....
3480: 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 7d 3b ....};........};
3490: 0a 09 09 09 09 09 09 09 6c 65 74 20 63 68 61 6e ........let chan
34a0: 6e 65 6c 5f 69 64 20 3d 20 69 36 34 3a 3a 66 72 nel_id = i64::fr
34b0: 6f 6d 28 63 6f 72 65 2e 74 67 2e 73 65 6e 64 28 om(core.tg.send(
34c0: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 47 65 telegram_bot::Ge
34d0: 74 43 68 61 74 3a 3a 6e 65 77 28 74 65 6c 65 67 tChat::new(teleg
34e0: 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a ram_bot::types::
34f0: 43 68 61 74 52 65 66 3a 3a 43 68 61 6e 6e 65 6c ChatRef::Channel
3500: 55 73 65 72 6e 61 6d 65 28 63 68 61 6e 6e 65 6c Username(channel
3510: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 29 29 2e .to_string()))).
3520: 61 77 61 69 74 3f 2e 69 64 28 29 29 3b 0a 09 09 await?.id());...
3530: 09 09 09 09 09 6c 65 74 20 63 68 61 6e 5f 61 64 .....let chan_ad
3540: 6d 20 3d 20 63 6f 72 65 2e 74 67 2e 73 65 6e 64 m = core.tg.send
3550: 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 47 (telegram_bot::G
3560: 65 74 43 68 61 74 41 64 6d 69 6e 69 73 74 72 61 etChatAdministra
3570: 74 6f 72 73 3a 3a 6e 65 77 28 74 65 6c 65 67 72 tors::new(telegr
3580: 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a 43 am_bot::types::C
3590: 68 61 74 52 65 66 3a 3a 43 68 61 6e 6e 65 6c 55 hatRef::ChannelU
35a0: 73 65 72 6e 61 6d 65 28 63 68 61 6e 6e 65 6c 2e sername(channel.
35b0: 74 6f 5f 73 74 72 69 6e 67 28 29 29 29 29 2e 61 to_string()))).a
35c0: 77 61 69 74 0a 09 09 09 09 09 09 09 09 2e 63 6f wait..........co
35d0: 6e 74 65 78 74 28 22 53 6f 72 72 79 2c 20 49 20 ntext("Sorry, I
35e0: 68 61 76 65 20 6e 6f 20 61 63 63 65 73 73 20 74 have no access t
35f0: 6f 20 74 68 61 74 20 63 68 61 74 5c 5c 2e 22 29 o that chat\\.")
3600: 3f 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 28 6d ?;........let (m
3610: 75 74 20 6d 65 2c 20 6d 75 74 20 75 73 65 72 29 ut me, mut user)
3620: 20 3d 20 28 66 61 6c 73 65 2c 20 66 61 6c 73 65 = (false, false
3630: 29 3b 0a 09 09 09 09 09 09 09 66 6f 72 20 61 64 );........for ad
3640: 6d 69 6e 20 69 6e 20 63 68 61 6e 5f 61 64 6d 20 min in chan_adm
3650: 7b 0a 09 09 09 09 09 09 09 09 69 66 20 61 64 6d {.........if adm
3660: 69 6e 2e 75 73 65 72 2e 69 64 20 3d 3d 20 63 6f in.user.id == co
3670: 72 65 2e 6d 79 2e 69 64 20 7b 0a 09 09 09 09 09 re.my.id {......
3680: 09 09 09 09 6d 65 20 3d 20 74 72 75 65 3b 0a 09 ....me = true;..
3690: 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 .......};.......
36a0: 09 09 69 66 20 61 64 6d 69 6e 2e 75 73 65 72 2e ..if admin.user.
36b0: 69 64 20 3d 3d 20 6d 65 73 73 61 67 65 2e 66 72 id == message.fr
36c0: 6f 6d 2e 69 64 20 7b 0a 09 09 09 09 09 09 09 09 om.id {.........
36d0: 09 75 73 65 72 20 3d 20 74 72 75 65 3b 0a 09 09 .user = true;...
36e0: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 ......};........
36f0: 7d 3b 0a 09 09 09 09 09 09 09 69 66 20 21 20 6d };........if ! m
3700: 65 20 20 20 7b 20 62 61 69 6c 21 28 22 49 20 6e e { bail!("I n
3710: 65 65 64 20 74 6f 20 62 65 20 61 64 6d 69 6e 20 eed to be admin
3720: 6f 6e 20 74 68 61 74 20 63 68 61 6e 6e 65 6c 5c on that channel\
3730: 5c 2e 22 29 3b 20 7d 3b 0a 09 09 09 09 09 09 09 \."); };........
3740: 69 66 20 21 20 75 73 65 72 20 7b 20 62 61 69 6c if ! user { bail
3750: 21 28 22 59 6f 75 20 73 68 6f 75 6c 64 20 62 65 !("You should be
3760: 20 61 64 6d 69 6e 20 6f 6e 20 74 68 61 74 20 63 admin on that c
3770: 68 61 6e 6e 65 6c 5c 5c 2e 22 29 3b 20 7d 3b 0a hannel\\."); };.
3780: 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 .......reply.pus
3790: 68 28 63 6f 72 65 2e 75 70 64 61 74 65 28 73 6f h(core.update(so
37a0: 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c urce_id, channel
37b0: 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 , channel_id, ur
37c0: 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6d 65 73 73 l, iv_hash, mess
37d0: 61 67 65 2e 66 72 6f 6d 2e 69 64 29 2e 61 77 61 age.from.id).awa
37e0: 69 74 3f 29 3b 0a 09 09 09 09 09 09 7d 2c 0a 0a it?);.......},..
37f0: 2f 2f 20 63 68 65 63 6b 0a 0a 09 09 09 09 09 09 // check........
3800: 22 2f 63 68 65 63 6b 22 20 3d 3e 20 7b 0a 09 09 "/check" => {...
3810: 09 09 09 09 09 6d 61 74 63 68 20 26 77 6f 72 64 .....match &word
3820: 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 70 28 s.next().unwrap(
3830: 29 2e 70 61 72 73 65 3a 3a 3c 69 33 32 3e 28 29 ).parse::<i32>()
3840: 20 7b 0a 09 09 09 09 09 09 09 09 45 72 72 28 65 {.........Err(e
3850: 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 rr) => {........
3860: 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 ..reply.push(for
3870: 6d 61 74 21 28 22 49 20 6e 65 65 64 20 61 20 6e mat!("I need a n
3880: 75 6d 62 65 72 5c 5c 2e 5c 6e 7b 7d 22 2c 20 26 umber\\.\n{}", &
3890: 65 72 72 29 29 3b 0a 09 09 09 09 09 09 09 09 7d err));.........}
38a0: 2c 0a 09 09 09 09 09 09 09 09 4f 6b 28 6e 75 6d ,.........Ok(num
38b0: 62 65 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 ber) => {.......
38c0: 09 09 09 63 6f 72 65 2e 63 68 65 63 6b 28 2a 6e ...core.check(*n
38d0: 75 6d 62 65 72 2c 20 6d 65 73 73 61 67 65 2e 66 umber, message.f
38e0: 72 6f 6d 2e 69 64 2c 20 66 61 6c 73 65 29 2e 61 rom.id, false).a
38f0: 77 61 69 74 0a 09 09 09 09 09 09 09 09 09 09 2e wait............
3900: 63 6f 6e 74 65 78 74 28 22 43 68 61 6e 6e 65 6c context("Channel
3910: 20 63 68 65 63 6b 20 66 61 69 6c 65 64 2e 22 29 check failed.")
3920: 3f 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 ?;.........},...
3930: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 2c .....};.......},
3940: 0a 0a 2f 2f 20 63 6c 65 61 6e 0a 0a 09 09 09 09 ..// clean......
3950: 09 09 22 2f 63 6c 65 61 6e 22 20 3d 3e 20 7b 0a .."/clean" => {.
3960: 09 09 09 09 09 09 09 6d 61 74 63 68 20 26 77 6f .......match &wo
3970: 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 rds.next().unwra
3980: 70 28 29 2e 70 61 72 73 65 3a 3a 3c 69 33 32 3e p().parse::<i32>
3990: 28 29 20 7b 0a 09 09 09 09 09 09 09 09 45 72 72 () {.........Err
39a0: 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 (err) => {......
39b0: 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 ....reply.push(f
39c0: 6f 72 6d 61 74 21 28 22 49 20 6e 65 65 64 20 61 ormat!("I need a
39d0: 20 6e 75 6d 62 65 72 5c 5c 2e 5c 6e 7b 7d 22 2c number\\.\n{}",
39e0: 20 26 65 72 72 29 29 3b 0a 09 09 09 09 09 09 09 &err));........
39f0: 09 7d 2c 0a 09 09 09 09 09 09 09 09 4f 6b 28 6e .},.........Ok(n
3a00: 75 6d 62 65 72 29 20 3d 3e 20 7b 0a 09 09 09 09 umber) => {.....
3a10: 09 09 09 09 09 6c 65 74 20 72 65 73 75 6c 74 20 .....let result
3a20: 3d 20 63 6f 72 65 2e 63 6c 65 61 6e 28 26 6e 75 = core.clean(&nu
3a30: 6d 62 65 72 2c 20 6d 65 73 73 61 67 65 2e 66 72 mber, message.fr
3a40: 6f 6d 2e 69 64 29 2e 61 77 61 69 74 3f 3b 0a 09 om.id).await?;..
3a50: 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 ........reply.pu
3a60: 73 68 28 72 65 73 75 6c 74 2e 74 6f 5f 73 74 72 sh(result.to_str
3a70: 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 ing());.........
3a80: 7d 2c 0a 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 },........};....
3a90: 09 09 09 7d 2c 0a 0a 2f 2f 20 65 6e 61 62 6c 65 ...},..// enable
3aa0: 0a 0a 09 09 09 09 09 09 22 2f 65 6e 61 62 6c 65 ........"/enable
3ab0: 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 6d 61 " => {........ma
3ac0: 74 63 68 20 26 77 6f 72 64 73 2e 6e 65 78 74 28 tch &words.next(
3ad0: 29 2e 75 6e 77 72 61 70 28 29 2e 70 61 72 73 65 ).unwrap().parse
3ae0: 3a 3a 3c 69 33 32 3e 28 29 20 7b 0a 09 09 09 09 ::<i32>() {.....
3af0: 09 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 ....Err(err) =>
3b00: 7b 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 {..........reply
3b10: 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 49 .push(format!("I
3b20: 20 6e 65 65 64 20 61 20 6e 75 6d 62 65 72 5c 5c need a number\\
3b30: 2e 5c 6e 7b 7d 22 2c 20 26 65 72 72 29 29 3b 0a .\n{}", &err));.
3b40: 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ........},......
3b50: 09 09 09 4f 6b 28 6e 75 6d 62 65 72 29 20 3d 3e ...Ok(number) =>
3b60: 20 7b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 {..........let
3b70: 72 65 73 75 6c 74 20 3d 20 63 6f 72 65 2e 65 6e result = core.en
3b80: 61 62 6c 65 28 26 6e 75 6d 62 65 72 2c 20 6d 65 able(&number, me
3b90: 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 29 2e 61 ssage.from.id).a
3ba0: 77 61 69 74 3f 3b 0a 09 09 09 09 09 09 09 09 09 wait?;..........
3bb0: 72 65 70 6c 79 2e 70 75 73 68 28 72 65 73 75 6c reply.push(resul
3bc0: 74 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a t.to_string());.
3bd0: 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ........},......
3be0: 09 09 7d 3b 0a 09 09 09 09 09 09 7d 2c 0a 0a 2f ..};.......},../
3bf0: 2f 20 64 65 6c 65 74 65 0a 0a 09 09 09 09 09 09 / delete........
3c00: 22 2f 64 65 6c 65 74 65 22 20 3d 3e 20 7b 0a 09 "/delete" => {..
3c10: 09 09 09 09 09 09 6d 61 74 63 68 20 26 77 6f 72 ......match &wor
3c20: 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 70 ds.next().unwrap
3c30: 28 29 2e 70 61 72 73 65 3a 3a 3c 69 33 32 3e 28 ().parse::<i32>(
3c40: 29 20 7b 0a 09 09 09 09 09 09 09 09 45 72 72 28 ) {.........Err(
3c50: 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 err) => {.......
3c60: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f ...reply.push(fo
3c70: 72 6d 61 74 21 28 22 49 20 6e 65 65 64 20 61 20 rmat!("I need a
3c80: 6e 75 6d 62 65 72 5c 5c 2e 5c 6e 7b 7d 22 2c 20 number\\.\n{}",
3c90: 26 65 72 72 29 29 3b 0a 09 09 09 09 09 09 09 09 &err));.........
3ca0: 7d 2c 0a 09 09 09 09 09 09 09 09 4f 6b 28 6e 75 },.........Ok(nu
3cb0: 6d 62 65 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 mber) => {......
3cc0: 09 09 09 09 6c 65 74 20 72 65 73 75 6c 74 20 3d ....let result =
3cd0: 20 63 6f 72 65 2e 64 65 6c 65 74 65 28 26 6e 75 core.delete(&nu
3ce0: 6d 62 65 72 2c 20 6d 65 73 73 61 67 65 2e 66 72 mber, message.fr
3cf0: 6f 6d 2e 69 64 29 2e 61 77 61 69 74 3f 3b 0a 09 om.id).await?;..
3d00: 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 ........reply.pu
3d10: 73 68 28 72 65 73 75 6c 74 2e 74 6f 5f 73 74 72 sh(result.to_str
3d20: 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 ing());.........
3d30: 7d 2c 0a 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 },........};....
3d40: 09 09 09 7d 2c 0a 0a 2f 2f 20 64 69 73 61 62 6c ...},..// disabl
3d50: 65 0a 0a 09 09 09 09 09 09 22 2f 64 69 73 61 62 e........"/disab
3d60: 6c 65 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 le" => {........
3d70: 6d 61 74 63 68 20 26 77 6f 72 64 73 2e 6e 65 78 match &words.nex
3d80: 74 28 29 2e 75 6e 77 72 61 70 28 29 2e 70 61 72 t().unwrap().par
3d90: 73 65 3a 3a 3c 69 33 32 3e 28 29 20 7b 0a 09 09 se::<i32>() {...
3da0: 09 09 09 09 09 09 45 72 72 28 65 72 72 29 20 3d ......Err(err) =
3db0: 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 72 65 70 > {..........rep
3dc0: 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 ly.push(format!(
3dd0: 22 49 20 6e 65 65 64 20 61 20 6e 75 6d 62 65 72 "I need a number
3de0: 5c 5c 2e 5c 6e 7b 7d 22 2c 20 26 65 72 72 29 29 \\.\n{}", &err))
3df0: 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 ;.........},....
3e00: 09 09 09 09 09 4f 6b 28 6e 75 6d 62 65 72 29 20 .....Ok(number)
3e10: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 6c 65 => {..........le
3e20: 74 20 72 65 73 75 6c 74 20 3d 20 63 6f 72 65 2e t result = core.
3e30: 64 69 73 61 62 6c 65 28 26 6e 75 6d 62 65 72 2c disable(&number,
3e40: 20 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 message.from.id
3e50: 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 09 09 09 ).await?;.......
3e60: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 72 65 ...reply.push(re
3e70: 73 75 6c 74 2e 74 6f 5f 73 74 72 69 6e 67 28 29 sult.to_string()
3e80: 29 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 );.........},...
3e90: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 2c .....};.......},
3ea0: 0a 0a 09 09 09 09 09 09 5f 20 3d 3e 20 7b 0a 09 ........_ => {..
3eb0: 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 7d 3b 0a .....},......};.
3ec0: 09 09 09 09 7d 2c 0a 09 09 09 09 5f 20 3d 3e 20 ....},....._ =>
3ed0: 7b 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 3b 0a 0a {.....},....};..
3ee0: 09 09 09 69 66 20 72 65 70 6c 79 2e 6c 65 6e 28 ...if reply.len(
3ef0: 29 20 3e 20 30 20 7b 0a 09 09 09 09 69 66 20 6c ) > 0 {.....if l
3f00: 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6f et Err(err) = co
3f10: 72 65 2e 74 67 2e 73 65 6e 64 28 6d 65 73 73 61 re.tg.send(messa
3f20: 67 65 2e 74 65 78 74 5f 72 65 70 6c 79 28 72 65 ge.text_reply(re
3f30: 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 2e ply.join("\n")).
3f40: 70 61 72 73 65 5f 6d 6f 64 65 28 74 79 70 65 73 parse_mode(types
3f50: 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 4d 61 72 ::ParseMode::Mar
3f60: 6b 64 6f 77 6e 56 32 29 29 2e 61 77 61 69 74 20 kdownV2)).await
3f70: 7b 0a 09 09 09 09 09 64 62 67 21 28 72 65 70 6c {......dbg!(repl
3f80: 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 3b 0a 09 y.join("\n"));..
3f90: 09 09 09 09 70 72 69 6e 74 6c 6e 21 28 22 7b 7d ....println!("{}
3fa0: 22 2c 20 65 72 72 29 3b 0a 09 09 09 09 7d 3b 0a ", err);.....};.
3fb0: 09 09 09 7d 3b 0a 09 09 7d 2c 0a 09 09 5f 20 3d ...};...},..._ =
3fc0: 3e 20 7b 7d 2c 0a 09 7d 3b 0a 0a 09 4f 6b 28 28 > {},..};...Ok((
3fd0: 29 29 0a 7d 0a )).}.