# Copyright 2013 OpenStack LLC.
# All 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.
#
# vim: tabstop=4 shiftwidth=4 softtabstop=4

import logging

from quantumclient.openstack.common.gettextutils import _
from quantumclient.quantum import v2_0 as quantumV20
from quantumclient.quantum.v2_0 import network
from quantumclient.quantum.v2_0 import router
PERFECT_TIME_FORMAT = "%Y-%m-%dT%H:%M:%S.%f"


class AddNetworkToDhcpAgent(quantumV20.QuantumCommand):
    """Add a network to a DHCP agent."""

    log = logging.getLogger(__name__ + '.AddNetworkToDhcpAgent')

    def get_parser(self, prog_name):
        parser = super(AddNetworkToDhcpAgent, self).get_parser(prog_name)
        parser.add_argument(
            'dhcp_agent',
            help='ID of the DHCP agent')
        parser.add_argument(
            'network',
            help='network to add')
        return parser

    def run(self, parsed_args):
        self.log.debug('run(%s)' % parsed_args)
        quantum_client = self.get_client()
        quantum_client.format = parsed_args.request_format
        _net_id = quantumV20.find_resourceid_by_name_or_id(
            quantum_client, 'network', parsed_args.network)
        quantum_client.add_network_to_dhcp_agent(parsed_args.dhcp_agent,
                                                 {'network_id': _net_id})
        print >>self.app.stdout, (
            _('Added network %s to DHCP agent') % parsed_args.network)


class RemoveNetworkFromDhcpAgent(quantumV20.QuantumCommand):
    """Remove a network from a DHCP agent."""
    log = logging.getLogger(__name__ + '.RemoveNetworkFromDhcpAgent')

    def get_parser(self, prog_name):
        parser = super(RemoveNetworkFromDhcpAgent, self).get_parser(prog_name)
        parser.add_argument(
            'dhcp_agent',
            help='ID of the DHCP agent')
        parser.add_argument(
            'network',
            help='network to remove')
        return parser

    def run(self, parsed_args):
        self.log.debug('run(%s)' % parsed_args)
        quantum_client = self.get_client()
        quantum_client.format = parsed_args.request_format
        _net_id = quantumV20.find_resourceid_by_name_or_id(
            quantum_client, 'network', parsed_args.network)
        quantum_client.remove_network_from_dhcp_agent(
            parsed_args.dhcp_agent, _net_id)
        print >>self.app.stdout, (
            _('Removed network %s to DHCP agent') % parsed_args.network)


class ListNetworksOnDhcpAgent(network.ListNetwork):
    """List the networks on a DHCP agent."""

    log = logging.getLogger(__name__ + '.ListNetworksOnDhcpAgent')
    unknown_parts_flag = False

    def get_parser(self, prog_name):
        parser = super(ListNetworksOnDhcpAgent,
                       self).get_parser(prog_name)
        parser.add_argument(
            'dhcp_agent',
            help='ID of the DHCP agent')
        return parser

    def call_server(self, quantum_client, search_opts, parsed_args):
        data = quantum_client.list_networks_on_dhcp_agent(
            parsed_args.dhcp_agent, **search_opts)
        return data


class ListDhcpAgentsHostingNetwork(quantumV20.ListCommand):
    """List DHCP agents hosting a network."""

    resource = 'agent'
    _formatters = {}
    log = logging.getLogger(__name__ + '.ListDhcpAgentsHostingNetwork')
    list_columns = ['id', 'host', 'admin_state_up', 'alive']
    unknown_parts_flag = False

    def get_parser(self, prog_name):
        parser = super(ListDhcpAgentsHostingNetwork,
                       self).get_parser(prog_name)
        parser.add_argument(
            'network',
            help='network to query')
        return parser

    def extend_list(self, data, parsed_args):
        for agent in data:
            agent['alive'] = ":-)" if agent['alive'] else 'xxx'

    def call_server(self, quantum_client, search_opts, parsed_args):
        _id = quantumV20.find_resourceid_by_name_or_id(quantum_client,
                                                       'network',
                                                       parsed_args.network)
        search_opts['network'] = _id
        data = quantum_client.list_dhcp_agent_hosting_networks(**search_opts)
        return data


class AddRouterToL3Agent(quantumV20.QuantumCommand):
    """Add a router to a L3 agent."""

    log = logging.getLogger(__name__ + '.AddRouterToL3Agent')

    def get_parser(self, prog_name):
        parser = super(AddRouterToL3Agent, self).get_parser(prog_name)
        parser.add_argument(
            'l3_agent',
            help='ID of the L3 agent')
        parser.add_argument(
            'router',
            help='router to add')
        return parser

    def run(self, parsed_args):
        self.log.debug('run(%s)' % parsed_args)
        quantum_client = self.get_client()
        quantum_client.format = parsed_args.request_format
        _id = quantumV20.find_resourceid_by_name_or_id(
            quantum_client, 'router', parsed_args.router)
        quantum_client.add_router_to_l3_agent(parsed_args.l3_agent,
                                              {'router_id': _id})
        print >>self.app.stdout, (
            _('Added router %s to L3 agent') % parsed_args.router)


class RemoveRouterFromL3Agent(quantumV20.QuantumCommand):
    """Remove a router from a L3 agent."""

    log = logging.getLogger(__name__ + '.RemoveRouterFromL3Agent')

    def get_parser(self, prog_name):
        parser = super(RemoveRouterFromL3Agent, self).get_parser(prog_name)
        parser.add_argument(
            'l3_agent',
            help='ID of the L3 agent')
        parser.add_argument(
            'router',
            help='router to remove')
        return parser

    def run(self, parsed_args):
        self.log.debug('run(%s)' % parsed_args)
        quantum_client = self.get_client()
        quantum_client.format = parsed_args.request_format
        _id = quantumV20.find_resourceid_by_name_or_id(
            quantum_client, 'router', parsed_args.router)
        quantum_client.remove_router_from_l3_agent(
            parsed_args.l3_agent, _id)
        print >>self.app.stdout, (
            _('Removed Router %s to L3 agent') % parsed_args.router)


class ListRoutersOnL3Agent(quantumV20.ListCommand):
    """List the routers on a L3 agent."""

    log = logging.getLogger(__name__ + '.ListRoutersOnL3Agent')
    _formatters = {'external_gateway_info':
                   router._format_external_gateway_info}
    list_columns = ['id', 'name', 'external_gateway_info']
    resource = 'router'
    unknown_parts_flag = False

    def get_parser(self, prog_name):
        parser = super(ListRoutersOnL3Agent,
                       self).get_parser(prog_name)
        parser.add_argument(
            'l3_agent',
            help='ID of the L3 agent to query')
        return parser

    def call_server(self, quantum_client, search_opts, parsed_args):
        data = quantum_client.list_routers_on_l3_agent(
            parsed_args.l3_agent, **search_opts)
        return data


class ListL3AgentsHostingRouter(quantumV20.ListCommand):
    """List L3 agents hosting a router."""

    resource = 'agent'
    _formatters = {}
    log = logging.getLogger(__name__ + '.ListL3AgentsHostingRouter')
    list_columns = ['id', 'host', 'admin_state_up', 'alive', 'default']
    unknown_parts_flag = False

    def get_parser(self, prog_name):
        parser = super(ListL3AgentsHostingRouter,
                       self).get_parser(prog_name)
        parser.add_argument('router',
                            help='router to query')
        return parser

    def extend_list(self, data, parsed_args):
        for agent in data:
            agent['alive'] = ":-)" if agent['alive'] else 'xxx'

    def call_server(self, quantum_client, search_opts, parsed_args):
        _id = quantumV20.find_resourceid_by_name_or_id(quantum_client,
                                                       'router',
                                                       parsed_args.router)
        search_opts['router'] = _id
        data = quantum_client.list_l3_agent_hosting_routers(**search_opts)
        return data