use super::user::Register; use super::*; #[derive(Serialize, Deserialize)] pub struct ListCategories {} #[derive(Serialize, Deserialize)] pub struct ListCategoriesResponse { categories: Vec, } #[derive(Serialize, Deserialize, Debug)] pub struct Search { q: String, type_: String, community_id: Option, sort: String, page: Option, limit: Option, auth: Option, } #[derive(Serialize, Deserialize, Debug)] pub struct SearchResponse { pub type_: String, pub comments: Vec, pub posts: Vec, pub communities: Vec, pub users: Vec, } #[derive(Serialize, Deserialize)] pub struct GetModlog { mod_user_id: Option, community_id: Option, page: Option, limit: Option, } #[derive(Serialize, Deserialize)] pub struct GetModlogResponse { removed_posts: Vec, locked_posts: Vec, stickied_posts: Vec, removed_comments: Vec, removed_communities: Vec, banned_from_community: Vec, banned: Vec, added_to_community: Vec, added: Vec, } #[derive(Serialize, Deserialize)] pub struct CreateSite { pub name: String, pub description: Option, pub enable_downvotes: bool, pub open_registration: bool, pub enable_nsfw: bool, pub auth: String, } #[derive(Serialize, Deserialize)] pub struct EditSite { name: String, description: Option, enable_downvotes: bool, open_registration: bool, enable_nsfw: bool, auth: String, } #[derive(Serialize, Deserialize)] pub struct GetSite {} #[derive(Serialize, Deserialize, Clone)] pub struct SiteResponse { site: SiteView, } #[derive(Serialize, Deserialize)] pub struct GetSiteResponse { site: Option, admins: Vec, banned: Vec, pub online: usize, } #[derive(Serialize, Deserialize)] pub struct TransferSite { user_id: i32, auth: String, } #[derive(Serialize, Deserialize)] pub struct GetSiteConfig { auth: String, } #[derive(Serialize, Deserialize)] pub struct GetSiteConfigResponse { config_hjson: String, } #[derive(Serialize, Deserialize)] pub struct SaveSiteConfig { config_hjson: String, auth: String, } impl Perform for Oper { type Response = ListCategoriesResponse; fn perform( &self, pool: Pool>, _websocket_info: Option, ) -> Result { let _data: &ListCategories = &self.data; let conn = pool.get()?; let categories: Vec = Category::list_all(&conn)?; // Return the jwt Ok(ListCategoriesResponse { categories }) } } impl Perform for Oper { type Response = GetModlogResponse; fn perform( &self, pool: Pool>, _websocket_info: Option, ) -> Result { let data: &GetModlog = &self.data; let conn = pool.get()?; let removed_posts = ModRemovePostView::list( &conn, data.community_id, data.mod_user_id, data.page, data.limit, )?; let locked_posts = ModLockPostView::list( &conn, data.community_id, data.mod_user_id, data.page, data.limit, )?; let stickied_posts = ModStickyPostView::list( &conn, data.community_id, data.mod_user_id, data.page, data.limit, )?; let removed_comments = ModRemoveCommentView::list( &conn, data.community_id, data.mod_user_id, data.page, data.limit, )?; let banned_from_community = ModBanFromCommunityView::list( &conn, data.community_id, data.mod_user_id, data.page, data.limit, )?; let added_to_community = ModAddCommunityView::list( &conn, data.community_id, data.mod_user_id, data.page, data.limit, )?; // These arrays are only for the full modlog, when a community isn't given let (removed_communities, banned, added) = if data.community_id.is_none() { ( ModRemoveCommunityView::list(&conn, data.mod_user_id, data.page, data.limit)?, ModBanView::list(&conn, data.mod_user_id, data.page, data.limit)?, ModAddView::list(&conn, data.mod_user_id, data.page, data.limit)?, ) } else { (Vec::new(), Vec::new(), Vec::new()) }; // Return the jwt Ok(GetModlogResponse { removed_posts, locked_posts, stickied_posts, removed_comments, removed_communities, banned_from_community, banned, added_to_community, added, }) } } impl Perform for Oper { type Response = SiteResponse; fn perform( &self, pool: Pool>, _websocket_info: Option, ) -> Result { let data: &CreateSite = &self.data; let claims = match Claims::decode(&data.auth) { Ok(claims) => claims.claims, Err(_e) => return Err(APIError::err("not_logged_in").into()), }; if let Err(slurs) = slur_check(&data.name) { return Err(APIError::err(&slurs_vec_to_str(slurs)).into()); } if let Some(description) = &data.description { if let Err(slurs) = slur_check(description) { return Err(APIError::err(&slurs_vec_to_str(slurs)).into()); } } let user_id = claims.id; let conn = pool.get()?; // Make sure user is an admin if !UserView::read(&conn, user_id)?.admin { return Err(APIError::err("not_an_admin").into()); } let site_form = SiteForm { name: data.name.to_owned(), description: data.description.to_owned(), creator_id: user_id, enable_downvotes: data.enable_downvotes, open_registration: data.open_registration, enable_nsfw: data.enable_nsfw, updated: None, }; match Site::create(&conn, &site_form) { Ok(site) => site, Err(_e) => return Err(APIError::err("site_already_exists").into()), }; let site_view = SiteView::read(&conn)?; Ok(SiteResponse { site: site_view }) } } impl Perform for Oper { type Response = SiteResponse; fn perform( &self, pool: Pool>, websocket_info: Option, ) -> Result { let data: &EditSite = &self.data; let claims = match Claims::decode(&data.auth) { Ok(claims) => claims.claims, Err(_e) => return Err(APIError::err("not_logged_in").into()), }; if let Err(slurs) = slur_check(&data.name) { return Err(APIError::err(&slurs_vec_to_str(slurs)).into()); } if let Some(description) = &data.description { if let Err(slurs) = slur_check(description) { return Err(APIError::err(&slurs_vec_to_str(slurs)).into()); } } let user_id = claims.id; let conn = pool.get()?; // Make sure user is an admin if !UserView::read(&conn, user_id)?.admin { return Err(APIError::err("not_an_admin").into()); } let found_site = Site::read(&conn, 1)?; let site_form = SiteForm { name: data.name.to_owned(), description: data.description.to_owned(), creator_id: found_site.creator_id, updated: Some(naive_now()), enable_downvotes: data.enable_downvotes, open_registration: data.open_registration, enable_nsfw: data.enable_nsfw, }; match Site::update(&conn, 1, &site_form) { Ok(site) => site, Err(_e) => return Err(APIError::err("couldnt_update_site").into()), }; let site_view = SiteView::read(&conn)?; let res = SiteResponse { site: site_view }; if let Some(ws) = websocket_info { ws.chatserver.do_send(SendAllMessage { op: UserOperation::EditSite, response: res.clone(), my_id: ws.id, }); } Ok(res) } } impl Perform for Oper { type Response = GetSiteResponse; fn perform( &self, pool: Pool>, websocket_info: Option, ) -> Result { let _data: &GetSite = &self.data; let conn = pool.get()?; // TODO refactor this a little let site_view = if let Ok(_site) = Site::read(&conn, 1) { Some(SiteView::read(&conn)?) } else if let Some(setup) = Settings::get().setup.as_ref() { let register = Register { username: setup.admin_username.to_owned(), email: setup.admin_email.to_owned(), password: setup.admin_password.to_owned(), password_verify: setup.admin_password.to_owned(), admin: true, show_nsfw: true, }; let login_response = Oper::new(register).perform(pool.clone(), websocket_info.clone())?; info!("Admin {} created", setup.admin_username); let create_site = CreateSite { name: setup.site_name.to_owned(), description: None, enable_downvotes: true, open_registration: true, enable_nsfw: true, auth: login_response.jwt, }; Oper::new(create_site).perform(pool, websocket_info.clone())?; info!("Site {} created", setup.site_name); Some(SiteView::read(&conn)?) } else { None }; let mut admins = UserView::admins(&conn)?; // Make sure the site creator is the top admin if let Some(site_view) = site_view.to_owned() { let site_creator_id = site_view.creator_id; // TODO investigate why this is sometimes coming back null // Maybe user_.admin isn't being set to true? if let Some(creator_index) = admins.iter().position(|r| r.id == site_creator_id) { let creator_user = admins.remove(creator_index); admins.insert(0, creator_user); } } let banned = UserView::banned(&conn)?; let online = if let Some(_ws) = websocket_info { // TODO 1 // let fut = async { // ws.chatserver.send(GetUsersOnline).await.unwrap() // }; // Runtime::new().unwrap().block_on(fut) } else { 0 }; Ok(GetSiteResponse { site: site_view, admins, banned, online, }) } } impl Perform for Oper { type Response = SearchResponse; fn perform( &self, pool: Pool>, _websocket_info: Option, ) -> Result { let data: &Search = &self.data; dbg!(&data); let conn = pool.get()?; match search_by_apub_id(&data.q, &conn) { Ok(r) => return Ok(r), Err(e) => debug!("Failed to resolve search query as activitypub ID: {}", e), } let user_id: Option = match &data.auth { Some(auth) => match Claims::decode(&auth) { Ok(claims) => { let user_id = claims.claims.id; Some(user_id) } Err(_e) => None, }, None => None, }; let sort = SortType::from_str(&data.sort)?; let type_ = SearchType::from_str(&data.type_)?; let mut posts = Vec::new(); let mut comments = Vec::new(); let mut communities = Vec::new(); let mut users = Vec::new(); // TODO no clean / non-nsfw searching rn match type_ { SearchType::Posts => { posts = PostQueryBuilder::create(&conn) .sort(&sort) .show_nsfw(true) .for_community_id(data.community_id) .search_term(data.q.to_owned()) .my_user_id(user_id) .page(data.page) .limit(data.limit) .list()?; } SearchType::Comments => { comments = CommentQueryBuilder::create(&conn) .sort(&sort) .search_term(data.q.to_owned()) .my_user_id(user_id) .page(data.page) .limit(data.limit) .list()?; } SearchType::Communities => { communities = CommunityQueryBuilder::create(&conn) .sort(&sort) .search_term(data.q.to_owned()) .page(data.page) .limit(data.limit) .list()?; } SearchType::Users => { users = UserQueryBuilder::create(&conn) .sort(&sort) .search_term(data.q.to_owned()) .page(data.page) .limit(data.limit) .list()?; } SearchType::All => { posts = PostQueryBuilder::create(&conn) .sort(&sort) .show_nsfw(true) .for_community_id(data.community_id) .search_term(data.q.to_owned()) .my_user_id(user_id) .page(data.page) .limit(data.limit) .list()?; comments = CommentQueryBuilder::create(&conn) .sort(&sort) .search_term(data.q.to_owned()) .my_user_id(user_id) .page(data.page) .limit(data.limit) .list()?; communities = CommunityQueryBuilder::create(&conn) .sort(&sort) .search_term(data.q.to_owned()) .page(data.page) .limit(data.limit) .list()?; users = UserQueryBuilder::create(&conn) .sort(&sort) .search_term(data.q.to_owned()) .page(data.page) .limit(data.limit) .list()?; } SearchType::Url => { posts = PostQueryBuilder::create(&conn) .sort(&sort) .show_nsfw(true) .for_community_id(data.community_id) .url_search(data.q.to_owned()) .page(data.page) .limit(data.limit) .list()?; } }; // Return the jwt Ok(SearchResponse { type_: data.type_.to_owned(), comments, posts, communities, users, }) } } impl Perform for Oper { type Response = GetSiteResponse; fn perform( &self, pool: Pool>, _websocket_info: Option, ) -> Result { let data: &TransferSite = &self.data; let claims = match Claims::decode(&data.auth) { Ok(claims) => claims.claims, Err(_e) => return Err(APIError::err("not_logged_in").into()), }; let user_id = claims.id; let conn = pool.get()?; let read_site = Site::read(&conn, 1)?; // Make sure user is the creator if read_site.creator_id != user_id { return Err(APIError::err("not_an_admin").into()); } let site_form = SiteForm { name: read_site.name, description: read_site.description, creator_id: data.user_id, updated: Some(naive_now()), enable_downvotes: read_site.enable_downvotes, open_registration: read_site.open_registration, enable_nsfw: read_site.enable_nsfw, }; match Site::update(&conn, 1, &site_form) { Ok(site) => site, Err(_e) => return Err(APIError::err("couldnt_update_site").into()), }; // Mod tables let form = ModAddForm { mod_user_id: user_id, other_user_id: data.user_id, removed: Some(false), }; ModAdd::create(&conn, &form)?; let site_view = SiteView::read(&conn)?; let mut admins = UserView::admins(&conn)?; let creator_index = admins .iter() .position(|r| r.id == site_view.creator_id) .unwrap(); let creator_user = admins.remove(creator_index); admins.insert(0, creator_user); let banned = UserView::banned(&conn)?; Ok(GetSiteResponse { site: Some(site_view), admins, banned, online: 0, }) } } impl Perform for Oper { type Response = GetSiteConfigResponse; fn perform( &self, pool: Pool>, _websocket_info: Option, ) -> Result { let data: &GetSiteConfig = &self.data; let claims = match Claims::decode(&data.auth) { Ok(claims) => claims.claims, Err(_e) => return Err(APIError::err("not_logged_in").into()), }; let user_id = claims.id; let conn = pool.get()?; // Only let admins read this let admins = UserView::admins(&conn)?; let admin_ids: Vec = admins.into_iter().map(|m| m.id).collect(); if !admin_ids.contains(&user_id) { return Err(APIError::err("not_an_admin").into()); } let config_hjson = Settings::read_config_file()?; Ok(GetSiteConfigResponse { config_hjson }) } } impl Perform for Oper { type Response = GetSiteConfigResponse; fn perform( &self, pool: Pool>, _websocket_info: Option, ) -> Result { let data: &SaveSiteConfig = &self.data; let claims = match Claims::decode(&data.auth) { Ok(claims) => claims.claims, Err(_e) => return Err(APIError::err("not_logged_in").into()), }; let user_id = claims.id; let conn = pool.get()?; // Only let admins read this let admins = UserView::admins(&conn)?; let admin_ids: Vec = admins.into_iter().map(|m| m.id).collect(); if !admin_ids.contains(&user_id) { return Err(APIError::err("not_an_admin").into()); } // Make sure docker doesn't have :ro at the end of the volume, so its not a read-only filesystem let config_hjson = match Settings::save_config_file(&data.config_hjson) { Ok(config_hjson) => config_hjson, Err(_e) => return Err(APIError::err("couldnt_update_site").into()), }; Ok(GetSiteConfigResponse { config_hjson }) } }