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: 44 61 74 65 54 69 6d 65 3b 0a 75 73 65 20 6c 61 DateTime;.use la
00c0: 7a 79 5f 73 74 61 74 69 63 3a 3a 6c 61 7a 79 5f zy_static::lazy_
00d0: 73 74 61 74 69 63 3b 0a 75 73 65 20 72 65 67 65 static;.use rege
00e0: 78 3a 3a 52 65 67 65 78 3b 0a 75 73 65 20 74 67 x::Regex;.use tg
00f0: 62 6f 74 3a 3a 7b 0a 09 61 70 69 3a 3a 43 6c 69 bot::{..api::Cli
0100: 65 6e 74 2c 0a 09 68 61 6e 64 6c 65 72 3a 3a 55 ent,..handler::U
0110: 70 64 61 74 65 48 61 6e 64 6c 65 72 2c 0a 09 74 pdateHandler,..t
0120: 79 70 65 73 3a 3a 7b 0a 09 09 42 6f 74 2c 0a 09 ypes::{...Bot,..
0130: 09 43 68 61 74 50 65 65 72 49 64 2c 0a 09 09 43 .ChatPeerId,...C
0140: 6f 6d 6d 61 6e 64 2c 0a 09 09 47 65 74 42 6f 74 ommand,...GetBot
0150: 2c 0a 09 09 4d 65 73 73 61 67 65 2c 0a 09 09 50 ,...Message,...P
0160: 61 72 73 65 4d 6f 64 65 2c 0a 09 09 53 65 6e 64 arseMode,...Send
0170: 4d 65 73 73 61 67 65 2c 0a 09 09 55 70 64 61 74 Message,...Updat
0180: 65 2c 0a 09 09 55 70 64 61 74 65 54 79 70 65 2c e,...UpdateType,
0190: 0a 09 09 55 73 65 72 50 65 65 72 49 64 2c 0a 09 ...UserPeerId,..
01a0: 7d 2c 0a 7d 3b 0a 75 73 65 20 73 74 61 63 6b 65 },.};.use stacke
01b0: 64 5f 65 72 72 6f 72 73 3a 3a 7b 0a 09 52 65 73 d_errors::{..Res
01c0: 75 6c 74 2c 0a 09 53 74 61 63 6b 61 62 6c 65 45 ult,..StackableE
01d0: 72 72 2c 0a 09 61 6e 79 68 6f 77 2c 0a 09 62 61 rr,..anyhow,..ba
01e0: 69 6c 2c 0a 7d 3b 0a 0a 6c 61 7a 79 5f 73 74 61 il,.};..lazy_sta
01f0: 74 69 63 21 7b 0a 09 70 75 62 20 73 74 61 74 69 tic!{..pub stati
0200: 63 20 72 65 66 20 52 45 5f 53 50 45 43 49 41 4c c ref RE_SPECIAL
0210: 3a 20 52 65 67 65 78 20 3d 20 52 65 67 65 78 3a : Regex = Regex:
0220: 3a 6e 65 77 28 72 22 28 5b 5c 2d 5f 2a 5c 5b 5c :new(r"([\-_*\[\
0230: 5d 28 29 7e 60 3e 23 2b 7c 7b 7d 5c 2e 21 5d 29 ]()~`>#+|{}\.!])
0240: 22 29 2e 75 6e 77 72 61 70 28 29 3b 0a 7d 0a 0a ").unwrap();.}..
0250: 2f 2f 2f 20 45 6e 63 6f 64 65 73 20 73 70 65 63 /// Encodes spec
0260: 69 61 6c 20 48 54 4d 4c 20 65 6e 74 69 74 69 65 ial HTML entitie
0270: 73 20 74 6f 20 70 72 65 76 65 6e 74 20 74 68 65 s to prevent the
0280: 6d 20 69 6e 74 65 72 66 65 72 69 6e 67 20 77 69 m interfering wi
0290: 74 68 20 54 65 6c 65 67 72 61 6d 20 48 54 4d 4c th Telegram HTML
02a0: 0a 70 75 62 20 66 6e 20 65 6e 63 6f 64 65 20 28 .pub fn encode (
02b0: 74 65 78 74 3a 20 26 73 74 72 29 20 2d 3e 20 43 text: &str) -> C
02c0: 6f 77 3c 27 5f 2c 20 73 74 72 3e 20 7b 0a 09 52 ow<'_, str> {..R
02d0: 45 5f 53 50 45 43 49 41 4c 2e 72 65 70 6c 61 63 E_SPECIAL.replac
02e0: 65 5f 61 6c 6c 28 74 65 78 74 2c 20 22 5c 5c 24 e_all(text, "\\$
02f0: 31 22 29 0a 7d 0a 0a 23 5b 64 65 72 69 76 65 28 1").}..#[derive(
0300: 43 6c 6f 6e 65 29 5d 0a 70 75 62 20 73 74 72 75 Clone)].pub stru
0310: 63 74 20 43 6f 72 65 20 7b 0a 09 6f 77 6e 65 72 ct Core {..owner
0320: 5f 63 68 61 74 3a 20 43 68 61 74 50 65 65 72 49 _chat: ChatPeerI
0330: 64 2c 0a 09 2f 2f 20 6d 61 78 5f 64 65 6c 61 79 d,..// max_delay
0340: 3a 20 75 31 36 2c 0a 09 70 75 62 20 74 67 3a 20 : u16,..pub tg:
0350: 43 6c 69 65 6e 74 2c 0a 09 70 75 62 20 6d 65 3a Client,..pub me:
0360: 20 42 6f 74 2c 0a 09 70 75 62 20 64 62 3a 20 44 Bot,..pub db: D
0370: 62 2c 0a 09 73 6f 75 72 63 65 73 3a 20 41 72 63 b,..sources: Arc
0380: 3c 4d 75 74 65 78 3c 48 61 73 68 53 65 74 3c 41 <Mutex<HashSet<A
0390: 72 63 3c 69 33 32 3e 3e 3e 3e 2c 0a 09 68 74 74 rc<i32>>>>,..htt
03a0: 70 5f 63 6c 69 65 6e 74 3a 20 72 65 71 77 65 73 p_client: reqwes
03b0: 74 3a 3a 43 6c 69 65 6e 74 2c 0a 7d 0a 0a 69 6d t::Client,.}..im
03c0: 70 6c 20 43 6f 72 65 20 7b 0a 09 70 75 62 20 61 pl Core {..pub a
03d0: 73 79 6e 63 20 66 6e 20 6e 65 77 28 73 65 74 74 sync fn new(sett
03e0: 69 6e 67 73 3a 20 63 6f 6e 66 69 67 3a 3a 43 6f ings: config::Co
03f0: 6e 66 69 67 29 20 2d 3e 20 52 65 73 75 6c 74 3c nfig) -> Result<
0400: 43 6f 72 65 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 Core> {...let ow
0410: 6e 65 72 5f 63 68 61 74 20 3d 20 43 68 61 74 50 ner_chat = ChatP
0420: 65 65 72 49 64 3a 3a 66 72 6f 6d 28 73 65 74 74 eerId::from(sett
0430: 69 6e 67 73 2e 67 65 74 5f 69 6e 74 28 22 6f 77 ings.get_int("ow
0440: 6e 65 72 22 29 2e 73 74 61 63 6b 28 29 3f 29 3b ner").stack()?);
0450: 0a 09 09 6c 65 74 20 61 70 69 5f 6b 65 79 20 3d ...let api_key =
0460: 20 73 65 74 74 69 6e 67 73 2e 67 65 74 5f 73 74 settings.get_st
0470: 72 69 6e 67 28 22 61 70 69 5f 6b 65 79 22 29 2e ring("api_key").
0480: 73 74 61 63 6b 28 29 3f 3b 0a 09 09 6c 65 74 20 stack()?;...let
0490: 74 67 20 3d 20 43 6c 69 65 6e 74 3a 3a 6e 65 77 tg = Client::new
04a0: 28 26 61 70 69 5f 6b 65 79 29 2e 73 74 61 63 6b (&api_key).stack
04b0: 28 29 3f 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 ()?;....let mut
04c0: 63 6c 69 65 6e 74 20 3d 20 72 65 71 77 65 73 74 client = reqwest
04d0: 3a 3a 43 6c 69 65 6e 74 3a 3a 62 75 69 6c 64 65 ::Client::builde
04e0: 72 28 29 3b 0a 09 09 69 66 20 6c 65 74 20 4f 6b r();...if let Ok
04f0: 28 70 72 6f 78 79 29 20 3d 20 73 65 74 74 69 6e (proxy) = settin
0500: 67 73 2e 67 65 74 5f 73 74 72 69 6e 67 28 22 70 gs.get_string("p
0510: 72 6f 78 79 22 29 20 7b 0a 09 09 09 6c 65 74 20 roxy") {....let
0520: 70 72 6f 78 79 20 3d 20 72 65 71 77 65 73 74 3a proxy = reqwest:
0530: 3a 50 72 6f 78 79 3a 3a 61 6c 6c 28 70 72 6f 78 :Proxy::all(prox
0540: 79 29 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 09 y).stack()?;....
0550: 63 6c 69 65 6e 74 20 3d 20 63 6c 69 65 6e 74 2e client = client.
0560: 70 72 6f 78 79 28 70 72 6f 78 79 29 3b 0a 09 09 proxy(proxy);...
0570: 7d 0a 09 09 6c 65 74 20 68 74 74 70 5f 63 6c 69 }...let http_cli
0580: 65 6e 74 20 3d 20 63 6c 69 65 6e 74 2e 62 75 69 ent = client.bui
0590: 6c 64 28 29 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 ld().stack()?;..
05a0: 09 6c 65 74 20 6d 65 20 3d 20 74 67 2e 65 78 65 .let me = tg.exe
05b0: 63 75 74 65 28 47 65 74 42 6f 74 29 2e 61 77 61 cute(GetBot).awa
05c0: 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 6c it.stack()?;...l
05d0: 65 74 20 63 6f 72 65 20 3d 20 43 6f 72 65 20 7b et core = Core {
05e0: 0a 09 09 09 74 67 2c 0a 09 09 09 6d 65 2c 0a 09 ....tg,....me,..
05f0: 09 09 6f 77 6e 65 72 5f 63 68 61 74 2c 0a 09 09 ..owner_chat,...
0600: 09 64 62 3a 20 44 62 3a 3a 6e 65 77 28 26 73 65 .db: Db::new(&se
0610: 74 74 69 6e 67 73 2e 67 65 74 5f 73 74 72 69 6e ttings.get_strin
0620: 67 28 22 70 67 22 29 2e 73 74 61 63 6b 28 29 3f g("pg").stack()?
0630: 29 3f 2c 0a 09 09 09 73 6f 75 72 63 65 73 3a 20 )?,....sources:
0640: 41 72 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a 3a Arc::new(Mutex::
0650: 6e 65 77 28 48 61 73 68 53 65 74 3a 3a 6e 65 77 new(HashSet::new
0660: 28 29 29 29 2c 0a 09 09 09 68 74 74 70 5f 63 6c ())),....http_cl
0670: 69 65 6e 74 2c 0a 09 09 09 2f 2f 20 6d 61 78 5f ient,....// max_
0680: 64 65 6c 61 79 3a 20 36 30 2c 0a 09 09 7d 3b 0a delay: 60,...};.
0690: 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 63 6f ..let clone = co
06a0: 72 65 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 74 61 re.clone();...ta
06b0: 73 6b 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 sk::spawn(async
06c0: 6d 6f 76 65 20 7b 0a 09 09 09 6c 6f 6f 70 20 7b move {....loop {
06d0: 0a 09 09 09 09 6c 65 74 20 64 65 6c 61 79 20 3d .....let delay =
06e0: 20 6d 61 74 63 68 20 26 63 6c 6f 6e 65 2e 61 75 match &clone.au
06f0: 74 6f 66 65 74 63 68 28 29 2e 61 77 61 69 74 20 tofetch().await
0700: 7b 0a 09 09 09 09 09 45 72 72 28 65 72 72 29 20 {......Err(err)
0710: 3d 3e 20 7b 0a 09 09 09 09 09 09 69 66 20 6c 65 => {.......if le
0720: 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c 6f t Err(err) = clo
0730: 6e 65 2e 73 65 6e 64 28 66 6f 72 6d 61 74 21 28 ne.send(format!(
0740: 22 f0 9f 9b 91 20 7b 65 72 72 7d 22 29 2c 20 4e "š {err}"), N
0750: 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 74 one, None).await
0760: 20 7b 0a 09 09 09 09 09 09 09 65 70 72 69 6e 74 {........eprint
0770: 6c 6e 21 28 22 41 75 74 6f 66 65 74 63 68 20 65 ln!("Autofetch e
0780: 72 72 6f 72 3a 20 7b 65 72 72 3a 3f 7d 22 29 3b rror: {err:?}");
0790: 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 .......};.......
07a0: 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 std::time::Durat
07b0: 69 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 63 73 28 36 ion::from_secs(6
07c0: 30 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 0)......},......
07d0: 4f 6b 28 74 69 6d 65 29 20 3d 3e 20 2a 74 69 6d Ok(time) => *tim
07e0: 65 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 74 61 e,.....};.....ta
07f0: 73 6b 3a 3a 73 6c 65 65 70 28 64 65 6c 61 79 29 sk::sleep(delay)
0800: 2e 61 77 61 69 74 3b 0a 09 09 09 7d 0a 09 09 7d .await;....}...}
0810: 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 0a 09 7d );...Ok(core)..}
0820: 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 ...pub async fn
0830: 73 65 6e 64 20 3c 53 3e 28 26 73 65 6c 66 2c 20 send <S>(&self,
0840: 6d 73 67 3a 20 53 2c 20 74 61 72 67 65 74 3a 20 msg: S, target:
0850: 4f 70 74 69 6f 6e 3c 43 68 61 74 50 65 65 72 49 Option<ChatPeerI
0860: 64 3e 2c 20 6d 6f 64 65 3a 20 4f 70 74 69 6f 6e d>, mode: Option
0870: 3c 50 61 72 73 65 4d 6f 64 65 3e 29 20 2d 3e 20 <ParseMode>) ->
0880: 52 65 73 75 6c 74 3c 4d 65 73 73 61 67 65 3e 0a Result<Message>.
0890: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 53 .where S: Into<S
08a0: 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 74 20 6d tring> {...let m
08b0: 73 67 20 3d 20 6d 73 67 2e 69 6e 74 6f 28 29 3b sg = msg.into();
08c0: 0a 0a 09 09 6c 65 74 20 6d 6f 64 65 20 3d 20 6d ....let mode = m
08d0: 6f 64 65 2e 75 6e 77 72 61 70 5f 6f 72 28 50 61 ode.unwrap_or(Pa
08e0: 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 3b 0a rseMode::Html);.
08f0: 09 09 6c 65 74 20 74 61 72 67 65 74 20 3d 20 74 ..let target = t
0900: 61 72 67 65 74 2e 75 6e 77 72 61 70 5f 6f 72 28 arget.unwrap_or(
0910: 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 61 74 29 self.owner_chat)
0920: 3b 0a 09 09 73 65 6c 66 2e 74 67 2e 65 78 65 63 ;...self.tg.exec
0930: 75 74 65 28 0a 09 09 09 53 65 6e 64 4d 65 73 73 ute(....SendMess
0940: 61 67 65 3a 3a 6e 65 77 28 74 61 72 67 65 74 2c age::new(target,
0950: 20 6d 73 67 29 0a 09 09 09 09 2e 77 69 74 68 5f msg)......with_
0960: 70 61 72 73 65 5f 6d 6f 64 65 28 6d 6f 64 65 29 parse_mode(mode)
0970: 0a 09 09 29 2e 61 77 61 69 74 2e 73 74 61 63 6b ...).await.stack
0980: 28 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e ()..}...pub asyn
0990: 63 20 66 6e 20 63 68 65 63 6b 20 28 26 73 65 6c c fn check (&sel
09a0: 66 2c 20 69 64 3a 20 69 33 32 2c 20 72 65 61 6c f, id: i32, real
09b0: 3a 20 62 6f 6f 6c 29 20 2d 3e 20 52 65 73 75 6c : bool) -> Resul
09c0: 74 3c 53 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 t<String> {...le
09d0: 74 20 6d 75 74 20 70 6f 73 74 65 64 3a 20 69 33 t mut posted: i3
09e0: 32 20 3d 20 30 3b 0a 09 09 6c 65 74 20 6d 75 74 2 = 0;...let mut
09f0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 64 62 2e conn = self.db.
0a00: 62 65 67 69 6e 28 29 2e 61 77 61 69 74 2e 73 74 begin().await.st
0a10: 61 63 6b 28 29 3f 3b 0a 0a 09 09 6c 65 74 20 69 ack()?;....let i
0a20: 64 20 3d 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 d = {....let mut
0a30: 20 73 65 74 20 3d 20 73 65 6c 66 2e 73 6f 75 72 set = self.sour
0a40: 63 65 73 2e 6c 6f 63 6b 5f 61 72 63 28 29 2e 61 ces.lock_arc().a
0a50: 77 61 69 74 3b 0a 09 09 09 6d 61 74 63 68 20 73 wait;....match s
0a60: 65 74 2e 67 65 74 28 26 69 64 29 20 7b 0a 09 09 et.get(&id) {...
0a70: 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 69 64 ..Some(id) => id
0a80: 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 09 4e 6f .clone(),.....No
0a90: 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 6c 65 74 ne => {......let
0aa0: 20 69 64 20 3d 20 41 72 63 3a 3a 6e 65 77 28 69 id = Arc::new(i
0ab0: 64 29 3b 0a 09 09 09 09 09 73 65 74 2e 69 6e 73 d);......set.ins
0ac0: 65 72 74 28 69 64 2e 63 6c 6f 6e 65 28 29 29 3b ert(id.clone());
0ad0: 0a 09 09 09 09 09 69 64 2e 63 6c 6f 6e 65 28 29 ......id.clone()
0ae0: 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 7d .....},....}...}
0af0: 3b 0a 09 09 6c 65 74 20 63 6f 75 6e 74 20 3d 20 ;...let count =
0b00: 41 72 63 3a 3a 73 74 72 6f 6e 67 5f 63 6f 75 6e Arc::strong_coun
0b10: 74 28 26 69 64 29 3b 0a 09 09 69 66 20 63 6f 75 t(&id);...if cou
0b20: 6e 74 20 3d 3d 20 32 20 7b 0a 09 09 09 6c 65 74 nt == 2 {....let
0b30: 20 73 6f 75 72 63 65 20 3d 20 63 6f 6e 6e 2e 67 source = conn.g
0b40: 65 74 5f 73 6f 75 72 63 65 28 2a 69 64 2c 20 73 et_source(*id, s
0b50: 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 61 74 29 2e elf.owner_chat).
0b60: 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a await.stack()?;.
0b70: 09 09 09 63 6f 6e 6e 2e 73 65 74 5f 73 63 72 61 ...conn.set_scra
0b80: 70 65 28 2a 69 64 29 2e 61 77 61 69 74 2e 73 74 pe(*id).await.st
0b90: 61 63 6b 28 29 3f 3b 0a 09 09 09 6c 65 74 20 64 ack()?;....let d
0ba0: 65 73 74 69 6e 61 74 69 6f 6e 20 3d 20 43 68 61 estination = Cha
0bb0: 74 50 65 65 72 49 64 3a 3a 66 72 6f 6d 28 6d 61 tPeerId::from(ma
0bc0: 74 63 68 20 72 65 61 6c 20 7b 0a 09 09 09 09 74 tch real {.....t
0bd0: 72 75 65 20 3d 3e 20 73 6f 75 72 63 65 2e 63 68 rue => source.ch
0be0: 61 6e 6e 65 6c 5f 69 64 2c 0a 09 09 09 09 66 61 annel_id,.....fa
0bf0: 6c 73 65 20 3d 3e 20 73 6f 75 72 63 65 2e 6f 77 lse => source.ow
0c00: 6e 65 72 2c 0a 09 09 09 7d 29 3b 0a 09 09 09 6c ner,....});....l
0c10: 65 74 20 6d 75 74 20 74 68 69 73 5f 66 65 74 63 et mut this_fetc
0c20: 68 3a 20 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 h: Option<DateTi
0c30: 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 me<chrono::Fixed
0c40: 4f 66 66 73 65 74 3e 3e 20 3d 20 4e 6f 6e 65 3b Offset>> = None;
0c50: 0a 09 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 ....let mut post
0c60: 73 3a 20 42 54 72 65 65 4d 61 70 3c 44 61 74 65 s: BTreeMap<Date
0c70: 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 Time<chrono::Fix
0c80: 65 64 4f 66 66 73 65 74 3e 2c 20 53 74 72 69 6e edOffset>, Strin
0c90: 67 3e 20 3d 20 42 54 72 65 65 4d 61 70 3a 3a 6e g> = BTreeMap::n
0ca0: 65 77 28 29 3b 0a 0a 09 09 09 6c 65 74 20 72 65 ew();.....let re
0cb0: 73 70 6f 6e 73 65 20 3d 20 73 65 6c 66 2e 68 74 sponse = self.ht
0cc0: 74 70 5f 63 6c 69 65 6e 74 2e 67 65 74 28 26 73 tp_client.get(&s
0cd0: 6f 75 72 63 65 2e 75 72 6c 29 2e 73 65 6e 64 28 ource.url).send(
0ce0: 29 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f ).await.stack()?
0cf0: 3b 0a 09 09 09 6c 65 74 20 73 74 61 74 75 73 20 ;....let status
0d00: 3d 20 72 65 73 70 6f 6e 73 65 2e 73 74 61 74 75 = response.statu
0d10: 73 28 29 3b 0a 09 09 09 6c 65 74 20 63 6f 6e 74 s();....let cont
0d20: 65 6e 74 20 3d 20 72 65 73 70 6f 6e 73 65 2e 62 ent = response.b
0d30: 79 74 65 73 28 29 2e 61 77 61 69 74 2e 73 74 61 ytes().await.sta
0d40: 63 6b 28 29 3f 3b 0a 09 09 09 6d 61 74 63 68 20 ck()?;....match
0d50: 72 73 73 3a 3a 43 68 61 6e 6e 65 6c 3a 3a 72 65 rss::Channel::re
0d60: 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 ad_from(&content
0d70: 5b 2e 2e 5d 29 20 7b 0a 09 09 09 09 4f 6b 28 66 [..]) {.....Ok(f
0d80: 65 65 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 66 eed) => {......f
0d90: 6f 72 20 69 74 65 6d 20 69 6e 20 66 65 65 64 2e or item in feed.
0da0: 69 74 65 6d 73 28 29 20 7b 0a 09 09 09 09 09 09 items() {.......
0db0: 69 66 20 6c 65 74 20 53 6f 6d 65 28 6c 69 6e 6b if let Some(link
0dc0: 29 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 28 29 20 ) = item.link()
0dd0: 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 64 61 74 {........let dat
0de0: 65 20 3d 20 6d 61 74 63 68 20 69 74 65 6d 2e 70 e = match item.p
0df0: 75 62 5f 64 61 74 65 28 29 20 7b 0a 09 09 09 09 ub_date() {.....
0e00: 09 09 09 09 53 6f 6d 65 28 66 65 65 64 5f 64 61 ....Some(feed_da
0e10: 74 65 29 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a te) => DateTime:
0e20: 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 32 :parse_from_rfc2
0e30: 38 32 32 28 66 65 65 64 5f 64 61 74 65 29 2c 0a 822(feed_date),.
0e40: 09 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 ........None =>
0e50: 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f DateTime::parse_
0e60: 66 72 6f 6d 5f 72 66 63 33 33 33 39 28 26 69 74 from_rfc3339(&it
0e70: 65 6d 2e 64 75 62 6c 69 6e 5f 63 6f 72 65 5f 65 em.dublin_core_e
0e80: 78 74 28 29 2e 75 6e 77 72 61 70 28 29 2e 64 61 xt().unwrap().da
0e90: 74 65 73 28 29 5b 30 5d 29 2c 0a 09 09 09 09 09 tes()[0]),......
0ea0: 09 09 7d 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 ..}.stack()?;...
0eb0: 09 09 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 6c .....let url = l
0ec0: 69 6e 6b 3b 0a 09 09 09 09 09 09 09 70 6f 73 74 ink;........post
0ed0: 73 2e 69 6e 73 65 72 74 28 64 61 74 65 2c 20 75 s.insert(date, u
0ee0: 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b rl.to_string());
0ef0: 0a 09 09 09 09 09 09 7d 0a 09 09 09 09 09 7d 3b .......}......};
0f00: 0a 09 09 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 .....},.....Err(
0f10: 65 72 72 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 err) => match er
0f20: 72 20 7b 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 r {......rss::Er
0f30: 72 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 ror::InvalidStar
0f40: 74 54 61 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 tTag => {.......
0f50: 6d 61 74 63 68 20 61 74 6f 6d 5f 73 79 6e 64 69 match atom_syndi
0f60: 63 61 74 69 6f 6e 3a 3a 46 65 65 64 3a 3a 72 65 cation::Feed::re
0f70: 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 ad_from(&content
0f80: 5b 2e 2e 5d 29 20 7b 0a 09 09 09 09 09 09 09 4f [..]) {........O
0f90: 6b 28 66 65 65 64 29 20 3d 3e 20 7b 0a 09 09 09 k(feed) => {....
0fa0: 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 69 6e .....for item in
0fb0: 20 66 65 65 64 2e 65 6e 74 72 69 65 73 28 29 20 feed.entries()
0fc0: 7b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 64 {..........let d
0fd0: 61 74 65 20 3d 20 69 74 65 6d 2e 70 75 62 6c 69 ate = item.publi
0fe0: 73 68 65 64 28 29 2e 75 6e 77 72 61 70 28 29 3b shed().unwrap();
0ff0: 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 75 72 ..........let ur
1000: 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 29 l = item.links()
1010: 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 09 09 [0].href();.....
1020: 09 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 .....posts.inser
1030: 74 28 2a 64 61 74 65 2c 20 75 72 6c 2e 74 6f 5f t(*date, url.to_
1040: 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 string());......
1050: 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 7d 2c 0a ...};........},.
1060: 09 09 09 09 09 09 09 45 72 72 28 65 72 72 29 20 .......Err(err)
1070: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 62 61 69 => {.........bai
1080: 6c 21 28 22 55 6e 73 75 70 70 6f 72 74 65 64 20 l!("Unsupported
1090: 6f 72 20 6d 61 6e 67 6c 65 64 20 63 6f 6e 74 65 or mangled conte
10a0: 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b 65 72 72 7d nt:\n{:?}\n{err}
10b0: 5c 6e 7b 73 74 61 74 75 73 3a 23 3f 7d 5c 6e 22 \n{status:#?}\n"
10c0: 2c 20 26 73 6f 75 72 63 65 2e 75 72 6c 29 0a 09 , &source.url)..
10d0: 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 7d ......},.......}
10e0: 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 72 73 ......},......rs
10f0: 73 3a 3a 45 72 72 6f 72 3a 3a 45 6f 66 20 3d 3e s::Error::Eof =>
1100: 20 28 29 2c 0a 09 09 09 09 09 5f 20 3d 3e 20 62 (),......_ => b
1110: 61 69 6c 21 28 22 55 6e 73 75 70 70 6f 72 74 65 ail!("Unsupporte
1120: 64 20 6f 72 20 6d 61 6e 67 6c 65 64 20 63 6f 6e d or mangled con
1130: 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b 65 72 tent:\n{:?}\n{er
1140: 72 7d 5c 6e 7b 73 74 61 74 75 73 3a 23 3f 7d 5c r}\n{status:#?}\
1150: 6e 22 2c 20 26 73 6f 75 72 63 65 2e 75 72 6c 29 n", &source.url)
1160: 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 .....}....};....
1170: 66 6f 72 20 28 64 61 74 65 2c 20 75 72 6c 29 20 for (date, url)
1180: 69 6e 20 70 6f 73 74 73 2e 69 74 65 72 28 29 20 in posts.iter()
1190: 7b 0a 09 09 09 09 6c 65 74 20 70 6f 73 74 5f 75 {.....let post_u
11a0: 72 6c 3a 20 43 6f 77 3c 73 74 72 3e 20 3d 20 6d rl: Cow<str> = m
11b0: 61 74 63 68 20 73 6f 75 72 63 65 2e 75 72 6c 5f atch source.url_
11c0: 72 65 20 7b 0a 09 09 09 09 09 53 6f 6d 65 28 72 re {......Some(r
11d0: 65 66 20 78 29 20 3d 3e 20 73 65 64 72 65 67 65 ef x) => sedrege
11e0: 78 3a 3a 52 65 70 6c 61 63 65 43 6f 6d 6d 61 6e x::ReplaceComman
11f0: 64 3a 3a 6e 65 77 28 78 29 2e 73 74 61 63 6b 28 d::new(x).stack(
1200: 29 3f 2e 65 78 65 63 75 74 65 28 75 72 6c 29 2c )?.execute(url),
1210: 0a 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 75 72 ......None => ur
1220: 6c 2e 69 6e 74 6f 28 29 2c 0a 09 09 09 09 7d 3b l.into(),.....};
1230: 0a 09 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 .....if let Some
1240: 28 65 78 69 73 74 73 29 20 3d 20 63 6f 6e 6e 2e (exists) = conn.
1250: 65 78 69 73 74 73 28 26 70 6f 73 74 5f 75 72 6c exists(&post_url
1260: 2c 20 2a 69 64 29 2e 61 77 61 69 74 2e 73 74 61 , *id).await.sta
1270: 63 6b 28 29 3f 20 7b 0a 09 09 09 09 09 69 66 20 ck()? {......if
1280: 21 20 65 78 69 73 74 73 20 7b 0a 09 09 09 09 09 ! exists {......
1290: 09 69 66 20 74 68 69 73 5f 66 65 74 63 68 2e 69 .if this_fetch.i
12a0: 73 5f 6e 6f 6e 65 28 29 20 7c 7c 20 2a 64 61 74 s_none() || *dat
12b0: 65 20 3e 20 74 68 69 73 5f 66 65 74 63 68 2e 75 e > this_fetch.u
12c0: 6e 77 72 61 70 28 29 20 7b 0a 09 09 09 09 09 09 nwrap() {.......
12d0: 09 74 68 69 73 5f 66 65 74 63 68 20 3d 20 53 6f .this_fetch = So
12e0: 6d 65 28 2a 64 61 74 65 29 3b 0a 09 09 09 09 09 me(*date);......
12f0: 09 7d 3b 0a 09 09 09 09 09 09 73 65 6c 66 2e 73 .};.......self.s
1300: 65 6e 64 28 20 6d 61 74 63 68 20 26 73 6f 75 72 end( match &sour
1310: 63 65 2e 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 ce.iv_hash {....
1320: 09 09 09 09 53 6f 6d 65 28 68 61 73 68 29 20 3d ....Some(hash) =
1330: 3e 20 66 6f 72 6d 61 74 21 28 22 3c 61 20 68 72 > format!("<a hr
1340: 65 66 3d 5c 22 68 74 74 70 73 3a 2f 2f 74 2e 6d ef=\"https://t.m
1350: 65 2f 69 76 3f 75 72 6c 3d 7b 70 6f 73 74 5f 75 e/iv?url={post_u
1360: 72 6c 7d 26 72 68 61 73 68 3d 7b 68 61 73 68 7d rl}&rhash={hash}
1370: 5c 22 3e 20 3c 2f 61 3e 7b 70 6f 73 74 5f 75 72 \"> </a>{post_ur
1380: 6c 7d 22 29 2c 0a 09 09 09 09 09 09 09 4e 6f 6e l}"),........Non
1390: 65 20 3d 3e 20 66 6f 72 6d 61 74 21 28 22 7b 70 e => format!("{p
13a0: 6f 73 74 5f 75 72 6c 7d 22 29 2c 0a 09 09 09 09 ost_url}"),.....
13b0: 09 09 7d 2c 20 53 6f 6d 65 28 64 65 73 74 69 6e ..}, Some(destin
13c0: 61 74 69 6f 6e 29 2c 20 53 6f 6d 65 28 50 61 72 ation), Some(Par
13d0: 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 2e 61 seMode::Html)).a
13e0: 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 wait.stack()?;..
13f0: 09 09 09 09 09 63 6f 6e 6e 2e 61 64 64 5f 70 6f .....conn.add_po
1400: 73 74 28 2a 69 64 2c 20 64 61 74 65 2c 20 26 70 st(*id, date, &p
1410: 6f 73 74 5f 75 72 6c 29 2e 61 77 61 69 74 2e 73 ost_url).await.s
1420: 74 61 63 6b 28 29 3f 3b 0a 09 09 09 09 09 7d 3b tack()?;......};
1430: 0a 09 09 09 09 7d 3b 0a 09 09 09 09 70 6f 73 74 .....};.....post
1440: 65 64 20 2b 3d 20 31 3b 0a 09 09 09 7d 3b 0a 09 ed += 1;....};..
1450: 09 09 70 6f 73 74 73 2e 63 6c 65 61 72 28 29 3b ..posts.clear();
1460: 0a 09 09 7d 3b 0a 09 09 4f 6b 28 66 6f 72 6d 61 ...};...Ok(forma
1470: 74 21 28 22 50 6f 73 74 65 64 3a 20 7b 70 6f 73 t!("Posted: {pos
1480: 74 65 64 7d 22 29 29 0a 09 7d 0a 0a 09 61 73 79 ted}"))..}...asy
1490: 6e 63 20 66 6e 20 61 75 74 6f 66 65 74 63 68 28 nc fn autofetch(
14a0: 26 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 &self) -> Result
14b0: 3c 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 <std::time::Dura
14c0: 74 69 6f 6e 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 tion> {...let mu
14d0: 74 20 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f t delay = chrono
14e0: 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 ::Duration::minu
14f0: 74 65 73 28 31 29 3b 0a 09 09 6c 65 74 20 6e 6f tes(1);...let no
1500: 77 20 3d 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 w = chrono::Loca
1510: 6c 3a 3a 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 20 l::now();...let
1520: 71 75 65 75 65 20 3d 20 7b 0a 09 09 09 6c 65 74 queue = {....let
1530: 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 mut conn = self
1540: 2e 64 62 2e 62 65 67 69 6e 28 29 2e 61 77 61 69 .db.begin().awai
1550: 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 09 63 t.stack()?;....c
1560: 6f 6e 6e 2e 67 65 74 5f 71 75 65 75 65 28 29 2e onn.get_queue().
1570: 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 0a 09 await.stack()?..
1580: 09 7d 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e .};...for row in
1590: 20 71 75 65 75 65 20 7b 0a 09 09 09 69 66 20 6c queue {....if l
15a0: 65 74 20 53 6f 6d 65 28 6e 65 78 74 5f 66 65 74 et Some(next_fet
15b0: 63 68 29 20 3d 20 72 6f 77 2e 6e 65 78 74 5f 66 ch) = row.next_f
15c0: 65 74 63 68 20 7b 0a 09 09 09 09 69 66 20 6e 65 etch {.....if ne
15d0: 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 20 7b xt_fetch < now {
15e0: 0a 09 09 09 09 09 69 66 20 6c 65 74 20 28 53 6f ......if let (So
15f0: 6d 65 28 6f 77 6e 65 72 29 2c 20 53 6f 6d 65 28 me(owner), Some(
1600: 73 6f 75 72 63 65 5f 69 64 29 29 20 3d 20 28 72 source_id)) = (r
1610: 6f 77 2e 6f 77 6e 65 72 2c 20 72 6f 77 2e 73 6f ow.owner, row.so
1620: 75 72 63 65 5f 69 64 29 20 7b 0a 09 09 09 09 09 urce_id) {......
1630: 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 .let clone = Cor
1640: 65 20 7b 0a 09 09 09 09 09 09 09 6f 77 6e 65 72 e {........owner
1650: 5f 63 68 61 74 3a 20 43 68 61 74 50 65 65 72 49 _chat: ChatPeerI
1660: 64 3a 3a 66 72 6f 6d 28 6f 77 6e 65 72 29 2c 0a d::from(owner),.
1670: 09 09 09 09 09 09 09 2e 2e 73 65 6c 66 2e 63 6c .........self.cl
1680: 6f 6e 65 28 29 0a 09 09 09 09 09 09 7d 3b 0a 09 one().......};..
1690: 09 09 09 09 09 6c 65 74 20 73 6f 75 72 63 65 20 .....let source
16a0: 3d 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 20 6d = {........let m
16b0: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 64 ut conn = self.d
16c0: 62 2e 62 65 67 69 6e 28 29 2e 61 77 61 69 74 2e b.begin().await.
16d0: 73 74 61 63 6b 28 29 3f 3b 0a 09 09 09 09 09 09 stack()?;.......
16e0: 09 6d 61 74 63 68 20 63 6f 6e 6e 2e 67 65 74 5f .match conn.get_
16f0: 6f 6e 65 28 6f 77 6e 65 72 2c 20 73 6f 75 72 63 one(owner, sourc
1700: 65 5f 69 64 29 2e 61 77 61 69 74 20 7b 0a 09 09 e_id).await {...
1710: 09 09 09 09 09 09 4f 6b 28 53 6f 6d 65 28 73 6f ......Ok(Some(so
1720: 75 72 63 65 29 29 20 3d 3e 20 73 6f 75 72 63 65 urce)) => source
1730: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 2c 0a 09 09 .to_string(),...
1740: 09 09 09 09 09 09 4f 6b 28 4e 6f 6e 65 29 20 3d ......Ok(None) =
1750: 3e 20 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f > "Source not fo
1760: 75 6e 64 20 69 6e 20 64 61 74 61 62 61 73 65 2e und in database.
1770: 73 74 61 63 6b 28 29 3f 22 2e 74 6f 5f 73 74 72 stack()?".to_str
1780: 69 6e 67 28 29 2c 0a 09 09 09 09 09 09 09 09 45 ing(),.........E
1790: 72 72 28 65 72 72 29 20 3d 3e 20 66 6f 72 6d 61 rr(err) => forma
17a0: 74 21 28 22 46 61 69 6c 65 64 20 74 6f 20 66 65 t!("Failed to fe
17b0: 74 63 68 20 73 6f 75 72 63 65 20 64 61 74 61 3a tch source data:
17c0: 5c 6e 7b 65 72 72 7d 22 29 2c 0a 09 09 09 09 09 \n{err}"),......
17d0: 09 09 7d 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 ..}.......};....
17e0: 09 09 09 74 61 73 6b 3a 3a 73 70 61 77 6e 28 61 ...task::spawn(a
17f0: 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 sync move {.....
1800: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
1810: 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b r) = clone.check
1820: 28 73 6f 75 72 63 65 5f 69 64 2c 20 74 72 75 65 (source_id, true
1830: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 ).await {.......
1840: 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 ..if let Err(err
1850: 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 26 ) = clone.send(&
1860: 66 6f 72 6d 61 74 21 28 22 7b 73 6f 75 72 63 65 format!("{source
1870: 7d 5c 6e 5c 6e f0 9f 9b 91 20 7b 7d 22 2c 20 65 }\n\nš {}", e
1880: 6e 63 6f 64 65 28 26 65 72 72 2e 74 6f 5f 73 74 ncode(&err.to_st
1890: 72 69 6e 67 28 29 29 29 2c 20 4e 6f 6e 65 2c 20 ring())), None,
18a0: 53 6f 6d 65 28 50 61 72 73 65 4d 6f 64 65 3a 3a Some(ParseMode::
18b0: 4d 61 72 6b 64 6f 77 6e 56 32 29 29 2e 61 77 61 MarkdownV2)).awa
18c0: 69 74 20 7b 0a 09 09 09 09 09 09 09 09 09 65 70 it {..........ep
18d0: 72 69 6e 74 6c 6e 21 28 22 43 68 65 63 6b 20 65 rintln!("Check e
18e0: 72 72 6f 72 3a 20 7b 65 72 72 3a 3f 7d 22 29 3b rror: {err:?}");
18f0: 0a 09 09 09 09 09 09 09 09 09 2f 2f 20 63 6c 6f ..........// clo
1900: 6e 65 2e 64 69 73 61 62 6c 65 28 26 73 6f 75 72 ne.disable(&sour
1910: 63 65 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 61 77 ce_id, owner).aw
1920: 61 69 74 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 ait.unwrap();...
1930: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 ......};........
1940: 7d 3b 0a 09 09 09 09 09 09 7d 29 3b 0a 09 09 09 };.......});....
1950: 09 09 7d 0a 09 09 09 09 7d 20 65 6c 73 65 20 69 ..}.....} else i
1960: 66 20 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 6e f next_fetch - n
1970: 6f 77 20 3c 20 64 65 6c 61 79 20 7b 0a 09 09 09 ow < delay {....
1980: 09 09 64 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 ..delay = next_f
1990: 65 74 63 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 etch - now;.....
19a0: 7d 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 64 65 }....}...};...de
19b0: 6c 61 79 2e 74 6f 5f 73 74 64 28 29 2e 73 74 61 lay.to_std().sta
19c0: 63 6b 28 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 ck()..}...pub as
19d0: 79 6e 63 20 66 6e 20 6c 69 73 74 20 28 26 73 65 ync fn list (&se
19e0: 6c 66 2c 20 6f 77 6e 65 72 3a 20 55 73 65 72 50 lf, owner: UserP
19f0: 65 65 72 49 64 29 20 2d 3e 20 52 65 73 75 6c 74 eerId) -> Result
1a00: 3c 53 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 74 <String> {...let
1a10: 20 6d 75 74 20 72 65 70 6c 79 3a 20 56 65 63 3c mut reply: Vec<
1a20: 53 74 72 69 6e 67 3e 20 3d 20 76 65 63 21 5b 5d String> = vec![]
1a30: 3b 0a 09 09 72 65 70 6c 79 2e 70 75 73 68 28 22 ;...reply.push("
1a40: 43 68 61 6e 6e 65 6c 73 3a 22 2e 69 6e 74 6f 28 Channels:".into(
1a50: 29 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f ));...let mut co
1a60: 6e 6e 20 3d 20 73 65 6c 66 2e 64 62 2e 62 65 67 nn = self.db.beg
1a70: 69 6e 28 29 2e 61 77 61 69 74 2e 73 74 61 63 6b in().await.stack
1a80: 28 29 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 ()?;...for row i
1a90: 6e 20 63 6f 6e 6e 2e 67 65 74 5f 6c 69 73 74 28 n conn.get_list(
1aa0: 6f 77 6e 65 72 29 2e 61 77 61 69 74 2e 73 74 61 owner).await.sta
1ab0: 63 6b 28 29 3f 20 7b 0a 09 09 09 72 65 70 6c 79 ck()? {....reply
1ac0: 2e 70 75 73 68 28 72 6f 77 2e 74 6f 5f 73 74 72 .push(row.to_str
1ad0: 69 6e 67 28 29 29 3b 0a 09 09 7d 3b 0a 09 09 4f ing());...};...O
1ae0: 6b 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e k(reply.join("\n
1af0: 5c 6e 22 29 29 0a 09 7d 0a 7d 0a 0a 69 6d 70 6c \n"))..}.}..impl
1b00: 20 55 70 64 61 74 65 48 61 6e 64 6c 65 72 20 66 UpdateHandler f
1b10: 6f 72 20 43 6f 72 65 20 7b 0a 09 61 73 79 6e 63 or Core {..async
1b20: 20 66 6e 20 68 61 6e 64 6c 65 20 28 26 73 65 6c fn handle (&sel
1b30: 66 2c 20 75 70 64 61 74 65 3a 20 55 70 64 61 74 f, update: Updat
1b40: 65 29 20 7b 0a 09 09 69 66 20 6c 65 74 20 55 70 e) {...if let Up
1b50: 64 61 74 65 54 79 70 65 3a 3a 4d 65 73 73 61 67 dateType::Messag
1b60: 65 28 6d 73 67 29 20 3d 20 75 70 64 61 74 65 2e e(msg) = update.
1b70: 75 70 64 61 74 65 5f 74 79 70 65 20 7b 0a 09 09 update_type {...
1b80: 09 69 66 20 6c 65 74 20 4f 6b 28 63 6d 64 29 20 .if let Ok(cmd)
1b90: 3d 20 43 6f 6d 6d 61 6e 64 3a 3a 74 72 79 5f 66 = Command::try_f
1ba0: 72 6f 6d 28 6d 73 67 29 20 7b 0a 09 09 09 09 6c rom(msg) {.....l
1bb0: 65 74 20 6d 73 67 20 3d 20 63 6d 64 2e 67 65 74 et msg = cmd.get
1bc0: 5f 6d 65 73 73 61 67 65 28 29 3b 0a 09 09 09 09 _message();.....
1bd0: 6c 65 74 20 77 6f 72 64 73 20 3d 20 63 6d 64 2e let words = cmd.
1be0: 67 65 74 5f 61 72 67 73 28 29 3b 0a 09 09 09 09 get_args();.....
1bf0: 6c 65 74 20 63 6f 6d 6d 61 6e 64 20 3d 20 63 6d let command = cm
1c00: 64 2e 67 65 74 5f 6e 61 6d 65 28 29 3b 0a 09 09 d.get_name();...
1c10: 09 09 6c 65 74 20 72 65 73 20 3d 20 6d 61 74 63 ..let res = matc
1c20: 68 20 63 6f 6d 6d 61 6e 64 20 7b 0a 09 09 09 09 h command {.....
1c30: 09 22 2f 63 68 65 63 6b 22 20 7c 20 22 2f 63 6c ."/check" | "/cl
1c40: 65 61 6e 22 20 7c 20 22 2f 65 6e 61 62 6c 65 22 ean" | "/enable"
1c50: 20 7c 20 22 2f 64 65 6c 65 74 65 22 20 7c 20 22 | "/delete" | "
1c60: 2f 64 69 73 61 62 6c 65 22 20 3d 3e 20 63 6f 6d /disable" => com
1c70: 6d 61 6e 64 3a 3a 63 6f 6d 6d 61 6e 64 28 73 65 mand::command(se
1c80: 6c 66 2c 20 63 6f 6d 6d 61 6e 64 2c 20 6d 73 67 lf, command, msg
1c90: 2c 20 77 6f 72 64 73 29 2e 61 77 61 69 74 2c 0a , words).await,.
1ca0: 09 09 09 09 09 22 2f 73 74 61 72 74 22 20 3d 3e ....."/start" =>
1cb0: 20 63 6f 6d 6d 61 6e 64 3a 3a 73 74 61 72 74 28 command::start(
1cc0: 73 65 6c 66 2c 20 6d 73 67 29 2e 61 77 61 69 74 self, msg).await
1cd0: 2c 0a 09 09 09 09 09 22 2f 6c 69 73 74 22 20 3d ,......"/list" =
1ce0: 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 6c 69 73 74 28 > command::list(
1cf0: 73 65 6c 66 2c 20 6d 73 67 29 2e 61 77 61 69 74 self, msg).await
1d00: 2c 0a 09 09 09 09 09 22 2f 61 64 64 22 20 7c 20 ,......"/add" |
1d10: 22 2f 75 70 64 61 74 65 22 20 3d 3e 20 63 6f 6d "/update" => com
1d20: 6d 61 6e 64 3a 3a 75 70 64 61 74 65 28 73 65 6c mand::update(sel
1d30: 66 2c 20 63 6f 6d 6d 61 6e 64 2c 20 6d 73 67 2c f, command, msg,
1d40: 20 77 6f 72 64 73 29 2e 61 77 61 69 74 2c 0a 09 words).await,..
1d50: 09 09 09 09 61 6e 79 20 3d 3e 20 45 72 72 28 61 ....any => Err(a
1d60: 6e 79 68 6f 77 21 28 22 55 6e 6b 6e 6f 77 6e 20 nyhow!("Unknown
1d70: 63 6f 6d 6d 61 6e 64 3a 20 7b 61 6e 79 7d 22 29 command: {any}")
1d80: 29 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 69 66 ),.....};.....if
1d90: 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 let Err(err) =
1da0: 72 65 73 20 7b 0a 09 09 09 09 09 69 66 20 6c 65 res {......if le
1db0: 74 20 45 72 72 28 65 72 72 32 29 20 3d 20 73 65 t Err(err2) = se
1dc0: 6c 66 2e 73 65 6e 64 28 66 6f 72 6d 61 74 21 28 lf.send(format!(
1dd0: 22 5c 5c 23 65 72 72 6f 72 5c 6e 60 60 60 5c 6e "\\#error\n```\n
1de0: 7b 65 72 72 7d 5c 6e 60 60 60 22 29 2c 0a 09 09 {err}\n```"),...
1df0: 09 09 09 09 53 6f 6d 65 28 6d 73 67 2e 63 68 61 ....Some(msg.cha
1e00: 74 2e 67 65 74 5f 69 64 28 29 29 2c 0a 09 09 09 t.get_id()),....
1e10: 09 09 09 53 6f 6d 65 28 50 61 72 73 65 4d 6f 64 ...Some(ParseMod
1e20: 65 3a 3a 4d 61 72 6b 64 6f 77 6e 56 32 29 0a 09 e::MarkdownV2)..
1e30: 09 09 09 09 29 2e 61 77 61 69 74 7b 0a 09 09 09 ....).await{....
1e40: 09 09 09 64 62 67 21 28 65 72 72 32 29 3b 0a 09 ...dbg!(err2);..
1e50: 09 09 09 09 7d 3b 0a 09 09 09 09 7d 0a 09 09 09 ....};.....}....
1e60: 7d 3b 0a 09 09 7d 3b 0a 09 7d 0a 7d 0a };...};..}.}.