makepad-shell 1.0.0

use std::{
    path::{Path, PathBuf},
    fs::File,
    fs,
    io::prelude::*,
    io::BufReader,
    process::{Command, Stdio, Child}
};

pub fn shell(cwd: &Path, cmd: &str, args: &[&str]) -> Result<(), String> {
    let mut cmd_build = Command::new(cmd);
    
    cmd_build.args(args)
        .current_dir(cwd);
    
    let mut child = cmd_build.spawn().map_err( | e | format!("Error starting {} in dir {:?} - {:?}", cmd, cwd, e)) ?;
    
    let r = child.wait().map_err( | e | format!("Process {} in dir {:?} returned error {:?} ", cmd, cwd, e)) ?;
    if !r.success() {
        return Err(format!("Process {} in dir {:?} returned error exit code {:?}", cmd, cwd, r.code()));
    }
    Ok(())
} 

pub fn shell_env(env: &[(&str, &str)], cwd: &Path, cmd: &str, args: &[&str]) -> Result<(), String> {
    let mut cmd_build = Command::new(cmd);
    
    cmd_build.args(args)
        .current_dir(cwd);
        
    for (key, value) in env {
        cmd_build.env(key, value);
    }
    let mut child = cmd_build.spawn().map_err( | e | format!("Error starting {} in dir {:?} - {:?}", cmd, cwd, e)) ?;
    
    let r = child.wait().map_err( | e | format!("Process {} in dir {:?} returned error {:?} ", cmd, cwd, e)) ?;
    if !r.success() {
        return Err(format!("Process {} in dir {:?} returned error exit code {:?}", cmd, cwd, r.code()));
    }
    Ok(())
}

pub fn shell_env_cap(env: &[(&str, &str)], cwd: &Path, cmd: &str, args: &[&str]) -> Result<String, String> {
    let mut cmd_build = Command::new(cmd);
    
    cmd_build.args(args)
        .stdout(Stdio::piped())
        .stderr(Stdio::piped())
        .current_dir(cwd);
        
    for (key, value) in env {
        cmd_build.env(key, value);
    }
    let child = cmd_build.spawn().map_err( | e | format!("Error starting {} in dir {:?} - {:?}", cmd, cwd, e)) ?;
    let r = child.wait_with_output().map_err( | e | {
        format!("Process {} in dir {:?} returned error {:?} ", cmd, cwd, e)
    }) ?;
    let stderr = std::str::from_utf8(&r.stderr).unwrap_or("could not decode utf8");
    let stdout = std::str::from_utf8(&r.stdout).unwrap_or("could not decode utf8");
    if !r.status.success() {
        return Err(format!("Process {} in dir {:?} returned error exit code {}\n{}\n{}", cmd, cwd, r.status, stderr, stdout));
    }
    Ok(format!("{}{}", stdout, stderr))
}

pub struct ShellChild{
    child: Child,
    cwd: PathBuf,
    cmd: String,
}

pub fn shell_child_create(env: &[(&str, &str)], cwd: &Path, cmd: &str, args: &[&str]) -> Result<ShellChild, String> {
    let mut cmd_build = Command::new(cmd);
        
    cmd_build.args(args)
    .stdout(Stdio::piped())
    .stderr(Stdio::piped())
    .current_dir(cwd);
            
    for (key, value) in env {
        cmd_build.env(key, value);
    }
    let child = cmd_build.spawn().map_err( | e | format!("Error starting {} in dir {:?} - {:?}", cmd, cwd, e)) ?;
    Ok(ShellChild{
        child,
        cwd: cwd.into(),
        cmd: cmd.into()
    })
}


pub fn shell_child_wait(child:ShellChild)-> Result<String, String> {
    let r = child.child.wait_with_output().map_err( | e | {
        format!("Process {} in dir {:?} returned error {:?} ", child.cmd, child.cwd, e)
    }) ?;
    let stderr = std::str::from_utf8(&r.stderr).unwrap_or("could not decode utf8");
    let stdout = std::str::from_utf8(&r.stdout).unwrap_or("could not decode utf8");
    if !r.status.success() {
        return Err(format!("Process {} in dir {:?} returned error exit code {}\n{}\n{}", child.cmd, child.cwd, r.status, stderr, stdout));
    }
    Ok(format!("{}{}", stdout, stderr))
}

pub fn shell_env_cap_split(env: &[(&str, &str)], cwd: &Path, cmd: &str, args: &[&str]) -> (String, String, bool) {
    let mut cmd_build = Command::new(cmd);
        
    cmd_build.args(args)
    .stdout(Stdio::piped())
    .stderr(Stdio::piped())
    .current_dir(cwd);
            
    for (key, value) in env {
        cmd_build.env(key, value);
    }
    let child = cmd_build.spawn(); 
    if let Err(e) = child{
        return ("".to_string(),format!("Cannot start process {}", e), false);
    }
    let r = child.unwrap().wait_with_output();
    if let Err(e) = r{
        return ("".to_string(),format!("Wait with output failed for process {}", e), false);
    }
    let r = r.unwrap();
    let stderr = std::str::from_utf8(&r.stderr).unwrap_or("could not decode utf8").to_string();
    let stdout = std::str::from_utf8(&r.stdout).unwrap_or("could not decode utf8").to_string();
    (stdout, stderr, r.status.success())
}

pub fn shell_env_filter(start:&str, minus:Vec<String>, env: &[(&str, &str)], cwd: &Path, cmd: &str,  args: &[&str]) -> Result<(), String> {

    let mut cmd_build = Command::new(cmd);
    
    cmd_build.args(args)
        .stdin(Stdio::piped())
        .stdout(Stdio::piped())
        .current_dir(cwd);
    
    for (key, value) in env {
        cmd_build.env(key, value);
    }
    
    let mut child = cmd_build.spawn().map_err( | e | format!("Error starting {} in dir {:?} - {:?}", cmd, cwd, e)) ?;
    
    let stdout = child.stdout.take().expect("stdout cannot be taken!");
    let start = start.to_string();
    let _stdout_thread = {
        std::thread::spawn(move || {
            let mut reader = BufReader::new(stdout);
            let mut output = false;
            'a: loop{
                let mut line = String::new();
                if let Ok(_) = reader.read_line(&mut line){
                    if line.contains(&start){
                        output = true;
                    }
                    if output{
                        for min in &minus{
                            if line.contains(min){
                                continue 'a;
                            }
                        }
                        println!("{}",line);
                    }
                }
            }
        })
    };

    let r = child.wait().map_err( | e | format!("Process {} in dir {:?} returned error {:?} ", cmd, cwd, e)) ?;
    if !r.success() {
        return Err(format!("Process {} in dir {:?} returned error exit code {} ", cmd, cwd, r));
    }
    Ok(())
}

pub fn write_text(path: &Path, data:&str) -> Result<(), String> {
    mkdir(path.parent().unwrap()) ?;
    match fs::File::create(path) { 
        Err(e) => {
            Err(format!("file create {:?} failed {:?}", path, e))
        },
        Ok(mut f) =>{
            f.write_all(data.as_bytes())
                .map_err( | _e | format!("Cant write file {:?}", path))
        }
    }
}

pub fn rmdir(path: &Path) -> Result<(), String> {
    match fs::remove_dir_all(path) {
        Err(e) => {
            Err(format!("rmdir {:?} failed {:?}", path, e))
        },
        Ok(()) => Ok(())
    }
}


pub fn mkdir(path: &Path) -> Result<(), String> {
    match fs::create_dir_all(path) { 
        Err(e) => {
            Err(format!("mkdir {:?} failed {:?}", path, e))
        },
        Ok(()) => Ok(())
    }
}

pub fn rm(path: &Path) -> Result<(), String> {
    match fs::remove_file(path) {
        Err(e) => {
            Err(format!("remove_file {:?} failed {:?}", path, e))
        },
        Ok(()) => Ok(())
    }
}


#[allow(unused)]
pub fn cp(source_path: &Path, dest_path: &Path, exec: bool) -> Result<(), String> {
    let data = fs::read(source_path)
        .map_err( | _e | format!("Cant open input file {:?}", source_path)) ?;
    mkdir(dest_path.parent().unwrap()) ?;
    let mut output = File::create(dest_path)
        .map_err( | _e | format!("Cant open output file {:?}", dest_path)) ?;
    output.write(&data)
        .map_err( | _e | format!("Cant write output file {:?}", dest_path)) ?;
    #[cfg(any(target_os = "macos", target_os = "linux"))]
    if exec {
        use std::os::unix::fs::PermissionsExt;
        std::fs::set_permissions(dest_path, PermissionsExt::from_mode(0o744))
            .map_err( | _e | format!("Cant set exec permissions on output file {:?}", dest_path)) ?;
    }
    Ok(())
}

pub fn cp_all(source_dir: &Path, dest_dir: &Path, exec: bool) -> Result<(), String> {
    cp_all_recursive(source_dir, dest_dir, exec)?;
    Ok(())
}

fn cp_all_recursive(source_dir: &Path, dest_dir: &Path, exec: bool) -> Result<(), String> {
    if !source_dir.is_dir() {
        return Err(format!("{:?} is not a directory", source_dir));
    }

    mkdir(dest_dir) ?;

    for entry in fs::read_dir(source_dir).map_err(|_e| format!("Unable to read source directory {:?}", source_dir))? {
        let entry = entry.map_err(|_e| format!("Unable to process directory entry"))?;
        let source_path = entry.path();
        if source_path.is_file() {
            let dest_path = dest_dir.join(source_path.file_name()
                .ok_or_else(|| format!("Unable to get filename for {:?}", source_path))?);

            cp(&source_path, &dest_path, exec)?;
        } else if source_path.is_dir() {
            let dest_path = dest_dir.join(source_path.file_name()
                    .ok_or_else(|| format!("Unable to get folder name for {:?}", source_path))?);

            cp_all_recursive(&source_path, &dest_path, exec)?;
        }
    }

    Ok(())
}


pub fn walk_all(source_dir: &Path, dest_dir: &Path, cb:&mut dyn Fn(&Path, &Path)-> Result<(), String>) -> Result<(), String> {
    walk_all_recursive(source_dir, dest_dir, cb)?;
    Ok(())
}

fn walk_all_recursive(source_dir: &Path, dest_dir: &Path, cb:&mut dyn Fn(&Path, &Path)-> Result<(), String>) -> Result<(), String> {
    if !source_dir.is_dir() {
        return Err(format!("{:?} is not a directory", source_dir));
    }
    
    mkdir(dest_dir) ?;
    
    for entry in fs::read_dir(source_dir).map_err(|_e| format!("Unable to read source directory {:?}", source_dir))? {
        let entry = entry.map_err(|_e| format!("Unable to process directory entry"))?;
        let source_path = entry.path();
        if source_path.is_file() {
            cb(&source_path, dest_dir)?;
            /*let source_file_name = source_path.file_name().ok_or_else(|| format!("Unable to get filename for {:?}", source_path))?.to_string_lossy().to_string();
            let source_path2 = if let Some(tgt) = rename.get(&source_file_name){
                //println!("RENAMING {} {}", source_file_name, tgt);
                source_path.parent().unwrap().join(tgt)
            }
            else{
                source_path
            };
            let dest_path = dest_dir.join(source_file_name);
            cp(&source_path2, &dest_path, exec)?;*/
        } else if source_path.is_dir() {
            let dest_path = dest_dir.join(source_path.file_name()
            .ok_or_else(|| format!("Unable to get folder name for {:?}", source_path))?);
            
            walk_all_recursive(&source_path, &dest_path, cb)?;
        }
    }
    
    Ok(())
}

pub fn ls(dir: &Path) -> Result<Vec<PathBuf>, String> {
    let mut result = Vec::new();
    ls_recursive(dir, dir, &mut result)?;
    Ok(result)
}

fn ls_recursive(dir: &Path, prefix: &Path, result: &mut Vec<PathBuf>) -> Result<(), String> {
    for entry in fs::read_dir(dir).map_err(|_e| format!("Unable to read source directory {:?}", dir))? {
        let entry = entry.map_err(|_e| format!("Unable to process directory entry"))?;
        let source_path = entry.path();
        if source_path.is_file() {
            let file_name = source_path
                .file_name()
                .ok_or_else(|| format!("Unable to get filename"))?
                .to_str()
                .ok_or_else(|| format!( "Unable to convert filename to str"))?;

            if !file_name.starts_with('.') {
                let result_path = source_path.strip_prefix(prefix)
                    .map_err(|_e| format!("Unable to strip prefix"))?;
                result.push(result_path.to_path_buf());
            }
        } else if source_path.is_dir() {
            ls_recursive(&source_path, prefix, result)?;
        }
    }

    Ok(())
}