#
# Base class for all VM devices
#
# Copyright 2008, 2013  Red Hat, Inc.
# Cole Robinson <crobinso@redhat.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free  Software Foundation; either version 2 of the License, or
# (at your option)  any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
# MA 02110-1301 USA.

from virtinst.xmlbuilder import XMLBuilder, XMLProperty
import logging


class VirtualDevice(XMLBuilder):
    """
    Base class for all domain xml device objects.
    """

    VIRTUAL_DEV_DISK            = "disk"
    VIRTUAL_DEV_NET             = "interface"
    VIRTUAL_DEV_INPUT           = "input"
    VIRTUAL_DEV_GRAPHICS        = "graphics"
    VIRTUAL_DEV_AUDIO           = "sound"
    VIRTUAL_DEV_HOSTDEV         = "hostdev"
    VIRTUAL_DEV_SERIAL          = "serial"
    VIRTUAL_DEV_PARALLEL        = "parallel"
    VIRTUAL_DEV_CHANNEL         = "channel"
    VIRTUAL_DEV_CONSOLE         = "console"
    VIRTUAL_DEV_VIDEO           = "video"
    VIRTUAL_DEV_CONTROLLER      = "controller"
    VIRTUAL_DEV_WATCHDOG        = "watchdog"
    VIRTUAL_DEV_FILESYSTEM      = "filesystem"
    VIRTUAL_DEV_SMARTCARD       = "smartcard"
    VIRTUAL_DEV_REDIRDEV        = "redirdev"
    VIRTUAL_DEV_MEMBALLOON      = "memballoon"
    VIRTUAL_DEV_TPM             = "tpm"

    # Ordering in this list is important: it will be the order the
    # Guest class outputs XML. So changing this may upset the test suite
    virtual_device_types = [VIRTUAL_DEV_DISK,
                            VIRTUAL_DEV_CONTROLLER,
                            VIRTUAL_DEV_FILESYSTEM,
                            VIRTUAL_DEV_NET,
                            VIRTUAL_DEV_INPUT,
                            VIRTUAL_DEV_GRAPHICS,
                            VIRTUAL_DEV_SERIAL,
                            VIRTUAL_DEV_PARALLEL,
                            VIRTUAL_DEV_CONSOLE,
                            VIRTUAL_DEV_CHANNEL,
                            VIRTUAL_DEV_AUDIO,
                            VIRTUAL_DEV_VIDEO,
                            VIRTUAL_DEV_HOSTDEV,
                            VIRTUAL_DEV_WATCHDOG,
                            VIRTUAL_DEV_SMARTCARD,
                            VIRTUAL_DEV_REDIRDEV,
                            VIRTUAL_DEV_MEMBALLOON,
                            VIRTUAL_DEV_TPM]

    # General device type (disk, interface, etc.)
    virtual_device_type = None
    _XML_INDENT = 4

    def __init__(self, conn, parsexml=None, parsexmlnode=None):
        """
        Initialize device state

        @param conn: libvirt connection to validate device against
        """
        XMLBuilder.__init__(self, conn, parsexml, parsexmlnode)
        self._XML_ROOT_NAME = self.virtual_device_type

        self.alias = VirtualDeviceAlias(conn, parsexmlnode=parsexmlnode)
        self.address = VirtualDeviceAddress(conn, parsexmlnode=parsexmlnode)
        self._XML_SUB_ELEMENTS = ["alias", "address"]

        if not self.virtual_device_type:
            raise ValueError(_("Virtual device type must be set in subclass."))

        if self.virtual_device_type not in self.virtual_device_types:
            raise ValueError(_("Unknown virtual device type '%s'.") %
                             self.virtual_device_type)


    def setup(self, meter=None):
        """
        Perform potentially hazardous device initialization, like
        storage creation or host device reset

        @param meter: Optional progress meter to use
        """
        # Will be overwritten by subclasses if necessary.
        ignore = meter
        return

    def set_address(self, addrstr):
        self.address.set_addrstr(addrstr)


class VirtualDeviceAlias(XMLBuilder):
    _XML_ROOT_NAME = "alias"
    _XML_INDENT = 0

    name = XMLProperty(xpath="./alias/@name")


class VirtualDeviceAddress(XMLBuilder):
    """
    Examples:
    <address type='pci' domain='0x0000' bus='0x00' slot='0x04' function='0x0'/>
    <address type='drive' controller='0' bus='0' unit='0'/>
    <address type='ccid' controller='0' slot='0'/>
    <address type='virtio-serial' controller='1' bus='0' port='4'/>
    """

    ADDRESS_TYPE_PCI           = "pci"
    ADDRESS_TYPE_DRIVE         = "drive"
    ADDRESS_TYPE_VIRTIO_SERIAL = "virtio-serial"
    ADDRESS_TYPE_CCID          = "ccid"
    ADDRESS_TYPE_SPAPR_VIO     = "spapr-vio"

    TYPES = [ADDRESS_TYPE_PCI, ADDRESS_TYPE_DRIVE,
             ADDRESS_TYPE_VIRTIO_SERIAL, ADDRESS_TYPE_CCID,
             ADDRESS_TYPE_SPAPR_VIO]

    _XML_ROOT_NAME = "address"
    _XML_INDENT = 0
    _XML_XPATH_RELATIVE = True
    _XML_PROP_ORDER = ["type", "domain", "bus", "slot", "function"]

    def set_addrstr(self, addrstr):
        try:
            if addrstr.count(":") in [1, 2] and addrstr.count("."):
                self.type = self.ADDRESS_TYPE_PCI
                addrstr, self.function = addrstr.split(".", 1)
                addrstr, self.slot = addrstr.rsplit(":", 1)
                self.domain = "0"
                if addrstr.count(":"):
                    self.domain, self.bus = addrstr.split(":", 1)
            elif addrstr == "spapr-vio":
                self.type = self.ADDRESS_TYPE_SPAPR_VIO
            else:
                raise ValueError(_("Could not determine or unsupported "
                                   "format of '%s'") % addrstr)
        except:
            logging.exception("Error parsing address.")
            return None


    def clear(self):
        self.type = None
        self.bus = None
        self.domain = None
        self.slot = None
        self.function = None
        self.controller = None
        self.unit = None
        self.port = None


    type = XMLProperty(xpath="./address/@type")
    domain = XMLProperty(xpath="./address/@domain", is_int=True)
    bus = XMLProperty(xpath="./address/@bus", is_int=True)
    slot = XMLProperty(xpath="./address/@slot", is_int=True)
    function = XMLProperty(xpath="./address/@function", is_int=True)
    controller = XMLProperty(xpath="./address/@controller", is_int=True)
    unit = XMLProperty(xpath="./address/@unit", is_int=True)
    port = XMLProperty(xpath="./address/@port", is_int=True)