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 2d 5d 2b 2f 5b 2d 5f 61 2d 7a 41 2d 5a 2e -9-]+/[-_a-zA-Z.
1e00: 30 2d 39 2f 3f 3d 5d 2b 24 22 29 3f 3b 0a 09 6c 0-9/?=]+$")?;..l
1e10: 65 74 20 72 65 5f 69 76 5f 68 61 73 68 20 3d 20 et re_iv_hash =
1e20: 52 65 67 65 78 3a 3a 6e 65 77 28 72 22 5e 5b 61 Regex::new(r"^[a
1e30: 2d 66 30 2d 39 5d 7b 31 34 7d 24 22 29 3f 3b 0a -f0-9]{14}$")?;.
1e40: 0a 09 6c 65 74 20 63 6f 72 65 20 3d 20 43 6f 72 ..let core = Cor
1e50: 65 3a 3a 6e 65 77 28 73 65 74 74 69 6e 67 73 29 e::new(settings)
1e60: 2e 61 77 61 69 74 3f 3b 0a 0a 09 6c 65 74 20 6d .await?;...let m
1e70: 75 74 20 73 74 72 65 61 6d 20 3d 20 63 6f 72 65 ut stream = core
1e80: 2e 73 74 72 65 61 6d 28 29 3b 0a 0a 09 77 68 69 .stream();...whi
1e90: 6c 65 20 6c 65 74 20 53 6f 6d 65 28 75 70 64 61 le let Some(upda
1ea0: 74 65 29 20 3d 20 73 74 72 65 61 6d 2e 6e 65 78 te) = stream.nex
1eb0: 74 28 29 2e 61 77 61 69 74 20 7b 0a 09 09 6c 65 t().await {...le
1ec0: 74 20 75 70 64 61 74 65 20 3d 20 75 70 64 61 74 t update = updat
1ed0: 65 3f 3b 0a 09 09 6d 61 74 63 68 20 75 70 64 61 e?;...match upda
1ee0: 74 65 2e 6b 69 6e 64 20 7b 0a 09 09 09 55 70 64 te.kind {....Upd
1ef0: 61 74 65 4b 69 6e 64 3a 3a 4d 65 73 73 61 67 65 ateKind::Message
1f00: 28 6d 65 73 73 61 67 65 29 20 3d 3e 20 7b 0a 09 (message) => {..
1f10: 09 09 09 6c 65 74 20 6d 75 74 20 72 65 70 6c 79 ...let mut reply
1f20: 3a 20 56 65 63 3c 53 74 72 69 6e 67 3e 20 3d 20 : Vec<String> =
1f30: 76 65 63 21 5b 5d 3b 0a 09 09 09 09 6d 61 74 63 vec![];.....matc
1f40: 68 20 6d 65 73 73 61 67 65 2e 6b 69 6e 64 20 7b h message.kind {
1f50: 0a 09 09 09 09 09 4d 65 73 73 61 67 65 4b 69 6e ......MessageKin
1f60: 64 3a 3a 54 65 78 74 20 7b 20 72 65 66 20 64 61 d::Text { ref da
1f70: 74 61 2c 20 2e 2e 20 7d 20 3d 3e 20 7b 0a 09 09 ta, .. } => {...
1f80: 09 09 09 09 6c 65 74 20 6d 75 74 20 77 6f 72 64 ....let mut word
1f90: 73 20 3d 20 64 61 74 61 2e 73 70 6c 69 74 5f 77 s = data.split_w
1fa0: 68 69 74 65 73 70 61 63 65 28 29 3b 0a 09 09 09 hitespace();....
1fb0: 09 09 09 6c 65 74 20 63 6d 64 20 3d 20 77 6f 72 ...let cmd = wor
1fc0: 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 70 ds.next().unwrap
1fd0: 28 29 3b 0a 09 09 09 09 09 09 6d 61 74 63 68 20 ();.......match
1fe0: 63 6d 64 20 7b 0a 0a 2f 2f 20 73 74 61 72 74 0a cmd {..// start.
1ff0: 0a 09 09 09 09 09 09 09 22 2f 73 74 61 72 74 22 ........"/start"
2000: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 72 65 => {.........re
2010: 70 6c 79 2e 70 75 73 68 28 22 4e 6f 74 20 69 6e ply.push("Not in
2020: 20 73 65 72 76 69 63 65 20 79 65 74 2e 20 54 72 service yet. Tr
2030: 79 20 6c 61 74 65 72 2e 22 2e 74 6f 5f 73 74 72 y later.".to_str
2040: 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 7d ing());........}
2050: 2c 0a 0a 2f 2f 20 6c 69 73 74 0a 0a 09 09 09 09 ,..// list......
2060: 09 09 09 22 2f 6c 69 73 74 22 20 3d 3e 20 7b 0a ..."/list" => {.
2070: 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 ........reply.pu
2080: 73 68 28 22 43 68 61 6e 6e 65 6c 73 3a 22 2e 74 sh("Channels:".t
2090: 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 o_string());....
20a0: 09 09 09 09 09 6c 65 74 20 6d 75 74 20 72 6f 77 .....let mut row
20b0: 73 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 s = sqlx::query(
20c0: 22 73 65 6c 65 63 74 20 75 73 65 72 6e 61 6d 65 "select username
20d0: 2c 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 , enabled, url,
20e0: 69 76 5f 68 61 73 68 20 66 72 6f 6d 20 72 73 73 iv_hash from rss
20f0: 74 67 5f 73 6f 75 72 63 65 20 6c 65 66 74 20 6a tg_source left j
2100: 6f 69 6e 20 72 73 73 74 67 5f 63 68 61 6e 6e 65 oin rsstg_channe
2110: 6c 20 75 73 69 6e 67 20 28 63 68 61 6e 6e 65 6c l using (channel
2120: 5f 69 64 29 20 77 68 65 72 65 20 6f 77 6e 65 72 _id) where owner
2130: 20 3d 20 24 31 22 29 0a 09 09 09 09 09 09 09 09 = $1").........
2140: 09 2e 62 69 6e 64 28 69 36 34 3a 3a 66 72 6f 6d ..bind(i64::from
2150: 28 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 (message.from.id
2160: 29 29 0a 09 09 09 09 09 09 09 09 09 2e 66 65 74 ))...........fet
2170: 63 68 28 26 63 6f 72 65 2e 70 6f 6f 6c 29 3b 0a ch(&core.pool);.
2180: 09 09 09 09 09 09 09 09 77 68 69 6c 65 20 6c 65 ........while le
2190: 74 20 53 6f 6d 65 28 72 6f 77 29 20 3d 20 72 6f t Some(row) = ro
21a0: 77 73 2e 74 72 79 5f 6e 65 78 74 28 29 2e 61 77 ws.try_next().aw
21b0: 61 69 74 3f 20 7b 0a 09 09 09 09 09 09 09 09 09 ait? {..........
21c0: 6c 65 74 20 75 73 65 72 6e 61 6d 65 3a 20 26 73 let username: &s
21d0: 74 72 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 74 tr = row.try_get
21e0: 28 22 75 73 65 72 6e 61 6d 65 22 29 3f 3b 0a 09 ("username")?;..
21f0: 09 09 09 09 09 09 09 09 6c 65 74 20 65 6e 61 62 ........let enab
2200: 6c 65 64 3a 20 62 6f 6f 6c 20 3d 20 72 6f 77 2e led: bool = row.
2210: 74 72 79 5f 67 65 74 28 22 65 6e 61 62 6c 65 64 try_get("enabled
2220: 22 29 3f 3b 0a 09 09 09 09 09 09 09 09 09 6c 65 ")?;..........le
2230: 74 20 75 72 6c 3a 20 26 73 74 72 20 3d 20 72 6f t url: &str = ro
2240: 77 2e 74 72 79 5f 67 65 74 28 22 75 72 6c 22 29 w.try_get("url")
2250: 3f 3b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 20 ?;..........let
2260: 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c iv_hash: Option<
2270: 26 73 74 72 3e 20 3d 20 72 6f 77 2e 74 72 79 5f &str> = row.try_
2280: 67 65 74 28 22 69 76 5f 68 61 73 68 22 29 3f 3b get("iv_hash")?;
2290: 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e ..........reply.
22a0: 70 75 73 68 28 66 6f 72 6d 61 74 21 28 22 5c 6e push(format!("\n
22b0: 5c 5c 2a ef b8 8f e2 83 a3 20 60 7b 7d 60 20 7b \\*ļøā£ `{}` {
22c0: 7d 5c 6e f0 9f 94 97 20 60 7b 7d 60 22 2c 20 75 }\nš `{}`", u
22d0: 73 65 72 6e 61 6d 65 2c 20 20 0a 09 09 09 09 09 sername, ......
22e0: 09 09 09 09 09 6d 61 74 63 68 20 65 6e 61 62 6c .....match enabl
22f0: 65 64 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 ed {............
2300: 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 84 20 65 true => "š e
2310: 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 09 09 09 nabled",........
2320: 09 09 09 09 66 61 6c 73 65 20 3d 3e 20 22 e2 9b ....false => "ā
2330: 94 20 64 69 73 61 62 6c 65 64 22 2c 0a 09 09 09 disabled",....
2340: 09 09 09 09 09 09 09 7d 2c 20 75 72 6c 29 29 3b .......}, url));
2350: 0a 09 09 09 09 09 09 09 09 09 69 66 20 6c 65 74 ..........if let
2360: 20 53 6f 6d 65 28 68 61 73 68 29 20 3d 20 69 76 Some(hash) = iv
2370: 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 09 09 09 _hash {.........
2380: 09 09 72 65 70 6c 79 2e 70 75 73 68 28 66 6f 72 ..reply.push(for
2390: 6d 61 74 21 28 22 49 56 20 60 7b 7d 60 22 2c 20 mat!("IV `{}`",
23a0: 68 61 73 68 29 29 3b 0a 09 09 09 09 09 09 09 09 hash));.........
23b0: 09 7d 0a 09 09 09 09 09 09 09 09 7d 0a 09 09 09 .}.........}....
23c0: 09 09 09 09 7d 2c 0a 0a 2f 2f 20 61 64 64 0a 0a ....},..// add..
23d0: 09 09 09 09 09 09 09 22 2f 61 64 64 22 20 3d 3e ......."/add" =>
23e0: 20 7b 0a 09 09 09 09 09 09 09 09 6c 65 74 20 28 {.........let (
23f0: 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 2c 20 69 76 channel, url, iv
2400: 5f 68 61 73 68 29 20 3d 20 28 77 6f 72 64 73 2e _hash) = (words.
2410: 6e 65 78 74 28 29 2e 75 6e 77 72 61 70 28 29 2c next().unwrap(),
2420: 20 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e words.next().un
2430: 77 72 61 70 28 29 2c 20 77 6f 72 64 73 2e 6e 65 wrap(), words.ne
2440: 78 74 28 29 29 3b 0a 09 09 09 09 09 09 09 09 6c xt());.........l
2450: 65 74 20 6f 6b 5f 6c 69 6e 6b 20 3d 20 72 65 5f et ok_link = re_
2460: 6c 69 6e 6b 2e 69 73 5f 6d 61 74 63 68 28 26 75 link.is_match(&u
2470: 72 6c 29 3b 0a 09 09 09 09 09 09 09 09 6c 65 74 rl);.........let
2480: 20 6f 6b 5f 68 61 73 68 20 3d 20 6d 61 74 63 68 ok_hash = match
2490: 20 69 76 5f 68 61 73 68 20 7b 0a 09 09 09 09 09 iv_hash {......
24a0: 09 09 09 09 53 6f 6d 65 28 68 61 73 68 29 20 3d ....Some(hash) =
24b0: 3e 20 72 65 5f 69 76 5f 68 61 73 68 2e 69 73 5f > re_iv_hash.is_
24c0: 6d 61 74 63 68 28 26 68 61 73 68 29 2c 0a 09 09 match(&hash),...
24d0: 09 09 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 74 .......None => t
24e0: 72 75 65 2c 0a 09 09 09 09 09 09 09 09 7d 3b 0a rue,.........};.
24f0: 09 09 09 09 09 09 09 09 69 66 20 21 20 6f 6b 5f ........if ! ok_
2500: 6c 69 6e 6b 20 7b 0a 09 09 09 09 09 09 09 09 09 link {..........
2510: 72 65 70 6c 79 2e 70 75 73 68 28 22 4c 69 6e 6b reply.push("Link
2520: 20 73 68 6f 75 6c 64 20 62 65 20 6c 69 6e 6b 20 should be link
2530: 74 6f 20 61 74 6f 6d 2f 72 73 73 20 66 65 65 64 to atom/rss feed
2540: 2c 20 73 6f 6d 65 74 68 69 6e 67 20 6c 69 6b 65 , something like
2550: 20 5c 22 68 74 74 70 73 3a 2f 2f 64 6f 6d 61 69 \"https://domai
2560: 6e 2f 70 61 74 68 5c 22 5c 5c 2e 22 2e 74 6f 5f n/path\"\\.".to_
2570: 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 string());......
2580: 09 09 09 09 63 6f 72 65 2e 64 65 62 75 67 28 26 ....core.debug(&
2590: 66 6f 72 6d 61 74 21 28 22 55 72 6c 3a 20 7b 3a format!("Url: {:
25a0: 3f 7d 22 2c 20 26 75 72 6c 29 29 3f 3b 0a 09 09 ?}", &url))?;...
25b0: 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 09 09 ......}.........
25c0: 69 66 20 21 20 6f 6b 5f 68 61 73 68 20 7b 0a 09 if ! ok_hash {..
25d0: 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 ........reply.pu
25e0: 73 68 28 22 49 56 20 68 61 73 68 20 73 68 6f 75 sh("IV hash shou
25f0: 6c 64 20 62 65 20 31 34 20 68 65 78 20 64 69 67 ld be 14 hex dig
2600: 69 74 73 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 its.".to_string(
2610: 29 29 3b 0a 09 09 09 09 09 09 09 09 09 63 6f 72 ));..........cor
2620: 65 2e 64 65 62 75 67 28 26 66 6f 72 6d 61 74 21 e.debug(&format!
2630: 28 22 49 56 3a 20 7b 3a 3f 7d 22 2c 20 26 69 76 ("IV: {:?}", &iv
2640: 5f 68 61 73 68 29 29 3f 3b 0a 09 09 09 09 09 09 _hash))?;.......
2650: 09 09 7d 0a 09 09 09 09 09 09 09 09 69 66 20 6f ..}.........if o
2660: 6b 5f 6c 69 6e 6b 20 26 26 20 6f 6b 5f 68 61 73 k_link && ok_has
2670: 68 20 7b 0a 09 09 09 09 09 09 09 09 09 6c 65 74 h {..........let
2680: 20 63 68 61 6e 3a 20 4f 70 74 69 6f 6e 3c 69 36 chan: Option<i6
2690: 34 3e 20 3d 20 6d 61 74 63 68 20 73 71 6c 78 3a 4> = match sqlx:
26a0: 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 20 63 :query("select c
26b0: 68 61 6e 6e 65 6c 5f 69 64 20 66 72 6f 6d 20 72 hannel_id from r
26c0: 73 73 74 67 5f 63 68 61 6e 6e 65 6c 20 77 68 65 sstg_channel whe
26d0: 72 65 20 75 73 65 72 6e 61 6d 65 20 3d 20 24 31 re username = $1
26e0: 22 29 0a 09 09 09 09 09 09 09 09 09 09 2e 62 69 ")............bi
26f0: 6e 64 28 63 68 61 6e 6e 65 6c 29 0a 09 09 09 09 nd(channel).....
2700: 09 09 09 09 09 09 2e 66 65 74 63 68 5f 6f 6e 65 .......fetch_one
2710: 28 26 63 6f 72 65 2e 70 6f 6f 6c 29 2e 61 77 61 (&core.pool).awa
2720: 69 74 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 it {............
2730: 4f 6b 28 63 68 61 6e 29 20 3d 3e 20 53 6f 6d 65 Ok(chan) => Some
2740: 28 63 68 61 6e 2e 74 72 79 5f 67 65 74 28 22 63 (chan.try_get("c
2750: 68 61 6e 6e 65 6c 5f 69 64 22 29 3f 29 2c 0a 09 hannel_id")?),..
2760: 09 09 09 09 09 09 09 09 09 09 45 72 72 28 73 71 ..........Err(sq
2770: 6c 78 3a 3a 45 72 72 6f 72 3a 3a 52 6f 77 4e 6f lx::Error::RowNo
2780: 74 46 6f 75 6e 64 29 20 3d 3e 20 7b 0a 09 09 09 tFound) => {....
2790: 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 .........reply.p
27a0: 75 73 68 28 22 53 6f 72 72 79 2c 20 49 20 64 6f ush("Sorry, I do
27b0: 6e 27 74 20 6b 6e 6f 77 20 61 62 6f 75 74 20 74 n't know about t
27c0: 68 61 74 20 63 68 61 6e 6e 65 6c 2e 20 50 6c 65 hat channel. Ple
27d0: 61 73 65 2c 20 61 64 64 20 61 20 63 68 61 6e 6e ase, add a chann
27e0: 65 6c 20 77 69 74 68 20 2f 61 64 64 63 68 61 6e el with /addchan
27f0: 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b .".to_string());
2800: 0a 09 09 09 09 09 09 09 09 09 09 09 09 4e 6f 6e .............Non
2810: 65 0a 09 09 09 09 09 09 09 09 09 09 09 7d 2c 0a e............},.
2820: 09 09 09 09 09 09 09 09 09 09 09 45 72 72 28 65 ...........Err(e
2830: 72 72 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 rr) => {........
2840: 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 .....reply.push(
2850: 22 53 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 "Sorry, unknown
2860: 65 72 72 6f 72 5c 5c 2e 22 2e 74 6f 5f 73 74 72 error\\.".to_str
2870: 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 ing());.........
2880: 09 09 09 09 63 6f 72 65 2e 64 65 62 75 67 28 26 ....core.debug(&
2890: 66 6f 72 6d 61 74 21 28 22 53 6f 72 72 79 2c 20 format!("Sorry,
28a0: 75 6e 6b 6e 6f 77 6e 20 65 72 72 6f 72 3a 20 7b unknown error: {
28b0: 3a 23 3f 7d 5c 6e 22 2c 20 65 72 72 29 29 3f 3b :#?}\n", err))?;
28c0: 0a 09 09 09 09 09 09 09 09 09 09 09 09 4e 6f 6e .............Non
28d0: 65 0a 09 09 09 09 09 09 09 09 09 09 09 7d 2c 0a e............},.
28e0: 09 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 .........};.....
28f0: 09 09 09 09 09 6d 61 74 63 68 20 63 68 61 6e 20 .....match chan
2900: 7b 0a 09 09 09 09 09 09 09 09 09 09 53 6f 6d 65 {...........Some
2910: 28 63 68 61 6e 29 20 3d 3e 20 7b 0a 09 09 09 09 (chan) => {.....
2920: 09 09 09 09 09 09 09 6d 61 74 63 68 20 73 71 6c .......match sql
2930: 78 3a 3a 71 75 65 72 79 28 22 69 6e 73 65 72 74 x::query("insert
2940: 20 69 6e 74 6f 20 72 73 73 74 67 5f 73 6f 75 72 into rsstg_sour
2950: 63 65 20 28 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 ce (channel_id,
2960: 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 url, iv_hash, ow
2970: 6e 65 72 29 20 76 61 6c 75 65 73 20 28 24 31 2c ner) values ($1,
2980: 20 24 32 2c 20 24 33 2c 20 24 34 29 20 6f 6e 20 $2, $3, $4) on
2990: 63 6f 6e 66 6c 69 63 74 20 28 63 68 61 6e 6e 65 conflict (channe
29a0: 6c 5f 69 64 2c 20 6f 77 6e 65 72 29 20 64 6f 20 l_id, owner) do
29b0: 75 70 64 61 74 65 20 73 65 74 20 75 72 6c 20 3d update set url =
29c0: 20 65 78 63 6c 75 64 65 64 2e 75 72 6c 2c 20 69 excluded.url, i
29d0: 76 5f 68 61 73 68 20 3d 20 65 78 63 6c 75 64 65 v_hash = exclude
29e0: 64 2e 69 76 5f 68 61 73 68 3b 22 29 0a 09 09 09 d.iv_hash;")....
29f0: 09 09 09 09 09 09 09 09 09 2e 62 69 6e 64 28 63 ..........bind(c
2a00: 68 61 6e 29 0a 09 09 09 09 09 09 09 09 09 09 09 han)............
2a10: 09 2e 62 69 6e 64 28 75 72 6c 29 0a 09 09 09 09 ..bind(url).....
2a20: 09 09 09 09 09 09 09 09 2e 62 69 6e 64 28 69 76 .........bind(iv
2a30: 5f 68 61 73 68 29 0a 09 09 09 09 09 09 09 09 09 _hash)..........
2a40: 09 09 09 2e 62 69 6e 64 28 69 36 34 3a 3a 66 72 ....bind(i64::fr
2a50: 6f 6d 28 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e om(message.from.
2a60: 69 64 29 29 0a 09 09 09 09 09 09 09 09 09 09 09 id))............
2a70: 09 2e 65 78 65 63 75 74 65 28 26 63 6f 72 65 2e ..execute(&core.
2a80: 70 6f 6f 6c 29 2e 61 77 61 69 74 20 7b 0a 09 09 pool).await {...
2a90: 09 09 09 09 09 09 09 09 09 09 4f 6b 28 5f 29 20 ..........Ok(_)
2aa0: 3d 3e 20 72 65 70 6c 79 2e 70 75 73 68 28 22 43 => reply.push("C
2ab0: 68 61 6e 6e 65 6c 20 61 64 64 65 64 5c 5c 2e 22 hannel added\\."
2ac0: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 2c 0a 09 .to_string()),..
2ad0: 09 09 09 09 09 09 09 09 09 09 09 45 72 72 28 73 ...........Err(s
2ae0: 71 6c 78 3a 3a 45 72 72 6f 72 3a 3a 44 61 74 61 qlx::Error::Data
2af0: 62 61 73 65 28 65 72 72 29 29 20 3d 3e 20 7b 0a base(err)) => {.
2b00: 09 09 09 09 09 09 09 09 09 09 09 09 09 6d 61 74 .............mat
2b10: 63 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 74 3a ch err.downcast:
2b20: 3a 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 65 73 :<sqlx::postgres
2b30: 3a 3a 50 67 44 61 74 61 62 61 73 65 45 72 72 6f ::PgDatabaseErro
2b40: 72 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 20 7b r>().routine() {
2b50: 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 09 53 ...............S
2b60: 6f 6d 65 28 22 5f 62 74 5f 63 68 65 63 6b 5f 75 ome("_bt_check_u
2b70: 6e 69 71 75 65 22 2c 20 29 20 3d 3e 20 7b 0a 09 nique", ) => {..
2b80: 09 09 09 09 09 09 09 09 09 09 09 09 09 09 72 65 ..............re
2b90: 70 6c 79 2e 70 75 73 68 28 22 44 75 70 6c 69 63 ply.push("Duplic
2ba0: 61 74 65 20 6b 65 79 5c 5c 2e 22 2e 74 6f 5f 73 ate key\\.".to_s
2bb0: 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 tring());.......
2bc0: 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ........},......
2bd0: 09 09 09 09 09 09 09 09 09 53 6f 6d 65 28 5f 29 .........Some(_)
2be0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 09 => {...........
2bf0: 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 68 28 .....reply.push(
2c00: 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 5c "Database error\
2c10: 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 \.".to_string())
2c20: 3b 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 09 ;...............
2c30: 7d 2c 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 },..............
2c40: 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 09 09 .None => {......
2c50: 09 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e ..........reply.
2c60: 70 75 73 68 28 22 4e 6f 20 64 61 74 61 62 61 73 push("No databas
2c70: 65 20 65 72 72 6f 72 20 65 78 74 72 61 63 74 65 e error extracte
2c80: 64 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 d\\.".to_string(
2c90: 29 29 3b 0a 09 09 09 09 09 09 09 09 09 09 09 09 ));.............
2ca0: 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 09 09 09 ..},............
2cb0: 09 09 7d 3b 0a 09 09 09 09 09 09 09 09 09 09 09 ..};............
2cc0: 09 7d 2c 0a 09 09 09 09 09 09 09 09 09 09 09 09 .},.............
2cd0: 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 Err(err) => {...
2ce0: 09 09 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 ...........reply
2cf0: 2e 70 75 73 68 28 22 53 6f 72 72 79 2c 20 75 6e .push("Sorry, un
2d00: 6b 6e 6f 77 6e 20 65 72 72 6f 72 5c 5c 2e 22 2e known error\\.".
2d10: 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 to_string());...
2d20: 09 09 09 09 09 09 09 09 09 09 09 63 6f 72 65 2e ...........core.
2d30: 64 65 62 75 67 28 26 66 6f 72 6d 61 74 21 28 22 debug(&format!("
2d40: 53 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 Sorry, unknown e
2d50: 72 72 6f 72 3a 20 7b 3a 23 3f 7d 5c 6e 22 2c 20 rror: {:#?}\n",
2d60: 65 72 72 29 29 3f 3b 0a 09 09 09 09 09 09 09 09 err))?;.........
2d70: 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 09 ....},..........
2d80: 09 09 7d 3b 0a 09 09 09 09 09 09 09 09 09 09 7d ..};...........}
2d90: 2c 0a 09 09 09 09 09 09 09 09 09 09 4e 6f 6e 65 ,...........None
2da0: 20 3d 3e 20 7b 7d 2c 0a 09 09 09 09 09 09 09 09 => {},.........
2db0: 09 7d 3b 0a 09 09 09 09 09 09 09 09 7d 3b 0a 09 .};.........};..
2dc0: 09 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 61 64 64 ......},..// add
2dd0: 63 68 61 6e 0a 0a 09 09 09 09 09 09 09 22 2f 61 chan........."/a
2de0: 64 64 63 68 61 6e 22 20 3d 3e 20 7b 0a 09 09 09 ddchan" => {....
2df0: 09 09 09 09 09 6c 65 74 20 63 68 61 6e 6e 65 6c .....let channel
2e00: 20 3d 20 77 6f 72 64 73 2e 6e 65 78 74 28 29 2e = words.next().
2e10: 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 09 09 unwrap();.......
2e20: 09 09 69 66 20 21 20 72 65 5f 75 73 65 72 6e 61 ..if ! re_userna
2e30: 6d 65 2e 69 73 5f 6d 61 74 63 68 28 26 63 68 61 me.is_match(&cha
2e40: 6e 6e 65 6c 29 20 7b 0a 09 09 09 09 09 09 09 09 nnel) {.........
2e50: 09 72 65 70 6c 79 2e 70 75 73 68 28 22 55 73 65 .reply.push("Use
2e60: 72 6e 61 6d 65 73 20 73 68 6f 75 6c 64 20 62 65 rnames should be
2e70: 20 73 6f 6d 65 74 68 69 6e 67 20 6c 69 6b 65 20 something like
2e80: 5c 22 40 5c 5c 5b 61 5c 5c 2d 7a 41 5c 5c 2d 5a \"@\\[a\\-zA\\-Z
2e90: 5d 5c 5c 5b 61 5c 5c 2d 7a 41 5c 5c 2d 5a 30 5c ]\\[a\\-zA\\-Z0\
2ea0: 5c 2d 39 5c 5c 5f 5d 2b 5c 22 2c 20 61 72 65 6e \-9\\_]+\", aren
2eb0: 27 74 20 74 68 65 79 3f 22 2e 74 6f 5f 73 74 72 't they?".to_str
2ec0: 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 ing());.........
2ed0: 7d 20 65 6c 73 65 20 7b 0a 09 09 09 09 09 09 09 } else {........
2ee0: 09 09 6c 65 74 20 63 68 61 6e 3a 20 4f 70 74 69 ..let chan: Opti
2ef0: 6f 6e 3c 69 36 34 3e 20 3d 20 6d 61 74 63 68 20 on<i64> = match
2f00: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 73 65 6c sqlx::query("sel
2f10: 65 63 74 20 63 68 61 6e 6e 65 6c 5f 69 64 20 66 ect channel_id f
2f20: 72 6f 6d 20 72 73 73 74 67 5f 63 68 61 6e 6e 65 rom rsstg_channe
2f30: 6c 20 77 68 65 72 65 20 75 73 65 72 6e 61 6d 65 l where username
2f40: 20 3d 20 24 31 22 29 0a 09 09 09 09 09 09 09 09 = $1").........
2f50: 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c 29 ...bind(channel)
2f60: 0a 09 09 09 09 09 09 09 09 09 09 2e 66 65 74 63 ............fetc
2f70: 68 5f 6f 6e 65 28 26 63 6f 72 65 2e 70 6f 6f 6c h_one(&core.pool
2f80: 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 09 ).await {.......
2f90: 09 09 09 09 09 4f 6b 28 63 68 61 6e 29 20 3d 3e .....Ok(chan) =>
2fa0: 20 53 6f 6d 65 28 63 68 61 6e 2e 74 72 79 5f 67 Some(chan.try_g
2fb0: 65 74 28 22 63 68 61 6e 6e 65 6c 5f 69 64 22 29 et("channel_id")
2fc0: 3f 29 2c 0a 09 09 09 09 09 09 09 09 09 09 09 45 ?),............E
2fd0: 72 72 28 73 71 6c 78 3a 3a 45 72 72 6f 72 3a 3a rr(sqlx::Error::
2fe0: 52 6f 77 4e 6f 74 46 6f 75 6e 64 29 20 3d 3e 20 RowNotFound) =>
2ff0: 4e 6f 6e 65 2c 0a 09 09 09 09 09 09 09 09 09 09 None,...........
3000: 09 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 .Err(err) => {..
3010: 09 09 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 ...........reply
3020: 2e 70 75 73 68 28 22 53 6f 72 72 79 2c 20 75 6e .push("Sorry, un
3030: 6b 6e 6f 77 6e 20 65 72 72 6f 72 5c 5c 2e 22 2e known error\\.".
3040: 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 to_string());...
3050: 09 09 09 09 09 09 09 09 09 09 63 6f 72 65 2e 64 ..........core.d
3060: 65 62 75 67 28 26 66 6f 72 6d 61 74 21 28 22 53 ebug(&format!("S
3070: 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 72 orry, unknown er
3080: 72 6f 72 3a 20 7b 3a 23 3f 7d 5c 6e 22 2c 20 65 ror: {:#?}\n", e
3090: 72 72 29 29 3f 3b 0a 09 09 09 09 09 09 09 09 09 rr))?;..........
30a0: 09 09 09 4e 6f 6e 65 0a 09 09 09 09 09 09 09 09 ...None.........
30b0: 09 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 09 7d ...},..........}
30c0: 3b 0a 09 09 09 09 09 09 09 09 09 6d 61 74 63 68 ;..........match
30d0: 20 63 68 61 6e 20 7b 0a 09 09 09 09 09 09 09 09 chan {.........
30e0: 09 09 53 6f 6d 65 28 63 68 61 6e 29 20 3d 3e 20 ..Some(chan) =>
30f0: 7b 0a 09 09 09 09 09 09 09 09 09 09 09 6c 65 74 {............let
3100: 20 6e 65 77 5f 63 68 61 6e 20 3d 20 63 6f 72 65 new_chan = core
3110: 2e 74 67 2e 73 65 6e 64 28 74 65 6c 65 67 72 61 .tg.send(telegra
3120: 6d 5f 62 6f 74 3a 3a 47 65 74 43 68 61 74 3a 3a m_bot::GetChat::
3130: 6e 65 77 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 new(telegram_bot
3140: 3a 3a 74 79 70 65 73 3a 3a 43 68 61 74 49 64 3a ::types::ChatId:
3150: 3a 6e 65 77 28 63 68 61 6e 29 29 29 2e 61 77 61 :new(chan))).awa
3160: 69 74 3f 3b 0a 09 09 09 09 09 09 09 09 09 09 09 it?;............
3170: 69 66 20 69 36 34 3a 3a 66 72 6f 6d 28 6e 65 77 if i64::from(new
3180: 5f 63 68 61 6e 2e 69 64 28 29 29 20 3d 3d 20 63 _chan.id()) == c
3190: 68 61 6e 20 7b 0a 09 09 09 09 09 09 09 09 09 09 han {...........
31a0: 09 09 72 65 70 6c 79 2e 70 75 73 68 28 22 49 20 ..reply.push("I
31b0: 61 6c 72 65 61 64 79 20 6b 6e 6f 77 20 74 68 61 already know tha
31c0: 74 20 63 68 61 6e 6e 65 6c 5c 5c 2e 22 2e 74 6f t channel\\.".to
31d0: 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 _string());.....
31e0: 09 09 09 09 09 09 09 7d 20 65 6c 73 65 20 7b 0a .......} else {.
31f0: 09 09 09 09 09 09 09 09 09 09 09 09 72 65 70 6c ............repl
3200: 79 2e 70 75 73 68 28 22 48 6d 6d 2c 20 63 68 61 y.push("Hmm, cha
3210: 6e 6e 65 6c 20 68 61 73 20 63 68 61 6e 67 65 64 nnel has changed
3220: e2 80 a6 20 49 27 6c 6c 20 66 69 78 20 69 74 20 ā¦ I'll fix it
3230: 6c 61 74 65 72 5c 5c 2e 22 2e 74 6f 5f 73 74 72 later\\.".to_str
3240: 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 ing());.........
3250: 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 09 09 09 ...};...........
3260: 7d 2c 0a 09 09 09 09 09 09 09 09 09 09 4e 6f 6e },...........Non
3270: 65 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 e => {..........
3280: 09 09 6d 61 74 63 68 20 63 6f 72 65 2e 74 67 2e ..match core.tg.
3290: 73 65 6e 64 28 74 65 6c 65 67 72 61 6d 5f 62 6f send(telegram_bo
32a0: 74 3a 3a 47 65 74 43 68 61 74 41 64 6d 69 6e 69 t::GetChatAdmini
32b0: 73 74 72 61 74 6f 72 73 3a 3a 6e 65 77 28 74 65 strators::new(te
32c0: 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 legram_bot::type
32d0: 73 3a 3a 43 68 61 74 52 65 66 3a 3a 43 68 61 6e s::ChatRef::Chan
32e0: 6e 65 6c 55 73 65 72 6e 61 6d 65 28 63 68 61 6e nelUsername(chan
32f0: 6e 65 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 nel.to_string())
3300: 29 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 09 )).await {......
3310: 09 09 09 09 09 09 09 4f 6b 28 63 68 61 6e 5f 61 .......Ok(chan_a
3320: 64 6d 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 dm) => {........
3330: 09 09 09 09 09 09 6c 65 74 20 28 6d 75 74 20 6d ......let (mut m
3340: 65 2c 20 6d 75 74 20 75 73 65 72 29 20 3d 20 28 e, mut user) = (
3350: 66 61 6c 73 65 2c 20 66 61 6c 73 65 29 3b 0a 09 false, false);..
3360: 09 09 09 09 09 09 09 09 09 09 09 09 66 6f 72 20 ............for
3370: 61 64 6d 69 6e 20 69 6e 20 26 63 68 61 6e 5f 61 admin in &chan_a
3380: 64 6d 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 dm {............
3390: 09 09 09 69 66 20 61 64 6d 69 6e 2e 75 73 65 72 ...if admin.user
33a0: 2e 69 64 20 3d 3d 20 63 6f 72 65 2e 6d 79 2e 69 .id == core.my.i
33b0: 64 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 09 d {.............
33c0: 09 09 09 6d 65 20 3d 20 74 72 75 65 3b 0a 09 09 ...me = true;...
33d0: 09 09 09 09 09 09 09 09 09 09 09 09 7d 3b 0a 09 ............};..
33e0: 09 09 09 09 09 09 09 09 09 09 09 09 09 69 66 20 .............if
33f0: 61 64 6d 69 6e 2e 75 73 65 72 2e 69 64 20 3d 3d admin.user.id ==
3400: 20 6d 65 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 message.from.id
3410: 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 {..............
3420: 09 09 75 73 65 72 20 3d 20 74 72 75 65 3b 0a 09 ..user = true;..
3430: 09 09 09 09 09 09 09 09 09 09 09 09 09 7d 3b 0a .............};.
3440: 09 09 09 09 09 09 09 09 09 09 09 09 09 7d 3b 0a .............};.
3450: 09 09 09 09 09 09 09 09 09 09 09 09 09 69 66 20 .............if
3460: 21 20 6d 65 20 20 20 7b 20 72 65 70 6c 79 2e 70 ! me { reply.p
3470: 75 73 68 28 22 49 20 6e 65 65 64 20 74 6f 20 62 ush("I need to b
3480: 65 20 61 64 6d 69 6e 20 6f 6e 20 74 68 61 74 20 e admin on that
3490: 63 68 61 6e 6e 65 6c 5c 5c 2e 22 2e 74 6f 5f 73 channel\\.".to_s
34a0: 74 72 69 6e 67 28 29 29 3b 20 7d 3b 0a 09 09 09 tring()); };....
34b0: 09 09 09 09 09 09 09 09 09 09 69 66 20 21 20 75 ..........if ! u
34c0: 73 65 72 20 7b 20 72 65 70 6c 79 2e 70 75 73 68 ser { reply.push
34d0: 28 22 59 6f 75 20 73 68 6f 75 6c 64 20 62 65 20 ("You should be
34e0: 61 64 6d 69 6e 20 6f 6e 20 74 68 61 74 20 63 68 admin on that ch
34f0: 61 6e 6e 65 6c 5c 5c 2e 22 2e 74 6f 5f 73 74 72 annel\\.".to_str
3500: 69 6e 67 28 29 29 3b 20 7d 3b 0a 09 09 09 09 09 ing()); };......
3510: 09 09 09 09 09 09 09 09 69 66 20 6d 65 20 26 26 ........if me &&
3520: 20 75 73 65 72 20 7b 0a 09 09 09 09 09 09 09 09 user {.........
3530: 09 09 09 09 09 09 6c 65 74 20 63 68 61 6e 5f 69 ......let chan_i
3540: 64 20 3d 20 63 6f 72 65 2e 74 67 2e 73 65 6e 64 d = core.tg.send
3550: 28 74 65 6c 65 67 72 61 6d 5f 62 6f 74 3a 3a 47 (telegram_bot::G
3560: 65 74 43 68 61 74 3a 3a 6e 65 77 28 74 65 6c 65 etChat::new(tele
3570: 67 72 61 6d 5f 62 6f 74 3a 3a 74 79 70 65 73 3a gram_bot::types:
3580: 3a 43 68 61 74 52 65 66 3a 3a 43 68 61 6e 6e 65 :ChatRef::Channe
3590: 6c 55 73 65 72 6e 61 6d 65 28 63 68 61 6e 6e 65 lUsername(channe
35a0: 6c 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 29 29 l.to_string())))
35b0: 2e 61 77 61 69 74 3f 3b 0a 09 09 09 09 09 09 09 .await?;........
35c0: 09 09 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 .......sqlx::que
35d0: 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 ry("insert into
35e0: 72 73 73 74 67 5f 63 68 61 6e 6e 65 6c 20 28 63 rsstg_channel (c
35f0: 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 73 65 72 6e hannel_id, usern
3600: 61 6d 65 29 20 76 61 6c 75 65 73 20 28 24 31 2c ame) values ($1,
3610: 20 24 32 29 3b 22 29 0a 09 09 09 09 09 09 09 09 $2);").........
3620: 09 09 09 09 09 09 09 2e 62 69 6e 64 28 69 36 34 ........bind(i64
3630: 3a 3a 66 72 6f 6d 28 63 68 61 6e 5f 69 64 2e 69 ::from(chan_id.i
3640: 64 28 29 29 29 0a 09 09 09 09 09 09 09 09 09 09 d()))...........
3650: 09 09 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e ......bind(chann
3660: 65 6c 29 0a 09 09 09 09 09 09 09 09 09 09 09 09 el).............
3670: 09 09 09 2e 65 78 65 63 75 74 65 28 26 63 6f 72 ....execute(&cor
3680: 65 2e 70 6f 6f 6c 29 2e 61 77 61 69 74 3f 3b 0a e.pool).await?;.
3690: 09 09 09 09 09 09 09 09 09 09 09 09 09 09 72 65 ..............re
36a0: 70 6c 79 2e 70 75 73 68 28 22 47 6f 6f 64 2c 20 ply.push("Good,
36b0: 49 20 6b 6e 6f 77 20 74 68 61 74 20 63 68 61 6e I know that chan
36c0: 6e 65 6c 20 6e 6f 77 5c 5c 2e 5c 6e 22 2e 74 6f nel now\\.\n".to
36d0: 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 _string());.....
36e0: 09 09 09 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 .........};.....
36f0: 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ........},......
3700: 09 09 09 09 09 09 09 45 72 72 28 5f 29 20 3d 3e .......Err(_) =>
3710: 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 09 09 {..............
3720: 72 65 70 6c 79 2e 70 75 73 68 28 22 53 6f 72 72 reply.push("Sorr
3730: 79 2c 20 49 20 68 61 76 65 20 6e 6f 20 61 63 63 y, I have no acc
3740: 65 73 73 20 74 6f 20 74 68 61 74 20 63 68 61 74 ess to that chat
3750: 5c 5c 2e 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 \\.".to_string()
3760: 29 3b 0a 09 09 09 09 09 09 09 09 09 09 09 09 7d );.............}
3770: 2c 0a 09 09 09 09 09 09 09 09 09 09 09 7d 3b 0a ,............};.
3780: 09 09 09 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 ..........},....
3790: 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 09 09 ......};........
37a0: 09 7d 3b 0a 09 09 09 09 09 09 09 7d 2c 0a 0a 2f .};........},../
37b0: 2f 20 63 68 65 63 6b 0a 0a 09 09 09 09 09 09 09 / check.........
37c0: 22 2f 63 68 65 63 6b 22 20 3d 3e 20 7b 0a 09 09 "/check" => {...
37d0: 09 09 09 09 09 09 6c 65 74 20 63 68 61 6e 6e 65 ......let channe
37e0: 6c 20 3d 20 77 6f 72 64 73 2e 6e 65 78 74 28 29 l = words.next()
37f0: 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 09 .unwrap();......
3800: 09 09 09 69 66 20 21 20 72 65 5f 75 73 65 72 6e ...if ! re_usern
3810: 61 6d 65 2e 69 73 5f 6d 61 74 63 68 28 26 63 68 ame.is_match(&ch
3820: 61 6e 6e 65 6c 29 20 7b 0a 09 09 09 09 09 09 09 annel) {........
3830: 09 09 72 65 70 6c 79 2e 70 75 73 68 28 22 55 73 ..reply.push("Us
3840: 65 72 6e 61 6d 65 73 20 73 68 6f 75 6c 64 20 62 ernames should b
3850: 65 20 73 6f 6d 65 74 68 69 6e 67 20 6c 69 6b 65 e something like
3860: 20 5c 22 40 5c 5c 5b 61 2d 7a 5d 5c 5c 5b 61 2d \"@\\[a-z]\\[a-
3870: 7a 30 2d 39 5f 5d 2b 5c 22 2c 20 61 72 65 6e 27 z0-9_]+\", aren'
3880: 74 20 74 68 65 79 3f 22 2e 74 6f 5f 73 74 72 69 t they?".to_stri
3890: 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 09 09 7d ng());.........}
38a0: 20 65 6c 73 65 20 7b 0a 09 09 09 09 09 09 09 09 else {.........
38b0: 09 26 63 6f 72 65 2e 63 68 65 63 6b 28 63 68 61 .&core.check(cha
38c0: 6e 6e 65 6c 2c 20 4e 6f 6e 65 29 2e 61 77 61 69 nnel, None).awai
38d0: 74 3f 3b 0a 09 09 09 09 09 09 09 09 7d 0a 09 09 t?;.........}...
38e0: 09 09 09 09 09 7d 2c 0a 0a 2f 2f 20 63 6c 65 61 .....},..// clea
38f0: 72 0a 0a 09 09 09 09 09 09 09 22 2f 63 6c 65 61 r........."/clea
3900: 6e 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 n" => {.........
3910: 69 66 20 63 6f 72 65 2e 6f 77 6e 65 72 20 21 3d if core.owner !=
3920: 20 69 36 34 3a 3a 66 72 6f 6d 28 6d 65 73 73 61 i64::from(messa
3930: 67 65 2e 66 72 6f 6d 2e 69 64 29 20 7b 0a 09 09 ge.from.id) {...
3940: 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 75 73 .......reply.pus
3950: 68 28 22 52 65 73 65 72 76 65 64 20 66 6f 72 20 h("Reserved for
3960: 74 65 73 74 69 6e 67 5c 5c 2e 22 2e 74 6f 5f 73 testing\\.".to_s
3970: 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 tring());.......
3980: 09 09 7d 20 65 6c 73 65 20 7b 0a 09 09 09 09 09 ..} else {......
3990: 09 09 09 09 6c 65 74 20 73 6f 75 72 63 65 5f 69 ....let source_i
39a0: 64 20 3d 20 77 6f 72 64 73 2e 6e 65 78 74 28 29 d = words.next()
39b0: 2e 75 6e 77 72 61 70 28 29 2e 70 61 72 73 65 3a .unwrap().parse:
39c0: 3a 3c 69 33 32 3e 28 29 2e 75 6e 77 72 61 70 5f :<i32>().unwrap_
39d0: 6f 72 28 30 29 3b 0a 09 09 09 09 09 09 09 09 09 or(0);..........
39e0: 26 63 6f 72 65 2e 63 6c 65 61 6e 28 73 6f 75 72 &core.clean(sour
39f0: 63 65 5f 69 64 29 2e 61 77 61 69 74 3f 3b 0a 09 ce_id).await?;..
3a00: 09 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 09 .......}........
3a10: 7d 2c 0a 0a 2f 2f 20 65 6e 61 62 6c 65 0a 0a 09 },..// enable...
3a20: 09 09 09 09 09 09 22 2f 65 6e 61 62 6c 65 22 20 ......"/enable"
3a30: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 6c 65 74 => {.........let
3a40: 20 63 68 61 6e 6e 65 6c 20 3d 20 77 6f 72 64 73 channel = words
3a50: 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 70 28 29 .next().unwrap()
3a60: 3b 0a 09 09 09 09 09 09 09 09 69 66 20 21 20 72 ;.........if ! r
3a70: 65 5f 75 73 65 72 6e 61 6d 65 2e 69 73 5f 6d 61 e_username.is_ma
3a80: 74 63 68 28 26 63 68 61 6e 6e 65 6c 29 20 7b 0a tch(&channel) {.
3a90: 09 09 09 09 09 09 09 09 09 72 65 70 6c 79 2e 70 .........reply.p
3aa0: 75 73 68 28 22 55 73 65 72 6e 61 6d 65 73 20 73 ush("Usernames s
3ab0: 68 6f 75 6c 64 20 62 65 20 73 6f 6d 65 74 68 69 hould be somethi
3ac0: 6e 67 20 6c 69 6b 65 20 5c 22 40 5c 5c 5b 61 2d ng like \"@\\[a-
3ad0: 7a 5d 5c 5c 5b 61 2d 7a 30 2d 39 5f 5d 2b 5c 22 z]\\[a-z0-9_]+\"
3ae0: 2c 20 61 72 65 6e 27 74 20 74 68 65 79 3f 22 2e , aren't they?".
3af0: 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 09 to_string());...
3b00: 09 09 09 09 09 09 7d 20 65 6c 73 65 20 7b 0a 09 ......} else {..
3b10: 09 09 09 09 09 09 09 09 6d 61 74 63 68 20 63 6f ........match co
3b20: 72 65 2e 65 6e 61 62 6c 65 28 6d 65 73 73 61 67 re.enable(messag
3b30: 65 2e 66 72 6f 6d 2e 69 64 2c 20 63 68 61 6e 6e e.from.id, chann
3b40: 65 6c 29 2e 61 77 61 69 74 20 7b 0a 09 09 09 09 el).await {.....
3b50: 09 09 09 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 7b ......Ok(_) => {
3b60: 0a 09 09 09 09 09 09 09 09 09 09 09 72 65 70 6c ............repl
3b70: 79 2e 70 75 73 68 28 22 43 68 61 6e 6e 65 6c 20 y.push("Channel
3b80: 65 6e 61 62 6c 65 64 5c 5c 2e 22 2e 74 6f 5f 73 enabled\\.".to_s
3b90: 74 72 69 6e 67 28 29 29 3b 0a 09 09 09 09 09 09 tring());.......
3ba0: 09 09 09 09 7d 0a 09 09 09 09 09 09 09 09 09 09 ....}...........
3bb0: 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 Err(err) => {...
3bc0: 09 09 09 09 09 09 09 09 09 63 6f 72 65 2e 64 65 .........core.de
3bd0: 62 75 67 28 26 65 72 72 2e 74 6f 5f 73 74 72 69 bug(&err.to_stri
3be0: 6e 67 28 29 29 3f 3b 0a 09 09 09 09 09 09 09 09 ng())?;.........
3bf0: 09 09 7d 2c 0a 09 09 09 09 09 09 09 09 09 7d 0a ..},..........}.
3c00: 09 09 09 09 09 09 09 09 7d 0a 09 09 09 09 09 09 ........}.......
3c10: 09 7d 2c 0a 0a 2f 2f 20 64 69 73 61 62 6c 65 0a .},..// disable.
3c20: 0a 09 09 09 09 09 09 09 22 2f 64 69 73 61 62 6c ........"/disabl
3c30: 65 22 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 e" => {.........
3c40: 6c 65 74 20 63 68 61 6e 6e 65 6c 20 3d 20 77 6f let channel = wo
3c50: 72 64 73 2e 6e 65 78 74 28 29 2e 75 6e 77 72 61 rds.next().unwra
3c60: 70 28 29 3b 0a 09 09 09 09 09 09 09 09 69 66 20 p();.........if
3c70: 21 20 72 65 5f 75 73 65 72 6e 61 6d 65 2e 69 73 ! re_username.is
3c80: 5f 6d 61 74 63 68 28 26 63 68 61 6e 6e 65 6c 29 _match(&channel)
3c90: 20 7b 0a 09 09 09 09 09 09 09 09 09 72 65 70 6c {..........repl
3ca0: 79 2e 70 75 73 68 28 22 55 73 65 72 6e 61 6d 65 y.push("Username
3cb0: 73 20 73 68 6f 75 6c 64 20 62 65 20 73 6f 6d 65 s should be some
3cc0: 74 68 69 6e 67 20 6c 69 6b 65 20 5c 22 40 5c 5c thing like \"@\\
3cd0: 5b 61 2d 7a 5d 5c 5c 5b 61 2d 7a 30 2d 39 5f 5d [a-z]\\[a-z0-9_]
3ce0: 2b 5c 22 2c 20 61 72 65 6e 27 74 20 74 68 65 79 +\", aren't they
3cf0: 3f 22 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b ?".to_string());
3d00: 0a 09 09 09 09 09 09 09 09 7d 20 65 6c 73 65 20 .........} else
3d10: 7b 0a 09 09 09 09 09 09 09 09 09 6d 61 74 63 68 {..........match
3d20: 20 63 6f 72 65 2e 64 69 73 61 62 6c 65 28 6d 65 core.disable(me
3d30: 73 73 61 67 65 2e 66 72 6f 6d 2e 69 64 2c 20 63 ssage.from.id, c
3d40: 68 61 6e 6e 65 6c 29 2e 61 77 61 69 74 20 7b 0a hannel).await {.
3d50: 09 09 09 09 09 09 09 09 09 09 4f 6b 28 5f 29 20 ..........Ok(_)
3d60: 3d 3e 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 => {............
3d70: 72 65 70 6c 79 2e 70 75 73 68 28 22 43 68 61 6e reply.push("Chan
3d80: 6e 65 6c 20 64 69 73 61 62 6c 65 64 5c 5c 2e 22 nel disabled\\."
3d90: 2e 74 6f 5f 73 74 72 69 6e 67 28 29 29 3b 0a 09 .to_string());..
3da0: 09 09 09 09 09 09 09 09 09 7d 0a 09 09 09 09 09 .........}......
3db0: 09 09 09 09 09 45 72 72 28 65 72 72 29 20 3d 3e .....Err(err) =>
3dc0: 20 7b 0a 09 09 09 09 09 09 09 09 09 09 09 63 6f {............co
3dd0: 72 65 2e 64 65 62 75 67 28 26 65 72 72 2e 74 6f re.debug(&err.to
3de0: 5f 73 74 72 69 6e 67 28 29 29 3f 3b 0a 09 09 09 _string())?;....
3df0: 09 09 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 09 .......},.......
3e00: 09 09 09 7d 0a 09 09 09 09 09 09 09 09 7d 0a 09 ...}.........}..
3e10: 09 09 09 09 09 09 7d 2c 0a 0a 09 09 09 09 09 09 ......},........
3e20: 09 5f 20 3d 3e 20 7b 0a 09 09 09 09 09 09 09 7d ._ => {........}
3e30: 2c 0a 09 09 09 09 09 09 7d 3b 0a 09 09 09 09 09 ,.......};......
3e40: 7d 2c 0a 09 09 09 09 09 5f 20 3d 3e 20 7b 0a 09 },......_ => {..
3e50: 09 09 09 09 7d 2c 0a 09 09 09 09 7d 3b 0a 09 09 ....},.....};...
3e60: 09 09 69 66 20 72 65 70 6c 79 2e 6c 65 6e 28 29 ..if reply.len()
3e70: 20 3e 20 30 20 7b 0a 09 09 09 09 09 6d 61 74 63 > 0 {......matc
3e80: 68 20 63 6f 72 65 2e 74 67 2e 73 65 6e 64 28 6d h core.tg.send(m
3e90: 65 73 73 61 67 65 2e 74 65 78 74 5f 72 65 70 6c essage.text_repl
3ea0: 79 28 72 65 70 6c 79 2e 6a 6f 69 6e 28 22 5c 6e y(reply.join("\n
3eb0: 22 29 29 2e 70 61 72 73 65 5f 6d 6f 64 65 28 74 ")).parse_mode(t
3ec0: 79 70 65 73 3a 3a 50 61 72 73 65 4d 6f 64 65 3a ypes::ParseMode:
3ed0: 3a 4d 61 72 6b 64 6f 77 6e 56 32 29 29 2e 61 77 :MarkdownV2)).aw
3ee0: 61 69 74 20 7b 0a 09 09 09 09 09 09 4f 6b 28 5f ait {.......Ok(_
3ef0: 29 20 3d 3e 20 7b 7d 2c 0a 09 09 09 09 09 09 45 ) => {},.......E
3f00: 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 09 rr(err) => {....
3f10: 09 09 09 09 64 62 67 21 28 72 65 70 6c 79 2e 6a ....dbg!(reply.j
3f20: 6f 69 6e 28 22 5c 6e 22 29 29 3b 0a 09 09 09 09 oin("\n"));.....
3f30: 09 09 09 70 72 69 6e 74 6c 6e 21 28 22 7b 7d 22 ...println!("{}"
3f40: 2c 20 65 72 72 29 3b 0a 09 09 09 09 09 09 7d 2c , err);.......},
3f50: 0a 09 09 09 09 09 7d 0a 09 09 09 09 7d 0a 09 09 ......}.....}...
3f60: 09 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b 7d 2c 0a .},...._ => {},.
3f70: 09 09 7d 3b 0a 09 7d 0a 0a 09 4f 6b 28 28 29 29 ..};..}...Ok(())
3f80: 0a 7d 0a .}.