1
0
Fork 0
forked from navos/sheepd

mqtt connections

almost working, need encryption
This commit is contained in:
JMARyA 2025-04-28 23:33:11 +02:00
parent 20ab0153d1
commit 125d50530d
Signed by untrusted user: jmarya
GPG key ID: 901B2ADDF27C2263
15 changed files with 1259 additions and 42 deletions

View file

@ -1,37 +1,53 @@
use crate::{api, sheepd_core::config::AgentConfig};
use std::path::PathBuf;
use crate::{
api::{self, Identity, JoinResponse},
sheepd_core::config::AgentConfig,
};
use super::args::JoinCommand;
fn domain(host: &str) -> String {
if host.starts_with("http") {
return host.to_string();
} else {
format!("https://{host}")
}
}
pub fn join(conf: JoinCommand) {
// TODO : check for root
// TODO : check if joined somewhere already
log::info!("Joining to {}", conf.home);
let url = format!("http://{}/join", conf.home);
let _ = std::fs::create_dir_all("/etc/sheepd");
let i = if let Some(i) = Identity::try_load(&PathBuf::from("/etc/sheepd")) {
i
} else {
let i = Identity::new();
i.save(&PathBuf::from("/etc/sheepd"));
i
};
let url = format!("{}/join", domain(&conf.home));
println!("{url}");
let mut res = ureq::post(url)
.send_json(&api::JoinParams {
join_token: None,
machine_id: std::fs::read_to_string("/etc/machine-id").unwrap(),
hostname: std::fs::read_to_string("/etc/hostname").unwrap(),
identity: i.public(),
})
.unwrap();
let res: serde_json::Value = res.body_mut().read_json().unwrap();
let token = res
.as_object()
.unwrap()
.get("ok")
.unwrap()
.as_str()
.unwrap();
let res: JoinResponse = res.body_mut().read_json().unwrap();
log::info!("Joined {} successfully", conf.home);
std::fs::write(
"/etc/sheepd.toml",
toml::to_string(&AgentConfig::new(&conf.home, token)).unwrap(),
"/etc/sheepd/config.toml",
toml::to_string(&AgentConfig::new(&conf.home, res)).unwrap(),
)
.unwrap();
}