Added proper parsing of Gandi API Responses.

This also makes it possible to output prettier logs.
This commit is contained in:
jannikac 2023-02-04 15:03:25 +01:00
parent 36f6f6600e
commit 0d1e8fe1ae

View file

@ -8,7 +8,7 @@ use ip_source::seeip::IPSourceSeeIP;
use opts::Opts; use opts::Opts;
use reqwest::header::InvalidHeaderValue; use reqwest::header::InvalidHeaderValue;
use reqwest::{header, Client, ClientBuilder, StatusCode}; use reqwest::{header, Client, ClientBuilder, StatusCode};
use serde::Serialize; use serde::{Deserialize, Serialize};
use std::{num::NonZeroU32, sync::Arc, time::Duration}; use std::{num::NonZeroU32, sync::Arc, time::Duration};
use tokio::join; use tokio::join;
use tokio::{self, task::JoinHandle, time::sleep}; use tokio::{self, task::JoinHandle, time::sleep};
@ -74,6 +74,21 @@ pub struct APIPayload {
pub rrset_ttl: u32, pub rrset_ttl: u32,
} }
#[derive(Debug)]
struct ResponseFeedback {
entry_name: String,
entry_type: String,
response: Result<String, ApiError>,
}
#[derive(Deserialize)]
struct ApiResponse {
message: String,
cause: Option<String>,
code: Option<i32>,
object: Option<String>,
}
async fn run( async fn run(
base_url: &str, base_url: &str,
ip_source: &Box<dyn IPSource>, ip_source: &Box<dyn IPSource>,
@ -112,7 +127,7 @@ async fn run(
if !ipv4_same || !ipv6_same || conf.always_update { if !ipv4_same || !ipv6_same || conf.always_update {
let client = api_client(&conf.api_key)?; let client = api_client(&conf.api_key)?;
let mut tasks: Vec<JoinHandle<(StatusCode, String)>> = Vec::new(); let mut tasks: Vec<JoinHandle<Result<ResponseFeedback, ClientError>>> = Vec::new();
println!("Attempting to update DNS entries now"); println!("Attempting to update DNS entries now");
let governor = Arc::new(governor::RateLimiter::direct(governor::Quota::per_minute( let governor = Arc::new(governor::RateLimiter::direct(governor::Quota::per_minute(
@ -155,28 +170,82 @@ async fn run(
let req = client.put(url).json(&payload); let req = client.put(url).json(&payload);
let task_governor = governor.clone(); let task_governor = governor.clone();
let entry_type = entry_type.to_string(); let entry_type = entry_type.to_string();
let task = tokio::task::spawn(async move { let entry_name = entry.name.to_string();
task_governor.until_ready_with_jitter(retry_jitter).await;
println!("Updating {} record for {}", entry_type, &fqdn); let task: JoinHandle<Result<ResponseFeedback, ClientError>> =
match req.send().await { tokio::task::spawn(async move {
Ok(response) => ( task_governor.until_ready_with_jitter(retry_jitter).await;
response.status(), println!("Updating {} record for {}", entry_type, &fqdn);
response
.text() let resp = req.send().await?;
.await
.unwrap_or_else(|error| error.to_string()), let response_feedback = match resp.status() {
), StatusCode::CREATED => {
Err(error) => (StatusCode::IM_A_TEAPOT, error.to_string()), let body: ApiResponse = resp.json().await?;
} ResponseFeedback {
}); entry_name,
entry_type,
response: Ok(body.message),
}
}
StatusCode::UNAUTHORIZED => ResponseFeedback {
entry_name,
entry_type,
response: Err(ApiError::Unauthorized()),
},
StatusCode::FORBIDDEN => {
let body: ApiResponse = resp.json().await?;
ResponseFeedback {
entry_name: entry_name.clone(),
entry_type,
response: Err(ApiError::Forbidden {
message: body.message,
}),
}
}
_ => {
let status = resp.status();
let body: ApiResponse = resp.json().await?;
ResponseFeedback {
entry_name,
entry_type,
response: Err(ApiError::Unknown(status, body.message)),
}
}
};
Ok(response_feedback)
});
tasks.push(task); tasks.push(task);
} }
} }
let results = futures::future::try_join_all(tasks).await?; let results = futures::future::try_join_all(tasks).await?;
println!("Updates done for {} entries", results.len()); // Only count successfull requests
for (status, body) in results { println!(
println!("{} - {}", status, body); "Updates done for {} entries",
results
.iter()
.filter_map(|item| item.as_ref().ok())
.filter(|item| item.response.is_ok())
.count()
);
for item in results {
match item {
Ok(value) => println!(
"{}",
match value.response {
Ok(val) => format!(
"Record '{}' ({}): {}",
value.entry_name, value.entry_type, val
),
Err(err) => format!(
"Record '{}' ({}): {}",
value.entry_name, value.entry_type, err
),
}
),
Err(err) => println!("{}", err),
}
} }
} else { } else {
println!("IP address has not changed since last update"); println!("IP address has not changed since last update");