#!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