#include<list>
#include "nlsr_fe.hpp"
#include "nlsr_fib.hpp"
#include "nlsr_nhl.hpp"
#include "nlsr.hpp"

using namespace std;
using namespace ndn;

static bool
fibEntryNameCompare(FibEntry& fe, string name)
{
	return fe.getName() == name ;
}

void 
Fib::cancelScheduledFeExpiringEvent(nlsr& pnlsr, EventId eid)
{
	pnlsr.getScheduler().cancelEvent(eid);
}


ndn::EventId 
Fib::scheduleFibEntryRefreshing(nlsr& pnlsr, string name, int feSeqNum, int refreshTime)
{
		return pnlsr.getScheduler().scheduleEvent(ndn::time::seconds(refreshTime),
				 ndn::bind(&Fib::refreshFibEntry,this,name,feSeqNum));
}

void
Fib::refreshFibEntry(string name, int feSeqNum)
{
	
}

void 
Fib::removeFromFib(nlsr& pnlsr, string name)
{
	std::list<FibEntry >::iterator it = std::find_if( fibTable.begin(), 
									       fibTable.end(), bind(&fibEntryNameCompare, _1, name));
  if( it != fibTable.end() )
  {
  		for(std::list<NextHop>::iterator nhit=(*it).getNhl().getNextHopList().begin(); 
  		                    nhit != (*it).getNhl().getNextHopList().begin(); nhit++)
  		{
  			//remove entry from NDN-FIB
  		}
  		cancelScheduledFeExpiringEvent(pnlsr, (*it).getFeExpiringEventId());
  		fibTable.erase(it);
  }
}


void 
Fib::updateFib(nlsr& pnlsr,string name, Nhl& nextHopList, int maxFacesPerPrefix)
{
	int startFace=0;
	int endFace=getNumberOfFacesForName(nextHopList,maxFacesPerPrefix);
	std::list<FibEntry >::iterator it = std::find_if( fibTable.begin(), 
									       fibTable.end(), bind(&fibEntryNameCompare, _1, name));
  if( it != fibTable.end() )
  {
  		nextHopList.sortNhl();
  		if ( !(*it).isEqualNextHops(nextHopList) ) 
  		{
  			std::list<NextHop>::iterator nhit=nextHopList.getNextHopList().begin();
  			(*it).getNhl().addNextHop((*nhit));
			removeFibEntryHop((*it).getNhl(),(*nhit).getConnectingFace());
			startFace++;
  			nhit++;
  			for( int i=startFace;i< endFace;nhit++,i++)
  			{
  				(*it).getNhl().addNextHop((*nhit));
  			}

  			(*it).setTimeToRefresh(fibEntryRefreshTime);
  		}
  		(*it).getNhl().sortNhl();
  		cancelScheduledFeExpiringEvent(pnlsr, (*it).getFeExpiringEventId());
  		(*it).setFeSeqNo((*it).getFeSeqNo()+1);
  		(*it).setFeExpiringEventId(scheduleFibEntryRefreshing(pnlsr, 
  		                            (*it).getName() ,
  		                            (*it).getFeSeqNo(),fibEntryRefreshTime));
  		//update NDN-FIB
  }
  else
  {
  		nextHopList.sortNhl();
  		FibEntry newEntry(name);
  		std::list<NextHop>::iterator nhit=nextHopList.getNextHopList().begin();
  		for(int i=startFace; i< endFace ; i++)
  		{
  			newEntry.getNhl().addNextHop((*nhit));
  			++nhit;
  		}
  		newEntry.getNhl().sortNhl();
  		newEntry.setTimeToRefresh(fibEntryRefreshTime);
  		newEntry.setFeSeqNo(1);
  		fibTable.push_back(newEntry);	

  		//cancelScheduledFeExpiringEvent(pnlsr, newEntry().getFeExpiringEventId());
  		
  		//Update NDN-FIB
  }
}


void Fib::cleanFib(nlsr& pnlsr)
{
	for( std::list<FibEntry >::iterator it=fibTable.begin(); it != fibTable.end();
	                                                                         ++it)
	{
		for(std::list<NextHop>::iterator nhit=(*it).getNhl().getNextHopList().begin(); 
  		                    nhit != (*it).getNhl().getNextHopList().begin(); nhit++)
  		{
  			cancelScheduledFeExpiringEvent(pnlsr,(*it).getFeExpiringEventId());
  			//remove entry from NDN-FIB
  		}
	}

	if ( fibTable.size() > 0 )
	{
		fibTable.clear();
	}
}


void 
Fib::removeFibEntryHop(Nhl& nl, int doNotRemoveHopFaceId)
{
	for( std::list<NextHop >::iterator it=nl.getNextHopList().begin(); 
	                                      it != nl.getNextHopList().end();   ++it)
	{
		if ( (*it).getConnectingFace() != doNotRemoveHopFaceId )
		{
			nl.getNextHopList().erase(it);
		}
	}
}


int 
Fib::getNumberOfFacesForName(Nhl& nextHopList, int maxFacesPerPrefix)
{
	int endFace=0;
  	if((maxFacesPerPrefix == 0) || (nextHopList.getNhlSize() <= maxFacesPerPrefix))
  	{
  		return nextHopList.getNhlSize();
  	}
  	else
  	{
  		return maxFacesPerPrefix;
  	}

  	return endFace;
}

void
Fib::printFib()
{
	cout<<"-------------------FIB-----------------------------"<<endl;
	for(std::list<FibEntry>::iterator it = fibTable.begin(); it!=fibTable.end();
	                                                                         ++it)
	{
		cout<<(*it);
	}
}