/* vim: set ts=4 sw=4 tw=99 et:

 *

 * Copyright (C) 2012 David Anderson

 *

 * This file is part of SourcePawn.

 *

 * SourcePawn 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.

 * 

 * SourcePawn 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

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

 */

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include "../include/keima.h"

#include "Utility.h"

#include "Platform.h"

#ifdef WIN32

# include <Windows.h>

# include <direct.h>

#else

# include <unistd.h>

#endif

using namespace Ke;

class MessageReporter : public IMessageReporter

{

  public:

    void onOutOfMemory() {

        fprintf(stderr, "Out of memory!\n");

    }

    void onUncaughtException(const char *message) {

        fprintf(stderr, "Uncaught exception: %s\n", message);

        IStackFrameIterator *iter = Keima::NewStackFrameIterator();

        unsigned index = 0;

        const char *filename = iter->source();

        if (!filename)

            filename = "<anonymous>";

        if (const char *tail = strrchr(filename, '/'))

            filename = tail + 1;

#ifdef WIN32

        if (const char *tail = strrchr(filename, '\\'))

            filename = tail + 1;

#endif

        while (!iter->done() && index < 10) {

            fprintf(stderr, "[%d] %s (%s line %d)\n",

                    index,

                    iter->function(),

                    filename,

                    iter->lineNumber());

            iter->prev();

            index++;

        }

    }

    virtual void beginCompilationErrors(const char *file) {

        fprintf(stderr, "Displaying compilation errors for \"%s\"\n", file);

    }

    virtual void reportCompilationError(const char *message, unsigned line, unsigned col) {

        fprintf(stderr, "  line %d, col %d: %s\n", line, col, message);

    }

    virtual void endCompilationErrors() {

    }

} s_Reporter;

class MainCollector : public Ke::IPublicFunctionListener

{

  public:

    MainCollector()

    {

    }

    void registerPublic(Handle<Function> fun) {

        Local<String> str = fun->name();

        if (str && strcmp(str->chars(), "main") == 0)

            fun_ = fun;

    }

    void unregisterPublic(Handle<Function> fun) {

        if (fun_ == fun)

            fun_ = Handle<Function>();

    }

    Handle<Function> fun() const {

        return fun_;

    }

  private:

    Root<Function> fun_;

};

intptr_t PrintFloat(void *blah, intptr_t *params)

{

    fprintf(stdout, "%f", *reinterpret_cast<float *>(&params[1]));

    return 0;

}

intptr_t PrintNum(void *blah, intptr_t *params)

{

    fprintf(stdout, "%i", *reinterpret_cast<int *>(&params[1]));

    return 0;

}

intptr_t PrintString(void *blah, intptr_t *params)

{

    fprintf(stdout, "%s", *reinterpret_cast<const char **>(&params[1]));

    return 0;

}

int main(int argc, char **argv)

{

    if (argc < 2) {

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

        return 1;

    }

    if (!Keima::Initialize())

        return 1;

    ke::AutoFree<char> fullpath;

    if (!ke::OS::GetAbsolutePath(argv[1], fullpath))

        return 1;

    ke::AutoFree<char> path, name;

    if (!ke::OS::SplitPath(*fullpath, path, name))

        return 1;

    int rval = 0;

    Keima::InstallMessageReporter(&s_Reporter);

    do {

        MainCollector entry;

        Keima::SetPublicFunctionListener(&entry);

        HandleScope scope;

        Local<Package> package = Package::New(Handle<Package>(), Handle<String>(), *path, Package::Normal);

        if (!package) {

            rval = 1;

            break;

        }

        if (!Keima::Compile(package, *name)) {

            rval = 1;

            break;

        }

        Keima::CollectGarbage();

        Keima::BindNative("PrintFloat", (void *)PrintFloat);

        Keima::BindNative("PrintNum", (void *)PrintNum);

        Keima::BindNative("PrintString", (void *)PrintString);

        Handle<Function> fun = entry.fun();

        if (!fun) {

            fprintf(stdout, "Not running; no public 'main'.\n");

        } else {

            if (!Keima::Invoke(fun, NULL, 0, true, &rval))

                rval = 1;

        }

    } while (0);

    Keima::Shutdown();

    return rval;

}

/* Overload a few things to prevent libstdc++ linking */

#if defined __linux__ || defined __APPLE__

extern "C" void __cxa_pure_virtual(void)

{

}

void *operator new(size_t size)

{

    return malloc(size);

}

void *operator new[](size_t size) 

{

    return malloc(size);

}

void operator delete(void *ptr) 

{

    free(ptr);

}

void operator delete[](void * ptr)

{

    free(ptr);

}

#endif