/*
 * Copyright (c) 2014 Cisco Systems, 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.controller.sample.l2switch.md;

import org.opendaylight.controller.sample.l2switch.md.addresstracker.AddressTracker;
import org.opendaylight.controller.sample.l2switch.md.flow.FlowWriterService;
import org.opendaylight.controller.sample.l2switch.md.flow.FlowWriterServiceImpl;
import org.opendaylight.controller.sample.l2switch.md.inventory.InventoryService;
import org.opendaylight.controller.sample.l2switch.md.packet.PacketHandler;
import org.opendaylight.controller.sample.l2switch.md.topology.NetworkGraphDijkstra;
import org.opendaylight.controller.sample.l2switch.md.topology.NetworkGraphService;
import org.opendaylight.controller.sample.l2switch.md.topology.TopologyLinkDataChangeHandler;
import org.opendaylight.controller.sal.binding.api.AbstractBindingAwareConsumer;
import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
import org.opendaylight.controller.sal.binding.api.NotificationService;
import org.opendaylight.controller.sal.binding.api.data.DataBrokerService;
import org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.PacketProcessingService;
import org.opendaylight.yangtools.concepts.Registration;
import org.opendaylight.yangtools.yang.binding.NotificationListener;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * L2SwitchProvider serves as the Activator for our L2Switch OSGI bundle.
 */
public class L2SwitchProvider extends AbstractBindingAwareConsumer
                              implements AutoCloseable {

  private final static Logger _logger = LoggerFactory.getLogger(L2SwitchProvider.class);

  private Registration<NotificationListener> listenerRegistration;
  private AddressTracker addressTracker;
  private TopologyLinkDataChangeHandler topologyLinkDataChangeHandler;


  /**
   * Setup the L2Switch.
   * @param consumerContext  The context of the L2Switch.
   */
  @Override
  public void onSessionInitialized(BindingAwareBroker.ConsumerContext consumerContext) {
    DataBrokerService dataService = consumerContext.<DataBrokerService>getSALService(DataBrokerService.class);
    addressTracker = new AddressTracker(dataService);

    NetworkGraphService networkGraphService = new NetworkGraphDijkstra();
    FlowWriterService flowWriterService = new FlowWriterServiceImpl(dataService, networkGraphService);

    NotificationService notificationService =
            consumerContext.<NotificationService>getSALService(NotificationService.class);
    PacketProcessingService packetProcessingService =
            consumerContext.<PacketProcessingService>getRpcService(PacketProcessingService.class);
    PacketHandler packetHandler = new PacketHandler();
    packetHandler.setAddressTracker(addressTracker);
    packetHandler.setFlowWriterService(flowWriterService);
    packetHandler.setPacketProcessingService(packetProcessingService);
    packetHandler.setInventoryService(new InventoryService(dataService));

    this.listenerRegistration = notificationService.registerNotificationListener(packetHandler);
    this.topologyLinkDataChangeHandler = new TopologyLinkDataChangeHandler(dataService, networkGraphService);
    topologyLinkDataChangeHandler.registerAsDataChangeListener();
  }

  /**
   * Cleanup the L2Switch.
   * @throws Exception  occurs when the NotificationListener is closed
   */
  @Override
  public void close() throws Exception {
    if (listenerRegistration != null)
      listenerRegistration.close();
  }
}