0000: 75 73 65 20 61 6e 79 68 6f 77 3a 3a 7b 61 6e 79 use anyhow::{any
0010: 68 6f 77 2c 20 62 61 69 6c 2c 20 43 6f 6e 74 65 how, bail, Conte
0020: 78 74 2c 20 52 65 73 75 6c 74 7d 3b 0a 75 73 65 xt, Result};.use
0030: 20 61 73 79 6e 63 5f 73 74 64 3a 3a 74 61 73 6b async_std::task
0040: 3b 0a 75 73 65 20 63 68 72 6f 6e 6f 3a 3a 44 61 ;.use chrono::Da
0050: 74 65 54 69 6d 65 3b 0a 75 73 65 20 73 71 6c 78 teTime;.use sqlx
0060: 3a 3a 70 6f 73 74 67 72 65 73 3a 3a 50 67 50 6f ::postgres::PgPo
0070: 6f 6c 4f 70 74 69 6f 6e 73 3b 0a 75 73 65 20 74 olOptions;.use t
0080: 65 6c 6f 78 69 64 65 3a 3a 7b 0a 09 42 6f 74 2c eloxide::{..Bot,
0090: 0a 09 70 61 79 6c 6f 61 64 73 3a 3a 53 65 6e 64 ..payloads::Send
00a0: 4d 65 73 73 61 67 65 2c 0a 09 72 65 71 75 65 73 Message,..reques
00b0: 74 73 3a 3a 52 65 71 75 65 73 74 65 72 2c 0a 09 ts::Requester,..
00c0: 74 79 70 65 73 3a 3a 7b 0a 09 09 4d 65 2c 0a 09 types::{...Me,..
00d0: 09 55 73 65 72 49 64 2c 0a 09 7d 2c 0a 7d 3b 0a .UserId,..},.};.
00e0: 75 73 65 20 74 68 69 73 65 72 72 6f 72 3a 3a 45 use thiserror::E
00f0: 72 72 6f 72 3b 0a 0a 75 73 65 20 73 74 64 3a 3a rror;..use std::
0100: 7b 0a 09 62 6f 72 72 6f 77 3a 3a 43 6f 77 2c 0a {..borrow::Cow,.
0110: 09 63 6f 6c 6c 65 63 74 69 6f 6e 73 3a 3a 7b 0a .collections::{.
0120: 09 09 42 54 72 65 65 4d 61 70 2c 0a 09 09 48 61 ..BTreeMap,...Ha
0130: 73 68 53 65 74 2c 0a 09 7d 2c 0a 09 6e 75 6d 3a shSet,..},..num:
0140: 3a 54 72 79 46 72 6f 6d 49 6e 74 45 72 72 6f 72 :TryFromIntError
0150: 2c 0a 09 73 79 6e 63 3a 3a 7b 0a 09 09 41 72 63 ,..sync::{...Arc
0160: 2c 0a 09 09 4d 75 74 65 78 0a 09 7d 2c 0a 7d 3b ,...Mutex..},.};
0170: 0a 0a 23 5b 64 65 72 69 76 65 28 45 72 72 6f 72 ..#[derive(Error
0180: 2c 20 44 65 62 75 67 29 5d 0a 70 75 62 20 65 6e , Debug)].pub en
0190: 75 6d 20 52 73 73 45 72 72 6f 72 20 7b 0a 09 2f um RssError {../
01a0: 2f 20 23 5b 65 72 72 6f 72 28 74 72 61 6e 73 70 / #[error(transp
01b0: 61 72 65 6e 74 29 5d 0a 09 2f 2f 20 54 67 28 23 arent)]..// Tg(#
01c0: 5b 66 72 6f 6d 5d 20 54 67 45 72 72 6f 72 29 2c [from] TgError),
01d0: 0a 09 23 5b 65 72 72 6f 72 28 74 72 61 6e 73 70 ..#[error(transp
01e0: 61 72 65 6e 74 29 5d 0a 09 49 6e 74 28 23 5b 66 arent)]..Int(#[f
01f0: 72 6f 6d 5d 20 54 72 79 46 72 6f 6d 49 6e 74 45 rom] TryFromIntE
0200: 72 72 6f 72 29 2c 0a 7d 0a 0a 23 5b 64 65 72 69 rror),.}..#[deri
0210: 76 65 28 43 6c 6f 6e 65 29 5d 0a 70 75 62 20 73 ve(Clone)].pub s
0220: 74 72 75 63 74 20 43 6f 72 65 20 7b 0a 09 6f 77 truct Core {..ow
0230: 6e 65 72 5f 63 68 61 74 3a 20 55 73 65 72 49 64 ner_chat: UserId
0240: 2c 0a 09 70 75 62 20 74 67 3a 20 42 6f 74 2c 0a ,..pub tg: Bot,.
0250: 09 70 75 62 20 6d 79 3a 20 4d 65 2c 0a 09 70 6f .pub my: Me,..po
0260: 6f 6c 3a 20 73 71 6c 78 3a 3a 50 6f 6f 6c 3c 73 ol: sqlx::Pool<s
0270: 71 6c 78 3a 3a 50 6f 73 74 67 72 65 73 3e 2c 0a qlx::Postgres>,.
0280: 09 73 6f 75 72 63 65 73 3a 20 41 72 63 3c 4d 75 .sources: Arc<Mu
0290: 74 65 78 3c 48 61 73 68 53 65 74 3c 41 72 63 3c tex<HashSet<Arc<
02a0: 69 33 32 3e 3e 3e 3e 2c 0a 09 68 74 74 70 5f 63 i32>>>>,..http_c
02b0: 6c 69 65 6e 74 3a 20 72 65 71 77 65 73 74 3a 3a lient: reqwest::
02c0: 43 6c 69 65 6e 74 2c 0a 7d 0a 0a 69 6d 70 6c 20 Client,.}..impl
02d0: 43 6f 72 65 20 7b 0a 09 70 75 62 20 66 6e 20 6e Core {..pub fn n
02e0: 65 77 28 73 65 74 74 69 6e 67 73 3a 20 63 6f 6e ew(settings: con
02f0: 66 69 67 3a 3a 43 6f 6e 66 69 67 29 20 2d 3e 20 fig::Config) ->
0300: 52 65 73 75 6c 74 3c 41 72 63 3c 43 6f 72 65 3e Result<Arc<Core>
0310: 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a > {...let owner:
0320: 20 75 36 34 20 3d 20 73 65 74 74 69 6e 67 73 2e u64 = settings.
0330: 67 65 74 5f 69 6e 74 28 22 6f 77 6e 65 72 22 29 get_int("owner")
0340: 3f 2e 74 72 79 5f 69 6e 74 6f 28 29 3f 3b 0a 09 ?.try_into()?;..
0350: 09 6c 65 74 20 61 70 69 5f 6b 65 79 20 3d 20 73 .let api_key = s
0360: 65 74 74 69 6e 67 73 2e 67 65 74 5f 73 74 72 69 ettings.get_stri
0370: 6e 67 28 22 61 70 69 5f 6b 65 79 22 29 3f 3b 0a ng("api_key")?;.
0380: 09 09 6c 65 74 20 74 67 20 3d 20 42 6f 74 3a 3a ..let tg = Bot::
0390: 6e 65 77 28 61 70 69 5f 6b 65 79 29 3b 0a 09 09 new(api_key);...
03a0: 6c 65 74 20 74 67 5f 63 6c 6f 6e 65 64 20 3d 20 let tg_cloned =
03b0: 74 67 2e 63 6c 6f 6e 65 28 29 3b 0a 0a 09 09 6c tg.clone();....l
03c0: 65 74 20 6d 75 74 20 63 6c 69 65 6e 74 20 3d 20 et mut client =
03d0: 72 65 71 77 65 73 74 3a 3a 43 6c 69 65 6e 74 3a reqwest::Client:
03e0: 3a 62 75 69 6c 64 65 72 28 29 3b 0a 09 09 69 66 :builder();...if
03f0: 20 6c 65 74 20 4f 6b 28 70 72 6f 78 79 29 20 3d let Ok(proxy) =
0400: 20 73 65 74 74 69 6e 67 73 2e 67 65 74 5f 73 74 settings.get_st
0410: 72 69 6e 67 28 22 70 72 6f 78 79 22 29 20 7b 0a ring("proxy") {.
0420: 09 09 09 6c 65 74 20 70 72 6f 78 79 20 3d 20 72 ...let proxy = r
0430: 65 71 77 65 73 74 3a 3a 50 72 6f 78 79 3a 3a 61 eqwest::Proxy::a
0440: 6c 6c 28 70 72 6f 78 79 29 3f 3b 0a 09 09 09 63 ll(proxy)?;....c
0450: 6c 69 65 6e 74 20 3d 20 63 6c 69 65 6e 74 2e 70 lient = client.p
0460: 72 6f 78 79 28 70 72 6f 78 79 29 3b 0a 09 09 7d roxy(proxy);...}
0470: 0a 09 09 6c 65 74 20 68 74 74 70 5f 63 6c 69 65 ...let http_clie
0480: 6e 74 20 3d 20 63 6c 69 65 6e 74 2e 62 75 69 6c nt = client.buil
0490: 64 28 29 3f 3b 0a 09 09 6c 65 74 20 63 6f 72 65 d()?;...let core
04a0: 20 3d 20 41 72 63 3a 3a 6e 65 77 28 43 6f 72 65 = Arc::new(Core
04b0: 20 7b 0a 09 09 09 74 67 2c 0a 09 09 09 6d 79 3a {....tg,....my:
04c0: 20 74 61 73 6b 3a 3a 62 6c 6f 63 6b 5f 6f 6e 28 task::block_on(
04d0: 61 73 79 6e 63 20 7b 0a 09 09 09 09 74 67 5f 63 async {.....tg_c
04e0: 6c 6f 6e 65 64 2e 67 65 74 5f 6d 65 28 29 2e 61 loned.get_me().a
04f0: 77 61 69 74 0a 09 09 09 7d 29 3f 2c 0a 09 09 09 wait....})?,....
0500: 6f 77 6e 65 72 5f 63 68 61 74 3a 20 55 73 65 72 owner_chat: User
0510: 49 64 28 6f 77 6e 65 72 29 2c 0a 09 09 09 70 6f Id(owner),....po
0520: 6f 6c 3a 20 50 67 50 6f 6f 6c 4f 70 74 69 6f 6e ol: PgPoolOption
0530: 73 3a 3a 6e 65 77 28 29 0a 09 09 09 09 2e 6d 61 s::new()......ma
0540: 78 5f 63 6f 6e 6e 65 63 74 69 6f 6e 73 28 35 29 x_connections(5)
0550: 0a 09 09 09 09 2e 61 63 71 75 69 72 65 5f 74 69 ......acquire_ti
0560: 6d 65 6f 75 74 28 73 74 64 3a 3a 74 69 6d 65 3a meout(std::time:
0570: 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 33 :Duration::new(3
0580: 30 30 2c 20 30 29 29 0a 09 09 09 09 2e 69 64 6c 00, 0))......idl
0590: 65 5f 74 69 6d 65 6f 75 74 28 73 74 64 3a 3a 74 e_timeout(std::t
05a0: 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e ime::Duration::n
05b0: 65 77 28 36 30 2c 20 30 29 29 0a 09 09 09 09 2e ew(60, 0))......
05c0: 63 6f 6e 6e 65 63 74 5f 6c 61 7a 79 28 26 73 65 connect_lazy(&se
05d0: 74 74 69 6e 67 73 2e 67 65 74 5f 73 74 72 69 6e ttings.get_strin
05e0: 67 28 22 70 67 22 29 3f 29 3f 2c 0a 09 09 09 73 g("pg")?)?,....s
05f0: 6f 75 72 63 65 73 3a 20 41 72 63 3a 3a 6e 65 77 ources: Arc::new
0600: 28 4d 75 74 65 78 3a 3a 6e 65 77 28 48 61 73 68 (Mutex::new(Hash
0610: 53 65 74 3a 3a 6e 65 77 28 29 29 29 2c 0a 09 09 Set::new())),...
0620: 09 68 74 74 70 5f 63 6c 69 65 6e 74 2c 0a 09 09 .http_client,...
0630: 7d 29 3b 0a 09 09 2f 2a 20 6c 65 74 20 63 6c 6f });.../* let clo
0640: 6e 65 20 3d 20 63 6f 72 65 2e 63 6c 6f 6e 65 28 ne = core.clone(
0650: 29 3b 0a 09 09 74 61 73 6b 3a 3a 73 70 61 77 6e );...task::spawn
0660: 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 (async move {...
0670: 09 6c 6f 6f 70 20 7b 0a 09 09 09 09 6c 65 74 20 .loop {.....let
0680: 64 65 6c 61 79 20 3d 20 6d 61 74 63 68 20 26 63 delay = match &c
0690: 6c 6f 6e 65 2e 61 75 74 6f 66 65 74 63 68 28 29 lone.autofetch()
06a0: 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 45 72 .await {......Er
06b0: 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 r(err) => {.....
06c0: 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 ..if let Err(err
06d0: 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 66 ) = clone.send(f
06e0: 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b 3a 3f ormat!("š {:?
06f0: 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 2c 20 }", err), None,
0700: 4e 6f 6e 65 29 2e 61 77 61 69 74 20 7b 0a 09 09 None).await {...
0710: 09 09 09 09 09 65 70 72 69 6e 74 6c 6e 21 28 22 .....eprintln!("
0720: 41 75 74 6f 66 65 74 63 68 20 65 72 72 6f 72 3a Autofetch error:
0730: 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 09 {}", err);.....
0740: 09 09 7d 3b 0a 09 09 09 09 09 09 73 74 64 3a 3a ..};.......std::
0750: 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a time::Duration::
0760: 66 72 6f 6d 5f 73 65 63 73 28 36 30 29 0a 09 09 from_secs(60)...
0770: 09 09 09 7d 2c 0a 09 09 09 09 09 4f 6b 28 74 69 ...},......Ok(ti
0780: 6d 65 29 20 3d 3e 20 2a 74 69 6d 65 2c 0a 09 09 me) => *time,...
0790: 09 09 7d 3b 0a 09 09 09 09 74 61 73 6b 3a 3a 73 ..};.....task::s
07a0: 6c 65 65 70 28 64 65 6c 61 79 29 2e 61 77 61 69 leep(delay).awai
07b0: 74 3b 0a 09 09 09 7d 0a 09 09 7d 29 3b 20 2a 2f t;....}...}); */
07c0: 0a 09 09 4f 6b 28 63 6f 72 65 29 0a 09 7d 0a 0a ...Ok(core)..}..
07d0: 09 70 75 62 20 66 6e 20 73 74 72 65 61 6d 28 26 .pub fn stream(&
07e0: 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c self) -> Result<
07f0: 28 29 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 ()> {...let mut
0800: 6c 61 73 74 5f 75 70 64 61 74 65 3a 20 4f 70 74 last_update: Opt
0810: 69 6f 6e 3c 69 33 32 3e 20 3d 20 4e 6f 6e 65 3b ion<i32> = None;
0820: 0a 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 6c 65 74 ...loop {....let
0830: 20 75 70 64 61 74 65 73 20 3d 20 73 65 6c 66 2e updates = self.
0840: 74 67 2e 67 65 74 5f 75 70 64 61 74 65 73 28 6c tg.get_updates(l
0850: 61 73 74 5f 75 70 64 61 74 65 2c 20 4e 6f 6e 65 ast_update, None
0860: 2c 20 33 30 30 2c 20 53 6f 6d 65 28 76 65 63 21 , 300, Some(vec!
0870: 5b 22 6d 65 73 73 61 67 65 22 5d 29 29 3b 0a 09 ["message"]));..
0880: 09 7d 0a 09 09 4f 6b 28 28 29 29 0a 09 7d 0a 0a .}...Ok(())..}..
0890: 09 2f 2a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 ./*..pub async f
08a0: 6e 20 73 65 6e 64 3c 27 61 2c 20 53 3e 28 26 73 n send<'a, S>(&s
08b0: 65 6c 66 2c 20 6d 73 67 3a 20 53 2c 20 74 61 72 elf, msg: S, tar
08c0: 67 65 74 3a 20 4f 70 74 69 6f 6e 3c 74 65 6c 65 get: Option<tele
08d0: 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 gram_bot::UserId
08e0: 3e 2c 20 6d 6f 64 65 3a 20 4f 70 74 69 6f 6e 3c >, mode: Option<
08f0: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 telegram_bot::ty
0900: 70 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3e 29 pes::ParseMode>)
0910: 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 0a 09 -> Result<()>..
0920: 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 43 6f where S: Into<Co
0930: 77 3c 27 61 2c 20 73 74 72 3e 3e 20 7b 0a 09 09 w<'a, str>> {...
0940: 6c 65 74 20 6d 6f 64 65 20 3d 20 6d 6f 64 65 2e let mode = mode.
0950: 75 6e 77 72 61 70 5f 6f 72 28 74 65 6c 65 67 72 unwrap_or(telegr
0960: 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 am_bot::types::P
0970: 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 3b arseMode::Html);
0980: 0a 09 09 6c 65 74 20 74 61 72 67 65 74 20 3d 20 ...let target =
0990: 74 61 72 67 65 74 2e 75 6e 77 72 61 70 5f 6f 72 target.unwrap_or
09a0: 28 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 61 74 (self.owner_chat
09b0: 29 3b 0a 09 09 73 65 6c 66 2e 72 65 71 75 65 73 );...self.reques
09c0: 74 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a t(telegram_bot::
09d0: 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 SendMessage::new
09e0: 28 74 61 72 67 65 74 2c 20 6d 73 67 29 2e 70 61 (target, msg).pa
09f0: 72 73 65 5f 6d 6f 64 65 28 6d 6f 64 65 29 29 2e rse_mode(mode)).
0a00: 61 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 28 29 29 await?;...Ok(())
0a10: 0a 09 7d 20 2a 2f 0a 0a 09 2f 2a 20 70 75 62 20 ..} */.../* pub
0a20: 61 73 79 6e 63 20 66 6e 20 72 65 71 75 65 73 74 async fn request
0a30: 3c 52 65 71 3a 20 74 65 6c 65 67 72 61 6d 5f 62 <Req: telegram_b
0a40: 6f 74 3a 3a 52 65 71 75 65 73 74 3e 20 28 26 73 ot::Request> (&s
0a50: 65 6c 66 2c 20 72 65 71 3a 20 52 65 71 29 20 2d elf, req: Req) -
0a60: 3e 20 52 65 73 75 6c 74 3c 3c 52 65 71 3a 3a 52 > Result<<Req::R
0a70: 65 73 70 6f 6e 73 65 20 61 73 20 74 65 6c 65 67 esponse as teleg
0a80: 72 61 6d 5f 62 6f 74 3a 3a 52 65 73 70 6f 6e 73 ram_bot::Respons
0a90: 65 54 79 70 65 3e 3a 3a 54 79 70 65 2c 20 52 73 eType>::Type, Rs
0aa0: 73 45 72 72 6f 72 3e 20 7b 0a 09 09 6c 6f 6f 70 sError> {...loop
0ab0: 20 7b 0a 09 09 09 6c 65 74 20 72 65 73 20 3d 20 {....let res =
0ac0: 73 65 6c 66 2e 74 67 2e 73 65 6e 64 28 26 72 65 self.tg.send(&re
0ad0: 71 29 2e 61 77 61 69 74 3b 0a 09 09 09 6d 61 74 q).await;....mat
0ae0: 63 68 20 72 65 73 20 7b 0a 09 09 09 09 4f 6b 28 ch res {.....Ok(
0af0: 5f 29 20 3d 3e 20 72 65 74 75 72 6e 20 4f 6b 28 _) => return Ok(
0b00: 72 65 73 3f 29 2c 0a 09 09 09 09 45 72 72 28 65 res?),.....Err(e
0b10: 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 6d 61 rr) => {......ma
0b20: 74 63 68 20 26 65 72 72 20 7b 0a 09 09 09 09 09 tch &err {......
0b30: 09 54 67 45 72 72 6f 72 3a 3a 52 61 77 28 54 67 .TgError::Raw(Tg
0b40: 72 45 72 72 6f 72 3a 3a 54 65 6c 65 67 72 61 6d rError::Telegram
0b50: 45 72 72 6f 72 20 7b 20 64 65 73 63 72 69 70 74 Error { descript
0b60: 69 6f 6e 3a 20 5f 2c 20 70 61 72 61 6d 65 74 65 ion: _, paramete
0b70: 72 73 3a 20 53 6f 6d 65 28 70 61 72 61 6d 73 29 rs: Some(params)
0b80: 20 7d 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 }) => {........
0b90: 69 66 20 6c 65 74 20 53 6f 6d 65 28 64 65 6c 61 if let Some(dela
0ba0: 79 29 20 3d 20 70 61 72 61 6d 73 2e 72 65 74 72 y) = params.retr
0bb0: 79 5f 61 66 74 65 72 20 7b 0a 09 09 09 09 09 09 y_after {.......
0bc0: 09 09 70 72 69 6e 74 6c 6e 21 28 22 54 68 72 6f ..println!("Thro
0bd0: 74 74 6c 65 64 2c 20 77 61 69 74 69 6e 67 20 7b ttled, waiting {
0be0: 7d 20 73 65 6e 63 6f 6e 64 73 2e 22 2c 20 64 65 } senconds.", de
0bf0: 6c 61 79 29 3b 0a 09 09 09 09 09 09 09 09 74 61 lay);.........ta
0c00: 73 6b 3a 3a 73 6c 65 65 70 28 73 74 64 3a 3a 74 sk::sleep(std::t
0c10: 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 66 ime::Duration::f
0c20: 72 6f 6d 5f 73 65 63 73 28 64 65 6c 61 79 2e 74 rom_secs(delay.t
0c30: 72 79 5f 69 6e 74 6f 28 29 3f 29 29 2e 61 77 61 ry_into()?)).awa
0c40: 69 74 3b 0a 09 09 09 09 09 09 09 7d 20 65 6c 73 it;........} els
0c50: 65 20 7b 0a 09 09 09 09 09 09 09 09 72 65 74 75 e {.........retu
0c60: 72 6e 20 45 72 72 28 65 72 72 2e 69 6e 74 6f 28 rn Err(err.into(
0c70: 29 29 3b 0a 09 09 09 09 09 09 09 7d 0a 09 09 09 ));........}....
0c80: 09 09 09 7d 2c 0a 09 09 09 09 09 09 5f 20 3d 3e ...},......._ =>
0c90: 20 72 65 74 75 72 6e 20 45 72 72 28 65 72 72 2e return Err(err.
0ca0: 69 6e 74 6f 28 29 29 2c 0a 09 09 09 09 09 7d 0a into()),......}.
0cb0: 09 09 09 09 7d 2c 0a 09 09 09 7d 3b 0a 09 09 7d ....},....};...}
0cc0: 0a 09 7d 20 2a 2f 0a 0a 09 2f 2a 20 70 75 62 20 ..} */.../* pub
0cd0: 61 73 79 6e 63 20 66 6e 20 63 68 65 63 6b 3c 53 async fn check<S
0ce0: 3e 28 26 73 65 6c 66 2c 20 69 64 3a 20 26 69 33 >(&self, id: &i3
0cf0: 32 2c 20 6f 77 6e 65 72 3a 20 53 2c 20 72 65 61 2, owner: S, rea
0d00: 6c 3a 20 62 6f 6f 6c 29 20 2d 3e 20 52 65 73 75 l: bool) -> Resu
0d10: 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e lt<Cow<'_, str>>
0d20: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
0d30: 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e i64> {...let own
0d40: 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 er = owner.into(
0d50: 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 );...let mut pos
0d60: 74 65 64 3a 20 69 33 32 20 3d 20 30 3b 0a 09 09 ted: i32 = 0;...
0d70: 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 let mut conn = s
0d80: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
0d90: 28 29 2e 61 77 61 69 74 3f 3b 0a 0a 09 09 6c 65 ().await?;....le
0da0: 74 20 69 64 20 3d 20 7b 0a 09 09 09 6c 65 74 20 t id = {....let
0db0: 6d 75 74 20 73 65 74 20 3d 20 73 65 6c 66 2e 73 mut set = self.s
0dc0: 6f 75 72 63 65 73 2e 6c 6f 63 6b 28 29 2e 75 6e ources.lock().un
0dd0: 77 72 61 70 28 29 3b 0a 09 09 09 6d 61 74 63 68 wrap();....match
0de0: 20 73 65 74 2e 67 65 74 28 69 64 29 20 7b 0a 09 set.get(id) {..
0df0: 09 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 69 ...Some(id) => i
0e00: 64 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 09 4e d.clone(),.....N
0e10: 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 6c 65 one => {......le
0e20: 74 20 69 64 20 3d 20 41 72 63 3a 3a 6e 65 77 28 t id = Arc::new(
0e30: 2a 69 64 29 3b 0a 09 09 09 09 09 73 65 74 2e 69 *id);......set.i
0e40: 6e 73 65 72 74 28 69 64 2e 63 6c 6f 6e 65 28 29 nsert(id.clone()
0e50: 29 3b 0a 09 09 09 09 09 69 64 2e 63 6c 6f 6e 65 );......id.clone
0e60: 28 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 ().....},....}..
0e70: 09 7d 3b 0a 09 09 6c 65 74 20 63 6f 75 6e 74 20 .};...let count
0e80: 3d 20 41 72 63 3a 3a 73 74 72 6f 6e 67 5f 63 6f = Arc::strong_co
0e90: 75 6e 74 28 26 69 64 29 3b 0a 09 09 69 66 20 63 unt(&id);...if c
0ea0: 6f 75 6e 74 20 3d 3d 20 32 20 7b 0a 09 09 09 6c ount == 2 {....l
0eb0: 65 74 20 73 6f 75 72 63 65 20 3d 20 73 71 6c 78 et source = sqlx
0ec0: 3a 3a 71 75 65 72 79 21 28 22 73 65 6c 65 63 74 ::query!("select
0ed0: 20 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e source_id, chan
0ee0: 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f nel_id, url, iv_
0ef0: 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 75 72 6c hash, owner, url
0f00: 5f 72 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 _re from rsstg_s
0f10: 6f 75 72 63 65 20 77 68 65 72 65 20 73 6f 75 72 ource where sour
0f20: 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f ce_id = $1 and o
0f30: 77 6e 65 72 20 3d 20 24 32 22 2c 0a 09 09 09 09 wner = $2",.....
0f40: 2a 69 64 2c 20 6f 77 6e 65 72 29 2e 66 65 74 63 *id, owner).fetc
0f50: 68 5f 6f 6e 65 28 26 6d 75 74 20 2a 63 6f 6e 6e h_one(&mut *conn
0f60: 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 6c 65 74 ).await?;....let
0f70: 20 64 65 73 74 69 6e 61 74 69 6f 6e 20 3d 20 6d destination = m
0f80: 61 74 63 68 20 72 65 61 6c 20 7b 0a 09 09 09 09 atch real {.....
0f90: 74 72 75 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d true => telegram
0fa0: 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 _bot::UserId::ne
0fb0: 77 28 73 6f 75 72 63 65 2e 63 68 61 6e 6e 65 6c w(source.channel
0fc0: 5f 69 64 29 2c 0a 09 09 09 09 66 61 6c 73 65 20 _id),.....false
0fd0: 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a => telegram_bot:
0fe0: 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 73 6f 75 :UserId::new(sou
0ff0: 72 63 65 2e 6f 77 6e 65 72 29 2c 0a 09 09 09 7d rce.owner),....}
1000: 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 74 68 69 ;....let mut thi
1010: 73 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e 3c s_fetch: Option<
1020: 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a DateTime<chrono:
1030: 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 3e 20 3d :FixedOffset>> =
1040: 20 4e 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 6d 75 None;....let mu
1050: 74 20 70 6f 73 74 73 3a 20 42 54 72 65 65 4d 61 t posts: BTreeMa
1060: 70 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e p<DateTime<chron
1070: 6f 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 2c o::FixedOffset>,
1080: 20 53 74 72 69 6e 67 3e 20 3d 20 42 54 72 65 65 String> = BTree
1090: 4d 61 70 3a 3a 6e 65 77 28 29 3b 0a 0a 09 09 09 Map::new();.....
10a0: 6c 65 74 20 72 65 73 70 6f 6e 73 65 20 3d 20 73 let response = s
10b0: 65 6c 66 2e 68 74 74 70 5f 63 6c 69 65 6e 74 2e elf.http_client.
10c0: 67 65 74 28 26 73 6f 75 72 63 65 2e 75 72 6c 29 get(&source.url)
10d0: 2e 73 65 6e 64 28 29 2e 61 77 61 69 74 3f 3b 0a .send().await?;.
10e0: 09 09 09 6c 65 74 20 73 74 61 74 75 73 20 3d 20 ...let status =
10f0: 72 65 73 70 6f 6e 73 65 2e 73 74 61 74 75 73 28 response.status(
1100: 29 3b 0a 09 09 09 6c 65 74 20 63 6f 6e 74 65 6e );....let conten
1110: 74 20 3d 20 72 65 73 70 6f 6e 73 65 2e 62 79 74 t = response.byt
1120: 65 73 28 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 es().await?;....
1130: 6d 61 74 63 68 20 72 73 73 3a 3a 43 68 61 6e 6e match rss::Chann
1140: 65 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 el::read_from(&c
1150: 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 09 ontent[..]) {...
1160: 09 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 7b 0a ..Ok(feed) => {.
1170: 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 69 6e .....for item in
1180: 20 66 65 65 64 2e 69 74 65 6d 73 28 29 20 7b 0a feed.items() {.
1190: 09 09 09 09 09 09 69 66 20 6c 65 74 20 53 6f 6d ......if let Som
11a0: 65 28 6c 69 6e 6b 29 20 3d 20 69 74 65 6d 2e 6c e(link) = item.l
11b0: 69 6e 6b 28 29 20 7b 0a 09 09 09 09 09 09 09 6c ink() {........l
11c0: 65 74 20 64 61 74 65 20 3d 20 6d 61 74 63 68 20 et date = match
11d0: 69 74 65 6d 2e 70 75 62 5f 64 61 74 65 28 29 20 item.pub_date()
11e0: 7b 0a 09 09 09 09 09 09 09 09 53 6f 6d 65 28 66 {.........Some(f
11f0: 65 65 64 5f 64 61 74 65 29 20 3d 3e 20 44 61 74 eed_date) => Dat
1200: 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f eTime::parse_fro
1210: 6d 5f 72 66 63 32 38 32 32 28 66 65 65 64 5f 64 m_rfc2822(feed_d
1220: 61 74 65 29 2c 0a 09 09 09 09 09 09 09 09 4e 6f ate),.........No
1230: 6e 65 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a ne => DateTime::
1240: 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 33 33 parse_from_rfc33
1250: 33 39 28 26 69 74 65 6d 2e 64 75 62 6c 69 6e 5f 39(&item.dublin_
1260: 63 6f 72 65 5f 65 78 74 28 29 2e 75 6e 77 72 61 core_ext().unwra
1270: 70 28 29 2e 64 61 74 65 73 28 29 5b 30 5d 29 2c p().dates()[0]),
1280: 0a 09 09 09 09 09 09 09 7d 3f 3b 0a 09 09 09 09 ........}?;.....
1290: 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 6c 69 6e ...let url = lin
12a0: 6b 3b 0a 09 09 09 09 09 09 09 70 6f 73 74 73 2e k;........posts.
12b0: 69 6e 73 65 72 74 28 64 61 74 65 2c 20 75 72 6c insert(date, url
12c0: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 .to_string());..
12d0: 09 09 09 09 09 7d 0a 09 09 09 09 09 7d 3b 0a 09 .....}......};..
12e0: 09 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 65 72 ...},.....Err(er
12f0: 72 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 72 20 r) => match err
1300: 7b 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 6f {......rss::Erro
1310: 72 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 74 54 r::InvalidStartT
1320: 61 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 6d 61 ag => {.......ma
1330: 74 63 68 20 61 74 6f 6d 5f 73 79 6e 64 69 63 61 tch atom_syndica
1340: 74 69 6f 6e 3a 3a 46 65 65 64 3a 3a 72 65 61 64 tion::Feed::read
1350: 5f 66 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e _from(&content[.
1360: 2e 5d 29 20 7b 0a 09 09 09 09 09 09 09 4f 6b 28 .]) {........Ok(
1370: 66 65 65 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 feed) => {......
1380: 09 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 ...for item in f
1390: 65 65 64 2e 65 6e 74 72 69 65 73 28 29 20 7b 0a eed.entries() {.
13a0: 09 09 09 09 09 09 09 09 09 6c 65 74 20 64 61 74 .........let dat
13b0: 65 20 3d 20 69 74 65 6d 2e 70 75 62 6c 69 73 68 e = item.publish
13c0: 65 64 28 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 ed().unwrap();..
13d0: 09 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c 20 ........let url
13e0: 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 28 29 5b 30 = item.links()[0
13f0: 5d 2e 68 72 65 66 28 29 3b 0a 09 09 09 09 09 09 ].href();.......
1400: 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 ...posts.insert(
1410: 2a 64 61 74 65 2c 20 75 72 6c 2e 74 6f 5f 73 74 *date, url.to_st
1420: 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 ring());........
1430: 09 7d 3b 0a 09 09 09 09 09 09 09 7d 2c 0a 09 09 .};........},...
1440: 09 09 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e .....Err(err) =>
1450: 20 7b 0a 09 09 09 09 09 09 09 09 62 61 69 6c 21 {.........bail!
1460: 28 22 55 6e 73 75 70 70 6f 72 74 65 64 20 6f 72 ("Unsupported or
1470: 20 6d 61 6e 67 6c 65 64 20 63 6f 6e 74 65 6e 74 mangled content
1480: 3a 5c 6e 7b 3a 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e :\n{:?}\n{:#?}\n
1490: 7b 3a 23 3f 7d 5c 6e 22 2c 20 26 73 6f 75 72 63 {:#?}\n", &sourc
14a0: 65 2e 75 72 6c 2c 20 65 72 72 2c 20 73 74 61 74 e.url, err, stat
14b0: 75 73 29 0a 09 09 09 09 09 09 09 7d 2c 0a 09 09 us)........},...
14c0: 09 09 09 09 7d 0a 09 09 09 09 09 7d 2c 0a 09 09 ....}......},...
14d0: 09 09 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 ...rss::Error::E
14e0: 6f 66 20 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f of => (),......_
14f0: 20 3d 3e 20 62 61 69 6c 21 28 22 55 6e 73 75 70 => bail!("Unsup
1500: 70 6f 72 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 ported or mangle
1510: 64 20 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d d content:\n{:?}
1520: 5c 6e 7b 3a 23 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e \n{:#?}\n{:#?}\n
1530: 22 2c 20 26 73 6f 75 72 63 65 2e 75 72 6c 2c 20 ", &source.url,
1540: 65 72 72 2c 20 73 74 61 74 75 73 29 0a 09 09 09 err, status)....
1550: 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 66 6f 72 20 .}....};....for
1560: 28 64 61 74 65 2c 20 75 72 6c 29 20 69 6e 20 70 (date, url) in p
1570: 6f 73 74 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 osts.iter() {...
1580: 09 09 6c 65 74 20 70 6f 73 74 5f 75 72 6c 3a 20 ..let post_url:
1590: 43 6f 77 3c 73 74 72 3e 20 3d 20 6d 61 74 63 68 Cow<str> = match
15a0: 20 73 6f 75 72 63 65 2e 75 72 6c 5f 72 65 20 7b source.url_re {
15b0: 0a 09 09 09 09 09 53 6f 6d 65 28 72 65 66 20 78 ......Some(ref x
15c0: 29 20 3d 3e 20 73 65 64 72 65 67 65 78 3a 3a 52 ) => sedregex::R
15d0: 65 70 6c 61 63 65 43 6f 6d 6d 61 6e 64 3a 3a 6e eplaceCommand::n
15e0: 65 77 28 78 29 3f 2e 65 78 65 63 75 74 65 28 75 ew(x)?.execute(u
15f0: 72 6c 29 2c 0a 09 09 09 09 09 4e 6f 6e 65 20 3d rl),......None =
1600: 3e 20 75 72 6c 2e 69 6e 74 6f 28 29 2c 0a 09 09 > url.into(),...
1610: 09 09 7d 3b 0a 09 09 09 09 69 66 20 6c 65 74 20 ..};.....if let
1620: 53 6f 6d 65 28 65 78 69 73 74 73 29 20 3d 20 73 Some(exists) = s
1630: 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 73 65 6c qlx::query!("sel
1640: 65 63 74 20 65 78 69 73 74 73 28 73 65 6c 65 63 ect exists(selec
1650: 74 20 74 72 75 65 20 66 72 6f 6d 20 72 73 73 74 t true from rsst
1660: 67 5f 70 6f 73 74 20 77 68 65 72 65 20 75 72 6c g_post where url
1670: 20 3d 20 24 31 20 61 6e 64 20 73 6f 75 72 63 65 = $1 and source
1680: 5f 69 64 20 3d 20 24 32 29 20 61 73 20 65 78 69 _id = $2) as exi
1690: 73 74 73 3b 22 2c 0a 09 09 09 09 09 26 70 6f 73 sts;",......&pos
16a0: 74 5f 75 72 6c 2c 20 2a 69 64 29 2e 66 65 74 63 t_url, *id).fetc
16b0: 68 5f 6f 6e 65 28 26 6d 75 74 20 2a 63 6f 6e 6e h_one(&mut *conn
16c0: 29 2e 61 77 61 69 74 3f 2e 65 78 69 73 74 73 20 ).await?.exists
16d0: 7b 0a 09 09 09 09 09 69 66 20 21 20 65 78 69 73 {......if ! exis
16e0: 74 73 20 7b 0a 09 09 09 09 09 09 69 66 20 74 68 ts {.......if th
16f0: 69 73 5f 66 65 74 63 68 2e 69 73 5f 6e 6f 6e 65 is_fetch.is_none
1700: 28 29 20 7c 7c 20 2a 64 61 74 65 20 3e 20 74 68 () || *date > th
1710: 69 73 5f 66 65 74 63 68 2e 75 6e 77 72 61 70 28 is_fetch.unwrap(
1720: 29 20 7b 0a 09 09 09 09 09 09 09 74 68 69 73 5f ) {........this_
1730: 66 65 74 63 68 20 3d 20 53 6f 6d 65 28 2a 64 61 fetch = Some(*da
1740: 74 65 29 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 te);.......};...
1750: 09 09 09 09 73 65 6c 66 2e 72 65 71 75 65 73 74 ....self.request
1760: 28 20 6d 61 74 63 68 20 26 73 6f 75 72 63 65 2e ( match &source.
1770: 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 iv_hash {.......
1780: 09 09 53 6f 6d 65 28 68 61 73 68 29 20 3d 3e 20 ..Some(hash) =>
1790: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 telegram_bot::Se
17a0: 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 ndMessage::new(d
17b0: 65 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d estination, form
17c0: 61 74 21 28 22 3c 61 20 68 72 65 66 3d 5c 22 68 at!("<a href=\"h
17d0: 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 ttps://t.me/iv?u
17e0: 72 6c 3d 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c 22 rl={}&rhash={}\"
17f0: 3e 20 3c 2f 61 3e 7b 30 7d 22 2c 20 26 70 6f 73 > </a>{0}", &pos
1800: 74 5f 75 72 6c 2c 20 68 61 73 68 29 29 2c 0a 09 t_url, hash)),..
1810: 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 .......None => t
1820: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e elegram_bot::Sen
1830: 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 dMessage::new(de
1840: 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 stination, forma
1850: 74 21 28 22 7b 7d 22 2c 20 70 6f 73 74 5f 75 72 t!("{}", post_ur
1860: 6c 29 29 2c 0a 09 09 09 09 09 09 09 7d 2e 70 61 l)),........}.pa
1870: 72 73 65 5f 6d 6f 64 65 28 74 65 6c 65 67 72 61 rse_mode(telegra
1880: 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 61 m_bot::types::Pa
1890: 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 2e rseMode::Html)).
18a0: 61 77 61 69 74 0a 09 09 09 09 09 09 09 2e 63 6f await.........co
18b0: 6e 74 65 78 74 28 22 43 61 6e 27 74 20 70 6f 73 ntext("Can't pos
18c0: 74 20 6d 65 73 73 61 67 65 3a 22 29 3f 3b 0a 09 t message:")?;..
18d0: 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 .....sqlx::query
18e0: 21 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 !("insert into r
18f0: 73 73 74 67 5f 70 6f 73 74 20 28 73 6f 75 72 63 sstg_post (sourc
1900: 65 5f 69 64 2c 20 70 6f 73 74 65 64 2c 20 75 72 e_id, posted, ur
1910: 6c 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 l) values ($1, $
1920: 32 2c 20 24 33 29 3b 22 2c 0a 09 09 09 09 09 09 2, $3);",.......
1930: 09 2a 69 64 2c 20 64 61 74 65 2c 20 26 70 6f 73 .*id, date, &pos
1940: 74 5f 75 72 6c 29 2e 65 78 65 63 75 74 65 28 26 t_url).execute(&
1950: 6d 75 74 20 2a 63 6f 6e 6e 29 2e 61 77 61 69 74 mut *conn).await
1960: 3f 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 09 09 7d ?;......};.....}
1970: 3b 0a 09 09 09 09 70 6f 73 74 65 64 20 2b 3d 20 ;.....posted +=
1980: 31 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 6f 73 74 1;....};....post
1990: 73 2e 63 6c 65 61 72 28 29 3b 0a 09 09 7d 3b 0a s.clear();...};.
19a0: 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 ..sqlx::query!("
19b0: 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 update rsstg_sou
19c0: 72 63 65 20 73 65 74 20 6c 61 73 74 5f 73 63 72 rce set last_scr
19d0: 61 70 65 20 3d 20 6e 6f 77 28 29 20 77 68 65 72 ape = now() wher
19e0: 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 e source_id = $1
19f0: 3b 22 2c 0a 09 09 09 2a 69 64 29 2e 65 78 65 63 ;",....*id).exec
1a00: 75 74 65 28 26 6d 75 74 20 2a 63 6f 6e 6e 29 2e ute(&mut *conn).
1a10: 61 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 66 6f 72 await?;...Ok(for
1a20: 6d 61 74 21 28 22 50 6f 73 74 65 64 3a 20 7b 7d mat!("Posted: {}
1a30: 22 2c 20 26 70 6f 73 74 65 64 29 2e 69 6e 74 6f ", &posted).into
1a40: 28 29 29 0a 09 7d 20 2a 2f 0a 0a 09 2f 2a 20 70 ())..} */.../* p
1a50: 75 62 20 61 73 79 6e 63 20 66 6e 20 64 65 6c 65 ub async fn dele
1a60: 74 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 te<S>(&self, sou
1a70: 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 rce_id: &i32, ow
1a80: 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c ner: S) -> Resul
1a90: 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a t<Cow<'_, str>>.
1aa0: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
1ab0: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
1ac0: 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 r = owner.into()
1ad0: 3b 0a 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a ;....match sqlx:
1ae0: 3a 71 75 65 72 79 21 28 22 64 65 6c 65 74 65 20 :query!("delete
1af0: 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 from rsstg_sourc
1b00: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
1b10: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
1b20: 20 3d 20 24 32 3b 22 2c 0a 09 09 09 73 6f 75 72 = $2;",....sour
1b30: 63 65 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 65 78 ce_id, owner).ex
1b40: 65 63 75 74 65 28 26 6d 75 74 20 2a 73 65 6c 66 ecute(&mut *self
1b50: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
1b60: 61 77 61 69 74 3f 29 2e 61 77 61 69 74 3f 2e 72 await?).await?.r
1b70: 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b ows_affected() {
1b80: 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e ....0 => { Ok("N
1b90: 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 o data found fou
1ba0: 6e 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a nd.".into()) },.
1bb0: 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 ...x => { Ok(for
1bc0: 6d 61 74 21 28 22 7b 7d 20 73 6f 75 72 63 65 73 mat!("{} sources
1bd0: 20 72 65 6d 6f 76 65 64 2e 22 2c 20 78 29 2e 69 removed.", x).i
1be0: 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d nto()) },...}..}
1bf0: 20 2a 2f 0a 0a 09 2f 2a 20 70 75 62 20 61 73 79 */.../* pub asy
1c00: 6e 63 20 66 6e 20 63 6c 65 61 6e 3c 53 3e 28 26 nc fn clean<S>(&
1c10: 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a self, source_id:
1c20: 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 &i32, owner: S)
1c30: 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 3c 27 -> Result<Cow<'
1c40: 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 65 20 _, str>>..where
1c50: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
1c60: 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e .let owner = own
1c70: 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6d 61 er.into();....ma
1c80: 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 21 tch sqlx::query!
1c90: 28 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 ("delete from rs
1ca0: 73 74 67 5f 70 6f 73 74 20 70 20 75 73 69 6e 67 stg_post p using
1cb0: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 20 rsstg_source s
1cc0: 77 68 65 72 65 20 70 2e 73 6f 75 72 63 65 5f 69 where p.source_i
1cd0: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
1ce0: 20 3d 20 24 32 20 61 6e 64 20 70 2e 73 6f 75 72 = $2 and p.sour
1cf0: 63 65 5f 69 64 20 3d 20 73 2e 73 6f 75 72 63 65 ce_id = s.source
1d00: 5f 69 64 3b 22 2c 0a 09 09 09 73 6f 75 72 63 65 _id;",....source
1d10: 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 65 78 65 63 _id, owner).exec
1d20: 75 74 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e 70 ute(&mut *self.p
1d30: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
1d40: 61 69 74 3f 29 2e 61 77 61 69 74 3f 2e 72 6f 77 ait?).await?.row
1d50: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
1d60: 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 ..0 => { Ok("No
1d70: 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 data found found
1d80: 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 .".into()) },...
1d90: 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 .x => { Ok(forma
1da0: 74 21 28 22 7b 7d 20 70 6f 73 74 73 20 70 75 72 t!("{} posts pur
1db0: 67 65 64 2e 22 2c 20 78 29 2e 69 6e 74 6f 28 29 ged.", x).into()
1dc0: 29 20 7d 2c 0a 09 09 7d 0a 09 7d 20 2a 2f 0a 0a ) },...}..} */..
1dd0: 09 2f 2a 20 70 75 62 20 61 73 79 6e 63 20 66 6e ./* pub async fn
1de0: 20 65 6e 61 62 6c 65 3c 53 3e 28 26 73 65 6c 66 enable<S>(&self
1df0: 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 , source_id: &i3
1e00: 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 2, owner: S) ->
1e10: 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 Result<&str>..wh
1e20: 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e ere S: Into<i64>
1e30: 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d {...let owner =
1e40: 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a owner.into();..
1e50: 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 ..match sqlx::qu
1e60: 65 72 79 21 28 22 75 70 64 61 74 65 20 72 73 73 ery!("update rss
1e70: 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 6e tg_source set en
1e80: 61 62 6c 65 64 20 3d 20 74 72 75 65 20 77 68 65 abled = true whe
1e90: 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 re source_id = $
1ea0: 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 1 and owner = $2
1eb0: 22 2c 0a 09 09 09 73 6f 75 72 63 65 5f 69 64 2c ",....source_id,
1ec0: 20 6f 77 6e 65 72 29 2e 65 78 65 63 75 74 65 28 owner).execute(
1ed0: 26 6d 75 74 20 2a 73 65 6c 66 2e 70 6f 6f 6c 2e &mut *self.pool.
1ee0: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 3f acquire().await?
1ef0: 29 2e 61 77 61 69 74 3f 2e 72 6f 77 73 5f 61 66 ).await?.rows_af
1f00: 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 31 20 fected() {....1
1f10: 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 => { Ok("Source
1f20: 65 6e 61 62 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 enabled.") },...
1f30: 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 .0 => { Ok("Sour
1f40: 63 65 20 6e 6f 74 20 66 6f 75 6e 64 2e 22 29 20 ce not found.")
1f50: 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 45 72 72 },...._ => { Err
1f60: 28 61 6e 79 68 6f 77 21 28 22 44 61 74 61 62 61 (anyhow!("Databa
1f70: 73 65 20 65 72 72 6f 72 2e 22 29 29 20 7d 2c 0a se error.")) },.
1f80: 09 09 7d 0a 09 7d 20 2a 2f 0a 0a 09 2f 2a 20 70 ..}..} */.../* p
1f90: 75 62 20 61 73 79 6e 63 20 66 6e 20 64 69 73 61 ub async fn disa
1fa0: 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f ble<S>(&self, so
1fb0: 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f urce_id: &i32, o
1fc0: 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 wner: S) -> Resu
1fd0: 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 20 lt<&str>..where
1fe0: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
1ff0: 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e .let owner = own
2000: 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6d 61 er.into();....ma
2010: 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 21 tch sqlx::query!
2020: 28 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 ("update rsstg_s
2030: 6f 75 72 63 65 20 73 65 74 20 65 6e 61 62 6c 65 ource set enable
2040: 64 20 3d 20 66 61 6c 73 65 20 77 68 65 72 65 20 d = false where
2050: 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 source_id = $1 a
2060: 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 2c 0a nd owner = $2",.
2070: 09 09 09 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 ...source_id, ow
2080: 6e 65 72 29 2e 65 78 65 63 75 74 65 28 26 6d 75 ner).execute(&mu
2090: 74 20 2a 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 t *self.pool.acq
20a0: 75 69 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 uire().await?).a
20b0: 77 61 69 74 3f 2e 72 6f 77 73 5f 61 66 66 65 63 wait?.rows_affec
20c0: 74 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 ted() {....1 =>
20d0: 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 64 69 73 { Ok("Source dis
20e0: 61 62 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 09 30 abled.") },....0
20f0: 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 => { Ok("Source
2100: 20 6e 6f 74 20 66 6f 75 6e 64 2e 22 29 20 7d 2c not found.") },
2110: 0a 09 09 09 5f 20 3d 3e 20 7b 20 45 72 72 28 61 ...._ => { Err(a
2120: 6e 79 68 6f 77 21 28 22 44 61 74 61 62 61 73 65 nyhow!("Database
2130: 20 65 72 72 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 error.")) },...
2140: 7d 0a 09 7d 20 2a 2f 0a 0a 09 2f 2a 20 70 75 62 }..} */.../* pub
2150: 20 61 73 79 6e 63 20 66 6e 20 75 70 64 61 74 65 async fn update
2160: 3c 53 3e 28 26 73 65 6c 66 2c 20 75 70 64 61 74 <S>(&self, updat
2170: 65 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c 20 e: Option<i32>,
2180: 63 68 61 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 63 channel: &str, c
2190: 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c 20 hannel_id: i64,
21a0: 75 72 6c 3a 20 26 73 74 72 2c 20 69 76 5f 68 61 url: &str, iv_ha
21b0: 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e sh: Option<&str>
21c0: 2c 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f 6e , url_re: Option
21d0: 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 72 3a 20 53 <&str>, owner: S
21e0: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 ) -> Result<&str
21f0: 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f >..where S: Into
2200: 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 <i64> {...let ow
2210: 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f ner = owner.into
2220: 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f ();...let mut co
2230: 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 nn = self.pool.a
2240: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 3f 3b cquire().await?;
2250: 0a 0a 09 09 6d 61 74 63 68 20 6d 61 74 63 68 20 ....match match
2260: 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d update {.....Som
2270: 65 28 69 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 e(id) => {......
2280: 73 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 75 70 sqlx::query!("up
2290: 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 date rsstg_sourc
22a0: 65 20 73 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 e set channel_id
22b0: 20 3d 20 24 32 2c 20 75 72 6c 20 3d 20 24 33 2c = $2, url = $3,
22c0: 20 69 76 5f 68 61 73 68 20 3d 20 24 34 2c 20 6f iv_hash = $4, o
22d0: 77 6e 65 72 20 3d 20 24 35 2c 20 63 68 61 6e 6e wner = $5, chann
22e0: 65 6c 20 3d 20 24 36 2c 20 75 72 6c 5f 72 65 20 el = $6, url_re
22f0: 3d 20 24 37 20 77 68 65 72 65 20 73 6f 75 72 63 = $7 where sourc
2300: 65 5f 69 64 20 3d 20 24 31 22 2c 0a 09 09 09 09 e_id = $1",.....
2310: 09 09 69 64 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 ..id, channel_id
2320: 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 , url, iv_hash,
2330: 6f 77 6e 65 72 2c 20 63 68 61 6e 6e 65 6c 2c 20 owner, channel,
2340: 75 72 6c 5f 72 65 29 2e 65 78 65 63 75 74 65 28 url_re).execute(
2350: 26 6d 75 74 20 2a 63 6f 6e 6e 29 2e 61 77 61 69 &mut *conn).awai
2360: 74 0a 09 09 09 09 7d 2c 0a 09 09 09 09 4e 6f 6e t.....},.....Non
2370: 65 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 e => {......sqlx
2380: 3a 3a 71 75 65 72 79 21 28 22 69 6e 73 65 72 74 ::query!("insert
2390: 20 69 6e 74 6f 20 72 73 73 74 67 5f 73 6f 75 72 into rsstg_sour
23a0: 63 65 20 28 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 ce (channel_id,
23b0: 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 url, iv_hash, ow
23c0: 6e 65 72 2c 20 63 68 61 6e 6e 65 6c 2c 20 75 72 ner, channel, ur
23d0: 6c 5f 72 65 29 20 76 61 6c 75 65 73 20 28 24 31 l_re) values ($1
23e0: 2c 20 24 32 2c 20 24 33 2c 20 24 34 2c 20 24 35 , $2, $3, $4, $5
23f0: 2c 20 24 36 29 22 2c 0a 09 09 09 09 09 09 63 68 , $6)",.......ch
2400: 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 annel_id, url, i
2410: 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 63 v_hash, owner, c
2420: 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 2e hannel, url_re).
2430: 65 78 65 63 75 74 65 28 26 6d 75 74 20 2a 63 6f execute(&mut *co
2440: 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 7d 2c nn).await.....},
2450: 0a 09 09 09 7d 20 7b 0a 09 09 09 4f 6b 28 5f 29 ....} {....Ok(_)
2460: 20 3d 3e 20 4f 6b 28 6d 61 74 63 68 20 75 70 64 => Ok(match upd
2470: 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 5f ate {.....Some(_
2480: 29 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 75 70 ) => "Channel up
2490: 64 61 74 65 64 2e 22 2c 0a 09 09 09 09 4e 6f 6e dated.",.....Non
24a0: 65 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 61 64 e => "Channel ad
24b0: 64 65 64 2e 22 2c 0a 09 09 09 7d 29 2c 0a 09 09 ded.",....}),...
24c0: 09 45 72 72 28 73 71 6c 78 3a 3a 45 72 72 6f 72 .Err(sqlx::Error
24d0: 3a 3a 44 61 74 61 62 61 73 65 28 65 72 72 29 29 ::Database(err))
24e0: 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 74 63 68 20 => {.....match
24f0: 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a 3a 3c 73 err.downcast::<s
2500: 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a 3a 50 qlx::postgres::P
2510: 67 44 61 74 61 62 61 73 65 45 72 72 6f 72 3e 28 gDatabaseError>(
2520: 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b 0a 09 09 ).routine() {...
2530: 09 09 09 53 6f 6d 65 28 22 5f 62 74 5f 63 68 65 ...Some("_bt_che
2540: 63 6b 5f 75 6e 69 71 75 65 22 2c 20 29 20 3d 3e ck_unique", ) =>
2550: 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 44 75 70 {.......Ok("Dup
2560: 6c 69 63 61 74 65 20 6b 65 79 2e 22 29 0a 09 09 licate key.")...
2570: 09 09 09 7d 2c 0a 09 09 09 09 09 53 6f 6d 65 28 ...},......Some(
2580: 5f 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b _) => {.......Ok
2590: 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 ("Database error
25a0: 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 .")......},.....
25b0: 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 .None => {......
25c0: 09 4f 6b 28 22 4e 6f 20 64 61 74 61 62 61 73 65 .Ok("No database
25d0: 20 65 72 72 6f 72 20 65 78 74 72 61 63 74 65 64 error extracted
25e0: 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 .")......},.....
25f0: 7d 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 65 }....},....Err(e
2600: 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 62 61 69 rr) => {.....bai
2610: 6c 21 28 22 53 6f 72 72 79 2c 20 75 6e 6b 6e 6f l!("Sorry, unkno
2620: 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 3f 7d wn error:\n{:#?}
2630: 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 09 09 7d 2c \n", err);....},
2640: 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 ...}..}...async
2650: 66 6e 20 61 75 74 6f 66 65 74 63 68 28 26 73 65 fn autofetch(&se
2660: 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 73 74 lf) -> Result<st
2670: 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f d::time::Duratio
2680: 6e 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 64 n> {...let mut d
2690: 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 elay = chrono::D
26a0: 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 uration::minutes
26b0: 28 31 29 3b 0a 09 09 6c 65 74 20 6e 6f 77 20 3d (1);...let now =
26c0: 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a chrono::Local::
26d0: 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 now();...let mut
26e0: 20 71 75 65 75 65 20 3d 20 73 71 6c 78 3a 3a 71 queue = sqlx::q
26f0: 75 65 72 79 21 28 72 23 22 73 65 6c 65 63 74 20 uery!(r#"select
2700: 73 6f 75 72 63 65 5f 69 64 2c 20 6e 65 78 74 5f source_id, next_
2710: 66 65 74 63 68 20 61 73 20 22 6e 65 78 74 5f 66 fetch as "next_f
2720: 65 74 63 68 3a 20 44 61 74 65 54 69 6d 65 3c 63 etch: DateTime<c
2730: 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3e 22 2c 20 hrono::Local>",
2740: 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 73 74 67 owner from rsstg
2750: 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 6c 20 6c _order natural l
2760: 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 67 5f 73 eft join rsstg_s
2770: 6f 75 72 63 65 20 77 68 65 72 65 20 6e 65 78 74 ource where next
2780: 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 29 20 2b _fetch < now() +
2790: 20 69 6e 74 65 72 76 61 6c 20 27 31 20 6d 69 6e interval '1 min
27a0: 75 74 65 27 3b 22 23 29 0a 09 09 09 2e 66 65 74 ute';"#).....fet
27b0: 63 68 5f 61 6c 6c 28 26 6d 75 74 20 2a 73 65 6c ch_all(&mut *sel
27c0: 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 f.pool.acquire()
27d0: 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 74 3f 3b .await?).await?;
27e0: 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 71 75 ...for row in qu
27f0: 65 75 65 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 eue.iter() {....
2800: 69 66 20 6c 65 74 20 53 6f 6d 65 28 6e 65 78 74 if let Some(next
2810: 5f 66 65 74 63 68 29 20 3d 20 72 6f 77 2e 6e 65 _fetch) = row.ne
2820: 78 74 5f 66 65 74 63 68 20 7b 0a 09 09 09 09 69 xt_fetch {.....i
2830: 66 20 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 6e f next_fetch < n
2840: 6f 77 20 7b 0a 09 09 09 09 09 69 66 20 6c 65 74 ow {......if let
2850: 20 28 53 6f 6d 65 28 6f 77 6e 65 72 29 2c 20 53 (Some(owner), S
2860: 6f 6d 65 28 73 6f 75 72 63 65 5f 69 64 29 29 20 ome(source_id))
2870: 3d 20 28 72 6f 77 2e 6f 77 6e 65 72 2c 20 72 6f = (row.owner, ro
2880: 77 2e 73 6f 75 72 63 65 5f 69 64 29 20 7b 0a 09 w.source_id) {..
2890: 09 09 09 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d .....let clone =
28a0: 20 43 6f 72 65 20 7b 0a 09 09 09 09 09 09 09 6f Core {........o
28b0: 77 6e 65 72 5f 63 68 61 74 3a 20 74 65 6c 65 67 wner_chat: teleg
28c0: 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a ram_bot::UserId:
28d0: 3a 6e 65 77 28 6f 77 6e 65 72 29 2c 0a 09 09 09 :new(owner),....
28e0: 09 09 09 09 2e 2e 73 65 6c 66 2e 63 6c 6f 6e 65 ......self.clone
28f0: 28 29 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 ().......};.....
2900: 09 09 74 61 73 6b 3a 3a 73 70 61 77 6e 28 61 73 ..task::spawn(as
2910: 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 09 ync move {......
2920: 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 ..if let Err(err
2930: 29 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b 28 ) = clone.check(
2940: 26 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e 65 &source_id, owne
2950: 72 2c 20 74 72 75 65 29 2e 61 77 61 69 74 20 7b r, true).await {
2960: 0a 09 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 .........if let
2970: 45 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 Err(err) = clone
2980: 2e 73 65 6e 64 28 26 66 6f 72 6d 61 74 21 28 22 .send(&format!("
2990: f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 š {:?}", err)
29a0: 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 , None, None).aw
29b0: 61 69 74 20 7b 0a 09 09 09 09 09 09 09 09 09 64 ait {..........d
29c0: 62 67 21 28 22 43 68 65 63 6b 20 65 72 72 6f 72 bg!("Check error
29d0: 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 : {}", err);....
29e0: 09 09 09 09 09 09 2f 2f 20 63 6c 6f 6e 65 2e 64 ......// clone.d
29f0: 69 73 61 62 6c 65 28 26 73 6f 75 72 63 65 5f 69 isable(&source_i
2a00: 64 2c 20 6f 77 6e 65 72 29 2e 61 77 61 69 74 2e d, owner).await.
2a10: 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 09 09 unwrap();.......
2a20: 09 09 7d 3b 0a 09 09 09 09 09 09 09 7d 3b 0a 09 ..};........};..
2a30: 09 09 09 09 09 7d 29 3b 0a 09 09 09 09 09 7d 0a .....});......}.
2a40: 09 09 09 09 7d 20 65 6c 73 65 20 69 66 20 6e 65 ....} else if ne
2a50: 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 20 3c xt_fetch - now <
2a60: 20 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 64 65 delay {......de
2a70: 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 63 68 lay = next_fetch
2a80: 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 7d 0a 09 09 - now;.....}...
2a90: 09 7d 0a 09 09 7d 3b 0a 09 09 71 75 65 75 65 2e .}...};...queue.
2aa0: 63 6c 65 61 72 28 29 3b 0a 09 09 4f 6b 28 64 65 clear();...Ok(de
2ab0: 6c 61 79 2e 74 6f 5f 73 74 64 28 29 3f 29 0a 09 lay.to_std()?)..
2ac0: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
2ad0: 20 6c 69 73 74 3c 53 3e 28 26 73 65 6c 66 2c 20 list<S>(&self,
2ae0: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
2af0: 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 ult<String>..whe
2b00: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
2b10: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
2b20: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 owner.into();...
2b30: 09 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 3a 20 .let mut reply:
2b40: 56 65 63 3c 43 6f 77 3c 73 74 72 3e 3e 20 3d 20 Vec<Cow<str>> =
2b50: 76 65 63 21 5b 5d 3b 0a 09 09 72 65 70 6c 79 2e vec![];...reply.
2b60: 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 push("Channels:"
2b70: 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 6c 65 74 20 .into());...let
2b80: 72 6f 77 73 20 3d 20 73 71 6c 78 3a 3a 71 75 65 rows = sqlx::que
2b90: 72 79 21 28 22 73 65 6c 65 63 74 20 73 6f 75 72 ry!("select sour
2ba0: 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 ce_id, channel,
2bb0: 65 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 76 enabled, url, iv
2bc0: 5f 68 61 73 68 2c 20 75 72 6c 5f 72 65 20 66 72 _hash, url_re fr
2bd0: 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 om rsstg_source
2be0: 77 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 31 where owner = $1
2bf0: 20 6f 72 64 65 72 20 62 79 20 73 6f 75 72 63 65 order by source
2c00: 5f 69 64 22 2c 0a 09 09 09 6f 77 6e 65 72 29 2e _id",....owner).
2c10: 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 2a fetch_all(&mut *
2c20: 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 self.pool.acquir
2c30: 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 e().await?).awai
2c40: 74 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e t?;...for row in
2c50: 20 72 6f 77 73 2e 69 74 65 72 28 29 20 7b 0a 09 rows.iter() {..
2c60: 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 ..reply.push(for
2c70: 6d 61 74 21 28 22 5c 6e 5c 5c 23 ef b8 8f e2 83 mat!("\n\\#ļøā
2c80: a3 20 7b 7d 20 5c 5c 2a ef b8 8f e2 83 a3 20 60 £ {} \\*ļøā£ `
2c90: 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 97 20 60 7b 7d {}` {}\nš `{}
2ca0: 60 22 2c 20 72 6f 77 2e 73 6f 75 72 63 65 5f 69 `", row.source_i
2cb0: 64 2c 20 72 6f 77 2e 63 68 61 6e 6e 65 6c 2c 0a d, row.channel,.
2cc0: 09 09 09 09 6d 61 74 63 68 20 72 6f 77 2e 65 6e ....match row.en
2cd0: 61 62 6c 65 64 20 7b 0a 09 09 09 09 09 74 72 75 abled {......tru
2ce0: 65 20 20 3d 3e 20 22 f0 9f 94 84 20 65 6e 61 62 e => "š enab
2cf0: 6c 65 64 22 2c 0a 09 09 09 09 09 66 61 6c 73 65 led",......false
2d00: 20 3d 3e 20 22 e2 9b 94 20 64 69 73 61 62 6c 65 => "ā disable
2d10: 64 22 2c 0a 09 09 09 09 7d 2c 20 72 6f 77 2e 75 d",.....}, row.u
2d20: 72 6c 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 rl).into());....
2d30: 69 66 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 68 if let Some(hash
2d40: 29 20 3d 20 26 72 6f 77 2e 69 76 5f 68 61 73 68 ) = &row.iv_hash
2d50: 20 7b 0a 09 09 09 09 72 65 70 6c 79 2e 70 75 73 {.....reply.pus
2d60: 68 28 66 6f 72 6d 61 74 21 28 22 49 56 3a 20 60 h(format!("IV: `
2d70: 7b 7d 60 22 2c 20 68 61 73 68 29 2e 69 6e 74 6f {}`", hash).into
2d80: 28 29 29 3b 0a 09 09 09 7d 0a 09 09 09 69 66 20 ());....}....if
2d90: 6c 65 74 20 53 6f 6d 65 28 72 65 29 20 3d 20 26 let Some(re) = &
2da0: 72 6f 77 2e 75 72 6c 5f 72 65 20 7b 0a 09 09 09 row.url_re {....
2db0: 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d .reply.push(form
2dc0: 61 74 21 28 22 52 45 3a 20 60 7b 7d 60 22 2c 20 at!("RE: `{}`",
2dd0: 72 65 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 re).into());....
2de0: 7d 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 65 70 6c }...};...Ok(repl
2df0: 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 0a 09 7d y.join("\n"))..}
2e00: 20 2a 2f 0a 7d 0a */.}.