/*
----
This file is part of SECONDO.

Copyright (C) 2015,
Faculty of Mathematics and Computer Science,
Database Systems for New Applications.

SECONDO 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 2 of the License, or
(at your option) any later version.

SECONDO 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 SECONDO; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
----

*/

#ifndef KAFKA_SIGNALINGSOCKETS_H
#define KAFKA_SIGNALINGSOCKETS_H

static const int RESULT_OK = 0;

static const int ERROR_SRV_SOCKET_FAILED = 1;
static const int ERROR_SRV_SETSOCKOPT_FAILED = 2;
static const int ERROR_SRV_BIND_FAILED = 3;
static const int ERROR_SRV_LISTEN_FAILED = 4;

static const int ERROR_CLI_SOCKET_CREATION_ERROR = 1;
static const int ERROR_CLI_UNABLE_TO_RESOLVE_HOST = 2;
static const int ERROR_CLI_RESOLVED_HOST_TYPE_NOT_SUPPORTED = 3;
static const int ERROR_CLI_CONNECTION_FAILED = 4;

#define SOCKET_DEBUG 0

#include <iostream>

#include <unistd.h>
#include <cstdio>
#include <sys/socket.h>
#include <cstdlib>
#include <netinet/in.h>
#include <cstring>
#include <thread>
#include <arpa/inet.h>
#include <netdb.h>

class SignallingSocket {
private:
    int sockfd;
    struct sockaddr_in address;
    std::thread *sockThread;
    bool socketShutdown = false;
    bool signalReceived = false;

public:

    SignallingSocket() {
    }

    ~SignallingSocket() {
    }


    int open(int port) {
        // Creating socket file descriptor
        if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == 0) {
            perror("socket failed");
            return ERROR_SRV_SOCKET_FAILED;
        }

        int opt = 1;
        if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT,
                       &opt, sizeof(opt))) {
            perror("setsockopt");
            return ERROR_SRV_SETSOCKOPT_FAILED;
        }

        // Forcefully attaching socket to the port
        address.sin_family = AF_INET;
        address.sin_addr.s_addr = INADDR_ANY;
        address.sin_port = htons(port);
        if (bind(sockfd, (struct sockaddr *) &address,
                 sizeof(address)) < 0) {
            perror("bind failed");
            return ERROR_SRV_BIND_FAILED;
        }
        if (listen(sockfd, 3) < 0) {
            perror("listen");
            return ERROR_SRV_LISTEN_FAILED;
        }
        sockThread = new std::thread([this] { this->waitForConnection(); });
        return RESULT_OK;
    }

    int close() {
        if (SOCKET_DEBUG) std::cout << "Closing signaling socket" << std::endl;
        socketShutdown = true;
        shutdown(sockfd, SHUT_RD);
        ::close(sockfd);
        sockThread->join();
        delete sockThread;
        if (SOCKET_DEBUG) std::cout << "signaling socket closed" << std::endl;
        return 0;
    }

    bool isSignalReceived() {
        return signalReceived;
    }

private:
    void waitForConnection() {
        if (SOCKET_DEBUG)
            std::cout << "Starting waiting for connection"
                      << std::endl;
        int new_socket;
        int addrlen = sizeof(address);
        char const *expected = "Finish";
        char const *responseAccepted = "Finish accepted";
        char const *responseIgnored = "False request";

        while (!signalReceived) {
            if ((new_socket = accept(sockfd, (struct sockaddr *) &address,
                                     (socklen_t *) &addrlen)) < 0) {
                if (socketShutdown)
                    return;
                perror("accept");
                exit(EXIT_FAILURE);
            }
            char buffer[1024] = {0};
            int valread = read(new_socket, buffer, 1024);
            printf("Received signal: %s\n", buffer);
            signalReceived = strncmp(expected, buffer, valread) == 0;
            if (signalReceived)
                send(new_socket, responseAccepted, strlen(responseAccepted), 0);
            else
                send(new_socket, responseIgnored, strlen(responseIgnored), 0);
        }
        if (SOCKET_DEBUG)
            std::cout << "Exit waiting for connection" << std::endl;
    }

};


class SignallingClient {
private:

    int getServerAddressByHostname(const char *hostname, int port_number,
                                   sockaddr_in &serv_addr) {
        printf("resolving %s\n", hostname);
        struct hostent *host = gethostbyname(hostname);
        if (!host) {
            printf("unable to resolve host %s\n", hostname);
            return ERROR_CLI_UNABLE_TO_RESOLVE_HOST;
        }

        switch (host->h_addrtype) {
            case AF_INET:
            case AF_INET6:
                break;

            default:
                printf("resolved host type is not supported\n");
                return ERROR_CLI_RESOLVED_HOST_TYPE_NOT_SUPPORTED;
        }

        for (char **addr = host->h_addr_list; *addr != 0; ++addr) {
            bzero(&serv_addr, sizeof(serv_addr));

            switch (host->h_addrtype) {
                case AF_INET: {
                    struct sockaddr_in *saddr =
                            (struct sockaddr_in *) &serv_addr;
                    saddr->sin_family = AF_INET;
                    bcopy(*addr, &(saddr->sin_addr), host->h_length);
                    saddr->sin_port = htons(port_number);
                    break;
                }

                case AF_INET6: {
                    struct sockaddr_in6 *saddr =
                            (struct sockaddr_in6 *) &serv_addr;
                    saddr->sin6_family = AF_INET6;
                    bcopy(*addr, &(saddr->sin6_addr), host->h_length);
                    saddr->sin6_port = htons(port_number);
                    break;
                }
            }
        } // End For
        return 0;
    }

public:
    int sendSignal(std::string hostname, int port_number) {
        int sock = 0;
        struct sockaddr_in serv_addr;
        char const *message = "Finish";
        if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
            printf("\n Socket creation error \n");
            return ERROR_CLI_SOCKET_CREATION_ERROR;
        }

        serv_addr.sin_family = AF_INET;
        serv_addr.sin_port = htons(port_number);

        // Convert IPv4 and IPv6 addresses from text to binary form
        if (inet_pton(AF_INET, hostname.c_str(), &serv_addr.sin_addr) <= 0) {
////            printf("\nInvalid address/ Address not supported \n");
////            return 2;
            bzero(&serv_addr, sizeof(serv_addr));
            if (int res = getServerAddressByHostname(hostname.c_str(),
                                                     port_number, serv_addr))
                return res;
        }

        if (connect(sock, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) <
            0) {
            printf("\nConnection Failed \n");
            return ERROR_CLI_CONNECTION_FAILED;
        }
        send(sock, message, strlen(message), 0);
        printf("Signaling message sent\n");

        char buffer[1024] = {0};
        read(sock, buffer, 1024);
        printf("Response from server: %s\n", buffer);
        return RESULT_OK;
    }
};


#endif //KAFKA_SIGNALINGSOCKETS_H