Add getters for the attributes

This commit is contained in:
Dietrich 2020-09-29 12:14:07 +02:00
parent 5d83f1264f
commit a25092c7a0
2 changed files with 113 additions and 28 deletions

View File

@ -78,9 +78,9 @@ impl<'a> Passwd<'a> {
/// # Example /// # Example
/// ``` /// ```
/// let pwd = adduser::passwd::Passwd::new_from_string( /// let pwd = adduser::passwd::Passwd::new_from_string(
/// "testuser:testpassword:1001:1001:full Name,004,000342,001-2312,myemail@test.com:/home/test:/bin/test" /// "testuser:testpassword:1001:1001:full Name,,,,:/home/test:/bin/test"
/// ).unwrap(); /// ).unwrap();
/// assert!(format!("{:?}", pwd).find("testuser") > Some(0)); /// assert_eq!(pwd.get_username(), "testuser");
/// ``` /// ```
/// ///
/// # Errors /// # Errors
@ -89,23 +89,89 @@ impl<'a> Passwd<'a> {
let elements: Vec<&str> = line.split(':').collect(); let elements: Vec<&str> = line.split(':').collect();
if elements.len() == 7 { if elements.len() == 7 {
Ok(Passwd { Ok(Passwd {
username: Username::try_from(*elements.get(0).unwrap()) username: Username::try_from(*elements.get(0).unwrap())?,
.expect("failed to parse username."), password: Password::try_from(*elements.get(1).unwrap())?,
password: Password::try_from(*elements.get(1).unwrap()) uid: Uid::try_from(*elements.get(2).unwrap())?,
.expect("Failed to parse Password"), gid: Gid::try_from(*elements.get(3).unwrap())?,
uid: Uid::try_from(*elements.get(2).unwrap()).expect("Failed to parse uid"), gecos: Gecos::try_from(*elements.get(4).unwrap())?,
gid: Gid::try_from(*elements.get(3).unwrap()).expect("Failed to parse gid"), home_dir: HomeDir::try_from(*elements.get(5).unwrap())?,
gecos: Gecos::try_from(*elements.get(4).unwrap()) shell_dir: ShellDir::try_from(*elements.get(6).unwrap())?,
.expect("Failed to parse Gecos field"),
home_dir: HomeDir::try_from(*elements.get(5).unwrap())
.expect("Failed to parse home directory"),
shell_dir: ShellDir::try_from(*elements.get(6).unwrap())
.expect("Failed to parse shell directory"),
}) })
} else { } else {
Err(UserLibError::Message( Err("Failed to parse: not enough elements".into())
"Failed to parse: not enough elements".to_owned(), }
)) }
#[must_use]
pub const fn get_username(&self) -> &'a str {
self.username.username
}
#[must_use]
pub const fn get_password(&self) -> &'a str {
self.password.password
}
#[must_use]
pub const fn get_uid(&self) -> u32 {
self.uid.uid
}
#[must_use]
pub const fn get_gid(&self) -> u32 {
self.gid.gid
}
#[must_use]
pub const fn get_comment(&self) -> &Gecos {
&self.gecos
}
#[must_use]
pub const fn get_home_dir(&self) -> &'a str {
self.home_dir.dir
}
#[must_use]
pub const fn get_shell_dir(&self) -> &'a str {
self.shell_dir.shell
}
}
impl<'a> Gecos<'a> {
#[must_use]
pub const fn get_comment(&'a self) -> Option<&'a str> {
match *self {
Gecos::Simple { comment, .. } => Some(comment),
Gecos::Detail { .. } => None,
}
}
#[must_use]
pub const fn get_full_name(&'a self) -> Option<&'a str> {
match *self {
Gecos::Simple { .. } => None,
Gecos::Detail { full_name, .. } => Some(full_name),
}
}
#[must_use]
pub const fn get_room(&'a self) -> Option<&'a str> {
match *self {
Gecos::Simple { .. } => None,
Gecos::Detail { room, .. } => Some(room),
}
}
#[must_use]
pub const fn get_phone_work(&'a self) -> Option<&'a str> {
match *self {
Gecos::Simple { .. } => None,
Gecos::Detail { phone_work, .. } => Some(phone_work),
}
}
#[must_use]
pub const fn get_phone_home(&'a self) -> Option<&'a str> {
match *self {
Gecos::Simple { .. } => None,
Gecos::Detail { phone_home, .. } => Some(phone_home),
}
}
#[must_use]
pub const fn get_other(&'a self) -> Option<&'a str> {
match *self {
Gecos::Simple { .. } => None,
Gecos::Detail { other, .. } => Some(other),
} }
} }
} }
@ -155,7 +221,7 @@ impl Display for Username<'_> {
} }
impl<'a> TryFrom<&'a str> for Username<'a> { impl<'a> TryFrom<&'a str> for Username<'a> {
type Error = &'static str; type Error = UserLibError;
fn try_from(source: &'a str) -> std::result::Result<Self, Self::Error> { fn try_from(source: &'a str) -> std::result::Result<Self, Self::Error> {
Ok(Self { username: source }) Ok(Self { username: source })
} }
@ -168,7 +234,7 @@ impl Display for Password<'_> {
} }
impl<'a> TryFrom<&'a str> for Password<'a> { impl<'a> TryFrom<&'a str> for Password<'a> {
type Error = &'static str; type Error = UserLibError;
fn try_from(source: &'a str) -> std::result::Result<Self, Self::Error> { fn try_from(source: &'a str) -> std::result::Result<Self, Self::Error> {
Ok(Self { password: source }) Ok(Self { password: source })
} }
@ -181,7 +247,7 @@ impl Display for Uid {
} }
impl TryFrom<&str> for Uid { impl TryFrom<&str> for Uid {
type Error = &'static str; type Error = UserLibError;
fn try_from(source: &str) -> std::result::Result<Self, Self::Error> { fn try_from(source: &str) -> std::result::Result<Self, Self::Error> {
Ok(Self { Ok(Self {
uid: source.parse::<u32>().unwrap(), uid: source.parse::<u32>().unwrap(),
@ -196,7 +262,7 @@ impl Display for Gid {
} }
impl TryFrom<&str> for Gid { impl TryFrom<&str> for Gid {
type Error = &'static str; type Error = UserLibError;
fn try_from(source: &str) -> std::result::Result<Self, Self::Error> { fn try_from(source: &str) -> std::result::Result<Self, Self::Error> {
Ok(Self { Ok(Self {
gid: source.parse::<u32>().unwrap(), gid: source.parse::<u32>().unwrap(),
@ -224,7 +290,7 @@ impl Display for Gecos<'_> {
} }
impl<'a> TryFrom<&'a str> for Gecos<'a> { impl<'a> TryFrom<&'a str> for Gecos<'a> {
type Error = &'static str; type Error = UserLibError;
fn try_from(source: &'a str) -> std::result::Result<Self, Self::Error> { fn try_from(source: &'a str) -> std::result::Result<Self, Self::Error> {
let vals: Vec<&str> = source.split(',').collect(); let vals: Vec<&str> = source.split(',').collect();
if vals.len() == 5 { if vals.len() == 5 {
@ -260,7 +326,7 @@ impl Display for HomeDir<'_> {
} }
impl<'a> TryFrom<&'a str> for HomeDir<'a> { impl<'a> TryFrom<&'a str> for HomeDir<'a> {
type Error = &'static str; type Error = UserLibError;
fn try_from(source: &'a str) -> std::result::Result<Self, Self::Error> { fn try_from(source: &'a str) -> std::result::Result<Self, Self::Error> {
Ok(Self { dir: source }) Ok(Self { dir: source })
} }
@ -273,7 +339,7 @@ impl Display for ShellDir<'_> {
} }
impl<'a> TryFrom<&'a str> for ShellDir<'a> { impl<'a> TryFrom<&'a str> for ShellDir<'a> {
type Error = &'static str; type Error = UserLibError;
fn try_from(source: &'a str) -> std::result::Result<Self, Self::Error> { fn try_from(source: &'a str) -> std::result::Result<Self, Self::Error> {
Ok(Self { shell: source }) Ok(Self { shell: source })
} }
@ -340,6 +406,11 @@ fn test_parse_gecos() {
#[test] #[test]
fn test_new_from_string() { fn test_new_from_string() {
// Test if a single line can be parsed and if the resulting struct is populated correctly. // Test if a single line can be parsed and if the resulting struct is populated correctly.
let fail = Passwd::new_from_string("").err().unwrap();
assert_eq!(
fail,
UserLibError::Message("Failed to parse: not enough elements".into())
);
let pwd = let pwd =
Passwd::new_from_string("testuser:testpassword:1001:1001:testcomment:/home/test:/bin/test") Passwd::new_from_string("testuser:testpassword:1001:1001:testcomment:/home/test:/bin/test")
.unwrap(); .unwrap();

View File

@ -1,9 +1,21 @@
use std::error::Error; use std::error::Error;
use std::fmt::{self, Display}; use std::fmt::{self, Display};
#[derive(Debug)] #[derive(Debug, PartialEq)]
pub enum ParseError {
Username,
Password,
Uid,
Gid,
Gecos,
HomeDir,
ShellDir,
}
#[derive(Debug, PartialEq)]
pub enum UserLibError { pub enum UserLibError {
NotFound, NotFound,
ParseError,
Message(String), Message(String),
} }
@ -11,6 +23,7 @@ impl Display for UserLibError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self { match self {
Self::NotFound => write!(f, ""), Self::NotFound => write!(f, ""),
Self::ParseError => write!(f, "Failed to parse"), // TODO details
Self::Message(message) => write!(f, "{}", message), Self::Message(message) => write!(f, "{}", message),
} }
} }
@ -20,13 +33,14 @@ impl Error for UserLibError {
fn description(&self) -> &str { fn description(&self) -> &str {
match self { match self {
Self::NotFound => "not found", Self::NotFound => "not found",
Self::ParseError => "failed to parse",
Self::Message(message) => message, Self::Message(message) => message,
} }
} }
} }
impl From<String> for UserLibError { impl From<&str> for UserLibError {
fn from(err: String) -> Self { fn from(err: &str) -> Self {
Self::Message(err) Self::Message(err.to_owned())
} }
} }