83 lines
No EOL
2.3 KiB
Rust
83 lines
No EOL
2.3 KiB
Rust
#![feature(decl_macro)]
|
|
#![feature(lazy_cell)]
|
|
|
|
#[macro_use] extern crate maplit;
|
|
#[macro_use] extern crate rocket;
|
|
|
|
use std::fs;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
use rocket::fs::NamedFile;
|
|
use rocket::data::{Limits, ToByteUnit};
|
|
|
|
use rocket_dyn_templates::Template;
|
|
|
|
mod db;
|
|
use db::*;
|
|
|
|
mod helpers;
|
|
use helpers::*;
|
|
|
|
mod endpoints;
|
|
use endpoints::*;
|
|
|
|
mod template_endpoints;
|
|
use template_endpoints::*;
|
|
|
|
mod config;
|
|
use config::*;
|
|
|
|
#[get("/<file..>")]
|
|
async fn files(file: PathBuf) -> Option<NamedFile> {
|
|
NamedFile::open(Path::new("public/").join(file)).await.ok()
|
|
}
|
|
|
|
#[launch]
|
|
fn rocket() -> _ {
|
|
// init stuff
|
|
crate::helpers::reupload::init();
|
|
|
|
// data directories
|
|
// this is a bit scuffed
|
|
fs::create_dir_all(&CONFIG.db.data_folder).expect("failed to create data directory!");
|
|
fs::create_dir_all(format!("{}/levels", &CONFIG.db.data_folder)).expect("failed to create data directory for levels");
|
|
|
|
rocket::build()
|
|
// conf
|
|
.configure(rocket::Config::figment()
|
|
.merge(("port", CONFIG.general.port))
|
|
.merge(("ip_header", CONFIG.general.realip_header.as_str()))
|
|
.merge(("limits", Limits::new().limit("forms", 10.megabytes()))))
|
|
// actual website
|
|
.mount("/", routes![
|
|
template_endpoints::index::index,
|
|
|
|
template_endpoints::reupload::post_reupload,
|
|
template_endpoints::reupload::get_reupload,
|
|
|
|
template_endpoints::login::post_login,
|
|
template_endpoints::login::get_login,
|
|
|
|
template_endpoints::account_management::account_management,
|
|
|
|
template_endpoints::logout::logout
|
|
])
|
|
// assets
|
|
.mount("/", routes![
|
|
files
|
|
])
|
|
// https://www.youtube.com/watch?v=_pLrtsf5yfE
|
|
.mount(CONFIG.general.append_path.as_str(), routes![
|
|
endpoints::accounts::login_account::login_account,
|
|
endpoints::accounts::register_account::register_account,
|
|
endpoints::accounts::update_account_settings::update_account_settings,
|
|
|
|
endpoints::users::get_users::get_users,
|
|
|
|
endpoints::levels::download_level::download_level,
|
|
endpoints::levels::get_levels::get_levels,
|
|
endpoints::levels::upload_level::upload_level
|
|
])
|
|
// so templates work i think
|
|
.attach(Template::fairing())
|
|
} |