kaan/gandi-live-dns-rust
1
0
Fork 0
mirror of https://github.com/SeriousBug/gandi-live-dns-rust synced 2025-02-05 17:39:57 -06:00
Code Issues Projects Releases Wiki Activity

Add repeat flag (#67)

Browse source 
* Add a config option for the repeat flag

* Implement repeat flag
This commit is contained in:
jannikac 2022-12-18 06:53:34 +01:00 committed by GitHub
parent 28984e1b52
commit bbd7ce347a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 23 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/main.rs
View file

@ -8,6 +8,7 @@ use futures;
use ip_source::icanhazip::IPSourceIcanhazip;
use reqwest::{header, Client, ClientBuilder, StatusCode};
use serde::Serialize;
use std::thread;
use std::{num::NonZeroU32, sync::Arc, time::Duration};
use tokio::{self, task::JoinHandle};
mod config;
@ -42,7 +43,7 @@ pub struct APIPayload {
pub rrset_ttl: u32,
}
async fn run<IP: IPSource>(base_url: &str, conf: Config) -> anyhow::Result<()> {
async fn run<IP: IPSource>(base_url: &str, conf: &Config) -> anyhow::Result<()> {
config::validate_config(&conf).die_with(|error| format!("Invalid config: {}", error));
println!("Finding out the IP address...");
let ipv4_result = IP::get_ipv4().await;
@ -124,6 +125,21 @@ async fn main() -> anyhow::Result<()> {
let conf = config::load_config(&opts)
.die_with(|error| format!("Failed to read config file: {}", error));
// run indefinitely if repeat is given
if let Some(delay) = opts.repeat {
loop {
run_dispatch(&conf).await.ok();
thread::sleep(Duration::from_secs(delay))
}
}
// otherwise run just once
else {
run_dispatch(&conf).await?;
Ok(())
}
}
async fn run_dispatch(conf: &Config) -> anyhow::Result<()> {
match conf.ip_source {
IPSourceName::Ipify => run::<IPSourceIpify>("https://api.gandi.net", conf).await,
IPSourceName::Icanhazip => run::<IPSourceIcanhazip>("https://api.gandi.net", conf).await,
@ -182,7 +198,7 @@ mod tests {
..Opts::default()
};
let conf = config::load_config(&opts).expect("Failed to load config");
run::<IPSourceMock>(server.base_url().as_str(), conf)
run::<IPSourceMock>(server.base_url().as_str(), &conf)
.await
.expect("Failed when running the update");

5
src/opts.rs
View file

@ -17,4 +17,9 @@ pub struct Opts {
/// If enabled, any IPv6 (AAAA) records in the configuration file are ignored.
#[clap(action, long)]
pub skip_ipv6: bool,
/// Repeat after specified delay
///
/// If enabled waits for the given delay between updating DNS records
#[clap(long)]
pub repeat: Option<u64>,
}