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 73 olOptions;.use s
0080: 74 64 3a 3a 7b 0a 09 62 6f 72 72 6f 77 3a 3a 43 td::{..borrow::C
0090: 6f 77 2c 0a 09 63 6f 6c 6c 65 63 74 69 6f 6e 73 ow,..collections
00a0: 3a 3a 7b 0a 09 09 42 54 72 65 65 4d 61 70 2c 0a ::{...BTreeMap,.
00b0: 09 09 48 61 73 68 53 65 74 2c 0a 09 7d 2c 0a 09 ..HashSet,..},..
00c0: 73 79 6e 63 3a 3a 7b 41 72 63 2c 20 4d 75 74 65 sync::{Arc, Mute
00d0: 78 7d 2c 0a 7d 3b 0a 0a 23 5b 64 65 72 69 76 65 x},.};..#[derive
00e0: 28 43 6c 6f 6e 65 29 5d 0a 70 75 62 20 73 74 72 (Clone)].pub str
00f0: 75 63 74 20 43 6f 72 65 20 7b 0a 09 6f 77 6e 65 uct Core {..owne
0100: 72 5f 63 68 61 74 3a 20 74 65 6c 65 67 72 61 6d r_chat: telegram
0110: 5f 62 6f 74 3a 3a 55 73 65 72 49 64 2c 0a 09 70 _bot::UserId,..p
0120: 75 62 20 74 67 3a 20 74 65 6c 65 67 72 61 6d 5f ub tg: telegram_
0130: 62 6f 74 3a 3a 41 70 69 2c 0a 09 70 75 62 20 6d bot::Api,..pub m
0140: 79 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a y: telegram_bot:
0150: 3a 55 73 65 72 2c 0a 09 70 6f 6f 6c 3a 20 73 71 :User,..pool: sq
0160: 6c 78 3a 3a 50 6f 6f 6c 3c 73 71 6c 78 3a 3a 50 lx::Pool<sqlx::P
0170: 6f 73 74 67 72 65 73 3e 2c 0a 09 73 6f 75 72 63 ostgres>,..sourc
0180: 65 73 3a 20 41 72 63 3c 4d 75 74 65 78 3c 48 61 es: Arc<Mutex<Ha
0190: 73 68 53 65 74 3c 41 72 63 3c 69 33 32 3e 3e 3e shSet<Arc<i32>>>
01a0: 3e 2c 0a 09 68 74 74 70 5f 63 6c 69 65 6e 74 3a >,..http_client:
01b0: 20 72 65 71 77 65 73 74 3a 3a 43 6c 69 65 6e 74 reqwest::Client
01c0: 2c 0a 7d 0a 0a 69 6d 70 6c 20 43 6f 72 65 20 7b ,.}..impl Core {
01d0: 0a 09 70 75 62 20 66 6e 20 6e 65 77 28 73 65 74 ..pub fn new(set
01e0: 74 69 6e 67 73 3a 20 63 6f 6e 66 69 67 3a 3a 43 tings: config::C
01f0: 6f 6e 66 69 67 29 20 2d 3e 20 52 65 73 75 6c 74 onfig) -> Result
0200: 3c 41 72 63 3c 43 6f 72 65 3e 3e 20 7b 0a 09 09 <Arc<Core>> {...
0210: 6c 65 74 20 6f 77 6e 65 72 20 3d 20 73 65 74 74 let owner = sett
0220: 69 6e 67 73 2e 67 65 74 5f 69 6e 74 28 22 6f 77 ings.get_int("ow
0230: 6e 65 72 22 29 3f 3b 0a 09 09 6c 65 74 20 61 70 ner")?;...let ap
0240: 69 5f 6b 65 79 20 3d 20 73 65 74 74 69 6e 67 73 i_key = settings
0250: 2e 67 65 74 5f 73 74 72 69 6e 67 28 22 61 70 69 .get_string("api
0260: 5f 6b 65 79 22 29 3f 3b 0a 09 09 6c 65 74 20 74 _key")?;...let t
0270: 67 20 3d 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 g = telegram_bot
0280: 3a 3a 41 70 69 3a 3a 6e 65 77 28 61 70 69 5f 6b ::Api::new(api_k
0290: 65 79 29 3b 0a 09 09 6c 65 74 20 74 67 5f 63 6c ey);...let tg_cl
02a0: 6f 6e 65 64 20 3d 20 74 67 2e 63 6c 6f 6e 65 28 oned = tg.clone(
02b0: 29 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 6c );....let mut cl
02c0: 69 65 6e 74 20 3d 20 72 65 71 77 65 73 74 3a 3a ient = reqwest::
02d0: 43 6c 69 65 6e 74 3a 3a 62 75 69 6c 64 65 72 28 Client::builder(
02e0: 29 3b 0a 09 09 69 66 20 6c 65 74 20 4f 6b 28 70 );...if let Ok(p
02f0: 72 6f 78 79 29 20 3d 20 73 65 74 74 69 6e 67 73 roxy) = settings
0300: 2e 67 65 74 5f 73 74 72 69 6e 67 28 22 70 72 6f .get_string("pro
0310: 78 79 22 29 20 7b 0a 09 09 09 6c 65 74 20 70 72 xy") {....let pr
0320: 6f 78 79 20 3d 20 72 65 71 77 65 73 74 3a 3a 50 oxy = reqwest::P
0330: 72 6f 78 79 3a 3a 61 6c 6c 28 70 72 6f 78 79 29 roxy::all(proxy)
0340: 3f 3b 0a 09 09 09 63 6c 69 65 6e 74 20 3d 20 63 ?;....client = c
0350: 6c 69 65 6e 74 2e 70 72 6f 78 79 28 70 72 6f 78 lient.proxy(prox
0360: 79 29 3b 0a 09 09 7d 0a 09 09 6c 65 74 20 68 74 y);...}...let ht
0370: 74 70 5f 63 6c 69 65 6e 74 20 3d 20 63 6c 69 65 tp_client = clie
0380: 6e 74 2e 62 75 69 6c 64 28 29 3f 3b 0a 09 09 6c nt.build()?;...l
0390: 65 74 20 63 6f 72 65 20 3d 20 41 72 63 3a 3a 6e et core = Arc::n
03a0: 65 77 28 43 6f 72 65 20 7b 0a 09 09 09 74 67 2c ew(Core {....tg,
03b0: 0a 09 09 09 6d 79 3a 20 74 61 73 6b 3a 3a 62 6c ....my: task::bl
03c0: 6f 63 6b 5f 6f 6e 28 61 73 79 6e 63 20 7b 0a 09 ock_on(async {..
03d0: 09 09 09 74 67 5f 63 6c 6f 6e 65 64 2e 73 65 6e ...tg_cloned.sen
03e0: 64 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a d(telegram_bot::
03f0: 47 65 74 4d 65 29 2e 61 77 61 69 74 0a 09 09 09 GetMe).await....
0400: 7d 29 3f 2c 0a 09 09 09 6f 77 6e 65 72 5f 63 68 })?,....owner_ch
0410: 61 74 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 at: telegram_bot
0420: 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 ::UserId::new(ow
0430: 6e 65 72 29 2c 0a 09 09 09 70 6f 6f 6c 3a 20 50 ner),....pool: P
0440: 67 50 6f 6f 6c 4f 70 74 69 6f 6e 73 3a 3a 6e 65 gPoolOptions::ne
0450: 77 28 29 0a 09 09 09 09 2e 6d 61 78 5f 63 6f 6e w()......max_con
0460: 6e 65 63 74 69 6f 6e 73 28 35 29 0a 09 09 09 09 nections(5).....
0470: 2e 61 63 71 75 69 72 65 5f 74 69 6d 65 6f 75 74 .acquire_timeout
0480: 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 (std::time::Dura
0490: 74 69 6f 6e 3a 3a 6e 65 77 28 33 30 30 2c 20 30 tion::new(300, 0
04a0: 29 29 0a 09 09 09 09 2e 69 64 6c 65 5f 74 69 6d ))......idle_tim
04b0: 65 6f 75 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a eout(std::time::
04c0: 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 36 30 Duration::new(60
04d0: 2c 20 30 29 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 , 0))......conne
04e0: 63 74 5f 6c 61 7a 79 28 26 73 65 74 74 69 6e 67 ct_lazy(&setting
04f0: 73 2e 67 65 74 5f 73 74 72 69 6e 67 28 22 70 67 s.get_string("pg
0500: 22 29 3f 29 3f 2c 0a 09 09 09 73 6f 75 72 63 65 ")?)?,....source
0510: 73 3a 20 41 72 63 3a 3a 6e 65 77 28 4d 75 74 65 s: Arc::new(Mute
0520: 78 3a 3a 6e 65 77 28 48 61 73 68 53 65 74 3a 3a x::new(HashSet::
0530: 6e 65 77 28 29 29 29 2c 0a 09 09 09 68 74 74 70 new())),....http
0540: 5f 63 6c 69 65 6e 74 2c 0a 09 09 7d 29 3b 0a 09 _client,...});..
0550: 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 63 6f 72 .let clone = cor
0560: 65 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 74 61 73 e.clone();...tas
0570: 6b 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 6d k::spawn(async m
0580: 6f 76 65 20 7b 0a 09 09 09 6c 6f 6f 70 20 7b 0a ove {....loop {.
0590: 09 09 09 09 6c 65 74 20 64 65 6c 61 79 20 3d 20 ....let delay =
05a0: 6d 61 74 63 68 20 26 63 6c 6f 6e 65 2e 61 75 74 match &clone.aut
05b0: 6f 66 65 74 63 68 28 29 2e 61 77 61 69 74 20 7b ofetch().await {
05c0: 0a 09 09 09 09 09 45 72 72 28 65 72 72 29 20 3d ......Err(err) =
05d0: 3e 20 7b 0a 09 09 09 09 09 09 69 66 20 6c 65 74 > {.......if let
05e0: 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e Err(err) = clon
05f0: 65 2e 73 65 6e 64 28 66 6f 72 6d 61 74 21 28 22 e.send(format!("
0600: f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 š {:?}", err)
0610: 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 , None, None).aw
0620: 61 69 74 20 7b 0a 09 09 09 09 09 09 09 65 70 72 ait {........epr
0630: 69 6e 74 6c 6e 21 28 22 41 75 74 6f 66 65 74 63 intln!("Autofetc
0640: 68 20 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 h error: {}", er
0650: 72 29 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 r);.......};....
0660: 09 09 09 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 ...std::time::Du
0670: 72 61 74 69 6f 6e 3a 3a 66 72 6f 6d 5f 73 65 63 ration::from_sec
0680: 73 28 36 30 29 0a 09 09 09 09 09 7d 2c 0a 09 09 s(60)......},...
0690: 09 09 09 4f 6b 28 74 69 6d 65 29 20 3d 3e 20 2a ...Ok(time) => *
06a0: 74 69 6d 65 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 time,.....};....
06b0: 09 74 61 73 6b 3a 3a 73 6c 65 65 70 28 64 65 6c .task::sleep(del
06c0: 61 79 29 2e 61 77 61 69 74 3b 0a 09 09 09 7d 0a ay).await;....}.
06d0: 09 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f 72 65 29 ..});...Ok(core)
06e0: 0a 09 7d 0a 0a 09 70 75 62 20 66 6e 20 73 74 72 ..}...pub fn str
06f0: 65 61 6d 28 26 73 65 6c 66 29 20 2d 3e 20 74 65 eam(&self) -> te
0700: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 70 64 61 legram_bot::Upda
0710: 74 65 73 53 74 72 65 61 6d 20 7b 0a 09 09 73 65 tesStream {...se
0720: 6c 66 2e 74 67 2e 73 74 72 65 61 6d 28 29 0a 09 lf.tg.stream()..
0730: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
0740: 20 73 65 6e 64 3c 27 61 2c 20 53 3e 28 26 73 65 send<'a, S>(&se
0750: 6c 66 2c 20 6d 73 67 3a 20 53 2c 20 74 61 72 67 lf, msg: S, targ
0760: 65 74 3a 20 4f 70 74 69 6f 6e 3c 74 65 6c 65 67 et: Option<teleg
0770: 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3e ram_bot::UserId>
0780: 2c 20 6d 6f 64 65 3a 20 4f 70 74 69 6f 6e 3c 74 , mode: Option<t
0790: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 elegram_bot::typ
07a0: 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3e 29 20 es::ParseMode>)
07b0: 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 0a 09 77 -> Result<()>..w
07c0: 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 43 6f 77 here S: Into<Cow
07d0: 3c 27 61 2c 20 73 74 72 3e 3e 20 7b 0a 09 09 6c <'a, str>> {...l
07e0: 65 74 20 6d 6f 64 65 20 3d 20 6d 6f 64 65 2e 75 et mode = mode.u
07f0: 6e 77 72 61 70 5f 6f 72 28 74 65 6c 65 67 72 61 nwrap_or(telegra
0800: 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 61 m_bot::types::Pa
0810: 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 3b 0a rseMode::Html);.
0820: 09 09 6c 65 74 20 74 61 72 67 65 74 20 3d 20 74 ..let target = t
0830: 61 72 67 65 74 2e 75 6e 77 72 61 70 5f 6f 72 28 arget.unwrap_or(
0840: 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 61 74 29 self.owner_chat)
0850: 3b 0a 09 09 73 65 6c 66 2e 74 67 2e 73 65 6e 64 ;...self.tg.send
0860: 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 (telegram_bot::S
0870: 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 endMessage::new(
0880: 74 61 72 67 65 74 2c 20 6d 73 67 29 2e 70 61 72 target, msg).par
0890: 73 65 5f 6d 6f 64 65 28 6d 6f 64 65 29 29 2e 61 se_mode(mode)).a
08a0: 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 28 29 29 0a wait?;...Ok(()).
08b0: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
08c0: 6e 20 63 68 65 63 6b 3c 53 3e 28 26 73 65 6c 66 n check<S>(&self
08d0: 2c 20 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 , id: &i32, owne
08e0: 72 3a 20 53 2c 20 72 65 61 6c 3a 20 62 6f 6f 6c r: S, real: bool
08f0: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 3c ) -> Result<Cow<
0900: 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 65 '_, str>>..where
0910: 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a S: Into<i64> {.
0920: 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 ..let owner = ow
0930: 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c ner.into();....l
0940: 65 74 20 6d 75 74 20 70 6f 73 74 65 64 3a 20 69 et mut posted: i
0950: 33 32 20 3d 20 30 3b 0a 09 09 6c 65 74 20 69 64 32 = 0;...let id
0960: 20 3d 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 = {....let mut
0970: 73 65 74 20 3d 20 73 65 6c 66 2e 73 6f 75 72 63 set = self.sourc
0980: 65 73 2e 6c 6f 63 6b 28 29 2e 75 6e 77 72 61 70 es.lock().unwrap
0990: 28 29 3b 0a 09 09 09 6d 61 74 63 68 20 73 65 74 ();....match set
09a0: 2e 67 65 74 28 69 64 29 20 7b 0a 09 09 09 09 53 .get(id) {.....S
09b0: 6f 6d 65 28 69 64 29 20 3d 3e 20 69 64 2e 63 6c ome(id) => id.cl
09c0: 6f 6e 65 28 29 2c 0a 09 09 09 09 4e 6f 6e 65 20 one(),.....None
09d0: 3d 3e 20 7b 0a 09 09 09 09 09 6c 65 74 20 69 64 => {......let id
09e0: 20 3d 20 41 72 63 3a 3a 6e 65 77 28 2a 69 64 29 = Arc::new(*id)
09f0: 3b 0a 09 09 09 09 09 73 65 74 2e 69 6e 73 65 72 ;......set.inser
0a00: 74 28 69 64 2e 63 6c 6f 6e 65 28 29 29 3b 0a 09 t(id.clone());..
0a10: 09 09 09 09 69 64 2e 63 6c 6f 6e 65 28 29 0a 09 ....id.clone()..
0a20: 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 7d 3b 0a ...},....}...};.
0a30: 09 09 6c 65 74 20 63 6f 75 6e 74 20 3d 20 41 72 ..let count = Ar
0a40: 63 3a 3a 73 74 72 6f 6e 67 5f 63 6f 75 6e 74 28 c::strong_count(
0a50: 26 69 64 29 3b 0a 09 09 69 66 20 63 6f 75 6e 74 &id);...if count
0a60: 20 3d 3d 20 32 20 7b 0a 09 09 09 6c 65 74 20 73 == 2 {....let s
0a70: 6f 75 72 63 65 20 3d 20 73 71 6c 78 3a 3a 71 75 ource = sqlx::qu
0a80: 65 72 79 21 28 22 73 65 6c 65 63 74 20 73 6f 75 ery!("select sou
0a90: 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 5f rce_id, channel_
0aa0: 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 id, url, iv_hash
0ab0: 2c 20 6f 77 6e 65 72 2c 20 75 72 6c 5f 72 65 20 , owner, url_re
0ac0: 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 from rsstg_sourc
0ad0: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
0ae0: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
0af0: 20 3d 20 24 32 22 2c 0a 09 09 09 09 2a 69 64 2c = $2",.....*id,
0b00: 20 6f 77 6e 65 72 29 2e 66 65 74 63 68 5f 6f 6e owner).fetch_on
0b10: 65 28 26 6d 75 74 20 73 65 6c 66 2e 70 6f 6f 6c e(&mut self.pool
0b20: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
0b30: 3f 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 6c 65 ?).await?;....le
0b40: 74 20 64 65 73 74 69 6e 61 74 69 6f 6e 20 3d 20 t destination =
0b50: 6d 61 74 63 68 20 72 65 61 6c 20 7b 0a 09 09 09 match real {....
0b60: 09 74 72 75 65 20 3d 3e 20 74 65 6c 65 67 72 61 .true => telegra
0b70: 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e m_bot::UserId::n
0b80: 65 77 28 73 6f 75 72 63 65 2e 63 68 61 6e 6e 65 ew(source.channe
0b90: 6c 5f 69 64 29 2c 0a 09 09 09 09 66 61 6c 73 65 l_id),.....false
0ba0: 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 => telegram_bot
0bb0: 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 73 6f ::UserId::new(so
0bc0: 75 72 63 65 2e 6f 77 6e 65 72 29 2c 0a 09 09 09 urce.owner),....
0bd0: 7d 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 74 68 };....let mut th
0be0: 69 73 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e is_fetch: Option
0bf0: 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f <DateTime<chrono
0c00: 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 3e 20 ::FixedOffset>>
0c10: 3d 20 4e 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 6d = None;....let m
0c20: 75 74 20 70 6f 73 74 73 3a 20 42 54 72 65 65 4d ut posts: BTreeM
0c30: 61 70 3c 44 61 74 65 54 69 6d 65 3c 63 68 72 6f ap<DateTime<chro
0c40: 6e 6f 3a 3a 46 69 78 65 64 4f 66 66 73 65 74 3e no::FixedOffset>
0c50: 2c 20 53 74 72 69 6e 67 3e 20 3d 20 42 54 72 65 , String> = BTre
0c60: 65 4d 61 70 3a 3a 6e 65 77 28 29 3b 0a 0a 09 09 eMap::new();....
0c70: 09 6c 65 74 20 72 65 73 70 6f 6e 73 65 20 3d 20 .let response =
0c80: 73 65 6c 66 2e 68 74 74 70 5f 63 6c 69 65 6e 74 self.http_client
0c90: 2e 67 65 74 28 26 73 6f 75 72 63 65 2e 75 72 6c .get(&source.url
0ca0: 29 2e 73 65 6e 64 28 29 2e 61 77 61 69 74 3f 3b ).send().await?;
0cb0: 0a 09 09 09 6c 65 74 20 73 74 61 74 75 73 20 3d ....let status =
0cc0: 20 72 65 73 70 6f 6e 73 65 2e 73 74 61 74 75 73 response.status
0cd0: 28 29 3b 0a 09 09 09 6c 65 74 20 63 6f 6e 74 65 ();....let conte
0ce0: 6e 74 20 3d 20 72 65 73 70 6f 6e 73 65 2e 62 79 nt = response.by
0cf0: 74 65 73 28 29 2e 61 77 61 69 74 3f 3b 0a 09 09 tes().await?;...
0d00: 09 6d 61 74 63 68 20 72 73 73 3a 3a 43 68 61 6e .match rss::Chan
0d10: 6e 65 6c 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 nel::read_from(&
0d20: 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 20 7b 0a 09 content[..]) {..
0d30: 09 09 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 7b ...Ok(feed) => {
0d40: 0a 09 09 09 09 09 66 6f 72 20 69 74 65 6d 20 69 ......for item i
0d50: 6e 20 66 65 65 64 2e 69 74 65 6d 73 28 29 20 7b n feed.items() {
0d60: 0a 09 09 09 09 09 09 69 66 20 6c 65 74 20 53 6f .......if let So
0d70: 6d 65 28 6c 69 6e 6b 29 20 3d 20 69 74 65 6d 2e me(link) = item.
0d80: 6c 69 6e 6b 28 29 20 7b 0a 09 09 09 09 09 09 09 link() {........
0d90: 6c 65 74 20 64 61 74 65 20 3d 20 6d 61 74 63 68 let date = match
0da0: 20 69 74 65 6d 2e 70 75 62 5f 64 61 74 65 28 29 item.pub_date()
0db0: 20 7b 0a 09 09 09 09 09 09 09 09 53 6f 6d 65 28 {.........Some(
0dc0: 66 65 65 64 5f 64 61 74 65 29 20 3d 3e 20 44 61 feed_date) => Da
0dd0: 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 teTime::parse_fr
0de0: 6f 6d 5f 72 66 63 32 38 32 32 28 66 65 65 64 5f om_rfc2822(feed_
0df0: 64 61 74 65 29 2c 0a 09 09 09 09 09 09 09 09 4e date),.........N
0e00: 6f 6e 65 20 3d 3e 20 44 61 74 65 54 69 6d 65 3a one => DateTime:
0e10: 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 66 63 33 :parse_from_rfc3
0e20: 33 33 39 28 26 69 74 65 6d 2e 64 75 62 6c 69 6e 339(&item.dublin
0e30: 5f 63 6f 72 65 5f 65 78 74 28 29 2e 75 6e 77 72 _core_ext().unwr
0e40: 61 70 28 29 2e 64 61 74 65 73 28 29 5b 30 5d 29 ap().dates()[0])
0e50: 2c 0a 09 09 09 09 09 09 09 7d 3f 3b 0a 09 09 09 ,........}?;....
0e60: 09 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 6c 69 ....let url = li
0e70: 6e 6b 3b 0a 09 09 09 09 09 09 09 70 6f 73 74 73 nk;........posts
0e80: 2e 69 6e 73 65 72 74 28 64 61 74 65 2c 20 75 72 .insert(date, ur
0e90: 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a l.to_string());.
0ea0: 09 09 09 09 09 09 7d 0a 09 09 09 09 09 7d 3b 0a ......}......};.
0eb0: 09 09 09 09 7d 2c 0a 09 09 09 09 45 72 72 28 65 ....},.....Err(e
0ec0: 72 72 29 20 3d 3e 20 6d 61 74 63 68 20 65 72 72 rr) => match err
0ed0: 20 7b 0a 09 09 09 09 09 72 73 73 3a 3a 45 72 72 {......rss::Err
0ee0: 6f 72 3a 3a 49 6e 76 61 6c 69 64 53 74 61 72 74 or::InvalidStart
0ef0: 54 61 67 20 3d 3e 20 7b 0a 09 09 09 09 09 09 6c Tag => {.......l
0f00: 65 74 20 66 65 65 64 20 3d 20 61 74 6f 6d 5f 73 et feed = atom_s
0f10: 79 6e 64 69 63 61 74 69 6f 6e 3a 3a 46 65 65 64 yndication::Feed
0f20: 3a 3a 72 65 61 64 5f 66 72 6f 6d 28 26 63 6f 6e ::read_from(&con
0f30: 74 65 6e 74 5b 2e 2e 5d 29 0a 09 09 09 09 09 09 tent[..]).......
0f40: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
0f50: 7c 20 66 6f 72 6d 61 74 21 28 22 50 72 6f 62 6c | format!("Probl
0f60: 65 6d 20 6f 70 65 6e 69 6e 67 20 66 65 65 64 20 em opening feed
0f70: 75 72 6c 3a 5c 6e 7b 7d 5c 6e 7b 7d 22 2c 20 26 url:\n{}\n{}", &
0f80: 73 6f 75 72 63 65 2e 75 72 6c 2c 20 73 74 61 74 source.url, stat
0f90: 75 73 29 29 3f 3b 0a 09 09 09 09 09 09 66 6f 72 us))?;.......for
0fa0: 20 69 74 65 6d 20 69 6e 20 66 65 65 64 2e 65 6e item in feed.en
0fb0: 74 72 69 65 73 28 29 20 7b 0a 09 09 09 09 09 09 tries() {.......
0fc0: 09 6c 65 74 20 64 61 74 65 20 3d 20 69 74 65 6d .let date = item
0fd0: 2e 70 75 62 6c 69 73 68 65 64 28 29 2e 75 6e 77 .published().unw
0fe0: 72 61 70 28 29 3b 0a 09 09 09 09 09 09 09 6c 65 rap();........le
0ff0: 74 20 75 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e t url = item.lin
1000: 6b 73 28 29 5b 30 5d 2e 68 72 65 66 28 29 3b 0a ks()[0].href();.
1010: 09 09 09 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 .......posts.ins
1020: 65 72 74 28 2a 64 61 74 65 2c 20 75 72 6c 2e 74 ert(*date, url.t
1030: 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 o_string());....
1040: 09 09 09 7d 3b 0a 09 09 09 09 09 7d 2c 0a 09 09 ...};......},...
1050: 09 09 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 ...rss::Error::E
1060: 6f 66 20 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f of => (),......_
1070: 20 3d 3e 20 62 61 69 6c 21 28 22 55 6e 73 75 70 => bail!("Unsup
1080: 70 6f 72 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 ported or mangle
1090: 64 20 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d d content:\n{:?}
10a0: 5c 6e 7b 3a 23 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e \n{:#?}\n{:#?}\n
10b0: 22 2c 20 26 73 6f 75 72 63 65 2e 75 72 6c 2c 20 ", &source.url,
10c0: 65 72 72 2c 20 73 74 61 74 75 73 29 0a 09 09 09 err, status)....
10d0: 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 66 6f 72 20 .}....};....for
10e0: 28 64 61 74 65 2c 20 75 72 6c 29 20 69 6e 20 70 (date, url) in p
10f0: 6f 73 74 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 osts.iter() {...
1100: 09 09 6c 65 74 20 70 6f 73 74 5f 75 72 6c 3a 20 ..let post_url:
1110: 43 6f 77 3c 73 74 72 3e 20 3d 20 6d 61 74 63 68 Cow<str> = match
1120: 20 73 6f 75 72 63 65 2e 75 72 6c 5f 72 65 20 7b source.url_re {
1130: 0a 09 09 09 09 09 53 6f 6d 65 28 72 65 66 20 78 ......Some(ref x
1140: 29 20 3d 3e 20 73 65 64 72 65 67 65 78 3a 3a 52 ) => sedregex::R
1150: 65 70 6c 61 63 65 43 6f 6d 6d 61 6e 64 3a 3a 6e eplaceCommand::n
1160: 65 77 28 78 29 3f 2e 65 78 65 63 75 74 65 28 75 ew(x)?.execute(u
1170: 72 6c 29 2c 0a 09 09 09 09 09 4e 6f 6e 65 20 3d rl),......None =
1180: 3e 20 75 72 6c 2e 69 6e 74 6f 28 29 2c 0a 09 09 > url.into(),...
1190: 09 09 7d 3b 0a 09 09 09 09 69 66 20 6c 65 74 20 ..};.....if let
11a0: 53 6f 6d 65 28 65 78 69 73 74 73 29 20 3d 20 73 Some(exists) = s
11b0: 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 73 65 6c qlx::query!("sel
11c0: 65 63 74 20 65 78 69 73 74 73 28 73 65 6c 65 63 ect exists(selec
11d0: 74 20 74 72 75 65 20 66 72 6f 6d 20 72 73 73 74 t true from rsst
11e0: 67 5f 70 6f 73 74 20 77 68 65 72 65 20 75 72 6c g_post where url
11f0: 20 3d 20 24 31 20 61 6e 64 20 73 6f 75 72 63 65 = $1 and source
1200: 5f 69 64 20 3d 20 24 32 29 20 61 73 20 65 78 69 _id = $2) as exi
1210: 73 74 73 3b 22 2c 0a 09 09 09 09 09 26 70 6f 73 sts;",......&pos
1220: 74 5f 75 72 6c 2c 20 2a 69 64 29 2e 66 65 74 63 t_url, *id).fetc
1230: 68 5f 6f 6e 65 28 26 6d 75 74 20 73 65 6c 66 2e h_one(&mut self.
1240: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
1250: 77 61 69 74 3f 29 2e 61 77 61 69 74 3f 2e 65 78 wait?).await?.ex
1260: 69 73 74 73 20 7b 0a 09 09 09 09 09 69 66 20 21 ists {......if !
1270: 20 65 78 69 73 74 73 20 7b 0a 09 09 09 09 09 09 exists {.......
1280: 69 66 20 74 68 69 73 5f 66 65 74 63 68 2e 69 73 if this_fetch.is
1290: 5f 6e 6f 6e 65 28 29 20 7c 7c 20 2a 64 61 74 65 _none() || *date
12a0: 20 3e 20 74 68 69 73 5f 66 65 74 63 68 2e 75 6e > this_fetch.un
12b0: 77 72 61 70 28 29 20 7b 0a 09 09 09 09 09 09 09 wrap() {........
12c0: 74 68 69 73 5f 66 65 74 63 68 20 3d 20 53 6f 6d this_fetch = Som
12d0: 65 28 2a 64 61 74 65 29 3b 0a 09 09 09 09 09 09 e(*date);.......
12e0: 7d 3b 0a 09 09 09 09 09 09 73 65 6c 66 2e 74 67 };.......self.tg
12f0: 2e 73 65 6e 64 28 20 6d 61 74 63 68 20 26 73 6f .send( match &so
1300: 75 72 63 65 2e 69 76 5f 68 61 73 68 20 7b 0a 09 urce.iv_hash {..
1310: 09 09 09 09 09 09 09 53 6f 6d 65 28 68 61 73 68 .......Some(hash
1320: 29 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f ) => telegram_bo
1330: 74 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a t::SendMessage::
1340: 6e 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c new(destination,
1350: 20 66 6f 72 6d 61 74 21 28 22 3c 61 20 68 72 65 format!("<a hre
1360: 66 3d 5c 22 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 f=\"https://t.me
1370: 2f 69 76 3f 75 72 6c 3d 7b 7d 26 72 68 61 73 68 /iv?url={}&rhash
1380: 3d 7b 7d 5c 22 3e 20 3c 2f 61 3e 7b 30 7d 22 2c ={}\"> </a>{0}",
1390: 20 26 70 6f 73 74 5f 75 72 6c 2c 20 68 61 73 68 &post_url, hash
13a0: 29 29 2c 0a 09 09 09 09 09 09 09 09 4e 6f 6e 65 )),.........None
13b0: 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 => telegram_bot
13c0: 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e ::SendMessage::n
13d0: 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c 20 ew(destination,
13e0: 66 6f 72 6d 61 74 21 28 22 7b 7d 22 2c 20 70 6f format!("{}", po
13f0: 73 74 5f 75 72 6c 29 29 2c 0a 09 09 09 09 09 09 st_url)),.......
1400: 09 7d 2e 70 61 72 73 65 5f 6d 6f 64 65 28 74 65 .}.parse_mode(te
1410: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 legram_bot::type
1420: 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 s::ParseMode::Ht
1430: 6d 6c 29 29 2e 61 77 61 69 74 0a 09 09 09 09 09 ml)).await......
1440: 09 09 2e 63 6f 6e 74 65 78 74 28 22 43 61 6e 27 ...context("Can'
1450: 74 20 70 6f 73 74 20 6d 65 73 73 61 67 65 3a 22 t post message:"
1460: 29 3f 3b 0a 09 09 09 09 09 09 73 71 6c 78 3a 3a )?;.......sqlx::
1470: 71 75 65 72 79 21 28 22 69 6e 73 65 72 74 20 69 query!("insert i
1480: 6e 74 6f 20 72 73 73 74 67 5f 70 6f 73 74 20 28 nto rsstg_post (
1490: 73 6f 75 72 63 65 5f 69 64 2c 20 70 6f 73 74 65 source_id, poste
14a0: 64 2c 20 75 72 6c 29 20 76 61 6c 75 65 73 20 28 d, url) values (
14b0: 24 31 2c 20 24 32 2c 20 24 33 29 3b 22 2c 0a 09 $1, $2, $3);",..
14c0: 09 09 09 09 09 09 2a 69 64 2c 20 64 61 74 65 2c ......*id, date,
14d0: 20 26 70 6f 73 74 5f 75 72 6c 29 2e 65 78 65 63 &post_url).exec
14e0: 75 74 65 28 26 6d 75 74 20 73 65 6c 66 2e 70 6f ute(&mut self.po
14f0: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
1500: 69 74 3f 29 2e 61 77 61 69 74 3f 3b 0a 09 09 09 it?).await?;....
1510: 09 09 09 74 61 73 6b 3a 3a 73 6c 65 65 70 28 73 ...task::sleep(s
1520: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
1530: 6f 6e 3a 3a 6e 65 77 28 34 2c 20 30 29 29 2e 61 on::new(4, 0)).a
1540: 77 61 69 74 3b 0a 09 09 09 09 09 7d 3b 0a 09 09 wait;......};...
1550: 09 09 7d 3b 0a 09 09 09 09 70 6f 73 74 65 64 20 ..};.....posted
1560: 2b 3d 20 31 3b 0a 09 09 09 7d 3b 0a 09 09 09 70 += 1;....};....p
1570: 6f 73 74 73 2e 63 6c 65 61 72 28 29 3b 0a 09 09 osts.clear();...
1580: 7d 3b 0a 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 };...sqlx::query
1590: 21 28 22 75 70 64 61 74 65 20 72 73 73 74 67 5f !("update rsstg_
15a0: 73 6f 75 72 63 65 20 73 65 74 20 6c 61 73 74 5f source set last_
15b0: 73 63 72 61 70 65 20 3d 20 6e 6f 77 28 29 20 77 scrape = now() w
15c0: 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d here source_id =
15d0: 20 24 31 3b 22 2c 0a 09 09 09 2a 69 64 29 2e 65 $1;",....*id).e
15e0: 78 65 63 75 74 65 28 26 6d 75 74 20 73 65 6c 66 xecute(&mut self
15f0: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
1600: 61 77 61 69 74 3f 29 2e 61 77 61 69 74 3f 3b 0a await?).await?;.
1610: 09 09 4f 6b 28 66 6f 72 6d 61 74 21 28 22 50 6f ..Ok(format!("Po
1620: 73 74 65 64 3a 20 7b 7d 22 2c 20 26 70 6f 73 74 sted: {}", &post
1630: 65 64 29 2e 69 6e 74 6f 28 29 29 0a 09 7d 0a 0a ed).into())..}..
1640: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 64 65 .pub async fn de
1650: 6c 65 74 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 lete<S>(&self, s
1660: 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 ource_id: &i32,
1670: 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 owner: S) -> Res
1680: 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e ult<Cow<'_, str>
1690: 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f >..where S: Into
16a0: 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 <i64> {...let ow
16b0: 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f ner = owner.into
16c0: 28 29 3b 0a 0a 09 09 6d 61 74 63 68 20 73 71 6c ();....match sql
16d0: 78 3a 3a 71 75 65 72 79 21 28 22 64 65 6c 65 74 x::query!("delet
16e0: 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 e from rsstg_sou
16f0: 72 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 rce where source
1700: 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e _id = $1 and own
1710: 65 72 20 3d 20 24 32 3b 22 2c 0a 09 09 09 73 6f er = $2;",....so
1720: 75 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 29 2e urce_id, owner).
1730: 65 78 65 63 75 74 65 28 26 6d 75 74 20 73 65 6c execute(&mut sel
1740: 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 f.pool.acquire()
1750: 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 74 3f 2e .await?).await?.
1760: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 rows_affected()
1770: 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 {....0 => { Ok("
1780: 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f No data found fo
1790: 75 6e 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c und.".into()) },
17a0: 0a 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f ....x => { Ok(fo
17b0: 72 6d 61 74 21 28 22 7b 7d 20 73 6f 75 72 63 65 rmat!("{} source
17c0: 73 20 72 65 6d 6f 76 65 64 2e 22 2c 20 78 29 2e s removed.", x).
17d0: 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 into()) },...}..
17e0: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
17f0: 20 63 6c 65 61 6e 3c 53 3e 28 26 73 65 6c 66 2c clean<S>(&self,
1800: 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 33 32 source_id: &i32
1810: 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 , owner: S) -> R
1820: 65 73 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 esult<Cow<'_, st
1830: 72 3e 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e r>>..where S: In
1840: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 to<i64> {...let
1850: 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e owner = owner.in
1860: 74 6f 28 29 3b 0a 0a 09 09 6d 61 74 63 68 20 73 to();....match s
1870: 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 64 65 6c qlx::query!("del
1880: 65 74 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 ete from rsstg_p
1890: 6f 73 74 20 70 20 75 73 69 6e 67 20 72 73 73 74 ost p using rsst
18a0: 67 5f 73 6f 75 72 63 65 20 73 20 77 68 65 72 65 g_source s where
18b0: 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 p.source_id = $
18c0: 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 1 and owner = $2
18d0: 20 61 6e 64 20 70 2e 73 6f 75 72 63 65 5f 69 64 and p.source_id
18e0: 20 3d 20 73 2e 73 6f 75 72 63 65 5f 69 64 3b 22 = s.source_id;"
18f0: 2c 0a 09 09 09 73 6f 75 72 63 65 5f 69 64 2c 20 ,....source_id,
1900: 6f 77 6e 65 72 29 2e 65 78 65 63 75 74 65 28 26 owner).execute(&
1910: 6d 75 74 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 mut self.pool.ac
1920: 71 75 69 72 65 28 29 2e 61 77 61 69 74 3f 29 2e quire().await?).
1930: 61 77 61 69 74 3f 2e 72 6f 77 73 5f 61 66 66 65 await?.rows_affe
1940: 63 74 65 64 28 29 20 7b 0a 09 09 09 30 20 3d 3e cted() {....0 =>
1950: 20 7b 20 4f 6b 28 22 4e 6f 20 64 61 74 61 20 66 { Ok("No data f
1960: 6f 75 6e 64 20 66 6f 75 6e 64 2e 22 2e 69 6e 74 ound found.".int
1970: 6f 28 29 29 20 7d 2c 0a 09 09 09 78 20 3d 3e 20 o()) },....x =>
1980: 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 22 7b 7d { Ok(format!("{}
1990: 20 70 6f 73 74 73 20 70 75 72 67 65 64 2e 22 2c posts purged.",
19a0: 20 78 29 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 x).into()) },..
19b0: 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e .}..}...pub asyn
19c0: 63 20 66 6e 20 65 6e 61 62 6c 65 3c 53 3e 28 26 c fn enable<S>(&
19d0: 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a self, source_id:
19e0: 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 &i32, owner: S)
19f0: 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e -> Result<&str>
1a00: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
1a10: 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e i64> {...let own
1a20: 65 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 er = owner.into(
1a30: 29 3b 0a 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 );....match sqlx
1a40: 3a 3a 71 75 65 72 79 21 28 22 75 70 64 61 74 65 ::query!("update
1a50: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 rsstg_source se
1a60: 74 20 65 6e 61 62 6c 65 64 20 3d 20 74 72 75 65 t enabled = true
1a70: 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 where source_id
1a80: 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 = $1 and owner
1a90: 3d 20 24 32 22 2c 0a 09 09 09 73 6f 75 72 63 65 = $2",....source
1aa0: 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 65 78 65 63 _id, owner).exec
1ab0: 75 74 65 28 26 6d 75 74 20 73 65 6c 66 2e 70 6f ute(&mut self.po
1ac0: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
1ad0: 69 74 3f 29 2e 61 77 61 69 74 3f 2e 72 6f 77 73 it?).await?.rows
1ae0: 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 _affected() {...
1af0: 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 .1 => { Ok("Sour
1b00: 63 65 20 65 6e 61 62 6c 65 64 2e 22 29 20 7d 2c ce enabled.") },
1b10: 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 ....0 => { Ok("S
1b20: 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 2e ource not found.
1b30: 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 ") },...._ => {
1b40: 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 61 74 Err(anyhow!("Dat
1b50: 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 29 20 abase error."))
1b60: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 },...}..}...pub
1b70: 61 73 79 6e 63 20 66 6e 20 64 69 73 61 62 6c 65 async fn disable
1b80: 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 <S>(&self, sourc
1b90: 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 e_id: &i32, owne
1ba0: 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c r: S) -> Result<
1bb0: 26 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 &str>..where S:
1bc0: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
1bd0: 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e t owner = owner.
1be0: 69 6e 74 6f 28 29 3b 0a 0a 09 09 6d 61 74 63 68 into();....match
1bf0: 20 73 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 75 sqlx::query!("u
1c00: 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 pdate rsstg_sour
1c10: 63 65 20 73 65 74 20 65 6e 61 62 6c 65 64 20 3d ce set enabled =
1c20: 20 66 61 6c 73 65 20 77 68 65 72 65 20 73 6f 75 false where sou
1c30: 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 rce_id = $1 and
1c40: 6f 77 6e 65 72 20 3d 20 24 32 22 2c 0a 09 09 09 owner = $2",....
1c50: 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 source_id, owner
1c60: 29 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 73 ).execute(&mut s
1c70: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
1c80: 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 74 ().await?).await
1c90: 3f 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 ?.rows_affected(
1ca0: 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b ) {....1 => { Ok
1cb0: 28 22 53 6f 75 72 63 65 20 64 69 73 61 62 6c 65 ("Source disable
1cc0: 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 d.") },....0 =>
1cd0: 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 { Ok("Source not
1ce0: 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 found.") },....
1cf0: 5f 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 68 6f _ => { Err(anyho
1d00: 77 21 28 22 44 61 74 61 62 61 73 65 20 65 72 72 w!("Database err
1d10: 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d or.")) },...}..}
1d20: 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 ...pub async fn
1d30: 75 70 64 61 74 65 3c 53 3e 28 26 73 65 6c 66 2c update<S>(&self,
1d40: 20 75 70 64 61 74 65 3a 20 4f 70 74 69 6f 6e 3c update: Option<
1d50: 69 33 32 3e 2c 20 63 68 61 6e 6e 65 6c 3a 20 26 i32>, channel: &
1d60: 73 74 72 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 3a str, channel_id:
1d70: 20 69 36 34 2c 20 75 72 6c 3a 20 26 73 74 72 2c i64, url: &str,
1d80: 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e iv_hash: Option
1d90: 3c 26 73 74 72 3e 2c 20 75 72 6c 5f 72 65 3a 20 <&str>, url_re:
1da0: 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 6f 77 Option<&str>, ow
1db0: 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c ner: S) -> Resul
1dc0: 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 20 53 t<&str>..where S
1dd0: 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 : Into<i64> {...
1de0: 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 let owner = owne
1df0: 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6d 61 74 r.into();....mat
1e00: 63 68 20 6d 61 74 63 68 20 75 70 64 61 74 65 20 ch match update
1e10: 7b 0a 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d {.....Some(id) =
1e20: 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 > {......sqlx::q
1e30: 75 65 72 79 21 28 22 75 70 64 61 74 65 20 72 73 uery!("update rs
1e40: 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 63 stg_source set c
1e50: 68 61 6e 6e 65 6c 5f 69 64 20 3d 20 24 32 2c 20 hannel_id = $2,
1e60: 75 72 6c 20 3d 20 24 33 2c 20 69 76 5f 68 61 73 url = $3, iv_has
1e70: 68 20 3d 20 24 34 2c 20 6f 77 6e 65 72 20 3d 20 h = $4, owner =
1e80: 24 35 2c 20 63 68 61 6e 6e 65 6c 20 3d 20 24 36 $5, channel = $6
1e90: 2c 20 75 72 6c 5f 72 65 20 3d 20 24 37 20 77 68 , url_re = $7 wh
1ea0: 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 ere source_id =
1eb0: 24 31 22 2c 0a 09 09 09 09 09 09 69 64 2c 20 63 $1",.......id, c
1ec0: 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 hannel_id, url,
1ed0: 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 iv_hash, owner,
1ee0: 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 channel, url_re)
1ef0: 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 73 65 .execute(&mut se
1f00: 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 lf.pool.acquire(
1f10: 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 74 0a ).await?).await.
1f20: 09 09 09 09 7d 2c 0a 09 09 09 09 4e 6f 6e 65 20 ....},.....None
1f30: 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a => {......sqlx::
1f40: 71 75 65 72 79 21 28 22 69 6e 73 65 72 74 20 69 query!("insert i
1f50: 6e 74 6f 20 72 73 73 74 67 5f 73 6f 75 72 63 65 nto rsstg_source
1f60: 20 28 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 (channel_id, ur
1f70: 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 l, iv_hash, owne
1f80: 72 2c 20 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f r, channel, url_
1f90: 72 65 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 re) values ($1,
1fa0: 24 32 2c 20 24 33 2c 20 24 34 2c 20 24 35 2c 20 $2, $3, $4, $5,
1fb0: 24 36 29 22 2c 0a 09 09 09 09 09 09 63 68 61 6e $6)",.......chan
1fc0: 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f nel_id, url, iv_
1fd0: 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 hash, owner, cha
1fe0: 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 2e 65 78 nnel, url_re).ex
1ff0: 65 63 75 74 65 28 26 6d 75 74 20 73 65 6c 66 2e ecute(&mut self.
2000: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
2010: 77 61 69 74 3f 29 2e 61 77 61 69 74 0a 09 09 09 wait?).await....
2020: 09 7d 2c 0a 09 09 09 7d 20 7b 0a 09 09 09 4f 6b .},....} {....Ok
2030: 28 5f 29 20 3d 3e 20 4f 6b 28 6d 61 74 63 68 20 (_) => Ok(match
2040: 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d update {.....Som
2050: 65 28 5f 29 20 3d 3e 20 22 43 68 61 6e 6e 65 6c e(_) => "Channel
2060: 20 75 70 64 61 74 65 64 2e 22 2c 0a 09 09 09 09 updated.",.....
2070: 4e 6f 6e 65 20 3d 3e 20 22 43 68 61 6e 6e 65 6c None => "Channel
2080: 20 61 64 64 65 64 2e 22 2c 0a 09 09 09 7d 29 2c added.",....}),
2090: 0a 09 09 09 45 72 72 28 73 71 6c 78 3a 3a 45 72 ....Err(sqlx::Er
20a0: 72 6f 72 3a 3a 44 61 74 61 62 61 73 65 28 65 72 ror::Database(er
20b0: 72 29 29 20 3d 3e 20 7b 0a 09 09 09 09 6d 61 74 r)) => {.....mat
20c0: 63 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a ch err.downcast:
20d0: 3a 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 :<sqlx::postgres
20e0: 3a 3a 50 67 44 61 74 61 62 61 73 65 45 72 72 6f ::PgDatabaseErro
20f0: 72 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b r>().routine() {
2100: 0a 09 09 09 09 09 53 6f 6d 65 28 22 5f 62 74 5f ......Some("_bt_
2110: 63 68 65 63 6b 5f 75 6e 69 71 75 65 22 2c 20 29 check_unique", )
2120: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 => {.......Ok("
2130: 44 75 70 6c 69 63 61 74 65 20 6b 65 79 2e 22 29 Duplicate key.")
2140: 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 53 6f ......},......So
2150: 6d 65 28 5f 29 20 3d 3e 20 7b 0a 09 09 09 09 09 me(_) => {......
2160: 09 4f 6b 28 22 44 61 74 61 62 61 73 65 20 65 72 .Ok("Database er
2170: 72 6f 72 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 ror.")......},..
2180: 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 ....None => {...
2190: 09 09 09 09 4f 6b 28 22 4e 6f 20 64 61 74 61 62 ....Ok("No datab
21a0: 61 73 65 20 65 72 72 6f 72 20 65 78 74 72 61 63 ase error extrac
21b0: 74 65 64 2e 22 29 0a 09 09 09 09 09 7d 2c 0a 09 ted.")......},..
21c0: 09 09 09 7d 0a 09 09 09 7d 2c 0a 09 09 09 45 72 ...}....},....Er
21d0: 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 r(err) => {.....
21e0: 62 61 69 6c 21 28 22 53 6f 72 72 79 2c 20 75 6e bail!("Sorry, un
21f0: 6b 6e 6f 77 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a known error:\n{:
2200: 23 3f 7d 5c 6e 22 2c 20 65 72 72 29 3b 0a 09 09 #?}\n", err);...
2210: 09 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 .},...}..}...asy
2220: 6e 63 20 66 6e 20 61 75 74 6f 66 65 74 63 68 28 nc fn autofetch(
2230: 26 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 &self) -> Result
2240: 3c 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 <std::time::Dura
2250: 74 69 6f 6e 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 tion> {...let mu
2260: 74 20 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e 6f t delay = chrono
2270: 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 ::Duration::minu
2280: 74 65 73 28 31 29 3b 0a 09 09 6c 65 74 20 6e 6f tes(1);...let no
2290: 77 20 3d 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 w = chrono::Loca
22a0: 6c 3a 3a 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 20 l::now();...let
22b0: 6d 75 74 20 71 75 65 75 65 20 3d 20 73 71 6c 78 mut queue = sqlx
22c0: 3a 3a 71 75 65 72 79 21 28 72 23 22 73 65 6c 65 ::query!(r#"sele
22d0: 63 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 6e 65 ct source_id, ne
22e0: 78 74 5f 66 65 74 63 68 20 61 73 20 22 6e 65 78 xt_fetch as "nex
22f0: 74 5f 66 65 74 63 68 3a 20 44 61 74 65 54 69 6d t_fetch: DateTim
2300: 65 3c 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3e e<chrono::Local>
2310: 22 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 ", owner from rs
2320: 73 74 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 stg_order natura
2330: 6c 20 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 l left join rsst
2340: 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 6e g_source where n
2350: 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 ext_fetch < now(
2360: 29 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 20 ) + interval '1
2370: 6d 69 6e 75 74 65 27 3b 22 23 29 0a 09 09 09 2e minute';"#).....
2380: 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 73 fetch_all(&mut s
2390: 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 elf.pool.acquire
23a0: 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 69 74 ().await?).await
23b0: 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 ?;...for row in
23c0: 71 75 65 75 65 2e 69 74 65 72 28 29 20 7b 0a 09 queue.iter() {..
23d0: 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 6e 65 ..if let Some(ne
23e0: 78 74 5f 66 65 74 63 68 29 20 3d 20 72 6f 77 2e xt_fetch) = row.
23f0: 6e 65 78 74 5f 66 65 74 63 68 20 7b 0a 09 09 09 next_fetch {....
2400: 09 69 66 20 6e 65 78 74 5f 66 65 74 63 68 20 3c .if next_fetch <
2410: 20 6e 6f 77 20 7b 0a 09 09 09 09 09 69 66 20 6c now {......if l
2420: 65 74 20 28 53 6f 6d 65 28 6f 77 6e 65 72 29 2c et (Some(owner),
2430: 20 53 6f 6d 65 28 73 6f 75 72 63 65 5f 69 64 29 Some(source_id)
2440: 29 20 3d 20 28 72 6f 77 2e 6f 77 6e 65 72 2c 20 ) = (row.owner,
2450: 72 6f 77 2e 73 6f 75 72 63 65 5f 69 64 29 20 7b row.source_id) {
2460: 0a 09 09 09 09 09 09 6c 65 74 20 63 6c 6f 6e 65 .......let clone
2470: 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 09 09 09 = Core {.......
2480: 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 74 65 6c .owner_chat: tel
2490: 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 49 egram_bot::UserI
24a0: 64 3a 3a 6e 65 77 28 6f 77 6e 65 72 29 2c 0a 09 d::new(owner),..
24b0: 09 09 09 09 09 09 2e 2e 73 65 6c 66 2e 63 6c 6f ........self.clo
24c0: 6e 65 28 29 0a 09 09 09 09 09 09 7d 3b 0a 09 09 ne().......};...
24d0: 09 09 09 09 74 61 73 6b 3a 3a 73 70 61 77 6e 28 ....task::spawn(
24e0: 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 async move {....
24f0: 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 ....if let Err(e
2500: 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 rr) = clone.chec
2510: 6b 28 26 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 k(&source_id, ow
2520: 6e 65 72 2c 20 74 72 75 65 29 2e 61 77 61 69 74 ner, true).await
2530: 20 7b 0a 09 09 09 09 09 09 09 09 69 66 20 6c 65 {.........if le
2540: 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c 6f t Err(err) = clo
2550: 6e 65 2e 73 65 6e 64 28 26 66 6f 72 6d 61 74 21 ne.send(&format!
2560: 28 22 f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 ("š {:?}", er
2570: 72 29 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e r), None, None).
2580: 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 09 09 await {.........
2590: 09 65 70 72 69 6e 74 6c 6e 21 28 22 43 68 65 63 .eprintln!("Chec
25a0: 6b 20 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 k error: {}", er
25b0: 72 29 3b 0a 09 09 09 09 09 09 09 09 09 63 6c 6f r);..........clo
25c0: 6e 65 2e 64 69 73 61 62 6c 65 28 26 73 6f 75 72 ne.disable(&sour
25d0: 63 65 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 61 77 ce_id, owner).aw
25e0: 61 69 74 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 ait.unwrap();...
25f0: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 ......};........
2600: 7d 3b 0a 09 09 09 09 09 09 7d 29 3b 0a 09 09 09 };.......});....
2610: 09 09 7d 0a 09 09 09 09 7d 20 65 6c 73 65 20 69 ..}.....} else i
2620: 66 20 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 6e f next_fetch - n
2630: 6f 77 20 3c 20 64 65 6c 61 79 20 7b 0a 09 09 09 ow < delay {....
2640: 09 09 64 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 ..delay = next_f
2650: 65 74 63 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 etch - now;.....
2660: 7d 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 71 75 }....}...};...qu
2670: 65 75 65 2e 63 6c 65 61 72 28 29 3b 0a 09 09 4f eue.clear();...O
2680: 6b 28 64 65 6c 61 79 2e 74 6f 5f 73 74 64 28 29 k(delay.to_std()
2690: 3f 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e ?)..}...pub asyn
26a0: 63 20 66 6e 20 6c 69 73 74 3c 53 3e 28 26 73 65 c fn list<S>(&se
26b0: 6c 66 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e lf, owner: S) ->
26c0: 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e 0a Result<String>.
26d0: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
26e0: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
26f0: 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 r = owner.into()
2700: 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 72 65 70 ;....let mut rep
2710: 6c 79 3a 20 56 65 63 3c 43 6f 77 3c 73 74 72 3e ly: Vec<Cow<str>
2720: 3e 20 3d 20 76 65 63 21 5b 5d 3b 0a 09 09 72 65 > = vec![];...re
2730: 70 6c 79 2e 70 75 73 68 28 22 43 68 61 6e 6e 65 ply.push("Channe
2740: 6c 73 3a 22 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 ls:".into());...
2750: 6c 65 74 20 72 6f 77 73 20 3d 20 73 71 6c 78 3a let rows = sqlx:
2760: 3a 71 75 65 72 79 21 28 22 73 65 6c 65 63 74 20 :query!("select
2770: 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e source_id, chann
2780: 65 6c 2c 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c el, enabled, url
2790: 2c 20 69 76 5f 68 61 73 68 2c 20 75 72 6c 5f 72 , iv_hash, url_r
27a0: 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 e from rsstg_sou
27b0: 72 63 65 20 77 68 65 72 65 20 6f 77 6e 65 72 20 rce where owner
27c0: 3d 20 24 31 20 6f 72 64 65 72 20 62 79 20 73 6f = $1 order by so
27d0: 75 72 63 65 5f 69 64 22 2c 0a 09 09 09 6f 77 6e urce_id",....own
27e0: 65 72 29 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d er).fetch_all(&m
27f0: 75 74 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 ut self.pool.acq
2800: 75 69 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 uire().await?).a
2810: 77 61 69 74 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 wait?;...for row
2820: 20 69 6e 20 72 6f 77 73 2e 69 74 65 72 28 29 20 in rows.iter()
2830: 7b 0a 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 {....reply.push(
2840: 66 6f 72 6d 61 74 21 28 22 5c 6e 5c 5c 23 ef b8 format!("\n\\#ļø
2850: 8f e2 83 a3 20 7b 7d 20 5c 5c 2a ef b8 8f e2 83 ⣠{} \\*ļøā
2860: a3 20 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 97 20 £ `{}` {}\nš
2870: 60 7b 7d 60 22 2c 20 72 6f 77 2e 73 6f 75 72 63 `{}`", row.sourc
2880: 65 5f 69 64 2c 20 72 6f 77 2e 63 68 61 6e 6e 65 e_id, row.channe
2890: 6c 2c 20 20 0a 09 09 09 09 6d 61 74 63 68 20 72 l, .....match r
28a0: 6f 77 2e 65 6e 61 62 6c 65 64 20 7b 0a 09 09 09 ow.enabled {....
28b0: 09 09 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 ..true => "š
28c0: 20 65 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 enabled",......
28d0: 66 61 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 false => "ā di
28e0: 73 61 62 6c 65 64 22 2c 0a 09 09 09 09 7d 2c 20 sabled",.....},
28f0: 72 6f 77 2e 75 72 6c 29 2e 69 6e 74 6f 28 29 29 row.url).into())
2900: 3b 0a 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 ;....if let Some
2910: 28 68 61 73 68 29 20 3d 20 26 72 6f 77 2e 69 76 (hash) = &row.iv
2920: 5f 68 61 73 68 20 7b 0a 09 09 09 09 72 65 70 6c _hash {.....repl
2930: 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 y.push(format!("
2940: 49 56 3a 20 60 7b 7d 60 22 2c 20 68 61 73 68 29 IV: `{}`", hash)
2950: 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 7d 0a 09 .into());....}..
2960: 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 72 65 ..if let Some(re
2970: 29 20 3d 20 26 72 6f 77 2e 75 72 6c 5f 72 65 20 ) = &row.url_re
2980: 7b 0a 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 {.....reply.push
2990: 28 66 6f 72 6d 61 74 21 28 22 52 45 3a 20 60 7b (format!("RE: `{
29a0: 7d 60 22 2c 20 72 65 29 2e 69 6e 74 6f 28 29 29 }`", re).into())
29b0: 3b 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 4f 6b ;....}...};...Ok
29c0: 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 (reply.join("\n"
29d0: 29 29 0a 09 7d 0a 7d 0a ))..}.}.