crates/db_views/src/registration_application_view.rs

   1 use crate::structs::RegistrationApplicationView;
   2 use diesel::{
   3   dsl::count,
   4   result::Error,
   5   ExpressionMethods,
   6   JoinOnDsl,
   7   NullableExpressionMethods,
   8   QueryDsl,
   9 };
  10 use diesel_async::RunQueryDsl;
  11 use lemmy_db_schema::{
  12   schema::{local_user, person, registration_application},
  13   source::{
  14     local_user::LocalUser,
  15     person::Person,
  16     registration_application::RegistrationApplication,
  17   },
  18   traits::JoinView,
  19   utils::{get_conn, limit_and_offset, DbPool},
  20 };
  21
  22 type RegistrationApplicationViewTuple =
  23   (RegistrationApplication, LocalUser, Person, Option<Person>);
  24
  25 impl RegistrationApplicationView {
  26   pub async fn read(
  27     pool: &mut DbPool<'_>,
  28     registration_application_id: i32,
  29   ) -> Result<Self, Error> {
  30     let conn = &mut get_conn(pool).await?;
  31     let person_alias_1 = diesel::alias!(person as person1);
  32
  33     let (registration_application, creator_local_user, creator, admin) =
  34       registration_application::table
  35         .find(registration_application_id)
  36         .inner_join(
  37           local_user::table.on(registration_application::local_user_id.eq(local_user::id)),
  38         )
  39         .inner_join(person::table.on(local_user::person_id.eq(person::id)))
  40         .left_join(
  41           person_alias_1
  42             .on(registration_application::admin_id.eq(person_alias_1.field(person::id).nullable())),
  43         )
  44         .order_by(registration_application::published.desc())
  45         .select((
  46           registration_application::all_columns,
  47           local_user::all_columns,
  48           person::all_columns,
  49           person_alias_1.fields(person::all_columns).nullable(),
  50         ))
  51         .first::<RegistrationApplicationViewTuple>(conn)
  52         .await?;
  53
  54     Ok(RegistrationApplicationView {
  55       registration_application,
  56       creator_local_user,
  57       creator,
  58       admin,
  59     })
  60   }
  61
  62   /// Returns the current unread registration_application count
  63   pub async fn get_unread_count(
  64     pool: &mut DbPool<'_>,
  65     verified_email_only: bool,
  66   ) -> Result<i64, Error> {
  67     let conn = &mut get_conn(pool).await?;
  68     let person_alias_1 = diesel::alias!(person as person1);
  69
  70     let mut query = registration_application::table
  71       .inner_join(local_user::table.on(registration_application::local_user_id.eq(local_user::id)))
  72       .inner_join(person::table.on(local_user::person_id.eq(person::id)))
  73       .left_join(
  74         person_alias_1
  75           .on(registration_application::admin_id.eq(person_alias_1.field(person::id).nullable())),
  76       )
  77       .filter(registration_application::admin_id.is_null())
  78       .into_boxed();
  79
  80     if verified_email_only {
  81       query = query.filter(local_user::email_verified.eq(true))
  82     }
  83
  84     query
  85       .select(count(registration_application::id))
  86       .first::<i64>(conn)
  87       .await
  88   }
  89 }
  90
  91 #[derive(Default)]
  92 pub struct RegistrationApplicationQuery {
  93   pub unread_only: Option<bool>,
  94   pub verified_email_only: Option<bool>,
  95   pub page: Option<i64>,
  96   pub limit: Option<i64>,
  97 }
  98
  99 impl RegistrationApplicationQuery {
 100   pub async fn list(
 101     self,
 102     pool: &mut DbPool<'_>,
 103   ) -> Result<Vec<RegistrationApplicationView>, Error> {
 104     let conn = &mut get_conn(pool).await?;
 105     let person_alias_1 = diesel::alias!(person as person1);
 106
 107     let mut query = registration_application::table
 108       .inner_join(local_user::table.on(registration_application::local_user_id.eq(local_user::id)))
 109       .inner_join(person::table.on(local_user::person_id.eq(person::id)))
 110       .left_join(
 111         person_alias_1
 112           .on(registration_application::admin_id.eq(person_alias_1.field(person::id).nullable())),
 113       )
 114       .order_by(registration_application::published.desc())
 115       .select((
 116         registration_application::all_columns,
 117         local_user::all_columns,
 118         person::all_columns,
 119         person_alias_1.fields(person::all_columns).nullable(),
 120       ))
 121       .into_boxed();
 122
 123     if self.unread_only.unwrap_or(false) {
 124       query = query.filter(registration_application::admin_id.is_null())
 125     }
 126
 127     if self.verified_email_only.unwrap_or(false) {
 128       query = query.filter(local_user::email_verified.eq(true))
 129     }
 130
 131     let (limit, offset) = limit_and_offset(self.page, self.limit)?;
 132
 133     query = query
 134       .limit(limit)
 135       .offset(offset)
 136       .order_by(registration_application::published.desc());
 137
 138     let res = query.load::<RegistrationApplicationViewTuple>(conn).await?;
 139
 140     Ok(
 141       res
 142         .into_iter()
 143         .map(RegistrationApplicationView::from_tuple)
 144         .collect(),
 145     )
 146   }
 147 }
 148
 149 impl JoinView for RegistrationApplicationView {
 150   type JoinTuple = RegistrationApplicationViewTuple;
 151   fn from_tuple(a: Self::JoinTuple) -> Self {
 152     Self {
 153       registration_application: a.0,
 154       creator_local_user: a.1,
 155       creator: a.2,
 156       admin: a.3,
 157     }
 158   }
 159 }
 160
 161 #[cfg(test)]
 162 mod tests {
 163   use crate::registration_application_view::{
 164     RegistrationApplicationQuery,
 165     RegistrationApplicationView,
 166   };
 167   use lemmy_db_schema::{
 168     source::{
 169       instance::Instance,
 170       local_user::{LocalUser, LocalUserInsertForm, LocalUserUpdateForm},
 171       person::{Person, PersonInsertForm},
 172       registration_application::{
 173         RegistrationApplication,
 174         RegistrationApplicationInsertForm,
 175         RegistrationApplicationUpdateForm,
 176       },
 177     },
 178     traits::Crud,
 179     utils::build_db_pool_for_tests,
 180   };
 181   use serial_test::serial;
 182
 183   #[tokio::test]
 184   #[serial]
 185   async fn test_crud() {
 186     let pool = &build_db_pool_for_tests().await;
 187     let pool = &mut pool.into();
 188
 189     let inserted_instance = Instance::read_or_create(pool, "my_domain.tld".to_string())
 190       .await
 191       .unwrap();
 192
 193     let timmy_person_form = PersonInsertForm::builder()
 194       .name("timmy_rav".into())
 195       .admin(Some(true))
 196       .public_key("pubkey".to_string())
 197       .instance_id(inserted_instance.id)
 198       .build();
 199
 200     let inserted_timmy_person = Person::create(pool, &timmy_person_form).await.unwrap();
 201
 202     let timmy_local_user_form = LocalUserInsertForm::builder()
 203       .person_id(inserted_timmy_person.id)
 204       .password_encrypted("nada".to_string())
 205       .build();
 206
 207     let _inserted_timmy_local_user = LocalUser::create(pool, &timmy_local_user_form)
 208       .await
 209       .unwrap();
 210
 211     let sara_person_form = PersonInsertForm::builder()
 212       .name("sara_rav".into())
 213       .public_key("pubkey".to_string())
 214       .instance_id(inserted_instance.id)
 215       .build();
 216
 217     let inserted_sara_person = Person::create(pool, &sara_person_form).await.unwrap();
 218
 219     let sara_local_user_form = LocalUserInsertForm::builder()
 220       .person_id(inserted_sara_person.id)
 221       .password_encrypted("nada".to_string())
 222       .build();
 223
 224     let inserted_sara_local_user = LocalUser::create(pool, &sara_local_user_form)
 225       .await
 226       .unwrap();
 227
 228     // Sara creates an application
 229     let sara_app_form = RegistrationApplicationInsertForm {
 230       local_user_id: inserted_sara_local_user.id,
 231       answer: "LET ME IIIIINN".to_string(),
 232     };
 233
 234     let sara_app = RegistrationApplication::create(pool, &sara_app_form)
 235       .await
 236       .unwrap();
 237
 238     let read_sara_app_view = RegistrationApplicationView::read(pool, sara_app.id)
 239       .await
 240       .unwrap();
 241
 242     let jess_person_form = PersonInsertForm::builder()
 243       .name("jess_rav".into())
 244       .public_key("pubkey".to_string())
 245       .instance_id(inserted_instance.id)
 246       .build();
 247
 248     let inserted_jess_person = Person::create(pool, &jess_person_form).await.unwrap();
 249
 250     let jess_local_user_form = LocalUserInsertForm::builder()
 251       .person_id(inserted_jess_person.id)
 252       .password_encrypted("nada".to_string())
 253       .build();
 254
 255     let inserted_jess_local_user = LocalUser::create(pool, &jess_local_user_form)
 256       .await
 257       .unwrap();
 258
 259     // Sara creates an application
 260     let jess_app_form = RegistrationApplicationInsertForm {
 261       local_user_id: inserted_jess_local_user.id,
 262       answer: "LET ME IIIIINN".to_string(),
 263     };
 264
 265     let jess_app = RegistrationApplication::create(pool, &jess_app_form)
 266       .await
 267       .unwrap();
 268
 269     let read_jess_app_view = RegistrationApplicationView::read(pool, jess_app.id)
 270       .await
 271       .unwrap();
 272
 273     let mut expected_sara_app_view = RegistrationApplicationView {
 274       registration_application: sara_app.clone(),
 275       creator_local_user: LocalUser {
 276         id: inserted_sara_local_user.id,
 277         person_id: inserted_sara_local_user.person_id,
 278         email: inserted_sara_local_user.email,
 279         show_nsfw: inserted_sara_local_user.show_nsfw,
 280         theme: inserted_sara_local_user.theme,
 281         default_sort_type: inserted_sara_local_user.default_sort_type,
 282         default_listing_type: inserted_sara_local_user.default_listing_type,
 283         interface_language: inserted_sara_local_user.interface_language,
 284         show_avatars: inserted_sara_local_user.show_avatars,
 285         send_notifications_to_email: inserted_sara_local_user.send_notifications_to_email,
 286         validator_time: inserted_sara_local_user.validator_time,
 287         show_bot_accounts: inserted_sara_local_user.show_bot_accounts,
 288         show_scores: inserted_sara_local_user.show_scores,
 289         show_read_posts: inserted_sara_local_user.show_read_posts,
 290         show_new_post_notifs: inserted_sara_local_user.show_new_post_notifs,
 291         email_verified: inserted_sara_local_user.email_verified,
 292         accepted_application: inserted_sara_local_user.accepted_application,
 293         totp_2fa_secret: inserted_sara_local_user.totp_2fa_secret,
 294         totp_2fa_url: inserted_sara_local_user.totp_2fa_url,
 295         password_encrypted: inserted_sara_local_user.password_encrypted,
 296         open_links_in_new_tab: inserted_sara_local_user.open_links_in_new_tab,
 297         infinite_scroll_enabled: inserted_sara_local_user.infinite_scroll_enabled,
 298       },
 299       creator: Person {
 300         id: inserted_sara_person.id,
 301         name: inserted_sara_person.name.clone(),
 302         display_name: None,
 303         published: inserted_sara_person.published,
 304         avatar: None,
 305         actor_id: inserted_sara_person.actor_id.clone(),
 306         local: true,
 307         banned: false,
 308         ban_expires: None,
 309         deleted: false,
 310         admin: false,
 311         bot_account: false,
 312         bio: None,
 313         banner: None,
 314         updated: None,
 315         inbox_url: inserted_sara_person.inbox_url.clone(),
 316         shared_inbox_url: None,
 317         matrix_user_id: None,
 318         instance_id: inserted_instance.id,
 319         private_key: inserted_sara_person.private_key,
 320         public_key: inserted_sara_person.public_key,
 321         last_refreshed_at: inserted_sara_person.last_refreshed_at,
 322       },
 323       admin: None,
 324     };
 325
 326     assert_eq!(read_sara_app_view, expected_sara_app_view);
 327
 328     // Do a batch read of the applications
 329     let apps = RegistrationApplicationQuery {
 330       unread_only: (Some(true)),
 331       ..Default::default()
 332     }
 333     .list(pool)
 334     .await
 335     .unwrap();
 336
 337     assert_eq!(
 338       apps,
 339       [read_jess_app_view.clone(), expected_sara_app_view.clone()]
 340     );
 341
 342     // Make sure the counts are correct
 343     let unread_count = RegistrationApplicationView::get_unread_count(pool, false)
 344       .await
 345       .unwrap();
 346     assert_eq!(unread_count, 2);
 347
 348     // Approve the application
 349     let approve_form = RegistrationApplicationUpdateForm {
 350       admin_id: Some(Some(inserted_timmy_person.id)),
 351       deny_reason: None,
 352     };
 353
 354     RegistrationApplication::update(pool, sara_app.id, &approve_form)
 355       .await
 356       .unwrap();
 357
 358     // Update the local_user row
 359     let approve_local_user_form = LocalUserUpdateForm::builder()
 360       .accepted_application(Some(true))
 361       .build();
 362
 363     LocalUser::update(pool, inserted_sara_local_user.id, &approve_local_user_form)
 364       .await
 365       .unwrap();
 366
 367     let read_sara_app_view_after_approve = RegistrationApplicationView::read(pool, sara_app.id)
 368       .await
 369       .unwrap();
 370
 371     // Make sure the columns changed
 372     expected_sara_app_view
 373       .creator_local_user
 374       .accepted_application = true;
 375     expected_sara_app_view.registration_application.admin_id = Some(inserted_timmy_person.id);
 376
 377     expected_sara_app_view.admin = Some(Person {
 378       id: inserted_timmy_person.id,
 379       name: inserted_timmy_person.name.clone(),
 380       display_name: None,
 381       published: inserted_timmy_person.published,
 382       avatar: None,
 383       actor_id: inserted_timmy_person.actor_id.clone(),
 384       local: true,
 385       banned: false,
 386       ban_expires: None,
 387       deleted: false,
 388       admin: true,
 389       bot_account: false,
 390       bio: None,
 391       banner: None,
 392       updated: None,
 393       inbox_url: inserted_timmy_person.inbox_url.clone(),
 394       shared_inbox_url: None,
 395       matrix_user_id: None,
 396       instance_id: inserted_instance.id,
 397       private_key: inserted_timmy_person.private_key,
 398       public_key: inserted_timmy_person.public_key,
 399       last_refreshed_at: inserted_timmy_person.last_refreshed_at,
 400     });
 401     assert_eq!(read_sara_app_view_after_approve, expected_sara_app_view);
 402
 403     // Do a batch read of apps again
 404     // It should show only jessicas which is unresolved
 405     let apps_after_resolve = RegistrationApplicationQuery {
 406       unread_only: (Some(true)),
 407       ..Default::default()
 408     }
 409     .list(pool)
 410     .await
 411     .unwrap();
 412     assert_eq!(apps_after_resolve, vec![read_jess_app_view]);
 413
 414     // Make sure the counts are correct
 415     let unread_count_after_approve = RegistrationApplicationView::get_unread_count(pool, false)
 416       .await
 417       .unwrap();
 418     assert_eq!(unread_count_after_approve, 1);
 419
 420     // Make sure the not undenied_only has all the apps
 421     let all_apps = RegistrationApplicationQuery::default()
 422       .list(pool)
 423       .await
 424       .unwrap();
 425     assert_eq!(all_apps.len(), 2);
 426
 427     Person::delete(pool, inserted_timmy_person.id)
 428       .await
 429       .unwrap();
 430     Person::delete(pool, inserted_sara_person.id).await.unwrap();
 431     Person::delete(pool, inserted_jess_person.id).await.unwrap();
 432     Instance::delete(pool, inserted_instance.id).await.unwrap();
 433   }
 434 }