/*
 * Copyright (c) 2014, 2015 Red Hat, Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */

package org.opendaylight.neutron.transcriber;

import java.util.List;

import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
import org.opendaylight.neutron.spi.INeutronFirewallRuleCRUD;
import org.opendaylight.neutron.spi.NeutronFirewallRule;

import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.IpVersionBase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.IpVersionV4;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.IpVersionV6;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.NeutronUtils.FwProtocolMapper;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.fwaas.rev150712.ActionAllow;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.fwaas.rev150712.ActionBase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.fwaas.rev150712.ActionDeny;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.fwaas.rev150712.FirewallRuleAttributes.Protocol;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.fwaas.rev150712.rules.attributes.FirewallRules;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.fwaas.rev150712.rules.attributes.firewall.rules.FirewallRule;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.fwaas.rev150712.rules.attributes.firewall.rules.FirewallRuleBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.Neutron;
import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.types.rev160517.IpPrefixOrAddress;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;


import com.google.common.collect.ImmutableBiMap;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceRegistration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class NeutronFirewallRuleInterface extends AbstractNeutronInterface<FirewallRule, FirewallRules, NeutronFirewallRule> implements INeutronFirewallRuleCRUD {
    private static final Logger LOGGER = LoggerFactory.getLogger(NeutronFirewallRuleInterface.class);

    private static final ImmutableBiMap<Class<? extends ActionBase>, String> ACTION_MAP
            = new ImmutableBiMap.Builder<Class<? extends ActionBase>, String>()
            .put(ActionAllow.class, "allow")
            .put(ActionDeny.class, "deny")
            .build();

    private static final ImmutableBiMap<Class<? extends IpVersionBase>, Integer> IP_VERSION_MAP
            = new ImmutableBiMap.Builder<Class<? extends IpVersionBase>, Integer>()
            .put(IpVersionV4.class, Integer.valueOf(4))
            .put(IpVersionV6.class, Integer.valueOf(6))
            .build();

    NeutronFirewallRuleInterface(ProviderContext providerContext) {
        super(providerContext);
    }

    @Override
    protected List<FirewallRule> getDataObjectList(FirewallRules rules) {
        return rules.getFirewallRule();
    }

    @Override
    protected InstanceIdentifier<FirewallRule> createInstanceIdentifier(FirewallRule item) {
        return InstanceIdentifier.create(Neutron.class)
                .child(FirewallRules.class)
                .child(FirewallRule.class, item.getKey());
    }

    @Override
    protected InstanceIdentifier<FirewallRules> createInstanceIdentifier() {
        return InstanceIdentifier.create(Neutron.class)
                .child(FirewallRules.class);
    }

    protected NeutronFirewallRule fromMd(FirewallRule rule) {
        final NeutronFirewallRule answer = new NeutronFirewallRule();
        if (rule.getUuid() != null) {
            answer.setID(rule.getUuid().getValue());
        }
        if (rule.getName() != null) {
            answer.setFirewallRuleName(rule.getName());
        }
        if (rule.getTenantId() != null) {
            answer.setTenantID(rule.getTenantId());
        }
        if (rule.isShared() != null) {
            answer.setFirewallRuleIsShared(rule.isShared());
        }
        if (rule.isEnabled() != null) {
            answer.setFirewallRuleIsEnabled(rule.isEnabled());
        }
        if (rule.getFirewallPolicyId() != null) {
            answer.setFirewallRulePolicyID(rule.getFirewallPolicyId().getValue());
        }
        if (rule.getProtocol() != null) {
            final Protocol protocol = rule.getProtocol();
            if (protocol.getUint8() != null) {
                // uint8
                answer.setFirewallRuleProtocol(protocol.getUint8().toString());
            } else {
                // symbolic protocol name
                answer.setFirewallRuleProtocol(FwProtocolMapper.getName(protocol.getIdentityref()));
            }
        }
        if (rule.getIpVersion() != null) {
            answer.setFirewallRuleIpVer(IP_VERSION_MAP.get(rule.getIpVersion()));
        }
        if (rule.getSourceIpAddr() != null) {
            answer.setFirewallRuleSrcIpAddr(String.valueOf(rule.getSourceIpAddr().getValue()));
        }
        if (rule.getDestinationIpAddr() != null) {
            answer.setFirewallRuleDstIpAddr(String.valueOf(rule.getDestinationIpAddr().getValue()));
        }
        if (rule.getSourcePortRangeMin() != null) {
            answer.setFirewallRuleSrcPortRangeMin(rule.getSourcePortRangeMin());
        }
        if (rule.getSourcePortRangeMax() != null) {
            answer.setFirewallRuleSrcPortRangeMax(rule.getSourcePortRangeMax());
        }
        if (rule.getDestinationPortRangeMin() != null) {
            answer.setFirewallRuleDstPortRangeMin(rule.getDestinationPortRangeMin());
        }
        if (rule.getDestinationPortRangeMax() != null) {
            answer.setFirewallRuleDstPortRangeMax(rule.getDestinationPortRangeMax());
        }
        if (rule.getPosition() != null) {
            answer.setFirewallRulePosition(Integer.valueOf(rule.getPosition().intValue()));
        }
        if (rule.getAction() != null) {
            answer.setFirewallRuleAction(ACTION_MAP.get(rule.getAction()));
        }
        return answer;
    }

    @Override
    protected FirewallRule toMd(NeutronFirewallRule rule) {
        final FirewallRuleBuilder ruleBuilder = new FirewallRuleBuilder();
        if (rule.getID() != null) {
            ruleBuilder.setUuid(toUuid(rule.getID()));
        }
        if (rule.getFirewallRuleName() != null) {
            ruleBuilder.setName(rule.getFirewallRuleName());
        }
        if (rule.getTenantID() != null) {
            ruleBuilder.setTenantId(toUuid(rule.getTenantID()));
        }
        if (rule.getFirewallRuleIsShared() != null) {
            ruleBuilder.setShared(rule.getFirewallRuleIsShared());
        }
        if (rule.getFirewallRuleIsEnabled() != null) {
            ruleBuilder.setEnabled(rule.getFirewallRuleIsEnabled());
        }
        if (rule.getFirewallRulePolicyID() != null) {
            ruleBuilder.setFirewallPolicyId(toUuid(rule.getFirewallRulePolicyID()));
        }
        if (rule.getFirewallRuleProtocol() != null) {
            final String protocolString = rule.getFirewallRuleProtocol();
            final Protocol protocol = new Protocol(protocolString.toCharArray());
            ruleBuilder.setProtocol(protocol);
        }
        if (rule.getFirewallRuleIpVer() != null) {
            final ImmutableBiMap<Integer, Class<? extends IpVersionBase>> mapper = IP_VERSION_MAP.inverse();
            ruleBuilder.setIpVersion(mapper.get(rule.getFirewallRuleIpVer()));
        }
        if (rule.getFirewallRuleSrcIpAddr() != null) {
            final IpPrefixOrAddress ipAddress = new IpPrefixOrAddress(rule.getFirewallRuleSrcIpAddr().toCharArray());
            ruleBuilder.setSourceIpAddr(ipAddress);
        }
        if (rule.getFirewallRuleDstIpAddr() != null) {
            final IpPrefixOrAddress ipAddress = new IpPrefixOrAddress(rule.getFirewallRuleDstIpAddr().toCharArray());
            ruleBuilder.setDestinationIpAddr(ipAddress);
        }
        if (rule.getFirewallRuleSrcPortRangeMin() != null) {
            ruleBuilder.setSourcePortRangeMin(rule.getFirewallRuleSrcPortRangeMin());
        }
        if (rule.getFirewallRuleSrcPortRangeMax() != null) {
            ruleBuilder.setSourcePortRangeMax(rule.getFirewallRuleSrcPortRangeMax());
        }
        if (rule.getFirewallRuleDstPortRangeMin() != null) {
            ruleBuilder.setDestinationPortRangeMin(rule.getFirewallRuleDstPortRangeMin());
        }
        if (rule.getFirewallRuleDstPortRangeMax() != null) {
            ruleBuilder.setDestinationPortRangeMax(rule.getFirewallRuleDstPortRangeMax());
        }
        if (rule.getFirewallRulePosition() != null) {
            ruleBuilder.setPosition(rule.getFirewallRulePosition().shortValue());
        }
        if (rule.getFirewallRuleAction() != null) {
            final ImmutableBiMap<String, Class<? extends ActionBase>> mapper = ACTION_MAP.inverse();
            ruleBuilder.setAction(mapper.get(rule.getFirewallRuleAction()));
        }
        return ruleBuilder.build();
    }

    @Override
    protected FirewallRule toMd(String uuid) {
        final FirewallRuleBuilder ruleBuilder = new FirewallRuleBuilder();
        ruleBuilder.setUuid(toUuid(uuid));
        return ruleBuilder.build();
    }

    public static void registerNewInterface(BundleContext context,
                                            ProviderContext providerContext,
                                            List<ServiceRegistration<?>> registrations) {
        final NeutronFirewallRuleInterface neutronFirewallRuleInterface = new NeutronFirewallRuleInterface(providerContext);
        final ServiceRegistration<INeutronFirewallRuleCRUD> neutronFirewallRuleInterfaceRegistration = context.registerService(INeutronFirewallRuleCRUD.class, neutronFirewallRuleInterface, null);
        if(neutronFirewallRuleInterfaceRegistration != null) {
            registrations.add(neutronFirewallRuleInterfaceRegistration);
        }
    }
}