0000: 75 73 65 20 63 72 61 74 65 3a 3a 7b 0a 09 63 6f use crate::{..co
0010: 6d 6d 61 6e 64 2c 0a 09 73 71 6c 3a 3a 44 62 2c mmand,..sql::Db,
0020: 0a 7d 3b 0a 0a 75 73 65 20 73 74 64 3a 3a 7b 0a .};..use std::{.
0030: 09 62 6f 72 72 6f 77 3a 3a 43 6f 77 2c 0a 09 63 .borrow::Cow,..c
0040: 6f 6c 6c 65 63 74 69 6f 6e 73 3a 3a 7b 0a 09 09 ollections::{...
0050: 42 54 72 65 65 4d 61 70 2c 0a 09 09 48 61 73 68 BTreeMap,...Hash
0060: 53 65 74 2c 0a 09 7d 2c 0a 09 6e 75 6d 3a 3a 54 Set,..},..num::T
0070: 72 79 46 72 6f 6d 49 6e 74 45 72 72 6f 72 2c 0a ryFromIntError,.
0080: 09 73 79 6e 63 3a 3a 7b 0a 09 09 41 72 63 2c 0a .sync::{...Arc,.
0090: 09 09 4d 75 74 65 78 0a 09 7d 2c 0a 7d 3b 0a 0a ..Mutex..},.};..
00a0: 75 73 65 20 61 6e 79 68 6f 77 3a 3a 7b 0a 09 61 use anyhow::{..a
00b0: 6e 79 68 6f 77 2c 0a 09 62 61 69 6c 2c 0a 09 52 nyhow,..bail,..R
00c0: 65 73 75 6c 74 2c 0a 7d 3b 0a 75 73 65 20 61 73 esult,.};.use as
00d0: 79 6e 63 5f 73 74 64 3a 3a 74 61 73 6b 3b 0a 75 ync_std::task;.u
00e0: 73 65 20 63 68 72 6f 6e 6f 3a 3a 44 61 74 65 54 se chrono::DateT
00f0: 69 6d 65 3b 0a 75 73 65 20 74 67 62 6f 74 3a 3a ime;.use tgbot::
0100: 7b 0a 09 61 70 69 3a 3a 43 6c 69 65 6e 74 2c 0a {..api::Client,.
0110: 09 68 61 6e 64 6c 65 72 3a 3a 55 70 64 61 74 65 .handler::Update
0120: 48 61 6e 64 6c 65 72 2c 0a 09 74 79 70 65 73 3a Handler,..types:
0130: 3a 7b 0a 09 09 42 6f 74 2c 0a 09 09 43 68 61 74 :{...Bot,...Chat
0140: 50 65 65 72 49 64 2c 0a 09 09 43 6f 6d 6d 61 6e PeerId,...Comman
0150: 64 2c 0a 09 09 47 65 74 42 6f 74 2c 0a 09 09 4d d,...GetBot,...M
0160: 65 73 73 61 67 65 2c 0a 09 09 50 61 72 73 65 4d essage,...ParseM
0170: 6f 64 65 2c 0a 09 09 53 65 6e 64 4d 65 73 73 61 ode,...SendMessa
0180: 67 65 2c 0a 09 09 55 70 64 61 74 65 2c 0a 09 09 ge,...Update,...
0190: 55 70 64 61 74 65 54 79 70 65 2c 0a 09 09 55 73 UpdateType,...Us
01a0: 65 72 50 65 65 72 49 64 2c 0a 09 7d 2c 0a 7d 3b erPeerId,..},.};
01b0: 0a 75 73 65 20 74 68 69 73 65 72 72 6f 72 3a 3a .use thiserror::
01c0: 45 72 72 6f 72 3b 0a 0a 23 5b 64 65 72 69 76 65 Error;..#[derive
01d0: 28 45 72 72 6f 72 2c 20 44 65 62 75 67 29 5d 0a (Error, Debug)].
01e0: 70 75 62 20 65 6e 75 6d 20 52 73 73 45 72 72 6f pub enum RssErro
01f0: 72 20 7b 0a 09 2f 2f 20 23 5b 65 72 72 6f 72 28 r {..// #[error(
0200: 74 72 61 6e 73 70 61 72 65 6e 74 29 5d 0a 09 2f transparent)]../
0210: 2f 20 54 67 28 23 5b 66 72 6f 6d 5d 20 54 67 45 / Tg(#[from] TgE
0220: 72 72 6f 72 29 2c 0a 09 23 5b 65 72 72 6f 72 28 rror),..#[error(
0230: 74 72 61 6e 73 70 61 72 65 6e 74 29 5d 0a 09 49 transparent)]..I
0240: 6e 74 28 23 5b 66 72 6f 6d 5d 20 54 72 79 46 72 nt(#[from] TryFr
0250: 6f 6d 49 6e 74 45 72 72 6f 72 29 2c 0a 7d 0a 0a omIntError),.}..
0260: 23 5b 64 65 72 69 76 65 28 43 6c 6f 6e 65 29 5d #[derive(Clone)]
0270: 0a 70 75 62 20 73 74 72 75 63 74 20 43 6f 72 65 .pub struct Core
0280: 20 7b 0a 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 {..owner_chat:
0290: 43 68 61 74 50 65 65 72 49 64 2c 0a 09 2f 2f 20 ChatPeerId,..//
02a0: 6d 61 78 5f 64 65 6c 61 79 3a 20 75 31 36 2c 0a max_delay: u16,.
02b0: 09 70 75 62 20 74 67 3a 20 43 6c 69 65 6e 74 2c .pub tg: Client,
02c0: 0a 09 70 75 62 20 6d 65 3a 20 42 6f 74 2c 0a 09 ..pub me: Bot,..
02d0: 70 75 62 20 64 62 3a 20 44 62 2c 0a 09 73 6f 75 pub db: Db,..sou
02e0: 72 63 65 73 3a 20 41 72 63 3c 4d 75 74 65 78 3c rces: Arc<Mutex<
02f0: 48 61 73 68 53 65 74 3c 41 72 63 3c 69 33 32 3e HashSet<Arc<i32>
0300: 3e 3e 3e 2c 0a 09 68 74 74 70 5f 63 6c 69 65 6e >>>,..http_clien
0310: 74 3a 20 72 65 71 77 65 73 74 3a 3a 43 6c 69 65 t: reqwest::Clie
0320: 6e 74 2c 0a 7d 0a 0a 69 6d 70 6c 20 43 6f 72 65 nt,.}..impl Core
0330: 20 7b 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e {..pub async fn
0340: 20 6e 65 77 28 73 65 74 74 69 6e 67 73 3a 20 63 new(settings: c
0350: 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 29 20 2d onfig::Config) -
0360: 3e 20 52 65 73 75 6c 74 3c 43 6f 72 65 3e 20 7b > Result<Core> {
0370: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 5f 63 68 61 ...let owner_cha
0380: 74 20 3d 20 43 68 61 74 50 65 65 72 49 64 3a 3a t = ChatPeerId::
0390: 66 72 6f 6d 28 73 65 74 74 69 6e 67 73 2e 67 65 from(settings.ge
03a0: 74 5f 69 6e 74 28 22 6f 77 6e 65 72 22 29 3f 29 t_int("owner")?)
03b0: 3b 0a 09 09 6c 65 74 20 61 70 69 5f 6b 65 79 20 ;...let api_key
03c0: 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 5f 73 = settings.get_s
03d0: 74 72 69 6e 67 28 22 61 70 69 5f 6b 65 79 22 29 tring("api_key")
03e0: 3f 3b 0a 09 09 6c 65 74 20 74 67 20 3d 20 43 6c ?;...let tg = Cl
03f0: 69 65 6e 74 3a 3a 6e 65 77 28 26 61 70 69 5f 6b ient::new(&api_k
0400: 65 79 29 3f 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 ey)?;....let mut
0410: 20 63 6c 69 65 6e 74 20 3d 20 72 65 71 77 65 73 client = reqwes
0420: 74 3a 3a 43 6c 69 65 6e 74 3a 3a 62 75 69 6c 64 t::Client::build
0430: 65 72 28 29 3b 0a 09 09 69 66 20 6c 65 74 20 4f er();...if let O
0440: 6b 28 70 72 6f 78 79 29 20 3d 20 73 65 74 74 69 k(proxy) = setti
0450: 6e 67 73 2e 67 65 74 5f 73 74 72 69 6e 67 28 22 ngs.get_string("
0460: 70 72 6f 78 79 22 29 20 7b 0a 09 09 09 6c 65 74 proxy") {....let
0470: 20 70 72 6f 78 79 20 3d 20 72 65 71 77 65 73 74 proxy = reqwest
0480: 3a 3a 50 72 6f 78 79 3a 3a 61 6c 6c 28 70 72 6f ::Proxy::all(pro
0490: 78 79 29 3f 3b 0a 09 09 09 63 6c 69 65 6e 74 20 xy)?;....client
04a0: 3d 20 63 6c 69 65 6e 74 2e 70 72 6f 78 79 28 70 = client.proxy(p
04b0: 72 6f 78 79 29 3b 0a 09 09 7d 0a 09 09 6c 65 74 roxy);...}...let
04c0: 20 68 74 74 70 5f 63 6c 69 65 6e 74 20 3d 20 63 http_client = c
04d0: 6c 69 65 6e 74 2e 62 75 69 6c 64 28 29 3f 3b 0a lient.build()?;.
04e0: 09 09 6c 65 74 20 6d 65 20 3d 20 74 67 2e 65 78 ..let me = tg.ex
04f0: 65 63 75 74 65 28 47 65 74 42 6f 74 29 2e 61 77 ecute(GetBot).aw
0500: 61 69 74 3f 3b 0a 09 09 6c 65 74 20 63 6f 72 65 ait?;...let core
0510: 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 74 67 2c = Core {....tg,
0520: 0a 09 09 09 6d 65 2c 0a 09 09 09 6f 77 6e 65 72 ....me,....owner
0530: 5f 63 68 61 74 2c 0a 09 09 09 64 62 3a 20 44 62 _chat,....db: Db
0540: 3a 3a 6e 65 77 28 26 73 65 74 74 69 6e 67 73 2e ::new(&settings.
0550: 67 65 74 5f 73 74 72 69 6e 67 28 22 70 67 22 29 get_string("pg")
0560: 3f 29 3f 2c 0a 09 09 09 73 6f 75 72 63 65 73 3a ?)?,....sources:
0570: 20 41 72 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a Arc::new(Mutex:
0580: 3a 6e 65 77 28 48 61 73 68 53 65 74 3a 3a 6e 65 :new(HashSet::ne
0590: 77 28 29 29 29 2c 0a 09 09 09 68 74 74 70 5f 63 w())),....http_c
05a0: 6c 69 65 6e 74 2c 0a 09 09 09 2f 2f 20 6d 61 78 lient,....// max
05b0: 5f 64 65 6c 61 79 3a 20 36 30 2c 0a 09 09 7d 3b _delay: 60,...};
05c0: 0a 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 63 ...let clone = c
05d0: 6f 72 65 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 74 ore.clone();...t
05e0: 61 73 6b 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 ask::spawn(async
05f0: 20 6d 6f 76 65 20 7b 0a 09 09 09 6c 6f 6f 70 20 move {....loop
0600: 7b 0a 09 09 09 09 6c 65 74 20 64 65 6c 61 79 20 {.....let delay
0610: 3d 20 6d 61 74 63 68 20 26 63 6c 6f 6e 65 2e 61 = match &clone.a
0620: 75 74 6f 66 65 74 63 68 28 29 2e 61 77 61 69 74 utofetch().await
0630: 20 7b 0a 09 09 09 09 09 45 72 72 28 65 72 72 29 {......Err(err)
0640: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 69 66 20 6c => {.......if l
0650: 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c et Err(err) = cl
0660: 6f 6e 65 2e 73 65 6e 64 28 66 6f 72 6d 61 74 21 one.send(format!
0670: 28 22 f0 9f 9b 91 20 7b 65 72 72 3a 3f 7d 22 29 ("š {err:?}")
0680: 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 , None, None).aw
0690: 61 69 74 20 7b 0a 09 09 09 09 09 09 09 65 70 72 ait {........epr
06a0: 69 6e 74 6c 6e 21 28 22 41 75 74 6f 66 65 74 63 intln!("Autofetc
06b0: 68 20 65 72 72 6f 72 3a 20 7b 65 72 72 3a 3f 7d h error: {err:?}
06c0: 22 29 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 ");.......};....
06d0: 09 09 09 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 ...std::time::Du
06e0: 72 61 74 69 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 63 ration::from_sec
06f0: 73 28 36 30 29 0a 09 09 09 09 09 7d 2c 0a 09 09 s(60)......},...
0700: 09 09 09 4f 6b 28 74 69 6d 65 29 20 3d 3e 20 2a ...Ok(time) => *
0710: 74 69 6d 65 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 time,.....};....
0720: 09 74 61 73 6b 3a 3a 73 6c 65 65 70 28 64 65 6c .task::sleep(del
0730: 61 79 29 2e 61 77 61 69 74 3b 0a 09 09 09 7d 0a ay).await;....}.
0740: 09 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 ..});...Ok(core)
0750: 0a 09 7d 0a 0a 09 2f 2a 0a 09 70 75 62 20 61 73 ..}.../*..pub as
0760: 79 6e 63 20 66 6e 20 73 74 72 65 61 6d 28 26 73 ync fn stream(&s
0770: 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 elf) -> Result<(
0780: 29 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 6f )> {...let mut o
0790: 66 66 73 65 74 3a 20 69 36 34 20 3d 20 30 3b 0a ffset: i64 = 0;.
07a0: 09 09 6c 65 74 20 6d 75 74 20 70 61 72 61 6d 73 ..let mut params
07b0: 20 3d 20 47 65 74 55 70 64 61 74 65 73 50 61 72 = GetUpdatesPar
07c0: 61 6d 73 20 7b 0a 09 09 09 6f 66 66 73 65 74 3a ams {....offset:
07d0: 20 4e 6f 6e 65 2c 0a 09 09 09 6c 69 6d 69 74 3a None,....limit:
07e0: 20 53 6f 6d 65 28 31 30 30 29 2c 0a 09 09 09 74 Some(100),....t
07f0: 69 6d 65 6f 75 74 3a 20 53 6f 6d 65 28 33 30 30 imeout: Some(300
0800: 29 2c 0a 09 09 09 61 6c 6c 6f 77 65 64 5f 75 70 ),....allowed_up
0810: 64 61 74 65 73 3a 20 53 6f 6d 65 28 76 65 63 21 dates: Some(vec!
0820: 5b 41 6c 6c 6f 77 65 64 55 70 64 61 74 65 3a 3a [AllowedUpdate::
0830: 4d 65 73 73 61 67 65 5d 29 2c 0a 09 09 7d 3b 0a Message]),...};.
0840: 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 6c 65 74 20 ..loop {....let
0850: 75 70 64 61 74 65 73 20 3d 20 73 65 6c 66 2e 74 updates = self.t
0860: 67 2e 67 65 74 5f 75 70 64 61 74 65 73 28 26 70 g.get_updates(&p
0870: 61 72 61 6d 73 29 2e 61 77 61 69 74 3f 2e 72 65 arams).await?.re
0880: 73 75 6c 74 3b 0a 09 09 09 69 66 20 75 70 64 61 sult;....if upda
0890: 74 65 73 2e 69 73 5f 65 6d 70 74 79 28 29 20 7b tes.is_empty() {
08a0: 0a 09 09 09 09 6f 66 66 73 65 74 20 3d 20 30 3b .....offset = 0;
08b0: 0a 09 09 09 09 70 61 72 61 6d 73 2e 6f 66 66 73 .....params.offs
08c0: 65 74 20 3d 20 4e 6f 6e 65 3b 0a 09 09 09 09 63 et = None;.....c
08d0: 6f 6e 74 69 6e 75 65 3b 0a 09 09 09 7d 0a 09 09 ontinue;....}...
08e0: 09 66 6f 72 20 75 70 64 61 74 65 20 69 6e 20 75 .for update in u
08f0: 70 64 61 74 65 73 20 7b 0a 09 09 09 09 69 66 20 pdates {.....if
0900: 69 36 34 3a 3a 66 72 6f 6d 28 75 70 64 61 74 65 i64::from(update
0910: 2e 75 70 64 61 74 65 5f 69 64 29 20 3e 3d 20 6f .update_id) >= o
0920: 66 66 73 65 74 20 7b 0a 09 09 09 09 09 6f 66 66 ffset {......off
0930: 73 65 74 20 3d 20 69 36 34 3a 3a 66 72 6f 6d 28 set = i64::from(
0940: 75 70 64 61 74 65 2e 75 70 64 61 74 65 5f 69 64 update.update_id
0950: 29 20 2b 20 31 3b 0a 09 09 09 09 09 70 61 72 61 ) + 1;......para
0960: 6d 73 2e 6f 66 66 73 65 74 20 3d 20 53 6f 6d 65 ms.offset = Some
0970: 28 6f 66 66 73 65 74 29 3b 0a 09 09 09 09 7d 0a (offset);.....}.
0980: 09 09 09 09 69 66 20 6c 65 74 20 55 70 64 61 74 ....if let Updat
0990: 65 43 6f 6e 74 65 6e 74 3a 3a 4d 65 73 73 61 67 eContent::Messag
09a0: 65 28 6d 73 67 29 20 3d 20 75 70 64 61 74 65 2e e(msg) = update.
09b0: 63 6f 6e 74 65 6e 74 20 7b 0a 09 09 09 09 09 69 content {......i
09c0: 66 20 6c 65 74 20 53 6f 6d 65 28 74 65 78 74 29 f let Some(text)
09d0: 20 3d 20 6d 73 67 2e 74 65 78 74 20 7b 0a 09 09 = msg.text {...
09e0: 09 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 ....if let Some(
09f0: 65 6e 74 69 74 69 65 73 29 20 3d 20 6d 73 67 2e entities) = msg.
0a00: 65 6e 74 69 74 69 65 73 20 7b 0a 09 09 09 09 09 entities {......
0a10: 09 09 6c 65 74 20 63 68 61 72 73 3a 20 56 65 63 ..let chars: Vec
0a20: 3c 75 31 36 3e 20 3d 20 74 65 78 74 2e 65 6e 63 <u16> = text.enc
0a30: 6f 64 65 5f 75 74 66 31 36 28 29 2e 63 6f 6c 6c ode_utf16().coll
0a40: 65 63 74 28 29 3b 0a 09 09 09 09 09 09 09 66 6f ect();........fo
0a50: 72 20 65 6e 74 69 74 79 20 69 6e 20 65 6e 74 69 r entity in enti
0a60: 74 69 65 73 20 7b 0a 09 09 09 09 09 09 09 09 69 ties {.........i
0a70: 66 20 65 6e 74 69 74 79 2e 74 79 70 65 5f 66 69 f entity.type_fi
0a80: 65 6c 64 20 3d 3d 20 4d 65 73 73 61 67 65 45 6e eld == MessageEn
0a90: 74 69 74 79 54 79 70 65 3a 3a 42 6f 74 43 6f 6d tityType::BotCom
0aa0: 6d 61 6e 64 20 26 26 20 65 6e 74 69 74 79 2e 6f mand && entity.o
0ab0: 66 66 73 65 74 20 21 3d 20 30 20 7b 0a 09 09 09 ffset != 0 {....
0ac0: 09 09 09 09 09 09 62 61 69 6c 21 28 22 63 6f 6d ......bail!("com
0ad0: 6d 61 6e 64 73 20 73 68 6f 75 6c 64 20 62 65 20 mands should be
0ae0: 61 74 20 6d 65 73 73 61 67 65 20 73 74 61 72 74 at message start
0af0: 22 29 3b 0a 09 09 09 09 09 09 09 09 7d 3b 0a 09 ");.........};..
0b00: 09 09 09 09 09 09 09 6c 65 74 20 63 6d 64 20 3d .......let cmd =
0b10: 20 53 74 72 69 6e 67 3a 3a 66 72 6f 6d 5f 75 74 String::from_ut
0b20: 66 31 36 5f 6c 6f 73 73 79 28 26 63 68 61 72 73 f16_lossy(&chars
0b30: 5b 65 6e 74 69 74 79 2e 6f 66 66 73 65 74 20 61 [entity.offset a
0b40: 73 20 75 73 69 7a 65 2e 2e 65 6e 74 69 74 79 2e s usize..entity.
0b50: 6c 65 6e 67 74 68 20 61 73 20 75 73 69 7a 65 5d length as usize]
0b60: 29 3b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 77 );.........let w
0b70: 6f 72 64 73 3a 20 56 65 63 3c 26 73 74 72 3e 20 ords: Vec<&str>
0b80: 3d 20 74 65 78 74 2e 73 70 6c 69 74 5f 77 68 69 = text.split_whi
0b90: 74 65 73 70 61 63 65 28 29 2e 63 6f 6c 6c 65 63 tespace().collec
0ba0: 74 28 29 3b 0a 09 09 09 09 09 09 09 09 6c 65 74 t();.........let
0bb0: 20 72 65 73 20 3d 20 6d 61 74 63 68 20 63 6d 64 res = match cmd
0bc0: 2e 61 73 5f 72 65 66 28 29 20 7b 0a 09 09 09 09 .as_ref() {.....
0bd0: 09 09 09 09 09 22 2f 63 68 65 63 6b 22 20 7c 20 ....."/check" |
0be0: 22 2f 63 6c 65 61 6e 22 20 7c 20 22 2f 65 6e 61 "/clean" | "/ena
0bf0: 62 6c 65 22 20 7c 20 22 2f 64 65 6c 65 74 65 22 ble" | "/delete"
0c00: 20 7c 20 22 2f 64 69 73 61 62 6c 65 22 20 3d 3e | "/disable" =>
0c10: 20 63 6f 6d 6d 61 6e 64 3a 3a 63 6f 6d 6d 61 6e command::comman
0c20: 64 28 73 65 6c 66 2c 20 6d 73 67 2e 63 68 61 74 d(self, msg.chat
0c30: 2e 69 64 2c 20 77 6f 72 64 73 29 2e 61 77 61 69 .id, words).awai
0c40: 74 2c 0a 09 09 09 09 09 09 09 09 09 22 2f 73 74 t,.........."/st
0c50: 61 72 74 22 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a art" => command:
0c60: 3a 73 74 61 72 74 28 73 65 6c 66 2c 20 6d 73 67 :start(self, msg
0c70: 2e 63 68 61 74 2e 69 64 29 2e 61 77 61 69 74 2c .chat.id).await,
0c80: 0a 09 09 09 09 09 09 09 09 09 22 2f 6c 69 73 74 .........."/list
0c90: 22 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 6c 69 " => command::li
0ca0: 73 74 28 73 65 6c 66 2c 20 6d 73 67 2e 63 68 61 st(self, msg.cha
0cb0: 74 2e 69 64 29 2e 61 77 61 69 74 2c 0a 09 09 09 t.id).await,....
0cc0: 09 09 09 09 09 09 22 2f 61 64 64 22 20 7c 20 22 ......"/add" | "
0cd0: 2f 75 70 64 61 74 65 22 20 3d 3e 20 63 6f 6d 6d /update" => comm
0ce0: 61 6e 64 3a 3a 75 70 64 61 74 65 28 73 65 6c 66 and::update(self
0cf0: 2c 20 6d 73 67 2e 63 68 61 74 2e 69 64 2c 20 77 , msg.chat.id, w
0d00: 6f 72 64 73 29 2e 61 77 61 69 74 2c 0a 09 09 09 ords).await,....
0d10: 09 09 09 09 09 09 61 6e 79 20 3d 3e 20 45 72 72 ......any => Err
0d20: 28 61 6e 79 68 6f 77 21 28 22 55 6e 6b 6e 6f 77 (anyhow!("Unknow
0d30: 6e 20 63 6f 6d 6d 61 6e 64 3a 20 7b 61 6e 79 7d n command: {any}
0d40: 22 29 29 2c 0a 09 09 09 09 09 09 09 09 7d 3b 0a ")),.........};.
0d50: 09 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 ........if let E
0d60: 72 72 28 65 72 72 29 20 3d 20 72 65 73 20 7b 0a rr(err) = res {.
0d70: 09 09 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 .........if let
0d80: 45 72 72 28 65 72 72 32 29 20 3d 20 73 65 6c 66 Err(err2) = self
0d90: 2e 73 65 6e 64 28 66 6f 72 6d 61 74 21 28 22 5c .send(format!("\
0da0: 5c 23 65 72 72 6f 72 5c 6e 60 60 60 5c 6e 7b 65 \#error\n```\n{e
0db0: 72 72 3a 3f 7d 5c 6e 60 60 60 22 29 2c 0a 09 09 rr:?}\n```"),...
0dc0: 09 09 09 09 09 09 09 09 53 6f 6d 65 28 6d 73 67 ........Some(msg
0dd0: 2e 63 68 61 74 2e 69 64 29 2c 0a 09 09 09 09 09 .chat.id),......
0de0: 09 09 09 09 09 53 6f 6d 65 28 50 61 72 73 65 4d .....Some(ParseM
0df0: 6f 64 65 3a 3a 4d 61 72 6b 64 6f 77 6e 56 32 29 ode::MarkdownV2)
0e00: 0a 09 09 09 09 09 09 09 09 09 29 2e 61 77 61 69 ..........).awai
0e10: 74 7b 0a 09 09 09 09 09 09 09 09 09 09 64 62 67 t{...........dbg
0e20: 21 28 65 72 72 32 29 3b 0a 09 09 09 09 09 09 09 !(err2);........
0e30: 09 09 7d 3b 0a 09 09 09 09 09 09 09 09 7d 0a 09 ..};.........}..
0e40: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d ......};.......}
0e50: 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 7d 3b ;......};.....};
0e60: 0a 09 09 09 7d 0a 09 09 7d 0a 09 7d 0a 09 2a 2f ....}...}..}..*/
0e70: 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 ...pub async fn
0e80: 73 65 6e 64 20 3c 53 3e 28 26 73 65 6c 66 2c 20 send <S>(&self,
0e90: 6d 73 67 3a 20 53 2c 20 74 61 72 67 65 74 3a 20 msg: S, target:
0ea0: 4f 70 74 69 6f 6e 3c 43 68 61 74 50 65 65 72 49 Option<ChatPeerI
0eb0: 64 3e 2c 20 6d 6f 64 65 3a 20 4f 70 74 69 6f 6e d>, mode: Option
0ec0: 3c 50 61 72 73 65 4d 6f 64 65 3e 29 20 2d 3e 20 <ParseMode>) ->
0ed0: 52 65 73 75 6c 74 3c 4d 65 73 73 61 67 65 3e 0a Result<Message>.
0ee0: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 53 .where S: Into<S
0ef0: 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 74 20 6d tring> {...let m
0f00: 73 67 20 3d 20 6d 73 67 2e 69 6e 74 6f 28 29 3b sg = msg.into();
0f10: 0a 0a 09 09 6c 65 74 20 6d 6f 64 65 20 3d 20 6d ....let mode = m
0f20: 6f 64 65 2e 75 6e 77 72 61 70 5f 6f 72 28 50 61 ode.unwrap_or(Pa
0f30: 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 3b 0a rseMode::Html);.
0f40: 09 09 6c 65 74 20 74 61 72 67 65 74 20 3d 20 74 ..let target = t
0f50: 61 72 67 65 74 2e 75 6e 77 72 61 70 5f 6f 72 28 arget.unwrap_or(
0f60: 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 61 74 29 self.owner_chat)
0f70: 3b 0a 09 09 4f 6b 28 73 65 6c 66 2e 74 67 2e 65 ;...Ok(self.tg.e
0f80: 78 65 63 75 74 65 28 0a 09 09 09 53 65 6e 64 4d xecute(....SendM
0f90: 65 73 73 61 67 65 3a 3a 6e 65 77 28 74 61 72 67 essage::new(targ
0fa0: 65 74 2c 20 6d 73 67 29 0a 09 09 09 09 2e 77 69 et, msg)......wi
0fb0: 74 68 5f 70 61 72 73 65 5f 6d 6f 64 65 28 6d 6f th_parse_mode(mo
0fc0: 64 65 29 0a 09 09 29 2e 61 77 61 69 74 3f 29 0a de)...).await?).
0fd0: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
0fe0: 6e 20 63 68 65 63 6b 20 28 26 73 65 6c 66 2c 20 n check (&self,
0ff0: 69 64 3a 20 69 33 32 2c 20 72 65 61 6c 3a 20 62 id: i32, real: b
1000: 6f 6f 6c 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 ool) -> Result<S
1010: 74 72 69 6e 67 3e 20 7b 0a 09 09 6c 65 74 20 6d tring> {...let m
1020: 75 74 20 70 6f 73 74 65 64 3a 20 69 33 32 20 3d ut posted: i32 =
1030: 20 30 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 0;...let mut co
1040: 6e 6e 20 3d 20 73 65 6c 66 2e 64 62 2e 62 65 67 nn = self.db.beg
1050: 69 6e 28 29 2e 61 77 61 69 74 3f 3b 0a 0a 09 09 in().await?;....
1060: 6c 65 74 20 69 64 20 3d 20 7b 0a 09 09 09 6c 65 let id = {....le
1070: 74 20 6d 75 74 20 73 65 74 20 3d 20 73 65 6c 66 t mut set = self
1080: 2e 73 6f 75 72 63 65 73 2e 6c 6f 63 6b 28 29 2e .sources.lock().
1090: 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 6d 61 74 unwrap();....mat
10a0: 63 68 20 73 65 74 2e 67 65 74 28 26 69 64 29 20 ch set.get(&id)
10b0: 7b 0a 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d {.....Some(id) =
10c0: 3e 20 69 64 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 > id.clone(),...
10d0: 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 ..None => {.....
10e0: 09 6c 65 74 20 69 64 20 3d 20 41 72 63 3a 3a 6e .let id = Arc::n
10f0: 65 77 28 69 64 29 3b 0a 09 09 09 09 09 73 65 74 ew(id);......set
1100: 2e 69 6e 73 65 72 74 28 69 64 2e 63 6c 6f 6e 65 .insert(id.clone
1110: 28 29 29 3b 0a 09 09 09 09 09 69 64 2e 63 6c 6f ());......id.clo
1120: 6e 65 28 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d ne().....},....}
1130: 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 63 6f 75 6e ...};...let coun
1140: 74 20 3d 20 41 72 63 3a 3a 73 74 72 6f 6e 67 5f t = Arc::strong_
1150: 63 6f 75 6e 74 28 26 69 64 29 3b 0a 09 09 69 66 count(&id);...if
1160: 20 63 6f 75 6e 74 20 3d 3d 20 32 20 7b 0a 09 09 count == 2 {...
1170: 09 6c 65 74 20 73 6f 75 72 63 65 20 3d 20 63 6f .let source = co
1180: 6e 6e 2e 67 65 74 5f 73 6f 75 72 63 65 28 2a 69 nn.get_source(*i
1190: 64 2c 20 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 d, self.owner_ch
11a0: 61 74 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 63 at).await?;....c
11b0: 6f 6e 6e 2e 73 65 74 5f 73 63 72 61 70 65 28 2a onn.set_scrape(*
11c0: 69 64 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 6c id).await?;....l
11d0: 65 74 20 64 65 73 74 69 6e 61 74 69 6f 6e 20 3d et destination =
11e0: 20 43 68 61 74 50 65 65 72 49 64 3a 3a 66 72 6f ChatPeerId::fro
11f0: 6d 28 6d 61 74 63 68 20 72 65 61 6c 20 7b 0a 09 m(match real {..
1200: 09 09 09 74 72 75 65 20 3d 3e 20 73 6f 75 72 63 ...true => sourc
1210: 65 2e 63 68 61 6e 6e 65 6c 5f 69 64 2c 0a 09 09 e.channel_id,...
1220: 09 09 66 61 6c 73 65 20 3d 3e 20 73 6f 75 72 63 ..false => sourc
1230: 65 2e 6f 77 6e 65 72 2c 0a 09 09 09 7d 29 3b 0a e.owner,....});.
1240: 09 09 09 6c 65 74 20 6d 75 74 20 74 68 69 73 5f ...let mut this_
1250: 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e 3c 44 61 fetch: Option<Da
1260: 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 teTime<chrono::F
1270: 69 78 65 64 4f 66 66 73 65 74 3e 3e 20 3d 20 4e ixedOffset>> = N
1280: 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 one;....let mut
1290: 70 6f 73 74 73 3a 20 42 54 72 65 65 4d 61 70 3c posts: BTreeMap<
12a0: 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a DateTime<chrono:
12b0: 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 2c 20 53 :FixedOffset>, S
12c0: 74 72 69 6e 67 3e 20 3d 20 42 54 72 65 65 4d 61 tring> = BTreeMa
12d0: 70 3a 3a 6e 65 77 28 29 3b 0a 0a 09 09 09 6c 65 p::new();.....le
12e0: 74 20 72 65 73 70 6f 6e 73 65 20 3d 20 73 65 6c t response = sel
12f0: 66 2e 68 74 74 70 5f 63 6c 69 65 6e 74 2e 67 65 f.http_client.ge
1300: 74 28 26 73 6f 75 72 63 65 2e 75 72 6c 29 2e 73 t(&source.url).s
1310: 65 6e 64 28 29 2e 61 77 61 69 74 3f 3b 0a 09 09 end().await?;...
1320: 09 6c 65 74 20 73 74 61 74 75 73 20 3d 20 72 65 .let status = re
1330: 73 70 6f 6e 73 65 2e 73 74 61 74 75 73 28 29 3b sponse.status();
1340: 0a 09 09 09 6c 65 74 20 63 6f 6e 74 65 6e 74 20 ....let content
1350: 3d 20 72 65 73 70 6f 6e 73 65 2e 62 79 74 65 73 = response.bytes
1360: 28 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 6d 61 ().await?;....ma
1370: 74 63 68 20 72 73 73 3a 3a 43 68 61 6e 6e 65 6c tch rss::Channel
1380: 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f 6e ::read_from(&con
1390: 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 09 09 09 tent[..]) {.....
13a0: 4f 6b 28 66 65 65 64 29 20 3d 3e 20 7b 0a 09 09 Ok(feed) => {...
13b0: 09 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 ...for item in f
13c0: 65 65 64 2e 69 74 65 6d 73 28 29 20 7b 0a 09 09 eed.items() {...
13d0: 09 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 ....if let Some(
13e0: 6c 69 6e 6b 29 20 3d 20 69 74 65 6d 2e 6c 69 6e link) = item.lin
13f0: 6b 28 29 20 7b 0a 09 09 09 09 09 09 09 6c 65 74 k() {........let
1400: 20 64 61 74 65 20 3d 20 6d 61 74 63 68 20 69 74 date = match it
1410: 65 6d 2e 70 75 62 5f 64 61 74 65 28 29 20 7b 0a em.pub_date() {.
1420: 09 09 09 09 09 09 09 09 53 6f 6d 65 28 66 65 65 ........Some(fee
1430: 64 5f 64 61 74 65 29 20 3d 3e 20 44 61 74 65 54 d_date) => DateT
1440: 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f ime::parse_from_
1450: 72 66 63 32 38 32 32 28 66 65 65 64 5f 64 61 74 rfc2822(feed_dat
1460: 65 29 2c 0a 09 09 09 09 09 09 09 09 4e 6f 6e 65 e),.........None
1470: 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 => DateTime::pa
1480: 72 73 65 5f 66 72 6f 6d 5f 72 66 63 33 33 33 39 rse_from_rfc3339
1490: 28 26 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 63 6f (&item.dublin_co
14a0: 72 65 5f 65 78 74 28 29 2e 75 6e 77 72 61 70 28 re_ext().unwrap(
14b0: 29 2e 64 61 74 65 73 28 29 5b 30 5d 29 2c 0a 09 ).dates()[0]),..
14c0: 09 09 09 09 09 09 7d 3f 3b 0a 09 09 09 09 09 09 ......}?;.......
14d0: 09 6c 65 74 20 75 72 6c 20 3d 20 6c 69 6e 6b 3b .let url = link;
14e0: 0a 09 09 09 09 09 09 09 70 6f 73 74 73 2e 69 6e ........posts.in
14f0: 73 65 72 74 28 64 61 74 65 2c 20 75 72 6c 2e 74 sert(date, url.t
1500: 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 o_string());....
1510: 09 09 09 7d 0a 09 09 09 09 09 7d 3b 0a 09 09 09 ...}......};....
1520: 09 7d 2c 0a 09 09 09 09 45 72 72 28 65 72 72 29 .},.....Err(err)
1530: 20 3d 3e 20 6d 61 74 63 68 20 65 72 72 20 7b 0a => match err {.
1540: 09 09 09 09 09 72 73 73 3a 3a 45 72 72 6f 72 3a .....rss::Error:
1550: 3a 49 6e 76 61 6c 69 64 53 74 61 72 74 54 61 67 :InvalidStartTag
1560: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 6d 61 74 63 => {.......matc
1570: 68 20 61 74 6f 6d 5f 73 79 6e 64 69 63 61 74 69 h atom_syndicati
1580: 6f 6e 3a 3a 46 65 65 64 3a 3a 72 65 61 64 5f 66 on::Feed::read_f
1590: 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d rom(&content[..]
15a0: 29 20 7b 0a 09 09 09 09 09 09 09 4f 6b 28 66 65 ) {........Ok(fe
15b0: 65 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 ed) => {........
15c0: 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 65 65 .for item in fee
15d0: 64 2e 65 6e 74 72 69 65 73 28 29 20 7b 0a 09 09 d.entries() {...
15e0: 09 09 09 09 09 09 09 6c 65 74 20 64 61 74 65 20 .......let date
15f0: 3d 20 69 74 65 6d 2e 70 75 62 6c 69 73 68 65 64 = item.published
1600: 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 ().unwrap();....
1610: 09 09 09 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 ......let url =
1620: 69 74 65 6d 2e 6c 69 6e 6b 73 28 29 5b 30 5d 2e item.links()[0].
1630: 68 72 65 66 28 29 3b 0a 09 09 09 09 09 09 09 09 href();.........
1640: 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 2a 64 .posts.insert(*d
1650: 61 74 65 2c 20 75 72 6c 2e 74 6f 5f 73 74 72 69 ate, url.to_stri
1660: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 7d ng());.........}
1670: 3b 0a 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 ;........},.....
1680: 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 7b ...Err(err) => {
1690: 0a 09 09 09 09 09 09 09 09 62 61 69 6c 21 28 22 .........bail!("
16a0: 55 6e 73 75 70 70 6f 72 74 65 64 20 6f 72 20 6d Unsupported or m
16b0: 61 6e 67 6c 65 64 20 63 6f 6e 74 65 6e 74 3a 5c angled content:\
16c0: 6e 7b 3a 3f 7d 5c 6e 7b 65 72 72 3a 23 3f 7d 5c n{:?}\n{err:#?}\
16d0: 6e 7b 73 74 61 74 75 73 3a 23 3f 7d 5c 6e 22 2c n{status:#?}\n",
16e0: 20 26 73 6f 75 72 63 65 2e 75 72 6c 29 0a 09 09 &source.url)...
16f0: 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 7d 0a .....},.......}.
1700: 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 72 73 73 .....},......rss
1710: 3a 3a 45 72 72 6f 72 3a 3a 45 6f 66 20 3d 3e 20 ::Error::Eof =>
1720: 28 29 2c 0a 09 09 09 09 09 5f 20 3d 3e 20 62 61 (),......_ => ba
1730: 69 6c 21 28 22 55 6e 73 75 70 70 6f 72 74 65 64 il!("Unsupported
1740: 20 6f 72 20 6d 61 6e 67 6c 65 64 20 63 6f 6e 74 or mangled cont
1750: 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b 65 72 72 ent:\n{:?}\n{err
1760: 3a 23 3f 7d 5c 6e 7b 73 74 61 74 75 73 3a 23 3f :#?}\n{status:#?
1770: 7d 5c 6e 22 2c 20 26 73 6f 75 72 63 65 2e 75 72 }\n", &source.ur
1780: 6c 29 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 l).....}....};..
1790: 09 09 66 6f 72 20 28 64 61 74 65 2c 20 75 72 6c ..for (date, url
17a0: 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 65 72 28 ) in posts.iter(
17b0: 29 20 7b 0a 09 09 09 09 6c 65 74 20 70 6f 73 74 ) {.....let post
17c0: 5f 75 72 6c 3a 20 43 6f 77 3c 73 74 72 3e 20 3d _url: Cow<str> =
17d0: 20 6d 61 74 63 68 20 73 6f 75 72 63 65 2e 75 72 match source.ur
17e0: 6c 5f 72 65 20 7b 0a 09 09 09 09 09 53 6f 6d 65 l_re {......Some
17f0: 28 72 65 66 20 78 29 20 3d 3e 20 73 65 64 72 65 (ref x) => sedre
1800: 67 65 78 3a 3a 52 65 70 6c 61 63 65 43 6f 6d 6d gex::ReplaceComm
1810: 61 6e 64 3a 3a 6e 65 77 28 78 29 3f 2e 65 78 65 and::new(x)?.exe
1820: 63 75 74 65 28 75 72 6c 29 2c 0a 09 09 09 09 09 cute(url),......
1830: 4e 6f 6e 65 20 3d 3e 20 75 72 6c 2e 69 6e 74 6f None => url.into
1840: 28 29 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 09 69 (),.....};.....i
1850: 66 20 6c 65 74 20 53 6f 6d 65 28 65 78 69 73 74 f let Some(exist
1860: 73 29 20 3d 20 63 6f 6e 6e 2e 65 78 69 73 74 73 s) = conn.exists
1870: 28 26 70 6f 73 74 5f 75 72 6c 2c 20 2a 69 64 29 (&post_url, *id)
1880: 2e 61 77 61 69 74 3f 20 7b 0a 09 09 09 09 09 69 .await? {......i
1890: 66 20 21 20 65 78 69 73 74 73 20 7b 0a 09 09 09 f ! exists {....
18a0: 09 09 09 69 66 20 74 68 69 73 5f 66 65 74 63 68 ...if this_fetch
18b0: 2e 69 73 5f 6e 6f 6e 65 28 29 20 7c 7c 20 2a 64 .is_none() || *d
18c0: 61 74 65 20 3e 20 74 68 69 73 5f 66 65 74 63 68 ate > this_fetch
18d0: 2e 75 6e 77 72 61 70 28 29 20 7b 0a 09 09 09 09 .unwrap() {.....
18e0: 09 09 09 74 68 69 73 5f 66 65 74 63 68 20 3d 20 ...this_fetch =
18f0: 53 6f 6d 65 28 2a 64 61 74 65 29 3b 0a 09 09 09 Some(*date);....
1900: 09 09 09 7d 3b 0a 09 09 09 09 09 09 73 65 6c 66 ...};.......self
1910: 2e 73 65 6e 64 28 20 6d 61 74 63 68 20 26 73 6f .send( match &so
1920: 75 72 63 65 2e 69 76 5f 68 61 73 68 20 7b 0a 09 urce.iv_hash {..
1930: 09 09 09 09 09 09 53 6f 6d 65 28 68 61 73 68 29 ......Some(hash)
1940: 20 3d 3e 20 66 6f 72 6d 61 74 21 28 22 3c 61 20 => format!("<a
1950: 68 72 65 66 3d 5c 22 68 74 74 70 73 3a 2f 2f 74 href=\"https://t
1960: 2e 6d 65 2f 69 76 3f 75 72 6c 3d 7b 70 6f 73 74 .me/iv?url={post
1970: 5f 75 72 6c 7d 26 72 68 61 73 68 3d 7b 68 61 73 _url}&rhash={has
1980: 68 7d 5c 22 3e 20 3c 2f 61 3e 7b 70 6f 73 74 5f h}\"> </a>{post_
1990: 75 72 6c 7d 22 29 2c 0a 09 09 09 09 09 09 09 4e url}"),........N
19a0: 6f 6e 65 20 3d 3e 20 66 6f 72 6d 61 74 21 28 22 one => format!("
19b0: 7b 70 6f 73 74 5f 75 72 6c 7d 22 29 2c 0a 09 09 {post_url}"),...
19c0: 09 09 09 09 7d 2c 20 53 6f 6d 65 28 64 65 73 74 ....}, Some(dest
19d0: 69 6e 61 74 69 6f 6e 29 2c 20 53 6f 6d 65 28 50 ination), Some(P
19e0: 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 arseMode::Html))
19f0: 2e 61 77 61 69 74 3f 3b 0a 09 09 09 09 09 09 63 .await?;.......c
1a00: 6f 6e 6e 2e 61 64 64 5f 70 6f 73 74 28 2a 69 64 onn.add_post(*id
1a10: 2c 20 64 61 74 65 2c 20 26 70 6f 73 74 5f 75 72 , date, &post_ur
1a20: 6c 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 09 09 l).await?;......
1a30: 7d 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 09 70 6f };.....};.....po
1a40: 73 74 65 64 20 2b 3d 20 31 3b 0a 09 09 09 7d 3b sted += 1;....};
1a50: 0a 09 09 09 70 6f 73 74 73 2e 63 6c 65 61 72 28 ....posts.clear(
1a60: 29 3b 0a 09 09 7d 3b 0a 09 09 4f 6b 28 66 6f 72 );...};...Ok(for
1a70: 6d 61 74 21 28 22 50 6f 73 74 65 64 3a 20 7b 70 mat!("Posted: {p
1a80: 6f 73 74 65 64 7d 22 29 29 0a 09 7d 0a 0a 09 61 osted}"))..}...a
1a90: 73 79 6e 63 20 66 6e 20 61 75 74 6f 66 65 74 63 sync fn autofetc
1aa0: 68 28 26 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 h(&self) -> Resu
1ab0: 6c 74 3c 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 lt<std::time::Du
1ac0: 72 61 74 69 6f 6e 3e 20 7b 0a 09 09 6c 65 74 20 ration> {...let
1ad0: 6d 75 74 20 64 65 6c 61 79 20 3d 20 63 68 72 6f mut delay = chro
1ae0: 6e 6f 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 no::Duration::mi
1af0: 6e 75 74 65 73 28 31 29 3b 0a 09 09 6c 65 74 20 nutes(1);...let
1b00: 6e 6f 77 20 3d 20 63 68 72 6f 6e 6f 3a 3a 4c 6f now = chrono::Lo
1b10: 63 61 6c 3a 3a 6e 6f 77 28 29 3b 0a 09 09 6c 65 cal::now();...le
1b20: 74 20 71 75 65 75 65 20 3d 20 7b 0a 09 09 09 6c t queue = {....l
1b30: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
1b40: 6c 66 2e 64 62 2e 62 65 67 69 6e 28 29 2e 61 77 lf.db.begin().aw
1b50: 61 69 74 3f 3b 0a 09 09 09 63 6f 6e 6e 2e 67 65 ait?;....conn.ge
1b60: 74 5f 71 75 65 75 65 28 29 2e 61 77 61 69 74 3f t_queue().await?
1b70: 0a 09 09 7d 3b 0a 09 09 66 6f 72 20 72 6f 77 20 ...};...for row
1b80: 69 6e 20 71 75 65 75 65 20 7b 0a 09 09 09 69 66 in queue {....if
1b90: 20 6c 65 74 20 53 6f 6d 65 28 6e 65 78 74 5f 66 let Some(next_f
1ba0: 65 74 63 68 29 20 3d 20 72 6f 77 2e 6e 65 78 74 etch) = row.next
1bb0: 5f 66 65 74 63 68 20 7b 0a 09 09 09 09 69 66 20 _fetch {.....if
1bc0: 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 next_fetch < now
1bd0: 20 7b 0a 09 09 09 09 09 69 66 20 6c 65 74 20 28 {......if let (
1be0: 53 6f 6d 65 28 6f 77 6e 65 72 29 2c 20 53 6f 6d Some(owner), Som
1bf0: 65 28 73 6f 75 72 63 65 5f 69 64 29 29 20 3d 20 e(source_id)) =
1c00: 28 72 6f 77 2e 6f 77 6e 65 72 2c 20 72 6f 77 2e (row.owner, row.
1c10: 73 6f 75 72 63 65 5f 69 64 29 20 7b 0a 09 09 09 source_id) {....
1c20: 09 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 ...let clone = C
1c30: 6f 72 65 20 7b 0a 09 09 09 09 09 09 09 6f 77 6e ore {........own
1c40: 65 72 5f 63 68 61 74 3a 20 43 68 61 74 50 65 65 er_chat: ChatPee
1c50: 72 49 64 3a 3a 66 72 6f 6d 28 6f 77 6e 65 72 29 rId::from(owner)
1c60: 2c 0a 09 09 09 09 09 09 09 2e 2e 73 65 6c 66 2e ,..........self.
1c70: 63 6c 6f 6e 65 28 29 0a 09 09 09 09 09 09 7d 3b clone().......};
1c80: 0a 09 09 09 09 09 09 74 61 73 6b 3a 3a 73 70 61 .......task::spa
1c90: 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a wn(async move {.
1ca0: 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 .......if let Er
1cb0: 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 r(err) = clone.c
1cc0: 68 65 63 6b 28 73 6f 75 72 63 65 5f 69 64 2c 20 heck(source_id,
1cd0: 74 72 75 65 29 2e 61 77 61 69 74 20 7b 0a 09 09 true).await {...
1ce0: 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 ......if let Err
1cf0: 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 (err) = clone.se
1d00: 6e 64 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b nd(&format!("š
1d10: 91 20 7b 65 72 72 3a 3f 7d 22 29 2c 20 4e 6f 6e {err:?}"), Non
1d20: 65 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 74 20 7b e, None).await {
1d30: 0a 09 09 09 09 09 09 09 09 09 65 70 72 69 6e 74 ..........eprint
1d40: 6c 6e 21 28 22 43 68 65 63 6b 20 65 72 72 6f 72 ln!("Check error
1d50: 3a 20 7b 65 72 72 3a 3f 7d 22 29 3b 0a 09 09 09 : {err:?}");....
1d60: 09 09 09 09 09 09 2f 2f 20 63 6c 6f 6e 65 2e 64 ......// clone.d
1d70: 69 73 61 62 6c 65 28 26 73 6f 75 72 63 65 5f 69 isable(&source_i
1d80: 64 2c 20 6f 77 6e 65 72 29 2e 61 77 61 69 74 2e d, owner).await.
1d90: 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 09 09 unwrap();.......
1da0: 09 09 7d 3b 0a 09 09 09 09 09 09 09 7d 3b 0a 09 ..};........};..
1db0: 09 09 09 09 09 7d 29 3b 0a 09 09 09 09 09 7d 0a .....});......}.
1dc0: 09 09 09 09 7d 20 65 6c 73 65 20 69 66 20 6e 65 ....} else if ne
1dd0: 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 20 3c xt_fetch - now <
1de0: 20 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 64 65 delay {......de
1df0: 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 63 68 lay = next_fetch
1e00: 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 7d 0a 09 09 - now;.....}...
1e10: 09 7d 0a 09 09 7d 3b 0a 09 09 4f 6b 28 64 65 6c .}...};...Ok(del
1e20: 61 79 2e 74 6f 5f 73 74 64 28 29 3f 29 0a 09 7d ay.to_std()?)..}
1e30: 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 ...pub async fn
1e40: 6c 69 73 74 20 28 26 73 65 6c 66 2c 20 6f 77 6e list (&self, own
1e50: 65 72 3a 20 55 73 65 72 50 65 65 72 49 64 29 20 er: UserPeerId)
1e60: 2d 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 -> Result<String
1e70: 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 72 65 > {...let mut re
1e80: 70 6c 79 3a 20 56 65 63 3c 43 6f 77 3c 73 74 72 ply: Vec<Cow<str
1e90: 3e 3e 20 3d 20 76 65 63 21 5b 5d 3b 0a 09 09 72 >> = vec![];...r
1ea0: 65 70 6c 79 2e 70 75 73 68 28 22 43 68 61 6e 6e eply.push("Chann
1eb0: 65 6c 73 3a 22 2e 69 6e 74 6f 28 29 29 3b 0a 09 els:".into());..
1ec0: 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 .let mut conn =
1ed0: 73 65 6c 66 2e 64 62 2e 62 65 67 69 6e 28 29 2e self.db.begin().
1ee0: 61 77 61 69 74 3f 3b 0a 09 09 66 6f 72 20 72 6f await?;...for ro
1ef0: 77 20 69 6e 20 63 6f 6e 6e 2e 67 65 74 5f 6c 69 w in conn.get_li
1f00: 73 74 28 6f 77 6e 65 72 29 2e 61 77 61 69 74 3f st(owner).await?
1f10: 20 7b 0a 09 09 09 72 65 70 6c 79 2e 70 75 73 68 {....reply.push
1f20: 28 66 6f 72 6d 61 74 21 28 22 5c 6e 5c 5c 23 ef (format!("\n\\#ļ
1f30: b8 8f e2 83 a3 20 7b 7d 20 5c 5c 2a ef b8 8f e2 øā£ {} \\*ļøā
1f40: 83 a3 20 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 97 £ `{}` {}\nš
1f50: 20 60 7b 7d 60 22 2c 20 72 6f 77 2e 73 6f 75 72 `{}`", row.sour
1f60: 63 65 5f 69 64 2c 20 72 6f 77 2e 63 68 61 6e 6e ce_id, row.chann
1f70: 65 6c 2c 0a 09 09 09 09 6d 61 74 63 68 20 72 6f el,.....match ro
1f80: 77 2e 65 6e 61 62 6c 65 64 20 7b 0a 09 09 09 09 w.enabled {.....
1f90: 09 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 20 .true => "š
1fa0: 65 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 66 enabled",......f
1fb0: 61 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 73 alse => "ā dis
1fc0: 61 62 6c 65 64 22 2c 0a 09 09 09 09 7d 2c 20 72 abled",.....}, r
1fd0: 6f 77 2e 75 72 6c 29 2e 69 6e 74 6f 28 29 29 3b ow.url).into());
1fe0: 0a 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 ....if let Some(
1ff0: 68 61 73 68 29 20 3d 20 26 72 6f 77 2e 69 76 5f hash) = &row.iv_
2000: 68 61 73 68 20 7b 0a 09 09 09 09 72 65 70 6c 79 hash {.....reply
2010: 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 49 .push(format!("I
2020: 56 3a 20 60 7b 68 61 73 68 7d 60 22 29 2e 69 6e V: `{hash}`").in
2030: 74 6f 28 29 29 3b 0a 09 09 09 7d 0a 09 09 09 69 to());....}....i
2040: 66 20 6c 65 74 20 53 6f 6d 65 28 72 65 29 20 3d f let Some(re) =
2050: 20 26 72 6f 77 2e 75 72 6c 5f 72 65 20 7b 0a 09 &row.url_re {..
2060: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f ...reply.push(fo
2070: 72 6d 61 74 21 28 22 52 45 3a 20 60 7b 72 65 7d rmat!("RE: `{re}
2080: 60 22 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 `").into());....
2090: 7d 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 65 70 6c }...};...Ok(repl
20a0: 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 0a 09 7d y.join("\n"))..}
20b0: 0a 7d 0a 0a 69 6d 70 6c 20 55 70 64 61 74 65 48 .}..impl UpdateH
20c0: 61 6e 64 6c 65 72 20 66 6f 72 20 43 6f 72 65 20 andler for Core
20d0: 7b 0a 09 61 73 79 6e 63 20 66 6e 20 68 61 6e 64 {..async fn hand
20e0: 6c 65 20 28 26 73 65 6c 66 2c 20 75 70 64 61 74 le (&self, updat
20f0: 65 3a 20 55 70 64 61 74 65 29 20 7b 0a 09 09 69 e: Update) {...i
2100: 66 20 6c 65 74 20 55 70 64 61 74 65 54 79 70 65 f let UpdateType
2110: 3a 3a 4d 65 73 73 61 67 65 28 6d 73 67 29 20 3d ::Message(msg) =
2120: 20 75 70 64 61 74 65 2e 75 70 64 61 74 65 5f 74 update.update_t
2130: 79 70 65 20 7b 0a 09 09 09 69 66 20 6c 65 74 20 ype {....if let
2140: 4f 6b 28 63 6d 64 29 20 3d 20 43 6f 6d 6d 61 6e Ok(cmd) = Comman
2150: 64 3a 3a 74 72 79 5f 66 72 6f 6d 28 6d 73 67 29 d::try_from(msg)
2160: 20 7b 0a 09 09 09 09 6c 65 74 20 6d 73 67 20 3d {.....let msg =
2170: 20 63 6d 64 2e 67 65 74 5f 6d 65 73 73 61 67 65 cmd.get_message
2180: 28 29 3b 0a 09 09 09 09 6c 65 74 20 77 6f 72 64 ();.....let word
2190: 73 20 3d 20 63 6d 64 2e 67 65 74 5f 61 72 67 73 s = cmd.get_args
21a0: 28 29 3b 0a 09 09 09 09 6c 65 74 20 72 65 73 20 ();.....let res
21b0: 3d 20 6d 61 74 63 68 20 63 6d 64 2e 67 65 74 5f = match cmd.get_
21c0: 6e 61 6d 65 28 29 20 7b 0a 09 09 09 09 09 22 2f name() {......"/
21d0: 63 68 65 63 6b 22 20 7c 20 22 2f 63 6c 65 61 6e check" | "/clean
21e0: 22 20 7c 20 22 2f 65 6e 61 62 6c 65 22 20 7c 20 " | "/enable" |
21f0: 22 2f 64 65 6c 65 74 65 22 20 7c 20 22 2f 64 69 "/delete" | "/di
2200: 73 61 62 6c 65 22 20 3d 3e 20 63 6f 6d 6d 61 6e sable" => comman
2210: 64 3a 3a 63 6f 6d 6d 61 6e 64 28 73 65 6c 66 2c d::command(self,
2220: 20 6d 73 67 2c 20 77 6f 72 64 73 29 2e 61 77 61 msg, words).awa
2230: 69 74 2c 0a 09 09 09 09 09 22 2f 73 74 61 72 74 it,......"/start
2240: 22 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 73 74 " => command::st
2250: 61 72 74 28 73 65 6c 66 2c 20 6d 73 67 29 2e 61 art(self, msg).a
2260: 77 61 69 74 2c 0a 09 09 09 09 09 22 2f 6c 69 73 wait,......"/lis
2270: 74 22 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 6c t" => command::l
2280: 69 73 74 28 73 65 6c 66 2c 20 6d 73 67 29 2e 61 ist(self, msg).a
2290: 77 61 69 74 2c 0a 09 09 09 09 09 22 2f 61 64 64 wait,......"/add
22a0: 22 20 7c 20 22 2f 75 70 64 61 74 65 22 20 3d 3e " | "/update" =>
22b0: 20 63 6f 6d 6d 61 6e 64 3a 3a 75 70 64 61 74 65 command::update
22c0: 28 73 65 6c 66 2c 20 6d 73 67 2c 20 77 6f 72 64 (self, msg, word
22d0: 73 29 2e 61 77 61 69 74 2c 0a 09 09 09 09 09 61 s).await,......a
22e0: 6e 79 20 3d 3e 20 45 72 72 28 61 6e 79 68 6f 77 ny => Err(anyhow
22f0: 21 28 22 55 6e 6b 6e 6f 77 6e 20 63 6f 6d 6d 61 !("Unknown comma
2300: 6e 64 3a 20 7b 61 6e 79 7d 22 29 29 2c 0a 09 09 nd: {any}")),...
2310: 09 09 7d 3b 0a 09 09 09 09 69 66 20 6c 65 74 20 ..};.....if let
2320: 45 72 72 28 65 72 72 29 20 3d 20 72 65 73 20 7b Err(err) = res {
2330: 0a 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 ......if let Err
2340: 28 65 72 72 32 29 20 3d 20 73 65 6c 66 2e 73 65 (err2) = self.se
2350: 6e 64 28 66 6f 72 6d 61 74 21 28 22 5c 5c 23 65 nd(format!("\\#e
2360: 72 72 6f 72 5c 6e 60 60 60 5c 6e 7b 65 72 72 3a rror\n```\n{err:
2370: 3f 7d 5c 6e 60 60 60 22 29 2c 0a 09 09 09 09 09 ?}\n```"),......
2380: 09 53 6f 6d 65 28 6d 73 67 2e 63 68 61 74 2e 67 .Some(msg.chat.g
2390: 65 74 5f 69 64 28 29 29 2c 0a 09 09 09 09 09 09 et_id()),.......
23a0: 53 6f 6d 65 28 50 61 72 73 65 4d 6f 64 65 3a 3a Some(ParseMode::
23b0: 4d 61 72 6b 64 6f 77 6e 56 32 29 0a 09 09 09 09 MarkdownV2).....
23c0: 09 29 2e 61 77 61 69 74 7b 0a 09 09 09 09 09 09 .).await{.......
23d0: 64 62 67 21 28 65 72 72 32 29 3b 0a 09 09 09 09 dbg!(err2);.....
23e0: 09 7d 3b 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a .};.....}....};.
23f0: 09 09 7d 3b 0a 09 7d 0a 7d 0a ..};..}.}.