94 lines
2.6 KiB
Rust
94 lines
2.6 KiB
Rust
use std::{
|
|
cell::OnceCell,
|
|
path::{Path, PathBuf},
|
|
str::FromStr,
|
|
};
|
|
use tokio::sync::mpsc::channel;
|
|
|
|
use notify::{Config, RecommendedWatcher, RecursiveMode, Watcher};
|
|
|
|
mod observe {
|
|
use tracing::level_filters::LevelFilter;
|
|
use tracing_subscriber::EnvFilter;
|
|
|
|
pub fn setup_logging() {
|
|
color_eyre::install().expect("Failed to install color_eyre");
|
|
|
|
let filter = EnvFilter::builder()
|
|
.with_default_directive(LevelFilter::TRACE.into())
|
|
.parse_lossy("info,chatbot=trace");
|
|
|
|
tracing_subscriber::fmt().with_env_filter(filter).init();
|
|
}
|
|
}
|
|
|
|
mod prelude {
|
|
#![allow(unused_imports)]
|
|
pub use tracing::{debug, error, info, warn};
|
|
}
|
|
|
|
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)))
|
|
}
|