0000: 75 73 65 20 63 72 61 74 65 3a 3a 7b 0a 09 63 6f use crate::{..co
0010: 6d 6d 61 6e 64 2c 0a 09 73 71 6c 3a 3a 44 62 2c mmand,..sql::Db,
0020: 0a 7d 3b 0a 0a 75 73 65 20 73 74 64 3a 3a 7b 0a .};..use std::{.
0030: 09 62 6f 72 72 6f 77 3a 3a 43 6f 77 2c 0a 09 63 .borrow::Cow,..c
0040: 6f 6c 6c 65 63 74 69 6f 6e 73 3a 3a 7b 0a 09 09 ollections::{...
0050: 42 54 72 65 65 4d 61 70 2c 0a 09 09 48 61 73 68 BTreeMap,...Hash
0060: 53 65 74 2c 0a 09 7d 2c 0a 09 6e 75 6d 3a 3a 54 Set,..},..num::T
0070: 72 79 46 72 6f 6d 49 6e 74 45 72 72 6f 72 2c 0a ryFromIntError,.
0080: 09 73 79 6e 63 3a 3a 7b 0a 09 09 41 72 63 2c 0a .sync::{...Arc,.
0090: 09 09 4d 75 74 65 78 0a 09 7d 2c 0a 7d 3b 0a 0a ..Mutex..},.};..
00a0: 75 73 65 20 61 6e 79 68 6f 77 3a 3a 7b 0a 09 61 use anyhow::{..a
00b0: 6e 79 68 6f 77 2c 0a 09 62 61 69 6c 2c 0a 09 52 nyhow,..bail,..R
00c0: 65 73 75 6c 74 2c 0a 7d 3b 0a 75 73 65 20 61 73 esult,.};.use as
00d0: 79 6e 63 5f 73 74 64 3a 3a 74 61 73 6b 3b 0a 75 ync_std::task;.u
00e0: 73 65 20 63 68 72 6f 6e 6f 3a 3a 44 61 74 65 54 se chrono::DateT
00f0: 69 6d 65 3b 0a 75 73 65 20 66 72 61 6e 6b 65 6e ime;.use franken
0100: 73 74 65 69 6e 3a 3a 7b 0a 09 41 73 79 6e 63 54 stein::{..AsyncT
0110: 65 6c 65 67 72 61 6d 41 70 69 2c 0a 09 45 72 72 elegramApi,..Err
0120: 6f 72 20 61 73 20 46 72 61 6e 6b 45 72 72 6f 72 or as FrankError
0130: 2c 0a 09 50 61 72 73 65 4d 6f 64 65 2c 0a 09 63 ,..ParseMode,..c
0140: 6c 69 65 6e 74 5f 72 65 71 77 65 73 74 3a 3a 42 lient_reqwest::B
0150: 6f 74 2c 0a 09 6d 65 74 68 6f 64 73 3a 3a 7b 0a ot,..methods::{.
0160: 09 09 47 65 74 55 70 64 61 74 65 73 50 61 72 61 ..GetUpdatesPara
0170: 6d 73 2c 0a 09 09 53 65 6e 64 4d 65 73 73 61 67 ms,...SendMessag
0180: 65 50 61 72 61 6d 73 0a 09 7d 2c 0a 09 74 79 70 eParams..},..typ
0190: 65 73 3a 3a 7b 0a 09 09 41 6c 6c 6f 77 65 64 55 es::{...AllowedU
01a0: 70 64 61 74 65 2c 0a 09 09 4d 65 73 73 61 67 65 pdate,...Message
01b0: 45 6e 74 69 74 79 54 79 70 65 2c 0a 09 09 55 73 EntityType,...Us
01c0: 65 72 2c 0a 09 7d 2c 0a 09 75 70 64 61 74 65 73 er,..},..updates
01d0: 3a 3a 55 70 64 61 74 65 43 6f 6e 74 65 6e 74 2c ::UpdateContent,
01e0: 0a 7d 3b 0a 75 73 65 20 74 68 69 73 65 72 72 6f .};.use thiserro
01f0: 72 3a 3a 45 72 72 6f 72 3b 0a 0a 23 5b 64 65 72 r::Error;..#[der
0200: 69 76 65 28 45 72 72 6f 72 2c 20 44 65 62 75 67 ive(Error, Debug
0210: 29 5d 0a 70 75 62 20 65 6e 75 6d 20 52 73 73 45 )].pub enum RssE
0220: 72 72 6f 72 20 7b 0a 09 2f 2f 20 23 5b 65 72 72 rror {..// #[err
0230: 6f 72 28 74 72 61 6e 73 70 61 72 65 6e 74 29 5d or(transparent)]
0240: 0a 09 2f 2f 20 54 67 28 23 5b 66 72 6f 6d 5d 20 ..// Tg(#[from]
0250: 54 67 45 72 72 6f 72 29 2c 0a 09 23 5b 65 72 72 TgError),..#[err
0260: 6f 72 28 74 72 61 6e 73 70 61 72 65 6e 74 29 5d or(transparent)]
0270: 0a 09 49 6e 74 28 23 5b 66 72 6f 6d 5d 20 54 72 ..Int(#[from] Tr
0280: 79 46 72 6f 6d 49 6e 74 45 72 72 6f 72 29 2c 0a yFromIntError),.
0290: 7d 0a 0a 23 5b 64 65 72 69 76 65 28 43 6c 6f 6e }..#[derive(Clon
02a0: 65 29 5d 0a 70 75 62 20 73 74 72 75 63 74 20 43 e)].pub struct C
02b0: 6f 72 65 20 7b 0a 09 6f 77 6e 65 72 5f 63 68 61 ore {..owner_cha
02c0: 74 3a 20 69 36 34 2c 0a 09 6d 61 78 5f 64 65 6c t: i64,..max_del
02d0: 61 79 3a 20 75 31 36 2c 0a 09 70 75 62 20 74 67 ay: u16,..pub tg
02e0: 3a 20 42 6f 74 2c 0a 09 70 75 62 20 6d 65 3a 20 : Bot,..pub me:
02f0: 55 73 65 72 2c 0a 09 70 75 62 20 64 62 3a 20 44 User,..pub db: D
0300: 62 2c 0a 09 73 6f 75 72 63 65 73 3a 20 41 72 63 b,..sources: Arc
0310: 3c 4d 75 74 65 78 3c 48 61 73 68 53 65 74 3c 41 <Mutex<HashSet<A
0320: 72 63 3c 69 33 32 3e 3e 3e 3e 2c 0a 09 68 74 74 rc<i32>>>>,..htt
0330: 70 5f 63 6c 69 65 6e 74 3a 20 72 65 71 77 65 73 p_client: reqwes
0340: 74 3a 3a 43 6c 69 65 6e 74 2c 0a 7d 0a 0a 69 6d t::Client,.}..im
0350: 70 6c 20 43 6f 72 65 20 7b 0a 09 70 75 62 20 61 pl Core {..pub a
0360: 73 79 6e 63 20 66 6e 20 6e 65 77 28 73 65 74 74 sync fn new(sett
0370: 69 6e 67 73 3a 20 63 6f 6e 66 69 67 3a 3a 43 6f ings: config::Co
0380: 6e 66 69 67 29 20 2d 3e 20 52 65 73 75 6c 74 3c nfig) -> Result<
0390: 43 6f 72 65 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 Core> {...let ow
03a0: 6e 65 72 5f 63 68 61 74 20 3d 20 73 65 74 74 69 ner_chat = setti
03b0: 6e 67 73 2e 67 65 74 5f 69 6e 74 28 22 6f 77 6e ngs.get_int("own
03c0: 65 72 22 29 3f 3b 0a 09 09 6c 65 74 20 61 70 69 er")?;...let api
03d0: 5f 6b 65 79 20 3d 20 73 65 74 74 69 6e 67 73 2e _key = settings.
03e0: 67 65 74 5f 73 74 72 69 6e 67 28 22 61 70 69 5f get_string("api_
03f0: 6b 65 79 22 29 3f 3b 0a 09 09 6c 65 74 20 74 67 key")?;...let tg
0400: 20 3d 20 42 6f 74 3a 3a 6e 65 77 28 26 61 70 69 = Bot::new(&api
0410: 5f 6b 65 79 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 _key);....let mu
0420: 74 20 63 6c 69 65 6e 74 20 3d 20 72 65 71 77 65 t client = reqwe
0430: 73 74 3a 3a 43 6c 69 65 6e 74 3a 3a 62 75 69 6c st::Client::buil
0440: 64 65 72 28 29 3b 0a 09 09 69 66 20 6c 65 74 20 der();...if let
0450: 4f 6b 28 70 72 6f 78 79 29 20 3d 20 73 65 74 74 Ok(proxy) = sett
0460: 69 6e 67 73 2e 67 65 74 5f 73 74 72 69 6e 67 28 ings.get_string(
0470: 22 70 72 6f 78 79 22 29 20 7b 0a 09 09 09 6c 65 "proxy") {....le
0480: 74 20 70 72 6f 78 79 20 3d 20 72 65 71 77 65 73 t proxy = reqwes
0490: 74 3a 3a 50 72 6f 78 79 3a 3a 61 6c 6c 28 70 72 t::Proxy::all(pr
04a0: 6f 78 79 29 3f 3b 0a 09 09 09 63 6c 69 65 6e 74 oxy)?;....client
04b0: 20 3d 20 63 6c 69 65 6e 74 2e 70 72 6f 78 79 28 = client.proxy(
04c0: 70 72 6f 78 79 29 3b 0a 09 09 7d 0a 09 09 6c 65 proxy);...}...le
04d0: 74 20 68 74 74 70 5f 63 6c 69 65 6e 74 20 3d 20 t http_client =
04e0: 63 6c 69 65 6e 74 2e 62 75 69 6c 64 28 29 3f 3b client.build()?;
04f0: 0a 09 09 6c 65 74 20 6d 65 20 3d 20 74 67 2e 67 ...let me = tg.g
0500: 65 74 5f 6d 65 28 29 2e 61 77 61 69 74 3f 3b 0a et_me().await?;.
0510: 09 09 6c 65 74 20 6d 65 20 3d 20 6d 65 2e 72 65 ..let me = me.re
0520: 73 75 6c 74 3b 0a 09 09 6c 65 74 20 63 6f 72 65 sult;...let core
0530: 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 74 67 2c = Core {....tg,
0540: 0a 09 09 09 6d 65 2c 0a 09 09 09 6f 77 6e 65 72 ....me,....owner
0550: 5f 63 68 61 74 2c 0a 09 09 09 64 62 3a 20 44 62 _chat,....db: Db
0560: 3a 3a 6e 65 77 28 26 73 65 74 74 69 6e 67 73 2e ::new(&settings.
0570: 67 65 74 5f 73 74 72 69 6e 67 28 22 70 67 22 29 get_string("pg")
0580: 3f 29 3f 2c 0a 09 09 09 73 6f 75 72 63 65 73 3a ?)?,....sources:
0590: 20 41 72 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a Arc::new(Mutex:
05a0: 3a 6e 65 77 28 48 61 73 68 53 65 74 3a 3a 6e 65 :new(HashSet::ne
05b0: 77 28 29 29 29 2c 0a 09 09 09 68 74 74 70 5f 63 w())),....http_c
05c0: 6c 69 65 6e 74 2c 0a 09 09 09 6d 61 78 5f 64 65 lient,....max_de
05d0: 6c 61 79 3a 20 36 30 2c 0a 09 09 7d 3b 0a 09 09 lay: 60,...};...
05e0: 6c 65 74 20 6d 75 74 20 63 6c 6f 6e 65 20 3d 20 let mut clone =
05f0: 63 6f 72 65 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 core.clone();...
0600: 74 61 73 6b 3a 3a 73 70 61 77 6e 28 61 73 79 6e task::spawn(asyn
0610: 63 20 6d 6f 76 65 20 7b 0a 09 09 09 6c 6f 6f 70 c move {....loop
0620: 20 7b 0a 09 09 09 09 6c 65 74 20 64 65 6c 61 79 {.....let delay
0630: 20 3d 20 6d 61 74 63 68 20 26 63 6c 6f 6e 65 2e = match &clone.
0640: 61 75 74 6f 66 65 74 63 68 28 29 2e 61 77 61 69 autofetch().awai
0650: 74 20 7b 0a 09 09 09 09 09 45 72 72 28 65 72 72 t {......Err(err
0660: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 69 66 20 ) => {.......if
0670: 6c 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 let Err(err) = c
0680: 6c 6f 6e 65 2e 73 65 6e 64 28 66 6f 72 6d 61 74 lone.send(format
0690: 21 28 22 f0 9f 9b 91 20 7b 65 72 72 3a 3f 7d 22 !("š {err:?}"
06a0: 29 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 ), None, None).a
06b0: 77 61 69 74 20 7b 0a 09 09 09 09 09 09 09 65 70 wait {........ep
06c0: 72 69 6e 74 6c 6e 21 28 22 41 75 74 6f 66 65 74 rintln!("Autofet
06d0: 63 68 20 65 72 72 6f 72 3a 20 7b 65 72 72 3a 3f ch error: {err:?
06e0: 7d 22 29 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 }");.......};...
06f0: 09 09 09 09 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 ....std::time::D
0700: 75 72 61 74 69 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 uration::from_se
0710: 63 73 28 36 30 29 0a 09 09 09 09 09 7d 2c 0a 09 cs(60)......},..
0720: 09 09 09 09 4f 6b 28 74 69 6d 65 29 20 3d 3e 20 ....Ok(time) =>
0730: 2a 74 69 6d 65 2c 0a 09 09 09 09 7d 3b 0a 09 09 *time,.....};...
0740: 09 09 74 61 73 6b 3a 3a 73 6c 65 65 70 28 64 65 ..task::sleep(de
0750: 6c 61 79 29 2e 61 77 61 69 74 3b 0a 09 09 09 7d lay).await;....}
0760: 0a 09 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 ...});...Ok(core
0770: 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 )..}...pub async
0780: 20 66 6e 20 73 74 72 65 61 6d 28 26 6d 75 74 20 fn stream(&mut
0790: 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c self) -> Result<
07a0: 28 29 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 ()> {...let mut
07b0: 6f 66 66 73 65 74 3a 20 69 36 34 20 3d 20 30 3b offset: i64 = 0;
07c0: 0a 09 09 6c 65 74 20 6d 75 74 20 70 61 72 61 6d ...let mut param
07d0: 73 20 3d 20 47 65 74 55 70 64 61 74 65 73 50 61 s = GetUpdatesPa
07e0: 72 61 6d 73 20 7b 0a 09 09 09 6f 66 66 73 65 74 rams {....offset
07f0: 3a 20 4e 6f 6e 65 2c 0a 09 09 09 6c 69 6d 69 74 : None,....limit
0800: 3a 20 53 6f 6d 65 28 31 30 30 29 2c 0a 09 09 09 : Some(100),....
0810: 74 69 6d 65 6f 75 74 3a 20 53 6f 6d 65 28 33 30 timeout: Some(30
0820: 30 29 2c 0a 09 09 09 61 6c 6c 6f 77 65 64 5f 75 0),....allowed_u
0830: 70 64 61 74 65 73 3a 20 53 6f 6d 65 28 76 65 63 pdates: Some(vec
0840: 21 5b 41 6c 6c 6f 77 65 64 55 70 64 61 74 65 3a ![AllowedUpdate:
0850: 3a 4d 65 73 73 61 67 65 5d 29 2c 0a 09 09 7d 3b :Message]),...};
0860: 0a 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 6c 65 74 ...loop {....let
0870: 20 75 70 64 61 74 65 73 20 3d 20 73 65 6c 66 2e updates = self.
0880: 74 67 2e 67 65 74 5f 75 70 64 61 74 65 73 28 26 tg.get_updates(&
0890: 70 61 72 61 6d 73 29 2e 61 77 61 69 74 3f 2e 72 params).await?.r
08a0: 65 73 75 6c 74 3b 0a 09 09 09 69 66 20 75 70 64 esult;....if upd
08b0: 61 74 65 73 2e 69 73 5f 65 6d 70 74 79 28 29 20 ates.is_empty()
08c0: 7b 0a 09 09 09 09 6f 66 66 73 65 74 20 3d 20 30 {.....offset = 0
08d0: 3b 0a 09 09 09 09 70 61 72 61 6d 73 2e 6f 66 66 ;.....params.off
08e0: 73 65 74 20 3d 20 4e 6f 6e 65 3b 0a 09 09 09 09 set = None;.....
08f0: 63 6f 6e 74 69 6e 75 65 3b 0a 09 09 09 7d 0a 09 continue;....}..
0900: 09 09 66 6f 72 20 75 70 64 61 74 65 20 69 6e 20 ..for update in
0910: 75 70 64 61 74 65 73 20 7b 0a 09 09 09 09 69 66 updates {.....if
0920: 20 69 36 34 3a 3a 66 72 6f 6d 28 75 70 64 61 74 i64::from(updat
0930: 65 2e 75 70 64 61 74 65 5f 69 64 29 20 3e 3d 20 e.update_id) >=
0940: 6f 66 66 73 65 74 20 7b 0a 09 09 09 09 09 6f 66 offset {......of
0950: 66 73 65 74 20 3d 20 69 36 34 3a 3a 66 72 6f 6d fset = i64::from
0960: 28 75 70 64 61 74 65 2e 75 70 64 61 74 65 5f 69 (update.update_i
0970: 64 29 20 2b 20 31 3b 0a 09 09 09 09 09 70 61 72 d) + 1;......par
0980: 61 6d 73 2e 6f 66 66 73 65 74 20 3d 20 53 6f 6d ams.offset = Som
0990: 65 28 6f 66 66 73 65 74 29 3b 0a 09 09 09 09 7d e(offset);.....}
09a0: 0a 09 09 09 09 69 66 20 6c 65 74 20 55 70 64 61 .....if let Upda
09b0: 74 65 43 6f 6e 74 65 6e 74 3a 3a 4d 65 73 73 61 teContent::Messa
09c0: 67 65 28 6d 73 67 29 20 3d 20 75 70 64 61 74 65 ge(msg) = update
09d0: 2e 63 6f 6e 74 65 6e 74 20 7b 0a 09 09 09 09 09 .content {......
09e0: 69 66 20 6c 65 74 20 53 6f 6d 65 28 74 65 78 74 if let Some(text
09f0: 29 20 3d 20 6d 73 67 2e 74 65 78 74 20 7b 0a 09 ) = msg.text {..
0a00: 09 09 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 .....if let Some
0a10: 28 65 6e 74 69 74 69 65 73 29 20 3d 20 6d 73 67 (entities) = msg
0a20: 2e 65 6e 74 69 74 69 65 73 20 7b 0a 09 09 09 09 .entities {.....
0a30: 09 09 09 6c 65 74 20 63 68 61 72 73 3a 20 56 65 ...let chars: Ve
0a40: 63 3c 75 31 36 3e 20 3d 20 74 65 78 74 2e 65 6e c<u16> = text.en
0a50: 63 6f 64 65 5f 75 74 66 31 36 28 29 2e 63 6f 6c code_utf16().col
0a60: 6c 65 63 74 28 29 3b 0a 09 09 09 09 09 09 09 66 lect();........f
0a70: 6f 72 20 65 6e 74 69 74 79 20 69 6e 20 65 6e 74 or entity in ent
0a80: 69 74 69 65 73 20 7b 0a 09 09 09 09 09 09 09 09 ities {.........
0a90: 69 66 20 65 6e 74 69 74 79 2e 74 79 70 65 5f 66 if entity.type_f
0aa0: 69 65 6c 64 20 3d 3d 20 4d 65 73 73 61 67 65 45 ield == MessageE
0ab0: 6e 74 69 74 79 54 79 70 65 3a 3a 42 6f 74 43 6f ntityType::BotCo
0ac0: 6d 6d 61 6e 64 20 26 26 20 65 6e 74 69 74 79 2e mmand && entity.
0ad0: 6f 66 66 73 65 74 20 21 3d 20 30 20 7b 0a 09 09 offset != 0 {...
0ae0: 09 09 09 09 09 09 09 62 61 69 6c 21 28 22 63 6f .......bail!("co
0af0: 6d 6d 61 6e 64 73 20 73 68 6f 75 6c 64 20 62 65 mmands should be
0b00: 20 61 74 20 6d 65 73 73 61 67 65 20 73 74 61 72 at message star
0b10: 74 22 29 3b 0a 09 09 09 09 09 09 09 09 7d 3b 0a t");.........};.
0b20: 09 09 09 09 09 09 09 09 6c 65 74 20 63 6d 64 20 ........let cmd
0b30: 3d 20 53 74 72 69 6e 67 3a 3a 66 72 6f 6d 5f 75 = String::from_u
0b40: 74 66 31 36 5f 6c 6f 73 73 79 28 26 63 68 61 72 tf16_lossy(&char
0b50: 73 5b 65 6e 74 69 74 79 2e 6f 66 66 73 65 74 20 s[entity.offset
0b60: 61 73 20 75 73 69 7a 65 2e 2e 65 6e 74 69 74 79 as usize..entity
0b70: 2e 6c 65 6e 67 74 68 20 61 73 20 75 73 69 7a 65 .length as usize
0b80: 5d 29 3b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 ]);.........let
0b90: 77 6f 72 64 73 3a 20 56 65 63 3c 26 73 74 72 3e words: Vec<&str>
0ba0: 20 3d 20 74 65 78 74 2e 73 70 6c 69 74 5f 77 68 = text.split_wh
0bb0: 69 74 65 73 70 61 63 65 28 29 2e 63 6f 6c 6c 65 itespace().colle
0bc0: 63 74 28 29 3b 0a 09 09 09 09 09 09 09 09 6c 65 ct();.........le
0bd0: 74 20 72 65 73 20 3d 20 6d 61 74 63 68 20 63 6d t res = match cm
0be0: 64 2e 61 73 5f 72 65 66 28 29 20 7b 0a 09 09 09 d.as_ref() {....
0bf0: 09 09 09 09 09 09 22 2f 63 68 65 63 6b 22 20 7c ......"/check" |
0c00: 20 22 2f 63 6c 65 61 6e 22 20 7c 20 22 2f 65 6e "/clean" | "/en
0c10: 61 62 6c 65 22 20 7c 20 22 2f 64 65 6c 65 74 65 able" | "/delete
0c20: 22 20 7c 20 22 2f 64 69 73 61 62 6c 65 22 20 3d " | "/disable" =
0c30: 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 63 6f 6d 6d 61 > command::comma
0c40: 6e 64 28 73 65 6c 66 2c 20 6d 73 67 2e 63 68 61 nd(self, msg.cha
0c50: 74 2e 69 64 2c 20 77 6f 72 64 73 29 2e 61 77 61 t.id, words).awa
0c60: 69 74 2c 0a 09 09 09 09 09 09 09 09 09 22 2f 73 it,.........."/s
0c70: 74 61 72 74 22 20 3d 3e 20 63 6f 6d 6d 61 6e 64 tart" => command
0c80: 3a 3a 73 74 61 72 74 28 73 65 6c 66 2c 20 6d 73 ::start(self, ms
0c90: 67 2e 63 68 61 74 2e 69 64 29 2e 61 77 61 69 74 g.chat.id).await
0ca0: 2c 0a 09 09 09 09 09 09 09 09 09 22 2f 6c 69 73 ,.........."/lis
0cb0: 74 22 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 6c t" => command::l
0cc0: 69 73 74 28 73 65 6c 66 2c 20 6d 73 67 2e 63 68 ist(self, msg.ch
0cd0: 61 74 2e 69 64 29 2e 61 77 61 69 74 2c 0a 09 09 at.id).await,...
0ce0: 09 09 09 09 09 09 09 22 2f 61 64 64 22 20 7c 20 ......."/add" |
0cf0: 22 2f 75 70 64 61 74 65 22 20 3d 3e 20 63 6f 6d "/update" => com
0d00: 6d 61 6e 64 3a 3a 75 70 64 61 74 65 28 73 65 6c mand::update(sel
0d10: 66 2c 20 6d 73 67 2e 63 68 61 74 2e 69 64 2c 20 f, msg.chat.id,
0d20: 77 6f 72 64 73 29 2e 61 77 61 69 74 2c 0a 09 09 words).await,...
0d30: 09 09 09 09 09 09 09 61 6e 79 20 3d 3e 20 45 72 .......any => Er
0d40: 72 28 61 6e 79 68 6f 77 21 28 22 55 6e 6b 6e 6f r(anyhow!("Unkno
0d50: 77 6e 20 63 6f 6d 6d 61 6e 64 3a 20 7b 61 6e 79 wn command: {any
0d60: 7d 22 29 29 2c 0a 09 09 09 09 09 09 09 09 7d 3b }")),.........};
0d70: 0a 09 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 .........if let
0d80: 45 72 72 28 65 72 72 29 20 3d 20 72 65 73 20 7b Err(err) = res {
0d90: 0a 09 09 09 09 09 09 09 09 09 69 66 20 6c 65 74 ..........if let
0da0: 20 45 72 72 28 65 72 72 32 29 20 3d 20 73 65 6c Err(err2) = sel
0db0: 66 2e 73 65 6e 64 28 66 6f 72 6d 61 74 21 28 22 f.send(format!("
0dc0: 5c 5c 23 65 72 72 6f 72 5c 6e 60 60 60 5c 6e 7b \\#error\n```\n{
0dd0: 65 72 72 3a 3f 7d 5c 6e 60 60 60 22 29 2c 0a 09 err:?}\n```"),..
0de0: 09 09 09 09 09 09 09 09 09 53 6f 6d 65 28 6d 73 .........Some(ms
0df0: 67 2e 63 68 61 74 2e 69 64 29 2c 0a 09 09 09 09 g.chat.id),.....
0e00: 09 09 09 09 09 09 53 6f 6d 65 28 50 61 72 73 65 ......Some(Parse
0e10: 4d 6f 64 65 3a 3a 4d 61 72 6b 64 6f 77 6e 56 32 Mode::MarkdownV2
0e20: 29 0a 09 09 09 09 09 09 09 09 09 29 2e 61 77 61 )..........).awa
0e30: 69 74 7b 0a 09 09 09 09 09 09 09 09 09 09 64 62 it{...........db
0e40: 67 21 28 65 72 72 32 29 3b 0a 09 09 09 09 09 09 g!(err2);.......
0e50: 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 09 7d 0a ...};.........}.
0e60: 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 .......};.......
0e70: 7d 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 7d };......};.....}
0e80: 3b 0a 09 09 09 7d 0a 09 09 7d 0a 09 7d 0a 0a 09 ;....}...}..}...
0e90: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 73 65 6e pub async fn sen
0ea0: 64 20 3c 53 3e 28 26 73 65 6c 66 2c 20 6d 73 67 d <S>(&self, msg
0eb0: 3a 20 53 2c 20 74 61 72 67 65 74 3a 20 4f 70 74 : S, target: Opt
0ec0: 69 6f 6e 3c 69 36 34 3e 2c 20 6d 6f 64 65 3a 20 ion<i64>, mode:
0ed0: 4f 70 74 69 6f 6e 3c 50 61 72 73 65 4d 6f 64 65 Option<ParseMode
0ee0: 3e 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e >) -> Result<()>
0ef0: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
0f00: 53 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 74 20 String> {...let
0f10: 6d 73 67 20 3d 20 6d 73 67 2e 69 6e 74 6f 28 29 msg = msg.into()
0f20: 3b 0a 0a 09 09 6c 65 74 20 6d 6f 64 65 20 3d 20 ;....let mode =
0f30: 6d 6f 64 65 2e 75 6e 77 72 61 70 5f 6f 72 28 50 mode.unwrap_or(P
0f40: 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 3b arseMode::Html);
0f50: 0a 09 09 6c 65 74 20 74 61 72 67 65 74 20 3d 20 ...let target =
0f60: 74 61 72 67 65 74 2e 75 6e 77 72 61 70 5f 6f 72 target.unwrap_or
0f70: 28 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 61 74 (self.owner_chat
0f80: 29 3b 0a 09 09 6c 65 74 20 73 65 6e 64 20 3d 20 );...let send =
0f90: 53 65 6e 64 4d 65 73 73 61 67 65 50 61 72 61 6d SendMessageParam
0fa0: 73 3a 3a 62 75 69 6c 64 65 72 28 29 0a 09 09 09 s::builder()....
0fb0: 2e 63 68 61 74 5f 69 64 28 74 61 72 67 65 74 29 .chat_id(target)
0fc0: 0a 09 09 09 2e 74 65 78 74 28 6d 73 67 29 0a 09 .....text(msg)..
0fd0: 09 09 2e 70 61 72 73 65 5f 6d 6f 64 65 28 6d 6f ...parse_mode(mo
0fe0: 64 65 29 0a 09 09 09 2e 62 75 69 6c 64 28 29 3b de).....build();
0ff0: 0a 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 6d 61 74 ...loop {....mat
1000: 63 68 20 73 65 6c 66 2e 74 67 2e 73 65 6e 64 5f ch self.tg.send_
1010: 6d 65 73 73 61 67 65 28 26 73 65 6e 64 29 2e 61 message(&send).a
1020: 77 61 69 74 20 7b 0a 09 09 09 09 4f 6b 28 5f 29 wait {.....Ok(_)
1030: 20 3d 3e 20 62 72 65 61 6b 2c 0a 09 09 09 09 45 => break,.....E
1040: 72 72 28 65 72 72 29 20 3d 3e 20 6d 61 74 63 68 rr(err) => match
1050: 20 65 72 72 20 7b 0a 09 09 09 09 09 46 72 61 6e err {......Fran
1060: 6b 45 72 72 6f 72 3a 3a 41 70 69 28 72 65 66 20 kError::Api(ref
1070: 72 65 73 70 29 20 3d 3e 20 7b 0a 09 09 09 09 09 resp) => {......
1080: 09 69 66 20 72 65 73 70 2e 65 72 72 6f 72 5f 63 .if resp.error_c
1090: 6f 64 65 20 3d 3d 20 34 32 39 20 7b 0a 09 09 09 ode == 429 {....
10a0: 09 09 09 09 6c 65 74 20 6d 75 74 20 6d 79 5f 64 ....let mut my_d
10b0: 65 6c 61 79 20 3d 20 73 65 6c 66 2e 6d 61 78 5f elay = self.max_
10c0: 64 65 6c 61 79 3b 0a 09 09 09 09 09 09 09 69 66 delay;........if
10d0: 20 6c 65 74 20 53 6f 6d 65 28 70 61 72 61 6d 73 let Some(params
10e0: 29 20 3d 20 72 65 73 70 2e 70 61 72 61 6d 65 74 ) = resp.paramet
10f0: 65 72 73 20 7b 0a 09 09 09 09 09 09 09 09 69 66 ers {.........if
1100: 20 6c 65 74 20 53 6f 6d 65 28 64 65 6c 61 79 29 let Some(delay)
1110: 20 3d 20 70 61 72 61 6d 73 2e 72 65 74 72 79 5f = params.retry_
1120: 61 66 74 65 72 20 7b 0a 09 09 09 09 09 09 09 09 after {.........
1130: 09 69 66 20 64 65 6c 61 79 20 3c 20 6d 79 5f 64 .if delay < my_d
1140: 65 6c 61 79 20 7b 0a 09 09 09 09 09 09 09 09 09 elay {..........
1150: 09 6d 79 5f 64 65 6c 61 79 20 3d 20 64 65 6c 61 .my_delay = dela
1160: 79 3b 0a 09 09 09 09 09 09 09 09 09 7d 0a 09 09 y;..........}...
1170: 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 09 7d ......}........}
1180: 0a 09 09 09 09 09 09 09 74 61 73 6b 3a 3a 73 6c ........task::sl
1190: 65 65 70 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 eep(std::time::D
11a0: 75 72 61 74 69 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 uration::from_se
11b0: 63 73 28 6d 79 5f 64 65 6c 61 79 2e 69 6e 74 6f cs(my_delay.into
11c0: 28 29 29 29 2e 61 77 61 69 74 3b 0a 09 09 09 09 ())).await;.....
11d0: 09 09 7d 20 65 6c 73 65 20 7b 0a 09 09 09 09 09 ..} else {......
11e0: 09 09 72 65 74 75 72 6e 20 45 72 72 28 65 72 72 ..return Err(err
11f0: 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 09 09 09 .into());.......
1200: 7d 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 5f }......},......_
1210: 20 3d 3e 20 72 65 74 75 72 6e 20 45 72 72 28 65 => return Err(e
1220: 72 72 2e 69 6e 74 6f 28 29 29 2c 0a 09 09 09 09 rr.into()),.....
1230: 7d 2c 0a 09 09 09 7d 0a 09 09 7d 0a 09 09 4f 6b },....}...}...Ok
1240: 28 28 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 (())..}...pub as
1250: 79 6e 63 20 66 6e 20 63 68 65 63 6b 20 28 26 6d ync fn check (&m
1260: 75 74 20 73 65 6c 66 2c 20 69 64 3a 20 26 69 33 ut self, id: &i3
1270: 32 2c 20 6f 77 6e 65 72 3a 20 69 36 34 2c 20 72 2, owner: i64, r
1280: 65 61 6c 3a 20 62 6f 6f 6c 29 20 2d 3e 20 52 65 eal: bool) -> Re
1290: 73 75 6c 74 3c 53 74 72 69 6e 67 3e 20 7b 0a 09 sult<String> {..
12a0: 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 65 64 3a .let mut posted:
12b0: 20 69 33 32 20 3d 20 30 3b 0a 09 09 6c 65 74 20 i32 = 0;...let
12c0: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
12d0: 64 62 2e 62 65 67 69 6e 28 29 2e 61 77 61 69 74 db.begin().await
12e0: 3f 3b 0a 0a 09 09 6c 65 74 20 69 64 20 3d 20 7b ?;....let id = {
12f0: 0a 09 09 09 6c 65 74 20 6d 75 74 20 73 65 74 20 ....let mut set
1300: 3d 20 73 65 6c 66 2e 73 6f 75 72 63 65 73 2e 6c = self.sources.l
1310: 6f 63 6b 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a ock().unwrap();.
1320: 09 09 09 6d 61 74 63 68 20 73 65 74 2e 67 65 74 ...match set.get
1330: 28 69 64 29 20 7b 0a 09 09 09 09 53 6f 6d 65 28 (id) {.....Some(
1340: 69 64 29 20 3d 3e 20 69 64 2e 63 6c 6f 6e 65 28 id) => id.clone(
1350: 29 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b ),.....None => {
1360: 0a 09 09 09 09 09 6c 65 74 20 69 64 20 3d 20 41 ......let id = A
1370: 72 63 3a 3a 6e 65 77 28 2a 69 64 29 3b 0a 09 09 rc::new(*id);...
1380: 09 09 09 73 65 74 2e 69 6e 73 65 72 74 28 69 64 ...set.insert(id
1390: 2e 63 6c 6f 6e 65 28 29 29 3b 0a 09 09 09 09 09 .clone());......
13a0: 69 64 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 7d id.clone().....}
13b0: 2c 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 6c 65 ,....}...};...le
13c0: 74 20 63 6f 75 6e 74 20 3d 20 41 72 63 3a 3a 73 t count = Arc::s
13d0: 74 72 6f 6e 67 5f 63 6f 75 6e 74 28 26 69 64 29 trong_count(&id)
13e0: 3b 0a 09 09 69 66 20 63 6f 75 6e 74 20 3d 3d 20 ;...if count ==
13f0: 32 20 7b 0a 09 09 09 6c 65 74 20 73 6f 75 72 63 2 {....let sourc
1400: 65 20 3d 20 63 6f 6e 6e 2e 67 65 74 5f 73 6f 75 e = conn.get_sou
1410: 72 63 65 28 2a 69 64 2c 20 6f 77 6e 65 72 29 2e rce(*id, owner).
1420: 61 77 61 69 74 3f 3b 0a 09 09 09 63 6f 6e 6e 2e await?;....conn.
1430: 73 65 74 5f 73 63 72 61 70 65 28 2a 69 64 29 2e set_scrape(*id).
1440: 61 77 61 69 74 3f 3b 0a 09 09 09 6c 65 74 20 64 await?;....let d
1450: 65 73 74 69 6e 61 74 69 6f 6e 20 3d 20 6d 61 74 estination = mat
1460: 63 68 20 72 65 61 6c 20 7b 0a 09 09 09 09 74 72 ch real {.....tr
1470: 75 65 20 3d 3e 20 73 6f 75 72 63 65 2e 63 68 61 ue => source.cha
1480: 6e 6e 65 6c 5f 69 64 2c 0a 09 09 09 09 66 61 6c nnel_id,.....fal
1490: 73 65 20 3d 3e 20 73 6f 75 72 63 65 2e 6f 77 6e se => source.own
14a0: 65 72 2c 0a 09 09 09 7d 3b 0a 09 09 09 6c 65 74 er,....};....let
14b0: 20 6d 75 74 20 74 68 69 73 5f 66 65 74 63 68 3a mut this_fetch:
14c0: 20 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 6d 65 Option<DateTime
14d0: 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 <chrono::FixedOf
14e0: 66 73 65 74 3e 3e 20 3d 20 4e 6f 6e 65 3b 0a 09 fset>> = None;..
14f0: 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 73 3a ..let mut posts:
1500: 20 42 54 72 65 65 4d 61 70 3c 44 61 74 65 54 69 BTreeMap<DateTi
1510: 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 me<chrono::Fixed
1520: 4f 66 66 73 65 74 3e 2c 20 53 74 72 69 6e 67 3e Offset>, String>
1530: 20 3d 20 42 54 72 65 65 4d 61 70 3a 3a 6e 65 77 = BTreeMap::new
1540: 28 29 3b 0a 0a 09 09 09 6c 65 74 20 72 65 73 70 ();.....let resp
1550: 6f 6e 73 65 20 3d 20 73 65 6c 66 2e 68 74 74 70 onse = self.http
1560: 5f 63 6c 69 65 6e 74 2e 67 65 74 28 26 73 6f 75 _client.get(&sou
1570: 72 63 65 2e 75 72 6c 29 2e 73 65 6e 64 28 29 2e rce.url).send().
1580: 61 77 61 69 74 3f 3b 0a 09 09 09 6c 65 74 20 73 await?;....let s
1590: 74 61 74 75 73 20 3d 20 72 65 73 70 6f 6e 73 65 tatus = response
15a0: 2e 73 74 61 74 75 73 28 29 3b 0a 09 09 09 6c 65 .status();....le
15b0: 74 20 63 6f 6e 74 65 6e 74 20 3d 20 72 65 73 70 t content = resp
15c0: 6f 6e 73 65 2e 62 79 74 65 73 28 29 2e 61 77 61 onse.bytes().awa
15d0: 69 74 3f 3b 0a 09 09 09 6d 61 74 63 68 20 72 73 it?;....match rs
15e0: 73 3a 3a 43 68 61 6e 6e 65 6c 3a 3a 72 65 61 64 s::Channel::read
15f0: 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e _from(&content[.
1600: 2e 5d 29 20 7b 0a 09 09 09 09 4f 6b 28 66 65 65 .]) {.....Ok(fee
1610: 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 66 6f 72 d) => {......for
1620: 20 69 74 65 6d 20 69 6e 20 66 65 65 64 2e 69 74 item in feed.it
1630: 65 6d 73 28 29 20 7b 0a 09 09 09 09 09 09 69 66 ems() {.......if
1640: 20 6c 65 74 20 53 6f 6d 65 28 6c 69 6e 6b 29 20 let Some(link)
1650: 3d 20 69 74 65 6d 2e 6c 69 6e 6b 28 29 20 7b 0a = item.link() {.
1660: 09 09 09 09 09 09 09 6c 65 74 20 64 61 74 65 20 .......let date
1670: 3d 20 6d 61 74 63 68 20 69 74 65 6d 2e 70 75 62 = match item.pub
1680: 5f 64 61 74 65 28 29 20 7b 0a 09 09 09 09 09 09 _date() {.......
1690: 09 09 53 6f 6d 65 28 66 65 65 64 5f 64 61 74 65 ..Some(feed_date
16a0: 29 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 ) => DateTime::p
16b0: 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 32 38 32 arse_from_rfc282
16c0: 32 28 66 65 65 64 5f 64 61 74 65 29 2c 0a 09 09 2(feed_date),...
16d0: 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 44 61 ......None => Da
16e0: 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 teTime::parse_fr
16f0: 6f 6d 5f 72 66 63 33 33 33 39 28 26 69 74 65 6d om_rfc3339(&item
1700: 2e 64 75 62 6c 69 6e 5f 63 6f 72 65 5f 65 78 74 .dublin_core_ext
1710: 28 29 2e 75 6e 77 72 61 70 28 29 2e 64 61 74 65 ().unwrap().date
1720: 73 28 29 5b 30 5d 29 2c 0a 09 09 09 09 09 09 09 s()[0]),........
1730: 7d 3f 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 75 }?;........let u
1740: 72 6c 20 3d 20 6c 69 6e 6b 3b 0a 09 09 09 09 09 rl = link;......
1750: 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 64 ..posts.insert(d
1760: 61 74 65 2c 20 75 72 6c 2e 74 6f 5f 73 74 72 69 ate, url.to_stri
1770: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 7d 0a 09 ng());.......}..
1780: 09 09 09 09 7d 3b 0a 09 09 09 09 7d 2c 0a 09 09 ....};.....},...
1790: 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 6d 61 ..Err(err) => ma
17a0: 74 63 68 20 65 72 72 20 7b 0a 09 09 09 09 09 72 tch err {......r
17b0: 73 73 3a 3a 45 72 72 6f 72 3a 3a 49 6e 76 61 6c ss::Error::Inval
17c0: 69 64 53 74 61 72 74 54 61 67 20 3d 3e 20 7b 0a idStartTag => {.
17d0: 09 09 09 09 09 09 6d 61 74 63 68 20 61 74 6f 6d ......match atom
17e0: 5f 73 79 6e 64 69 63 61 74 69 6f 6e 3a 3a 46 65 _syndication::Fe
17f0: 65 64 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 ed::read_from(&c
1800: 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 09 ontent[..]) {...
1810: 09 09 09 09 09 4f 6b 28 66 65 65 64 29 20 3d 3e .....Ok(feed) =>
1820: 20 7b 0a 09 09 09 09 09 09 09 09 66 6f 72 20 69 {.........for i
1830: 74 65 6d 20 69 6e 20 66 65 65 64 2e 65 6e 74 72 tem in feed.entr
1840: 69 65 73 28 29 20 7b 0a 09 09 09 09 09 09 09 09 ies() {.........
1850: 09 6c 65 74 20 64 61 74 65 20 3d 20 69 74 65 6d .let date = item
1860: 2e 70 75 62 6c 69 73 68 65 64 28 29 2e 75 6e 77 .published().unw
1870: 72 61 70 28 29 3b 0a 09 09 09 09 09 09 09 09 09 rap();..........
1880: 6c 65 74 20 75 72 6c 20 3d 20 69 74 65 6d 2e 6c let url = item.l
1890: 69 6e 6b 73 28 29 5b 30 5d 2e 68 72 65 66 28 29 inks()[0].href()
18a0: 3b 0a 09 09 09 09 09 09 09 09 09 70 6f 73 74 73 ;..........posts
18b0: 2e 69 6e 73 65 72 74 28 2a 64 61 74 65 2c 20 75 .insert(*date, u
18c0: 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b rl.to_string());
18d0: 0a 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 .........};.....
18e0: 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 45 72 72 ...},........Err
18f0: 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 (err) => {......
1900: 09 09 09 62 61 69 6c 21 28 22 55 6e 73 75 70 70 ...bail!("Unsupp
1910: 6f 72 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 64 orted or mangled
1920: 20 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c content:\n{:?}\
1930: 6e 7b 65 72 72 3a 23 3f 7d 5c 6e 7b 73 74 61 74 n{err:#?}\n{stat
1940: 75 73 3a 23 3f 7d 5c 6e 22 2c 20 26 73 6f 75 72 us:#?}\n", &sour
1950: 63 65 2e 75 72 6c 29 0a 09 09 09 09 09 09 09 7d ce.url)........}
1960: 2c 0a 09 09 09 09 09 09 7d 0a 09 09 09 09 09 7d ,.......}......}
1970: 2c 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 6f ,......rss::Erro
1980: 72 3a 3a 45 6f 66 20 3d 3e 20 28 29 2c 0a 09 09 r::Eof => (),...
1990: 09 09 09 5f 20 3d 3e 20 62 61 69 6c 21 28 22 55 ..._ => bail!("U
19a0: 6e 73 75 70 70 6f 72 74 65 64 20 6f 72 20 6d 61 nsupported or ma
19b0: 6e 67 6c 65 64 20 63 6f 6e 74 65 6e 74 3a 5c 6e ngled content:\n
19c0: 7b 3a 3f 7d 5c 6e 7b 65 72 72 3a 23 3f 7d 5c 6e {:?}\n{err:#?}\n
19d0: 7b 73 74 61 74 75 73 3a 23 3f 7d 5c 6e 22 2c 20 {status:#?}\n",
19e0: 26 73 6f 75 72 63 65 2e 75 72 6c 29 0a 09 09 09 &source.url)....
19f0: 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 66 6f 72 20 .}....};....for
1a00: 28 64 61 74 65 2c 20 75 72 6c 29 20 69 6e 20 70 (date, url) in p
1a10: 6f 73 74 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 osts.iter() {...
1a20: 09 09 6c 65 74 20 70 6f 73 74 5f 75 72 6c 3a 20 ..let post_url:
1a30: 43 6f 77 3c 73 74 72 3e 20 3d 20 6d 61 74 63 68 Cow<str> = match
1a40: 20 73 6f 75 72 63 65 2e 75 72 6c 5f 72 65 20 7b source.url_re {
1a50: 0a 09 09 09 09 09 53 6f 6d 65 28 72 65 66 20 78 ......Some(ref x
1a60: 29 20 3d 3e 20 73 65 64 72 65 67 65 78 3a 3a 52 ) => sedregex::R
1a70: 65 70 6c 61 63 65 43 6f 6d 6d 61 6e 64 3a 3a 6e eplaceCommand::n
1a80: 65 77 28 78 29 3f 2e 65 78 65 63 75 74 65 28 75 ew(x)?.execute(u
1a90: 72 6c 29 2c 0a 09 09 09 09 09 4e 6f 6e 65 20 3d rl),......None =
1aa0: 3e 20 75 72 6c 2e 69 6e 74 6f 28 29 2c 0a 09 09 > url.into(),...
1ab0: 09 09 7d 3b 0a 09 09 09 09 69 66 20 6c 65 74 20 ..};.....if let
1ac0: 53 6f 6d 65 28 65 78 69 73 74 73 29 20 3d 20 63 Some(exists) = c
1ad0: 6f 6e 6e 2e 65 78 69 73 74 73 28 26 70 6f 73 74 onn.exists(&post
1ae0: 5f 75 72 6c 2c 20 2a 69 64 29 2e 61 77 61 69 74 _url, *id).await
1af0: 3f 20 7b 0a 09 09 09 09 09 69 66 20 21 20 65 78 ? {......if ! ex
1b00: 69 73 74 73 20 7b 0a 09 09 09 09 09 09 69 66 20 ists {.......if
1b10: 74 68 69 73 5f 66 65 74 63 68 2e 69 73 5f 6e 6f this_fetch.is_no
1b20: 6e 65 28 29 20 7c 7c 20 2a 64 61 74 65 20 3e 20 ne() || *date >
1b30: 74 68 69 73 5f 66 65 74 63 68 2e 75 6e 77 72 61 this_fetch.unwra
1b40: 70 28 29 20 7b 0a 09 09 09 09 09 09 09 74 68 69 p() {........thi
1b50: 73 5f 66 65 74 63 68 20 3d 20 53 6f 6d 65 28 2a s_fetch = Some(*
1b60: 64 61 74 65 29 3b 0a 09 09 09 09 09 09 7d 3b 0a date);.......};.
1b70: 09 09 09 09 09 09 73 65 6c 66 2e 73 65 6e 64 28 ......self.send(
1b80: 20 6d 61 74 63 68 20 26 73 6f 75 72 63 65 2e 69 match &source.i
1b90: 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 09 v_hash {........
1ba0: 53 6f 6d 65 28 68 61 73 68 29 20 3d 3e 20 66 6f Some(hash) => fo
1bb0: 72 6d 61 74 21 28 22 3c 61 20 68 72 65 66 3d 5c rmat!("<a href=\
1bc0: 22 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 "https://t.me/iv
1bd0: 3f 75 72 6c 3d 7b 70 6f 73 74 5f 75 72 6c 7d 26 ?url={post_url}&
1be0: 72 68 61 73 68 3d 7b 68 61 73 68 7d 5c 22 3e 20 rhash={hash}\">
1bf0: 3c 2f 61 3e 7b 70 6f 73 74 5f 75 72 6c 7d 22 29 </a>{post_url}")
1c00: 2c 0a 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e ,........None =>
1c10: 20 66 6f 72 6d 61 74 21 28 22 7b 70 6f 73 74 5f format!("{post_
1c20: 75 72 6c 7d 22 29 2c 0a 09 09 09 09 09 09 7d 2c url}"),.......},
1c30: 20 53 6f 6d 65 28 64 65 73 74 69 6e 61 74 69 6f Some(destinatio
1c40: 6e 29 2c 20 53 6f 6d 65 28 50 61 72 73 65 4d 6f n), Some(ParseMo
1c50: 64 65 3a 3a 48 74 6d 6c 29 29 2e 61 77 61 69 74 de::Html)).await
1c60: 3f 3b 0a 09 09 09 09 09 09 63 6f 6e 6e 2e 61 64 ?;.......conn.ad
1c70: 64 5f 70 6f 73 74 28 2a 69 64 2c 20 64 61 74 65 d_post(*id, date
1c80: 2c 20 26 70 6f 73 74 5f 75 72 6c 29 2e 61 77 61 , &post_url).awa
1c90: 69 74 3f 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 it?;......};....
1ca0: 09 7d 3b 0a 09 09 09 09 70 6f 73 74 65 64 20 2b .};.....posted +
1cb0: 3d 20 31 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 6f = 1;....};....po
1cc0: 73 74 73 2e 63 6c 65 61 72 28 29 3b 0a 09 09 7d sts.clear();...}
1cd0: 3b 0a 09 09 4f 6b 28 66 6f 72 6d 61 74 21 28 22 ;...Ok(format!("
1ce0: 50 6f 73 74 65 64 3a 20 7b 70 6f 73 74 65 64 7d Posted: {posted}
1cf0: 22 29 29 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 66 "))..}...async f
1d00: 6e 20 61 75 74 6f 66 65 74 63 68 28 26 6d 75 74 n autofetch(&mut
1d10: 20 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 self) -> Result
1d20: 3c 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 <std::time::Dura
1d30: 74 69 6f 6e 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 tion> {...let mu
1d40: 74 20 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f t delay = chrono
1d50: 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 ::Duration::minu
1d60: 74 65 73 28 31 29 3b 0a 09 09 6c 65 74 20 6e 6f tes(1);...let no
1d70: 77 20 3d 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 w = chrono::Loca
1d80: 6c 3a 3a 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 20 l::now();...let
1d90: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
1da0: 64 62 2e 62 65 67 69 6e 28 29 2e 61 77 61 69 74 db.begin().await
1db0: 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 ?;...for row in
1dc0: 63 6f 6e 6e 2e 67 65 74 5f 71 75 65 75 65 28 29 conn.get_queue()
1dd0: 2e 61 77 61 69 74 3f 20 7b 0a 09 09 09 69 66 20 .await? {....if
1de0: 6c 65 74 20 53 6f 6d 65 28 6e 65 78 74 5f 66 65 let Some(next_fe
1df0: 74 63 68 29 20 3d 20 72 6f 77 2e 6e 65 78 74 5f tch) = row.next_
1e00: 66 65 74 63 68 20 7b 0a 09 09 09 09 69 66 20 6e fetch {.....if n
1e10: 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 20 ext_fetch < now
1e20: 7b 0a 09 09 09 09 09 69 66 20 6c 65 74 20 28 53 {......if let (S
1e30: 6f 6d 65 28 6f 77 6e 65 72 29 2c 20 53 6f 6d 65 ome(owner), Some
1e40: 28 73 6f 75 72 63 65 5f 69 64 29 29 20 3d 20 28 (source_id)) = (
1e50: 72 6f 77 2e 6f 77 6e 65 72 2c 20 72 6f 77 2e 73 row.owner, row.s
1e60: 6f 75 72 63 65 5f 69 64 29 20 7b 0a 09 09 09 09 ource_id) {.....
1e70: 09 09 6c 65 74 20 6d 75 74 20 63 6c 6f 6e 65 20 ..let mut clone
1e80: 3d 20 43 6f 72 65 20 7b 0a 09 09 09 09 09 09 09 = Core {........
1e90: 6f 77 6e 65 72 5f 63 68 61 74 3a 20 6f 77 6e 65 owner_chat: owne
1ea0: 72 2c 0a 09 09 09 09 09 09 09 2e 2e 73 65 6c 66 r,..........self
1eb0: 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 09 09 7d .clone().......}
1ec0: 3b 0a 09 09 09 09 09 09 74 61 73 6b 3a 3a 73 70 ;.......task::sp
1ed0: 61 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b awn(async move {
1ee0: 0a 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 ........if let E
1ef0: 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e rr(err) = clone.
1f00: 63 68 65 63 6b 28 26 73 6f 75 72 63 65 5f 69 64 check(&source_id
1f10: 2c 20 6f 77 6e 65 72 2c 20 74 72 75 65 29 2e 61 , owner, true).a
1f20: 77 61 69 74 20 7b 0a 09 09 09 09 09 09 09 09 69 wait {.........i
1f30: 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 3d f let Err(err) =
1f40: 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 26 66 6f 72 clone.send(&for
1f50: 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 65 72 72 3a mat!("š {err:
1f60: 3f 7d 22 29 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 ?}"), None, None
1f70: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 ).await {.......
1f80: 09 09 09 65 70 72 69 6e 74 6c 6e 21 28 22 43 68 ...eprintln!("Ch
1f90: 65 63 6b 20 65 72 72 6f 72 3a 20 7b 65 72 72 3a eck error: {err:
1fa0: 3f 7d 22 29 3b 0a 09 09 09 09 09 09 09 09 09 2f ?}");........../
1fb0: 2f 20 63 6c 6f 6e 65 2e 64 69 73 61 62 6c 65 28 / clone.disable(
1fc0: 26 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e 65 &source_id, owne
1fd0: 72 29 2e 61 77 61 69 74 2e 75 6e 77 72 61 70 28 r).await.unwrap(
1fe0: 29 3b 0a 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 );.........};...
1ff0: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 29 .....};.......})
2000: 3b 0a 09 09 09 09 09 7d 0a 09 09 09 09 7d 20 65 ;......}.....} e
2010: 6c 73 65 20 69 66 20 6e 65 78 74 5f 66 65 74 63 lse if next_fetc
2020: 68 20 2d 20 6e 6f 77 20 3c 20 64 65 6c 61 79 20 h - now < delay
2030: 7b 0a 09 09 09 09 09 64 65 6c 61 79 20 3d 20 6e {......delay = n
2040: 65 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 3b ext_fetch - now;
2050: 0a 09 09 09 09 7d 0a 09 09 09 7d 0a 09 09 7d 3b .....}....}...};
2060: 0a 09 09 4f 6b 28 64 65 6c 61 79 2e 74 6f 5f 73 ...Ok(delay.to_s
2070: 74 64 28 29 3f 29 0a 09 7d 0a 0a 09 70 75 62 20 td()?)..}...pub
2080: 61 73 79 6e 63 20 66 6e 20 6c 69 73 74 20 28 26 async fn list (&
2090: 6d 75 74 20 73 65 6c 66 2c 20 6f 77 6e 65 72 3a mut self, owner:
20a0: 20 69 36 34 29 20 2d 3e 20 52 65 73 75 6c 74 3c i64) -> Result<
20b0: 53 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 74 20 String> {...let
20c0: 6d 75 74 20 72 65 70 6c 79 3a 20 56 65 63 3c 43 mut reply: Vec<C
20d0: 6f 77 3c 73 74 72 3e 3e 20 3d 20 76 65 63 21 5b ow<str>> = vec![
20e0: 5d 3b 0a 09 09 72 65 70 6c 79 2e 70 75 73 68 28 ];...reply.push(
20f0: 22 43 68 61 6e 6e 65 6c 73 3a 22 2e 69 6e 74 6f "Channels:".into
2100: 28 29 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 ());...let mut c
2110: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 64 62 2e 62 65 onn = self.db.be
2120: 67 69 6e 28 29 2e 61 77 61 69 74 3f 3b 0a 09 09 gin().await?;...
2130: 66 6f 72 20 72 6f 77 20 69 6e 20 63 6f 6e 6e 2e for row in conn.
2140: 67 65 74 5f 6c 69 73 74 28 6f 77 6e 65 72 29 2e get_list(owner).
2150: 61 77 61 69 74 3f 20 7b 0a 09 09 09 72 65 70 6c await? {....repl
2160: 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 y.push(format!("
2170: 5c 6e 5c 5c 23 ef b8 8f e2 83 a3 20 7b 7d 20 5c \n\\#ļøā£ {} \
2180: 5c 2a ef b8 8f e2 83 a3 20 60 7b 7d 60 20 7b 7d \*ļøā£ `{}` {}
2190: 5c 6e f0 9f 94 97 20 60 7b 7d 60 22 2c 20 72 6f \nš `{}`", ro
21a0: 77 2e 73 6f 75 72 63 65 5f 69 64 2c 20 72 6f 77 w.source_id, row
21b0: 2e 63 68 61 6e 6e 65 6c 2c 0a 09 09 09 09 6d 61 .channel,.....ma
21c0: 74 63 68 20 72 6f 77 2e 65 6e 61 62 6c 65 64 20 tch row.enabled
21d0: 7b 0a 09 09 09 09 09 74 72 75 65 20 20 3d 3e 20 {......true =>
21e0: 22 f0 9f 94 84 20 65 6e 61 62 6c 65 64 22 2c 0a "š enabled",.
21f0: 09 09 09 09 09 66 61 6c 73 65 20 3d 3e 20 22 e2 .....false => "ā
2200: 9b 94 20 64 69 73 61 62 6c 65 64 22 2c 0a 09 09 disabled",...
2210: 09 09 7d 2c 20 72 6f 77 2e 75 72 6c 29 2e 69 6e ..}, row.url).in
2220: 74 6f 28 29 29 3b 0a 09 09 09 69 66 20 6c 65 74 to());....if let
2230: 20 53 6f 6d 65 28 68 61 73 68 29 20 3d 20 26 72 Some(hash) = &r
2240: 6f 77 2e 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 ow.iv_hash {....
2250: 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d .reply.push(form
2260: 61 74 21 28 22 49 56 3a 20 60 7b 68 61 73 68 7d at!("IV: `{hash}
2270: 60 22 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 `").into());....
2280: 7d 0a 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 }....if let Some
2290: 28 72 65 29 20 3d 20 26 72 6f 77 2e 75 72 6c 5f (re) = &row.url_
22a0: 72 65 20 7b 0a 09 09 09 09 72 65 70 6c 79 2e 70 re {.....reply.p
22b0: 75 73 68 28 66 6f 72 6d 61 74 21 28 22 52 45 3a ush(format!("RE:
22c0: 20 60 7b 72 65 7d 60 22 29 2e 69 6e 74 6f 28 29 `{re}`").into()
22d0: 29 3b 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 4f );....}...};...O
22e0: 6b 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e k(reply.join("\n
22f0: 22 29 29 0a 09 7d 0a 7d 0a "))..}.}.