use crate::{ Result, AdapterPool, db::{ ChannelId, ItemId, }, fetch::FetchedRSSItem, }; use chrono::{DateTime, Utc}; pub struct UnparsedItem { pub id: i64, pub channel_id: i64, pub fetched_at: Option, pub title: Option, pub description: Option, pub content: Option, } impl UnparsedItem { pub fn parse(self) -> Result { Ok(Item { id: ItemId(self.id), channel_id: ChannelId(self.channel_id), fetched_at: self.fetched_at.as_deref() .map(DateTime::parse_from_rfc2822) .transpose()? .map(|dt| dt.with_timezone(&Utc)), title: self.title, description: self.description, content: self.content, }) } } pub struct Item { id: ItemId, channel_id: ChannelId, fetched_at: Option>, title: Option, description: Option, content: Option, } impl Item { pub fn id(&self) -> ItemId { self.id } pub fn channel(&self) -> ChannelId { self.channel_id } pub fn title(&self) -> Option<&str> { self.title.as_deref() } pub fn description(&self) -> Option<&str> { self.description.as_deref() } pub fn content(&self) -> Option<&str> { self.content.as_deref() } pub async fn get_or_create( pool: &AdapterPool, from_channel: ChannelId, guid: &str ) -> Result { let int_channel_id = i64::from(from_channel); let item = sqlx::query_as!( UnparsedItem, "INSERT INTO items (channel_id, guid) VALUES(?, ?) ON CONFLICT(id) DO UPDATE SET id = id RETURNING id as `id!`, channel_id, fetched_at, title, description, content", int_channel_id, guid ).fetch_one(&pool.0).await?.parse(); item } pub async fn update_content( &self, pool: &AdapterPool, fetched: &FetchedRSSItem, fetched_at: &DateTime ) -> Result<()> { let title = fetched.title(); let description = fetched.description(); let content = fetched.content(); let string_fetched_at = fetched_at.to_rfc2822(); sqlx::query!( "UPDATE items SET title = ?, description = ?, content = ?, fetched_at = ? WHERE id = ?", title, description, content, string_fetched_at, self.id.0 ).execute(&pool.0).await?; Ok(()) } }