mitsuo_makuda/linutil
1
0
Fork 0
mirror of https://github.com/ChrisTitusTech/linutil.git synced 2024-11-25 14:30:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2 from lj3954/root

Browse Source 
refactor: Reduce nesting in root module
This commit is contained in:
JEEVITHA KANNAN K S 2024-10-26 07:27:46 +05:30 committed by GitHub
commit 8ba2fdd23e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 27 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
tui/src/root.rs
View File

@ -8,62 +8,50 @@ use ratatui::{
};
use std::io;
pub fn check_root(terminal: &mut Terminal<CrosstermBackend<io::Stdout>>) -> io::Result<bool> {
if nix::unistd::geteuid().is_root() {
terminal.draw(|frame| {
let root_warn = Paragraph::new(
r#"
const ROOT_WARNING: &str = r#"
!!!!!!!!!!!!!! YOU ARE ABOUT TO RUN LINUTIL AS ROOT !!!!!!!!!!!!!!
This utility prioritizes compatibility with non-root environments.
Some scripts may work without any issues, some may not.
You have been warned!
!!!!!!!!!!!!!!!!!!!!!! PROCEED WITH CAUTION !!!!!!!!!!!!!!!!!!!!!!
Press [y] to continue, [n] to abort
"#,
)
"#;
pub fn check_root(terminal: &mut Terminal<CrosstermBackend<io::Stdout>>) -> io::Result<bool> {
if !nix::unistd::geteuid().is_root() {
return Ok(true);
}
terminal.draw(|frame| {
let root_warn = Paragraph::new(ROOT_WARNING)
.white()
.on_black()
.alignment(Alignment::Center)
.style(Style::default().bold())
.wrap(Wrap { trim: true });
let rects = Layout::vertical([
Constraint::Fill(1),
Constraint::Length(10),
Constraint::Fill(1),
])
.split(frame.area());
let rects = Layout::vertical([
Constraint::Fill(1),
Constraint::Length(10),
Constraint::Fill(1),
])
.split(frame.area());
let centered = rects[1];
let centered = rects[1];
frame.render_widget(root_warn, centered);
})?;
frame.render_widget(root_warn, centered);
})?;
loop {
match event::read()? {
Event::Key(
KeyEvent {
code: KeyCode::Char('y'),
..
}
| KeyEvent {
code: KeyCode::Char('Y'),
..
},
) => break,
Event::Key(
KeyEvent {
code: KeyCode::Char('n'),
..
}
| KeyEvent {
code: KeyCode::Char('N'),
..
},
) => return Ok(false),
loop {
if let Event::Key(KeyEvent {
code: KeyCode::Char(ch),
..
}) = event::read()?
{
match ch.to_ascii_lowercase() {
'y' => return Ok(true),
'n' => return Ok(false),
_ => {}
}
}
}
Ok(true)
}