grsh/src/ls.rs

use std::env;
use std::fs;

fn main(){
    let args: Vec<String> = env::args().collect();
    let pwd = env::current_dir().unwrap();
    let mut show_hidden = false;
    let mut dir = pwd.to_str();
    for arg in 1..args.len() {
        match args[arg].as_str() {
            "--help" => {
                println!("List files in directory");
            },
            "-a" | "--all" => {
                show_hidden = true;
            },
            arg => {
                dir = Some(arg);
            },
        }
    }
    list_dir(dir.unwrap().to_string(), show_hidden);
}         
fn list_dir(arg: String, show_hidden: bool) {
    let files = fs::read_dir(arg.clone()).unwrap();
    if show_hidden {
        println!(".");
        println!("..");
    }
    for entry in files {
        let path = entry.unwrap().path();
        let file = path.strip_prefix(arg.clone());
        if file.clone().unwrap().display().to_string().starts_with('.') && !show_hidden {
            continue;
        }
        println!("{}", file.unwrap().display());
    }
}