Public/PowerTree.ps1
|
function PowerTree { [CmdletBinding()] param( [Parameter(Position=0)] [string]$Path = ".", [Parameter()] [Alias("l", "level")] [int]$Depth = -1, [Parameter()] [Alias("h", "?")] [switch]$Help, [Parameter()] [Alias("ex", "example")] [switch]$Examples, [Parameter()] [Alias("dm", "m")] [switch]$DisplayMode, [Parameter()] [Alias("prune", "p")] [switch]$PruneEmptyFolders, [Parameter()] [Alias("dmd")] [switch]$DisplayModificationDate, [Parameter()] [Alias("dcd")] [switch]$DisplayCreationDate, [Parameter()] [Alias("dla")] [switch]$DisplayLastAccessDate, [Parameter()] [Alias("d", "dir")] [switch]$DirectoryOnly, [Parameter()] [Alias("e", "exclude")] [string[]]$ExcludeDirectories = @(), [Parameter()] [ValidateSet("size", "name", "version", "md", "cd", "la")] [string]$Sort, [Parameter()] [Alias("smd")] [switch]$SortByModificationDate, [Parameter()] [Alias("scd")] [switch]$SortByCreationDate, [Parameter()] [Alias("sla", "sld")] [switch]$SortByLastAccessDate, [Parameter()] [Alias("ss")] [switch]$SortBySize, [Parameter()] [Alias("sn")] [switch]$SortByName, [Parameter()] [Alias("sv")] [switch]$SortByVersion, [Parameter()] [Alias("des", "desc")] [switch]$Descending, [Parameter()] [ValidateScript({ # Validate format of lower bound size filter $_ -match '^\d+(?:\.\d+)?(b|kb|mb|gb|tb)?$' })] [Alias('fsmi')] [string]$FileSizeMinimum = "-1kb", [Parameter()] [ValidateScript({ # Validate format of upper bound size filter $_ -match '^\d+(?:\.\d+)?(b|kb|mb|gb|tb)?$' })] [Alias('fsma')] [string]$FileSizeMaximum = "-1kb", [Alias('fs', 'filesize')] [string]$FileSizeFilter, [Parameter()] [Alias("s", "size")] [switch]$DisplaySize, [Parameter()] [Alias("ef")] [string[]]$ExcludeExtensions = @(), [Parameter()] [Alias("if")] [string[]]$IncludeExtensions = @(), [Parameter()] [Alias("force")] [switch]$ShowHiddenFiles, [Parameter()] [Alias("o", "of")] [string]$OutFile, [Parameter()] [Alias("q", "silent")] [switch]$Quiet ) if ($Help -or $PSBoundParameters.ContainsKey('h') -or $PSBoundParameters.ContainsKey('?') -or $args -contains "--help" -or $args -contains "-help") { Write-PSTreeHelp return } if ($Examples) { Write-ExamplesHelp return } $treeStats = [TreeStats]::new() $jsonSettings = Get-PowerTreeSettingsFromJson $treeConfig = [TreeConfig]::new() $treeConfig.Path = $Path $treeConfig.LineStyle = Build-TreeLineStyle -Style $jsonSettings.LineStyle $treeConfig.DirectoryOnly = $DirectoryOnly $treeConfig.ExcludeDirectories = Build-ExcludedDirectoryParams -CommandLineExcludedDir $ExcludeDirectories ` -Settings $jsonSettings $treeConfig.SortBy = Get-SortingMethod -SortBySize $SortBySize ` -SortByVersion $SortByVersion ` -SortByName $SortByName ` -SortByCreationDate $SortByCreationDate ` -SortByLastAccessDate $SortByLastAccessDate ` -SortByModificationDate $SortByModificationDate ` -DefaultSort $jsonSettings.Sorting.By ` -Sort $Sort $treeConfig.SortDescending = $Descending $treeConfig.SortFolders = $jsonSettings.Sorting.SortFolders $treeConfig.HeaderTable = Get-HeaderTable -DisplayCreationDate $DisplayCreationDate ` -DisplayLastAccessDate $DisplayLastAccessDate ` -DisplayModificationDate $DisplayModificationDate ` -DisplaySize $DisplaySize ` -DisplayMode $DisplayMode ` -LineStyle $treeConfig.LineStyle $treeConfig.ShowConnectorLines = $jsonSettings.ShowConnectorLines $treeConfig.ShowHiddenFiles = $ShowHiddenFiles $treeConfig.MaxDepth = if ($Depth -ne -1) { $Depth } else { $jsonSettings.MaxDepth } $treeConfig.FileSizeBounds = Build-FileSizeParams -CommandLineMaxSize $FileSizeMaximum ` -CommandlineMinSize $FileSizeMinimum ` -SettingsLineMaxSize $jsonSettings.Files.FileSizeMaximum ` -SettingsLineMinSize $jsonSettings.Files.FileSizeMinimum $treeConfig.Quiet = $Quiet $treeConfig.OutFile = Add-DefaultExtension -FilePath $OutFile -Quiet $treeConfig.Quiet $treeConfig.PruneEmptyFolders = $PruneEmptyFolders $outputBuilder = Initialize-OutputBuilder -TreeConfig $treeConfig -ShowExecutionStats $jsonSettings.ShowExecutionStats # Main entry point $executionResultTime = Measure-Command { try { if (-not (Test-Path $Path)) { throw "Cannot find path '$Path'" } Write-Verbose "Loaded settings: $($jsonSettings | ConvertTo-Json -Depth 5)" $ChildItemDirectoryParams = Build-ChildItemDirectoryParams $ShowHiddenFiles $ChildItemFileParams = Build-ChildItemFileParams -ShowHiddenFiles $ShowHiddenFiles ` -CommandLineIncludeExt $IncludeExtensions ` -CommandLineExcludeExt $ExcludeExtensions ` -FileSettings $jsonSettings.Files Write-Configuration -TreeConfig $treeConfig Write-HeaderToOutput -HeaderTable $treeConfig.HeaderTable ` -OutputBuilder $outputBuilder ` -Quiet $treeConfig.Quiet ` -LineStyle $treeConfig.LineStyle Get-TreeView -TreeConfig $treeConfig ` -TreeStats $treeStats ` -ChildItemDirectoryParams $ChildItemDirectoryParams ` -ChildItemFileParams $ChildItemFileParams ` -OutputBuilder $outputBuilder } catch { Write-Error "Details: $($_.Exception.Message)" Write-Error "Location: $($_.InvocationInfo.ScriptLineNumber), $($_.InvocationInfo.PositionMessage)" Write-Verbose "Exception details: $($_ | Format-List * -Force | Out-String)" } } if($jsonSettings.ShowExecutionStats) { $treeStats.DisplaySummary($executionResultTime, $outputBuilder, $treeConfig.Quiet, $treeConfig.LineStyle) } if($null -ne $outputBuilder) { $outputBuilder.ToString() | Write-ToFile -FilePath $treeConfig.OutFile -OpenOutputFileOnFinish $jsonSettings.Files.OpenOutputFileOnFinish $fullOutputPath = Resolve-Path $treeConfig.OutFile -ErrorAction SilentlyContinue if ($null -eq $fullOutputPath) { $fullOutputPath = $treeConfig.OutFile } Write-Host "" Write-Host "Output saved to: $($fullOutputPath)" -ForegroundColor Cyan } Write-Host "" } |