ansible-later/testenv/lib/python2.7/site-packages/ansible/modules/windows/win_xml.ps1
2019-04-23 13:04:27 +02:00

239 lines
8.4 KiB
PowerShell

#!powershell
# Copyright: (c) 2018, Ansible Project
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
#Requires -Module Ansible.ModuleUtils.Legacy
Set-StrictMode -Version 2
function Copy-Xml($dest, $src, $xmlorig) {
if ($src.get_NodeType() -eq "Text") {
$dest.set_InnerText($src.get_InnerText())
}
if ($src.get_HasAttributes()) {
foreach ($attr in $src.get_Attributes()) {
$dest.SetAttribute($attr.get_Name(), $attr.get_Value())
}
}
if ($src.get_HasChildNodes()) {
foreach ($childnode in $src.get_ChildNodes()) {
if ($childnode.get_NodeType() -eq "Element") {
$newnode = $xmlorig.CreateElement($childnode.get_Name(), $xmlorig.get_DocumentElement().get_NamespaceURI())
Copy-Xml $newnode $childnode $xmlorig
$dest.AppendChild($newnode) | Out-Null
} elseif ($childnode.get_NodeType() -eq "Text") {
$dest.set_InnerText($childnode.get_InnerText())
}
}
}
}
function Compare-XmlDocs($actual, $expected) {
if ($actual.get_Name() -ne $expected.get_Name()) {
throw "Actual name not same as expected: actual=" + $actual.get_Name() + ", expected=" + $expected.get_Name()
}
##attributes...
if (($actual.get_NodeType() -eq "Element") -and ($expected.get_NodeType() -eq "Element")) {
if ($actual.get_HasAttributes() -and $expected.get_HasAttributes()) {
if ($actual.get_Attributes().Count -ne $expected.get_Attributes().Count) {
throw "attribute mismatch for actual=" + $actual.get_Name()
}
for ($i=0;$i -lt $expected.get_Attributes().Count; $i =$i+1) {
if ($expected.get_Attributes()[$i].get_Name() -ne $actual.get_Attributes()[$i].get_Name()) {
throw "attribute name mismatch for actual=" + $actual.get_Name()
}
if ($expected.get_Attributes()[$i].get_Value() -ne $actual.get_Attributes()[$i].get_Value()) {
throw "attribute value mismatch for actual=" + $actual.get_Name()
}
}
}
if (($actual.get_HasAttributes() -and !$expected.get_HasAttributes()) -or (!$actual.get_HasAttributes() -and $expected.get_HasAttributes())) {
throw "attribute presence mismatch for actual=" + $actual.get_Name()
}
}
##children
if ($expected.get_ChildNodes().Count -ne $actual.get_ChildNodes().Count) {
throw "child node mismatch. for actual=" + $actual.get_Name()
}
for ($i=0;$i -lt $expected.get_ChildNodes().Count; $i =$i+1) {
if (-not $actual.get_ChildNodes()[$i]) {
throw "actual missing child nodes. for actual=" + $actual.get_Name()
}
Compare-XmlDocs $expected.get_ChildNodes()[$i] $actual.get_ChildNodes()[$i]
}
if ($expected.get_InnerText()) {
if ($expected.get_InnerText() -ne $actual.get_InnerText()) {
throw "inner text mismatch for actual=" + $actual.get_Name()
}
}
elseif ($actual.get_InnerText()) {
throw "actual has inner text but expected does not for actual=" + $actual.get_Name()
}
}
function BackupFile($path) {
$backuppath = $path + "." + [DateTime]::Now.ToString("yyyyMMdd-HHmmss");
Copy-Item $path $backuppath;
return $backuppath;
}
$params = Parse-Args $args -supports_check_mode $true
$check_mode = Get-AnsibleParam -obj $params -name "_ansible_check_mode" -type "bool" -default $false
$debug_level = Get-AnsibleParam -obj $params -name "_ansible_verbosity" -type "int"
$debug = $debug_level -gt 2
$dest = Get-AnsibleParam $params "path" -type "path" -FailIfEmpty $true -aliases "dest", "file"
$fragment = Get-AnsibleParam $params "fragment" -type "str" -FailIfEmpty $true -aliases "xmlstring"
$xpath = Get-AnsibleParam $params "xpath" -type "str" -FailIfEmpty $true
$backup = Get-AnsibleParam $params "backup" -type "bool" -Default $false
$type = Get-AnsibleParam $params "type" -type "str" -Default "element" -ValidateSet "element", "attribute", "text"
$attribute = Get-AnsibleParam $params "attribute" -type "str" -FailIfEmpty ($type -eq "attribute")
$state = Get-AnsibleParam $params "state" -type "str" -Default "present"
$result = @{
changed = $false
}
If (-Not (Test-Path -Path $dest -PathType Leaf)){
Fail-Json $result "Specified path $dest does not exist or is not a file."
}
[xml]$xmlorig = $null
Try {
[xml]$xmlorig = Get-Content -Path $dest
}
Catch {
Fail-Json $result "Failed to parse file at '$dest' as an XML document: $($_.Exception.Message)"
}
$namespaceMgr = New-Object System.Xml.XmlNamespaceManager $xmlorig.NameTable
$namespace = $xmlorig.DocumentElement.NamespaceURI
$localname = $xmlorig.DocumentElement.LocalName
$namespaceMgr.AddNamespace($xmlorig.$localname.SchemaInfo.Prefix, $namespace)
if ($type -eq "element") {
$xmlchild = $null
Try {
$xmlchild = [xml]$fragment
} Catch {
Fail-Json $result "Failed to parse fragment as XML: $($_.Exception.Message)"
}
$child = $xmlorig.CreateElement($xmlchild.get_DocumentElement().get_Name(), $xmlorig.get_DocumentElement().get_NamespaceURI())
Copy-Xml $child $xmlchild.DocumentElement $xmlorig
$node = $xmlorig.SelectSingleNode($xpath, $namespaceMgr)
if ($node.get_NodeType() -eq "Document") {
$node = $node.get_DocumentElement()
}
$elements = $node.get_ChildNodes()
[bool]$present = $false
[bool]$changed = $false
if ($elements.get_Count()) {
if ($debug) {
$err = @()
$result.err = {$err}.Invoke()
}
foreach ($element in $elements) {
try {
Compare-XmlDocs $child $element
$present = $true
break
} catch {
if ($debug) {
$result.err.Add($_.Exception.ToString())
}
}
}
if (!$present -and ($state -eq "present")) {
[void]$node.AppendChild($child)
$result.msg = "xml added"
$changed = $true
} elseif ($present -and ($state -eq "absent")) {
[void]$node.RemoveChild($element)
$result.msg = "xml removed"
$changed = $true
}
} else {
if ($state -eq "present") {
[void]$node.AppendChild($child)
$result.msg = "xml added"
$changed = $true
}
}
if ($changed) {
$result.changed = $true
if (!$check_mode) {
if ($backup) {
$result.backup = BackupFile($dest)
}
$xmlorig.Save($dest)
} else {
$result.msg += " check mode"
}
} else {
$result.msg = "not changed"
}
} elseif ($type -eq "text") {
$node = $xmlorig.SelectSingleNode($xpath, $namespaceMgr)
[bool]$add = ($node.get_InnerText() -ne $fragment)
if ($add) {
$result.changed = $true
if (-Not $check_mode) {
if ($backup) {
$result.backup = BackupFile($dest)
}
$node.set_InnerText($fragment)
$xmlorig.Save($dest)
$result.msg = "text changed"
} else {
$result.msg = "text changed check mode"
}
} else {
$result.msg = "not changed"
}
} elseif ($type -eq "attribute") {
$node = $xmlorig.SelectSingleNode($xpath, $namespaceMgr)
[bool]$add = !$node.HasAttribute($attribute) -Or ($node.$attribute -ne $fragment)
if ($add -And ($state -eq "present")) {
$result.changed = $true
if (-Not $check_mode) {
if ($backup) {
$result.backup = BackupFile($dest)
}
if (!$node.HasAttribute($attribute)) {
$node.SetAttributeNode($attribute, $xmlorig.get_DocumentElement().get_NamespaceURI())
}
$node.SetAttribute($attribute, $fragment)
$xmlorig.Save($dest)
$result.msg = "text changed"
} else {
$result.msg = "text changed check mode"
}
} elseif (!$add -And ($state -eq "absent")) {
$result.changed = $true
if (-Not $check_mode) {
if ($backup) {
$result.backup = BackupFile($dest)
}
$node.RemoveAttribute($attribute)
$xmlorig.Save($dest)
$result.msg = "text changed"
}
} else {
$result.msg = "not changed"
}
}
Exit-Json $result