/**

   This file is part of CWRT package **

   Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies). **

   This program 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, version 2.1 of the License.

   This program 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 this program. If not, see <http://www.gnu.org/licenses/>.

*/

#include "serviceipclocalsocketsession.h"

#include "serviceipcobserver.h"

#include "serviceipcrequest.h"

#include <QtNetwork>

#include <clientinfo.h>

namespace WRT

{

const char KIPCSeparator = ';';

const char* REQUEST_COMPLETE_TOKEN = ";ROK";

/*!

 \class LocalSocketSession

 Class managing local socket server side sessions

 */

/*!

 Destructor

 */

LocalSocketSession::~LocalSocketSession()

{

    delete m_curRequest;

    m_curRequest = NULL;

    if (m_clientInfo) {

        delete m_clientInfo;

        m_clientInfo = NULL;

    }

}

/*!

 Constructor

 @param aNewSocket new socket connection to be handled

 @param aObserver observer to the IPC module

 */

LocalSocketSession::LocalSocketSession(QLocalSocket* aNewSocket,

                                       MServiceIPCObserver* aObserver) 

    : ServiceIPCSession(aObserver)

    , m_socket(aNewSocket)

{

    // Take ownership of the socket

    m_socket->setParent(this);

    QObject::connect(m_socket, SIGNAL( readyRead() ),

    this, SLOT( handleRequest() ) );

    QObject::connect(m_socket, SIGNAL( disconnected() ),

    this, SLOT( handleDisconnect() ) );

}

/*!

 Handle a new request

 */

void LocalSocketSession::handleRequest()

{

    // Process data

    QByteArray data = m_socket->readAll();

    // TODO: Get Client info

    ClientInfo *client = new ClientInfo();

    // New Request

    if (!m_curRequest) {

        // Should be very fast, the sperator is at the front

        int separator = data.indexOf(KIPCSeparator);

        int separator2 = data.indexOf(KIPCSeparator, separator + 1);

        // How long is the data, ensure separators are found, ie valid data

        //

        if (separator != -1 && separator2 != -1) {

            bool lengthConverted;

            QByteArray lengthData = data.left(separator);

            qint64 length = lengthData.toLong(&lengthConverted);

            if (lengthConverted) {

                // Compute the data 

                QByteArray operation = data.mid(separator + 1, separator2

                                - separator - 1);

                QByteArray requestData = data.right(data.length() - separator2

                                - 1);

                // New request

                delete m_curRequest;

                m_curRequest = NULL;

                m_curRequest = new ServiceIPCRequest(this, length, operation);

                // Call to observer with request if the length is correct

                // Otherwise wait until the data packet is re-assembled

                //

                if (m_curRequest->addRequestdata(requestData)) {

                    client->setSessionId(m_clientInfo->sessionId());

                    m_curRequest->setClientInfo(client); // ownership passed

                    //m_observer->handleRequest(m_curRequest);

                    handleReq();

                }

            }

        }

    }

    // More data available from the buffer

    else {

        // If all the data has been added, call back to the observer

        //

        if (m_curRequest->addRequestdata(data)) {

            m_curRequest->setClientInfo(client); // ownership passed

            //m_observer->handleRequest(m_curRequest);

            handleReq();

        }

    }

}

/*!

 Write data to the socket to send to the client

 @param aData data to write to the socket

 */

bool LocalSocketSession::write(const QByteArray& aData)

{

    int written = m_socket->write(aData);

    return (written != -1);

}

/*!

 Complete the outstanding IPC request

 */

bool LocalSocketSession::completeRequest()

{

    // Write a request complete token and wait until all data has been written to the socket

    m_socket->write(REQUEST_COMPLETE_TOKEN);

    bool done = m_socket->waitForBytesWritten(-1);

    delete m_curRequest;

    m_curRequest = NULL;

    return done;

}

/*!

 Handles when a client disconnect

 This slot function is connected to the underlying QLocalSocket

 */

void LocalSocketSession::handleDisconnect()

{

    if (m_appendToBList) {

        ((ServiceFwIPCServerLocalSocket*) parent())->removeBroadcastList(m_clientInfo->sessionId());  

    }

    doCancelRequest();

    // Emit signal back to server to cleanup

    emit disconnected(this);

    m_observer->handleClientDisconnect(m_clientInfo);

    ((ServiceFwIPCServerLocalSocket*) parent())->releaseSessionId(m_clientInfo->sessionId()); //release sessionId

}

/*!

 Closes and shutsdown the local socket session

 */

void LocalSocketSession::close()

{

    doCancelRequest();

    // Close the socket connection

    m_socket->abort();

}

/*! 

 Sends the cancel request callback to the client if a request was pending

 */

void LocalSocketSession::doCancelRequest()

{

    // If we had a request pending, make sure the client handles this

    // gracefully

    if (m_curRequest) {

        // TODO: Get client info

        ClientInfo *client = new ClientInfo();

        m_curRequest->setClientInfo(client); // ownership passed

        m_observer->handleCancelRequest(m_curRequest);

        delete m_curRequest;

        m_curRequest = NULL;

    }

}

}

/*!

 \fn WRT::LocalSocketSession::disconnected( ServiceIPCSession* aSession )

 Signal emitted when the session has been disconnected by the client

 @param aSession the session that is disconnected

 */

// END OF FILE