spiretf/dispenser
1
0
Fork 0
mirror of https://codeberg.org/spire/dispenser.git synced 2026-06-03 18:14:06 +02:00
Code Issues Projects Releases Packages Wiki Activity

update dns when adopting server

Browse source
This commit is contained in:
Robin Appelman 2022-08-21 22:52:00 +02:00
commit cf515a4acd
2 changed files with 25 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/config.rs
View file

@ -172,7 +172,7 @@ fn digital_ocean_default_plan() -> String {
String::from("s-2vcpu-2gb") String::from("s-2vcpu-2gb")
} }
#[derive(Deserialize, Debug)] #[derive(Deserialize, Debug, Clone)]
pub struct DynDnsConfig { pub struct DynDnsConfig {
pub update_url: String, pub update_url: String,
pub hostname: String, pub hostname: String,

39
src/main.rs
View file

@ -1,7 +1,7 @@
extern crate core; extern crate core;
use crate::cloud::{Cloud, CloudError, CreatedAuth, Server}; use crate::cloud::{Cloud, CloudError, CreatedAuth, Server};
use crate::config::{Config, ConfigError, ServerConfig}; use crate::config::{Config, ConfigError, DynDnsConfig, ServerConfig};
use crate::dns::{DynDnsClient, DynDnsError}; use crate::dns::{DynDnsClient, DynDnsError};
use crate::rcon::Rcon; use crate::rcon::Rcon;
use crate::ssh::SshError; use crate::ssh::SshError;
@ -15,9 +15,9 @@ use std::str::FromStr;
use std::sync::Arc; use std::sync::Arc;
use std::time::Duration; use std::time::Duration;
use thiserror::Error; use thiserror::Error;
use tokio::select;
use tokio::signal::ctrl_c; use tokio::signal::ctrl_c;
use tokio::time::sleep; use tokio::time::sleep;
use tokio::{select, spawn};
use tracing::{debug, error, info, instrument, warn}; use tracing::{debug, error, info, instrument, warn};
mod cloud; mod cloud;
@ -106,7 +106,7 @@ async fn setup(
} }
} }
debug!("starting container"); info!("starting container");
let result = ssh let result = ssh
.exec(format!( .exec(format!(
"docker run --name spire -d \ "docker run --name spire -d \
@ -136,13 +136,13 @@ async fn setup(
return Err(Error::SetupError(result.output())); return Err(Error::SetupError(result.output()));
} }
debug!("setting up swap"); info!("setting up swap");
ssh.exec("dd if=/dev/zero of=/swapfile bs=1M count=1024") ssh.exec("dd if=/dev/zero of=/swapfile bs=1M count=1024")
.await?; .await?;
ssh.exec("chmod 600 /swapfile && mkswap /swapfile && swapon /swapfile") ssh.exec("chmod 600 /swapfile && mkswap /swapfile && swapon /swapfile")
.await?; .await?;
debug!("setting up prometheus"); info!("setting up prometheus");
ssh.exec("wget https://github.com/icewind1991/palantir/raw/main/palantir.service -O /etc/systemd/system/palantir.service").await?; ssh.exec("wget https://github.com/icewind1991/palantir/raw/main/palantir.service -O /etc/systemd/system/palantir.service").await?;
ssh.exec("wget https://github.com/icewind1991/palantir/releases/download/v1.1.0/palantir-x86_64-unknown-linux-musl -O /usr/local/bin/palantir").await?; ssh.exec("wget https://github.com/icewind1991/palantir/releases/download/v1.1.0/palantir-x86_64-unknown-linux-musl -O /usr/local/bin/palantir").await?;
ssh.exec("chmod +x /usr/local/bin/palantir").await?; ssh.exec("chmod +x /usr/local/bin/palantir").await?;
@ -262,6 +262,9 @@ async fn run_loop(
server = debug(&server), server = debug(&server),
"Taking ownership of existing server" "Taking ownership of existing server"
); );
if let Some(dns_config) = config.dyndns.as_ref() {
spawn(set_dyndns(dns_config.clone(), server.ip));
}
active_server = Some(server); active_server = Some(server);
} }
Err(e) => eprintln!("{:#}", e), Err(e) => eprintln!("{:#}", e),
@ -331,16 +334,7 @@ async fn start(cloud: &dyn Cloud, config: &Config) -> Result<Server, Error> {
println!(" Root Password: {}", created.auth); println!(" Root Password: {}", created.auth);
let connect_host = if let Some(dns_config) = config.dyndns.as_ref() { let connect_host = if let Some(dns_config) = config.dyndns.as_ref() {
let dns = DynDnsClient::new( spawn(set_dyndns(dns_config.clone(), server.ip));
dns_config.update_url.to_string(),
dns_config.username.to_string(),
dns_config.password.to_string(),
);
println!(
"Updating DynDNS entry for {} to {}",
dns_config.hostname, server.ip
);
dns.update(&dns_config.hostname, server.ip).await?;
dns_config.hostname.to_string() dns_config.hostname.to_string()
} else { } else {
format!("{}", server.ip) format!("{}", server.ip)
@ -364,6 +358,21 @@ async fn start(cloud: &dyn Cloud, config: &Config) -> Result<Server, Error> {
Ok(server) Ok(server)
} }
async fn set_dyndns(dns_config: DynDnsConfig, ip: IpAddr) {
let dns = DynDnsClient::new(
dns_config.update_url.to_string(),
dns_config.username.to_string(),
dns_config.password.to_string(),
);
println!(
"Updating DynDNS entry for {} to {}",
dns_config.hostname, ip
);
if let Err(e) = dns.update(&dns_config.hostname, ip).await {
eprintln!("Error while updating DynDNS: {}", e);
}
}
async fn connect_ssh(ip: IpAddr, auth: &CreatedAuth) -> Result<SshSession, Error> { async fn connect_ssh(ip: IpAddr, auth: &CreatedAuth) -> Result<SshSession, Error> {
let mut tries = 0; let mut tries = 0;