//
// Készítette: Toldi Balázs Ádám
// Dátum:  2019. 10. 14..
//
#include "webio.h"

#ifdef RANDOM_PORT
#define DEFAULT_INTERFACE_PORT "0"
#else
#define DEFAULT_INTERFACE_PORT "5081"
#endif


int webio_create(int port,char* folder,struct Node_data myData,bool wildcard,WebIO *webIo){

    struct addrinfo *result = NULL;
    SOCKET listening;
    result = tcp_createIPv4Socket(&listening,port,wildcard);
    if(result == NULL){
        return 1;
    }
    tcp_bindnlisten(listening,result,SOMAXCONN);
    WebIO wio;

    wio.sockaddr = tcp_getAddr_in(listening);
    wio.socket= listening;
    wio.nodeData = myData;
    strcpy(wio.folder,folder);
    *webIo = wio;
    return 0;
}

int webio_handleRequest(WebIO wio,const PeerList *list){
    SOCKET client = accept(wio.socket,NULL,NULL);
    char buf[8192];
    memset(buf,0,8192);
    int res = recv(client,buf,8192,0);
    if(res <=0){
        logger_log("Error with web interface!");
        close(client);
        return -1;
    }
    char req[10];
    sscanf(buf,"%s",req);

    if(strcmp(req,"GET") == 0) {
        char file[50];
        sscanf(buf, "%*s %s", file);
        res = webio_handleGETrequest(client,wio,file, list);
		
    }else if(strcmp(req,"POST") == 0) {
        int i = strlen(buf)-1;
        while(buf[i] != '\n'){
            i--;
        }
        i++;
        char tmp[8196];
        strcpy(tmp,buf+i);
        Map post = getPacketData(tmp);

        res = webio_handlePOSTrequest(client, wio,list,post);
        free(post.pairs);
    }else
        res = -1;

    return res;

}

char* webio_getMIMEtype(char* filename){
    char* ext = webio_getFiletype(filename);
    if (strcmp(ext,"html") == 0)
        return "text/html";
    else if (strcmp(ext, "json") == 0)
        return "application/json";
    else if (strcmp(ext, "svg") == 0)
        return "image/svg+xml";
    else if (strcmp(ext, "js")== 0)
        return "application/javascript";
    else if (strcmp(ext, "css")== 0)
        return "text/css";
    else if (strcmp(ext, "jpg")== 0)
        return "image/jpeg";
    else if (strcmp(ext, "png")== 0)
        return "image/png";
    else if (strcmp(ext, "bmp")== 0)
        return "image/bmp";
    else if (strcmp(ext, "ico")== 0)
        return "image/x-icon";
    else if (strcmp(ext, "xml")== 0)
        return "text/xml";
    else if (strcmp(ext, "zip")== 0)
        return "application/zip";
    else return "text/plain";
}

char* webio_getFiletype(char* filename){
    char *type = (char*) malloc(10);
    char* tmp = strtok(filename,".");
    while(tmp != NULL){
        type =tmp;
        tmp = strtok(NULL,".");
    }
    return type;
}

int webio_handleGETrequest(SOCKET client,WebIO wio,char* file,const PeerList *list){

    char buf[8192];
    sscanf(buf,"%*s %s",file);

    char path[129];

    memset(path,0,sizeof(path));
    strcat(path,wio.folder);
    char *response = (char *) malloc(sizeof(char) * 8192);
    int len = 0;
    if (file[0] == '/')
        memmove(file, file+1, strlen(file));
    if(strcmp(file,"") == 0) {
		char index[8192] = "";
        webio_getIndex(wio.folder, list, index);
		
		//strcat(index,"<h1>IT WORKS LIKE A DUCK!</h1>");
        sprintf(response, "HTTP/1.1 200 OK "
                          "Content-Encoding: gzip\r\n"
                          "Content-Language: en\r\n"
                          "Content-Type: text/html\r\n\r\n%s",index);
        len = strlen(response);

    }else if(peer_ID_isFound(*list,file) || webio_isPeerFound(wio.folder,file)){
		char cnt[8192] = "";
        webio_getPeerPage(wio.folder, file, (peer_ID_getPeer(*list, file) != -1), cnt);
        sprintf(response, "HTTP/1.1 200 OK "
                          "Content-Encoding: gzip\r\n"
                          "Content-Language: en\r\n"
                          "Content-Type: text/html\r\n\r\n%s"
                ,cnt);
        len = strlen(response);
    }else if(strcmp(file,"kill") == 0){

        logger_log("Exit sign received.Exiting...");
        return -2;
    } else{
        strcat(path, file);
        FILE *fp;
        fp = fopen(path, "rb");

        if (fp != NULL) {
			char* content = (char*)calloc(sizeof(char) * 16384, sizeof(char));

			char buf[4096];
			int counter = 0;

			while (fgets(buf, 4096, fp) != NULL) {
				int len = strlen(content);
				strcat(content, buf);
				counter += 4096;
				if (counter >= len) {
					content = realloc(content, counter + 16384);
				}
			}

			fclose(fp);
			int len = counter + 1024;
			response = (char*)realloc(response, sizeof(char) * len);

			sprintf(response, "HTTP/1.1 200 OK "
				"Content-Encoding: gzip\r\n"
				"Content-Language: en\r\n"
				"Content-Type: %s\r\n\r\n%s"
				, webio_getMIMEtype(file), content);

			free(content);
        } else {

            sprintf(response, "HTTP/1.1 404 Not Found "
                              "Content-Encoding: gzip\r\n"
                              "Content-Language: en\r\n"
                              "Content-Type: text/html\r\n\r\n"
                              "<h1>Error 404 File not found!</h1>");
        }
    }
    int res = send(client,response,strlen(response),0);
    if(res == SOCKET_ERROR){
        logger_log("Sending failed!");
        return -1;
    }
    shutdown(client,SHUT_RDWR);
    close(client);

    free(response);
}

int webio_handlePOSTrequest(SOCKET client, WebIO wio, const PeerList *list, Map post){
    shutdown(client,SHUT_RD);
    char *response = "HTTP/1.1 304 Not Modified ";

    int res = send(client,response,strlen(response),0);
    if(res == SOCKET_ERROR ) {
        logger_log("Error with io");
        return -1;
    }
    shutdown(client,SHUT_RDWR);

    if(map_isFound(post,"id") && map_isFound(post,"message") && strcmp(map_getValue(post,"message"),"%0D%0A") != 0){
        char file[64];
        sprintf(file,"%speers/%s.txt",wio.folder,map_getValue(post,"id"));
        FILE * f;
        f = fopen(file,"a");
        fprintf(f,"Me: %s\n",map_getValue(post,"message"));
        fclose(f);
        char buf[DEFAULT_BUFLEN];
        int i = peer_ID_getPeer(*list,map_getValue(post,"id"));
        if(i <0){
            logger_log("Error sending message! Error: Peer offline!");
            return 1;
        }

        unsigned char  encrypted[DEFAULT_BUFLEN];
        sprintf(buf,"@message=%s",map_getValue(post,"message"));
        res = send(list->array[i].socket,buf,DEFAULT_BUFLEN,0);
        if(res == SOCKET_ERROR){
            logger_log("Error sending message.Error: %d",errno);
            return 2;
        }
        logger_log("Message sent to %s",map_getValue(post,"id"));
    }else map_dump(post);
}
void webio_getHeader(char* folder,char result[]) {

    char path[65];
    strcpy(path, folder);
    strcat(path, "/header.html");

    FILE* fp;
    fp = fopen(path, "r");

    if (fp != NULL) {
        char buf[1024];
        while (fgets(buf, 1024, fp) != NULL) {
            strcat(result, buf);
        }
        fclose(fp);
    }else
        result = "<html>";
}

void webio_getIndex(char* folder, const PeerList *list, char *outputBuffer){
    char content[8192] ="";
    char header[4096] ="";

    webio_getHeader(folder, header);
    strcpy(content,header);
	
    strcat(content,"<h1>Peers:</h1>\n");
    if(list->length > 0) {
        strcat(content, "<ul>\n");
        for (int i = 0; i < list->length; ++i) {
            sprintf(content, "%s<li>"
                             "<a href=\"%s\">",
            content, list->array[i].peerData.id);
            if(strcmp(list->array[i].peerData.nick,"") !=0){
                sprintf(content, "%s%s %d- ",content,list->array[i].peerData.nick,(int) strlen(list->array[i].peerData.nick));
            }
            sprintf(content,"%s%s</a></li>\n",content, list->array[i].peerData.id);
        }
        strcat(content, "</ul>\n");
    }else
        sprintf(content,"%s<div class=\"alert alert-warning\" role=\"alert\">\n"
                       "  No peers connected!\n"
                       "</div>\n",content);
    sprintf(content,"%s<script>setTimeout(function(){\n"
                   "   window.location.reload(1);\n"
                   "}, 5000);</script>\n"
                   "</div>\n"
                   "</body>\n"
    "</html>\n",content);
    strcpy(outputBuffer,content);
}

void webio_getPeerPage(char *folder, char *id, bool online, char *outputBuffer) {
	char content[8192] = "";
	char header[4096] = "";

	webio_getHeader(folder, header);
	strcpy(content, header);

    char *img = (online) ?"<img width=\"30\" height=\"30\" src=\"assets\\img\\on.svg\">" : "<img width=\"30\" height=\"30\" src=\"assets\\img\\off.svg\">";
    char* disabled = (online) ? "":"disabled";

    sprintf(content,"%s\n"
                    "<h1>%s%s</h1>\n"
                    "<div id=\"msgs\" style=\"margin-bottom:5em\"></div>\n"
                    "<div id=\"end\"></div>\n"
                    "    <form name=\"sendmsg\" class=\"form-inline\" style=\"margin: 7px;padding: 7px;position: fixed;bottom: 0;width: 100%%;\">"
                    "<textarea name=\"message\" id=\"message\" class=\"form-control\" style=\"width: 90%%;display: block;\" %s></textarea>"
                    "<input name=\"id\"class=\"form-control\" type=\"text\" style=\"display:none;\" value=\"%s\">"
                    "<input onclick=\"submitForm();\"name=\"s\" id=\"s\" type=\"button\" class=\"btn btn-primary\" style=\"margin: 7px;padding: 7px;display:inline;\" value=\"Send\" /></form>\n"
                    "<script src=\"assets/js/chat.js\"></script>"
                    "</body>\n"
                    "\n"
                    "</html>",header,img,id,disabled,id);
	strcpy(outputBuffer, content);
}

bool webio_isPeerFound(char* folder,char *id) {

    char file[129];
    sprintf(file,"%s/peers/%s.txt",folder,id);
    FILE* F = fopen(file,"r");
    if(F == NULL){
        return false;
    }else
    return true;
}