DNS.ps1
|
$hostFilePath = 'C:\Windows\System32\drivers\etc\hosts' function Add-DNSEntries() { [CmdletBinding()] param( [Parameter(Mandatory, Position = 0)] [string]$IPAddress, [Parameter(Mandatory, Position = 1)] [string]$Domain, [Parameter(Position = 2)] [string[]]$SubDomains ) $PSBoundParameters.Remove('SubDomains') > $null Invoke-Privileged -Function 'Add-DNSEntries' @PSBoundParameters ` ( $SubDomains ? "-SubDomains $($SubDomains -join ',')" : '') if (-not (Test-Admin)) { return } $isVerbose = $PSBoundParameters['Verbose'] -eq $true Remove-DNSEntries -Domain $Domain Write-Verbose "Writing DNS entries to location: $hostFilePath" $hasNewlime = (Get-Content $hostFilePath -Raw) -Match [System.Environment]::NewLine + '$' $entries = ($hasNewlime ? '' : [System.Environment]::NewLine) + "$IPAddress $Domain # Added by powershell devop tools" foreach ($subDomain in $SubDomains) { $entries += [System.Environment]::NewLine + "$IPAddress $subDomain.$Domain # Added by powershell devop tools" } Add-Content -Path $hostFilePath -Value $entries -NoNewline if ($isVerbose) { Get-Content $hostFilePath -Raw Write-Verbose 'Done ... Press Enter to exit:' Read-Host 1> $null } } function Remove-DNSEntries() { [CmdletBinding()] param( [Parameter(Mandatory)] [string]$Domain ) Invoke-Privileged -Function 'Remove-DNSEntries' @PSBoundParameters if (-not (Test-Admin)) { return } $isVerbose = $PSBoundParameters['Verbose'] -eq $true $lines = @() foreach ($line in Get-Content $hostFilePath) { if ($line -NotMatch "$Domain # Added by powershell devop tools") { $lines += $line } } $lines | Out-File $hostFilePath if ($isVerbose) { Get-Content $hostFilePath -Raw Write-Verbose 'Done ... Press Enter to exit:' Read-Host 1> $null } } |