feat: finish adding device name to login, register, and refresh endpoints

src/api/v1/auth/login.rs

@@ -7,7 +7,7 @@ use diesel_async::RunQueryDsl;
 use serde::Deserialize;
 
 use crate::{
-    error::Error, schema::*, utils::{generate_token, new_refresh_token_cookie, user_uuid_from_identifier, PASSWORD_REGEX}, generate_device_name::generate_device_name, Data
+    error::Error, schema::*, utils::{generate_device_name, generate_token, new_refresh_token_cookie, user_uuid_from_identifier, PASSWORD_REGEX}, Data
 };
 
 use super::Response;

src/api/v1/auth/refresh.rs

@@ -7,7 +7,7 @@ use std::time::{SystemTime, UNIX_EPOCH};
 use crate::{
     error::Error, schema::{
         access_tokens::{self, dsl},
-        refresh_tokens::{self, device_name, dsl as rdsl},
+        refresh_tokens::{self, dsl as rdsl},
     }, utils::{generate_token, new_refresh_token_cookie}, Data
 };
 
@@ -50,7 +50,7 @@ pub async fn res(req: HttpRequest, data: web::Data<Data>) -> Result<HttpResponse
         }
 
         let current_time = SystemTime::now().duration_since(UNIX_EPOCH)?.as_secs() as i64;
-        let existing_device_name: String;
+        let mut device_name: String = String::new();
 
         if lifetime > 1987200 {
             let new_refresh_token = generate_token::<32>()?;
@@ -65,9 +65,9 @@ pub async fn res(req: HttpRequest, data: web::Data<Data>) -> Result<HttpResponse
                 .get_result::<String>(&mut conn)
                 .await
             {
-                Ok(device_name) => {
+                Ok(existing_device_name) => {
                     refresh_token = new_refresh_token;
-                    existing_device_name = device_name.to_string();
+                    device_name = existing_device_name;
                 }
                 Err(error) => {
                     error!("{error}");
@@ -88,7 +88,7 @@ pub async fn res(req: HttpRequest, data: web::Data<Data>) -> Result<HttpResponse
 
         return Ok(HttpResponse::Ok()
             .cookie(new_refresh_token_cookie(&data.config, refresh_token))
-            .json(Response { access_token, existing_device_name }));
+            .json(Response { access_token, device_name }));
     }
 
     refresh_token_cookie.make_removal();

src/api/v1/auth/register.rs

@@ -17,8 +17,8 @@ use crate::{
         refresh_tokens::{self, dsl as rdsl},
         users::{self, dsl as udsl},
     }, utils::{
-        generate_token, new_refresh_token_cookie, EMAIL_REGEX, PASSWORD_REGEX, USERNAME_REGEX
-    }, generate_device_name::generate_device_name, Data
+        generate_device_name, generate_token, new_refresh_token_cookie, EMAIL_REGEX, PASSWORD_REGEX, USERNAME_REGEX
+    }, Data
 };
 
 #[derive(Deserialize)]

src/main.rs

@@ -9,7 +9,7 @@ use objects::MailClient;
 use simple_logger::SimpleLogger;
 use std::time::SystemTime;
 mod config;
-mod generate_device_name;
+mod wordlist;
 use config::{Config, ConfigBuilder};
 use diesel_migrations::{EmbeddedMigrations, MigrationHarness, embed_migrations};
 

src/schema.rs

@@ -103,7 +103,7 @@ diesel::table! {
         token -> Varchar,
         uuid -> Uuid,
         created_at -> Int8,
-        #[max_length = 16]
+        #[max_length = 64]
         device_name -> Varchar,
     }
 }

src/utils.rs

@@ -1,4 +1,5 @@
 use std::sync::LazyLock;
+use rand::Rng;
 
 use actix_web::{
     cookie::{Cookie, SameSite, time::Duration},
@@ -16,11 +17,7 @@ use serde::Serialize;
 use uuid::Uuid;
 
 use crate::{
-    Conn, Data,
-    config::Config,
-    error::Error,
-    objects::{HasIsAbove, HasUuid},
-    schema::users,
+    config::Config, error::Error, objects::{HasIsAbove, HasUuid}, schema::users, wordlist::{ADJECTIVES, ANIMALS}, Conn, Data
 };
 
 pub static EMAIL_REGEX: LazyLock<Regex> = LazyLock::new(|| {
@@ -282,3 +279,10 @@ impl Data {
             .await
     }
 }
+
+pub fn generate_device_name() -> String {
+    let adjective_index = rand::rng().random_range(0..ADJECTIVES.len()-1);
+    let animal_index = rand::rng().random_range(0..ANIMALS.len()-1);
+
+    return [ADJECTIVES[adjective_index], ANIMALS[animal_index]].join(" ")
+}

src/wordlist.rs

@@ -1,16 +1,4 @@
-use rand::Rng;
-
-pub fn generate_device_name() -> String {
-    let adjective_index = rand::rng().random_range(0..ADJECTIVES_LENGTH-1);
-    let animal_index = rand::rng().random_range(0..ANIMALS_LENGTH-1);
-
-    return [ADJECTIVES[adjective_index], ANIMALS[animal_index]].join(" ")
-}
-
-const ANIMALS_LENGTH: usize = 223;
-const ADJECTIVES_LENGTH: usize = 765;
-
-const ANIMALS: [&'static str; ANIMALS_LENGTH] = [
+pub const ANIMALS: [&'static str; 223] = [
     "Aardvark",
     "Albatross",
     "Alligator",
@@ -236,7 +224,7 @@ const ANIMALS: [&'static str; ANIMALS_LENGTH] = [
     "Zebra",
 ];
 
-const ADJECTIVES: [&'static str; ADJECTIVES_LENGTH] = [
+pub const ADJECTIVES: [&'static str; 765] = [
     "other",
     "such",
     "first",