self-replace 1.5.0

Utility crate that allows executables to replace or uninstall themselves
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use std::env;
use std::fs;
use std::io;
use std::path::Path;

/// On Unix a running executable can be safely deleted.
pub fn self_delete(exe: &Path) -> Result<(), io::Error> {
    let exe = exe.canonicalize()?;
    fs::remove_file(exe)?;
    Ok(())
}

pub fn self_replace(new_executable: &Path) -> Result<(), io::Error> {
    let mut exe = env::current_exe()?;
    if fs::symlink_metadata(&exe).map_or(false, |x| x.file_type().is_symlink()) {
        exe = fs::read_link(exe)?;
    }
    let old_permissions = exe.metadata()?.permissions();

    let prefix = if let Some(hint) = exe.file_stem().and_then(|x| x.to_str()) {
        format!(".{}.__temp__", hint)
    } else {
        ".__temp__".into()
    };

    let tmp = tempfile::Builder::new()
        .prefix(&prefix)
        .tempfile_in(exe.parent().ok_or_else(|| {
            io::Error::new(
                io::ErrorKind::Other,
                "executable has no known parent folder",
            )
        })?)?;
    fs::copy(new_executable, tmp.path())?;
    fs::set_permissions(tmp.path(), old_permissions)?;

    // if we made it this far, try to persist the temporary file and move it over.
    let (_, path) = tmp.keep()?;
    match fs::rename(&path, &exe) {
        Ok(()) => {}
        Err(err) => {
            fs::remove_file(&path).ok();
            return Err(err);
        }
    }

    Ok(())
}