Method Include

src/lua/LuaVFS.cpp:266–307

//============================================================================//

int LuaVFS::Include(lua_State* L) {
  lua_settop(L, 3);

  const char* path = luaL_checkstring(L, 1);
  std::string modes = luaL_optstring(L, 2, GetReadFilter(path, L));
  modes = BzVFS::allowModes(modes, L2ES(L)->vfsModes->readAllowed);

  std::string code;
  if (!bzVFS.readFile(path, modes, code)) {
    luaL_error(L, "file not found");
  }

  // compile the code
  int error = luaL_loadbuffer(L, code.data(), code.size(), path);
  if (error != 0) {
    lua_error(L);
  }
  const int funcIndex = 4;

  // setup the function's environment
  if (lua_istable(L, 3)) {
    lua_pushvalue(L, 3);
  }
  else if (lua_isnil(L, 3)) {
    LuaUtils::PushCurrentFuncEnv(L, "Include");
  }
  else {
    luaL_error(L, "bad fenv parameter");
  }

  if (lua_setfenv(L, -2) == 0) {
    luaL_error(L, "Include: error with setfenv");
  }

  // execute the code
  error = lua_pcall(L, 0, LUA_MULTRET, 0);
  if (error != 0) {
    lua_error(L);
  }

  return lua_gettop(L) - (funcIndex - 1);
}


//============================================================================//