1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2024-09-14 20:13:21 +02:00
community.general/lib/ansible/modules/windows/win_robocopy.ps1
Dag Wieers fc7d6a4cd5 win_robocopy: Small cleanup (#20515)
- Harmonizing the checkmode usage in windows modules
- Simplify the $result object
- Add parameter types
2017-01-24 10:40:14 -05:00

165 lines
4.5 KiB
PowerShell

#!powershell
# This file is part of Ansible
#
# Copyright 2015, Corwin Brown <corwin.brown@maxpoint.com>
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# WANT_JSON
# POWERSHELL_COMMON
$params = Parse-Args $args -supports_check_mode $true
$check_mode = Get-AnsibleParam -obj $params -name "_ansible_check_mode" -type "bool" -default $false
$src = Get-AnsibleParam -obj $params -name "src" -type "path" -failifempty $true
$dest = Get-AnsibleParam -obj $params -name "dest" -type "path" -failifempty $true
$purge = Get-AnsibleParam -obj $params -name "purge" -type "bool" -default $false
$recurse = Get-AnsibleParam -obj $params -name "recurse" -type "bool" -default $false
$flags = Get-AnsibleParam -obj $params -name "flags" -type "string" -default $null
$result = @{
src = $src
dest = $dest
recurse = $recurse
purge = $purge
changed = $false
}
# Search for an Error Message
# Robocopy seems to display an error after 3 '-----' separator lines
Function SearchForError($cmd_output, $default_msg) {
$separator_count = 0
$error_msg = $default_msg
ForEach ($line in $cmd_output) {
if (-Not $line) {
continue
}
if ($separator_count -ne 3) {
if (Select-String -InputObject $line -pattern "^(\s+)?(\-+)(\s+)?$") {
$separator_count += 1
}
}
Else {
If (Select-String -InputObject $line -pattern "error") {
$error_msg = $line
break
}
}
}
return $error_msg
}
# Build Arguments
$robocopy_opts = @()
if (-Not (Test-Path $src)) {
Fail-Json $result "$src does not exist!"
}
$robocopy_opts += $src
$robocopy_opts += $dest
if ($flags -eq $null) {
if ($purge) {
$robocopy_opts += "/purge"
}
if ($recurse) {
$robocopy_opts += "/e"
}
}
Else {
ForEach ($f in $flags.split(" ")) {
$robocopy_opts += $f
}
}
$result.flags = $flags
$robocopy_output = ""
$rc = 0
If ($check_mode -eq $true) {
$robocopy_output = "Would have copied the contents of $src to $dest"
$rc = 0
}
Else {
Try {
&robocopy $robocopy_opts | Tee-Object -Variable robocopy_output | Out-Null
$rc = $LASTEXITCODE
}
Catch {
$ErrorMessage = $_.Exception.Message
Fail-Json $result "Error synchronizing $src to $dest! Msg: $ErrorMessage"
}
}
$result.return_code = $rc
$result.output = $robocopy_output
$cmd_msg = "Success"
$changed = $false
switch ($rc)
{
0 {
$cmd_msg = "No files copied."
}
1 {
$cmd_msg = "Files copied successfully!"
$changed = $true
}
2 {
$cmd_msg = "Some Extra files or directories were detected. No files were copied."
$changed = $true
}
3 {
$cmd_msg = "(2+1) Some files were copied. Additional files were present."
$changed = $true
}
4 {
$cmd_msg = "Some mismatched files or directories were detected. Housekeeping might be required!"
$changed = $true
}
5 {
$cmd_msg = "(4+1) Some files were copied. Some files were mismatched."
$changed = $true
}
6 {
$cmd_msg = "(4+2) Additional files and mismatched files exist. No files were copied."
$changed = $true
}
7 {
$cmd_msg = "(4+1+2) Files were copied, a file mismatch was present, and additional files were present."
$changed = $true
}
8 {
$error_msg = SearchForError $robocopy_output "Some files or directories could not be copied!"
Fail-Json $result $error_msg
}
{ @(9, 10, 11, 12, 13, 14, 15) -contains $_ } {
$error_msg = SearchForError $robocopy_output "Fatal error. Check log message!"
Fail-Json $result $error_msg
}
16 {
$error_msg = SearchForError $robocopy_output "Serious Error! No files were copied! Do you have permissions to access $src and $dest?"
Fail-Json $result $error_msg
}
}
$result.msg = $cmd_msg
$result.changed = $changed
Exit-Json $result