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
49
50
51
52
53

use std::ffi::OsStr;
use std::fmt;
use std::path::Path;

use heim_common::prelude::*;

use crate::{sys, FileSystem};

/// Mounted disk partition.
#[derive(heim_derive::ImplWrap)]
pub struct Partition(sys::Partition);

impl Partition {
    pub fn device(&self) -> Option<&OsStr> {
        self.as_ref().device()
    }

    pub fn mount_point(&self) -> &Path {
        self.as_ref().mount_point()
    }

    /// Returns partition file system.
    pub fn file_system(&self) -> &FileSystem {
        self.as_ref().file_system()
    }
}

impl fmt::Debug for Partition {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        f.debug_struct("Partition")
            .field("device", &self.device())
            .field("mount_point", &self.mount_point())
            .field("file_system", &self.file_system())
            .finish()
    }
}

/// Returns stream which yields mounted disk [Partitions].
///
/// This includes all virtual partitions, such as `tmpfs`.
/// See [partitions_physical] for physical partitions stream.
///
/// [Partitions]: struct.Partition.html
pub fn partitions() -> impl Stream<Item = Result<Partition>> {
    sys::partitions().map_ok(Into::into)
}

/// Returns stream which yields physical only mounted disk [Partitions].
///
/// [Partitions]: struct.Partition.html
pub fn partitions_physical() -> impl Stream<Item = Result<Partition>> {
    sys::partitions_physical().map_ok(Into::into)
}