# -*- coding: UTF-8 -*-
from .pool import Pool, PoolElement, Template, extractString, XMLElement


class Quota(XMLElement):
    def __init__(self, xml):
        super(Quota, self).__init__(xml)
        self._convert_types()


class VMQuota(Quota):
    XML_TYPES = {
        'cpu': int,
        'cpu_used': int,
        'memory': int,
        'memory_used': int,
        'system_disk_size': int,
        'system_disk_size_used': int,
        'vms': int,
        'vms_used': int,
    }


class DatastoreQuota(Quota):
    XML_TYPES = {
        'images': int,
        'images_used': int,
        'size': int,
        'size_used': int,
    }


class NetworkQuota(Quota):
    XML_TYPES = {
        'leases': int,
        'leases_used': int,
    }


class VMQuotaList(Quota):
    XML_TYPES = {
        'vm': VMQuota,
    }


class DatastoreQuotaList(Quota):
    XML_TYPES = {
        'datastore': DatastoreQuota,
    }


class NetworkQuotaList(Quota):
    XML_TYPES = {
        'network': NetworkQuota,
    }


class User(PoolElement):
    METHODS = {
        'info': 'user.info',
        'allocate': 'user.allocate',
        'delete': 'user.delete',
        'passwd': 'user.passwd',
        'chgrp': 'user.chgrp'
    }

    XML_TYPES = {
        'id': int,
        'gid': int,
        'group_ids': ['GROUPS', lambda group_ids: [int(group_id.text) for group_id in group_ids]],
        'gname': extractString,
        'name': extractString,
        'password': extractString,
        'auth_driver': extractString,
        'enabled': bool,
        'template': ['TEMPLATE', Template],
        # 'network_quota': handled separately   # see http://dev.opennebula.org/issues/3849
        # 'image_quota'                         # see http://dev.opennebula.org/issues/3849
        # 'default_user_quotas'                 # see http://dev.opennebula.org/issues/3849
    }

    ELEMENT_NAME = 'USER'

    @staticmethod
    def allocate(client, user, password):
        """
        allocates a new user in OpenNebula

        ``user``
           username for the new user

        ``password``
           password for the new user
        """
        user_id = client.call(User.METHODS['allocate'], user, password)
        return user_id

    def __init__(self, xml, client):
        super(User, self).__init__(xml, client)
        self.id = self['ID'] if self['ID'] else None

    def change_passwd(self, new_password):
        """
        Changes the password for the given user.

        ``new_password``
           The new password
        """
        self.client.call(User.METHODS['passwd'], self.id, new_password)

    def chgrp(self, gid):
        """
        Changes the main group

        ``gid``
            New group id. Set to -1 to leave the current one
        """
        self.client.call(User.METHODS['chgrp'], self.id, gid)

    @property
    def vm_quota(self):
        self.info()
        return VMQuotaList(self.xml.find('VM_QUOTA')).vm

    @property
    def datastore_quota(self):
        self.info()
        return DatastoreQuotaList(self.xml.find('DATASTORE_QUOTA')).datastore

    @property
    def network_quota(self):
        self.info()
        return NetworkQuotaList(self.xml.find('NETWORK_QUOTA')).network

    def __repr__(self):
        return '<oca.User("%s")>' % self.name


class UserPool(Pool):
    METHODS = {
        'info': 'userpool.info',
    }

    def __init__(self, client):
        super(UserPool, self).__init__('USER_POOL', 'USER', client)

    def _factory(self, xml):
        u = User(xml, self.client)
        u._convert_types()
        return u