#include "../test.h"
#include "arc/std/errno.h"
#include "arc/std/parser.h"
#include "arc/std/lexer.h"
#include "arc/std/vector.h"
#include "arc/std/parser/parserlang.h"
#include <stddef.h>

#define LAMBDA        ARC_PARSER_TAG_LAMBDA
#define CHAR          ARC_LEXER_TOKEN_ALPHALOWERCHAR
#define NUM           ARC_LEXER_TOKEN_NUMBER
#define CHAR_OR_NUM   23
#define VARIABLE_NAME 24
#define VARIABLE      25

const uint32_t TEST_ALPHA_LAMBDA     = 0;
const uint32_t TEST_ALPHA_LOWER_CHAR = 1;
const uint32_t TEST_ALPHA_UPPER_CHAR = 2;
const uint32_t TEST_ALPHA_CHAR       = 3;

void TEST_ParserData_RecurseStringAdd(ARC_String **data, ARC_ParserTagToken *tagToken){
    if(tagToken->token != NULL){
        ARC_String_Append(data, tagToken->token->data);
        return;
    }

    //TODO: probs don't need this
    if(tagToken->tagTokens == NULL){
        return;
    }

    for(uint32_t index = 0; index < ARC_Vector_GetSize(tagToken->tagTokens); index++){
        TEST_ParserData_RecurseStringAdd(data, (ARC_ParserTagToken *)ARC_Vector_Get(tagToken->tagTokens, index));
    }
}

void TEST_ParserData_CreateStringFn(void **data, ARC_ParserTagToken *parsedData, void *userData){
    ARC_String_Create((ARC_String **)data, NULL, 0);

    TEST_ParserData_RecurseStringAdd((ARC_String **)data, parsedData);
}

void TEST_ParserData_DestroyStringFn(void *data, void *userData){
    ARC_String_Destroy((ARC_String *)data);
}

uint32_t *charOrNumTokens[]      = { (uint32_t[]){ 1, CHAR }, (uint32_t[]){ 1, NUM } };
uint32_t *variableNameTags[]     = { (uint32_t[]){ 2, CHAR_OR_NUM, VARIABLE_NAME }, (uint32_t[]){ 1, LAMBDA } };
uint32_t *variableTokensOrTags[] = { (uint32_t[]){ 2, CHAR, VARIABLE_NAME } };

//TODO: note how language function callbacks work, and how they use the parentData if createDataFn is NULL
ARC_ParserData_CreateFn  createStringFn  = TEST_ParserData_CreateStringFn;
ARC_ParserData_DestroyFn destroyStringFn = TEST_ParserData_DestroyStringFn;


ARC_ParserTag testTags[3] = {
    {
        VARIABLE,             //tagId
        variableTokensOrTags, //tokensOrTags
        1                     //tokenOrTagsSize
    },
    {
        VARIABLE_NAME,    //tagId
        variableNameTags, //tokensOrTags
        2                 //tokenOrTagsSize
    },
    {
        CHAR_OR_NUM,     //tagId
        charOrNumTokens, //tokensOrTags
        2                //tokenOrTagsSize
    }
};

ARC_Array languageArray = {
    3,       //size
    testTags //data
};

void TEST_Parser_InitLexerRulesFn(ARC_Lexer *lexer){
    ARC_Lexer_InitBasicTokenRules(lexer);
}

//ARC_TEST(Parser_Init){
//    ARC_Parser *parser;
//    ARC_Parser_Create(&parser, &languageArray, TEST_Parser_InitLexerRulesFn, NULL, NULL, NULL);
//
//    ARC_Parser_Destroy(parser);
//
//    ARC_CHECK(arc_errno == 0);
//}
//
//ARC_TEST(Parser_Basic_Parse){
//    ARC_Parser *parser;
//    ARC_Parser_Create(&parser, &languageArray, TEST_Parser_InitLexerRulesFn, NULL, NULL, NULL);
//
//    ARC_String *tempString;
//
//
//    /* ~ first test ~ */
//    ARC_String_CreateWithStrlen(&tempString, "myvar1");
//
//    //this destroys string, so no need for cleanup
//    ARC_Parser_Parse(parser, &tempString);
//
//    ARC_CHECK(arc_errno == 0);
//
//
//    /* ~ second test ~ */
//    ARC_String_CreateWithStrlen(&tempString, "z1xwvq");
//    //this destroys string, so no need for cleanup
//    ARC_Parser_Parse(parser, &tempString);
//
//
//    ARC_CHECK(arc_errno == 0);
//
//
//    /* ~ third test ~ */
//    ARC_String_CreateWithStrlen(&tempString, "z1234");
//
//    //this destroys string, so no need for cleanup
//    ARC_Parser_Parse(parser, &tempString);
//
//    ARC_CHECK(arc_errno == 0);
//
//
//    /* ~ fourth test ~ */
//    ARC_String_CreateWithStrlen(&tempString, "aaaaa");
//
//    //this destroys string, so no need for cleanup
//    ARC_Parser_Parse(parser, &tempString);
//
//    ARC_CHECK(arc_errno == 0);
//
//
//    /* ~ cleanup ~ */
//    ARC_Parser_Destroy(parser);
//}
//
//ARC_TEST(Parser_Basic_ParseError){
//    ARC_Parser *parser;
//    ARC_Parser_Create(&parser, &languageArray, TEST_Parser_InitLexerRulesFn, NULL, NULL, NULL);
//
//    ARC_String *tempString;
//
//
//    /* ~ first test ~ */
//    ARC_String_CreateWithStrlen(&tempString, "!myVar1");
//
//    //this destroys string, so no need for cleanup
//    ARC_Parser_Parse(parser, &tempString);
//
//    ARC_CHECK(arc_errno == ARC_ERRNO_DATA);
//
//
//    /* ~ second test ~ */
//    //check again with moved character
//    arc_errno = 0;
//    ARC_String_CreateWithStrlen(&tempString, "my!Var1");
//
//    //this destroys string, so no need for cleanup
//    ARC_Parser_Parse(parser, &tempString);
//
//    ARC_CHECK(arc_errno == ARC_ERRNO_DATA);
//
//
//    /* ~ third test ~ */
//    //check again with moved character
//    arc_errno = 0;
//    ARC_String_CreateWithStrlen(&tempString, "myVar1!");
//
//    //this destroys string, so no need for cleanup
//    ARC_Parser_Parse(parser, &tempString);
//
//    ARC_CHECK(arc_errno == ARC_ERRNO_DATA);
//
//
//    /* ~ cleanup ~ */
//    ARC_Parser_Destroy(parser);
//
//    //reset for next test
//    arc_errno = 0;
//}
//
//ARC_TEST(Parser_Basic_GetParsedValue){
//    ARC_Parser *parser;
//    ARC_Parser_Create(&parser, &languageArray, TEST_Parser_InitLexerRulesFn, &createStringFn, &destroyStringFn, NULL);
//
//    ARC_String *tempString;
//
//
//    /* ~ first test ~ */
//    ARC_String_CreateWithStrlen(&tempString, "myvar1");
//
//    //this destroys string, so no need for cleanup
//    ARC_Parser_Parse(parser, &tempString);
//
//    ARC_CHECK(arc_errno == 0);
//
//    ARC_String *checkValue = (ARC_String *)ARC_Parser_GetData(parser);
//    ARC_CHECK(ARC_String_EqualsCStringWithStrlen(checkValue, "myvar1"));
//
//
//    /* ~ cleanup ~ */
//    ARC_Parser_Destroy(parser);
//}

/* ~ parserlang tests ~ */
uint32_t TEST_ParserLang_GetIdFn(ARC_String *string){
    if(ARC_String_EqualsCStringWithStrlen(string, "<alphaChar>")){
        return TEST_ALPHA_CHAR;
    }

    if(ARC_String_EqualsCStringWithStrlen(string, "ALPHA_LOWER_CHAR")){
        return TEST_ALPHA_LOWER_CHAR;
    }

    if(ARC_String_EqualsCStringWithStrlen(string, "ALPHA_UPPER_CHAR")){
        return TEST_ALPHA_UPPER_CHAR;
    }

    printf("tag: %s\n", string->data);
    return 0;
}

ARC_TEST(Parser_ParserLang_BasicTest){
    ARC_Parser *parser;
    ARC_Parser_CreateAsParserLang(&parser, TEST_ParserLang_GetIdFn);

    ARC_String *tempString;
    ARC_String_CreateWithStrlen(&tempString, "<alphaChar> -> ALPHA_LOWER_CHAR | ALPHA_UPPER_CHAR\n");

    //this destroys string, so no need for cleanup
    ARC_Parser_Parse(parser, &tempString);
    ARC_CHECK(arc_errno == 0);

    void *data = ARC_Parser_GetData(parser);
    ARC_ParserTag *tag = ARC_Vector_Get((ARC_Vector *)data, 0);

    ARC_CHECK(tag->tagId == TEST_ALPHA_CHAR);

    ARC_CHECK(tag->tokensOrTagsSize   ==                     2);
    ARC_CHECK(tag->tokensOrTags[0][0] ==                     1);
    ARC_CHECK(tag->tokensOrTags[0][1] == TEST_ALPHA_LOWER_CHAR);
    ARC_CHECK(tag->tokensOrTags[1][0] ==                     1);
    ARC_CHECK(tag->tokensOrTags[1][1] == TEST_ALPHA_UPPER_CHAR);

    ARC_Parser_Destroy(parser);

}

//ARC_TEST(Parser_ParserLang_BasicVector){
//    ARC_Vector *testLanguage;
//    ARC_Vector_Create(&testLanguage, NULL, NULL);
//
//    ARC_Vector_Add(testLanguage, testTags + 0);
//    ARC_Vector_Add(testLanguage, testTags + 1);
//    ARC_Vector_Add(testLanguage, testTags + 2);
//
//    ARC_Parser *parser;
//    ARC_Parser_CreateFromVector(&parser, testLanguage, TEST_Parser_InitLexerRulesFn, NULL, NULL);
//
//    ARC_String *tempString;
//    ARC_String_CreateWithStrlen(&tempString, "variablename");
//
//    //this destroys string, so no need for cleanup
//    ARC_Parser_Parse(parser, &tempString);
//
//    //cleanup
//    ARC_Parser_Destroy(parser);
//    ARC_Vector_Destroy(testLanguage);
//
//    ARC_CHECK(arc_errno == 0);
//}