forked from mirrors/rmenu
65 lines
1.7 KiB
Rust
65 lines
1.7 KiB
Rust
use std::env;
|
|
use std::os::unix::fs::PermissionsExt;
|
|
|
|
use rmenu_plugin::Entry;
|
|
use walkdir::{DirEntry, WalkDir};
|
|
|
|
static PATH: &'static str = "PATH";
|
|
static DEFAULT_PATH: &'static str = "/bin:/usr/bin:/usr/sbin";
|
|
static EXEC_FLAG: u32 = 0o111;
|
|
|
|
/// Retrieve Search Paths from OS-VAR or Default
|
|
fn bin_paths() -> Vec<String> {
|
|
env::var(PATH)
|
|
.unwrap_or_else(|_| DEFAULT_PATH.to_string())
|
|
.split(":")
|
|
.map(|s| s.to_string())
|
|
.collect()
|
|
}
|
|
|
|
/// Ignore Entry if Hidden or Filename contains a `.`
|
|
fn should_ignore(entry: &DirEntry) -> bool {
|
|
entry
|
|
.file_name()
|
|
.to_str()
|
|
.map(|s| s.contains("."))
|
|
.unwrap_or(false)
|
|
}
|
|
|
|
/// Retrieve Binaries for the Specified Paths
|
|
fn find_binaries(path: String) -> Vec<Entry> {
|
|
WalkDir::new(path)
|
|
.follow_links(true)
|
|
.into_iter()
|
|
.filter_entry(|e| !should_ignore(e))
|
|
.filter_map(|e| e.ok())
|
|
.filter(|e| e.file_type().is_file())
|
|
.filter(|e| {
|
|
e.metadata()
|
|
.map(|m| m.permissions().mode() & EXEC_FLAG != 0)
|
|
.unwrap_or(false)
|
|
})
|
|
.map(|e| {
|
|
let path = e.path().to_string_lossy();
|
|
Entry::new(&e.file_name().to_string_lossy(), &path, Some(&path))
|
|
})
|
|
.collect()
|
|
}
|
|
|
|
fn main() {
|
|
// collect entries for sorting
|
|
let mut entries: Vec<Entry> = bin_paths()
|
|
.into_iter()
|
|
.map(find_binaries)
|
|
.flatten()
|
|
.collect();
|
|
// sort entries and render to json
|
|
entries.sort_by_cached_key(|e| e.name.clone());
|
|
entries
|
|
.into_iter()
|
|
.map(|e| serde_json::to_string(&e))
|
|
.filter_map(|r| r.ok())
|
|
.map(|s| println!("{}", s))
|
|
.last();
|
|
}
|