//=-------------------------------------------------------------------------=//
//  XML parsing module                                                       //
//                                                                           //
//  Copyright © 2021 The Gem-graph Project                                   //
//                                                                           //
//  This file is part of gem-graph.                                          //
//                                                                           //
//  This program is free software: you can redistribute it and/or modify     //
//  it under the terms of the GNU Affero General Public License as           //
//  published by the Free Software Foundation, either version 3 of the       //
//  License, or (at your option) any later version.                          //
//                                                                           //
//  This program is distributed in the hope that it will be useful,          //
//  but WITHOUT ANY WARRANTY; without even the implied warranty of           //
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the            //
//  GNU Affero General Public License for more details.                      //
//                                                                           //
//  You should have received a copy of the GNU Affero General Public License //
//  along with this program.  If not, see <https://www.gnu.org/licenses/>.   //
//=-------------------------------------------------------------------------=//

#include "../include/parsing.h"
#include "../include/model.h"

// -------------------------------------------------------------------------- //
// Parsing NOTHING (but yeah that prints)                                     //
// -------------------------------------------------------------------------- //
int parseStubFieldXML (xmlDocPtr doc,
                       ModelParserTableXML_t *ModelTable,
                       int currentParser,
                       xmlNodePtr currentNode)
{
    xmlChar *content, *contentText, *contentValueAttribute;

    contentText = xmlNodeListGetString(doc, currentNode->xmlChildrenNode, 1);
    contentValueAttribute = xmlGetProp(currentNode,
                                       (xmlChar*)"value");

    if (contentText) {
        content = contentText;
    } else if (contentValueAttribute) {
        content = contentValueAttribute;

        // Detect children
        if (currentNode->xmlChildrenNode) {
            printLog("%s (stub) has children\n",
                     ModelTable->table[currentParser].tag);
        }
    }

    printLog("%s (stub): %s\n", ModelTable->table[currentParser].tag,
                         content);

    if (!content) {
        xmlFree(contentText);
        xmlFree(contentValueAttribute);
        return -1;
    }

    xmlFree(contentText);
    xmlFree(contentValueAttribute);

    return 0;
}

// -------------------------------------------------------------------------- //
// Parsing a text field                                                       //
// -------------------------------------------------------------------------- //
int parseTextFieldXML (xmlDocPtr doc,
                       ModelParserTableXML_t *ModelTable,
                       int currentParser,
                       xmlNodePtr currentNode)
{
    xmlChar *content, *contentText, *contentValueAttribute;
    char *destination = (char*)ModelTable->table[currentParser].destination;

    contentText = xmlNodeListGetString(doc, currentNode->xmlChildrenNode, 1);
    contentValueAttribute = xmlGetProp(currentNode,
                                       (xmlChar*)"value");

    if (contentText) {
        content = contentText;
    } else if (contentValueAttribute) {
        content = contentValueAttribute;

        // Detect children
        if (currentNode->xmlChildrenNode) {
            printLog("%s has children\n", ModelTable->table[currentParser].tag);
        }
    }

    printLog("%s: %s\n", ModelTable->table[currentParser].tag,
                         content);

    if (!content) {
        xmlFree(contentText);
        xmlFree(contentValueAttribute);
        return -1;
    }

    strcat(destination, " ");
    strncpy(destination + strlen(destination),
           (char *)content, MODEL_STRING_SIZE - strlen(destination));

    xmlFree(contentText);
    xmlFree(contentValueAttribute);

    return 0;
}

// -------------------------------------------------------------------------- //
// Parsing an integer field                                                   //
// -------------------------------------------------------------------------- //
int parseIntFieldXML (xmlDocPtr doc,
                      ModelParserTableXML_t *ModelTable,
                      int currentParser,
                      xmlNodePtr currentNode)
{
    xmlChar *content, *contentText, *contentValueAttribute;
    int *destination = (int*)ModelTable->table[currentParser].destination;

    contentText = xmlNodeListGetString(doc, currentNode->xmlChildrenNode, 1);
    contentValueAttribute = xmlGetProp(currentNode,
                                       (xmlChar*)"value");

    if (contentText) {
        content = contentText;
    } else if (contentValueAttribute) {
        content = contentValueAttribute;

        // Detect children
        if (currentNode->xmlChildrenNode) {
            printLog("%s has children\n", ModelTable->table[currentParser].tag);
        }
    }

    printLog("%s: %s\n", ModelTable->table[currentParser].tag,
                         content);

    if (!content) {
        xmlFree(contentText);
        xmlFree(contentValueAttribute);
        return -1;
    }

    *destination = (int)atoi((char*)content);

    xmlFree(content);
    return 0;
}

// -------------------------------------------------------------------------- //
// Parsing a field that contains children fields                              //
// -------------------------------------------------------------------------- //
int parseParentFieldXML (xmlDocPtr doc,
                         ModelParserTableXML_t *ModelTable,
                         int currentParser,
                         xmlNodePtr currentNode)
{
    printLog("%s parsed\n",
             ModelTable->table[currentParser].tag);

    // Getting children content
    currentNode = currentNode->xmlChildrenNode;
    while (currentNode != NULL) {
        for (int i = 0; i < ModelTable->len; i++) {
            if ((!xmlStrcmp(currentNode->name,
                            ModelTable->table[i].tag))) {
                ModelTable->table[i].parse(doc,
                                           ModelTable,
                                           i,
                                           currentNode);
                break;
            }
        }
    currentNode = currentNode->next;
    }
    return 0;
}

// -------------------------------------------------------------------------- //
// Parsing a field that contains children fields with props                   //
// -------------------------------------------------------------------------- //
int parseParentFieldPropsXML (xmlDocPtr doc,
                              ModelParserTableXML_t *ModelTable,
                              int currentParser,
                              xmlNodePtr currentNode)
{
    xmlChar *id, *date, *author;
    // (truc*)destination = (truc*)ModelTable->table[currentParser].destination;

    // Getting field identity
    id = xmlGetProp(currentNode, (xmlChar*)"id");
    date = xmlGetProp(currentNode, (xmlChar*)"date");
    author = xmlGetProp(currentNode, (xmlChar*)"author");

    printLog("%s parsed (id=%s, date=%s, author=%s)\n",
             ModelTable->table[currentParser].tag, id, date, author);

    // Getting children content
    currentNode = currentNode->xmlChildrenNode;
    while (currentNode != NULL) {
        for (int i = 0; i < ModelTable->len; i++) {
            if ((!xmlStrcmp(currentNode->name,
                            ModelTable->table[i].tag))) {
                ModelTable->table[i].parse(doc,
                                           ModelTable,
                                           i,
                                           currentNode);
                break;
            }
        }
    currentNode = currentNode->next;
    }
    return 0;
}

// -------------------------------------------------------------------------- //
// Parsing a model file                                                       //
// -------------------------------------------------------------------------- //
int ParseModelXML(Model_t *model)
{
    ParserTableXML_t table[] =
    {
    // IDENTITY
        {(const xmlChar *)"identity", parseParentFieldXML, model},
        {(const xmlChar *)"name", parseTextFieldXML, model->name},
        {(const xmlChar *)"owner", parseTextFieldXML, model->owner},

        // TODO lacking implementation (model side)
        {(const xmlChar *)"owner_id", parseStubFieldXML, model->owner_id},

        {(const xmlChar *)"date", parseIntFieldXML, &model->date},
        {(const xmlChar *)"version", parseTextFieldXML, model->version},

    // PARAMETERS
        {(const xmlChar *)"parameters", parseParentFieldPropsXML, model},
        // MODELIZATION
        {(const xmlChar *)"modelization", parseParentFieldXML, model},
        {(const xmlChar *)"max_thread", parseStubFieldXML, model},
        {(const xmlChar *)"max_cycles", parseStubFieldXML, model},
        // SPACE
        {(const xmlChar *)"space_param", parseParentFieldXML, model},
        {(const xmlChar *)"dimension", parseStubFieldXML, model},
        {(const xmlChar *)"size", parseStubFieldXML, model},
        {(const xmlChar *)"site_multiplicity", parseStubFieldXML, model},

        // TODO lacking implementation (model side)
        {(const xmlChar *)"boundaries", parseStubFieldXML, model},

    // OBJECTS
        {(const xmlChar *)"objects", parseParentFieldXML, model},
        {(const xmlChar *)"object", parseParentFieldPropsXML, model},

    // SPACE
        {(const xmlChar *)"space", parseParentFieldPropsXML, model},

    // SAVESTATES
        {(const xmlChar *)"savestates", parseParentFieldPropsXML, model},

    // TRANSITIONS
        {(const xmlChar *)"transitions", parseParentFieldPropsXML, model},
        // TRANSITION
        // TODO probability
        {(const xmlChar *)"transition", parseParentFieldPropsXML, model},
        {(const xmlChar *)"if", parseParentFieldXML, model},
        {(const xmlChar *)"then", parseParentFieldXML, model},

    // ARROW
        {(const xmlChar *)"arrow", parseStubFieldXML, model},

    // REF
        {(const xmlChar *)"ref", parseStubFieldXML, model},

    // QUOTE
        {(const xmlChar *)"quote", parseStubFieldXML, model},
    };

    ModelParserTableXML_t modelParserTable =
    {
        LEN(table),
        &table[0]
    };

    printLog("Parsing model %s\n", model->name);
    xmlDocPtr xmlDocument;
    xmlNodePtr currentNode;

    xmlDocument = xmlParseFile(model->filename);

    if (xmlDocument == NULL) {
        printLog("Can't parse model file at '%s'.\n", model->filename);
        return -1;
    }

    currentNode = xmlDocGetRootElement(xmlDocument);

    if (currentNode == NULL) {
        printLog("Invalid model file at '%s', document empty !\n",
                 model->filename);
        xmlFreeDoc(xmlDocument);
        return -2;
    }

    if (xmlStrcmp(currentNode->name, (const xmlChar *) "gem-graph-model")) {
        printLog("Invalid model file at '%s', "
                 "root node is not <gem-graph-model> !\n",
                 model->filename);
        xmlFreeDoc(xmlDocument);
        return -3;
    }

    currentNode = currentNode->xmlChildrenNode;
    while (currentNode != NULL) {
        for (int i = 0; i < modelParserTable.len; i++) {
            if ((!xmlStrcmp(currentNode->name,
                            modelParserTable.table[i].tag))) {
                modelParserTable.table[i].parse(xmlDocument,
                                                &modelParserTable,
                                                i,
                                                currentNode);
            }
        }

    currentNode = currentNode->next;
    }

    xmlFreeDoc(xmlDocument);

    // validate when we're finished
    model->validated = true;
    return 0;
}