win_chocolatey_feature: new module (#42848)

* win_chocolatey_feature: new module

* Fixed up copyright header in PowerShell file

lib/ansible/modules/windows/win_chocolatey_feature.ps1

@@ -0,0 +1,74 @@
+#!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.CommandUtil
+#Requires -Module Ansible.ModuleUtils.Legacy
+
+$ErrorActionPreference = "Stop"
+
+$params = Parse-Args -arguments $args -supports_check_mode $true
+$check_mode = Get-AnsibleParam -obj $params -name "_ansible_check_mode" -type "bool" -default $false
+
+$name = Get-AnsibleParam -obj $params -name "name" -type "str" -failifempty $true
+$state = Get-AnsibleParam -obj $params -name "state" -type "str" -default "enabled" -validateset "disabled", "enabled"
+
+$result = @{
+    changed = $false
+}
+
+Function Get-ChocolateyFeatures {
+    param($choco_app)
+
+    $res = Run-Command -command "`"$($choco_app.Path)`" feature list -r"
+    if ($res.rc -ne 0) {
+        Fail-Json -obj $result -message "Failed to list Chocolatey features: $($res.stderr)"
+    }
+    $feature_info = @{}
+    $res.stdout -split "`r`n" | Where-Object { $_ -ne "" } | ForEach-Object {
+        $feature_split = $_ -split "\|"
+        $feature_info."$($feature_split[0])" = $feature_split[1] -eq "Enabled"
+    }
+
+    return ,$feature_info
+}
+
+Function Set-ChocolateyFeature {
+    param(
+        $choco_app,
+        $name,
+        $enabled
+    )
+
+    if ($enabled) {
+        $state_string = "enable"
+    } else {
+        $state_string = "disable"
+    }
+    $res = Run-Command -command "`"$($choco_app.Path)`" feature $state_string --name `"$name`""
+    if ($res.rc -ne 0) {
+        Fail-Json -obj $result -message "Failed to set Chocolatey feature $name to $($state_string): $($res.stderr)"
+    }
+}
+
+$choco_app = Get-Command -Name choco.exe -CommandType Application -ErrorAction SilentlyContinue
+if (-not $choco_app) {
+    Fail-Json -obj $result -message "Failed to find Chocolatey installation, make sure choco.exe is in the PATH env value"
+}
+
+$feature_info = Get-ChocolateyFeatures -choco_app $choco_app
+if ($name -notin $feature_info.keys) {
+    Fail-Json -obj $result -message "Invalid feature name '$name' specified, valid features are: $($feature_info.keys -join ', ')"
+}
+
+$expected_status = $state -eq "enabled"
+$feature_status = $feature_info.$name
+if ($feature_status -ne $expected_status) {
+    if (-not $check_mode) {
+        Set-ChocolateyFeature -choco_app $choco_app -name $name -enabled $expected_status
+    }
+    $result.changed = $true
+}
+
+Exit-Json -obj $result

lib/ansible/modules/windows/win_chocolatey_feature.py

@@ -0,0 +1,50 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# Copyright: (c) 2018, Ansible Project
+# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
+
+ANSIBLE_METADATA = {'metadata_version': '1.1',
+                    'status': ['preview'],
+                    'supported_by': 'community'}
+
+DOCUMENTATION = r'''
+---
+module: win_chocolatey_feature
+version_added: '2.7'
+short_description: Manages Chocolatey features
+description:
+- Used to enable or disable features in Chocolatey.
+options:
+  name:
+    description:
+    - The name of the feature to manage.
+    - Run C(choco.exe feature list) to get a list of features that can be
+      managed.
+    required: yes
+  state:
+    description:
+    - When C(disabled) then the feature will be disabled.
+    - When C(enabled) then the feature will be enabled.
+    choices:
+    - disabled
+    - enabled
+    default: enabled
+author:
+- Jordan Borean (@jborean93)
+'''
+
+EXAMPLES = r'''
+- name: disable file checksum matching
+  win_chocolatey_feature:
+    name: checksumFiles
+    state: disabled
+
+- name: stop Chocolatey on the first package failure
+  win_chocolatey_feature:
+    name: stopOnFirstPackageFailure
+    state: enabled
+'''
+
+RETURN = r'''
+'''