/*

 * Copyright (C) 2012 David Anderson

 *

 * This file is part of JITCraft.

 *

 * JITCraft is free software: you can redistribute it and/or modify it under

 * the terms of the GNU General Public License as published by the Free

 * Software Foundation, either version 3 of the License, or (at your option)

 * any later version.

 * 

 * Foobar 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License along with

 * JITCraft. If not, see http://www.gnu.org/licenses/.

 */

#include <stdio.h>

#include "Utility.h"

#include "Scanner.h"

#include "Parser.h"

#include "SemanticAnalysis.h"

#include "Zone.h"

#include "BytecodeCompiler.h"

#include "Code.h"

#include "Interpreter.h"

#include "Modules.h"

#include "RootScopes.h"

using namespace jc;

class File

{

    FILE *fp_;

  public:

    File(FILE *fp)

      : fp_(fp)

    {

    }

    ~File() {

        if (fp_)

            fclose(fp_);

    }

    operator FILE *() {

        return fp_;

    }

    operator bool () {

        return !!fp_;

    }

};

static bool

Compile(const char *stream, size_t length)

{

    if (!ZONE()->initialize())

        return false;

    RootScope roots(ZONE());

    CompileContext cc(ZONE());

    Parser parser(ZONE(), cc, stream, length);

    ParseTree *tree = parser.parse();

    if (!tree)

        return false;

    tree->dump(stdout);

    if (!Scope::Bind(tree))

        return false;

    SemanticAnalysis semantics(ZONE(), cc, tree);

    if (!semantics.analyze())

        return false;

    Scope::Allocate(tree);

    BytecodeCompiler compiler(ZONE(), tree, NULL);

    Module *module = compiler.compile();

    if (!module)

        return false;

    printf("\n ------------ \n\n");

    Disassemble(module->getFunction(1)->code(), stdout);

    //Disassemble(module->getFunction(1), stdout);

    printf("\n ------------ \n\n");

    Function *main = module->getFunction(1);

    Top top = ZONE()->stack().begin();

    LocalTop localTop(top);

    Frame *frame = ZONE()->stack().pushFrame(main);

    if (!frame)

        return false;

    LocalSlot result;

    if (!Interpret(ZONE(), frame, &result))

        return false;

    printf("Result: %d\n", result.i32);

    return true;

}

int main(int argc, char **argv)

{

    if (argc < 2) {

        fprintf(stderr, "Usage: <file>\n");

        return 1;

    }

    AutoVec<char> bytes(NULL);

    size_t length;

    {

        File fp = fopen(argv[1], "rb");

        if (!fp) {

            fprintf(stderr, "Can't open file: \"%s\"\n", argv[1]);

            return 1;

        }

        if (fseek(fp, 0, SEEK_END)) {

            fprintf(stderr, "Error seeking file!\n");

            return 1;

        }

        long size = ftell(fp);

        if (size < 0) {

            fprintf(stderr, "Error getting file length!\n");

            return 1;

        }

        rewind(fp);

        length = size_t(size);

        bytes = new char[length + 1];

        if (fread(bytes, 1, length, fp) != length) {

            fprintf(stderr, "Error reading file!\n");

            return 1;

        }

    }

    Compile(bytes, length);

}