Last known almost-working Rust code
This commit is contained in:
Gregory Ballantine
parent
49ad921c9d
commit
e38b263e57
7
Cargo.lock
generated
7
Cargo.lock
generated
@ -210,6 +210,7 @@ version = "0.1.0"
|
||||
dependencies = [
|
||||
"clap 2.32.0 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||
"ldap3 0.6.0 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||
"maplit 1.0.1 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||
"yaml-rust 0.4.0 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||
]
|
||||
|
||||
@ -262,6 +263,11 @@ dependencies = [
|
||||
"cfg-if 0.1.4 (registry+https://github.com/rust-lang/crates.io-index)",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "maplit"
|
||||
version = "1.0.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
|
||||
[[package]]
|
||||
name = "matches"
|
||||
version = "0.1.6"
|
||||
@ -834,6 +840,7 @@ dependencies = [
|
||||
"checksum linked-hash-map 0.5.1 (registry+https://github.com/rust-lang/crates.io-index)" = "70fb39025bc7cdd76305867c4eccf2f2dcf6e9a57f5b21a93e1c2d86cd03ec9e"
|
||||
"checksum log 0.3.9 (registry+https://github.com/rust-lang/crates.io-index)" = "e19e8d5c34a3e0e2223db8e060f9e8264aeeb5c5fc64a4ee9965c062211c024b"
|
||||
"checksum log 0.4.3 (registry+https://github.com/rust-lang/crates.io-index)" = "61bd98ae7f7b754bc53dca7d44b604f733c6bba044ea6f41bc8d89272d8161d2"
|
||||
"checksum maplit 1.0.1 (registry+https://github.com/rust-lang/crates.io-index)" = "08cbb6b4fef96b6d77bfc40ec491b1690c779e77b05cd9f07f787ed376fd4c43"
|
||||
"checksum matches 0.1.6 (registry+https://github.com/rust-lang/crates.io-index)" = "100aabe6b8ff4e4a7e32c1c13523379802df0772b82466207ac25b013f193376"
|
||||
"checksum memoffset 0.2.1 (registry+https://github.com/rust-lang/crates.io-index)" = "0f9dc261e2b62d7a622bf416ea3c5245cdd5d9a7fcc428c0d06804dfce1775b3"
|
||||
"checksum mio 0.6.15 (registry+https://github.com/rust-lang/crates.io-index)" = "4fcfcb32d63961fb6f367bfd5d21e4600b92cd310f71f9dca25acae196eb1560"
|
||||
|
||||
@ -7,5 +7,7 @@ authors = ["Gregory Ballantine <gballantine555@gmail.com>"]
|
||||
version = "2.32"
|
||||
[dependencies.ldap3]
|
||||
version = "0.6"
|
||||
[dependencies.maplit]
|
||||
version = "1.0"
|
||||
[dependencies.yaml-rust]
|
||||
version = "0.4"
|
||||
|
||||
25
src/ldap.rs
25
src/ldap.rs
@ -2,6 +2,7 @@ use std::collections::HashMap;
|
||||
use std::error::Error;
|
||||
|
||||
use ldap3::{LdapConn, Scope, SearchEntry};
|
||||
use yaml_rust::Yaml;
|
||||
|
||||
// create an LDAP connection object
|
||||
pub fn connect(ldap_connection_info: HashMap<&str, &str>) -> LdapConn {
|
||||
@ -43,3 +44,27 @@ pub fn do_search(ldap_conn: LdapConn, search_base: &str, search_filter: &str) ->
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
// perform an add user operation
|
||||
pub fn do_add_user(ldap_conn: LdapConn, user_conf: Yaml, user_info: &HashMap<&str, &str>) -> Result<(), Box<Error>> {
|
||||
let user_dn = format!("{}={},{}", user_conf["id_attr"].as_str().unwrap(), user_info.get("username").unwrap(), user_conf["base"].as_str().unwrap());
|
||||
let user_home = format!("/nfs/users/{}/home", user_info.get("username").unwrap());
|
||||
let user_display = format!("{} {}", user_info.get("first-name").unwrap(), user_info.get("last-name").unwrap());
|
||||
|
||||
let user_vec = vec![
|
||||
(user_conf["id_attr"].as_str().unwrap(), hashset!{user_info.get("username").unwrap().to_owned()}),
|
||||
("mail", hashset!{user_info.get("username").unwrap().to_owned()}),
|
||||
("displayName", hashset!{user_display.as_str()}),
|
||||
("givenName", hashset!{user_info.get("first-name").unwrap().to_owned()}),
|
||||
("sn", hashset!{user_info.get("last-name").unwrap().to_owned()}),
|
||||
("uidNumber", hashset!{user_info.get("id-number").unwrap().to_owned()}),
|
||||
("gidNumber", hashset!{user_info.get("id-number").unwrap().to_owned()}),
|
||||
("homeDirectory", hashset!{user_home.as_str()}),
|
||||
("objectClass", hashset!{user_conf["attrs"]["objectClass"].as_vec().unwrap()})
|
||||
];
|
||||
|
||||
let res = ldap_conn.add(&user_dn, user_vec)?.success()?;
|
||||
println!("{:?}", res);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
72
src/main.rs
72
src/main.rs
@ -1,5 +1,7 @@
|
||||
extern crate clap;
|
||||
extern crate ldap3;
|
||||
#[macro_use]
|
||||
extern crate maplit;
|
||||
extern crate yaml_rust;
|
||||
|
||||
mod ldap;
|
||||
@ -71,6 +73,49 @@ fn main() {
|
||||
.default_value("(&(objectClass=*))")
|
||||
.help("LDAP search filter")))
|
||||
|
||||
// LDAP add resource subcommand
|
||||
.subcommand(SubCommand::with_name("add")
|
||||
.about("Add an LDAP entry")
|
||||
|
||||
// add user entry subcommand
|
||||
.subcommand(SubCommand::with_name("user")
|
||||
.about("Add an LDAP user entry")
|
||||
// Username argument
|
||||
.arg(Arg::with_name("username")
|
||||
.short("u")
|
||||
.long("username")
|
||||
.takes_value(true)
|
||||
.required(true)
|
||||
.help("User's login username"))
|
||||
// Email address argument
|
||||
.arg(Arg::with_name("email")
|
||||
.short("e")
|
||||
.long("email")
|
||||
.takes_value(true)
|
||||
.required(true)
|
||||
.help("User's email address"))
|
||||
// First name argument
|
||||
.arg(Arg::with_name("first-name")
|
||||
.short("f")
|
||||
.long("first-name")
|
||||
.takes_value(true)
|
||||
.required(true)
|
||||
.help("User's first name"))
|
||||
// Last name argument
|
||||
.arg(Arg::with_name("last-name")
|
||||
.short("l")
|
||||
.long("last-name")
|
||||
.takes_value(true)
|
||||
.required(true)
|
||||
.help("User's last name"))
|
||||
// ID number argument
|
||||
.arg(Arg::with_name("id-number")
|
||||
.short("i")
|
||||
.long("id-number")
|
||||
.takes_value(true)
|
||||
.required(true)
|
||||
.help("User's ID number"))))
|
||||
|
||||
.get_matches();
|
||||
|
||||
// create a new hashmap for the connection info
|
||||
@ -90,6 +135,31 @@ fn main() {
|
||||
let search_filter = matches.value_of("filter").unwrap();
|
||||
|
||||
// run the LDAP search
|
||||
let _ = ldap::do_search(ldap_conn, search_base, search_filter);
|
||||
match ldap::do_search(ldap_conn, search_base, search_filter) {
|
||||
Ok(_) => {},
|
||||
Err(e) => println!("{:?}", e),
|
||||
}
|
||||
} else if let Some(matches) = matches.subcommand_matches("add") {
|
||||
// perform an LDAP bind
|
||||
let bind_dn = yaml["muldap"]["bind_dn"].as_str().unwrap();
|
||||
let bind_pw = yaml["muldap"]["bind_pw"].as_str().unwrap();
|
||||
ldap_conn.simple_bind(bind_dn, bind_pw);
|
||||
|
||||
if let Some(matches) = matches.subcommand_matches("user") {
|
||||
// create HashMap of user info
|
||||
let mut ldap_user_info = HashMap::new();
|
||||
ldap_user_info.insert("username", matches.value_of("username").unwrap());
|
||||
ldap_user_info.insert("email", matches.value_of("email").unwrap());
|
||||
ldap_user_info.insert("first-name", matches.value_of("first-name").unwrap());
|
||||
ldap_user_info.insert("last-name", matches.value_of("last-name").unwrap());
|
||||
ldap_user_info.insert("id-number", matches.value_of("id-number").unwrap());
|
||||
|
||||
let user_conf = &yaml["muldap"]["user"];
|
||||
|
||||
match ldap::do_add_user(ldap_conn, user_conf.to_owned(), &ldap_user_info) {
|
||||
Ok(_) => {},
|
||||
Err(e) => println!("{:?}", e),
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user