Llewellyn/starship
1
0
Fork 0
mirror of https://github.com/Llewellynvdm/starship.git synced 2024-12-26 03:27:30 +00:00
Code Issues Releases Activity

perf(username): get uid without external command (#2286)

Browse Source
This commit is contained in:
David Knaack 2021-02-09 18:41:09 +01:00 committed by GitHub
parent e7246cee82
commit e8a02e7d53
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/modules/username.rs
View File

@ -2,9 +2,7 @@ use super::{Context, Module, RootModuleConfig};
use crate::configs::username::UsernameConfig; use crate::configs::username::UsernameConfig;
use crate::formatter::StringFormatter; use crate::formatter::StringFormatter;
use crate::utils;
const ROOT_UID: Option<u32> = Some(0);
#[cfg(not(target_os = "windows"))] #[cfg(not(target_os = "windows"))]
const USERNAME_ENV_VAR: &str = "USER"; const USERNAME_ENV_VAR: &str = "USER";
@ -21,13 +19,7 @@ pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
let username = context.get_env(USERNAME_ENV_VAR)?; let username = context.get_env(USERNAME_ENV_VAR)?;
let logname = context.get_env("LOGNAME"); let logname = context.get_env("LOGNAME");
let is_root = if cfg!(not(target_os = "windows")) { let is_root = is_root_user();
let user_uid = get_uid();
user_uid == ROOT_UID
} else {
false
};
let is_not_login = logname.is_some() && username != logname.unwrap(); let is_not_login = logname.is_some() && username != logname.unwrap();
let mut module = context.new_module("username"); let mut module = context.new_module("username");
@ -67,19 +59,22 @@ pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
} }
} }
#[cfg(target_os = "windows")]
fn is_root_user() -> bool {
false
}
#[cfg(not(target_os = "windows"))]
fn is_root_user() -> bool {
let user_uid = nix::unistd::geteuid();
user_uid == nix::unistd::ROOT
}
fn is_ssh_connection(context: &Context) -> bool { fn is_ssh_connection(context: &Context) -> bool {
let ssh_env = ["SSH_CONNECTION", "SSH_CLIENT", "SSH_TTY"]; let ssh_env = ["SSH_CONNECTION", "SSH_CLIENT", "SSH_TTY"];
ssh_env.iter().any(|env| context.get_env(env).is_some()) ssh_env.iter().any(|env| context.get_env(env).is_some())
} }
fn get_uid() -> Option<u32> {
utils::exec_cmd("id", &["-u"])?
.stdout
.trim()
.parse::<u32>()
.ok()
}
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use crate::test::ModuleRenderer; use crate::test::ModuleRenderer;