mod handlers; mod models; use std::sync::RwLock; use rocket::{ fs::{relative, FileServer}, routes, serde::Deserialize, Build, Rocket, }; use rocket_dyn_templates::Template; type NoteStore = RwLock>; #[derive(Debug, Deserialize)] #[serde(crate = "rocket::serde")] pub struct Config { title: String, } fn build_rocket(notes_path: Option<&str>) -> Rocket { use std::{fs::File, io::prelude::*, path::Path}; let mut config_data = String::new(); let config_file_name = Path::new(env!("CARGO_MANIFEST_DIR")).join("config.toml"); let mut config_file = File::open(config_file_name).expect("Cannot find config file: config.toml"); config_file .read_to_string(&mut config_data) .expect("Cannot read config file: config.toml"); let config: Config = toml::from_str(&config_data).expect("Cannot parse config file: config.toml"); let notes = models::note::Note::load_all(notes_path); rocket::build() .manage(RwLock::new(notes)) .manage(config) .mount("/", FileServer::from(relative!("static"))) .mount("/", routes![handlers::home::index]) .mount( "/notes", routes![ handlers::note::index, handlers::note::show_html, handlers::note::show_json, handlers::note::update ], ) .attach(Template::fairing()) } #[rocket::launch] fn rocket() -> _ { build_rocket(None) }