2021-12-19 16:43:13 +00:00
|
|
|
mod api;
|
2021-07-12 20:14:39 +00:00
|
|
|
mod auth;
|
2021-08-06 02:06:50 +00:00
|
|
|
mod db;
|
2021-07-12 20:14:39 +00:00
|
|
|
mod env;
|
|
|
|
|
mod rpc;
|
|
|
|
|
|
2021-10-04 22:06:12 +00:00
|
|
|
use ::rpc::Peer;
|
2022-04-22 19:33:19 +00:00
|
|
|
use anyhow::Result;
|
2021-07-12 20:14:39 +00:00
|
|
|
use async_trait::async_trait;
|
2022-02-17 16:04:04 +00:00
|
|
|
use db::{Db, PostgresDb};
|
2022-04-23 00:46:31 +00:00
|
|
|
use hyper::{
|
|
|
|
|
server::conn::AddrStream,
|
|
|
|
|
service::{make_service_fn, service_fn},
|
|
|
|
|
Body, Request, Response, Server,
|
|
|
|
|
};
|
2022-04-21 14:57:49 +00:00
|
|
|
use serde::Deserialize;
|
2022-04-23 00:46:31 +00:00
|
|
|
use std::{convert::Infallible, net::TcpListener, sync::Arc};
|
2021-07-12 20:14:39 +00:00
|
|
|
|
2022-04-22 19:33:19 +00:00
|
|
|
// type Request = tide::Request<Arc<AppState>>;
|
2021-07-12 20:14:39 +00:00
|
|
|
|
|
|
|
|
#[derive(Default, Deserialize)]
|
|
|
|
|
pub struct Config {
|
|
|
|
|
pub http_port: u16,
|
|
|
|
|
pub database_url: String,
|
2021-12-19 16:43:13 +00:00
|
|
|
pub api_token: String,
|
2021-07-12 20:14:39 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub struct AppState {
|
2022-02-17 16:04:04 +00:00
|
|
|
db: Arc<dyn Db>,
|
2021-07-12 20:14:39 +00:00
|
|
|
config: Config,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl AppState {
|
2022-04-22 19:33:19 +00:00
|
|
|
async fn new(config: Config) -> Result<Arc<Self>> {
|
2022-02-17 16:04:04 +00:00
|
|
|
let db = PostgresDb::new(&config.database_url, 5).await?;
|
2021-07-12 20:14:39 +00:00
|
|
|
|
|
|
|
|
let this = Self {
|
2022-02-17 16:04:04 +00:00
|
|
|
db: Arc::new(db),
|
2021-07-12 20:14:39 +00:00
|
|
|
config,
|
|
|
|
|
};
|
|
|
|
|
Ok(Arc::new(this))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2022-04-22 19:33:19 +00:00
|
|
|
// #[async_trait]
|
|
|
|
|
// trait RequestExt {
|
|
|
|
|
// fn db(&self) -> &Arc<dyn Db>;
|
|
|
|
|
// }
|
2021-07-12 20:14:39 +00:00
|
|
|
|
2022-04-22 19:33:19 +00:00
|
|
|
// #[async_trait]
|
|
|
|
|
// impl RequestExt for Request {
|
|
|
|
|
// fn db(&self) -> &Arc<dyn Db> {
|
|
|
|
|
// &self.state().db
|
|
|
|
|
// }
|
|
|
|
|
// }
|
2021-07-12 20:14:39 +00:00
|
|
|
|
2022-04-22 19:33:19 +00:00
|
|
|
#[tokio::main]
|
|
|
|
|
async fn main() -> Result<()> {
|
2022-04-08 16:06:51 +00:00
|
|
|
if std::env::var("LOG_JSON").is_ok() {
|
|
|
|
|
json_env_logger::init();
|
|
|
|
|
} else {
|
2022-04-22 19:33:19 +00:00
|
|
|
env_logger::init();
|
2022-04-08 16:06:51 +00:00
|
|
|
}
|
2021-07-12 20:14:39 +00:00
|
|
|
|
|
|
|
|
if let Err(error) = env::load_dotenv() {
|
|
|
|
|
log::error!(
|
|
|
|
|
"error loading .env.toml (this is expected in production): {}",
|
|
|
|
|
error
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
let config = envy::from_env::<Config>().expect("error loading config");
|
|
|
|
|
let state = AppState::new(config).await?;
|
|
|
|
|
let rpc = Peer::new();
|
|
|
|
|
run_server(
|
|
|
|
|
state.clone(),
|
|
|
|
|
rpc,
|
2022-04-23 00:46:31 +00:00
|
|
|
TcpListener::bind(&format!("0.0.0.0:{}", state.config.http_port))
|
|
|
|
|
.expect("failed to bind TCP listener"),
|
2021-07-12 20:14:39 +00:00
|
|
|
)
|
|
|
|
|
.await?;
|
|
|
|
|
Ok(())
|
|
|
|
|
}
|
|
|
|
|
|
2022-04-22 19:33:19 +00:00
|
|
|
pub async fn run_server(state: Arc<AppState>, rpc: Arc<Peer>, listener: TcpListener) -> Result<()> {
|
2022-04-23 00:46:31 +00:00
|
|
|
let make_service = make_service_fn(|_: &AddrStream| async move {
|
|
|
|
|
Ok::<_, Infallible>(service_fn(|_: Request<Body>| async move {
|
|
|
|
|
Response::new(Body::from(format!("hello"))
|
|
|
|
|
}))
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
Server::from_tcp(listener)
|
|
|
|
|
.expect("could not create server")
|
|
|
|
|
.serve(make_service);
|
|
|
|
|
|
2022-04-22 19:33:19 +00:00
|
|
|
// let mut app = tide::with_state(state.clone());
|
|
|
|
|
// rpc::add_routes(&mut app, &rpc);
|
2022-04-21 15:30:08 +00:00
|
|
|
|
2022-04-22 19:33:19 +00:00
|
|
|
// let mut web = tide::with_state(state.clone());
|
|
|
|
|
// web.with(CompressMiddleware::new());
|
|
|
|
|
// api::add_routes(&mut web);
|
2021-09-29 17:15:19 +00:00
|
|
|
|
2022-04-22 19:33:19 +00:00
|
|
|
// app.at("/").nest(web);
|
2021-07-12 20:14:39 +00:00
|
|
|
|
2022-04-22 19:33:19 +00:00
|
|
|
// app.listen(listener).await?;
|
2021-07-12 20:14:39 +00:00
|
|
|
|
|
|
|
|
Ok(())
|
|
|
|
|
}
|
