88 lines
3.1 KiB
Rust
88 lines
3.1 KiB
Rust
// SPDX-FileCopyrightText: 2023 Sayantan Santra <sayantan.santra689@gmail.com>
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
use actix_files::Files;
|
|
use actix_session::{storage::CookieSessionStore, SessionMiddleware};
|
|
use actix_web::{cookie::Key, middleware, web, App, HttpServer};
|
|
use rusqlite::Connection;
|
|
use std::{env, io::Result};
|
|
|
|
// Import modules
|
|
mod auth;
|
|
mod database;
|
|
mod services;
|
|
mod utils;
|
|
|
|
// This struct represents state
|
|
struct AppState {
|
|
db: Connection,
|
|
}
|
|
|
|
#[actix_web::main]
|
|
async fn main() -> Result<()> {
|
|
env_logger::init_from_env(env_logger::Env::new().default_filter_or("warn"));
|
|
|
|
// Generate session key in runtime so that restart invalidates older logins
|
|
let secret_key = Key::generate();
|
|
|
|
let db_location = env::var("db_url")
|
|
.ok()
|
|
.filter(|s| !s.trim().is_empty())
|
|
.unwrap_or(String::from("urls.sqlite"));
|
|
|
|
|
|
let port = env::var("port")
|
|
.unwrap_or(String::from("4567"))
|
|
.parse::<u16>()
|
|
.expect("Supplied port is not an integer");
|
|
|
|
let cache_control_header = env::var("cache_control_header")
|
|
.ok()
|
|
.filter(|s| !s.trim().is_empty());
|
|
|
|
// If an API key is set, check the security
|
|
if let Ok(key) = env::var("api_key") {
|
|
if !auth::is_key_secure() {
|
|
eprintln!("API key is insecure! Please change it. Current key is: {}. Generated secure key which you may use: {}", key, auth::gen_key())
|
|
}
|
|
}
|
|
|
|
// Tell the user that the server has started, and where it is listening to, rather than simply outputting nothing
|
|
eprintln!("Server has started at 0.0.0.0 on port {}. Configured Site URL is: {}", port, env::var("site_url").unwrap_or(String::from("http://localhost")));
|
|
|
|
// Actually start the server
|
|
HttpServer::new(move || {
|
|
App::new()
|
|
.wrap(middleware::Logger::default())
|
|
.wrap(middleware::Compress::default())
|
|
.wrap(
|
|
SessionMiddleware::builder(CookieSessionStore::default(), secret_key.clone())
|
|
.cookie_same_site(actix_web::cookie::SameSite::Strict)
|
|
.cookie_secure(false)
|
|
.build(),
|
|
)
|
|
// Maintain a single instance of database throughout
|
|
.app_data(web::Data::new(AppState {
|
|
db: database::open_db(db_location.clone()),
|
|
}))
|
|
.wrap(if let Some(header) = &cache_control_header {
|
|
middleware::DefaultHeaders::new().add(("Cache-Control", header.to_owned()))
|
|
} else {
|
|
middleware::DefaultHeaders::new()
|
|
})
|
|
.service(services::link_handler)
|
|
.service(services::getall)
|
|
.service(services::siteurl)
|
|
.service(services::version)
|
|
.service(services::add_link)
|
|
.service(services::delete_link)
|
|
.service(services::login)
|
|
.service(services::logout)
|
|
.service(Files::new("/", "./resources/").index_file("index.html"))
|
|
.default_service(actix_web::web::get().to(services::error404))
|
|
})
|
|
.bind(("0.0.0.0", port))?
|
|
.run()
|
|
.await
|
|
}
|