Tree.psm1

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
# UTF8 with BOM
# http://gnuwin32.sourceforge.net/packages/tree.htm
# https://linux.die.net/man/1/tree
# Get-ChildItemTree D:\Mount\AppsBasic\Console\Powershell\Root -a -I .git

[String]$defaultColor = [ConsoleColor]::darkGray
[String]$leafColor = [ConsoleColor]::DarkGray
[String]$directoryColor = [ConsoleColor]::White
[String]$directoryBackground = [ConsoleColor]::DarkRed

$extensionTypes = @{
    picture = @([ConsoleColor]::Green, @(".png", ".gif", ".jpg"))
    archive = @([ConsoleColor]::Yellow, @(".7z", ".gz", ".rar", ".tar", ".zip"))
    dll = @([ConsoleColor]::Darkgreen, @(".dll", ".pdb"))
    text = @([ConsoleColor]::Cyan, @(".csv", ".lg", "markdown", ".rst", ".txt", ".md"))
    config = @([ConsoleColor]::DarkYellow, @(".cfg", ".config", ".conf", ".ini", ".gitignore", ".yml"))
    executable = @([ConsoleColor]::Blue, @(".exe", ".exe*", ".bat", ".cmd", ".py", ".pl", ".ps1",
                     ".psm1", ".vbs", ".rb", ".reg", ".fsx", ".js"))
}

$nameTypes = @{
    text = @([ConsoleColor]::DarkYellow, @("LICENSE"))
}

function Get-ChildItemTree {

    $tree = tree.exe -F @args
    $tree = $tree -Replace '`-', '└─' -Replace '\|-', '├─' -Replace '\|', '│' -Replace '-', '─'

    cprint yellow . print cr

    foreach ($line in $tree)
    {
        if (-Not ($line -match '(?<leaf>^.*?─\s)(?<filename>.*?$)')) { continue }

        $color = $defaultColor

        if ($matches.filename[-1] -eq '/')
        {
            cprint $matches.leaf $leafColor print ($matches.filename -Replace '/', '') `
            $directoryColor on $directoryBackground print cr
            continue
        }

        foreach ($filesGroup in $extensionTypes.values)
        {
            if ($filesGroup[1].contains([IO.Path]::GetExtension($matches.filename)))
            {
                $color = $filesGroup[0]; break
            }
        }

        foreach ($filesGroup in $nameTypes.values)
        {
            if ($filesGroup[1] -eq $matches.filename) { $color = $filesGroup[0]; break }
        }

        cprint $matches.leaf $leafColor print $matches.filename $color print cr
    }
}

set-alias tree Get-ChildItemTree

export-modulemember -function Get-ChildItemTree -alias tree