0000: 75 73 65 20 73 74 64 3a 3a 62 6f 72 72 6f 77 3a use std::borrow:
0010: 3a 43 6f 77 3b 0a 0a 75 73 65 20 61 6e 79 68 6f :Cow;..use anyho
0020: 77 3a 3a 7b 0a 09 52 65 73 75 6c 74 2c 0a 09 62 w::{..Result,..b
0030: 61 69 6c 2c 0a 7d 3b 0a 75 73 65 20 61 73 79 6e ail,.};.use asyn
0040: 63 5f 73 74 64 3a 3a 73 79 6e 63 3a 3a 7b 0a 09 c_std::sync::{..
0050: 41 72 63 2c 0a 09 4d 75 74 65 78 2c 0a 7d 3b 0a Arc,..Mutex,.};.
0060: 75 73 65 20 63 68 72 6f 6e 6f 3a 3a 7b 0a 09 44 use chrono::{..D
0070: 61 74 65 54 69 6d 65 2c 0a 09 46 69 78 65 64 4f ateTime,..FixedO
0080: 66 66 73 65 74 2c 0a 09 4c 6f 63 61 6c 2c 0a 7d ffset,..Local,.}
0090: 3b 0a 75 73 65 20 73 71 6c 78 3a 3a 7b 0a 09 50 ;.use sqlx::{..P
00a0: 6f 73 74 67 72 65 73 2c 0a 09 52 6f 77 2c 0a 09 ostgres,..Row,..
00b0: 70 6f 73 74 67 72 65 73 3a 3a 50 67 50 6f 6f 6c postgres::PgPool
00c0: 4f 70 74 69 6f 6e 73 2c 0a 09 70 6f 6f 6c 3a 3a Options,..pool::
00d0: 50 6f 6f 6c 43 6f 6e 6e 65 63 74 69 6f 6e 2c 0a PoolConnection,.
00e0: 7d 3b 0a 0a 23 5b 64 65 72 69 76 65 28 73 71 6c };..#[derive(sql
00f0: 78 3a 3a 46 72 6f 6d 52 6f 77 2c 20 44 65 62 75 x::FromRow, Debu
0100: 67 29 5d 0a 70 75 62 20 73 74 72 75 63 74 20 4c g)].pub struct L
0110: 69 73 74 20 7b 0a 09 70 75 62 20 73 6f 75 72 63 ist {..pub sourc
0120: 65 5f 69 64 3a 20 69 33 32 2c 0a 09 70 75 62 20 e_id: i32,..pub
0130: 63 68 61 6e 6e 65 6c 3a 20 53 74 72 69 6e 67 2c channel: String,
0140: 0a 09 70 75 62 20 65 6e 61 62 6c 65 64 3a 20 62 ..pub enabled: b
0150: 6f 6f 6c 2c 0a 09 70 75 62 20 75 72 6c 3a 20 53 ool,..pub url: S
0160: 74 72 69 6e 67 2c 0a 09 70 75 62 20 69 76 5f 68 tring,..pub iv_h
0170: 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 53 74 72 69 ash: Option<Stri
0180: 6e 67 3e 2c 0a 09 70 75 62 20 75 72 6c 5f 72 65 ng>,..pub url_re
0190: 3a 20 4f 70 74 69 6f 6e 3c 53 74 72 69 6e 67 3e : Option<String>
01a0: 2c 0a 7d 0a 0a 23 5b 64 65 72 69 76 65 28 73 71 ,.}..#[derive(sq
01b0: 6c 78 3a 3a 46 72 6f 6d 52 6f 77 2c 20 44 65 62 lx::FromRow, Deb
01c0: 75 67 29 5d 0a 70 75 62 20 73 74 72 75 63 74 20 ug)].pub struct
01d0: 53 6f 75 72 63 65 20 7b 0a 09 70 75 62 20 63 68 Source {..pub ch
01e0: 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c 0a 09 annel_id: i64,..
01f0: 70 75 62 20 75 72 6c 3a 20 53 74 72 69 6e 67 2c pub url: String,
0200: 0a 09 70 75 62 20 69 76 5f 68 61 73 68 3a 20 4f ..pub iv_hash: O
0210: 70 74 69 6f 6e 3c 53 74 72 69 6e 67 3e 2c 0a 09 ption<String>,..
0220: 70 75 62 20 6f 77 6e 65 72 3a 20 69 36 34 2c 0a pub owner: i64,.
0230: 09 70 75 62 20 75 72 6c 5f 72 65 3a 20 4f 70 74 .pub url_re: Opt
0240: 69 6f 6e 3c 53 74 72 69 6e 67 3e 2c 0a 7d 0a 0a ion<String>,.}..
0250: 23 5b 64 65 72 69 76 65 28 73 71 6c 78 3a 3a 46 #[derive(sqlx::F
0260: 72 6f 6d 52 6f 77 29 5d 0a 70 75 62 20 73 74 72 romRow)].pub str
0270: 75 63 74 20 51 75 65 75 65 20 7b 0a 09 70 75 62 uct Queue {..pub
0280: 20 73 6f 75 72 63 65 5f 69 64 3a 20 4f 70 74 69 source_id: Opti
0290: 6f 6e 3c 69 33 32 3e 2c 0a 09 70 75 62 20 6e 65 on<i32>,..pub ne
02a0: 78 74 5f 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e xt_fetch: Option
02b0: 3c 44 61 74 65 54 69 6d 65 3c 4c 6f 63 61 6c 3e <DateTime<Local>
02c0: 3e 2c 0a 09 70 75 62 20 6f 77 6e 65 72 3a 20 4f >,..pub owner: O
02d0: 70 74 69 6f 6e 3c 69 36 34 3e 2c 0a 7d 0a 0a 23 ption<i64>,.}..#
02e0: 5b 64 65 72 69 76 65 28 43 6c 6f 6e 65 29 5d 0a [derive(Clone)].
02f0: 70 75 62 20 73 74 72 75 63 74 20 44 62 20 7b 0a pub struct Db {.
0300: 09 70 6f 6f 6c 3a 20 41 72 63 3c 4d 75 74 65 78 .pool: Arc<Mutex
0310: 3c 73 71 6c 78 3a 3a 50 6f 6f 6c 3c 73 71 6c 78 <sqlx::Pool<sqlx
0320: 3a 3a 50 6f 73 74 67 72 65 73 3e 3e 3e 2c 0a 7d ::Postgres>>>,.}
0330: 0a 0a 70 75 62 20 73 74 72 75 63 74 20 43 6f 6e ..pub struct Con
0340: 6e 7b 0a 09 63 6f 6e 6e 3a 20 50 6f 6f 6c 43 6f n{..conn: PoolCo
0350: 6e 6e 65 63 74 69 6f 6e 3c 50 6f 73 74 67 72 65 nnection<Postgre
0360: 73 3e 2c 0a 7d 0a 0a 69 6d 70 6c 20 44 62 20 7b s>,.}..impl Db {
0370: 0a 09 70 75 62 20 66 6e 20 6e 65 77 20 28 70 67 ..pub fn new (pg
0380: 75 72 69 3a 20 26 73 74 72 29 20 2d 3e 20 52 65 uri: &str) -> Re
0390: 73 75 6c 74 3c 44 62 3e 20 7b 0a 09 09 4f 6b 28 sult<Db> {...Ok(
03a0: 44 62 7b 0a 09 09 09 70 6f 6f 6c 3a 20 41 72 63 Db{....pool: Arc
03b0: 3a 3a 6e 65 77 28 4d 75 74 65 78 3a 3a 6e 65 77 ::new(Mutex::new
03c0: 28 50 67 50 6f 6f 6c 4f 70 74 69 6f 6e 73 3a 3a (PgPoolOptions::
03d0: 6e 65 77 28 29 0a 09 09 09 09 2e 6d 61 78 5f 63 new()......max_c
03e0: 6f 6e 6e 65 63 74 69 6f 6e 73 28 35 29 0a 09 09 onnections(5)...
03f0: 09 09 2e 61 63 71 75 69 72 65 5f 74 69 6d 65 6f ...acquire_timeo
0400: 75 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 75 ut(std::time::Du
0410: 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 33 30 30 2c ration::new(300,
0420: 20 30 29 29 0a 09 09 09 09 2e 69 64 6c 65 5f 74 0))......idle_t
0430: 69 6d 65 6f 75 74 28 73 74 64 3a 3a 74 69 6d 65 imeout(std::time
0440: 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 ::Duration::new(
0450: 36 30 2c 20 30 29 29 0a 09 09 09 09 2e 63 6f 6e 60, 0))......con
0460: 6e 65 63 74 5f 6c 61 7a 79 28 70 67 75 72 69 29 nect_lazy(pguri)
0470: 3f 29 29 2c 0a 09 09 7d 29 0a 09 7d 0a 0a 09 70 ?)),...})..}...p
0480: 75 62 20 61 73 79 6e 63 20 66 6e 20 62 65 67 69 ub async fn begi
0490: 6e 28 26 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 n(&self) -> Resu
04a0: 6c 74 3c 43 6f 6e 6e 3e 20 7b 0a 09 09 6c 65 74 lt<Conn> {...let
04b0: 20 70 6f 6f 6c 20 3d 20 73 65 6c 66 2e 70 6f 6f pool = self.poo
04c0: 6c 2e 6c 6f 63 6b 5f 61 72 63 28 29 2e 61 77 61 l.lock_arc().awa
04d0: 69 74 3b 0a 09 09 6c 65 74 20 63 6f 6e 6e 20 3d it;...let conn =
04e0: 20 43 6f 6e 6e 3a 3a 6e 65 77 28 70 6f 6f 6c 2e Conn::new(pool.
04f0: 61 63 71 75 69 72 65 28 29 2e 61 77 61 69 74 3f acquire().await?
0500: 29 2e 61 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 63 ).await?;...Ok(c
0510: 6f 6e 6e 29 0a 09 7d 0a 7d 0a 0a 69 6d 70 6c 20 onn)..}.}..impl
0520: 43 6f 6e 6e 20 7b 0a 09 70 75 62 20 61 73 79 6e Conn {..pub asyn
0530: 63 20 66 6e 20 6e 65 77 20 28 63 6f 6e 6e 3a 20 c fn new (conn:
0540: 50 6f 6f 6c 43 6f 6e 6e 65 63 74 69 6f 6e 3c 50 PoolConnection<P
0550: 6f 73 74 67 72 65 73 3e 29 20 2d 3e 20 52 65 73 ostgres>) -> Res
0560: 75 6c 74 3c 43 6f 6e 6e 3e 20 7b 0a 09 09 4f 6b ult<Conn> {...Ok
0570: 28 43 6f 6e 6e 7b 0a 09 09 09 63 6f 6e 6e 2c 0a (Conn{....conn,.
0580: 09 09 7d 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 ..})..}...pub as
0590: 79 6e 63 20 66 6e 20 61 64 64 5f 70 6f 73 74 20 ync fn add_post
05a0: 28 26 6d 75 74 20 73 65 6c 66 2c 20 73 6f 75 72 (&mut self, sour
05b0: 63 65 5f 69 64 3a 20 69 33 32 2c 20 64 61 74 65 ce_id: i32, date
05c0: 3a 20 26 44 61 74 65 54 69 6d 65 3c 46 69 78 65 : &DateTime<Fixe
05d0: 64 4f 66 66 73 65 74 3e 2c 20 70 6f 73 74 5f 75 dOffset>, post_u
05e0: 72 6c 3a 20 26 73 74 72 29 20 2d 3e 20 52 65 73 rl: &str) -> Res
05f0: 75 6c 74 3c 28 29 3e 20 7b 0a 09 09 73 71 6c 78 ult<()> {...sqlx
0600: 3a 3a 71 75 65 72 79 28 22 69 6e 73 65 72 74 20 ::query("insert
0610: 69 6e 74 6f 20 72 73 73 74 67 5f 70 6f 73 74 20 into rsstg_post
0620: 28 73 6f 75 72 63 65 5f 69 64 2c 20 70 6f 73 74 (source_id, post
0630: 65 64 2c 20 75 72 6c 29 20 76 61 6c 75 65 73 20 ed, url) values
0640: 28 24 31 2c 20 24 32 2c 20 24 33 29 3b 22 29 0a ($1, $2, $3);").
0650: 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f ....bind(source_
0660: 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 64 61 74 id).....bind(dat
0670: 65 29 0a 09 09 09 2e 62 69 6e 64 28 70 6f 73 74 e).....bind(post
0680: 5f 75 72 6c 29 0a 09 09 09 2e 65 78 65 63 75 74 _url).....execut
0690: 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e 63 6f 6e e(&mut *self.con
06a0: 6e 29 2e 61 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 n).await?;...Ok(
06b0: 28 29 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ())..}...pub asy
06c0: 6e 63 20 66 6e 20 63 6c 65 61 6e 20 3c 49 3e 20 nc fn clean <I>
06d0: 28 26 6d 75 74 20 73 65 6c 66 2c 20 73 6f 75 72 (&mut self, sour
06e0: 63 65 5f 69 64 3a 20 69 33 32 2c 20 6f 77 6e 65 ce_id: i32, owne
06f0: 72 3a 20 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c r: I) -> Result<
0700: 43 6f 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 Cow<'_, str>>..w
0710: 68 65 72 65 20 49 3a 20 49 6e 74 6f 3c 69 36 34 here I: Into<i64
0720: 3e 20 7b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 > {...match sqlx
0730: 3a 3a 71 75 65 72 79 28 22 64 65 6c 65 74 65 20 ::query("delete
0740: 66 72 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 from rsstg_post
0750: 70 20 75 73 69 6e 67 20 72 73 73 74 67 5f 73 6f p using rsstg_so
0760: 75 72 63 65 20 73 20 77 68 65 72 65 20 70 2e 73 urce s where p.s
0770: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e ource_id = $1 an
0780: 64 20 6f 77 6e 65 72 20 3d 20 24 32 20 61 6e 64 d owner = $2 and
0790: 20 70 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 73 p.source_id = s
07a0: 2e 73 6f 75 72 63 65 5f 69 64 3b 22 29 0a 09 09 .source_id;")...
07b0: 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 ..bind(source_id
07c0: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
07d0: 2e 69 6e 74 6f 28 29 29 0a 09 09 09 2e 65 78 65 .into()).....exe
07e0: 63 75 74 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e cute(&mut *self.
07f0: 63 6f 6e 6e 29 2e 61 77 61 69 74 3f 2e 72 6f 77 conn).await?.row
0800: 73 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 s_affected() {..
0810: 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 ..0 => { Ok("No
0820: 64 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 data found found
0830: 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 .".into()) },...
0840: 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 .x => { Ok(forma
0850: 74 21 28 22 7b 78 7d 20 70 6f 73 74 73 20 70 75 t!("{x} posts pu
0860: 72 67 65 64 2e 22 29 2e 69 6e 74 6f 28 29 29 20 rged.").into())
0870: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 },...}..}...pub
0880: 61 73 79 6e 63 20 66 6e 20 64 65 6c 65 74 65 20 async fn delete
0890: 3c 49 3e 20 28 26 6d 75 74 20 73 65 6c 66 2c 20 <I> (&mut self,
08a0: 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 2c 20 source_id: i32,
08b0: 6f 77 6e 65 72 3a 20 49 29 20 2d 3e 20 52 65 73 owner: I) -> Res
08c0: 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 74 72 3e ult<Cow<'_, str>
08d0: 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 6e 74 6f >..where I: Into
08e0: 3c 69 36 34 3e 20 7b 0a 09 09 6d 61 74 63 68 20 <i64> {...match
08f0: 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 64 65 6c sqlx::query("del
0900: 65 74 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 ete from rsstg_s
0910: 6f 75 72 63 65 20 77 68 65 72 65 20 73 6f 75 72 ource where sour
0920: 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f ce_id = $1 and o
0930: 77 6e 65 72 20 3d 20 24 32 3b 22 29 0a 09 09 09 wner = $2;")....
0940: 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 .bind(source_id)
0950: 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 2e .....bind(owner.
0960: 69 6e 74 6f 28 29 29 0a 09 09 09 2e 65 78 65 63 into()).....exec
0970: 75 74 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e 63 ute(&mut *self.c
0980: 6f 6e 6e 29 2e 61 77 61 69 74 3f 2e 72 6f 77 73 onn).await?.rows
0990: 5f 61 66 66 65 63 74 65 64 28 29 20 7b 0a 09 09 _affected() {...
09a0: 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 4e 6f 20 64 .0 => { Ok("No d
09b0: 61 74 61 20 66 6f 75 6e 64 20 66 6f 75 6e 64 2e ata found found.
09c0: 22 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 09 ".into()) },....
09d0: 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f 72 6d 61 74 x => { Ok(format
09e0: 21 28 22 7b 7d 20 73 6f 75 72 63 65 73 20 72 65 !("{} sources re
09f0: 6d 6f 76 65 64 2e 22 2c 20 78 29 2e 69 6e 74 6f moved.", x).into
0a00: 28 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 ()) },...}..}...
0a10: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 64 69 73 pub async fn dis
0a20: 61 62 6c 65 20 3c 49 3e 20 28 26 6d 75 74 20 73 able <I> (&mut s
0a30: 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 elf, source_id:
0a40: 69 33 32 2c 20 6f 77 6e 65 72 3a 20 49 29 20 2d i32, owner: I) -
0a50: 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 > Result<&str>..
0a60: 77 68 65 72 65 20 49 3a 20 49 6e 74 6f 3c 69 36 where I: Into<i6
0a70: 34 3e 20 7b 0a 09 09 6d 61 74 63 68 20 73 71 6c 4> {...match sql
0a80: 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 x::query("update
0a90: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 rsstg_source se
0aa0: 74 20 65 6e 61 62 6c 65 64 20 3d 20 66 61 6c 73 t enabled = fals
0ab0: 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 e where source_i
0ac0: 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 d = $1 and owner
0ad0: 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 = $2").....bind
0ae0: 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e (source_id).....
0af0: 62 69 6e 64 28 6f 77 6e 65 72 2e 69 6e 74 6f 28 bind(owner.into(
0b00: 29 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 )).....execute(&
0b10: 6d 75 74 20 2a 73 65 6c 66 2e 63 6f 6e 6e 29 2e mut *self.conn).
0b20: 61 77 61 69 74 3f 2e 72 6f 77 73 5f 61 66 66 65 await?.rows_affe
0b30: 63 74 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e cted() {....1 =>
0b40: 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 64 69 { Ok("Source di
0b50: 73 61 62 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 09 sabled.") },....
0b60: 30 20 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 0 => { Ok("Sourc
0b70: 65 20 6e 6f 74 20 66 6f 75 6e 64 2e 22 29 20 7d e not found.") }
0b80: 2c 0a 09 09 09 5f 20 3d 3e 20 7b 20 62 61 69 6c ,...._ => { bail
0b90: 21 28 22 44 61 74 61 62 61 73 65 20 65 72 72 6f !("Database erro
0ba0: 72 2e 22 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a r.") },...}..}..
0bb0: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 65 6e .pub async fn en
0bc0: 61 62 6c 65 20 3c 49 3e 20 28 26 6d 75 74 20 73 able <I> (&mut s
0bd0: 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 elf, source_id:
0be0: 69 33 32 2c 20 6f 77 6e 65 72 3a 20 49 29 20 2d i32, owner: I) -
0bf0: 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 > Result<&str>..
0c00: 77 68 65 72 65 20 49 3a 20 49 6e 74 6f 3c 69 36 where I: Into<i6
0c10: 34 3e 20 7b 0a 09 09 6d 61 74 63 68 20 73 71 6c 4> {...match sql
0c20: 78 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 x::query("update
0c30: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 rsstg_source se
0c40: 74 20 65 6e 61 62 6c 65 64 20 3d 20 74 72 75 65 t enabled = true
0c50: 20 77 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 where source_id
0c60: 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e 65 72 20 = $1 and owner
0c70: 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 28 = $2").....bind(
0c80: 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 2e 62 source_id).....b
0c90: 69 6e 64 28 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 ind(owner.into()
0ca0: 29 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d ).....execute(&m
0cb0: 75 74 20 2a 73 65 6c 66 2e 63 6f 6e 6e 29 2e 61 ut *self.conn).a
0cc0: 77 61 69 74 3f 2e 72 6f 77 73 5f 61 66 66 65 63 wait?.rows_affec
0cd0: 74 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 ted() {....1 =>
0ce0: 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 65 6e 61 { Ok("Source ena
0cf0: 62 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 bled.") },....0
0d00: 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 => { Ok("Source
0d10: 6e 6f 74 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a not found.") },.
0d20: 09 09 09 5f 20 3d 3e 20 7b 20 62 61 69 6c 21 28 ..._ => { bail!(
0d30: 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e "Database error.
0d40: 22 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 ") },...}..}...p
0d50: 75 62 20 61 73 79 6e 63 20 66 6e 20 65 78 69 73 ub async fn exis
0d60: 74 73 20 3c 49 3e 20 28 26 6d 75 74 20 73 65 6c ts <I> (&mut sel
0d70: 66 2c 20 70 6f 73 74 5f 75 72 6c 3a 20 26 73 74 f, post_url: &st
0d80: 72 2c 20 69 64 3a 20 49 29 20 2d 3e 20 52 65 73 r, id: I) -> Res
0d90: 75 6c 74 3c 4f 70 74 69 6f 6e 3c 62 6f 6f 6c 3e ult<Option<bool>
0da0: 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 6e 74 6f >..where I: Into
0db0: 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 72 6f <i64> {...let ro
0dc0: 77 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 w = sqlx::query(
0dd0: 22 73 65 6c 65 63 74 20 65 78 69 73 74 73 28 73 "select exists(s
0de0: 65 6c 65 63 74 20 74 72 75 65 20 66 72 6f 6d 20 elect true from
0df0: 72 73 73 74 67 5f 70 6f 73 74 20 77 68 65 72 65 rsstg_post where
0e00: 20 75 72 6c 20 3d 20 24 31 20 61 6e 64 20 73 6f url = $1 and so
0e10: 75 72 63 65 5f 69 64 20 3d 20 24 32 29 20 61 73 urce_id = $2) as
0e20: 20 65 78 69 73 74 73 3b 22 29 0a 09 09 09 2e 62 exists;").....b
0e30: 69 6e 64 28 70 6f 73 74 5f 75 72 6c 29 0a 09 09 ind(post_url)...
0e40: 09 2e 62 69 6e 64 28 69 64 2e 69 6e 74 6f 28 29 ..bind(id.into()
0e50: 29 0a 09 09 09 2e 66 65 74 63 68 5f 6f 6e 65 28 ).....fetch_one(
0e60: 26 6d 75 74 20 2a 73 65 6c 66 2e 63 6f 6e 6e 29 &mut *self.conn)
0e70: 2e 61 77 61 69 74 3f 3b 0a 09 09 6c 65 74 20 65 .await?;...let e
0e80: 78 69 73 74 73 3a 20 4f 70 74 69 6f 6e 3c 62 6f xists: Option<bo
0e90: 6f 6c 3e 20 3d 20 72 6f 77 2e 74 72 79 5f 67 65 ol> = row.try_ge
0ea0: 74 28 22 65 78 69 73 74 73 22 29 3f 3b 0a 09 09 t("exists")?;...
0eb0: 4f 6b 28 65 78 69 73 74 73 29 0a 09 7d 0a 0a 09 Ok(exists)..}...
0ec0: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 67 65 74 pub async fn get
0ed0: 5f 71 75 65 75 65 20 28 26 6d 75 74 20 73 65 6c _queue (&mut sel
0ee0: 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 56 65 63 f) -> Result<Vec
0ef0: 3c 51 75 65 75 65 3e 3e 20 7b 0a 09 09 6c 65 74 <Queue>> {...let
0f00: 20 62 6c 6f 63 6b 3a 20 56 65 63 3c 51 75 65 75 block: Vec<Queu
0f10: 65 3e 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 e> = sqlx::query
0f20: 5f 61 73 28 22 73 65 6c 65 63 74 20 73 6f 75 72 _as("select sour
0f30: 63 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 63 ce_id, next_fetc
0f40: 68 2c 20 6f 77 6e 65 72 20 66 72 6f 6d 20 72 73 h, owner from rs
0f50: 73 74 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 stg_order natura
0f60: 6c 20 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 l left join rsst
0f70: 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 6e g_source where n
0f80: 65 78 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 ext_fetch < now(
0f90: 29 20 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 20 ) + interval '1
0fa0: 6d 69 6e 75 74 65 27 3b 22 29 0a 09 09 09 2e 66 minute';").....f
0fb0: 65 74 63 68 5f 61 6c 6c 28 26 6d 75 74 20 2a 73 etch_all(&mut *s
0fc0: 65 6c 66 2e 63 6f 6e 6e 29 2e 61 77 61 69 74 3f elf.conn).await?
0fd0: 3b 0a 09 09 4f 6b 28 62 6c 6f 63 6b 29 0a 09 7d ;...Ok(block)..}
0fe0: 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 ...pub async fn
0ff0: 67 65 74 5f 6c 69 73 74 20 3c 49 3e 20 28 26 6d get_list <I> (&m
1000: 75 74 20 73 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 ut self, owner:
1010: 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c 56 65 63 I) -> Result<Vec
1020: 3c 4c 69 73 74 3e 3e 0a 09 77 68 65 72 65 20 49 <List>>..where I
1030: 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 : Into<i64> {...
1040: 6c 65 74 20 73 6f 75 72 63 65 3a 20 56 65 63 3c let source: Vec<
1050: 4c 69 73 74 3e 20 3d 20 73 71 6c 78 3a 3a 71 75 List> = sqlx::qu
1060: 65 72 79 5f 61 73 28 22 73 65 6c 65 63 74 20 73 ery_as("select s
1070: 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 ource_id, channe
1080: 6c 2c 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c 2c l, enabled, url,
1090: 20 69 76 5f 68 61 73 68 2c 20 75 72 6c 5f 72 65 iv_hash, url_re
10a0: 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 from rsstg_sour
10b0: 63 65 20 77 68 65 72 65 20 6f 77 6e 65 72 20 3d ce where owner =
10c0: 20 24 31 20 6f 72 64 65 72 20 62 79 20 73 6f 75 $1 order by sou
10d0: 72 63 65 5f 69 64 22 29 0a 09 09 09 2e 62 69 6e rce_id").....bin
10e0: 64 28 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 29 0a d(owner.into()).
10f0: 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 26 6d ....fetch_all(&m
1100: 75 74 20 2a 73 65 6c 66 2e 63 6f 6e 6e 29 2e 61 ut *self.conn).a
1110: 77 61 69 74 3f 3b 0a 09 09 4f 6b 28 73 6f 75 72 wait?;...Ok(sour
1120: 63 65 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 ce)..}...pub asy
1130: 6e 63 20 66 6e 20 67 65 74 5f 73 6f 75 72 63 65 nc fn get_source
1140: 20 3c 49 3e 20 28 26 6d 75 74 20 73 65 6c 66 2c <I> (&mut self,
1150: 20 69 64 3a 20 69 33 32 2c 20 6f 77 6e 65 72 3a id: i32, owner:
1160: 20 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c 53 6f I) -> Result<So
1170: 75 72 63 65 3e 0a 09 77 68 65 72 65 20 49 3a 20 urce>..where I:
1180: 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 Into<i64> {...le
1190: 74 20 73 6f 75 72 63 65 3a 20 53 6f 75 72 63 65 t source: Source
11a0: 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 5f 61 = sqlx::query_a
11b0: 73 28 22 73 65 6c 65 63 74 20 63 68 61 6e 6e 65 s("select channe
11c0: 6c 5f 69 64 2c 20 75 72 6c 2c 20 69 76 5f 68 61 l_id, url, iv_ha
11d0: 73 68 2c 20 6f 77 6e 65 72 2c 20 75 72 6c 5f 72 sh, owner, url_r
11e0: 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 e from rsstg_sou
11f0: 72 63 65 20 77 68 65 72 65 20 73 6f 75 72 63 65 rce where source
1200: 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 6f 77 6e _id = $1 and own
1210: 65 72 20 3d 20 24 32 22 29 0a 09 09 09 2e 62 69 er = $2").....bi
1220: 6e 64 28 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 nd(id).....bind(
1230: 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 29 0a 09 09 owner.into())...
1240: 09 2e 66 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 ..fetch_one(&mut
1250: 20 2a 73 65 6c 66 2e 63 6f 6e 6e 29 2e 61 77 61 *self.conn).awa
1260: 69 74 3f 3b 0a 09 09 4f 6b 28 73 6f 75 72 63 65 it?;...Ok(source
1270: 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 )..}...pub async
1280: 20 66 6e 20 73 65 74 5f 73 63 72 61 70 65 20 3c fn set_scrape <
1290: 49 3e 20 28 26 6d 75 74 20 73 65 6c 66 2c 20 69 I> (&mut self, i
12a0: 64 3a 20 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c d: I) -> Result<
12b0: 28 29 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 6e ()>..where I: In
12c0: 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 73 71 6c 78 to<i64> {...sqlx
12d0: 3a 3a 71 75 65 72 79 28 22 75 70 64 61 74 65 20 ::query("update
12e0: 72 73 73 74 67 5f 73 6f 75 72 63 65 20 73 65 74 rsstg_source set
12f0: 20 6c 61 73 74 5f 73 63 72 61 70 65 20 3d 20 6e last_scrape = n
1300: 6f 77 28 29 20 77 68 65 72 65 20 73 6f 75 72 63 ow() where sourc
1310: 65 5f 69 64 20 3d 20 24 31 3b 22 29 0a 09 09 09 e_id = $1;")....
1320: 2e 62 69 6e 64 28 69 64 2e 69 6e 74 6f 28 29 29 .bind(id.into())
1330: 0a 09 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 .....execute(&mu
1340: 74 20 2a 73 65 6c 66 2e 63 6f 6e 6e 29 2e 61 77 t *self.conn).aw
1350: 61 69 74 3f 3b 0a 09 09 4f 6b 28 28 29 29 0a 09 ait?;...Ok(())..
1360: 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e }...pub async fn
1370: 20 75 70 64 61 74 65 20 3c 49 3e 20 28 26 6d 75 update <I> (&mu
1380: 74 20 73 65 6c 66 2c 20 75 70 64 61 74 65 3a 20 t self, update:
1390: 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c 20 63 68 61 Option<i32>, cha
13a0: 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 63 68 61 6e nnel: &str, chan
13b0: 6e 65 6c 5f 69 64 3a 20 69 36 34 2c 20 75 72 6c nel_id: i64, url
13c0: 3a 20 26 73 74 72 2c 20 69 76 5f 68 61 73 68 3a : &str, iv_hash:
13d0: 20 4f 70 74 69 6f 6e 3c 26 73 74 72 3e 2c 20 75 Option<&str>, u
13e0: 72 6c 5f 72 65 3a 20 4f 70 74 69 6f 6e 3c 26 73 rl_re: Option<&s
13f0: 74 72 3e 2c 20 6f 77 6e 65 72 3a 20 49 29 20 2d tr>, owner: I) -
1400: 3e 20 52 65 73 75 6c 74 3c 26 73 74 72 3e 0a 09 > Result<&str>..
1410: 77 68 65 72 65 20 49 3a 20 49 6e 74 6f 3c 69 36 where I: Into<i6
1420: 34 3e 20 7b 0a 09 09 6d 61 74 63 68 20 6d 61 74 4> {...match mat
1430: 63 68 20 75 70 64 61 74 65 20 7b 0a 09 09 09 09 ch update {.....
1440: 53 6f 6d 65 28 69 64 29 20 3d 3e 20 7b 0a 09 09 Some(id) => {...
1450: 09 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ...sqlx::query("
1460: 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 update rsstg_sou
1470: 72 63 65 20 73 65 74 20 63 68 61 6e 6e 65 6c 5f rce set channel_
1480: 69 64 20 3d 20 24 32 2c 20 75 72 6c 20 3d 20 24 id = $2, url = $
1490: 33 2c 20 69 76 5f 68 61 73 68 20 3d 20 24 34 2c 3, iv_hash = $4,
14a0: 20 6f 77 6e 65 72 20 3d 20 24 35 2c 20 63 68 61 owner = $5, cha
14b0: 6e 6e 65 6c 20 3d 20 24 36 2c 20 75 72 6c 5f 72 nnel = $6, url_r
14c0: 65 20 3d 20 24 37 20 77 68 65 72 65 20 73 6f 75 e = $7 where sou
14d0: 72 63 65 5f 69 64 20 3d 20 24 31 22 29 0a 09 09 rce_id = $1")...
14e0: 09 09 09 09 2e 62 69 6e 64 28 69 64 29 0a 09 09 .....bind(id)...
14f0: 09 09 7d 2c 0a 09 09 09 09 4e 6f 6e 65 20 3d 3e ..},.....None =>
1500: 20 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 {......sqlx::qu
1510: 65 72 79 28 22 69 6e 73 65 72 74 20 69 6e 74 6f ery("insert into
1520: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 28 63 rsstg_source (c
1530: 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 hannel_id, url,
1540: 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 iv_hash, owner,
1550: 63 68 61 6e 6e 65 6c 2c 20 75 72 6c 5f 72 65 29 channel, url_re)
1560: 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 2c values ($1, $2,
1570: 20 24 33 2c 20 24 34 2c 20 24 35 2c 20 24 36 29 $3, $4, $5, $6)
1580: 22 29 0a 09 09 09 09 7d 2c 0a 09 09 09 7d 0a 09 ").....},....}..
1590: 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c ....bind(channel
15a0: 5f 69 64 29 0a 09 09 09 09 2e 62 69 6e 64 28 75 _id)......bind(u
15b0: 72 6c 29 0a 09 09 09 09 2e 62 69 6e 64 28 69 76 rl)......bind(iv
15c0: 5f 68 61 73 68 29 0a 09 09 09 09 2e 62 69 6e 64 _hash)......bind
15d0: 28 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 29 0a 09 (owner.into())..
15e0: 09 09 09 2e 62 69 6e 64 28 63 68 61 6e 6e 65 6c ....bind(channel
15f0: 29 0a 09 09 09 09 2e 62 69 6e 64 28 75 72 6c 5f )......bind(url_
1600: 72 65 29 0a 09 09 09 09 2e 65 78 65 63 75 74 65 re)......execute
1610: 28 26 6d 75 74 20 2a 73 65 6c 66 2e 63 6f 6e 6e (&mut *self.conn
1620: 29 2e 61 77 61 69 74 0a 09 09 09 7b 0a 09 09 09 ).await....{....
1630: 4f 6b 28 5f 29 20 3d 3e 20 4f 6b 28 6d 61 74 63 Ok(_) => Ok(matc
1640: 68 20 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 h update {.....S
1650: 6f 6d 65 28 5f 29 20 3d 3e 20 22 43 68 61 6e 6e ome(_) => "Chann
1660: 65 6c 20 75 70 64 61 74 65 64 2e 22 2c 0a 09 09 el updated.",...
1670: 09 09 4e 6f 6e 65 20 3d 3e 20 22 43 68 61 6e 6e ..None => "Chann
1680: 65 6c 20 61 64 64 65 64 2e 22 2c 0a 09 09 09 7d el added.",....}
1690: 29 2c 0a 09 09 09 45 72 72 28 73 71 6c 78 3a 3a ),....Err(sqlx::
16a0: 45 72 72 6f 72 3a 3a 44 61 74 61 62 61 73 65 28 Error::Database(
16b0: 65 72 72 29 29 20 3d 3e 20 7b 0a 09 09 09 09 6d err)) => {.....m
16c0: 61 74 63 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 atch err.downcas
16d0: 74 3a 3a 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 t::<sqlx::postgr
16e0: 65 73 3a 3a 50 67 44 61 74 61 62 61 73 65 45 72 es::PgDatabaseEr
16f0: 72 6f 72 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 ror>().routine()
1700: 20 7b 0a 09 09 09 09 09 53 6f 6d 65 28 22 5f 62 {......Some("_b
1710: 74 5f 63 68 65 63 6b 5f 75 6e 69 71 75 65 22 2c t_check_unique",
1720: 20 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b ) => {.......Ok
1730: 28 22 44 75 70 6c 69 63 61 74 65 20 6b 65 79 2e ("Duplicate key.
1740: 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ")......},......
1750: 53 6f 6d 65 28 5f 29 20 3d 3e 20 7b 0a 09 09 09 Some(_) => {....
1760: 09 09 09 4f 6b 28 22 44 61 74 61 62 61 73 65 20 ...Ok("Database
1770: 65 72 72 6f 72 2e 22 29 0a 09 09 09 09 09 7d 2c error.")......},
1780: 0a 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a ......None => {.
1790: 09 09 09 09 09 09 4f 6b 28 22 4e 6f 20 64 61 74 ......Ok("No dat
17a0: 61 62 61 73 65 20 65 72 72 6f 72 20 65 78 74 72 abase error extr
17b0: 61 63 74 65 64 2e 22 29 0a 09 09 09 09 09 7d 2c acted.")......},
17c0: 0a 09 09 09 09 7d 0a 09 09 09 7d 2c 0a 09 09 09 .....}....},....
17d0: 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 Err(err) => {...
17e0: 09 09 62 61 69 6c 21 28 22 53 6f 72 72 79 2c 20 ..bail!("Sorry,
17f0: 75 6e 6b 6e 6f 77 6e 20 65 72 72 6f 72 3a 5c 6e unknown error:\n
1800: 7b 65 72 72 3a 23 3f 7d 5c 6e 22 29 3b 0a 09 09 {err:#?}\n");...
1810: 09 7d 2c 0a 09 09 7d 0a 09 7d 0a 7d 0a .},...}..}.}.