#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/ip.h>
#include <errno.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include "../include/ipaddress.h"
#include "../include/server.h"

// This code has been borrowed from this github repo https://github.com/wasmerio/c-http-server and modified by me.

void send_response(FILE *f, char *html)
{
    fprintf(f, "HTTP/1.1 200 OK\r\n");
    fprintf(f, "Content-Type: application/html\r\n");
    fprintf(f, "\r\n");
    fprintf(f, "%s", html);
}

int open_connection(int port)
{
    int sock;
    struct sockaddr_in addr_in;
    
    printf("simpleserver::Creating socket: (%d, %d, 0)\n", AF_INET, SOCK_STREAM);
    sock = socket(AF_INET, SOCK_STREAM, 0);
    printf("simpleserver::Socket created (%d)\n", sock);
    
    if (sock == -1)
    {
        printf("Failed to create socket (%d)\n", errno);
        exit(EXIT_FAILURE);
    }
    
    addr_in.sin_family = AF_INET;
    addr_in.sin_port = htons(port);
    addr_in.sin_addr.s_addr = INADDR_ANY;
    
    printf("simpleserver::bind (%d, (%d, %d, %d), %lu)\n", sock, addr_in.sin_family, addr_in.sin_port, addr_in.sin_addr.s_addr, sizeof(struct sockaddr_in));
    
    int lol = bind(sock, (struct sockaddr *)&addr_in, sizeof(struct sockaddr_in));
    
    printf("simpleserver::listen(%d, 8)\n", sock);
    
    if (listen(sock, 8) == -1)
    {
        printf("Failed to get socket to listen (%d)\n", errno);
        exit(EXIT_FAILURE);
    }
    
    return sock;
}

void accept_client(int sock, char *key)
{
    struct sockaddr_in client_addr;
    socklen_t clientaddr_len;
    FILE *f;
    
    printf("simpleserver::accept(%d)\n", sock);
    
    int client_sock = accept(sock, (struct sockaddr *)&client_addr, &clientaddr_len);
    printf("simpleserver::accepted (%d)\n", client_sock);
    
    if (client_sock == -1)
    {
        printf("Failed to accept connection (%d)\n", errno);
        exit(EXIT_FAILURE);
    }
    
    printf("simpleserver::fdopen(%d)\n", client_sock);
    
    f = fdopen(client_sock, "w+");
    
    if (f == NULL){
        printf("simpleserver::Error during fdopen\n");
    }
    
    printf("simpleserver::send_response(%s)\n", key);
    send_response(f, "<h1>Hello</h1>");
    
    printf("simpleserver::fclose()\n");
    fclose(f);
    
    printf("simpleserver::fflush\n");
    // stdout needs to be flushed in order for heroku to read the logs
    fflush(stdout);
    printf("simpleserver::endfflush\n");
}

void send_only(int port, char *key)
{
    int sock = open_connection(port);
    
    while (1)
    {
        accept_client(sock, key);
    }
    
    close(sock);
}