cloudflare-ddns-service/src/network.rs

95 lines
2.8 KiB
Rust

use serde::{Serialize, Deserialize};
use failure::{Error, format_err};
#[derive(Deserialize, Debug)]
struct CloudflareResponse {
success: bool,
errors: Vec<String>,
result: Vec<ObjectWithId>,
}
#[derive(Deserialize, Debug, PartialEq)]
struct ObjectWithId {
id: String,
}
#[derive(Serialize, Debug)]
struct UpdateIpData {
id: String,
r#type: String,
name: String,
content: String,
}
pub fn get_zone_identifier(zone: &str, email: &str, key: &str) -> Result<String, Error> {
let client = reqwest::Client::new();
let url = format!("https://api.cloudflare.com/client/v4/zones?name={}", zone);
let response: CloudflareResponse = client
.get(&url)
.header("X-Auth-Email", email)
.header("X-Auth-Key", key)
.header("Content-Type", "application/json")
.send()?
.json()?;
if !response.success {
let err: String = response.errors.iter().map(|s| format!("{}\n", s.to_owned())).collect();
return Err(format_err!("API Error: {}", err));
}
Ok(response.result[0].id.clone())
}
pub fn get_dns_record_id(zone_id: &str, domain: &str, email: &str, key: &str) -> Result<String, Error> {
let client = reqwest::Client::new();
let url = format!("https://api.cloudflare.com/client/v4/zones/{}/dns_records?name={}", zone_id, domain);
let response: CloudflareResponse = client
.get(&url)
.header("X-Auth-Email", email)
.header("X-Auth-Key", key)
.header("Content-Type", "application/json")
.send()?
.json()?;
if !response.success {
let err: String = response.errors.iter().map(|s| format!("{}\n", s.to_owned())).collect();
return Err(format_err!("API Error: {}", err));
}
Ok(response.result[0].id.clone())
}
pub fn get_current_ip() -> Result<String, Error> {
Ok(reqwest::Client::new()
.get("http://ipv4.icanhazip.com")
.send()?
.text()?)
}
pub fn update_ddns(ip: &str, domain: &str, zone_id: &str, record_id: &str, email: &str, key: &str) -> Result<(), Error> {
let client = reqwest::Client::new();
let url = format!("https://api.cloudflare.com/client/v4/zones/{}/dns_records/{}", zone_id, record_id);
let update_data = UpdateIpData {
id: zone_id.to_owned(),
r#type: "A".to_owned(),
name: domain.to_owned(),
content: ip.to_owned(),
};
let response: CloudflareResponse = client
.put(&url)
.header("X-Auth-Email", email)
.header("X-Auth-Key", key)
.header("Content-Type", "application/json")
.json(&update_data)
.send()?
.json()?;
if !response.success {
let err: String = response.errors.iter().map(|s| format!("{}\n", s.to_owned())).collect();
return Err(format_err!("Unsuccessful update of DNS record: {}", err));
}
Ok(())
}