atuin-common 18.23.0

common library for atuin
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

#![allow(unsafe_code)]

use std::io;
use std::os::windows::ffi::OsStrExt;
use std::path::Path;

use windows_sys::Win32::Storage::FileSystem::GetDiskFreeSpaceExW;

use super::fallible_do;
use crate::units::ByteSize;

/// How much total space does the disk behind `path` have.
pub fn total_space(path: &Path) -> io::Result<ByteSize> {
    let directory: Vec<u16> = path.as_os_str().encode_wide().chain(std::iter::once(0)).collect();

    let mut total: u64 = 0;
    fallible_do(|| unsafe {
        // Classic win32 misnomer -- this returns both the total and available space.
        GetDiskFreeSpaceExW(
            directory.as_ptr(),
            std::ptr::null_mut(),
            &mut total,
            std::ptr::null_mut(),
        )
    })?;

    Ok(ByteSize::b(total))
}