use super::*; use crate::cache::Cache; use crate::model::communities::{CommunityContext, CommunityMembership}; use crate::model::communities_permissions::CommunityPermission; use crate::model::{ Error, Result, auth::User, communities::Community, communities::{CommunityReadAccess, CommunityWriteAccess}, permissions::FinePermission, }; use crate::{auto_method, execute, get, query_row}; #[cfg(feature = "sqlite")] use rusqlite::Row; #[cfg(feature = "postgres")] use tokio_postgres::Row; impl DataManager { /// Get a [`Community`] from an SQL row. pub(crate) fn get_community_from_row( #[cfg(feature = "sqlite")] x: &Row<'_>, #[cfg(feature = "postgres")] x: &Row, ) -> Community { Community { id: get!(x->0(isize)) as usize, created: get!(x->1(isize)) as usize, title: get!(x->2(String)), context: serde_json::from_str(&get!(x->3(String))).unwrap(), owner: get!(x->4(isize)) as usize, read_access: serde_json::from_str(&get!(x->5(String))).unwrap(), write_access: serde_json::from_str(&get!(x->6(String))).unwrap(), // likes likes: get!(x->7(isize)) as isize, dislikes: get!(x->8(isize)) as isize, // counts member_count: get!(x->9(isize)) as usize, } } auto_method!(get_community_by_id()@get_community_from_row -> "SELECT * FROM communities WHERE id = $1" --name="community" --returns=Community --cache-key-tmpl="atto.community:{}"); auto_method!(get_community_by_title(&str)@get_community_from_row -> "SELECT * FROM communities WHERE title = $1" --name="community" --returns=Community --cache-key-tmpl="atto.community:{}"); /// Create a new community in the database. /// /// # Arguments /// * `data` - a mock [`Community`] to insert pub async fn create_community(&self, data: Community) -> Result { // check values if data.title.len() < 2 { return Err(Error::DataTooShort("title".to_string())); } else if data.title.len() > 32 { return Err(Error::DataTooLong("title".to_string())); } if !data.title.is_ascii() | data.title.contains(" ") { return Err(Error::MiscError( "Title contains characters which aren't allowed".to_string(), )); } // make sure community doesn't already exist with title if self.get_community_by_title(&data.title).await.is_ok() { return Err(Error::MiscError("Title already in use".to_string())); } // ... let conn = match self.connect().await { Ok(c) => c, Err(e) => return Err(Error::DatabaseConnection(e.to_string())), }; let res = execute!( &conn, "INSERT INTO communities VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)", &[ &data.id.to_string().as_str(), &data.created.to_string().as_str(), &data.title.as_str(), &serde_json::to_string(&data.context).unwrap().as_str(), &data.owner.to_string().as_str(), &serde_json::to_string(&data.read_access).unwrap().as_str(), &serde_json::to_string(&data.write_access).unwrap().as_str(), &0.to_string().as_str(), &0.to_string().as_str(), &0.to_string().as_str() ] ); if let Err(e) = res { return Err(Error::DatabaseError(e.to_string())); } // add journal page owner as admin self.create_membership(CommunityMembership::new( data.owner, data.id, CommunityPermission::ADMINISTRATOR, )) .await .unwrap(); // return Ok(data.title) } pub async fn cache_clear_community(&self, community: &Community) { self.2 .remove(format!("atto.community:{}", community.id)) .await; self.2 .remove(format!("atto.community:{}", community.title)) .await; } auto_method!(delete_community()@get_community_by_id:MANAGE_COMMUNITIES -> "DELETE communities pages WHERE id = $1" --cache-key-tmpl=cache_clear_community); auto_method!(update_community_title(String)@get_community_by_id:MANAGE_COMMUNITIES -> "UPDATE communities SET title = $1 WHERE id = $2" --cache-key-tmpl=cache_clear_community); auto_method!(update_community_context(CommunityContext)@get_community_by_id:MANAGE_COMMUNITIES -> "UPDATE communities SET context = $1 WHERE id = $2" --serde --cache-key-tmpl=cache_clear_community); auto_method!(update_community_read_access(CommunityReadAccess)@get_community_by_id:MANAGE_COMMUNITIES -> "UPDATE communities SET read_access = $1 WHERE id = $2" --serde --cache-key-tmpl=cache_clear_community); auto_method!(update_community_write_access(CommunityWriteAccess)@get_community_by_id:MANAGE_COMMUNITIES -> "UPDATE communities SET write_access = $1 WHERE id = $2" --serde --cache-key-tmpl=cache_clear_community); auto_method!(incr_community_likes()@get_community_by_id -> "UPDATE communities SET likes = likes + 1 WHERE id = $1" --cache-key-tmpl=cache_clear_community --incr); auto_method!(incr_community_dislikes()@get_community_by_id -> "UPDATE communities SET dislikes = dislikes + 1 WHERE id = $1" --cache-key-tmpl=cache_clear_community --incr); auto_method!(decr_community_likes()@get_community_by_id -> "UPDATE communities SET likes = likes - 1 WHERE id = $1" --cache-key-tmpl=cache_clear_community --decr); auto_method!(decr_community_dislikes()@get_community_by_id -> "UPDATE communities SET dislikes = dislikes - 1 WHERE id = $1" --cache-key-tmpl=cache_clear_community --decr); auto_method!(incr_community_member_count()@get_community_by_id -> "UPDATE communities SET member_count = member_count + 1 WHERE id = $1" --cache-key-tmpl=cache_clear_community --incr); auto_method!(decr_community_member_count()@get_community_by_id -> "UPDATE communities SET member_count = member_count - 1 WHERE id = $1" --cache-key-tmpl=cache_clear_community --decr); }