From 18c9be898a290049fa8f69095ded712c0c0c845b Mon Sep 17 00:00:00 2001
From: Hamcha <hamcha@crunchy.rocks>
Date: Thu, 6 Jul 2023 13:48:03 +0200
Subject: [PATCH] nicer schema for get_site

---
 src/content.rs      | 20 +++++++-------------
 src/routes/sites.rs | 29 +++++++++++++++++++++++------
 2 files changed, 30 insertions(+), 19 deletions(-)

diff --git a/src/content.rs b/src/content.rs
index 940417f..f00a348 100644
--- a/src/content.rs
+++ b/src/content.rs
@@ -1,4 +1,4 @@
-use chrono::{serde::*, DateTime, Utc};
+use chrono::{serde::*, DateTime, NaiveDateTime, Utc};
 use serde::{Deserialize, Serialize};
 use sqlx::{types::Json, FromRow};
 use uuid::Uuid;
@@ -21,12 +21,9 @@ pub struct Site {
     pub description: Option<String>,
 
     /// Times
-    #[serde(with = "ts_seconds")]
-    pub created_at: DateTime<Utc>,
-    #[serde(with = "ts_seconds_option")]
-    pub modified_at: Option<DateTime<Utc>>,
-    #[serde(with = "ts_seconds_option")]
-    pub deleted_at: Option<DateTime<Utc>>,
+    pub created_at: NaiveDateTime,
+    pub modified_at: Option<NaiveDateTime>,
+    pub deleted_at: Option<NaiveDateTime>,
 }
 
 #[derive(Deserialize, Serialize, FromRow)]
@@ -56,12 +53,9 @@ pub struct Page {
     pub blocks: Json<Vec<PageBlock>>,
 
     /// Times
-    #[serde(with = "ts_seconds")]
-    pub created_at: DateTime<Utc>,
-    #[serde(with = "ts_seconds_option")]
-    pub modified_at: Option<DateTime<Utc>>,
-    #[serde(with = "ts_seconds_option")]
-    pub deleted_at: Option<DateTime<Utc>>,
+    pub created_at: NaiveDateTime,
+    pub modified_at: Option<NaiveDateTime>,
+    pub deleted_at: Option<NaiveDateTime>,
 }
 
 #[derive(Deserialize, Serialize)]
diff --git a/src/routes/sites.rs b/src/routes/sites.rs
index 61f1d92..2628d91 100644
--- a/src/routes/sites.rs
+++ b/src/routes/sites.rs
@@ -1,5 +1,6 @@
 use axum::{
     extract::{Path, State},
+    http::StatusCode,
     response::IntoResponse,
     routing::{get, post},
     Json, Router,
@@ -10,7 +11,6 @@ use std::sync::Arc;
 
 use crate::{
     auth::http::RequireUser,
-    content::Site,
     http::{
         error::{ApiError, ERR_NOT_FOUND},
         extract::JsonBody,
@@ -22,15 +22,22 @@ async fn get_site(
     State(state): State<Arc<AppState>>,
     Path(name): Path<String>,
 ) -> Result<impl IntoResponse, ApiError<'static>> {
-    let site_query = sqlx::query_as("SELECT * FROM sites WHERE name = $1").bind(name);
-    let site: Site = site_query
-        .fetch_one(&state.database)
+    let site = sqlx::query!(
+        "SELECT sites.id, sites.name, users.name as owner, title, description, sites.created_at FROM sites JOIN users ON sites.owner = users.id WHERE sites.name = $1 ",
+        name
+    ).fetch_one(&state.database)
         .await
         .map_err(|e| match e {
             sqlx::Error::RowNotFound => ERR_NOT_FOUND,
             _ => e.into(),
         })?;
-    Ok(Json(site))
+    Ok(Json(json!({
+        "name": site.name,
+        "owner": site.owner,
+        "title": site.title,
+        "description": site.description,
+        "created_at": site.created_at,
+    })))
 }
 
 #[derive(Deserialize)]
@@ -51,7 +58,17 @@ async fn create_site(
         payload.title,
     )
     .execute(&state.database)
-    .await?;
+    .await
+    .map_err(|err| match err {
+        sqlx::Error::Database(dberr) if dberr.constraint() == Some("sites_name_key") => {
+            ApiError::ClientError {
+                status: StatusCode::CONFLICT,
+                code: "name-not-available",
+                message: "the chosen name is not available",
+            }
+        }
+        _ => err.into(),
+    })?;
     Ok(Json(json!({"ok": true, "url": "test"})))
 }