/*
 * Connection.h
 *
 * Copyright (C) 2008 Matthias Schiffer <matthias@gamezock.de>
 *
 * 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 3 of the License, or
 * (at your option) any later version.
 *
 * 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef MAD_NET_CONNECTION_H_
#define MAD_NET_CONNECTION_H_

#include <queue>
#include <gnutls/gnutls.h>
#include <sigc++/signal.h>
#include "Packet.h"

namespace Mad {
namespace Net {

class IPAddress;
class Packet;

class Connection {
  private:
    struct Transmission {
      unsigned long length;
      unsigned long transmitted;
      
      unsigned char *data;
      
      sigc::slot<void,const void*,unsigned long> notify;
    };
    
    Transmission transR;
    std::queue<Transmission> transS;
    
    Packet::Data header;
    
    sigc::signal<void,const Connection*,const Packet&> signal;
    
    void packetHeaderReceiveHandler(const void *data, unsigned long length);
    void packetDataReceiveHandler(const void *data, unsigned long length);
    
    void doReceive();
    void doSend();
    
    bool receiveComplete() const {
      return (transR.length == transR.transmitted);
    }
    
  protected:
    virtual gnutls_session_t& getSession() = 0;
    
    bool rawReceive(unsigned long length,
        const sigc::slot<void,const void*,unsigned long> &notify);
    bool rawSend(const unsigned char *data, unsigned long length);
    
    bool enterReceiveLoop() {
      return rawReceive(sizeof(Packet::Data), sigc::mem_fun(this, &Connection::packetHeaderReceiveHandler));
    }
    
  public:
    Connection() {
      transR.length = transR.transmitted = 0;
    }
    
    virtual ~Connection() {}
    
    virtual bool isConnected() const = 0;
    virtual const IPAddress* getPeer() const = 0;
    
    virtual bool dataPending() const = 0;
    
    bool send(const Packet &packet) {
      return rawSend(reinterpret_cast<const unsigned char*>(packet.getRawData()), packet.getRawDataLength());
    }
    
    void sendReceive() {
      doReceive();
      doSend();
    }
    
    bool sendQueueEmpty() {return transS.empty();}
    
    sigc::signal<void,const Connection*,const Packet&> signalReceive() const {return signal;}
    
    static void init() {
      gnutls_global_init();
    }
    
    static void deinit() {
      gnutls_global_deinit();
    }
};

}
}

#endif /*MAD_NET_CONNECTION_H_*/