sbuild-dirstream.h

Go to the documentation of this file.

/* Copyright © 2003,2006-2007  Roger Leigh <rleigh@debian.org>
 *
 * schroot is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * schroot is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see
 * <http://www.gnu.org/licenses/>.
 *
 *********************************************************************/

#ifndef SBUILD_DIRSTREAM_H
#define SBUILD_DIRSTREAM_H

#include <sbuild/sbuild-custom-error.h>

#include <iostream>
#include <deque>
#include <string>

#include <sys/types.h>
#include <dirent.h>

namespace sbuild
{

  class direntry
  {
  public:
    direntry()
    { std::memset(&this->data, 0, sizeof(struct dirent)); }

    direntry(const struct dirent *entry)
    { std::memcpy(&this->data, entry, sizeof(struct dirent)); }

    direntry(direntry const& orig)
    { memcpy(&this->data, &orig.data, sizeof(struct dirent)); }

    virtual ~direntry()
    {}

    long inode() const
    { return this->data.d_ino; }

    unsigned char type() const
    { return this->data.d_type; }

    std::string name() const
    { return this->data.d_name; }

    struct dirent const& dirent()
    { return this->data; }

  private:
    struct dirent data;
  }; // class direntry

  class dirstream
    {
    public:
      enum error_code
        {
          DIR_OPEN, 
          DIR_READ  
        };

    typedef custom_error<error_code> error;

      dirstream(std::string const& dir);

      virtual ~dirstream();

      void open(std::string const& dirname);

      void close();

      bool eof() const;

      bool bad() const;

      operator bool ();

      bool
      operator ! ();

      friend dirstream&
      operator >> (dirstream& stream,
                   direntry&  entry);

    private:
      void read (int quantity=1);

      std::string dirname;

      DIR *dir;

      std::deque<direntry> data;

      bool error_status;

      bool eof_status;
  };

  dirstream&
  operator >> (dirstream& stream,
               direntry&  entry);

}

#endif /* SBUILD_DIRSTREAM_H */

/*
 * Local Variables:
 * mode:C++
 * End:
 */

---