webu

1 files changed, 4 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index 5096ae8..354d303 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,3 +1,4 @@
+pub mod helper;
 pub mod webui;
 pub mod worker_ws;
 
@@ -14,10 +15,10 @@ use tokio::{
     net::TcpListener,
     sync::{RwLock, mpsc::Sender},
 };
-use webui::webui;
+use webui::{webui, webui_style};
 use worker_ws::{WorkerID, WorkerResponse, worker_websocket};
 
-struct Worker {
+pub struct Worker {
     accept: usize,
     name: String,
     sources: Vec<String>,
@@ -42,6 +43,7 @@ async fn main() -> Result<()> {
     state.load().await?;
     let router = Router::new()
         .route("/", get(webui))
+        .route("/style.css", get(webui_style))
         .route("/worker_ws", get(worker_websocket))
         .with_state(Arc::new(RwLock::new(state)));
     let listener = TcpListener::bind("127.0.0.1:8080").await?;