Merge pull request #14 from FxQnLr/axum7

Axum7 & config changes

1 files changed, 24 insertions, 17 deletions

diff --git a/src/main.rs b/src/main.rs
index e96b736..4ef129b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -8,12 +8,12 @@ use time::util::local_offset;
 use tokio::sync::broadcast::{channel, Sender};
 use tracing::{info, level_filters::LevelFilter};
 use tracing_subscriber::{EnvFilter, fmt::{self, time::LocalTime}, prelude::*};
-use crate::config::SETTINGS;
+use crate::config::Config;
 use crate::db::init_db_pool;
-use crate::routes::device::{get_device, post_device, put_device};
+use crate::routes::device;
 use crate::routes::start::start;
 use crate::routes::status::status;
-use crate::services::ping::{BroadcastCommands, PingMap};
+use crate::services::ping::{BroadcastCommands, StatusMap};
 
 mod auth;
 mod config;
@@ -24,7 +24,10 @@ mod error;
 mod services;
 
 #[tokio::main]
-async fn main() {
+async fn main() -> color_eyre::eyre::Result<()> {
+
+    color_eyre::install()?;
+
     unsafe { local_offset::set_soundness(local_offset::Soundness::Unsound); }
     let time_format =
         time::macros::format_description!("[year]-[month]-[day] [hour]:[minute]:[second]");
@@ -43,35 +46,39 @@ async fn main() {
 
     let version = env!("CARGO_PKG_VERSION");
 
+    let config = Config::load()?;
+
     info!("start webol v{}", version);
 
-    let db = init_db_pool().await;
+    let db = init_db_pool(&config.database_url).await;
     sqlx::migrate!().run(&db).await.unwrap();
 
     let (tx, _) = channel(32);
 
-    let ping_map: PingMap = DashMap::new();
+    let ping_map: StatusMap = DashMap::new();
     
-    let shared_state = Arc::new(AppState { db, ping_send: tx, ping_map });
+    let shared_state = Arc::new(AppState { db, config: config.clone(), ping_send: tx, ping_map });
 
     let app = Router::new()
         .route("/start", post(start))
-        .route("/device", get(get_device))
-        .route("/device", put(put_device))
-        .route("/device", post(post_device))
+        .route("/device", get(device::get))
+        .route("/device", put(device::put))
+        .route("/device", post(device::post))
         .route("/status", get(status))
         .with_state(shared_state);
 
-    let addr = SETTINGS.get_string("serveraddr").unwrap_or("0.0.0.0:7229".to_string());
+    let addr = config.serveraddr;
     info!("start server on {}", addr);
-    axum::Server::bind(&addr.parse().unwrap())
-        .serve(app.into_make_service())
-        .await
-        .unwrap();
+    let listener = tokio::net::TcpListener::bind(addr)
+        .await?;
+    axum::serve(listener, app).await?;
+
+    Ok(())
 }
 
 pub struct AppState {
     db: PgPool,
+    config: Config,
     ping_send: Sender<BroadcastCommands>,
-    ping_map: PingMap,
-}
\ No newline at end of file
+    ping_map: StatusMap,
+}