#include <arpa/inet.h>
#include <ctype.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/poll.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <time.h>
#include <unistd.h>

#include "hsosstp.h"

int main(int argc, char* argv[]) {
    int serverSocket, chunkSize, sessionId, m;
    unsigned short int command;
    struct sockaddr_in serverAddress;
    char buffer[HSOSSTP_CHUNK_MAX_SIZE + HSOSSTP_COMMANDLENGTH + 10 + 2 + 1], * param1, * param2,
         request[HSOSSTP_COMMANDLENGTH + 10 + 2 + 1];
    FILE *file;


    if (argc == 1) {
        puts("Give filename and chunksize");
        return 0;
    }
    else if (argc == 2) {
        puts("Give chunksize");
        return 0;
    }

    memset((void*)&serverAddress, '\0', sizeof(serverAddress));
    serverSocket = socket(AF_INET, SOCK_STREAM, 0);
    serverAddress.sin_family      = AF_INET;
    serverAddress.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
    serverAddress.sin_port        = htons(HSOSSTP_SERVER_PORT);

    file = fopen(argv[1], "a");
    chunkSize = atoi(argv[2]);

    if (chunkSize <= 0) {
        puts("Chunksize must be larger than 0");
        return 0;
    }

    if (argc > 3) {
        serverAddress.sin_addr.s_addr = inet_addr(argv[3]);
    }


    if (connect(serverSocket, (const struct sockaddr*)&serverAddress, (socklen_t)sizeof(serverAddress)) < 0) {
        puts("Could not connect to server");
        return 0;
    }

    m = sprintf(request, "HSOSSTP_INITX;%d;%s", chunkSize, argv[1]);
    if (send(serverSocket, request, m + 1, 0) < 0) {
        puts("Error writing to server");
        return 0;
    }
    if (recv(serverSocket, buffer, sizeof(buffer), 0) < 0) {
        puts("Could not read from server");
        return 0;
    }

    puts("Connected to server, starting session");
    command = parse(buffer, &param1, &param2);
    if (command != HSOSSTP_COMMAND_SIDXX) {
        puts("Error starting session");
        return 0;
    }
    printf("Started Session, ID: %s\n", param1);
    sessionId = atoi(param1);


    puts("Starting data transfer");
    m = sprintf(request, "HSOSSTP_GETXX;%d", sessionId);
    while ((send(serverSocket, request, m + 1, 0) > 0) && (recv(serverSocket, buffer, sizeof(buffer), 0) > 0)) {
        command = parse(buffer, &param1, &param2);
        if ((command == HSOSSTP_COMMAND_DATAX) && (param2 != NULL)) {
            fwrite(param2, strlen(param2), 1, file);
        }
        else if (command == HSOSSTP_COMMAND_ERROR) {
            puts("Error");
        }
        memset(buffer, '\0', sizeof(buffer));
        param1 = param2 = NULL;
    }


    fclose(file);
    return 0;
}