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 73 79 6e 63 3a 3a Set,..},..sync::
0070: 41 72 63 2c 0a 7d 3b 0a 0a 75 73 65 20 61 73 79 Arc,.};..use asy
0080: 6e 63 5f 63 6f 6d 70 61 74 3a 3a 43 6f 6d 70 61 nc_compat::Compa
0090: 74 3b 0a 75 73 65 20 63 68 72 6f 6e 6f 3a 3a 7b t;.use chrono::{
00a0: 0a 09 44 61 74 65 54 69 6d 65 2c 0a 09 4c 6f 63 ..DateTime,..Loc
00b0: 61 6c 2c 0a 7d 3b 0a 75 73 65 20 6c 61 7a 79 5f al,.};.use lazy_
00c0: 73 74 61 74 69 63 3a 3a 6c 61 7a 79 5f 73 74 61 static::lazy_sta
00d0: 74 69 63 3b 0a 75 73 65 20 72 65 67 65 78 3a 3a tic;.use regex::
00e0: 52 65 67 65 78 3b 0a 75 73 65 20 72 65 71 77 65 Regex;.use reqwe
00f0: 73 74 3a 3a 68 65 61 64 65 72 3a 3a 7b 0a 09 43 st::header::{..C
0100: 41 43 48 45 5f 43 4f 4e 54 52 4f 4c 2c 0a 09 45 ACHE_CONTROL,..E
0110: 58 50 49 52 45 53 2c 0a 09 4c 41 53 54 5f 4d 4f XPIRES,..LAST_MO
0120: 44 49 46 49 45 44 0a 7d 3b 0a 75 73 65 20 73 6d DIFIED.};.use sm
0130: 6f 6c 3a 3a 7b 0a 09 54 69 6d 65 72 2c 0a 09 6c ol::{..Timer,..l
0140: 6f 63 6b 3a 3a 4d 75 74 65 78 2c 0a 7d 3b 0a 75 ock::Mutex,.};.u
0150: 73 65 20 74 67 62 6f 74 3a 3a 7b 0a 09 61 70 69 se tgbot::{..api
0160: 3a 3a 43 6c 69 65 6e 74 2c 0a 09 68 61 6e 64 6c ::Client,..handl
0170: 65 72 3a 3a 55 70 64 61 74 65 48 61 6e 64 6c 65 er::UpdateHandle
0180: 72 2c 0a 09 74 79 70 65 73 3a 3a 7b 0a 09 09 42 r,..types::{...B
0190: 6f 74 2c 0a 09 09 43 68 61 74 50 65 65 72 49 64 ot,...ChatPeerId
01a0: 2c 0a 09 09 43 6f 6d 6d 61 6e 64 2c 0a 09 09 47 ,...Command,...G
01b0: 65 74 42 6f 74 2c 0a 09 09 4d 65 73 73 61 67 65 etBot,...Message
01c0: 2c 0a 09 09 50 61 72 73 65 4d 6f 64 65 2c 0a 09 ,...ParseMode,..
01d0: 09 53 65 6e 64 4d 65 73 73 61 67 65 2c 0a 09 09 .SendMessage,...
01e0: 55 70 64 61 74 65 2c 0a 09 09 55 70 64 61 74 65 Update,...Update
01f0: 54 79 70 65 2c 0a 09 09 55 73 65 72 50 65 65 72 Type,...UserPeer
0200: 49 64 2c 0a 09 7d 2c 0a 7d 3b 0a 75 73 65 20 73 Id,..},.};.use s
0210: 74 61 63 6b 65 64 5f 65 72 72 6f 72 73 3a 3a 7b tacked_errors::{
0220: 0a 09 52 65 73 75 6c 74 2c 0a 09 53 74 61 63 6b ..Result,..Stack
0230: 61 62 6c 65 45 72 72 2c 0a 09 61 6e 79 68 6f 77 ableErr,..anyhow
0240: 2c 0a 09 62 61 69 6c 2c 0a 7d 3b 0a 0a 6c 61 7a ,..bail,.};..laz
0250: 79 5f 73 74 61 74 69 63 21 7b 0a 09 70 75 62 20 y_static!{..pub
0260: 73 74 61 74 69 63 20 72 65 66 20 52 45 5f 53 50 static ref RE_SP
0270: 45 43 49 41 4c 3a 20 52 65 67 65 78 20 3d 20 52 ECIAL: Regex = R
0280: 65 67 65 78 3a 3a 6e 65 77 28 72 22 28 5b 5c 2d egex::new(r"([\-
0290: 5f 2a 5c 5b 5c 5d 28 29 7e 60 3e 23 2b 7c 7b 7d _*\[\]()~`>#+|{}
02a0: 5c 2e 21 5d 29 22 29 2e 75 6e 77 72 61 70 28 29 \.!])").unwrap()
02b0: 3b 0a 7d 0a 0a 2f 2f 2f 20 45 6e 63 6f 64 65 73 ;.}../// Encodes
02c0: 20 73 70 65 63 69 61 6c 20 48 54 4d 4c 20 65 6e special HTML en
02d0: 74 69 74 69 65 73 20 74 6f 20 70 72 65 76 65 6e tities to preven
02e0: 74 20 74 68 65 6d 20 69 6e 74 65 72 66 65 72 69 t them interferi
02f0: 6e 67 20 77 69 74 68 20 54 65 6c 65 67 72 61 6d ng with Telegram
0300: 20 48 54 4d 4c 0a 70 75 62 20 66 6e 20 65 6e 63 HTML.pub fn enc
0310: 6f 64 65 20 28 74 65 78 74 3a 20 26 73 74 72 29 ode (text: &str)
0320: 20 2d 3e 20 43 6f 77 3c 27 5f 2c 20 73 74 72 3e -> Cow<'_, str>
0330: 20 7b 0a 09 52 45 5f 53 50 45 43 49 41 4c 2e 72 {..RE_SPECIAL.r
0340: 65 70 6c 61 63 65 5f 61 6c 6c 28 74 65 78 74 2c eplace_all(text,
0350: 20 22 5c 5c 24 31 22 29 0a 7d 0a 0a 2f 2f 20 54 "\\$1").}..// T
0360: 68 69 73 20 6f 6e 65 20 64 6f 65 73 20 6e 6f 74 his one does not
0370: 68 69 6e 67 20 65 78 63 65 70 74 20 6d 61 6b 69 hing except maki
0380: 6e 67 20 73 75 72 65 20 6f 6e 6c 79 20 6f 6e 65 ng sure only one
0390: 20 74 6f 6b 65 6e 20 65 78 69 73 74 73 20 66 6f token exists fo
03a0: 72 20 65 61 63 68 20 69 64 0a 70 75 62 20 73 74 r each id.pub st
03b0: 72 75 63 74 20 54 6f 6b 65 6e 20 7b 0a 09 72 75 ruct Token {..ru
03c0: 6e 6e 69 6e 67 3a 20 41 72 63 3c 4d 75 74 65 78 nning: Arc<Mutex
03d0: 3c 48 61 73 68 53 65 74 3c 69 33 32 3e 3e 3e 2c <HashSet<i32>>>,
03e0: 0a 09 6d 79 5f 69 64 3a 20 69 33 32 2c 0a 7d 0a ..my_id: i32,.}.
03f0: 0a 69 6d 70 6c 20 54 6f 6b 65 6e 20 7b 0a 09 66 .impl Token {..f
0400: 6e 20 6e 65 77 20 28 72 75 6e 6e 69 6e 67 3a 20 n new (running:
0410: 26 41 72 63 3c 4d 75 74 65 78 3c 48 61 73 68 53 &Arc<Mutex<HashS
0420: 65 74 3c 69 33 32 3e 3e 3e 2c 20 6d 79 5f 69 64 et<i32>>>, my_id
0430: 3a 20 69 33 32 29 20 2d 3e 20 4f 70 74 69 6f 6e : i32) -> Option
0440: 3c 54 6f 6b 65 6e 3e 20 7b 0a 09 09 6c 65 74 20 <Token> {...let
0450: 72 75 6e 6e 69 6e 67 20 3d 20 72 75 6e 6e 69 6e running = runnin
0460: 67 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 73 6d 6f g.clone();...smo
0470: 6c 3a 3a 62 6c 6f 63 6b 5f 6f 6e 28 61 73 79 6e l::block_on(asyn
0480: 63 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 73 c {....let mut s
0490: 65 74 20 3d 20 72 75 6e 6e 69 6e 67 2e 6c 6f 63 et = running.loc
04a0: 6b 5f 61 72 63 28 29 2e 61 77 61 69 74 3b 0a 09 k_arc().await;..
04b0: 09 09 69 66 20 73 65 74 2e 63 6f 6e 74 61 69 6e ..if set.contain
04c0: 73 28 26 6d 79 5f 69 64 29 20 7b 0a 09 09 09 09 s(&my_id) {.....
04d0: 4e 6f 6e 65 0a 09 09 09 7d 20 65 6c 73 65 20 7b None....} else {
04e0: 0a 09 09 09 09 73 65 74 2e 69 6e 73 65 72 74 28 .....set.insert(
04f0: 6d 79 5f 69 64 29 3b 0a 09 09 09 09 53 6f 6d 65 my_id);.....Some
0500: 28 54 6f 6b 65 6e 20 7b 0a 09 09 09 09 09 72 75 (Token {......ru
0510: 6e 6e 69 6e 67 2c 0a 09 09 09 09 09 6d 79 5f 69 nning,......my_i
0520: 64 2c 0a 09 09 09 09 7d 29 0a 09 09 09 7d 0a 09 d,.....})....}..
0530: 09 7d 29 0a 09 7d 0a 7d 0a 0a 69 6d 70 6c 20 44 .})..}.}..impl D
0540: 72 6f 70 20 66 6f 72 20 54 6f 6b 65 6e 20 7b 0a rop for Token {.
0550: 09 66 6e 20 64 72 6f 70 20 28 26 6d 75 74 20 73 .fn drop (&mut s
0560: 65 6c 66 29 20 7b 0a 09 09 73 6d 6f 6c 3a 3a 62 elf) {...smol::b
0570: 6c 6f 63 6b 5f 6f 6e 28 61 73 79 6e 63 20 7b 0a lock_on(async {.
0580: 09 09 09 6c 65 74 20 6d 75 74 20 73 65 74 20 3d ...let mut set =
0590: 20 73 65 6c 66 2e 72 75 6e 6e 69 6e 67 2e 6c 6f self.running.lo
05a0: 63 6b 5f 61 72 63 28 29 2e 61 77 61 69 74 3b 0a ck_arc().await;.
05b0: 09 09 09 73 65 74 2e 72 65 6d 6f 76 65 28 26 73 ...set.remove(&s
05c0: 65 6c 66 2e 6d 79 5f 69 64 29 3b 0a 09 09 7d 29 elf.my_id);...})
05d0: 0a 09 7d 0a 7d 0a 0a 23 5b 64 65 72 69 76 65 28 ..}.}..#[derive(
05e0: 43 6c 6f 6e 65 29 5d 0a 70 75 62 20 73 74 72 75 Clone)].pub stru
05f0: 63 74 20 43 6f 72 65 20 7b 0a 09 6f 77 6e 65 72 ct Core {..owner
0600: 5f 63 68 61 74 3a 20 43 68 61 74 50 65 65 72 49 _chat: ChatPeerI
0610: 64 2c 0a 09 2f 2f 20 6d 61 78 5f 64 65 6c 61 79 d,..// max_delay
0620: 3a 20 75 31 36 2c 0a 09 70 75 62 20 74 67 3a 20 : u16,..pub tg:
0630: 43 6c 69 65 6e 74 2c 0a 09 70 75 62 20 6d 65 3a Client,..pub me:
0640: 20 42 6f 74 2c 0a 09 70 75 62 20 64 62 3a 20 44 Bot,..pub db: D
0650: 62 2c 0a 09 72 75 6e 6e 69 6e 67 3a 20 41 72 63 b,..running: Arc
0660: 3c 4d 75 74 65 78 3c 48 61 73 68 53 65 74 3c 69 <Mutex<HashSet<i
0670: 33 32 3e 3e 3e 2c 0a 09 68 74 74 70 5f 63 6c 69 32>>>,..http_cli
0680: 65 6e 74 3a 20 72 65 71 77 65 73 74 3a 3a 43 6c ent: reqwest::Cl
0690: 69 65 6e 74 2c 0a 7d 0a 0a 70 75 62 20 73 74 72 ient,.}..pub str
06a0: 75 63 74 20 50 6f 73 74 20 7b 0a 09 75 72 69 3a uct Post {..uri:
06b0: 20 53 74 72 69 6e 67 2c 0a 09 74 69 74 6c 65 3a String,..title:
06c0: 20 53 74 72 69 6e 67 2c 0a 09 61 75 74 68 6f 72 String,..author
06d0: 73 3a 20 53 74 72 69 6e 67 2c 0a 09 73 75 6d 6d s: String,..summ
06e0: 61 72 79 3a 20 53 74 72 69 6e 67 2c 0a 7d 0a 0a ary: String,.}..
06f0: 69 6d 70 6c 20 43 6f 72 65 20 7b 0a 09 70 75 62 impl Core {..pub
0700: 20 61 73 79 6e 63 20 66 6e 20 6e 65 77 28 73 65 async fn new(se
0710: 74 74 69 6e 67 73 3a 20 63 6f 6e 66 69 67 3a 3a ttings: config::
0720: 43 6f 6e 66 69 67 29 20 2d 3e 20 52 65 73 75 6c Config) -> Resul
0730: 74 3c 43 6f 72 65 3e 20 7b 0a 09 09 6c 65 74 20 t<Core> {...let
0740: 6f 77 6e 65 72 5f 63 68 61 74 20 3d 20 43 68 61 owner_chat = Cha
0750: 74 50 65 65 72 49 64 3a 3a 66 72 6f 6d 28 73 65 tPeerId::from(se
0760: 74 74 69 6e 67 73 2e 67 65 74 5f 69 6e 74 28 22 ttings.get_int("
0770: 6f 77 6e 65 72 22 29 2e 73 74 61 63 6b 28 29 3f owner").stack()?
0780: 29 3b 0a 09 09 6c 65 74 20 61 70 69 5f 6b 65 79 );...let api_key
0790: 20 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 5f = settings.get_
07a0: 73 74 72 69 6e 67 28 22 61 70 69 5f 6b 65 79 22 string("api_key"
07b0: 29 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 6c 65 ).stack()?;...le
07c0: 74 20 74 67 20 3d 20 43 6c 69 65 6e 74 3a 3a 6e t tg = Client::n
07d0: 65 77 28 26 61 70 69 5f 6b 65 79 29 2e 73 74 61 ew(&api_key).sta
07e0: 63 6b 28 29 3f 0a 09 09 09 2e 77 69 74 68 5f 68 ck()?.....with_h
07f0: 6f 73 74 28 73 65 74 74 69 6e 67 73 2e 67 65 74 ost(settings.get
0800: 5f 73 74 72 69 6e 67 28 22 61 70 69 5f 67 61 74 _string("api_gat
0810: 65 77 61 79 22 29 2e 73 74 61 63 6b 28 29 3f 29 eway").stack()?)
0820: 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 6c 69 ;....let mut cli
0830: 65 6e 74 20 3d 20 72 65 71 77 65 73 74 3a 3a 43 ent = reqwest::C
0840: 6c 69 65 6e 74 3a 3a 62 75 69 6c 64 65 72 28 29 lient::builder()
0850: 3b 0a 09 09 69 66 20 6c 65 74 20 4f 6b 28 70 72 ;...if let Ok(pr
0860: 6f 78 79 29 20 3d 20 73 65 74 74 69 6e 67 73 2e oxy) = settings.
0870: 67 65 74 5f 73 74 72 69 6e 67 28 22 70 72 6f 78 get_string("prox
0880: 79 22 29 20 7b 0a 09 09 09 6c 65 74 20 70 72 6f y") {....let pro
0890: 78 79 20 3d 20 72 65 71 77 65 73 74 3a 3a 50 72 xy = reqwest::Pr
08a0: 6f 78 79 3a 3a 61 6c 6c 28 70 72 6f 78 79 29 2e oxy::all(proxy).
08b0: 73 74 61 63 6b 28 29 3f 3b 0a 09 09 09 63 6c 69 stack()?;....cli
08c0: 65 6e 74 20 3d 20 63 6c 69 65 6e 74 2e 70 72 6f ent = client.pro
08d0: 78 79 28 70 72 6f 78 79 29 3b 0a 09 09 7d 0a 09 xy(proxy);...}..
08e0: 09 6c 65 74 20 68 74 74 70 5f 63 6c 69 65 6e 74 .let http_client
08f0: 20 3d 20 63 6c 69 65 6e 74 2e 62 75 69 6c 64 28 = client.build(
0900: 29 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 6c 65 ).stack()?;...le
0910: 74 20 6d 65 20 3d 20 74 67 2e 65 78 65 63 75 74 t me = tg.execut
0920: 65 28 47 65 74 42 6f 74 29 2e 61 77 61 69 74 2e e(GetBot).await.
0930: 73 74 61 63 6b 28 29 3f 3b 0a 09 09 6c 65 74 20 stack()?;...let
0940: 63 6f 72 65 20 3d 20 43 6f 72 65 20 7b 0a 09 09 core = Core {...
0950: 09 74 67 2c 0a 09 09 09 6d 65 2c 0a 09 09 09 6f .tg,....me,....o
0960: 77 6e 65 72 5f 63 68 61 74 2c 0a 09 09 09 64 62 wner_chat,....db
0970: 3a 20 44 62 3a 3a 6e 65 77 28 26 73 65 74 74 69 : Db::new(&setti
0980: 6e 67 73 2e 67 65 74 5f 73 74 72 69 6e 67 28 22 ngs.get_string("
0990: 70 67 22 29 2e 73 74 61 63 6b 28 29 3f 29 3f 2c pg").stack()?)?,
09a0: 0a 09 09 09 72 75 6e 6e 69 6e 67 3a 20 41 72 63 ....running: Arc
09b0: 3a 3a 6e 65 77 28 4d 75 74 65 78 3a 3a 6e 65 77 ::new(Mutex::new
09c0: 28 48 61 73 68 53 65 74 3a 3a 6e 65 77 28 29 29 (HashSet::new())
09d0: 29 2c 0a 09 09 09 68 74 74 70 5f 63 6c 69 65 6e ),....http_clien
09e0: 74 2c 0a 09 09 09 2f 2f 20 6d 61 78 5f 64 65 6c t,....// max_del
09f0: 61 79 3a 20 36 30 2c 0a 09 09 7d 3b 0a 09 09 6c ay: 60,...};...l
0a00: 65 74 20 63 6c 6f 6e 65 20 3d 20 63 6f 72 65 2e et clone = core.
0a10: 63 6c 6f 6e 65 28 29 3b 0a 09 09 73 6d 6f 6c 3a clone();...smol:
0a20: 3a 73 70 61 77 6e 28 43 6f 6d 70 61 74 3a 3a 6e :spawn(Compat::n
0a30: 65 77 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a ew(async move {.
0a40: 09 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 09 6c 65 ...loop {.....le
0a50: 74 20 64 65 6c 61 79 20 3d 20 6d 61 74 63 68 20 t delay = match
0a60: 26 63 6c 6f 6e 65 2e 61 75 74 6f 66 65 74 63 68 &clone.autofetch
0a70: 28 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 ().await {......
0a80: 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 Err(err) => {...
0a90: 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 ....if let Err(e
0aa0: 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 rr) = clone.send
0ab0: 28 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b (format!("🛑 {
0ac0: 65 72 72 7d 22 29 2c 20 4e 6f 6e 65 2c 20 4e 6f err}"), None, No
0ad0: 6e 65 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 ne).await {.....
0ae0: 09 09 09 65 70 72 69 6e 74 6c 6e 21 28 22 41 75 ...eprintln!("Au
0af0: 74 6f 66 65 74 63 68 20 65 72 72 6f 72 3a 20 7b tofetch error: {
0b00: 65 72 72 3a 3f 7d 22 29 3b 0a 09 09 09 09 09 09 err:?}");.......
0b10: 7d 3b 0a 09 09 09 09 09 09 73 74 64 3a 3a 74 69 };.......std::ti
0b20: 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 66 72 me::Duration::fr
0b30: 6f 6d 5f 73 65 63 73 28 36 30 29 0a 09 09 09 09 om_secs(60).....
0b40: 09 7d 2c 0a 09 09 09 09 09 4f 6b 28 74 69 6d 65 .},......Ok(time
0b50: 29 20 3d 3e 20 2a 74 69 6d 65 2c 0a 09 09 09 09 ) => *time,.....
0b60: 7d 3b 0a 09 09 09 09 54 69 6d 65 72 3a 3a 61 66 };.....Timer::af
0b70: 74 65 72 28 64 65 6c 61 79 29 2e 61 77 61 69 74 ter(delay).await
0b80: 3b 0a 09 09 09 7d 0a 09 09 7d 29 29 2e 64 65 74 ;....}...})).det
0b90: 61 63 68 28 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 ach();...Ok(core
0ba0: 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 )..}...pub async
0bb0: 20 66 6e 20 73 65 6e 64 20 3c 53 3e 28 26 73 65 fn send <S>(&se
0bc0: 6c 66 2c 20 6d 73 67 3a 20 53 2c 20 74 61 72 67 lf, msg: S, targ
0bd0: 65 74 3a 20 4f 70 74 69 6f 6e 3c 43 68 61 74 50 et: Option<ChatP
0be0: 65 65 72 49 64 3e 2c 20 6d 6f 64 65 3a 20 4f 70 eerId>, mode: Op
0bf0: 74 69 6f 6e 3c 50 61 72 73 65 4d 6f 64 65 3e 29 tion<ParseMode>)
0c00: 20 2d 3e 20 52 65 73 75 6c 74 3c 4d 65 73 73 61 -> Result<Messa
0c10: 67 65 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e ge>..where S: In
0c20: 74 6f 3c 53 74 72 69 6e 67 3e 20 7b 0a 09 09 6c to<String> {...l
0c30: 65 74 20 6d 73 67 20 3d 20 6d 73 67 2e 69 6e 74 et msg = msg.int
0c40: 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d 6f 64 65 o();....let mode
0c50: 20 3d 20 6d 6f 64 65 2e 75 6e 77 72 61 70 5f 6f = mode.unwrap_o
0c60: 72 28 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d r(ParseMode::Htm
0c70: 6c 29 3b 0a 09 09 6c 65 74 20 74 61 72 67 65 74 l);...let target
0c80: 20 3d 20 74 61 72 67 65 74 2e 75 6e 77 72 61 70 = target.unwrap
0c90: 5f 6f 72 28 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 _or(self.owner_c
0ca0: 68 61 74 29 3b 0a 09 09 73 65 6c 66 2e 74 67 2e hat);...self.tg.
0cb0: 65 78 65 63 75 74 65 28 0a 09 09 09 53 65 6e 64 execute(....Send
0cc0: 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 74 61 72 Message::new(tar
0cd0: 67 65 74 2c 20 6d 73 67 29 0a 09 09 09 09 2e 77 get, msg)......w
0ce0: 69 74 68 5f 70 61 72 73 65 5f 6d 6f 64 65 28 6d ith_parse_mode(m
0cf0: 6f 64 65 29 0a 09 09 29 2e 61 77 61 69 74 2e 73 ode)...).await.s
0d00: 74 61 63 6b 28 29 0a 09 7d 0a 0a 09 70 75 62 20 tack()..}...pub
0d10: 61 73 79 6e 63 20 66 6e 20 63 68 65 63 6b 20 28 async fn check (
0d20: 26 73 65 6c 66 2c 20 69 64 3a 20 69 33 32 2c 20 &self, id: i32,
0d30: 72 65 61 6c 3a 20 62 6f 6f 6c 2c 20 6c 61 73 74 real: bool, last
0d40: 5f 73 63 72 61 70 65 3a 20 4f 70 74 69 6f 6e 3c _scrape: Option<
0d50: 44 61 74 65 54 69 6d 65 3c 4c 6f 63 61 6c 3e 3e DateTime<Local>>
0d60: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 ) -> Result<Stri
0d70: 6e 67 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 ng> {...let mut
0d80: 70 6f 73 74 65 64 3a 20 69 33 32 20 3d 20 30 3b posted: i32 = 0;
0d90: 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 ...let mut conn
0da0: 3d 20 73 65 6c 66 2e 64 62 2e 62 65 67 69 6e 28 = self.db.begin(
0db0: 29 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f ).await.stack()?
0dc0: 3b 0a 0a 09 09 6c 65 74 20 74 6f 6b 65 6e 20 3d ;....let token =
0dd0: 20 54 6f 6b 65 6e 3a 3a 6e 65 77 28 26 73 65 6c Token::new(&sel
0de0: 66 2e 72 75 6e 6e 69 6e 67 2c 20 69 64 29 3b 0a f.running, id);.
0df0: 09 09 69 66 20 74 6f 6b 65 6e 2e 69 73 5f 6e 6f ..if token.is_no
0e00: 6e 65 28 29 20 7b 0a 09 09 09 62 61 69 6c 21 28 ne() {....bail!(
0e10: 22 63 68 65 63 6b 20 69 73 20 61 6c 72 65 61 64 "check is alread
0e20: 79 20 72 75 6e 6e 69 6e 67 22 29 3b 0a 09 09 7d y running");...}
0e30: 0a 09 09 6c 65 74 20 73 6f 75 72 63 65 20 3d 20 ...let source =
0e40: 63 6f 6e 6e 2e 67 65 74 5f 73 6f 75 72 63 65 28 conn.get_source(
0e50: 69 64 2c 20 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 id, self.owner_c
0e60: 68 61 74 29 2e 61 77 61 69 74 2e 73 74 61 63 6b hat).await.stack
0e70: 28 29 3f 3b 0a 09 09 63 6f 6e 6e 2e 73 65 74 5f ()?;...conn.set_
0e80: 73 63 72 61 70 65 28 69 64 29 2e 61 77 61 69 74 scrape(id).await
0e90: 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 6c 65 74 .stack()?;...let
0ea0: 20 64 65 73 74 69 6e 61 74 69 6f 6e 20 3d 20 43 destination = C
0eb0: 68 61 74 50 65 65 72 49 64 3a 3a 66 72 6f 6d 28 hatPeerId::from(
0ec0: 6d 61 74 63 68 20 72 65 61 6c 20 7b 0a 09 09 09 match real {....
0ed0: 74 72 75 65 20 3d 3e 20 73 6f 75 72 63 65 2e 63 true => source.c
0ee0: 68 61 6e 6e 65 6c 5f 69 64 2c 0a 09 09 09 66 61 hannel_id,....fa
0ef0: 6c 73 65 20 3d 3e 20 73 6f 75 72 63 65 2e 6f 77 lse => source.ow
0f00: 6e 65 72 2c 0a 09 09 7d 29 3b 0a 09 09 6c 65 74 ner,...});...let
0f10: 20 6d 75 74 20 74 68 69 73 5f 66 65 74 63 68 3a mut this_fetch:
0f20: 20 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 6d 65 Option<DateTime
0f30: 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 <chrono::FixedOf
0f40: 66 73 65 74 3e 3e 20 3d 20 4e 6f 6e 65 3b 0a 09 fset>> = None;..
0f50: 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 73 3a 20 .let mut posts:
0f60: 42 54 72 65 65 4d 61 70 3c 44 61 74 65 54 69 6d BTreeMap<DateTim
0f70: 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f e<chrono::FixedO
0f80: 66 66 73 65 74 3e 2c 20 50 6f 73 74 3e 20 3d 20 ffset>, Post> =
0f90: 42 54 72 65 65 4d 61 70 3a 3a 6e 65 77 28 29 3b BTreeMap::new();
0fa0: 0a 0a 09 09 6c 65 74 20 6d 75 74 20 62 75 69 6c ....let mut buil
0fb0: 64 65 72 20 3d 20 73 65 6c 66 2e 68 74 74 70 5f der = self.http_
0fc0: 63 6c 69 65 6e 74 2e 67 65 74 28 26 73 6f 75 72 client.get(&sour
0fd0: 63 65 2e 75 72 6c 29 3b 0a 09 09 69 66 20 6c 65 ce.url);...if le
0fe0: 74 20 53 6f 6d 65 28 6c 61 73 74 5f 73 63 72 61 t Some(last_scra
0ff0: 70 65 29 20 3d 20 6c 61 73 74 5f 73 63 72 61 70 pe) = last_scrap
1000: 65 20 7b 0a 09 09 09 62 75 69 6c 64 65 72 20 3d e {....builder =
1010: 20 62 75 69 6c 64 65 72 2e 68 65 61 64 65 72 28 builder.header(
1020: 4c 41 53 54 5f 4d 4f 44 49 46 49 45 44 2c 20 6c LAST_MODIFIED, l
1030: 61 73 74 5f 73 63 72 61 70 65 2e 74 6f 5f 72 66 ast_scrape.to_rf
1040: 63 32 38 32 32 28 29 29 3b 0a 09 09 7d 3b 0a 09 c2822());...};..
1050: 09 6c 65 74 20 72 65 73 70 6f 6e 73 65 20 3d 20 .let response =
1060: 62 75 69 6c 64 65 72 2e 73 65 6e 64 28 29 2e 61 builder.send().a
1070: 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 wait.stack()?;..
1080: 09 7b 0a 09 09 09 6c 65 74 20 68 65 61 64 65 72 .{....let header
1090: 73 20 3d 20 72 65 73 70 6f 6e 73 65 2e 68 65 61 s = response.hea
10a0: 64 65 72 73 28 29 3b 0a 09 09 09 6c 65 74 20 65 ders();....let e
10b0: 78 70 69 72 65 73 20 3d 20 68 65 61 64 65 72 73 xpires = headers
10c0: 2e 67 65 74 28 45 58 50 49 52 45 53 29 3b 0a 09 .get(EXPIRES);..
10d0: 09 09 6c 65 74 20 63 61 63 68 65 20 3d 20 68 65 ..let cache = he
10e0: 61 64 65 72 73 2e 67 65 74 28 43 41 43 48 45 5f aders.get(CACHE_
10f0: 43 4f 4e 54 52 4f 4c 29 3b 0a 09 09 09 69 66 20 CONTROL);....if
1100: 65 78 70 69 72 65 73 2e 69 73 5f 73 6f 6d 65 28 expires.is_some(
1110: 29 20 7c 7c 20 63 61 63 68 65 2e 69 73 5f 73 6f ) || cache.is_so
1120: 6d 65 28 29 20 7b 0a 09 09 09 09 70 72 69 6e 74 me() {.....print
1130: 6c 6e 21 28 22 7b 7d 20 7b 7d 20 7b 3a 3f 7d 20 ln!("{} {} {:?}
1140: 7b 3a 3f 7d 20 7b 3a 3f 7d 22 2c 20 4c 6f 63 61 {:?} {:?}", Loca
1150: 6c 3a 3a 6e 6f 77 28 29 2e 74 6f 5f 72 66 63 32 l::now().to_rfc2
1160: 38 32 32 28 29 2c 20 26 73 6f 75 72 63 65 2e 75 822(), &source.u
1170: 72 6c 2c 20 6c 61 73 74 5f 73 63 72 61 70 65 2c rl, last_scrape,
1180: 20 65 78 70 69 72 65 73 2c 20 63 61 63 68 65 29 expires, cache)
1190: 3b 0a 09 09 09 7d 0a 09 09 7d 0a 09 09 6c 65 74 ;....}...}...let
11a0: 20 73 74 61 74 75 73 20 3d 20 72 65 73 70 6f 6e status = respon
11b0: 73 65 2e 73 74 61 74 75 73 28 29 3b 0a 09 09 6c se.status();...l
11c0: 65 74 20 63 6f 6e 74 65 6e 74 20 3d 20 72 65 73 et content = res
11d0: 70 6f 6e 73 65 2e 62 79 74 65 73 28 29 2e 61 77 ponse.bytes().aw
11e0: 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 ait.stack()?;...
11f0: 6d 61 74 63 68 20 72 73 73 3a 3a 43 68 61 6e 6e match rss::Chann
1200: 65 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 el::read_from(&c
1210: 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 09 ontent[..]) {...
1220: 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 7b 0a 09 .Ok(feed) => {..
1230: 09 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 ...for item in f
1240: 65 65 64 2e 69 74 65 6d 73 28 29 20 7b 0a 09 09 eed.items() {...
1250: 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 6c ...if let Some(l
1260: 69 6e 6b 29 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b ink) = item.link
1270: 28 29 20 7b 0a 09 09 09 09 09 09 6c 65 74 20 64 () {.......let d
1280: 61 74 65 20 3d 20 6d 61 74 63 68 20 69 74 65 6d ate = match item
1290: 2e 70 75 62 5f 64 61 74 65 28 29 20 7b 0a 09 09 .pub_date() {...
12a0: 09 09 09 09 09 53 6f 6d 65 28 66 65 65 64 5f 64 .....Some(feed_d
12b0: 61 74 65 29 20 3d 3e 20 44 61 74 65 54 69 6d 65 ate) => DateTime
12c0: 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 ::parse_from_rfc
12d0: 32 38 32 32 28 66 65 65 64 5f 64 61 74 65 29 2c 2822(feed_date),
12e0: 0a 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 ........None =>
12f0: 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f DateTime::parse_
1300: 66 72 6f 6d 5f 72 66 63 33 33 33 39 28 6d 61 74 from_rfc3339(mat
1310: 63 68 20 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 63 ch item.dublin_c
1320: 6f 72 65 5f 65 78 74 28 29 20 7b 0a 09 09 09 09 ore_ext() {.....
1330: 09 09 09 09 53 6f 6d 65 28 64 61 74 65 73 29 20 ....Some(dates)
1340: 3d 3e 20 26 64 61 74 65 73 2e 64 61 74 65 73 28 => &dates.dates(
1350: 29 5b 30 5d 2c 0a 09 09 09 09 09 09 09 09 4e 6f )[0],.........No
1360: 6e 65 20 3d 3e 20 62 61 69 6c 21 28 22 46 65 65 ne => bail!("Fee
1370: 64 20 69 74 65 6d 20 6d 69 73 73 65 73 20 70 6f d item misses po
1380: 73 74 69 6e 67 20 64 61 74 65 2e 22 29 2c 0a 09 sting date."),..
1390: 09 09 09 09 09 09 7d 29 2c 0a 09 09 09 09 09 09 ......}),.......
13a0: 7d 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 09 09 }.stack()?;.....
13b0: 09 09 6c 65 74 20 75 72 69 20 3d 20 6c 69 6e 6b ..let uri = link
13c0: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 3b 0a 09 09 .to_string();...
13d0: 09 09 09 09 6c 65 74 20 74 69 74 6c 65 20 3d 20 ....let title =
13e0: 69 74 65 6d 2e 74 69 74 6c 65 28 29 2e 75 6e 77 item.title().unw
13f0: 72 61 70 5f 6f 72 28 22 22 29 2e 74 6f 5f 73 74 rap_or("").to_st
1400: 72 69 6e 67 28 29 3b 0a 09 09 09 09 09 09 6c 65 ring();.......le
1410: 74 20 61 75 74 68 6f 72 73 20 3d 20 69 74 65 6d t authors = item
1420: 2e 61 75 74 68 6f 72 28 29 2e 75 6e 77 72 61 70 .author().unwrap
1430: 5f 6f 72 28 22 22 29 2e 74 6f 5f 73 74 72 69 6e _or("").to_strin
1440: 67 28 29 3b 0a 09 09 09 09 09 09 6c 65 74 20 73 g();.......let s
1450: 75 6d 6d 61 72 79 20 3d 20 69 74 65 6d 2e 63 6f ummary = item.co
1460: 6e 74 65 6e 74 28 29 2e 75 6e 77 72 61 70 5f 6f ntent().unwrap_o
1470: 72 28 22 22 29 2e 74 6f 5f 73 74 72 69 6e 67 28 r("").to_string(
1480: 29 3b 0a 09 09 09 09 09 09 70 6f 73 74 73 2e 69 );.......posts.i
1490: 6e 73 65 72 74 28 64 61 74 65 2c 20 50 6f 73 74 nsert(date, Post
14a0: 7b 0a 09 09 09 09 09 09 09 75 72 69 2c 0a 09 09 {........uri,...
14b0: 09 09 09 09 09 74 69 74 6c 65 2c 0a 09 09 09 09 .....title,.....
14c0: 09 09 09 61 75 74 68 6f 72 73 2c 0a 09 09 09 09 ...authors,.....
14d0: 09 09 09 73 75 6d 6d 61 72 79 2c 0a 09 09 09 09 ...summary,.....
14e0: 09 09 7d 29 3b 0a 09 09 09 09 09 7d 0a 09 09 09 ..});......}....
14f0: 09 7d 3b 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 .};....},....Err
1500: 28 65 72 72 29 20 3d 3e 20 6d 61 74 63 68 20 65 (err) => match e
1510: 72 72 20 7b 0a 09 09 09 09 72 73 73 3a 3a 45 72 rr {.....rss::Er
1520: 72 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 ror::InvalidStar
1530: 74 54 61 67 20 3d 3e 20 7b 0a 09 09 09 09 09 6d tTag => {......m
1540: 61 74 63 68 20 61 74 6f 6d 5f 73 79 6e 64 69 63 atch atom_syndic
1550: 61 74 69 6f 6e 3a 3a 46 65 65 64 3a 3a 72 65 61 ation::Feed::rea
1560: 64 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b d_from(&content[
1570: 2e 2e 5d 29 20 7b 0a 09 09 09 09 09 09 4f 6b 28 ..]) {.......Ok(
1580: 66 65 65 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 feed) => {......
1590: 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 65 ..for item in fe
15a0: 65 64 2e 65 6e 74 72 69 65 73 28 29 20 7b 0a 09 ed.entries() {..
15b0: 09 09 09 09 09 09 09 6c 65 74 20 64 61 74 65 20 .......let date
15c0: 3d 20 69 74 65 6d 2e 70 75 62 6c 69 73 68 65 64 = item.published
15d0: 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 ().unwrap();....
15e0: 09 09 09 09 09 6c 65 74 20 75 72 69 20 3d 20 69 .....let uri = i
15f0: 74 65 6d 2e 6c 69 6e 6b 73 28 29 5b 30 5d 2e 68 tem.links()[0].h
1600: 72 65 66 28 29 2e 74 6f 5f 73 74 72 69 6e 67 28 ref().to_string(
1610: 29 3b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 74 );.........let t
1620: 69 74 6c 65 20 3d 20 69 74 65 6d 2e 74 69 74 6c itle = item.titl
1630: 65 28 29 2e 74 6f 5f 73 74 72 69 6e 67 28 29 3b e().to_string();
1640: 0a 09 09 09 09 09 09 09 09 6c 65 74 20 61 75 74 .........let aut
1650: 68 6f 72 73 20 3d 20 69 74 65 6d 2e 61 75 74 68 hors = item.auth
1660: 6f 72 73 28 29 2e 69 74 65 72 28 29 2e 6d 61 70 ors().iter().map
1670: 28 7c 78 7c 20 66 6f 72 6d 61 74 21 28 22 7b 7d (|x| format!("{}
1680: 20 3c 7b 3a 3f 7d 3e 22 2c 20 78 2e 6e 61 6d 65 <{:?}>", x.name
1690: 28 29 2c 20 78 2e 65 6d 61 69 6c 28 29 29 29 2e (), x.email())).
16a0: 63 6f 6c 6c 65 63 74 3a 3a 3c 56 65 63 3c 53 74 collect::<Vec<St
16b0: 72 69 6e 67 3e 3e 28 29 2e 6a 6f 69 6e 28 22 2c ring>>().join(",
16c0: 20 22 29 3b 0a 09 09 09 09 09 09 09 09 6c 65 74 ");.........let
16d0: 20 73 75 6d 6d 61 72 79 20 3d 20 69 66 20 6c 65 summary = if le
16e0: 74 20 53 6f 6d 65 28 73 75 6d 29 20 3d 20 69 74 t Some(sum) = it
16f0: 65 6d 2e 73 75 6d 6d 61 72 79 28 29 20 7b 20 73 em.summary() { s
1700: 75 6d 2e 76 61 6c 75 65 2e 63 6c 6f 6e 65 28 29 um.value.clone()
1710: 20 7d 20 65 6c 73 65 20 7b 20 53 74 72 69 6e 67 } else { String
1720: 3a 3a 6e 65 77 28 29 20 7d 3b 0a 09 09 09 09 09 ::new() };......
1730: 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 ...posts.insert(
1740: 2a 64 61 74 65 2c 20 50 6f 73 74 7b 0a 09 09 09 *date, Post{....
1750: 09 09 09 09 09 09 75 72 69 2c 0a 09 09 09 09 09 ......uri,......
1760: 09 09 09 09 74 69 74 6c 65 2c 0a 09 09 09 09 09 ....title,......
1770: 09 09 09 09 61 75 74 68 6f 72 73 2c 0a 09 09 09 ....authors,....
1780: 09 09 09 09 09 09 73 75 6d 6d 61 72 79 2c 0a 09 ......summary,..
1790: 09 09 09 09 09 09 09 7d 29 3b 0a 09 09 09 09 09 .......});......
17a0: 09 09 7d 3b 0a 09 09 09 09 09 09 7d 2c 0a 09 09 ..};.......},...
17b0: 09 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 ....Err(err) =>
17c0: 7b 0a 09 09 09 09 09 09 09 62 61 69 6c 21 28 22 {........bail!("
17d0: 55 6e 73 75 70 70 6f 72 74 65 64 20 6f 72 20 6d Unsupported or m
17e0: 61 6e 67 6c 65 64 20 63 6f 6e 74 65 6e 74 3a 5c angled content:\
17f0: 6e 7b 3a 3f 7d 5c 6e 7b 65 72 72 7d 5c 6e 7b 73 n{:?}\n{err}\n{s
1800: 74 61 74 75 73 3a 23 3f 7d 5c 6e 22 2c 20 26 73 tatus:#?}\n", &s
1810: 6f 75 72 63 65 2e 75 72 6c 29 0a 09 09 09 09 09 ource.url)......
1820: 09 7d 2c 0a 09 09 09 09 09 7d 0a 09 09 09 09 7d .},......}.....}
1830: 2c 0a 09 09 09 09 72 73 73 3a 3a 45 72 72 6f 72 ,.....rss::Error
1840: 3a 3a 45 6f 66 20 3d 3e 20 28 29 2c 0a 09 09 09 ::Eof => (),....
1850: 09 5f 20 3d 3e 20 62 61 69 6c 21 28 22 55 6e 73 ._ => bail!("Uns
1860: 75 70 70 6f 72 74 65 64 20 6f 72 20 6d 61 6e 67 upported or mang
1870: 6c 65 64 20 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a led content:\n{:
1880: 3f 7d 5c 6e 7b 65 72 72 7d 5c 6e 7b 73 74 61 74 ?}\n{err}\n{stat
1890: 75 73 3a 23 3f 7d 5c 6e 22 2c 20 26 73 6f 75 72 us:#?}\n", &sour
18a0: 63 65 2e 75 72 6c 29 0a 09 09 09 7d 0a 09 09 7d ce.url)....}...}
18b0: 3b 0a 09 09 66 6f 72 20 28 64 61 74 65 2c 20 70 ;...for (date, p
18c0: 6f 73 74 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 ost) in posts.it
18d0: 65 72 28 29 20 7b 0a 09 09 09 6c 65 74 20 70 6f er() {....let po
18e0: 73 74 5f 75 72 6c 3a 20 43 6f 77 3c 73 74 72 3e st_url: Cow<str>
18f0: 20 3d 20 6d 61 74 63 68 20 73 6f 75 72 63 65 2e = match source.
1900: 75 72 6c 5f 72 65 20 7b 0a 09 09 09 09 53 6f 6d url_re {.....Som
1910: 65 28 72 65 66 20 78 29 20 3d 3e 20 73 65 64 72 e(ref x) => sedr
1920: 65 67 65 78 3a 3a 52 65 70 6c 61 63 65 43 6f 6d egex::ReplaceCom
1930: 6d 61 6e 64 3a 3a 6e 65 77 28 78 29 2e 73 74 61 mand::new(x).sta
1940: 63 6b 28 29 3f 2e 65 78 65 63 75 74 65 28 26 70 ck()?.execute(&p
1950: 6f 73 74 2e 75 72 69 29 2c 0a 09 09 09 09 4e 6f ost.uri),.....No
1960: 6e 65 20 3d 3e 20 70 6f 73 74 2e 75 72 69 2e 63 ne => post.uri.c
1970: 6c 6f 6e 65 28 29 2e 69 6e 74 6f 28 29 2c 0a 09 lone().into(),..
1980: 09 09 7d 3b 0a 09 09 09 69 66 20 6c 65 74 20 53 ..};....if let S
1990: 6f 6d 65 28 65 78 69 73 74 73 29 20 3d 20 63 6f ome(exists) = co
19a0: 6e 6e 2e 65 78 69 73 74 73 28 26 70 6f 73 74 5f nn.exists(&post_
19b0: 75 72 6c 2c 20 69 64 29 2e 61 77 61 69 74 2e 73 url, id).await.s
19c0: 74 61 63 6b 28 29 3f 20 7b 0a 09 09 09 09 69 66 tack()? {.....if
19d0: 20 21 20 65 78 69 73 74 73 20 7b 0a 09 09 09 09 ! exists {.....
19e0: 09 69 66 20 74 68 69 73 5f 66 65 74 63 68 2e 69 .if this_fetch.i
19f0: 73 5f 6e 6f 6e 65 28 29 20 7c 7c 20 2a 64 61 74 s_none() || *dat
1a00: 65 20 3e 20 74 68 69 73 5f 66 65 74 63 68 2e 75 e > this_fetch.u
1a10: 6e 77 72 61 70 28 29 20 7b 0a 09 09 09 09 09 09 nwrap() {.......
1a20: 74 68 69 73 5f 66 65 74 63 68 20 3d 20 53 6f 6d this_fetch = Som
1a30: 65 28 2a 64 61 74 65 29 3b 0a 09 09 09 09 09 7d e(*date);......}
1a40: 3b 0a 09 09 09 09 09 73 65 6c 66 2e 73 65 6e 64 ;......self.send
1a50: 28 20 6d 61 74 63 68 20 26 73 6f 75 72 63 65 2e ( match &source.
1a60: 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 iv_hash {.......
1a70: 53 6f 6d 65 28 68 61 73 68 29 20 3d 3e 20 66 6f Some(hash) => fo
1a80: 72 6d 61 74 21 28 22 3c 61 20 68 72 65 66 3d 5c rmat!("<a href=\
1a90: 22 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 "https://t.me/iv
1aa0: 3f 75 72 6c 3d 7b 70 6f 73 74 5f 75 72 6c 7d 26 ?url={post_url}&
1ab0: 72 68 61 73 68 3d 7b 68 61 73 68 7d 5c 22 3e 20 rhash={hash}\">
1ac0: 3c 2f 61 3e 7b 70 6f 73 74 5f 75 72 6c 7d 22 29 </a>{post_url}")
1ad0: 2c 0a 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 ,.......None =>
1ae0: 66 6f 72 6d 61 74 21 28 22 7b 70 6f 73 74 5f 75 format!("{post_u
1af0: 72 6c 7d 22 29 2c 0a 09 09 09 09 09 7d 2c 20 53 rl}"),......}, S
1b00: 6f 6d 65 28 64 65 73 74 69 6e 61 74 69 6f 6e 29 ome(destination)
1b10: 2c 20 53 6f 6d 65 28 50 61 72 73 65 4d 6f 64 65 , Some(ParseMode
1b20: 3a 3a 48 74 6d 6c 29 29 2e 61 77 61 69 74 2e 73 ::Html)).await.s
1b30: 74 61 63 6b 28 29 3f 3b 0a 09 09 09 09 09 63 6f tack()?;......co
1b40: 6e 6e 2e 61 64 64 5f 70 6f 73 74 28 69 64 2c 20 nn.add_post(id,
1b50: 64 61 74 65 2c 20 26 70 6f 73 74 5f 75 72 6c 29 date, &post_url)
1b60: 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b .await.stack()?;
1b70: 0a 09 09 09 09 7d 3b 0a 09 09 09 7d 3b 0a 09 09 .....};....};...
1b80: 09 70 6f 73 74 65 64 20 2b 3d 20 31 3b 0a 09 09 .posted += 1;...
1b90: 7d 3b 0a 09 09 70 6f 73 74 73 2e 63 6c 65 61 72 };...posts.clear
1ba0: 28 29 3b 0a 09 09 4f 6b 28 66 6f 72 6d 61 74 21 ();...Ok(format!
1bb0: 28 22 50 6f 73 74 65 64 3a 20 7b 70 6f 73 74 65 ("Posted: {poste
1bc0: 64 7d 22 29 29 0a 09 7d 0a 0a 09 61 73 79 6e 63 d}"))..}...async
1bd0: 20 66 6e 20 61 75 74 6f 66 65 74 63 68 28 26 73 fn autofetch(&s
1be0: 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 73 elf) -> Result<s
1bf0: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
1c00: 6f 6e 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 on> {...let mut
1c10: 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a delay = chrono::
1c20: 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 Duration::minute
1c30: 73 28 31 29 3b 0a 09 09 6c 65 74 20 6e 6f 77 20 s(1);...let now
1c40: 3d 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a = chrono::Local:
1c50: 3a 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 20 71 75 :now();...let qu
1c60: 65 75 65 20 3d 20 7b 0a 09 09 09 6c 65 74 20 6d eue = {....let m
1c70: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 64 ut conn = self.d
1c80: 62 2e 62 65 67 69 6e 28 29 2e 61 77 61 69 74 2e b.begin().await.
1c90: 73 74 61 63 6b 28 29 3f 3b 0a 09 09 09 63 6f 6e stack()?;....con
1ca0: 6e 2e 67 65 74 5f 71 75 65 75 65 28 29 2e 61 77 n.get_queue().aw
1cb0: 61 69 74 2e 73 74 61 63 6b 28 29 3f 0a 09 09 7d ait.stack()?...}
1cc0: 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 71 ;...for row in q
1cd0: 75 65 75 65 20 7b 0a 09 09 09 69 66 20 6c 65 74 ueue {....if let
1ce0: 20 53 6f 6d 65 28 6e 65 78 74 5f 66 65 74 63 68 Some(next_fetch
1cf0: 29 20 3d 20 72 6f 77 2e 6e 65 78 74 5f 66 65 74 ) = row.next_fet
1d00: 63 68 20 7b 0a 09 09 09 09 69 66 20 6e 65 78 74 ch {.....if next
1d10: 5f 66 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a 09 _fetch < now {..
1d20: 09 09 09 09 69 66 20 6c 65 74 20 28 53 6f 6d 65 ....if let (Some
1d30: 28 6f 77 6e 65 72 29 2c 20 53 6f 6d 65 28 73 6f (owner), Some(so
1d40: 75 72 63 65 5f 69 64 29 2c 20 6c 61 73 74 5f 73 urce_id), last_s
1d50: 63 72 61 70 65 29 20 3d 20 28 72 6f 77 2e 6f 77 crape) = (row.ow
1d60: 6e 65 72 2c 20 72 6f 77 2e 73 6f 75 72 63 65 5f ner, row.source_
1d70: 69 64 2c 20 72 6f 77 2e 6c 61 73 74 5f 73 63 72 id, row.last_scr
1d80: 61 70 65 29 20 7b 0a 09 09 09 09 09 09 6c 65 74 ape) {.......let
1d90: 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 65 20 7b 0a clone = Core {.
1da0: 09 09 09 09 09 09 09 6f 77 6e 65 72 5f 63 68 61 .......owner_cha
1db0: 74 3a 20 43 68 61 74 50 65 65 72 49 64 3a 3a 66 t: ChatPeerId::f
1dc0: 72 6f 6d 28 6f 77 6e 65 72 29 2c 0a 09 09 09 09 rom(owner),.....
1dd0: 09 09 09 2e 2e 73 65 6c 66 2e 63 6c 6f 6e 65 28 .....self.clone(
1de0: 29 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ).......};......
1df0: 09 6c 65 74 20 73 6f 75 72 63 65 20 3d 20 7b 0a .let source = {.
1e00: 09 09 09 09 09 09 09 6c 65 74 20 6d 75 74 20 63 .......let mut c
1e10: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 64 62 2e 62 65 onn = self.db.be
1e20: 67 69 6e 28 29 2e 61 77 61 69 74 2e 73 74 61 63 gin().await.stac
1e30: 6b 28 29 3f 3b 0a 09 09 09 09 09 09 09 6d 61 74 k()?;........mat
1e40: 63 68 20 63 6f 6e 6e 2e 67 65 74 5f 6f 6e 65 28 ch conn.get_one(
1e50: 6f 77 6e 65 72 2c 20 73 6f 75 72 63 65 5f 69 64 owner, source_id
1e60: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 ).await {.......
1e70: 09 09 4f 6b 28 53 6f 6d 65 28 73 6f 75 72 63 65 ..Ok(Some(source
1e80: 29 29 20 3d 3e 20 73 6f 75 72 63 65 2e 74 6f 5f )) => source.to_
1e90: 73 74 72 69 6e 67 28 29 2c 0a 09 09 09 09 09 09 string(),.......
1ea0: 09 09 4f 6b 28 4e 6f 6e 65 29 20 3d 3e 20 22 53 ..Ok(None) => "S
1eb0: 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 20 ource not found
1ec0: 69 6e 20 64 61 74 61 62 61 73 65 3f 22 2e 74 6f in database?".to
1ed0: 5f 73 74 72 69 6e 67 28 29 2c 0a 09 09 09 09 09 _string(),......
1ee0: 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 66 ...Err(err) => f
1ef0: 6f 72 6d 61 74 21 28 22 46 61 69 6c 65 64 20 74 ormat!("Failed t
1f00: 6f 20 66 65 74 63 68 20 73 6f 75 72 63 65 20 64 o fetch source d
1f10: 61 74 61 3a 5c 6e 7b 65 72 72 7d 22 29 2c 0a 09 ata:\n{err}"),..
1f20: 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 7d 3b ......}.......};
1f30: 0a 09 09 09 09 09 09 73 6d 6f 6c 3a 3a 73 70 61 .......smol::spa
1f40: 77 6e 28 43 6f 6d 70 61 74 3a 3a 6e 65 77 28 61 wn(Compat::new(a
1f50: 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 sync move {.....
1f60: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
1f70: 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b r) = clone.check
1f80: 28 73 6f 75 72 63 65 5f 69 64 2c 20 74 72 75 65 (source_id, true
1f90: 2c 20 53 6f 6d 65 28 6c 61 73 74 5f 73 63 72 61 , Some(last_scra
1fa0: 70 65 29 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 pe)).await {....
1fb0: 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 .....if let Err(
1fc0: 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e err) = clone.sen
1fd0: 64 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 d(&format!("🛑
1fe0: 20 7b 73 6f 75 72 63 65 7d 5c 6e 7b 7d 22 2c 20 {source}\n{}",
1ff0: 65 6e 63 6f 64 65 28 26 65 72 72 2e 74 6f 5f 73 encode(&err.to_s
2000: 74 72 69 6e 67 28 29 29 29 2c 20 4e 6f 6e 65 2c tring())), None,
2010: 20 53 6f 6d 65 28 50 61 72 73 65 4d 6f 64 65 3a Some(ParseMode:
2020: 3a 4d 61 72 6b 64 6f 77 6e 56 32 29 29 2e 61 77 :MarkdownV2)).aw
2030: 61 69 74 20 7b 0a 09 09 09 09 09 09 09 09 09 65 ait {..........e
2040: 70 72 69 6e 74 6c 6e 21 28 22 43 68 65 63 6b 20 println!("Check
2050: 65 72 72 6f 72 3a 20 7b 65 72 72 7d 22 29 3b 0a error: {err}");.
2060: 09 09 09 09 09 09 09 09 09 2f 2f 20 63 6c 6f 6e .........// clon
2070: 65 2e 64 69 73 61 62 6c 65 28 26 73 6f 75 72 63 e.disable(&sourc
2080: 65 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 61 77 61 e_id, owner).awa
2090: 69 74 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 it.unwrap();....
20a0: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 7d .....};........}
20b0: 3b 0a 09 09 09 09 09 09 7d 29 29 2e 64 65 74 61 ;.......})).deta
20c0: 63 68 28 29 3b 0a 09 09 09 09 09 7d 0a 09 09 09 ch();......}....
20d0: 09 7d 20 65 6c 73 65 20 69 66 20 6e 65 78 74 5f .} else if next_
20e0: 66 65 74 63 68 20 2d 20 6e 6f 77 20 3c 20 64 65 fetch - now < de
20f0: 6c 61 79 20 7b 0a 09 09 09 09 09 64 65 6c 61 79 lay {......delay
2100: 20 3d 20 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 = next_fetch -
2110: 6e 6f 77 3b 0a 09 09 09 09 7d 0a 09 09 09 7d 0a now;.....}....}.
2120: 09 09 7d 3b 0a 09 09 64 65 6c 61 79 2e 74 6f 5f ..};...delay.to_
2130: 73 74 64 28 29 2e 73 74 61 63 6b 28 29 0a 09 7d std().stack()..}
2140: 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 ...pub async fn
2150: 6c 69 73 74 20 28 26 73 65 6c 66 2c 20 6f 77 6e list (&self, own
2160: 65 72 3a 20 55 73 65 72 50 65 65 72 49 64 29 20 er: UserPeerId)
2170: 2d 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 -> Result<String
2180: 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 72 65 > {...let mut re
2190: 70 6c 79 3a 20 56 65 63 3c 53 74 72 69 6e 67 3e ply: Vec<String>
21a0: 20 3d 20 76 65 63 21 5b 5d 3b 0a 09 09 72 65 70 = vec![];...rep
21b0: 6c 79 2e 70 75 73 68 28 22 43 68 61 6e 6e 65 6c ly.push("Channel
21c0: 73 3a 22 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 6c s:".into());...l
21d0: 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 et mut conn = se
21e0: 6c 66 2e 64 62 2e 62 65 67 69 6e 28 29 2e 61 77 lf.db.begin().aw
21f0: 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 ait.stack()?;...
2200: 66 6f 72 20 72 6f 77 20 69 6e 20 63 6f 6e 6e 2e for row in conn.
2210: 67 65 74 5f 6c 69 73 74 28 6f 77 6e 65 72 29 2e get_list(owner).
2220: 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 20 7b await.stack()? {
2230: 0a 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 72 ....reply.push(r
2240: 6f 77 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b ow.to_string());
2250: 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 65 70 6c 79 ...};...Ok(reply
2260: 2e 6a 6f 69 6e 28 22 5c 6e 5c 6e 22 29 29 0a 09 .join("\n\n"))..
2270: 7d 0a 7d 0a 0a 69 6d 70 6c 20 55 70 64 61 74 65 }.}..impl Update
2280: 48 61 6e 64 6c 65 72 20 66 6f 72 20 43 6f 72 65 Handler for Core
2290: 20 7b 0a 09 61 73 79 6e 63 20 66 6e 20 68 61 6e {..async fn han
22a0: 64 6c 65 20 28 26 73 65 6c 66 2c 20 75 70 64 61 dle (&self, upda
22b0: 74 65 3a 20 55 70 64 61 74 65 29 20 7b 0a 09 09 te: Update) {...
22c0: 69 66 20 6c 65 74 20 55 70 64 61 74 65 54 79 70 if let UpdateTyp
22d0: 65 3a 3a 4d 65 73 73 61 67 65 28 6d 73 67 29 20 e::Message(msg)
22e0: 3d 20 75 70 64 61 74 65 2e 75 70 64 61 74 65 5f = update.update_
22f0: 74 79 70 65 20 7b 0a 09 09 09 69 66 20 6c 65 74 type {....if let
2300: 20 4f 6b 28 63 6d 64 29 20 3d 20 43 6f 6d 6d 61 Ok(cmd) = Comma
2310: 6e 64 3a 3a 74 72 79 5f 66 72 6f 6d 28 6d 73 67 nd::try_from(msg
2320: 29 20 7b 0a 09 09 09 09 6c 65 74 20 6d 73 67 20 ) {.....let msg
2330: 3d 20 63 6d 64 2e 67 65 74 5f 6d 65 73 73 61 67 = cmd.get_messag
2340: 65 28 29 3b 0a 09 09 09 09 6c 65 74 20 77 6f 72 e();.....let wor
2350: 64 73 20 3d 20 63 6d 64 2e 67 65 74 5f 61 72 67 ds = cmd.get_arg
2360: 73 28 29 3b 0a 09 09 09 09 6c 65 74 20 63 6f 6d s();.....let com
2370: 6d 61 6e 64 20 3d 20 63 6d 64 2e 67 65 74 5f 6e mand = cmd.get_n
2380: 61 6d 65 28 29 3b 0a 09 09 09 09 6c 65 74 20 72 ame();.....let r
2390: 65 73 20 3d 20 6d 61 74 63 68 20 63 6f 6d 6d 61 es = match comma
23a0: 6e 64 20 7b 0a 09 09 09 09 09 22 2f 63 68 65 63 nd {......"/chec
23b0: 6b 22 20 7c 20 22 2f 63 6c 65 61 6e 22 20 7c 20 k" | "/clean" |
23c0: 22 2f 65 6e 61 62 6c 65 22 20 7c 20 22 2f 64 65 "/enable" | "/de
23d0: 6c 65 74 65 22 20 7c 20 22 2f 64 69 73 61 62 6c lete" | "/disabl
23e0: 65 22 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 63 e" => command::c
23f0: 6f 6d 6d 61 6e 64 28 73 65 6c 66 2c 20 63 6f 6d ommand(self, com
2400: 6d 61 6e 64 2c 20 6d 73 67 2c 20 77 6f 72 64 73 mand, msg, words
2410: 29 2e 61 77 61 69 74 2c 0a 09 09 09 09 09 22 2f ).await,......"/
2420: 73 74 61 72 74 22 20 3d 3e 20 63 6f 6d 6d 61 6e start" => comman
2430: 64 3a 3a 73 74 61 72 74 28 73 65 6c 66 2c 20 6d d::start(self, m
2440: 73 67 29 2e 61 77 61 69 74 2c 0a 09 09 09 09 09 sg).await,......
2450: 22 2f 6c 69 73 74 22 20 3d 3e 20 63 6f 6d 6d 61 "/list" => comma
2460: 6e 64 3a 3a 6c 69 73 74 28 73 65 6c 66 2c 20 6d nd::list(self, m
2470: 73 67 29 2e 61 77 61 69 74 2c 0a 09 09 09 09 09 sg).await,......
2480: 22 2f 61 64 64 22 20 7c 20 22 2f 75 70 64 61 74 "/add" | "/updat
2490: 65 22 20 3d 3e 20 63 6f 6d 6d 61 6e 64 3a 3a 75 e" => command::u
24a0: 70 64 61 74 65 28 73 65 6c 66 2c 20 63 6f 6d 6d pdate(self, comm
24b0: 61 6e 64 2c 20 6d 73 67 2c 20 77 6f 72 64 73 29 and, msg, words)
24c0: 2e 61 77 61 69 74 2c 0a 09 09 09 09 09 61 6e 79 .await,......any
24d0: 20 3d 3e 20 45 72 72 28 61 6e 79 68 6f 77 21 28 => Err(anyhow!(
24e0: 22 55 6e 6b 6e 6f 77 6e 20 63 6f 6d 6d 61 6e 64 "Unknown command
24f0: 3a 20 7b 61 6e 79 7d 22 29 29 2c 0a 09 09 09 09 : {any}")),.....
2500: 7d 3b 0a 09 09 09 09 69 66 20 6c 65 74 20 45 72 };.....if let Er
2510: 72 28 65 72 72 29 20 3d 20 72 65 73 20 7b 0a 09 r(err) = res {..
2520: 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 ....if let Err(e
2530: 72 72 32 29 20 3d 20 73 65 6c 66 2e 73 65 6e 64 rr2) = self.send
2540: 28 66 6f 72 6d 61 74 21 28 22 5c 5c 23 65 72 72 (format!("\\#err
2550: 6f 72 5c 6e 60 60 60 5c 6e 7b 65 72 72 7d 5c 6e or\n```\n{err}\n
2560: 60 60 60 22 29 2c 0a 09 09 09 09 09 09 53 6f 6d ```"),.......Som
2570: 65 28 6d 73 67 2e 63 68 61 74 2e 67 65 74 5f 69 e(msg.chat.get_i
2580: 64 28 29 29 2c 0a 09 09 09 09 09 09 53 6f 6d 65 d()),.......Some
2590: 28 50 61 72 73 65 4d 6f 64 65 3a 3a 4d 61 72 6b (ParseMode::Mark
25a0: 64 6f 77 6e 56 32 29 0a 09 09 09 09 09 29 2e 61 downV2)......).a
25b0: 77 61 69 74 7b 0a 09 09 09 09 09 09 64 62 67 21 wait{.......dbg!
25c0: 28 65 72 72 32 29 3b 0a 09 09 09 09 09 7d 3b 0a (err2);......};.
25d0: 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 7d 3b ....}....};...};
25e0: 0a 09 7d 0a 7d 0a ..}.}.