rebecca-windows 0.2.0

Windows-specific adapters for Rebecca cleanup, discovery, and Recycle Bin execution.
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

use rebecca_core::error::Result;
use rebecca_core::executor::{CleanupBackend, ExecutionOutcome};
use rebecca_core::plan::CleanupTarget;

pub mod apps;
pub mod steam;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum PrivilegeLevel {
    StandardUser,
    Elevated,
    Unknown,
}

pub fn current_privilege_level() -> PrivilegeLevel {
    platform::current_privilege_level()
}

#[derive(Debug, Default, Clone, Copy)]
pub struct WindowsRecycleBinBackend;

impl WindowsRecycleBinBackend {
    pub fn new() -> Self {
        Self
    }
}

impl CleanupBackend for WindowsRecycleBinBackend {
    fn delete(&self, target: &CleanupTarget) -> Result<ExecutionOutcome> {
        platform::delete_to_recycle_bin(&target.path, target.estimated_bytes, target.deletion_style)
    }
}

#[cfg(windows)]
mod platform {
    use std::fs;
    use std::path::Path;

    use rebecca_core::error::{RebeccaError, Result};
    use rebecca_core::executor::ExecutionOutcome;
    use windows::Win32::UI::Shell::IsUserAnAdmin;

    pub fn current_privilege_level() -> super::PrivilegeLevel {
        unsafe {
            if IsUserAnAdmin().as_bool() {
                super::PrivilegeLevel::Elevated
            } else {
                super::PrivilegeLevel::StandardUser
            }
        }
    }

    pub fn delete_to_recycle_bin(
        path: &Path,
        estimated_bytes: u64,
        deletion_style: rebecca_core::CleanupTargetDeletionStyle,
    ) -> Result<ExecutionOutcome> {
        match deletion_style {
            rebecca_core::CleanupTargetDeletionStyle::DeleteWholePath => {
                trash::delete(path)
                    .map_err(|err| RebeccaError::ExecutionFailed(err.to_string()))?;
            }
            rebecca_core::CleanupTargetDeletionStyle::PreserveRootContents => {
                if path.is_dir() {
                    let entries = fs::read_dir(path)?
                        .map(|entry| entry.map(|entry| entry.path()))
                        .collect::<std::io::Result<Vec<_>>>()?;
                    if !entries.is_empty() {
                        trash::delete_all(entries)
                            .map_err(|err| RebeccaError::ExecutionFailed(err.to_string()))?;
                    }
                } else {
                    trash::delete(path)
                        .map_err(|err| RebeccaError::ExecutionFailed(err.to_string()))?;
                }
            }
        }

        Ok(ExecutionOutcome {
            freed_bytes: 0,
            pending_reclaim_bytes: estimated_bytes,
            note: Some("moved to Recycle Bin".to_string()),
        })
    }
}

#[cfg(not(windows))]
mod platform {
    use std::path::Path;

    use rebecca_core::error::{RebeccaError, Result};
    use rebecca_core::executor::ExecutionOutcome;

    pub fn current_privilege_level() -> super::PrivilegeLevel {
        super::PrivilegeLevel::Unknown
    }

    pub fn delete_to_recycle_bin(
        _path: &Path,
        _estimated_bytes: u64,
        _deletion_style: rebecca_core::CleanupTargetDeletionStyle,
    ) -> Result<ExecutionOutcome> {
        Err(RebeccaError::PlatformUnavailable(
            "Windows recycle bin deletion is not available on this platform".to_string(),
        ))
    }
}