surrealpatch/src/net/mod.rs

mod export;
mod fail;
mod head;
mod health;
mod import;
mod index;
mod key;
mod log;
mod output;
mod rpc;
mod session;
mod signin;
mod signup;
mod sql;
mod status;
mod sync;
mod version;

use crate::cli::CF;
use crate::err::Error;
use warp::Filter;

const LOG: &str = "surrealdb::net";

pub async fn init() -> Result<(), Error> {
	// Setup web routes
	let net = index::config()
		// Version endpoint
		.or(version::config())
		// Status endpoint
		.or(status::config())
		// Health endpoint
		.or(health::config())
		// Signup endpoint
		.or(signup::config())
		// Signin endpoint
		.or(signin::config())
		// Export endpoint
		.or(export::config())
		// Import endpoint
		.or(import::config())
		// Backup endpoint
		.or(sync::config())
		// RPC query endpoint
		.or(rpc::config())
		// SQL query endpoint
		.or(sql::config())
		// API query endpoint
		.or(key::config())
		// Catch all errors
		.recover(fail::recover)
		// End routes setup
	;
	// Specify a generic version header
	let net = net.with(head::version());
	// Specify a generic server header
	let net = net.with(head::server());
	// Set cors headers on all requests
	let net = net.with(head::cors());
	// Log all requests to the console
	let net = net.with(log::write());

	// Get local copy of options
	let opt = CF.get().unwrap();

	info!(target: LOG, "Starting web server on {}", &opt.bind);

	if let (Some(c), Some(k)) = (&opt.crt, &opt.key) {
		// Bind the server to the desired port
		let (adr, srv) = warp::serve(net)
			.tls()
			.cert_path(c)
			.key_path(k)
			.bind_with_graceful_shutdown(opt.bind, async move {
				tokio::signal::ctrl_c().await.expect("Failed to listen to shutdown signal");
			});
		// Log the server startup status
		info!(target: LOG, "Started web server on {}", &adr);
		// Run the server forever
		srv.await
	} else {
		// Bind the server to the desired port
		let (adr, srv) = warp::serve(net).bind_with_graceful_shutdown(opt.bind, async move {
			tokio::signal::ctrl_c().await.expect("Failed to listen to shutdown signal");
		});
		// Log the server startup status
		info!(target: LOG, "Started web server on {}", &adr);
		// Run the server forever
		srv.await
	};

	Ok(())
}