"""Init file for HassIO host rest api."""
import logging

import voluptuous as vol

from .util import api_process_hostcontroll, api_process, api_validate
from ..const import ATTR_VERSION

_LOGGER = logging.getLogger(__name__)

UNKNOWN = 'unknown'

SCHEMA_VERSION = vol.Schema({
    vol.Optional(ATTR_VERSION): vol.Coerce(str),
})


class APIHost(object):
    """Handle rest api for host functions."""

    def __init__(self, config, loop, host_controll):
        """Initialize host rest api part."""
        self.config = config
        self.loop = loop
        self.host_controll = host_controll

    @api_process
    async def info(self, request):
        """Return host information."""
        if not self.host_controll.active:
            info = {
                'os': UNKNOWN,
                'version': UNKNOWN,
                'current': UNKNOWN,
                'level': 0,
                'hostname': UNKNOWN,
            }
            return info

        return await self.host_controll.info()

    @api_process_hostcontroll
    def reboot(self, request):
        """Reboot host."""
        return self.host_controll.reboot()

    @api_process_hostcontroll
    def shutdown(self, request):
        """Poweroff host."""
        return self.host_controll.shutdown()

    @api_process_hostcontroll
    async def update(self, request):
        """Update host OS."""
        body = await api_validate(SCHEMA_VERSION, request)
        version = body.get(ATTR_VERSION)

        if version == self.host_controll.version:
            raise RuntimeError("Version is already in use")

        return await self.host_controll.host_update(version=version)