Changes

src/endpoints/folder/create.rs

@@ -7,16 +7,16 @@ pub struct Params {
 }
 
 pub async fn create(
-    State(state): State<AppState>,
+    State(pool): State<Pool>,
     claims: Claims,
     Json(params): Json<Params>,
 ) -> Result<Json<Uuid>, StatusCode> {
-    db::folder::get_permissions(params.parent_folder_id, claims.user_id, &state.pool)
+    db::folder::get_permissions(params.parent_folder_id, claims.user_id, &pool)
         .await
         .handle_internal()?
         .can_write_guard()?;
 
-    let exists = db::folder::name_exists(params.parent_folder_id, &params.folder_name, &state.pool)
+    let exists = db::folder::name_exists(params.parent_folder_id, &params.folder_name, &pool)
         .await
         .handle_internal()?;
     if exists {
@@ -27,7 +27,7 @@ pub async fn create(
         params.parent_folder_id,
         claims.user_id,
         &params.folder_name,
-        &state.pool,
+        &pool,
     )
     .await
     .handle_internal()?;

src/endpoints/folder/list.rs

@@ -16,17 +16,17 @@ pub struct Response {
 
 pub async fn list(
     Query(params): Query<Params>,
-    State(state): State<AppState>,
+    State(pool): State<Pool>,
     claims: Claims,
 ) -> Result<Json<Response>, StatusCode> {
-    let folder_id = db::folder::get_by_id(params.folder_id, claims.user_id, &state.pool)
+    let folder_id = db::folder::get_by_id(params.folder_id, claims.user_id, &pool)
         .await
         .handle_internal()?
         .ok_or(StatusCode::NOT_FOUND)?;
 
     let (files, folders) = try_join!(
-        db::file::get_files(folder_id, &state.pool),
-        db::folder::get_folders(folder_id, &state.pool)
+        db::file::get_files(folder_id, &pool),
+        db::folder::get_folders(folder_id, &pool)
     )
     .handle_internal()?;
 

src/endpoints/permissions/delete.rs

@@ -7,18 +7,18 @@ pub struct Params {
 }
 
 pub async fn delete(
+    State(pool): State<Pool>,
     claims: Claims,
-    State(state): State<AppState>,
     Json(params): Json<Params>,
 ) -> Result<StatusCode, StatusCode> {
     if params.user_id != claims.user_id {
-        db::folder::get_permissions(params.folder_id, claims.user_id, &state.pool)
+        db::folder::get_permissions(params.folder_id, claims.user_id, &pool)
             .await
             .handle_internal()?
             .can_manage_guard()?;
     }
 
-    db::permissions::delete_for_folder(params.folder_id, params.user_id, &state.pool)
+    db::permissions::delete_for_folder(params.folder_id, params.user_id, &pool)
         .await
         .handle_internal()?;
 

src/endpoints/permissions/get.rs

@@ -10,16 +10,16 @@ pub struct Params {
 }
 
 pub async fn get(
+    State(pool): State<Pool>,
     Query(params): Query<Params>,
     claims: Claims,
-    State(state): State<AppState>,
 ) -> Result<Json<HashMap<String, PermissionRaw>>, StatusCode> {
-    db::folder::get_permissions(params.folder_id, claims.user_id, &state.pool)
+    db::folder::get_permissions(params.folder_id, claims.user_id, &pool)
         .await
         .handle_internal()?
         .can_manage_guard()?;
 
-    let permissions = db::permissions::get_all_for_folder(params.folder_id, &state.pool)
+    let permissions = db::permissions::get_all_for_folder(params.folder_id, &pool)
         .await
         .handle_internal()?;
     Ok(Json(permissions))

src/endpoints/permissions/get_top_level.rs

@@ -1,10 +1,10 @@
 use crate::prelude::*;
 
 pub async fn get_top_level(
-    State(state): State<AppState>,
+    State(pool): State<Pool>,
     claims: Claims,
 ) -> Result<Json<Vec<Uuid>>, StatusCode> {
-    let folders = db::permissions::get_top_level_permitted_folders(claims.user_id, &state.pool)
+    let folders = db::permissions::get_top_level_permitted_folders(claims.user_id, &pool)
         .await
         .handle_internal()?;
     Ok(Json(folders))

src/endpoints/permissions/set.rs

@@ -11,17 +11,17 @@ pub struct Params {
 
 pub async fn set(
     claims: Claims,
-    State(state): State<AppState>,
+    State(pool): State<Pool>,
     Json(params): Json<Params>,
 ) -> Result<StatusCode, StatusCode> {
-    let root = db::folder::get_root(claims.user_id, &state.pool)
+    let root = db::folder::get_root(claims.user_id, &pool)
         .await
         .handle_internal()?;
     if params.folder_id == root {
         return Err(StatusCode::BAD_REQUEST);
     }
 
-    db::folder::get_permissions(params.folder_id, claims.user_id, &state.pool)
+    db::folder::get_permissions(params.folder_id, claims.user_id, &pool)
         .await
         .handle_internal()?
         .can_manage_guard()?;
@@ -30,7 +30,7 @@ pub async fn set(
         params.user_id,
         params.folder_id,
         params.permission_type,
-        &state.pool,
+        &pool,
     )
     .await
     .handle_internal()?;

src/endpoints/users/create.rs

@@ -7,10 +7,10 @@ pub struct Params {
 }
 
 pub async fn create(
-    State(state): State<AppState>,
+    State(pool): State<Pool>,
     Json(params): Json<Params>,
 ) -> Result<Json<i32>, StatusCode> {
-    let id = db::users::create_user(&params.username, &params.email, &state.pool)
+    let id = db::users::create_user(&params.username, &params.email, &pool)
         .await
         .handle_internal()?;
     Ok(Json(id))

src/endpoints/users/delete.rs

@@ -2,9 +2,11 @@ use futures::TryStreamExt;
 
 use crate::prelude::*;
 
-pub async fn delete(State(state): State<AppState>, claims: Claims) -> Result<(), StatusCode> {
-    let storage = &state.storage;
-    db::users::delete_user(claims.user_id, &state.pool)
+pub async fn delete(
+    State(AppState { pool, ref storage }): State<AppState>,
+    claims: Claims,
+) -> Result<(), StatusCode> {
+    db::users::delete_user(claims.user_id, &pool)
         .try_for_each_concurrent(5, |file_id| async move {
             let _ = storage.delete(file_id).await;
             Ok(())

src/endpoints/users/get.rs

@@ -6,10 +6,10 @@ pub struct Params {
 }
 
 pub async fn get(
-    State(state): State<AppState>,
+    State(pool): State<Pool>,
     Query(params): Query<Params>,
 ) -> Result<Json<db::users::UserInfo>, StatusCode> {
-    let info = db::users::get(params.user_id, &state.pool)
+    let info = db::users::get(params.user_id, &pool)
         .await
         .handle_internal()?;
     Ok(Json(info))

src/endpoints/users/put.rs

@@ -7,11 +7,11 @@ pub struct Params {
 }
 
 pub async fn put(
-    State(state): State<AppState>,
+    State(pool): State<Pool>,
     claims: Claims,
     Json(params): Json<Params>,
 ) -> Result<Json<db::users::UserInfo>, StatusCode> {
-    let info = db::users::update(claims.user_id, &params.username, &params.email, &state.pool)
+    let info = db::users::update(claims.user_id, &params.username, &params.email, &pool)
         .await
         .handle_internal()?;
     Ok(Json(info))

src/endpoints/users/search.rs

@@ -8,11 +8,11 @@ pub struct Params {
 }
 
 pub async fn search(
-    State(state): State<AppState>,
+    State(pool): State<Pool>,
     Query(params): Query<Params>,
 ) -> sqlx::Result<Json<Vec<db::users::UserInfo>>, StatusCode> {
-    let users = db::users::search_for_user(&params.search_string, &state.pool)
-        .take(5)
+    let users = db::users::search_for_user(&params.search_string, &pool)
+        .take(20)
         .try_collect()
         .await
         .handle_internal()?;

src/file_storage.rs

@@ -21,7 +21,7 @@ impl FileStorage {
     pub fn new() -> anyhow::Result<Self> {
         let var = env::var("DRIVE_STORAGE_PATH");
         let path_str = match var {
-            Ok(ref string) => string,
+            Ok(ref string) => string.trim(),
             Err(err) => {
                 tracing::info!(
                     %err,
@@ -55,7 +55,7 @@ impl FileStorage {
     pub async fn create(&self) -> anyhow::Result<(Uuid, impl tokio::io::AsyncWrite)> {
         let mut error = anyhow::anyhow!("Error creating a file");
         for _ in 0..3 {
-            let file_id = Uuid::new_v4();
+            let file_id = Uuid::now_v7();
             match self.create_inner(file_id).await {
                 Ok(file) => return Ok((file_id, file)),
                 Err(err) => error = error.context(err),

src/main.rs

@@ -7,7 +7,7 @@ mod prelude;
 
 use std::{env, net::Ipv4Addr};
 
-use axum::Router;
+use axum::{extract::FromRef, Router};
 use file_storage::FileStorage;
 use tokio::net::TcpListener;
 
@@ -19,7 +19,13 @@ struct AppState {
     storage: FileStorage,
 }
 
-async fn create_debug_users(pool: &Pool) -> anyhow::Result<()> {
+impl FromRef<AppState> for Pool {
+    fn from_ref(input: &AppState) -> Self {
+        input.pool.clone()
+    }
+}
+
+async fn create_test_users(pool: &Pool) -> anyhow::Result<()> {
     let count = sqlx::query!("SELECT count(user_id) FROM users")
         .fetch_one(pool)
         .await?
@@ -49,11 +55,14 @@ async fn main() -> anyhow::Result<()> {
         Err(err) => anyhow::bail!("Error getting database url: {err}"),
     };
     sqlx::migrate!().run(&pool).await?;
-    create_debug_users(&pool).await?;
+    if let Ok("1") = env::var("DEVELOPMENT").as_deref().map(str::trim_ascii) {
+        create_test_users(&pool).await?;
+    }
 
-    let storage = file_storage::FileStorage::new()?;
-
-    let state = AppState { pool, storage };
+    let state = AppState {
+        pool,
+        storage: FileStorage::new()?,
+    };
 
     let router = app(state);
     let addr = (Ipv4Addr::UNSPECIFIED, 3000);