0000: 75 73 65 20 73 74 64 3a 3a 63 6f 6c 6c 65 63 74 use std::collect
0010: 69 6f 6e 73 3a 3a 42 54 72 65 65 4d 61 70 3b 0a ions::BTreeMap;.
0020: 0a 75 73 65 20 63 6f 6e 66 69 67 3b 0a 0a 75 73 .use config;..us
0030: 65 20 74 6f 6b 69 6f 3b 0a 0a 75 73 65 20 72 73 e tokio;..use rs
0040: 73 3b 0a 0a 75 73 65 20 63 68 72 6f 6e 6f 3a 3a s;..use chrono::
0050: 44 61 74 65 54 69 6d 65 3b 0a 0a 75 73 65 20 72 DateTime;..use r
0060: 65 67 65 78 3a 3a 52 65 67 65 78 3b 0a 0a 75 73 egex::Regex;..us
0070: 65 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a e telegram_bot::
0080: 2a 3b 0a 75 73 65 20 74 6f 6b 69 6f 3a 3a 73 74 *;.use tokio::st
0090: 72 65 61 6d 3a 3a 53 74 72 65 61 6d 45 78 74 3b ream::StreamExt;
00a0: 0a 0a 75 73 65 20 73 71 6c 78 3a 3a 70 6f 73 74 ..use sqlx::post
00b0: 67 72 65 73 3a 3a 50 67 50 6f 6f 6c 4f 70 74 69 gres::PgPoolOpti
00c0: 6f 6e 73 3b 0a 75 73 65 20 73 71 6c 78 3a 3a 52 ons;.use sqlx::R
00d0: 6f 77 3b 0a 75 73 65 20 73 71 6c 78 3a 3a 44 6f ow;.use sqlx::Do
00e0: 6e 65 3b 20 2f 2f 20 2e 72 6f 77 73 5f 61 66 66 ne; // .rows_aff
00f0: 65 63 74 65 64 28 29 0a 0a 23 5b 6d 61 63 72 6f ected()..#[macro
0100: 5f 75 73 65 5d 0a 65 78 74 65 72 6e 20 63 72 61 _use].extern cra
0110: 74 65 20 6c 61 7a 79 5f 73 74 61 74 69 63 3b 0a te lazy_static;.
0120: 0a 75 73 65 20 61 6e 79 68 6f 77 3a 3a 7b 61 6e .use anyhow::{an
0130: 79 68 6f 77 2c 20 62 61 69 6c 2c 20 43 6f 6e 74 yhow, bail, Cont
0140: 65 78 74 2c 20 52 65 73 75 6c 74 7d 3b 0a 0a 23 ext, Result};..#
0150: 5b 64 65 72 69 76 65 28 43 6c 6f 6e 65 29 5d 0a [derive(Clone)].
0160: 73 74 72 75 63 74 20 43 6f 72 65 20 7b 0a 09 6f struct Core {..o
0170: 77 6e 65 72 3a 20 69 36 34 2c 0a 09 61 70 69 5f wner: i64,..api_
0180: 6b 65 79 3a 20 53 74 72 69 6e 67 2c 0a 09 6f 77 key: String,..ow
0190: 6e 65 72 5f 63 68 61 74 3a 20 55 73 65 72 49 64 ner_chat: UserId
01a0: 2c 0a 09 74 67 3a 20 74 65 6c 65 67 72 61 6d 5f ,..tg: telegram_
01b0: 62 6f 74 3a 3a 41 70 69 2c 0a 09 6d 79 3a 20 55 bot::Api,..my: U
01c0: 73 65 72 2c 0a 09 70 6f 6f 6c 3a 20 73 71 6c 78 ser,..pool: sqlx
01d0: 3a 3a 50 6f 6f 6c 3c 73 71 6c 78 3a 3a 50 6f 73 ::Pool<sqlx::Pos
01e0: 74 67 72 65 73 3e 2c 0a 7d 0a 0a 69 6d 70 6c 20 tgres>,.}..impl
01f0: 43 6f 72 65 20 7b 0a 09 61 73 79 6e 63 20 66 6e Core {..async fn
0200: 20 6e 65 77 28 73 65 74 74 69 6e 67 73 3a 20 63 new(settings: c
0210: 6f 6e 66 69 67 3a 3a 43 6f 6e 66 69 67 29 20 2d onfig::Config) -
0220: 3e 20 52 65 73 75 6c 74 3c 43 6f 72 65 3e 20 7b > Result<Core> {
0230: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 73 ...let owner = s
0240: 65 74 74 69 6e 67 73 2e 67 65 74 5f 69 6e 74 28 ettings.get_int(
0250: 22 6f 77 6e 65 72 22 29 3f 3b 0a 09 09 6c 65 74 "owner")?;...let
0260: 20 61 70 69 5f 6b 65 79 20 3d 20 73 65 74 74 69 api_key = setti
0270: 6e 67 73 2e 67 65 74 5f 73 74 72 28 22 61 70 69 ngs.get_str("api
0280: 5f 6b 65 79 22 29 3f 3b 0a 09 09 6c 65 74 20 74 _key")?;...let t
0290: 67 20 3d 20 41 70 69 3a 3a 6e 65 77 28 26 61 70 g = Api::new(&ap
02a0: 69 5f 6b 65 79 29 3b 0a 09 09 6c 65 74 20 63 6f i_key);...let co
02b0: 72 65 20 3d 20 43 6f 72 65 20 7b 0a 09 09 09 6f re = Core {....o
02c0: 77 6e 65 72 3a 20 6f 77 6e 65 72 2c 0a 09 09 09 wner: owner,....
02d0: 61 70 69 5f 6b 65 79 3a 20 61 70 69 5f 6b 65 79 api_key: api_key
02e0: 2e 63 6c 6f 6e 65 28 29 2c 0a 09 09 09 6d 79 3a .clone(),....my:
02f0: 20 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 72 61 tg.send(telegra
0300: 6d 5f 62 6f 74 3a 3a 47 65 74 4d 65 29 2e 61 77 m_bot::GetMe).aw
0310: 61 69 74 3f 2c 0a 09 09 09 74 67 3a 20 74 67 2c ait?,....tg: tg,
0320: 0a 09 09 09 6f 77 6e 65 72 5f 63 68 61 74 3a 20 ....owner_chat:
0330: 55 73 65 72 49 64 3a 3a 6e 65 77 28 6f 77 6e 65 UserId::new(owne
0340: 72 29 2c 0a 09 09 09 70 6f 6f 6c 3a 20 50 67 50 r),....pool: PgP
0350: 6f 6f 6c 4f 70 74 69 6f 6e 73 3a 3a 6e 65 77 28 oolOptions::new(
0360: 29 0a 09 09 09 09 2e 6d 61 78 5f 63 6f 6e 6e 65 )......max_conne
0370: 63 74 69 6f 6e 73 28 35 29 0a 09 09 09 09 2e 63 ctions(5)......c
0380: 6f 6e 6e 65 63 74 5f 74 69 6d 65 6f 75 74 28 73 onnect_timeout(s
0390: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
03a0: 6f 6e 3a 3a 6e 65 77 28 33 30 30 2c 20 30 29 29 on::new(300, 0))
03b0: 0a 09 09 09 09 2e 69 64 6c 65 5f 74 69 6d 65 6f ......idle_timeo
03c0: 75 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 ut(std::time::Du
03d0: 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 36 30 2c 20 ration::new(60,
03e0: 30 29 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 74 0))......connect
03f0: 5f 6c 61 7a 79 28 26 73 65 74 74 69 6e 67 73 2e _lazy(&settings.
0400: 67 65 74 5f 73 74 72 28 22 70 67 22 29 3f 29 3f get_str("pg")?)?
0410: 2c 0a 09 09 7d 3b 0a 09 09 6c 65 74 20 63 6c 6f ,...};...let clo
0420: 6e 65 20 3d 20 63 6f 72 65 2e 63 6c 6f 6e 65 28 ne = core.clone(
0430: 29 3b 0a 09 09 74 6f 6b 69 6f 3a 3a 73 70 61 77 );...tokio::spaw
0440: 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b 0a 09 n(async move {..
0450: 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 ..if let Err(err
0460: 29 20 3d 20 26 63 6c 6f 6e 65 2e 61 75 74 6f 66 ) = &clone.autof
0470: 65 74 63 68 28 29 2e 61 77 61 69 74 20 7b 0a 09 etch().await {..
0480: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
0490: 72 29 20 3d 20 63 6c 6f 6e 65 2e 64 65 62 75 67 r) = clone.debug
04a0: 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 (&format!("š
04b0: 7b 3a 3f 7d 22 2c 20 65 72 72 29 29 20 7b 0a 09 {:?}", err)) {..
04c0: 09 09 09 09 65 70 72 69 6e 74 6c 6e 21 28 22 41 ....eprintln!("A
04d0: 75 74 6f 66 65 74 63 68 20 65 72 72 6f 72 3a 20 utofetch error:
04e0: 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 09 7d {}", err);.....}
04f0: 3b 0a 09 09 09 7d 0a 09 09 7d 29 3b 0a 09 09 4f ;....}...});...O
0500: 6b 28 63 6f 72 65 29 0a 09 7d 0a 0a 09 66 6e 20 k(core)..}...fn
0510: 73 74 72 65 61 6d 28 26 73 65 6c 66 29 20 2d 3e stream(&self) ->
0520: 20 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 telegram_bot::U
0530: 70 64 61 74 65 73 53 74 72 65 61 6d 20 7b 0a 09 pdatesStream {..
0540: 09 73 65 6c 66 2e 74 67 2e 73 74 72 65 61 6d 28 .self.tg.stream(
0550: 29 0a 09 7d 0a 0a 09 66 6e 20 64 65 62 75 67 28 )..}...fn debug(
0560: 26 73 65 6c 66 2c 20 6d 73 67 3a 20 26 73 74 72 &self, msg: &str
0570: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 20 ) -> Result<()>
0580: 7b 0a 09 09 73 65 6c 66 2e 74 67 2e 73 70 61 77 {...self.tg.spaw
0590: 6e 28 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e n(SendMessage::n
05a0: 65 77 28 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 ew(self.owner_ch
05b0: 61 74 2c 20 6d 73 67 29 29 3b 0a 09 09 4f 6b 28 at, msg));...Ok(
05c0: 28 29 29 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 66 ())..}...async f
05d0: 6e 20 63 68 65 63 6b 3c 53 3e 28 26 73 65 6c 66 n check<S>(&self
05e0: 2c 20 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 , id: &i32, owne
05f0: 72 3a 20 53 2c 20 72 65 61 6c 3a 20 62 6f 6f 6c r: S, real: bool
0600: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 0a ) -> Result<()>.
0610: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 69 .where S: Into<i
0620: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e 65 64> {...let owne
0630: 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e 69 r: i64 = owner.i
0640: 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 74 nto();...let mut
0650: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
0660: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
0670: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
0680: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 51 xt(|| format!("Q
0690: 75 65 72 79 20 71 75 65 75 65 20 66 65 74 63 68 uery queue fetch
06a0: 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 conn:\n{:?}", &
06b0: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 self.pool))?;...
06c0: 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c 78 3a 3a let row = sqlx::
06d0: 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f query("select so
06e0: 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c urce_id, channel
06f0: 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 _id, url, iv_has
0700: 68 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 h, owner from rs
0710: 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 stg_source where
0720: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 source_id = $1
0730: 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 and owner = $2")
0740: 0a 09 09 09 2e 62 69 6e 64 28 69 64 29 0a 09 09 .....bind(id)...
0750: 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 ..bind(owner)...
0760: 09 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 ..fetch_one(&mut
0770: 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 conn).await....
0780: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
0790: 20 66 6f 72 6d 61 74 21 28 22 51 75 65 72 79 20 format!("Query
07a0: 73 6f 75 72 63 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 source:\n{:?}",
07b0: 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 &self.pool))?;..
07c0: 09 64 72 6f 70 28 63 6f 6e 6e 29 3b 0a 09 09 6c .drop(conn);...l
07d0: 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 et channel_id: i
07e0: 36 34 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 64 = row.try_get
07f0: 28 22 63 68 61 6e 6e 65 6c 5f 69 64 22 29 3f 3b ("channel_id")?;
0800: 0a 09 09 6c 65 74 20 64 65 73 74 69 6e 61 74 69 ...let destinati
0810: 6f 6e 20 3d 20 6d 61 74 63 68 20 72 65 61 6c 20 on = match real
0820: 7b 0a 09 09 09 74 72 75 65 20 3d 3e 20 55 73 65 {....true => Use
0830: 72 49 64 3a 3a 6e 65 77 28 63 68 61 6e 6e 65 6c rId::new(channel
0840: 5f 69 64 29 2c 0a 09 09 09 66 61 6c 73 65 20 3d _id),....false =
0850: 3e 20 55 73 65 72 49 64 3a 3a 6e 65 77 28 72 6f > UserId::new(ro
0860: 77 2e 74 72 79 5f 67 65 74 28 22 6f 77 6e 65 72 w.try_get("owner
0870: 22 29 3f 29 2c 0a 09 09 7d 3b 0a 09 09 6c 65 74 ")?),...};...let
0880: 20 75 72 6c 3a 20 26 73 74 72 20 3d 20 72 6f 77 url: &str = row
0890: 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 22 29 3f .try_get("url")?
08a0: 3b 0a 09 09 6c 65 74 20 6d 75 74 20 74 68 69 73 ;...let mut this
08b0: 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e 3c 44 _fetch: Option<D
08c0: 61 74 65 54 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a ateTime<chrono::
08d0: 46 69 78 65 64 4f 66 66 73 65 74 3e 3e 20 3d 20 FixedOffset>> =
08e0: 4e 6f 6e 65 3b 0a 09 09 6c 65 74 20 69 76 5f 68 None;...let iv_h
08f0: 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 ash: Option<&str
0900: 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 > = row.try_get(
0910: 22 69 76 5f 68 61 73 68 22 29 3f 3b 0a 09 09 6c "iv_hash")?;...l
0920: 65 74 20 6d 75 74 20 70 6f 73 74 73 3a 20 42 54 et mut posts: BT
0930: 72 65 65 4d 61 70 3c 44 61 74 65 54 69 6d 65 3c reeMap<DateTime<
0940: 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 66 chrono::FixedOff
0950: 73 65 74 3e 2c 20 53 74 72 69 6e 67 3e 20 3d 20 set>, String> =
0960: 42 54 72 65 65 4d 61 70 3a 3a 6e 65 77 28 29 3b BTreeMap::new();
0970: 0a 09 09 6c 65 74 20 66 65 65 64 20 3d 20 72 73 ...let feed = rs
0980: 73 3a 3a 43 68 61 6e 6e 65 6c 3a 3a 66 72 6f 6d s::Channel::from
0990: 5f 75 72 6c 28 75 72 6c 29 0a 09 09 09 2e 77 69 _url(url).....wi
09a0: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
09b0: 72 6d 61 74 21 28 22 50 72 6f 62 6c 65 6d 20 6f rmat!("Problem o
09c0: 70 65 6e 69 6e 67 20 66 65 65 64 20 75 72 6c 3a pening feed url:
09d0: 5c 6e 7b 7d 22 2c 20 26 75 72 6c 29 29 3f 3b 0a \n{}", &url))?;.
09e0: 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 65 ..for item in fe
09f0: 65 64 2e 69 74 65 6d 73 28 29 20 7b 0a 09 09 09 ed.items() {....
0a00: 6c 65 74 20 64 61 74 65 20 3d 20 6d 61 74 63 68 let date = match
0a10: 20 69 74 65 6d 2e 70 75 62 5f 64 61 74 65 28 29 item.pub_date()
0a20: 20 7b 0a 09 09 09 09 53 6f 6d 65 28 66 65 65 64 {.....Some(feed
0a30: 5f 64 61 74 65 29 20 3d 3e 20 44 61 74 65 54 69 _date) => DateTi
0a40: 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f 72 me::parse_from_r
0a50: 66 63 32 38 32 32 28 66 65 65 64 5f 64 61 74 65 fc2822(feed_date
0a60: 29 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 44 ),.....None => D
0a70: 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 65 5f 66 ateTime::parse_f
0a80: 72 6f 6d 5f 72 66 63 33 33 33 39 28 26 69 74 65 rom_rfc3339(&ite
0a90: 6d 2e 64 75 62 6c 69 6e 5f 63 6f 72 65 5f 65 78 m.dublin_core_ex
0aa0: 74 28 29 2e 75 6e 77 72 61 70 28 29 2e 64 61 74 t().unwrap().dat
0ab0: 65 73 28 29 5b 30 5d 29 2c 0a 09 09 09 7d 3f 3b es()[0]),....}?;
0ac0: 0a 09 09 09 6c 65 74 20 75 72 6c 20 3d 20 69 74 ....let url = it
0ad0: 65 6d 2e 6c 69 6e 6b 28 29 2e 75 6e 77 72 61 70 em.link().unwrap
0ae0: 28 29 2e 74 6f 5f 73 74 72 69 6e 67 28 29 3b 0a ().to_string();.
0af0: 09 09 09 70 6f 73 74 73 2e 69 6e 73 65 72 74 28 ...posts.insert(
0b00: 64 61 74 65 2e 63 6c 6f 6e 65 28 29 2c 20 75 72 date.clone(), ur
0b10: 6c 2e 63 6c 6f 6e 65 28 29 29 3b 0a 09 09 7d 3b l.clone());...};
0b20: 0a 09 09 66 6f 72 20 28 64 61 74 65 2c 20 75 72 ...for (date, ur
0b30: 6c 29 20 69 6e 20 70 6f 73 74 73 2e 69 74 65 72 l) in posts.iter
0b40: 28 29 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 () {....let mut
0b50: 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c conn = self.pool
0b60: 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 .acquire().await
0b70: 0a 09 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 ......with_conte
0b80: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 xt(|| format!("C
0b90: 68 65 63 6b 20 70 6f 73 74 20 66 65 74 63 68 20 heck post fetch
0ba0: 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 conn:\n{:?}", &s
0bb0: 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 elf.pool))?;....
0bc0: 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c 78 3a 3a let row = sqlx::
0bd0: 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 65 78 query("select ex
0be0: 69 73 74 73 28 73 65 6c 65 63 74 20 74 72 75 65 ists(select true
0bf0: 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 from rsstg_post
0c00: 20 77 68 65 72 65 20 75 72 6c 20 3d 20 24 31 20 where url = $1
0c10: 61 6e 64 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 and source_id =
0c20: 24 32 29 20 61 73 20 65 78 69 73 74 73 3b 22 29 $2) as exists;")
0c30: 0a 09 09 09 09 2e 62 69 6e 64 28 26 75 72 6c 29 ......bind(&url)
0c40: 0a 09 09 09 09 2e 62 69 6e 64 28 69 64 29 0a 09 ......bind(id)..
0c50: 09 09 09 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d ....fetch_one(&m
0c60: 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a 09 ut conn).await..
0c70: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
0c80: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 43 68 65 (|| format!("Che
0c90: 63 6b 20 70 6f 73 74 3a 5c 6e 7b 3a 3f 7d 22 2c ck post:\n{:?}",
0ca0: 20 26 63 6f 6e 6e 29 29 3f 3b 0a 09 09 09 6c 65 &conn))?;....le
0cb0: 74 20 65 78 69 73 74 73 3a 20 62 6f 6f 6c 20 3d t exists: bool =
0cc0: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 65 78 row.try_get("ex
0cd0: 69 73 74 73 22 29 3f 3b 0a 09 09 09 69 66 20 21 ists")?;....if !
0ce0: 20 65 78 69 73 74 73 20 7b 0a 09 09 09 09 69 66 exists {.....if
0cf0: 20 74 68 69 73 5f 66 65 74 63 68 20 3d 3d 20 4e this_fetch == N
0d00: 6f 6e 65 20 7c 7c 20 2a 64 61 74 65 20 3e 20 74 one || *date > t
0d10: 68 69 73 5f 66 65 74 63 68 2e 75 6e 77 72 61 70 his_fetch.unwrap
0d20: 28 29 20 7b 0a 09 09 09 09 09 74 68 69 73 5f 66 () {......this_f
0d30: 65 74 63 68 20 3d 20 53 6f 6d 65 28 2a 64 61 74 etch = Some(*dat
0d40: 65 29 3b 0a 09 09 09 09 7d 3b 0a 09 09 09 09 73 e);.....};.....s
0d50: 65 6c 66 2e 74 67 2e 73 65 6e 64 28 20 6d 61 74 elf.tg.send( mat
0d60: 63 68 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 ch iv_hash {....
0d70: 09 09 09 53 6f 6d 65 28 78 29 20 3d 3e 20 53 65 ...Some(x) => Se
0d80: 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 28 64 ndMessage::new(d
0d90: 65 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f 72 6d estination, form
0da0: 61 74 21 28 22 3c 61 20 68 72 65 66 3d 5c 22 68 at!("<a href=\"h
0db0: 74 74 70 73 3a 2f 2f 74 2e 6d 65 2f 69 76 3f 75 ttps://t.me/iv?u
0dc0: 72 6c 3d 7b 7d 26 72 68 61 73 68 3d 7b 7d 5c 22 rl={}&rhash={}\"
0dd0: 3e 20 3c 2f 61 3e 7b 30 7d 22 2c 20 75 72 6c 2c > </a>{0}", url,
0de0: 20 78 29 29 2c 0a 09 09 09 09 09 09 4e 6f 6e 65 x)),.......None
0df0: 20 3d 3e 20 53 65 6e 64 4d 65 73 73 61 67 65 3a => SendMessage:
0e00: 3a 6e 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e :new(destination
0e10: 2c 20 66 6f 72 6d 61 74 21 28 22 7b 7d 22 2c 20 , format!("{}",
0e20: 75 72 6c 29 29 2c 0a 09 09 09 09 09 7d 2e 70 61 url)),......}.pa
0e30: 72 73 65 5f 6d 6f 64 65 28 74 79 70 65 73 3a 3a rse_mode(types::
0e40: 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 6d 6c 29 ParseMode::Html)
0e50: 29 2e 61 77 61 69 74 0a 09 09 09 09 09 2e 63 6f ).await.......co
0e60: 6e 74 65 78 74 28 22 43 61 6e 27 74 20 70 6f 73 ntext("Can't pos
0e70: 74 20 6d 65 73 73 61 67 65 3a 22 29 3f 3b 0a 09 t message:")?;..
0e80: 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ...sqlx::query("
0e90: 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 insert into rsst
0ea0: 67 5f 70 6f 73 74 20 28 73 6f 75 72 63 65 5f 69 g_post (source_i
0eb0: 64 2c 20 70 6f 73 74 65 64 2c 20 75 72 6c 29 20 d, posted, url)
0ec0: 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 2c 20 values ($1, $2,
0ed0: 24 33 29 3b 22 29 0a 09 09 09 09 09 2e 62 69 6e $3);").......bin
0ee0: 64 28 69 64 29 0a 09 09 09 09 09 2e 62 69 6e 64 d(id).......bind
0ef0: 28 64 61 74 65 29 0a 09 09 09 09 09 2e 62 69 6e (date).......bin
0f00: 64 28 75 72 6c 29 0a 09 09 09 09 09 2e 65 78 65 d(url).......exe
0f10: 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e cute(&mut conn).
0f20: 61 77 61 69 74 0a 09 09 09 09 09 2e 77 69 74 68 await.......with
0f30: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
0f40: 61 74 21 28 22 52 65 63 6f 72 64 20 70 6f 73 74 at!("Record post
0f50: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e 6e 29 :\n{:?}", &conn)
0f60: 29 3f 3b 0a 09 09 09 09 64 72 6f 70 28 63 6f 6e )?;.....drop(con
0f70: 6e 29 3b 0a 09 09 09 09 74 6f 6b 69 6f 3a 3a 74 n);.....tokio::t
0f80: 69 6d 65 3a 3a 64 65 6c 61 79 5f 66 6f 72 28 73 ime::delay_for(s
0f90: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
0fa0: 6f 6e 3a 3a 6e 65 77 28 34 2c 20 30 29 29 2e 61 on::new(4, 0)).a
0fb0: 77 61 69 74 3b 0a 09 09 09 7d 3b 0a 09 09 7d 3b wait;....};...};
0fc0: 0a 09 09 70 6f 73 74 73 2e 63 6c 65 61 72 28 29 ...posts.clear()
0fd0: 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ;...let mut conn
0fe0: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
0ff0: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
1000: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
1010: 20 66 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 format!("Update
1020: 20 73 63 72 61 70 65 20 66 65 74 63 68 20 63 6f scrape fetch co
1030: 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c nn:\n{:?}", &sel
1040: 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 73 71 6c f.pool))?;...sql
1050: 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 x::query("update
1060: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 rsstg_source se
1070: 74 20 6c 61 73 74 5f 73 63 72 61 70 65 20 3d 20 t last_scrape =
1080: 6e 6f 77 28 29 20 77 68 65 72 65 20 73 6f 75 72 now() where sour
1090: 63 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a 09 09 ce_id = $1;")...
10a0: 09 2e 62 69 6e 64 28 69 64 29 0a 09 09 09 2e 65 ..bind(id).....e
10b0: 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e xecute(&mut conn
10c0: 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 ).await.....with
10d0: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
10e0: 61 74 21 28 22 55 70 64 61 74 65 20 73 63 72 61 at!("Update scra
10f0: 70 65 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 63 6f 6e pe:\n{:?}", &con
1100: 6e 29 29 3f 3b 0a 09 09 4f 6b 28 28 29 29 0a 09 n))?;...Ok(())..
1110: 7d 0a 0a 09 61 73 79 6e 63 20 66 6e 20 64 65 6c }...async fn del
1120: 65 74 65 3c 53 3e 28 26 73 65 6c 66 2c 20 73 6f ete<S>(&self, so
1130: 75 72 63 65 5f 69 64 3a 20 26 69 33 32 2c 20 6f urce_id: &i32, o
1140: 77 6e 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 wner: S) -> Resu
1150: 6c 74 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 72 lt<String>..wher
1160: 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e S: Into<i64> {
1170: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 ...let owner: i6
1180: 34 20 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 4 = owner.into()
1190: 3b 0a 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e ;...let mut conn
11a0: 20 3d 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 = self.pool.acq
11b0: 75 69 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 uire().await....
11c0: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
11d0: 20 66 6f 72 6d 61 74 21 28 22 44 65 6c 65 74 65 format!("Delete
11e0: 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a fetch conn:\n{:
11f0: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
1200: 29 3f 3b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 )?;...match sqlx
1210: 3a 3a 71 75 65 72 79 28 22 64 65 6c 65 74 65 20 ::query("delete
1220: 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 from rsstg_sourc
1230: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
1240: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
1250: 20 3d 20 24 32 3b 22 29 0a 09 09 09 2e 62 69 6e = $2;").....bin
1260: 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 d(source_id)....
1270: 2e 62 69 6e 64 28 6f 77 6e 65 72 29 0a 09 09 09 .bind(owner)....
1280: 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 63 6f .execute(&mut co
1290: 6e 6e 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 nn).await.....wi
12a0: 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f th_context(|| fo
12b0: 72 6d 61 74 21 28 22 44 65 6c 65 74 65 20 73 6f rmat!("Delete so
12c0: 75 72 63 65 20 72 75 6c 65 3a 5c 6e 7b 3a 3f 7d urce rule:\n{:?}
12d0: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
12e0: 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 63 74 .....rows_affect
12f0: 65 64 28 29 20 7b 0a 09 09 09 30 20 3d 3e 20 7b ed() {....0 => {
1300: 20 4f 6b 28 22 4e 6f 20 64 61 74 61 20 66 6f 75 Ok("No data fou
1310: 6e 64 20 66 6f 75 6e 64 5c 5c 2e 22 2e 74 6f 5f nd found\\.".to_
1320: 73 74 72 69 6e 67 28 29 29 20 7d 2c 0a 09 09 09 string()) },....
1330: 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 x => { Ok(format
1340: 21 28 22 7b 7d 20 73 6f 75 72 63 65 73 20 72 65 !("{} sources re
1350: 6d 6f 76 65 64 5c 5c 2e 22 2c 20 78 29 29 20 7d moved\\.", x)) }
1360: 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e 63 ,...}..}...async
1370: 20 66 6e 20 63 6c 65 61 6e 3c 53 3e 28 26 73 65 fn clean<S>(&se
1380: 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 26 lf, source_id: &
1390: 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 29 20 2d i32, owner: S) -
13a0: 3e 20 52 65 73 75 6c 74 3c 53 74 72 69 6e 67 3e > Result<String>
13b0: 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c ..where S: Into<
13c0: 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 6e i64> {...let own
13d0: 65 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 2e er: i64 = owner.
13e0: 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d 75 into();...let mu
13f0: 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f t conn = self.po
1400: 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 ol.acquire().awa
1410: 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 it.....with_cont
1420: 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 ext(|| format!("
1430: 43 6c 65 61 6e 20 66 65 74 63 68 20 63 6f 6e 6e Clean fetch conn
1440: 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e :\n{:?}", &self.
1450: 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 74 63 68 pool))?;...match
1460: 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 sqlx::query("de
1470: 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 74 67 5f lete from rsstg_
1480: 70 6f 73 74 20 70 20 75 73 69 6e 67 20 72 73 73 post p using rss
1490: 74 67 5f 73 6f 75 72 63 65 20 73 20 77 68 65 72 tg_source s wher
14a0: 65 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 e p.source_id =
14b0: 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 $1 and owner = $
14c0: 32 20 61 6e 64 20 70 2e 73 6f 75 72 63 65 5f 69 2 and p.source_i
14d0: 64 20 3d 20 73 2e 73 6f 75 72 63 65 5f 69 64 3b d = s.source_id;
14e0: 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 ").....bind(sour
14f0: 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 ce_id).....bind(
1500: 6f 77 6e 65 72 29 0a 09 09 09 2e 65 78 65 63 75 owner).....execu
1510: 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 te(&mut conn).aw
1520: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
1530: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1540: 22 43 6c 65 61 6e 20 73 65 65 6e 20 70 6f 73 74 "Clean seen post
1550: 73 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 s:\n{:?}", &self
1560: 2e 70 6f 6f 6c 29 29 3f 0a 09 09 09 2e 72 6f 77 .pool))?.....row
1570: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
1580: 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 ..0 => { Ok("No
1590: 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 data found found
15a0: 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 \\.".to_string()
15b0: 29 20 7d 2c 0a 09 09 09 78 20 3d 3e 20 7b 20 4f ) },....x => { O
15c0: 6b 28 66 6f 72 6d 61 74 21 28 22 7b 7d 20 70 6f k(format!("{} po
15d0: 73 74 73 20 70 75 72 67 65 64 5c 5c 2e 22 2c 20 sts purged\\.",
15e0: 78 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 x)) },...}..}...
15f0: 61 73 79 6e 63 20 66 6e 20 65 6e 61 62 6c 65 3c async fn enable<
1600: 53 3e 28 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 S>(&self, source
1610: 5f 69 64 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 _id: &i32, owner
1620: 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 : S) -> Result<&
1630: 73 74 72 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 str>..where S: I
1640: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
1650: 20 6f 77 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 owner: i64 = ow
1660: 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 ner.into();...le
1670: 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c t mut conn = sel
1680: 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 f.pool.acquire()
1690: 2e 61 77 61 69 74 0a 09 09 09 2e 77 69 74 68 5f .await.....with_
16a0: 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 context(|| forma
16b0: 74 21 28 22 45 6e 61 62 6c 65 20 66 65 74 63 68 t!("Enable fetch
16c0: 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 conn:\n{:?}", &
16d0: 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 self.pool))?;...
16e0: 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 match sqlx::quer
16f0: 79 28 22 75 70 64 61 74 65 20 72 73 73 74 67 5f y("update rsstg_
1700: 73 6f 75 72 63 65 20 73 65 74 20 65 6e 61 62 6c source set enabl
1710: 65 64 20 3d 20 74 72 75 65 20 77 68 65 72 65 20 ed = true where
1720: 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 source_id = $1 a
1730: 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a nd owner = $2").
1740: 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f ....bind(source_
1750: 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e id).....bind(own
1760: 65 72 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 er).....execute(
1770: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
1780: 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 .....with_contex
1790: 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 45 6e t(|| format!("En
17a0: 61 62 6c 65 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a able source:\n{:
17b0: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
17c0: 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 )?.....rows_affe
17d0: 63 74 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e cted() {....1 =>
17e0: 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 64 69 { Ok("Source di
17f0: 73 61 62 6c 65 64 5c 5c 2e 22 29 20 7d 2c 0a 09 sabled\\.") },..
1800: 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 ..0 => { Ok("Sou
1810: 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 5c 5c 2e rce not found\\.
1820: 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 ") },...._ => {
1830: 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 61 74 Err(anyhow!("Dat
1840: 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 29 20 abase error."))
1850: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e },...}..}...asyn
1860: 63 20 66 6e 20 64 69 73 61 62 6c 65 3c 53 3e 28 c fn disable<S>(
1870: 26 73 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 &self, source_id
1880: 3a 20 26 69 33 32 2c 20 6f 77 6e 65 72 3a 20 53 : &i32, owner: S
1890: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 ) -> Result<&str
18a0: 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f >..where S: Into
18b0: 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 6f 77 <i64> {...let ow
18c0: 6e 65 72 3a 20 69 36 34 20 3d 20 6f 77 6e 65 72 ner: i64 = owner
18d0: 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c 65 74 20 6d .into();...let m
18e0: 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 ut conn = self.p
18f0: 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 ool.acquire().aw
1900: 61 69 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e ait.....with_con
1910: 74 65 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 text(|| format!(
1920: 22 44 69 73 61 62 6c 65 20 66 65 74 63 68 20 63 "Disable fetch c
1930: 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 onn:\n{:?}", &se
1940: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 6d 61 lf.pool))?;...ma
1950: 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 tch sqlx::query(
1960: 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f "update rsstg_so
1970: 75 72 63 65 20 73 65 74 20 65 6e 61 62 6c 65 64 urce set enabled
1980: 20 3d 20 66 61 6c 73 65 20 77 68 65 72 65 20 73 = false where s
1990: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e ource_id = $1 an
19a0: 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 d owner = $2")..
19b0: 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 ...bind(source_i
19c0: 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 d).....bind(owne
19d0: 72 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 r).....execute(&
19e0: 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 0a mut conn).await.
19f0: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
1a00: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 44 69 73 (|| format!("Dis
1a10: 61 62 6c 65 20 73 6f 75 72 63 65 3a 5c 6e 7b 3a able source:\n{:
1a20: 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 ?}", &self.pool)
1a30: 29 3f 0a 09 09 09 2e 72 6f 77 73 5f 61 66 66 65 )?.....rows_affe
1a40: 63 74 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e cted() {....1 =>
1a50: 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 64 69 { Ok("Source di
1a60: 73 61 62 6c 65 64 5c 5c 2e 22 29 20 7d 2c 0a 09 sabled\\.") },..
1a70: 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 ..0 => { Ok("Sou
1a80: 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 5c 5c 2e rce not found\\.
1a90: 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 ") },...._ => {
1aa0: 45 72 72 28 61 6e 79 68 6f 77 21 28 22 44 61 74 Err(anyhow!("Dat
1ab0: 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 29 20 abase error."))
1ac0: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 61 73 79 6e },...}..}...asyn
1ad0: 63 20 66 6e 20 75 70 64 61 74 65 3c 53 3e 28 26 c fn update<S>(&
1ae0: 73 65 6c 66 2c 20 75 70 64 61 74 65 3a 20 4f 70 self, update: Op
1af0: 74 69 6f 6e 3c 69 33 32 3e 2c 20 63 68 61 6e 6e tion<i32>, chann
1b00: 65 6c 3a 20 26 73 74 72 2c 20 63 68 61 6e 6e 65 el: &str, channe
1b10: 6c 5f 69 64 3a 20 69 36 34 2c 20 75 72 6c 3a 20 l_id: i64, url:
1b20: 26 73 74 72 2c 20 69 76 5f 68 61 73 68 3a 20 4f &str, iv_hash: O
1b30: 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e ption<&str>, own
1b40: 65 72 3a 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 er: S) -> Result
1b50: 3c 53 74 72 69 6e 67 3e 0a 09 77 68 65 72 65 20 <String>..where
1b60: 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 S: Into<i64> {..
1b70: 09 6c 65 74 20 6f 77 6e 65 72 3a 20 69 36 34 20 .let owner: i64
1b80: 3d 20 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a = owner.into();.
1b90: 09 09 6c 65 74 20 6d 75 74 20 63 6f 6e 6e 20 3d ..let mut conn =
1ba0: 20 73 65 6c 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 self.pool.acqui
1bb0: 72 65 28 29 2e 61 77 61 69 74 0a 09 09 09 2e 77 re().await.....w
1bc0: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
1bd0: 6f 72 6d 61 74 21 28 22 55 70 64 61 74 65 20 66 ormat!("Update f
1be0: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
1bf0: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
1c00: 3b 0a 0a 09 09 6d 61 74 63 68 20 6d 61 74 63 68 ;....match match
1c10: 20 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 6f update {.....So
1c20: 6d 65 28 69 64 29 20 3d 3e 20 7b 0a 09 09 09 09 me(id) => {.....
1c30: 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 .sqlx::query("up
1c40: 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 date rsstg_sourc
1c50: 65 20 73 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 e set channel_id
1c60: 20 3d 20 24 32 2c 20 75 72 6c 20 3d 20 24 33 2c = $2, url = $3,
1c70: 20 69 76 5f 68 61 73 68 20 3d 20 24 34 2c 20 6f iv_hash = $4, o
1c80: 77 6e 65 72 20 3d 20 24 35 2c 20 63 68 61 6e 6e wner = $5, chann
1c90: 65 6c 20 3d 20 24 36 20 77 68 65 72 65 20 73 6f el = $6 where so
1ca0: 75 72 63 65 5f 69 64 20 3d 20 24 31 22 29 2e 62 urce_id = $1").b
1cb0: 69 6e 64 28 69 64 29 0a 09 09 09 09 7d 2c 0a 09 ind(id).....},..
1cc0: 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 ...None => {....
1cd0: 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 69 ..sqlx::query("i
1ce0: 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 67 nsert into rsstg
1cf0: 5f 73 6f 75 72 63 65 20 28 63 68 61 6e 6e 65 6c _source (channel
1d00: 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 _id, url, iv_has
1d10: 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 6e 6e 65 h, owner, channe
1d20: 6c 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 l) values ($1, $
1d30: 32 2c 20 24 33 2c 20 24 34 2c 20 24 35 29 22 29 2, $3, $4, $5)")
1d40: 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 09 .....},....}....
1d50: 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c 5f 69 64 .bind(channel_id
1d60: 29 0a 09 09 09 2e 62 69 6e 64 28 75 72 6c 29 0a ).....bind(url).
1d70: 09 09 09 2e 62 69 6e 64 28 69 76 5f 68 61 73 68 ....bind(iv_hash
1d80: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
1d90: 29 0a 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e ).....bind(chann
1da0: 65 6c 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 el).....execute(
1db0: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
1dc0: 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 72 {....Ok(_) => r
1dd0: 65 74 75 72 6e 20 4f 6b 28 53 74 72 69 6e 67 3a eturn Ok(String:
1de0: 3a 66 72 6f 6d 28 22 43 68 61 6e 6e 65 6c 20 61 :from("Channel a
1df0: 64 64 65 64 5c 5c 2e 22 29 29 2c 0a 09 09 09 45 dded\\.")),....E
1e00: 72 72 28 73 71 6c 78 3a 3a 45 72 72 6f 72 3a 3a rr(sqlx::Error::
1e10: 44 61 74 61 62 61 73 65 28 65 72 72 29 29 20 3d Database(err)) =
1e20: 3e 20 7b 0a 09 09 09 09 6d 61 74 63 68 20 65 72 > {.....match er
1e30: 72 2e 64 6f 77 6e 63 61 73 74 3a 3a 3c 73 71 6c r.downcast::<sql
1e40: 78 3a 3a 70 6f 73 74 67 72 65 73 3a 3a 50 67 44 x::postgres::PgD
1e50: 61 74 61 62 61 73 65 45 72 72 6f 72 3e 28 29 2e atabaseError>().
1e60: 72 6f 75 74 69 6e 65 28 29 20 7b 0a 09 09 09 09 routine() {.....
1e70: 09 53 6f 6d 65 28 22 5f 62 74 5f 63 68 65 63 6b .Some("_bt_check
1e80: 5f 75 6e 69 71 75 65 22 2c 20 29 20 3d 3e 20 7b _unique", ) => {
1e90: 0a 09 09 09 09 09 09 72 65 74 75 72 6e 20 4f 6b .......return Ok
1ea0: 28 22 44 75 70 6c 69 63 61 74 65 20 6b 65 79 5c ("Duplicate key\
1eb0: 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 \.".to_string())
1ec0: 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 53 6f ......},......So
1ed0: 6d 65 28 5f 29 20 3d 3e 20 7b 0a 09 09 09 09 09 me(_) => {......
1ee0: 09 72 65 74 75 72 6e 20 4f 6b 28 22 44 61 74 61 .return Ok("Data
1ef0: 62 61 73 65 20 65 72 72 6f 72 5c 5c 2e 22 2e 74 base error\\.".t
1f00: 6f 5f 73 74 72 69 6e 67 28 29 29 0a 09 09 09 09 o_string()).....
1f10: 09 7d 2c 0a 09 09 09 09 09 4e 6f 6e 65 20 3d 3e .},......None =>
1f20: 20 7b 0a 09 09 09 09 09 09 72 65 74 75 72 6e 20 {.......return
1f30: 4f 6b 28 22 4e 6f 20 64 61 74 61 62 61 73 65 20 Ok("No database
1f40: 65 72 72 6f 72 20 65 78 74 72 61 63 74 65 64 5c error extracted\
1f50: 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 \.".to_string())
1f60: 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 7d 3b 0a ......},.....};.
1f70: 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 65 72 72 ...},....Err(err
1f80: 29 20 3d 3e 20 7b 0a 09 09 09 09 62 61 69 6c 21 ) => {.....bail!
1f90: 28 22 53 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e ("Sorry, unknown
1fa0: 20 65 72 72 6f 72 3a 5c 6e 7b 3a 23 3f 7d 5c 6e error:\n{:#?}\n
1fb0: 22 2c 20 65 72 72 29 3b 0a 09 09 09 7d 2c 0a 09 ", err);....},..
1fc0: 09 7d 3b 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 66 .};..}...async f
1fd0: 6e 20 61 75 74 6f 66 65 74 63 68 28 26 73 65 6c n autofetch(&sel
1fe0: 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e f) -> Result<()>
1ff0: 20 7b 0a 09 09 6c 65 74 20 6d 75 74 20 64 65 6c {...let mut del
2000: 61 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 72 ay = chrono::Dur
2010: 61 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 35 ation::minutes(5
2020: 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 6e 6f 77 );...let mut now
2030: 3b 0a 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 6c 65 ;...loop {....le
2040: 74 20 6d 75 74 20 63 6f 6e 6e 20 3d 20 73 65 6c t mut conn = sel
2050: 66 2e 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 29 f.pool.acquire()
2060: 2e 61 77 61 69 74 0a 09 09 09 09 2e 77 69 74 68 .await......with
2070: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
2080: 61 74 21 28 22 41 75 74 6f 66 65 74 63 68 20 66 at!("Autofetch f
2090: 65 74 63 68 20 63 6f 6e 6e 3a 5c 6e 7b 3a 3f 7d etch conn:\n{:?}
20a0: 22 2c 20 26 73 65 6c 66 2e 70 6f 6f 6c 29 29 3f ", &self.pool))?
20b0: 3b 0a 09 09 09 6e 6f 77 20 3d 20 63 68 72 6f 6e ;....now = chron
20c0: 6f 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f 77 28 29 3b o::Local::now();
20d0: 0a 09 09 09 6c 65 74 20 6d 75 74 20 71 75 65 75 ....let mut queu
20e0: 65 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 e = sqlx::query(
20f0: 22 73 65 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 "select source_i
2100: 64 2c 20 6e 65 78 74 5f 66 65 74 63 68 2c 20 6f d, next_fetch, o
2110: 77 6e 65 72 20 66 72 6f 6d 20 72 73 73 74 67 5f wner from rsstg_
2120: 6f 72 64 65 72 20 6e 61 74 75 72 61 6c 20 6c 65 order natural le
2130: 66 74 20 6a 6f 69 6e 20 72 73 73 74 67 5f 73 6f ft join rsstg_so
2140: 75 72 63 65 20 77 68 65 72 65 20 6e 65 78 74 5f urce where next_
2150: 66 65 74 63 68 20 3c 20 6e 6f 77 28 29 3b 22 29 fetch < now();")
2160: 0a 09 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 ......fetch_all(
2170: 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 &mut conn).await
2180: 3f 3b 0a 09 09 09 66 6f 72 20 72 6f 77 20 69 6e ?;....for row in
2190: 20 71 75 65 75 65 2e 69 74 65 72 28 29 20 7b 0a queue.iter() {.
21a0: 09 09 09 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 ....let source_i
21b0: 64 3a 20 69 33 32 20 3d 20 72 6f 77 2e 74 72 79 d: i32 = row.try
21c0: 5f 67 65 74 28 22 73 6f 75 72 63 65 5f 69 64 22 _get("source_id"
21d0: 29 3f 3b 0a 09 09 09 09 6c 65 74 20 6f 77 6e 65 )?;.....let owne
21e0: 72 3a 20 69 36 34 20 3d 20 72 6f 77 2e 74 72 79 r: i64 = row.try
21f0: 5f 67 65 74 28 22 6f 77 6e 65 72 22 29 3f 3b 0a _get("owner")?;.
2200: 09 09 09 09 6c 65 74 20 6e 65 78 74 5f 66 65 74 ....let next_fet
2210: 63 68 3a 20 44 61 74 65 54 69 6d 65 3c 63 68 72 ch: DateTime<chr
2220: 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3e 20 3d 20 72 6f ono::Local> = ro
2230: 77 2e 74 72 79 5f 67 65 74 28 22 6e 65 78 74 5f w.try_get("next_
2240: 66 65 74 63 68 22 29 3f 3b 0a 09 09 09 09 69 66 fetch")?;.....if
2250: 20 6e 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f next_fetch < no
2260: 77 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 w {......sqlx::q
2270: 75 65 72 79 28 22 75 70 64 61 74 65 20 72 73 73 uery("update rss
2280: 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 6c 61 tg_source set la
2290: 73 74 5f 73 63 72 61 70 65 20 3d 20 6e 6f 77 28 st_scrape = now(
22a0: 29 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 20 ) + interval '1
22b0: 68 6f 75 72 27 20 77 68 65 72 65 20 73 6f 75 72 hour' where sour
22c0: 63 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a 09 09 ce_id = $1;")...
22d0: 09 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 .....bind(source
22e0: 5f 69 64 29 0a 09 09 09 09 09 09 2e 65 78 65 63 _id)........exec
22f0: 75 74 65 28 26 6d 75 74 20 63 6f 6e 6e 29 2e 61 ute(&mut conn).a
2300: 77 61 69 74 0a 09 09 09 09 09 09 2e 77 69 74 68 wait........with
2310: 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 6f 72 6d _context(|| form
2320: 61 74 21 28 22 20 4c 6f 63 6b 20 73 6f 75 72 63 at!(" Lock sourc
2330: 65 3a 5c 6e 5c 6e 7b 3a 3f 7d 22 2c 20 26 73 65 e:\n\n{:?}", &se
2340: 6c 66 2e 70 6f 6f 6c 29 29 3f 3b 0a 09 09 09 09 lf.pool))?;.....
2350: 09 6c 65 74 20 63 6c 6f 6e 65 20 3d 20 73 65 6c .let clone = sel
2360: 66 2e 63 6c 6f 6e 65 28 29 3b 0a 09 09 09 09 09 f.clone();......
2370: 74 6f 6b 69 6f 3a 3a 73 70 61 77 6e 28 61 73 79 tokio::spawn(asy
2380: 6e 63 20 6d 6f 76 65 20 7b 0a 09 09 09 09 09 09 nc move {.......
2390: 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 29 20 if let Err(err)
23a0: 3d 20 63 6c 6f 6e 65 2e 63 68 65 63 6b 28 26 73 = clone.check(&s
23b0: 6f 75 72 63 65 5f 69 64 2c 20 6f 77 6e 65 72 2c ource_id, owner,
23c0: 20 74 72 75 65 29 2e 61 77 61 69 74 20 7b 0a 09 true).await {..
23d0: 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 72 ......if let Err
23e0: 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 64 65 (err) = clone.de
23f0: 62 75 67 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f bug(&format!("š
2400: 9b 91 20 7b 3a 3f 7d 22 2c 20 65 72 72 29 29 20 {:?}", err))
2410: 7b 0a 09 09 09 09 09 09 09 09 65 70 72 69 6e 74 {.........eprint
2420: 6c 6e 21 28 22 43 68 65 63 6b 20 65 72 72 6f 72 ln!("Check error
2430: 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a 09 09 09 : {}", err);....
2440: 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 3b 0a ....};.......};.
2450: 09 09 09 09 09 7d 29 3b 0a 09 09 09 09 7d 20 65 .....});.....} e
2460: 6c 73 65 20 7b 0a 09 09 09 09 09 69 66 20 6e 65 lse {......if ne
2470: 78 74 5f 66 65 74 63 68 20 2d 20 6e 6f 77 20 3c xt_fetch - now <
2480: 20 64 65 6c 61 79 20 7b 0a 09 09 09 09 09 09 64 delay {.......d
2490: 65 6c 61 79 20 3d 20 6e 65 78 74 5f 66 65 74 63 elay = next_fetc
24a0: 68 20 2d 20 6e 6f 77 3b 0a 09 09 09 09 09 7d 0a h - now;......}.
24b0: 09 09 09 09 7d 0a 09 09 09 7d 3b 0a 09 09 09 71 ....}....};....q
24c0: 75 65 75 65 2e 63 6c 65 61 72 28 29 3b 0a 09 09 ueue.clear();...
24d0: 09 74 6f 6b 69 6f 3a 3a 74 69 6d 65 3a 3a 64 65 .tokio::time::de
24e0: 6c 61 79 5f 66 6f 72 28 64 65 6c 61 79 2e 74 6f lay_for(delay.to
24f0: 5f 73 74 64 28 29 3f 29 2e 61 77 61 69 74 3b 0a _std()?).await;.
2500: 09 09 09 64 65 6c 61 79 20 3d 20 63 68 72 6f 6e ...delay = chron
2510: 6f 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6d 69 6e o::Duration::min
2520: 75 74 65 73 28 35 29 3b 0a 09 09 7d 0a 09 7d 0a utes(5);...}..}.
2530: 0a 09 61 73 79 6e 63 20 66 6e 20 6c 69 73 74 3c ..async fn list<
2540: 53 3e 28 26 73 65 6c 66 2c 20 6f 77 6e 65 72 3a S>(&self, owner:
2550: 20 53 29 20 2d 3e 20 52 65 73 75 6c 74 3c 56 65 S) -> Result<Ve
2560: 63 3c 53 74 72 69 6e 67 3e 3e 0a 09 77 68 65 72 c<String>>..wher
2570: 65 20 53 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e S: Into<i64> {
2580: 0a 09 09 6c 65 74 20 6f 77 6e 65 72 20 3d 20 6f ...let owner = o
2590: 77 6e 65 72 2e 69 6e 74 6f 28 29 3b 0a 09 09 6c wner.into();...l
25a0: 65 74 20 6d 75 74 20 72 65 70 6c 79 20 3d 20 76 et mut reply = v
25b0: 65 63 21 5b 5d 3b 0a 09 09 6c 65 74 20 6d 75 74 ec![];...let mut
25c0: 20 63 6f 6e 6e 20 3d 20 73 65 6c 66 2e 70 6f 6f conn = self.poo
25d0: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
25e0: 74 0a 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 t.....with_conte
25f0: 78 74 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 4c xt(|| format!("L
2600: 69 73 74 20 66 65 74 63 68 20 63 6f 6e 6e 3a 5c ist fetch conn:\
2610: 6e 7b 3a 3f 7d 22 2c 20 26 73 65 6c 66 2e 70 6f n{:?}", &self.po
2620: 6f 6c 29 29 3f 3b 0a 09 09 72 65 70 6c 79 2e 70 ol))?;...reply.p
2630: 75 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 2e ush("Channels:".
2640: 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 to_string());...
2650: 6c 65 74 20 72 6f 77 73 20 3d 20 73 71 6c 78 3a let rows = sqlx:
2660: 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 73 :query("select s
2670: 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 ource_id, channe
2680: 6c 2c 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c 2c l, enabled, url,
2690: 20 69 76 5f 68 61 73 68 20 66 72 6f 6d 20 72 73 iv_hash from rs
26a0: 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 stg_source where
26b0: 20 6f 77 6e 65 72 20 3d 20 24 31 20 6f 72 64 65 owner = $1 orde
26c0: 72 20 62 79 20 73 6f 75 72 63 65 5f 69 64 22 29 r by source_id")
26d0: 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 29 .....bind(owner)
26e0: 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 26 .....fetch_all(&
26f0: 6d 75 74 20 63 6f 6e 6e 29 2e 61 77 61 69 74 3f mut conn).await?
2700: 3b 0a 09 09 66 6f 72 20 72 6f 77 20 69 6e 20 72 ;...for row in r
2710: 6f 77 73 2e 69 74 65 72 28 29 20 7b 0a 09 09 09 ows.iter() {....
2720: 6c 65 74 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 let source_id: i
2730: 33 32 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 32 = row.try_get
2740: 28 22 73 6f 75 72 63 65 5f 69 64 22 29 3f 3b 0a ("source_id")?;.
2750: 09 09 09 6c 65 74 20 75 73 65 72 6e 61 6d 65 3a ...let username:
2760: 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f &str = row.try_
2770: 67 65 74 28 22 63 68 61 6e 6e 65 6c 22 29 3f 3b get("channel")?;
2780: 0a 09 09 09 6c 65 74 20 65 6e 61 62 6c 65 64 3a ....let enabled:
2790: 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 5f bool = row.try_
27a0: 67 65 74 28 22 65 6e 61 62 6c 65 64 22 29 3f 3b get("enabled")?;
27b0: 0a 09 09 09 6c 65 74 20 75 72 6c 3a 20 26 73 74 ....let url: &st
27c0: 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 r = row.try_get(
27d0: 22 75 72 6c 22 29 3f 3b 0a 09 09 09 6c 65 74 20 "url")?;....let
27e0: 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c iv_hash: Option<
27f0: 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f &str> = row.try_
2800: 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f 3b get("iv_hash")?;
2810: 0a 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 ....reply.push(f
2820: 6f 72 6d 61 74 21 28 22 5c 6e 5c 5c 23 ef b8 8f ormat!("\n\\#ļø
2830: e2 83 a3 20 7b 7d 20 5c 5c 2a ef b8 8f e2 83 a3 ā£ {} \\*ļøā£
2840: 20 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 97 20 60 `{}` {}\nš `
2850: 7b 7d 60 22 2c 20 73 6f 75 72 63 65 5f 69 64 2c {}`", source_id,
2860: 20 75 73 65 72 6e 61 6d 65 2c 20 20 0a 09 09 09 username, ....
2870: 09 6d 61 74 63 68 20 65 6e 61 62 6c 65 64 20 7b .match enabled {
2880: 0a 09 09 09 09 09 74 72 75 65 20 20 3d 3e 20 22 ......true => "
2890: f0 9f 94 84 20 65 6e 61 62 6c 65 64 22 2c 0a 09 š enabled",..
28a0: 09 09 09 09 66 61 6c 73 65 20 3d 3e 20 22 e2 9b ....false => "ā
28b0: 94 20 64 69 73 61 62 6c 65 64 22 2c 0a 09 09 09 disabled",....
28c0: 09 7d 2c 20 75 72 6c 29 29 3b 0a 09 09 09 69 66 .}, url));....if
28d0: 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 68 29 20 let Some(hash)
28e0: 3d 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 = iv_hash {.....
28f0: 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d 61 reply.push(forma
2900: 74 21 28 22 49 56 20 60 7b 7d 60 22 2c 20 68 61 t!("IV `{}`", ha
2910: 73 68 29 29 3b 0a 09 09 09 7d 0a 09 09 7d 3b 0a sh));....}...};.
2920: 09 09 4f 6b 28 72 65 70 6c 79 29 0a 09 7d 0a 7d ..Ok(reply)..}.}
2930: 0a 0a 23 5b 74 6f 6b 69 6f 3a 3a 6d 61 69 6e 5d ..#[tokio::main]
2940: 0a 61 73 79 6e 63 20 66 6e 20 6d 61 69 6e 28 29 .async fn main()
2950: 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 20 7b -> Result<()> {
2960: 0a 09 6c 65 74 20 6d 75 74 20 73 65 74 74 69 6e ..let mut settin
2970: 67 73 20 3d 20 63 6f 6e 66 69 67 3a 3a 43 6f 6e gs = config::Con
2980: 66 69 67 3a 3a 64 65 66 61 75 6c 74 28 29 3b 0a fig::default();.
2990: 09 73 65 74 74 69 6e 67 73 2e 6d 65 72 67 65 28 .settings.merge(
29a0: 63 6f 6e 66 69 67 3a 3a 46 69 6c 65 3a 3a 77 69 config::File::wi
29b0: 74 68 5f 6e 61 6d 65 28 22 72 73 73 74 67 22 29 th_name("rsstg")
29c0: 29 3f 3b 0a 0a 09 6c 65 74 20 63 6f 72 65 20 3d )?;...let core =
29d0: 20 43 6f 72 65 3a 3a 6e 65 77 28 73 65 74 74 69 Core::new(setti
29e0: 6e 67 73 29 2e 61 77 61 69 74 3f 3b 0a 0a 09 6c ngs).await?;...l
29f0: 65 74 20 6d 75 74 20 73 74 72 65 61 6d 20 3d 20 et mut stream =
2a00: 63 6f 72 65 2e 73 74 72 65 61 6d 28 29 3b 0a 0a core.stream();..
2a10: 09 77 68 69 6c 65 20 6c 65 74 20 53 6f 6d 65 28 .while let Some(
2a20: 75 70 64 61 74 65 29 20 3d 20 73 74 72 65 61 6d update) = stream
2a30: 2e 6e 65 78 74 28 29 2e 61 77 61 69 74 20 7b 0a .next().await {.
2a40: 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 72 ..if let Err(err
2a50: 29 20 3d 20 68 61 6e 64 6c 65 28 75 70 64 61 74 ) = handle(updat
2a60: 65 3f 2c 20 26 63 6f 72 65 29 2e 61 77 61 69 74 e?, &core).await
2a70: 20 7b 0a 09 09 09 63 6f 72 65 2e 64 65 62 75 67 {....core.debug
2a80: 28 26 66 6f 72 6d 61 74 21 28 22 f0 9f 9b 91 20 (&format!("š
2a90: 7b 3a 3f 7d 22 2c 20 65 72 72 29 29 3f 3b 0a 09 {:?}", err))?;..
2aa0: 09 7d 3b 0a 09 7d 0a 0a 09 4f 6b 28 28 29 29 0a .};..}...Ok(()).
2ab0: 7d 0a 0a 61 73 79 6e 63 20 66 6e 20 68 61 6e 64 }..async fn hand
2ac0: 6c 65 28 75 70 64 61 74 65 3a 20 74 65 6c 65 67 le(update: teleg
2ad0: 72 61 6d 5f 62 6f 74 3a 3a 55 70 64 61 74 65 2c ram_bot::Update,
2ae0: 20 63 6f 72 65 3a 20 26 43 6f 72 65 29 20 2d 3e core: &Core) ->
2af0: 20 52 65 73 75 6c 74 3c 28 29 3e 20 7b 0a 09 6c Result<()> {..l
2b00: 61 7a 79 5f 73 74 61 74 69 63 21 20 7b 0a 09 09 azy_static! {...
2b10: 73 74 61 74 69 63 20 72 65 66 20 52 45 5f 55 53 static ref RE_US
2b20: 45 52 4e 41 4d 45 3a 20 52 65 67 65 78 20 3d 20 ERNAME: Regex =
2b30: 52 65 67 65 78 3a 3a 6e 65 77 28 72 22 5e 40 5b Regex::new(r"^@[
2b40: 61 2d 7a 41 2d 5a 5d 5b 61 2d 7a 41 2d 5a 30 2d a-zA-Z][a-zA-Z0-
2b50: 39 5f 5d 2b 24 22 29 2e 75 6e 77 72 61 70 28 29 9_]+$").unwrap()
2b60: 3b 0a 09 09 73 74 61 74 69 63 20 72 65 66 20 52 ;...static ref R
2b70: 45 5f 4c 49 4e 4b 3a 20 52 65 67 65 78 20 3d 20 E_LINK: Regex =
2b80: 52 65 67 65 78 3a 3a 6e 65 77 28 72 22 5e 68 74 Regex::new(r"^ht
2b90: 74 70 73 3f 3a 2f 2f 5b 61 2d 7a 41 2d 5a 2e 30 tps?://[a-zA-Z.0
2ba0: 2d 39 2d 5d 2b 2f 5b 2d 5f 61 2d 7a 41 2d 5a 2e -9-]+/[-_a-zA-Z.
2bb0: 30 2d 39 2f 3f 3d 5d 2b 24 22 29 2e 75 6e 77 72 0-9/?=]+$").unwr
2bc0: 61 70 28 29 3b 0a 09 09 73 74 61 74 69 63 20 72 ap();...static r
2bd0: 65 66 20 52 45 5f 49 56 5f 48 41 53 48 3a 20 52 ef RE_IV_HASH: R
2be0: 65 67 65 78 20 3d 20 52 65 67 65 78 3a 3a 6e 65 egex = Regex::ne
2bf0: 77 28 72 22 5e 5b 61 2d 66 30 2d 39 5d 7b 31 34 w(r"^[a-f0-9]{14
2c00: 7d 24 22 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 }$").unwrap();..
2c10: 7d 0a 0a 09 6d 61 74 63 68 20 75 70 64 61 74 65 }...match update
2c20: 2e 6b 69 6e 64 20 7b 0a 09 09 55 70 64 61 74 65 .kind {...Update
2c30: 4b 69 6e 64 3a 3a 4d 65 73 73 61 67 65 28 6d 65 Kind::Message(me
2c40: 73 73 61 67 65 29 20 3d 3e 20 7b 0a 09 09 09 6c ssage) => {....l
2c50: 65 74 20 6d 75 74 20 72 65 70 6c 79 3a 20 56 65 et mut reply: Ve
2c60: 63 3c 53 74 72 69 6e 67 3e 20 3d 20 76 65 63 21 c<String> = vec!
2c70: 5b 5d 3b 0a 09 09 09 6d 61 74 63 68 20 6d 65 73 [];....match mes
2c80: 73 61 67 65 2e 6b 69 6e 64 20 7b 0a 09 09 09 09 sage.kind {.....
2c90: 4d 65 73 73 61 67 65 4b 69 6e 64 3a 3a 54 65 78 MessageKind::Tex
2ca0: 74 20 7b 20 72 65 66 20 64 61 74 61 2c 20 2e 2e t { ref data, ..
2cb0: 20 7d 20 3d 3e 20 7b 0a 09 09 09 09 09 6c 65 74 } => {......let
2cc0: 20 6d 75 74 20 77 6f 72 64 73 20 3d 20 64 61 74 mut words = dat
2cd0: 61 2e 73 70 6c 69 74 5f 77 68 69 74 65 73 70 61 a.split_whitespa
2ce0: 63 65 28 29 3b 0a 09 09 09 09 09 6c 65 74 20 63 ce();......let c
2cf0: 6d 64 20 3d 20 77 6f 72 64 73 2e 6e 65 78 74 28 md = words.next(
2d00: 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 ).unwrap();.....
2d10: 09 6d 61 74 63 68 20 63 6d 64 20 7b 0a 0a 2f 2f .match cmd {..//
2d20: 20 73 74 61 72 74 0a 0a 09 09 09 09 09 09 22 2f start........"/
2d30: 73 74 61 72 74 22 20 3d 3e 20 7b 0a 09 09 09 09 start" => {.....
2d40: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 22 57 ...reply.push("W
2d50: 65 20 61 72 65 20 6f 70 65 6e 5c 5c 2e 20 50 72 e are open\\. Pr
2d60: 6f 62 61 62 6c 79 5c 5c 2e 20 56 69 73 69 74 20 obably\\. Visit
2d70: 5b 63 68 61 6e 6e 65 6c 5d 28 68 74 74 70 73 3a [channel](https:
2d80: 2f 2f 74 2e 6d 65 2f 72 73 73 74 67 5f 62 6f 74 //t.me/rsstg_bot
2d90: 5f 68 65 6c 70 2f 33 29 20 66 6f 72 20 64 65 74 _help/3) for det
2da0: 61 69 6c 73 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 ails\\.".to_stri
2db0: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 7d 2c 0a ng());.......},.
2dc0: 0a 2f 2f 20 6c 69 73 74 0a 0a 09 09 09 09 09 09 .// list........
2dd0: 22 2f 6c 69 73 74 22 20 3d 3e 20 7b 0a 09 09 09 "/list" => {....
2de0: 09 09 09 09 72 65 70 6c 79 2e 61 70 70 65 6e 64 ....reply.append
2df0: 28 26 6d 75 74 20 63 6f 72 65 2e 6c 69 73 74 28 (&mut core.list(
2e00: 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 29 message.from.id)
2e10: 2e 61 77 61 69 74 3f 29 3b 0a 09 09 09 09 09 09 .await?);.......
2e20: 7d 2c 0a 0a 2f 2f 20 61 64 64 0a 0a 09 09 09 09 },..// add......
2e30: 09 09 22 2f 61 64 64 22 20 7c 20 22 2f 75 70 64 .."/add" | "/upd
2e40: 61 74 65 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 ate" => {.......
2e50: 09 6c 65 74 20 6d 75 74 20 73 6f 75 72 63 65 5f .let mut source_
2e60: 69 64 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 20 id: Option<i32>
2e70: 3d 20 4e 6f 6e 65 3b 0a 09 09 09 09 09 09 09 6c = None;........l
2e80: 65 74 20 61 74 5f 6c 65 61 73 74 20 3d 20 22 52 et at_least = "R
2e90: 65 71 75 69 72 65 73 20 61 74 20 6c 65 61 73 74 equires at least
2ea0: 20 33 20 70 61 72 61 6d 65 74 65 72 73 2e 22 3b 3 parameters.";
2eb0: 0a 09 09 09 09 09 09 09 69 66 20 63 6d 64 20 3d ........if cmd =
2ec0: 3d 20 22 2f 75 70 64 61 74 65 22 20 7b 0a 09 09 = "/update" {...
2ed0: 09 09 09 09 09 09 6c 65 74 20 66 69 72 73 74 5f ......let first_
2ee0: 77 6f 72 64 20 3d 20 77 6f 72 64 73 2e 6e 65 78 word = words.nex
2ef0: 74 28 29 0a 09 09 09 09 09 09 09 09 09 2e 63 6f t()...........co
2f00: 6e 74 65 78 74 28 61 74 5f 6c 65 61 73 74 29 3f ntext(at_least)?
2f10: 3b 0a 09 09 09 09 09 09 09 09 73 6f 75 72 63 65 ;.........source
2f20: 5f 69 64 20 3d 20 53 6f 6d 65 28 66 69 72 73 74 _id = Some(first
2f30: 5f 77 6f 72 64 2e 70 61 72 73 65 3a 3a 3c 69 33 _word.parse::<i3
2f40: 32 3e 28 29 0a 09 09 09 09 09 09 09 09 09 2e 77 2>()...........w
2f50: 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c 20 66 ith_context(|| f
2f60: 6f 72 6d 61 74 21 28 22 49 20 6e 65 65 64 20 61 ormat!("I need a
2f70: 20 6e 75 6d 62 65 72 2c 20 62 75 74 20 67 6f 74 number, but got
2f80: 20 7b 7d 2e 22 2c 20 66 69 72 73 74 5f 77 6f 72 {}.", first_wor
2f90: 64 29 29 3f 29 3b 0a 09 09 09 09 09 09 09 7d 0a d))?);........}.
2fa0: 09 09 09 09 09 09 09 6c 65 74 20 28 63 68 61 6e .......let (chan
2fb0: 6e 65 6c 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 nel, url, iv_has
2fc0: 68 29 20 3d 20 28 0a 09 09 09 09 09 09 09 09 77 h) = (.........w
2fd0: 6f 72 64 73 2e 6e 65 78 74 28 29 2e 63 6f 6e 74 ords.next().cont
2fe0: 65 78 74 28 61 74 5f 6c 65 61 73 74 29 3f 2c 0a ext(at_least)?,.
2ff0: 09 09 09 09 09 09 09 09 77 6f 72 64 73 2e 6e 65 ........words.ne
3000: 78 74 28 29 2e 63 6f 6e 74 65 78 74 28 61 74 5f xt().context(at_
3010: 6c 65 61 73 74 29 3f 2c 0a 09 09 09 09 09 09 09 least)?,........
3020: 09 77 6f 72 64 73 2e 6e 65 78 74 28 29 29 3b 0a .words.next());.
3030: 09 09 09 09 09 09 09 69 66 20 21 20 52 45 5f 55 .......if ! RE_U
3040: 53 45 52 4e 41 4d 45 2e 69 73 5f 6d 61 74 63 68 SERNAME.is_match
3050: 28 26 63 68 61 6e 6e 65 6c 29 20 7b 0a 09 09 09 (&channel) {....
3060: 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 .....reply.push(
3070: 22 55 73 65 72 6e 61 6d 65 73 20 73 68 6f 75 6c "Usernames shoul
3080: 64 20 62 65 20 73 6f 6d 65 74 68 69 6e 67 20 6c d be something l
3090: 69 6b 65 20 5c 22 40 5c 5c 5b 61 5c 5c 2d 7a 41 ike \"@\\[a\\-zA
30a0: 5c 5c 2d 5a 5d 5c 5c 5b 61 5c 5c 2d 7a 41 5c 5c \\-Z]\\[a\\-zA\\
30b0: 2d 5a 30 5c 5c 2d 39 5c 5c 5f 5d 2b 5c 22 2c 20 -Z0\\-9\\_]+\",
30c0: 61 72 65 6e 27 74 20 74 68 65 79 3f 22 2e 74 6f aren't they?".to
30d0: 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 _string());.....
30e0: 09 09 09 09 62 61 69 6c 21 28 22 57 72 6f 6e 67 ....bail!("Wrong
30f0: 20 75 73 65 72 6e 61 6d 65 20 7b 3a 3f 7d 2e 22 username {:?}."
3100: 2c 20 26 63 68 61 6e 6e 65 6c 29 3b 0a 09 09 09 , &channel);....
3110: 09 09 09 09 7d 0a 09 09 09 09 09 09 09 69 66 20 ....}........if
3120: 21 20 52 45 5f 4c 49 4e 4b 2e 69 73 5f 6d 61 74 ! RE_LINK.is_mat
3130: 63 68 28 26 75 72 6c 29 20 7b 0a 09 09 09 09 09 ch(&url) {......
3140: 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 22 4c ...reply.push("L
3150: 69 6e 6b 20 73 68 6f 75 6c 64 20 62 65 20 6c 69 ink should be li
3160: 6e 6b 20 74 6f 20 61 74 6f 6d 2f 72 73 73 20 66 nk to atom/rss f
3170: 65 65 64 2c 20 73 6f 6d 65 74 68 69 6e 67 20 6c eed, something l
3180: 69 6b 65 20 5c 22 68 74 74 70 73 3a 2f 2f 64 6f ike \"https://do
3190: 6d 61 69 6e 2f 70 61 74 68 5c 22 5c 5c 2e 22 2e main/path\"\\.".
31a0: 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 to_string());...
31b0: 09 09 09 09 09 09 62 61 69 6c 21 28 22 55 72 6c ......bail!("Url
31c0: 3a 20 7b 3a 3f 7d 22 2c 20 26 75 72 6c 29 3b 0a : {:?}", &url);.
31d0: 09 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 09 .......}........
31e0: 69 66 20 6c 65 74 20 53 6f 6d 65 28 68 61 73 68 if let Some(hash
31f0: 29 20 3d 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 ) = iv_hash {...
3200: 09 09 09 09 09 09 69 66 20 21 20 52 45 5f 49 56 ......if ! RE_IV
3210: 5f 48 41 53 48 2e 69 73 5f 6d 61 74 63 68 28 26 _HASH.is_match(&
3220: 68 61 73 68 29 20 7b 0a 09 09 09 09 09 09 09 09 hash) {.........
3230: 09 72 65 70 6c 79 2e 70 75 73 68 28 22 49 56 20 .reply.push("IV
3240: 68 61 73 68 20 73 68 6f 75 6c 64 20 62 65 20 31 hash should be 1
3250: 34 20 68 65 78 20 64 69 67 69 74 73 2e 22 2e 74 4 hex digits.".t
3260: 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 o_string());....
3270: 09 09 09 09 09 09 62 61 69 6c 21 28 22 49 56 3a ......bail!("IV:
3280: 20 7b 3a 3f 7d 22 2c 20 26 69 76 5f 68 61 73 68 {:?}", &iv_hash
3290: 29 3b 0a 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 );.........};...
32a0: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 6c .....};........l
32b0: 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 20 3d 20 et channel_id =
32c0: 69 36 34 3a 3a 66 72 6f 6d 28 63 6f 72 65 2e 74 i64::from(core.t
32d0: 67 2e 73 65 6e 64 28 74 65 6c 65 67 72 61 6d 5f g.send(telegram_
32e0: 62 6f 74 3a 3a 47 65 74 43 68 61 74 3a 3a 6e 65 bot::GetChat::ne
32f0: 77 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a w(telegram_bot::
3300: 74 79 70 65 73 3a 3a 43 68 61 74 52 65 66 3a 3a types::ChatRef::
3310: 43 68 61 6e 6e 65 6c 55 73 65 72 6e 61 6d 65 28 ChannelUsername(
3320: 63 68 61 6e 6e 65 6c 2e 74 6f 5f 73 74 72 69 6e channel.to_strin
3330: 67 28 29 29 29 29 2e 61 77 61 69 74 3f 2e 69 64 g()))).await?.id
3340: 28 29 29 3b 0a 09 09 09 09 09 09 09 6c 65 74 20 ());........let
3350: 63 68 61 6e 5f 61 64 6d 20 3d 20 63 6f 72 65 2e chan_adm = core.
3360: 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 72 61 6d tg.send(telegram
3370: 5f 62 6f 74 3a 3a 47 65 74 43 68 61 74 41 64 6d _bot::GetChatAdm
3380: 69 6e 69 73 74 72 61 74 6f 72 73 3a 3a 6e 65 77 inistrators::new
3390: 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 (telegram_bot::t
33a0: 79 70 65 73 3a 3a 43 68 61 74 52 65 66 3a 3a 43 ypes::ChatRef::C
33b0: 68 61 6e 6e 65 6c 55 73 65 72 6e 61 6d 65 28 63 hannelUsername(c
33c0: 68 61 6e 6e 65 6c 2e 74 6f 5f 73 74 72 69 6e 67 hannel.to_string
33d0: 28 29 29 29 29 2e 61 77 61 69 74 0a 09 09 09 09 ()))).await.....
33e0: 09 09 09 09 2e 63 6f 6e 74 65 78 74 28 22 53 6f .....context("So
33f0: 72 72 79 2c 20 49 20 68 61 76 65 20 6e 6f 20 61 rry, I have no a
3400: 63 63 65 73 73 20 74 6f 20 74 68 61 74 20 63 68 ccess to that ch
3410: 61 74 5c 5c 2e 22 29 3f 3b 0a 09 09 09 09 09 09 at\\.")?;.......
3420: 09 6c 65 74 20 28 6d 75 74 20 6d 65 2c 20 6d 75 .let (mut me, mu
3430: 74 20 75 73 65 72 29 20 3d 20 28 66 61 6c 73 65 t user) = (false
3440: 2c 20 66 61 6c 73 65 29 3b 0a 09 09 09 09 09 09 , false);.......
3450: 09 66 6f 72 20 61 64 6d 69 6e 20 69 6e 20 63 68 .for admin in ch
3460: 61 6e 5f 61 64 6d 20 7b 0a 09 09 09 09 09 09 09 an_adm {........
3470: 09 69 66 20 61 64 6d 69 6e 2e 75 73 65 72 2e 69 .if admin.user.i
3480: 64 20 3d 3d 20 63 6f 72 65 2e 6d 79 2e 69 64 20 d == core.my.id
3490: 7b 0a 09 09 09 09 09 09 09 09 09 6d 65 20 3d 20 {..........me =
34a0: 74 72 75 65 3b 0a 09 09 09 09 09 09 09 09 7d 3b true;.........};
34b0: 0a 09 09 09 09 09 09 09 09 69 66 20 61 64 6d 69 .........if admi
34c0: 6e 2e 75 73 65 72 2e 69 64 20 3d 3d 20 6d 65 73 n.user.id == mes
34d0: 73 61 67 65 2e 66 72 6f 6d 2e 69 64 20 7b 0a 09 sage.from.id {..
34e0: 09 09 09 09 09 09 09 09 75 73 65 72 20 3d 20 74 ........user = t
34f0: 72 75 65 3b 0a 09 09 09 09 09 09 09 09 7d 3b 0a rue;.........};.
3500: 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 .......};.......
3510: 09 69 66 20 21 20 6d 65 20 20 20 7b 20 62 61 69 .if ! me { bai
3520: 6c 21 28 22 49 20 6e 65 65 64 20 74 6f 20 62 65 l!("I need to be
3530: 20 61 64 6d 69 6e 20 6f 6e 20 74 68 61 74 20 63 admin on that c
3540: 68 61 6e 6e 65 6c 5c 5c 2e 22 29 3b 20 7d 3b 0a hannel\\."); };.
3550: 09 09 09 09 09 09 09 69 66 20 21 20 75 73 65 72 .......if ! user
3560: 20 7b 20 62 61 69 6c 21 28 22 59 6f 75 20 73 68 { bail!("You sh
3570: 6f 75 6c 64 20 62 65 20 61 64 6d 69 6e 20 6f 6e ould be admin on
3580: 20 74 68 61 74 20 63 68 61 6e 6e 65 6c 5c 5c 2e that channel\\.
3590: 22 29 3b 20 7d 3b 0a 09 09 09 09 09 09 09 72 65 "); };........re
35a0: 70 6c 79 2e 70 75 73 68 28 63 6f 72 65 2e 75 70 ply.push(core.up
35b0: 64 61 74 65 28 73 6f 75 72 63 65 5f 69 64 2c 20 date(source_id,
35c0: 63 68 61 6e 6e 65 6c 2c 20 63 68 61 6e 6e 65 6c channel, channel
35d0: 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 _id, url, iv_has
35e0: 68 2c 20 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e h, message.from.
35f0: 69 64 29 2e 61 77 61 69 74 3f 29 3b 0a 09 09 09 id).await?);....
3600: 09 09 09 7d 2c 0a 0a 2f 2f 20 63 68 65 63 6b 0a ...},..// check.
3610: 0a 09 09 09 09 09 09 22 2f 63 68 65 63 6b 22 20 ......."/check"
3620: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 6d 61 74 63 => {........matc
3630: 68 20 26 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e h &words.next().
3640: 75 6e 77 72 61 70 28 29 2e 70 61 72 73 65 3a 3a unwrap().parse::
3650: 3c 69 33 32 3e 28 29 20 7b 0a 09 09 09 09 09 09 <i32>() {.......
3660: 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a ..Err(err) => {.
3670: 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 .........reply.p
3680: 75 73 68 28 66 6f 72 6d 61 74 21 28 22 49 20 6e ush(format!("I n
3690: 65 65 64 20 61 20 6e 75 6d 62 65 72 5c 5c 2e 5c eed a number\\.\
36a0: 6e 7b 7d 22 2c 20 26 65 72 72 29 29 3b 0a 09 09 n{}", &err));...
36b0: 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 ......},........
36c0: 09 4f 6b 28 6e 75 6d 62 65 72 29 20 3d 3e 20 7b .Ok(number) => {
36d0: 0a 09 09 09 09 09 09 09 09 09 63 6f 72 65 2e 63 ..........core.c
36e0: 68 65 63 6b 28 26 6e 75 6d 62 65 72 2c 20 6d 65 heck(&number, me
36f0: 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 2c 20 66 ssage.from.id, f
3700: 61 6c 73 65 29 2e 61 77 61 69 74 0a 09 09 09 09 alse).await.....
3710: 09 09 09 09 09 09 2e 63 6f 6e 74 65 78 74 28 22 .......context("
3720: 43 68 61 6e 6e 65 6c 20 63 68 65 63 6b 20 66 61 Channel check fa
3730: 69 6c 65 64 2e 22 29 3f 3b 0a 09 09 09 09 09 09 iled.")?;.......
3740: 09 09 7d 2c 0a 09 09 09 09 09 09 09 7d 3b 0a 09 ..},........};..
3750: 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 63 6c 65 61 .....},..// clea
3760: 6e 0a 0a 09 09 09 09 09 09 22 2f 63 6c 65 61 6e n........"/clean
3770: 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 6d 61 " => {........ma
3780: 74 63 68 20 26 77 6f 72 64 73 2e 6e 65 78 74 28 tch &words.next(
3790: 29 2e 75 6e 77 72 61 70 28 29 2e 70 61 72 73 65 ).unwrap().parse
37a0: 3a 3a 3c 69 33 32 3e 28 29 20 7b 0a 09 09 09 09 ::<i32>() {.....
37b0: 09 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 ....Err(err) =>
37c0: 7b 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 {..........reply
37d0: 2e 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 49 .push(format!("I
37e0: 20 6e 65 65 64 20 61 20 6e 75 6d 62 65 72 5c 5c need a number\\
37f0: 2e 5c 6e 7b 7d 22 2c 20 26 65 72 72 29 29 3b 0a .\n{}", &err));.
3800: 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ........},......
3810: 09 09 09 4f 6b 28 6e 75 6d 62 65 72 29 20 3d 3e ...Ok(number) =>
3820: 20 7b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 {..........let
3830: 72 65 73 75 6c 74 20 3d 20 63 6f 72 65 2e 63 6c result = core.cl
3840: 65 61 6e 28 26 6e 75 6d 62 65 72 2c 20 6d 65 73 ean(&number, mes
3850: 73 61 67 65 2e 66 72 6f 6d 2e 69 64 29 2e 61 77 sage.from.id).aw
3860: 61 69 74 3f 3b 0a 09 09 09 09 09 09 09 09 09 72 ait?;..........r
3870: 65 70 6c 79 2e 70 75 73 68 28 72 65 73 75 6c 74 eply.push(result
3880: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 .to_string());..
3890: 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 .......},.......
38a0: 09 7d 3b 0a 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f .};.......},..//
38b0: 20 65 6e 61 62 6c 65 0a 0a 09 09 09 09 09 09 22 enable........"
38c0: 2f 65 6e 61 62 6c 65 22 20 3d 3e 20 7b 0a 09 09 /enable" => {...
38d0: 09 09 09 09 09 6d 61 74 63 68 20 26 77 6f 72 64 .....match &word
38e0: 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 70 28 s.next().unwrap(
38f0: 29 2e 70 61 72 73 65 3a 3a 3c 69 33 32 3e 28 29 ).parse::<i32>()
3900: 20 7b 0a 09 09 09 09 09 09 09 09 45 72 72 28 65 {.........Err(e
3910: 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 rr) => {........
3920: 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 ..reply.push(for
3930: 6d 61 74 21 28 22 49 20 6e 65 65 64 20 61 20 6e mat!("I need a n
3940: 75 6d 62 65 72 5c 5c 2e 5c 6e 7b 7d 22 2c 20 26 umber\\.\n{}", &
3950: 65 72 72 29 29 3b 0a 09 09 09 09 09 09 09 09 7d err));.........}
3960: 2c 0a 09 09 09 09 09 09 09 09 4f 6b 28 6e 75 6d ,.........Ok(num
3970: 62 65 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 ber) => {.......
3980: 09 09 09 6c 65 74 20 72 65 73 75 6c 74 20 3d 20 ...let result =
3990: 63 6f 72 65 2e 65 6e 61 62 6c 65 28 26 6e 75 6d core.enable(&num
39a0: 62 65 72 2c 20 6d 65 73 73 61 67 65 2e 66 72 6f ber, message.fro
39b0: 6d 2e 69 64 29 2e 61 77 61 69 74 3f 3b 0a 09 09 m.id).await?;...
39c0: 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 .......reply.pus
39d0: 68 28 72 65 73 75 6c 74 2e 74 6f 5f 73 74 72 69 h(result.to_stri
39e0: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 7d ng());.........}
39f0: 2c 0a 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 ,........};.....
3a00: 09 09 7d 2c 0a 0a 2f 2f 20 64 65 6c 65 74 65 0a ..},..// delete.
3a10: 0a 09 09 09 09 09 09 22 2f 64 65 6c 65 74 65 22 ......."/delete"
3a20: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 6d 61 74 => {........mat
3a30: 63 68 20 26 77 6f 72 64 73 2e 6e 65 78 74 28 29 ch &words.next()
3a40: 2e 75 6e 77 72 61 70 28 29 2e 70 61 72 73 65 3a .unwrap().parse:
3a50: 3a 3c 69 33 32 3e 28 29 20 7b 0a 09 09 09 09 09 :<i32>() {......
3a60: 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 7b ...Err(err) => {
3a70: 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e ..........reply.
3a80: 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 49 20 push(format!("I
3a90: 6e 65 65 64 20 61 20 6e 75 6d 62 65 72 5c 5c 2e need a number\\.
3aa0: 5c 6e 7b 7d 22 2c 20 26 65 72 72 29 29 3b 0a 09 \n{}", &err));..
3ab0: 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 .......},.......
3ac0: 09 09 4f 6b 28 6e 75 6d 62 65 72 29 20 3d 3e 20 ..Ok(number) =>
3ad0: 7b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 72 {..........let r
3ae0: 65 73 75 6c 74 20 3d 20 63 6f 72 65 2e 64 65 6c esult = core.del
3af0: 65 74 65 28 26 6e 75 6d 62 65 72 2c 20 6d 65 73 ete(&number, mes
3b00: 73 61 67 65 2e 66 72 6f 6d 2e 69 64 29 2e 61 77 sage.from.id).aw
3b10: 61 69 74 3f 3b 0a 09 09 09 09 09 09 09 09 09 72 ait?;..........r
3b20: 65 70 6c 79 2e 70 75 73 68 28 72 65 73 75 6c 74 eply.push(result
3b30: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 .to_string());..
3b40: 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 .......},.......
3b50: 09 7d 3b 0a 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f .};.......},..//
3b60: 20 64 69 73 61 62 6c 65 0a 0a 09 09 09 09 09 09 disable........
3b70: 22 2f 64 69 73 61 62 6c 65 22 20 3d 3e 20 7b 0a "/disable" => {.
3b80: 09 09 09 09 09 09 09 6d 61 74 63 68 20 26 77 6f .......match &wo
3b90: 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 rds.next().unwra
3ba0: 70 28 29 2e 70 61 72 73 65 3a 3a 3c 69 33 32 3e p().parse::<i32>
3bb0: 28 29 20 7b 0a 09 09 09 09 09 09 09 09 45 72 72 () {.........Err
3bc0: 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 (err) => {......
3bd0: 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 ....reply.push(f
3be0: 6f 72 6d 61 74 21 28 22 49 20 6e 65 65 64 20 61 ormat!("I need a
3bf0: 20 6e 75 6d 62 65 72 5c 5c 2e 5c 6e 7b 7d 22 2c number\\.\n{}",
3c00: 20 26 65 72 72 29 29 3b 0a 09 09 09 09 09 09 09 &err));........
3c10: 09 7d 2c 0a 09 09 09 09 09 09 09 09 4f 6b 28 6e .},.........Ok(n
3c20: 75 6d 62 65 72 29 20 3d 3e 20 7b 0a 09 09 09 09 umber) => {.....
3c30: 09 09 09 09 09 6c 65 74 20 72 65 73 75 6c 74 20 .....let result
3c40: 3d 20 63 6f 72 65 2e 64 69 73 61 62 6c 65 28 26 = core.disable(&
3c50: 6e 75 6d 62 65 72 2c 20 6d 65 73 73 61 67 65 2e number, message.
3c60: 66 72 6f 6d 2e 69 64 29 2e 61 77 61 69 74 3f 3b from.id).await?;
3c70: 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e ..........reply.
3c80: 70 75 73 68 28 72 65 73 75 6c 74 2e 74 6f 5f 73 push(result.to_s
3c90: 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 tring());.......
3ca0: 09 09 7d 2c 0a 09 09 09 09 09 09 09 7d 3b 0a 09 ..},........};..
3cb0: 09 09 09 09 09 7d 2c 0a 0a 09 09 09 09 09 09 5f .....},........_
3cc0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 7d 2c 0a 09 => {.......},..
3cd0: 09 09 09 09 7d 3b 0a 09 09 09 09 7d 2c 0a 09 09 ....};.....},...
3ce0: 09 09 5f 20 3d 3e 20 7b 0a 09 09 09 09 7d 2c 0a .._ => {.....},.
3cf0: 09 09 09 7d 3b 0a 0a 09 09 09 69 66 20 72 65 70 ...};.....if rep
3d00: 6c 79 2e 6c 65 6e 28 29 20 3e 20 30 20 7b 0a 09 ly.len() > 0 {..
3d10: 09 09 09 69 66 20 6c 65 74 20 45 72 72 28 65 72 ...if let Err(er
3d20: 72 29 20 3d 20 63 6f 72 65 2e 74 67 2e 73 65 6e r) = core.tg.sen
3d30: 64 28 6d 65 73 73 61 67 65 2e 74 65 78 74 5f 72 d(message.text_r
3d40: 65 70 6c 79 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 eply(reply.join(
3d50: 22 5c 6e 22 29 29 2e 70 61 72 73 65 5f 6d 6f 64 "\n")).parse_mod
3d60: 65 28 74 79 70 65 73 3a 3a 50 61 72 73 65 4d 6f e(types::ParseMo
3d70: 64 65 3a 3a 4d 61 72 6b 64 6f 77 6e 56 32 29 29 de::MarkdownV2))
3d80: 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 64 62 .await {......db
3d90: 67 21 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c g!(reply.join("\
3da0: 6e 22 29 29 3b 0a 09 09 09 09 09 70 72 69 6e 74 n"));......print
3db0: 6c 6e 21 28 22 7b 7d 22 2c 20 65 72 72 29 3b 0a ln!("{}", err);.
3dc0: 09 09 09 09 7d 3b 0a 09 09 09 7d 3b 0a 09 09 7d ....};....};...}
3dd0: 2c 0a 09 09 5f 20 3d 3e 20 7b 7d 2c 0a 09 7d 3b ,..._ => {},..};
3de0: 0a 0a 09 4f 6b 28 28 29 29 0a 7d 0a ...Ok(()).}.