mirror of
https://github.com/tw93/Mole.git
synced 2026-02-04 19:09:43 +00:00
301 lines
8.2 KiB
PowerShell
301 lines
8.2 KiB
PowerShell
# Mole - Clean Command
|
|
# Deep cleanup for Windows with dry-run support and whitelist
|
|
|
|
#Requires -Version 5.1
|
|
[CmdletBinding()]
|
|
param(
|
|
[switch]$DryRun,
|
|
[switch]$System,
|
|
[switch]$DebugMode,
|
|
[switch]$Whitelist,
|
|
[switch]$ShowHelp
|
|
)
|
|
|
|
Set-StrictMode -Version Latest
|
|
$ErrorActionPreference = "Stop"
|
|
|
|
# Script location
|
|
$scriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
|
|
$libDir = Join-Path (Split-Path -Parent $scriptDir) "lib"
|
|
|
|
# Import core modules
|
|
. "$libDir\core\base.ps1"
|
|
. "$libDir\core\log.ps1"
|
|
. "$libDir\core\ui.ps1"
|
|
. "$libDir\core\file_ops.ps1"
|
|
|
|
# Import cleanup modules
|
|
. "$libDir\clean\user.ps1"
|
|
. "$libDir\clean\caches.ps1"
|
|
. "$libDir\clean\dev.ps1"
|
|
. "$libDir\clean\apps.ps1"
|
|
. "$libDir\clean\system.ps1"
|
|
|
|
# ============================================================================
|
|
# Configuration
|
|
# ============================================================================
|
|
|
|
$script:ExportListFile = "$env:USERPROFILE\.config\mole\clean-list.txt"
|
|
|
|
# ============================================================================
|
|
# Help
|
|
# ============================================================================
|
|
|
|
function Show-CleanHelp {
|
|
$esc = [char]27
|
|
Write-Host ""
|
|
Write-Host "$esc[1;35mMole Clean$esc[0m - Deep cleanup for Windows"
|
|
Write-Host ""
|
|
Write-Host "$esc[33mUsage:$esc[0m mole clean [options]"
|
|
Write-Host ""
|
|
Write-Host "$esc[33mOptions:$esc[0m"
|
|
Write-Host " -DryRun Preview changes without deleting (recommended first run)"
|
|
Write-Host " -System Include system-level cleanup (requires admin)"
|
|
Write-Host " -Whitelist Manage protected paths"
|
|
Write-Host " -DebugMode Enable debug logging"
|
|
Write-Host " -ShowHelp Show this help message"
|
|
Write-Host ""
|
|
Write-Host "$esc[33mExamples:$esc[0m"
|
|
Write-Host " mole clean -DryRun # Preview what would be cleaned"
|
|
Write-Host " mole clean # Run standard cleanup"
|
|
Write-Host " mole clean -System # Include system cleanup (as admin)"
|
|
Write-Host ""
|
|
}
|
|
|
|
# ============================================================================
|
|
# Whitelist Management
|
|
# ============================================================================
|
|
|
|
function Edit-Whitelist {
|
|
$whitelistPath = $script:Config.WhitelistFile
|
|
$whitelistDir = Split-Path -Parent $whitelistPath
|
|
|
|
# Ensure directory exists
|
|
if (-not (Test-Path $whitelistDir)) {
|
|
New-Item -ItemType Directory -Path $whitelistDir -Force | Out-Null
|
|
}
|
|
|
|
# Create default whitelist if doesn't exist
|
|
if (-not (Test-Path $whitelistPath)) {
|
|
$defaultContent = @"
|
|
# Mole Whitelist - Paths listed here will never be cleaned
|
|
# Use full paths or patterns with wildcards (*)
|
|
#
|
|
# Examples:
|
|
# C:\Users\YourName\Documents\ImportantProject
|
|
# C:\Users\*\AppData\Local\MyApp
|
|
# $env:LOCALAPPDATA\CriticalApp
|
|
#
|
|
# Add your protected paths below:
|
|
|
|
"@
|
|
Set-Content -Path $whitelistPath -Value $defaultContent
|
|
}
|
|
|
|
# Open in default editor
|
|
Write-Info "Opening whitelist file: $whitelistPath"
|
|
Start-Process notepad.exe -ArgumentList $whitelistPath -Wait
|
|
|
|
Write-Success "Whitelist saved"
|
|
}
|
|
|
|
# ============================================================================
|
|
# Cleanup Summary
|
|
# ============================================================================
|
|
|
|
function Show-CleanupSummary {
|
|
param(
|
|
[hashtable]$Stats,
|
|
[bool]$IsDryRun
|
|
)
|
|
|
|
$esc = [char]27
|
|
|
|
Write-Host ""
|
|
Write-Host "$esc[1;35m" -NoNewline
|
|
if ($IsDryRun) {
|
|
Write-Host "Dry run complete - no changes made" -NoNewline
|
|
}
|
|
else {
|
|
Write-Host "Cleanup complete" -NoNewline
|
|
}
|
|
Write-Host "$esc[0m"
|
|
Write-Host ""
|
|
|
|
if ($Stats.TotalSizeKB -gt 0) {
|
|
$sizeGB = [Math]::Round($Stats.TotalSizeKB / 1024 / 1024, 2)
|
|
|
|
if ($IsDryRun) {
|
|
Write-Host " Potential space: $esc[32m${sizeGB}GB$esc[0m"
|
|
Write-Host " Items found: $($Stats.FilesCleaned)"
|
|
Write-Host " Categories: $($Stats.TotalItems)"
|
|
Write-Host ""
|
|
Write-Host " Detailed list: $esc[90m$($script:ExportListFile)$esc[0m"
|
|
Write-Host " Run without -DryRun to apply cleanup"
|
|
}
|
|
else {
|
|
Write-Host " Space freed: $esc[32m${sizeGB}GB$esc[0m"
|
|
Write-Host " Items cleaned: $($Stats.FilesCleaned)"
|
|
Write-Host " Categories: $($Stats.TotalItems)"
|
|
Write-Host ""
|
|
Write-Host " Free space now: $(Get-FreeSpace)"
|
|
}
|
|
}
|
|
else {
|
|
if ($IsDryRun) {
|
|
Write-Host " No significant reclaimable space detected."
|
|
}
|
|
else {
|
|
Write-Host " System was already clean; no additional space freed."
|
|
}
|
|
Write-Host " Free space now: $(Get-FreeSpace)"
|
|
}
|
|
|
|
Write-Host ""
|
|
}
|
|
|
|
# ============================================================================
|
|
# Main Cleanup Flow
|
|
# ============================================================================
|
|
|
|
function Start-Cleanup {
|
|
param(
|
|
[bool]$IsDryRun,
|
|
[bool]$IncludeSystem
|
|
)
|
|
|
|
$esc = [char]27
|
|
|
|
# Clear screen
|
|
Clear-Host
|
|
Write-Host ""
|
|
Write-Host "$esc[1;35mClean Your Windows$esc[0m"
|
|
Write-Host ""
|
|
|
|
# Show mode
|
|
if ($IsDryRun) {
|
|
Write-Host "$esc[33mDry Run Mode$esc[0m - Preview only, no deletions"
|
|
Write-Host ""
|
|
|
|
# Prepare export file
|
|
$exportDir = Split-Path -Parent $script:ExportListFile
|
|
if (-not (Test-Path $exportDir)) {
|
|
New-Item -ItemType Directory -Path $exportDir -Force | Out-Null
|
|
}
|
|
|
|
$header = @"
|
|
# Mole Cleanup Preview - $(Get-Date -Format 'yyyy-MM-dd HH:mm:ss')
|
|
#
|
|
# How to protect files:
|
|
# 1. Copy any path below to $($script:Config.WhitelistFile)
|
|
# 2. Run: mole clean -Whitelist
|
|
#
|
|
|
|
"@
|
|
Set-Content -Path $script:ExportListFile -Value $header
|
|
}
|
|
else {
|
|
Write-Host "$esc[90m$($script:Icons.Solid) Use -DryRun to preview, -Whitelist to manage protected paths$esc[0m"
|
|
Write-Host ""
|
|
}
|
|
|
|
# System cleanup confirmation
|
|
if ($IncludeSystem -and -not $IsDryRun) {
|
|
if (-not (Test-IsAdmin)) {
|
|
Write-MoleWarning "System cleanup requires administrator privileges"
|
|
Write-Host " Run PowerShell as Administrator for full cleanup"
|
|
Write-Host ""
|
|
$IncludeSystem = $false
|
|
}
|
|
else {
|
|
Write-Host "$esc[32m$($script:Icons.Success)$esc[0m Running with Administrator privileges"
|
|
Write-Host ""
|
|
}
|
|
}
|
|
|
|
# Show system info
|
|
$winVer = Get-WindowsVersion
|
|
Write-Host "$esc[34m$($script:Icons.Admin)$esc[0m $($winVer.Name) | Free space: $(Get-FreeSpace)"
|
|
Write-Host ""
|
|
|
|
# Reset stats
|
|
Reset-CleanupStats
|
|
Set-DryRunMode -Enabled $IsDryRun
|
|
|
|
# Run cleanup modules
|
|
try {
|
|
# User essentials (temp, logs, etc.)
|
|
Invoke-UserCleanup -TempDaysOld 7 -LogDaysOld 7
|
|
|
|
# Browser caches
|
|
Clear-BrowserCaches
|
|
|
|
# Application caches
|
|
Clear-AppCaches
|
|
|
|
# Developer tools
|
|
Invoke-DevToolsCleanup
|
|
|
|
# Applications cleanup
|
|
Invoke-AppCleanup
|
|
|
|
# System cleanup (if requested and admin)
|
|
if ($IncludeSystem -and (Test-IsAdmin)) {
|
|
Invoke-SystemCleanup
|
|
}
|
|
}
|
|
catch {
|
|
Write-MoleError "Cleanup error: $_"
|
|
}
|
|
|
|
# Get final stats
|
|
$stats = Get-CleanupStats
|
|
|
|
# Show summary
|
|
Show-CleanupSummary -Stats $stats -IsDryRun $IsDryRun
|
|
}
|
|
|
|
# ============================================================================
|
|
# Main Entry Point
|
|
# ============================================================================
|
|
|
|
function Main {
|
|
# Enable debug if requested
|
|
if ($DebugMode) {
|
|
$env:MOLE_DEBUG = "1"
|
|
$DebugPreference = "Continue"
|
|
}
|
|
|
|
# Show help
|
|
if ($ShowHelp) {
|
|
Show-CleanHelp
|
|
return
|
|
}
|
|
|
|
# Manage whitelist
|
|
if ($Whitelist) {
|
|
Edit-Whitelist
|
|
return
|
|
}
|
|
|
|
# Set dry-run mode
|
|
if ($DryRun) {
|
|
$env:MOLE_DRY_RUN = "1"
|
|
}
|
|
else {
|
|
$env:MOLE_DRY_RUN = "0"
|
|
}
|
|
|
|
# Run cleanup
|
|
try {
|
|
Start-Cleanup -IsDryRun $DryRun -IncludeSystem $System
|
|
}
|
|
finally {
|
|
# Cleanup temp files
|
|
Clear-TempFiles
|
|
}
|
|
}
|
|
|
|
# Run main
|
|
Main
|