Initial Implementation of ResourceGuard Macro

Cargo.lock

@@ -497,14 +497,16 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "523dc4f511e55ab87b694dc30d0f820d60906ef06413f93d4d7a1385599cc149"
 
 [[package]]
-name = "message-hideyoshi_com"
+name = "message-hideyoshi-com"
 version = "0.1.0"
 dependencies = [
  "axum",
  "chrono",
+ "proc-utils",
  "serde",
  "serde_with",
  "tokio",
+ "utils",
  "uuid",
 ]
 
@@ -627,6 +629,14 @@ dependencies = [
  "unicode-ident",
 ]
 
+[[package]]
+name = "proc-utils"
+version = "0.1.0"
+dependencies = [
+ "quote",
+ "syn",
+]
+
 [[package]]
 name = "quote"
 version = "1.0.35"
@@ -912,6 +922,10 @@ version = "1.0.12"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "3354b9ac3fae1ff6755cb6db53683adb661634f67557942dea4facebec0fee4b"
 
+[[package]]
+name = "utils"
+version = "0.1.0"
+
 [[package]]
 name = "uuid"
 version = "1.7.0"

Cargo.toml

@@ -1,11 +1,15 @@
+workspace = { members = ["proc-utils", "utils"] }
 [package]
-name = "message-hideyoshi_com"
+name = "message-hideyoshi-com"
 version = "0.1.0"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+proc-utils = { path = "proc-utils" }
+utils = { path = "utils" }
+
 axum = "0.7.4"
 tokio = { version = "1.0.0", features = ["rt", "rt-multi-thread", "macros"] }
 uuid = "1.7.0"

proc-utils/Cargo.toml

@@ -0,0 +1,13 @@
+[package]
+name = "proc-utils"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[lib]
+proc-macro = true
+
+[dependencies]
+syn = "2.0.50"
+quote = "1.0.35"

proc-utils/src/lib.rs

@@ -0,0 +1,11 @@
+use proc_macro::TokenStream;
+use quote::quote;
+use syn::{parse_macro_input, ItemFn};
+
+#[proc_macro_attribute]
+pub fn guard_resource(_attr: TokenStream, item: TokenStream) -> TokenStream {
+    let input = parse_macro_input!(item as ItemFn);
+    println!("{} defined", input.sig.ident);
+    println!("Args received: {}", _attr.to_string());
+    return TokenStream::from(quote!(#input));
+}

src/handler/health.rs

@@ -1,10 +1,8 @@
-use axum::{
+use crate::model::generic_response::GenericResponse;
-    http::StatusCode,
+use axum::{http::StatusCode, response::IntoResponse, Json};
-    response::IntoResponse,
+use proc_utils::guard_resource;
-    Json,
-};
-use crate::model::generic_response::{GenericResponse};
 
+#[guard_resource(ResourceType::OPEN)]
 pub async fn health_check() -> impl IntoResponse {
     const MESSAGE: &str = "Server is running";
     let response = GenericResponse {
@@ -13,4 +11,4 @@ pub async fn health_check() -> impl IntoResponse {
     };
 
     (StatusCode::OK, Json(response))
-}
+}

src/handler/message.rs

@@ -1,10 +1,8 @@
-use axum::{
+use crate::model::send_message::SendMessage;
-    http::StatusCode,
+use axum::{http::StatusCode, response::IntoResponse, Json};
-    response::IntoResponse,
+use proc_utils::guard_resource;
-    Json,
-};
-use crate::model::send_message::{SendMessage};
 
+#[guard_resource(ResourceType::OPEN)]
 pub async fn send_message(Json(payload): Json<SendMessage>) -> impl IntoResponse {
     (StatusCode::OK, Json(payload))
-}
+}

src/handler/mod.rs

@@ -1,2 +1,2 @@
 pub mod health;
-pub mod message;
+pub mod message;

src/interceptor/mod.rs

@@ -0,0 +1 @@
+

src/main.rs

@@ -1,11 +1,7 @@
-mod route;
-mod model;
 mod handler;
-
+mod interceptor;
-use axum::http::{
+mod model;
-    header::{ACCEPT, AUTHORIZATION, CONTENT_TYPE},
+mod route;
-    HeaderValue, Method
-};
 
 #[tokio::main]
 async fn main() {
@@ -14,4 +10,4 @@ async fn main() {
     println!("🚀 Server started successfully");
     let listener = tokio::net::TcpListener::bind("0.0.0.0:8500").await.unwrap();
     axum::serve(listener, app).await.unwrap();
-}
+}

src/model/generic_response.rs

@@ -1,8 +1,7 @@
 use serde::Serialize;
 
-
 #[derive(Serialize)]
 pub struct GenericResponse {
     pub status: String,
     pub message: String,
-}
+}

src/model/mod.rs

@@ -1,2 +1,2 @@
 pub mod generic_response;
-pub mod send_message;
+pub mod send_message;

src/model/send_message.rs

@@ -1,18 +1,15 @@
+use chrono::naive::serde::ts_seconds::deserialize as from_ts;
 use chrono::NaiveDateTime;
 use serde::{Deserialize, Serialize};
 use serde_with::serde_as;
-use chrono::naive::serde::ts_seconds::deserialize as from_ts;
-
-
 
 #[derive(Deserialize, Serialize)]
 pub struct MessageAuthor {
     name: String,
     username: String,
-    email: String
+    email: String,
 }
 
-
 #[serde_as]
 #[derive(Deserialize, Serialize)]
 pub struct SendMessage {
@@ -23,5 +20,5 @@ pub struct SendMessage {
     message: String,
 
     #[serde(deserialize_with = "from_ts")]
-    timestamp: NaiveDateTime
+    timestamp: NaiveDateTime,
-}
+}

src/route.rs

@@ -1,13 +1,12 @@
+use crate::handler::health::health_check;
+use crate::handler::message::send_message;
 use axum::{
     routing::{get, post},
-    Router
+    Router,
 };
-use crate::handler::health::{health_check};
-use crate::handler::message::{send_message};
-
 
 pub fn create_route() -> Router {
     Router::new()
         .route("/health", get(health_check))
         .route("/message", post(send_message))
-}
+}

utils/Cargo.toml

@@ -0,0 +1,8 @@
+[package]
+name = "utils"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]

utils/src/lib.rs

@@ -0,0 +1,24 @@
+pub enum ResourceType {
+    OPEN,
+    USER,
+    ADMIN,
+}
+
+impl ResourceType {
+    fn as_str(&self) -> &str {
+        match self {
+            ResourceType::OPEN => "OPEN",
+            ResourceType::USER => "USER",
+            ResourceType::ADMIN => "ADMIN",
+        }
+    }
+
+    fn from_str(s: &str) -> ResourceType {
+        match s {
+            "OPEN" => ResourceType::OPEN,
+            "USER" => ResourceType::USER,
+            "ADMIN" => ResourceType::ADMIN,
+            _ => panic!("Invalid resource type"),
+        }
+    }
+}