tdenetwork/kopete/protocols/irc/libkirc/kirctransfer.h

/*
    kirctransfer.h - DCC Handler

    Copyright (c) 2003      by Michel Hermier <michel.hermier@wanadoo.fr>

    Kopete    (c) 2003      by the Kopete developers <kopete-devel@kde.org>

    *************************************************************************
    *                                                                       *
    * This program 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 2 of the License, or     *
    * (at your option) any later version.                                   *
    *                                                                       *
    *************************************************************************
*/

#ifndef KIRCTRANSFER_H
#define KIRCTRANSFER_H

#include <tqdatastream.h>
#include <tqfile.h>
#include <tqhostaddress.h>
#include <tqobject.h>
#include <tqtextstream.h>

class KExtendedSocket;

class TQFile;
class TQTextCodec;

namespace KIRC
{
class Engine;

class Transfer
	: public TQObject
{
	TQ_OBJECT
  

public:
	enum Type {
		Unknown,
		Chat,
		FileOutgoing,
		FileIncoming
	};

	enum Status {
		Error_NoSocket	= -2,
		Error		= -1,
		Idle		= 0,
		HostLookup,
		Connecting,
		Connected,
		Closed
	};
public:
	Transfer(	KIRC::Engine *engine, TQString nick,// TQString nick_peer_adress
			Type type = Unknown,
			TQObject *parent = 0L, const char *name = 0L );

	Transfer(	KIRC::Engine *engine, TQString nick,// TQString nick_peer_adress,
			TQHostAddress peer_address, TQ_UINT16 peer_port,
			Transfer::Type type,
			TQObject *parent = 0L, const char *name = 0L );

	Transfer(	KIRC::Engine *engine, TQString nick,// TQString nick_peer_adress,
			Transfer::Type type,
			TQString fileName, TQ_UINT32 fileSize,
			TQObject *parent = 0L, const char *name = 0L );

	Transfer(	KIRC::Engine *engine, TQString nick,// TQString nick_peer_adress,
			TQHostAddress peer_address, TQ_UINT16 peer_port,
			Transfer::Type type,
			TQString fileName, TQ_UINT32 fileSize,
			TQObject *parent = 0L, const char *name = 0L );
/*
	For a file transfer properties are:

		KExntendedSocket	*socket
	or
		TQHostAddress		peerAddress
		TQ_UINT16		peerPort
	for determining the socket.

		TQString			fileName
		TQ_UINT32		fileSize
	for detemining the file propeties.
*//*
	Transfer(	KIRC *engine, TQString nick,// TQString nick_peer_adress,
			Transfer::Type type, TQVariant properties,
			TQObject *parent = 0L, const char *name = 0L );
*/
	~Transfer();

	KIRC::Engine *engine() const
		{ return m_engine; }
	TQString nick() const
		{ return m_nick; }
	Type type() const
		{ return m_type; }
	Status status() const;

	/* Start the transfer.
	 * If not connected connect to client.
	 * Allow receiving/emitting data.
	 */
	bool initiate();

	TQString fileName() const
		{ return m_fileName; }
	/* Change the file name.
	 */
	void setFileName(TQString fileName)
		{ m_fileName = fileName; }
	unsigned long fileSize() const
		{ return m_fileSize; }
public slots:
	bool setSocket( KExtendedSocket *socket );
	void closeSocket();

	void setCodec( TQTextCodec *codec );
	void writeLine( const TQString &msg );

	void flush();

	void userAbort(TQString);

signals:
	void readLine( const TQString &msg );

	void fileSizeCurrent( unsigned int );
	void fileSizeAcknowledge( unsigned int );

//	void received(TQ_UINT32);
//	void sent(TQ_UINT32);

	void abort(TQString);

	/* Emited when the transfer is complete.
	 * Usually it means that the file transfer has successfully finished.
	 */
	void complete();

protected slots:
	void slotError(int);

	void readyReadLine();

	void readyReadFileIncoming();

	void writeFileOutgoing();
	void readyReadFileOutgoing();

protected:
//	void emitSignals();
	void checkFileTransferEnd( TQ_UINT32 fileSizeAck );

	KIRC::Engine *	m_engine;
	TQString		m_nick;

	Type		m_type;
	KExtendedSocket *m_socket;
	bool		m_initiated;

	// Text member data
	TQTextStream	m_socket_textStream;
//	TQTextCodec *	m_socket_codec;

	// File member data
	TQFile		m_file;
	TQString		m_fileName;
	TQ_UINT32	m_fileSize;
	TQ_UINT32 /*usize_t*/	m_fileSizeCur;
	TQ_UINT32 /*usize_t*/	m_fileSizeAck;
	TQDataStream	m_socketDataStream;
	char		m_buffer[1024];
	int		m_bufferLength;

	// Data transfer measures
	TQ_UINT32	m_receivedBytes;
	TQ_UINT32	m_receivedBytesLimit;

	TQ_UINT32	m_sentBytes;
	TQ_UINT32	m_sentBytesLimit;
};

}

#endif