feat: Add helper functions for network requests and caching

This commit is contained in:
Rostislav Raykov 2019-09-17 17:33:35 +02:00
parent d080f1b364
commit f01cba4dc5
2 changed files with 108 additions and 0 deletions

19
src/file.rs Normal file
View file

@ -0,0 +1,19 @@
use failure::Error;
use std::fs::File;
use std::path::PathBuf;
use std::io::prelude::*;
pub fn read_cache_file(path: &PathBuf) -> Result<String, Error> {
let mut file = File::open(&path)?;
let mut s = String::new();
file.read_to_string(&mut s)?;
Ok(s.clone())
}
pub fn write_cache_file(path: &PathBuf, ip: &str) -> Result<(), Error> {
let mut file = File::create(&path)?;
file.write_all(ip.as_bytes())?;
Ok(())
}

89
src/network.rs Normal file
View file

@ -0,0 +1,89 @@
use serde::{Serialize, Deserialize};
use failure::{Error, format_err};
#[derive(Deserialize, Debug)]
struct ListResponse {
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: ListResponse = 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| 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: ListResponse = 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| 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(),
};
client
.put(&url)
.header("X-Auth-Email", email)
.header("X-Auth-Key", key)
.header("Content-Type", "application/json")
.json(&update_data)
.send()?;
Ok(())
}