0000: 75 73 65 20 63 6f 6e 66 69 67 3b 0a 0a 75 73 65 use config;..use
0010: 20 74 6f 6b 69 6f 3b 0a 75 73 65 20 72 73 73 3b tokio;.use rss;
0020: 0a 75 73 65 20 63 68 72 6f 6e 6f 3a 3a 44 61 74 .use chrono::Dat
0030: 65 54 69 6d 65 3b 0a 0a 75 73 65 20 72 65 67 65 eTime;..use rege
0040: 78 3a 3a 52 65 67 65 78 3b 0a 0a 75 73 65 20 74 x::Regex;..use t
0050: 6f 6b 69 6f 3a 3a 73 74 72 65 61 6d 3a 3a 53 74 okio::stream::St
0060: 72 65 61 6d 45 78 74 3b 0a 75 73 65 20 74 65 6c reamExt;.use tel
0070: 65 67 72 61 6d 5f 62 6f 74 3a 3a 2a 3b 0a 0a 75 egram_bot::*;..u
0080: 73 65 20 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 se sqlx::postgre
0090: 73 3a 3a 50 67 50 6f 6f 6c 4f 70 74 69 6f 6e 73 s::PgPoolOptions
00a0: 3b 0a 75 73 65 20 73 71 6c 78 3a 3a 52 6f 77 3b ;.use sqlx::Row;
00b0: 0a 0a 74 79 70 65 20 52 65 73 75 6c 74 3c 54 3e ..type Result<T>
00c0: 20 3d 20 73 74 64 3a 3a 72 65 73 75 6c 74 3a 3a = std::result::
00d0: 52 65 73 75 6c 74 3c 54 2c 20 42 6f 78 3c 64 79 Result<T, Box<dy
00e0: 6e 20 73 74 64 3a 3a 65 72 72 6f 72 3a 3a 45 72 n std::error::Er
00f0: 72 6f 72 3e 3e 3b 0a 0a 23 5b 64 65 72 69 76 65 ror>>;..#[derive
0100: 28 43 6c 6f 6e 65 29 5d 0a 73 74 72 75 63 74 20 (Clone)].struct
0110: 43 6f 72 65 20 7b 0a 09 6f 77 6e 65 72 3a 20 69 Core {..owner: i
0120: 36 34 2c 0a 09 61 70 69 5f 6b 65 79 3a 20 53 74 64,..api_key: St
0130: 72 69 6e 67 2c 0a 09 6f 77 6e 65 72 5f 63 68 61 ring,..owner_cha
0140: 74 3a 20 55 73 65 72 49 64 2c 0a 09 74 67 3a 20 t: UserId,..tg:
0150: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 41 70 telegram_bot::Ap
0160: 69 2c 0a 09 6d 79 3a 20 55 73 65 72 2c 0a 09 70 i,..my: User,..p
0170: 6f 6f 6c 3a 20 73 71 6c 78 3a 3a 50 6f 6f 6c 3c ool: sqlx::Pool<
0180: 73 71 6c 78 3a 3a 50 6f 73 74 67 72 65 73 3e 2c sqlx::Postgres>,
0190: 0a 7d 0a 0a 69 6d 70 6c 20 43 6f 72 65 20 7b 0a .}..impl Core {.
01a0: 09 61 73 79 6e 63 20 66 6e 20 6e 65 77 28 73 65 .async fn new(se
01b0: 74 74 69 6e 67 73 3a 20 63 6f 6e 66 69 67 3a 3a ttings: config::
01c0: 43 6f 6e 66 69 67 29 20 2d 3e 20 52 65 73 75 6c Config) -> Resul
01d0: 74 3c 43 6f 72 65 3e 20 7b 0a 09 09 6c 65 74 20 t<Core> {...let
01e0: 6f 77 6e 65 72 20 3d 20 73 65 74 74 69 6e 67 73 owner = settings
01f0: 2e 67 65 74 5f 69 6e 74 28 22 6f 77 6e 65 72 22 .get_int("owner"
0200: 29 3f 3b 0a 09 09 6c 65 74 20 61 70 69 5f 6b 65 )?;...let api_ke
0210: 79 20 3d 20 73 65 74 74 69 6e 67 73 2e 67 65 74 y = settings.get
0220: 5f 73 74 72 28 22 61 70 69 5f 6b 65 79 22 29 3f _str("api_key")?
0230: 3b 0a 09 09 6c 65 74 20 74 67 20 3d 20 41 70 69 ;...let tg = Api
0240: 3a 3a 6e 65 77 28 26 61 70 69 5f 6b 65 79 29 3b ::new(&api_key);
0250: 0a 09 09 6c 65 74 20 63 6f 72 65 20 3d 20 43 6f ...let core = Co
0260: 72 65 20 7b 0a 09 09 09 6f 77 6e 65 72 3a 20 6f re {....owner: o
0270: 77 6e 65 72 2c 0a 09 09 09 61 70 69 5f 6b 65 79 wner,....api_key
0280: 3a 20 61 70 69 5f 6b 65 79 2e 63 6c 6f 6e 65 28 : api_key.clone(
0290: 29 2c 0a 09 09 09 6d 79 3a 20 74 67 2e 73 65 6e ),....my: tg.sen
02a0: 64 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a d(telegram_bot::
02b0: 47 65 74 4d 65 29 2e 61 77 61 69 74 3f 2c 0a 09 GetMe).await?,..
02c0: 09 09 74 67 3a 20 74 67 2c 0a 09 09 09 6f 77 6e ..tg: tg,....own
02d0: 65 72 5f 63 68 61 74 3a 20 55 73 65 72 49 64 3a er_chat: UserId:
02e0: 3a 6e 65 77 28 6f 77 6e 65 72 29 2c 0a 09 09 09 :new(owner),....
02f0: 70 6f 6f 6c 3a 20 50 67 50 6f 6f 6c 4f 70 74 69 pool: PgPoolOpti
0300: 6f 6e 73 3a 3a 6e 65 77 28 29 2e 6d 61 78 5f 63 ons::new().max_c
0310: 6f 6e 6e 65 63 74 69 6f 6e 73 28 35 29 2e 63 6f onnections(5).co
0320: 6e 6e 65 63 74 28 26 73 65 74 74 69 6e 67 73 2e nnect(&settings.
0330: 67 65 74 5f 73 74 72 28 22 70 67 22 29 3f 29 2e get_str("pg")?).
0340: 61 77 61 69 74 3f 2c 0a 09 09 7d 3b 0a 09 09 6c await?,...};...l
0350: 65 74 20 63 6c 6f 6e 65 20 3d 20 63 6f 72 65 2e et clone = core.
0360: 63 6c 6f 6e 65 28 29 3b 0a 09 09 74 6f 6b 69 6f clone();...tokio
0370: 3a 3a 73 70 61 77 6e 28 61 73 79 6e 63 20 6d 6f ::spawn(async mo
0380: 76 65 20 7b 0a 09 09 09 69 66 20 6c 65 74 20 45 ve {....if let E
0390: 72 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e rr(err) = clone.
03a0: 61 75 74 6f 66 65 74 63 68 28 29 2e 61 77 61 69 autofetch().awai
03b0: 74 20 7b 0a 09 09 09 09 65 70 72 69 6e 74 6c 6e t {.....eprintln
03c0: 21 28 22 63 6f 6e 6e 65 63 74 69 6f 6e 20 65 72 !("connection er
03d0: 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 29 3b 0a ror: {}", err);.
03e0: 09 09 09 7d 0a 09 09 7d 29 3b 0a 09 09 4f 6b 28 ...}...});...Ok(
03f0: 63 6f 72 65 29 0a 09 7d 0a 0a 09 66 6e 20 73 74 core)..}...fn st
0400: 72 65 61 6d 28 26 73 65 6c 66 29 20 2d 3e 20 74 ream(&self) -> t
0410: 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 55 70 64 elegram_bot::Upd
0420: 61 74 65 73 53 74 72 65 61 6d 20 7b 0a 09 09 73 atesStream {...s
0430: 65 6c 66 2e 74 67 2e 73 74 72 65 61 6d 28 29 0a elf.tg.stream().
0440: 09 7d 0a 0a 09 66 6e 20 64 65 62 75 67 28 26 73 .}...fn debug(&s
0450: 65 6c 66 2c 20 6d 73 67 3a 20 26 73 74 72 29 20 elf, msg: &str)
0460: 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 20 7b 0a -> Result<()> {.
0470: 09 09 73 65 6c 66 2e 74 67 2e 73 70 61 77 6e 28 ..self.tg.spawn(
0480: 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 SendMessage::new
0490: 28 73 65 6c 66 2e 6f 77 6e 65 72 5f 63 68 61 74 (self.owner_chat
04a0: 2c 20 6d 73 67 29 29 3b 0a 09 09 4f 6b 28 28 29 , msg));...Ok(()
04b0: 29 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 66 6e 20 )..}...async fn
04c0: 63 68 65 63 6b 28 26 73 65 6c 66 2c 20 63 68 61 check(&self, cha
04d0: 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 72 65 61 6c nnel: &str, real
04e0: 3a 20 4f 70 74 69 6f 6e 3c 62 6f 6f 6c 3e 29 20 : Option<bool>)
04f0: 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 20 7b 0a -> Result<()> {.
0500: 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 ..match sqlx::qu
0510: 65 72 79 28 22 73 65 6c 65 63 74 20 73 6f 75 72 ery("select sour
0520: 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 5f 69 ce_id, channel_i
0530: 64 2c 20 75 72 6c 2c 20 6c 61 73 74 5f 66 65 74 d, url, last_fet
0540: 63 68 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e ch, iv_hash, own
0550: 65 72 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f er from rsstg_so
0560: 75 72 63 65 20 6e 61 74 75 72 61 6c 20 6c 65 66 urce natural lef
0570: 74 20 6a 6f 69 6e 20 72 73 73 74 67 5f 63 68 61 t join rsstg_cha
0580: 6e 6e 65 6c 20 77 68 65 72 65 20 75 73 65 72 6e nnel where usern
0590: 61 6d 65 20 3d 20 24 31 22 29 0a 09 09 09 2e 62 ame = $1").....b
05a0: 69 6e 64 28 63 68 61 6e 6e 65 6c 29 0a 09 09 09 ind(channel)....
05b0: 2e 66 65 74 63 68 5f 6f 6e 65 28 26 73 65 6c 66 .fetch_one(&self
05c0: 2e 70 6f 6f 6c 29 2e 61 77 61 69 74 20 7b 0a 09 .pool).await {..
05d0: 09 09 4f 6b 28 72 6f 77 29 20 3d 3e 20 7b 0a 09 ..Ok(row) => {..
05e0: 09 09 09 6c 65 74 20 69 64 3a 20 69 33 32 20 3d ...let id: i32 =
05f0: 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 73 6f row.try_get("so
0600: 75 72 63 65 5f 69 64 22 29 3f 3b 0a 09 09 09 09 urce_id")?;.....
0610: 6c 65 74 20 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 let channel_id:
0620: 69 36 34 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 i64 = row.try_ge
0630: 74 28 22 63 68 61 6e 6e 65 6c 5f 69 64 22 29 3f t("channel_id")?
0640: 3b 0a 09 09 09 09 6c 65 74 20 64 65 73 74 69 6e ;.....let destin
0650: 61 74 69 6f 6e 20 3d 20 6d 61 74 63 68 20 72 65 ation = match re
0660: 61 6c 20 7b 0a 09 09 09 09 09 53 6f 6d 65 28 74 al {......Some(t
0670: 72 75 65 29 20 3d 3e 20 55 73 65 72 49 64 3a 3a rue) => UserId::
0680: 6e 65 77 28 63 68 61 6e 6e 65 6c 5f 69 64 29 2c new(channel_id),
0690: 0a 09 09 09 09 09 53 6f 6d 65 28 66 61 6c 73 65 ......Some(false
06a0: 29 20 7c 20 4e 6f 6e 65 20 3d 3e 20 55 73 65 72 ) | None => User
06b0: 49 64 3a 3a 6e 65 77 28 72 6f 77 2e 74 72 79 5f Id::new(row.try_
06c0: 67 65 74 28 22 6f 77 6e 65 72 22 29 3f 29 2c 0a get("owner")?),.
06d0: 09 09 09 09 7d 3b 0a 09 09 09 09 6c 65 74 20 75 ....};.....let u
06e0: 72 6c 3a 20 26 73 74 72 20 3d 20 72 6f 77 2e 74 rl: &str = row.t
06f0: 72 79 5f 67 65 74 28 22 75 72 6c 22 29 3f 3b 0a ry_get("url")?;.
0700: 09 09 09 09 6c 65 74 20 6c 61 73 74 5f 66 65 74 ....let last_fet
0710: 63 68 3a 20 4f 70 74 69 6f 6e 3c 44 61 74 65 54 ch: Option<DateT
0720: 69 6d 65 3c 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 ime<chrono::Fixe
0730: 64 4f 66 66 73 65 74 3e 3e 20 3d 20 72 6f 77 2e dOffset>> = row.
0740: 74 72 79 5f 67 65 74 28 22 6c 61 73 74 5f 66 65 try_get("last_fe
0750: 74 63 68 22 29 3f 3b 0a 09 09 09 09 6c 65 74 20 tch")?;.....let
0760: 6d 75 74 20 74 68 69 73 5f 66 65 74 63 68 3a 20 mut this_fetch:
0770: 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 6d 65 3c Option<DateTime<
0780: 63 68 72 6f 6e 6f 3a 3a 46 69 78 65 64 4f 66 66 chrono::FixedOff
0790: 73 65 74 3e 3e 20 3d 20 4e 6f 6e 65 3b 0a 09 09 set>> = None;...
07a0: 09 09 6c 65 74 20 69 76 5f 68 61 73 68 3a 20 4f ..let iv_hash: O
07b0: 70 74 69 6f 6e 3c 26 73 74 72 3e 20 3d 20 72 6f ption<&str> = ro
07c0: 77 2e 74 72 79 5f 67 65 74 28 22 69 76 5f 68 61 w.try_get("iv_ha
07d0: 73 68 22 29 3f 3b 0a 09 09 09 09 6d 61 74 63 68 sh")?;.....match
07e0: 20 72 73 73 3a 3a 43 68 61 6e 6e 65 6c 3a 3a 66 rss::Channel::f
07f0: 72 6f 6d 5f 75 72 6c 28 75 72 6c 29 20 7b 0a 09 rom_url(url) {..
0800: 09 09 09 09 4f 6b 28 66 65 65 64 29 20 3d 3e 20 ....Ok(feed) =>
0810: 7b 0a 09 09 09 09 09 09 73 65 6c 66 2e 64 65 62 {.......self.deb
0820: 75 67 28 26 66 6f 72 6d 61 74 21 28 22 23 20 74 ug(&format!("# t
0830: 69 74 6c 65 3a 7b 3a 3f 7d 20 74 74 6c 3a 7b 3a itle:{:?} ttl:{:
0840: 3f 7d 20 68 6f 75 72 73 3a 7b 3a 3f 7d 20 64 61 ?} hours:{:?} da
0850: 79 73 3a 7b 3a 3f 7d 22 2c 20 66 65 65 64 2e 74 ys:{:?}", feed.t
0860: 69 74 6c 65 28 29 2c 20 66 65 65 64 2e 74 74 6c itle(), feed.ttl
0870: 28 29 2c 20 66 65 65 64 2e 73 6b 69 70 5f 68 6f (), feed.skip_ho
0880: 75 72 73 28 29 2c 20 66 65 65 64 2e 73 6b 69 70 urs(), feed.skip
0890: 5f 64 61 79 73 28 29 29 29 3f 3b 0a 09 09 09 09 _days()))?;.....
08a0: 09 09 66 6f 72 20 69 74 65 6d 20 69 6e 20 66 65 ..for item in fe
08b0: 65 64 2e 69 74 65 6d 73 28 29 20 7b 0a 09 09 09 ed.items() {....
08c0: 09 09 09 09 6c 65 74 20 64 61 74 65 20 3d 20 6d ....let date = m
08d0: 61 74 63 68 20 69 74 65 6d 2e 70 75 62 5f 64 61 atch item.pub_da
08e0: 74 65 28 29 20 7b 0a 09 09 09 09 09 09 09 09 53 te() {.........S
08f0: 6f 6d 65 28 66 65 65 64 5f 64 61 74 65 29 20 3d ome(feed_date) =
0900: 3e 20 44 61 74 65 54 69 6d 65 3a 3a 70 61 72 73 > DateTime::pars
0910: 65 5f 66 72 6f 6d 5f 72 66 63 32 38 32 32 28 66 e_from_rfc2822(f
0920: 65 65 64 5f 64 61 74 65 29 2c 0a 09 09 09 09 09 eed_date),......
0930: 09 09 09 4e 6f 6e 65 20 3d 3e 20 44 61 74 65 54 ...None => DateT
0940: 69 6d 65 3a 3a 70 61 72 73 65 5f 66 72 6f 6d 5f ime::parse_from_
0950: 72 66 63 33 33 33 39 28 26 69 74 65 6d 2e 64 75 rfc3339(&item.du
0960: 62 6c 69 6e 5f 63 6f 72 65 5f 65 78 74 28 29 2e blin_core_ext().
0970: 75 6e 77 72 61 70 28 29 2e 64 61 74 65 73 28 29 unwrap().dates()
0980: 5b 30 5d 29 2c 0a 09 09 09 09 09 09 09 7d 3f 3b [0]),........}?;
0990: 0a 09 09 09 09 09 09 09 6c 65 74 20 75 72 6c 20 ........let url
09a0: 3d 20 69 74 65 6d 2e 6c 69 6e 6b 28 29 2e 75 6e = item.link().un
09b0: 77 72 61 70 28 29 2e 74 6f 5f 73 74 72 69 6e 67 wrap().to_string
09c0: 28 29 3b 0a 09 09 09 09 09 09 09 69 66 20 6c 61 ();........if la
09d0: 73 74 5f 66 65 74 63 68 20 3d 3d 20 4e 6f 6e 65 st_fetch == None
09e0: 20 7c 7c 20 64 61 74 65 20 3e 20 6c 61 73 74 5f || date > last_
09f0: 66 65 74 63 68 2e 75 6e 77 72 61 70 28 29 20 7b fetch.unwrap() {
0a00: 0a 09 09 09 09 09 09 09 09 6d 61 74 63 68 20 73 .........match s
0a10: 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 qlx::query("sele
0a20: 63 74 20 65 78 69 73 74 73 28 73 65 6c 65 63 74 ct exists(select
0a30: 20 74 72 75 65 20 66 72 6f 6d 20 72 73 73 74 67 true from rsstg
0a40: 5f 70 6f 73 74 20 77 68 65 72 65 20 75 72 6c 20 _post where url
0a50: 3d 20 24 31 20 61 6e 64 20 73 6f 75 72 63 65 5f = $1 and source_
0a60: 69 64 20 3d 20 24 32 29 20 61 73 20 65 78 69 73 id = $2) as exis
0a70: 74 73 3b 22 29 0a 09 09 09 09 09 09 09 09 09 2e ts;")...........
0a80: 62 69 6e 64 28 26 75 72 6c 29 0a 09 09 09 09 09 bind(&url)......
0a90: 09 09 09 09 2e 62 69 6e 64 28 69 64 29 0a 09 09 .....bind(id)...
0aa0: 09 09 09 09 09 09 09 2e 66 65 74 63 68 5f 6f 6e ........fetch_on
0ab0: 65 28 26 73 65 6c 66 2e 70 6f 6f 6c 29 2e 61 77 e(&self.pool).aw
0ac0: 61 69 74 20 7b 0a 09 09 09 09 09 09 09 09 09 4f ait {..........O
0ad0: 6b 28 72 6f 77 29 20 3d 3e 20 7b 0a 09 09 09 09 k(row) => {.....
0ae0: 09 09 09 09 09 09 6c 65 74 20 65 78 69 73 74 73 ......let exists
0af0: 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 72 79 : bool = row.try
0b00: 5f 67 65 74 28 22 65 78 69 73 74 73 22 29 3f 3b _get("exists")?;
0b10: 0a 09 09 09 09 09 09 09 09 09 09 69 66 20 21 20 ...........if !
0b20: 65 78 69 73 74 73 20 7b 0a 09 09 09 09 09 09 09 exists {........
0b30: 09 09 09 09 69 66 20 74 68 69 73 5f 66 65 74 63 ....if this_fetc
0b40: 68 20 3d 3d 20 4e 6f 6e 65 20 7c 7c 20 64 61 74 h == None || dat
0b50: 65 20 3e 20 74 68 69 73 5f 66 65 74 63 68 2e 75 e > this_fetch.u
0b60: 6e 77 72 61 70 28 29 20 7b 0a 09 09 09 09 09 09 nwrap() {.......
0b70: 09 09 09 09 09 09 74 68 69 73 5f 66 65 74 63 68 ......this_fetch
0b80: 20 3d 20 53 6f 6d 65 28 64 61 74 65 29 3b 0a 09 = Some(date);..
0b90: 09 09 09 09 09 09 09 09 09 09 7d 0a 09 09 09 09 ..........}.....
0ba0: 09 09 09 09 09 09 09 6d 61 74 63 68 20 73 65 6c .......match sel
0bb0: 66 2e 74 67 2e 73 65 6e 64 28 20 6d 61 74 63 68 f.tg.send( match
0bc0: 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 iv_hash {......
0bd0: 09 09 09 09 09 09 09 09 53 6f 6d 65 28 78 29 20 ........Some(x)
0be0: 3d 3e 20 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a => SendMessage::
0bf0: 6e 65 77 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c new(destination,
0c00: 20 66 6f 72 6d 61 74 21 28 22 3c 61 20 68 72 65 format!("<a hre
0c10: 66 3d 5c 22 68 74 74 70 73 3a 2f 2f 74 2e 6d 65 f=\"https://t.me
0c20: 2f 69 76 3f 75 72 6c 3d 7b 7d 26 72 68 61 73 68 /iv?url={}&rhash
0c30: 3d 7b 7d 5c 22 3e 20 3c 2f 61 3e 7b 30 7d 22 2c ={}\"> </a>{0}",
0c40: 20 75 72 6c 2c 20 78 29 29 2c 0a 09 09 09 09 09 url, x)),......
0c50: 09 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 ........None =>
0c60: 53 65 6e 64 4d 65 73 73 61 67 65 3a 3a 6e 65 77 SendMessage::new
0c70: 28 64 65 73 74 69 6e 61 74 69 6f 6e 2c 20 66 6f (destination, fo
0c80: 72 6d 61 74 21 28 22 7b 7d 22 2c 20 75 72 6c 29 rmat!("{}", url)
0c90: 29 2c 0a 09 09 09 09 09 09 09 09 09 09 09 09 7d ),.............}
0ca0: 2e 70 61 72 73 65 5f 6d 6f 64 65 28 74 79 70 65 .parse_mode(type
0cb0: 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a 48 74 s::ParseMode::Ht
0cc0: 6d 6c 29 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 ml)).await {....
0cd0: 09 09 09 09 09 09 09 09 09 4f 6b 28 5f 29 20 3d .........Ok(_) =
0ce0: 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 09 > {.............
0cf0: 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 .match sqlx::que
0d00: 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 ry("insert into
0d10: 72 73 73 74 67 5f 70 6f 73 74 20 28 73 6f 75 72 rsstg_post (sour
0d20: 63 65 5f 69 64 2c 20 70 6f 73 74 65 64 2c 20 75 ce_id, posted, u
0d30: 72 6c 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 rl) values ($1,
0d40: 24 32 2c 20 24 33 29 3b 22 29 0a 09 09 09 09 09 $2, $3);")......
0d50: 09 09 09 09 09 09 09 09 09 2e 62 69 6e 64 28 69 ..........bind(i
0d60: 64 29 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 d)..............
0d70: 09 2e 62 69 6e 64 28 64 61 74 65 29 0a 09 09 09 ..bind(date)....
0d80: 09 09 09 09 09 09 09 09 09 09 09 2e 62 69 6e 64 ............bind
0d90: 28 75 72 6c 29 0a 09 09 09 09 09 09 09 09 09 09 (url)...........
0da0: 09 09 09 09 2e 65 78 65 63 75 74 65 28 26 73 65 .....execute(&se
0db0: 6c 66 2e 70 6f 6f 6c 29 2e 61 77 61 69 74 20 7b lf.pool).await {
0dc0: 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 09 09 ................
0dd0: 4f 6b 28 5f 29 20 3d 3e 20 7b 7d 2c 0a 09 09 09 Ok(_) => {},....
0de0: 09 09 09 09 09 09 09 09 09 09 09 09 45 72 72 28 ............Err(
0df0: 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 err) => {.......
0e00: 09 09 09 09 09 09 09 09 09 09 73 65 6c 66 2e 64 ..........self.d
0e10: 65 62 75 67 28 26 65 72 72 2e 74 6f 5f 73 74 72 ebug(&err.to_str
0e20: 69 6e 67 28 29 29 3f 3b 0a 09 09 09 09 09 09 09 ing())?;........
0e30: 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ........},......
0e40: 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ........};......
0e50: 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 .......},.......
0e60: 09 09 09 09 09 09 45 72 72 28 65 72 72 29 20 3d ......Err(err) =
0e70: 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 09 > {.............
0e80: 09 73 65 6c 66 2e 64 65 62 75 67 28 26 65 72 72 .self.debug(&err
0e90: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3f 3b 0a .to_string())?;.
0ea0: 09 09 09 09 09 09 09 09 09 09 09 09 7d 2c 0a 09 ............},..
0eb0: 09 09 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 ..........};....
0ec0: 09 09 09 09 09 09 09 09 74 6f 6b 69 6f 3a 3a 74 ........tokio::t
0ed0: 69 6d 65 3a 3a 64 65 6c 61 79 5f 66 6f 72 28 73 ime::delay_for(s
0ee0: 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 td::time::Durati
0ef0: 6f 6e 3a 3a 6e 65 77 28 34 2c 20 30 29 29 2e 61 on::new(4, 0)).a
0f00: 77 61 69 74 3b 0a 09 09 09 09 09 09 09 09 09 09 wait;...........
0f10: 7d 0a 09 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 }..........},...
0f20: 09 09 09 09 09 09 09 45 72 72 28 65 72 72 29 20 .......Err(err)
0f30: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 09 73 => {...........s
0f40: 65 6c 66 2e 64 65 62 75 67 28 26 65 72 72 2e 74 elf.debug(&err.t
0f50: 6f 5f 73 74 72 69 6e 67 28 29 29 3f 3b 0a 09 09 o_string())?;...
0f60: 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 .......},.......
0f70: 09 09 7d 3b 0a 09 09 09 09 09 09 09 7d 3b 0a 09 ..};........};..
0f80: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 2f 2f .....};.......//
0f90: 20 75 70 64 61 74 65 20 6c 61 73 74 5f 66 65 74 update last_fet
0fa0: 63 68 0a 09 09 09 09 09 09 69 66 20 74 68 69 73 ch.......if this
0fb0: 5f 66 65 74 63 68 20 21 3d 20 4e 6f 6e 65 20 26 _fetch != None &
0fc0: 26 20 28 6c 61 73 74 5f 66 65 74 63 68 20 3d 3d & (last_fetch ==
0fd0: 20 4e 6f 6e 65 20 7c 7c 20 74 68 69 73 5f 66 65 None || this_fe
0fe0: 74 63 68 2e 75 6e 77 72 61 70 28 29 20 3e 20 6c tch.unwrap() > l
0ff0: 61 73 74 5f 66 65 74 63 68 2e 75 6e 77 72 61 70 ast_fetch.unwrap
1000: 28 29 29 20 7b 0a 09 09 09 09 09 09 09 6d 61 74 ()) {........mat
1010: 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ch sqlx::query("
1020: 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 update rsstg_sou
1030: 72 63 65 20 73 65 74 20 6c 61 73 74 5f 66 65 74 rce set last_fet
1040: 63 68 20 3d 20 63 61 73 65 20 77 68 65 6e 20 28 ch = case when (
1050: 6c 61 73 74 5f 66 65 74 63 68 20 3c 20 24 31 29 last_fetch < $1)
1060: 20 74 68 65 6e 20 24 31 20 65 6c 73 65 20 6c 61 then $1 else la
1070: 73 74 5f 66 65 74 63 68 20 65 6e 64 20 77 68 65 st_fetch end whe
1080: 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 re source_id = $
1090: 32 3b 22 29 0a 09 09 09 09 09 09 09 09 2e 62 69 2;")..........bi
10a0: 6e 64 28 74 68 69 73 5f 66 65 74 63 68 2e 75 6e nd(this_fetch.un
10b0: 77 72 61 70 28 29 29 0a 09 09 09 09 09 09 09 09 wrap()).........
10c0: 2e 62 69 6e 64 28 69 64 29 0a 09 09 09 09 09 09 .bind(id).......
10d0: 09 09 2e 65 78 65 63 75 74 65 28 26 73 65 6c 66 ...execute(&self
10e0: 2e 70 6f 6f 6c 29 2e 61 77 61 69 74 20 7b 0a 09 .pool).await {..
10f0: 09 09 09 09 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 .......Ok(_) =>
1100: 7b 7d 2c 0a 09 09 09 09 09 09 09 09 45 72 72 28 {},.........Err(
1110: 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 err) => {.......
1120: 09 09 09 73 65 6c 66 2e 64 65 62 75 67 28 26 65 ...self.debug(&e
1130: 72 72 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3f rr.to_string())?
1140: 3b 0a 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 ;.........},....
1150: 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 7d 0a 09 ....};.......}..
1160: 09 09 09 09 7d 2c 0a 09 09 09 09 09 45 72 72 28 ....},......Err(
1170: 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 err) => {.......
1180: 73 65 6c 66 2e 64 65 62 75 67 28 26 65 72 72 2e self.debug(&err.
1190: 74 6f 5f 73 74 72 69 6e 67 28 29 29 3f 3b 0a 09 to_string())?;..
11a0: 09 09 09 09 7d 2c 0a 09 09 09 09 7d 3b 0a 09 09 ....},.....};...
11b0: 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 ..match sqlx::qu
11c0: 65 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 ery("update rsst
11d0: 67 5f 73 6f 75 72 63 65 20 73 65 74 20 6c 61 73 g_source set las
11e0: 74 5f 73 63 72 61 70 65 20 3d 20 6e 6f 77 28 29 t_scrape = now()
11f0: 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 where source_id
1200: 20 3d 20 24 31 3b 22 29 0a 09 09 09 09 09 2e 62 = $1;").......b
1210: 69 6e 64 28 69 64 29 0a 09 09 09 09 09 2e 65 78 ind(id).......ex
1220: 65 63 75 74 65 28 26 73 65 6c 66 2e 70 6f 6f 6c ecute(&self.pool
1230: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 4f ).await {......O
1240: 6b 28 5f 29 20 3d 3e 20 7b 7d 2c 0a 09 09 09 09 k(_) => {},.....
1250: 09 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 .Err(err) => {..
1260: 09 09 09 09 09 73 65 6c 66 2e 64 65 62 75 67 28 .....self.debug(
1270: 26 65 72 72 2e 74 6f 5f 73 74 72 69 6e 67 28 29 &err.to_string()
1280: 29 3f 3b 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 )?;......},.....
1290: 7d 3b 0a 09 09 09 7d 2c 0a 09 09 09 45 72 72 28 };....},....Err(
12a0: 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 73 65 err) => {.....se
12b0: 6c 66 2e 64 65 62 75 67 28 26 65 72 72 2e 74 6f lf.debug(&err.to
12c0: 5f 73 74 72 69 6e 67 28 29 29 3f 3b 0a 09 09 09 _string())?;....
12d0: 7d 2c 0a 09 09 7d 3b 0a 09 09 4f 6b 28 28 29 29 },...};...Ok(())
12e0: 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 66 6e 20 63 ..}...async fn c
12f0: 6c 65 61 6e 28 26 73 65 6c 66 2c 20 73 6f 75 72 lean(&self, sour
1300: 63 65 5f 69 64 3a 20 69 33 32 29 20 2d 3e 20 52 ce_id: i32) -> R
1310: 65 73 75 6c 74 3c 28 29 3e 20 7b 0a 09 09 66 6f esult<()> {...fo
1320: 72 20 71 75 65 72 79 20 69 6e 20 76 65 63 21 5b r query in vec![
1330: 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 "delete from rss
1340: 74 67 5f 70 6f 73 74 20 77 68 65 72 65 20 73 6f tg_post where so
1350: 75 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 2c 20 urce_id = $1;",
1360: 22 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f "update rsstg_so
1370: 75 72 63 65 20 73 65 74 20 6c 61 73 74 5f 66 65 urce set last_fe
1380: 74 63 68 20 3d 20 4e 55 4c 4c 20 77 68 65 72 65 tch = NULL where
1390: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 3b source_id = $1;
13a0: 22 5d 20 7b 0a 09 09 09 6d 61 74 63 68 20 73 71 "] {....match sq
13b0: 6c 78 3a 3a 71 75 65 72 79 28 71 75 65 72 79 29 lx::query(query)
13c0: 0a 09 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 ......bind(sourc
13d0: 65 5f 69 64 29 0a 09 09 09 09 2e 65 78 65 63 75 e_id)......execu
13e0: 74 65 28 26 73 65 6c 66 2e 70 6f 6f 6c 29 2e 61 te(&self.pool).a
13f0: 77 61 69 74 20 7b 0a 09 09 09 09 4f 6b 28 5f 29 wait {.....Ok(_)
1400: 20 3d 3e 20 7b 7d 2c 0a 09 09 09 09 45 72 72 28 => {},.....Err(
1410: 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 73 err) => {......s
1420: 65 6c 66 2e 64 65 62 75 67 28 26 65 72 72 2e 74 elf.debug(&err.t
1430: 6f 5f 73 74 72 69 6e 67 28 29 29 3f 3b 0a 09 09 o_string())?;...
1440: 09 09 7d 2c 0a 09 09 09 7d 0a 09 09 7d 0a 09 09 ..},....}...}...
1450: 4f 6b 28 28 29 29 0a 09 7d 0a 0a 09 61 73 79 6e Ok(())..}...asyn
1460: 63 20 66 6e 20 65 6e 61 62 6c 65 28 26 73 65 6c c fn enable(&sel
1470: 66 2c 20 75 73 65 72 3a 20 55 73 65 72 49 64 2c f, user: UserId,
1480: 20 63 68 61 6e 6e 65 6c 3a 20 26 73 74 72 29 20 channel: &str)
1490: 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 20 7b 0a -> Result<()> {.
14a0: 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 ..match sqlx::qu
14b0: 65 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 ery("update rsst
14c0: 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 6e 61 g_source set ena
14d0: 62 6c 65 64 20 3d 20 74 72 75 65 20 66 72 6f 6d bled = true from
14e0: 20 72 73 73 74 67 5f 63 68 61 6e 6e 65 6c 20 77 rsstg_channel w
14f0: 68 65 72 65 20 72 73 73 74 67 5f 63 68 61 6e 6e here rsstg_chann
1500: 65 6c 2e 63 68 61 6e 6e 65 6c 5f 69 64 20 3d 20 el.channel_id =
1510: 72 73 73 74 67 5f 73 6f 75 72 63 65 2e 63 68 61 rsstg_source.cha
1520: 6e 6e 65 6c 5f 69 64 20 61 6e 64 20 72 73 73 74 nnel_id and rsst
1530: 67 5f 63 68 61 6e 6e 65 6c 2e 75 73 65 72 6e 61 g_channel.userna
1540: 6d 65 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 me = $1 and owne
1550: 72 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e r = $2").....bin
1560: 64 28 63 68 61 6e 6e 65 6c 29 0a 09 09 09 2e 62 d(channel).....b
1570: 69 6e 64 28 69 36 34 3a 3a 66 72 6f 6d 28 75 73 ind(i64::from(us
1580: 65 72 29 29 0a 09 09 09 2e 65 78 65 63 75 74 65 er)).....execute
1590: 28 26 73 65 6c 66 2e 70 6f 6f 6c 29 2e 61 77 61 (&self.pool).awa
15a0: 69 74 20 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e it {....Ok(_) =>
15b0: 20 7b 7d 2c 0a 09 09 09 45 72 72 28 65 72 72 29 {},....Err(err)
15c0: 20 3d 3e 20 7b 0a 09 09 09 09 73 65 6c 66 2e 64 => {.....self.d
15d0: 65 62 75 67 28 26 65 72 72 2e 74 6f 5f 73 74 72 ebug(&err.to_str
15e0: 69 6e 67 28 29 29 3f 3b 0a 09 09 09 7d 2c 0a 09 ing())?;....},..
15f0: 09 7d 0a 09 09 4f 6b 28 28 29 29 0a 09 7d 0a 0a .}...Ok(())..}..
1600: 09 61 73 79 6e 63 20 66 6e 20 64 69 73 61 62 6c .async fn disabl
1610: 65 28 26 73 65 6c 66 2c 20 75 73 65 72 3a 20 55 e(&self, user: U
1620: 73 65 72 49 64 2c 20 63 68 61 6e 6e 65 6c 3a 20 serId, channel:
1630: 26 73 74 72 29 20 2d 3e 20 52 65 73 75 6c 74 3c &str) -> Result<
1640: 28 29 3e 20 7b 0a 09 09 6d 61 74 63 68 20 73 71 ()> {...match sq
1650: 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 lx::query("updat
1660: 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 e rsstg_source s
1670: 65 74 20 65 6e 61 62 6c 65 64 20 3d 20 66 61 6c et enabled = fal
1680: 73 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 63 68 se from rsstg_ch
1690: 61 6e 6e 65 6c 20 77 68 65 72 65 20 72 73 73 74 annel where rsst
16a0: 67 5f 63 68 61 6e 6e 65 6c 2e 63 68 61 6e 6e 65 g_channel.channe
16b0: 6c 5f 69 64 20 3d 20 72 73 73 74 67 5f 73 6f 75 l_id = rsstg_sou
16c0: 72 63 65 2e 63 68 61 6e 6e 65 6c 5f 69 64 20 61 rce.channel_id a
16d0: 6e 64 20 72 73 73 74 67 5f 63 68 61 6e 6e 65 6c nd rsstg_channel
16e0: 2e 75 73 65 72 6e 61 6d 65 20 3d 20 24 31 20 61 .username = $1 a
16f0: 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a nd owner = $2").
1700: 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c ....bind(channel
1710: 29 0a 09 09 09 2e 62 69 6e 64 28 69 36 34 3a 3a ).....bind(i64::
1720: 66 72 6f 6d 28 75 73 65 72 29 29 0a 09 09 09 2e from(user)).....
1730: 65 78 65 63 75 74 65 28 26 73 65 6c 66 2e 70 6f execute(&self.po
1740: 6f 6c 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 4f ol).await {....O
1750: 6b 28 5f 29 20 3d 3e 20 7b 7d 2c 0a 09 09 09 45 k(_) => {},....E
1760: 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 rr(err) => {....
1770: 09 73 65 6c 66 2e 64 65 62 75 67 28 26 65 72 72 .self.debug(&err
1780: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3f 3b 0a .to_string())?;.
1790: 09 09 09 7d 2c 0a 09 09 7d 0a 09 09 4f 6b 28 28 ...},...}...Ok((
17a0: 29 29 0a 09 7d 0a 0a 09 61 73 79 6e 63 20 66 6e ))..}...async fn
17b0: 20 61 75 74 6f 66 65 74 63 68 28 26 73 65 6c 66 autofetch(&self
17c0: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 20 ) -> Result<()>
17d0: 7b 0a 09 09 6c 65 74 20 6d 75 74 20 64 65 6c 61 {...let mut dela
17e0: 79 20 3d 20 63 68 72 6f 6e 6f 3a 3a 44 75 72 61 y = chrono::Dura
17f0: 74 69 6f 6e 3a 3a 6d 69 6e 75 74 65 73 28 35 29 tion::minutes(5)
1800: 3b 0a 09 09 6c 65 74 20 6d 75 74 20 6e 65 78 74 ;...let mut next
1810: 5f 66 65 74 63 68 3a 20 44 61 74 65 54 69 6d 65 _fetch: DateTime
1820: 3c 63 68 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3e 3b <chrono::Local>;
1830: 0a 09 09 6c 65 74 20 6d 75 74 20 6e 6f 77 3b 0a ...let mut now;.
1840: 09 09 6c 6f 6f 70 20 7b 0a 09 09 09 6c 65 74 20 ..loop {....let
1850: 6d 75 74 20 72 6f 77 73 20 3d 20 73 71 6c 78 3a mut rows = sqlx:
1860: 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 73 :query("select s
1870: 6f 75 72 63 65 5f 69 64 2c 20 75 73 65 72 6e 61 ource_id, userna
1880: 6d 65 2c 20 6e 65 78 74 5f 66 65 74 63 68 20 66 me, next_fetch f
1890: 72 6f 6d 20 72 73 73 74 67 5f 6f 72 64 65 72 20 rom rsstg_order
18a0: 6e 61 74 75 72 61 6c 20 6c 65 66 74 20 6a 6f 69 natural left joi
18b0: 6e 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 6e n rsstg_source n
18c0: 61 74 75 72 61 6c 20 6c 65 66 74 20 6a 6f 69 6e atural left join
18d0: 20 72 73 73 74 67 5f 63 68 61 6e 6e 65 6c 3b 22 rsstg_channel;"
18e0: 29 0a 09 09 09 09 2e 66 65 74 63 68 28 26 73 65 )......fetch(&se
18f0: 6c 66 2e 70 6f 6f 6c 29 3b 0a 09 09 09 77 68 69 lf.pool);....whi
1900: 6c 65 20 6c 65 74 20 53 6f 6d 65 28 72 6f 77 29 le let Some(row)
1910: 20 3d 20 72 6f 77 73 2e 74 72 79 5f 6e 65 78 74 = rows.try_next
1920: 28 29 2e 61 77 61 69 74 2e 75 6e 77 72 61 70 28 ().await.unwrap(
1930: 29 20 7b 0a 09 09 09 09 6e 6f 77 20 3d 20 63 68 ) {.....now = ch
1940: 72 6f 6e 6f 3a 3a 4c 6f 63 61 6c 3a 3a 6e 6f 77 rono::Local::now
1950: 28 29 3b 0a 09 09 09 09 6c 65 74 20 73 6f 75 72 ();.....let sour
1960: 63 65 5f 69 64 3a 20 69 33 32 20 3d 20 72 6f 77 ce_id: i32 = row
1970: 2e 74 72 79 5f 67 65 74 28 22 73 6f 75 72 63 65 .try_get("source
1980: 5f 69 64 22 29 3f 3b 0a 09 09 09 09 6e 65 78 74 _id")?;.....next
1990: 5f 66 65 74 63 68 20 3d 20 72 6f 77 2e 74 72 79 _fetch = row.try
19a0: 5f 67 65 74 28 22 6e 65 78 74 5f 66 65 74 63 68 _get("next_fetch
19b0: 22 29 3f 3b 0a 09 09 09 09 69 66 20 6e 65 78 74 ")?;.....if next
19c0: 5f 66 65 74 63 68 20 3c 20 6e 6f 77 20 7b 0a 09 _fetch < now {..
19d0: 09 09 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a ....match sqlx::
19e0: 71 75 65 72 79 28 22 75 70 64 61 74 65 20 72 73 query("update rs
19f0: 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 20 6c stg_source set l
1a00: 61 73 74 5f 73 63 72 61 70 65 20 3d 20 6e 6f 77 ast_scrape = now
1a10: 28 29 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 () + interval '1
1a20: 20 68 6f 75 72 27 20 77 68 65 72 65 20 73 6f 75 hour' where sou
1a30: 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a 09 rce_id = $1;")..
1a40: 09 09 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 ......bind(sourc
1a50: 65 5f 69 64 29 0a 09 09 09 09 09 09 2e 65 78 65 e_id)........exe
1a60: 63 75 74 65 28 26 73 65 6c 66 2e 70 6f 6f 6c 29 cute(&self.pool)
1a70: 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 4f .await {.......O
1a80: 6b 28 5f 29 20 3d 3e 20 7b 7d 2c 0a 09 09 09 09 k(_) => {},.....
1a90: 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a ..Err(err) => {.
1aa0: 09 09 09 09 09 09 09 73 65 6c 66 2e 64 65 62 75 .......self.debu
1ab0: 67 28 26 65 72 72 2e 74 6f 5f 73 74 72 69 6e 67 g(&err.to_string
1ac0: 28 29 29 3f 3b 0a 09 09 09 09 09 09 7d 2c 0a 09 ())?;.......},..
1ad0: 09 09 09 09 7d 3b 0a 09 09 09 09 09 6c 65 74 20 ....};......let
1ae0: 63 6c 6f 6e 65 20 3d 20 73 65 6c 66 2e 63 6c 6f clone = self.clo
1af0: 6e 65 28 29 3b 0a 09 09 09 09 09 6c 65 74 20 75 ne();......let u
1b00: 73 65 72 6e 61 6d 65 3a 20 53 74 72 69 6e 67 20 sername: String
1b10: 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 75 = row.try_get("u
1b20: 73 65 72 6e 61 6d 65 22 29 3f 3b 0a 09 09 09 09 sername")?;.....
1b30: 09 6c 65 74 20 75 73 65 72 6e 61 6d 65 20 3d 20 .let username =
1b40: 75 73 65 72 6e 61 6d 65 2e 63 6c 6f 6e 65 28 29 username.clone()
1b50: 3b 0a 09 09 09 09 09 74 6f 6b 69 6f 3a 3a 73 70 ;......tokio::sp
1b60: 61 77 6e 28 61 73 79 6e 63 20 6d 6f 76 65 20 7b awn(async move {
1b70: 0a 09 09 09 09 09 09 69 66 20 6c 65 74 20 45 72 .......if let Er
1b80: 72 28 65 72 72 29 20 3d 20 63 6c 6f 6e 65 2e 63 r(err) = clone.c
1b90: 68 65 63 6b 28 26 75 73 65 72 6e 61 6d 65 2c 20 heck(&username,
1ba0: 53 6f 6d 65 28 74 72 75 65 29 29 2e 61 77 61 69 Some(true)).awai
1bb0: 74 20 7b 0a 09 09 09 09 09 09 09 65 70 72 69 6e t {........eprin
1bc0: 74 6c 6e 21 28 22 63 6f 6e 6e 65 63 74 69 6f 6e tln!("connection
1bd0: 20 65 72 72 6f 72 3a 20 7b 7d 22 2c 20 65 72 72 error: {}", err
1be0: 29 3b 0a 09 09 09 09 09 09 7d 0a 09 09 09 09 09 );.......}......
1bf0: 7d 29 3b 0a 09 09 09 09 09 2f 2f 26 73 65 6c 66 });......//&self
1c00: 2e 63 68 65 63 6b 28 72 6f 77 2e 74 72 79 5f 67 .check(row.try_g
1c10: 65 74 28 22 75 73 65 72 6e 61 6d 65 22 29 3f 2c et("username")?,
1c20: 20 53 6f 6d 65 28 74 72 75 65 29 29 2e 61 77 61 Some(true)).awa
1c30: 69 74 3f 3b 0a 09 09 09 09 7d 20 65 6c 73 65 20 it?;.....} else
1c40: 7b 0a 09 09 09 09 09 69 66 20 6e 65 78 74 5f 66 {......if next_f
1c50: 65 74 63 68 20 2d 20 6e 6f 77 20 3c 20 64 65 6c etch - now < del
1c60: 61 79 20 7b 0a 09 09 09 09 09 09 64 65 6c 61 79 ay {.......delay
1c70: 20 3d 20 6e 65 78 74 5f 66 65 74 63 68 20 2d 20 = next_fetch -
1c80: 6e 6f 77 3b 0a 09 09 09 09 09 7d 0a 09 09 09 09 now;......}.....
1c90: 7d 0a 09 09 09 7d 3b 0a 09 09 09 74 6f 6b 69 6f }....};....tokio
1ca0: 3a 3a 74 69 6d 65 3a 3a 64 65 6c 61 79 5f 66 6f ::time::delay_fo
1cb0: 72 28 64 65 6c 61 79 2e 74 6f 5f 73 74 64 28 29 r(delay.to_std()
1cc0: 3f 29 2e 61 77 61 69 74 3b 0a 09 09 7d 0a 09 09 ?).await;...}...
1cd0: 2f 2f 4f 6b 28 28 29 29 0a 09 7d 0a 0a 7d 0a 0a //Ok(())..}..}..
1ce0: 23 5b 74 6f 6b 69 6f 3a 3a 6d 61 69 6e 28 62 61 #[tokio::main(ba
1cf0: 73 69 63 5f 73 63 68 65 64 75 6c 65 72 29 5d 0a sic_scheduler)].
1d00: 61 73 79 6e 63 20 66 6e 20 6d 61 69 6e 28 29 20 async fn main()
1d10: 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 20 7b 0a -> Result<()> {.
1d20: 09 6c 65 74 20 6d 75 74 20 73 65 74 74 69 6e 67 .let mut setting
1d30: 73 20 3d 20 63 6f 6e 66 69 67 3a 3a 43 6f 6e 66 s = config::Conf
1d40: 69 67 3a 3a 64 65 66 61 75 6c 74 28 29 3b 0a 09 ig::default();..
1d50: 73 65 74 74 69 6e 67 73 2e 6d 65 72 67 65 28 63 settings.merge(c
1d60: 6f 6e 66 69 67 3a 3a 46 69 6c 65 3a 3a 77 69 74 onfig::File::wit
1d70: 68 5f 6e 61 6d 65 28 22 72 73 73 74 67 22 29 29 h_name("rsstg"))
1d80: 3f 3b 0a 0a 09 6c 65 74 20 72 65 5f 75 73 65 72 ?;...let re_user
1d90: 6e 61 6d 65 20 3d 20 52 65 67 65 78 3a 3a 6e 65 name = Regex::ne
1da0: 77 28 72 22 5e 40 5b 61 2d 7a 41 2d 5a 5d 5b 61 w(r"^@[a-zA-Z][a
1db0: 2d 7a 41 2d 5a 30 2d 39 5f 5d 2b 24 22 29 3f 3b -zA-Z0-9_]+$")?;
1dc0: 0a 09 6c 65 74 20 72 65 5f 6c 69 6e 6b 20 3d 20 ..let re_link =
1dd0: 52 65 67 65 78 3a 3a 6e 65 77 28 72 22 5e 68 74 Regex::new(r"^ht
1de0: 74 70 73 3f 3a 2f 2f 5b 61 2d 7a 41 2d 5a 2e 30 tps?://[a-zA-Z.0
1df0: 2d 39 5d 2b 2f 5b 2d 5f 61 2d 7a 41 2d 5a 2e 30 -9]+/[-_a-zA-Z.0
1e00: 2d 39 2f 3f 3d 5d 2b 24 22 29 3f 3b 0a 09 6c 65 -9/?=]+$")?;..le
1e10: 74 20 72 65 5f 69 76 5f 68 61 73 68 20 3d 20 52 t re_iv_hash = R
1e20: 65 67 65 78 3a 3a 6e 65 77 28 72 22 5e 5b 61 2d egex::new(r"^[a-
1e30: 66 30 2d 39 5d 7b 31 34 7d 24 22 29 3f 3b 0a 0a f0-9]{14}$")?;..
1e40: 09 6c 65 74 20 63 6f 72 65 20 3d 20 43 6f 72 65 .let core = Core
1e50: 3a 3a 6e 65 77 28 73 65 74 74 69 6e 67 73 29 2e ::new(settings).
1e60: 61 77 61 69 74 3f 3b 0a 0a 09 6c 65 74 20 6d 75 await?;...let mu
1e70: 74 20 73 74 72 65 61 6d 20 3d 20 63 6f 72 65 2e t stream = core.
1e80: 73 74 72 65 61 6d 28 29 3b 0a 0a 09 77 68 69 6c stream();...whil
1e90: 65 20 6c 65 74 20 53 6f 6d 65 28 75 70 64 61 74 e let Some(updat
1ea0: 65 29 20 3d 20 73 74 72 65 61 6d 2e 6e 65 78 74 e) = stream.next
1eb0: 28 29 2e 61 77 61 69 74 20 7b 0a 09 09 6c 65 74 ().await {...let
1ec0: 20 75 70 64 61 74 65 20 3d 20 75 70 64 61 74 65 update = update
1ed0: 3f 3b 0a 09 09 6d 61 74 63 68 20 75 70 64 61 74 ?;...match updat
1ee0: 65 2e 6b 69 6e 64 20 7b 0a 09 09 09 55 70 64 61 e.kind {....Upda
1ef0: 74 65 4b 69 6e 64 3a 3a 4d 65 73 73 61 67 65 28 teKind::Message(
1f00: 6d 65 73 73 61 67 65 29 20 3d 3e 20 7b 0a 09 09 message) => {...
1f10: 09 09 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 3a ..let mut reply:
1f20: 20 56 65 63 3c 53 74 72 69 6e 67 3e 20 3d 20 76 Vec<String> = v
1f30: 65 63 21 5b 5d 3b 0a 09 09 09 09 6d 61 74 63 68 ec![];.....match
1f40: 20 6d 65 73 73 61 67 65 2e 6b 69 6e 64 20 7b 0a message.kind {.
1f50: 09 09 09 09 09 4d 65 73 73 61 67 65 4b 69 6e 64 .....MessageKind
1f60: 3a 3a 54 65 78 74 20 7b 20 72 65 66 20 64 61 74 ::Text { ref dat
1f70: 61 2c 20 2e 2e 20 7d 20 3d 3e 20 7b 0a 09 09 09 a, .. } => {....
1f80: 09 09 09 6c 65 74 20 6d 75 74 20 77 6f 72 64 73 ...let mut words
1f90: 20 3d 20 64 61 74 61 2e 73 70 6c 69 74 5f 77 68 = data.split_wh
1fa0: 69 74 65 73 70 61 63 65 28 29 3b 0a 09 09 09 09 itespace();.....
1fb0: 09 09 6c 65 74 20 63 6d 64 20 3d 20 77 6f 72 64 ..let cmd = word
1fc0: 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 70 28 s.next().unwrap(
1fd0: 29 3b 0a 09 09 09 09 09 09 6d 61 74 63 68 20 63 );.......match c
1fe0: 6d 64 20 7b 0a 0a 2f 2f 20 73 74 61 72 74 0a 0a md {..// start..
1ff0: 09 09 09 09 09 09 09 22 2f 73 74 61 72 74 22 20 ......."/start"
2000: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 72 65 70 => {.........rep
2010: 6c 79 2e 70 75 73 68 28 22 4e 6f 74 20 69 6e 20 ly.push("Not in
2020: 73 65 72 76 69 63 65 20 79 65 74 2e 20 54 72 79 service yet. Try
2030: 20 6c 61 74 65 72 2e 22 2e 74 6f 5f 73 74 72 69 later.".to_stri
2040: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 7d 2c ng());........},
2050: 0a 0a 2f 2f 20 6c 69 73 74 0a 0a 09 09 09 09 09 ..// list.......
2060: 09 09 22 2f 6c 69 73 74 22 20 3d 3e 20 7b 0a 09 .."/list" => {..
2070: 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 .......reply.pus
2080: 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 2e 74 6f h("Channels:".to
2090: 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 _string());.....
20a0: 09 09 09 09 6c 65 74 20 6d 75 74 20 72 6f 77 73 ....let mut rows
20b0: 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 = sqlx::query("
20c0: 73 65 6c 65 63 74 20 75 73 65 72 6e 61 6d 65 2c select username,
20d0: 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 enabled, url, i
20e0: 76 5f 68 61 73 68 20 66 72 6f 6d 20 72 73 73 74 v_hash from rsst
20f0: 67 5f 73 6f 75 72 63 65 20 6c 65 66 74 20 6a 6f g_source left jo
2100: 69 6e 20 72 73 73 74 67 5f 63 68 61 6e 6e 65 6c in rsstg_channel
2110: 20 75 73 69 6e 67 20 28 63 68 61 6e 6e 65 6c 5f using (channel_
2120: 69 64 29 20 77 68 65 72 65 20 6f 77 6e 65 72 20 id) where owner
2130: 3d 20 24 31 22 29 0a 09 09 09 09 09 09 09 09 09 = $1")..........
2140: 2e 62 69 6e 64 28 69 36 34 3a 3a 66 72 6f 6d 28 .bind(i64::from(
2150: 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 29 message.from.id)
2160: 29 0a 09 09 09 09 09 09 09 09 09 2e 66 65 74 63 )...........fetc
2170: 68 28 26 63 6f 72 65 2e 70 6f 6f 6c 29 3b 0a 09 h(&core.pool);..
2180: 09 09 09 09 09 09 09 77 68 69 6c 65 20 6c 65 74 .......while let
2190: 20 53 6f 6d 65 28 72 6f 77 29 20 3d 20 72 6f 77 Some(row) = row
21a0: 73 2e 74 72 79 5f 6e 65 78 74 28 29 2e 61 77 61 s.try_next().awa
21b0: 69 74 3f 20 7b 0a 09 09 09 09 09 09 09 09 09 6c it? {..........l
21c0: 65 74 20 75 73 65 72 6e 61 6d 65 3a 20 26 73 74 et username: &st
21d0: 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 28 r = row.try_get(
21e0: 22 75 73 65 72 6e 61 6d 65 22 29 3f 3b 0a 09 09 "username")?;...
21f0: 09 09 09 09 09 09 09 6c 65 74 20 65 6e 61 62 6c .......let enabl
2200: 65 64 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e 74 ed: bool = row.t
2210: 72 79 5f 67 65 74 28 22 65 6e 61 62 6c 65 64 22 ry_get("enabled"
2220: 29 3f 3b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 )?;..........let
2230: 20 75 72 6c 3a 20 26 73 74 72 20 3d 20 72 6f 77 url: &str = row
2240: 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 22 29 3f .try_get("url")?
2250: 3b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 69 ;..........let i
2260: 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 v_hash: Option<&
2270: 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 str> = row.try_g
2280: 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f 3b 0a et("iv_hash")?;.
2290: 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 .........reply.p
22a0: 75 73 68 28 66 6f 72 6d 61 74 21 28 22 5c 6e 5c ush(format!("\n\
22b0: 5c 2a ef b8 8f e2 83 a3 20 60 7b 7d 60 20 7b 7d \*️⃣ `{}` {}
22c0: 5c 6e f0 9f 94 97 20 60 7b 7d 60 22 2c 20 75 73 \n🔗 `{}`", us
22d0: 65 72 6e 61 6d 65 2c 20 20 0a 09 09 09 09 09 09 ername, .......
22e0: 09 09 09 09 6d 61 74 63 68 20 65 6e 61 62 6c 65 ....match enable
22f0: 64 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 74 d {............t
2300: 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 20 65 6e rue => "🔄 en
2310: 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 09 09 09 abled",.........
2320: 09 09 09 66 61 6c 73 65 20 3d 3e 20 22 e2 9b 94 ...false => "⛔
2330: 20 64 69 73 61 62 6c 65 64 22 2c 0a 09 09 09 09 disabled",.....
2340: 09 09 09 09 09 09 7d 2c 20 75 72 6c 29 29 3b 0a ......}, url));.
2350: 09 09 09 09 09 09 09 09 09 69 66 20 6c 65 74 20 .........if let
2360: 53 6f 6d 65 28 68 61 73 68 29 20 3d 20 69 76 5f Some(hash) = iv_
2370: 68 61 73 68 20 7b 0a 09 09 09 09 09 09 09 09 09 hash {..........
2380: 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 6d .reply.push(form
2390: 61 74 21 28 22 49 56 20 60 7b 7d 60 22 2c 20 68 at!("IV `{}`", h
23a0: 61 73 68 29 29 3b 0a 09 09 09 09 09 09 09 09 09 ash));..........
23b0: 7d 0a 09 09 09 09 09 09 09 09 7d 0a 09 09 09 09 }.........}.....
23c0: 09 09 09 7d 2c 0a 0a 2f 2f 20 61 64 64 0a 0a 09 ...},..// add...
23d0: 09 09 09 09 09 09 22 2f 61 64 64 22 20 3d 3e 20 ......"/add" =>
23e0: 7b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 28 63 {.........let (c
23f0: 68 61 6e 6e 65 6c 2c 20 75 72 6c 2c 20 69 76 5f hannel, url, iv_
2400: 68 61 73 68 29 20 3d 20 28 77 6f 72 64 73 2e 6e hash) = (words.n
2410: 65 78 74 28 29 2e 75 6e 77 72 61 70 28 29 2c 20 ext().unwrap(),
2420: 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 words.next().unw
2430: 72 61 70 28 29 2c 20 77 6f 72 64 73 2e 6e 65 78 rap(), words.nex
2440: 74 28 29 29 3b 0a 09 09 09 09 09 09 09 09 6c 65 t());.........le
2450: 74 20 6f 6b 5f 6c 69 6e 6b 20 3d 20 72 65 5f 6c t ok_link = re_l
2460: 69 6e 6b 2e 69 73 5f 6d 61 74 63 68 28 26 75 72 ink.is_match(&ur
2470: 6c 29 3b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 l);.........let
2480: 6f 6b 5f 68 61 73 68 20 3d 20 6d 61 74 63 68 20 ok_hash = match
2490: 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 iv_hash {.......
24a0: 09 09 09 53 6f 6d 65 28 68 61 73 68 29 20 3d 3e ...Some(hash) =>
24b0: 20 72 65 5f 69 76 5f 68 61 73 68 2e 69 73 5f 6d re_iv_hash.is_m
24c0: 61 74 63 68 28 26 68 61 73 68 29 2c 0a 09 09 09 atch(&hash),....
24d0: 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 72 ......None => tr
24e0: 75 65 2c 0a 09 09 09 09 09 09 09 09 7d 3b 0a 09 ue,.........};..
24f0: 09 09 09 09 09 09 09 69 66 20 21 20 6f 6b 5f 6c .......if ! ok_l
2500: 69 6e 6b 20 7b 0a 09 09 09 09 09 09 09 09 09 72 ink {..........r
2510: 65 70 6c 79 2e 70 75 73 68 28 22 4c 69 6e 6b 20 eply.push("Link
2520: 73 68 6f 75 6c 64 20 62 65 20 6c 69 6e 6b 20 74 should be link t
2530: 6f 20 61 74 6f 6d 2f 72 73 73 20 66 65 65 64 2c o atom/rss feed,
2540: 20 73 6f 6d 65 74 68 69 6e 67 20 6c 69 6b 65 20 something like
2550: 5c 22 68 74 74 70 73 3a 2f 2f 64 6f 6d 61 69 6e \"https://domain
2560: 2f 70 61 74 68 5c 22 5c 5c 2e 22 2e 74 6f 5f 73 /path\"\\.".to_s
2570: 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 tring());.......
2580: 09 09 09 63 6f 72 65 2e 64 65 62 75 67 28 26 66 ...core.debug(&f
2590: 6f 72 6d 61 74 21 28 22 55 72 6c 3a 20 7b 3a 3f ormat!("Url: {:?
25a0: 7d 22 2c 20 26 75 72 6c 29 29 3f 3b 0a 09 09 09 }", &url))?;....
25b0: 09 09 09 09 09 7d 0a 09 09 09 09 09 09 09 09 69 .....}.........i
25c0: 66 20 21 20 6f 6b 5f 68 61 73 68 20 7b 0a 09 09 f ! ok_hash {...
25d0: 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 .......reply.pus
25e0: 68 28 22 49 56 20 68 61 73 68 20 73 68 6f 75 6c h("IV hash shoul
25f0: 64 20 62 65 20 31 34 20 68 65 78 20 64 69 67 69 d be 14 hex digi
2600: 74 73 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 ts.".to_string()
2610: 29 3b 0a 09 09 09 09 09 09 09 09 09 63 6f 72 65 );..........core
2620: 2e 64 65 62 75 67 28 26 66 6f 72 6d 61 74 21 28 .debug(&format!(
2630: 22 49 56 3a 20 7b 3a 3f 7d 22 2c 20 26 69 76 5f "IV: {:?}", &iv_
2640: 68 61 73 68 29 29 3f 3b 0a 09 09 09 09 09 09 09 hash))?;........
2650: 09 7d 0a 09 09 09 09 09 09 09 09 69 66 20 6f 6b .}.........if ok
2660: 5f 6c 69 6e 6b 20 26 26 20 6f 6b 5f 68 61 73 68 _link && ok_hash
2670: 20 7b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 {..........let
2680: 63 68 61 6e 3a 20 4f 70 74 69 6f 6e 3c 69 36 34 chan: Option<i64
2690: 3e 20 3d 20 6d 61 74 63 68 20 73 71 6c 78 3a 3a > = match sqlx::
26a0: 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 63 68 query("select ch
26b0: 61 6e 6e 65 6c 5f 69 64 20 66 72 6f 6d 20 72 73 annel_id from rs
26c0: 73 74 67 5f 63 68 61 6e 6e 65 6c 20 77 68 65 72 stg_channel wher
26d0: 65 20 75 73 65 72 6e 61 6d 65 20 3d 20 24 31 22 e username = $1"
26e0: 29 0a 09 09 09 09 09 09 09 09 09 09 2e 62 69 6e )............bin
26f0: 64 28 63 68 61 6e 6e 65 6c 29 0a 09 09 09 09 09 d(channel)......
2700: 09 09 09 09 09 2e 66 65 74 63 68 5f 6f 6e 65 28 ......fetch_one(
2710: 26 63 6f 72 65 2e 70 6f 6f 6c 29 2e 61 77 61 69 &core.pool).awai
2720: 74 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 4f t {............O
2730: 6b 28 63 68 61 6e 29 20 3d 3e 20 53 6f 6d 65 28 k(chan) => Some(
2740: 63 68 61 6e 2e 74 72 79 5f 67 65 74 28 22 63 68 chan.try_get("ch
2750: 61 6e 6e 65 6c 5f 69 64 22 29 3f 29 2c 0a 09 09 annel_id")?),...
2760: 09 09 09 09 09 09 09 09 09 45 72 72 28 73 71 6c .........Err(sql
2770: 78 3a 3a 45 72 72 6f 72 3a 3a 52 6f 77 4e 6f 74 x::Error::RowNot
2780: 46 6f 75 6e 64 29 20 3d 3e 20 7b 0a 09 09 09 09 Found) => {.....
2790: 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 ........reply.pu
27a0: 73 68 28 22 53 6f 72 72 79 2c 20 49 20 64 6f 6e sh("Sorry, I don
27b0: 27 74 20 6b 6e 6f 77 20 61 62 6f 75 74 20 74 68 't know about th
27c0: 61 74 20 63 68 61 6e 6e 65 6c 2e 20 50 6c 65 61 at channel. Plea
27d0: 73 65 2c 20 61 64 64 20 61 20 63 68 61 6e 6e 65 se, add a channe
27e0: 6c 20 77 69 74 68 20 2f 61 64 64 63 68 61 6e 2e l with /addchan.
27f0: 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a ".to_string());.
2800: 09 09 09 09 09 09 09 09 09 09 09 09 4e 6f 6e 65 ............None
2810: 0a 09 09 09 09 09 09 09 09 09 09 09 7d 2c 0a 09 ............},..
2820: 09 09 09 09 09 09 09 09 09 09 45 72 72 28 65 72 ..........Err(er
2830: 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 r) => {.........
2840: 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 22 ....reply.push("
2850: 53 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 Sorry, unknown e
2860: 72 72 6f 72 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 rror\\.".to_stri
2870: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 09 ng());..........
2880: 09 09 09 63 6f 72 65 2e 64 65 62 75 67 28 26 66 ...core.debug(&f
2890: 6f 72 6d 61 74 21 28 22 53 6f 72 72 79 2c 20 75 ormat!("Sorry, u
28a0: 6e 6b 6e 6f 77 6e 20 65 72 72 6f 72 3a 20 7b 3a nknown error: {:
28b0: 23 3f 7d 5c 6e 22 2c 20 65 72 72 29 29 3f 3b 0a #?}\n", err))?;.
28c0: 09 09 09 09 09 09 09 09 09 09 09 09 4e 6f 6e 65 ............None
28d0: 0a 09 09 09 09 09 09 09 09 09 09 09 7d 2c 0a 09 ............},..
28e0: 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ........};......
28f0: 09 09 09 09 6d 61 74 63 68 20 63 68 61 6e 20 7b ....match chan {
2900: 0a 09 09 09 09 09 09 09 09 09 09 53 6f 6d 65 28 ...........Some(
2910: 63 68 61 6e 29 20 3d 3e 20 7b 0a 09 09 09 09 09 chan) => {......
2920: 09 09 09 09 09 09 6d 61 74 63 68 20 73 71 6c 78 ......match sqlx
2930: 3a 3a 71 75 65 72 79 28 22 69 6e 73 65 72 74 20 ::query("insert
2940: 69 6e 74 6f 20 72 73 73 74 67 5f 73 6f 75 72 63 into rsstg_sourc
2950: 65 20 28 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 e (channel_id, u
2960: 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e rl, iv_hash, own
2970: 65 72 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 er) values ($1,
2980: 24 32 2c 20 24 33 2c 20 24 34 29 20 6f 6e 20 63 $2, $3, $4) on c
2990: 6f 6e 66 6c 69 63 74 20 28 63 68 61 6e 6e 65 6c onflict (channel
29a0: 5f 69 64 2c 20 6f 77 6e 65 72 29 20 64 6f 20 75 _id, owner) do u
29b0: 70 64 61 74 65 20 73 65 74 20 75 72 6c 20 3d 20 pdate set url =
29c0: 65 78 63 6c 75 64 65 64 2e 75 72 6c 2c 20 69 76 excluded.url, iv
29d0: 5f 68 61 73 68 20 3d 20 65 78 63 6c 75 64 65 64 _hash = excluded
29e0: 2e 69 76 5f 68 61 73 68 3b 22 29 0a 09 09 09 09 .iv_hash;").....
29f0: 09 09 09 09 09 09 09 09 2e 62 69 6e 64 28 63 68 .........bind(ch
2a00: 61 6e 29 0a 09 09 09 09 09 09 09 09 09 09 09 09 an).............
2a10: 2e 62 69 6e 64 28 75 72 6c 29 0a 09 09 09 09 09 .bind(url)......
2a20: 09 09 09 09 09 09 09 2e 62 69 6e 64 28 69 76 5f ........bind(iv_
2a30: 68 61 73 68 29 0a 09 09 09 09 09 09 09 09 09 09 hash)...........
2a40: 09 09 2e 62 69 6e 64 28 69 36 34 3a 3a 66 72 6f ...bind(i64::fro
2a50: 6d 28 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 m(message.from.i
2a60: 64 29 29 0a 09 09 09 09 09 09 09 09 09 09 09 09 d)).............
2a70: 2e 65 78 65 63 75 74 65 28 26 63 6f 72 65 2e 70 .execute(&core.p
2a80: 6f 6f 6c 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 ool).await {....
2a90: 09 09 09 09 09 09 09 09 09 4f 6b 28 5f 29 20 3d .........Ok(_) =
2aa0: 3e 20 72 65 70 6c 79 2e 70 75 73 68 28 22 43 68 > reply.push("Ch
2ab0: 61 6e 6e 65 6c 20 61 64 64 65 64 5c 5c 2e 22 2e annel added\\.".
2ac0: 74 6f 5f 73 74 72 69 6e 67 28 29 29 2c 0a 09 09 to_string()),...
2ad0: 09 09 09 09 09 09 09 09 09 09 45 72 72 28 73 71 ..........Err(sq
2ae0: 6c 78 3a 3a 45 72 72 6f 72 3a 3a 44 61 74 61 62 lx::Error::Datab
2af0: 61 73 65 28 65 72 72 29 29 20 3d 3e 20 7b 0a 09 ase(err)) => {..
2b00: 09 09 09 09 09 09 09 09 09 09 09 09 6d 61 74 63 ............matc
2b10: 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a 3a h err.downcast::
2b20: 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 3a <sqlx::postgres:
2b30: 3a 50 67 44 61 74 61 62 61 73 65 45 72 72 6f 72 :PgDatabaseError
2b40: 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b 0a >().routine() {.
2b50: 09 09 09 09 09 09 09 09 09 09 09 09 09 09 53 6f ..............So
2b60: 6d 65 28 22 5f 62 74 5f 63 68 65 63 6b 5f 75 6e me("_bt_check_un
2b70: 69 71 75 65 22 2c 20 29 20 3d 3e 20 7b 0a 09 09 ique", ) => {...
2b80: 09 09 09 09 09 09 09 09 09 09 09 09 09 72 65 70 .............rep
2b90: 6c 79 2e 70 75 73 68 28 22 44 75 70 6c 69 63 61 ly.push("Duplica
2ba0: 74 65 20 6b 65 79 5c 5c 2e 22 2e 74 6f 5f 73 74 te key\\.".to_st
2bb0: 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 ring());........
2bc0: 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 .......},.......
2bd0: 09 09 09 09 09 09 09 09 53 6f 6d 65 28 5f 29 20 ........Some(_)
2be0: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 => {............
2bf0: 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 22 ....reply.push("
2c00: 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 5c 5c Database error\\
2c10: 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b .".to_string());
2c20: 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 09 7d ...............}
2c30: 2c 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 09 ,...............
2c40: 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 09 None => {.......
2c50: 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 .........reply.p
2c60: 75 73 68 28 22 4e 6f 20 64 61 74 61 62 61 73 65 ush("No database
2c70: 20 65 72 72 6f 72 20 65 78 74 72 61 63 74 65 64 error extracted
2c80: 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 \\.".to_string()
2c90: 29 3b 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 );..............
2ca0: 09 7d 2c 0a 09 09 09 09 09 09 09 09 09 09 09 09 .},.............
2cb0: 09 7d 3b 0a 09 09 09 09 09 09 09 09 09 09 09 09 .};.............
2cc0: 7d 2c 0a 09 09 09 09 09 09 09 09 09 09 09 09 45 },.............E
2cd0: 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 rr(err) => {....
2ce0: 09 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e ..........reply.
2cf0: 70 75 73 68 28 22 53 6f 72 72 79 2c 20 75 6e 6b push("Sorry, unk
2d00: 6e 6f 77 6e 20 65 72 72 6f 72 5c 5c 2e 22 2e 74 nown error\\.".t
2d10: 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 o_string());....
2d20: 09 09 09 09 09 09 09 09 09 09 63 6f 72 65 2e 64 ..........core.d
2d30: 65 62 75 67 28 26 66 6f 72 6d 61 74 21 28 22 53 ebug(&format!("S
2d40: 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 72 orry, unknown er
2d50: 72 6f 72 3a 20 7b 3a 23 3f 7d 5c 6e 22 2c 20 65 ror: {:#?}\n", e
2d60: 72 72 29 29 3f 3b 0a 09 09 09 09 09 09 09 09 09 rr))?;..........
2d70: 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 09 09 ...},...........
2d80: 09 7d 3b 0a 09 09 09 09 09 09 09 09 09 09 7d 2c .};...........},
2d90: 0a 09 09 09 09 09 09 09 09 09 09 4e 6f 6e 65 20 ...........None
2da0: 3d 3e 20 7b 7d 2c 0a 09 09 09 09 09 09 09 09 09 => {},..........
2db0: 7d 3b 0a 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 };.........};...
2dc0: 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 61 64 64 63 .....},..// addc
2dd0: 68 61 6e 0a 0a 09 09 09 09 09 09 09 22 2f 61 64 han........."/ad
2de0: 64 63 68 61 6e 22 20 3d 3e 20 7b 0a 09 09 09 09 dchan" => {.....
2df0: 09 09 09 09 6c 65 74 20 63 68 61 6e 6e 65 6c 20 ....let channel
2e00: 3d 20 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e 75 = words.next().u
2e10: 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 09 09 09 nwrap();........
2e20: 09 69 66 20 21 20 72 65 5f 75 73 65 72 6e 61 6d .if ! re_usernam
2e30: 65 2e 69 73 5f 6d 61 74 63 68 28 26 63 68 61 6e e.is_match(&chan
2e40: 6e 65 6c 29 20 7b 0a 09 09 09 09 09 09 09 09 09 nel) {..........
2e50: 72 65 70 6c 79 2e 70 75 73 68 28 22 55 73 65 72 reply.push("User
2e60: 6e 61 6d 65 73 20 73 68 6f 75 6c 64 20 62 65 20 names should be
2e70: 73 6f 6d 65 74 68 69 6e 67 20 6c 69 6b 65 20 5c something like \
2e80: 22 40 5c 5c 5b 61 5c 5c 2d 7a 41 5c 5c 2d 5a 5d "@\\[a\\-zA\\-Z]
2e90: 5c 5c 5b 61 5c 5c 2d 7a 41 5c 5c 2d 5a 30 5c 5c \\[a\\-zA\\-Z0\\
2ea0: 2d 39 5c 5c 5f 5d 2b 5c 22 2c 20 61 72 65 6e 27 -9\\_]+\", aren'
2eb0: 74 20 74 68 65 79 3f 22 2e 74 6f 5f 73 74 72 69 t they?".to_stri
2ec0: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 7d ng());.........}
2ed0: 20 65 6c 73 65 20 7b 0a 09 09 09 09 09 09 09 09 else {.........
2ee0: 09 6c 65 74 20 63 68 61 6e 3a 20 4f 70 74 69 6f .let chan: Optio
2ef0: 6e 3c 69 36 34 3e 20 3d 20 6d 61 74 63 68 20 73 n<i64> = match s
2f00: 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 qlx::query("sele
2f10: 63 74 20 63 68 61 6e 6e 65 6c 5f 69 64 20 66 72 ct channel_id fr
2f20: 6f 6d 20 72 73 73 74 67 5f 63 68 61 6e 6e 65 6c om rsstg_channel
2f30: 20 77 68 65 72 65 20 75 73 65 72 6e 61 6d 65 20 where username
2f40: 3d 20 24 31 22 29 0a 09 09 09 09 09 09 09 09 09 = $1")..........
2f50: 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c 29 0a ..bind(channel).
2f60: 09 09 09 09 09 09 09 09 09 09 2e 66 65 74 63 68 ...........fetch
2f70: 5f 6f 6e 65 28 26 63 6f 72 65 2e 70 6f 6f 6c 29 _one(&core.pool)
2f80: 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 09 .await {........
2f90: 09 09 09 09 4f 6b 28 63 68 61 6e 29 20 3d 3e 20 ....Ok(chan) =>
2fa0: 53 6f 6d 65 28 63 68 61 6e 2e 74 72 79 5f 67 65 Some(chan.try_ge
2fb0: 74 28 22 63 68 61 6e 6e 65 6c 5f 69 64 22 29 3f t("channel_id")?
2fc0: 29 2c 0a 09 09 09 09 09 09 09 09 09 09 09 45 72 ),............Er
2fd0: 72 28 73 71 6c 78 3a 3a 45 72 72 6f 72 3a 3a 52 r(sqlx::Error::R
2fe0: 6f 77 4e 6f 74 46 6f 75 6e 64 29 20 3d 3e 20 4e owNotFound) => N
2ff0: 6f 6e 65 2c 0a 09 09 09 09 09 09 09 09 09 09 09 one,............
3000: 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 Err(err) => {...
3010: 09 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e ..........reply.
3020: 70 75 73 68 28 22 53 6f 72 72 79 2c 20 75 6e 6b push("Sorry, unk
3030: 6e 6f 77 6e 20 65 72 72 6f 72 5c 5c 2e 22 2e 74 nown error\\.".t
3040: 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 o_string());....
3050: 09 09 09 09 09 09 09 09 09 63 6f 72 65 2e 64 65 .........core.de
3060: 62 75 67 28 26 66 6f 72 6d 61 74 21 28 22 53 6f bug(&format!("So
3070: 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 72 72 rry, unknown err
3080: 6f 72 3a 20 7b 3a 23 3f 7d 5c 6e 22 2c 20 65 72 or: {:#?}\n", er
3090: 72 29 29 3f 3b 0a 09 09 09 09 09 09 09 09 09 09 r))?;...........
30a0: 09 09 4e 6f 6e 65 0a 09 09 09 09 09 09 09 09 09 ..None..........
30b0: 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 09 7d 3b ..},..........};
30c0: 0a 09 09 09 09 09 09 09 09 09 6d 61 74 63 68 20 ..........match
30d0: 63 68 61 6e 20 7b 0a 09 09 09 09 09 09 09 09 09 chan {..........
30e0: 09 53 6f 6d 65 28 63 68 61 6e 29 20 3d 3e 20 7b .Some(chan) => {
30f0: 0a 09 09 09 09 09 09 09 09 09 09 09 6c 65 74 20 ............let
3100: 6e 65 77 5f 63 68 61 6e 20 3d 20 63 6f 72 65 2e new_chan = core.
3110: 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 72 61 6d tg.send(telegram
3120: 5f 62 6f 74 3a 3a 47 65 74 43 68 61 74 3a 3a 6e _bot::GetChat::n
3130: 65 77 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a ew(telegram_bot:
3140: 3a 74 79 70 65 73 3a 3a 43 68 61 74 49 64 3a 3a :types::ChatId::
3150: 6e 65 77 28 63 68 61 6e 29 29 29 2e 61 77 61 69 new(chan))).awai
3160: 74 3f 3b 0a 09 09 09 09 09 09 09 09 09 09 09 69 t?;............i
3170: 66 20 69 36 34 3a 3a 66 72 6f 6d 28 6e 65 77 5f f i64::from(new_
3180: 63 68 61 6e 2e 69 64 28 29 29 20 3d 3d 20 63 68 chan.id()) == ch
3190: 61 6e 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 an {............
31a0: 09 72 65 70 6c 79 2e 70 75 73 68 28 22 49 20 61 .reply.push("I a
31b0: 6c 72 65 61 64 79 20 6b 6e 6f 77 20 74 68 61 74 lready know that
31c0: 20 63 68 61 6e 6e 65 6c 5c 5c 2e 22 2e 74 6f 5f channel\\.".to_
31d0: 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 string());......
31e0: 09 09 09 09 09 09 7d 20 65 6c 73 65 20 7b 0a 09 ......} else {..
31f0: 09 09 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 ...........reply
3200: 2e 70 75 73 68 28 22 48 6d 6d 2c 20 63 68 61 6e .push("Hmm, chan
3210: 6e 65 6c 20 68 61 73 20 63 68 61 6e 67 65 64 e2 nel has changed
3220: 80 a6 20 49 27 6c 6c 20 66 69 78 20 69 74 20 6c I'll fix it l
3230: 61 74 65 72 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 ater\\.".to_stri
3240: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 09 ng());..........
3250: 09 09 7d 3b 0a 09 09 09 09 09 09 09 09 09 09 7d ..};...........}
3260: 2c 0a 09 09 09 09 09 09 09 09 09 09 4e 6f 6e 65 ,...........None
3270: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 09 => {...........
3280: 09 6d 61 74 63 68 20 63 6f 72 65 2e 74 67 2e 73 .match core.tg.s
3290: 65 6e 64 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 end(telegram_bot
32a0: 3a 3a 47 65 74 43 68 61 74 41 64 6d 69 6e 69 73 ::GetChatAdminis
32b0: 74 72 61 74 6f 72 73 3a 3a 6e 65 77 28 74 65 6c trators::new(tel
32c0: 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 egram_bot::types
32d0: 3a 3a 43 68 61 74 52 65 66 3a 3a 43 68 61 6e 6e ::ChatRef::Chann
32e0: 65 6c 55 73 65 72 6e 61 6d 65 28 63 68 61 6e 6e elUsername(chann
32f0: 65 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 29 el.to_string()))
3300: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 ).await {.......
3310: 09 09 09 09 09 09 4f 6b 28 63 68 61 6e 5f 61 64 ......Ok(chan_ad
3320: 6d 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 m) => {.........
3330: 09 09 09 09 09 6c 65 74 20 28 6d 75 74 20 6d 65 .....let (mut me
3340: 2c 20 6d 75 74 20 75 73 65 72 29 20 3d 20 28 66 , mut user) = (f
3350: 61 6c 73 65 2c 20 66 61 6c 73 65 29 3b 0a 09 09 alse, false);...
3360: 09 09 09 09 09 09 09 09 09 09 09 66 6f 72 20 61 ...........for a
3370: 64 6d 69 6e 20 69 6e 20 26 63 68 61 6e 5f 61 64 dmin in &chan_ad
3380: 6d 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 09 m {.............
3390: 09 09 69 66 20 61 64 6d 69 6e 2e 75 73 65 72 2e ..if admin.user.
33a0: 69 64 20 3d 3d 20 63 6f 72 65 2e 6d 79 2e 69 64 id == core.my.id
33b0: 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 {..............
33c0: 09 09 6d 65 20 3d 20 74 72 75 65 3b 0a 09 09 09 ..me = true;....
33d0: 09 09 09 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 ...........};...
33e0: 09 09 09 09 09 09 09 09 09 09 09 09 69 66 20 61 ............if a
33f0: 64 6d 69 6e 2e 75 73 65 72 2e 69 64 20 3d 3d 20 dmin.user.id ==
3400: 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 20 message.from.id
3410: 7b 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 09 {...............
3420: 09 75 73 65 72 20 3d 20 74 72 75 65 3b 0a 09 09 .user = true;...
3430: 09 09 09 09 09 09 09 09 09 09 09 09 7d 3b 0a 09 ............};..
3440: 09 09 09 09 09 09 09 09 09 09 09 09 7d 3b 0a 09 ............};..
3450: 09 09 09 09 09 09 09 09 09 09 09 09 69 66 20 21 ............if !
3460: 20 6d 65 20 20 20 7b 20 72 65 70 6c 79 2e 70 75 me { reply.pu
3470: 73 68 28 22 49 20 6e 65 65 64 20 74 6f 20 62 65 sh("I need to be
3480: 20 61 64 6d 69 6e 20 6f 6e 20 74 68 61 74 20 63 admin on that c
3490: 68 61 6e 6e 65 6c 5c 5c 2e 22 2e 74 6f 5f 73 74 hannel\\.".to_st
34a0: 72 69 6e 67 28 29 29 3b 20 7d 3b 0a 09 09 09 09 ring()); };.....
34b0: 09 09 09 09 09 09 09 09 09 69 66 20 21 20 75 73 .........if ! us
34c0: 65 72 20 7b 20 72 65 70 6c 79 2e 70 75 73 68 28 er { reply.push(
34d0: 22 59 6f 75 20 73 68 6f 75 6c 64 20 62 65 20 61 "You should be a
34e0: 64 6d 69 6e 20 6f 6e 20 74 68 61 74 20 63 68 61 dmin on that cha
34f0: 6e 6e 65 6c 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 nnel\\.".to_stri
3500: 6e 67 28 29 29 3b 20 7d 3b 0a 09 09 09 09 09 09 ng()); };.......
3510: 09 09 09 09 09 09 09 69 66 20 6d 65 20 26 26 20 .......if me &&
3520: 75 73 65 72 20 7b 0a 09 09 09 09 09 09 09 09 09 user {..........
3530: 09 09 09 09 09 6c 65 74 20 63 68 61 6e 5f 69 64 .....let chan_id
3540: 20 3d 20 63 6f 72 65 2e 74 67 2e 73 65 6e 64 28 = core.tg.send(
3550: 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 47 65 telegram_bot::Ge
3560: 74 43 68 61 74 3a 3a 6e 65 77 28 74 65 6c 65 67 tChat::new(teleg
3570: 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a 3a ram_bot::types::
3580: 43 68 61 74 52 65 66 3a 3a 43 68 61 6e 6e 65 6c ChatRef::Channel
3590: 55 73 65 72 6e 61 6d 65 28 63 68 61 6e 6e 65 6c Username(channel
35a0: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 29 29 2e .to_string()))).
35b0: 61 77 61 69 74 3f 3b 0a 09 09 09 09 09 09 09 09 await?;.........
35c0: 09 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 ......sqlx::quer
35d0: 79 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 y("insert into r
35e0: 73 73 74 67 5f 63 68 61 6e 6e 65 6c 20 28 63 68 sstg_channel (ch
35f0: 61 6e 6e 65 6c 5f 69 64 2c 20 75 73 65 72 6e 61 annel_id, userna
3600: 6d 65 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 me) values ($1,
3610: 24 32 29 3b 22 29 0a 09 09 09 09 09 09 09 09 09 $2);")..........
3620: 09 09 09 09 09 09 2e 62 69 6e 64 28 69 36 34 3a .......bind(i64:
3630: 3a 66 72 6f 6d 28 63 68 61 6e 5f 69 64 2e 69 64 :from(chan_id.id
3640: 28 29 29 29 0a 09 09 09 09 09 09 09 09 09 09 09 ()))............
3650: 09 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 .....bind(channe
3660: 6c 29 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 l)..............
3670: 09 09 2e 65 78 65 63 75 74 65 28 26 63 6f 72 65 ...execute(&core
3680: 2e 70 6f 6f 6c 29 2e 61 77 61 69 74 3f 3b 0a 09 .pool).await?;..
3690: 09 09 09 09 09 09 09 09 09 09 09 09 09 72 65 70 .............rep
36a0: 6c 79 2e 70 75 73 68 28 22 47 6f 6f 64 2c 20 49 ly.push("Good, I
36b0: 20 6b 6e 6f 77 20 74 68 61 74 20 63 68 61 6e 6e know that chann
36c0: 65 6c 20 6e 6f 77 5c 5c 2e 5c 6e 22 2e 74 6f 5f el now\\.\n".to_
36d0: 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 string());......
36e0: 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ........};......
36f0: 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 .......},.......
3700: 09 09 09 09 09 09 45 72 72 28 5f 29 20 3d 3e 20 ......Err(_) =>
3710: 7b 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 72 {..............r
3720: 65 70 6c 79 2e 70 75 73 68 28 22 53 6f 72 72 79 eply.push("Sorry
3730: 2c 20 49 20 68 61 76 65 20 6e 6f 20 61 63 63 65 , I have no acce
3740: 73 73 20 74 6f 20 74 68 61 74 20 63 68 61 74 5c ss to that chat\
3750: 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 \.".to_string())
3760: 3b 0a 09 09 09 09 09 09 09 09 09 09 09 09 7d 2c ;.............},
3770: 0a 09 09 09 09 09 09 09 09 09 09 09 7d 3b 0a 09 ............};..
3780: 09 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 .........},.....
3790: 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 09 .....};.........
37a0: 7d 3b 0a 09 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f };........},..//
37b0: 20 63 68 65 63 6b 0a 0a 09 09 09 09 09 09 09 22 check........."
37c0: 2f 63 68 65 63 6b 22 20 3d 3e 20 7b 0a 09 09 09 /check" => {....
37d0: 09 09 09 09 09 6c 65 74 20 63 68 61 6e 6e 65 6c .....let channel
37e0: 20 3d 20 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e = words.next().
37f0: 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 09 09 unwrap();.......
3800: 09 09 69 66 20 21 20 72 65 5f 75 73 65 72 6e 61 ..if ! re_userna
3810: 6d 65 2e 69 73 5f 6d 61 74 63 68 28 26 63 68 61 me.is_match(&cha
3820: 6e 6e 65 6c 29 20 7b 0a 09 09 09 09 09 09 09 09 nnel) {.........
3830: 09 72 65 70 6c 79 2e 70 75 73 68 28 22 55 73 65 .reply.push("Use
3840: 72 6e 61 6d 65 73 20 73 68 6f 75 6c 64 20 62 65 rnames should be
3850: 20 73 6f 6d 65 74 68 69 6e 67 20 6c 69 6b 65 20 something like
3860: 5c 22 40 5c 5c 5b 61 2d 7a 5d 5c 5c 5b 61 2d 7a \"@\\[a-z]\\[a-z
3870: 30 2d 39 5f 5d 2b 5c 22 2c 20 61 72 65 6e 27 74 0-9_]+\", aren't
3880: 20 74 68 65 79 3f 22 2e 74 6f 5f 73 74 72 69 6e they?".to_strin
3890: 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 7d 20 g());.........}
38a0: 65 6c 73 65 20 7b 0a 09 09 09 09 09 09 09 09 09 else {..........
38b0: 26 63 6f 72 65 2e 63 68 65 63 6b 28 63 68 61 6e &core.check(chan
38c0: 6e 65 6c 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 74 nel, None).await
38d0: 3f 3b 0a 09 09 09 09 09 09 09 09 7d 0a 09 09 09 ?;.........}....
38e0: 09 09 09 09 7d 2c 0a 0a 2f 2f 20 63 6c 65 61 72 ....},..// clear
38f0: 0a 0a 09 09 09 09 09 09 09 22 2f 63 6c 65 61 6e ........."/clean
3900: 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 69 " => {.........i
3910: 66 20 63 6f 72 65 2e 6f 77 6e 65 72 20 21 3d 20 f core.owner !=
3920: 69 36 34 3a 3a 66 72 6f 6d 28 6d 65 73 73 61 67 i64::from(messag
3930: 65 2e 66 72 6f 6d 2e 69 64 29 20 7b 0a 09 09 09 e.from.id) {....
3940: 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 ......reply.push
3950: 28 22 52 65 73 65 72 76 65 64 20 66 6f 72 20 74 ("Reserved for t
3960: 65 73 74 69 6e 67 5c 5c 2e 22 2e 74 6f 5f 73 74 esting\\.".to_st
3970: 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 ring());........
3980: 09 7d 20 65 6c 73 65 20 7b 0a 09 09 09 09 09 09 .} else {.......
3990: 09 09 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 64 ...let source_id
39a0: 20 3d 20 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e = words.next().
39b0: 75 6e 77 72 61 70 28 29 2e 70 61 72 73 65 3a 3a unwrap().parse::
39c0: 3c 69 33 32 3e 28 29 2e 75 6e 77 72 61 70 5f 6f <i32>().unwrap_o
39d0: 72 28 30 29 3b 0a 09 09 09 09 09 09 09 09 09 26 r(0);..........&
39e0: 63 6f 72 65 2e 63 6c 65 61 6e 28 73 6f 75 72 63 core.clean(sourc
39f0: 65 5f 69 64 29 2e 61 77 61 69 74 3f 3b 0a 09 09 e_id).await?;...
3a00: 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 09 7d ......}........}
3a10: 2c 0a 0a 2f 2f 20 65 6e 61 62 6c 65 0a 0a 09 09 ,..// enable....
3a20: 09 09 09 09 09 22 2f 65 6e 61 62 6c 65 22 20 3d ....."/enable" =
3a30: 3e 20 7b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 > {.........let
3a40: 63 68 61 6e 6e 65 6c 20 3d 20 77 6f 72 64 73 2e channel = words.
3a50: 6e 65 78 74 28 29 2e 75 6e 77 72 61 70 28 29 3b next().unwrap();
3a60: 0a 09 09 09 09 09 09 09 09 69 66 20 21 20 72 65 .........if ! re
3a70: 5f 75 73 65 72 6e 61 6d 65 2e 69 73 5f 6d 61 74 _username.is_mat
3a80: 63 68 28 26 63 68 61 6e 6e 65 6c 29 20 7b 0a 09 ch(&channel) {..
3a90: 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 ........reply.pu
3aa0: 73 68 28 22 55 73 65 72 6e 61 6d 65 73 20 73 68 sh("Usernames sh
3ab0: 6f 75 6c 64 20 62 65 20 73 6f 6d 65 74 68 69 6e ould be somethin
3ac0: 67 20 6c 69 6b 65 20 5c 22 40 5c 5c 5b 61 2d 7a g like \"@\\[a-z
3ad0: 5d 5c 5c 5b 61 2d 7a 30 2d 39 5f 5d 2b 5c 22 2c ]\\[a-z0-9_]+\",
3ae0: 20 61 72 65 6e 27 74 20 74 68 65 79 3f 22 2e 74 aren't they?".t
3af0: 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 o_string());....
3b00: 09 09 09 09 09 7d 20 65 6c 73 65 20 7b 0a 09 09 .....} else {...
3b10: 09 09 09 09 09 09 09 6d 61 74 63 68 20 63 6f 72 .......match cor
3b20: 65 2e 65 6e 61 62 6c 65 28 6d 65 73 73 61 67 65 e.enable(message
3b30: 2e 66 72 6f 6d 2e 69 64 2c 20 63 68 61 6e 6e 65 .from.id, channe
3b40: 6c 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 l).await {......
3b50: 09 09 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 7b 0a .....Ok(_) => {.
3b60: 09 09 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 ...........reply
3b70: 2e 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 20 65 .push("Channel e
3b80: 6e 61 62 6c 65 64 5c 5c 2e 22 2e 74 6f 5f 73 74 nabled\\.".to_st
3b90: 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 ring());........
3ba0: 09 09 09 7d 0a 09 09 09 09 09 09 09 09 09 09 45 ...}...........E
3bb0: 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 rr(err) => {....
3bc0: 09 09 09 09 09 09 09 09 63 6f 72 65 2e 64 65 62 ........core.deb
3bd0: 75 67 28 26 65 72 72 2e 74 6f 5f 73 74 72 69 6e ug(&err.to_strin
3be0: 67 28 29 29 3f 3b 0a 09 09 09 09 09 09 09 09 09 g())?;..........
3bf0: 09 7d 2c 0a 09 09 09 09 09 09 09 09 09 7d 0a 09 .},..........}..
3c00: 09 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 09 .......}........
3c10: 7d 2c 0a 0a 2f 2f 20 64 69 73 61 62 6c 65 0a 0a },..// disable..
3c20: 09 09 09 09 09 09 09 22 2f 64 69 73 61 62 6c 65 ......."/disable
3c30: 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 6c " => {.........l
3c40: 65 74 20 63 68 61 6e 6e 65 6c 20 3d 20 77 6f 72 et channel = wor
3c50: 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 70 ds.next().unwrap
3c60: 28 29 3b 0a 09 09 09 09 09 09 09 09 69 66 20 21 ();.........if !
3c70: 20 72 65 5f 75 73 65 72 6e 61 6d 65 2e 69 73 5f re_username.is_
3c80: 6d 61 74 63 68 28 26 63 68 61 6e 6e 65 6c 29 20 match(&channel)
3c90: 7b 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 {..........reply
3ca0: 2e 70 75 73 68 28 22 55 73 65 72 6e 61 6d 65 73 .push("Usernames
3cb0: 20 73 68 6f 75 6c 64 20 62 65 20 73 6f 6d 65 74 should be somet
3cc0: 68 69 6e 67 20 6c 69 6b 65 20 5c 22 40 5c 5c 5b hing like \"@\\[
3cd0: 61 2d 7a 5d 5c 5c 5b 61 2d 7a 30 2d 39 5f 5d 2b a-z]\\[a-z0-9_]+
3ce0: 5c 22 2c 20 61 72 65 6e 27 74 20 74 68 65 79 3f \", aren't they?
3cf0: 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a ".to_string());.
3d00: 09 09 09 09 09 09 09 09 7d 20 65 6c 73 65 20 7b ........} else {
3d10: 0a 09 09 09 09 09 09 09 09 09 6d 61 74 63 68 20 ..........match
3d20: 63 6f 72 65 2e 64 69 73 61 62 6c 65 28 6d 65 73 core.disable(mes
3d30: 73 61 67 65 2e 66 72 6f 6d 2e 69 64 2c 20 63 68 sage.from.id, ch
3d40: 61 6e 6e 65 6c 29 2e 61 77 61 69 74 20 7b 0a 09 annel).await {..
3d50: 09 09 09 09 09 09 09 09 09 4f 6b 28 5f 29 20 3d .........Ok(_) =
3d60: 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 72 > {............r
3d70: 65 70 6c 79 2e 70 75 73 68 28 22 43 68 61 6e 6e eply.push("Chann
3d80: 65 6c 20 64 69 73 61 62 6c 65 64 5c 5c 2e 22 2e el disabled\\.".
3d90: 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 to_string());...
3da0: 09 09 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 ........}.......
3db0: 09 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e 20 ....Err(err) =>
3dc0: 7b 0a 09 09 09 09 09 09 09 09 09 09 09 63 6f 72 {............cor
3dd0: 65 2e 64 65 62 75 67 28 26 65 72 72 2e 74 6f 5f e.debug(&err.to_
3de0: 73 74 72 69 6e 67 28 29 29 3f 3b 0a 09 09 09 09 string())?;.....
3df0: 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 ......},........
3e00: 09 09 7d 0a 09 09 09 09 09 09 09 09 7d 0a 09 09 ..}.........}...
3e10: 09 09 09 09 09 7d 2c 0a 0a 09 09 09 09 09 09 09 .....},.........
3e20: 5f 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 7d 2c _ => {........},
3e30: 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 7d .......};......}
3e40: 2c 0a 09 09 09 09 09 5f 20 3d 3e 20 7b 0a 09 09 ,......_ => {...
3e50: 09 09 09 7d 2c 0a 09 09 09 09 7d 3b 0a 09 09 09 ...},.....};....
3e60: 09 69 66 20 72 65 70 6c 79 2e 6c 65 6e 28 29 20 .if reply.len()
3e70: 3e 20 30 20 7b 0a 09 09 09 09 09 6d 61 74 63 68 > 0 {......match
3e80: 20 63 6f 72 65 2e 74 67 2e 73 65 6e 64 28 6d 65 core.tg.send(me
3e90: 73 73 61 67 65 2e 74 65 78 74 5f 72 65 70 6c 79 ssage.text_reply
3ea0: 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e 22 (reply.join("\n"
3eb0: 29 29 2e 70 61 72 73 65 5f 6d 6f 64 65 28 74 79 )).parse_mode(ty
3ec0: 70 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3a 3a pes::ParseMode::
3ed0: 4d 61 72 6b 64 6f 77 6e 56 32 29 29 2e 61 77 61 MarkdownV2)).awa
3ee0: 69 74 20 7b 0a 09 09 09 09 09 09 4f 6b 28 5f 29 it {.......Ok(_)
3ef0: 20 3d 3e 20 7b 7d 2c 0a 09 09 09 09 09 09 45 72 => {},.......Er
3f00: 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 r(err) => {.....
3f10: 09 09 09 64 62 67 21 28 72 65 70 6c 79 2e 6a 6f ...dbg!(reply.jo
3f20: 69 6e 28 22 5c 6e 22 29 29 3b 0a 09 09 09 09 09 in("\n"));......
3f30: 09 09 70 72 69 6e 74 6c 6e 21 28 22 7b 7d 22 2c ..println!("{}",
3f40: 20 65 72 72 29 3b 0a 09 09 09 09 09 09 7d 2c 0a err);.......},.
3f50: 09 09 09 09 09 7d 0a 09 09 09 09 7d 0a 09 09 09 .....}.....}....
3f60: 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 7d 2c 0a 09 },...._ => {},..
3f70: 09 7d 3b 0a 09 7d 0a 0a 09 4f 6b 28 28 29 29 0a .};..}...Ok(()).
3f80: 7d 0a }.