pyapp 0.29.0

Runtime installer for Python applications
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

use std::fs;
use std::path::PathBuf;

use anyhow::{Context, Result};
use fs4::fs_std::FileExt;

use crate::terminal;

pub fn move_temp_file(temp_file: &PathBuf, destination: &PathBuf) -> Result<()> {
    if fs::rename(temp_file, destination).is_err() {
        // If on different devices / mountpoints, copy the file, then remove the temp file
        fs::copy(temp_file, destination).with_context(|| {
            format!(
                "unable to copy {} to {}",
                temp_file.display(),
                destination.display()
            )
        })?;
        fs::remove_file(temp_file)
            .with_context(|| format!("unable to remove temporary file {}", temp_file.display()))?;
    }

    Ok(())
}

pub fn acquire_lock(file_path: &PathBuf) -> Result<fs::File> {
    let locks_dir = file_path.parent().unwrap();
    fs::create_dir_all(locks_dir)
        .with_context(|| format!("unable to create lock directory {}", &locks_dir.display()))?;

    let lock_file = fs::OpenOptions::new()
        .read(true)
        .write(true)
        .create(true)
        .truncate(true)
        .open(file_path)
        .with_context(|| format!("unable to open lock file {}", file_path.display()))?;

    if lock_file.try_lock_exclusive().is_err() {
        let spinner = terminal::spinner("Waiting on shared resource".to_string());
        let result = lock_file.lock_exclusive();
        spinner.finish_and_clear();
        result.with_context(|| format!("unable to acquire lock file {}", file_path.display()))?;
    }

    Ok(lock_file)
}