/*
 UIPClient.h - Arduino implementation of a uIP wrapper class.
 Copyright (c) 2013 Norbert Truchsess <norbert.truchsess@t-online.de>
 All rights reserved.

 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 UIPCLIENT_H
#define UIPCLIENT_H

#include <Client.h>
#include "utility/mempool.h"

extern "C" {
  #include "utility/uip.h"
}

#define UIP_SOCKET_DATALEN UIP_TCP_MSS

#define UIP_CLIENT_CONNECTED 0x01
#define UIP_CLIENT_CLOSE 0x02
#define UIP_CLIENT_REMOTECLOSED 0x04
#define UIP_CLIENT_RESTART 0x08
#define UIP_CLIENT_ACCEPTED 0x10

typedef struct {
  uint8_t conn_index;
  uint8_t state;
  memhandle packets_in[UIP_SOCKET_NUMPACKETS];
  uint16_t lport;        /**< The local TCP port, in network byte order. */
} uip_userdata_closed_t;

typedef struct {
  uint8_t conn_index;
  uint8_t state;
  memhandle packets_in[UIP_SOCKET_NUMPACKETS] = {NOBLOCK};
  memhandle packets_out[UIP_SOCKET_NUMPACKETS] = {NOBLOCK};
  memaddress out_pos;
} uip_userdata_t;

class EthernetClient : public Client {

public:
  EthernetClient();
  int connect(IPAddress ip, uint16_t port);
  int connect(const char *host, uint16_t port);
  int read(uint8_t *buf, size_t size);
  void stop();
  uint8_t connected();
  operator bool();
  virtual bool operator==(const EthernetClient&);
  virtual bool operator!=(const EthernetClient& rhs) { return !this->operator==(rhs); };

  size_t write(uint8_t);
  size_t write(const uint8_t *buf, size_t size);
  int availableForWrite();
  void flush(); // flush sends the buffered data

  int available();
  int read();
  int peek();
  void discardReceived();

  using Print::write;

  IPAddress remoteIP();
  uint16_t remotePort();

  uint8_t status();

  void setConnectionTimeout(uint16_t millis) {connectTimeout = millis;}

private:
  EthernetClient(struct uip_conn *_conn);
  EthernetClient(uip_userdata_t* conn_data);

  uip_userdata_t* data;
  uint16_t connectTimeout = 1000 * UIP_CONNECT_TIMEOUT;

  static uip_userdata_t all_data[UIP_CONNS];
  static uip_userdata_t* _allocateData();

  static uint8_t _currentBlock(memhandle* blocks);
  static void _eatBlock(memhandle* blocks);
  static void _flushBlocks(memhandle* blocks);

#ifdef UIPETHERNET_DEBUG_CLIENT
  static void _dumpAllData();
#endif

  friend class UIPEthernetClass;
  friend class EthernetServer;

  friend void uipclient_appcall(void);

};

#endif