/*
 *  Copyright 2014 The Luvit Authors. All Rights Reserved.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 */

#define LUA_LIB
#include "luvi.h"
#include "luv.h"
#include "lenv.c"
#include "luvi.c"
#include "lminiz.c"
#include "snapshot.c"
#ifdef WITH_PCRE
int luaopen_rex_pcre(lua_State* L);
#endif
#include "../deps/strlib.c"
#ifdef WITH_PLAIN_LUA
#include "../deps/bit.c"
#endif

#ifdef WITH_CUSTOM
int luvi_custom(lua_State* L);
#endif

static int luvi_traceback(lua_State *L) {
  if (!lua_isstring(L, 1))  /* 'message' not a string? */
    return 1;  /* keep it intact */
  lua_pushglobaltable(L);
  lua_getfield(L, -1, "debug");
  lua_remove(L, -2);
  if (!lua_istable(L, -1)) {
    lua_pop(L, 1);
    return 1;
  }
  lua_getfield(L, -1, "traceback");
  if (!lua_isfunction(L, -1)) {
    lua_pop(L, 2);
    return 1;
  }
  lua_pushvalue(L, 1);  /* pass error message */
  lua_pushinteger(L, 2);  /* skip this function and traceback */
  lua_call(L, 2, 1);  /* call debug.traceback */
  return 1;
}

static lua_State* vm_acquire(){
  lua_State*L = luaL_newstate();
  if (L == NULL)
    return L;

  // Add in the lua standard libraries
  luaL_openlibs(L);

  // Add string lua 5.3 compat apis, pack,unpack,packsize
#if LUA_VERSION_NUM < 503
  make_compat53_string(L);
#endif

  // Get package.preload so we can store builtins in it.
  lua_getglobal(L, "package");
  lua_getfield(L, -1, "preload");
  lua_remove(L, -2); // Remove package

  // Store uv module definition at preload.uv
  lua_pushcfunction(L, luaopen_luv);
  lua_setfield(L, -2, "uv");

  lua_pushcfunction(L, luaopen_env);
  lua_setfield(L, -2, "env");

  lua_pushcfunction(L, luaopen_miniz);
  lua_setfield(L, -2, "miniz");

  lua_pushcfunction(L, luaopen_snapshot);
  lua_setfield(L, -2, "snapshot");

#ifdef WITH_LPEG
  lua_pushcfunction(L, luaopen_lpeg);
  lua_setfield(L, -2, "lpeg");
#endif

#ifdef WITH_PCRE
  lua_pushcfunction(L, luaopen_rex_pcre);
  lua_setfield(L, -2, "rex");
#endif

  // Store luvi module definition at preload.luvi
  lua_pushcfunction(L, luaopen_luvi);
  lua_setfield(L, -2, "luvi");

#ifdef WITH_OPENSSL
  // Store openssl module definition at preload.openssl
  lua_pushcfunction(L, luaopen_openssl);
  lua_setfield(L, -2, "openssl");
#endif

#ifdef WITH_ZLIB
  // Store zlib module definition at preload.zlib
  lua_pushcfunction(L, luaopen_zlib);
  lua_setfield(L, -2, "zlib");
#endif

#ifdef WITH_PLAIN_LUA
  {
      LUALIB_API int luaopen_init(lua_State *L);
      LUALIB_API int luaopen_luvibundle(lua_State *L);
      LUALIB_API int luaopen_luvipath(lua_State *L);
      lua_pushcfunction(L, luaopen_init);
      lua_setfield(L, -2, "init");
      lua_pushcfunction(L, luaopen_luvibundle);
      lua_setfield(L, -2, "luvibundle");
      lua_pushcfunction(L, luaopen_luvipath);
      lua_setfield(L, -2, "luvipath");
      luaL_requiref(L, "bit", luaopen_bit, 1);
  }
#endif

#ifdef WITH_CUSTOM
  luvi_custom(L);
#endif
  return L;
}

static void vm_release(lua_State*L) {
  lua_close(L);
}

int main(int argc, char* argv[] ) {

  lua_State* L;
  int index;
  int res;
  int errfunc;

  // Hooks in libuv that need to be done in main.
  argv = uv_setup_args(argc, argv);

  // Create the lua state.
  L = vm_acquire();
  if (L == NULL) {
    fprintf(stderr, "luaL_newstate has failed\n");
    return 1;
  }

  luv_set_thread_cb(vm_acquire, vm_release);

#ifdef WITH_WINSVC
  // Store luvi module definition at preload.openssl
  lua_pushcfunction(L, luaopen_winsvc);
  lua_setfield(L, -2, "winsvc");
  lua_pushcfunction(L, luaopen_winsvcaux);
  lua_setfield(L, -2, "winsvcaux");
#endif

  /* push debug function */
  lua_pushcfunction(L, luvi_traceback);
  errfunc = lua_gettop(L);

  // Load the init.lua script
  if (luaL_loadstring(L, "return require('init')(...)")) {
    fprintf(stderr, "%s\n", lua_tostring(L, -1));
    vm_release(L);
    return -1;
  }

  // Pass the command-line arguments to init as a zero-indexed table
  lua_createtable (L, argc, 0);
  for (index = 0; index < argc; index++) {
    lua_pushstring(L, argv[index]);
    lua_rawseti(L, -2, index);
  }

  // Start the main script.
  if (lua_pcall(L, 1, 1, errfunc)) {
    fprintf(stderr, "%s\n", lua_tostring(L, -1));
    vm_release(L);
    return -1;
  }

  // Use the return value from the script as process exit code.
  res = 0;
  if (lua_type(L, -1) == LUA_TNUMBER) {
    res = (int)lua_tointeger(L, -1);
  }
  vm_release(L);
  return res;
}