IOStream.h

Go to the documentation of this file.

/*---------------------------------------------------------------------\
|                          ____ _   __ __ ___                          |
|                         |__  / \ / / . \ . \                         |
|                           / / \ V /|  _/  _/                         |
|                          / /__ | | | | | |                           |
|                         /_____||_| |_| |_|                           |
|                                                                      |
\---------------------------------------------------------------------*/
#ifndef ZYPP_BASE_IOSTREAM_H
#define ZYPP_BASE_IOSTREAM_H

#include <iosfwd>
#include <boost/io/ios_state.hpp>

#include "zypp/base/PtrTypes.h"
#include <zypp/base/SafeBool.h>
#include <zypp/base/NonCopyable.h>

namespace zypp
{ 


  namespace iostr
  { 

    typedef boost::io::ios_base_all_saver IosFmtFlagsSaver;


    std::string getline( std::istream & str );

    inline std::ostream & copy( std::istream & from_r, std::ostream & to_r )
    {
      if ( from_r && to_r )
      {
        char ch;
        while ( from_r && from_r.get( ch ) )
          to_r.put( ch );
      }
      return to_r;
    }

    inline std::ostream & copyIndent( std::istream & from_r, std::ostream & to_r, const std::string & indent_r = "> " )
    {
      if ( from_r && to_r )
      {
        char ch;
        bool indent = true;
        while ( from_r && from_r.get( ch ) )
        {
          if ( indent )
            to_r << indent_r;
          indent = ( ch == '\n' );
          to_r.put( ch );
        }
      }
      return to_r;
    }

    inline void tee( std::istream & from_r, std::ostream & to1_r, std::ostream & to2_r )
    {
      if ( from_r && ( to1_r ||to2_r ) )
      {
        char ch;
        while ( from_r && from_r.get( ch ) )
        {
          to1_r.put( ch );
          to2_r.put( ch );
        }
      }
    }

    //
    //  CLASS NAME : EachLine
    //
    class EachLine : private base::SafeBool<EachLine>, private base::NonCopyable
    {
      typedef base::SafeBool<EachLine> SafeBool;

      public:
        EachLine( std::istream & str_r, unsigned lineNo_r = 0 );

        using SafeBool::operator bool_type;

        bool valid() const
        { return boolTest(); }

        unsigned lineNo() const
        { return _lineNo; }

        std::streamoff lineStart() const
        { return _lineStart; };

        void setLineNo( unsigned lineNo_r )
        { _lineNo = lineNo_r; }

        const std::string & operator*() const
        { return _line; }
        std::string & operator*()
        { return _line; }

        const std::string * operator->() const
        { return &_line; }

        bool next();

        bool next( unsigned num_r )
        {
          while ( num_r-- && next() )
            ; /* EMPTY */
          return valid();
        }

      private:
        friend SafeBool::operator bool_type() const;
        bool boolTest() const
        { return _valid; }

      private:
        std::istream & _str;
        std::string    _line;
        std::streamoff _lineStart;
        unsigned       _lineNo;
        bool           _valid;
    };

    template<class _Function>
      _Function & forEachLine( std::istream & str_r, _Function & consume_r )
      {
        while ( str_r )
          {
            std::string l = getline( str_r );
            if ( ! (str_r.fail() || str_r.bad()) )
              {
                // l contains valid data to be consumed.
                if ( ! consume_r( l ) )
                  break;
              }
          }
        return consume_r;
      }

  } // namespace iostr
} // namespace zypp
#endif // ZYPP_BASE_IOSTREAM_H