#include "BuddyList.h"
using namespace std;

BuddyList::BuddyList()
{
}

void BuddyList::addBuddy(const string& name, const string& buddy)
{
  // Make sure this buddy isn't already there.
  // We don't want to insert an identical copy of the
  // key/value pair.
  if (!isBuddy(name, buddy)) {
    mBuddies.insert(make_pair(name, buddy));
  }
}

void BuddyList::removeBuddy(const string& name, const string& buddy)
{
  // declare two iterators into the map
  multimap<string, string>::iterator start, end;

  // Obtain the beginning and end of the range of elements with
  // key name. Use both lower_bound() and upper_bound() to demonstrate
  // their use. Otherwise, could just call equal_range().
  start = mBuddies.lower_bound(name);
  end = mBuddies.upper_bound(name); 
  
  // iterate through the elements with key name looking
  // for a value buddy
  for (start; start != end; ++start) {
    if (start->second == buddy) {
      // We found a match! Remove it from the map
      mBuddies.erase(start);
      break;
    }
  }
}

bool BuddyList::isBuddy(const string& name, const string& buddy) const
{
  // declare two iterators into the map
  multimap<string, string>::const_iterator start, end;
  // obtain the beginning and end of the range of elements with
  // key name. Use both lower_bound() and upper_bound() to demonstrate
  // their use. Otherwise, could just call equal_range().
  start = mBuddies.lower_bound(name);
  end = mBuddies.upper_bound(name);

  // iterate through the elements with key name looking
  // for a value buddy. If there are no elements with key name,
  // start equals end, so the loop body doesn't execute.
  for (start; start != end; ++start) {
    if (start->second == buddy) {
      // We found a match!
      return (true);
    }
  }
  // No matches
  return (false);
}

list<string> BuddyList::getBuddies(const string& name) const
{
  // Create a variable to store the pair of iterators.
  pair<multimap<string, string>::const_iterator,
    multimap<string, string>::const_iterator> its;

  // Obtain the pair of iterators marking the range containing
  // elements with key name
  its = mBuddies.equal_range(name);

  // Create a list with all the names in the range
  // (all the buddies of name).
  list<string> buddies;
  for (its.first; its.first != its.second; ++its.first) {
    buddies.push_back((its.first)->second);
  }

  return (buddies);
}