# Copyright 2013 Big Switch Networks
# 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.
#

from neutronclient.i18n import _
from neutronclient.neutron import v2_0 as neutronv20


class ListFirewall(neutronv20.ListCommand):
    """List firewalls that belong to a given tenant."""

    resource = 'firewall'
    list_columns = ['id', 'name', 'firewall_policy_id']
    _formatters = {}
    pagination_support = True
    sorting_support = True


class ShowFirewall(neutronv20.ShowCommand):
    """Show information of a given firewall."""

    resource = 'firewall'


class CreateFirewall(neutronv20.CreateCommand):
    """Create a firewall."""

    resource = 'firewall'

    def add_known_arguments(self, parser):
        parser.add_argument(
            'firewall_policy_id', metavar='POLICY',
            help=_('Firewall policy name or ID.'))
        parser.add_argument(
            '--name',
            help=_('Name for the firewall.'))
        parser.add_argument(
            '--description',
            help=_('Description for the firewall rule.'))
        parser.add_argument(
            '--admin-state-down',
            dest='admin_state',
            action='store_false',
            help=_('Set admin state up to false.'))
        parser.add_argument(
            '--router',
            dest='routers',
            metavar='ROUTER',
            action='append',
            help=_('Firewall associated router names or IDs (requires FWaaS '
                   'router insertion extension, this option can be repeated)'))

    def args2body(self, parsed_args):
        client = self.get_client()
        _policy_id = neutronv20.find_resourceid_by_name_or_id(
            client, 'firewall_policy',
            parsed_args.firewall_policy_id)
        body = {'firewall_policy_id': _policy_id,
                'admin_state_up': parsed_args.admin_state, }
        if parsed_args.routers:
            body['router_ids'] = [
                neutronv20.find_resourceid_by_name_or_id(client, 'router', r)
                for r in parsed_args.routers]
        neutronv20.update_dict(parsed_args, body,
                               ['name', 'description', 'tenant_id'])
        return {self.resource: body}


class UpdateFirewall(neutronv20.UpdateCommand):
    """Update a given firewall."""

    resource = 'firewall'

    def add_known_arguments(self, parser):
        parser.add_argument(
            '--policy', metavar='POLICY',
            help=_('Firewall policy name or ID.'))
        router_sg = parser.add_mutually_exclusive_group()
        router_sg.add_argument(
            '--router',
            dest='routers',
            metavar='ROUTER',
            action='append',
            help=_('Firewall associated router names or IDs (requires FWaaS '
                   'router insertion extension, this option can be repeated)'))
        router_sg.add_argument(
            '--no-routers',
            action='store_true',
            help=_('Associate no routers with the firewall (requires FWaaS '
                   'router insertion extension)'))

    def args2body(self, parsed_args):
        data = {}
        client = self.get_client()
        if parsed_args.policy:
            _policy_id = neutronv20.find_resourceid_by_name_or_id(
                client, 'firewall_policy',
                parsed_args.policy)
            data['firewall_policy_id'] = _policy_id
        if parsed_args.routers:
            data['router_ids'] = [
                neutronv20.find_resourceid_by_name_or_id(client, 'router', r)
                for r in parsed_args.routers]
        elif parsed_args.no_routers:
            data['router_ids'] = []
        return {self.resource: data}


class DeleteFirewall(neutronv20.DeleteCommand):
    """Delete a given firewall."""

    resource = 'firewall'