]> Untitled Git - lemmy.git/blob - crates/api_crud/src/custom_emoji/create.rs
Sanitize html (#3708)
[lemmy.git] / crates / api_crud / src / custom_emoji / create.rs
1 use crate::PerformCrud;
2 use actix_web::web::Data;
3 use lemmy_api_common::{
4   context::LemmyContext,
5   custom_emoji::{CreateCustomEmoji, CustomEmojiResponse},
6   utils::{is_admin, local_user_view_from_jwt, sanitize_html},
7 };
8 use lemmy_db_schema::source::{
9   custom_emoji::{CustomEmoji, CustomEmojiInsertForm},
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 CreateCustomEmoji {
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: &CreateCustomEmoji = 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 shortcode = sanitize_html(data.shortcode.to_lowercase().trim());
30     let alt_text = sanitize_html(&data.alt_text);
31     let category = sanitize_html(&data.category);
32
33     let emoji_form = CustomEmojiInsertForm::builder()
34       .local_site_id(local_site.id)
35       .shortcode(shortcode)
36       .alt_text(alt_text)
37       .category(category)
38       .image_url(data.clone().image_url.into())
39       .build();
40     let emoji = CustomEmoji::create(&mut context.pool(), &emoji_form).await?;
41     let mut keywords = vec![];
42     for keyword in &data.keywords {
43       let keyword_form = CustomEmojiKeywordInsertForm::builder()
44         .custom_emoji_id(emoji.id)
45         .keyword(keyword.to_lowercase().trim().to_string())
46         .build();
47       keywords.push(keyword_form);
48     }
49     CustomEmojiKeyword::create(&mut context.pool(), keywords).await?;
50     let view = CustomEmojiView::get(&mut context.pool(), emoji.id).await?;
51     Ok(CustomEmojiResponse { custom_emoji: view })
52   }
53 }