3 files changed, 48 insertions, 7 deletions

diff --git a/src/main.rs b/src/main.rs
index ff04b57..c53664a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,3 +1,5 @@
+mod error;
+mod response;
 mod routers;
 
 use tokio::net::TcpListener;
diff --git a/src/response.rs b/src/response.rs
new file mode 100644
index 0000000..8d505a5
--- /dev/null
+++ b/src/response.rs
@@ -0,0 +1,32 @@
+use axum::response::{IntoResponse, Response};
+use serde::Serialize;
+use serde_json::json;
+
+pub enum ApiResponse<T> {
+    Success(T),
+    Fail(T),
+    Error(String),
+}
+
+impl<T> IntoResponse for ApiResponse<T>
+where
+    T: Serialize,
+{
+    fn into_response(self) -> Response {
+        axum::Json(match self {
+            ApiResponse::Success(data) => json!({
+                "status": "success",
+                "data": data
+            }),
+            ApiResponse::Fail(data) => json!({
+                "status": "fail",
+                "data": data
+            }),
+            ApiResponse::Error(message) => json!({
+                "status": "error",
+                "message": message
+            }),
+        })
+        .into_response()
+    }
+}
diff --git a/src/routers/account.rs b/src/routers/account.rs
index e8c9753..8192133 100644
--- a/src/routers/account.rs
+++ b/src/routers/account.rs
@@ -2,16 +2,23 @@ use axum::Router;
 use axum::response::IntoResponse;
 use axum::routing::{get, post};
 
-pub(crate) fn router() -> Router {
-    Router::new()
-        .route("/me", get(me))
-        .route("/login", post(login))
-}
+use crate::response::ApiResponse;
 
 async fn me() -> impl IntoResponse {
-    "Me"
+    ApiResponse::Success("Me")
+}
+
+async fn register() -> impl IntoResponse {
+    ApiResponse::Success("Register")
 }
 
 async fn login() -> impl IntoResponse {
-    "Login"
+    ApiResponse::Success("Login")
+}
+
+pub(crate) fn router() -> Router {
+    Router::new()
+        .route("/me", get(me))
+        .route("/register", post(register))
+        .route("/login", post(login))
 }