fcgistream.hpp

Go to the documentation of this file.

/***************************************************************************
* Copyright (C) 2007 Eddie Carle [eddie@erctech.org]                       *
*                                                                          *
* This file is part of fastcgi++.                                          *
*                                                                          *
* fastcgi++ is free software: you can redistribute it and/or modify it     *
* under the terms of the GNU Lesser General Public License as  published   *
* by the Free Software Foundation, either version 3 of the License, or (at *
* your option) any later version.                                          *
*                                                                          *
* fastcgi++ 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 Lesser General Public     *
* License for more details.                                                *
*                                                                          *
* You should have received a copy of the GNU Lesser General Public License *
* along with fastcgi++.  If not, see <http://www.gnu.org/licenses/>.       *
****************************************************************************/


#ifndef FCGISTREAM_HPP
#define FCGISTREAM_HPP

#include <iosfwd>
#include <boost/iostreams/filtering_stream.hpp>
#include <boost/iostreams/categories.hpp>
#include <boost/iostreams/concepts.hpp>

#include <fastcgi++/protocol.hpp>
#include <fastcgi++/transceiver.hpp>

namespace Fastcgipp
{

   enum OutputEncoding {NONE, HTML, URL};

   class FcgistreamSink: public boost::iostreams::device<boost::iostreams::output, char>
   {
   private:
      Protocol::FullId m_id;
      Protocol::RecordType m_type;
      Transceiver* m_transceiver;
   public:
      std::streamsize write(const char* s, std::streamsize n);

      void set(Protocol::FullId id, Transceiver &transceiver, Protocol::RecordType type) {m_id=id, m_type=type, m_transceiver=&transceiver;}
      void dump(const char* data, size_t size) { write(data, size); }
      void dump(std::basic_istream<char>& stream);
   };


   template <typename charT> class Fcgistream: public boost::iostreams::filtering_stream<boost::iostreams::output, charT>
   {
   private:
      struct Encoder: public boost::iostreams::multichar_filter<boost::iostreams::output, charT>
      {
         template<typename Sink> std::streamsize write(Sink& dest, const charT* s, std::streamsize n);
         OutputEncoding m_state;
         Encoder(): m_state(NONE) {}
      };
      
      Encoder& m_encoder;

      FcgistreamSink& m_sink;

   public:
      Fcgistream();
      void set(Protocol::FullId id, Transceiver& transceiver, Protocol::RecordType type) { m_sink.set(id, transceiver, type); }

      void flush() { boost::iostreams::filtering_stream<boost::iostreams::output, charT>::strict_sync(); }
      

      void dump(const char* data, size_t size) { flush(); m_sink.dump(data, size); }

      void dump(std::basic_istream<char>& stream) { flush(); m_sink.dump(stream); }


      void setEncoding(OutputEncoding x) { m_encoder.m_state=x; }
   };


   struct encoding
   {
      const OutputEncoding m_type;
      encoding(OutputEncoding type): m_type(type) {}
   };

   template<class charT, class Traits> std::basic_ostream<charT, Traits>& operator<<(std::basic_ostream<charT, Traits>& os, const encoding& enc);
}

#endif