🩹 [Refactor]: Update parameters in Build-PSModule scripts for improved clarity and functionality

MariusStorhaug · MariusStorhaug · commit b5b42c0e7ed3 · 2025-03-10T10:02:02.000+01:00
diff --git a/scripts/helpers/Build-PSModule.ps1 b/scripts/helpers/Build-PSModule.ps1
@@ -44,7 +44,7 @@
     Build-PSModuleBase -ModuleName $ModuleName -ModuleSourceFolder $moduleSourceFolder -ModuleOutputFolder $moduleOutputFolder
     Build-PSModuleManifest -ModuleName $ModuleName -ModuleOutputFolder $moduleOutputFolder
     Build-PSModuleRootModule -ModuleName $ModuleName -ModuleOutputFolder $moduleOutputFolder
-    Update-PSModuleManifestAliasesToExport -ModuleName $ModuleName -ModuleOutputFolder $moduleOutputFolder
+    Update-PSModuleManifestAliasesToExport -ModuleName $ModuleName -ModuleSourceFolder $moduleSourceFolder
 
     LogGroup 'Build manifest file - Final Result' {
         $outputManifestPath = Join-Path -Path $ModuleOutputFolder -ChildPath "$ModuleName.psd1"
diff --git a/scripts/helpers/Build/Build-PSModuleRootModule.ps1 b/scripts/helpers/Build/Build-PSModuleRootModule.ps1
@@ -256,11 +256,11 @@ Export-ModuleMember @exports
         Write-Host (Show-FileContent -Path $rootModuleFile)
     }
 
-    LogGroup 'Build root module - Validate - Import' {
-        Install-PSModule -Path $ModuleOutputFolder
-    }
+    # LogGroup 'Build root module - Validate - Import' {
+    #     Install-PSModule -Path $ModuleOutputFolder
+    # }
 
-    LogGroup 'Build root module - Validate - File list' {
-        Get-ChildItem -Path $ModuleOutputFolder -Recurse -Force | Resolve-Path -Relative | Sort-Object
-    }
+    # LogGroup 'Build root module - Validate - File list' {
+    #     Get-ChildItem -Path $ModuleOutputFolder -Recurse -Force | Resolve-Path -Relative | Sort-Object
+    # }
 }
diff --git a/scripts/helpers/Build/Update-PSModuleManifestAliasesToExport.ps1 b/scripts/helpers/Build/Update-PSModuleManifestAliasesToExport.ps1
@@ -25,19 +25,71 @@
 
         # Folder where the module is outputted.
         [Parameter(Mandatory)]
-        [System.IO.DirectoryInfo] $ModuleOutputFolder
+        [System.IO.DirectoryInfo] $ModuleSourceFolder
     )
     LogGroup 'Updating aliases to export in module manifest' {
         Write-Host "Module name: [$ModuleName]"
-        Write-Host "Module output folder: [$ModuleOutputFolder]"
-        $aliases = Get-Command -Module $ModuleName -CommandType Alias
-        Write-Host "Found aliases: [$($aliases.Count)]"
-        foreach ($alias in $aliases) {
-            Write-Host "Alias: [$($alias.Name)]"
+        Write-Host "Module output folder: [$ModuleSourceFolder]"
+
+        $publicFunctionsPath = Join-Path -Path $ModuleSourceFolder -ChildPath 'public/functions'
+        Write-Host "Public functions path: [$publicFunctionsPath]"
+        if (-not (Test-Path -Path $publicFunctionsPath)) {
+            Write-Host "Public functions path does not exist: [$publicFunctionsPath]"
+            return
+        }
+
+        # Get all child items in the module source folder of a powershell file type
+        $files = Get-ChildItem -Path $publicFunctionsPath -Recurse -File -Include '*.ps1', '*.psm1' | Select-Object -ExpandProperty Path
+
+        # Initialize an array to store all found aliases
+        $allAliases = @()
+
+        foreach ($file in $files) {
+            Write-Host "Parsing file: [$file]"
+
+            # Parse the file using AST
+            $ast = [System.Management.Automation.Language.Parser]::ParseFile(
+                $file,
+                [ref]$null,
+                [ref]$null
+            )
+
+            # Get all function definitions
+            $functionAsts = $ast.FindAll(
+                {
+                    param($node)
+                    $node -is [System.Management.Automation.Language.FunctionDefinitionAst]
+                }, $false)
+
+            Write-Host "   Found functions: [$($functionAsts.Count)]"
+            foreach ($functionAst in $functionAsts) {
+                # Get the function name
+                $functionName = $functionAst.Name
+                Write-Host "   Processing: [$functionName]"
+
+                $functionAst | ForEach-Object {
+                    $funcName = $_.Name
+                    $funcAttributes = $_.Body.FindAll({ $args[0] -is [System.Management.Automation.Language.AttributeAst] }, $true) | Where-Object {
+                        $_.Parent -is [System.Management.Automation.Language.ParamBlockAst]
+                    }
+                    $aliasAttr = $funcAttributes | Where-Object { $_.TypeName.Name -eq 'Alias' }
+
+                    if ($aliasAttr) {
+                        $allAliases = $aliasAttr.PositionalArguments | ForEach-Object { $_.ToString().Trim('"', "'") }
+                        Write-Host "  Found alias [$allAliases] for function [$funcName]"
+                        $allAliases += $aliases
+                    }
+                }
+            }
+        }
+
+        Write-Host "Found aliases: [$($allAliases.Count)]"
+        foreach ($alias in $allAliases) {
+            Write-Host " - [$($alias.Name)]"
         }
-        $outputManifestPath = Join-Path -Path $ModuleOutputFolder -ChildPath "$ModuleName.psd1"
+        $outputManifestPath = Join-Path -Path $ModuleSourceFolder -ChildPath "$ModuleName.psd1"
         Write-Host "Output manifest path: [$outputManifestPath]"
         Write-Host 'Setting module manifest with AliasesToExport'
-        Set-ModuleManifest -Path $outputManifestPath -AliasesToExport $aliases.Name -Verbose
+        Set-ModuleManifest -Path $outputManifestPath -AliasesToExport $allAliases.Name -Verbose
     }
 }
