/home/mhx/git/github/libmoost/include/moost/iterator/glob_iterator.hpp

Go to the documentation of this file.

/* vim:set ts=3 sw=3 sts=3 et: */
#ifndef MOOST_ITERATOR_GLOB_ITERATOR_HPP__
#define MOOST_ITERATOR_GLOB_ITERATOR_HPP__

#include <string>
#include <sstream>

#include <boost/filesystem.hpp>
#include <boost/iterator/filter_iterator.hpp>
#include <boost/regex.hpp>

namespace moost { namespace iterator {

class glob_predicate
{
public:
   glob_predicate()
      : glob_(false)
   {
   }

   glob_predicate(boost::regex const & re)
      : filter_(re)
      , glob_(false)
   {
   }

   glob_predicate(std::string const & glob)
      : filter_(glob2re(glob))
      , glob_(true)
   {
   }

   bool operator()(boost::filesystem::path p) const
   {
      return glob_ ?
         // when we're globbing we want to force an exact match
         boost::regex_match(p.string(), filter_) :
         // when we're pattern matching use search for flexibility
         boost::regex_search(p.string(), filter_);
   }

private:
   boost::regex glob2re(std::string const & glob)
   {
      std::ostringstream oss;
      bool lesc = false;

      // iterate glob and convert to regex
      for(std::string::const_iterator itr = glob.begin() ;
          itr != glob.end(); ++itr)
      {
         bool oesc = false;

         switch(*itr)
         {
            // slash is an esc-char unless it's the 2nd one else its a literal
            case '\\':
               if(lesc) { oss << "\\\\"; }
               oesc = !lesc;
               break;
            // match 0 or more chars
            case '*':
               oss << (lesc ? "\\*" : ".*");
               break;
            // match 1 char
            case '?':
               oss << (lesc ? "\\?" : ".");
               break;
            // literal dot
            case '.':
               oss << "\\.";
               break;
            // everything else
            default:
               oss << *itr;
               break;
         };

         lesc = oesc;
      }

      return boost::regex(oss.str());
   }

   boost::regex filter_;
   bool glob_;
};

typedef boost::filter_iterator<
   glob_predicate,
   boost::filesystem::directory_iterator
   > glob_iterator;

}}


#endif // MOOST_ITERATOR_GLOB_ITERATOR_HPP__