X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=src%2Fmain.rs;h=6665155fb5afa94d17c3c00bbbaad89d6774ec3a;hb=HEAD;hp=496fd986e178e0b3e9dcd520028ece98692c80e4;hpb=99c5a6de3ef69a61dcff181bb4b11fc4141e3c81;p=gandi_dns_update.git diff --git a/src/main.rs b/src/main.rs index 496fd98..6665155 100644 --- a/src/main.rs +++ b/src/main.rs @@ -9,28 +9,40 @@ #![cfg_attr(debug_assertions, allow(unused_variables, unused_imports, dead_code))] -use std::{ net::{ IpAddr, Ipv4Addr }, thread, time }; -use serde_json::{ Value, json }; +use std::{ + net::{IpAddr, Ipv4Addr}, + thread, time, +}; -mod error; -mod config; +use serde_json::{json, Value}; + +use crate::{ + config::Config, + error::{Error, Result}, +}; -use crate::error::{ Result, Error }; -use crate::config::Config; +mod config; +mod error; const FILE_CONF: &str = "config.ron"; fn main() -> Result<()> { - println!("=== GANDI LiveDNS updater ==="); + println!("GANDI LiveDNS updater"); let config = Config::read(FILE_CONF)?; - println!("Configuration: {:?}", Config { api_key: String::from("*****"), ..config.clone() }); + println!( + "Configuration: {:?}", + Config { + api_key: String::from("*****"), + ..config.clone() + } + ); loop { let time_beginning_loop = time::Instant::now(); - if let Err(err) = check_and_update_dns(&config.api_key, &config.fqdn, &config.domains, config.ttl) { + if let Err(err) = check_and_update_dns(&config.api_key, &config.domains, config.ttl) { println!("!! {}", err); } @@ -43,16 +55,19 @@ fn main() -> Result<()> { } } -fn check_and_update_dns(api_key: &str, fqdn: &str, domains: &Vec, ttl: i32) -> Result<()> { +fn check_and_update_dns(api_key: &str, domains: &Vec<(String, String)>, ttl: i32) -> Result<()> { let real_ip = get_real_ip()?; - for domain in domains { - let current_ip = get_current_record_ip(api_key, domain, fqdn)?; + for (hostname, domain) in domains { + let current_ip = get_current_record_ip(api_key, hostname, domain)?; if real_ip != current_ip { - println!("IP addresses don't match for domain {}: real = {}, dns = {}. Renewing DNS...", domain, real_ip, current_ip); - update_record_ip(api_key, domain, fqdn, real_ip, ttl)?; - println!("Renewing of {} successfully", domain); + println!( + "IP addresses don't match for domain {}: real = {}, dns = {}. Renewing DNS...", + domain, real_ip, current_ip + ); + update_record_ip(api_key, hostname, domain, real_ip, ttl)?; + println!("Renewing of {}.{} successfully", hostname, domain); } } @@ -64,44 +79,58 @@ fn get_real_ip() -> Result { let client = reqwest::blocking::Client::new(); match client.get(url).send() { - Ok(resp) => + Ok(resp) => { if resp.status().is_success() { let content = resp.text().unwrap(); match content.parse() { Ok(IpAddr::V4(ip_v4)) => Ok(ip_v4), - _ => Err(Box::new(Error { message: String::from("Can't parse IPv4 from ipify") })) + _ => Err(Box::new(Error { + message: String::from("Can't parse IPv4 from ipify"), + })), } } else { - Err(Box::new(Error { message: format!("Request unsuccessful to {}: {:#?}", url, resp) })) - }, - - Err(error) => { - Err(Box::new(Error { message: format!("Error during request to {}: {:?}", url, error) })) + Err(Box::new(Error { + message: format!("Request unsuccessful to {}: {:#?}", url, resp), + })) + } } + Err(error) => Err(Box::new(Error { + message: format!("Error during request to {}: {:?}", url, error), + })), } } fn get_current_record_ip(api_key: &str, name: &str, fqdn: &str) -> Result { - let json_value = request_livedns_gandi(api_key, &format!("domains/{}/records/{}/A", fqdn, name), Method::Get)?; + let json_value = request_livedns_gandi( + api_key, + &format!("domains/{}/records/{}/A", fqdn, name), + Method::Get, + )?; match &json_value["rrset_values"][0] { - Value::String(ip_str) => - Ok(ip_str.parse()?), - _ => - Result::Err(Box::new(Error { message: format!("Unable to extract the IP from the JSON answer: {}", json_value) })) + Value::String(ip_str) => Ok(ip_str.parse()?), + _ => Result::Err(Box::new(Error { + message: format!( + "Unable to extract the IP from the JSON answer: {}", + json_value + ), + })), } } fn update_record_ip(api_key: &str, name: &str, fqdn: &str, ip: Ipv4Addr, ttl: i32) -> Result<()> { - let json_body = - json!( - { - "rrset_values": [ format!("{}", ip) ], - "rrset_ttl": ttl - } - ); + let json_body = json!( + { + "rrset_values": [ format!("{}", ip) ], + "rrset_ttl": ttl + } + ); - let json_value = request_livedns_gandi(api_key, &format!("domains/{}/records/{}/A", fqdn, name), Method::Put(json_body.to_string()))?; + let json_value = request_livedns_gandi( + api_key, + &format!("domains/{}/records/{}/A", fqdn, name), + Method::Put(json_body.to_string()), + )?; println!("Update response: {}", json_value); @@ -110,28 +139,34 @@ fn update_record_ip(api_key: &str, name: &str, fqdn: &str, ip: Ipv4Addr, ttl: i3 enum Method { Put(String), - Get + Get, } fn request_livedns_gandi(api_key: &str, url_fragment: &str, method: Method) -> Result { let url = format!("https://api.gandi.net/v5/livedns/{}", url_fragment); let client = reqwest::blocking::Client::new(); - let request_builder = - match method { - Method::Put(body) => client.put(&url).body(body), - Method::Get => client.get(&url) - }; + let request_builder = match method { + Method::Put(body) => client.put(&url).body(body), + Method::Get => client.get(&url), + }; - match request_builder.header("Authorization", format!("Apikey {}", api_key)).send() { - Ok(resp) => + match request_builder + .header("Authorization", format!("Apikey {}", api_key)) + .send() + { + Ok(resp) => { if resp.status().is_success() { let content = resp.text().unwrap(); Ok(serde_json::from_str(&content).unwrap()) } else { - Err(Box::new(Error { message: format!("Request unsuccessful to {}: {:#?}", &url, resp) })) - }, - Err(error) => - Err(Box::new(Error { message: format!("Error during request to {}: {:?}", &url, error) })) + Err(Box::new(Error { + message: format!("Request unsuccessful to {}: {:#?}", &url, resp), + })) + } + } + Err(error) => Err(Box::new(Error { + message: format!("Error during request to {}: {:?}", &url, error), + })), } }