-
Notifications
You must be signed in to change notification settings - Fork 86
/
Locksmith.psm1
96 lines (92 loc) · 3.48 KB
/
Locksmith.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
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
# Get public and private function definition files.
$Public = @( Get-ChildItem -Path $PSScriptRoot\Public\*.ps1 -ErrorAction SilentlyContinue -Recurse )
$Private = @( Get-ChildItem -Path $PSScriptRoot\Private\*.ps1 -ErrorAction SilentlyContinue -Recurse )
$Classes = @( Get-ChildItem -Path $PSScriptRoot\Classes\*.ps1 -ErrorAction SilentlyContinue -Recurse )
$Enums = @( Get-ChildItem -Path $PSScriptRoot\Enums\*.ps1 -ErrorAction SilentlyContinue -Recurse )
# Get all assemblies
$AssemblyFolders = Get-ChildItem -Path $PSScriptRoot\Lib -Directory -ErrorAction SilentlyContinue
# Lets find which libraries we need to load
$Default = $false
$Core = $false
$Standard = $false
foreach ($A in $AssemblyFolders.Name) {
if ($A -eq 'Default') {
$Default = $true
} elseif ($A -eq 'Core') {
$Core = $true
} elseif ($A -eq 'Standard') {
$Standard = $true
}
}
if ($Standard -and $Core -and $Default) {
$FrameworkNet = 'Default'
$Framework = 'Standard'
} elseif ($Standard -and $Core) {
$Framework = 'Standard'
$FrameworkNet = 'Standard'
} elseif ($Core -and $Default) {
$Framework = 'Core'
$FrameworkNet = 'Default'
} elseif ($Standard -and $Default) {
$Framework = 'Standard'
$FrameworkNet = 'Default'
} elseif ($Standard) {
$Framework = 'Standard'
$FrameworkNet = 'Standard'
} elseif ($Core) {
$Framework = 'Core'
$FrameworkNet = ''
} elseif ($Default) {
$Framework = ''
$FrameworkNet = 'Default'
} else {
#Write-Error -Message 'No assemblies found'
}
$Assembly = @(
if ($Framework -and $PSEdition -eq 'Core') {
Get-ChildItem -Path $PSScriptRoot\Lib\$Framework\*.dll -ErrorAction SilentlyContinue -Recurse
}
if ($FrameworkNet -and $PSEdition -ne 'Core') {
Get-ChildItem -Path $PSScriptRoot\Lib\$FrameworkNet\*.dll -ErrorAction SilentlyContinue -Recurse
}
)
$FoundErrors = @(
Foreach ($Import in @($Assembly)) {
try {
Write-Verbose -Message $Import.FullName
Add-Type -Path $Import.Fullname -ErrorAction Stop
# }
} catch [System.Reflection.ReflectionTypeLoadException] {
Write-Warning "Processing $($Import.Name) Exception: $($_.Exception.Message)"
$LoaderExceptions = $($_.Exception.LoaderExceptions) | Sort-Object -Unique
foreach ($E in $LoaderExceptions) {
Write-Warning "Processing $($Import.Name) LoaderExceptions: $($E.Message)"
}
$true
#Write-Error -Message "StackTrace: $($_.Exception.StackTrace)"
} catch {
Write-Warning "Processing $($Import.Name) Exception: $($_.Exception.Message)"
$LoaderExceptions = $($_.Exception.LoaderExceptions) | Sort-Object -Unique
foreach ($E in $LoaderExceptions) {
Write-Warning "Processing $($Import.Name) LoaderExceptions: $($E.Message)"
}
$true
#Write-Error -Message "StackTrace: $($_.Exception.StackTrace)"
}
}
#Dot source the files
Foreach ($Import in @($Classes + $Enums + $Private + $Public)) {
Try {
. $Import.Fullname
} Catch {
Write-Error -Message "Failed to import functions from $($import.Fullname): $_"
$true
}
}
)
if ($FoundErrors.Count -gt 0) {
$ModuleName = (Get-ChildItem $PSScriptRoot\*.psd1).BaseName
Write-Warning "Importing module $ModuleName failed. Fix errors before continuing."
break
}
Export-ModuleMember -Function '*' -Alias '*'