Import-ModuleFunctions.ps1

function Import-ModuleFunctions(
    [Parameter()]
    [array]
    $Roots = @(
        "$PSScriptRoot/public"
        "$PSScriptRoot/private"
    ),

    [Parameter()]
    [switch]
    $All
) {
    <#
        .SYNOPSIS
        Import module functions
    #>
    $PublicFunctions = $()

    $Roots | ForEach-Object {
        if (Test-Path -Path $_) {
            Get-ChildItem -Path $_ -Filter "*.ps1" -Recurse | `
                ForEach-Object {
                if ($_ -match 'public') {
                    $PublicFunctions += $_.BaseName
                }
                elseif ($All) {
                    $PublicFunctions += $_.BaseName
                }
                $AllFunctions += $_.BaseName
                . $_.FullName
            }
        }
    }
}