#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <why2.h>

int main(void)
{
    //VARIABLES
    char *textBuffer = malloc(128);
    char *keyBuffer;
    int exitCode = 0;
    unsigned long timeBuffer;

    //FLAGS
    inputFlags flags =
    {
        0, //SKIP CHECK
        0, //NO OUTPUT
        1 //NO UPDATE
    };

    //SET KEY_LENGTH TO 100
    setKeyLength(100);
    keyBuffer = malloc(getKeyLength() + 1);

    //SET ENCRYPTION_SEPARATOR TO '|'
    setEncryptionSeparator('|');

    //ENCRYPT
    outputFlags encrypted = encryptText(TEST_TEXT, NULL, flags);

    strcpy(textBuffer, encrypted.outputText); //GET ENCRYPTED TEXT
    strcpy(keyBuffer, encrypted.usedKey); //GET KEY
    timeBuffer = encrypted.elapsedTime; //GET TIME 1

    //DEALLOCATE BUFFER
    deallocateOutput(encrypted);

    //DECRYPT
    encrypted = decryptText(textBuffer, keyBuffer, flags);

    timeBuffer += encrypted.elapsedTime; //GET TIME 1

    //COMPARE DIFFERENCE
    if (strcmp(encrypted.outputText, TEST_TEXT) == 0)
    {
        printf
        (
            "Test successful!\n\n"

            "TEXT: \t\t\"%s\"\n"
            "OUTPUT: \t\"%s\"\n"
            "KEY: \t\t\"%s\"\n"
            "TIME: \t\t\"%lums\"\n"
            "UNUSED KEY: \t\"%lu\"\n"
            "EXIT CODE: \t\"%d\"\n"

            , TEST_TEXT, textBuffer, encrypted.usedKey, timeBuffer / 1000, encrypted.unusedKeySize, encrypted.exitCode
        );
    }
    else
    {
        fprintf(stderr, "Test failed!\n\n%s // %s\n", encrypted.outputText, TEST_TEXT);
        exitCode = 1;
    }

    //DEALLOCATION
    free(textBuffer);
    free(keyBuffer);
    deallocateOutput(encrypted);

    return exitCode;
}