prefix-discovery.hpp

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil -*- */
#ifndef NDN_PREFIX_DISCOVERY_HPP
#define NDN_PREFIX_DISCOVERY_HPP

#include <ndn-cpp/face.hpp>

namespace ndntools {

class PrefixDiscovery {
public:
  typedef ndn::func_lib::function<void
    (const ndn::ptr_lib::shared_ptr<std::vector<ndn::Name> >& prefixes)> OnPrefixes;

  PrefixDiscovery
    (OnPrefixes onPrefixes, ndn::Face* face,
     ndn::Milliseconds periodMilliseconds = 60000.0)
  : impl_(new Impl(onPrefixes, face, periodMilliseconds))
  {
  }

  void
  start() { impl_->start(); }

  void
  stop() { impl_->stop(); }

private:
  class Impl : public ndn::ptr_lib::enable_shared_from_this<Impl> {
  public:
    Impl(OnPrefixes onPrefixes, ndn::Face* face,
         ndn::Milliseconds periodMilliseconds)
    : onPrefixes_(onPrefixes), face_(face),
      periodMilliseconds_(periodMilliseconds), isEnabled_(false)
    {
    }

    void
    start()
    {
      if (isEnabled_)
        // Already started.
        return;

      isEnabled_ = true;
      express();
    }

    void stop() { isEnabled_ = false; }

  private:
    void
    express();

    void
    onData
      (const ndn::ptr_lib::shared_ptr<const ndn::Interest>& interest,
       const ndn::ptr_lib::shared_ptr<ndn::Data>& data);

    void
    onTimeout(const ndn::ptr_lib::shared_ptr<const ndn::Interest>& interest);

    OnPrefixes onPrefixes_;
    ndn::Face* face_;
    ndn::Milliseconds periodMilliseconds_;
    std::vector<ndn::Name> prefixes_;
    bool isEnabled_;
  };

  ndn::ptr_lib::shared_ptr<Impl> impl_;
};

}

#endif