57 lines
1.5 KiB
Rust
57 lines
1.5 KiB
Rust
#![feature(proc_macro_hygiene, decl_macro)]
|
|
|
|
use rocket::{routes, Rocket};
|
|
use serde_derive::Deserialize;
|
|
use std::sync::RwLock;
|
|
use toml;
|
|
|
|
mod handlers;
|
|
mod models;
|
|
|
|
type NoteStore = RwLock<Vec<models::note::Note>>;
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
pub struct Config {
|
|
title: String,
|
|
}
|
|
|
|
fn rocket(notes_path: Option<&str>) -> Rocket {
|
|
use std::fs::File;
|
|
use std::io::prelude::*;
|
|
use std::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::ignite()
|
|
.manage(RwLock::new(notes))
|
|
.manage(config)
|
|
.mount(
|
|
"/",
|
|
routes![handlers::home::index, handlers::static_files::all],
|
|
)
|
|
.mount(
|
|
"/notes",
|
|
routes![
|
|
handlers::note::index,
|
|
handlers::note::show_html,
|
|
handlers::note::show_json,
|
|
handlers::note::update
|
|
],
|
|
)
|
|
.attach(rocket_contrib::templates::Template::fairing())
|
|
}
|
|
|
|
fn main() {
|
|
rocket(None).launch();
|
|
}
|