#ifndef NETWORK_H
#define NETWORK_H

#include <arpa/inet.h>
#include <SDL.h>

#define NETWORK_UPDATE_SIZE  (256 * 1024)
enum
{
	HEADER             = 0,
	DISPLAY_UPDATE_RAW = 1,
	DISPLAY_UPDATE_RLE = 2,
	SOUND_UPDATE_RAW   = 3,
	SOUND_UPDATE_RLE   = 4,
	KEYBOARD_UPDATE    = 5,
	JOYSTICK_UPDATE    = 6,
};

struct NetworkDisplayUpdate
{
	Uint8 type;    /* raw data or RLE */
	Uint8 square;
	Uint16 size;
	Uint8 data[];
};

struct NetworkSoundUpdate
{
	Uint8 type;    /* raw data or RLE */
	Uint8 dummy;
	Uint16 size;
	Uint8 data[];
};

struct NetworkJoystickUpdate
{
	Uint8 type;
	Uint8 which;
	Uint8 data;   /* New value */
};

struct NetworkUpdate 
{
	Uint8 type;
	Uint8 dummy;
	Uint16 size;

	/* The rest is just data of some type */
	Uint8 data[];
};

class Network
{
public:
	/** Encode part of a screen into @a dst
	 * 
	 * @param dst the destination update structure
	 * @param screen the screen to encode
	 * @param square the square index of the screen to encode
	 *
	 * @return the size of the encoded message
	 */
	size_t EncodeDisplaySquare(struct NetworkDisplayUpdate *dst,
			Uint8 *screen, int square);

	/**
	 * Encode the @a buf sound buffer into @a dst
	 *
	 * @param dst the destination update structure
	 * @param buf the buffer to encode
	 * @param len the length of the in-buffer
	 *
	 * @return the size of the encoded message
	 */
	size_t EncodeSoundBuffer(struct NetworkSoundUpdate *dst,
			Uint8 *buf, size_t len);

	void EncodeJoystickUpdate(struct NetworkJoystickUpdate *dst,
			Uint8 which, Uint8 v);

	/**
	 * Decode a display update message onto @a screen
	 *
	 * @param screen the screen to draw to
	 * @param src the message to decode
	 */
	bool DecodeDisplayUpdate(Uint8 *screen, struct NetworkDisplayUpdate *src);
	
	size_t DecodeSoundUpdate(struct NetworkSoundUpdate *src, char *buf);

	NetworkUpdate *GetNetworkUpdate(void);

	bool SendUpdate(NetworkUpdate *src, int sock);

	bool ReceiveUpdate(NetworkUpdate *dst, int sock);

	bool ReceiveUpdateBlock(NetworkUpdate *dst, int sock);

	NetworkUpdate *IterateFirst(NetworkUpdate *p, unsigned int *cookie);

	NetworkUpdate *IterateNext(NetworkUpdate *p, unsigned int *cookie);

private:
	size_t EncodeDisplayRLE(struct NetworkDisplayUpdate *dst, Uint8 *screen,
			int x, int y);
	size_t EncodeDisplayRaw(struct NetworkDisplayUpdate *dst, Uint8 *screen,
			int x, int y);
	size_t EncodeSoundRLE(struct NetworkSoundUpdate *dst,
			Uint8 *buffer, size_t len);
	size_t EncodeSoundRaw(struct NetworkSoundUpdate *dst,
			Uint8 *buffer, size_t len);

	bool DecodeDisplayRLE(Uint8 *screen, struct NetworkDisplayUpdate *src,
			int x, int y);
	bool DecodeDisplayRaw(Uint8 *screen, struct NetworkDisplayUpdate *src,
			int x, int y);

	bool ReceiveUpdate(NetworkUpdate *dst, int sock, struct timeval *tv);

	void MarshalData(NetworkUpdate *ud);

	void DeMarshalData(NetworkUpdate *ud);
};

class NetworkClient : public Network
{
public:
	NetworkClient(int sock);

	bool ConnectToServer(const char *hostname, int port);

	Uint8 *screen;
	int joystick_port;
private:
	int sock;
};

class NetworkServer : public Network
{
public:
	NetworkServer();

	NetworkClient *CheckNewConnection();

	NetworkClient **clients;
	int n_clients;

private:
	int listen_sock;
	fd_set listen_fds;
};

#endif /* NETWORK_H */