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 7d 3b 0a 0a 75 73 65 Set,..},.};..use
0070: 20 61 73 79 6e 63 5f 73 74 64 3a 3a 7b 0a 09 74 async_std::{..t
0080: 61 73 6b 2c 0a 09 73 79 6e 63 3a 3a 7b 0a 09 09 ask,..sync::{...
0090: 41 72 63 2c 0a 09 09 4d 75 74 65 78 0a 09 7d 2c Arc,...Mutex..},
00a0: 0a 7d 3b 0a 75 73 65 20 63 68 72 6f 6e 6f 3a 3a .};.use chrono::
00b0: 7b 0a 09 44 61 74 65 54 69 6d 65 2c 0a 09 4c 6f {..DateTime,..Lo
00c0: 63 61 6c 2c 0a 7d 3b 0a 75 73 65 20 6c 61 7a 79 cal,.};.use lazy
00d0: 5f 73 74 61 74 69 63 3a 3a 6c 61 7a 79 5f 73 74 _static::lazy_st
00e0: 61 74 69 63 3b 0a 75 73 65 20 72 65 67 65 78 3a atic;.use regex:
00f0: 3a 52 65 67 65 78 3b 0a 75 73 65 20 72 65 71 77 :Regex;.use reqw
0100: 65 73 74 3a 3a 68 65 61 64 65 72 3a 3a 7b 0a 09 est::header::{..
0110: 43 41 43 48 45 5f 43 4f 4e 54 52 4f 4c 2c 0a 09 CACHE_CONTROL,..
0120: 45 58 50 49 52 45 53 2c 0a 09 4c 41 53 54 5f 4d EXPIRES,..LAST_M
0130: 4f 44 49 46 49 45 44 0a 7d 3b 0a 75 73 65 20 74 ODIFIED.};.use t
0140: 67 62 6f 74 3a 3a 7b 0a 09 61 70 69 3a 3a 43 6c gbot::{..api::Cl
0150: 69 65 6e 74 2c 0a 09 68 61 6e 64 6c 65 72 3a 3a ient,..handler::
0160: 55 70 64 61 74 65 48 61 6e 64 6c 65 72 2c 0a 09 UpdateHandler,..
0170: 74 79 70 65 73 3a 3a 7b 0a 09 09 42 6f 74 2c 0a types::{...Bot,.
0180: 09 09 43 68 61 74 50 65 65 72 49 64 2c 0a 09 09 ..ChatPeerId,...
0190: 43 6f 6d 6d 61 6e 64 2c 0a 09 09 47 65 74 42 6f Command,...GetBo
01a0: 74 2c 0a 09 09 4d 65 73 73 61 67 65 2c 0a 09 09 t,...Message,...
01b0: 50 61 72 73 65 4d 6f 64 65 2c 0a 09 09 53 65 6e ParseMode,...Sen
01c0: 64 4d 65 73 73 61 67 65 2c 0a 09 09 55 70 64 61 dMessage,...Upda
01d0: 74 65 2c 0a 09 09 55 70 64 61 74 65 54 79 70 65 te,...UpdateType
01e0: 2c 0a 09 09 55 73 65 72 50 65 65 72 49 64 2c 0a ,...UserPeerId,.
01f0: 09 7d 2c 0a 7d 3b 0a 75 73 65 20 73 74 61 63 6b .},.};.use stack
0200: 65 64 5f 65 72 72 6f 72 73 3a 3a 7b 0a 09 52 65 ed_errors::{..Re
0210: 73 75 6c 74 2c 0a 09 53 74 61 63 6b 61 62 6c 65 sult,..Stackable
0220: 45 72 72 2c 0a 09 61 6e 79 68 6f 77 2c 0a 09 62 Err,..anyhow,..b
0230: 61 69 6c 2c 0a 7d 3b 0a 0a 6c 61 7a 79 5f 73 74 ail,.};..lazy_st
0240: 61 74 69 63 21 7b 0a 09 70 75 62 20 73 74 61 74 atic!{..pub stat
0250: 69 63 20 72 65 66 20 52 45 5f 53 50 45 43 49 41 ic ref RE_SPECIA
0260: 4c 3a 20 52 65 67 65 78 20 3d 20 52 65 67 65 78 L: Regex = Regex
0270: 3a 3a 6e 65 77 28 72 22 28 5b 5c 2d 5f 2a 5c 5b ::new(r"([\-_*\[
0280: 5c 5d 28 29 7e 60 3e 23 2b 7c 7b 7d 5c 2e 21 5d \]()~`>#+|{}\.!]
0290: 29 22 29 2e 75 6e 77 72 61 70 28 29 3b 0a 7d 0a )").unwrap();.}.
02a0: 0a 2f 2f 2f 20 45 6e 63 6f 64 65 73 20 73 70 65 ./// Encodes spe
02b0: 63 69 61 6c 20 48 54 4d 4c 20 65 6e 74 69 74 69 cial HTML entiti
02c0: 65 73 20 74 6f 20 70 72 65 76 65 6e 74 20 74 68 es to prevent th
02d0: 65 6d 20 69 6e 74 65 72 66 65 72 69 6e 67 20 77 em interfering w
02e0: 69 74 68 20 54 65 6c 65 67 72 61 6d 20 48 54 4d ith Telegram HTM
02f0: 4c 0a 70 75 62 20 66 6e 20 65 6e 63 6f 64 65 20 L.pub fn encode
0300: 28 74 65 78 74 3a 20 26 73 74 72 29 20 2d 3e 20 (text: &str) ->
0310: 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 20 7b 0a 09 Cow<'_, str> {..
0320: 52 45 5f 53 50 45 43 49 41 4c 2e 72 65 70 6c 61 RE_SPECIAL.repla
0330: 63 65 5f 61 6c 6c 28 74 65 78 74 2c 20 22 5c 5c ce_all(text, "\\
0340: 24 31 22 29 0a 7d 0a 0a 23 5b 64 65 72 69 76 65 $1").}..#[derive
0350: 28 43 6c 6f 6e 65 29 5d 0a 70 75 62 20 73 74 72 (Clone)].pub str
0360: 75 63 74 20 43 6f 72 65 20 7b 0a 09 6f 77 6e 65 uct Core {..owne
0370: 72 5f 63 68 61 74 3a 20 43 68 61 74 50 65 65 72 r_chat: ChatPeer
0380: 49 64 2c 0a 09 2f 2f 20 6d 61 78 5f 64 65 6c 61 Id,..// max_dela
0390: 79 3a 20 75 31 36 2c 0a 09 70 75 62 20 74 67 3a y: u16,..pub tg:
03a0: 20 43 6c 69 65 6e 74 2c 0a 09 70 75 62 20 6d 65 Client,..pub me
03b0: 3a 20 42 6f 74 2c 0a 09 70 75 62 20 64 62 3a 20 : Bot,..pub db:
03c0: 44 62 2c 0a 09 73 6f 75 72 63 65 73 3a 20 41 72 Db,..sources: Ar
03d0: 63 3c 4d 75 74 65 78 3c 48 61 73 68 53 65 74 3c c<Mutex<HashSet<
03e0: 41 72 63 3c 69 33 32 3e 3e 3e 3e 2c 0a 09 68 74 Arc<i32>>>>,..ht
03f0: 74 70 5f 63 6c 69 65 6e 74 3a 20 72 65 71 77 65 tp_client: reqwe
0400: 73 74 3a 3a 43 6c 69 65 6e 74 2c 0a 7d 0a 0a 69 st::Client,.}..i
0410: 6d 70 6c 20 43 6f 72 65 20 7b 0a 09 70 75 62 20 mpl Core {..pub
0420: 61 73 79 6e 63 20 66 6e 20 6e 65 77 28 73 65 74 async fn new(set
0430: 74 69 6e 67 73 3a 20 63 6f 6e 66 69 67 3a 3a 43 tings: config::C
0440: 6f 6e 66 69 67 29 20 2d 3e 20 52 65 73 75 6c 74 onfig) -> Result
0450: 3c 43 6f 72 65 3e 20 7b 0a 09 09 6c 65 74 20 6f <Core> {...let o
0460: 77 6e 65 72 5f 63 68 61 74 20 3d 20 43 68 61 74 wner_chat = Chat
0470: 50 65 65 72 49 64 3a 3a 66 72 6f 6d 28 73 65 74 PeerId::from(set
0480: 74 69 6e 67 73 2e 67 65 74 5f 69 6e 74 28 22 6f tings.get_int("o
0490: 77 6e 65 72 22 29 2e 73 74 61 63 6b 28 29 3f 29 wner").stack()?)
04a0: 3b 0a 09 09 6c 65 74 20 61 70 69 5f 6b 65 79 20 ;...let api_key
04b0: 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 5f 73 = settings.get_s
04c0: 74 72 69 6e 67 28 22 61 70 69 5f 6b 65 79 22 29 tring("api_key")
04d0: 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 6c 65 74 .stack()?;...let
04e0: 20 74 67 20 3d 20 43 6c 69 65 6e 74 3a 3a 6e 65 tg = Client::ne
04f0: 77 28 26 61 70 69 5f 6b 65 79 29 2e 73 74 61 63 w(&api_key).stac
0500: 6b 28 29 3f 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 k()?;....let mut
0510: 20 63 6c 69 65 6e 74 20 3d 20 72 65 71 77 65 73 client = reqwes
0520: 74 3a 3a 43 6c 69 65 6e 74 3a 3a 62 75 69 6c 64 t::Client::build
0530: 65 72 28 29 3b 0a 09 09 69 66 20 6c 65 74 20 4f er();...if let O
0540: 6b 28 70 72 6f 78 79 29 20 3d 20 73 65 74 74 69 k(proxy) = setti
0550: 6e 67 73 2e 67 65 74 5f 73 74 72 69 6e 67 28 22 ngs.get_string("
0560: 70 72 6f 78 79 22 29 20 7b 0a 09 09 09 6c 65 74 proxy") {....let
0570: 20 70 72 6f 78 79 20 3d 20 72 65 71 77 65 73 74 proxy = reqwest
0580: 3a 3a 50 72 6f 78 79 3a 3a 61 6c 6c 28 70 72 6f ::Proxy::all(pro
0590: 78 79 29 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 xy).stack()?;...
05a0: 09 63 6c 69 65 6e 74 20 3d 20 63 6c 69 65 6e 74 .client = client
05b0: 2e 70 72 6f 78 79 28 70 72 6f 78 79 29 3b 0a 09 .proxy(proxy);..
05c0: 09 7d 0a 09 09 6c 65 74 20 68 74 74 70 5f 63 6c .}...let http_cl
05d0: 69 65 6e 74 20 3d 20 63 6c 69 65 6e 74 2e 62 75 ient = client.bu
05e0: 69 6c 64 28 29 2e 73 74 61 63 6b 28 29 3f 3b 0a ild().stack()?;.
05f0: 09 09 6c 65 74 20 6d 65 20 3d 20 74 67 2e 65 78 ..let me = tg.ex
0600: 65 63 75 74 65 28 47 65 74 42 6f 74 29 2e 61 77 ecute(GetBot).aw
0610: 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 ait.stack()?;...
0620: 6c 65 74 20 63 6f 72 65 20 3d 20 43 6f 72 65 20 let core = Core
0630: 7b 0a 09 09 09 74 67 2c 0a 09 09 09 6d 65 2c 0a {....tg,....me,.
0640: 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 2c 0a 09 ...owner_chat,..
0650: 09 09 64 62 3a 20 44 62 3a 3a 6e 65 77 28 26 73 ..db: Db::new(&s
0660: 65 74 74 69 6e 67 73 2e 67 65 74 5f 73 74 72 69 ettings.get_stri
0670: 6e 67 28 22 70 67 22 29 2e 73 74 61 63 6b 28 29 ng("pg").stack()
0680: 3f 29 3f 2c 0a 09 09 09 73 6f 75 72 63 65 73 3a ?)?,....sources:
0690: 20 41 72 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a Arc::new(Mutex:
06a0: 3a 6e 65 77 28 48 61 73 68 53 65 74 3a 3a 6e 65 :new(HashSet::ne
06b0: 77 28 29 29 29 2c 0a 09 09 09 68 74 74 70 5f 63 w())),....http_c
06c0: 6c 69 65 6e 74 2c 0a 09 09 09 2f 2f 20 6d 61 78 lient,....// max
06d0: 5f 64 65 6c 61 79 3a 20 36 30 2c 0a 09 09 7d 3b _delay: 60,...};
06e0: 0a 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 63 ...let clone = c
06f0: 6f 72 65 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 74 ore.clone();...t
0700: 61 73 6b 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 ask::spawn(async
0710: 20 6d 6f 76 65 20 7b 0a 09 09 09 6c 6f 6f 70 20 move {....loop
0720: 7b 0a 09 09 09 09 6c 65 74 20 64 65 6c 61 79 20 {.....let delay
0730: 3d 20 6d 61 74 63 68 20 26 63 6c 6f 6e 65 2e 61 = match &clone.a
0740: 75 74 6f 66 65 74 63 68 28 29 2e 61 77 61 69 74 utofetch().await
0750: 20 7b 0a 09 09 09 09 09 45 72 72 28 65 72 72 29 {......Err(err)
0760: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 69 66 20 6c => {.......if l
0770: 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c et Err(err) = cl
0780: 6f 6e 65 2e 73 65 6e 64 28 66 6f 72 6d 61 74 21 one.send(format!
0790: 28 22 f0 9f 9b 91 20 7b 65 72 72 7d 22 29 2c 20 ("š {err}"),
07a0: 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 None, None).awai
07b0: 74 20 7b 0a 09 09 09 09 09 09 09 65 70 72 69 6e t {........eprin
07c0: 74 6c 6e 21 28 22 41 75 74 6f 66 65 74 63 68 20 tln!("Autofetch
07d0: 65 72 72 6f 72 3a 20 7b 65 72 72 3a 3f 7d 22 29 error: {err:?}")
07e0: 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ;.......};......
07f0: 09 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 .std::time::Dura
0800: 74 69 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 63 73 28 tion::from_secs(
0810: 36 30 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 60)......},.....
0820: 09 4f 6b 28 74 69 6d 65 29 20 3d 3e 20 2a 74 69 .Ok(time) => *ti
0830: 6d 65 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 74 me,.....};.....t
0840: 61 73 6b 3a 3a 73 6c 65 65 70 28 64 65 6c 61 79 ask::sleep(delay
0850: 29 2e 61 77 61 69 74 3b 0a 09 09 09 7d 0a 09 09 ).await;....}...
0860: 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 0a 09 });...Ok(core)..
0870: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
0880: 20 73 65 6e 64 20 3c 53 3e 28 26 73 65 6c 66 2c send <S>(&self,
0890: 20 6d 73 67 3a 20 53 2c 20 74 61 72 67 65 74 3a msg: S, target:
08a0: 20 4f 70 74 69 6f 6e 3c 43 68 61 74 50 65 65 72 Option<ChatPeer
08b0: 49 64 3e 2c 20 6d 6f 64 65 3a 20 4f 70 74 69 6f Id>, mode: Optio
08c0: 6e 3c 50 61 72 73 65 4d 6f 64 65 3e 29 20 2d 3e n<ParseMode>) ->
08d0: 20 52 65 73 75 6c 74 3c 4d 65 73 73 61 67 65 3e Result<Message>
08e0: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
08f0: 53 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 74 20 String> {...let
0900: 6d 73 67 20 3d 20 6d 73 67 2e 69 6e 74 6f 28 29 msg = msg.into()
0910: 3b 0a 0a 09 09 6c 65 74 20 6d 6f 64 65 20 3d 20 ;....let mode =
0920: 6d 6f 64 65 2e 75 6e 77 72 61 70 5f 6f 72 28 50 mode.unwrap_or(P
0930: 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 3b arseMode::Html);
0940: 0a 09 09 6c 65 74 20 74 61 72 67 65 74 20 3d 20 ...let target =
0950: 74 61 72 67 65 74 2e 75 6e 77 72 61 70 5f 6f 72 target.unwrap_or
0960: 28 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 61 74 (self.owner_chat
0970: 29 3b 0a 09 09 73 65 6c 66 2e 74 67 2e 65 78 65 );...self.tg.exe
0980: 63 75 74 65 28 0a 09 09 09 53 65 6e 64 4d 65 73 cute(....SendMes
0990: 73 61 67 65 3a 3a 6e 65 77 28 74 61 72 67 65 74 sage::new(target
09a0: 2c 20 6d 73 67 29 0a 09 09 09 09 2e 77 69 74 68 , msg)......with
09b0: 5f 70 61 72 73 65 5f 6d 6f 64 65 28 6d 6f 64 65 _parse_mode(mode
09c0: 29 0a 09 09 29 2e 61 77 61 69 74 2e 73 74 61 63 )...).await.stac
09d0: 6b 28 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 k()..}...pub asy
09e0: 6e 63 20 66 6e 20 63 68 65 63 6b 20 28 26 73 65 nc fn check (&se
09f0: 6c 66 2c 20 69 64 3a 20 69 33 32 2c 20 72 65 61 lf, id: i32, rea
0a00: 6c 3a 20 62 6f 6f 6c 2c 20 6c 61 73 74 5f 73 63 l: bool, last_sc
0a10: 72 61 70 65 3a 20 4f 70 74 69 6f 6e 3c 44 61 74 rape: Option<Dat
0a20: 65 54 69 6d 65 3c 4c 6f 63 61 6c 3e 3e 29 20 2d eTime<Local>>) -
0a30: 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e > Result<String>
0a40: 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 {...let mut pos
0a50: 74 65 64 3a 20 69 33 32 20 3d 20 30 3b 0a 09 09 ted: i32 = 0;...
0a60: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
0a70: 65 6c 66 2e 64 62 2e 62 65 67 69 6e 28 29 2e 61 elf.db.begin().a
0a80: 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 0a wait.stack()?;..
0a90: 09 09 6c 65 74 20 69 64 20 3d 20 7b 0a 09 09 09 ..let id = {....
0aa0: 6c 65 74 20 6d 75 74 20 73 65 74 20 3d 20 73 65 let mut set = se
0ab0: 6c 66 2e 73 6f 75 72 63 65 73 2e 6c 6f 63 6b 5f lf.sources.lock_
0ac0: 61 72 63 28 29 2e 61 77 61 69 74 3b 0a 09 09 09 arc().await;....
0ad0: 6d 61 74 63 68 20 73 65 74 2e 67 65 74 28 26 69 match set.get(&i
0ae0: 64 29 20 7b 0a 09 09 09 09 53 6f 6d 65 28 69 64 d) {.....Some(id
0af0: 29 20 3d 3e 20 69 64 2e 63 6c 6f 6e 65 28 29 2c ) => id.clone(),
0b00: 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 .....None => {..
0b10: 09 09 09 09 6c 65 74 20 69 64 20 3d 20 41 72 63 ....let id = Arc
0b20: 3a 3a 6e 65 77 28 69 64 29 3b 0a 09 09 09 09 09 ::new(id);......
0b30: 73 65 74 2e 69 6e 73 65 72 74 28 69 64 2e 63 6c set.insert(id.cl
0b40: 6f 6e 65 28 29 29 3b 0a 09 09 09 09 09 69 64 2e one());......id.
0b50: 63 6c 6f 6e 65 28 29 0a 09 09 09 09 7d 2c 0a 09 clone().....},..
0b60: 09 09 7d 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 63 ..}...};...let c
0b70: 6f 75 6e 74 20 3d 20 41 72 63 3a 3a 73 74 72 6f ount = Arc::stro
0b80: 6e 67 5f 63 6f 75 6e 74 28 26 69 64 29 3b 0a 09 ng_count(&id);..
0b90: 09 69 66 20 63 6f 75 6e 74 20 3d 3d 20 32 20 7b .if count == 2 {
0ba0: 0a 09 09 09 6c 65 74 20 73 6f 75 72 63 65 20 3d ....let source =
0bb0: 20 63 6f 6e 6e 2e 67 65 74 5f 73 6f 75 72 63 65 conn.get_source
0bc0: 28 2a 69 64 2c 20 73 65 6c 66 2e 6f 77 6e 65 72 (*id, self.owner
0bd0: 5f 63 68 61 74 29 2e 61 77 61 69 74 2e 73 74 61 _chat).await.sta
0be0: 63 6b 28 29 3f 3b 0a 09 09 09 63 6f 6e 6e 2e 73 ck()?;....conn.s
0bf0: 65 74 5f 73 63 72 61 70 65 28 2a 69 64 29 2e 61 et_scrape(*id).a
0c00: 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 wait.stack()?;..
0c10: 09 09 6c 65 74 20 64 65 73 74 69 6e 61 74 69 6f ..let destinatio
0c20: 6e 20 3d 20 43 68 61 74 50 65 65 72 49 64 3a 3a n = ChatPeerId::
0c30: 66 72 6f 6d 28 6d 61 74 63 68 20 72 65 61 6c 20 from(match real
0c40: 7b 0a 09 09 09 09 74 72 75 65 20 3d 3e 20 73 6f {.....true => so
0c50: 75 72 63 65 2e 63 68 61 6e 6e 65 6c 5f 69 64 2c urce.channel_id,
0c60: 0a 09 09 09 09 66 61 6c 73 65 20 3d 3e 20 73 6f .....false => so
0c70: 75 72 63 65 2e 6f 77 6e 65 72 2c 0a 09 09 09 7d urce.owner,....}
0c80: 29 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 74 68 );....let mut th
0c90: 69 73 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e is_fetch: Option
0ca0: 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f <DateTime<chrono
0cb0: 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 3e 20 ::FixedOffset>>
0cc0: 3d 20 4e 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 6d = None;....let m
0cd0: 75 74 20 70 6f 73 74 73 3a 20 42 54 72 65 65 4d ut posts: BTreeM
0ce0: 61 70 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f ap<DateTime<chro
0cf0: 6e 6f 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e no::FixedOffset>
0d00: 2c 20 53 74 72 69 6e 67 3e 20 3d 20 42 54 72 65 , String> = BTre
0d10: 65 4d 61 70 3a 3a 6e 65 77 28 29 3b 0a 0a 09 09 eMap::new();....
0d20: 09 6c 65 74 20 6d 75 74 20 62 75 69 6c 64 65 72 .let mut builder
0d30: 20 3d 20 73 65 6c 66 2e 68 74 74 70 5f 63 6c 69 = self.http_cli
0d40: 65 6e 74 2e 67 65 74 28 26 73 6f 75 72 63 65 2e ent.get(&source.
0d50: 75 72 6c 29 3b 0a 09 09 09 69 66 20 6c 65 74 20 url);....if let
0d60: 53 6f 6d 65 28 6c 61 73 74 5f 73 63 72 61 70 65 Some(last_scrape
0d70: 29 20 3d 20 6c 61 73 74 5f 73 63 72 61 70 65 20 ) = last_scrape
0d80: 7b 0a 09 09 09 09 62 75 69 6c 64 65 72 20 3d 20 {.....builder =
0d90: 62 75 69 6c 64 65 72 2e 68 65 61 64 65 72 28 4c builder.header(L
0da0: 41 53 54 5f 4d 4f 44 49 46 49 45 44 2c 20 6c 61 AST_MODIFIED, la
0db0: 73 74 5f 73 63 72 61 70 65 2e 74 6f 5f 72 66 63 st_scrape.to_rfc
0dc0: 32 38 32 32 28 29 29 3b 0a 09 09 09 7d 3b 0a 09 2822());....};..
0dd0: 09 09 6c 65 74 20 72 65 73 70 6f 6e 73 65 20 3d ..let response =
0de0: 20 62 75 69 6c 64 65 72 2e 73 65 6e 64 28 29 2e builder.send().
0df0: 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a await.stack()?;.
0e00: 09 09 09 7b 0a 09 09 09 09 6c 65 74 20 68 65 61 ...{.....let hea
0e10: 64 65 72 73 20 3d 20 72 65 73 70 6f 6e 73 65 2e ders = response.
0e20: 68 65 61 64 65 72 73 28 29 3b 0a 09 09 09 09 6c headers();.....l
0e30: 65 74 20 65 78 70 69 72 65 73 20 3d 20 68 65 61 et expires = hea
0e40: 64 65 72 73 2e 67 65 74 28 45 58 50 49 52 45 53 ders.get(EXPIRES
0e50: 29 3b 0a 09 09 09 09 6c 65 74 20 63 61 63 68 65 );.....let cache
0e60: 20 3d 20 68 65 61 64 65 72 73 2e 67 65 74 28 43 = headers.get(C
0e70: 41 43 48 45 5f 43 4f 4e 54 52 4f 4c 29 3b 0a 09 ACHE_CONTROL);..
0e80: 09 09 09 69 66 20 65 78 70 69 72 65 73 2e 69 73 ...if expires.is
0e90: 5f 73 6f 6d 65 28 29 20 7c 7c 20 63 61 63 68 65 _some() || cache
0ea0: 2e 69 73 5f 73 6f 6d 65 28 29 20 7b 0a 09 09 09 .is_some() {....
0eb0: 09 09 70 72 69 6e 74 6c 6e 21 28 22 7b 7d 20 7b ..println!("{} {
0ec0: 7d 20 7b 3a 3f 7d 20 7b 3a 3f 7d 20 7b 3a 3f 7d } {:?} {:?} {:?}
0ed0: 22 2c 20 4c 6f 63 61 6c 3a 3a 6e 6f 77 28 29 2e ", Local::now().
0ee0: 74 6f 5f 72 66 63 32 38 32 32 28 29 2c 20 26 73 to_rfc2822(), &s
0ef0: 6f 75 72 63 65 2e 75 72 6c 2c 20 6c 61 73 74 5f ource.url, last_
0f00: 73 63 72 61 70 65 2c 20 65 78 70 69 72 65 73 2c scrape, expires,
0f10: 20 63 61 63 68 65 29 3b 0a 09 09 09 09 7d 0a 09 cache);.....}..
0f20: 09 09 7d 0a 09 09 09 6c 65 74 20 73 74 61 74 75 ..}....let statu
0f30: 73 20 3d 20 72 65 73 70 6f 6e 73 65 2e 73 74 61 s = response.sta
0f40: 74 75 73 28 29 3b 0a 09 09 09 6c 65 74 20 63 6f tus();....let co
0f50: 6e 74 65 6e 74 20 3d 20 72 65 73 70 6f 6e 73 65 ntent = response
0f60: 2e 62 79 74 65 73 28 29 2e 61 77 61 69 74 2e 73 .bytes().await.s
0f70: 74 61 63 6b 28 29 3f 3b 0a 09 09 09 6d 61 74 63 tack()?;....matc
0f80: 68 20 72 73 73 3a 3a 43 68 61 6e 6e 65 6c 3a 3a h rss::Channel::
0f90: 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 read_from(&conte
0fa0: 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 09 09 09 4f 6b nt[..]) {.....Ok
0fb0: 28 66 65 65 64 29 20 3d 3e 20 7b 0a 09 09 09 09 (feed) => {.....
0fc0: 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 65 65 .for item in fee
0fd0: 64 2e 69 74 65 6d 73 28 29 20 7b 0a 09 09 09 09 d.items() {.....
0fe0: 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 6c 69 ..if let Some(li
0ff0: 6e 6b 29 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 28 nk) = item.link(
1000: 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 64 ) {........let d
1010: 61 74 65 20 3d 20 6d 61 74 63 68 20 69 74 65 6d ate = match item
1020: 2e 70 75 62 5f 64 61 74 65 28 29 20 7b 0a 09 09 .pub_date() {...
1030: 09 09 09 09 09 09 53 6f 6d 65 28 66 65 65 64 5f ......Some(feed_
1040: 64 61 74 65 29 20 3d 3e 20 44 61 74 65 54 69 6d date) => DateTim
1050: 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 e::parse_from_rf
1060: 63 32 38 32 32 28 66 65 65 64 5f 64 61 74 65 29 c2822(feed_date)
1070: 2c 0a 09 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d ,.........None =
1080: 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 > DateTime::pars
1090: 65 5f 66 72 6f 6d 5f 72 66 63 33 33 33 39 28 26 e_from_rfc3339(&
10a0: 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 63 6f 72 65 item.dublin_core
10b0: 5f 65 78 74 28 29 2e 75 6e 77 72 61 70 28 29 2e _ext().unwrap().
10c0: 64 61 74 65 73 28 29 5b 30 5d 29 2c 0a 09 09 09 dates()[0]),....
10d0: 09 09 09 09 7d 2e 73 74 61 63 6b 28 29 3f 3b 0a ....}.stack()?;.
10e0: 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c 20 3d .......let url =
10f0: 20 6c 69 6e 6b 3b 0a 09 09 09 09 09 09 09 70 6f link;........po
1100: 73 74 73 2e 69 6e 73 65 72 74 28 64 61 74 65 2c sts.insert(date,
1110: 20 75 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 url.to_string()
1120: 29 3b 0a 09 09 09 09 09 09 7d 0a 09 09 09 09 09 );.......}......
1130: 7d 3b 0a 09 09 09 09 7d 2c 0a 09 09 09 09 45 72 };.....},.....Er
1140: 72 28 65 72 72 29 20 3d 3e 20 6d 61 74 63 68 20 r(err) => match
1150: 65 72 72 20 7b 0a 09 09 09 09 09 72 73 73 3a 3a err {......rss::
1160: 45 72 72 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 74 Error::InvalidSt
1170: 61 72 74 54 61 67 20 3d 3e 20 7b 0a 09 09 09 09 artTag => {.....
1180: 09 09 6d 61 74 63 68 20 61 74 6f 6d 5f 73 79 6e ..match atom_syn
1190: 64 69 63 61 74 69 6f 6e 3a 3a 46 65 65 64 3a 3a dication::Feed::
11a0: 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 read_from(&conte
11b0: 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 09 09 09 09 09 nt[..]) {.......
11c0: 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 7b 0a 09 .Ok(feed) => {..
11d0: 09 09 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 .......for item
11e0: 69 6e 20 66 65 65 64 2e 65 6e 74 72 69 65 73 28 in feed.entries(
11f0: 29 20 7b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 ) {..........let
1200: 20 64 61 74 65 20 3d 20 69 74 65 6d 2e 70 75 62 date = item.pub
1210: 6c 69 73 68 65 64 28 29 2e 75 6e 77 72 61 70 28 lished().unwrap(
1220: 29 3b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 );..........let
1230: 75 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 url = item.links
1240: 28 29 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 ()[0].href();...
1250: 09 09 09 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 .......posts.ins
1260: 65 72 74 28 2a 64 61 74 65 2c 20 75 72 6c 2e 74 ert(*date, url.t
1270: 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 o_string());....
1280: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 7d .....};........}
1290: 2c 0a 09 09 09 09 09 09 09 45 72 72 28 65 72 72 ,........Err(err
12a0: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 62 ) => {.........b
12b0: 61 69 6c 21 28 22 55 6e 73 75 70 70 6f 72 74 65 ail!("Unsupporte
12c0: 64 20 6f 72 20 6d 61 6e 67 6c 65 64 20 63 6f 6e d or mangled con
12d0: 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b 65 72 tent:\n{:?}\n{er
12e0: 72 7d 5c 6e 7b 73 74 61 74 75 73 3a 23 3f 7d 5c r}\n{status:#?}\
12f0: 6e 22 2c 20 26 73 6f 75 72 63 65 2e 75 72 6c 29 n", &source.url)
1300: 0a 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ........},......
1310: 09 7d 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 .}......},......
1320: 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 6f 66 20 rss::Error::Eof
1330: 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f 20 3d 3e => (),......_ =>
1340: 20 62 61 69 6c 21 28 22 55 6e 73 75 70 70 6f 72 bail!("Unsuppor
1350: 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 64 20 63 ted or mangled c
1360: 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b ontent:\n{:?}\n{
1370: 65 72 72 7d 5c 6e 7b 73 74 61 74 75 73 3a 23 3f err}\n{status:#?
1380: 7d 5c 6e 22 2c 20 26 73 6f 75 72 63 65 2e 75 72 }\n", &source.ur
1390: 6c 29 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 l).....}....};..
13a0: 09 09 66 6f 72 20 28 64 61 74 65 2c 20 75 72 6c ..for (date, url
13b0: 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 65 72 28 ) in posts.iter(
13c0: 29 20 7b 0a 09 09 09 09 6c 65 74 20 70 6f 73 74 ) {.....let post
13d0: 5f 75 72 6c 3a 20 43 6f 77 3c 73 74 72 3e 20 3d _url: Cow<str> =
13e0: 20 6d 61 74 63 68 20 73 6f 75 72 63 65 2e 75 72 match source.ur
13f0: 6c 5f 72 65 20 7b 0a 09 09 09 09 09 53 6f 6d 65 l_re {......Some
1400: 28 72 65 66 20 78 29 20 3d 3e 20 73 65 64 72 65 (ref x) => sedre
1410: 67 65 78 3a 3a 52 65 70 6c 61 63 65 43 6f 6d 6d gex::ReplaceComm
1420: 61 6e 64 3a 3a 6e 65 77 28 78 29 2e 73 74 61 63 and::new(x).stac
1430: 6b 28 29 3f 2e 65 78 65 63 75 74 65 28 75 72 6c k()?.execute(url
1440: 29 2c 0a 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 ),......None =>
1450: 75 72 6c 2e 69 6e 74 6f 28 29 2c 0a 09 09 09 09 url.into(),.....
1460: 7d 3b 0a 09 09 09 09 69 66 20 6c 65 74 20 53 6f };.....if let So
1470: 6d 65 28 65 78 69 73 74 73 29 20 3d 20 63 6f 6e me(exists) = con
1480: 6e 2e 65 78 69 73 74 73 28 26 70 6f 73 74 5f 75 n.exists(&post_u
1490: 72 6c 2c 20 2a 69 64 29 2e 61 77 61 69 74 2e 73 rl, *id).await.s
14a0: 74 61 63 6b 28 29 3f 20 7b 0a 09 09 09 09 09 69 tack()? {......i
14b0: 66 20 21 20 65 78 69 73 74 73 20 7b 0a 09 09 09 f ! exists {....
14c0: 09 09 09 69 66 20 74 68 69 73 5f 66 65 74 63 68 ...if this_fetch
14d0: 2e 69 73 5f 6e 6f 6e 65 28 29 20 7c 7c 20 2a 64 .is_none() || *d
14e0: 61 74 65 20 3e 20 74 68 69 73 5f 66 65 74 63 68 ate > this_fetch
14f0: 2e 75 6e 77 72 61 70 28 29 20 7b 0a 09 09 09 09 .unwrap() {.....
1500: 09 09 09 74 68 69 73 5f 66 65 74 63 68 20 3d 20 ...this_fetch =
1510: 53 6f 6d 65 28 2a 64 61 74 65 29 3b 0a 09 09 09 Some(*date);....
1520: 09 09 09 7d 3b 0a 09 09 09 09 09 09 73 65 6c 66 ...};.......self
1530: 2e 73 65 6e 64 28 20 6d 61 74 63 68 20 26 73 6f .send( match &so
1540: 75 72 63 65 2e 69 76 5f 68 61 73 68 20 7b 0a 09 urce.iv_hash {..
1550: 09 09 09 09 09 09 53 6f 6d 65 28 68 61 73 68 29 ......Some(hash)
1560: 20 3d 3e 20 66 6f 72 6d 61 74 21 28 22 3c 61 20 => format!("<a
1570: 68 72 65 66 3d 5c 22 68 74 74 70 73 3a 2f 2f 74 href=\"https://t
1580: 2e 6d 65 2f 69 76 3f 75 72 6c 3d 7b 70 6f 73 74 .me/iv?url={post
1590: 5f 75 72 6c 7d 26 72 68 61 73 68 3d 7b 68 61 73 _url}&rhash={has
15a0: 68 7d 5c 22 3e 20 3c 2f 61 3e 7b 70 6f 73 74 5f h}\"> </a>{post_
15b0: 75 72 6c 7d 22 29 2c 0a 09 09 09 09 09 09 09 4e url}"),........N
15c0: 6f 6e 65 20 3d 3e 20 66 6f 72 6d 61 74 21 28 22 one => format!("
15d0: 7b 70 6f 73 74 5f 75 72 6c 7d 22 29 2c 0a 09 09 {post_url}"),...
15e0: 09 09 09 09 7d 2c 20 53 6f 6d 65 28 64 65 73 74 ....}, Some(dest
15f0: 69 6e 61 74 69 6f 6e 29 2c 20 53 6f 6d 65 28 50 ination), Some(P
1600: 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 arseMode::Html))
1610: 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b .await.stack()?;
1620: 0a 09 09 09 09 09 09 63 6f 6e 6e 2e 61 64 64 5f .......conn.add_
1630: 70 6f 73 74 28 2a 69 64 2c 20 64 61 74 65 2c 20 post(*id, date,
1640: 26 70 6f 73 74 5f 75 72 6c 29 2e 61 77 61 69 74 &post_url).await
1650: 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 09 09 09 .stack()?;......
1660: 7d 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 09 70 6f };.....};.....po
1670: 73 74 65 64 20 2b 3d 20 31 3b 0a 09 09 09 7d 3b sted += 1;....};
1680: 0a 09 09 09 70 6f 73 74 73 2e 63 6c 65 61 72 28 ....posts.clear(
1690: 29 3b 0a 09 09 7d 3b 0a 09 09 4f 6b 28 66 6f 72 );...};...Ok(for
16a0: 6d 61 74 21 28 22 50 6f 73 74 65 64 3a 20 7b 70 mat!("Posted: {p
16b0: 6f 73 74 65 64 7d 22 29 29 0a 09 7d 0a 0a 09 61 osted}"))..}...a
16c0: 73 79 6e 63 20 66 6e 20 61 75 74 6f 66 65 74 63 sync fn autofetc
16d0: 68 28 26 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 h(&self) -> Resu
16e0: 6c 74 3c 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 lt<std::time::Du
16f0: 72 61 74 69 6f 6e 3e 20 7b 0a 09 09 6c 65 74 20 ration> {...let
1700: 6d 75 74 20 64 65 6c 61 79 20 3d 20 63 68 72 6f mut delay = chro
1710: 6e 6f 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 no::Duration::mi
1720: 6e 75 74 65 73 28 31 29 3b 0a 09 09 6c 65 74 20 nutes(1);...let
1730: 6e 6f 77 20 3d 20 63 68 72 6f 6e 6f 3a 3a 4c 6f now = chrono::Lo
1740: 63 61 6c 3a 3a 6e 6f 77 28 29 3b 0a 09 09 6c 65 cal::now();...le
1750: 74 20 71 75 65 75 65 20 3d 20 7b 0a 09 09 09 6c t queue = {....l
1760: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
1770: 6c 66 2e 64 62 2e 62 65 67 69 6e 28 29 2e 61 77 lf.db.begin().aw
1780: 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 ait.stack()?;...
1790: 09 63 6f 6e 6e 2e 67 65 74 5f 71 75 65 75 65 28 .conn.get_queue(
17a0: 29 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f ).await.stack()?
17b0: 0a 09 09 7d 3b 0a 09 09 66 6f 72 20 72 6f 77 20 ...};...for row
17c0: 69 6e 20 71 75 65 75 65 20 7b 0a 09 09 09 69 66 in queue {....if
17d0: 20 6c 65 74 20 53 6f 6d 65 28 6e 65 78 74 5f 66 let Some(next_f
17e0: 65 74 63 68 29 20 3d 20 72 6f 77 2e 6e 65 78 74 etch) = row.next
17f0: 5f 66 65 74 63 68 20 7b 0a 09 09 09 09 69 66 20 _fetch {.....if
1800: 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 next_fetch < now
1810: 20 7b 0a 09 09 09 09 09 69 66 20 6c 65 74 20 28 {......if let (
1820: 53 6f 6d 65 28 6f 77 6e 65 72 29 2c 20 53 6f 6d Some(owner), Som
1830: 65 28 73 6f 75 72 63 65 5f 69 64 29 2c 20 6c 61 e(source_id), la
1840: 73 74 5f 73 63 72 61 70 65 29 20 3d 20 28 72 6f st_scrape) = (ro
1850: 77 2e 6f 77 6e 65 72 2c 20 72 6f 77 2e 73 6f 75 w.owner, row.sou
1860: 72 63 65 5f 69 64 2c 20 72 6f 77 2e 6c 61 73 74 rce_id, row.last
1870: 5f 73 63 72 61 70 65 29 20 7b 0a 09 09 09 09 09 _scrape) {......
1880: 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 .let clone = Cor
1890: 65 20 7b 0a 09 09 09 09 09 09 09 6f 77 6e 65 72 e {........owner
18a0: 5f 63 68 61 74 3a 20 43 68 61 74 50 65 65 72 49 _chat: ChatPeerI
18b0: 64 3a 3a 66 72 6f 6d 28 6f 77 6e 65 72 29 2c 0a d::from(owner),.
18c0: 09 09 09 09 09 09 09 2e 2e 73 65 6c 66 2e 63 6c .........self.cl
18d0: 6f 6e 65 28 29 0a 09 09 09 09 09 09 7d 3b 0a 09 one().......};..
18e0: 09 09 09 09 09 6c 65 74 20 73 6f 75 72 63 65 20 .....let source
18f0: 3d 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 6d = {........let m
1900: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 64 ut conn = self.d
1910: 62 2e 62 65 67 69 6e 28 29 2e 61 77 61 69 74 2e b.begin().await.
1920: 73 74 61 63 6b 28 29 3f 3b 0a 09 09 09 09 09 09 stack()?;.......
1930: 09 6d 61 74 63 68 20 63 6f 6e 6e 2e 67 65 74 5f .match conn.get_
1940: 6f 6e 65 28 6f 77 6e 65 72 2c 20 73 6f 75 72 63 one(owner, sourc
1950: 65 5f 69 64 29 2e 61 77 61 69 74 20 7b 0a 09 09 e_id).await {...
1960: 09 09 09 09 09 09 4f 6b 28 53 6f 6d 65 28 73 6f ......Ok(Some(so
1970: 75 72 63 65 29 29 20 3d 3e 20 73 6f 75 72 63 65 urce)) => source
1980: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 2c 0a 09 09 .to_string(),...
1990: 09 09 09 09 09 09 4f 6b 28 4e 6f 6e 65 29 20 3d ......Ok(None) =
19a0: 3e 20 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f > "Source not fo
19b0: 75 6e 64 20 69 6e 20 64 61 74 61 62 61 73 65 2e und in database.
19c0: 73 74 61 63 6b 28 29 3f 22 2e 74 6f 5f 73 74 72 stack()?".to_str
19d0: 69 6e 67 28 29 2c 0a 09 09 09 09 09 09 09 09 45 ing(),.........E
19e0: 72 72 28 65 72 72 29 20 3d 3e 20 66 6f 72 6d 61 rr(err) => forma
19f0: 74 21 28 22 46 61 69 6c 65 64 20 74 6f 20 66 65 t!("Failed to fe
1a00: 74 63 68 20 73 6f 75 72 63 65 20 64 61 74 61 3a tch source data:
1a10: 5c 6e 7b 65 72 72 7d 22 29 2c 0a 09 09 09 09 09 \n{err}"),......
1a20: 09 09 7d 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 ..}.......};....
1a30: 09 09 09 74 61 73 6b 3a 3a 73 70 61 77 6e 28 61 ...task::spawn(a
1a40: 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 sync move {.....
1a50: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
1a60: 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b r) = clone.check
1a70: 28 73 6f 75 72 63 65 5f 69 64 2c 20 74 72 75 65 (source_id, true
1a80: 2c 20 53 6f 6d 65 28 6c 61 73 74 5f 73 63 72 61 , Some(last_scra
1a90: 70 65 29 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 pe)).await {....
1aa0: 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 .....if let Err(
1ab0: 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e err) = clone.sen
1ac0: 64 28 26 66 6f 72 6d 61 74 21 28 22 7b 73 6f 75 d(&format!("{sou
1ad0: 72 63 65 7d 5c 6e 5c 6e f0 9f 9b 91 20 7b 7d 22 rce}\n\nš {}"
1ae0: 2c 20 65 6e 63 6f 64 65 28 26 65 72 72 2e 74 6f , encode(&err.to
1af0: 5f 73 74 72 69 6e 67 28 29 29 29 2c 20 4e 6f 6e _string())), Non
1b00: 65 2c 20 53 6f 6d 65 28 50 61 72 73 65 4d 6f 64 e, Some(ParseMod
1b10: 65 3a 3a 4d 61 72 6b 64 6f 77 6e 56 32 29 29 2e e::MarkdownV2)).
1b20: 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 09 09 await {.........
1b30: 09 65 70 72 69 6e 74 6c 6e 21 28 22 43 68 65 63 .eprintln!("Chec
1b40: 6b 20 65 72 72 6f 72 3a 20 7b 65 72 72 3a 3f 7d k error: {err:?}
1b50: 22 29 3b 0a 09 09 09 09 09 09 09 09 09 2f 2f 20 ");..........//
1b60: 63 6c 6f 6e 65 2e 64 69 73 61 62 6c 65 28 26 73 clone.disable(&s
1b70: 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 29 ource_id, owner)
1b80: 2e 61 77 61 69 74 2e 75 6e 77 72 61 70 28 29 3b .await.unwrap();
1b90: 0a 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 .........};.....
1ba0: 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 29 3b 0a ...};.......});.
1bb0: 09 09 09 09 09 7d 0a 09 09 09 09 7d 20 65 6c 73 .....}.....} els
1bc0: 65 20 69 66 20 6e 65 78 74 5f 66 65 74 63 68 20 e if next_fetch
1bd0: 2d 20 6e 6f 77 20 3c 20 64 65 6c 61 79 20 7b 0a - now < delay {.
1be0: 09 09 09 09 09 64 65 6c 61 79 20 3d 20 6e 65 78 .....delay = nex
1bf0: 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 3b 0a 09 t_fetch - now;..
1c00: 09 09 09 7d 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 ...}....}...};..
1c10: 09 64 65 6c 61 79 2e 74 6f 5f 73 74 64 28 29 2e .delay.to_std().
1c20: 73 74 61 63 6b 28 29 0a 09 7d 0a 0a 09 70 75 62 stack()..}...pub
1c30: 20 61 73 79 6e 63 20 66 6e 20 6c 69 73 74 20 28 async fn list (
1c40: 26 73 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 55 73 &self, owner: Us
1c50: 65 72 50 65 65 72 49 64 29 20 2d 3e 20 52 65 73 erPeerId) -> Res
1c60: 75 6c 74 3c 53 74 72 69 6e 67 3e 20 7b 0a 09 09 ult<String> {...
1c70: 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 3a 20 56 let mut reply: V
1c80: 65 63 3c 53 74 72 69 6e 67 3e 20 3d 20 76 65 63 ec<String> = vec
1c90: 21 5b 5d 3b 0a 09 09 72 65 70 6c 79 2e 70 75 73 ![];...reply.pus
1ca0: 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 2e 69 6e h("Channels:".in
1cb0: 74 6f 28 29 29 3b 0a 09 09 6c 65 74 20 6d 75 74 to());...let mut
1cc0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 64 62 2e conn = self.db.
1cd0: 62 65 67 69 6e 28 29 2e 61 77 61 69 74 2e 73 74 begin().await.st
1ce0: 61 63 6b 28 29 3f 3b 0a 09 09 66 6f 72 20 72 6f ack()?;...for ro
1cf0: 77 20 69 6e 20 63 6f 6e 6e 2e 67 65 74 5f 6c 69 w in conn.get_li
1d00: 73 74 28 6f 77 6e 65 72 29 2e 61 77 61 69 74 2e st(owner).await.
1d10: 73 74 61 63 6b 28 29 3f 20 7b 0a 09 09 09 72 65 stack()? {....re
1d20: 70 6c 79 2e 70 75 73 68 28 72 6f 77 2e 74 6f 5f ply.push(row.to_
1d30: 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 7d 3b 0a string());...};.
1d40: 09 09 4f 6b 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 ..Ok(reply.join(
1d50: 22 5c 6e 5c 6e 22 29 29 0a 09 7d 0a 7d 0a 0a 69 "\n\n"))..}.}..i
1d60: 6d 70 6c 20 55 70 64 61 74 65 48 61 6e 64 6c 65 mpl UpdateHandle
1d70: 72 20 66 6f 72 20 43 6f 72 65 20 7b 0a 09 61 73 r for Core {..as
1d80: 79 6e 63 20 66 6e 20 68 61 6e 64 6c 65 20 28 26 ync fn handle (&
1d90: 73 65 6c 66 2c 20 75 70 64 61 74 65 3a 20 55 70 self, update: Up
1da0: 64 61 74 65 29 20 7b 0a 09 09 69 66 20 6c 65 74 date) {...if let
1db0: 20 55 70 64 61 74 65 54 79 70 65 3a 3a 4d 65 73 UpdateType::Mes
1dc0: 73 61 67 65 28 6d 73 67 29 20 3d 20 75 70 64 61 sage(msg) = upda
1dd0: 74 65 2e 75 70 64 61 74 65 5f 74 79 70 65 20 7b te.update_type {
1de0: 0a 09 09 09 69 66 20 6c 65 74 20 4f 6b 28 63 6d ....if let Ok(cm
1df0: 64 29 20 3d 20 43 6f 6d 6d 61 6e 64 3a 3a 74 72 d) = Command::tr
1e00: 79 5f 66 72 6f 6d 28 6d 73 67 29 20 7b 0a 09 09 y_from(msg) {...
1e10: 09 09 6c 65 74 20 6d 73 67 20 3d 20 63 6d 64 2e ..let msg = cmd.
1e20: 67 65 74 5f 6d 65 73 73 61 67 65 28 29 3b 0a 09 get_message();..
1e30: 09 09 09 6c 65 74 20 77 6f 72 64 73 20 3d 20 63 ...let words = c
1e40: 6d 64 2e 67 65 74 5f 61 72 67 73 28 29 3b 0a 09 md.get_args();..
1e50: 09 09 09 6c 65 74 20 63 6f 6d 6d 61 6e 64 20 3d ...let command =
1e60: 20 63 6d 64 2e 67 65 74 5f 6e 61 6d 65 28 29 3b cmd.get_name();
1e70: 0a 09 09 09 09 6c 65 74 20 72 65 73 20 3d 20 6d .....let res = m
1e80: 61 74 63 68 20 63 6f 6d 6d 61 6e 64 20 7b 0a 09 atch command {..
1e90: 09 09 09 09 22 2f 63 68 65 63 6b 22 20 7c 20 22 ...."/check" | "
1ea0: 2f 63 6c 65 61 6e 22 20 7c 20 22 2f 65 6e 61 62 /clean" | "/enab
1eb0: 6c 65 22 20 7c 20 22 2f 64 65 6c 65 74 65 22 20 le" | "/delete"
1ec0: 7c 20 22 2f 64 69 73 61 62 6c 65 22 20 3d 3e 20 | "/disable" =>
1ed0: 63 6f 6d 6d 61 6e 64 3a 3a 63 6f 6d 6d 61 6e 64 command::command
1ee0: 28 73 65 6c 66 2c 20 63 6f 6d 6d 61 6e 64 2c 20 (self, command,
1ef0: 6d 73 67 2c 20 77 6f 72 64 73 29 2e 61 77 61 69 msg, words).awai
1f00: 74 2c 0a 09 09 09 09 09 22 2f 73 74 61 72 74 22 t,......"/start"
1f10: 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 73 74 61 => command::sta
1f20: 72 74 28 73 65 6c 66 2c 20 6d 73 67 29 2e 61 77 rt(self, msg).aw
1f30: 61 69 74 2c 0a 09 09 09 09 09 22 2f 6c 69 73 74 ait,......"/list
1f40: 22 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 6c 69 " => command::li
1f50: 73 74 28 73 65 6c 66 2c 20 6d 73 67 29 2e 61 77 st(self, msg).aw
1f60: 61 69 74 2c 0a 09 09 09 09 09 22 2f 61 64 64 22 ait,......"/add"
1f70: 20 7c 20 22 2f 75 70 64 61 74 65 22 20 3d 3e 20 | "/update" =>
1f80: 63 6f 6d 6d 61 6e 64 3a 3a 75 70 64 61 74 65 28 command::update(
1f90: 73 65 6c 66 2c 20 63 6f 6d 6d 61 6e 64 2c 20 6d self, command, m
1fa0: 73 67 2c 20 77 6f 72 64 73 29 2e 61 77 61 69 74 sg, words).await
1fb0: 2c 0a 09 09 09 09 09 61 6e 79 20 3d 3e 20 45 72 ,......any => Er
1fc0: 72 28 61 6e 79 68 6f 77 21 28 22 55 6e 6b 6e 6f r(anyhow!("Unkno
1fd0: 77 6e 20 63 6f 6d 6d 61 6e 64 3a 20 7b 61 6e 79 wn command: {any
1fe0: 7d 22 29 29 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 }")),.....};....
1ff0: 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 .if let Err(err)
2000: 20 3d 20 72 65 73 20 7b 0a 09 09 09 09 09 69 66 = res {......if
2010: 20 6c 65 74 20 45 72 72 28 65 72 72 32 29 20 3d let Err(err2) =
2020: 20 73 65 6c 66 2e 73 65 6e 64 28 66 6f 72 6d 61 self.send(forma
2030: 74 21 28 22 5c 5c 23 65 72 72 6f 72 5c 6e 60 60 t!("\\#error\n``
2040: 60 5c 6e 7b 65 72 72 7d 5c 6e 60 60 60 22 29 2c `\n{err}\n```"),
2050: 0a 09 09 09 09 09 09 53 6f 6d 65 28 6d 73 67 2e .......Some(msg.
2060: 63 68 61 74 2e 67 65 74 5f 69 64 28 29 29 2c 0a chat.get_id()),.
2070: 09 09 09 09 09 09 53 6f 6d 65 28 50 61 72 73 65 ......Some(Parse
2080: 4d 6f 64 65 3a 3a 4d 61 72 6b 64 6f 77 6e 56 32 Mode::MarkdownV2
2090: 29 0a 09 09 09 09 09 29 2e 61 77 61 69 74 7b 0a )......).await{.
20a0: 09 09 09 09 09 09 64 62 67 21 28 65 72 72 32 29 ......dbg!(err2)
20b0: 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 7d 0a ;......};.....}.
20c0: 09 09 09 7d 3b 0a 09 09 7d 3b 0a 09 7d 0a 7d 0a ...};...};..}.}.