#   Copyright 2017 Intel, Inc.
#
#   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.
#

import os

from rsdclient.common import base
from rsdclient.common import utils


class NodeManager(base.Manager):
    # resource_class = Node
    _resource_name = 'nodes'

    def __init__(self, *args, **kwargs):
        super(NodeManager, self).__init__(*args, **kwargs)
        self.nodes_path = self.client._nodes_path

    def _get_node_uri(self, node_id):
        return os.path.join(self.nodes_path, node_id)

    def compose(self, name=None, description=None, processor_req=None,
                memory_req=None, remote_drive_req=None, local_drive_req=None,
                ethernet_interface_req=None):
        node_uri = self.client.get_node_collection().compose_node(
            name=name, description=description, processor_req=processor_req,
            memory_req=memory_req, remote_drive_req=remote_drive_req,
            local_drive_req=local_drive_req,
            ethernet_interface_req=ethernet_interface_req)
        return node_uri[len(self.nodes_path) + 1:]

    def delete(self, node_id):
        self.client.get_node(self._get_node_uri(node_id)).delete_node()

    def show(self, node_id):
        node = self.client.get_node(self._get_node_uri(node_id))
        node_info = utils.extract_attr(node)

        node_info['allowed_attach_endpoints'] = \
            list(node.get_allowed_attach_endpoints())
        node_info['allowed_detach_endpoints'] = \
            list(node.get_allowed_detach_endpoints())
        node_info['allowed_boot_source'] = \
            list(node.get_allowed_node_boot_source_values())
        node_info['allowed_reset_node_values'] = \
            list(node.get_allowed_reset_node_values())

        return node_info

    def list(self):
        node_collection = self.client.get_node_collection()
        nodes = [utils.extract_attr(self.client.get_node(node_uri))
                 for node_uri in node_collection.members_identities]
        node_info_table = utils.print_dict(
            nodes, ["Identity", "Name", "UUID", "Description"])
        return node_info_table

    def attach(self, node_id, endpoint=None, capacity=None):
        node = self.client.get_node(self._get_node_uri(node_id))
        node.attach_endpoint(endpoint, capacity)

    def detach(self, node_id, endpoint):
        node = self.client.get_node(self._get_node_uri(node_id))
        node.detach_endpoint(endpoint)