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 6e 79 68 6f 77 3a 3a 7b 0a 09 61 6e 79 68 anyhow::{..anyh
0080: 6f 77 2c 0a 09 62 61 69 6c 2c 0a 09 52 65 73 75 ow,..bail,..Resu
0090: 6c 74 2c 0a 7d 3b 0a 75 73 65 20 61 73 79 6e 63 lt,.};.use async
00a0: 5f 73 74 64 3a 3a 7b 0a 09 74 61 73 6b 2c 0a 09 _std::{..task,..
00b0: 73 79 6e 63 3a 3a 7b 0a 09 09 41 72 63 2c 0a 09 sync::{...Arc,..
00c0: 09 4d 75 74 65 78 0a 09 7d 2c 0a 7d 3b 0a 75 73 .Mutex..},.};.us
00d0: 65 20 63 68 72 6f 6e 6f 3a 3a 44 61 74 65 54 69 e chrono::DateTi
00e0: 6d 65 3b 0a 75 73 65 20 6c 61 7a 79 5f 73 74 61 me;.use lazy_sta
00f0: 74 69 63 3a 3a 6c 61 7a 79 5f 73 74 61 74 69 63 tic::lazy_static
0100: 3b 0a 75 73 65 20 72 65 67 65 78 3a 3a 52 65 67 ;.use regex::Reg
0110: 65 78 3b 0a 75 73 65 20 74 67 62 6f 74 3a 3a 7b ex;.use tgbot::{
0120: 0a 09 61 70 69 3a 3a 43 6c 69 65 6e 74 2c 0a 09 ..api::Client,..
0130: 68 61 6e 64 6c 65 72 3a 3a 55 70 64 61 74 65 48 handler::UpdateH
0140: 61 6e 64 6c 65 72 2c 0a 09 74 79 70 65 73 3a 3a andler,..types::
0150: 7b 0a 09 09 42 6f 74 2c 0a 09 09 43 68 61 74 50 {...Bot,...ChatP
0160: 65 65 72 49 64 2c 0a 09 09 43 6f 6d 6d 61 6e 64 eerId,...Command
0170: 2c 0a 09 09 47 65 74 42 6f 74 2c 0a 09 09 4d 65 ,...GetBot,...Me
0180: 73 73 61 67 65 2c 0a 09 09 50 61 72 73 65 4d 6f ssage,...ParseMo
0190: 64 65 2c 0a 09 09 53 65 6e 64 4d 65 73 73 61 67 de,...SendMessag
01a0: 65 2c 0a 09 09 55 70 64 61 74 65 2c 0a 09 09 55 e,...Update,...U
01b0: 70 64 61 74 65 54 79 70 65 2c 0a 09 09 55 73 65 pdateType,...Use
01c0: 72 50 65 65 72 49 64 2c 0a 09 7d 2c 0a 7d 3b 0a rPeerId,..},.};.
01d0: 0a 6c 61 7a 79 5f 73 74 61 74 69 63 21 7b 0a 09 .lazy_static!{..
01e0: 70 75 62 20 73 74 61 74 69 63 20 72 65 66 20 52 pub static ref R
01f0: 45 5f 53 50 45 43 49 41 4c 3a 20 52 65 67 65 78 E_SPECIAL: Regex
0200: 20 3d 20 52 65 67 65 78 3a 3a 6e 65 77 28 72 22 = Regex::new(r"
0210: 28 5b 5c 2d 5f 2a 5c 5b 5c 5d 28 29 7e 60 3e 23 ([\-_*\[\]()~`>#
0220: 2b 7c 7b 7d 5c 2e 21 5d 29 22 29 2e 75 6e 77 72 +|{}\.!])").unwr
0230: 61 70 28 29 3b 0a 7d 0a 0a 2f 2f 2f 20 45 6e 63 ap();.}../// Enc
0240: 6f 64 65 73 20 73 70 65 63 69 61 6c 20 48 54 4d odes special HTM
0250: 4c 20 65 6e 74 69 74 69 65 73 20 74 6f 20 70 72 L entities to pr
0260: 65 76 65 6e 74 20 74 68 65 6d 20 69 6e 74 65 72 event them inter
0270: 66 65 72 69 6e 67 20 77 69 74 68 20 54 65 6c 65 fering with Tele
0280: 67 72 61 6d 20 48 54 4d 4c 0a 70 75 62 20 66 6e gram HTML.pub fn
0290: 20 65 6e 63 6f 64 65 20 28 74 65 78 74 3a 20 26 encode (text: &
02a0: 73 74 72 29 20 2d 3e 20 43 6f 77 3c 27 5f 2c 20 str) -> Cow<'_,
02b0: 73 74 72 3e 20 7b 0a 09 52 45 5f 53 50 45 43 49 str> {..RE_SPECI
02c0: 41 4c 2e 72 65 70 6c 61 63 65 5f 61 6c 6c 28 74 AL.replace_all(t
02d0: 65 78 74 2c 20 22 5c 5c 24 31 22 29 0a 7d 0a 0a ext, "\\$1").}..
02e0: 23 5b 64 65 72 69 76 65 28 43 6c 6f 6e 65 29 5d #[derive(Clone)]
02f0: 0a 70 75 62 20 73 74 72 75 63 74 20 43 6f 72 65 .pub struct Core
0300: 20 7b 0a 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 {..owner_chat:
0310: 43 68 61 74 50 65 65 72 49 64 2c 0a 09 2f 2f 20 ChatPeerId,..//
0320: 6d 61 78 5f 64 65 6c 61 79 3a 20 75 31 36 2c 0a max_delay: u16,.
0330: 09 70 75 62 20 74 67 3a 20 43 6c 69 65 6e 74 2c .pub tg: Client,
0340: 0a 09 70 75 62 20 6d 65 3a 20 42 6f 74 2c 0a 09 ..pub me: Bot,..
0350: 70 75 62 20 64 62 3a 20 44 62 2c 0a 09 73 6f 75 pub db: Db,..sou
0360: 72 63 65 73 3a 20 41 72 63 3c 4d 75 74 65 78 3c rces: Arc<Mutex<
0370: 48 61 73 68 53 65 74 3c 41 72 63 3c 69 33 32 3e HashSet<Arc<i32>
0380: 3e 3e 3e 2c 0a 09 68 74 74 70 5f 63 6c 69 65 6e >>>,..http_clien
0390: 74 3a 20 72 65 71 77 65 73 74 3a 3a 43 6c 69 65 t: reqwest::Clie
03a0: 6e 74 2c 0a 7d 0a 0a 69 6d 70 6c 20 43 6f 72 65 nt,.}..impl Core
03b0: 20 7b 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e {..pub async fn
03c0: 20 6e 65 77 28 73 65 74 74 69 6e 67 73 3a 20 63 new(settings: c
03d0: 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 29 20 2d onfig::Config) -
03e0: 3e 20 52 65 73 75 6c 74 3c 43 6f 72 65 3e 20 7b > Result<Core> {
03f0: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 5f 63 68 61 ...let owner_cha
0400: 74 20 3d 20 43 68 61 74 50 65 65 72 49 64 3a 3a t = ChatPeerId::
0410: 66 72 6f 6d 28 73 65 74 74 69 6e 67 73 2e 67 65 from(settings.ge
0420: 74 5f 69 6e 74 28 22 6f 77 6e 65 72 22 29 3f 29 t_int("owner")?)
0430: 3b 0a 09 09 6c 65 74 20 61 70 69 5f 6b 65 79 20 ;...let api_key
0440: 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 5f 73 = settings.get_s
0450: 74 72 69 6e 67 28 22 61 70 69 5f 6b 65 79 22 29 tring("api_key")
0460: 3f 3b 0a 09 09 6c 65 74 20 74 67 20 3d 20 43 6c ?;...let tg = Cl
0470: 69 65 6e 74 3a 3a 6e 65 77 28 26 61 70 69 5f 6b ient::new(&api_k
0480: 65 79 29 3f 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 ey)?;....let mut
0490: 20 63 6c 69 65 6e 74 20 3d 20 72 65 71 77 65 73 client = reqwes
04a0: 74 3a 3a 43 6c 69 65 6e 74 3a 3a 62 75 69 6c 64 t::Client::build
04b0: 65 72 28 29 3b 0a 09 09 69 66 20 6c 65 74 20 4f er();...if let O
04c0: 6b 28 70 72 6f 78 79 29 20 3d 20 73 65 74 74 69 k(proxy) = setti
04d0: 6e 67 73 2e 67 65 74 5f 73 74 72 69 6e 67 28 22 ngs.get_string("
04e0: 70 72 6f 78 79 22 29 20 7b 0a 09 09 09 6c 65 74 proxy") {....let
04f0: 20 70 72 6f 78 79 20 3d 20 72 65 71 77 65 73 74 proxy = reqwest
0500: 3a 3a 50 72 6f 78 79 3a 3a 61 6c 6c 28 70 72 6f ::Proxy::all(pro
0510: 78 79 29 3f 3b 0a 09 09 09 63 6c 69 65 6e 74 20 xy)?;....client
0520: 3d 20 63 6c 69 65 6e 74 2e 70 72 6f 78 79 28 70 = client.proxy(p
0530: 72 6f 78 79 29 3b 0a 09 09 7d 0a 09 09 6c 65 74 roxy);...}...let
0540: 20 68 74 74 70 5f 63 6c 69 65 6e 74 20 3d 20 63 http_client = c
0550: 6c 69 65 6e 74 2e 62 75 69 6c 64 28 29 3f 3b 0a lient.build()?;.
0560: 09 09 6c 65 74 20 6d 65 20 3d 20 74 67 2e 65 78 ..let me = tg.ex
0570: 65 63 75 74 65 28 47 65 74 42 6f 74 29 2e 61 77 ecute(GetBot).aw
0580: 61 69 74 3f 3b 0a 09 09 6c 65 74 20 63 6f 72 65 ait?;...let core
0590: 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 74 67 2c = Core {....tg,
05a0: 0a 09 09 09 6d 65 2c 0a 09 09 09 6f 77 6e 65 72 ....me,....owner
05b0: 5f 63 68 61 74 2c 0a 09 09 09 64 62 3a 20 44 62 _chat,....db: Db
05c0: 3a 3a 6e 65 77 28 26 73 65 74 74 69 6e 67 73 2e ::new(&settings.
05d0: 67 65 74 5f 73 74 72 69 6e 67 28 22 70 67 22 29 get_string("pg")
05e0: 3f 29 3f 2c 0a 09 09 09 73 6f 75 72 63 65 73 3a ?)?,....sources:
05f0: 20 41 72 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a Arc::new(Mutex:
0600: 3a 6e 65 77 28 48 61 73 68 53 65 74 3a 3a 6e 65 :new(HashSet::ne
0610: 77 28 29 29 29 2c 0a 09 09 09 68 74 74 70 5f 63 w())),....http_c
0620: 6c 69 65 6e 74 2c 0a 09 09 09 2f 2f 20 6d 61 78 lient,....// max
0630: 5f 64 65 6c 61 79 3a 20 36 30 2c 0a 09 09 7d 3b _delay: 60,...};
0640: 0a 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 63 ...let clone = c
0650: 6f 72 65 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 74 ore.clone();...t
0660: 61 73 6b 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 ask::spawn(async
0670: 20 6d 6f 76 65 20 7b 0a 09 09 09 6c 6f 6f 70 20 move {....loop
0680: 7b 0a 09 09 09 09 6c 65 74 20 64 65 6c 61 79 20 {.....let delay
0690: 3d 20 6d 61 74 63 68 20 26 63 6c 6f 6e 65 2e 61 = match &clone.a
06a0: 75 74 6f 66 65 74 63 68 28 29 2e 61 77 61 69 74 utofetch().await
06b0: 20 7b 0a 09 09 09 09 09 45 72 72 28 65 72 72 29 {......Err(err)
06c0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 69 66 20 6c => {.......if l
06d0: 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c et Err(err) = cl
06e0: 6f 6e 65 2e 73 65 6e 64 28 66 6f 72 6d 61 74 21 one.send(format!
06f0: 28 22 f0 9f 9b 91 20 7b 65 72 72 3a 3f 7d 22 29 ("š {err:?}")
0700: 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 , None, None).aw
0710: 61 69 74 20 7b 0a 09 09 09 09 09 09 09 65 70 72 ait {........epr
0720: 69 6e 74 6c 6e 21 28 22 41 75 74 6f 66 65 74 63 intln!("Autofetc
0730: 68 20 65 72 72 6f 72 3a 20 7b 65 72 72 3a 3f 7d h error: {err:?}
0740: 22 29 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 ");.......};....
0750: 09 09 09 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 ...std::time::Du
0760: 72 61 74 69 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 63 ration::from_sec
0770: 73 28 36 30 29 0a 09 09 09 09 09 7d 2c 0a 09 09 s(60)......},...
0780: 09 09 09 4f 6b 28 74 69 6d 65 29 20 3d 3e 20 2a ...Ok(time) => *
0790: 74 69 6d 65 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 time,.....};....
07a0: 09 74 61 73 6b 3a 3a 73 6c 65 65 70 28 64 65 6c .task::sleep(del
07b0: 61 79 29 2e 61 77 61 69 74 3b 0a 09 09 09 7d 0a ay).await;....}.
07c0: 09 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 ..});...Ok(core)
07d0: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
07e0: 66 6e 20 73 65 6e 64 20 3c 53 3e 28 26 73 65 6c fn send <S>(&sel
07f0: 66 2c 20 6d 73 67 3a 20 53 2c 20 74 61 72 67 65 f, msg: S, targe
0800: 74 3a 20 4f 70 74 69 6f 6e 3c 43 68 61 74 50 65 t: Option<ChatPe
0810: 65 72 49 64 3e 2c 20 6d 6f 64 65 3a 20 4f 70 74 erId>, mode: Opt
0820: 69 6f 6e 3c 50 61 72 73 65 4d 6f 64 65 3e 29 20 ion<ParseMode>)
0830: 2d 3e 20 52 65 73 75 6c 74 3c 4d 65 73 73 61 67 -> Result<Messag
0840: 65 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 e>..where S: Int
0850: 6f 3c 53 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 o<String> {...le
0860: 74 20 6d 73 67 20 3d 20 6d 73 67 2e 69 6e 74 6f t msg = msg.into
0870: 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 6f 64 65 20 ();....let mode
0880: 3d 20 6d 6f 64 65 2e 75 6e 77 72 61 70 5f 6f 72 = mode.unwrap_or
0890: 28 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c (ParseMode::Html
08a0: 29 3b 0a 09 09 6c 65 74 20 74 61 72 67 65 74 20 );...let target
08b0: 3d 20 74 61 72 67 65 74 2e 75 6e 77 72 61 70 5f = target.unwrap_
08c0: 6f 72 28 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 or(self.owner_ch
08d0: 61 74 29 3b 0a 09 09 4f 6b 28 73 65 6c 66 2e 74 at);...Ok(self.t
08e0: 67 2e 65 78 65 63 75 74 65 28 0a 09 09 09 53 65 g.execute(....Se
08f0: 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 74 ndMessage::new(t
0900: 61 72 67 65 74 2c 20 6d 73 67 29 0a 09 09 09 09 arget, msg).....
0910: 2e 77 69 74 68 5f 70 61 72 73 65 5f 6d 6f 64 65 .with_parse_mode
0920: 28 6d 6f 64 65 29 0a 09 09 29 2e 61 77 61 69 74 (mode)...).await
0930: 3f 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e ?)..}...pub asyn
0940: 63 20 66 6e 20 63 68 65 63 6b 20 28 26 73 65 6c c fn check (&sel
0950: 66 2c 20 69 64 3a 20 69 33 32 2c 20 72 65 61 6c f, id: i32, real
0960: 3a 20 62 6f 6f 6c 29 20 2d 3e 20 52 65 73 75 6c : bool) -> Resul
0970: 74 3c 53 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 t<String> {...le
0980: 74 20 6d 75 74 20 70 6f 73 74 65 64 3a 20 69 33 t mut posted: i3
0990: 32 20 3d 20 30 3b 0a 09 09 6c 65 74 20 6d 75 74 2 = 0;...let mut
09a0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 64 62 2e conn = self.db.
09b0: 62 65 67 69 6e 28 29 2e 61 77 61 69 74 3f 3b 0a begin().await?;.
09c0: 0a 09 09 6c 65 74 20 69 64 20 3d 20 7b 0a 09 09 ...let id = {...
09d0: 09 6c 65 74 20 6d 75 74 20 73 65 74 20 3d 20 73 .let mut set = s
09e0: 65 6c 66 2e 73 6f 75 72 63 65 73 2e 6c 6f 63 6b elf.sources.lock
09f0: 5f 61 72 63 28 29 2e 61 77 61 69 74 3b 0a 09 09 _arc().await;...
0a00: 09 6d 61 74 63 68 20 73 65 74 2e 67 65 74 28 26 .match set.get(&
0a10: 69 64 29 20 7b 0a 09 09 09 09 53 6f 6d 65 28 69 id) {.....Some(i
0a20: 64 29 20 3d 3e 20 69 64 2e 63 6c 6f 6e 65 28 29 d) => id.clone()
0a30: 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a ,.....None => {.
0a40: 09 09 09 09 09 6c 65 74 20 69 64 20 3d 20 41 72 .....let id = Ar
0a50: 63 3a 3a 6e 65 77 28 69 64 29 3b 0a 09 09 09 09 c::new(id);.....
0a60: 09 73 65 74 2e 69 6e 73 65 72 74 28 69 64 2e 63 .set.insert(id.c
0a70: 6c 6f 6e 65 28 29 29 3b 0a 09 09 09 09 09 69 64 lone());......id
0a80: 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 7d 2c 0a .clone().....},.
0a90: 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 ...}...};...let
0aa0: 63 6f 75 6e 74 20 3d 20 41 72 63 3a 3a 73 74 72 count = Arc::str
0ab0: 6f 6e 67 5f 63 6f 75 6e 74 28 26 69 64 29 3b 0a ong_count(&id);.
0ac0: 09 09 69 66 20 63 6f 75 6e 74 20 3d 3d 20 32 20 ..if count == 2
0ad0: 7b 0a 09 09 09 6c 65 74 20 73 6f 75 72 63 65 20 {....let source
0ae0: 3d 20 63 6f 6e 6e 2e 67 65 74 5f 73 6f 75 72 63 = conn.get_sourc
0af0: 65 28 2a 69 64 2c 20 73 65 6c 66 2e 6f 77 6e 65 e(*id, self.owne
0b00: 72 5f 63 68 61 74 29 2e 61 77 61 69 74 3f 3b 0a r_chat).await?;.
0b10: 09 09 09 63 6f 6e 6e 2e 73 65 74 5f 73 63 72 61 ...conn.set_scra
0b20: 70 65 28 2a 69 64 29 2e 61 77 61 69 74 3f 3b 0a pe(*id).await?;.
0b30: 09 09 09 6c 65 74 20 64 65 73 74 69 6e 61 74 69 ...let destinati
0b40: 6f 6e 20 3d 20 43 68 61 74 50 65 65 72 49 64 3a on = ChatPeerId:
0b50: 3a 66 72 6f 6d 28 6d 61 74 63 68 20 72 65 61 6c :from(match real
0b60: 20 7b 0a 09 09 09 09 74 72 75 65 20 3d 3e 20 73 {.....true => s
0b70: 6f 75 72 63 65 2e 63 68 61 6e 6e 65 6c 5f 69 64 ource.channel_id
0b80: 2c 0a 09 09 09 09 66 61 6c 73 65 20 3d 3e 20 73 ,.....false => s
0b90: 6f 75 72 63 65 2e 6f 77 6e 65 72 2c 0a 09 09 09 ource.owner,....
0ba0: 7d 29 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 74 });....let mut t
0bb0: 68 69 73 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f his_fetch: Optio
0bc0: 6e 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e n<DateTime<chron
0bd0: 6f 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 3e o::FixedOffset>>
0be0: 20 3d 20 4e 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 = None;....let
0bf0: 6d 75 74 20 70 6f 73 74 73 3a 20 42 54 72 65 65 mut posts: BTree
0c00: 4d 61 70 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 Map<DateTime<chr
0c10: 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 ono::FixedOffset
0c20: 3e 2c 20 53 74 72 69 6e 67 3e 20 3d 20 42 54 72 >, String> = BTr
0c30: 65 65 4d 61 70 3a 3a 6e 65 77 28 29 3b 0a 0a 09 eeMap::new();...
0c40: 09 09 6c 65 74 20 72 65 73 70 6f 6e 73 65 20 3d ..let response =
0c50: 20 73 65 6c 66 2e 68 74 74 70 5f 63 6c 69 65 6e self.http_clien
0c60: 74 2e 67 65 74 28 26 73 6f 75 72 63 65 2e 75 72 t.get(&source.ur
0c70: 6c 29 2e 73 65 6e 64 28 29 2e 61 77 61 69 74 3f l).send().await?
0c80: 3b 0a 09 09 09 6c 65 74 20 73 74 61 74 75 73 20 ;....let status
0c90: 3d 20 72 65 73 70 6f 6e 73 65 2e 73 74 61 74 75 = response.statu
0ca0: 73 28 29 3b 0a 09 09 09 6c 65 74 20 63 6f 6e 74 s();....let cont
0cb0: 65 6e 74 20 3d 20 72 65 73 70 6f 6e 73 65 2e 62 ent = response.b
0cc0: 79 74 65 73 28 29 2e 61 77 61 69 74 3f 3b 0a 09 ytes().await?;..
0cd0: 09 09 6d 61 74 63 68 20 72 73 73 3a 3a 43 68 61 ..match rss::Cha
0ce0: 6e 6e 65 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 nnel::read_from(
0cf0: 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a &content[..]) {.
0d00: 09 09 09 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 ....Ok(feed) =>
0d10: 7b 0a 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 {......for item
0d20: 69 6e 20 66 65 65 64 2e 69 74 65 6d 73 28 29 20 in feed.items()
0d30: 7b 0a 09 09 09 09 09 09 69 66 20 6c 65 74 20 53 {.......if let S
0d40: 6f 6d 65 28 6c 69 6e 6b 29 20 3d 20 69 74 65 6d ome(link) = item
0d50: 2e 6c 69 6e 6b 28 29 20 7b 0a 09 09 09 09 09 09 .link() {.......
0d60: 09 6c 65 74 20 64 61 74 65 20 3d 20 6d 61 74 63 .let date = matc
0d70: 68 20 69 74 65 6d 2e 70 75 62 5f 64 61 74 65 28 h item.pub_date(
0d80: 29 20 7b 0a 09 09 09 09 09 09 09 09 53 6f 6d 65 ) {.........Some
0d90: 28 66 65 65 64 5f 64 61 74 65 29 20 3d 3e 20 44 (feed_date) => D
0da0: 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 ateTime::parse_f
0db0: 72 6f 6d 5f 72 66 63 32 38 32 32 28 66 65 65 64 rom_rfc2822(feed
0dc0: 5f 64 61 74 65 29 2c 0a 09 09 09 09 09 09 09 09 _date),.........
0dd0: 4e 6f 6e 65 20 3d 3e 20 44 61 74 65 54 69 6d 65 None => DateTime
0de0: 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 ::parse_from_rfc
0df0: 33 33 33 39 28 26 69 74 65 6d 2e 64 75 62 6c 69 3339(&item.dubli
0e00: 6e 5f 63 6f 72 65 5f 65 78 74 28 29 2e 75 6e 77 n_core_ext().unw
0e10: 72 61 70 28 29 2e 64 61 74 65 73 28 29 5b 30 5d rap().dates()[0]
0e20: 29 2c 0a 09 09 09 09 09 09 09 7d 3f 3b 0a 09 09 ),........}?;...
0e30: 09 09 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 6c .....let url = l
0e40: 69 6e 6b 3b 0a 09 09 09 09 09 09 09 70 6f 73 74 ink;........post
0e50: 73 2e 69 6e 73 65 72 74 28 64 61 74 65 2c 20 75 s.insert(date, u
0e60: 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b rl.to_string());
0e70: 0a 09 09 09 09 09 09 7d 0a 09 09 09 09 09 7d 3b .......}......};
0e80: 0a 09 09 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 .....},.....Err(
0e90: 65 72 72 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 err) => match er
0ea0: 72 20 7b 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 r {......rss::Er
0eb0: 72 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 ror::InvalidStar
0ec0: 74 54 61 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 tTag => {.......
0ed0: 6d 61 74 63 68 20 61 74 6f 6d 5f 73 79 6e 64 69 match atom_syndi
0ee0: 63 61 74 69 6f 6e 3a 3a 46 65 65 64 3a 3a 72 65 cation::Feed::re
0ef0: 61 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 ad_from(&content
0f00: 5b 2e 2e 5d 29 20 7b 0a 09 09 09 09 09 09 09 4f [..]) {........O
0f10: 6b 28 66 65 65 64 29 20 3d 3e 20 7b 0a 09 09 09 k(feed) => {....
0f20: 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 69 6e .....for item in
0f30: 20 66 65 65 64 2e 65 6e 74 72 69 65 73 28 29 20 feed.entries()
0f40: 7b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 64 {..........let d
0f50: 61 74 65 20 3d 20 69 74 65 6d 2e 70 75 62 6c 69 ate = item.publi
0f60: 73 68 65 64 28 29 2e 75 6e 77 72 61 70 28 29 3b shed().unwrap();
0f70: 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 75 72 ..........let ur
0f80: 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 29 l = item.links()
0f90: 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 09 09 [0].href();.....
0fa0: 09 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 .....posts.inser
0fb0: 74 28 2a 64 61 74 65 2c 20 75 72 6c 2e 74 6f 5f t(*date, url.to_
0fc0: 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 string());......
0fd0: 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 7d 2c 0a ...};........},.
0fe0: 09 09 09 09 09 09 09 45 72 72 28 65 72 72 29 20 .......Err(err)
0ff0: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 62 61 69 => {.........bai
1000: 6c 21 28 22 55 6e 73 75 70 70 6f 72 74 65 64 20 l!("Unsupported
1010: 6f 72 20 6d 61 6e 67 6c 65 64 20 63 6f 6e 74 65 or mangled conte
1020: 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b 65 72 72 3a nt:\n{:?}\n{err:
1030: 23 3f 7d 5c 6e 7b 73 74 61 74 75 73 3a 23 3f 7d #?}\n{status:#?}
1040: 5c 6e 22 2c 20 26 73 6f 75 72 63 65 2e 75 72 6c \n", &source.url
1050: 29 0a 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 )........},.....
1060: 09 09 7d 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 ..}......},.....
1070: 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 6f 66 .rss::Error::Eof
1080: 20 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f 20 3d => (),......_ =
1090: 3e 20 62 61 69 6c 21 28 22 55 6e 73 75 70 70 6f > bail!("Unsuppo
10a0: 72 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 64 20 rted or mangled
10b0: 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e content:\n{:?}\n
10c0: 7b 65 72 72 3a 23 3f 7d 5c 6e 7b 73 74 61 74 75 {err:#?}\n{statu
10d0: 73 3a 23 3f 7d 5c 6e 22 2c 20 26 73 6f 75 72 63 s:#?}\n", &sourc
10e0: 65 2e 75 72 6c 29 0a 09 09 09 09 7d 0a 09 09 09 e.url).....}....
10f0: 7d 3b 0a 09 09 09 66 6f 72 20 28 64 61 74 65 2c };....for (date,
1100: 20 75 72 6c 29 20 69 6e 20 70 6f 73 74 73 2e 69 url) in posts.i
1110: 74 65 72 28 29 20 7b 0a 09 09 09 09 6c 65 74 20 ter() {.....let
1120: 70 6f 73 74 5f 75 72 6c 3a 20 43 6f 77 3c 73 74 post_url: Cow<st
1130: 72 3e 20 3d 20 6d 61 74 63 68 20 73 6f 75 72 63 r> = match sourc
1140: 65 2e 75 72 6c 5f 72 65 20 7b 0a 09 09 09 09 09 e.url_re {......
1150: 53 6f 6d 65 28 72 65 66 20 78 29 20 3d 3e 20 73 Some(ref x) => s
1160: 65 64 72 65 67 65 78 3a 3a 52 65 70 6c 61 63 65 edregex::Replace
1170: 43 6f 6d 6d 61 6e 64 3a 3a 6e 65 77 28 78 29 3f Command::new(x)?
1180: 2e 65 78 65 63 75 74 65 28 75 72 6c 29 2c 0a 09 .execute(url),..
1190: 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 75 72 6c 2e ....None => url.
11a0: 69 6e 74 6f 28 29 2c 0a 09 09 09 09 7d 3b 0a 09 into(),.....};..
11b0: 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 65 ...if let Some(e
11c0: 78 69 73 74 73 29 20 3d 20 63 6f 6e 6e 2e 65 78 xists) = conn.ex
11d0: 69 73 74 73 28 26 70 6f 73 74 5f 75 72 6c 2c 20 ists(&post_url,
11e0: 2a 69 64 29 2e 61 77 61 69 74 3f 20 7b 0a 09 09 *id).await? {...
11f0: 09 09 09 69 66 20 21 20 65 78 69 73 74 73 20 7b ...if ! exists {
1200: 0a 09 09 09 09 09 09 69 66 20 74 68 69 73 5f 66 .......if this_f
1210: 65 74 63 68 2e 69 73 5f 6e 6f 6e 65 28 29 20 7c etch.is_none() |
1220: 7c 20 2a 64 61 74 65 20 3e 20 74 68 69 73 5f 66 | *date > this_f
1230: 65 74 63 68 2e 75 6e 77 72 61 70 28 29 20 7b 0a etch.unwrap() {.
1240: 09 09 09 09 09 09 09 74 68 69 73 5f 66 65 74 63 .......this_fetc
1250: 68 20 3d 20 53 6f 6d 65 28 2a 64 61 74 65 29 3b h = Some(*date);
1260: 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 .......};.......
1270: 73 65 6c 66 2e 73 65 6e 64 28 20 6d 61 74 63 68 self.send( match
1280: 20 26 73 6f 75 72 63 65 2e 69 76 5f 68 61 73 68 &source.iv_hash
1290: 20 7b 0a 09 09 09 09 09 09 09 53 6f 6d 65 28 68 {........Some(h
12a0: 61 73 68 29 20 3d 3e 20 66 6f 72 6d 61 74 21 28 ash) => format!(
12b0: 22 3c 61 20 68 72 65 66 3d 5c 22 68 74 74 70 73 "<a href=\"https
12c0: 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 72 6c 3d 7b ://t.me/iv?url={
12d0: 70 6f 73 74 5f 75 72 6c 7d 26 72 68 61 73 68 3d post_url}&rhash=
12e0: 7b 68 61 73 68 7d 5c 22 3e 20 3c 2f 61 3e 7b 70 {hash}\"> </a>{p
12f0: 6f 73 74 5f 75 72 6c 7d 22 29 2c 0a 09 09 09 09 ost_url}"),.....
1300: 09 09 09 4e 6f 6e 65 20 3d 3e 20 66 6f 72 6d 61 ...None => forma
1310: 74 21 28 22 7b 70 6f 73 74 5f 75 72 6c 7d 22 29 t!("{post_url}")
1320: 2c 0a 09 09 09 09 09 09 7d 2c 20 53 6f 6d 65 28 ,.......}, Some(
1330: 64 65 73 74 69 6e 61 74 69 6f 6e 29 2c 20 53 6f destination), So
1340: 6d 65 28 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 me(ParseMode::Ht
1350: 6d 6c 29 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 ml)).await?;....
1360: 09 09 09 63 6f 6e 6e 2e 61 64 64 5f 70 6f 73 74 ...conn.add_post
1370: 28 2a 69 64 2c 20 64 61 74 65 2c 20 26 70 6f 73 (*id, date, &pos
1380: 74 5f 75 72 6c 29 2e 61 77 61 69 74 3f 3b 0a 09 t_url).await?;..
1390: 09 09 09 09 7d 3b 0a 09 09 09 09 7d 3b 0a 09 09 ....};.....};...
13a0: 09 09 70 6f 73 74 65 64 20 2b 3d 20 31 3b 0a 09 ..posted += 1;..
13b0: 09 09 7d 3b 0a 09 09 09 70 6f 73 74 73 2e 63 6c ..};....posts.cl
13c0: 65 61 72 28 29 3b 0a 09 09 7d 3b 0a 09 09 4f 6b ear();...};...Ok
13d0: 28 66 6f 72 6d 61 74 21 28 22 50 6f 73 74 65 64 (format!("Posted
13e0: 3a 20 7b 70 6f 73 74 65 64 7d 22 29 29 0a 09 7d : {posted}"))..}
13f0: 0a 0a 09 61 73 79 6e 63 20 66 6e 20 61 75 74 6f ...async fn auto
1400: 66 65 74 63 68 28 26 73 65 6c 66 29 20 2d 3e 20 fetch(&self) ->
1410: 52 65 73 75 6c 74 3c 73 74 64 3a 3a 74 69 6d 65 Result<std::time
1420: 3a 3a 44 75 72 61 74 69 6f 6e 3e 20 7b 0a 09 09 ::Duration> {...
1430: 6c 65 74 20 6d 75 74 20 64 65 6c 61 79 20 3d 20 let mut delay =
1440: 63 68 72 6f 6e 6f 3a 3a 44 75 72 61 74 69 6f 6e chrono::Duration
1450: 3a 3a 6d 69 6e 75 74 65 73 28 31 29 3b 0a 09 09 ::minutes(1);...
1460: 6c 65 74 20 6e 6f 77 20 3d 20 63 68 72 6f 6e 6f let now = chrono
1470: 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f 77 28 29 3b 0a ::Local::now();.
1480: 09 09 6c 65 74 20 71 75 65 75 65 20 3d 20 7b 0a ..let queue = {.
1490: 09 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
14a0: 3d 20 73 65 6c 66 2e 64 62 2e 62 65 67 69 6e 28 = self.db.begin(
14b0: 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 63 6f 6e ).await?;....con
14c0: 6e 2e 67 65 74 5f 71 75 65 75 65 28 29 2e 61 77 n.get_queue().aw
14d0: 61 69 74 3f 0a 09 09 7d 3b 0a 09 09 66 6f 72 20 ait?...};...for
14e0: 72 6f 77 20 69 6e 20 71 75 65 75 65 20 7b 0a 09 row in queue {..
14f0: 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 6e 65 ..if let Some(ne
1500: 78 74 5f 66 65 74 63 68 29 20 3d 20 72 6f 77 2e xt_fetch) = row.
1510: 6e 65 78 74 5f 66 65 74 63 68 20 7b 0a 09 09 09 next_fetch {....
1520: 09 69 66 20 6e 65 78 74 5f 66 65 74 63 68 20 3c .if next_fetch <
1530: 20 6e 6f 77 20 7b 0a 09 09 09 09 09 69 66 20 6c now {......if l
1540: 65 74 20 28 53 6f 6d 65 28 6f 77 6e 65 72 29 2c et (Some(owner),
1550: 20 53 6f 6d 65 28 73 6f 75 72 63 65 5f 69 64 29 Some(source_id)
1560: 29 20 3d 20 28 72 6f 77 2e 6f 77 6e 65 72 2c 20 ) = (row.owner,
1570: 72 6f 77 2e 73 6f 75 72 63 65 5f 69 64 29 20 7b row.source_id) {
1580: 0a 09 09 09 09 09 09 6c 65 74 20 63 6c 6f 6e 65 .......let clone
1590: 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 09 09 09 = Core {.......
15a0: 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 43 68 61 .owner_chat: Cha
15b0: 74 50 65 65 72 49 64 3a 3a 66 72 6f 6d 28 6f 77 tPeerId::from(ow
15c0: 6e 65 72 29 2c 0a 09 09 09 09 09 09 09 2e 2e 73 ner),..........s
15d0: 65 6c 66 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 elf.clone().....
15e0: 09 09 7d 3b 0a 09 09 09 09 09 09 6c 65 74 20 73 ..};.......let s
15f0: 6f 75 72 63 65 20 3d 20 7b 0a 09 09 09 09 09 09 ource = {.......
1600: 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 .let mut conn =
1610: 73 65 6c 66 2e 64 62 2e 62 65 67 69 6e 28 29 2e self.db.begin().
1620: 61 77 61 69 74 3f 3b 0a 09 09 09 09 09 09 09 6d await?;........m
1630: 61 74 63 68 20 63 6f 6e 6e 2e 67 65 74 5f 6f 6e atch conn.get_on
1640: 65 28 6f 77 6e 65 72 2c 20 73 6f 75 72 63 65 5f e(owner, source_
1650: 69 64 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 id).await {.....
1660: 09 09 09 09 4f 6b 28 53 6f 6d 65 28 73 6f 75 72 ....Ok(Some(sour
1670: 63 65 29 29 20 3d 3e 20 73 6f 75 72 63 65 2e 74 ce)) => source.t
1680: 6f 5f 73 74 72 69 6e 67 28 29 2c 0a 09 09 09 09 o_string(),.....
1690: 09 09 09 09 4f 6b 28 4e 6f 6e 65 29 20 3d 3e 20 ....Ok(None) =>
16a0: 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e "Source not foun
16b0: 64 20 69 6e 20 64 61 74 61 62 61 73 65 3f 22 2e d in database?".
16c0: 74 6f 5f 73 74 72 69 6e 67 28 29 2c 0a 09 09 09 to_string(),....
16d0: 09 09 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e .....Err(err) =>
16e0: 20 66 6f 72 6d 61 74 21 28 22 46 61 69 6c 65 64 format!("Failed
16f0: 20 74 6f 20 66 65 74 63 68 20 73 6f 75 72 63 65 to fetch source
1700: 20 64 61 74 61 3a 5c 6e 7b 65 72 72 7d 22 29 2c data:\n{err}"),
1710: 0a 09 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 ........}.......
1720: 7d 3b 0a 09 09 09 09 09 09 74 61 73 6b 3a 3a 73 };.......task::s
1730: 70 61 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 pawn(async move
1740: 7b 0a 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 {........if let
1750: 45 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 Err(err) = clone
1760: 2e 63 68 65 63 6b 28 73 6f 75 72 63 65 5f 69 64 .check(source_id
1770: 2c 20 74 72 75 65 29 2e 61 77 61 69 74 20 7b 0a , true).await {.
1780: 09 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 ........if let E
1790: 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e rr(err) = clone.
17a0: 73 65 6e 64 28 26 66 6f 72 6d 61 74 21 28 22 7b send(&format!("{
17b0: 73 6f 75 72 63 65 7d 5c 6e 5c 6e f0 9f 9b 91 20 source}\n\nš
17c0: 7b 7d 22 2c 20 65 6e 63 6f 64 65 28 26 65 72 72 {}", encode(&err
17d0: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 29 2c 20 .to_string())),
17e0: 4e 6f 6e 65 2c 20 53 6f 6d 65 28 50 61 72 73 65 None, Some(Parse
17f0: 4d 6f 64 65 3a 3a 4d 61 72 6b 64 6f 77 6e 56 32 Mode::MarkdownV2
1800: 29 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 )).await {......
1810: 09 09 09 09 65 70 72 69 6e 74 6c 6e 21 28 22 43 ....eprintln!("C
1820: 68 65 63 6b 20 65 72 72 6f 72 3a 20 7b 65 72 72 heck error: {err
1830: 3a 3f 7d 22 29 3b 0a 09 09 09 09 09 09 09 09 09 :?}");..........
1840: 2f 2f 20 63 6c 6f 6e 65 2e 64 69 73 61 62 6c 65 // clone.disable
1850: 28 26 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e (&source_id, own
1860: 65 72 29 2e 61 77 61 69 74 2e 75 6e 77 72 61 70 er).await.unwrap
1870: 28 29 3b 0a 09 09 09 09 09 09 09 09 7d 3b 0a 09 ();.........};..
1880: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d ......};.......}
1890: 29 3b 0a 09 09 09 09 09 7d 0a 09 09 09 09 7d 20 );......}.....}
18a0: 65 6c 73 65 20 69 66 20 6e 65 78 74 5f 66 65 74 else if next_fet
18b0: 63 68 20 2d 20 6e 6f 77 20 3c 20 64 65 6c 61 79 ch - now < delay
18c0: 20 7b 0a 09 09 09 09 09 64 65 6c 61 79 20 3d 20 {......delay =
18d0: 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 next_fetch - now
18e0: 3b 0a 09 09 09 09 7d 0a 09 09 09 7d 0a 09 09 7d ;.....}....}...}
18f0: 3b 0a 09 09 4f 6b 28 64 65 6c 61 79 2e 74 6f 5f ;...Ok(delay.to_
1900: 73 74 64 28 29 3f 29 0a 09 7d 0a 0a 09 70 75 62 std()?)..}...pub
1910: 20 61 73 79 6e 63 20 66 6e 20 6c 69 73 74 20 28 async fn list (
1920: 26 73 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 55 73 &self, owner: Us
1930: 65 72 50 65 65 72 49 64 29 20 2d 3e 20 52 65 73 erPeerId) -> Res
1940: 75 6c 74 3c 53 74 72 69 6e 67 3e 20 7b 0a 09 09 ult<String> {...
1950: 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 3a 20 56 let mut reply: V
1960: 65 63 3c 53 74 72 69 6e 67 3e 20 3d 20 76 65 63 ec<String> = vec
1970: 21 5b 5d 3b 0a 09 09 72 65 70 6c 79 2e 70 75 73 ![];...reply.pus
1980: 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 2e 69 6e h("Channels:".in
1990: 74 6f 28 29 29 3b 0a 09 09 6c 65 74 20 6d 75 74 to());...let mut
19a0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 64 62 2e conn = self.db.
19b0: 62 65 67 69 6e 28 29 2e 61 77 61 69 74 3f 3b 0a begin().await?;.
19c0: 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 63 6f 6e ..for row in con
19d0: 6e 2e 67 65 74 5f 6c 69 73 74 28 6f 77 6e 65 72 n.get_list(owner
19e0: 29 2e 61 77 61 69 74 3f 20 7b 0a 09 09 09 72 65 ).await? {....re
19f0: 70 6c 79 2e 70 75 73 68 28 72 6f 77 2e 74 6f 5f ply.push(row.to_
1a00: 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 7d 3b 0a string());...};.
1a10: 09 09 4f 6b 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 ..Ok(reply.join(
1a20: 22 5c 6e 5c 6e 22 29 29 0a 09 7d 0a 7d 0a 0a 69 "\n\n"))..}.}..i
1a30: 6d 70 6c 20 55 70 64 61 74 65 48 61 6e 64 6c 65 mpl UpdateHandle
1a40: 72 20 66 6f 72 20 43 6f 72 65 20 7b 0a 09 61 73 r for Core {..as
1a50: 79 6e 63 20 66 6e 20 68 61 6e 64 6c 65 20 28 26 ync fn handle (&
1a60: 73 65 6c 66 2c 20 75 70 64 61 74 65 3a 20 55 70 self, update: Up
1a70: 64 61 74 65 29 20 7b 0a 09 09 69 66 20 6c 65 74 date) {...if let
1a80: 20 55 70 64 61 74 65 54 79 70 65 3a 3a 4d 65 73 UpdateType::Mes
1a90: 73 61 67 65 28 6d 73 67 29 20 3d 20 75 70 64 61 sage(msg) = upda
1aa0: 74 65 2e 75 70 64 61 74 65 5f 74 79 70 65 20 7b te.update_type {
1ab0: 0a 09 09 09 69 66 20 6c 65 74 20 4f 6b 28 63 6d ....if let Ok(cm
1ac0: 64 29 20 3d 20 43 6f 6d 6d 61 6e 64 3a 3a 74 72 d) = Command::tr
1ad0: 79 5f 66 72 6f 6d 28 6d 73 67 29 20 7b 0a 09 09 y_from(msg) {...
1ae0: 09 09 6c 65 74 20 6d 73 67 20 3d 20 63 6d 64 2e ..let msg = cmd.
1af0: 67 65 74 5f 6d 65 73 73 61 67 65 28 29 3b 0a 09 get_message();..
1b00: 09 09 09 6c 65 74 20 77 6f 72 64 73 20 3d 20 63 ...let words = c
1b10: 6d 64 2e 67 65 74 5f 61 72 67 73 28 29 3b 0a 09 md.get_args();..
1b20: 09 09 09 6c 65 74 20 63 6f 6d 6d 61 6e 64 20 3d ...let command =
1b30: 20 63 6d 64 2e 67 65 74 5f 6e 61 6d 65 28 29 3b cmd.get_name();
1b40: 0a 09 09 09 09 6c 65 74 20 72 65 73 20 3d 20 6d .....let res = m
1b50: 61 74 63 68 20 63 6f 6d 6d 61 6e 64 20 7b 0a 09 atch command {..
1b60: 09 09 09 09 22 2f 63 68 65 63 6b 22 20 7c 20 22 ...."/check" | "
1b70: 2f 63 6c 65 61 6e 22 20 7c 20 22 2f 65 6e 61 62 /clean" | "/enab
1b80: 6c 65 22 20 7c 20 22 2f 64 65 6c 65 74 65 22 20 le" | "/delete"
1b90: 7c 20 22 2f 64 69 73 61 62 6c 65 22 20 3d 3e 20 | "/disable" =>
1ba0: 63 6f 6d 6d 61 6e 64 3a 3a 63 6f 6d 6d 61 6e 64 command::command
1bb0: 28 73 65 6c 66 2c 20 63 6f 6d 6d 61 6e 64 2c 20 (self, command,
1bc0: 6d 73 67 2c 20 77 6f 72 64 73 29 2e 61 77 61 69 msg, words).awai
1bd0: 74 2c 0a 09 09 09 09 09 22 2f 73 74 61 72 74 22 t,......"/start"
1be0: 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 73 74 61 => command::sta
1bf0: 72 74 28 73 65 6c 66 2c 20 6d 73 67 29 2e 61 77 rt(self, msg).aw
1c00: 61 69 74 2c 0a 09 09 09 09 09 22 2f 6c 69 73 74 ait,......"/list
1c10: 22 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 6c 69 " => command::li
1c20: 73 74 28 73 65 6c 66 2c 20 6d 73 67 29 2e 61 77 st(self, msg).aw
1c30: 61 69 74 2c 0a 09 09 09 09 09 22 2f 61 64 64 22 ait,......"/add"
1c40: 20 7c 20 22 2f 75 70 64 61 74 65 22 20 3d 3e 20 | "/update" =>
1c50: 63 6f 6d 6d 61 6e 64 3a 3a 75 70 64 61 74 65 28 command::update(
1c60: 73 65 6c 66 2c 20 63 6f 6d 6d 61 6e 64 2c 20 6d self, command, m
1c70: 73 67 2c 20 77 6f 72 64 73 29 2e 61 77 61 69 74 sg, words).await
1c80: 2c 0a 09 09 09 09 09 61 6e 79 20 3d 3e 20 45 72 ,......any => Er
1c90: 72 28 61 6e 79 68 6f 77 21 28 22 55 6e 6b 6e 6f r(anyhow!("Unkno
1ca0: 77 6e 20 63 6f 6d 6d 61 6e 64 3a 20 7b 61 6e 79 wn command: {any
1cb0: 7d 22 29 29 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 }")),.....};....
1cc0: 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 .if let Err(err)
1cd0: 20 3d 20 72 65 73 20 7b 0a 09 09 09 09 09 69 66 = res {......if
1ce0: 20 6c 65 74 20 45 72 72 28 65 72 72 32 29 20 3d let Err(err2) =
1cf0: 20 73 65 6c 66 2e 73 65 6e 64 28 66 6f 72 6d 61 self.send(forma
1d00: 74 21 28 22 5c 5c 23 65 72 72 6f 72 5c 6e 60 60 t!("\\#error\n``
1d10: 60 5c 6e 7b 65 72 72 3a 3f 7d 5c 6e 60 60 60 22 `\n{err:?}\n```"
1d20: 29 2c 0a 09 09 09 09 09 09 53 6f 6d 65 28 6d 73 ),.......Some(ms
1d30: 67 2e 63 68 61 74 2e 67 65 74 5f 69 64 28 29 29 g.chat.get_id())
1d40: 2c 0a 09 09 09 09 09 09 53 6f 6d 65 28 50 61 72 ,.......Some(Par
1d50: 73 65 4d 6f 64 65 3a 3a 4d 61 72 6b 64 6f 77 6e seMode::Markdown
1d60: 56 32 29 0a 09 09 09 09 09 29 2e 61 77 61 69 74 V2)......).await
1d70: 7b 0a 09 09 09 09 09 09 64 62 67 21 28 65 72 72 {.......dbg!(err
1d80: 32 29 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 2);......};.....
1d90: 7d 0a 09 09 09 7d 3b 0a 09 09 7d 3b 0a 09 7d 0a }....};...};..}.
1da0: 7d 0a }.