/*
 * Copyright (C) 2008-2010 Manjeet Dahiya
 *
 * Author:
 *	Manjeet Dahiya
 *
 * Source:
 *	http://code.google.com/p/qxmpp
 *
 * This file is a part of QXmpp library.
 *
 * This library 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 2.1 of the License, or (at your option) any later version.
 *
 * This library 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.
 *
 */


#ifndef QXMPPSTREAM_H
#define QXMPPSTREAM_H

#include <QObject>
#include <QSslSocket>
#include "QXmppConfiguration.h"
#include "QXmppRoster.h"
#include "QXmppStanza.h"
#include "QXmppVCardManager.h"
#include "QXmppArchiveManager.h"
#include "QXmppTransferManager.h"

class QDomElement;

class QXmppRoster;
class QXmppClient;
class QXmppPacket;
class QXmppPresence;
class QXmppIq;
class QXmppBind;
class QXmppRosterIq;
class QXmppVCard;
class QXmppMessage;
class QXmppRpcResponseIq;
class QXmppRpcErrorIq;
class QXmppArchiveChatIq;
class QXmppArchiveListIq;
class QXmppArchivePrefIq;
class QXmppByteStreamIq;
class QXmppDiscoveryIq;
class QXmppVersionIq;

class QXmppStream : public QObject
{
    Q_OBJECT

public:
    QXmppStream(QXmppClient* client);
    ~QXmppStream();
    void connect();
    void acceptSubscriptionRequest(const QString& from, bool accept = true);
    void sendSubscriptionRequest(const QString& to);
    void disconnect();
    QXmppArchiveManager& getArchiveManager();
    QXmppRoster& getRoster();
    QXmppTransferManager& getTransferManager();
    QXmppVCardManager& getVCardManager();
    void sendPacket(const QXmppPacket&);

    QAbstractSocket::SocketError getSocketError();
    QXmppClient::StreamError getXmppStreamError();

signals:
    // socket host found
    void hostFound();

    // socket connected
    void connected();

    // socket disconnected
    void disconnected();

    // xmpp connected
    void xmppConnected();

    void error(QXmppClient::Error);
    void subscriptionRequestReceived(const QString& from);
    void presenceReceived(const QXmppPresence&);
    void messageReceived(const QXmppMessage&);
    void iqReceived(const QXmppIq&);
    void rosterIqReceived(const QXmppRosterIq&);
    void rosterRequestIqReceived(const QXmppRosterIq&);
    void vCardIqReceived(const QXmppVCard&);

    void rpcCallResponse(const QXmppRpcResponseIq& result );
    void rpcCallError(const QXmppRpcErrorIq& err );

    void archiveChatIqReceived(const QXmppArchiveChatIq&);
    void archiveListIqReceived(const QXmppArchiveListIq&);
    void archivePrefIqReceived(const QXmppArchivePrefIq&);

    void discoveryIqReceived(const QXmppDiscoveryIq&);
    void versionIqReceived(const QXmppVersionIq&);

    void byteStreamIqReceived(const QXmppByteStreamIq&);
    void ibbCloseIqReceived(const QXmppIbbCloseIq&);
    void ibbDataIqReceived(const QXmppIbbDataIq&);
    void ibbOpenIqReceived(const QXmppIbbOpenIq&);
    void streamInitiationIqReceived(const QXmppStreamInitiationIq&);

private slots:
    void socketHostFound();
    void socketReadReady();
    void socketEncrypted();
    void socketConnected();
    void socketDisconnected();
    void socketError(QAbstractSocket::SocketError);
    void socketSslErrors(const QList<QSslError>&);

private:
    QXmppClient* m_client; // reverse pointer
    QXmppRoster m_roster;
    QString m_sessionId;
    QString m_bindId;
    QString m_rosterReqId;
    QByteArray m_dataBuffer;
    QSslSocket m_socket;
    bool m_sessionAvaliable;
    QAbstractSocket::SocketError m_socketError;
    QString m_streamId;
    QString m_nonSASLAuthId;
    QString m_XMPPVersion;
    QXmppClient::StreamError m_xmppStreamError;
//    m_xmppStanzaError;

    QXmppArchiveManager m_archiveManager;
    QXmppTransferManager m_transferManager;
    QXmppVCardManager m_vCardManager;
    int m_authStep;

    QXmppConfiguration& getConfiguration();
    void debug(const QString&);
    void info(const QString&);
    void warning(const QString&);
    void parser(const QByteArray&);
    void sendStartStream();
    void sendEndStream();
    void sendStartTls();
    void sendNonSASLAuth(bool);
    void sendNonSASLAuthQuery( const QString &to );
    void sendAuthPlain();
    void sendAuthDigestMD5();
    void sendAuthDigestMD5ResponseStep1(const QString& challenge);
    void sendAuthDigestMD5ResponseStep2();
    void sendBindIQ();
    void sendSessionIQ();
    void sendInitialPresence();
    void sendRosterRequest();
    void sendToServer(const QByteArray&);
    bool hasStartStreamElement(const QByteArray&);
    bool hasEndStreamElement(const QByteArray&);

    void processPresence(const QXmppPresence&);
    void processMessage(const QXmppMessage&);
    void processIq(const QXmppIq&);
    void processBindIq(const QXmppBind&);
    void processRosterIq(const QXmppRosterIq&);

    void flushDataBuffer();
};

#endif // QXMPPSTREAM_H