Private/FSvc.Persist.ps1
|
# Persisting Set-FSvcConfig values across sessions. # # Settings are written to a delimited block in the PowerShell profile, which # sets the FSVC_* environment variables when the profile loads. The current # process is updated by Set-FSvcConfig itself. $script:FSvcBlockStart = '# >>> fsvc (managed) >>>' $script:FSvcBlockEnd = '# <<< fsvc (managed) <<<' # Maps friendly setting names to their FSVC_* environment variable names. # Unknown keys are ignored; empty values are kept so they can clear a setting. function ConvertTo-FSvcEnvironmentMap { param([hashtable]$Settings) $map = @{} if (-not $Settings) { return $map } foreach ($key in $Settings.Keys) { if ($script:FSvcEnvNames.ContainsKey($key)) { $map[$script:FSvcEnvNames[$key]] = [string]$Settings[$key] } } return $map } # Renders the managed profile block for an env-name -> value map. function New-FSvcProfileBlock { param([hashtable]$Map) if (-not $Map -or $Map.Count -eq 0) { return "" } $lines = @() foreach ($key in ($Map.Keys | Sort-Object)) { $value = ([string]$Map[$key]).Replace("'", "''") $lines += ("`$env:{0} = '{1}'" -f $key, $value) } return ($script:FSvcBlockStart + "`n" + ($lines -join "`n") + "`n" + $script:FSvcBlockEnd + "`n") } # Reads the managed block back into an env-name -> value map. function Read-FSvcProfileSettings { param([string]$ProfilePath) $map = @{} if (-not $ProfilePath -or -not (Test-Path -LiteralPath $ProfilePath)) { return $map } $text = Get-Content -LiteralPath $ProfilePath -Raw $pattern = "(?ms)" + [regex]::Escape($script:FSvcBlockStart) + "(.*?)" + [regex]::Escape($script:FSvcBlockEnd) $match = [regex]::Match($text, $pattern) if (-not $match.Success) { return $map } foreach ($line in ($match.Groups[1].Value -split "`n")) { $lm = [regex]::Match($line.Trim(), '^\$env:(?<n>[A-Za-z0-9_]+)\s*=\s*''(?<v>.*)''$') if ($lm.Success) { $map[$lm.Groups['n'].Value] = $lm.Groups['v'].Value.Replace("''", "'") } } return $map } # Inserts, replaces, or (with an empty $Block) removes the managed block, # preserving the rest of the profile. function Update-FSvcProfile { param([string]$ProfilePath, [AllowNull()][string]$Block) $existing = "" if (Test-Path -LiteralPath $ProfilePath) { $existing = Get-Content -LiteralPath $ProfilePath -Raw } $pattern = "(?ms)" + [regex]::Escape($script:FSvcBlockStart) + ".*?" + [regex]::Escape($script:FSvcBlockEnd) + "\r?\n?" $cleaned = [regex]::Replace($existing, $pattern, "").TrimEnd() if ($Block) { $cleaned = $cleaned + "`n`n" + $Block } elseif ($cleaned) { $cleaned = $cleaned + "`n" } $dir = Split-Path -Parent $ProfilePath if ($dir -and -not (Test-Path -LiteralPath $dir)) { New-Item -ItemType Directory -Path $dir -Force | Out-Null } Set-Content -LiteralPath $ProfilePath -Value $cleaned -NoNewline } # Merges settings into the managed profile block for future sessions. Empty # values clear a persisted setting; the block is removed when nothing is left. function Set-FSvcPersistentSettings { param( [hashtable]$Settings, [string]$ProfilePath ) $map = ConvertTo-FSvcEnvironmentMap -Settings $Settings if ($map.Count -eq 0) { return } $merged = Read-FSvcProfileSettings -ProfilePath $ProfilePath foreach ($key in $map.Keys) { if ($map[$key]) { $merged[$key] = $map[$key] } else { [void]$merged.Remove($key) } } Update-FSvcProfile -ProfilePath $ProfilePath -Block (New-FSvcProfileBlock -Map $merged) } |