/**
Lightweight profiler library for c++
Copyright(C) 2016  Sergey Yagovtsev

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

#include <stdint.h>
#include "profiler/profiler.h"
#ifndef _WIN32
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#else

#define WIN32_LEAN_AND_MEAN

#include <winsock2.h>
#include <windows.h>
#include <ws2tcpip.h>
#include <stdlib.h>
#endif

class PROFILER_API EasySocket
{
public:
    enum ConnectionState
    {
        CONNECTION_STATE_UNKNOWN,
        CONNECTION_STATE_SUCCESS,

        CONNECTION_STATE_DISCONNECTED
    };
private:

    void checkResult(int result);

#ifdef _WIN32
    typedef SOCKET socket_t;
#else
    typedef int socket_t;
#endif
    
    socket_t m_socket = 0;
    socket_t m_replySocket = 0;
    uint16_t m_port = 0;

    int wsaret = -1;

#ifndef _WIN32
    struct sockaddr_in serv_addr;
    struct hostent *server = nullptr;
#else
    struct addrinfo  *result = NULL;
    struct addrinfo   hints;
#endif


    ConnectionState m_state = CONNECTION_STATE_UNKNOWN;
public:
    EasySocket();
    ~EasySocket();

    int send(const void *buf, size_t nbyte);
    int receive(void *buf, size_t nbyte);
    int listen(int count=5);
    int accept();
    int bind(uint16_t portno);

    bool setAddress(const char* serv, uint16_t port);
    int connect();

    void flush();
    void init();

    void setState(ConnectionState state){m_state=state;}
    ConnectionState state() const{return m_state;}
};

#endif // EASY________SOCKET_________H