PowerProfile.psm1

#Requires -Version 5.1
#Requires -Modules 'PowerProfile.Core'

$ExecutionContext.SessionState.Module.OnRemove = {
    Remove-Module -Force -Name PowerProfile-Dyn.* -ErrorAction Ignore
    Remove-Module -Force -Name PowerProfile.Core.Load -ErrorAction Ignore
    Remove-Module -Force -Name PowerProfile.Core -ErrorAction Ignore
}

#region PowerProfile Environment
if ($null -eq $env:PSLVL) {
    if ((Get-PoProfileContent).Modules.Count -gt 0) {
        [array]::Reverse((Get-PoProfileContent).Modules)
        foreach ($p in (Get-PoProfileContent).Modules) {
            $env:PSModulePath += [System.IO.Path]::PathSeparator + $p
        }
    }
    if ((Get-PoProfileContent).Scripts.Count -gt 0) {
        [array]::Reverse((Get-PoProfileContent).Scripts)
        foreach ($p in (Get-PoProfileContent).Scripts) {
            $env:PATH += [System.IO.Path]::PathSeparator + $p
        }
    }
}
#endregion

#region Profiles
function Initialize-Profiles {
    if (Get-PoProfileState 'PowerProfile.Profiles') {
        $ProfilesState = Get-PoProfileState 'PowerProfile.Profiles'
    } else {
        $ProfilesState = @{ State = 'Incomplete' }
    }
    $ProfilesState.LastUpdate = Get-Date
    if ($null -ne $ProfilesState.Exceptions) {
        $ProfilesState.Exceptions = $null
    }

    Write-PoProfileProgress -ProfileTitle 'Loading profile'

    #region User functions
    if ((Get-PoProfileContent).Functions.Count -gt 0) {
        New-Module -Name 'PowerProfile-Dyn.Functions' -ScriptBlock {
            $ErrorActionPreference = 'Stop'

            foreach ($key in (Get-PoProfileContent).Functions.Keys) {
                try {
                    $null = . (Get-PoProfileContent).Functions.$key
                }
                catch {
                    Write-PoProfileItemProgress -ItemTitle 'Failed to import functions' -ItemTextColor $PSStyle.Foreground.BrightRed -Depth 1 -ItemText ((Split-Path -LeafBase $key) -replace '(^[0-9][\w.]*-)')
                }
            }
            Export-ModuleMember -Function * -Alias * -Variable @()
        } | Import-Module -Global -DisableNameChecking
    }
    #endregion

    #region User profiles
    $ProfileId = 1
    foreach ($Global:PoProfileTmpThisProfile in @(Get-PoProfileProfilesList)) {
        if ($null -eq (Get-PoProfileContent).Profiles.$($PoProfileTmpThisProfile)) {
            continue
        }

        switch ($ProfileId) {
            1 {
                $DynModName = 'PowerProfile-Dyn.Scripts'
            }
            2 {
                $DynModName = 'PowerProfile-Dyn.Scripts.' + (($env:LC_PSHOST -replace ' ','.') -replace '_','')
                Write-PoProfileProgress -ProfileTitle "Loading profile for $env:LC_PSHOST"
                if ([System.IO.File]::Exists($PROFILE.CurrentUserCurrentHost)) {
                    Write-PoProfileProgress -ScriptTitleType Warning -ScriptTitle ((Split-Path -Leaf $PROFILE.CurrentUserCurrentHost)+' might interfere with PowerProfile.'),'You should delete the file and transform its functionality to PowerProfile directory format.'
                }
            }
            3 {
                $DynModName = 'PowerProfile-Dyn.Scripts.' + (($env:LC_TERMINAL -replace ' ','.') -replace '_','')
                Write-PoProfileProgress -ProfileTitle "Loading profile for $env:LC_TERMINAL"
            }
        }

        New-Module -Name $DynModName -ScriptBlock {
            $ErrorActionPreference = 'Stop'

            foreach ($key in (Get-PoProfileContent).Profiles.$($PoProfileTmpThisProfile).keys) {

                $ScriptProperties = $null

                if ($key -match '^([0-9]{4,})?((?:\.\w+)+)?-?([\w ]+)((?:\.\w+)+)?\.(\w+)$') {
                    if ($null -ne $Matches[1]) {
                        $ScriptOrderNumber = $Matches[1]
                    }
                    if ($null -ne $Matches[2]) {
                        $ScriptProviderDetails = $Matches[2].Substring(1).Split('.')
                    }
                    $PoProfileSciptName = $Matches[3] -replace '_',' '
                    if ($null -ne $Matches[4]) {
                        $ScriptProperties = $Matches[4].Substring(1).Split('.')
                        if (
                            (
                                $ScriptProperties -contains 'IsProfileSetup' -and
                                (
                                    $null -eq (Get-PoProfileState 'PSPackageManagement') -or
                                    ((Get-PoProfileState('PSPackageManagement')).State -eq 'Complete')
                                ) -and
                                (
                                    ((Get-PoProfileState('PowerProfile.Profiles')).State -eq 'Complete') -or
                                    ($null -ne $env:PSLVL)
                                )
                            ) -or
                            ($null -ne $env:PSLVL -and $ScriptProperties -contains 'IsNotPSLVL') -or
                            ($null -eq $env:PSLVL -and $ScriptProperties -contains 'IsPSLVL') -or
                            ($IsNonInteractive -and $ScriptProperties -contains 'IsInteractive')
                        ) {
                            continue
                        }
                        elseif (
                            ($env:IsElevated -and $ScriptProperties -contains 'IsNotElevated') -or
                            (-Not $env:IsElevated -and $ScriptProperties -contains 'IsElevated') -or
                            ($IsCommand -and -not $IsNoExit -and $ScriptProperties -contains 'IsNoCommand') -or
                            (-Not $IsCommand -and $ScriptProperties -contains 'IsCommand') -or
                            ($IsLogin -and $ScriptProperties -contains 'IsNoLogin') -or
                            (-Not $IsLogin -and $ScriptProperties -contains 'IsLogin') -or
                            (-Not $IsNonInteractive -and $ScriptProperties -contains 'IsNonInteractive')
                        ) {
                            if ($ScriptProperties -notcontains 'IsHidden') {
                                Write-PoProfileProgress -ScriptTitle ($PSStyle.Foreground.Yellow + $PSStyle.BoldOff + '[Skipped] ' + $PSStyle.Foreground.White + $PoProfileSciptName) -NoCounter
                            }
                            continue
                        }
                    }
                } else {
                    Write-PoProfileProgress -ScriptTitle ($PSStyle.Foreground.BrightYellow + $PSStyle.BoldOff + '[Ignored] ' + $PSStyle.Foreground.White + $key + $PSStyle.Foreground.Yellow + ' (file name order number?)') -NoCounter
                    continue
                }

                if ($ScriptProperties -notcontains 'IsHidden') {
                    Write-PoProfileProgress -ScriptTitle $PoProfileSciptName
                }

                try {
                    if ($ScriptIsHidden) {
                        $null = . (Get-PoProfileContent).Profiles.$($PoProfileTmpThisProfile).$key
                    } else {
                        . (Get-PoProfileContent).Profiles.$($PoProfileTmpThisProfile).$key
                    }
                }
                catch {
                    if ($ScriptIsHidden) {
                        Write-PoProfileProgress -ScriptTitle $PoProfileSciptName
                    }
                    Write-PoProfileProgress -ScriptTitle @(($_.Exception.Message).Trim() -split [System.Environment]::NewLine) -ScriptTitleType Error
                }
            }
        } | Import-Module -Global -DisableNameChecking

        $ProfileId++
    }
    #endregion

    if ($null -eq $env:PSLVL) {
        if ($null -eq $ProfilesState.Exceptions) {
            $ProfilesState.State = 'Complete'
        } elseif ($ProfilesState.State -eq 'Incomplete') {
            $ProfilesState.State = 'Error'
        }
        Set-PoProfileState 'PowerProfile.Profiles' $ProfilesState
    }
}

# Load narrow profile
if ($IsCommand) {
    Initialize-Profiles
}

# Load full interactive profile
if(-Not $IsCommand -or $IsNoExit) {

    # Environment notices
    if ($null -eq $env:PSLVL -and $null -eq $IsNoExit) {
        Write-PoProfileProgress -ProfileTitle 'NOTICE' -ScriptCategory ''
        if ($null -ne $env:IsElevated) {
            Write-PoProfileProgress -ScriptTitle 'Careful! Running session with ELEVATED PRIVILEGES' -ScriptTitleType Warning
        }
        elseif ($null -ne $env:IsProfileRedirected) {
            Write-PoProfileProgress -ScriptTitle 'Your profile is synced using OneDrive:','PowerShell modules will preferably be installed in global machine scpope instead of user scope.' -ScriptTitleType Information
            if ($IsWindows) {
                Push-Location $PROFILEHOME
                $null = attrib +P /s
                Pop-Location
            }
        }
    }

    # Load scripts from narrow profile if not loaded already
    if ($null -ne $IsCommand) {
        Initialize-Profiles
    }

    Write-PoProfileProgress -ProfileTitle 'INITIALIZATION COMPLETED'
    Remove-Variable PoProfileTmp* -Scope Global -ErrorAction Ignore
}
#endregion