From c012264363d026e93c7683c079095d452c3eee80 Mon Sep 17 00:00:00 2001 From: pbspbsingh Date: Sun, 23 Oct 2022 13:32:57 -0700 Subject: Replace hardcoded loop with actual reading of directory and picking the pattern files --- src/ui.rs | 27 +++++++++++++++++++++------ 1 file changed, 21 insertions(+), 6 deletions(-) (limited to 'src/ui.rs') diff --git a/src/ui.rs b/src/ui.rs index f9ddbb9..73a345f 100644 --- a/src/ui.rs +++ b/src/ui.rs @@ -82,14 +82,29 @@ pub struct App { impl App { pub fn new() -> App { - let mut v = vec![]; - for i in 1..=513 { - let name = format!("pattern{}", i); - let p = format!("./presets/patterns/pattern{}.txt", i); - v.push((name.to_owned(), Path::new(p.as_str()).to_owned())); + fn read_presets() -> io::Result> { + let mut result = Vec::new(); + for path in std::fs::read_dir("./presets/patterns")? { + let path = path?.path(); + if !path.is_file() { + continue; + } + if let Some(file_name) = path.file_name() { + let file_name = file_name.to_string_lossy(); + if file_name.starts_with("pattern") && file_name.ends_with(".txt") { + result.push((file_name.trim_end_matches(".txt").to_owned(), path)); + } + } + } + result.sort_by_key(|(name, _)| { + name.trim_start_matches("pattern") + .parse::() + .unwrap_or(0) + }); + Ok(result) } App { - items: StatefulList::with_items(v), + items: StatefulList::with_items(read_presets().unwrap_or_else(|_| Vec::new())), flag_cur: false, layout: Layout::default() .direction(Direction::Horizontal) -- cgit v1.2.3