2017-05-20 17:39:37 +00:00
|
|
|
use users::{Users, Groups};
|
|
|
|
|
|
|
|
use fs::fields as f;
|
2017-05-20 19:29:22 +00:00
|
|
|
use output::colours::Colours;
|
2017-05-20 17:39:37 +00:00
|
|
|
use output::cell::TextCell;
|
|
|
|
|
|
|
|
|
2017-05-20 19:29:22 +00:00
|
|
|
impl f::Group {
|
|
|
|
pub fn render<U: Users+Groups>(&self, colours: &Colours, users: &U) -> TextCell {
|
2017-05-20 17:39:37 +00:00
|
|
|
use users::os::unix::GroupExt;
|
|
|
|
|
2017-05-20 19:29:22 +00:00
|
|
|
let mut style = colours.users.group_not_yours;
|
2017-05-20 17:39:37 +00:00
|
|
|
|
2017-05-20 19:29:22 +00:00
|
|
|
let group = match users.get_group_by_gid(self.0) {
|
2017-05-20 17:39:37 +00:00
|
|
|
Some(g) => (*g).clone(),
|
2017-05-20 19:29:22 +00:00
|
|
|
None => return TextCell::paint(style, self.0.to_string()),
|
2017-05-20 17:39:37 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
let current_uid = users.get_current_uid();
|
|
|
|
if let Some(current_user) = users.get_user_by_uid(current_uid) {
|
|
|
|
if current_user.primary_group_id() == group.gid()
|
|
|
|
|| group.members().contains(¤t_user.name().to_owned()) {
|
2017-05-20 19:29:22 +00:00
|
|
|
style = colours.users.group_yours;
|
2017-05-20 17:39:37 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
TextCell::paint(style, group.name().to_owned())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
#[allow(unused_results)]
|
|
|
|
pub mod test {
|
|
|
|
use fs::fields as f;
|
|
|
|
use output::cell::TextCell;
|
2017-06-24 21:39:15 +00:00
|
|
|
use output::colours::Colours;
|
2017-05-20 17:39:37 +00:00
|
|
|
|
|
|
|
use users::{User, Group};
|
|
|
|
use users::mock::MockUsers;
|
|
|
|
use users::os::unix::GroupExt;
|
|
|
|
use ansi_term::Colour::*;
|
|
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn named() {
|
2017-06-24 21:39:15 +00:00
|
|
|
let mut colours = Colours::default();
|
|
|
|
colours.users.group_not_yours = Fixed(101).normal();
|
2017-05-20 17:39:37 +00:00
|
|
|
|
|
|
|
let mut users = MockUsers::with_current_uid(1000);
|
|
|
|
users.add_group(Group::new(100, "folk"));
|
|
|
|
|
|
|
|
let group = f::Group(100);
|
|
|
|
let expected = TextCell::paint_str(Fixed(101).normal(), "folk");
|
2017-06-24 21:39:15 +00:00
|
|
|
assert_eq!(expected, group.render(&colours, &users))
|
2017-05-20 17:39:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn unnamed() {
|
2017-06-24 21:39:15 +00:00
|
|
|
let mut colours = Colours::default();
|
|
|
|
colours.users.group_not_yours = Fixed(87).normal();
|
2017-05-20 17:39:37 +00:00
|
|
|
|
|
|
|
let users = MockUsers::with_current_uid(1000);
|
|
|
|
|
|
|
|
let group = f::Group(100);
|
|
|
|
let expected = TextCell::paint_str(Fixed(87).normal(), "100");
|
2017-06-24 21:39:15 +00:00
|
|
|
assert_eq!(expected, group.render(&colours, &users));
|
2017-05-20 17:39:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn primary() {
|
2017-06-24 21:39:15 +00:00
|
|
|
let mut colours = Colours::default();
|
|
|
|
colours.users.group_yours = Fixed(64).normal();
|
2017-05-20 17:39:37 +00:00
|
|
|
|
|
|
|
let mut users = MockUsers::with_current_uid(2);
|
|
|
|
users.add_user(User::new(2, "eve", 100));
|
|
|
|
users.add_group(Group::new(100, "folk"));
|
|
|
|
|
|
|
|
let group = f::Group(100);
|
|
|
|
let expected = TextCell::paint_str(Fixed(64).normal(), "folk");
|
2017-06-24 21:39:15 +00:00
|
|
|
assert_eq!(expected, group.render(&colours, &users))
|
2017-05-20 17:39:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn secondary() {
|
2017-06-24 21:39:15 +00:00
|
|
|
let mut colours = Colours::default();
|
|
|
|
colours.users.group_yours = Fixed(31).normal();
|
2017-05-20 17:39:37 +00:00
|
|
|
|
|
|
|
let mut users = MockUsers::with_current_uid(2);
|
|
|
|
users.add_user(User::new(2, "eve", 666));
|
|
|
|
|
|
|
|
let test_group = Group::new(100, "folk").add_member("eve");
|
|
|
|
users.add_group(test_group);
|
|
|
|
|
|
|
|
let group = f::Group(100);
|
|
|
|
let expected = TextCell::paint_str(Fixed(31).normal(), "folk");
|
2017-06-24 21:39:15 +00:00
|
|
|
assert_eq!(expected, group.render(&colours, &users))
|
2017-05-20 17:39:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn overflow() {
|
2017-06-24 21:39:15 +00:00
|
|
|
let mut colours = Colours::default();
|
|
|
|
colours.users.group_not_yours = Blue.underline();
|
2017-05-20 17:39:37 +00:00
|
|
|
|
|
|
|
let group = f::Group(2_147_483_648);
|
|
|
|
let expected = TextCell::paint_str(Blue.underline(), "2147483648");
|
2017-06-24 21:39:15 +00:00
|
|
|
assert_eq!(expected, group.render(&colours, &MockUsers::with_current_uid(0)));
|
2017-05-20 17:39:37 +00:00
|
|
|
}
|
|
|
|
}
|