Implemented request to update the records.

This commit is contained in:
shad0wflame 2021-12-22 16:19:59 +01:00
parent 218dbd3770
commit e20d7b0d86
2 changed files with 127 additions and 24 deletions

View File

@ -1,5 +1,52 @@
use std::fs::read_to_string;
use std::path::Path;
use serde::{Deserialize, Serialize};
use crate::ip_handler::get_ip_to_publish;
const RECORDS_FILE_NAME: &'static str = "records.json";
#[derive(Debug, Deserialize)]
struct DNSRecordsHolder {
records: Vec<DNSRecord>,
}
#[derive(Debug, Serialize, Deserialize)]
struct DNSRecord {
name: String,
record_type: String,
data: Option<String>,
ttl: u32,
}
#[derive(Debug, Serialize, Deserialize)]
struct DNSRecordCreateTypeName {
data: String,
#[serde(skip_serializing_if = "Option::is_none")]
#[serde(default)]
port: Option<u16>, // SRV Only.
#[serde(skip_serializing_if = "Option::is_none")]
#[serde(default)]
priority: Option<u32>, // MX and SRV only.
#[serde(skip_serializing_if = "Option::is_none")]
#[serde(default)]
protocol: Option<String>, // SRV only.
#[serde(skip_serializing_if = "Option::is_none")]
#[serde(default)]
service: Option<String>, // SRV only.
ttl: u32,
#[serde(skip_serializing_if = "Option::is_none")]
#[serde(default)]
weight: Option<u32>, // SRV only.
}
/// Updates the DNS records if the IP has changed and returns the result of the execution.
///
/// # Arguments
@ -12,19 +59,78 @@ use crate::ip_handler::get_ip_to_publish;
pub async fn exec(domain: &str, key: &str, secret: &str) -> Result<(), Box<dyn std::error::Error>> {
let new_ip = get_ip_to_publish().await;
/// There's no need to do anything here. So we stop the execution.
// There's no need to do anything here. So we stop the execution.
if Option::is_none(&new_ip) {
return Ok(());
}
// TODO: Create a struct representing the structure of
// a JSON file including the DNS Records we want to update.
let records = get_records();
// TODO: Create that JSON file.
// TODO: Read that JSON file and deserialize it with serde.
// TODO: Update the DNS Records.
for record in records {
update_record(&record, &new_ip.clone().unwrap(), domain, key, secret).await;
}
Ok(())
}
/// Gets a vector of DNSRecord from RECORDS_FILE_NAME and returns it.
fn get_records() -> Vec<DNSRecord> {
let path = Path::new(RECORDS_FILE_NAME);
let content = read_to_string(path).unwrap();
let base: DNSRecordsHolder =
serde_json::from_str(&content).expect("Failed to deserialize JSON");
base.records
}
/// Sends a put request to the GoDaddy API to update a DNS record.
///
/// # Arguments
///
/// * `record` - A &DNSRecord holding the record to update.
///
/// * `value` - A &str holding the current WAN ip.
///
/// * `domain` - A &str holding the domain to update.
///
/// * `key` - A &str holding the GoDaddy developer key.
///
/// * `secret` - A &str holding the GoDaddy developer secret.
async fn update_record(record: &DNSRecord, value: &str, domain: &str, key: &str, secret: &str) -> () {
let url = format!(
"https://api.godaddy.com/v1/domains/{domain}/records/{record_type}/{name}",
domain = domain,
record_type = record.record_type,
name = record.name
);
let data = match &record.data {
Some(x) => String::from(x),
None => String::from(value),
};
let body = vec![DNSRecordCreateTypeName {
data,
port: None,
priority: None,
protocol: None,
service: None,
ttl: record.ttl,
weight: None,
}];
let header = format!("sso-key {}:{}", key, secret);
let client = reqwest::Client::new();
let req = client
.put(url)
.json(&body)
.header("accept", "application/json")
.header("content-type", "application/json")
.header("authorization", &header);
req.send().await.expect("Error updating record.");
}

View File

@ -1,9 +1,9 @@
use serde::Deserialize;
use std::fs::File;
use std::io::{Read, Write};
use std::fs::{read_to_string, File};
use std::io::Write;
use std::path::Path;
const FILE_NAME: &'static str = "ddns_ip";
const IP_FILE_NAME: &'static str = "ddns_ip";
const WEBSITE_URL: &'static str = "https://httpbin.org/ip";
#[derive(Deserialize)]
@ -13,16 +13,14 @@ struct IP {
/// Returns an Option holding the current IP address if the value has changed, otherwise returns None.
pub async fn get_ip_to_publish() -> Option<String> {
let current_ip = check_current_ip()
.await
.expect("Error getting the current IP.");
let previous_ip = match check_previous_ip() {
Some(x) => x,
None => String::new(),
};
println!("Current IP: {}, Previous IP: {}", current_ip, previous_ip);
let current_ip = check_current_ip()
.await
.expect("Error getting the current IP.");
if current_ip.eq(&previous_ip) {
return None;
@ -47,22 +45,21 @@ async fn check_current_ip() -> Result<String, Box<dyn std::error::Error>> {
/// # Arguments
///
/// * `current_ip` - A &str holding the current IP value.
fn record_current_ip(current_ip: &str) {
let mut file = File::create(FILE_NAME).expect("Error creating file.");
fn record_current_ip(current_ip: &str) -> () {
let mut file = File::create(IP_FILE_NAME).expect("Error creating file.");
file.write_all(current_ip.as_ref())
.expect("Error writing file.");
}
/// Reads the current IP value from the FILE_NAME and returns it.
fn check_previous_ip() -> Option<String> {
if !Path::new(FILE_NAME).exists() {
let path = Path::new(IP_FILE_NAME);
if !path.exists() {
return None;
}
let mut file = File::open(FILE_NAME).ok()?;
let mut contents = String::new();
file.read_to_string(&mut contents)
.expect("Error reading file.");
let contents = read_to_string(path).expect("Error reading file.");
return Some(contents);
Some(contents)
}