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 7b 0a 09 70 6f 73 74 67 72 65 73 3a 3a 50 ::{..postgres::P
0070: 67 50 6f 6f 6c 4f 70 74 69 6f 6e 73 2c 0a 09 52 gPoolOptions,..R
0080: 6f 77 2c 0a 7d 3b 0a 75 73 65 20 73 74 64 3a 3a ow,.};.use std::
0090: 7b 0a 09 62 6f 72 72 6f 77 3a 3a 43 6f 77 2c 0a {..borrow::Cow,.
00a0: 09 63 6f 6c 6c 65 63 74 69 6f 6e 73 3a 3a 7b 0a .collections::{.
00b0: 09 09 42 54 72 65 65 4d 61 70 2c 0a 09 09 48 61 ..BTreeMap,...Ha
00c0: 73 68 53 65 74 2c 0a 09 7d 2c 0a 09 73 79 6e 63 shSet,..},..sync
00d0: 3a 3a 7b 41 72 63 2c 20 4d 75 74 65 78 7d 2c 0a ::{Arc, Mutex},.
00e0: 7d 3b 0a 0a 23 5b 64 65 72 69 76 65 28 43 6c 6f };..#[derive(Clo
00f0: 6e 65 29 5d 0a 70 75 62 20 73 74 72 75 63 74 20 ne)].pub struct
0100: 43 6f 72 65 20 7b 0a 09 6f 77 6e 65 72 5f 63 68 Core {..owner_ch
0110: 61 74 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 at: telegram_bot
0120: 3a 3a 55 73 65 72 49 64 2c 0a 09 70 75 62 20 74 ::UserId,..pub t
0130: 67 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a g: telegram_bot:
0140: 3a 41 70 69 2c 0a 09 70 75 62 20 6d 79 3a 20 74 :Api,..pub my: t
0150: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 elegram_bot::Use
0160: 72 2c 0a 09 70 6f 6f 6c 3a 20 73 71 6c 78 3a 3a r,..pool: sqlx::
0170: 50 6f 6f 6c 3c 73 71 6c 78 3a 3a 50 6f 73 74 67 Pool<sqlx::Postg
0180: 72 65 73 3e 2c 0a 09 73 6f 75 72 63 65 73 3a 20 res>,..sources:
0190: 41 72 63 3c 4d 75 74 65 78 3c 48 61 73 68 53 65 Arc<Mutex<HashSe
01a0: 74 3c 41 72 63 3c 69 33 32 3e 3e 3e 3e 2c 0a 7d t<Arc<i32>>>>,.}
01b0: 0a 0a 69 6d 70 6c 20 43 6f 72 65 20 7b 0a 09 70 ..impl Core {..p
01c0: 75 62 20 66 6e 20 6e 65 77 28 73 65 74 74 69 6e ub fn new(settin
01d0: 67 73 3a 20 63 6f 6e 66 69 67 3a 3a 43 6f 6e 66 gs: config::Conf
01e0: 69 67 29 20 2d 3e 20 52 65 73 75 6c 74 3c 41 72 ig) -> Result<Ar
01f0: 63 3c 43 6f 72 65 3e 3e 20 7b 0a 09 09 6c 65 74 c<Core>> {...let
0200: 20 6f 77 6e 65 72 20 3d 20 73 65 74 74 69 6e 67 owner = setting
0210: 73 2e 67 65 74 5f 69 6e 74 28 22 6f 77 6e 65 72 s.get_int("owner
0220: 22 29 3f 3b 0a 09 09 6c 65 74 20 61 70 69 5f 6b ")?;...let api_k
0230: 65 79 20 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 ey = settings.ge
0240: 74 5f 73 74 72 69 6e 67 28 22 61 70 69 5f 6b 65 t_string("api_ke
0250: 79 22 29 3f 3b 0a 09 09 6c 65 74 20 74 67 20 3d y")?;...let tg =
0260: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 41 telegram_bot::A
0270: 70 69 3a 3a 6e 65 77 28 61 70 69 5f 6b 65 79 29 pi::new(api_key)
0280: 3b 0a 09 09 6c 65 74 20 74 67 5f 63 6c 6f 6e 65 ;...let tg_clone
0290: 64 20 3d 20 74 67 2e 63 6c 6f 6e 65 28 29 3b 0a d = tg.clone();.
02a0: 09 09 6c 65 74 20 63 6f 72 65 20 3d 20 41 72 63 ..let core = Arc
02b0: 3a 3a 6e 65 77 28 43 6f 72 65 20 7b 0a 09 09 09 ::new(Core {....
02c0: 74 67 2c 0a 09 09 09 6d 79 3a 20 74 61 73 6b 3a tg,....my: task:
02d0: 3a 62 6c 6f 63 6b 5f 6f 6e 28 61 73 79 6e 63 20 :block_on(async
02e0: 7b 0a 09 09 09 09 74 67 5f 63 6c 6f 6e 65 64 2e {.....tg_cloned.
02f0: 73 65 6e 64 28 74 65 6c 65 67 72 61 6d 5f 62 6f send(telegram_bo
0300: 74 3a 3a 47 65 74 4d 65 29 2e 61 77 61 69 74 0a t::GetMe).await.
0310: 09 09 09 7d 29 3f 2c 0a 09 09 09 6f 77 6e 65 72 ...})?,....owner
0320: 5f 63 68 61 74 3a 20 74 65 6c 65 67 72 61 6d 5f _chat: telegram_
0330: 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 bot::UserId::new
0340: 28 6f 77 6e 65 72 29 2c 0a 09 09 09 70 6f 6f 6c (owner),....pool
0350: 3a 20 50 67 50 6f 6f 6c 4f 70 74 69 6f 6e 73 3a : PgPoolOptions:
0360: 3a 6e 65 77 28 29 0a 09 09 09 09 2e 6d 61 78 5f :new()......max_
0370: 63 6f 6e 6e 65 63 74 69 6f 6e 73 28 35 29 0a 09 connections(5)..
0380: 09 09 09 2e 61 63 71 75 69 72 65 5f 74 69 6d 65 ....acquire_time
0390: 6f 75 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 out(std::time::D
03a0: 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 33 30 30 uration::new(300
03b0: 2c 20 30 29 29 0a 09 09 09 09 2e 69 64 6c 65 5f , 0))......idle_
03c0: 74 69 6d 65 6f 75 74 28 73 74 64 3a 3a 74 69 6d timeout(std::tim
03d0: 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 e::Duration::new
03e0: 28 36 30 2c 20 30 29 29 0a 09 09 09 09 2e 63 6f (60, 0))......co
03f0: 6e 6e 65 63 74 5f 6c 61 7a 79 28 26 73 65 74 74 nnect_lazy(&sett
0400: 69 6e 67 73 2e 67 65 74 5f 73 74 72 69 6e 67 28 ings.get_string(
0410: 22 70 67 22 29 3f 29 3f 2c 0a 09 09 09 73 6f 75 "pg")?)?,....sou
0420: 72 63 65 73 3a 20 41 72 63 3a 3a 6e 65 77 28 4d rces: Arc::new(M
0430: 75 74 65 78 3a 3a 6e 65 77 28 48 61 73 68 53 65 utex::new(HashSe
0440: 74 3a 3a 6e 65 77 28 29 29 29 2c 0a 09 09 7d 29 t::new())),...})
0450: 3b 0a 09 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 ;...let clone =
0460: 63 6f 72 65 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 core.clone();...
0470: 74 61 73 6b 3a 3a 73 70 61 77 6e 28 61 73 79 6e task::spawn(asyn
0480: 63 20 6d 6f 76 65 20 7b 0a 09 09 09 6c 6f 6f 70 c move {....loop
0490: 20 7b 0a 09 09 09 09 6c 65 74 20 64 65 6c 61 79 {.....let delay
04a0: 20 3d 20 6d 61 74 63 68 20 26 63 6c 6f 6e 65 2e = match &clone.
04b0: 61 75 74 6f 66 65 74 63 68 28 29 2e 61 77 61 69 autofetch().awai
04c0: 74 20 7b 0a 09 09 09 09 09 45 72 72 28 65 72 72 t {......Err(err
04d0: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 69 66 20 ) => {.......if
04e0: 6c 65 74 20 45 72 72 28 65 72 72 29 20 3d 20 63 let Err(err) = c
04f0: 6c 6f 6e 65 2e 73 65 6e 64 28 66 6f 72 6d 61 74 lone.send(format
0500: 21 28 22 f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 65 !("š {:?}", e
0510: 72 72 29 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 rr), None, None)
0520: 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 09 .await {........
0530: 65 70 72 69 6e 74 6c 6e 21 28 22 41 75 74 6f 66 eprintln!("Autof
0540: 65 74 63 68 20 65 72 72 6f 72 3a 20 7b 7d 22 2c etch error: {}",
0550: 20 65 72 72 29 3b 0a 09 09 09 09 09 09 7d 3b 0a err);.......};.
0560: 09 09 09 09 09 09 73 74 64 3a 3a 74 69 6d 65 3a ......std::time:
0570: 3a 44 75 72 61 74 69 6f 6e 3a 3a 66 72 6f 6d 5f :Duration::from_
0580: 73 65 63 73 28 36 30 29 0a 09 09 09 09 09 7d 2c secs(60)......},
0590: 0a 09 09 09 09 09 4f 6b 28 74 69 6d 65 29 20 3d ......Ok(time) =
05a0: 3e 20 2a 74 69 6d 65 2c 0a 09 09 09 09 7d 3b 0a > *time,.....};.
05b0: 09 09 09 09 74 61 73 6b 3a 3a 73 6c 65 65 70 28 ....task::sleep(
05c0: 64 65 6c 61 79 29 2e 61 77 61 69 74 3b 0a 09 09 delay).await;...
05d0: 09 7d 0a 09 09 7d 29 3b 0a 09 09 4f 6b 28 63 6f .}...});...Ok(co
05e0: 72 65 29 0a 09 7d 0a 0a 09 70 75 62 20 66 6e 20 re)..}...pub fn
05f0: 73 74 72 65 61 6d 28 26 73 65 6c 66 29 20 2d 3e stream(&self) ->
0600: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 telegram_bot::U
0610: 70 64 61 74 65 73 53 74 72 65 61 6d 20 7b 0a 09 pdatesStream {..
0620: 09 73 65 6c 66 2e 74 67 2e 73 74 72 65 61 6d 28 .self.tg.stream(
0630: 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 )..}...pub async
0640: 20 66 6e 20 73 65 6e 64 3c 27 61 2c 20 53 3e 28 fn send<'a, S>(
0650: 26 73 65 6c 66 2c 20 6d 73 67 3a 20 53 2c 20 74 &self, msg: S, t
0660: 61 72 67 65 74 3a 20 4f 70 74 69 6f 6e 3c 74 65 arget: Option<te
0670: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 73 65 72 legram_bot::User
0680: 49 64 3e 2c 20 6d 6f 64 65 3a 20 4f 70 74 69 6f Id>, mode: Optio
0690: 6e 3c 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a n<telegram_bot::
06a0: 74 79 70 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 types::ParseMode
06b0: 3e 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e >) -> Result<()>
06c0: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
06d0: 43 6f 77 3c 27 61 2c 20 73 74 72 3e 3e 20 7b 0a Cow<'a, str>> {.
06e0: 09 09 6c 65 74 20 6d 6f 64 65 20 3d 20 6d 6f 64 ..let mode = mod
06f0: 65 2e 75 6e 77 72 61 70 5f 6f 72 28 74 65 6c 65 e.unwrap_or(tele
0700: 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a gram_bot::types:
0710: 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c :ParseMode::Html
0720: 29 3b 0a 09 09 6c 65 74 20 74 61 72 67 65 74 20 );...let target
0730: 3d 20 74 61 72 67 65 74 2e 75 6e 77 72 61 70 5f = target.unwrap_
0740: 6f 72 28 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 or(self.owner_ch
0750: 61 74 29 3b 0a 09 09 73 65 6c 66 2e 74 67 2e 73 at);...self.tg.s
0760: 65 6e 64 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 end(telegram_bot
0770: 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e ::SendMessage::n
0780: 65 77 28 74 61 72 67 65 74 2c 20 6d 73 67 29 2e ew(target, msg).
0790: 70 61 72 73 65 5f 6d 6f 64 65 28 6d 6f 64 65 29 parse_mode(mode)
07a0: 29 2e 61 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 28 ).await?;...Ok((
07b0: 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e ))..}...pub asyn
07c0: 63 20 66 6e 20 63 68 65 63 6b 3c 53 3e 28 26 73 c fn check<S>(&s
07d0: 65 6c 66 2c 20 69 64 3a 20 26 69 33 32 2c 20 6f elf, id: &i32, o
07e0: 77 6e 65 72 3a 20 53 2c 20 72 65 61 6c 3a 20 62 wner: S, real: b
07f0: 6f 6f 6c 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 ool) -> Result<C
0800: 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 ow<'_, str>>..wh
0810: 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e ere S: Into<i64>
0820: 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d {...let owner =
0830: 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a owner.into();..
0840: 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 74 65 64 ..let mut posted
0850: 3a 20 69 33 32 20 3d 20 30 3b 0a 09 09 6c 65 74 : i32 = 0;...let
0860: 20 69 64 20 3d 20 7b 0a 09 09 09 6c 65 74 20 6d id = {....let m
0870: 75 74 20 73 65 74 20 3d 20 73 65 6c 66 2e 73 6f ut set = self.so
0880: 75 72 63 65 73 2e 6c 6f 63 6b 28 29 2e 75 6e 77 urces.lock().unw
0890: 72 61 70 28 29 3b 0a 09 09 09 6d 61 74 63 68 20 rap();....match
08a0: 73 65 74 2e 67 65 74 28 69 64 29 20 7b 0a 09 09 set.get(id) {...
08b0: 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 69 64 ..Some(id) => id
08c0: 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 09 4e 6f .clone(),.....No
08d0: 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 6c 65 74 ne => {......let
08e0: 20 69 64 20 3d 20 41 72 63 3a 3a 6e 65 77 28 2a id = Arc::new(*
08f0: 69 64 29 3b 0a 09 09 09 09 09 73 65 74 2e 69 6e id);......set.in
0900: 73 65 72 74 28 69 64 2e 63 6c 6f 6e 65 28 29 29 sert(id.clone())
0910: 3b 0a 09 09 09 09 09 69 64 2e 63 6c 6f 6e 65 28 ;......id.clone(
0920: 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 ).....},....}...
0930: 7d 3b 0a 09 09 6c 65 74 20 63 6f 75 6e 74 20 3d };...let count =
0940: 20 41 72 63 3a 3a 73 74 72 6f 6e 67 5f 63 6f 75 Arc::strong_cou
0950: 6e 74 28 26 69 64 29 3b 0a 09 09 69 66 20 63 6f nt(&id);...if co
0960: 75 6e 74 20 3d 3d 20 32 20 7b 0a 09 09 09 6c 65 unt == 2 {....le
0970: 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c t mut conn = sel
0980: 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 f.pool.acquire()
0990: 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 74 68 .await......with
09a0: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
09b0: 61 74 21 28 22 51 75 65 72 79 20 71 75 65 75 65 at!("Query queue
09c0: 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a fetch conn:\n{:
09d0: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
09e0: 29 3f 3b 0a 09 09 09 6c 65 74 20 72 6f 77 20 3d )?;....let row =
09f0: 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 sqlx::query("se
0a00: 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 lect source_id,
0a10: 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c channel_id, url,
0a20: 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c iv_hash, owner,
0a30: 20 75 72 6c 5f 72 65 20 66 72 6f 6d 20 72 73 73 url_re from rss
0a40: 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 tg_source where
0a50: 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 source_id = $1 a
0a60: 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a nd owner = $2").
0a70: 09 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 .....bind(*id)..
0a80: 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a ....bind(owner).
0a90: 09 09 09 09 2e 66 65 74 63 68 5f 6f 6e 65 28 26 .....fetch_one(&
0aa0: 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a mut conn).await.
0ab0: 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
0ac0: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 51 75 t(|| format!("Qu
0ad0: 65 72 79 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f ery source:\n{:?
0ae0: 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 }", &self.pool))
0af0: 3f 3b 0a 09 09 09 64 72 6f 70 28 63 6f 6e 6e 29 ?;....drop(conn)
0b00: 3b 0a 09 09 09 6c 65 74 20 63 68 61 6e 6e 65 6c ;....let channel
0b10: 5f 69 64 3a 20 69 36 34 20 3d 20 72 6f 77 2e 74 _id: i64 = row.t
0b20: 72 79 5f 67 65 74 28 22 63 68 61 6e 6e 65 6c 5f ry_get("channel_
0b30: 69 64 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 id")?;....let ur
0b40: 6c 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 l: &str = row.tr
0b50: 79 5f 67 65 74 28 22 75 72 6c 22 29 3f 3b 0a 09 y_get("url")?;..
0b60: 09 09 6c 65 74 20 69 76 5f 68 61 73 68 3a 20 4f ..let iv_hash: O
0b70: 70 74 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 72 6f ption<&str> = ro
0b80: 77 2e 74 72 79 5f 67 65 74 28 22 69 76 5f 68 61 w.try_get("iv_ha
0b90: 73 68 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 sh")?;....let ur
0ba0: 6c 5f 72 65 20 3d 20 6d 61 74 63 68 20 72 6f 77 l_re = match row
0bb0: 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 5f 72 65 .try_get("url_re
0bc0: 22 29 3f 20 7b 0a 09 09 09 09 53 6f 6d 65 28 78 ")? {.....Some(x
0bd0: 29 20 3d 3e 20 53 6f 6d 65 28 73 65 64 72 65 67 ) => Some(sedreg
0be0: 65 78 3a 3a 52 65 70 6c 61 63 65 43 6f 6d 6d 61 ex::ReplaceComma
0bf0: 6e 64 3a 3a 6e 65 77 28 78 29 3f 29 2c 0a 09 09 nd::new(x)?),...
0c00: 09 09 4e 6f 6e 65 20 3d 3e 20 4e 6f 6e 65 2c 0a ..None => None,.
0c10: 09 09 09 7d 3b 0a 09 09 09 6c 65 74 20 64 65 73 ...};....let des
0c20: 74 69 6e 61 74 69 6f 6e 20 3d 20 6d 61 74 63 68 tination = match
0c30: 20 72 65 61 6c 20 7b 0a 09 09 09 09 74 72 75 65 real {.....true
0c40: 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 => telegram_bot
0c50: 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 63 68 ::UserId::new(ch
0c60: 61 6e 6e 65 6c 5f 69 64 29 2c 0a 09 09 09 09 66 annel_id),.....f
0c70: 61 6c 73 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d alse => telegram
0c80: 5f 62 6f 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 _bot::UserId::ne
0c90: 77 28 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 6f w(row.try_get("o
0ca0: 77 6e 65 72 22 29 3f 29 2c 0a 09 09 09 7d 3b 0a wner")?),....};.
0cb0: 09 09 09 6c 65 74 20 6d 75 74 20 74 68 69 73 5f ...let mut this_
0cc0: 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e 3c 44 61 fetch: Option<Da
0cd0: 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 teTime<chrono::F
0ce0: 69 78 65 64 4f 66 66 73 65 74 3e 3e 20 3d 20 4e ixedOffset>> = N
0cf0: 6f 6e 65 3b 0a 09 09 09 6c 65 74 20 6d 75 74 20 one;....let mut
0d00: 70 6f 73 74 73 3a 20 42 54 72 65 65 4d 61 70 3c posts: BTreeMap<
0d10: 44 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a DateTime<chrono:
0d20: 3a 46 69 78 65 64 4f 66 66 73 65 74 3e 2c 20 53 :FixedOffset>, S
0d30: 74 72 69 6e 67 3e 20 3d 20 42 54 72 65 65 4d 61 tring> = BTreeMa
0d40: 70 3a 3a 6e 65 77 28 29 3b 0a 09 09 09 6c 65 74 p::new();....let
0d50: 20 72 65 73 70 6f 6e 73 65 20 3d 20 72 65 71 77 response = reqw
0d60: 65 73 74 3a 3a 67 65 74 28 75 72 6c 29 2e 61 77 est::get(url).aw
0d70: 61 69 74 3f 3b 0a 09 09 09 6c 65 74 20 73 74 61 ait?;....let sta
0d80: 74 75 73 20 3d 20 72 65 73 70 6f 6e 73 65 2e 73 tus = response.s
0d90: 74 61 74 75 73 28 29 3b 0a 09 09 09 6c 65 74 20 tatus();....let
0da0: 63 6f 6e 74 65 6e 74 20 3d 20 72 65 73 70 6f 6e content = respon
0db0: 73 65 2e 62 79 74 65 73 28 29 2e 61 77 61 69 74 se.bytes().await
0dc0: 3f 3b 0a 09 09 09 6d 61 74 63 68 20 72 73 73 3a ?;....match rss:
0dd0: 3a 43 68 61 6e 6e 65 6c 3a 3a 72 65 61 64 5f 66 :Channel::read_f
0de0: 72 6f 6d 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d rom(&content[..]
0df0: 29 20 7b 0a 09 09 09 09 4f 6b 28 66 65 65 64 29 ) {.....Ok(feed)
0e00: 20 3d 3e 20 7b 0a 09 09 09 09 09 66 6f 72 20 69 => {......for i
0e10: 74 65 6d 20 69 6e 20 66 65 65 64 2e 69 74 65 6d tem in feed.item
0e20: 73 28 29 20 7b 0a 09 09 09 09 09 09 69 66 20 6c s() {.......if l
0e30: 65 74 20 53 6f 6d 65 28 6c 69 6e 6b 29 20 3d 20 et Some(link) =
0e40: 69 74 65 6d 2e 6c 69 6e 6b 28 29 20 7b 0a 09 09 item.link() {...
0e50: 09 09 09 09 09 6c 65 74 20 64 61 74 65 20 3d 20 .....let date =
0e60: 6d 61 74 63 68 20 69 74 65 6d 2e 70 75 62 5f 64 match item.pub_d
0e70: 61 74 65 28 29 20 7b 0a 09 09 09 09 09 09 09 09 ate() {.........
0e80: 53 6f 6d 65 28 66 65 65 64 5f 64 61 74 65 29 20 Some(feed_date)
0e90: 3d 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 => DateTime::par
0ea0: 73 65 5f 66 72 6f 6d 5f 72 66 63 32 38 32 32 28 se_from_rfc2822(
0eb0: 66 65 65 64 5f 64 61 74 65 29 2c 0a 09 09 09 09 feed_date),.....
0ec0: 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 44 61 74 65 ....None => Date
0ed0: 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d Time::parse_from
0ee0: 5f 72 66 63 33 33 33 39 28 26 69 74 65 6d 2e 64 _rfc3339(&item.d
0ef0: 75 62 6c 69 6e 5f 63 6f 72 65 5f 65 78 74 28 29 ublin_core_ext()
0f00: 2e 75 6e 77 72 61 70 28 29 2e 64 61 74 65 73 28 .unwrap().dates(
0f10: 29 5b 30 5d 29 2c 0a 09 09 09 09 09 09 09 7d 3f )[0]),........}?
0f20: 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c ;........let url
0f30: 20 3d 20 6c 69 6e 6b 3b 0a 09 09 09 09 09 09 09 = link;........
0f40: 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 64 61 74 posts.insert(dat
0f50: 65 2c 20 75 72 6c 2e 74 6f 5f 73 74 72 69 6e 67 e, url.to_string
0f60: 28 29 29 3b 0a 09 09 09 09 09 09 7d 0a 09 09 09 ());.......}....
0f70: 09 09 7d 3b 0a 09 09 09 09 7d 2c 0a 09 09 09 09 ..};.....},.....
0f80: 45 72 72 28 65 72 72 29 20 3d 3e 20 6d 61 74 63 Err(err) => matc
0f90: 68 20 65 72 72 20 7b 0a 09 09 09 09 09 72 73 73 h err {......rss
0fa0: 3a 3a 45 72 72 6f 72 3a 3a 49 6e 76 61 6c 69 64 ::Error::Invalid
0fb0: 53 74 61 72 74 54 61 67 20 3d 3e 20 7b 0a 09 09 StartTag => {...
0fc0: 09 09 09 09 6c 65 74 20 66 65 65 64 20 3d 20 61 ....let feed = a
0fd0: 74 6f 6d 5f 73 79 6e 64 69 63 61 74 69 6f 6e 3a tom_syndication:
0fe0: 3a 46 65 65 64 3a 3a 72 65 61 64 5f 66 72 6f 6d :Feed::read_from
0ff0: 28 26 63 6f 6e 74 65 6e 74 5b 2e 2e 5d 29 0a 09 (&content[..])..
1000: 09 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 .......with_cont
1010: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1020: 50 72 6f 62 6c 65 6d 20 6f 70 65 6e 69 6e 67 20 Problem opening
1030: 66 65 65 64 20 75 72 6c 3a 5c 6e 7b 7d 5c 6e 7b feed url:\n{}\n{
1040: 7d 22 2c 20 26 75 72 6c 2c 20 73 74 61 74 75 73 }", &url, status
1050: 29 29 3f 3b 0a 09 09 09 09 09 09 66 6f 72 20 69 ))?;.......for i
1060: 74 65 6d 20 69 6e 20 66 65 65 64 2e 65 6e 74 72 tem in feed.entr
1070: 69 65 73 28 29 20 7b 0a 09 09 09 09 09 09 09 6c ies() {........l
1080: 65 74 20 64 61 74 65 20 3d 20 69 74 65 6d 2e 70 et date = item.p
1090: 75 62 6c 69 73 68 65 64 28 29 2e 75 6e 77 72 61 ublished().unwra
10a0: 70 28 29 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 p();........let
10b0: 75 72 6c 20 3d 20 69 74 65 6d 2e 6c 69 6e 6b 73 url = item.links
10c0: 28 29 5b 30 5d 2e 68 72 65 66 28 29 3b 0a 09 09 ()[0].href();...
10d0: 09 09 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 .....posts.inser
10e0: 74 28 2a 64 61 74 65 2c 20 75 72 6c 2e 74 6f 5f t(*date, url.to_
10f0: 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 string());......
1100: 09 7d 3b 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 .};......},.....
1110: 09 72 73 73 3a 3a 45 72 72 6f 72 3a 3a 45 6f 66 .rss::Error::Eof
1120: 20 3d 3e 20 28 29 2c 0a 09 09 09 09 09 5f 20 3d => (),......_ =
1130: 3e 20 62 61 69 6c 21 28 22 55 6e 73 75 70 70 6f > bail!("Unsuppo
1140: 72 74 65 64 20 6f 72 20 6d 61 6e 67 6c 65 64 20 rted or mangled
1150: 63 6f 6e 74 65 6e 74 3a 5c 6e 7b 3a 3f 7d 5c 6e content:\n{:?}\n
1160: 7b 3a 23 3f 7d 5c 6e 7b 3a 23 3f 7d 5c 6e 22 2c {:#?}\n{:#?}\n",
1170: 20 26 75 72 6c 2c 20 65 72 72 2c 20 73 74 61 74 &url, err, stat
1180: 75 73 29 0a 09 09 09 09 7d 0a 09 09 09 7d 3b 0a us).....}....};.
1190: 09 09 09 66 6f 72 20 28 64 61 74 65 2c 20 75 72 ...for (date, ur
11a0: 6c 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 65 72 l) in posts.iter
11b0: 28 29 20 7b 0a 09 09 09 09 6c 65 74 20 6d 75 74 () {.....let mut
11c0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
11d0: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
11e0: 74 0a 09 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e t.......with_con
11f0: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1200: 22 43 68 65 63 6b 20 70 6f 73 74 20 66 65 74 63 "Check post fetc
1210: 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 h conn:\n{:?}",
1220: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 &self.pool))?;..
1230: 09 09 09 6c 65 74 20 70 6f 73 74 5f 75 72 6c 3a ...let post_url:
1240: 20 43 6f 77 3c 73 74 72 3e 20 3d 20 6d 61 74 63 Cow<str> = matc
1250: 68 20 75 72 6c 5f 72 65 20 7b 0a 09 09 09 09 09 h url_re {......
1260: 53 6f 6d 65 28 72 65 66 20 78 29 20 3d 3e 20 78 Some(ref x) => x
1270: 2e 65 78 65 63 75 74 65 28 75 72 6c 29 2c 0a 09 .execute(url),..
1280: 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 75 72 6c 2e ....None => url.
1290: 69 6e 74 6f 28 29 2c 0a 09 09 09 09 7d 3b 0a 09 into(),.....};..
12a0: 09 09 09 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c ...let row = sql
12b0: 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 x::query("select
12c0: 20 65 78 69 73 74 73 28 73 65 6c 65 63 74 20 74 exists(select t
12d0: 72 75 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 rue from rsstg_p
12e0: 6f 73 74 20 77 68 65 72 65 20 75 72 6c 20 3d 20 ost where url =
12f0: 24 31 20 61 6e 64 20 73 6f 75 72 63 65 5f 69 64 $1 and source_id
1300: 20 3d 20 24 32 29 20 61 73 20 65 78 69 73 74 73 = $2) as exists
1310: 3b 22 29 0a 09 09 09 09 09 2e 62 69 6e 64 28 26 ;").......bind(&
1320: 2a 70 6f 73 74 5f 75 72 6c 29 0a 09 09 09 09 09 *post_url)......
1330: 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 09 09 09 .bind(*id)......
1340: 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 .fetch_one(&mut
1350: 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 09 conn).await.....
1360: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
1370: 7c 20 66 6f 72 6d 61 74 21 28 22 43 68 65 63 6b | format!("Check
1380: 20 70 6f 73 74 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 post:\n{:?}", &
1390: 63 6f 6e 6e 29 29 3f 3b 0a 09 09 09 09 6c 65 74 conn))?;.....let
13a0: 20 65 78 69 73 74 73 3a 20 62 6f 6f 6c 20 3d 20 exists: bool =
13b0: 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 65 78 69 row.try_get("exi
13c0: 73 74 73 22 29 3f 3b 0a 09 09 09 09 69 66 20 21 sts")?;.....if !
13d0: 20 65 78 69 73 74 73 20 7b 0a 09 09 09 09 09 69 exists {......i
13e0: 66 20 74 68 69 73 5f 66 65 74 63 68 2e 69 73 5f f this_fetch.is_
13f0: 6e 6f 6e 65 28 29 20 7c 7c 20 2a 64 61 74 65 20 none() || *date
1400: 3e 20 74 68 69 73 5f 66 65 74 63 68 2e 75 6e 77 > this_fetch.unw
1410: 72 61 70 28 29 20 7b 0a 09 09 09 09 09 09 74 68 rap() {.......th
1420: 69 73 5f 66 65 74 63 68 20 3d 20 53 6f 6d 65 28 is_fetch = Some(
1430: 2a 64 61 74 65 29 3b 0a 09 09 09 09 09 7d 3b 0a *date);......};.
1440: 09 09 09 09 09 73 65 6c 66 2e 74 67 2e 73 65 6e .....self.tg.sen
1450: 64 28 20 6d 61 74 63 68 20 69 76 5f 68 61 73 68 d( match iv_hash
1460: 20 7b 0a 09 09 09 09 09 09 09 53 6f 6d 65 28 68 {........Some(h
1470: 61 73 68 29 20 3d 3e 20 74 65 6c 65 67 72 61 6d ash) => telegram
1480: 5f 62 6f 74 3a 3a 53 65 6e 64 4d 65 73 73 61 67 _bot::SendMessag
1490: 65 3a 3a 6e 65 77 28 64 65 73 74 69 6e 61 74 69 e::new(destinati
14a0: 6f 6e 2c 20 66 6f 72 6d 61 74 21 28 22 3c 61 20 on, format!("<a
14b0: 68 72 65 66 3d 5c 22 68 74 74 70 73 3a 2f 2f 74 href=\"https://t
14c0: 2e 6d 65 2f 69 76 3f 75 72 6c 3d 7b 7d 26 72 68 .me/iv?url={}&rh
14d0: 61 73 68 3d 7b 7d 5c 22 3e 20 3c 2f 61 3e 7b 30 ash={}\"> </a>{0
14e0: 7d 22 2c 20 26 70 6f 73 74 5f 75 72 6c 2c 20 68 }", &post_url, h
14f0: 61 73 68 29 29 2c 0a 09 09 09 09 09 09 09 4e 6f ash)),........No
1500: 6e 65 20 3d 3e 20 74 65 6c 65 67 72 61 6d 5f 62 ne => telegram_b
1510: 6f 74 3a 3a 53 65 6e 64 4d 65 73 73 61 67 65 3a ot::SendMessage:
1520: 3a 6e 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e :new(destination
1530: 2c 20 66 6f 72 6d 61 74 21 28 22 7b 7d 22 2c 20 , format!("{}",
1540: 70 6f 73 74 5f 75 72 6c 29 29 2c 0a 09 09 09 09 post_url)),.....
1550: 09 09 7d 2e 70 61 72 73 65 5f 6d 6f 64 65 28 74 ..}.parse_mode(t
1560: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 elegram_bot::typ
1570: 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 es::ParseMode::H
1580: 74 6d 6c 29 29 2e 61 77 61 69 74 0a 09 09 09 09 tml)).await.....
1590: 09 09 2e 63 6f 6e 74 65 78 74 28 22 43 61 6e 27 ...context("Can'
15a0: 74 20 70 6f 73 74 20 6d 65 73 73 61 67 65 3a 22 t post message:"
15b0: 29 3f 3b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 )?;......sqlx::q
15c0: 75 65 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 uery("insert int
15d0: 6f 20 72 73 73 74 67 5f 70 6f 73 74 20 28 73 6f o rsstg_post (so
15e0: 75 72 63 65 5f 69 64 2c 20 70 6f 73 74 65 64 2c urce_id, posted,
15f0: 20 75 72 6c 29 20 76 61 6c 75 65 73 20 28 24 31 url) values ($1
1600: 2c 20 24 32 2c 20 24 33 29 3b 22 29 0a 09 09 09 , $2, $3);")....
1610: 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 09 ....bind(*id)...
1620: 09 09 09 09 2e 62 69 6e 64 28 64 61 74 65 29 0a .....bind(date).
1630: 09 09 09 09 09 09 2e 62 69 6e 64 28 26 2a 70 6f .......bind(&*po
1640: 73 74 5f 75 72 6c 29 0a 09 09 09 09 09 09 2e 65 st_url)........e
1650: 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e xecute(&mut conn
1660: 29 2e 61 77 61 69 74 0a 09 09 09 09 09 09 2e 77 ).await........w
1670: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
1680: 6f 72 6d 61 74 21 28 22 52 65 63 6f 72 64 20 70 ormat!("Record p
1690: 6f 73 74 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f ost:\n{:?}", &co
16a0: 6e 6e 29 29 3f 3b 0a 09 09 09 09 09 64 72 6f 70 nn))?;......drop
16b0: 28 63 6f 6e 6e 29 3b 0a 09 09 09 09 09 74 61 73 (conn);......tas
16c0: 6b 3a 3a 73 6c 65 65 70 28 73 74 64 3a 3a 74 69 k::sleep(std::ti
16d0: 6d 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 me::Duration::ne
16e0: 77 28 34 2c 20 30 29 29 2e 61 77 61 69 74 3b 0a w(4, 0)).await;.
16f0: 09 09 09 09 7d 3b 0a 09 09 09 09 70 6f 73 74 65 ....};.....poste
1700: 64 20 2b 3d 20 31 3b 0a 09 09 09 7d 3b 0a 09 09 d += 1;....};...
1710: 09 70 6f 73 74 73 2e 63 6c 65 61 72 28 29 3b 0a .posts.clear();.
1720: 09 09 7d 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 ..};...let mut c
1730: 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e onn = self.pool.
1740: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a acquire().await.
1750: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
1760: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 55 70 64 (|| format!("Upd
1770: 61 74 65 20 73 63 72 61 70 65 20 66 65 74 63 68 ate scrape fetch
1780: 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 conn:\n{:?}", &
1790: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 self.pool))?;...
17a0: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 sqlx::query("upd
17b0: 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 ate rsstg_source
17c0: 20 73 65 74 20 6c 61 73 74 5f 73 63 72 61 70 65 set last_scrape
17d0: 20 3d 20 6e 6f 77 28 29 20 77 68 65 72 65 20 73 = now() where s
17e0: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 29 ource_id = $1;")
17f0: 0a 09 09 09 2e 62 69 6e 64 28 2a 69 64 29 0a 09 .....bind(*id)..
1800: 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 ...execute(&mut
1810: 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e conn).await.....
1820: 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 with_context(||
1830: 66 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 format!("Update
1840: 73 63 72 61 70 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 scrape:\n{:?}",
1850: 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 4f 6b 28 66 &conn))?;...Ok(f
1860: 6f 72 6d 61 74 21 28 22 50 6f 73 74 65 64 3a 20 ormat!("Posted:
1870: 7b 7d 22 2c 20 26 70 6f 73 74 65 64 29 2e 69 6e {}", &posted).in
1880: 74 6f 28 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 to())..}...pub a
1890: 73 79 6e 63 20 66 6e 20 64 65 6c 65 74 65 3c 53 sync fn delete<S
18a0: 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f >(&self, source_
18b0: 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a id: &i32, owner:
18c0: 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f S) -> Result<Co
18d0: 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 w<'_, str>>..whe
18e0: 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re S: Into<i64>
18f0: 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 {...let owner =
1900: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 owner.into();...
1910: 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 .let mut conn =
1920: 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 self.pool.acquir
1930: 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 e().await.....wi
1940: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
1950: 72 6d 61 74 21 28 22 44 65 6c 65 74 65 20 66 65 rmat!("Delete fe
1960: 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 tch conn:\n{:?}"
1970: 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b , &self.pool))?;
1980: 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 ...match sqlx::q
1990: 75 65 72 79 28 22 64 65 6c 65 74 65 20 66 72 6f uery("delete fro
19a0: 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 m rsstg_source w
19b0: 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d here source_id =
19c0: 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 $1 and owner =
19d0: 24 32 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 $2;").....bind(s
19e0: 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 ource_id).....bi
19f0: 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 nd(owner).....ex
1a00: 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 ecute(&mut conn)
1a10: 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f .await.....with_
1a20: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
1a30: 74 21 28 22 44 65 6c 65 74 65 20 73 6f 75 72 63 t!("Delete sourc
1a40: 65 20 72 75 6c 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 e rule:\n{:?}",
1a50: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 &self.pool))?...
1a60: 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 ..rows_affected(
1a70: 29 20 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b ) {....0 => { Ok
1a80: 28 22 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 ("No data found
1a90: 66 6f 75 6e 64 2e 22 2e 69 6e 74 6f 28 29 29 20 found.".into())
1aa0: 7d 2c 0a 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 },....x => { Ok(
1ab0: 66 6f 72 6d 61 74 21 28 22 7b 7d 20 73 6f 75 72 format!("{} sour
1ac0: 63 65 73 20 72 65 6d 6f 76 65 64 2e 22 2c 20 78 ces removed.", x
1ad0: 29 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 7d ).into()) },...}
1ae0: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
1af0: 66 6e 20 63 6c 65 61 6e 3c 53 3e 28 26 73 65 6c fn clean<S>(&sel
1b00: 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 69 f, source_id: &i
1b10: 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 32, owner: S) ->
1b20: 20 52 65 73 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 Result<Cow<'_,
1b30: 73 74 72 3e 3e 0a 09 77 68 65 72 65 20 53 3a 20 str>>..where S:
1b40: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
1b50: 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 2e t owner = owner.
1b60: 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 6d into();....let m
1b70: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 ut conn = self.p
1b80: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
1b90: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
1ba0: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1bb0: 22 43 6c 65 61 6e 20 66 65 74 63 68 20 63 6f 6e "Clean fetch con
1bc0: 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 n:\n{:?}", &self
1bd0: 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 .pool))?;...matc
1be0: 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 64 h sqlx::query("d
1bf0: 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 74 67 elete from rsstg
1c00: 5f 70 6f 73 74 20 70 20 75 73 69 6e 67 20 72 73 _post p using rs
1c10: 73 74 67 5f 73 6f 75 72 63 65 20 73 20 77 68 65 stg_source s whe
1c20: 72 65 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d re p.source_id =
1c30: 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 $1 and owner =
1c40: 24 32 20 61 6e 64 20 70 2e 73 6f 75 72 63 65 5f $2 and p.source_
1c50: 69 64 20 3d 20 73 2e 73 6f 75 72 63 65 5f 69 64 id = s.source_id
1c60: 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 ;").....bind(sou
1c70: 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 rce_id).....bind
1c80: 28 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 (owner).....exec
1c90: 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 ute(&mut conn).a
1ca0: 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f wait.....with_co
1cb0: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
1cc0: 28 22 43 6c 65 61 6e 20 73 65 65 6e 20 70 6f 73 ("Clean seen pos
1cd0: 74 73 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c ts:\n{:?}", &sel
1ce0: 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f f.pool))?.....ro
1cf0: 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a ws_affected() {.
1d00: 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f ...0 => { Ok("No
1d10: 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e data found foun
1d20: 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 d.".into()) },..
1d30: 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d ..x => { Ok(form
1d40: 61 74 21 28 22 7b 7d 20 70 6f 73 74 73 20 70 75 at!("{} posts pu
1d50: 72 67 65 64 2e 22 2c 20 78 29 2e 69 6e 74 6f 28 rged.", x).into(
1d60: 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 )) },...}..}...p
1d70: 75 62 20 61 73 79 6e 63 20 66 6e 20 65 6e 61 62 ub async fn enab
1d80: 6c 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 le<S>(&self, sou
1d90: 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 rce_id: &i32, ow
1da0: 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c ner: S) -> Resul
1db0: 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 20 53 t<&str>..where S
1dc0: 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 : Into<i64> {...
1dd0: 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 let owner = owne
1de0: 72 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 r.into();....let
1df0: 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 mut conn = self
1e00: 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e .pool.acquire().
1e10: 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 await.....with_c
1e20: 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 ontext(|| format
1e30: 21 28 22 45 6e 61 62 6c 65 20 66 65 74 63 68 20 !("Enable fetch
1e40: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
1e50: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d elf.pool))?;...m
1e60: 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 atch sqlx::query
1e70: 28 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 ("update rsstg_s
1e80: 6f 75 72 63 65 20 73 65 74 20 65 6e 61 62 6c 65 ource set enable
1e90: 64 20 3d 20 74 72 75 65 20 77 68 65 72 65 20 73 d = true where s
1ea0: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e ource_id = $1 an
1eb0: 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 d owner = $2")..
1ec0: 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 ...bind(source_i
1ed0: 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 d).....bind(owne
1ee0: 72 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 r).....execute(&
1ef0: 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a mut conn).await.
1f00: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
1f10: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 45 6e 61 (|| format!("Ena
1f20: 62 6c 65 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f ble source:\n{:?
1f30: 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 }", &self.pool))
1f40: 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 ?.....rows_affec
1f50: 74 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 ted() {....1 =>
1f60: 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 65 6e 61 { Ok("Source ena
1f70: 62 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 bled.") },....0
1f80: 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 => { Ok("Source
1f90: 6e 6f 74 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a not found.") },.
1fa0: 09 09 09 5f 20 3d 3e 20 7b 20 45 72 72 28 61 6e ..._ => { Err(an
1fb0: 79 68 6f 77 21 28 22 44 61 74 61 62 61 73 65 20 yhow!("Database
1fc0: 65 72 72 6f 72 2e 22 29 29 20 7d 2c 0a 09 09 7d error.")) },...}
1fd0: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
1fe0: 66 6e 20 64 69 73 61 62 6c 65 3c 53 3e 28 26 73 fn disable<S>(&s
1ff0: 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 elf, source_id:
2000: 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 &i32, owner: S)
2010: 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a -> Result<&str>.
2020: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
2030: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
2040: 72 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 r = owner.into()
2050: 3b 0a 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e ;....let mut con
2060: 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 n = self.pool.ac
2070: 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 quire().await...
2080: 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c ..with_context(|
2090: 7c 20 66 6f 72 6d 61 74 21 28 22 44 69 73 61 62 | format!("Disab
20a0: 6c 65 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e le fetch conn:\n
20b0: 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f {:?}", &self.poo
20c0: 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 l))?;...match sq
20d0: 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 lx::query("updat
20e0: 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 e rsstg_source s
20f0: 65 74 20 65 6e 61 62 6c 65 64 20 3d 20 66 61 6c et enabled = fal
2100: 73 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f se where source_
2110: 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 id = $1 and owne
2120: 72 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e r = $2").....bin
2130: 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 d(source_id)....
2140: 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 .bind(owner)....
2150: 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f .execute(&mut co
2160: 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 nn).await.....wi
2170: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
2180: 72 6d 61 74 21 28 22 44 69 73 61 62 6c 65 20 73 rmat!("Disable s
2190: 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 ource:\n{:?}", &
21a0: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 self.pool))?....
21b0: 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 .rows_affected()
21c0: 20 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 {....1 => { Ok(
21d0: 22 53 6f 75 72 63 65 20 64 69 73 61 62 6c 65 64 "Source disabled
21e0: 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b .") },....0 => {
21f0: 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 Ok("Source not
2200: 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f found.") },...._
2210: 20 3d 3e 20 7b 20 45 72 72 28 61 6e 79 68 6f 77 => { Err(anyhow
2220: 21 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f !("Database erro
2230: 72 2e 22 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a r.")) },...}..}.
2240: 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 75 ..pub async fn u
2250: 70 64 61 74 65 3c 53 3e 28 26 73 65 6c 66 2c 20 pdate<S>(&self,
2260: 75 70 64 61 74 65 3a 20 4f 70 74 69 6f 6e 3c 69 update: Option<i
2270: 33 32 3e 2c 20 63 68 61 6e 6e 65 6c 3a 20 26 73 32>, channel: &s
2280: 74 72 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 tr, channel_id:
2290: 69 36 34 2c 20 75 72 6c 3a 20 26 73 74 72 2c 20 i64, url: &str,
22a0: 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c iv_hash: Option<
22b0: 26 73 74 72 3e 2c 20 75 72 6c 5f 72 65 3a 20 4f &str>, url_re: O
22c0: 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e ption<&str>, own
22d0: 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 er: S) -> Result
22e0: 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a <&str>..where S:
22f0: 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c Into<i64> {...l
2300: 65 74 20 6f 77 6e 65 72 20 3d 20 6f 77 6e 65 72 et owner = owner
2310: 2e 69 6e 74 6f 28 29 3b 0a 0a 09 09 6c 65 74 20 .into();....let
2320: 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e mut conn = self.
2330: 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 pool.acquire().a
2340: 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f wait.....with_co
2350: 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 ntext(|| format!
2360: 28 22 55 70 64 61 74 65 20 66 65 74 63 68 20 63 ("Update fetch c
2370: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
2380: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 0a 09 09 6d lf.pool))?;....m
2390: 61 74 63 68 20 6d 61 74 63 68 20 75 70 64 61 74 atch match updat
23a0: 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 69 64 29 e {.....Some(id)
23b0: 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a => {......sqlx:
23c0: 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 :query("update r
23d0: 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 sstg_source set
23e0: 63 68 61 6e 6e 65 6c 5f 69 64 20 3d 20 24 32 2c channel_id = $2,
23f0: 20 75 72 6c 20 3d 20 24 33 2c 20 69 76 5f 68 61 url = $3, iv_ha
2400: 73 68 20 3d 20 24 34 2c 20 6f 77 6e 65 72 20 3d sh = $4, owner =
2410: 20 24 35 2c 20 63 68 61 6e 6e 65 6c 20 3d 20 24 $5, channel = $
2420: 36 2c 20 75 72 6c 5f 72 65 20 3d 20 24 37 20 77 6, url_re = $7 w
2430: 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d here source_id =
2440: 20 24 31 22 29 2e 62 69 6e 64 28 69 64 29 0a 09 $1").bind(id)..
2450: 09 09 09 7d 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d ...},.....None =
2460: 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 > {......sqlx::q
2470: 75 65 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 uery("insert int
2480: 6f 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 28 o rsstg_source (
2490: 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c channel_id, url,
24a0: 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c iv_hash, owner,
24b0: 20 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 channel, url_re
24c0: 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 ) values ($1, $2
24d0: 2c 20 24 33 2c 20 24 34 2c 20 24 35 2c 20 24 36 , $3, $4, $5, $6
24e0: 29 22 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a )").....},....}.
24f0: 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c ....bind(channel
2500: 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 _id).....bind(ur
2510: 6c 29 0a 09 09 09 2e 62 69 6e 64 28 69 76 5f 68 l).....bind(iv_h
2520: 61 73 68 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 ash).....bind(ow
2530: 6e 65 72 29 0a 09 09 09 2e 62 69 6e 64 28 63 68 ner).....bind(ch
2540: 61 6e 6e 65 6c 29 0a 09 09 09 2e 62 69 6e 64 28 annel).....bind(
2550: 75 72 6c 5f 72 65 29 0a 09 09 09 2e 65 78 65 63 url_re).....exec
2560: 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 ute(&mut conn).a
2570: 77 61 69 74 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 wait {....Ok(_)
2580: 3d 3e 20 4f 6b 28 6d 61 74 63 68 20 75 70 64 61 => Ok(match upda
2590: 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 5f 29 te {.....Some(_)
25a0: 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 75 70 64 => "Channel upd
25b0: 61 74 65 64 2e 22 2c 0a 09 09 09 09 4e 6f 6e 65 ated.",.....None
25c0: 20 3d 3e 20 22 43 68 61 6e 6e 65 6c 20 61 64 64 => "Channel add
25d0: 65 64 2e 22 2c 0a 09 09 09 7d 29 2c 0a 09 09 09 ed.",....}),....
25e0: 45 72 72 28 73 71 6c 78 3a 3a 45 72 72 6f 72 3a Err(sqlx::Error:
25f0: 3a 44 61 74 61 62 61 73 65 28 65 72 72 29 29 20 :Database(err))
2600: 3d 3e 20 7b 0a 09 09 09 09 6d 61 74 63 68 20 65 => {.....match e
2610: 72 72 2e 64 6f 77 6e 63 61 73 74 3a 3a 3c 73 71 rr.downcast::<sq
2620: 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a 3a 50 67 lx::postgres::Pg
2630: 44 61 74 61 62 61 73 65 45 72 72 6f 72 3e 28 29 DatabaseError>()
2640: 2e 72 6f 75 74 69 6e 65 28 29 20 7b 0a 09 09 09 .routine() {....
2650: 09 09 53 6f 6d 65 28 22 5f 62 74 5f 63 68 65 63 ..Some("_bt_chec
2660: 6b 5f 75 6e 69 71 75 65 22 2c 20 29 20 3d 3e 20 k_unique", ) =>
2670: 7b 0a 09 09 09 09 09 09 4f 6b 28 22 44 75 70 6c {.......Ok("Dupl
2680: 69 63 61 74 65 20 6b 65 79 2e 22 29 0a 09 09 09 icate key.")....
2690: 09 09 7d 2c 0a 09 09 09 09 09 53 6f 6d 65 28 5f ..},......Some(_
26a0: 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 ) => {.......Ok(
26b0: 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e "Database error.
26c0: 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ")......},......
26d0: 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 09 None => {.......
26e0: 4f 6b 28 22 4e 6f 20 64 61 74 61 62 61 73 65 20 Ok("No database
26f0: 65 72 72 6f 72 20 65 78 74 72 61 63 74 65 64 2e error extracted.
2700: 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 7d ")......},.....}
2710: 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 65 72 ....},....Err(er
2720: 72 29 20 3d 3e 20 7b 0a 09 09 09 09 62 61 69 6c r) => {.....bail
2730: 21 28 22 53 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 !("Sorry, unknow
2740: 6e 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 3f 7d 5c n error:\n{:#?}\
2750: 6e 22 2c 20 65 72 72 29 3b 0a 09 09 09 7d 2c 0a n", err);....},.
2760: 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 66 ..}..}...async f
2770: 6e 20 61 75 74 6f 66 65 74 63 68 28 26 73 65 6c n autofetch(&sel
2780: 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 73 74 64 f) -> Result<std
2790: 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f 6e ::time::Duration
27a0: 3e 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 64 65 > {...let mut de
27b0: 6c 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 lay = chrono::Du
27c0: 72 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 ration::minutes(
27d0: 31 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 1);...let mut co
27e0: 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 nn = self.pool.a
27f0: 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 cquire().await..
2800: 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 ...with_context(
2810: 7c 7c 20 66 6f 72 6d 61 74 21 28 22 41 75 74 6f || format!("Auto
2820: 66 65 74 63 68 20 66 65 74 63 68 20 63 6f 6e 6e fetch fetch conn
2830: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
2840: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6c 65 74 20 6e pool))?;...let n
2850: 6f 77 20 3d 20 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 ow = chrono::Loc
2860: 61 6c 3a 3a 6e 6f 77 28 29 3b 0a 09 09 6c 65 74 al::now();...let
2870: 20 6d 75 74 20 71 75 65 75 65 20 3d 20 73 71 6c mut queue = sql
2880: 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 x::query("select
2890: 20 73 6f 75 72 63 65 5f 69 64 2c 20 6e 65 78 74 source_id, next
28a0: 5f 66 65 74 63 68 2c 20 6f 77 6e 65 72 20 66 72 _fetch, owner fr
28b0: 6f 6d 20 72 73 73 74 67 5f 6f 72 64 65 72 20 6e om rsstg_order n
28c0: 61 74 75 72 61 6c 20 6c 65 66 74 20 6a 6f 69 6e atural left join
28d0: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 rsstg_source wh
28e0: 65 72 65 20 6e 65 78 74 5f 66 65 74 63 68 20 3c ere next_fetch <
28f0: 20 6e 6f 77 28 29 20 2b 20 69 6e 74 65 72 76 61 now() + interva
2900: 6c 20 27 31 20 6d 69 6e 75 74 65 27 3b 22 29 0a l '1 minute';").
2910: 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d ....fetch_all(&m
2920: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 3f 3b ut conn).await?;
2930: 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 71 75 ...for row in qu
2940: 65 75 65 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 eue.iter() {....
2950: 6c 65 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 let source_id: i
2960: 33 32 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 32 = row.try_get
2970: 28 22 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a ("source_id")?;.
2980: 09 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 ...let owner: i6
2990: 34 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 4 = row.try_get(
29a0: 22 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 09 6c 65 "owner")?;....le
29b0: 74 20 6e 65 78 74 5f 66 65 74 63 68 3a 20 44 61 t next_fetch: Da
29c0: 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 4c teTime<chrono::L
29d0: 6f 63 61 6c 3e 20 3d 20 72 6f 77 2e 74 72 79 5f ocal> = row.try_
29e0: 67 65 74 28 22 6e 65 78 74 5f 66 65 74 63 68 22 get("next_fetch"
29f0: 29 3f 3b 0a 09 09 09 69 66 20 6e 65 78 74 5f 66 )?;....if next_f
2a00: 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a 09 09 09 etch < now {....
2a10: 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 43 6f 72 .let clone = Cor
2a20: 65 20 7b 0a 09 09 09 09 09 6f 77 6e 65 72 5f 63 e {......owner_c
2a30: 68 61 74 3a 20 74 65 6c 65 67 72 61 6d 5f 62 6f hat: telegram_bo
2a40: 74 3a 3a 55 73 65 72 49 64 3a 3a 6e 65 77 28 6f t::UserId::new(o
2a50: 77 6e 65 72 29 2c 0a 09 09 09 09 09 2e 2e 73 65 wner),........se
2a60: 6c 66 2e 63 6c 6f 6e 65 28 29 0a 09 09 09 09 7d lf.clone().....}
2a70: 3b 0a 09 09 09 09 74 61 73 6b 3a 3a 73 70 61 77 ;.....task::spaw
2a80: 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 n(async move {..
2a90: 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 ....if let Err(e
2aa0: 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 rr) = clone.chec
2ab0: 6b 28 26 73 6f 75 72 63 65 5f 69 64 2c 20 6f 77 k(&source_id, ow
2ac0: 6e 65 72 2c 20 74 72 75 65 29 2e 61 77 61 69 74 ner, true).await
2ad0: 20 7b 0a 09 09 09 09 09 09 69 66 20 6c 65 74 20 {.......if let
2ae0: 45 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 Err(err) = clone
2af0: 2e 73 65 6e 64 28 26 66 6f 72 6d 61 74 21 28 22 .send(&format!("
2b00: f0 9f 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 š {:?}", err)
2b10: 2c 20 4e 6f 6e 65 2c 20 4e 6f 6e 65 29 2e 61 77 , None, None).aw
2b20: 61 69 74 20 7b 0a 09 09 09 09 09 09 09 65 70 72 ait {........epr
2b30: 69 6e 74 6c 6e 21 28 22 43 68 65 63 6b 20 65 72 intln!("Check er
2b40: 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a ror: {}", err);.
2b50: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 7d 3b ......};......};
2b60: 0a 09 09 09 09 7d 29 3b 0a 09 09 09 7d 20 65 6c .....});....} el
2b70: 73 65 20 69 66 20 6e 65 78 74 5f 66 65 74 63 68 se if next_fetch
2b80: 20 2d 20 6e 6f 77 20 3c 20 64 65 6c 61 79 20 7b - now < delay {
2b90: 0a 09 09 09 09 64 65 6c 61 79 20 3d 20 6e 65 78 .....delay = nex
2ba0: 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 3b 0a 09 t_fetch - now;..
2bb0: 09 09 7d 0a 09 09 7d 3b 0a 09 09 71 75 65 75 65 ..}...};...queue
2bc0: 2e 63 6c 65 61 72 28 29 3b 0a 09 09 4f 6b 28 64 .clear();...Ok(d
2bd0: 65 6c 61 79 2e 74 6f 5f 73 74 64 28 29 3f 29 0a elay.to_std()?).
2be0: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
2bf0: 6e 20 6c 69 73 74 3c 53 3e 28 26 73 65 6c 66 2c n list<S>(&self,
2c00: 20 6f 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 owner: S) -> Re
2c10: 73 75 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 sult<String>..wh
2c20: 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e ere S: Into<i64>
2c30: 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d {...let owner =
2c40: 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 0a owner.into();..
2c50: 09 09 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 3a ..let mut reply:
2c60: 20 56 65 63 3c 43 6f 77 3c 73 74 72 3e 3e 20 3d Vec<Cow<str>> =
2c70: 20 76 65 63 21 5b 5d 3b 0a 09 09 6c 65 74 20 6d vec![];...let m
2c80: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 ut conn = self.p
2c90: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
2ca0: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
2cb0: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
2cc0: 22 4c 69 73 74 20 66 65 74 63 68 20 63 6f 6e 6e "List fetch conn
2cd0: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
2ce0: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 72 65 70 6c 79 pool))?;...reply
2cf0: 2e 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a .push("Channels:
2d00: 22 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 6c 65 74 ".into());...let
2d10: 20 72 6f 77 73 20 3d 20 73 71 6c 78 3a 3a 71 75 rows = sqlx::qu
2d20: 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 ery("select sour
2d30: 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 ce_id, channel,
2d40: 65 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 76 enabled, url, iv
2d50: 5f 68 61 73 68 2c 20 75 72 6c 5f 72 65 20 66 72 _hash, url_re fr
2d60: 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 om rsstg_source
2d70: 77 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 31 where owner = $1
2d80: 20 6f 72 64 65 72 20 62 79 20 73 6f 75 72 63 65 order by source
2d90: 5f 69 64 22 29 0a 09 09 09 2e 62 69 6e 64 28 6f _id").....bind(o
2da0: 77 6e 65 72 29 0a 09 09 09 2e 66 65 74 63 68 5f wner).....fetch_
2db0: 61 6c 6c 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 all(&mut conn).a
2dc0: 77 61 69 74 3f 3b 0a 09 09 66 6f 72 20 72 6f 77 wait?;...for row
2dd0: 20 69 6e 20 72 6f 77 73 2e 69 74 65 72 28 29 20 in rows.iter()
2de0: 7b 0a 09 09 09 6c 65 74 20 73 6f 75 72 63 65 5f {....let source_
2df0: 69 64 3a 20 69 33 32 20 3d 20 72 6f 77 2e 74 72 id: i32 = row.tr
2e00: 79 5f 67 65 74 28 22 73 6f 75 72 63 65 5f 69 64 y_get("source_id
2e10: 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 73 65 72 ")?;....let user
2e20: 6e 61 6d 65 3a 20 26 73 74 72 20 3d 20 72 6f 77 name: &str = row
2e30: 2e 74 72 79 5f 67 65 74 28 22 63 68 61 6e 6e 65 .try_get("channe
2e40: 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 65 6e 61 l")?;....let ena
2e50: 62 6c 65 64 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 bled: bool = row
2e60: 2e 74 72 79 5f 67 65 74 28 22 65 6e 61 62 6c 65 .try_get("enable
2e70: 64 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c d")?;....let url
2e80: 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 : &str = row.try
2e90: 5f 67 65 74 28 22 75 72 6c 22 29 3f 3b 0a 09 09 _get("url")?;...
2ea0: 09 6c 65 74 20 69 76 5f 68 61 73 68 3a 20 4f 70 .let iv_hash: Op
2eb0: 74 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 72 6f 77 tion<&str> = row
2ec0: 2e 74 72 79 5f 67 65 74 28 22 69 76 5f 68 61 73 .try_get("iv_has
2ed0: 68 22 29 3f 3b 0a 09 09 09 6c 65 74 20 75 72 6c h")?;....let url
2ee0: 5f 72 65 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 _re: Option<&str
2ef0: 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 > = row.try_get(
2f00: 22 75 72 6c 5f 72 65 22 29 3f 3b 0a 09 09 09 72 "url_re")?;....r
2f10: 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 74 eply.push(format
2f20: 21 28 22 5c 6e 5c 5c 23 ef b8 8f e2 83 a3 20 7b !("\n\\#ļøā£ {
2f30: 7d 20 5c 5c 2a ef b8 8f e2 83 a3 20 60 7b 7d 60 } \\*ļøā£ `{}`
2f40: 20 7b 7d 5c 6e f0 9f 94 97 20 60 7b 7d 60 22 2c {}\nš `{}`",
2f50: 20 73 6f 75 72 63 65 5f 69 64 2c 20 75 73 65 72 source_id, user
2f60: 6e 61 6d 65 2c 20 20 0a 09 09 09 09 6d 61 74 63 name, .....matc
2f70: 68 20 65 6e 61 62 6c 65 64 20 7b 0a 09 09 09 09 h enabled {.....
2f80: 09 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 20 .true => "š
2f90: 65 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 66 enabled",......f
2fa0: 61 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 73 alse => "ā dis
2fb0: 61 62 6c 65 64 22 2c 0a 09 09 09 09 7d 2c 20 75 abled",.....}, u
2fc0: 72 6c 29 2e 69 6e 74 6f 28 29 29 3b 0a 09 09 09 rl).into());....
2fd0: 69 66 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 68 if let Some(hash
2fe0: 29 20 3d 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 ) = iv_hash {...
2ff0: 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 ..reply.push(for
3000: 6d 61 74 21 28 22 49 56 3a 20 60 7b 7d 60 22 2c mat!("IV: `{}`",
3010: 20 68 61 73 68 29 2e 69 6e 74 6f 28 29 29 3b 0a hash).into());.
3020: 09 09 09 7d 0a 09 09 09 69 66 20 6c 65 74 20 53 ...}....if let S
3030: 6f 6d 65 28 72 65 29 20 3d 20 75 72 6c 5f 72 65 ome(re) = url_re
3040: 20 7b 0a 09 09 09 09 72 65 70 6c 79 2e 70 75 73 {.....reply.pus
3050: 68 28 66 6f 72 6d 61 74 21 28 22 52 45 3a 20 60 h(format!("RE: `
3060: 7b 7d 60 22 2c 20 72 65 29 2e 69 6e 74 6f 28 29 {}`", re).into()
3070: 29 3b 0a 09 09 09 7d 0a 09 09 7d 3b 0a 09 09 4f );....}...};...O
3080: 6b 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e k(reply.join("\n
3090: 22 29 29 0a 09 7d 0a 7d 0a "))..}.}.