Merge pull request #12 from dante-gpu/multiOS

Major System Overhaul

Author: virjilakrum

Cargo.toml

@@ -2,10 +2,13 @@
 name = "gpu-share-vm-manager"
 version = "0.1.0"
 edition = "2021"
+resolver = "2"
+default-run = "gpu-share-vm-manager"
 
 [dependencies]
+
 tokio = { version = "1.36", features = ["full"] }
-virt = "0.4.1"
+# virt = "0.4.1" 
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 tracing = "0.1"
@@ -14,16 +17,20 @@ anyhow = "1.0"
 async-trait = "0.1"
 config = "0.15.6"
 axum = { version = "0.8.0", features = ["macros"] }
-hyper = { version = "1.0", features = ["full"] }
+hyper = { version = "0.14.32", features = ["full"] }
 tower = { version = "0.5.2", features = ["limit", "util"] }
 tower-http = { version = "0.6.2", features = ["trace", "limit", "add-extension"] }
 clap = { version = "4.4", features = ["derive"] }
 colored = "3.0"
 thiserror = "2.0.11"
-chrono = "0.4"
+chrono = { version = "0.4", features = ["serde"] }
 uuid = { version = "1.8.0", features = ["v4"] }
 governor = { version = "0.8", features = ["dashmap"] }
-jsonwebtoken = "8.3.0"
+jsonwebtoken = "9.3.0"
+bollard = "0.15.0"
+futures-util = "0.3"
+ratatui = "0.26"
+crossterm = "0.27"
 
 [target.'cfg(target_os = "linux")'.dependencies]
 nvml-wrapper = { version = "0.10.0", optional = true }
@@ -40,4 +47,12 @@ windows = { version = "0.48", features = ["Win32_Graphics_Dxgi"] }
 [features]
 default = ["metal"]
 metal = ["dep:core-graphics", "dep:metal"]
-windows = ["dep:dxgi", "winapi"]
+windows = ["dep:dxgi", "winapi"]
+
+[[bin]]
+name = "gpu-share-vm-manager"
+path = "src/main.rs"
+
+[dev-dependencies]
+tokio = { version = "1.0", features = ["full"] }
+rand = "0.8"

src/api/error.rs

@@ -0,0 +1,24 @@
+#[derive(Debug)]
+pub struct ErrorResponse {
+    pub code: ErrorNumber,
+    pub message: String,
+}
+
+impl ErrorResponse {
+    pub fn new(code: ErrorNumber, message: String) -> Self {
+        Self { code, message }
+    }
+}
+
+impl IntoResponse for ErrorResponse {
+    fn into_response(self) -> Response {
+        (
+            StatusCode::INTERNAL_SERVER_ERROR,
+            Json(json!({
+                "error_code": self.code as u32,
+                "message": self.message
+            })),
+        )
+            .into_response()
+    }
+} 

src/api/middleware/mod.rs

@@ -1 +1,5 @@
-pub mod rate_limit; 
+// src/api/middleware/mod.rs
+//! This module groups middleware for the API.
+//! Currently it only re-exports the rate_limit middleware.
+
+pub mod rate_limit;

src/api/middleware/rate_limit.rs

@@ -112,7 +112,52 @@ impl fmt::Display for RateLimitExceeded {
         write!(f, "Rate limit exceeded")
     }
 }
+#[derive(Clone)]
+pub struct RateLimit<T> {
+    inner: T,
+}
+
+// wrapper for RateLimitLayer
+#[derive(Clone)]
+pub struct CustomRateLimitLayer {
+    rate: u64,
+    per: Duration,
+    inner: RateLimitLayer,
+}
+
+impl CustomRateLimitLayer {
+    pub fn new(rate: u64, per: Duration) -> Self {
+        Self {
+            rate,
+            per,
+            inner: RateLimitLayer::new(rate, per),
+        }
+    }
+
+    pub fn get_rate(&self) -> u64 {
+        self.rate
+    }
 
+    pub fn get_per(&self) -> Duration {
+        self.per
+    }
+
+    pub fn into_inner(self) -> RateLimitLayer {
+        self.inner
+    }
+}
+
+impl From<RateLimitLayer> for CustomRateLimitLayer {
+    fn from(_layer: RateLimitLayer) -> Self {
+        Self::new(100, Duration::from_secs(1))
+    }
+}
+
+impl From<CustomRateLimitLayer> for RateLimitLayer {
+    fn from(custom: CustomRateLimitLayer) -> Self {
+        custom.into_inner()
+    }
+}
 
 #[cfg(test)]
 mod tests {

src/api/mod.rs

@@ -1,4 +1,4 @@
 pub mod middleware;
 pub mod routes;
 
-pub use routes::{create_router, AppState};
+pub use routes::{AppState, create_router};