# This code is part of Ansible, but is an independent component.
# This particular file snippet, and this file snippet only, is BSD licensed.
# Copyright: (c) 2018, NetApp Ansible Team <ng-ansibleteam@netapp.com>

from __future__ import (absolute_import, division, print_function)
__metaclass__ = type

from ansible.module_utils._text import to_native

HAS_SF_SDK = False
try:
    import solidfire.common
    HAS_SF_SDK = True
except ImportError:
    HAS_SF_SDK = False


def has_sf_sdk():
    return HAS_SF_SDK


class NaElementSWModule(object):
    ''' Support class for common or shared functions '''
    def __init__(self, elem):
        self.elem_connect = elem
        self.parameters = dict()

    def get_volume(self, volume_id):
        """
            Return volume details if volume exists for given volume_id

            :param volume_id: volume ID
            :type volume_id: int
            :return: Volume dict if found, None if not found
            :rtype: dict
        """
        volume_list = self.elem_connect.list_volumes(volume_ids=[volume_id])
        for volume in volume_list.volumes:
            if volume.volume_id == volume_id:
                if str(volume.delete_time) == "":
                    return volume
        return None

    def get_volume_id(self, vol_name, account_id):
        """
            Return volume id from the given (valid) account_id if found
            Return None if not found

            :param vol_name: Name of the volume
            :type vol_name: str
            :param account_id: Account ID
            :type account_id: int

            :return: Volume ID of the first matching volume if found. None if not found.
            :rtype: int
        """
        volume_list = self.elem_connect.list_volumes_for_account(account_id=account_id)
        for volume in volume_list.volumes:
            if volume.name == vol_name:
                # return volume_id
                if str(volume.delete_time) == "":
                    return volume.volume_id
        return None

    def volume_id_exists(self, volume_id):
        """
            Return volume_id if volume exists for given volume_id

            :param volume_id: volume ID
            :type volume_id: int
            :return: Volume ID if found, None if not found
            :rtype: int
        """
        volume_list = self.elem_connect.list_volumes(volume_ids=[volume_id])
        for volume in volume_list.volumes:
            if volume.volume_id == volume_id:
                if str(volume.delete_time) == "":
                    return volume.volume_id
        return None

    def volume_exists(self, volume, account_id):
        """
            Return volume_id if exists, None if not found

            :param volume: Volume ID or Name
            :type volume: str
            :param account_id: Account ID (valid)
            :type account_id: int
            :return: Volume ID if found, None if not found
        """
        # If volume is an integer, get_by_id
        if str(volume).isdigit():
            volume_id = int(volume)
            try:
                if self.volume_id_exists(volume_id):
                    return volume_id
            except solidfire.common.ApiServerError:
                # don't fail, continue and try get_by_name
                pass
        # get volume by name
        volume_id = self.get_volume_id(volume, account_id)
        return volume_id

    def get_snapshot(self, snapshot_id, volume_id):
        """
            Return snapshot details if found

            :param snapshot_id: Snapshot ID or Name
            :type snapshot_id: str
            :param volume_id: Account ID (valid)
            :type volume_id: int
            :return: Snapshot dict if found, None if not found
            :rtype: dict
        """
        # mandate src_volume_id although not needed by sdk
        snapshot_list = self.elem_connect.list_snapshots(
            volume_id=volume_id)
        for snapshot in snapshot_list.snapshots:
            # if actual id is provided
            if str(snapshot_id).isdigit() and snapshot.snapshot_id == int(snapshot_id):
                return snapshot
            # if snapshot name is provided
            elif snapshot.name == snapshot_id:
                return snapshot
        return None

    @staticmethod
    def map_qos_obj_to_dict(qos_obj):
        ''' Take a QOS object and return a key, normalize the key names
            Interestingly, the APIs are using different ids for create and get
        '''
        mappings = [
            ('burst_iops', 'burstIOPS'),
            ('min_iops', 'minIOPS'),
            ('max_iops', 'maxIOPS'),
        ]
        qos_dict = vars(qos_obj)
        # Align names to create API and module interface
        for read, send in mappings:
            if read in qos_dict:
                qos_dict[send] = qos_dict.pop(read)
        return qos_dict

    def get_qos_policy(self, name):
        """
        Get QOS Policy
            :description: Get QOS Policy object for a given name
            :return: object, error
                Policy object converted to dict if found, else None
                Error text if error, else None
            :rtype: dict/None, str/None
        """
        try:
            qos_policy_list_obj = self.elem_connect.list_qos_policies()
        except (solidfire.common.ApiServerError, solidfire.common.ApiConnectionError) as exc:
            error = "Error getting list of qos policies: %s" % to_native(exc)
            return None, error

        policy_dict = dict()
        if hasattr(qos_policy_list_obj, 'qos_policies'):
            for policy in qos_policy_list_obj.qos_policies:
                # Check and get policy object for a given name
                if str(policy.qos_policy_id) == name:
                    policy_dict = vars(policy)
                elif policy.name == name:
                    policy_dict = vars(policy)
        if 'qos' in policy_dict:
            policy_dict['qos'] = self.map_qos_obj_to_dict(policy_dict['qos'])

        return policy_dict if policy_dict else None, None

    def account_exists(self, account):
        """
            Return account_id if account exists for given account id or name
            Raises an exception if account does not exist

            :param account: Account ID or Name
            :type account: str
            :return: Account ID if found, None if not found
        """
        # If account is an integer, get_by_id
        if account.isdigit():
            account_id = int(account)
            try:
                result = self.elem_connect.get_account_by_id(account_id=account_id)
                if result.account.account_id == account_id:
                    return account_id
            except solidfire.common.ApiServerError:
                # don't fail, continue and try get_by_name
                pass
        # get account by name, the method returns an Exception if account doesn't exist
        result = self.elem_connect.get_account_by_name(username=account)
        return result.account.account_id

    def set_element_attributes(self, source):
        """
            Return telemetry attributes for the current execution

            :param source: name of the module
            :type source: str
            :return: a dict containing telemetry attributes
        """
        attributes = {}
        attributes['config-mgmt'] = 'ansible'
        attributes['event-source'] = source
        return attributes