1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
use axum::extract::Extension;
use axum::{
routing::{get, post, put},
Router, Server,
};
use dotenvy::dotenv;
use log::info;
use sqlx::postgres::PgPoolOptions;
use std::env;
use std::net::SocketAddr;
use std::sync::Arc;
use tower::ServiceBuilder;
use tower_http::trace::TraceLayer;
mod db;
mod extractors;
mod features;
mod routes;
#[tokio::main]
async fn main() {
// Load configuration from .env
dotenv().ok();
if std::env::args().nth(1) == Some("--version".to_string()) {
println!(
"{}",
option_env!("CARGO_PKG_VERSION").unwrap_or_else(|| "unknown")
);
return;
}
// Set logging levels if not already set
if env::var_os("RUST_LOG").is_none() {
env::set_var("RUST_LOG", "fdns_api=debug,tower_http=debug");
}
// Initialize tracing with previously set logging levels
tracing_subscriber::fmt::init();
// Connect to Postgres
let pg_pool = Arc::new(
PgPoolOptions::new()
.max_connections(5)
.connect(&env::var("DATABASE_URL").unwrap())
.await
.unwrap(),
);
info!("Postgres pool initialized");
let app = Router::new()
.nest(
"/api",
Router::new().nest(
"/v1",
Router::new()
.route("/features", get(routes::v1::features::get_features))
.nest(
"/users",
Router::new()
.route("/", post(routes::v1::users::create_user))
.route("/", get(routes::v1::users::get_all_users))
.route("/totp", get(routes::v1::users::needs_totp))
.route("/login", post(routes::v1::users::login))
.route("/whoami", get(routes::v1::users::whoami)),
)
.nest(
"/zones",
Router::new()
.route("/", get(routes::v1::zones::list_zones))
.route(
"/:zone_id",
get(routes::v1::records::get_records)
.post(routes::v1::zones::create_zone)
.put(routes::v1::records::create_record),
)
.route(
"/:zone_id/:record_id",
put(routes::v1::records::update_record)
.delete(routes::v1::records::delete_record),
),
),
),
)
.layer(
ServiceBuilder::new()
.layer(TraceLayer::new_for_http())
.layer(Extension(pg_pool)),
);
let addr = SocketAddr::from(([0, 0, 0, 0], 8000));
info!("Binding to {addr}");
Server::bind(&addr)
.serve(app.into_make_service_with_connect_info::<SocketAddr>())
.await
.unwrap();
}
|
