D53/src/provider/gandi.rs

103 lines
2.7 KiB
Rust

use std::net::{Ipv4Addr, Ipv6Addr};
use anyhow::{anyhow, Result};
use async_trait::async_trait;
use reqwest::header;
use serde::Serialize;
use tracing::{info, warn};
use crate::provider::DnsProvider;
use crate::Opt;
pub struct GandiProvider {
client: reqwest::Client,
}
impl GandiProvider {
pub fn new(opts: &Opt) -> Result<Self> {
let api_key = opts
.gandi_api_key
.clone()
.ok_or_else(|| anyhow!("Must specify D53_GANDI_API_KEY"))?;
let mut headers = header::HeaderMap::new();
let mut auth_value = header::HeaderValue::from_str(&format!("Apikey {}", api_key))?;
auth_value.set_sensitive(true);
headers.insert(header::AUTHORIZATION, auth_value);
let client = reqwest::Client::builder()
.default_headers(headers)
.use_rustls_tls()
.build()?;
Ok(Self { client })
}
async fn put_rrset(&self, url: &str, rrset: &GandiRrset) -> Result<()> {
info!("PUT {} with {:?}", url, rrset);
let http = self.client.put(url).json(rrset).send().await?;
if !http.status().is_success() {
warn!("PUT {} returned {}", url, http.status());
}
http.error_for_status()?;
Ok(())
}
}
#[async_trait]
impl DnsProvider for GandiProvider {
fn provider(&self) -> &'static str {
"gandi"
}
async fn update_a(&self, domain: &str, subdomain: &str, targets: &[Ipv4Addr]) -> Result<()> {
let url = format!(
"https://api.gandi.net/v5/livedns/domains/{}/records/{}/A",
domain, subdomain
);
let rrset = GandiRrset {
rrset_values: targets.iter().map(ToString::to_string).collect::<Vec<_>>(),
rrset_ttl: 300,
};
self.put_rrset(&url, &rrset).await
}
async fn update_aaaa(&self, domain: &str, subdomain: &str, targets: &[Ipv6Addr]) -> Result<()> {
let url = format!(
"https://api.gandi.net/v5/livedns/domains/{}/records/{}/AAAA",
domain, subdomain
);
let rrset = GandiRrset {
rrset_values: targets.iter().map(ToString::to_string).collect::<Vec<_>>(),
rrset_ttl: 300,
};
self.put_rrset(&url, &rrset).await
}
async fn update_cname(&self, domain: &str, subdomain: &str, target: &str) -> Result<()> {
let url = format!(
"https://api.gandi.net/v5/livedns/domains/{}/records/{}/CNAME",
domain, subdomain
);
let rrset = GandiRrset {
rrset_values: vec![target.to_string()],
rrset_ttl: 300,
};
self.put_rrset(&url, &rrset).await
}
}
#[derive(Serialize, Debug)]
struct GandiRrset {
rrset_values: Vec<String>,
rrset_ttl: u32,
}