/*
 *  This file is part of the KDE libraries
 *  Copyright (C) 1999-2000 Harri Porten (porten@kde.org)
 *  Copyright (C) 2004-2006 Apple Computer, Inc.
 *  Copyright (C) 2006 Björn Graf (bjoern.graf@gmail.com)
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Library General Public
 *  License as published by the Free Software Foundation; either
 *  version 2 of the License, or (at your option) any later version.
 *
 *  This library 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
 *  Library General Public License for more details.
 *
 *  You should have received a copy of the GNU Library General Public License
 *  along with this library; see the file COPYING.LIB.  If not, write to
 *  the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 *  Boston, MA 02110-1301, USA.
 *
 */

#include "collector.h"

#include <wtf/HashTraits.h>
#include "JSLock.h"
#include "object.h"
#include "JSVariableObject.h"
#include "Parser.h"

#include <math.h>
#include <stdio.h>
#include <string.h>

#include "global.h"
#if HAVE_SYS_TIME_H
#include <sys/time.h>
#endif

#include "protect.h"

#if defined(WTF_COMPILER_MSVC)
#include <windows.h>
#include <timeapi.h>
#if HAVE_CRTDBG_H
#include <crtdbg.h>
#endif
#endif

#if PLATFORM(WIN_OS) && ! defined(WTF_COMPILER_MSVC)
#ifndef timersub
# define timersub(a, b, result)                                               \
  do {                                                                        \
    (result)->tv_sec = (a)->tv_sec - (b)->tv_sec;                             \
    (result)->tv_usec = (a)->tv_usec - (b)->tv_usec;                          \
    if ((result)->tv_usec < 0) {                                              \
      --(result)->tv_sec;                                                     \
      (result)->tv_usec += 1000000;                                           \
    }                                                                         \
  } while (0)
#endif
#endif

using namespace KJS;
using namespace WTF;

static void testIsInteger();
static void testUString();
static char *createStringWithContentsOfFile(const char *fileName);

class StopWatch
{
public:
    void start();
    void stop();
    long getElapsedMS(); // call stop() first

private:
#if ! HAVE_GETTIMEOFDAY
    DWORD m_startTime;
    DWORD m_stopTime;
#else
    // Windows does not have timeval, disabling this class for now (bug 7399)
    timeval m_startTime;
    timeval m_stopTime;
#endif
};

void StopWatch::start()
{
#if ! HAVE_GETTIMEOFDAY
    m_startTime = timeGetTime();
#else
    gettimeofday(&m_startTime, nullptr);
#endif
}

void StopWatch::stop()
{
#if ! HAVE_GETTIMEOFDAY
    m_stopTime = timeGetTime();
#else
    gettimeofday(&m_stopTime, nullptr);
#endif
}

long StopWatch::getElapsedMS()
{
#if ! HAVE_GETTIMEOFDAY
    return m_stopTime - m_startTime;
#else
    timeval elapsedTime;
    timersub(&m_stopTime, &m_startTime, &elapsedTime);

    return elapsedTime.tv_sec * 1000 + lroundf(elapsedTime.tv_usec / 1000.0f);
#endif
}

class GlobalImp : public JSGlobalObject
{
public:
    UString className() const override
    {
        return "global";
    }
};

class TestFunctionImp : public JSObject
{
public:
    TestFunctionImp(int i, int length);
    bool implementsCall() const override
    {
        return true;
    }
    JSValue *callAsFunction(ExecState *exec, JSObject *thisObj, const List &args) override;

    enum { Print, Debug, Quit, GC, Version, Run };

private:
    int id;
};

TestFunctionImp::TestFunctionImp(int i, int length) : JSObject(), id(i)
{
    putDirect(Identifier("length"), length, DontDelete | ReadOnly | DontEnum);
}

JSValue *TestFunctionImp::callAsFunction(ExecState *exec, JSObject *, const List &args)
{
    switch (id) {
    case Print:
        printf("--> %s\n", JSValue::toString(args[0], exec).UTF8String().c_str());
        return jsUndefined();
    case Debug:
        fprintf(stderr, "--> %s\n", JSValue::toString(args[0], exec).UTF8String().c_str());
        return jsUndefined();
    case GC: {
        JSLock lock;
        Interpreter::collect();
        return jsUndefined();
    }
    case Version:
        // We need this function for compatibility with the Mozilla JS tests but for now
        // we don't actually do any version-specific handling
        return jsUndefined();
    case Run: {
        StopWatch stopWatch;
        char *fileName = strdup(JSValue::toString(args[0], exec).UTF8String().c_str());
        char *script = createStringWithContentsOfFile(fileName);
        if (!script) {
            return throwError(exec, GeneralError, "Could not open file.");
        }

        stopWatch.start();
        exec->dynamicInterpreter()->evaluate(fileName, 0, script);
        stopWatch.stop();

        free(script);
        free(fileName);

        return jsNumber(stopWatch.getElapsedMS());
    }
    case Quit:
        exit(0);
    default:
        abort();
    }
    return nullptr;
}

#if PLATFORM(WIN_OS) && defined(HAVE_CRTDBG_H) && !defined(__MINGW32__)
// Use SEH for Release builds only to get rid of the crash report dialog
// (luckily the same tests fail in Release and Debug builds so far). Need to
// be in a separate main function because the kjsmain function requires object
// unwinding.

#if defined(_DEBUG)
#define TRY
#define EXCEPT(x)
#else
#define TRY       __try {
#define EXCEPT(x) } __except (EXCEPTION_EXECUTE_HANDLER) { x; }
#endif

#else

#define TRY
#define EXCEPT(x)

#endif

int kjsmain(int argc, char **argv);

int main(int argc, char **argv)
{
#if defined(_DEBUG) && PLATFORM(WIN_OS)
    _CrtSetReportFile(_CRT_WARN, _CRTDBG_FILE_STDERR);
    _CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE);
    _CrtSetReportFile(_CRT_ERROR, _CRTDBG_FILE_STDERR);
    _CrtSetReportMode(_CRT_ERROR, _CRTDBG_MODE_FILE);
    _CrtSetReportFile(_CRT_ASSERT, _CRTDBG_FILE_STDERR);
    _CrtSetReportMode(_CRT_ASSERT, _CRTDBG_MODE_FILE);
#endif

    int res = 0;
    TRY
    res = kjsmain(argc, argv);
    EXCEPT(res = 3)
    return res;
}

bool doIt(int argc, char **argv)
{
    bool success = true;
#if 0
    bool prettyPrint = false;
#endif
    GlobalImp *global = new GlobalImp();

    // create interpreter
    RefPtr<Interpreter> interp = new Interpreter(global);
    // add debug() function
    global->put(interp->globalExec(), "debug", new TestFunctionImp(TestFunctionImp::Debug, 1));
    // add "print" for compatibility with the mozilla js shell
    global->put(interp->globalExec(), "print", new TestFunctionImp(TestFunctionImp::Print, 1));
    // add "quit" for compatibility with the mozilla js shell
    global->put(interp->globalExec(), "quit", new TestFunctionImp(TestFunctionImp::Quit, 0));
    // add "gc" for compatibility with the mozilla js shell
    global->put(interp->globalExec(), "gc", new TestFunctionImp(TestFunctionImp::GC, 0));
    // add "version" for compatibility with the mozilla js shell
    global->put(interp->globalExec(), "version", new TestFunctionImp(TestFunctionImp::Version, 1));
    global->put(interp->globalExec(), "run", new TestFunctionImp(TestFunctionImp::Run, 1));

    // abort evaluation of single script file after 5 seconds
    interp->setTimeoutTime(5000);

    Interpreter::setShouldPrintExceptions(true);

    for (int i = 1; i < argc; i++) {
        const char *fileName = argv[i];
        if (strcmp(fileName, "-f") == 0) { // mozilla test driver script uses "-f" prefix for files
            continue;
        }
#if 0
        if (strcmp(fileName, "-p") == 0) {
            prettyPrint = true;
            continue;
        }
#endif

        char *script = createStringWithContentsOfFile(fileName);
        if (!script) {
            success = false;
            break; // fail early so we can catch missing files
        }

#if 0
        if (prettyPrint) {
            int errLine = 0;
            UString errMsg;
            UString s = Parser::prettyPrint(script, &errLine, &errMsg);
            if (s.isNull()) {
                fprintf(stderr, "%s:%d: %s.\n", fileName, errLine, errMsg.UTF8String().c_str());
                success = false;
                break;
            }

            printf("%s\n", s.UTF8String().c_str());
        } else
#endif
        {
            interp->startTimeoutCheck();
            Completion completion = interp->evaluate(fileName, 0, script);
            interp->stopTimeoutCheck();
            success = success && completion.complType() != Throw;
        }
        free(script);
    }

    return success;
}

int kjsmain(int argc, char **argv)
{
    if (argc < 2) {
        fprintf(stderr, "Usage: testkjs file1 [file2...]\n");
        return -1;
    }

    testIsInteger();
    testUString();

    JSLock lock;

    bool success = doIt(argc, argv);

#ifndef NDEBUG
    Collector::collect();
#endif

    if (success) {
        fprintf(stderr, "OK.\n");
    }

    return success ? 0 : 3;
}

static void testIsInteger()
{
    // Unit tests for WTF::IsInteger. Don't have a better place for them now.
    // FIXME: move these once we create a unit test directory for WTF.

    assert(IsInteger<bool>::value);
    assert(IsInteger<char>::value);
    assert(IsInteger<signed char>::value);
    assert(IsInteger<unsigned char>::value);
    assert(IsInteger<short>::value);
    assert(IsInteger<unsigned short>::value);
    assert(IsInteger<int>::value);
    assert(IsInteger<unsigned int>::value);
    assert(IsInteger<long>::value);
    assert(IsInteger<unsigned long>::value);
    assert(IsInteger<long long>::value);
    assert(IsInteger<unsigned long long>::value);

    assert(!IsInteger<char *>::value);
    assert(!IsInteger<const char * >::value);
    assert(!IsInteger<volatile char * >::value);
    assert(!IsInteger<double>::value);
    assert(!IsInteger<float>::value);
    assert(!IsInteger<GlobalImp>::value);
}

// not a good place either
void testUString()
{
    // bug #141720
    UString s1 = "abc";
    UString s2 = s1;
    s1.append("xxx");
    s2.append((unsigned short)0x64);
    assert(s2.size() == 4);
}

static char *createStringWithContentsOfFile(const char *fileName)
{
    char *buffer;

    size_t buffer_size = 0;
    size_t buffer_capacity = 1024;
    buffer = (char *)malloc(buffer_capacity);

    FILE *f = fopen(fileName, "r");
    if (!f) {
        fprintf(stderr, "Could not open file: %s\n", fileName);
        return nullptr;
    }

    while (!feof(f) && !ferror(f)) {
        buffer_size += fread(buffer + buffer_size, 1, buffer_capacity - buffer_size, f);
        if (buffer_size == buffer_capacity) { // guarantees space for trailing '\0'
            buffer_capacity *= 2;
            buffer = (char *)realloc(buffer, buffer_capacity);
            assert(buffer);
        }

        assert(buffer_size < buffer_capacity);
    }
    fclose(f);
    buffer[buffer_size] = '\0';

    return buffer;
}
