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 23 66 65 ..write!(f, "#fe
0240: 65 64 5f 7b 7d 20 2a ef b8 8f e2 83 a3 20 3c 63 ed_{} *ļøā£ <c
0250: 6f 64 65 3e 7b 7d 3c 2f 63 6f 64 65 3e 20 7b 7d ode>{}</code> {}
0260: 5c 6e f0 9f 94 97 20 3c 63 6f 64 65 3e 7b 7d 3c \nš <code>{}<
0270: 2f 63 6f 64 65 3e 22 2c 20 73 65 6c 66 2e 73 6f /code>", self.so
0280: 75 72 63 65 5f 69 64 2c 20 73 65 6c 66 2e 63 68 urce_id, self.ch
0290: 61 6e 6e 65 6c 2c 0a 09 09 09 6d 61 74 63 68 20 annel,....match
02a0: 73 65 6c 66 2e 65 6e 61 62 6c 65 64 20 7b 0a 09 self.enabled {..
02b0: 09 09 09 74 72 75 65 20 20 3d 3e 20 22 f0 9f 94 ...true => "š
02c0: 84 20 65 6e 61 62 6c 65 64 22 2c 0a 09 09 09 09 enabled",.....
02d0: 66 61 6c 73 65 20 3d 3e 20 22 e2 9b 94 20 64 69 false => "ā di
02e0: 73 61 62 6c 65 64 22 2c 0a 09 09 09 7d 2c 20 73 sabled",....}, s
02f0: 65 6c 66 2e 75 72 6c 29 3f 3b 0a 09 09 69 66 20 elf.url)?;...if
0300: 6c 65 74 20 53 6f 6d 65 28 69 76 5f 68 61 73 68 let Some(iv_hash
0310: 29 20 3d 20 26 73 65 6c 66 2e 69 76 5f 68 61 73 ) = &self.iv_has
0320: 68 20 7b 0a 09 09 09 77 72 69 74 65 21 28 66 2c h {....write!(f,
0330: 20 22 5c 6e 49 56 3a 20 3c 63 6f 64 65 3e 7b 69 "\nIV: <code>{i
0340: 76 5f 68 61 73 68 7d 3c 2f 63 6f 64 65 3e 22 29 v_hash}</code>")
0350: 3f 3b 0a 09 09 7d 0a 09 09 69 66 20 6c 65 74 20 ?;...}...if let
0360: 53 6f 6d 65 28 75 72 6c 5f 72 65 29 20 3d 20 26 Some(url_re) = &
0370: 73 65 6c 66 2e 75 72 6c 5f 72 65 20 7b 0a 09 09 self.url_re {...
0380: 09 77 72 69 74 65 21 28 66 2c 20 22 5c 6e 52 45 .write!(f, "\nRE
0390: 3a 20 3c 63 6f 64 65 3e 7b 75 72 6c 5f 72 65 7d : <code>{url_re}
03a0: 3c 2f 63 6f 64 65 3e 22 29 3f 3b 0a 09 09 7d 0a </code>")?;...}.
03b0: 09 09 4f 6b 28 28 29 29 0a 09 7d 0a 7d 0a 0a 2f ..Ok(())..}.}../
03c0: 2f 2f 20 4f 6e 65 20 66 65 65 64 2c 20 75 73 65 // One feed, use
03d0: 64 20 66 6f 72 20 63 61 63 68 69 6e 67 20 61 6e d for caching an
03e0: 64 20 6d 65 6e 75 20 6e 61 76 69 67 61 74 69 6f d menu navigatio
03f0: 6e 0a 23 5b 64 65 72 69 76 65 28 73 71 6c 78 3a n.#[derive(sqlx:
0400: 3a 46 72 6f 6d 52 6f 77 2c 20 44 65 62 75 67 29 :FromRow, Debug)
0410: 5d 0a 70 75 62 20 73 74 72 75 63 74 20 46 65 65 ].pub struct Fee
0420: 64 20 7b 0a 09 70 75 62 20 73 6f 75 72 63 65 5f d {..pub source_
0430: 69 64 3a 20 69 33 32 2c 0a 09 70 75 62 20 63 68 id: i32,..pub ch
0440: 61 6e 6e 65 6c 3a 20 53 74 72 69 6e 67 2c 0a 7d annel: String,.}
0450: 0a 0a 23 5b 64 65 72 69 76 65 28 73 71 6c 78 3a ..#[derive(sqlx:
0460: 3a 46 72 6f 6d 52 6f 77 2c 20 44 65 62 75 67 29 :FromRow, Debug)
0470: 5d 0a 70 75 62 20 73 74 72 75 63 74 20 53 6f 75 ].pub struct Sou
0480: 72 63 65 20 7b 0a 09 70 75 62 20 63 68 61 6e 6e rce {..pub chann
0490: 65 6c 5f 69 64 3a 20 69 36 34 2c 0a 09 70 75 62 el_id: i64,..pub
04a0: 20 75 72 6c 3a 20 53 74 72 69 6e 67 2c 0a 09 70 url: String,..p
04b0: 75 62 20 69 76 5f 68 61 73 68 3a 20 4f 70 74 69 ub iv_hash: Opti
04c0: 6f 6e 3c 53 74 72 69 6e 67 3e 2c 0a 09 70 75 62 on<String>,..pub
04d0: 20 6f 77 6e 65 72 3a 20 69 36 34 2c 0a 09 70 75 owner: i64,..pu
04e0: 62 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f 6e b url_re: Option
04f0: 3c 53 74 72 69 6e 67 3e 2c 0a 7d 0a 0a 23 5b 64 <String>,.}..#[d
0500: 65 72 69 76 65 28 73 71 6c 78 3a 3a 46 72 6f 6d erive(sqlx::From
0510: 52 6f 77 29 5d 0a 70 75 62 20 73 74 72 75 63 74 Row)].pub struct
0520: 20 51 75 65 75 65 20 7b 0a 09 70 75 62 20 73 6f Queue {..pub so
0530: 75 72 63 65 5f 69 64 3a 20 4f 70 74 69 6f 6e 3c urce_id: Option<
0540: 69 33 32 3e 2c 0a 09 70 75 62 20 6e 65 78 74 5f i32>,..pub next_
0550: 66 65 74 63 68 3a 20 4f 70 74 69 6f 6e 3c 44 61 fetch: Option<Da
0560: 74 65 54 69 6d 65 3c 4c 6f 63 61 6c 3e 3e 2c 0a teTime<Local>>,.
0570: 09 70 75 62 20 6f 77 6e 65 72 3a 20 4f 70 74 69 .pub owner: Opti
0580: 6f 6e 3c 69 36 34 3e 2c 0a 09 70 75 62 20 6c 61 on<i64>,..pub la
0590: 73 74 5f 73 63 72 61 70 65 3a 20 44 61 74 65 54 st_scrape: DateT
05a0: 69 6d 65 3c 4c 6f 63 61 6c 3e 2c 0a 7d 0a 0a 23 ime<Local>,.}..#
05b0: 5b 64 65 72 69 76 65 28 43 6c 6f 6e 65 29 5d 0a [derive(Clone)].
05c0: 70 75 62 20 73 74 72 75 63 74 20 44 62 20 28 0a pub struct Db (.
05d0: 09 41 72 63 3c 4d 75 74 65 78 3c 73 71 6c 78 3a .Arc<Mutex<sqlx:
05e0: 3a 50 6f 6f 6c 3c 73 71 6c 78 3a 3a 50 6f 73 74 :Pool<sqlx::Post
05f0: 67 72 65 73 3e 3e 3e 2c 0a 29 3b 0a 0a 69 6d 70 gres>>>,.);..imp
0600: 6c 20 44 62 20 7b 0a 09 70 75 62 20 66 6e 20 6e l Db {..pub fn n
0610: 65 77 20 28 70 67 75 72 69 3a 20 26 73 74 72 29 ew (pguri: &str)
0620: 20 2d 3e 20 52 65 73 75 6c 74 3c 44 62 3e 20 7b -> Result<Db> {
0630: 0a 09 09 4f 6b 28 44 62 20 28 0a 09 09 09 41 72 ...Ok(Db (....Ar
0640: 63 3a 3a 6e 65 77 28 4d 75 74 65 78 3a 3a 6e 65 c::new(Mutex::ne
0650: 77 28 50 67 50 6f 6f 6c 4f 70 74 69 6f 6e 73 3a w(PgPoolOptions:
0660: 3a 6e 65 77 28 29 0a 09 09 09 09 2e 6d 61 78 5f :new()......max_
0670: 63 6f 6e 6e 65 63 74 69 6f 6e 73 28 35 29 0a 09 connections(5)..
0680: 09 09 09 2e 61 63 71 75 69 72 65 5f 74 69 6d 65 ....acquire_time
0690: 6f 75 74 28 73 74 64 3a 3a 74 69 6d 65 3a 3a 44 out(std::time::D
06a0: 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 28 33 30 30 uration::new(300
06b0: 2c 20 30 29 29 0a 09 09 09 09 2e 69 64 6c 65 5f , 0))......idle_
06c0: 74 69 6d 65 6f 75 74 28 73 74 64 3a 3a 74 69 6d timeout(std::tim
06d0: 65 3a 3a 44 75 72 61 74 69 6f 6e 3a 3a 6e 65 77 e::Duration::new
06e0: 28 36 30 2c 20 30 29 29 0a 09 09 09 09 2e 63 6f (60, 0))......co
06f0: 6e 6e 65 63 74 5f 6c 61 7a 79 28 70 67 75 72 69 nnect_lazy(pguri
0700: 29 0a 09 09 09 09 2e 73 74 61 63 6b 28 29 3f 29 )......stack()?)
0710: 29 2c 0a 09 09 29 29 0a 09 7d 0a 0a 09 70 75 62 ),...))..}...pub
0720: 20 61 73 79 6e 63 20 66 6e 20 62 65 67 69 6e 28 async fn begin(
0730: 26 73 65 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 &self) -> Result
0740: 3c 43 6f 6e 6e 3e 20 7b 0a 09 09 6c 65 74 20 70 <Conn> {...let p
0750: 6f 6f 6c 20 3d 20 73 65 6c 66 2e 30 2e 6c 6f 63 ool = self.0.loc
0760: 6b 5f 61 72 63 28 29 2e 61 77 61 69 74 3b 0a 09 k_arc().await;..
0770: 09 6c 65 74 20 63 6f 6e 6e 20 3d 20 43 6f 6e 6e .let conn = Conn
0780: 20 28 20 70 6f 6f 6c 2e 61 63 71 75 69 72 65 28 ( pool.acquire(
0790: 29 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f ).await.stack()?
07a0: 20 29 3b 0a 09 09 4f 6b 28 63 6f 6e 6e 29 0a 09 );...Ok(conn)..
07b0: 7d 0a 7d 0a 0a 70 75 62 20 73 74 72 75 63 74 20 }.}..pub struct
07c0: 43 6f 6e 6e 20 28 0a 09 50 6f 6f 6c 43 6f 6e 6e Conn (..PoolConn
07d0: 65 63 74 69 6f 6e 3c 50 6f 73 74 67 72 65 73 3e ection<Postgres>
07e0: 2c 0a 29 3b 0a 0a 69 6d 70 6c 20 43 6f 6e 6e 20 ,.);..impl Conn
07f0: 7b 0a 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 {..pub async fn
0800: 61 64 64 5f 70 6f 73 74 20 28 26 6d 75 74 20 73 add_post (&mut s
0810: 65 6c 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 elf, source_id:
0820: 69 33 32 2c 20 64 61 74 65 3a 20 26 44 61 74 65 i32, date: &Date
0830: 54 69 6d 65 3c 46 69 78 65 64 4f 66 66 73 65 74 Time<FixedOffset
0840: 3e 2c 20 70 6f 73 74 5f 75 72 6c 3a 20 26 73 74 >, post_url: &st
0850: 72 29 20 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e r) -> Result<()>
0860: 20 7b 0a 09 09 73 71 6c 78 3a 3a 71 75 65 72 79 {...sqlx::query
0870: 28 22 69 6e 73 65 72 74 20 69 6e 74 6f 20 72 73 ("insert into rs
0880: 73 74 67 5f 70 6f 73 74 20 28 73 6f 75 72 63 65 stg_post (source
0890: 5f 69 64 2c 20 70 6f 73 74 65 64 2c 20 75 72 6c _id, posted, url
08a0: 29 20 76 61 6c 75 65 73 20 28 24 31 2c 20 24 32 ) values ($1, $2
08b0: 2c 20 24 33 29 3b 22 29 0a 09 09 09 2e 62 69 6e , $3);").....bin
08c0: 64 28 73 6f 75 72 63 65 5f 69 64 29 0a 09 09 09 d(source_id)....
08d0: 2e 62 69 6e 64 28 64 61 74 65 29 0a 09 09 09 2e .bind(date).....
08e0: 62 69 6e 64 28 70 6f 73 74 5f 75 72 6c 29 0a 09 bind(post_url)..
08f0: 09 09 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 ...execute(&mut
0900: 2a 73 65 6c 66 2e 30 29 2e 61 77 61 69 74 2e 73 *self.0).await.s
0910: 74 61 63 6b 28 29 3f 3b 0a 09 09 4f 6b 28 28 29 tack()?;...Ok(()
0920: 29 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 )..}...pub async
0930: 20 66 6e 20 63 6c 65 61 6e 20 3c 49 3e 20 28 26 fn clean <I> (&
0940: 6d 75 74 20 73 65 6c 66 2c 20 73 6f 75 72 63 65 mut self, source
0950: 5f 69 64 3a 20 69 33 32 2c 20 6f 77 6e 65 72 3a _id: i32, owner:
0960: 20 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c 43 6f I) -> Result<Co
0970: 77 3c 27 5f 2c 20 73 74 72 3e 3e 0a 09 77 68 65 w<'_, str>>..whe
0980: 72 65 20 49 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 re I: Into<i64>
0990: 7b 0a 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a {...match sqlx::
09a0: 71 75 65 72 79 28 22 64 65 6c 65 74 65 20 66 72 query("delete fr
09b0: 6f 6d 20 72 73 73 74 67 5f 70 6f 73 74 20 70 20 om rsstg_post p
09c0: 75 73 69 6e 67 20 72 73 73 74 67 5f 73 6f 75 72 using rsstg_sour
09d0: 63 65 20 73 20 77 68 65 72 65 20 70 2e 73 6f 75 ce s where p.sou
09e0: 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 20 rce_id = $1 and
09f0: 6f 77 6e 65 72 20 3d 20 24 32 20 61 6e 64 20 70 owner = $2 and p
0a00: 2e 73 6f 75 72 63 65 5f 69 64 20 3d 20 73 2e 73 .source_id = s.s
0a10: 6f 75 72 63 65 5f 69 64 3b 22 29 0a 09 09 09 2e ource_id;").....
0a20: 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 29 0a bind(source_id).
0a30: 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 2e 69 ....bind(owner.i
0a40: 6e 74 6f 28 29 29 0a 09 09 09 2e 65 78 65 63 75 nto()).....execu
0a50: 74 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e 30 29 te(&mut *self.0)
0a60: 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 2e .await.stack()?.
0a70: 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 29 20 rows_affected()
0a80: 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 4f 6b 28 22 {....0 => { Ok("
0a90: 4e 6f 20 64 61 74 61 20 66 6f 75 6e 64 20 66 6f No data found fo
0aa0: 75 6e 64 2e 22 2e 69 6e 74 6f 28 29 29 20 7d 2c und.".into()) },
0ab0: 0a 09 09 09 78 20 3d 3e 20 7b 20 4f 6b 28 66 6f ....x => { Ok(fo
0ac0: 72 6d 61 74 21 28 22 7b 78 7d 20 70 6f 73 74 73 rmat!("{x} posts
0ad0: 20 70 75 72 67 65 64 2e 22 29 2e 69 6e 74 6f 28 purged.").into(
0ae0: 29 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 )) },...}..}...p
0af0: 75 62 20 61 73 79 6e 63 20 66 6e 20 64 65 6c 65 ub async fn dele
0b00: 74 65 20 3c 49 3e 20 28 26 6d 75 74 20 73 65 6c te <I> (&mut sel
0b10: 66 2c 20 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 f, source_id: i3
0b20: 32 2c 20 6f 77 6e 65 72 3a 20 49 29 20 2d 3e 20 2, owner: I) ->
0b30: 52 65 73 75 6c 74 3c 43 6f 77 3c 27 5f 2c 20 73 Result<Cow<'_, s
0b40: 74 72 3e 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 tr>>..where I: I
0b50: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6d 61 74 nto<i64> {...mat
0b60: 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ch sqlx::query("
0b70: 64 65 6c 65 74 65 20 66 72 6f 6d 20 72 73 73 74 delete from rsst
0b80: 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 73 g_source where s
0b90: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e ource_id = $1 an
0ba0: 64 20 6f 77 6e 65 72 20 3d 20 24 32 3b 22 29 0a d owner = $2;").
0bb0: 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f ....bind(source_
0bc0: 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e id).....bind(own
0bd0: 65 72 2e 69 6e 74 6f 28 29 29 0a 09 09 09 2e 65 er.into()).....e
0be0: 78 65 63 75 74 65 28 26 6d 75 74 20 2a 73 65 6c xecute(&mut *sel
0bf0: 66 2e 30 29 2e 61 77 61 69 74 2e 73 74 61 63 6b f.0).await.stack
0c00: 28 29 3f 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 ()?.rows_affecte
0c10: 64 28 29 20 7b 0a 09 09 09 30 20 3d 3e 20 7b 20 d() {....0 => {
0c20: 4f 6b 28 22 4e 6f 20 64 61 74 61 20 66 6f 75 6e Ok("No data foun
0c30: 64 20 66 6f 75 6e 64 2e 22 2e 69 6e 74 6f 28 29 d found.".into()
0c40: 29 20 7d 2c 0a 09 09 09 78 20 3d 3e 20 7b 20 4f ) },....x => { O
0c50: 6b 28 66 6f 72 6d 61 74 21 28 22 7b 78 7d 20 73 k(format!("{x} s
0c60: 6f 75 72 63 65 73 20 72 65 6d 6f 76 65 64 2e 22 ources removed."
0c70: 29 2e 69 6e 74 6f 28 29 29 20 7d 2c 0a 09 09 7d ).into()) },...}
0c80: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
0c90: 66 6e 20 64 69 73 61 62 6c 65 20 3c 49 3e 20 28 fn disable <I> (
0ca0: 26 6d 75 74 20 73 65 6c 66 2c 20 73 6f 75 72 63 &mut self, sourc
0cb0: 65 5f 69 64 3a 20 69 33 32 2c 20 6f 77 6e 65 72 e_id: i32, owner
0cc0: 3a 20 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 : I) -> Result<&
0cd0: 73 74 72 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 str>..where I: I
0ce0: 6e 74 6f 3c 69 36 34 3e 20 7b 0a 09 09 6d 61 74 nto<i64> {...mat
0cf0: 63 68 20 73 71 6c 78 3a 3a 71 75 65 72 79 28 22 ch sqlx::query("
0d00: 75 70 64 61 74 65 20 72 73 73 74 67 5f 73 6f 75 update rsstg_sou
0d10: 72 63 65 20 73 65 74 20 65 6e 61 62 6c 65 64 20 rce set enabled
0d20: 3d 20 66 61 6c 73 65 20 77 68 65 72 65 20 73 6f = false where so
0d30: 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e 64 urce_id = $1 and
0d40: 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 09 owner = $2")...
0d50: 09 2e 62 69 6e 64 28 73 6f 75 72 63 65 5f 69 64 ..bind(source_id
0d60: 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 65 72 ).....bind(owner
0d70: 2e 69 6e 74 6f 28 29 29 0a 09 09 09 2e 65 78 65 .into()).....exe
0d80: 63 75 74 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e cute(&mut *self.
0d90: 30 29 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 0).await.stack()
0da0: 3f 2e 72 6f 77 73 5f 61 66 66 65 63 74 65 64 28 ?.rows_affected(
0db0: 29 20 7b 0a 09 09 09 31 20 3d 3e 20 7b 20 4f 6b ) {....1 => { Ok
0dc0: 28 22 53 6f 75 72 63 65 20 64 69 73 61 62 6c 65 ("Source disable
0dd0: 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 3d 3e 20 d.") },....0 =>
0de0: 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 6e 6f 74 { Ok("Source not
0df0: 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a 09 09 09 found.") },....
0e00: 5f 20 3d 3e 20 7b 20 62 61 69 6c 21 28 22 44 61 _ => { bail!("Da
0e10: 74 61 62 61 73 65 20 65 72 72 6f 72 2e 22 29 20 tabase error.")
0e20: 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 70 75 62 20 },...}..}...pub
0e30: 61 73 79 6e 63 20 66 6e 20 65 6e 61 62 6c 65 20 async fn enable
0e40: 3c 49 3e 20 28 26 6d 75 74 20 73 65 6c 66 2c 20 <I> (&mut self,
0e50: 73 6f 75 72 63 65 5f 69 64 3a 20 69 33 32 2c 20 source_id: i32,
0e60: 6f 77 6e 65 72 3a 20 49 29 20 2d 3e 20 52 65 73 owner: I) -> Res
0e70: 75 6c 74 3c 26 73 74 72 3e 0a 09 77 68 65 72 65 ult<&str>..where
0e80: 20 49 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a I: Into<i64> {.
0e90: 09 09 6d 61 74 63 68 20 73 71 6c 78 3a 3a 71 75 ..match sqlx::qu
0ea0: 65 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 ery("update rsst
0eb0: 67 5f 73 6f 75 72 63 65 20 73 65 74 20 65 6e 61 g_source set ena
0ec0: 62 6c 65 64 20 3d 20 74 72 75 65 20 77 68 65 72 bled = true wher
0ed0: 65 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 e source_id = $1
0ee0: 20 61 6e 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 and owner = $2"
0ef0: 29 0a 09 09 09 2e 62 69 6e 64 28 73 6f 75 72 63 ).....bind(sourc
0f00: 65 5f 69 64 29 0a 09 09 09 2e 62 69 6e 64 28 6f e_id).....bind(o
0f10: 77 6e 65 72 2e 69 6e 74 6f 28 29 29 0a 09 09 09 wner.into())....
0f20: 2e 65 78 65 63 75 74 65 28 26 6d 75 74 20 2a 73 .execute(&mut *s
0f30: 65 6c 66 2e 30 29 2e 61 77 61 69 74 2e 73 74 61 elf.0).await.sta
0f40: 63 6b 28 29 3f 2e 72 6f 77 73 5f 61 66 66 65 63 ck()?.rows_affec
0f50: 74 65 64 28 29 20 7b 0a 09 09 09 31 20 3d 3e 20 ted() {....1 =>
0f60: 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 65 6e 61 { Ok("Source ena
0f70: 62 6c 65 64 2e 22 29 20 7d 2c 0a 09 09 09 30 20 bled.") },....0
0f80: 3d 3e 20 7b 20 4f 6b 28 22 53 6f 75 72 63 65 20 => { Ok("Source
0f90: 6e 6f 74 20 66 6f 75 6e 64 2e 22 29 20 7d 2c 0a not found.") },.
0fa0: 09 09 09 5f 20 3d 3e 20 7b 20 62 61 69 6c 21 28 ..._ => { bail!(
0fb0: 22 44 61 74 61 62 61 73 65 20 65 72 72 6f 72 2e "Database error.
0fc0: 22 29 20 7d 2c 0a 09 09 7d 0a 09 7d 0a 0a 09 2f ") },...}..}.../
0fd0: 2f 2f 20 43 68 65 63 6b 73 20 77 68 65 74 68 65 // Checks whethe
0fe0: 72 20 61 20 70 6f 73 74 20 77 69 74 68 20 74 68 r a post with th
0ff0: 65 20 67 69 76 65 6e 20 55 52 4c 20 65 78 69 73 e given URL exis
1000: 74 73 20 66 6f 72 20 74 68 65 20 73 70 65 63 69 ts for the speci
1010: 66 69 65 64 20 73 6f 75 72 63 65 2e 0a 09 2f 2f fied source...//
1020: 2f 0a 09 2f 2f 2f 20 23 20 50 61 72 61 6d 65 74 /../// # Paramet
1030: 65 72 73 0a 09 2f 2f 2f 20 2d 20 60 70 6f 73 74 ers../// - `post
1040: 5f 75 72 6c 60 3a 20 54 68 65 20 55 52 4c 20 6f _url`: The URL o
1050: 66 20 74 68 65 20 70 6f 73 74 20 74 6f 20 63 68 f the post to ch
1060: 65 63 6b 2e 0a 09 2f 2f 2f 20 2d 20 60 69 64 60 eck.../// - `id`
1070: 3a 20 54 68 65 20 73 6f 75 72 63 65 20 69 64 65 : The source ide
1080: 6e 74 69 66 69 65 72 20 28 63 6f 6e 76 65 72 74 ntifier (convert
1090: 65 64 20 74 6f 20 60 69 36 34 60 29 2e 0a 09 2f ed to `i64`).../
10a0: 2f 2f 0a 09 2f 2f 2f 20 23 20 52 65 74 75 72 6e //../// # Return
10b0: 73 0a 09 2f 2f 2f 20 60 74 72 75 65 60 20 69 66 s../// `true` if
10c0: 20 61 20 70 6f 73 74 20 77 69 74 68 20 74 68 65 a post with the
10d0: 20 55 52 4c 20 65 78 69 73 74 73 20 66 6f 72 20 URL exists for
10e0: 74 68 65 20 73 6f 75 72 63 65 2c 20 60 66 61 6c the source, `fal
10f0: 73 65 60 20 6f 74 68 65 72 77 69 73 65 2e 0a 09 se` otherwise...
1100: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 65 78 69 pub async fn exi
1110: 73 74 73 20 3c 49 3e 20 28 26 6d 75 74 20 73 65 sts <I> (&mut se
1120: 6c 66 2c 20 70 6f 73 74 5f 75 72 6c 3a 20 26 73 lf, post_url: &s
1130: 74 72 2c 20 69 64 3a 20 49 29 20 2d 3e 20 52 65 tr, id: I) -> Re
1140: 73 75 6c 74 3c 62 6f 6f 6c 3e 0a 09 77 68 65 72 sult<bool>..wher
1150: 65 20 49 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e I: Into<i64> {
1160: 0a 09 09 6c 65 74 20 72 6f 77 20 3d 20 73 71 6c ...let row = sql
1170: 78 3a 3a 71 75 65 72 79 28 22 73 65 6c 65 63 74 x::query("select
1180: 20 65 78 69 73 74 73 28 73 65 6c 65 63 74 20 74 exists(select t
1190: 72 75 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 70 rue from rsstg_p
11a0: 6f 73 74 20 77 68 65 72 65 20 75 72 6c 20 3d 20 ost where url =
11b0: 24 31 20 61 6e 64 20 73 6f 75 72 63 65 5f 69 64 $1 and source_id
11c0: 20 3d 20 24 32 29 20 61 73 20 65 78 69 73 74 73 = $2) as exists
11d0: 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 70 6f 73 ;").....bind(pos
11e0: 74 5f 75 72 6c 29 0a 09 09 09 2e 62 69 6e 64 28 t_url).....bind(
11f0: 69 64 2e 69 6e 74 6f 28 29 29 0a 09 09 09 2e 66 id.into()).....f
1200: 65 74 63 68 5f 6f 6e 65 28 26 6d 75 74 20 2a 73 etch_one(&mut *s
1210: 65 6c 66 2e 30 29 2e 61 77 61 69 74 2e 73 74 61 elf.0).await.sta
1220: 63 6b 28 29 3f 3b 0a 09 09 72 6f 77 2e 74 72 79 ck()?;...row.try
1230: 5f 67 65 74 28 22 65 78 69 73 74 73 22 29 0a 09 _get("exists")..
1240: 09 09 2e 73 74 61 63 6b 5f 65 72 72 28 22 44 61 ...stack_err("Da
1250: 74 61 62 61 73 65 20 65 72 72 6f 72 3a 20 63 61 tabase error: ca
1260: 6e 27 74 20 63 68 65 63 6b 20 77 68 65 74 68 65 n't check whethe
1270: 72 20 70 6f 73 74 20 65 78 69 73 74 73 2e 22 29 r post exists.")
1280: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
1290: 66 6e 20 67 65 74 5f 66 65 65 64 73 20 3c 49 3e fn get_feeds <I>
12a0: 28 26 6d 75 74 20 73 65 6c 66 2c 20 6f 77 6e 65 (&mut self, owne
12b0: 72 3a 20 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c r: I) -> Result<
12c0: 56 65 63 3c 46 65 65 64 3e 3e 0a 09 77 68 65 72 Vec<Feed>>..wher
12d0: 65 20 49 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e I: Into<i64> {
12e0: 0a 09 09 6c 65 74 20 62 6c 6f 63 6b 3a 20 56 65 ...let block: Ve
12f0: 63 3c 46 65 65 64 3e 20 3d 20 73 71 6c 78 3a 3a c<Feed> = sqlx::
1300: 71 75 65 72 79 5f 61 73 28 22 73 65 6c 65 63 74 query_as("select
1310: 20 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e source_id, chan
1320: 6e 65 6c 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 nel from rsstg_s
1330: 6f 75 72 63 65 20 77 68 65 72 65 20 6f 77 6e 65 ource where owne
1340: 72 20 3d 20 24 31 20 6f 72 64 65 72 20 62 79 20 r = $1 order by
1350: 73 6f 75 72 63 65 5f 69 64 22 29 0a 09 09 09 2e source_id").....
1360: 62 69 6e 64 28 6f 77 6e 65 72 2e 69 6e 74 6f 28 bind(owner.into(
1370: 29 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c )).....fetch_all
1380: 28 26 6d 75 74 20 2a 73 65 6c 66 2e 30 29 2e 61 (&mut *self.0).a
1390: 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 wait.stack()?;..
13a0: 09 4f 6b 28 62 6c 6f 63 6b 29 0a 09 7d 0a 0a 09 .Ok(block)..}...
13b0: 2f 2f 2f 20 47 65 74 20 61 6c 6c 20 70 65 6e 64 /// Get all pend
13c0: 69 6e 67 20 65 76 65 6e 74 73 20 66 6f 72 20 28 ing events for (
13d0: 6e 6f 77 20 2b 20 31 20 6d 69 6e 75 74 65 29 0a now + 1 minute).
13e0: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 67 65 .pub async fn ge
13f0: 74 5f 71 75 65 75 65 20 28 26 6d 75 74 20 73 65 t_queue (&mut se
1400: 6c 66 29 20 2d 3e 20 52 65 73 75 6c 74 3c 56 65 lf) -> Result<Ve
1410: 63 3c 51 75 65 75 65 3e 3e 20 7b 0a 09 09 6c 65 c<Queue>> {...le
1420: 74 20 62 6c 6f 63 6b 3a 20 56 65 63 3c 51 75 65 t block: Vec<Que
1430: 75 65 3e 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 ue> = sqlx::quer
1440: 79 5f 61 73 28 22 73 65 6c 65 63 74 20 73 6f 75 y_as("select sou
1450: 72 63 65 5f 69 64 2c 20 6e 65 78 74 5f 66 65 74 rce_id, next_fet
1460: 63 68 2c 20 6f 77 6e 65 72 2c 20 6c 61 73 74 5f ch, owner, last_
1470: 73 63 72 61 70 65 20 66 72 6f 6d 20 72 73 73 74 scrape from rsst
1480: 67 5f 6f 72 64 65 72 20 6e 61 74 75 72 61 6c 20 g_order natural
1490: 6c 65 66 74 20 6a 6f 69 6e 20 72 73 73 74 67 5f left join rsstg_
14a0: 73 6f 75 72 63 65 20 77 68 65 72 65 20 6e 65 78 source where nex
14b0: 74 5f 66 65 74 63 68 20 3c 20 6e 6f 77 28 29 20 t_fetch < now()
14c0: 2b 20 69 6e 74 65 72 76 61 6c 20 27 31 20 6d 69 + interval '1 mi
14d0: 6e 75 74 65 27 3b 22 29 0a 09 09 09 2e 66 65 74 nute';").....fet
14e0: 63 68 5f 61 6c 6c 28 26 6d 75 74 20 2a 73 65 6c ch_all(&mut *sel
14f0: 66 2e 30 29 2e 61 77 61 69 74 2e 73 74 61 63 6b f.0).await.stack
1500: 28 29 3f 3b 0a 09 09 4f 6b 28 62 6c 6f 63 6b 29 ()?;...Ok(block)
1510: 0a 09 7d 0a 0a 09 70 75 62 20 61 73 79 6e 63 20 ..}...pub async
1520: 66 6e 20 67 65 74 5f 6c 69 73 74 20 3c 49 3e 28 fn get_list <I>(
1530: 26 6d 75 74 20 73 65 6c 66 2c 20 6f 77 6e 65 72 &mut self, owner
1540: 3a 20 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c 56 : I) -> Result<V
1550: 65 63 3c 4c 69 73 74 3e 3e 0a 09 77 68 65 72 65 ec<List>>..where
1560: 20 49 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b 0a I: Into<i64> {.
1570: 09 09 6c 65 74 20 73 6f 75 72 63 65 3a 20 56 65 ..let source: Ve
1580: 63 3c 4c 69 73 74 3e 20 3d 20 73 71 6c 78 3a 3a c<List> = sqlx::
1590: 71 75 65 72 79 5f 61 73 28 22 73 65 6c 65 63 74 query_as("select
15a0: 20 73 6f 75 72 63 65 5f 69 64 2c 20 63 68 61 6e source_id, chan
15b0: 6e 65 6c 2c 20 65 6e 61 62 6c 65 64 2c 20 75 72 nel, enabled, ur
15c0: 6c 2c 20 69 76 5f 68 61 73 68 2c 20 75 72 6c 5f l, iv_hash, url_
15d0: 72 65 20 66 72 6f 6d 20 72 73 73 74 67 5f 73 6f re from rsstg_so
15e0: 75 72 63 65 20 77 68 65 72 65 20 6f 77 6e 65 72 urce where owner
15f0: 20 3d 20 24 31 20 6f 72 64 65 72 20 62 79 20 73 = $1 order by s
1600: 6f 75 72 63 65 5f 69 64 22 29 0a 09 09 09 2e 62 ource_id").....b
1610: 69 6e 64 28 6f 77 6e 65 72 2e 69 6e 74 6f 28 29 ind(owner.into()
1620: 29 0a 09 09 09 2e 66 65 74 63 68 5f 61 6c 6c 28 ).....fetch_all(
1630: 26 6d 75 74 20 2a 73 65 6c 66 2e 30 29 2e 61 77 &mut *self.0).aw
1640: 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 ait.stack()?;...
1650: 4f 6b 28 73 6f 75 72 63 65 29 0a 09 7d 0a 0a 09 Ok(source)..}...
1660: 70 75 62 20 61 73 79 6e 63 20 66 6e 20 67 65 74 pub async fn get
1670: 5f 6f 6e 65 20 3c 49 3e 20 28 26 6d 75 74 20 73 _one <I> (&mut s
1680: 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 49 2c 20 69 elf, owner: I, i
1690: 64 3a 20 69 33 32 29 20 2d 3e 20 52 65 73 75 6c d: i32) -> Resul
16a0: 74 3c 4f 70 74 69 6f 6e 3c 4c 69 73 74 3e 3e 0a t<Option<List>>.
16b0: 09 77 68 65 72 65 20 49 3a 20 49 6e 74 6f 3c 69 .where I: Into<i
16c0: 36 34 3e 20 7b 0a 09 09 6c 65 74 20 73 6f 75 72 64> {...let sour
16d0: 63 65 3a 20 4f 70 74 69 6f 6e 3c 4c 69 73 74 3e ce: Option<List>
16e0: 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 79 5f 61 = sqlx::query_a
16f0: 73 28 22 73 65 6c 65 63 74 20 73 6f 75 72 63 65 s("select source
1700: 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c 20 65 6e _id, channel, en
1710: 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 76 5f 68 abled, url, iv_h
1720: 61 73 68 2c 20 75 72 6c 5f 72 65 20 66 72 6f 6d ash, url_re from
1730: 20 72 73 73 74 67 5f 73 6f 75 72 63 65 20 77 68 rsstg_source wh
1740: 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 31 20 61 ere owner = $1 a
1750: 6e 64 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 nd source_id = $
1760: 32 22 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 6e 2").....bind(own
1770: 65 72 2e 69 6e 74 6f 28 29 29 0a 09 09 09 2e 62 er.into()).....b
1780: 69 6e 64 28 69 64 29 0a 09 09 09 2e 66 65 74 63 ind(id).....fetc
1790: 68 5f 6f 70 74 69 6f 6e 61 6c 28 26 6d 75 74 20 h_optional(&mut
17a0: 2a 73 65 6c 66 2e 30 29 2e 61 77 61 69 74 2e 73 *self.0).await.s
17b0: 74 61 63 6b 28 29 3f 3b 0a 09 09 4f 6b 28 73 6f tack()?;...Ok(so
17c0: 75 72 63 65 29 0a 09 7d 0a 0a 09 70 75 62 20 61 urce)..}...pub a
17d0: 73 79 6e 63 20 66 6e 20 67 65 74 5f 6f 6e 65 5f sync fn get_one_
17e0: 6e 61 6d 65 20 3c 49 3e 20 28 26 6d 75 74 20 73 name <I> (&mut s
17f0: 65 6c 66 2c 20 6f 77 6e 65 72 3a 20 49 2c 20 6e elf, owner: I, n
1800: 61 6d 65 3a 20 26 73 74 72 29 20 2d 3e 20 52 65 ame: &str) -> Re
1810: 73 75 6c 74 3c 4f 70 74 69 6f 6e 3c 4c 69 73 74 sult<Option<List
1820: 3e 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 6e 74 >>..where I: Int
1830: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6c 65 74 20 73 o<i64> {...let s
1840: 6f 75 72 63 65 3a 20 4f 70 74 69 6f 6e 3c 4c 69 ource: Option<Li
1850: 73 74 3e 20 3d 20 73 71 6c 78 3a 3a 71 75 65 72 st> = sqlx::quer
1860: 79 5f 61 73 28 22 73 65 6c 65 63 74 20 73 6f 75 y_as("select sou
1870: 72 63 65 5f 69 64 2c 20 63 68 61 6e 6e 65 6c 2c rce_id, channel,
1880: 20 65 6e 61 62 6c 65 64 2c 20 75 72 6c 2c 20 69 enabled, url, i
1890: 76 5f 68 61 73 68 2c 20 75 72 6c 5f 72 65 20 66 v_hash, url_re f
18a0: 72 6f 6d 20 72 73 73 74 67 5f 73 6f 75 72 63 65 rom rsstg_source
18b0: 20 77 68 65 72 65 20 6f 77 6e 65 72 20 3d 20 24 where owner = $
18c0: 31 20 61 6e 64 20 63 68 61 6e 6e 65 6c 20 3d 20 1 and channel =
18d0: 24 32 22 29 0a 09 09 09 2e 62 69 6e 64 28 6f 77 $2").....bind(ow
18e0: 6e 65 72 2e 69 6e 74 6f 28 29 29 0a 09 09 09 2e ner.into()).....
18f0: 62 69 6e 64 28 6e 61 6d 65 29 0a 09 09 09 2e 66 bind(name).....f
1900: 65 74 63 68 5f 6f 70 74 69 6f 6e 61 6c 28 26 6d etch_optional(&m
1910: 75 74 20 2a 73 65 6c 66 2e 30 29 2e 61 77 61 69 ut *self.0).awai
1920: 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 4f 6b t.stack()?;...Ok
1930: 28 73 6f 75 72 63 65 29 0a 09 7d 0a 0a 09 70 75 (source)..}...pu
1940: 62 20 61 73 79 6e 63 20 66 6e 20 67 65 74 5f 73 b async fn get_s
1950: 6f 75 72 63 65 20 3c 49 3e 20 28 26 6d 75 74 20 ource <I> (&mut
1960: 73 65 6c 66 2c 20 69 64 3a 20 69 33 32 2c 20 6f self, id: i32, o
1970: 77 6e 65 72 3a 20 49 29 20 2d 3e 20 52 65 73 75 wner: I) -> Resu
1980: 6c 74 3c 53 6f 75 72 63 65 3e 0a 09 77 68 65 72 lt<Source>..wher
1990: 65 20 49 3a 20 49 6e 74 6f 3c 69 36 34 3e 20 7b e I: Into<i64> {
19a0: 0a 09 09 6c 65 74 20 73 6f 75 72 63 65 3a 20 53 ...let source: S
19b0: 6f 75 72 63 65 20 3d 20 73 71 6c 78 3a 3a 71 75 ource = sqlx::qu
19c0: 65 72 79 5f 61 73 28 22 73 65 6c 65 63 74 20 63 ery_as("select c
19d0: 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 72 6c 2c 20 hannel_id, url,
19e0: 69 76 5f 68 61 73 68 2c 20 6f 77 6e 65 72 2c 20 iv_hash, owner,
19f0: 75 72 6c 5f 72 65 20 66 72 6f 6d 20 72 73 73 74 url_re from rsst
1a00: 67 5f 73 6f 75 72 63 65 20 77 68 65 72 65 20 73 g_source where s
1a10: 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 20 61 6e ource_id = $1 an
1a20: 64 20 6f 77 6e 65 72 20 3d 20 24 32 22 29 0a 09 d owner = $2")..
1a30: 09 09 2e 62 69 6e 64 28 69 64 29 0a 09 09 09 2e ...bind(id).....
1a40: 62 69 6e 64 28 6f 77 6e 65 72 2e 69 6e 74 6f 28 bind(owner.into(
1a50: 29 29 0a 09 09 09 2e 66 65 74 63 68 5f 6f 6e 65 )).....fetch_one
1a60: 28 26 6d 75 74 20 2a 73 65 6c 66 2e 30 29 2e 61 (&mut *self.0).a
1a70: 77 61 69 74 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 wait.stack()?;..
1a80: 09 4f 6b 28 73 6f 75 72 63 65 29 0a 09 7d 0a 0a .Ok(source)..}..
1a90: 09 70 75 62 20 61 73 79 6e 63 20 66 6e 20 73 65 .pub async fn se
1aa0: 74 5f 73 63 72 61 70 65 20 3c 49 3e 20 28 26 6d t_scrape <I> (&m
1ab0: 75 74 20 73 65 6c 66 2c 20 69 64 3a 20 49 29 20 ut self, id: I)
1ac0: 2d 3e 20 52 65 73 75 6c 74 3c 28 29 3e 0a 09 77 -> Result<()>..w
1ad0: 68 65 72 65 20 49 3a 20 49 6e 74 6f 3c 69 36 34 here I: Into<i64
1ae0: 3e 20 7b 0a 09 09 73 71 6c 78 3a 3a 71 75 65 72 > {...sqlx::quer
1af0: 79 28 22 75 70 64 61 74 65 20 72 73 73 74 67 5f y("update rsstg_
1b00: 73 6f 75 72 63 65 20 73 65 74 20 6c 61 73 74 5f source set last_
1b10: 73 63 72 61 70 65 20 3d 20 6e 6f 77 28 29 20 77 scrape = now() w
1b20: 68 65 72 65 20 73 6f 75 72 63 65 5f 69 64 20 3d here source_id =
1b30: 20 24 31 3b 22 29 0a 09 09 09 2e 62 69 6e 64 28 $1;").....bind(
1b40: 69 64 2e 69 6e 74 6f 28 29 29 0a 09 09 09 2e 65 id.into()).....e
1b50: 78 65 63 75 74 65 28 26 6d 75 74 20 2a 73 65 6c xecute(&mut *sel
1b60: 66 2e 30 29 2e 61 77 61 69 74 2e 73 74 61 63 6b f.0).await.stack
1b70: 28 29 3f 3b 0a 09 09 4f 6b 28 28 29 29 0a 09 7d ()?;...Ok(())..}
1b80: 0a 0a 09 23 5b 61 6c 6c 6f 77 28 63 6c 69 70 70 ...#[allow(clipp
1b90: 79 3a 3a 74 6f 6f 5f 6d 61 6e 79 5f 61 72 67 75 y::too_many_argu
1ba0: 6d 65 6e 74 73 29 5d 20 2f 2f 20 58 58 58 20 64 ments)] // XXX d
1bb0: 6f 20 49 20 6e 65 65 64 20 74 6f 20 6d 61 6b 65 o I need to make
1bc0: 20 69 74 20 76 61 72 69 61 64 69 63 3f 20 49 20 it variadic? I
1bd0: 67 75 65 73 73 20 69 74 20 77 6f 72 6b 20 66 69 guess it work fi
1be0: 6e 65 0a 09 09 09 09 09 09 09 09 09 09 20 2f 2f ne........... //
1bf0: 20 74 68 69 73 20 77 61 79 20 66 6f 72 20 6e 6f this way for no
1c00: 77 2c 20 75 6e 6c 65 73 73 20 74 68 65 72 65 20 w, unless there
1c10: 77 6f 75 6c 64 20 62 65 20 61 20 67 6f 6f 64 20 would be a good
1c20: 73 74 72 75 63 74 0a 09 09 09 09 09 09 09 09 09 struct..........
1c30: 09 20 2f 2f 20 77 69 74 68 20 61 6c 6c 20 73 6f . // with all so
1c40: 75 72 63 65 20 66 69 65 6c 64 73 20 74 6f 20 75 urce fields to u
1c50: 73 65 20 74 68 61 74 20 61 73 20 61 6e 20 61 72 se that as an ar
1c60: 67 75 6d 65 6e 74 0a 09 70 75 62 20 61 73 79 6e gument..pub asyn
1c70: 63 20 66 6e 20 75 70 64 61 74 65 20 3c 49 3e 20 c fn update <I>
1c80: 28 26 6d 75 74 20 73 65 6c 66 2c 20 75 70 64 61 (&mut self, upda
1c90: 74 65 3a 20 4f 70 74 69 6f 6e 3c 69 33 32 3e 2c te: Option<i32>,
1ca0: 20 63 68 61 6e 6e 65 6c 3a 20 26 73 74 72 2c 20 channel: &str,
1cb0: 63 68 61 6e 6e 65 6c 5f 69 64 3a 20 69 36 34 2c channel_id: i64,
1cc0: 20 75 72 6c 3a 20 26 73 74 72 2c 20 69 76 5f 68 url: &str, iv_h
1cd0: 61 73 68 3a 20 4f 70 74 69 6f 6e 3c 26 73 74 72 ash: Option<&str
1ce0: 3e 2c 20 75 72 6c 5f 72 65 3a 20 4f 70 74 69 6f >, url_re: Optio
1cf0: 6e 3c 26 73 74 72 3e 2c 20 6f 77 6e 65 72 3a 20 n<&str>, owner:
1d00: 49 29 20 2d 3e 20 52 65 73 75 6c 74 3c 26 73 74 I) -> Result<&st
1d10: 72 3e 0a 09 77 68 65 72 65 20 49 3a 20 49 6e 74 r>..where I: Int
1d20: 6f 3c 69 36 34 3e 20 7b 0a 09 09 6d 61 74 63 68 o<i64> {...match
1d30: 20 6d 61 74 63 68 20 75 70 64 61 74 65 20 7b 0a match update {.
1d40: 09 09 09 09 53 6f 6d 65 28 69 64 29 20 3d 3e 20 ....Some(id) =>
1d50: 7b 0a 09 09 09 09 09 73 71 6c 78 3a 3a 71 75 65 {......sqlx::que
1d60: 72 79 28 22 75 70 64 61 74 65 20 72 73 73 74 67 ry("update rsstg
1d70: 5f 73 6f 75 72 63 65 20 73 65 74 20 63 68 61 6e _source set chan
1d80: 6e 65 6c 5f 69 64 20 3d 20 24 32 2c 20 75 72 6c nel_id = $2, url
1d90: 20 3d 20 24 33 2c 20 69 76 5f 68 61 73 68 20 3d = $3, iv_hash =
1da0: 20 24 34 2c 20 6f 77 6e 65 72 20 3d 20 24 35 2c $4, owner = $5,
1db0: 20 63 68 61 6e 6e 65 6c 20 3d 20 24 36 2c 20 75 channel = $6, u
1dc0: 72 6c 5f 72 65 20 3d 20 24 37 20 77 68 65 72 65 rl_re = $7 where
1dd0: 20 73 6f 75 72 63 65 5f 69 64 20 3d 20 24 31 22 source_id = $1"
1de0: 29 0a 09 09 09 09 09 09 2e 62 69 6e 64 28 69 64 )........bind(id
1df0: 29 0a 09 09 09 09 7d 2c 0a 09 09 09 09 4e 6f 6e ).....},.....Non
1e00: 65 20 3d 3e 20 7b 0a 09 09 09 09 09 73 71 6c 78 e => {......sqlx
1e10: 3a 3a 71 75 65 72 79 28 22 69 6e 73 65 72 74 20 ::query("insert
1e20: 69 6e 74 6f 20 72 73 73 74 67 5f 73 6f 75 72 63 into rsstg_sourc
1e30: 65 20 28 63 68 61 6e 6e 65 6c 5f 69 64 2c 20 75 e (channel_id, u
1e40: 72 6c 2c 20 69 76 5f 68 61 73 68 2c 20 6f 77 6e rl, iv_hash, own
1e50: 65 72 2c 20 63 68 61 6e 6e 65 6c 2c 20 75 72 6c er, channel, url
1e60: 5f 72 65 29 20 76 61 6c 75 65 73 20 28 24 31 2c _re) values ($1,
1e70: 20 24 32 2c 20 24 33 2c 20 24 34 2c 20 24 35 2c $2, $3, $4, $5,
1e80: 20 24 36 29 22 29 0a 09 09 09 09 7d 2c 0a 09 09 $6)").....},...
1e90: 09 7d 0a 09 09 09 09 2e 62 69 6e 64 28 63 68 61 .}......bind(cha
1ea0: 6e 6e 65 6c 5f 69 64 29 0a 09 09 09 09 2e 62 69 nnel_id)......bi
1eb0: 6e 64 28 75 72 6c 29 0a 09 09 09 09 2e 62 69 6e nd(url)......bin
1ec0: 64 28 69 76 5f 68 61 73 68 29 0a 09 09 09 09 2e d(iv_hash)......
1ed0: 62 69 6e 64 28 6f 77 6e 65 72 2e 69 6e 74 6f 28 bind(owner.into(
1ee0: 29 29 0a 09 09 09 09 2e 62 69 6e 64 28 63 68 61 ))......bind(cha
1ef0: 6e 6e 65 6c 29 0a 09 09 09 09 2e 62 69 6e 64 28 nnel)......bind(
1f00: 75 72 6c 5f 72 65 29 0a 09 09 09 09 2e 65 78 65 url_re)......exe
1f10: 63 75 74 65 28 26 6d 75 74 20 2a 73 65 6c 66 2e cute(&mut *self.
1f20: 30 29 2e 61 77 61 69 74 0a 09 09 7b 0a 09 09 09 0).await...{....
1f30: 4f 6b 28 5f 29 20 3d 3e 20 4f 6b 28 6d 61 74 63 Ok(_) => Ok(matc
1f40: 68 20 75 70 64 61 74 65 20 7b 0a 09 09 09 09 53 h update {.....S
1f50: 6f 6d 65 28 5f 29 20 3d 3e 20 22 43 68 61 6e 6e ome(_) => "Chann
1f60: 65 6c 20 75 70 64 61 74 65 64 2e 22 2c 0a 09 09 el updated.",...
1f70: 09 09 4e 6f 6e 65 20 3d 3e 20 22 43 68 61 6e 6e ..None => "Chann
1f80: 65 6c 20 61 64 64 65 64 2e 22 2c 0a 09 09 09 7d el added.",....}
1f90: 29 2c 0a 09 09 09 45 72 72 28 73 71 6c 78 3a 3a ),....Err(sqlx::
1fa0: 45 72 72 6f 72 3a 3a 44 61 74 61 62 61 73 65 28 Error::Database(
1fb0: 65 72 72 29 29 20 3d 3e 20 7b 0a 09 09 09 09 6d err)) => {.....m
1fc0: 61 74 63 68 20 65 72 72 2e 64 6f 77 6e 63 61 73 atch err.downcas
1fd0: 74 3a 3a 3c 73 71 6c 78 3a 3a 70 6f 73 74 67 72 t::<sqlx::postgr
1fe0: 65 73 3a 3a 50 67 44 61 74 61 62 61 73 65 45 72 es::PgDatabaseEr
1ff0: 72 6f 72 3e 28 29 2e 72 6f 75 74 69 6e 65 28 29 ror>().routine()
2000: 20 7b 0a 09 09 09 09 09 53 6f 6d 65 28 22 5f 62 {......Some("_b
2010: 74 5f 63 68 65 63 6b 5f 75 6e 69 71 75 65 22 2c t_check_unique",
2020: 20 29 20 3d 3e 20 7b 0a 09 09 09 09 09 09 4f 6b ) => {.......Ok
2030: 28 22 44 75 70 6c 69 63 61 74 65 20 6b 65 79 2e ("Duplicate key.
2040: 22 29 0a 09 09 09 09 09 7d 2c 0a 09 09 09 09 09 ")......},......
2050: 53 6f 6d 65 28 5f 29 20 3d 3e 20 7b 0a 09 09 09 Some(_) => {....
2060: 09 09 09 4f 6b 28 22 44 61 74 61 62 61 73 65 20 ...Ok("Database
2070: 65 72 72 6f 72 2e 22 29 0a 09 09 09 09 09 7d 2c error.")......},
2080: 0a 09 09 09 09 09 4e 6f 6e 65 20 3d 3e 20 7b 0a ......None => {.
2090: 09 09 09 09 09 09 4f 6b 28 22 4e 6f 20 64 61 74 ......Ok("No dat
20a0: 61 62 61 73 65 20 65 72 72 6f 72 20 65 78 74 72 abase error extr
20b0: 61 63 74 65 64 2e 22 29 0a 09 09 09 09 09 7d 2c acted.")......},
20c0: 0a 09 09 09 09 7d 0a 09 09 09 7d 2c 0a 09 09 09 .....}....},....
20d0: 45 72 72 28 65 72 72 29 20 3d 3e 20 7b 0a 09 09 Err(err) => {...
20e0: 09 09 62 61 69 6c 21 28 22 53 6f 72 72 79 2c 20 ..bail!("Sorry,
20f0: 75 6e 6b 6e 6f 77 6e 20 65 72 72 6f 72 3a 5c 6e unknown error:\n
2100: 7b 65 72 72 3a 23 3f 7d 5c 6e 22 29 3b 0a 09 09 {err:#?}\n");...
2110: 09 7d 2c 0a 09 09 7d 0a 09 7d 0a 7d 0a .},...}..}.}.