#!/usr/bin/python
# coding: utf-8 -*-

# (c) 2020, Bianca Henderson <bianca@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: tower_inventory_source_update
author: "Bianca Henderson (@beeankha)"
short_description: Update inventory source(s).
description:
    - Update Ansible Tower inventory source(s). See
      U(https://www.ansible.com/tower) for an overview.
options:
    inventory:
      description:
        - Name of the inventory that contains the inventory source(s) to update.
      required: True
      type: str
    inventory_source:
      description:
        - The name of the inventory source to update.
      required: True
      type: str
    organization:
      description:
        - Name of organization for project.
      type: str
      required: False
extends_documentation_fragment: awx.awx.auth
'''

EXAMPLES = '''
- name: Update a single inventory source
  tower_inventory_source_update:
    inventory: "My Inventory"
    inventory_source: "Example Inventory Source"
    organization: Default

- name: Update all inventory sources
  tower_inventory_source_update:
    inventory: "My Other Inventory"
    inventory_source: "{{ item }}"
  loop: "{{ query('awx.awx.tower_api', 'inventory_sources', query_params={ 'inventory': 30 }, return_ids=True ) }}"
'''

RETURN = '''
id:
    description: id of the inventory update
    returned: success
    type: int
    sample: 86
status:
    description: status of the inventory update
    returned: success
    type: str
    sample: pending
'''

from ..module_utils.tower_api import TowerAPIModule


def main():
    # Any additional arguments that are not fields of the item can be added here
    argument_spec = dict(
        inventory=dict(required=True),
        inventory_source=dict(required=True),
        organization=dict(),
    )

    # Create a module for ourselves
    module = TowerAPIModule(argument_spec=argument_spec)

    # Extract our parameters
    inventory = module.params.get('inventory')
    inventory_source = module.params.get('inventory_source')

    new_fields = {}
    organization_id = None
    organization = module.params.get('organization')
    if organization:
        organization_id = module.get_one_by_name_or_id('organizations', organization)

    # Attempt to look up the inventory the user specified (these will fail the module if not found)
    inventory_object = module.get_one_by_name_or_id('inventories', inventory)
    # Return all inventory sources related to the specified inventory
    inventory_source_object = module.get_one_by_name_or_id(inventory_object['related']['inventory_sources'], inventory_source)

    # Sync the inventory source(s)
    inventory_source_update_results = module.post_endpoint(inventory_source_object['related']['update'], **{'data': {}})

    if inventory_source_update_results['status_code'] != 202:
        module.fail_json(msg="Failed to update inventory source, see response for details", **{'response': inventory_source_update_results})

    module.exit_json(**{
        'changed': True,
        'id': inventory_source_update_results['json']['id'],
        'status': inventory_source_update_results['json']['status'],
    })


if __name__ == '__main__':
    main()