]> Untitled Git - lemmy.git/blob - crates/api_crud/src/custom_emoji/update.rs
Sanitize html (#3708)
[lemmy.git] / crates / api_crud / src / custom_emoji / update.rs
1 use crate::PerformCrud;
2 use actix_web::web::Data;
3 use lemmy_api_common::{
4   context::LemmyContext,
5   custom_emoji::{CustomEmojiResponse, EditCustomEmoji},
6   utils::{is_admin, local_user_view_from_jwt, sanitize_html},
7 };
8 use lemmy_db_schema::source::{
9   custom_emoji::{CustomEmoji, CustomEmojiUpdateForm},
10   custom_emoji_keyword::{CustomEmojiKeyword, CustomEmojiKeywordInsertForm},
11   local_site::LocalSite,
12 };
13 use lemmy_db_views::structs::CustomEmojiView;
14 use lemmy_utils::error::LemmyError;
15
16 #[async_trait::async_trait(?Send)]
17 impl PerformCrud for EditCustomEmoji {
18   type Response = CustomEmojiResponse;
19
20   #[tracing::instrument(skip(self, context))]
21   async fn perform(&self, context: &Data<LemmyContext>) -> Result<CustomEmojiResponse, LemmyError> {
22     let data: &EditCustomEmoji = self;
23     let local_user_view = local_user_view_from_jwt(&data.auth, context).await?;
24
25     let local_site = LocalSite::read(&mut context.pool()).await?;
26     // Make sure user is an admin
27     is_admin(&local_user_view)?;
28
29     let alt_text = sanitize_html(&data.alt_text);
30     let category = sanitize_html(&data.category);
31
32     let emoji_form = CustomEmojiUpdateForm::builder()
33       .local_site_id(local_site.id)
34       .alt_text(alt_text)
35       .category(category)
36       .image_url(data.clone().image_url.into())
37       .build();
38     let emoji = CustomEmoji::update(&mut context.pool(), data.id, &emoji_form).await?;
39     CustomEmojiKeyword::delete(&mut context.pool(), data.id).await?;
40     let mut keywords = vec![];
41     for keyword in &data.keywords {
42       let keyword_form = CustomEmojiKeywordInsertForm::builder()
43         .custom_emoji_id(emoji.id)
44         .keyword(keyword.to_lowercase().trim().to_string())
45         .build();
46       keywords.push(keyword_form);
47     }
48     CustomEmojiKeyword::create(&mut context.pool(), keywords).await?;
49     let view = CustomEmojiView::get(&mut context.pool(), emoji.id).await?;
50     Ok(CustomEmojiResponse { custom_emoji: view })
51   }
52 }