boost-info-parser.hpp

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

#include <istream>
#include <string>
#include <vector>
#include <utility>
#include <ndn-cpp/common.hpp>

namespace ndn {

class BoostInfoTree
{
public:
  BoostInfoTree(const std::string& value = "", BoostInfoTree* parent = 0)
  : value_(value), parent_(parent), lastChild_(0)
  {
  }

  void
  addSubtree
    (const std::string& treeName, ptr_lib::shared_ptr<BoostInfoTree> newTree);

  const BoostInfoTree&
  createSubtree(const std::string& treeName, const std::string& value = "");

  std::vector<const BoostInfoTree*>
  operator [] (const std::string& key) const;

  const std::string*
  getFirstValue(const std::string& key) const
  {
    std::vector<const BoostInfoTree*> list = (*this)[key];
    if (list.size() >= 1)
      return &list[0]->value_;
    else
      return 0;
  }

  const std::string&
  getValue() const { return value_; }

  BoostInfoTree*
  getParent() { return parent_; }

  BoostInfoTree*
  getLastChild() { return lastChild_; }

  std::string
  prettyPrint(int indentLevel = 1) const;

private:
  std::vector<ptr_lib::shared_ptr<BoostInfoTree> >*
  find(const std::string& treeName);

  static std::vector<std::string>
  split(const std::string &input, char separator);

  // We can't use a map for subtrees_ because we want the keys to be in order.
  std::vector<std::pair<std::string, std::vector<ptr_lib::shared_ptr<BoostInfoTree> > > > subtrees_;
  std::string value_;
  BoostInfoTree* parent_;
  BoostInfoTree* lastChild_;
};

inline std::ostream&
operator << (std::ostream& os, const BoostInfoTree& tree)
{
  os << tree.prettyPrint();
  return os;
}

class BoostInfoParser
{
public:
  BoostInfoParser()
  : root_(new BoostInfoTree())
  {
  }

  const BoostInfoTree&
  read(const std::string& fileName);

  const BoostInfoTree&
  read(const std::string& input, const std::string& inputName);

  // TODO: Implement readPropertyList.

  void
  write(const std::string& fileName) const;

  const BoostInfoTree&
  getRoot() const { return *root_; }

private:
  static void
  shlex_split(const std::string& s, std::vector<std::string>& result);

  BoostInfoTree*
  read(std::istream& stream, BoostInfoTree* ctx);

  BoostInfoTree*
  parseLine(const std::string& line, BoostInfoTree* context);

  ptr_lib::shared_ptr<BoostInfoTree> root_;
};

}

#endif