Merge pull request #30 from PowershellFrameworkCollective/find-psmdtype

Find-PSMDType Attribute parameter

Author: FriedrichWeinmann

PSModuleDevelopment/changelog.md

@@ -1,4 +1,8 @@
 # Changelog
+## 2.2.2.14 (May 03rd, 2018)
+ - Upd: Find-PSMDType - add `-Attribute` parameter to filter by class attributes (#27)
+ - Fix: Find-PSMDType - suppress error that gets thrown on empty assemblies.
+
 ## 2.2.1.12 (March 08th, 2018)
  - Added out-of-the box templates
  - fix: Verious bugfixes around the template system

PSModuleDevelopment/functions/assembly/Find-PSMDType.ps1

@@ -37,6 +37,10 @@
 		The type must directly inherit from this type.
 		Accepts wildcards.
 	
+	.PARAMETER Attribute
+		The type must have this attribute assigned.
+		Accepts wildcards.
+	
 	.EXAMPLE
 		Find-PSMDType -Name "*String*"
 	
@@ -70,7 +74,10 @@
 		$Implements,
 
 		[string]
-		$InheritsFrom
+		$InheritsFrom,
+		
+		[string]
+		$Attribute
 	)
 
 	begin
@@ -85,11 +92,18 @@
 			if ($boundPublic)
 			{
 				if ($Public) { $types = $item.ExportedTypes }
-				else { $types = $item.GetTypes() | Where-Object IsPublic -EQ $false }
+				else
+				{
+					# Empty Assemblies will error on this, which is not really an issue and can be safely ignored
+					try { $types = $item.GetTypes() | Where-Object IsPublic -EQ $false }
+					catch { Write-PSFMessage -Message "Failed to enumerate types on $item" -Level InternalComment -Tag 'fail','assembly','type','enumerate' -ErrorRecord $_ }
+				}
 			}
 			else
 			{
-				$types = $item.GetTypes()
+				# Empty Assemblies will error on this, which is not really an issue and can be safely ignored
+				try { $types = $item.GetTypes() }
+				catch { Write-PSFMessage -Message "Failed to enumerate types on $item" -Level InternalComment -Tag 'fail', 'assembly', 'type', 'enumerate' -ErrorRecord $_ }
 			}
 
 			foreach ($type in $types)
@@ -99,6 +113,7 @@
 				if ($Implements -and ($type.ImplementedInterfaces.Name -notcontains $Implements)) { continue }
 				if ($boundEnum -and ($Enum -ne $type.IsEnum)) { continue }
 				if ($InheritsFrom -and ($type.BaseType.FullName -notlike $InheritsFrom)) { continue }
+				if ($Attribute -and ($type.CustomAttributes.AttributeType.Name -notlike $Attribute)) { continue }
 
 				$type
 			}