0000: 75 73 65 20 63 72 61 74 65 3a 3a 7b 0a 09 41 72 use crate::{..Ar
0010: 63 2c 0a 09 4d 75 74 65 78 2c 0a 7d 3b 0a 0a 75 c,..Mutex,.};..u
0020: 73 65 20 73 74 64 3a 3a 7b 0a 09 62 6f 72 72 6f se std::{..borro
0030: 77 3a 3a 43 6f 77 2c 0a 09 66 6d 74 2c 0a 7d 3b w::Cow,..fmt,.};
0040: 0a 0a 75 73 65 20 63 68 72 6f 6e 6f 3a 3a 7b 0a ..use chrono::{.
0050: 09 44 61 74 65 54 69 6d 65 2c 0a 09 46 69 78 65 .DateTime,..Fixe
0060: 64 4f 66 66 73 65 74 2c 0a 09 4c 6f 63 61 6c 2c dOffset,..Local,
0070: 0a 7d 3b 0a 75 73 65 20 73 71 6c 78 3a 3a 7b 0a .};.use sqlx::{.
0080: 09 50 6f 73 74 67 72 65 73 2c 0a 09 52 6f 77 2c .Postgres,..Row,
0090: 0a 09 70 6f 73 74 67 72 65 73 3a 3a 50 67 50 6f ..postgres::PgPo
00a0: 6f 6c 4f 70 74 69 6f 6e 73 2c 0a 09 70 6f 6f 6c olOptions,..pool
00b0: 3a 3a 50 6f 6f 6c 43 6f 6e 6e 65 63 74 69 6f 6e ::PoolConnection
00c0: 2c 0a 7d 3b 0a 75 73 65 20 73 74 61 63 6b 65 64 ,.};.use stacked
00d0: 5f 65 72 72 6f 72 73 3a 3a 7b 0a 09 52 65 73 75 _errors::{..Resu
00e0: 6c 74 2c 0a 09 53 74 61 63 6b 61 62 6c 65 45 72 lt,..StackableEr
00f0: 72 2c 0a 09 62 61 69 6c 2c 0a 7d 3b 0a 0a 23 5b r,..bail,.};..#[
0100: 64 65 72 69 76 65 28 73 71 6c 78 3a 3a 46 72 6f derive(sqlx::Fro
0110: 6d 52 6f 77 2c 20 44 65 62 75 67 29 5d 0a 70 75 mRow, Debug)].pu
0120: 62 20 73 74 72 75 63 74 20 4c 69 73 74 20 7b 0a b struct List {.
0130: 09 70 75 62 20 73 6f 75 72 63 65 5f 69 64 3a 20 .pub source_id:
0140: 69 33 32 2c 0a 09 70 75 62 20 63 68 61 6e 6e 65 i32,..pub channe
0150: 6c 3a 20 53 74 72 69 6e 67 2c 0a 09 70 75 62 20 l: String,..pub
0160: 65 6e 61 62 6c 65 64 3a 20 62 6f 6f 6c 2c 0a 09 enabled: bool,..
0170: 70 75 62 20 75 72 6c 3a 20 53 74 72 69 6e 67 2c pub url: String,
0180: 0a 09 70 75 62 20 69 76 5f 68 61 73 68 3a 20 4f ..pub iv_hash: O
0190: 70 74 69 6f 6e 3c 53 74 72 69 6e 67 3e 2c 0a 09 ption<String>,..
01a0: 70 75 62 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 pub url_re: Opti
01b0: 6f 6e 3c 53 74 72 69 6e 67 3e 2c 0a 7d 0a 0a 69 on<String>,.}..i
01c0: 6d 70 6c 20 66 6d 74 3a 3a 44 69 73 70 6c 61 79 mpl fmt::Display
01d0: 20 66 6f 72 20 4c 69 73 74 20 7b 0a 09 66 6e 20 for List {..fn
01e0: 66 6d 74 28 26 73 65 6c 66 2c 20 66 3a 20 26 6d fmt(&self, f: &m
01f0: 75 74 20 66 6d 74 3a 3a 46 6f 72 6d 61 74 74 65 ut fmt::Formatte
0200: 72 3c 27 5f 3e 29 20 2d 3e 20 73 74 64 3a 3a 72 r<'_>) -> std::r
0210: 65 73 75 6c 74 3a 3a 52 65 73 75 6c 74 3c 28 29 esult::Result<()
0220: 2c 20 66 6d 74 3a 3a 45 72 72 6f 72 3e 20 7b 0a , fmt::Error> {.
0230: 09 09 77 72 69 74 65 21 28 66 2c 20 22 5c 5c 23 ..write!(f, "\\#
0240: 66 65 65 64 5c 5c 5f 7b 7d 20 5c 5c 2a ef b8 8f feed\\_{} \\*ï¸
0250: e2 83 a3 20 60 7b 7d 60 20 7b 7d 5c 6e f0 9f 94 ⃣ `{}` {}\nðŸ”
0260: 97 20 60 7b 7d 60 22 2c 20 73 65 6c 66 2e 73 6f — `{}`", self.so
0270: 75 72 63 65 5f 69 64 2c 20 73 65 6c 66 2e 63 68 urce_id, self.ch
0280: 61 6e 6e 65 6c 2c 0a 09 09 09 6d 61 74 63 68 20 annel,....match
0290: 73 65 6c 66 2e 65 6e 61 62 6c 65 64 20 7b 0a 09 self.enabled {..
02a0: 09 09 09 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 ...true => "ðŸ”
02b0: 84 20 65 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 „ enabled",.....
02c0: 66 61 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 false => "â›” di
02d0: 73 61 62 6c 65 64 22 2c 0a 09 09 09 7d 2c 20 73 sabled",....}, s
02e0: 65 6c 66 2e 75 72 6c 29 3f 3b 0a 09 09 69 66 20 elf.url)?;...if
02f0: 6c 65 74 20 53 6f 6d 65 28 69 76 5f 68 61 73 68 let Some(iv_hash
0300: 29 20 3d 20 26 73 65 6c 66 2e 69 76 5f 68 61 73 ) = &self.iv_has
0310: 68 20 7b 0a 09 09 09 77 72 69 74 65 21 28 66 2c h {....write!(f,
0320: 20 22 5c 6e 49 56 3a 20 60 7b 69 76 5f 68 61 73 "\nIV: `{iv_has
0330: 68 7d 60 22 29 3f 3b 0a 09 09 7d 0a 09 09 69 66 h}`")?;...}...if
0340: 20 6c 65 74 20 53 6f 6d 65 28 75 72 6c 5f 72 65 let Some(url_re
0350: 29 20 3d 20 26 73 65 6c 66 2e 75 72 6c 5f 72 65 ) = &self.url_re
0360: 20 7b 0a 09 09 09 77 72 69 74 65 21 28 66 2c 20 {....write!(f,
0370: 22 5c 6e 52 45 3a 20 60 7b 75 72 6c 5f 72 65 7d "\nRE: `{url_re}
0380: 60 22 29 3f 3b 0a 09 09 7d 0a 09 09 4f 6b 28 28 `")?;...}...Ok((
0390: 29 29 0a 09 7d 0a 7d 0a 0a 2f 2f 2f 20 4f 6e 65 ))..}.}../// One
03a0: 20 66 65 65 64 2c 20 75 73 65 64 20 66 6f 72 20 feed, used for
03b0: 63 61 63 68 69 6e 67 20 61 6e 64 20 6d 65 6e 75 caching and menu
03c0: 20 6e 61 76 69 67 61 74 69 6f 6e 0a 23 5b 64 65 navigation.#[de
03d0: 72 69 76 65 28 73 71 6c 78 3a 3a 46 72 6f 6d 52 rive(sqlx::FromR
03e0: 6f 77 2c 20 44 65 62 75 67 29 5d 0a 70 75 62 20 ow, Debug)].pub
03f0: 73 74 72 75 63 74 20 46 65 65 64 20 7b 0a 09 70 struct Feed {..p
0400: 75 62 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 ub source_id: i3
0410: 32 2c 0a 09 70 75 62 20 63 68 61 6e 6e 65 6c 3a 2,..pub channel:
0420: 20 53 74 72 69 6e 67 2c 0a 7d 0a 0a 23 5b 64 65 String,.}..#[de
0430: 72 69 76 65 28 73 71 6c 78 3a 3a 46 72 6f 6d 52 rive(sqlx::FromR
0440: 6f 77 2c 20 44 65 62 75 67 29 5d 0a 70 75 62 20 ow, Debug)].pub
0450: 73 74 72 75 63 74 20 53 6f 75 72 63 65 20 7b 0a struct Source {.
0460: 09 70 75 62 20 63 68 61 6e 6e 65 6c 5f 69 64 3a .pub channel_id:
0470: 20 69 36 34 2c 0a 09 70 75 62 20 75 72 6c 3a 20 i64,..pub url:
0480: 53 74 72 69 6e 67 2c 0a 09 70 75 62 20 69 76 5f String,..pub iv_
0490: 68 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 53 74 72 hash: Option<Str
04a0: 69 6e 67 3e 2c 0a 09 70 75 62 20 6f 77 6e 65 72 ing>,..pub owner
04b0: 3a 20 69 36 34 2c 0a 09 70 75 62 20 75 72 6c 5f : i64,..pub url_
04c0: 72 65 3a 20 4f 70 74 69 6f 6e 3c 53 74 72 69 6e re: Option<Strin
04d0: 67 3e 2c 0a 7d 0a 0a 23 5b 64 65 72 69 76 65 28 g>,.}..#[derive(
04e0: 73 71 6c 78 3a 3a 46 72 6f 6d 52 6f 77 29 5d 0a sqlx::FromRow)].
04f0: 70 75 62 20 73 74 72 75 63 74 20 51 75 65 75 65 pub struct Queue
0500: 20 7b 0a 09 70 75 62 20 73 6f 75 72 63 65 5f 69 {..pub source_i
0510: 64 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c 0a d: Option<i32>,.
0520: 09 70 75 62 20 6e 65 78 74 5f 66 65 74 63 68 3a .pub next_fetch:
0530: 20 4f 70 74 69 6f 6e 3c 44 61 74 65 54 69 6d 65 Option<DateTime
0540: 3c 4c 6f 63 61 6c 3e 3e 2c 0a 09 70 75 62 20 6f <Local>>,..pub o
0550: 77 6e 65 72 3a 20 4f 70 74 69 6f 6e 3c 69 36 34 wner: Option<i64
0560: 3e 2c 0a 09 70 75 62 20 6c 61 73 74 5f 73 63 72 >,..pub last_scr
0570: 61 70 65 3a 20 44 61 74 65 54 69 6d 65 3c 4c 6f ape: DateTime<Lo
0580: 63 61 6c 3e 2c 0a 7d 0a 0a 23 5b 64 65 72 69 76 cal>,.}..#[deriv
0590: 65 28 43 6c 6f 6e 65 29 5d 0a 70 75 62 20 73 74 e(Clone)].pub st
05a0: 72 75 63 74 20 44 62 20 28 0a 09 41 72 63 3c 4d ruct Db (..Arc<M
05b0: 75 74 65 78 3c 73 71 6c 78 3a 3a 50 6f 6f 6c 3c utex<sqlx::Pool<
05c0: 73 71 6c 78 3a 3a 50 6f 73 74 67 72 65 73 3e 3e sqlx::Postgres>>
05d0: 3e 2c 0a 29 3b 0a 0a 69 6d 70 6c 20 44 62 20 7b >,.);..impl Db {
05e0: 0a 09 70 75 62 20 66 6e 20 6e 65 77 20 28 70 67 ..pub fn new (pg
05f0: 75 72 69 3a 20 26 73 74 72 29 20 2d 3e 20 52 65 uri: &str) -> Re
0600: 73 75 6c 74 3c 44 62 3e 20 7b 0a 09 09 4f 6b 28 sult<Db> {...Ok(
0610: 44 62 20 28 0a 09 09 09 41 72 63 3a 3a 6e 65 77 Db (....Arc::new
0620: 28 4d 75 74 65 78 3a 3a 6e 65 77 28 50 67 50 6f (Mutex::new(PgPo
0630: 6f 6c 4f 70 74 69 6f 6e 73 3a 3a 6e 65 77 28 29 olOptions::new()
0640: 0a 09 09 09 09 2e 6d 61 78 5f 63 6f 6e 6e 65 63 ......max_connec
0650: 74 69 6f 6e 73 28 35 29 0a 09 09 09 09 2e 61 63 tions(5)......ac
0660: 71 75 69 72 65 5f 74 69 6d 65 6f 75 74 28 73 74 quire_timeout(st
0670: 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 61 74 69 6f d::time::Duratio
0680: 6e 3a 3a 6e 65 77 28 33 30 30 2c 20 30 29 29 0a n::new(300, 0)).
0690: 09 09 09 09 2e 69 64 6c 65 5f 74 69 6d 65 6f 75 .....idle_timeou
06a0: 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 72 t(std::time::Dur
06b0: 61 74 69 6f 6e 3a 3a 6e 65 77 28 36 30 2c 20 30 ation::new(60, 0
06c0: 29 29 0a 09 09 09 09 2e 63 6f 6e 6e 65 63 74 5f ))......connect_
06d0: 6c 61 7a 79 28 70 67 75 72 69 29 0a 09 09 09 09 lazy(pguri).....
06e0: 2e 73 74 61 63 6b 28 29 3f 29 29 2c 0a 09 09 29 .stack()?)),...)
06f0: 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 )..}...pub async
0700: 20 66 6e 20 62 65 67 69 6e 28 26 73 65 6c 66 29 fn begin(&self)
0710: 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f 6e 6e 3e -> Result<Conn>
0720: 20 7b 0a 09 09 6c 65 74 20 70 6f 6f 6c 20 3d 20 {...let pool =
0730: 73 65 6c 66 2e 30 2e 6c 6f 63 6b 5f 61 72 63 28 self.0.lock_arc(
0740: 29 2e 61 77 61 69 74 3b 0a 09 09 6c 65 74 20 63 ).await;...let c
0750: 6f 6e 6e 20 3d 20 43 6f 6e 6e 20 28 20 70 6f 6f onn = Conn ( poo
0760: 6c 2e 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 l.acquire().awai
0770: 74 2e 73 74 61 63 6b 28 29 3f 20 29 3b 0a 09 09 t.stack()? );...
0780: 4f 6b 28 63 6f 6e 6e 29 0a 09 7d 0a 7d 0a 0a 70 Ok(conn)..}.}..p
0790: 75 62 20 73 74 72 75 63 74 20 43 6f 6e 6e 20 28 ub struct Conn (
07a0: 0a 09 50 6f 6f 6c 43 6f 6e 6e 65 63 74 69 6f 6e ..PoolConnection
07b0: 3c 50 6f 73 74 67 72 65 73 3e 2c 0a 29 3b 0a 0a <Postgres>,.);..
07c0: 69 6d 70 6c 20 43 6f 6e 6e 20 7b 0a 09 70 75 62 impl Conn {..pub
07d0: 20 61 73 79 6e 63 20 66 6e 20 61 64 64 5f 70 6f async fn add_po
07e0: 73 74 20 28 26 6d 75 74 20 73 65 6c 66 2c 20 73 st (&mut self, s
07f0: 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 2c 20 64 ource_id: i32, d
0800: 61 74 65 3a 20 26 44 61 74 65 54 69 6d 65 3c 46 ate: &DateTime<F
0810: 69 78 65 64 4f 66 66 73 65 74 3e 2c 20 70 6f 73 ixedOffset>, pos
0820: 74 5f 75 72 6c 3a 20 26 73 74 72 29 20 2d 3e 20 t_url: &str) ->
0830: 52 65 73 75 6c 74 3c 28 29 3e 20 7b 0a 09 09 73 Result<()> {...s
0840: 71 6c 78 3a 3a 71 75 65 72 79 28 22 69 6e 73 65 qlx::query("inse
0850: 72 74 20 69 6e 74 6f 20 72 73 73 74 67 5f 70 6f rt into rsstg_po
0860: 73 74 20 28 73 6f 75 72 63 65 5f 69 64 2c 20 70 st (source_id, p
0870: 6f 73 74 65 64 2c 20 75 72 6c 29 20 76 61 6c 75 osted, url) valu
0880: 65 73 20 28 24 31 2c 20 24 32 2c 20 24 33 29 3b es ($1, $2, $3);
0890: 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 ").....bind(sour
08a0: 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 ce_id).....bind(
08b0: 64 61 74 65 29 0a 09 09 09 2e 62 69 6e 64 28 70 date).....bind(p
08c0: 6f 73 74 5f 75 72 6c 29 0a 09 09 09 2e 65 78 65 ost_url).....exe
08d0: 63 75 74 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e cute(&mut *self.
08e0: 30 29 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 0).await.stack()
08f0: 3f 3b 0a 09 09 4f 6b 28 28 29 29 0a 09 7d 0a 0a ?;...Ok(())..}..
0900: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 63 6c .pub async fn cl
0910: 65 61 6e 20 3c 49 3e 20 28 26 6d 75 74 20 73 65 ean <I> (&mut se
0920: 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 lf, source_id: i
0930: 33 32 2c 20 6f 77 6e 65 72 3a 20 49 29 20 2d 3e 32, owner: I) ->
0940: 20 52 65 73 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 Result<Cow<'_,
0950: 73 74 72 3e 3e 0a 09 77 68 65 72 65 20 49 3a 20 str>>..where I:
0960: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6d 61 Into<i64> {...ma
0970: 74 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 tch sqlx::query(
0980: 22 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 "delete from rss
0990: 74 67 5f 70 6f 73 74 20 70 20 75 73 69 6e 67 20 tg_post p using
09a0: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 20 77 rsstg_source s w
09b0: 68 65 72 65 20 70 2e 73 6f 75 72 63 65 5f 69 64 here p.source_id
09c0: 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 = $1 and owner
09d0: 3d 20 24 32 20 61 6e 64 20 70 2e 73 6f 75 72 63 = $2 and p.sourc
09e0: 65 5f 69 64 20 3d 20 73 2e 73 6f 75 72 63 65 5f e_id = s.source_
09f0: 69 64 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 73 id;").....bind(s
0a00: 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 69 ource_id).....bi
0a10: 6e 64 28 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 29 nd(owner.into())
0a20: 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 .....execute(&mu
0a30: 74 20 2a 73 65 6c 66 2e 30 29 2e 61 77 61 69 74 t *self.0).await
0a40: 2e 73 74 61 63 6b 28 29 3f 2e 72 6f 77 73 5f 61 .stack()?.rows_a
0a50: 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 09 30 ffected() {....0
0a60: 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 61 74 => { Ok("No dat
0a70: 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 2e 22 2e a found found.".
0a80: 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 09 78 20 into()) },....x
0a90: 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 21 28 => { Ok(format!(
0aa0: 22 7b 78 7d 20 70 6f 73 74 73 20 70 75 72 67 65 "{x} posts purge
0ab0: 64 2e 22 29 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a d.").into()) },.
0ac0: 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ..}..}...pub asy
0ad0: 6e 63 20 66 6e 20 64 65 6c 65 74 65 20 3c 49 3e nc fn delete <I>
0ae0: 20 28 26 6d 75 74 20 73 65 6c 66 2c 20 73 6f 75 (&mut self, sou
0af0: 72 63 65 5f 69 64 3a 20 69 33 32 2c 20 6f 77 6e rce_id: i32, own
0b00: 65 72 3a 20 49 29 20 2d 3e 20 52 65 73 75 6c 74 er: I) -> Result
0b10: 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 <Cow<'_, str>>..
0b20: 77 68 65 72 65 20 49 3a 20 49 6e 74 6f 3c 69 36 where I: Into<i6
0b30: 34 3e 20 7b 0a 09 09 6d 61 74 63 68 20 73 71 6c 4> {...match sql
0b40: 78 3a 3a 71 75 65 72 79 28 22 64 65 6c 65 74 65 x::query("delete
0b50: 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 from rsstg_sour
0b60: 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f ce where source_
0b70: 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 id = $1 and owne
0b80: 72 20 3d 20 24 32 3b 22 29 0a 09 09 09 2e 62 69 r = $2;").....bi
0b90: 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 nd(source_id)...
0ba0: 09 2e 62 69 6e 64 28 6f 77 6e 65 72 2e 69 6e 74 ..bind(owner.int
0bb0: 6f 28 29 29 0a 09 09 09 2e 65 78 65 63 75 74 65 o()).....execute
0bc0: 28 26 6d 75 74 20 2a 73 65 6c 66 2e 30 29 2e 61 (&mut *self.0).a
0bd0: 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 2e 72 6f wait.stack()?.ro
0be0: 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a ws_affected() {.
0bf0: 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f ...0 => { Ok("No
0c00: 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e data found foun
0c10: 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 d.".into()) },..
0c20: 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d ..x => { Ok(form
0c30: 61 74 21 28 22 7b 78 7d 20 73 6f 75 72 63 65 73 at!("{x} sources
0c40: 20 72 65 6d 6f 76 65 64 2e 22 29 2e 69 6e 74 6f removed.").into
0c50: 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 ()) },...}..}...
0c60: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 64 69 73 pub async fn dis
0c70: 61 62 6c 65 20 3c 49 3e 20 28 26 6d 75 74 20 73 able <I> (&mut s
0c80: 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 elf, source_id:
0c90: 69 33 32 2c 20 6f 77 6e 65 72 3a 20 49 29 20 2d i32, owner: I) -
0ca0: 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 > Result<&str>..
0cb0: 77 68 65 72 65 20 49 3a 20 49 6e 74 6f 3c 69 36 where I: Into<i6
0cc0: 34 3e 20 7b 0a 09 09 6d 61 74 63 68 20 73 71 6c 4> {...match sql
0cd0: 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 x::query("update
0ce0: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 rsstg_source se
0cf0: 74 20 65 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 t enabled = fals
0d00: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
0d10: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
0d20: 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 = $2").....bind
0d30: 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e (source_id).....
0d40: 62 69 6e 64 28 6f 77 6e 65 72 2e 69 6e 74 6f 28 bind(owner.into(
0d50: 29 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 )).....execute(&
0d60: 6d 75 74 20 2a 73 65 6c 66 2e 30 29 2e 61 77 61 mut *self.0).awa
0d70: 69 74 2e 73 74 61 63 6b 28 29 3f 2e 72 6f 77 73 it.stack()?.rows
0d80: 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 _affected() {...
0d90: 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 .1 => { Ok("Sour
0da0: 63 65 20 64 69 73 61 62 6c 65 64 2e 22 29 20 7d ce disabled.") }
0db0: 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 ,....0 => { Ok("
0dc0: 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f 75 6e 64 Source not found
0dd0: 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d 3e 20 7b .") },...._ => {
0de0: 20 62 61 69 6c 21 28 22 44 61 74 61 62 61 73 65 bail!("Database
0df0: 20 65 72 72 6f 72 2e 22 29 20 7d 2c 0a 09 09 7d error.") },...}
0e00: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
0e10: 66 6e 20 65 6e 61 62 6c 65 20 3c 49 3e 20 28 26 fn enable <I> (&
0e20: 6d 75 74 20 73 65 6c 66 2c 20 73 6f 75 72 63 65 mut self, source
0e30: 5f 69 64 3a 20 69 33 32 2c 20 6f 77 6e 65 72 3a _id: i32, owner:
0e40: 20 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 I) -> Result<&s
0e50: 74 72 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 6e tr>..where I: In
0e60: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6d 61 74 63 to<i64> {...matc
0e70: 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 h sqlx::query("u
0e80: 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 pdate rsstg_sour
0e90: 63 65 20 73 65 74 20 65 6e 61 62 6c 65 64 20 3d ce set enabled =
0ea0: 20 74 72 75 65 20 77 68 65 72 65 20 73 6f 75 72 true where sour
0eb0: 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f ce_id = $1 and o
0ec0: 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 09 2e wner = $2").....
0ed0: 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a bind(source_id).
0ee0: 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 2e 69 ....bind(owner.i
0ef0: 6e 74 6f 28 29 29 0a 09 09 09 2e 65 78 65 63 75 nto()).....execu
0f00: 74 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e 30 29 te(&mut *self.0)
0f10: 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 2e .await.stack()?.
0f20: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 rows_affected()
0f30: 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b 28 22 {....1 => { Ok("
0f40: 53 6f 75 72 63 65 20 65 6e 61 62 6c 65 64 2e 22 Source enabled."
0f50: 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 7b 20 4f ) },....0 => { O
0f60: 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 20 66 6f k("Source not fo
0f70: 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 5f 20 3d und.") },...._ =
0f80: 3e 20 7b 20 62 61 69 6c 21 28 22 44 61 74 61 62 > { bail!("Datab
0f90: 61 73 65 20 65 72 72 6f 72 2e 22 29 20 7d 2c 0a ase error.") },.
0fa0: 09 09 7d 0a 09 7d 0a 0a 09 2f 2f 2f 20 43 68 65 ..}..}.../// Che
0fb0: 63 6b 73 20 77 68 65 74 68 65 72 20 61 20 70 6f cks whether a po
0fc0: 73 74 20 77 69 74 68 20 74 68 65 20 67 69 76 65 st with the give
0fd0: 6e 20 55 52 4c 20 65 78 69 73 74 73 20 66 6f 72 n URL exists for
0fe0: 20 74 68 65 20 73 70 65 63 69 66 69 65 64 20 73 the specified s
0ff0: 6f 75 72 63 65 2e 0a 09 2f 2f 2f 0a 09 2f 2f 2f ource...///..///
1000: 20 23 20 50 61 72 61 6d 65 74 65 72 73 0a 09 2f # Parameters../
1010: 2f 2f 20 2d 20 60 70 6f 73 74 5f 75 72 6c 60 3a // - `post_url`:
1020: 20 54 68 65 20 55 52 4c 20 6f 66 20 74 68 65 20 The URL of the
1030: 70 6f 73 74 20 74 6f 20 63 68 65 63 6b 2e 0a 09 post to check...
1040: 2f 2f 2f 20 2d 20 60 69 64 60 3a 20 54 68 65 20 /// - `id`: The
1050: 73 6f 75 72 63 65 20 69 64 65 6e 74 69 66 69 65 source identifie
1060: 72 20 28 63 6f 6e 76 65 72 74 65 64 20 74 6f 20 r (converted to
1070: 60 69 36 34 60 29 2e 0a 09 2f 2f 2f 0a 09 2f 2f `i64`)...///..//
1080: 2f 20 23 20 52 65 74 75 72 6e 73 0a 09 2f 2f 2f / # Returns..///
1090: 20 60 74 72 75 65 60 20 69 66 20 61 20 70 6f 73 `true` if a pos
10a0: 74 20 77 69 74 68 20 74 68 65 20 55 52 4c 20 65 t with the URL e
10b0: 78 69 73 74 73 20 66 6f 72 20 74 68 65 20 73 6f xists for the so
10c0: 75 72 63 65 2c 20 60 66 61 6c 73 65 60 20 6f 74 urce, `false` ot
10d0: 68 65 72 77 69 73 65 2e 0a 09 70 75 62 20 61 73 herwise...pub as
10e0: 79 6e 63 20 66 6e 20 65 78 69 73 74 73 20 3c 49 ync fn exists <I
10f0: 3e 20 28 26 6d 75 74 20 73 65 6c 66 2c 20 70 6f > (&mut self, po
1100: 73 74 5f 75 72 6c 3a 20 26 73 74 72 2c 20 69 64 st_url: &str, id
1110: 3a 20 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c 62 : I) -> Result<b
1120: 6f 6f 6c 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 ool>..where I: I
1130: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
1140: 20 72 6f 77 20 3d 20 73 71 6c 78 3a 3a 71 75 65 row = sqlx::que
1150: 72 79 28 22 73 65 6c 65 63 74 20 65 78 69 73 74 ry("select exist
1160: 73 28 73 65 6c 65 63 74 20 74 72 75 65 20 66 72 s(select true fr
1170: 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 77 68 om rsstg_post wh
1180: 65 72 65 20 75 72 6c 20 3d 20 24 31 20 61 6e 64 ere url = $1 and
1190: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 32 29 source_id = $2)
11a0: 20 61 73 20 65 78 69 73 74 73 3b 22 29 0a 09 09 as exists;")...
11b0: 09 2e 62 69 6e 64 28 70 6f 73 74 5f 75 72 6c 29 ..bind(post_url)
11c0: 0a 09 09 09 2e 62 69 6e 64 28 69 64 2e 69 6e 74 .....bind(id.int
11d0: 6f 28 29 29 0a 09 09 09 2e 66 65 74 63 68 5f 6f o()).....fetch_o
11e0: 6e 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e 30 29 ne(&mut *self.0)
11f0: 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b .await.stack()?;
1200: 0a 09 09 72 6f 77 2e 74 72 79 5f 67 65 74 28 22 ...row.try_get("
1210: 65 78 69 73 74 73 22 29 0a 09 09 09 2e 73 74 61 exists").....sta
1220: 63 6b 5f 65 72 72 28 22 44 61 74 61 62 61 73 65 ck_err("Database
1230: 20 65 72 72 6f 72 3a 20 63 61 6e 27 74 20 63 68 error: can't ch
1240: 65 63 6b 20 77 68 65 74 68 65 72 20 70 6f 73 74 eck whether post
1250: 20 65 78 69 73 74 73 2e 22 29 0a 09 7d 0a 0a 09 exists.")..}...
1260: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 67 65 74 pub async fn get
1270: 5f 66 65 65 64 73 20 3c 49 3e 28 26 6d 75 74 20 _feeds <I>(&mut
1280: 73 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 49 29 20 self, owner: I)
1290: 2d 3e 20 52 65 73 75 6c 74 3c 56 65 63 3c 46 65 -> Result<Vec<Fe
12a0: 65 64 3e 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 ed>>..where I: I
12b0: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
12c0: 20 62 6c 6f 63 6b 3a 20 56 65 63 3c 46 65 65 64 block: Vec<Feed
12d0: 3e 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 5f > = sqlx::query_
12e0: 61 73 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 as("select sourc
12f0: 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 20 66 72 e_id, channel fr
1300: 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 om rsstg_source
1310: 77 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 31 where owner = $1
1320: 20 6f 72 64 65 72 20 62 79 20 73 6f 75 72 63 65 order by source
1330: 5f 69 64 22 29 0a 09 09 09 2e 62 69 6e 64 28 6f _id").....bind(o
1340: 77 6e 65 72 2e 69 6e 74 6f 28 29 29 0a 09 09 09 wner.into())....
1350: 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 .fetch_all(&mut
1360: 2a 73 65 6c 66 2e 30 29 2e 61 77 61 69 74 2e 73 *self.0).await.s
1370: 74 61 63 6b 28 29 3f 3b 0a 09 09 4f 6b 28 62 6c tack()?;...Ok(bl
1380: 6f 63 6b 29 0a 09 7d 0a 0a 09 2f 2f 2f 20 47 65 ock)..}.../// Ge
1390: 74 20 61 6c 6c 20 70 65 6e 64 69 6e 67 20 65 76 t all pending ev
13a0: 65 6e 74 73 20 66 6f 72 20 28 6e 6f 77 20 2b 20 ents for (now +
13b0: 31 20 6d 69 6e 75 74 65 29 0a 09 70 75 62 20 61 1 minute)..pub a
13c0: 73 79 6e 63 20 66 6e 20 67 65 74 5f 71 75 65 75 sync fn get_queu
13d0: 65 20 28 26 6d 75 74 20 73 65 6c 66 29 20 2d 3e e (&mut self) ->
13e0: 20 52 65 73 75 6c 74 3c 56 65 63 3c 51 75 65 75 Result<Vec<Queu
13f0: 65 3e 3e 20 7b 0a 09 09 6c 65 74 20 62 6c 6f 63 e>> {...let bloc
1400: 6b 3a 20 56 65 63 3c 51 75 65 75 65 3e 20 3d 20 k: Vec<Queue> =
1410: 73 71 6c 78 3a 3a 71 75 65 72 79 5f 61 73 28 22 sqlx::query_as("
1420: 73 65 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 64 select source_id
1430: 2c 20 6e 65 78 74 5f 66 65 74 63 68 2c 20 6f 77 , next_fetch, ow
1440: 6e 65 72 2c 20 6c 61 73 74 5f 73 63 72 61 70 65 ner, last_scrape
1450: 20 66 72 6f 6d 20 72 73 73 74 67 5f 6f 72 64 65 from rsstg_orde
1460: 72 20 6e 61 74 75 72 61 6c 20 6c 65 66 74 20 6a r natural left j
1470: 6f 69 6e 20 72 73 73 74 67 5f 73 6f 75 72 63 65 oin rsstg_source
1480: 20 77 68 65 72 65 20 6e 65 78 74 5f 66 65 74 63 where next_fetc
1490: 68 20 3c 20 6e 6f 77 28 29 20 2b 20 69 6e 74 65 h < now() + inte
14a0: 72 76 61 6c 20 27 31 20 6d 69 6e 75 74 65 27 3b rval '1 minute';
14b0: 22 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c ").....fetch_all
14c0: 28 26 6d 75 74 20 2a 73 65 6c 66 2e 30 29 2e 61 (&mut *self.0).a
14d0: 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 wait.stack()?;..
14e0: 09 4f 6b 28 62 6c 6f 63 6b 29 0a 09 7d 0a 0a 09 .Ok(block)..}...
14f0: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 67 65 74 pub async fn get
1500: 5f 6c 69 73 74 20 3c 49 3e 28 26 6d 75 74 20 73 _list <I>(&mut s
1510: 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 49 29 20 2d elf, owner: I) -
1520: 3e 20 52 65 73 75 6c 74 3c 56 65 63 3c 4c 69 73 > Result<Vec<Lis
1530: 74 3e 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 6e t>>..where I: In
1540: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 to<i64> {...let
1550: 73 6f 75 72 63 65 3a 20 56 65 63 3c 4c 69 73 74 source: Vec<List
1560: 3e 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 5f > = sqlx::query_
1570: 61 73 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 as("select sourc
1580: 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 65 e_id, channel, e
1590: 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 76 5f nabled, url, iv_
15a0: 68 61 73 68 2c 20 75 72 6c 5f 72 65 20 66 72 6f hash, url_re fro
15b0: 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 m rsstg_source w
15c0: 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 31 20 here owner = $1
15d0: 6f 72 64 65 72 20 62 79 20 73 6f 75 72 63 65 5f order by source_
15e0: 69 64 22 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 id").....bind(ow
15f0: 6e 65 72 2e 69 6e 74 6f 28 29 29 0a 09 09 09 2e ner.into()).....
1600: 66 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 2a fetch_all(&mut *
1610: 73 65 6c 66 2e 30 29 2e 61 77 61 69 74 2e 73 74 self.0).await.st
1620: 61 63 6b 28 29 3f 3b 0a 09 09 4f 6b 28 73 6f 75 ack()?;...Ok(sou
1630: 72 63 65 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 rce)..}...pub as
1640: 79 6e 63 20 66 6e 20 67 65 74 5f 6f 6e 65 20 3c ync fn get_one <
1650: 49 3e 20 28 26 6d 75 74 20 73 65 6c 66 2c 20 6f I> (&mut self, o
1660: 77 6e 65 72 3a 20 49 2c 20 69 64 3a 20 69 33 32 wner: I, id: i32
1670: 29 20 2d 3e 20 52 65 73 75 6c 74 3c 4f 70 74 69 ) -> Result<Opti
1680: 6f 6e 3c 4c 69 73 74 3e 3e 0a 09 77 68 65 72 65 on<List>>..where
1690: 20 49 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a I: Into<i64> {.
16a0: 09 09 6c 65 74 20 73 6f 75 72 63 65 3a 20 4f 70 ..let source: Op
16b0: 74 69 6f 6e 3c 4c 69 73 74 3e 20 3d 20 73 71 6c tion<List> = sql
16c0: 78 3a 3a 71 75 65 72 79 5f 61 73 28 22 73 65 6c x::query_as("sel
16d0: 65 63 74 20 73 6f 75 72 63 65 5f 69 64 2c 20 63 ect source_id, c
16e0: 68 61 6e 6e 65 6c 2c 20 65 6e 61 62 6c 65 64 2c hannel, enabled,
16f0: 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 75 url, iv_hash, u
1700: 72 6c 5f 72 65 20 66 72 6f 6d 20 72 73 73 74 67 rl_re from rsstg
1710: 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 6f 77 _source where ow
1720: 6e 65 72 20 3d 20 24 31 20 61 6e 64 20 73 6f 75 ner = $1 and sou
1730: 72 63 65 5f 69 64 20 3d 20 24 32 22 29 0a 09 09 rce_id = $2")...
1740: 09 2e 62 69 6e 64 28 6f 77 6e 65 72 2e 69 6e 74 ..bind(owner.int
1750: 6f 28 29 29 0a 09 09 09 2e 62 69 6e 64 28 69 64 o()).....bind(id
1760: 29 0a 09 09 09 2e 66 65 74 63 68 5f 6f 70 74 69 ).....fetch_opti
1770: 6f 6e 61 6c 28 26 6d 75 74 20 2a 73 65 6c 66 2e onal(&mut *self.
1780: 30 29 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 0).await.stack()
1790: 3f 3b 0a 09 09 4f 6b 28 73 6f 75 72 63 65 29 0a ?;...Ok(source).
17a0: 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 .}...pub async f
17b0: 6e 20 67 65 74 5f 6f 6e 65 5f 6e 61 6d 65 20 3c n get_one_name <
17c0: 49 3e 20 28 26 6d 75 74 20 73 65 6c 66 2c 20 6f I> (&mut self, o
17d0: 77 6e 65 72 3a 20 49 2c 20 6e 61 6d 65 3a 20 26 wner: I, name: &
17e0: 73 74 72 29 20 2d 3e 20 52 65 73 75 6c 74 3c 4f str) -> Result<O
17f0: 70 74 69 6f 6e 3c 4c 69 73 74 3e 3e 0a 09 77 68 ption<List>>..wh
1800: 65 72 65 20 49 3a 20 49 6e 74 6f 3c 69 36 34 3e ere I: Into<i64>
1810: 20 7b 0a 09 09 6c 65 74 20 73 6f 75 72 63 65 3a {...let source:
1820: 20 4f 70 74 69 6f 6e 3c 4c 69 73 74 3e 20 3d 20 Option<List> =
1830: 73 71 6c 78 3a 3a 71 75 65 72 79 5f 61 73 28 22 sqlx::query_as("
1840: 73 65 6c 65 63 74 20 73 6f 75 72 63 65 5f 69 64 select source_id
1850: 2c 20 63 68 61 6e 6e 65 6c 2c 20 65 6e 61 62 6c , channel, enabl
1860: 65 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 68 ed, url, iv_hash
1870: 2c 20 75 72 6c 5f 72 65 20 66 72 6f 6d 20 72 73 , url_re from rs
1880: 73 74 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 stg_source where
1890: 20 6f 77 6e 65 72 20 3d 20 24 31 20 61 6e 64 20 owner = $1 and
18a0: 63 68 61 6e 6e 65 6c 20 3d 20 24 32 22 29 0a 09 channel = $2")..
18b0: 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 2e 69 6e ...bind(owner.in
18c0: 74 6f 28 29 29 0a 09 09 09 2e 62 69 6e 64 28 6e to()).....bind(n
18d0: 61 6d 65 29 0a 09 09 09 2e 66 65 74 63 68 5f 6f ame).....fetch_o
18e0: 70 74 69 6f 6e 61 6c 28 26 6d 75 74 20 2a 73 65 ptional(&mut *se
18f0: 6c 66 2e 30 29 2e 61 77 61 69 74 2e 73 74 61 63 lf.0).await.stac
1900: 6b 28 29 3f 3b 0a 09 09 4f 6b 28 73 6f 75 72 63 k()?;...Ok(sourc
1910: 65 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e e)..}...pub asyn
1920: 63 20 66 6e 20 67 65 74 5f 73 6f 75 72 63 65 20 c fn get_source
1930: 3c 49 3e 20 28 26 6d 75 74 20 73 65 6c 66 2c 20 <I> (&mut self,
1940: 69 64 3a 20 69 33 32 2c 20 6f 77 6e 65 72 3a 20 id: i32, owner:
1950: 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 6f 75 I) -> Result<Sou
1960: 72 63 65 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 rce>..where I: I
1970: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 nto<i64> {...let
1980: 20 73 6f 75 72 63 65 3a 20 53 6f 75 72 63 65 20 source: Source
1990: 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 5f 61 73 = sqlx::query_as
19a0: 28 22 73 65 6c 65 63 74 20 63 68 61 6e 6e 65 6c ("select channel
19b0: 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 _id, url, iv_has
19c0: 68 2c 20 6f 77 6e 65 72 2c 20 75 72 6c 5f 72 65 h, owner, url_re
19d0: 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 from rsstg_sour
19e0: 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f ce where source_
19f0: 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 id = $1 and owne
1a00: 72 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e r = $2").....bin
1a10: 64 28 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f d(id).....bind(o
1a20: 77 6e 65 72 2e 69 6e 74 6f 28 29 29 0a 09 09 09 wner.into())....
1a30: 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 .fetch_one(&mut
1a40: 2a 73 65 6c 66 2e 30 29 2e 61 77 61 69 74 2e 73 *self.0).await.s
1a50: 74 61 63 6b 28 29 3f 3b 0a 09 09 4f 6b 28 73 6f tack()?;...Ok(so
1a60: 75 72 63 65 29 0a 09 7d 0a 0a 09 70 75 62 20 61 urce)..}...pub a
1a70: 73 79 6e 63 20 66 6e 20 73 65 74 5f 73 63 72 61 sync fn set_scra
1a80: 70 65 20 3c 49 3e 20 28 26 6d 75 74 20 73 65 6c pe <I> (&mut sel
1a90: 66 2c 20 69 64 3a 20 49 29 20 2d 3e 20 52 65 73 f, id: I) -> Res
1aa0: 75 6c 74 3c 28 29 3e 0a 09 77 68 65 72 65 20 49 ult<()>..where I
1ab0: 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 : Into<i64> {...
1ac0: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 75 70 64 sqlx::query("upd
1ad0: 61 74 65 20 72 73 73 74 67 5f 73 6f 75 72 63 65 ate rsstg_source
1ae0: 20 73 65 74 20 6c 61 73 74 5f 73 63 72 61 70 65 set last_scrape
1af0: 20 3d 20 6e 6f 77 28 29 20 77 68 65 72 65 20 73 = now() where s
1b00: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 3b 22 29 ource_id = $1;")
1b10: 0a 09 09 09 2e 62 69 6e 64 28 69 64 2e 69 6e 74 .....bind(id.int
1b20: 6f 28 29 29 0a 09 09 09 2e 65 78 65 63 75 74 65 o()).....execute
1b30: 28 26 6d 75 74 20 2a 73 65 6c 66 2e 30 29 2e 61 (&mut *self.0).a
1b40: 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 wait.stack()?;..
1b50: 09 4f 6b 28 28 29 29 0a 09 7d 0a 0a 09 23 5b 61 .Ok(())..}...#[a
1b60: 6c 6c 6f 77 28 63 6c 69 70 70 79 3a 3a 74 6f 6f llow(clippy::too
1b70: 5f 6d 61 6e 79 5f 61 72 67 75 6d 65 6e 74 73 29 _many_arguments)
1b80: 5d 20 2f 2f 20 58 58 58 20 61 74 20 6c 65 61 73 ] // XXX at leas
1b90: 74 20 66 6f 72 20 6e 6f 77 e2 80 a6 0a 09 70 75 t for now…..pu
1ba0: 62 20 61 73 79 6e 63 20 66 6e 20 75 70 64 61 74 b async fn updat
1bb0: 65 20 3c 49 3e 20 28 26 6d 75 74 20 73 65 6c 66 e <I> (&mut self
1bc0: 2c 20 75 70 64 61 74 65 3a 20 4f 70 74 69 6f 6e , update: Option
1bd0: 3c 69 33 32 3e 2c 20 63 68 61 6e 6e 65 6c 3a 20 <i32>, channel:
1be0: 26 73 74 72 2c 20 63 68 61 6e 6e 65 6c 5f 69 64 &str, channel_id
1bf0: 3a 20 69 36 34 2c 20 75 72 6c 3a 20 26 73 74 72 : i64, url: &str
1c00: 2c 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 6f , iv_hash: Optio
1c10: 6e 3c 26 73 74 72 3e 2c 20 75 72 6c 5f 72 65 3a n<&str>, url_re:
1c20: 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 6f Option<&str>, o
1c30: 77 6e 65 72 3a 20 49 29 20 2d 3e 20 52 65 73 75 wner: I) -> Resu
1c40: 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 20 lt<&str>..where
1c50: 49 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 I: Into<i64> {..
1c60: 09 6d 61 74 63 68 20 6d 61 74 63 68 20 75 70 64 .match match upd
1c70: 61 74 65 20 7b 0a 09 09 09 09 53 6f 6d 65 28 69 ate {.....Some(i
1c80: 64 29 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c d) => {......sql
1c90: 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 x::query("update
1ca0: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 rsstg_source se
1cb0: 74 20 63 68 61 6e 6e 65 6c 5f 69 64 20 3d 20 24 t channel_id = $
1cc0: 32 2c 20 75 72 6c 20 3d 20 24 33 2c 20 69 76 5f 2, url = $3, iv_
1cd0: 68 61 73 68 20 3d 20 24 34 2c 20 6f 77 6e 65 72 hash = $4, owner
1ce0: 20 3d 20 24 35 2c 20 63 68 61 6e 6e 65 6c 20 3d = $5, channel =
1cf0: 20 24 36 2c 20 75 72 6c 5f 72 65 20 3d 20 24 37 $6, url_re = $7
1d00: 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 where source_id
1d10: 20 3d 20 24 31 22 29 0a 09 09 09 09 09 09 2e 62 = $1")........b
1d20: 69 6e 64 28 69 64 29 0a 09 09 09 09 7d 2c 0a 09 ind(id).....},..
1d30: 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a 09 09 09 ...None => {....
1d40: 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 69 ..sqlx::query("i
1d50: 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 73 74 67 nsert into rsstg
1d60: 5f 73 6f 75 72 63 65 20 28 63 68 61 6e 6e 65 6c _source (channel
1d70: 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 73 _id, url, iv_has
1d80: 68 2c 20 6f 77 6e 65 72 2c 20 63 68 61 6e 6e 65 h, owner, channe
1d90: 6c 2c 20 75 72 6c 5f 72 65 29 20 76 61 6c 75 65 l, url_re) value
1da0: 73 20 28 24 31 2c 20 24 32 2c 20 24 33 2c 20 24 s ($1, $2, $3, $
1db0: 34 2c 20 24 35 2c 20 24 36 29 22 29 0a 09 09 09 4, $5, $6)")....
1dc0: 09 7d 2c 0a 09 09 09 7d 0a 09 09 09 09 2e 62 69 .},....}......bi
1dd0: 6e 64 28 63 68 61 6e 6e 65 6c 5f 69 64 29 0a 09 nd(channel_id)..
1de0: 09 09 09 2e 62 69 6e 64 28 75 72 6c 29 0a 09 09 ....bind(url)...
1df0: 09 09 2e 62 69 6e 64 28 69 76 5f 68 61 73 68 29 ...bind(iv_hash)
1e00: 0a 09 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ......bind(owner
1e10: 2e 69 6e 74 6f 28 29 29 0a 09 09 09 09 2e 62 69 .into())......bi
1e20: 6e 64 28 63 68 61 6e 6e 65 6c 29 0a 09 09 09 09 nd(channel).....
1e30: 2e 62 69 6e 64 28 75 72 6c 5f 72 65 29 0a 09 09 .bind(url_re)...
1e40: 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 ...execute(&mut
1e50: 2a 73 65 6c 66 2e 30 29 2e 61 77 61 69 74 0a 09 *self.0).await..
1e60: 09 7b 0a 09 09 09 4f 6b 28 5f 29 20 3d 3e 20 4f .{....Ok(_) => O
1e70: 6b 28 6d 61 74 63 68 20 75 70 64 61 74 65 20 7b k(match update {
1e80: 0a 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e 20 .....Some(_) =>
1e90: 22 43 68 61 6e 6e 65 6c 20 75 70 64 61 74 65 64 "Channel updated
1ea0: 2e 22 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 .",.....None =>
1eb0: 22 43 68 61 6e 6e 65 6c 20 61 64 64 65 64 2e 22 "Channel added."
1ec0: 2c 0a 09 09 09 7d 29 2c 0a 09 09 09 45 72 72 28 ,....}),....Err(
1ed0: 73 71 6c 78 3a 3a 45 72 72 6f 72 3a 3a 44 61 74 sqlx::Error::Dat
1ee0: 61 62 61 73 65 28 65 72 72 29 29 20 3d 3e 20 7b abase(err)) => {
1ef0: 0a 09 09 09 09 6d 61 74 63 68 20 65 72 72 2e 64 .....match err.d
1f00: 6f 77 6e 63 61 73 74 3a 3a 3c 73 71 6c 78 3a 3a owncast::<sqlx::
1f10: 70 6f 73 74 67 72 65 73 3a 3a 50 67 44 61 74 61 postgres::PgData
1f20: 62 61 73 65 45 72 72 6f 72 3e 28 29 2e 72 6f 75 baseError>().rou
1f30: 74 69 6e 65 28 29 20 7b 0a 09 09 09 09 09 53 6f tine() {......So
1f40: 6d 65 28 22 5f 62 74 5f 63 68 65 63 6b 5f 75 6e me("_bt_check_un
1f50: 69 71 75 65 22 2c 20 29 20 3d 3e 20 7b 0a 09 09 ique", ) => {...
1f60: 09 09 09 09 4f 6b 28 22 44 75 70 6c 69 63 61 74 ....Ok("Duplicat
1f70: 65 20 6b 65 79 2e 22 29 0a 09 09 09 09 09 7d 2c e key.")......},
1f80: 0a 09 09 09 09 09 53 6f 6d 65 28 5f 29 20 3d 3e ......Some(_) =>
1f90: 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 44 61 74 {.......Ok("Dat
1fa0: 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 0a 09 abase error.")..
1fb0: 09 09 09 09 7d 2c 0a 09 09 09 09 09 4e 6f 6e 65 ....},......None
1fc0: 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b 28 22 => {.......Ok("
1fd0: 4e 6f 20 64 61 74 61 62 61 73 65 20 65 72 72 6f No database erro
1fe0: 72 20 65 78 74 72 61 63 74 65 64 2e 22 29 0a 09 r extracted.")..
1ff0: 09 09 09 09 7d 2c 0a 09 09 09 09 7d 0a 09 09 09 ....},.....}....
2000: 7d 2c 0a 09 09 09 45 72 72 28 65 72 72 29 20 3d },....Err(err) =
2010: 3e 20 7b 0a 09 09 09 09 62 61 69 6c 21 28 22 53 > {.....bail!("S
2020: 6f 72 72 79 2c 20 75 6e 6b 6e 6f 77 6e 20 65 72 orry, unknown er
2030: 72 6f 72 3a 5c 6e 7b 65 72 72 3a 23 3f 7d 5c 6e ror:\n{err:#?}\n
2040: 22 29 3b 0a 09 09 09 7d 2c 0a 09 09 7d 0a 09 7d ");....},...}..}
2050: 0a 7d 0a .}.