lyrs/src/main.rs

76 lines
2.1 KiB
Rust
Raw Normal View History

use std::{
cell::OnceCell,
path::{Path, PathBuf},
str::FromStr,
};
use tokio::sync::mpsc::channel;
use notify::{Config, RecommendedWatcher, RecursiveMode, Watcher};
mod observe;
mod prelude;
use axum::{serve, Router};
use tower_http::services::ServeDir;
use tower_livereload::LiveReloadLayer;
use crate::prelude::*;
const LIVE_RELOADING: bool = true;
const STATIC_FILE_PATH: OnceCell<PathBuf> = OnceCell::new();
fn static_file_dir() -> &'static Path {
STATIC_FILE_PATH.get_or_init(|| PathBuf::from_str("static").unwrap())
}
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
// load configuration?
observe::setup_logging();
info!("Creating async watcher...");
let (tx, mut rx) = channel(1);
info!("Creating watcher...");
let mut watcher = RecommendedWatcher::new(
move |res| {
info!("Res from watcher: {res:#?}");
futures::executor::block_on(async {
tx.send(res).await.unwrap();
})
},
Config::default(),
)?;
info!("Created watcher");
watcher.watch(static_file_dir(), RecursiveMode::Recursive)?;
let rl_layer = LiveReloadLayer::new();
let rl = rl_layer.reloader();
tokio::spawn(async move {
info!("Recieving...");
while let Some(res) = rx.recv().await {
info!("Recieved! {res:#?}");
match res {
Ok(event) => {
info!("fs event: {event:#?}");
rl.reload()
}
Err(e) => println!("watch error: {:?}", e),
}
}
});
let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
info!("Listening on {listener:#?}");
Ok(serve(listener, static_files()?.layer(rl_layer)).await?)
}
fn static_files() -> Result<Router, notify::Error> {
let static_file_path = Path::new("./static/");
// serve the file in the "assets" directory under `/assets`
info!("Starting static file server...");
Ok(Router::new().nest_service("/static", ServeDir::new(static_file_path)))
2024-05-05 09:47:13 -05:00
}