connectioncompression.h

/*
 * Copyright (c) 2009 by Jakob Schroeter <js@camaya.net>
 * This file is part of the glooxd library. http://camaya.net/glooxd
 *
 * This software is distributed under a license. The full license
 * agreement can be found in the file LICENSE in this distribution.
 * This software may not be copied, modified, sold or distributed
 * other than expressed in the named license agreement.
 *
 * This software is distributed without any warranty.
 */

#ifndef CONNECTIONCOMPRESSION_H__
#define CONNECTIONCOMPRESSION_H__

#include "glooxd.h"

#include <gloox/gloox.h>
#include <gloox/compressiondatahandler.h>
#include <gloox/connectionbase.h>
#include <gloox/connectiondatahandler.h>

#include <string>

namespace gloox {
  class CompressionDefault;
  class LogSink;
}

namespace glooxd
{

  class GLOOXD_API ConnectionCompression : public gloox::CompressionDataHandler,
                                           public gloox::ConnectionBase,
                                           gloox::ConnectionDataHandler
  {
    public:
      ConnectionCompression( gloox::ConnectionDataHandler* cdh, gloox::ConnectionBase* conn,
                             const gloox::LogSink& log );

      ConnectionCompression( gloox::ConnectionBase* conn, const gloox::LogSink& log );

      virtual ~ConnectionCompression();

      void setConnectionImpl( gloox::ConnectionBase* connection );

      // reimplemented from ConnectionBase
      virtual gloox::ConnectionError connect();

      // reimplemented from ConnectionBase
      virtual gloox::ConnectionError recv( int timeout = -1 );

      // reimplemented from ConnectionBase
      virtual bool send( const std::string& data );

      // reimplemented from ConnectionBase
      virtual gloox::ConnectionError receive();

      // reimplemented from ConnectionBase
      virtual void disconnect();

      // reimplemented from ConnectionBase
      virtual void cleanup();

      // reimplemented from ConnectionBase
      virtual void getStatistics( long int& totalIn, long int& totalOut );

      // reimplemented from ConnectionDataHandler
      virtual void handleReceivedData( const gloox::ConnectionBase* connection, const std::string& data );

      // reimplemented from ConnectionDataHandler
      virtual void handleConnect( const gloox::ConnectionBase* connection );

      // reimplemented from ConnectionDataHandler
      virtual void handleDisconnect( const gloox::ConnectionBase* connection, gloox::ConnectionError reason );

      // reimplemented from ConnectionDataHandler
      virtual gloox::ConnectionBase* newInstance() const;

      // reimplemented from CompressionDataHandler
      virtual void handleCompressedData( const std::string& data );

      // reimplemented from CompressionDataHandler
      virtual void handleDecompressedData( const std::string& data );


    private:
      gloox::ConnectionBase* m_connection;
      gloox::CompressionDefault* m_compression;
      const gloox::LogSink& m_log;

  };

}

#endif // CONNECTIONCOMPRESSION_H__