feat: add support for disable https to run http server

This commit is contained in:
Oskar 2022-09-26 20:04:28 +00:00
parent 1b9529223a
commit ff02a1e180
3 changed files with 42 additions and 20 deletions

View File

@ -1,6 +1,7 @@
[http]
host = "0.0.0.0"
httpPort = 8080 # http server port (recommended 80)
enableHttps = true # if true start https server and http will be redirect to https
httpsPort = 8443 # https server port (recommended 443)
cors = [ "localhost:8000" ] # CORS domains
tlsCert = "./cert.pem" # TLS certificate file

View File

@ -14,6 +14,7 @@ pub struct Config {
pub struct ConfigHTTP {
pub host: String,
pub http_port: u16,
pub enable_https: bool,
pub https_port: u16,
pub cors: Vec<String>,
pub tls_cert: String,

View File

@ -2,7 +2,10 @@ mod api;
pub mod error;
pub mod utils;
use std::{path::{PathBuf, Path}, process::exit};
use std::{
path::{Path, PathBuf},
process::exit,
};
use anyhow::anyhow;
use axum::{
@ -18,21 +21,28 @@ use tower_http::{
cors::{AllowOrigin, CorsLayer},
BoxError,
};
use tracing::{debug, info, error};
use tracing::{debug, error, info};
use crate::{config::Config, database::Database};
pub async fn start_server(config: Config, db: Database) -> anyhow::Result<()> {
let host = format!("{}:{}", config.http.host, config.http.https_port);
let host = if config.http.enable_https {
format!("{}:{}", config.http.host, config.http.https_port)
} else {
format!("{}:{}", config.http.host, config.http.http_port)
};
// check if tls cert and key file exists
if !Path::new(&config.http.tls_cert).exists() || !Path::new(&config.http.tls_key).exists() {
error!("TLS cert or/and key file not found!");
exit(1);
if config.http.enable_https {
// check if tls cert and key file exists
if !Path::new(&config.http.tls_cert).exists() || !Path::new(&config.http.tls_key).exists() {
error!("TLS cert or/and key file not found!");
exit(1);
}
// start http redirect to https
tokio::spawn(redirect_http_to_https(config.clone()));
}
tokio::spawn(redirect_http_to_https(config.clone()));
info!("🚀 Server has launched on https://{host}");
// change the type from Vec<String> to Vec<HeaderValue> so that the http server can correctly detect CORS hosts
@ -43,23 +53,33 @@ pub async fn start_server(config: Config, db: Database) -> anyhow::Result<()> {
.map(|e| e.parse().expect("Failed to parse CORS hosts"))
.collect::<Vec<HeaderValue>>();
let tls_config = RustlsConfig::from_pem_file(
PathBuf::from("").join("").join(&config.http.tls_cert),
PathBuf::from("").join("").join(&config.http.tls_key),
)
.await
.unwrap();
let app = Router::new()
.nest("/api", api::app())
.route("/", get(api::health))
.layer(CorsLayer::new().allow_origin(AllowOrigin::list(origins)))
.layer(Extension(config))
.layer(Extension(config.clone()))
.layer(Extension(db));
axum_server::bind_rustls(host.parse()?, tls_config)
.serve(app.into_make_service())
.await?;
// if https is enabled, run it, otherwise run server http
if config.http.enable_https {
// start https server
let tls_config = RustlsConfig::from_pem_file(
PathBuf::from("").join("").join(&config.http.tls_cert),
PathBuf::from("").join("").join(&config.http.tls_key),
)
.await
.unwrap();
axum_server::bind_rustls(host.parse()?, tls_config)
.serve(app.into_make_service())
.await?;
} else {
// start http server
axum::Server::bind(&host.parse().unwrap())
.serve(app.into_make_service())
.await
.unwrap();
}
Err(anyhow!("Server unexpected stopped!"))
}