Joshua Higgins
41b27f018d
gonna move on now, kind of want testing but its gonna have to be another day
85 lines
3.4 KiB
Rust
85 lines
3.4 KiB
Rust
use std::env;
|
|
use std::future::Future;
|
|
use std::net::{IpAddr, Ipv6Addr};
|
|
use dotenvy::dotenv;
|
|
use futures::{future, StreamExt};
|
|
use sqlx::{migrate, Sqlite, SqlitePool};
|
|
use sqlx::migrate::MigrateDatabase;
|
|
use tarpc::server::{BaseChannel, Channel};
|
|
use tarpc::server::incoming::Incoming;
|
|
use tarpc::tokio_serde::formats::Json;
|
|
use realm_auth::server::RealmAuthServer;
|
|
use realm_auth::types::{AuthEmail, RealmAuth};
|
|
use tracing::*;
|
|
|
|
async fn spawn(fut: impl Future<Output = ()> + Send + 'static) {
|
|
tokio::spawn(fut);
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
dotenv().ok();
|
|
|
|
let subscriber = tracing_subscriber::fmt()
|
|
.compact()
|
|
.with_file(true)
|
|
.with_line_number(true)
|
|
.with_thread_ids(true)
|
|
.with_target(false)
|
|
.finish();
|
|
|
|
subscriber::set_global_default(subscriber).unwrap();
|
|
|
|
let auth_email = AuthEmail {
|
|
server_address: env::var("SERVER_MAIL_ADDRESS").expect("SERVER_MAIL_ADDRESS must be set"),
|
|
server_port: env::var("SERVER_MAIL_PORT").expect("SERVER_MAIL_PORT must be set").parse::<u16>().expect("SERVER_MAIL_ADDRESS must be a number"),
|
|
auth_name: env::var("SERVER_MAIL_NAME").expect("SERVER_MAIL_NAME must be set"),
|
|
auth_from_address: env::var("SERVER_MAIL_FROM_ADDRESS").expect("SERVER_MAIL_FROM_ADDRESS must be set"),
|
|
auth_username: env::var("SERVER_MAIL_USERNAME").expect("SERVER_MAIL_USERNAME must be set"),
|
|
auth_password: env::var("SERVER_MAIL_PASSWORD").expect("SERVER_MAIL_PASSWORD must be set"),
|
|
};
|
|
|
|
let DB_URL: &str = &env::var("DATABASE_URL").expect("DATABASE_URL must be set");
|
|
|
|
if !Sqlite::database_exists(DB_URL).await.unwrap_or(false) {
|
|
info!("Creating database {}", DB_URL);
|
|
match Sqlite::create_database(DB_URL).await {
|
|
Ok(_) => info!("Create db success"),
|
|
Err(error) => panic!("error: {}", error),
|
|
}
|
|
} else {
|
|
warn!("Database already exists");
|
|
} // TODO: Do in Docker with Sqlx-cli
|
|
|
|
let db_pool = SqlitePool::connect(DB_URL).await.unwrap();
|
|
|
|
info!("Running migrations...");
|
|
migrate!().run(&db_pool).await?; // TODO: Do in Docker with Sqlx-cli
|
|
info!("Migrations complete!");
|
|
|
|
let server_addr = (IpAddr::V6(Ipv6Addr::LOCALHOST), env::var("PORT").expect("PORT must be set").parse::<u16>().unwrap());
|
|
|
|
// JSON transport is provided by the json_transport tarpc module. It makes it easy
|
|
// to start up a serde-powered json serialization strategy over TCP.
|
|
let mut listener = tarpc::serde_transport::tcp::listen(&server_addr, Json::default).await?;
|
|
info!("Listening on port {}", listener.local_addr().port());
|
|
listener.config_mut().max_frame_length(usize::MAX);
|
|
listener
|
|
// Ignore accept errors.
|
|
.filter_map(|r| future::ready(r.ok()))
|
|
.map(BaseChannel::with_defaults)
|
|
// Limit channels to 1 per IP.
|
|
.max_channels_per_key(1, |t| t.transport().peer_addr().unwrap().ip())
|
|
// serve is generated by the service attribute. It takes as input any type implementing
|
|
// the generated World trait.
|
|
.map(|channel| {
|
|
let server = RealmAuthServer::new(channel.transport().peer_addr().unwrap(), db_pool.clone(), auth_email.clone());
|
|
channel.execute(server.serve()).for_each(spawn)
|
|
})
|
|
// Max 10 channels.
|
|
.buffer_unordered(10)
|
|
.for_each(|_| async {})
|
|
.await;
|
|
|
|
Ok(())
|
|
} |