#!powershell
# This file is part of Ansible.
#
# Copyright 2014, Paul Durivage <paul.durivage@rackspace.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

Import-Module Servermanager;

$params = Parse-Args $args;

$result = New-Object PSObject -Property @{
    changed = $false
}

$name = Get-Attr $params "name" -failifempty $true

$name = $name -split ',' | % { $_.Trim() }

$state = Get-Attr $params "state" "present"
$state = $state.ToString().ToLower()
If (($state -ne 'present') -and ($state -ne 'absent')) {
    Fail-Json $result "state is '$state'; must be 'present' or 'absent'"
}

$restart = Get-Attr $params "restart" $false | ConvertTo-Bool
$includesubfeatures = Get-Attr $params "include_sub_features" $false | ConvertTo-Bool
$includemanagementtools = Get-Attr $params "include_management_tools" $false | ConvertTo-Bool
$source = Get-Attr $params "source" $false

If ($state -eq "present") {
    if ($source)
    {
        if (!(test-path $source))
        {
            Fail-Json $result "Failed to find source path $source"
        }
    }
    
    $InstallParams = @{
        "Name"=$name;
        "Restart"=$Restart;
        "IncludeAllSubFeature"=$includesubfeatures;
        "ErrorAction"="SilentlyContinue"
    }
    
    if ($IncludeManagementTools -eq $true)
    {
        $InstallParams.add("IncludeManagementTools",$includemanagementtools)
    }
    
    if ($source -ne $null)
    {
        $InstallParams.add("Source",$source)
    }
    
    
    
    try {
        If (Get-Command "Install-WindowsFeature" -ErrorAction SilentlyContinue) {
            $featureresult = Install-WindowsFeature @InstallParams
        }
        ElseIf (Get-Command "Add-WindowsFeature" -ErrorAction SilentlyContinue) {
            if ($IncludeManagementTools)
            {
                $Params.Remove("IncludeManagementTools")
            }
            $featureresult = Add-WindowsFeature @InstallParams
        }
        Else {
            Fail-Json $result "Not supported on this version of Windows"
        }
    }
    catch {
        Fail-Json $result $_.Exception.Message
    }
}
ElseIf ($state -eq "absent") {
    try {
        If (Get-Command "Uninstall-WindowsFeature" -ErrorAction SilentlyContinue) {
            $featureresult = Uninstall-WindowsFeature -Name $name -Restart:$restart -ErrorAction SilentlyContinue
        }
        ElseIf (Get-Command "Remove-WindowsFeature" -ErrorAction SilentlyContinue) {
            $featureresult = Remove-WindowsFeature -Name $name -Restart:$restart -ErrorAction SilentlyContinue
        }
        Else {
            Fail-Json $result "Not supported on this version of Windows"
        }
    }
    catch {
        Fail-Json $result $_.Exception.Message
    }
}

# Loop through results and create a hash containing details about
# each role/feature that is installed/removed
$installed_features = @()
#$featureresult.featureresult is filled if anything was changed
If ($featureresult.FeatureResult)
{
    ForEach ($item in $featureresult.FeatureResult) {
        $message = @()
        ForEach ($msg in $item.Message) {
            $message += New-Object PSObject -Property @{
                message_type = $msg.MessageType.ToString()
                error_code = $msg.ErrorCode
                text = $msg.Text
            }
        }
        $installed_features += New-Object PSObject -Property @{
            id = $item.Id
            display_name = $item.DisplayName
            message = $message
            restart_needed = $item.RestartNeeded.ToString() | ConvertTo-Bool
            skip_reason = $item.SkipReason.ToString()
            success = $item.Success.ToString() | ConvertTo-Bool
        }
    }
    $result.changed = $true
}

Set-Attr $result "feature_result" $installed_features
Set-Attr $result "success" ($featureresult.Success.ToString() | ConvertTo-Bool)
Set-Attr $result "exitcode" $featureresult.ExitCode.ToString()
Set-Attr $result "restart_needed" ($featureresult.RestartNeeded.ToString() | ConvertTo-Bool)

If ($result.success) {
    Exit-Json $result
}
ElseIf ($state -eq "present") {
    Fail-Json $result "Failed to add feature"
}
Else {
    Fail-Json $result "Failed to remove feature"
}
