From 65f0155d8a4648c6b431c5bca1537ede6f9f9701 Mon Sep 17 00:00:00 2001
From: Ner'zhul <nerzhul@users.noreply.github.com>
Date: Thu, 29 Dec 2016 15:05:22 +0100
Subject: [PATCH] Add icinga2_feature module (#19732)

This module permits to tell icinga to enable or disable features
---
 CHANGELOG.md                                  |   1 +
 .../modules/monitoring/icinga2_feature.py     | 137 ++++++++++++++++++
 2 files changed, 138 insertions(+)
 create mode 100644 lib/ansible/modules/monitoring/icinga2_feature.py

diff --git a/CHANGELOG.md b/CHANGELOG.md
index aaf276987e..451da3136f 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -45,6 +45,7 @@ Ansible Changes By Release
   * ipa_sudocmd
   * ipa_sudorule
   * ipa_user
+- icinga2_feature
 - infinibox:
   * infini_export
   * infini_export_client
diff --git a/lib/ansible/modules/monitoring/icinga2_feature.py b/lib/ansible/modules/monitoring/icinga2_feature.py
new file mode 100644
index 0000000000..f44f93003f
--- /dev/null
+++ b/lib/ansible/modules/monitoring/icinga2_feature.py
@@ -0,0 +1,137 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+"""
+
+Ansible module to manage Icinga2 feature state
+(c) 2016, Loic Blot <loic.blot@unix-experience.fr>
+
+Sponsored by Infopro Digital. http://www.infopro-digital.com/
+Sponsored by E.T.A.I. http://www.etai.fr/
+
+This file is part of Ansible
+
+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/>.
+"""
+
+DOCUMENTATION = '''
+---
+module: icinga2_feature
+
+short_description: Manage Icinga2 feature
+description:
+    - Enable or disable an Icinga2 feature
+version_added: "2.3"
+author: "Loic Blot (@nerzhul)"
+options:
+    name:
+        description:
+            - This is the feature name to enable or disable.
+        required: True
+    state:
+        description:
+            - Apply feature state.
+        required: false
+        choices: [ "present", "absent" ]
+        default: present
+'''
+
+EXAMPLES = '''
+- name: Enable ido-pgsql feature
+  icinga2_feature:
+    name: ido-pgsql
+    state: present
+'''
+
+RETURN = '''
+changed:
+    description: If module has done a change
+    returned: success
+    type: string
+'''
+
+from ansible.module_utils.basic import AnsibleModule
+import re
+
+
+class Icinga2FeatureHelper:
+    def __init__(self, module):
+        self.module = module
+        self._icinga2 = module.get_bin_path('icinga2', True)
+
+    def _exec(self, args):
+        if not self.module.check_mode:
+            cmd = [self._icinga2, 'feature']
+            rc, out, err = self.module.run_command(cmd + args, check_rc=True)
+            return rc, out
+        return 0, list()
+
+    def manage(self):
+        rc, out = self._exec(["list"])
+        if rc != 0:
+            self.module.fail_json(msg="Unable to list icinga2 features. "
+                                      "Ensure icinga2 is installed and present in binary path.")
+
+        else:
+            # If feature is already in good state, just exit
+            if re.search("Disabled features:.* %s[ \n]" % self.module.params["name"], out) \
+                    and self.module.params["state"] == "absent" or \
+                    re.search("Enabled features:.* %s[ \n]" % self.module.params["name"], out) \
+                    and self.module.params["state"] == "present":
+                self.module.exit_json(changed=False)
+
+            if self.module.check_mode:
+                self.module.exit_json(changed=True)
+
+        if self.module.params["state"] == "present":
+            feature_enable_str = "enable"
+        else:
+            feature_enable_str = "disable"
+
+        rc, out = self._exec([feature_enable_str, self.module.params["name"]])
+
+        if self.module.params["state"] == "present":
+            if rc != 0:
+                self.module.fail_json("Fail to %s feature %s. icinga2 command returned %s"
+                                      % (feature_enable_str, self.module.params["name"], out))
+
+            if re.search("already enabled", out) is None:
+                change_applied = True
+            else:
+                change_applied = False
+        else:
+            if rc == 0:
+                change_applied = True
+            # RC is not 0 for this already disabled feature, handle it as no change applied
+            elif re.search("Cannot disable feature '%s'. Target file .* does not exist"
+                                             % self.module.params["name"]):
+                change_applied = False
+            else:
+                self.module.fail_json(msg="Fail to disable feature. Command returns %s" % out)
+
+        self.module.exit_json(changed=change_applied)
+
+
+def main():
+    module = AnsibleModule(
+        argument_spec=dict(
+            name=dict(required=True, type='str'),
+            state=dict(required=False, type='str', choices=["present", "absent"], default="present")
+        ),
+        supports_check_mode=True
+    )
+
+    Icinga2FeatureHelper(module).manage()
+
+if __name__ == '__main__':
+    main()