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 26 ew(x)?.execute(&
1170: 73 6f 75 72 63 65 2e 75 72 6c 29 2c 0a 09 09 09 source.url),....
1180: 09 09 4e 6f 6e 65 20 3d 3e 20 75 72 6c 2e 69 6e ..None => url.in
1190: 74 6f 28 29 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 to(),.....};....
11a0: 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 65 78 69 .if let Some(exi
11b0: 73 74 73 29 20 3d 20 73 71 6c 78 3a 3a 71 75 65 sts) = sqlx::que
11c0: 72 79 21 28 22 73 65 6c 65 63 74 20 65 78 69 73 ry!("select exis
11d0: 74 73 28 73 65 6c 65 63 74 20 74 72 75 65 20 66 ts(select true f
11e0: 72 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 77 rom rsstg_post w
11f0: 68 65 72 65 20 75 72 6c 20 3d 20 24 31 20 61 6e here url = $1 an
1200: 64 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 32 d source_id = $2
1210: 29 20 61 73 20 65 78 69 73 74 73 3b 22 2c 0a 09 ) as exists;",..
1220: 09 09 09 09 26 70 6f 73 74 5f 75 72 6c 2c 20 2a ....&post_url, *
1230: 69 64 29 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d id).fetch_one(&m
1240: 75 74 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 ut self.pool.acq
1250: 75 69 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 uire().await?).a
1260: 77 61 69 74 3f 2e 65 78 69 73 74 73 20 7b 0a 09 wait?.exists {..
1270: 09 09 09 09 69 66 20 21 20 65 78 69 73 74 73 20 ....if ! exists
1280: 7b 0a 09 09 09 09 09 09 69 66 20 74 68 69 73 5f {.......if this_
1290: 66 65 74 63 68 2e 69 73 5f 6e 6f 6e 65 28 29 20 fetch.is_none()
12a0: 7c 7c 20 2a 64 61 74 65 20 3e 20 74 68 69 73 5f || *date > this_
12b0: 66 65 74 63 68 2e 75 6e 77 72 61 70 28 29 20 7b fetch.unwrap() {
12c0: 0a 09 09 09 09 09 09 09 74 68 69 73 5f 66 65 74 ........this_fet
12d0: 63 68 20 3d 20 53 6f 6d 65 28 2a 64 61 74 65 29 ch = Some(*date)
12e0: 3b 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ;.......};......
12f0: 09 73 65 6c 66 2e 74 67 2e 73 65 6e 64 28 20 6d .self.tg.send( m
1300: 61 74 63 68 20 26 73 6f 75 72 63 65 2e 69 76 5f atch &source.iv_
1310: 68 61 73 68 20 7b 0a 09 09 09 09 09 09 09 09 53 hash {.........S
1320: 6f 6d 65 28 68 61 73 68 29 20 3d 3e 20 74 65 6c ome(hash) => tel
1330: 65 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d egram_bot::SendM
1340: 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 73 74 essage::new(dest
1350: 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 21 ination, format!
1360: 28 22 3c 61 20 68 72 65 66 3d 5c 22 68 74 74 70 ("<a href=\"http
1370: 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 72 6c 3d s://t.me/iv?url=
1380: 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c 22 3e 20 3c {}&rhash={}\"> <
1390: 2f 61 3e 7b 30 7d 22 2c 20 26 70 6f 73 74 5f 75 /a>{0}", &post_u
13a0: 72 6c 2c 20 68 61 73 68 29 29 2c 0a 09 09 09 09 rl, hash)),.....
13b0: 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 65 6c 65 ....None => tele
13c0: 67 72 61 6d 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 gram_bot::SendMe
13d0: 73 73 61 67 65 3a 3a 6e 65 77 28 64 65 73 74 69 ssage::new(desti
13e0: 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d 61 74 21 28 nation, format!(
13f0: 22 7b 7d 22 2c 20 70 6f 73 74 5f 75 72 6c 29 29 "{}", post_url))
1400: 2c 0a 09 09 09 09 09 09 09 7d 2e 70 61 72 73 65 ,........}.parse
1410: 5f 6d 6f 64 65 28 74 65 6c 65 67 72 61 6d 5f 62 _mode(telegram_b
1420: 6f 74 3a 3a 74 79 70 65 73 3a 3a 50 61 72 73 65 ot::types::Parse
1430: 4d 6f 64 65 3a 3a 48 74 6d 6c 29 29 2e 61 77 61 Mode::Html)).awa
1440: 69 74 0a 09 09 09 09 09 09 09 2e 63 6f 6e 74 65 it.........conte
1450: 78 74 28 22 43 61 6e 27 74 20 70 6f 73 74 20 6d xt("Can't post m
1460: 65 73 73 61 67 65 3a 22 29 3f 3b 0a 09 09 09 09 essage:")?;.....
1470: 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 ..sqlx::query!("
1480: 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 insert into rsst
1490: 67 5f 70 6f 73 74 20 28 73 6f 75 72 63 65 5f 69 g_post (source_i
14a0: 64 2c 20 70 6f 73 74 65 64 2c 20 75 72 6c 29 20 d, posted, url)
14b0: 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 values ($1, $2,
14c0: 24 33 29 3b 22 2c 0a 09 09 09 09 09 09 09 2a 69 $3);",........*i
14d0: 64 2c 20 64 61 74 65 2c 20 26 70 6f 73 74 5f 75 d, date, &post_u
14e0: 72 6c 29 2e 65 78 65 63 75 74 65 28 26 6d 75 74 rl).execute(&mut
14f0: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
1500: 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 re().await?).awa
1510: 69 74 3f 3b 0a 09 09 09 09 09 09 74 61 73 6b 3a it?;.......task:
1520: 3a 73 6c 65 65 70 28 73 74 64 3a 3a 74 69 6d 65 :sleep(std::time
1530: 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 ::Duration::new(
1540: 34 2c 20 30 29 29 2e 61 77 61 69 74 3b 0a 09 09 4, 0)).await;...
1550: 09 09 09 7d 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 ...};.....};....
1560: 09 70 6f 73 74 65 64 20 2b 3d 20 31 3b 0a 09 09 .posted += 1;...
1570: 09 7d 3b 0a 09 09 09 70 6f 73 74 73 2e 63 6c 65 .};....posts.cle
1580: 61 72 28 29 3b 0a 09 09 7d 3b 0a 09 09 73 71 6c ar();...};...sql
1590: 78 3a 3a 71 75 65 72 79 21 28 22 75 70 64 61 74 x::query!("updat
15a0: 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 e rsstg_source s
15b0: 65 74 20 6c 61 73 74 5f 73 63 72 61 70 65 20 3d et last_scrape =
15c0: 20 6e 6f 77 28 29 20 77 68 65 72 65 20 73 6f 75 now() where sou
15d0: 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 2c 0a 09 rce_id = $1;",..
15e0: 09 09 2a 69 64 29 2e 65 78 65 63 75 74 65 28 26 ..*id).execute(&
15f0: 6d 75 74 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 mut self.pool.ac
1600: 71 75 69 72 65 28 29 2e 61 77 61 69 74 3f 29 2e quire().await?).
1610: 61 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 66 6f 72 await?;...Ok(for
1620: 6d 61 74 21 28 22 50 6f 73 74 65 64 3a 20 7b 7d mat!("Posted: {}
1630: 22 2c 20 26 70 6f 73 74 65 64 29 2e 69 6e 74 6f ", &posted).into
1640: 28 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ())..}...pub asy
1650: 6e 63 20 66 6e 20 64 65 6c 65 74 65 3c 53 3e 28 nc fn delete<S>(
1660: 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 &self, source_id
1670: 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 : &i32, owner: S
1680: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 77 3c ) -> Result<Cow<
1690: 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 72 65 '_, str>>..where
16a0: 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a S: Into<i64> {.
16b0: 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 ..let owner = ow
16c0: 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6d ner.into();....m
16d0: 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 atch sqlx::query
16e0: 21 28 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 !("delete from r
16f0: 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 sstg_source wher
1700: 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 e source_id = $1
1710: 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 3b and owner = $2;
1720: 22 2c 0a 09 09 09 73 6f 75 72 63 65 5f 69 64 2c ",....source_id,
1730: 20 6f 77 6e 65 72 29 2e 65 78 65 63 75 74 65 28 owner).execute(
1740: 26 6d 75 74 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 &mut self.pool.a
1750: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 3f 29 cquire().await?)
1760: 2e 61 77 61 69 74 3f 2e 72 6f 77 73 5f 61 66 66 .await?.rows_aff
1770: 65 63 74 65 64 28 29 20 7b 0a 09 09 09 30 20 3d ected() {....0 =
1780: 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 61 74 61 20 > { Ok("No data
1790: 66 6f 75 6e 64 20 66 6f 75 6e 64 2e 22 2e 69 6e found found.".in
17a0: 74 6f 28 29 29 20 7d 2c 0a 09 09 09 78 20 3d 3e to()) },....x =>
17b0: 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 22 7b { Ok(format!("{
17c0: 7d 20 73 6f 75 72 63 65 73 20 72 65 6d 6f 76 65 } sources remove
17d0: 64 2e 22 2c 20 78 29 2e 69 6e 74 6f 28 29 29 20 d.", x).into())
17e0: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 },...}..}...pub
17f0: 61 73 79 6e 63 20 66 6e 20 63 6c 65 61 6e 3c 53 async fn clean<S
1800: 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f >(&self, source_
1810: 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a id: &i32, owner:
1820: 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f S) -> Result<Co
1830: 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 w<'_, str>>..whe
1840: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
1850: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
1860: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 owner.into();...
1870: 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 .match sqlx::que
1880: 72 79 21 28 22 64 65 6c 65 74 65 20 66 72 6f 6d ry!("delete from
1890: 20 72 73 73 74 67 5f 70 6f 73 74 20 70 20 75 73 rsstg_post p us
18a0: 69 6e 67 20 72 73 73 74 67 5f 73 6f 75 72 63 65 ing rsstg_source
18b0: 20 73 20 77 68 65 72 65 20 70 2e 73 6f 75 72 63 s where p.sourc
18c0: 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 e_id = $1 and ow
18d0: 6e 65 72 20 3d 20 24 32 20 61 6e 64 20 70 2e 73 ner = $2 and p.s
18e0: 6f 75 72 63 65 5f 69 64 20 3d 20 73 2e 73 6f 75 ource_id = s.sou
18f0: 72 63 65 5f 69 64 3b 22 2c 0a 09 09 09 73 6f 75 rce_id;",....sou
1900: 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 29 2e 65 rce_id, owner).e
1910: 78 65 63 75 74 65 28 26 6d 75 74 20 73 65 6c 66 xecute(&mut self
1920: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
1930: 61 77 61 69 74 3f 29 2e 61 77 61 69 74 3f 2e 72 await?).await?.r
1940: 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b ows_affected() {
1950: 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e ....0 => { Ok("N
1960: 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 o data found fou
1970: 6e 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a nd.".into()) },.
1980: 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 ...x => { Ok(for
1990: 6d 61 74 21 28 22 7b 7d 20 70 6f 73 74 73 20 70 mat!("{} posts p
19a0: 75 72 67 65 64 2e 22 2c 20 78 29 2e 69 6e 74 6f urged.", x).into
19b0: 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 ()) },...}..}...
19c0: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 65 6e 61 pub async fn ena
19d0: 62 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f ble<S>(&self, so
19e0: 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f urce_id: &i32, o
19f0: 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 wner: S) -> Resu
1a00: 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 20 lt<&str>..where
1a10: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
1a20: 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e .let owner = own
1a30: 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6d 61 er.into();....ma
1a40: 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 21 tch sqlx::query!
1a50: 28 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 ("update rsstg_s
1a60: 6f 75 72 63 65 20 73 65 74 20 65 6e 61 62 6c 65 ource set enable
1a70: 64 20 3d 20 74 72 75 65 20 77 68 65 72 65 20 73 d = true where s
1a80: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e ource_id = $1 an
1a90: 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 2c 0a 09 d owner = $2",..
1aa0: 09 09 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e ..source_id, own
1ab0: 65 72 29 2e 65 78 65 63 75 74 65 28 26 6d 75 74 er).execute(&mut
1ac0: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
1ad0: 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 re().await?).awa
1ae0: 69 74 3f 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 it?.rows_affecte
1af0: 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 d() {....1 => {
1b00: 4f 6b 28 22 53 6f 75 72 63 65 20 65 6e 61 62 6c Ok("Source enabl
1b10: 65 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e ed.") },....0 =>
1b20: 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f { Ok("Source no
1b30: 74 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 t found.") },...
1b40: 09 5f 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 68 ._ => { Err(anyh
1b50: 6f 77 21 28 22 44 61 74 61 62 61 73 65 20 65 72 ow!("Database er
1b60: 72 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 ror.")) },...}..
1b70: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
1b80: 20 64 69 73 61 62 6c 65 3c 53 3e 28 26 73 65 6c disable<S>(&sel
1b90: 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 f, source_id: &i
1ba0: 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 32, owner: S) ->
1bb0: 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 77 Result<&str>..w
1bc0: 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 here S: Into<i64
1bd0: 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 > {...let owner
1be0: 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a = owner.into();.
1bf0: 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 ...match sqlx::q
1c00: 75 65 72 79 21 28 22 75 70 64 61 74 65 20 72 73 uery!("update rs
1c10: 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 stg_source set e
1c20: 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 65 20 77 nabled = false w
1c30: 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d here source_id =
1c40: 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 $1 and owner =
1c50: 24 32 22 2c 0a 09 09 09 73 6f 75 72 63 65 5f 69 $2",....source_i
1c60: 64 2c 20 6f 77 6e 65 72 29 2e 65 78 65 63 75 74 d, owner).execut
1c70: 65 28 26 6d 75 74 20 73 65 6c 66 2e 70 6f 6f 6c e(&mut self.pool
1c80: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
1c90: 3f 29 2e 61 77 61 69 74 3f 2e 72 6f 77 73 5f 61 ?).await?.rows_a
1ca0: 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 31 ffected() {....1
1cb0: 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 => { Ok("Source
1cc0: 20 64 69 73 61 62 6c 65 64 2e 22 29 20 7d 2c 0a disabled.") },.
1cd0: 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f ...0 => { Ok("So
1ce0: 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 2e 22 urce not found."
1cf0: 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 45 ) },...._ => { E
1d00: 72 72 28 61 6e 79 68 6f 77 21 28 22 44 61 74 61 rr(anyhow!("Data
1d10: 62 61 73 65 20 65 72 72 6f 72 2e 22 29 29 20 7d base error.")) }
1d20: 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 ,...}..}...pub a
1d30: 73 79 6e 63 20 66 6e 20 75 70 64 61 74 65 3c 53 sync fn update<S
1d40: 3e 28 26 73 65 6c 66 2c 20 75 70 64 61 74 65 3a >(&self, update:
1d50: 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c 20 63 68 Option<i32>, ch
1d60: 61 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 63 68 61 annel: &str, cha
1d70: 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c 20 75 72 nnel_id: i64, ur
1d80: 6c 3a 20 26 73 74 72 2c 20 69 76 5f 68 61 73 68 l: &str, iv_hash
1d90: 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 : Option<&str>,
1da0: 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f 6e 3c 26 url_re: Option<&
1db0: 73 74 72 3e 2c 20 6f 77 6e 65 72 3a 20 53 29 20 str>, owner: S)
1dc0: 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a -> Result<&str>.
1dd0: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
1de0: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
1df0: 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 r = owner.into()
1e00: 3b 0a 0a 09 09 6d 61 74 63 68 20 6d 61 74 63 68 ;....match match
1e10: 20 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f update {.....So
1e20: 6d 65 28 69 64 29 20 3d 3e 20 7b 0a 09 09 09 09 me(id) => {.....
1e30: 09 73 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 75 .sqlx::query!("u
1e40: 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 pdate rsstg_sour
1e50: 63 65 20 73 65 74 20 63 68 61 6e 6e 65 6c 5f 69 ce set channel_i
1e60: 64 20 3d 20 24 32 2c 20 75 72 6c 20 3d 20 24 33 d = $2, url = $3
1e70: 2c 20 69 76 5f 68 61 73 68 20 3d 20 24 34 2c 20 , iv_hash = $4,
1e80: 6f 77 6e 65 72 20 3d 20 24 35 2c 20 63 68 61 6e owner = $5, chan
1e90: 6e 65 6c 20 3d 20 24 36 2c 20 75 72 6c 5f 72 65 nel = $6, url_re
1ea0: 20 3d 20 24 37 20 77 68 65 72 65 20 73 6f 75 72 = $7 where sour
1eb0: 63 65 5f 69 64 20 3d 20 24 31 22 2c 0a 09 09 09 ce_id = $1",....
1ec0: 09 09 09 69 64 2c 20 63 68 61 6e 6e 65 6c 5f 69 ...id, channel_i
1ed0: 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c d, url, iv_hash,
1ee0: 20 6f 77 6e 65 72 2c 20 63 68 61 6e 6e 65 6c 2c owner, channel,
1ef0: 20 75 72 6c 5f 72 65 29 2e 65 78 65 63 75 74 65 url_re).execute
1f00: 28 26 6d 75 74 20 73 65 6c 66 2e 70 6f 6f 6c 2e (&mut self.pool.
1f10: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 3f acquire().await?
1f20: 29 2e 61 77 61 69 74 0a 09 09 09 09 7d 2c 0a 09 ).await.....},..
1f30: 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 ...None => {....
1f40: 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 21 28 22 ..sqlx::query!("
1f50: 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 insert into rsst
1f60: 67 5f 73 6f 75 72 63 65 20 28 63 68 61 6e 6e 65 g_source (channe
1f70: 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 l_id, url, iv_ha
1f80: 73 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 6e 6e sh, owner, chann
1f90: 65 6c 2c 20 75 72 6c 5f 72 65 29 20 76 61 6c 75 el, url_re) valu
1fa0: 65 73 20 28 24 31 2c 20 24 32 2c 20 24 33 2c 20 es ($1, $2, $3,
1fb0: 24 34 2c 20 24 35 2c 20 24 36 29 22 2c 0a 09 09 $4, $5, $6)",...
1fc0: 09 09 09 09 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 ....channel_id,
1fd0: 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 url, iv_hash, ow
1fe0: 6e 65 72 2c 20 63 68 61 6e 6e 65 6c 2c 20 75 72 ner, channel, ur
1ff0: 6c 5f 72 65 29 2e 65 78 65 63 75 74 65 28 26 6d l_re).execute(&m
2000: 75 74 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 ut self.pool.acq
2010: 75 69 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 uire().await?).a
2020: 77 61 69 74 0a 09 09 09 09 7d 2c 0a 09 09 09 7d wait.....},....}
2030: 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 4f {....Ok(_) => O
2040: 6b 28 6d 61 74 63 68 20 75 70 64 61 74 65 20 7b k(match update {
2050: 0a 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 .....Some(_) =>
2060: 22 43 68 61 6e 6e 65 6c 20 75 70 64 61 74 65 64 "Channel updated
2070: 2e 22 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 .",.....None =>
2080: 22 43 68 61 6e 6e 65 6c 20 61 64 64 65 64 2e 22 "Channel added."
2090: 2c 0a 09 09 09 7d 29 2c 0a 09 09 09 45 72 72 28 ,....}),....Err(
20a0: 73 71 6c 78 3a 3a 45 72 72 6f 72 3a 3a 44 61 74 sqlx::Error::Dat
20b0: 61 62 61 73 65 28 65 72 72 29 29 20 3d 3e 20 7b abase(err)) => {
20c0: 0a 09 09 09 09 6d 61 74 63 68 20 65 72 72 2e 64 .....match err.d
20d0: 6f 77 6e 63 61 73 74 3a 3a 3c 73 71 6c 78 3a 3a owncast::<sqlx::
20e0: 70 6f 73 74 67 72 65 73 3a 3a 50 67 44 61 74 61 postgres::PgData
20f0: 62 61 73 65 45 72 72 6f 72 3e 28 29 2e 72 6f 75 baseError>().rou
2100: 74 69 6e 65 28 29 20 7b 0a 09 09 09 09 09 53 6f tine() {......So
2110: 6d 65 28 22 5f 62 74 5f 63 68 65 63 6b 5f 75 6e me("_bt_check_un
2120: 69 71 75 65 22 2c 20 29 20 3d 3e 20 7b 0a 09 09 ique", ) => {...
2130: 09 09 09 09 4f 6b 28 22 44 75 70 6c 69 63 61 74 ....Ok("Duplicat
2140: 65 20 6b 65 79 2e 22 29 0a 09 09 09 09 09 7d 2c e key.")......},
2150: 0a 09 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e ......Some(_) =>
2160: 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 44 61 74 {.......Ok("Dat
2170: 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 0a 09 abase error.")..
2180: 09 09 09 09 7d 2c 0a 09 09 09 09 09 4e 6f 6e 65 ....},......None
2190: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 => {.......Ok("
21a0: 4e 6f 20 64 61 74 61 62 61 73 65 20 65 72 72 6f No database erro
21b0: 72 20 65 78 74 72 61 63 74 65 64 2e 22 29 0a 09 r extracted.")..
21c0: 09 09 09 09 7d 2c 0a 09 09 09 09 7d 0a 09 09 09 ....},.....}....
21d0: 7d 2c 0a 09 09 09 45 72 72 28 65 72 72 29 20 3d },....Err(err) =
21e0: 3e 20 7b 0a 09 09 09 09 62 61 69 6c 21 28 22 53 > {.....bail!("S
21f0: 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 72 orry, unknown er
2200: 72 6f 72 3a 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c 20 ror:\n{:#?}\n",
2210: 65 72 72 29 3b 0a 09 09 09 7d 2c 0a 09 09 7d 0a err);....},...}.
2220: 09 7d 0a 0a 09 61 73 79 6e 63 20 66 6e 20 61 75 .}...async fn au
2230: 74 6f 66 65 74 63 68 28 26 73 65 6c 66 29 20 2d tofetch(&self) -
2240: 3e 20 52 65 73 75 6c 74 3c 73 74 64 3a 3a 74 69 > Result<std::ti
2250: 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3e 20 7b 0a me::Duration> {.
2260: 09 09 6c 65 74 20 6d 75 74 20 64 65 6c 61 79 20 ..let mut delay
2270: 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 72 61 74 69 = chrono::Durati
2280: 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 31 29 3b 0a on::minutes(1);.
2290: 09 09 6c 65 74 20 6e 6f 77 20 3d 20 63 68 72 6f ..let now = chro
22a0: 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f 77 28 29 no::Local::now()
22b0: 3b 0a 09 09 6c 65 74 20 6d 75 74 20 71 75 65 75 ;...let mut queu
22c0: 65 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 21 e = sqlx::query!
22d0: 28 72 23 22 73 65 6c 65 63 74 20 73 6f 75 72 63 (r#"select sourc
22e0: 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 63 68 e_id, next_fetch
22f0: 20 61 73 20 22 6e 65 78 74 5f 66 65 74 63 68 3a as "next_fetch:
2300: 20 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f DateTime<chrono
2310: 3a 3a 4c 6f 63 61 6c 3e 22 2c 20 6f 77 6e 65 72 ::Local>", owner
2320: 20 66 72 6f 6d 20 72 73 73 74 67 5f 6f 72 64 65 from rsstg_orde
2330: 72 20 6e 61 74 75 72 61 6c 20 6c 65 66 74 20 6a r natural left j
2340: 6f 69 6e 20 72 73 73 74 67 5f 73 6f 75 72 63 65 oin rsstg_source
2350: 20 77 68 65 72 65 20 6e 65 78 74 5f 66 65 74 63 where next_fetc
2360: 68 20 3c 20 6e 6f 77 28 29 20 2b 20 69 6e 74 65 h < now() + inte
2370: 72 76 61 6c 20 27 31 20 6d 69 6e 75 74 65 27 3b rval '1 minute';
2380: 22 23 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c "#).....fetch_al
2390: 6c 28 26 6d 75 74 20 73 65 6c 66 2e 70 6f 6f 6c l(&mut self.pool
23a0: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
23b0: 3f 29 2e 61 77 61 69 74 3f 3b 0a 09 09 66 6f 72 ?).await?;...for
23c0: 20 72 6f 77 20 69 6e 20 71 75 65 75 65 2e 69 74 row in queue.it
23d0: 65 72 28 29 20 7b 0a 09 09 09 69 66 20 6c 65 74 er() {....if let
23e0: 20 53 6f 6d 65 28 6e 65 78 74 5f 66 65 74 63 68 Some(next_fetch
23f0: 29 20 3d 20 72 6f 77 2e 6e 65 78 74 5f 66 65 74 ) = row.next_fet
2400: 63 68 20 7b 0a 09 09 09 09 69 66 20 6e 65 78 74 ch {.....if next
2410: 5f 66 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a 09 _fetch < now {..
2420: 09 09 09 09 69 66 20 6c 65 74 20 28 53 6f 6d 65 ....if let (Some
2430: 28 6f 77 6e 65 72 29 2c 20 53 6f 6d 65 28 73 6f (owner), Some(so
2440: 75 72 63 65 5f 69 64 29 29 20 3d 20 28 72 6f 77 urce_id)) = (row
2450: 2e 6f 77 6e 65 72 2c 20 72 6f 77 2e 73 6f 75 72 .owner, row.sour
2460: 63 65 5f 69 64 29 20 7b 0a 09 09 09 09 09 09 6c ce_id) {.......l
2470: 65 74 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 65 20 et clone = Core
2480: 7b 0a 09 09 09 09 09 09 09 6f 77 6e 65 72 5f 63 {........owner_c
2490: 68 61 74 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f hat: telegram_bo
24a0: 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 6f t::UserId::new(o
24b0: 77 6e 65 72 29 2c 0a 09 09 09 09 09 09 09 2e 2e wner),..........
24c0: 73 65 6c 66 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 self.clone()....
24d0: 09 09 09 7d 3b 0a 09 09 09 09 09 09 74 61 73 6b ...};.......task
24e0: 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 6d 6f ::spawn(async mo
24f0: 76 65 20 7b 0a 09 09 09 09 09 09 09 69 66 20 6c ve {........if l
2500: 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 6c et Err(err) = cl
2510: 6f 6e 65 2e 63 68 65 63 6b 28 26 73 6f 75 72 63 one.check(&sourc
2520: 65 5f 69 64 2c 20 6f 77 6e 65 72 2c 20 74 72 75 e_id, owner, tru
2530: 65 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 e).await {......
2540: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
2550: 72 29 20 3d 20 63 6c 6f 6e 65 2e 73 65 6e 64 28 r) = clone.send(
2560: 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 7b &format!("š {
2570: 3a 3f 7d 22 2c 20 65 72 72 29 2c 20 4e 6f 6e 65 :?}", err), None
2580: 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 74 20 7b 0a , None).await {.
2590: 09 09 09 09 09 09 09 09 09 65 70 72 69 6e 74 6c .........eprintl
25a0: 6e 21 28 22 43 68 65 63 6b 20 65 72 72 6f 72 3a n!("Check error:
25b0: 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 09 {}", err);.....
25c0: 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 7d 3b ....};........};
25d0: 0a 09 09 09 09 09 09 7d 29 3b 0a 09 09 09 09 09 .......});......
25e0: 7d 0a 09 09 09 09 7d 20 65 6c 73 65 20 69 66 20 }.....} else if
25f0: 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 next_fetch - now
2600: 20 3c 20 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 < delay {......
2610: 64 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 delay = next_fet
2620: 63 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 7d 0a ch - now;.....}.
2630: 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 71 75 65 75 ...}...};...queu
2640: 65 2e 63 6c 65 61 72 28 29 3b 0a 09 09 4f 6b 28 e.clear();...Ok(
2650: 64 65 6c 61 79 2e 74 6f 5f 73 74 64 28 29 3f 29 delay.to_std()?)
2660: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
2670: 66 6e 20 6c 69 73 74 3c 53 3e 28 26 73 65 6c 66 fn list<S>(&self
2680: 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 , owner: S) -> R
2690: 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 esult<String>..w
26a0: 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 here S: Into<i64
26b0: 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 > {...let owner
26c0: 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a = owner.into();.
26d0: 0a 09 09 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 ...let mut reply
26e0: 3a 20 56 65 63 3c 43 6f 77 3c 73 74 72 3e 3e 20 : Vec<Cow<str>>
26f0: 3d 20 76 65 63 21 5b 5d 3b 0a 09 09 72 65 70 6c = vec![];...repl
2700: 79 2e 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 y.push("Channels
2710: 3a 22 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 6c 65 :".into());...le
2720: 74 20 72 6f 77 73 20 3d 20 73 71 6c 78 3a 3a 71 t rows = sqlx::q
2730: 75 65 72 79 21 28 22 73 65 6c 65 63 74 20 73 6f uery!("select so
2740: 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c urce_id, channel
2750: 2c 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 , enabled, url,
2760: 69 76 5f 68 61 73 68 2c 20 75 72 6c 5f 72 65 20 iv_hash, url_re
2770: 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 from rsstg_sourc
2780: 65 20 77 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 e where owner =
2790: 24 31 20 6f 72 64 65 72 20 62 79 20 73 6f 75 72 $1 order by sour
27a0: 63 65 5f 69 64 22 2c 0a 09 09 09 6f 77 6e 65 72 ce_id",....owner
27b0: 29 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 ).fetch_all(&mut
27c0: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
27d0: 72 65 28 29 2e 61 77 61 69 74 3f 29 2e 61 77 61 re().await?).awa
27e0: 69 74 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 it?;...for row i
27f0: 6e 20 72 6f 77 73 2e 69 74 65 72 28 29 20 7b 0a n rows.iter() {.
2800: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f ...reply.push(fo
2810: 72 6d 61 74 21 28 22 5c 6e 5c 5c 23 ef b8 8f e2 rmat!("\n\\#ļøā
2820: 83 a3 20 7b 7d 20 5c 5c 2a ef b8 8f e2 83 a3 20 £ {} \\*ļøā£
2830: 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 97 20 60 7b `{}` {}\nš `{
2840: 7d 60 22 2c 20 72 6f 77 2e 73 6f 75 72 63 65 5f }`", row.source_
2850: 69 64 2c 20 72 6f 77 2e 63 68 61 6e 6e 65 6c 2c id, row.channel,
2860: 20 20 0a 09 09 09 09 6d 61 74 63 68 20 72 6f 77 .....match row
2870: 2e 65 6e 61 62 6c 65 64 20 7b 0a 09 09 09 09 09 .enabled {......
2880: 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 20 65 true => "š e
2890: 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 66 61 nabled",......fa
28a0: 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 73 61 lse => "ā disa
28b0: 62 6c 65 64 22 2c 0a 09 09 09 09 7d 2c 20 72 6f bled",.....}, ro
28c0: 77 2e 75 72 6c 29 2e 69 6e 74 6f 28 29 29 3b 0a w.url).into());.
28d0: 09 09 09 69 66 20 6c 65 74 20 53 6f 6d 65 28 68 ...if let Some(h
28e0: 61 73 68 29 20 3d 20 26 72 6f 77 2e 69 76 5f 68 ash) = &row.iv_h
28f0: 61 73 68 20 7b 0a 09 09 09 09 72 65 70 6c 79 2e ash {.....reply.
2900: 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 49 56 push(format!("IV
2910: 3a 20 60 7b 7d 60 22 2c 20 68 61 73 68 29 2e 69 : `{}`", hash).i
2920: 6e 74 6f 28 29 29 3b 0a 09 09 09 7d 0a 09 09 09 nto());....}....
2930: 69 66 20 6c 65 74 20 53 6f 6d 65 28 72 65 29 20 if let Some(re)
2940: 3d 20 26 72 6f 77 2e 75 72 6c 5f 72 65 20 7b 0a = &row.url_re {.
2950: 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 ....reply.push(f
2960: 6f 72 6d 61 74 21 28 22 52 45 3a 20 60 7b 7d 60 ormat!("RE: `{}`
2970: 22 2c 20 72 65 29 2e 69 6e 74 6f 28 29 29 3b 0a ", re).into());.
2980: 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 4f 6b 28 72 ...}...};...Ok(r
2990: 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 29 29 eply.join("\n"))
29a0: 0a 09 7d 0a 7d 0a ..}.}.