/*
    Copyright (c) 2007-2009 FastMQ Inc.

    This file is part of 0MQ.

    0MQ is free software; you can redistribute it and/or modify it under
    the terms of the Lesser GNU General Public License as published by
    the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.

    0MQ 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
    Lesser GNU General Public License for more details.

    You should have received a copy of the Lesser GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <string>

#include "../include/zs.h"

#include "session_stub.hpp"
#include "i_engine.hpp"
#include "listener.hpp"
#include "err.hpp"

zs::session_stub_t::session_stub_t (listener_t *listener_) :
    state (reading_identity),
    engine (NULL),
    listener (listener_),
    index (-1)
{
}

void zs::session_stub_t::terminate ()
{
    if (engine)
        engine->terminate ();
    delete this;
}

void zs::session_stub_t::shutdown ()
{
    if (engine)
        engine->shutdown ();
    delete this;
}

zs::session_stub_t::~session_stub_t ()
{
}

void zs::session_stub_t::set_engine (i_engine *engine_)
{
    zs_assert (!engine_ || !engine);
    engine = engine_;
}

bool zs::session_stub_t::read (struct zs_msg *msg_)
{
    //  No messages are sent to the connecting peer.
    return false;
}

bool zs::session_stub_t::write (struct zs_msg *msg_)
{
    //  The first message arrived is the connection identity.
    if (state == reading_identity) {
        identity = std::string ((const char*) zs_msg_data (msg_),
            zs_msg_size (msg_));
        state = has_identity;
        return true;
    }

    //  We are not interested in any subsequent messages.
    return false;
}

void zs::session_stub_t::flush ()
{
    //  We have the identity. At this point we can find the correct session and
    //  attach it to the connection.
    if (state == has_identity) {

        //  At this point the stub will be deleted. Return immediately without
        //  touching 'this' pointer.
        listener->got_identity (this, identity.c_str ());
        return;
    }
}

zs::i_engine *zs::session_stub_t::detach_engine ()
{
    //  Ask engine to unregister from the poller.
    i_engine *e = engine;
    engine->detach ();
    return e;
}

void zs::session_stub_t::set_index (int index_)
{
    index = index_;
}

int zs::session_stub_t::get_index ()
{
    return index;
}