summaryrefslogtreecommitdiff
path: root/lib/ansible/modules/storage/purestorage/purefa_alert.py
blob: 23c683a4b627ebb84c826294b4161d1c1444279c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
#!/usr/bin/python
# -*- coding: utf-8 -*-

# (c) 2018, Simon Dodsley (simon@purestorage.com)
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)

from __future__ import absolute_import, division, print_function
__metaclass__ = type

ANSIBLE_METADATA = {'metadata_version': '1.1',
                    'status': ['preview'],
                    'supported_by': 'community'}

DOCUMENTATION = r'''
---
module: purefa_alert
version_added: '2.9'
short_description: Configure Pure Storage FlashArray alert email settings
description:
- Configure alert email configuration for Pure Storage FlashArrays.
- Add or delete an individual syslog server to the existing
  list of serves.
author:
- Simon Dodsley (@sdodsley)
options:
  state:
    type: str
    description:
    - Create or delete alert email
    default: present
    choices: [ absent, present ]
  address:
    type: str
    description:
    - Email address (valid format required)
    required: true
  enabled:
    type: bool
    default: true
    description:
    - Set specified email address to be enabled or disabled
extends_documentation_fragment:
- purestorage.fa
'''

EXAMPLES = r'''
- name: Add new email recipient and enable, or enable existing email
  purefa_alert:
    address: "user@domain.com"
    enabled: true
    state: present
    fa_url: 10.10.10.2
    api_token: e31060a7-21fc-e277-6240-25983c6c4592

- name: Delete existing email recipient
  purefa_alert:
    state: absent
    address: "user@domain.com"
    fa_url: 10.10.10.2
    api_token: e31060a7-21fc-e277-6240-25983c6c4592
'''

RETURN = r'''
'''

import re
from ansible.module_utils.basic import AnsibleModule
from ansible.module_utils.pure import get_system, purefa_argument_spec


def create_alert(module, array):
    """Create Alert Email"""
    changed = False
    if not module.check_mode:
        try:
            array.create_alert_recipient(module.params['address'])
            changed = True
        except Exception:
            module.fail_json(msg='Failed to create alert email: {0}'.format(module.params['address']))

        if not module.params['enabled']:
            try:
                array.disable_alert_recipient(module.params['address'])
            except Exception:
                module.fail_json(msg='Failed to create alert email: {0}'.format(module.params['address']))
    changed = True

    module.exit_json(changed=changed)


def enable_alert(module, array):
    """Enable Alert Email"""
    changed = False
    if not module.check_mode:
        try:
            array.enable_alert_recipient(module.params['address'])
        except Exception:
            module.fail_json(msg='Failed to enable alert email: {0}'.format(module.params['address']))
    changed = True

    module.exit_json(changed=changed)


def disable_alert(module, array):
    """Disable Alert Email"""
    changed = False
    if not module.check_mode:
        try:
            array.disable_alert_recipient(module.params['address'])
        except Exception:
            module.fail_json(msg='Failed to disable alert email: {0}'.format(module.params['address']))
    changed = True

    module.exit_json(changed=changed)


def delete_alert(module, array):
    """Delete Alert Email"""
    changed = False
    if module.params['address'] == "flasharray-alerts@purestorage.com":
        module.fail_json(msg='Built-in address {0} cannot be deleted.'.format(module.params['address']))
    if not module.check_mode:
        try:
            array.delete_alert_recipient(module.params['address'])
        except Exception:
            module.fail_json(msg='Failed to delete alert email: {0}'.format(module.params['address']))
    changed = True

    module.exit_json(changed=changed)


def main():
    argument_spec = purefa_argument_spec()
    argument_spec.update(dict(
        address=dict(type='str', required=True),
        enabled=dict(type='bool', default=True),
        state=dict(type='str', default='present', choices=['absent', 'present']),
    ))

    module = AnsibleModule(argument_spec,
                           supports_check_mode=True)

    pattern = re.compile(r"^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$")
    if not pattern.match(module.params['address']):
        module.fail_json(msg='Valid email address not provided.')

    array = get_system(module)

    exists = False
    try:
        emails = array.list_alert_recipients()
    except Exception:
        module.fail_json(msg='Failed to get existing email list')
    for email in range(0, len(emails)):
        if emails[email]['name'] == module.params['address']:
            exists = True
            enabled = emails[email]['enabled']
            break
    if module.params['state'] == 'present' and not exists:
        create_alert(module, array)
    elif module.params['state'] == 'present' and exists and not enabled and module.params['enabled']:
        enable_alert(module, array)
    elif module.params['state'] == 'present' and exists and enabled and not module.params['enabled']:
        disable_alert(module, array)
    elif module.params['state'] == 'absent' and exists:
        delete_alert(module, array)

    module.exit_json(changed=False)


if __name__ == '__main__':
    main()