Seperated API Key into different config file
This commit is contained in:
parent
ac2db79f21
commit
2433907dd6
3 changed files with 111 additions and 36 deletions
2
.gitignore
vendored
2
.gitignore
vendored
|
@ -3,4 +3,4 @@ target/
|
||||||
venv/
|
venv/
|
||||||
.idea/
|
.idea/
|
||||||
.vscode/
|
.vscode/
|
||||||
config.json
|
api-key.json
|
||||||
|
|
50
config.json
Normal file
50
config.json
Normal file
|
@ -0,0 +1,50 @@
|
||||||
|
{
|
||||||
|
"ipv6_interface": ":da5e:d3ff:feeb:4346",
|
||||||
|
"zones": [
|
||||||
|
{
|
||||||
|
"email": "neshura@proton.me",
|
||||||
|
"name": "neshura.net",
|
||||||
|
"id": "0183f167a051f1e432c0d931478638b5",
|
||||||
|
"dns_entries": [
|
||||||
|
{
|
||||||
|
"name": "*.neshura.net",
|
||||||
|
"type4": true,
|
||||||
|
"type6": true,
|
||||||
|
"interface": ":da5e:d3ff:feeb:4346"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "neshura.net",
|
||||||
|
"type4": true,
|
||||||
|
"type6": true,
|
||||||
|
"interface": ":da5e:d3ff:feeb:4346"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "test.neshura.net",
|
||||||
|
"type4": false,
|
||||||
|
"type6": true,
|
||||||
|
"interface": ":da5e:d3ff:feeb:4346"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "test2.neshura.net",
|
||||||
|
"type4": true,
|
||||||
|
"type6": false,
|
||||||
|
"interface": ":da5e:d3ff:feeb:4346"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"email": "neshura@proton.me",
|
||||||
|
"name": "neshura-server.net",
|
||||||
|
"id": "146d4cd6a1777376b423aaedc6824818",
|
||||||
|
"dns_entries": [
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"email": "neshura@proton.me",
|
||||||
|
"name": "neshweb.net",
|
||||||
|
"id": "75b0d52229357478b734ae0f6d075c15",
|
||||||
|
"dns_entries": [
|
||||||
|
]
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
77
src/main.rs
77
src/main.rs
|
@ -1,14 +1,18 @@
|
||||||
use serde_derive::{Deserialize, Serialize};
|
|
||||||
use reqwest::blocking::get;
|
use reqwest::blocking::get;
|
||||||
|
use serde_derive::{Deserialize, Serialize};
|
||||||
use std::{fs, process::exit};
|
use std::{fs, process::exit};
|
||||||
mod cloudflare;
|
mod cloudflare;
|
||||||
|
|
||||||
|
fn default_key() -> String {
|
||||||
|
return "".to_string();
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize)]
|
#[derive(Serialize, Deserialize)]
|
||||||
struct DnsEntry {
|
struct DnsEntry {
|
||||||
name: String,
|
name: String,
|
||||||
type4: bool,
|
type4: bool,
|
||||||
type6: bool,
|
type6: bool,
|
||||||
interface: Option<String>
|
interface: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize)]
|
#[derive(Serialize, Deserialize)]
|
||||||
|
@ -16,14 +20,15 @@ struct DnsZone {
|
||||||
email: String,
|
email: String,
|
||||||
name: String,
|
name: String,
|
||||||
id: String,
|
id: String,
|
||||||
dns_entries: Vec<DnsEntry>
|
dns_entries: Vec<DnsEntry>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize)]
|
#[derive(Serialize, Deserialize)]
|
||||||
struct DnsConfig {
|
struct DnsConfig {
|
||||||
ipv6_interface: String,
|
ipv6_interface: String,
|
||||||
|
#[serde(default = "default_key")]
|
||||||
api_key: String,
|
api_key: String,
|
||||||
zones: Vec<DnsZone>
|
zones: Vec<DnsZone>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl DnsConfig {
|
impl DnsConfig {
|
||||||
|
@ -31,32 +36,44 @@ impl DnsConfig {
|
||||||
DnsConfig {
|
DnsConfig {
|
||||||
ipv6_interface: "".to_string(),
|
ipv6_interface: "".to_string(),
|
||||||
api_key: "".to_string(),
|
api_key: "".to_string(),
|
||||||
zones: Vec::new() }
|
zones: Vec::new(),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn load(&mut self) {
|
fn load(&mut self) {
|
||||||
let file_contents = match fs::read_to_string("config.json") {
|
let file_contents = match fs::read_to_string("config.json") {
|
||||||
Ok(data) => {data},
|
Ok(data) => data,
|
||||||
Err(e) => panic!("{:#?}", e)
|
Err(e) => panic!("{:#?}", e),
|
||||||
};
|
};
|
||||||
let config_parse: DnsConfig = match serde_json::from_str(&file_contents) {
|
let config_parse: DnsConfig = match serde_json::from_str(&file_contents) {
|
||||||
Ok(data) => {data},
|
Ok(data) => data,
|
||||||
Err(e) => panic!("{:#?}", e)
|
Err(e) => panic!("{:#?}", e),
|
||||||
};
|
};
|
||||||
|
|
||||||
|
let key_file_contents = match fs::read_to_string("api-key.json") {
|
||||||
|
Ok(data) => data,
|
||||||
|
Err(e) => panic!("{:#?}", e),
|
||||||
|
};
|
||||||
|
let api_key_parse: serde_json::Value = match serde_json::from_str(&key_file_contents) {
|
||||||
|
Ok(data) => data,
|
||||||
|
Err(e) => panic!("{:#?}", e),
|
||||||
|
};
|
||||||
|
|
||||||
*self = config_parse;
|
*self = config_parse;
|
||||||
|
self.api_key = api_key_parse["api_key"].as_str().unwrap().to_owned();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
struct Ips {
|
struct Ips {
|
||||||
ipv4: String,
|
ipv4: String,
|
||||||
ipv6base: String
|
ipv6base: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Ips {
|
impl Ips {
|
||||||
fn new() -> Ips {
|
fn new() -> Ips {
|
||||||
Ips {
|
Ips {
|
||||||
ipv4: "0.0.0.0".to_string(),
|
ipv4: "0.0.0.0".to_string(),
|
||||||
ipv6base: ":".to_string()
|
ipv6base: ":".to_string(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -80,11 +97,11 @@ impl Ips {
|
||||||
Ok(data) => {
|
Ok(data) => {
|
||||||
let ipv6 = data.text().expect(":");
|
let ipv6 = data.text().expect(":");
|
||||||
let stripped = match ipv6.strip_suffix(ipv6_interface) {
|
let stripped = match ipv6.strip_suffix(ipv6_interface) {
|
||||||
Some(string) => {string.to_string()},
|
Some(string) => string.to_string(),
|
||||||
None => {":".to_string()}
|
None => ":".to_string(),
|
||||||
};
|
};
|
||||||
self.ipv6base = stripped;
|
self.ipv6base = stripped;
|
||||||
},
|
}
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
println!("Could not fetch IP6, quitting");
|
println!("Could not fetch IP6, quitting");
|
||||||
exit(75);
|
exit(75);
|
||||||
|
@ -94,10 +111,11 @@ impl Ips {
|
||||||
}
|
}
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
|
|
||||||
let mut config = DnsConfig::new();
|
let mut config = DnsConfig::new();
|
||||||
config.load();
|
config.load();
|
||||||
|
|
||||||
|
/* while (true) {} */
|
||||||
|
|
||||||
let mut ips = Ips::new();
|
let mut ips = Ips::new();
|
||||||
ips.get(&config.ipv6_interface);
|
ips.get(&config.ipv6_interface);
|
||||||
|
|
||||||
|
@ -116,38 +134,42 @@ fn main() {
|
||||||
|
|
||||||
agent.load_entries(&zone.email, &zone.id);
|
agent.load_entries(&zone.email, &zone.id);
|
||||||
|
|
||||||
|
|
||||||
for entry in zone.dns_entries {
|
for entry in zone.dns_entries {
|
||||||
let mut found4 = false;
|
let mut found4 = false;
|
||||||
let mut found6 = false;
|
let mut found6 = false;
|
||||||
|
|
||||||
for cloudflare_entry in &agent.dns_entries {
|
for cloudflare_entry in &agent.dns_entries {
|
||||||
|
|
||||||
if cloudflare_entry.is_equal(&entry.name) {
|
if cloudflare_entry.is_equal(&entry.name) {
|
||||||
if cloudflare_entry.r#type == cloudflare::CloudflareDnsType::A && entry.type4 {
|
if cloudflare_entry.r#type == cloudflare::CloudflareDnsType::A && entry.type4 {
|
||||||
found4 = true;
|
found4 = true;
|
||||||
if cloudflare_entry.is_ip_new(&ips.ipv4) {
|
if cloudflare_entry.is_ip_new(&ips.ipv4) {
|
||||||
let success = agent.update_entry(cloudflare_entry, &ips.ipv4);
|
let success = agent.update_entry(cloudflare_entry, &ips.ipv4);
|
||||||
if success { updated4 += 1 } else { error4 += 1 }
|
if success {
|
||||||
|
updated4 += 1
|
||||||
|
} else {
|
||||||
|
error4 += 1
|
||||||
}
|
}
|
||||||
else {
|
} else {
|
||||||
unchanged4 += 1;
|
unchanged4 += 1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if cloudflare_entry.r#type == cloudflare::CloudflareDnsType::AAAA && entry.type6 {
|
if cloudflare_entry.r#type == cloudflare::CloudflareDnsType::AAAA && entry.type6
|
||||||
|
{
|
||||||
found6 = true;
|
found6 = true;
|
||||||
let ipv6 = ips.ipv6base.clone() + entry.interface.as_ref().unwrap();
|
let ipv6 = ips.ipv6base.clone() + entry.interface.as_ref().unwrap();
|
||||||
|
|
||||||
if cloudflare_entry.is_ip_new(&ipv6) {
|
if cloudflare_entry.is_ip_new(&ipv6) {
|
||||||
let success = agent.update_entry(cloudflare_entry, &ipv6);
|
let success = agent.update_entry(cloudflare_entry, &ipv6);
|
||||||
if success { updated6 += 1 } else { error6 += 1 }
|
if success {
|
||||||
|
updated6 += 1
|
||||||
|
} else {
|
||||||
|
error6 += 1
|
||||||
}
|
}
|
||||||
else {
|
} else {
|
||||||
unchanged6 += 1;
|
unchanged6 += 1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
//println!("{:#?}", entry.subdomain); // DEBUG
|
//println!("{:#?}", entry.subdomain); // DEBUG
|
||||||
|
@ -155,13 +177,17 @@ fn main() {
|
||||||
|
|
||||||
if !found4 && entry.type4 {
|
if !found4 && entry.type4 {
|
||||||
let success = agent.create_entry(&zone.id, "A", &entry.name, &ips.ipv4);
|
let success = agent.create_entry(&zone.id, "A", &entry.name, &ips.ipv4);
|
||||||
if success { created4 += 1 };
|
if success {
|
||||||
|
created4 += 1
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
if !found6 && entry.type6 {
|
if !found6 && entry.type6 {
|
||||||
let ipv6 = ips.ipv6base.clone() + entry.interface.as_ref().unwrap();
|
let ipv6 = ips.ipv6base.clone() + entry.interface.as_ref().unwrap();
|
||||||
let success = agent.create_entry(&zone.id, "AAAA", &entry.name, &ipv6);
|
let success = agent.create_entry(&zone.id, "AAAA", &entry.name, &ipv6);
|
||||||
if success { created6 += 1 };
|
if success {
|
||||||
|
created6 += 1
|
||||||
|
};
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -177,5 +203,4 @@ fn main() {
|
||||||
println!("{} updated", updated6);
|
println!("{} updated", updated6);
|
||||||
println!("{} unchanged", unchanged6);
|
println!("{} unchanged", unchanged6);
|
||||||
println!("{} errors", error6);
|
println!("{} errors", error6);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue