crates/api_crud/src/site/mod.rs

   1 use lemmy_db_schema::{ListingType, RegistrationMode};
   2 use lemmy_utils::error::{LemmyErrorType, LemmyResult};
   3
   4 mod create;
   5 mod read;
   6 mod update;
   7
   8 /// Checks whether the default post listing type is valid for a site.
   9 pub fn site_default_post_listing_type_check(
  10   default_post_listing_type: &Option<ListingType>,
  11 ) -> LemmyResult<()> {
  12   if let Some(listing_type) = default_post_listing_type {
  13     // Only allow all or local as default listing types...
  14     if listing_type != &ListingType::All && listing_type != &ListingType::Local {
  15       Err(LemmyErrorType::InvalidDefaultPostListingType)?
  16     } else {
  17       Ok(())
  18     }
  19   } else {
  20     Ok(())
  21   }
  22 }
  23
  24 /// Checks whether the application question and registration mode align.
  25 pub fn application_question_check(
  26   current_application_question: &Option<String>,
  27   new_application_question: &Option<String>,
  28   registration_mode: RegistrationMode,
  29 ) -> LemmyResult<()> {
  30   let has_no_question: bool =
  31     current_application_question.is_none() && new_application_question.is_none();
  32   let is_nullifying_question: bool = new_application_question == &Some(String::new());
  33
  34   if registration_mode == RegistrationMode::RequireApplication
  35     && (has_no_question || is_nullifying_question)
  36   {
  37     Err(LemmyErrorType::ApplicationQuestionRequired)?
  38   } else {
  39     Ok(())
  40   }
  41 }
  42
  43 #[cfg(test)]
  44 mod tests {
  45   use crate::site::{application_question_check, site_default_post_listing_type_check};
  46   use lemmy_db_schema::{ListingType, RegistrationMode};
  47
  48   #[test]
  49   fn test_site_default_post_listing_type_check() {
  50     assert!(site_default_post_listing_type_check(&None::<ListingType>).is_ok());
  51     assert!(site_default_post_listing_type_check(&Some(ListingType::All)).is_ok());
  52     assert!(site_default_post_listing_type_check(&Some(ListingType::Local)).is_ok());
  53     assert!(site_default_post_listing_type_check(&Some(ListingType::Subscribed)).is_err());
  54   }
  55
  56   #[test]
  57   fn test_application_question_check() {
  58     assert!(
  59       application_question_check(&Some(String::from("q")), &Some(String::new()), RegistrationMode::RequireApplication).is_err(),
  60       "Expected application to be invalid because an application is required, current question: {:?}, new question: {:?}",
  61       "q",
  62       String::new(),
  63     );
  64     assert!(
  65       application_question_check(&None, &None, RegistrationMode::RequireApplication).is_err(),
  66       "Expected application to be invalid because an application is required, current question: {:?}, new question: {:?}",
  67       None::<String>,
  68       None::<String>
  69     );
  70
  71     assert!(
  72       application_question_check(&None, &None, RegistrationMode::Open).is_ok(),
  73       "Expected application to be valid because no application required, current question: {:?}, new question: {:?}, mode: {:?}",
  74       None::<String>,
  75       None::<String>,
  76       RegistrationMode::Open
  77     );
  78     assert!(
  79       application_question_check(&None, &Some(String::from("q")), RegistrationMode::RequireApplication).is_ok(),
  80       "Expected application to be valid because new application provided, current question: {:?}, new question: {:?}, mode: {:?}",
  81       None::<String>,
  82       Some(String::from("q")),
  83       RegistrationMode::RequireApplication
  84     );
  85     assert!(
  86       application_question_check(&Some(String::from("q")), &None, RegistrationMode::RequireApplication).is_ok(),
  87       "Expected application to be valid because application existed, current question: {:?}, new question: {:?}, mode: {:?}",
  88       Some(String::from("q")),
  89       None::<String>,
  90       RegistrationMode::RequireApplication
  91     );
  92   }
  93 }