# 查看當(dāng)前目錄下所有子文件夾的大小
Get-ChildItem -Directory | ForEach-Object {
    $folder = $_
    # 遞歸獲取所有文件并計(jì)算大小總和
    $size = Get-ChildItem $_.FullName -Recurse -File | 
             Measure-Object -Property Length -Sum | 
             Select-Object -ExpandProperty Sum
    
    # 如果沒有文件，大小為0
    if (-not $size) { $size = 0 }
    
    # 創(chuàng)建自定義對(duì)象返回結(jié)果
    [PSCustomObject]@{
        FolderName = $folder.Name
        Size_Bytes = $size
        Size_MB    = [math]::Round($size / 1MB, 2)
        Size_GB    = [math]::Round($size / 1GB, 2)
    }
}

function Get-BasicFolderSize {
    [CmdletBinding()]
    param(
        [Parameter(Position=0)]
        [string]$Path = ".",
        
        [ValidateSet("KB", "MB", "GB")]
        [string]$Unit = "MB"
    )
    
    # 驗(yàn)證路徑是否存在
    if (-not (Test-Path $Path)) {
        Write-Error "路徑 '$Path' 不存在"
        return
    }
    
    # 獲取目標(biāo)路徑信息
    $targetPath = Get-Item $Path
    
    # 如果是文件，直接返回文件大小
    if (-not $targetPath.PSIsContainer) {
        $size = $targetPath.Length
        $formattedSize = switch ($Unit) {
            "KB" { [math]::Round($size / 1KB, 2); break }
            "MB" { [math]::Round($size / 1MB, 2); break }
            "GB" { [math]::Round($size / 1GB, 2); break }
        }
        
        return [PSCustomObject]@{
            Name = $targetPath.Name
            Path = $targetPath.FullName
            "Size($Unit)" = $formattedSize
            Bytes = $size
            Type = "File"
        }
    }
    
    # 如果是文件夾，計(jì)算其大小
    $results = @()
    $folders = Get-ChildItem -Path $Path -Directory -ErrorAction SilentlyContinue
    
    foreach ($folder in $folders) {
        Write-Verbose "正在處理文件夾: $($folder.FullName)"
        
        try {
            $files = Get-ChildItem -Path $folder.FullName -Recurse -File -ErrorAction Stop
            $size = ($files | Measure-Object -Property Length -Sum).Sum
            
            if (-not $size) { $size = 0 }
            
            $formattedSize = switch ($Unit) {
                "KB" { [math]::Round($size / 1KB, 2); break }
                "MB" { [math]::Round($size / 1MB, 2); break }
                "GB" { [math]::Round($size / 1GB, 2); break }
            }
            
            $results += [PSCustomObject]@{
                Name = $folder.Name
                Path = $folder.FullName
                "Size($Unit)" = $formattedSize
                Bytes = $size
                Type = "Folder"
            }
        }
        catch {
            Write-Warning "無法訪問文件夾: $($folder.FullName)"
            Write-Warning "錯(cuò)誤信息: $_"
        }
    }
    
    # 按大小降序排序
    return $results | Sort-Object Bytes -Descending
}

# 使用示例
Get-BasicFolderSize -Path "C:\Users" -Unit GB -Verbose

function Get-OptimizedFolderSize {
    [CmdletBinding()]
    param(
        [string]$Path = ".",
        [switch]$IncludeHidden,
        [switch]$IncludeSystem
    )
    
    # 創(chuàng)建文件系統(tǒng)枚舉選項(xiàng)
    $enumerationOptions = [System.IO.SearchOption]::AllDirectories
    
    # 獲取文件夾信息
    $directoryInfo = New-Object System.IO.DirectoryInfo($Path)
    
    # 如果文件夾不存在
    if (-not $directoryInfo.Exists) {
        Write-Error "文件夾 '$Path' 不存在"
        return
    }
    
    $results = @()
    $subDirectories = $directoryInfo.GetDirectories()
    
    foreach ($subDir in $subDirectories) {
        # 檢查是否需要跳過隱藏/系統(tǒng)文件夾
        $attributes = $subDir.Attributes
        if ((-not $IncludeHidden) -and ($attributes -band [System.IO.FileAttributes]::Hidden)) {
            continue
        }
        if ((-not $IncludeSystem) -and ($attributes -band [System.IO.FileAttributes]::System)) {
            continue
        }
        
        $totalSize = 0
        $fileCount = 0
        $folderCount = 0
        
        try {
            # 使用EnumerateFiles進(jìn)行更高效的遍歷
            $files = [System.IO.Directory]::EnumerateFiles(
                $subDir.FullName, 
                "*.*", 
                [System.IO.SearchOption]::AllDirectories
            )
            
            foreach ($file in $files) {
                try {
                    $fileInfo = New-Object System.IO.FileInfo($file)
                    $totalSize += $fileInfo.Length
                    $fileCount++
                }
                catch {
                    # 跳過無法訪問的文件
                }
            }
            
            # 計(jì)算子文件夾數(shù)量（不遞歸計(jì)算，避免性能開銷）
            $folderCount = [System.IO.Directory]::GetDirectories($subDir.FullName, "*", [System.IO.SearchOption]::AllDirectories).Count
        }
        catch {
            Write-Warning "無法完全訪問文件夾: $($subDir.FullName)"
        }
        
        $results += [PSCustomObject]@{
            Name = $subDir.Name
            Path = $subDir.FullName
            Size_GB = [math]::Round($totalSize / 1GB, 3)
            Size_MB = [math]::Round($totalSize / 1MB, 2)
            Size_KB = [math]::Round($totalSize / 1KB, 2)
            Bytes = $totalSize
            FileCount = $fileCount
            FolderCount = $folderCount
            LastModified = $subDir.LastWriteTime
        }
    }
    
    # 按大小排序并返回
    return $results | Sort-Object Bytes -Descending
}

function Get-ParallelFolderSize {
    [CmdletBinding()]
    param(
        [string]$Path = ".",
        [int]$MaxThreads = 5
    )
    
    # 獲取所有子文件夾
    $folders = Get-ChildItem -Path $Path -Directory
    
    # 創(chuàng)建運(yùn)行空間池
    $runspacePool = [RunspaceFactory]::CreateRunspacePool(1, $MaxThreads)
    $runspacePool.Open()
    
    $runspaces = @()
    $results = @()
    
    # 為每個(gè)文件夾創(chuàng)建運(yùn)行空間
    foreach ($folder in $folders) {
        $powershell = [PowerShell]::Create()
        $powershell.RunspacePool = $runspacePool
        
        # 添加腳本塊
        [void]$powershell.AddScript({
            param($folderPath)
            
            $size = 0
            $fileCount = 0
            
            try {
                $files = [System.IO.Directory]::EnumerateFiles(
                    $folderPath, 
                    "*.*", 
                    [System.IO.SearchOption]::AllDirectories
                )
                
                foreach ($file in $files) {
                    try {
                        $fileInfo = New-Object System.IO.FileInfo($file)
                        $size += $fileInfo.Length
                        $fileCount++
                    }
                    catch {
                        # 忽略錯(cuò)誤
                    }
                }
            }
            catch {
                # 忽略文件夾訪問錯(cuò)誤
            }
            
            return [PSCustomObject]@{
                Name = (Get-Item $folderPath).Name
                Path = $folderPath
                Bytes = $size
                FileCount = $fileCount
            }
        })
        
        [void]$powershell.AddArgument($folder.FullName)
        
        # 異步執(zhí)行
        $runspace = [PSCustomObject]@{
            PowerShell = $powershell
            AsyncResult = $powershell.BeginInvoke()
            Folder = $folder
        }
        
        $runspaces += $runspace
    }
    
    # 等待所有任務(wù)完成并收集結(jié)果
    while ($runspaces.AsyncResult.IsCompleted -contains $false) {
        Start-Sleep -Milliseconds 100
    }
    
    foreach ($runspace in $runspaces) {
        $result = $runspace.PowerShell.EndInvoke($runspace.AsyncResult)
        $results += $result
        $runspace.PowerShell.Dispose()
    }
    
    $runspacePool.Close()
    $runspacePool.Dispose()
    
    # 格式化輸出
    return $results | ForEach-Object {
        [PSCustomObject]@{
            FolderName = $_.Name
            Path = $_.Path
            Size_GB = [math]::Round($_.Bytes / 1GB, 3)
            Size_MB = [math]::Round($_.Bytes / 1MB, 2)
            FileCount = $_.FileCount
        }
    } | Sort-Object Size_MB -Descending
}

function Get-EnhancedFolderSize {
    [CmdletBinding()]
    param(
        [string]$Path = ".",
        [string[]]$ExcludeFolders,
        [string[]]$IncludePatterns,
        [int]$MinSizeMB = 0,
        [int]$MaxDepth = 10,
        [switch]$ShowProgress
    )
    
    # 遞歸函數(shù)計(jì)算文件夾大小
    function Get-FolderSizeRecursive {
        param(
            [string]$FolderPath,
            [int]$CurrentDepth
        )
        
        # 檢查深度限制
        if ($CurrentDepth -ge $MaxDepth) {
            return @{Size = 0; FileCount = 0}
        }
        
        $totalSize = 0
        $fileCount = 0
        
        try {
            # 獲取當(dāng)前文件夾下的文件
            $files = Get-ChildItem -Path $FolderPath -File -ErrorAction Stop
            foreach ($file in $files) {
                $totalSize += $file.Length
                $fileCount++
            }
            
            # 遞歸處理子文件夾
            $subFolders = Get-ChildItem -Path $FolderPath -Directory -ErrorAction Stop
            foreach ($folder in $subFolders) {
                # 檢查是否排除
                if ($ExcludeFolders -contains $folder.Name) {
                    continue
                }
                
                # 檢查是否包含
                $shouldInclude = $true
                if ($IncludePatterns) {
                    $shouldInclude = $false
                    foreach ($pattern in $IncludePatterns) {
                        if ($folder.Name -like $pattern) {
                            $shouldInclude = $true
                            break
                        }
                    }
                }
                
                if ($shouldInclude) {
                    $subResult = Get-FolderSizeRecursive -FolderPath $folder.FullName -CurrentDepth ($CurrentDepth + 1)
                    $totalSize += $subResult.Size
                    $fileCount += $subResult.FileCount
                }
            }
        }
        catch {
            Write-Verbose "訪問文件夾失敗: $FolderPath"
        }
        
        return @{Size = $totalSize; FileCount = $fileCount}
    }
    
    # 主邏輯
    $results = @()
    $folders = Get-ChildItem -Path $Path -Directory
    
    $i = 0
    $totalFolders = $folders.Count
    
    foreach ($folder in $folders) {
        $i++
        
        if ($ShowProgress) {
            $percent = [math]::Round(($i / $totalFolders) * 100, 1)
            Write-Progress -Activity "計(jì)算文件夾大小" -Status "$percent% 完成" `
                -CurrentOperation "正在處理: $($folder.Name)" `
                -PercentComplete $percent
        }
        
        # 檢查排除列表
        if ($ExcludeFolders -contains $folder.Name) {
            continue
        }
        
        # 檢查包含模式
        if ($IncludePatterns) {
            $shouldInclude = $false
            foreach ($pattern in $IncludePatterns) {
                if ($folder.Name -like $pattern) {
                    $shouldInclude = $true
                    break
                }
            }
            if (-not $shouldInclude) {
                continue
            }
        }
        
        $result = Get-FolderSizeRecursive -FolderPath $folder.FullName -CurrentDepth 0
        $sizeMB = [math]::Round($result.Size / 1MB, 2)
        
        # 檢查最小大小限制
        if ($sizeMB -ge $MinSizeMB) {
            $results += [PSCustomObject]@{
                FolderName = $folder.Name
                Path = $folder.FullName
                Size_MB = $sizeMB
                Size_GB = [math]::Round($result.Size / 1GB, 3)
                FileCount = $result.FileCount
                LastModified = $folder.LastWriteTime
            }
        }
    }
    
    if ($ShowProgress) {
        Write-Progress -Activity "計(jì)算文件夾大小" -Completed
    }
    
    return $results | Sort-Object Size_MB -Descending
}

function Get-FolderSizeReport {
    [CmdletBinding()]
    param(
        [string]$Path = ".",
        [string]$OutputFormat = "Table",
        [string]$ExportPath
    )
    
    # 獲取文件夾大小數(shù)據(jù)
    $data = Get-EnhancedFolderSize -Path $Path -ShowProgress
    
    if (-not $data) {
        Write-Output "沒有找到符合條件的文件夾"
        return
    }
    
    # 統(tǒng)計(jì)信息
    $totalSizeMB = ($data | Measure-Object -Property Size_MB -Sum).Sum
    $averageSizeMB = [math]::Round($totalSizeMB / $data.Count, 2)
    $largestFolder = $data[0]
    $smallestFolder = $data[-1]
    
    # 生成報(bào)告
    $report = @"
文件夾大小分析報(bào)告
====================
分析路徑: $Path
分析時(shí)間: $(Get-Date -Format 'yyyy-MM-dd HH:mm:ss')
分析文件夾數(shù)量: $($data.Count)

總體統(tǒng)計(jì):
- 總大小: $totalSizeMB MB
- 平均大小: $averageSizeMB MB
- 最大文件夾: $($largestFolder.FolderName) ($($largestFolder.Size_MB) MB)
- 最小文件夾: $($smallestFolder.FolderName) ($($smallestFolder.Size_MB) MB)

詳細(xì)列表:
"@
    
    # 添加詳細(xì)信息
    $rank = 1
    foreach ($item in $data) {
        $report += "`n$rank. $($item.FolderName) - $($item.Size_MB) MB ($($item.FileCount) 個(gè)文件)"
        $rank++
    }
    
    # 輸出報(bào)告
    switch ($OutputFormat.ToLower()) {
        "table" {
            $data | Format-Table -AutoSize
        }
        "list" {
            $data | Format-List
        }
        "csv" {
            if ($ExportPath) {
                $data | Export-Csv -Path $ExportPath -NoTypeInformation
                Write-Host "報(bào)告已導(dǎo)出到: $ExportPath" -ForegroundColor Green
            } else {
                $data | ConvertTo-Csv -NoTypeInformation
            }
        }
        "html" {
            $html = $data | ConvertTo-Html -Title "文件夾大小報(bào)告" -PreContent "<h1>文件夾大小報(bào)告</h1>"
            if ($ExportPath) {
                $html | Out-File -FilePath $ExportPath
                Write-Host "HTML報(bào)告已導(dǎo)出到: $ExportPath" -ForegroundColor Green
            } else {
                $html
            }
        }
        default {
            Write-Output $report
        }
    }
    
    # 輸出統(tǒng)計(jì)信息
    Write-Host "`n=== 統(tǒng)計(jì)摘要 ===" -ForegroundColor Cyan
    Write-Host "總文件夾數(shù): $($data.Count)" -ForegroundColor Yellow
    Write-Host "總大小: $totalSizeMB MB" -ForegroundColor Yellow
    Write-Host "平均大小: $averageSizeMB MB" -ForegroundColor Yellow
}

# 查找可以清理的大文件夾
function Find-LargeFoldersToClean {
    param(
        [string]$Path = "C:\",
        [int]$ThresholdGB = 10,
        [string[]]$ExcludePaths = @("Windows", "Program Files", "Program Files (x86)")
    )
    
    Write-Host "正在查找大于 ${ThresholdGB}GB 的文件夾..." -ForegroundColor Cyan
    
    $largeFolders = Get-EnhancedFolderSize -Path $Path -MinSizeMB ($ThresholdGB * 1024) -ExcludeFolders $ExcludePaths
    
    if ($largeFolders) {
        Write-Host "找到 $($largeFolders.Count) 個(gè)大文件夾:" -ForegroundColor Yellow
        
        foreach ($folder in $largeFolders) {
            $color = if ($folder.Size_GB -gt $ThresholdGB * 2) { "Red" } 
                    elseif ($folder.Size_GB -gt $ThresholdGB * 1.5) { "Magenta" }
                    else { "Green" }
            
            Write-Host "  $($folder.FolderName)" -ForegroundColor $color -NoNewline
            Write-Host " - $($folder.Size_GB) GB ($($folder.FileCount) 個(gè)文件)"
        }
        
        # 生成清理建議
        Write-Host "`n清理建議:" -ForegroundColor Cyan
        foreach ($folder in $largeFolders) {
            $suggestion = switch -Wildcard ($folder.FolderName) {
                "*Log*" { "檢查日志文件，可清理舊日志" }
                "*Temp*" { "臨時(shí)文件夾，可安全清理" }
                "*Cache*" { "緩存文件夾，可清理" }
                "*Download*" { "檢查下載內(nèi)容，可刪除不必要的文件" }
                default { "檢查文件夾內(nèi)容，確認(rèn)是否可清理" }
            }
            
            Write-Host "  $($folder.FolderName): $suggestion"
        }
    } else {
        Write-Host "未找到大于 ${ThresholdGB}GB 的文件夾。" -ForegroundColor Green
    }
}

# 監(jiān)控文件夾大小變化
function Monitor-FolderGrowth {
    param(
        [string]$Path = ".",
        [string]$LogFile = "folder_growth.log",
        [int]$IntervalHours = 24
    )
    
    # 創(chuàng)建日志文件頭
    if (-not (Test-Path $LogFile)) {
        "時(shí)間戳,文件夾路徑,大小_MB,文件數(shù)量,變化_MB" | Out-File -FilePath $LogFile
    }
    
    # 獲取初始大小
    $initialSizes = @{}
    $folders = Get-ChildItem -Path $Path -Directory
    
    foreach ($folder in $folders) {
        $result = Get-FolderSizeRecursive -FolderPath $folder.FullName -CurrentDepth 0
        $initialSizes[$folder.FullName] = @{
            SizeMB = [math]::Round($result.Size / 1MB, 2)
            FileCount = $result.FileCount
            LastCheck = Get-Date
        }
    }
    
    Write-Host "開始監(jiān)控文件夾大小變化..." -ForegroundColor Cyan
    Write-Host "按 Ctrl+C 停止監(jiān)控" -ForegroundColor Yellow
    
    try {
        while ($true) {
            Start-Sleep -Seconds ($IntervalHours * 3600)
            
            $currentTime = Get-Date
            
            foreach ($folder in $folders) {
                $result = Get-FolderSizeRecursive -FolderPath $folder.FullName -CurrentDepth 0
                $currentSizeMB = [math]::Round($result.Size / 1MB, 2)
                $previousSizeMB = $initialSizes[$folder.FullName].SizeMB
                $changeMB = $currentSizeMB - $previousSizeMB
                
                if ([math]::Abs($changeMB) -gt 1) {  # 只記錄變化大于1MB的
                    $logEntry = "$($currentTime.ToString('yyyy-MM-dd HH:mm:ss')),$($folder.FullName),$currentSizeMB,$($result.FileCount),$changeMB"
                    $logEntry | Out-File -FilePath $LogFile -Append
                    
                    if ($changeMB -gt 0) {
                        Write-Host "$($folder.Name) 增加了 ${changeMB}MB" -ForegroundColor Yellow
                    } elseif ($changeMB -lt 0) {
                        Write-Host "$($folder.Name) 減少了 $([math]::Abs($changeMB))MB" -ForegroundColor Green
                    }
                }
                
                # 更新記錄
                $initialSizes[$folder.FullName].SizeMB = $currentSizeMB
                $initialSizes[$folder.FullName].FileCount = $result.FileCount
                $initialSizes[$folder.FullName].LastCheck = $currentTime
            }
        }
    }
    catch {
        Write-Host "監(jiān)控已停止。" -ForegroundColor Red
    }
}

# 以管理員身份運(yùn)行PowerShell
Start-Process PowerShell -Verb RunAs

# 或在腳本中添加錯(cuò)誤處理
try {
    # 嘗試訪問
} catch [System.UnauthorizedAccessException] {
    Write-Warning "無法訪問: $_"
    # 記錄日志或跳過
}

# 使用流式處理而不是一次性加載所有文件
function Get-StreamedFolderSize {
    param([string]$Path)
    
    $totalSize = 0
    $files = [System.IO.Directory]::EnumerateFiles($Path, "*", [System.IO.SearchOption]::AllDirectories)
    
    foreach ($file in $files) {
        try {
            $fileInfo = New-Object System.IO.FileInfo($file)
            $totalSize += $fileInfo.Length
        } catch {
            # 處理錯(cuò)誤
        }
    }
    
    return $totalSize
}

# 創(chuàng)建模塊目錄結(jié)構(gòu)
New-Item -ItemType Directory -Path "FolderSizeAnalyzer"
Set-Location "FolderSizeAnalyzer"

# 創(chuàng)建模塊文件
@'
# FolderSizeAnalyzer.psm1

function Get-FolderSize {
    # 基礎(chǔ)功能
}

function Get-FolderSizeReport {
    # 報(bào)告功能
}

function Find-LargeFolders {
    # 查找大文件夾
}

Export-ModuleMember -Function Get-FolderSize, Get-FolderSizeReport, Find-LargeFolders
'@ | Out-File -FilePath "FolderSizeAnalyzer.psm1"

# 創(chuàng)建模塊清單
New-ModuleManifest -Path "FolderSizeAnalyzer.psd1" `
    -RootModule "FolderSizeAnalyzer.psm1" `
    -Author "Your Name" `
    -Description "高級(jí)文件夾大小分析工具" `
    -FunctionsToExport "Get-FolderSize", "Get-FolderSizeReport", "Find-LargeFolders"

# 創(chuàng)建定期檢查的腳本
$monitorScript = @'
# 每周一早上8點(diǎn)檢查磁盤使用情況
$reportPath = "C:\Reports\DiskUsage_$(Get-Date -Format 'yyyy-MM-dd').html"
Get-FolderSizeReport -Path "C:\" -OutputFormat HTML -ExportPath $reportPath

# 發(fā)送郵件通知（可選）
Send-MailMessage -To "admin@example.com" `
    -Subject "磁盤使用報(bào)告" `
    -Body "本周磁盤使用報(bào)告已生成，請(qǐng)查看附件。" `
    -Attachments $reportPath `
    -SmtpServer "smtp.example.com"
'@

$monitorScript | Out-File -FilePath "C:\Scripts\Monitor-DiskUsage.ps1"

# 創(chuàng)建計(jì)劃任務(wù)
$action = New-ScheduledTaskAction -Execute "PowerShell.exe" `
    -Argument "-File `"C:\Scripts\Monitor-DiskUsage.ps1`""
$trigger = New-ScheduledTaskTrigger -Weekly -DaysOfWeek Monday -At 8am
Register-ScheduledTask -TaskName "磁盤使用監(jiān)控" `
    -Action $action -Trigger $trigger `
    -Description "每周檢查磁盤使用情況" -RunLevel Highest