# Copyright 2017 AT&T Intellectual Property.  All other rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Model for MaaS API boot_resource type."""

import drydock_provisioner.error as errors
import drydock_provisioner.drivers.node.maasdriver.models.base as model_base


class BootResource(model_base.ResourceBase):

    resource_url = 'boot-resources/{resource_id}/'
    fields = [
        'resource_id',
        'name',
        'type',
        'subarches',
        'architecture',
    ]
    json_fields = [
        'name',
        'type',
        'subarches',
        'architecture',
    ]

    def __init__(self, api_client, **kwargs):
        super().__init__(api_client, **kwargs)

    def get_image_name(self):
        """Return the name that would be specified in a deployment.

        Return None if this is not an ubuntu image, otherwise
        the distro series name
        """
        (os, release) = self.name.split('/')

        # Only supply image names for ubuntu-based images
        if os == 'ubuntu':
            return release
        else:
            # Non-ubuntu images such as the uefi bootloader
            # should never be selectable
            return None

    def get_kernel_name(self):
        """Return the kernel name that would be specified in a deployment."""
        (_, kernel) = self.architecture.split('/')

        return kernel


class BootResources(model_base.ResourceCollectionBase):

    collection_url = 'boot-resources/'
    collection_resource = BootResource

    def __init__(self, api_client, **kwargs):
        super().__init__(api_client)

    def is_importing(self):
        """Check if boot resources are importing."""
        url = self.interpolate_url()

        self.logger.debug("Checking if boot resources are importing.")
        resp = self.api_client.get(url, op='is_importing')

        if resp.status_code == 200:
            resp_json = resp.json()
            self.logger.debug("Boot resource importing status: %s" % resp_json)
            return resp_json
        else:
            msg = "Error checking import status of boot resources: %s - %s" % (
                resp.status_code, resp.text)
            self.logger.error(msg)
            raise errors.DriverError(msg)

    def get_available_images(self):
        """Get list of available deployable images."""
        image_options = list()
        for k, v in self.resources.items():
            if v.get_image_name() not in image_options:
                image_options.append(v.get_image_name())
        return image_options

    def get_available_kernels(self, image_name):
        """Get kernels available for image_name

        Return list of kernel names available for
        ``image_name``.

        :param image_name: str image_name (e.g. 'xenial')
        """
        kernel_options = list()
        for k, v in self.resources.items():
            if (v.get_image_name() == image_name
                    and v.get_kernel_name() not in kernel_options):
                kernel_options.append(v.get_kernel_name())
        return kernel_options