#include <iostream>

#include <cstring>

#include <string>

#include <sstream>

#include <vector>

#include "tinyxml2.h"

static void testfunction(const tinyxml2::XMLElement *node, const std::string &functionName);

static std::vector<std::string> splitString(const char *str, char delim)

{

    std::vector<std::string> ret;

    while (const char *p = std::strchr(str,delim)) {

        ret.push_back(std::string(str, p-str));

        str = p + 1;

    }

    ret.push_back(str);

    return ret;

}

static std::string replaceCharInString(std::string s, char from, char to)

{

    for (int i = 0; i < s.size(); ++i) {

        if (s[i] == from)

            s[i] = to;

    }

    return s;

}

int main(int argc, char **argv)

{

    if (argc != 2 || !std::strstr(argv[1], ".cfg")) {

        std::cerr << "no cfg file\n";

        return 1;

    }

    const char *cfgfile = argv[1];

    tinyxml2::XMLDocument doc;

    if (tinyxml2::XML_SUCCESS != doc.LoadFile(cfgfile)) {

        std::cerr << "failed to load cfg file\n";

        return 1;

    }

    std::string testfile(cfgfile);

    if (testfile.find('/') != std::string::npos)

        testfile = testfile.substr(testfile.rfind('/')+1);

    testfile = "generated-cfg-tests-" + testfile.substr(0,testfile.find('.')) + ".cpp";

    std::cout << "// auto generated tests from " << cfgfile << std::endl;

    std::cout << "//" << std::endl;

    std::cout << "// Generated by command:" << std::endl;

    std::cout << "// " << argv[0] << ' ' << cfgfile << " > " << testfile << std::endl;

    std::cout << "//" << std::endl;

    std::cout << "// Recommended cppcheck command line:" << std::endl;

    std::cout << "// $ cppcheck --enable=warning,information --inline-suppr --platform=unix64 " << testfile << std::endl;

    std::cout << "// => 'unmatched suppression' warnings are false negatives." << std::endl;

    std::cout << "//" << std::endl << std::endl ;

    const tinyxml2::XMLElement * const rootnode = doc.FirstChildElement();

    for (const tinyxml2::XMLElement *node = rootnode->FirstChildElement(); node; node = node->NextSiblingElement()) {

        const std::string nodename = node->Name();

        if (nodename == "memory" || nodename == "resource") {

            //testalloc(node);

        } else if (nodename == "function") {

            for (const std::string &name : splitString(node->Attribute("name"), ','))

                testfunction(node, name);

        }

    }

    return 0;

}

static std::string functionCall(const std::string &functionName, bool useretval, int argNr, int numberOfArgs, const char badArg[])

{

    std::ostringstream ostr;

    if (useretval)

        ostr << "result = ";

    ostr << functionName << '(';

    for (int i = 1; i <= numberOfArgs; ++i) {

        if (i > 1)

            ostr << ", ";

        if (i == argNr)

            ostr << badArg;

        else

            ostr << "arg" << i;

    }

    ostr << ')';

    return ostr.str();

}

static std::string testFunctionArg(const std::string &functionName, bool useretval, const char type[], int argNr, int numberOfArgs, const char code[], const char suppress[], const char badArg[])

{

    std::ostringstream ostr;

    ostr << "void test__" << replaceCharInString(functionName,':','_') << "__arg" << argNr << "__" << type << "() {" << std::endl;

    if (*code)

        ostr << "  " << code << std::endl;

    ostr << "  // cppcheck-suppress " << suppress << std::endl;

    ostr << "  " << functionCall(functionName, useretval, argNr, numberOfArgs, badArg) << ';' << std::endl;

    ostr << '}';

    return ostr.str();

}

static void testfunction(const tinyxml2::XMLElement *node, const std::string &functionName)

{

    // How many args does this function take?

    int numberOfArgs = 0;

    for (const tinyxml2::XMLElement *child = node->FirstChildElement(); child; child = child->NextSiblingElement()) {

        if (std::strcmp(child->Name(), "arg") == 0) {

            const char *nr = child->Attribute("nr");

            if (nr && std::isdigit(*nr) && std::atoi(nr) > numberOfArgs)

                numberOfArgs = std::atoi(nr);

        }

    }

    bool noreturn = false;

    bool useretval = false;

    bool pure = false;

    //bool constant = false;

    bool leakignore = false;

    for (const tinyxml2::XMLElement *child = node->FirstChildElement(); child; child = child->NextSiblingElement()) {

        if (std::strcmp(child->Name(), "noreturn") == 0)

            noreturn = true;

        else if (std::strcmp(child->Name(), "use-retval") == 0)

            useretval = true;

        else if (std::strcmp(child->Name(), "pure") == 0)

            pure = true;

        //else if (std::strcmp(child->Name(), "const") == 0)

        //    constant = true;

        else if (std::strcmp(child->Name(), "leak-ignore") == 0)

            leakignore = true;

    }

    if (noreturn) {

        std::cout << "void test__" << replaceCharInString(functionName,':','_') << "__noreturn() {" << std::endl;

        std::cout << "  int x = 1;" << std::endl;

        std::cout << "  if (cond) { x=100; " << functionCall(functionName, useretval, -1, numberOfArgs, "") << "; }" << std::endl;

        std::cout << "  // cppcheck-suppress shiftTooManyBits" << std::endl;

        std::cout << "  x = 1 << x;" << std::endl;

        std::cout << "}" << std::endl << std::endl;

    }

    if (useretval) {

        std::cout << "void test__" << replaceCharInString(functionName,':','_') << "__useretval() {" << std::endl;

        std::cout << "  // cppcheck-suppress ignoredReturnValue" << std::endl;

        std::cout << "  " << functionCall(functionName, false, -1, numberOfArgs, "") << ';' << std::endl;

        std::cout << "}" << std::endl << std::endl;

    }

    if (pure) {

        std::cout << "void test__" << replaceCharInString(functionName,':','_') << "__pure(";

        for (int i = 1; i <= numberOfArgs; ++i)

            std::cout << (i>1?",":"") << "int arg" << i;

        std::cout << ") {" << std::endl;

        std::cout << "  // cppcheck-suppress incorrectLogicOperator" << std::endl;

        std::cout << "  if ((" << functionCall(functionName, false, -1, numberOfArgs, "") << " > 10) || (";

        std::cout << functionCall(functionName, false, -1, numberOfArgs, "") <<  " < 100)) {}" << std::endl;

        std::cout << "}" << std::endl << std::endl;

    }

    if (leakignore && functionName.find("::") == std::string::npos) {

        std::cout << "void test__" << replaceCharInString(functionName,':','_') << "__leakignore() {" << std::endl;

        std::cout << "  char *p = malloc(10); *p=0;" << std::endl;

        std::cout << "  " << functionCall(functionName, useretval, 1, numberOfArgs, "p") << ';' << std::endl;

        std::cout << "  // cppcheck-suppress memleak" << std::endl;

        std::cout << "}" << std::endl << std::endl;

    }

    // Generate tests for args

    for (int argNr = 1; argNr <= numberOfArgs; ++argNr) {

        for (const tinyxml2::XMLElement *child = node->FirstChildElement(); child; child = child->NextSiblingElement()) {

            if (std::strcmp(child->Name(), "arg") != 0)

                continue;

            const char *nrAttr = child->Attribute("nr");

            if (!nrAttr || std::atoi(nrAttr) != argNr)

                continue;

            bool notbool = false;

            bool notuninit = false;

            bool notnull = false;

            for (const tinyxml2::XMLElement *argCheck = child->FirstChildElement(); argCheck; argCheck = argCheck->NextSiblingElement()) {

                if (std::strcmp(argCheck->Name(), "not-bool") == 0)

                    notbool = true;

                else if (std::strcmp(argCheck->Name(), "not-null") == 0)

                    notnull = true;

                else if (std::strcmp(argCheck->Name(), "not-uninit") == 0)

                    notuninit = true;

            }

            if (notbool) {

                std::cout << testFunctionArg(functionName, useretval, "notbool", argNr, numberOfArgs, "", "invalidFunctionArgBool", "!x") << std::endl << std::endl;

            }

            if (notnull) {

                std::cout << testFunctionArg(functionName, useretval, "notnull", argNr, numberOfArgs, "", "nullPointer", "NULL") << std::endl << std::endl;

            }

            if (notuninit) {

                const char *code = notnull ? "int x[10];" : "int x;";

                std::cout << testFunctionArg(functionName, useretval, "notuninit", argNr, numberOfArgs, code, "uninitvar", "x") << std::endl << std::endl;

            }

        }

    }

}