Skynet/ldap_backend
6
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity Actions

fmt: fmt and clippy

Browse source
This commit is contained in:
silver 2023-06-18 18:34:27 +01:00
parent a990f31a45
commit 9cba5ce94a
1 changed files with 16 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/bin/update_groups.rs
View file

@ -1,7 +1,7 @@
use std::env;
use dotenv::dotenv;
use ldap3::{LdapConn, Mod};
use skynet_ldap_server::{Config, get_config};
use skynet_ldap_server::{get_config, Config};
use std::env;
#[async_std::main]
async fn main() -> tide::Result<()> {
@ -13,36 +13,35 @@ async fn main() -> tide::Result<()> {
Ok(())
}
//async fn update_users(config: &Config) {
//}
fn uid_to_dn(uid: &str) -> String{
fn uid_to_dn(uid: &str) -> String {
format!("uid={},ou=users,dc=skynet,dc=ie", uid)
}
async fn update_admin(config: &Config) -> tide::Result<()>{
async fn update_admin(config: &Config) -> tide::Result<()> {
dotenv().ok();
// read from teh env
if let Ok(x) = env::var("USERS_ADMIN") {
let users = x.split(',').collect::<Vec<&str>>();
update_group(config,"skynet-admins", &users, true).await?;
update_group(config, "skynet-admins", &users, true).await?;
// admins automatically get added as users
update_group(config,"skynet-users", &users, false).await?;
update_group(config, "skynet-users", &users, false).await?;
}
Ok(())
}
async fn update_group(config: &Config, group: &str, users: &Vec<&str>, replace: bool) -> tide::Result<()>{
async fn update_group(config: &Config, group: &str, users: &[&str], replace: bool) -> tide::Result<()> {
let mut ldap = LdapConn::new(&config.ldap_host)?;
// use the admin account
ldap.simple_bind(&config.ldap_admin, &config.ldap_admin_pw)?.success()?;
let dn = format!("cn={},ou=groups,dc=skynet,dc=ie", group);
let members = users.clone().into_iter().map(|uid| uid_to_dn(uid)).collect();
let members = users.iter().map(|uid| uid_to_dn(uid)).collect();
let mods = if replace {
vec![Mod::Replace("member".to_string(), members)]
} else {
@ -54,13 +53,13 @@ async fn update_group(config: &Config, group: &str, users: &Vec<&str>, replace:
}
let dn_linux = format!("cn={}-linux,ou=groups,dc=skynet,dc=ie", group);
let members_linux = users.clone().into_iter().map(|uid| uid.to_string()).collect();
let members_linux = users.iter().map(|uid| uid.to_string()).collect();
let mods = if replace {
vec![Mod::Replace("memberUid".to_string(), members_linux)]
} else {
vec![Mod::Add("memberUid".to_string(), members_linux)]
};
if let Err(x) = ldap.modify(&dn_linux, mods){
if let Err(x) = ldap.modify(&dn_linux, mods) {
println!("{:?}", x);
};