You cannot select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
	
	
		
			117 lines
		
	
	
		
			3.9 KiB
		
	
	
	
		
			Rust
		
	
			
		
		
	
	
			117 lines
		
	
	
		
			3.9 KiB
		
	
	
	
		
			Rust
		
	
| use log::error;
 | |
| use rezervator::backend::company::check_company;
 | |
| use rezervator::backend::mail::check_messages;
 | |
| use rezervator::backend::user::create_admin;
 | |
| 
 | |
| #[cfg(feature = "ssr")]
 | |
| #[actix_web::main]
 | |
| async fn main() -> std::io::Result<()> {
 | |
|     use actix_files::Files;
 | |
|     use actix_web::*;
 | |
|     use leptos::*;
 | |
|     use leptos_actix::{generate_route_list, LeptosRoutes};
 | |
|     use rezervator::app::*;
 | |
|     use rezervator::backend::*;
 | |
|     use actix_session::storage::CookieSessionStore;
 | |
|     use actix_session::SessionMiddleware;
 | |
|     use actix_web::cookie::Key;
 | |
|     use actix_web::web::Data;
 | |
|     use sqlx::migrate;
 | |
|     use sqlx::postgres::PgPoolOptions;
 | |
|     use rezervator::backend::auth_middleware::Authentication;
 | |
|     use rezervator::config::load_config;
 | |
|     use env_logger::Env;
 | |
|     use std::env::args;
 | |
|     use getopts::Options;
 | |
|     use log::info;
 | |
| 
 | |
|     let args: Vec<String> = args().collect();
 | |
|     let mut opts = Options::new();
 | |
|     opts.optopt("c", "config", "Path to config file. Default is ./config.toml", "");
 | |
|     opts.optflag("h", "help", "Show help");
 | |
|     let matches = opts.parse(&args[1..]).expect("Can't parse commandline");
 | |
| 
 | |
|     if matches.opt_present("h") {
 | |
|         println!("{}",opts.usage(format!("Usage: {} [options]",
 | |
|                                          args.get(0).unwrap_or(&"".to_string())).as_str()));
 | |
|         return Ok(());
 | |
|     }
 | |
| 
 | |
|     let cfg_path = matches.opt_str("c").unwrap_or("config.toml".to_string());
 | |
| 
 | |
|     env_logger::Builder::from_env(Env::default().default_filter_or("debug")).init();
 | |
|     info!("Starting server");
 | |
| 
 | |
|     let conf = get_configuration(None).await.unwrap();
 | |
|     let srv_conf = load_config(&cfg_path);
 | |
|     // Generate the list of routes in your Leptos App
 | |
|     let routes = generate_route_list(|| view! { <App/> });
 | |
|     let key = Key::from(srv_conf.session().key());
 | |
| 
 | |
|     let pool = PgPoolOptions::new()
 | |
|         .max_connections(10)
 | |
|         .connect(&srv_conf.database().con_string()).await.unwrap();
 | |
| 
 | |
|     migrate!().run(&pool).await.expect("could not run SQLx migrations");
 | |
| 
 | |
|     if let Err(e) = create_admin(&pool).await {
 | |
|         error!("Error while checking admin user: {:?}", e);
 | |
|     }
 | |
|     if let Err(e) = check_company(&pool).await {
 | |
|         error!("Error while checking company: {:?}", e);
 | |
|     }
 | |
|     if let Err(e) = check_messages(&pool).await {
 | |
|         error!("Error while checking messages: {:?}", e);
 | |
|     }
 | |
| 
 | |
|     HttpServer::new(move || {
 | |
|         let leptos_options = &conf.leptos_options;
 | |
|         let site_root = &leptos_options.site_root;
 | |
| 
 | |
|         App::new()
 | |
|             .app_data(Data::new(AppData::new(pool.clone())))
 | |
|             .wrap(Authentication)
 | |
|             .wrap(SessionMiddleware::new(
 | |
|                 CookieSessionStore::default(),
 | |
|                 key.clone()
 | |
|             ))
 | |
|             .route("/api/{tail:.*}", leptos_actix::handle_server_fns())
 | |
|             .leptos_routes(
 | |
|                 leptos_options.to_owned(),
 | |
|                 routes.to_owned(),
 | |
|                 || view! { <App/> },
 | |
|             )
 | |
|             .service(Files::new("/", site_root))
 | |
|         //.wrap(middleware::Compress::default())
 | |
|     })
 | |
|     .bind(srv_conf.network().bind_address())?
 | |
|     .run()
 | |
|     .await
 | |
| }
 | |
| 
 | |
| #[cfg(not(any(feature = "ssr", feature = "csr")))]
 | |
| pub fn main() {
 | |
|     // no client-side main function
 | |
|     // unless we want this to work with e.g., Trunk for pure client-side testing
 | |
|     // see lib.rs for hydration function instead
 | |
|     // see optional feature `ssg` instead
 | |
| }
 | |
| 
 | |
| #[cfg(all(not(feature = "ssr"), feature = "csr"))]
 | |
| pub fn main() {
 | |
|     // a client-side main function is required for using `trunk serve`
 | |
|     // prefer using `cargo leptos serve` instead
 | |
|     // to run: `trunk serve --open --features ssg`
 | |
|     use leptos::*;
 | |
|     use leptos_start::app::*;
 | |
|     use wasm_bindgen::prelude::wasm_bindgen;
 | |
| 
 | |
|     console_error_panic_hook::set_once();
 | |
| 
 | |
|     leptos::mount_to_body(move || {
 | |
|         // note: for testing it may be preferrable to replace this with a
 | |
|         // more specific component, although leptos_router should still work
 | |
|         view! {<App/> }
 | |
|     });
 | |
| }
 |