From 529bc0edc67a5d4f2c9d00f5325ddfa19a267c7a Mon Sep 17 00:00:00 2001
From: Abhijeet Kasurde <akasurde@redhat.com>
Date: Wed, 28 Feb 2018 20:11:53 +0530
Subject: [PATCH] New module: digital_ocean_size_facts (#35898)

Signed-off-by: Abhijeet Kasurde <akasurde@redhat.com>
---
 .../digital_ocean/digital_ocean_size_facts.py | 115 ++++++++++++++++++
 1 file changed, 115 insertions(+)
 create mode 100644 lib/ansible/modules/cloud/digital_ocean/digital_ocean_size_facts.py

diff --git a/lib/ansible/modules/cloud/digital_ocean/digital_ocean_size_facts.py b/lib/ansible/modules/cloud/digital_ocean/digital_ocean_size_facts.py
new file mode 100644
index 00000000000..ca1b5f59612
--- /dev/null
+++ b/lib/ansible/modules/cloud/digital_ocean/digital_ocean_size_facts.py
@@ -0,0 +1,115 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+# Copyright: (c) 2018, Ansible Project
+# Copyright: (c) 2018, Abhijeet Kasurde <akasurde@redhat.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 = '''
+---
+module: digital_ocean_size_facts
+short_description: Gather facts about DigitalOcean Droplet sizes
+description:
+    - This module can be used to gather facts about droplet sizes.
+author: "Abhijeet Kasurde (@akasurde)"
+version_added: "2.6"
+requirements:
+  - "python >= 2.6"
+extends_documentation_fragment: digital_ocean.documentation
+'''
+
+
+EXAMPLES = '''
+- name: Gather facts about all droplet sizes
+  digital_ocean_size_facts:
+    oauth_token: "{{ oauth_token }}"
+
+- name: Get droplet Size Slug where vcpus is 1
+  digital_ocean_size_facts:
+    oauth_token: "{{ oauth_token }}"
+  register: resp_out
+- debug: var=resp_out
+- set_fact:
+    size_slug: "{{ item.slug }}"
+  with_items: "{{ resp_out.data|json_query(name) }}"
+  vars:
+    name: "[?vcpus==`1`]"
+- debug: var=size_slug
+
+
+'''
+
+
+RETURN = '''
+data:
+    description: DigitalOcean droplet size facts
+    returned: success
+    type: list
+    sample: [
+        {
+            "available": true,
+            "disk": 20,
+            "memory": 512,
+            "price_hourly": 0.00744,
+            "price_monthly": 5.0,
+            "regions": [
+                "ams2",
+                "ams3",
+                "blr1",
+                "fra1",
+                "lon1",
+                "nyc1",
+                "nyc2",
+                "nyc3",
+                "sfo1",
+                "sfo2",
+                "sgp1",
+                "tor1"
+            ],
+            "slug": "512mb",
+            "transfer": 1.0,
+            "vcpus": 1
+        },
+    ]
+'''
+
+from traceback import format_exc
+from ansible.module_utils.basic import AnsibleModule
+from ansible.module_utils.digital_ocean import DigitalOceanHelper
+from ansible.module_utils._text import to_native
+
+
+def core(module):
+    rest = DigitalOceanHelper(module)
+
+    response = rest.get('sizes')
+    if response.status_code != 200:
+        module.fail_json(msg="Failed to fetch 'sizes' facts due to error : %s" % response.json['message'])
+
+    module.exit_json(changed=False, data=response.json['sizes'])
+
+
+def main():
+    argument_spec = DigitalOceanHelper.digital_ocean_argument_spec()
+    module = AnsibleModule(
+        argument_spec=argument_spec,
+    )
+
+    try:
+        core(module)
+    except Exception as e:
+        module.fail_json(msg=to_native(e), exception=format_exc())
+
+
+if __name__ == '__main__':
+    main()