Commit b758c875 authored by bksp's avatar bksp
Browse files

change architecture

parent 13a16342
pub mod providers;
pub mod routes;
// enum Providers {
// Ldap(LdapProvider),
// Local(LocalProvider),
// }
// struct LdapProvider {
// user_filter: String,
// group_filter: String,
// // ...
// }
// struct LocalProvider;
...@@ -4,62 +4,23 @@ ...@@ -4,62 +4,23 @@
#[macro_use] extern crate rocket_contrib; #[macro_use] extern crate rocket_contrib;
#[macro_use] extern crate diesel; #[macro_use] extern crate diesel;
use rocket::State; use trust_dns_client::client::SyncClient;
use rocket::http::Status;
use rocket_contrib::json::Json;
use trust_dns_client::client::{Client, SyncClient};
use trust_dns_client::tcp::TcpClientConnection; use trust_dns_client::tcp::TcpClientConnection;
use trust_dns_client::op::{DnsResponse, ResponseCode};
use trust_dns_client::rr::{DNSClass, Name, RecordType};
mod models; mod models;
mod config; mod config;
mod auth;
mod schema; mod schema;
mod routes;
use models::errors::ErrorResponse; use routes::users::*;
use models::users::UserInfo; use routes::zones::*;
use auth::routes::*;
#[database("db")] #[database("db")]
pub struct DbConn(diesel::SqliteConnection); pub struct DbConn(diesel::SqliteConnection);
type DnsClient = SyncClient<TcpClientConnection>;
#[get("/zones/<zone>/records")]
fn get_zone_records(
client: State<SyncClient<TcpClientConnection>>,
_user_info: UserInfo,
zone: String
) -> Result<Json<Vec<models::dns::Record>>, ErrorResponse<()>> {
// TODO: Implement FromParam for Name
let name = Name::from_utf8(&zone).unwrap();
let response: DnsResponse = client.query(&name, DNSClass::IN, RecordType::AXFR).unwrap();
if response.response_code() != ResponseCode::NoError {
return ErrorResponse::new(
Status::NotFound,
format!("zone {} could not be found", name.to_utf8())
).err()
}
let answers = response.answers();
let mut records: Vec<_> = answers.to_vec().into_iter()
.map(|record| models::dns::Record::from(record))
.filter(|record| match record.rdata {
models::dns::RData::NULL { .. } | models::dns::RData::DNSSEC(_) => false,
_ => true,
}).collect();
// AXFR response ends with SOA, we remove it so it is not doubled in the response.
records.pop();
Ok(Json(records))
}
#[launch] #[launch]
fn rocket() -> rocket::Rocket { fn rocket() -> rocket::Rocket {
......
pub mod users;
pub mod zones;
use rocket::State;
use rocket::http::Status;
use rocket_contrib::json::Json;
use trust_dns_client::client::{Client};
use trust_dns_client::op::{DnsResponse, ResponseCode};
use trust_dns_client::rr::{DNSClass, Name, RecordType};
use crate::models::dns;
use crate::models::errors::ErrorResponse;
use crate::models::users::UserInfo;
use crate::DnsClient;
#[get("/zones/<zone>/records")]
pub fn get_zone_records(
client: State<DnsClient>,
_user_info: UserInfo,
zone: String
) -> Result<Json<Vec<dns::Record>>, ErrorResponse<()>> {
// TODO: Implement FromParam for Name
let name = Name::from_utf8(&zone).unwrap();
let response: DnsResponse = client.query(&name, DNSClass::IN, RecordType::AXFR).unwrap();
if response.response_code() != ResponseCode::NoError {
return ErrorResponse::new(
Status::NotFound,
format!("zone {} could not be found", name.to_utf8())
).err()
}
let answers = response.answers();
let mut records: Vec<_> = answers.to_vec().into_iter()
.map(|record| dns::Record::from(record))
.filter(|record| match record.rdata {
dns::RData::NULL { .. } | dns::RData::DNSSEC(_) => false,
_ => true,
}).collect();
// AXFR response ends with SOA, we remove it so it is not doubled in the response.
records.pop();
Ok(Json(records))
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment