#include "Statements.h"
#include "tokenize.h"

#include "CommonCheck.h"    // <- IsName

#include <iostream>
#include <sstream>

std::vector<std::string> VariableNames;
std::list<STATEMENT> Statements;

extern bool Debug;
extern bool ShowAll;




//---------------------------------------------------------------------------
// Create statement list
//---------------------------------------------------------------------------

static void AppendStatement(STATEMENT::etype Type, TOKEN *tok, std::string Var="")
{
    STATEMENT NewStatement;
    NewStatement.Type = Type;
    NewStatement.Token = tok;
    if (Var.empty())
    {
        NewStatement.VarIndex = 0;
    }
    else
    {
        bool Found = false;
        for (unsigned int i = 0; i < VariableNames.size(); i++)
        {
            if (VariableNames[i] == Var)
            {
                Found = true;
                NewStatement.VarIndex = i;
                break;
            }
        }

        if ( ! Found )
        {
            NewStatement.VarIndex = VariableNames.size();
            VariableNames.push_back(Var);
        }
    }

    Statements.push_back(NewStatement);
}

TOKEN *GotoNextStatement(TOKEN *tok)
{
    if (tok && (tok->str[0]=='{' || tok->str[0]=='}'))
        return tok->next;

    if (tok)
        tok = tok->next;
    int parlevel = 0;
    for (; tok; tok = tok->next)
    {
        if (tok->str[0] == '(')
            parlevel++;
        else if (tok->str[0] == ')')
            parlevel--;
        if (strchr("{}", tok->str[0]))
            break;
        if (parlevel==0 && tok->str[0] == ';')
        {
            while (tok && tok->str[0] == ';')
                tok = tok->next;
            break;
        }
    }
    return tok;
}


void GetVariableName(TOKEN * &Token, std::string &varname)
{
    varname = "";

    if (Token == NULL)
        return;

    if (!IsName(Token->str))
        return;

    // Get variable name..
    std::ostringstream ostr;
    bool array = false;
    bool name  = false;
    for (TOKEN *tok = Token; tok; tok = tok->next)
    {
        const char *str = tok->str;

        if ((array) || (str[0]=='['))
        {
            ostr << str;
            array = (str[0] != ']');
        }

        else
        {
            if (name && IsName(str))
                return;
            name = IsName(str);
            if (!name && strcmp(str,".") && strcmp(str,"->") && strcmp(str,"::"))
            {
                if (str[0] == '(')
                    return;
                Token = tok;
                break;
            }
            ostr << str;
        }
    }

    varname = ostr.str();
}


void CreateStatementList()
{
    // Clear lists..
    VariableNames.clear();
    Statements.clear();

    int indentlevel = 0;
    for (TOKEN *tok = tokens; tok; tok = GotoNextStatement(tok))
    {
        if (tok->str[0] == '{')
        {
            AppendStatement(STATEMENT::OBRACE, tok);
            indentlevel++;
        }
        else if (tok->str[0] == '}')
        {
            AppendStatement(STATEMENT::EBRACE, tok);
            indentlevel--;
        }
        else if (indentlevel >= 1)
        {
            bool hasif = false, hasloop = false, hasswitch = false;

            if (strcmp(tok->str,"if")==0)
            {
                hasif = true;
                AppendStatement(STATEMENT::IF, tok);
            }
            else if (strcmp(tok->str,"else")==0)
            {
                hasif = true;
                if (strcmp(getstr(tok,1),"if")==0)
                    AppendStatement(STATEMENT::ELSEIF, tok);
                else
                    AppendStatement(STATEMENT::ELSE, tok);
            }

            else if (strcmp(tok->str,"do")==0 ||
                     strcmp(tok->str,"while")==0 ||
                     strcmp(tok->str,"for")==0)
            {
                AppendStatement(STATEMENT::LOOP, tok);
                hasloop = true;
            }

            else if (strcmp(tok->str,"switch")==0)
            {
                AppendStatement(STATEMENT::SWITCH, tok);
                hasswitch = true;
            }

            // Declaring variables..
            if (IsName(tok->str) && strcmp(tok->str,"delete") && strcmp(tok->str,"return"))
            {
                const char *str1 = getstr(tok, 1);
                bool decl = IsName(str1) || str1[0]=='*';
                for (TOKEN *tok2 = decl ? tok->next : NULL; tok2; tok2 = tok2->next)
                {
                    if (strchr("{};.", tok2->str[0]))
                        break;

                    const char *str1 = getstr(tok2, 1);
                    if (IsName(tok2->str) && strchr("[=,;",str1[0]))
                    {
                        AppendStatement(STATEMENT::DECL, tok2, tok2->str);
                        while (tok2 && !strchr(",;", tok2->str[0]))
                            tok2 = tok2->next;
                        if (tok2->str[0] == ';')
                            break;
                    }
                }
            }


            // Assign..
            for (TOKEN *tok2 = tok; tok2; tok2 = tok2->next)
            {
                if (strchr("{};", tok2->str[0]))
                    break;

                TOKEN *eq = tok2;
                std::string varname = "";
                GetVariableName(eq, varname);

                // Equal with..
                if (eq && strcmp(eq->str,"=")==0 && !varname.empty())
                {
                    TOKEN *rs = eq->next;

                    bool ismalloc = false;
                    if (Match(rs, "strdup ("))
                    {
                        ismalloc = true;
                    }
                    else if (rs->str[0]=='(' && IsName(getstr(rs,1)))
                    {
                        ismalloc |= Match(rs, "( %type% * ) malloc (");
                        ismalloc |= Match(rs, "( %type% * * ) malloc (");
                        ismalloc |= Match(rs, "( %type% %type% * ) malloc (");
                        ismalloc |= Match(rs, "( %type% %type% * * ) malloc (");
                        ismalloc |= Match(rs, "( %type% * ) kmalloc (");
                        ismalloc |= Match(rs, "( %type% * * ) kmalloc (");
                        ismalloc |= Match(rs, "( %type% %type% * ) kmalloc (");
                        ismalloc |= Match(rs, "( %type% %type% * * ) kmalloc (");
                    }

                    if ( ismalloc )
                        AppendStatement(STATEMENT::MALLOC, tok2, varname);

                    else if ( Match(rs,"new %type% ;") )
                        AppendStatement(STATEMENT::NEW, tok2, varname);

                    else if ( Match(rs, "new %type% (") )
                        AppendStatement(STATEMENT::NEW, tok2, varname);

                    else if ( Match(rs, "new %type% [") )
                        AppendStatement(STATEMENT::NEWARRAY, tok2, varname);

                    else
                        AppendStatement(STATEMENT::ASSIGN, tok2, varname);

                    tok2 = eq;
                }
            }

            // Delete..
            for (TOKEN *tok2 = tok; tok2; tok2 = tok2->next)
            {
                if (strchr("{};", tok2->str[0]))
                    break;

                if (Match(tok2, "free ( %var% ) ;"))
                    AppendStatement(STATEMENT::FREE, tok2, getstr(tok2, 2));

                if (Match(tok2, "kfree ( %var% ) ;"))
                    AppendStatement(STATEMENT::FREE, tok2, getstr(tok2, 2));

                if (Match(tok2, "delete %var% ;"))
                    AppendStatement(STATEMENT::DELETE, tok2, getstr(tok2,1));

                if (Match(tok2, "delete [ ] %var% ;"))
                    AppendStatement(STATEMENT::DELETEARRAY, tok2, getstr(tok2,3));
            }

            // Use..
            bool UseVar = false;
            int parlevel = 0;
            for (TOKEN *tok2 = tok; tok2; tok2 = tok2->next)
            {
                if (parlevel==0 && strchr("{};", tok2->str[0]))
                    break;

                if (Match(tok2,"free ( %var% )"))
                    break;

                if (tok2->str[0] == '(')
                    parlevel++;
                if (tok2->str[0] == ')')
                    parlevel--;
                if (parlevel == 0 && tok2->str[0] == ',')
                    UseVar = false;

                else if (tok2->str[0]=='=' || tok2->str[0]=='(')
                    UseVar = true;

                else if (UseVar && IsName(tok2->str))
                {
                    std::string varname = "";
                    GetVariableName(tok2, varname);

                    if (!varname.empty() &&
                        varname!="continue" &&
                        varname!="break" &&
                        varname!="return")
                    {
                        AppendStatement(STATEMENT::USE, tok2, varname);
                    }

                    if (tok2->str[0] == ')')
                        parlevel--;

                    if (parlevel==0 && tok2->str[0]==';')
                        break;
                }
            }

            // Return, continue, break..
            for (TOKEN *tok2 = tok; tok2; tok2 = tok2->next)
            {
                if (strchr("{};", tok2->str[0]))
                    break;

                if (strcmp(tok2->str,"continue")==0)
                {
                    AppendStatement(STATEMENT::CONTINUE, tok2);
                    break;
                }

                if (strcmp(tok2->str,"break")==0)
                {
                    AppendStatement(STATEMENT::BREAK, tok2);
                    break;
                }

                if (strcmp(tok2->str,"return")==0)
                {
                    if (IsName(getstr(tok2,1)) && strcmp(getstr(tok2,2),";")==0)
                        AppendStatement(STATEMENT::RETURN, tok2, getstr(tok2,1));

                    else
                        AppendStatement(STATEMENT::RETURN, tok2, ";");

                    break;
                }
            }

            if (hasif)
            {
                AppendStatement(STATEMENT::ENDIF, tok);
            }
            else if (hasloop)
            {
                AppendStatement(STATEMENT::ENDLOOP, tok);
            }
            else if (hasswitch)
            {
                AppendStatement(STATEMENT::ENDSWITCH, tok);
            }
        }
    }

    if (Debug)
    {
        OutputStatementList( std::cout );
    }
}


void OutputStatementList( std::ostream &ostr )
{
    std::list<STATEMENT>::const_iterator it;
    for (it = Statements.begin(); it != Statements.end(); it++)
    {
        STATEMENT s = *it;
        // ostr << it->Token->linenr << " : ";
        switch (s.Type)
        {
            case STATEMENT::OBRACE:
                ostr << "{";
                break;

            case STATEMENT::EBRACE:
                ostr << "}";
                break;

            case STATEMENT::DECL:
                ostr << "decl " << VariableNames[s.VarIndex];
                break;

            case STATEMENT::ASSIGN:
                ostr << "assign " << VariableNames[s.VarIndex];
                break;

            case STATEMENT::MALLOC:
                ostr << "malloc " << VariableNames[s.VarIndex];
                break;

            case STATEMENT::FREE:
                ostr << "free " << VariableNames[s.VarIndex];
                break;

            case STATEMENT::NEW:
                ostr << "new " << VariableNames[s.VarIndex];
                break;

            case STATEMENT::NEWARRAY:
                ostr << "new[] " << VariableNames[s.VarIndex];
                break;

            case STATEMENT::DELETE:
                ostr << "delete " << VariableNames[s.VarIndex];
                break;

            case STATEMENT::DELETEARRAY:
                ostr << "delete[] " << VariableNames[s.VarIndex];
                break;

            case STATEMENT::USE:
                ostr << "use " << VariableNames[s.VarIndex];
                break;


            case STATEMENT::LOOP:
                ostr << "loop";
                break;

            case STATEMENT::ENDLOOP:
                ostr << "endloop";
                break;


            case STATEMENT::SWITCH:
                ostr << "switch";
                break;

            case STATEMENT::ENDSWITCH:
                ostr << "endswitch";
                break;


            case STATEMENT::IF:
                ostr << "if";
                break;

            case STATEMENT::ELSEIF:
                ostr << "elseif";
                break;

            case STATEMENT::ELSE:
                ostr << "else";
                break;

            case STATEMENT::ENDIF:
                ostr << "endif";
                break;

            case STATEMENT::RETURN:
                ostr << "return " << VariableNames[s.VarIndex];
                break;

            case STATEMENT::CONTINUE:
                ostr << "continue";
                break;

            case STATEMENT::BREAK:
                ostr << "break";
                break;

            default:
                ostr << "ERROR. Unknown code!!";
                break;
        }
        ostr << "\n";
    }
}
//---------------------------------------------------------------------------