releasing moonplus as a lib.

1 files changed, 121 insertions, 7 deletions

diff --git a/src/moonc.cpp b/src/moonp.cpp
index d04ce47..a6c99df 100644
--- a/src/moonc.cpp
+++ b/src/moonp.cpp
@@ -13,9 +13,11 @@ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLI
 #include "MoonP/moon_compiler.h"
 #include "MoonP/moon_parser.h"
 
+#ifndef LIBMOONP
+
 int main(int narg, const char** args) {
         const char* help =
-"Usage: moonc [options|files] ...\n\n"
+"Usage: moonp [options|files] ...\n\n"
 "    -h          Print this message\n"
 "    -t path     Specify where to place compiled files\n"
 "    -o file     Write output to file\n"
@@ -82,7 +84,7 @@ int main(int narg, const char** args) {
                 std::cout << "Error: -o can not be used with multiple input files.\n";
                 std::cout << help;
         }
-        if (targetPath.back() != '/' && targetPath.back() != '\\') {
+        if (!targetPath.empty() && targetPath.back() != '/' && targetPath.back() != '\\') {
                 targetPath.append("/");
         }
         std::list<std::future<std::result_of_t<std::decay_t<int()>()>>> results;
@@ -121,14 +123,9 @@ int main(int narg, const char** args) {
                                         } else {
                                                 std::string targetFile;
                                                 if (resultFile.empty()) {
-                                                        std::string ext;
                                                         targetFile = file;
                                                         size_t pos = file.rfind('.');
                                                         if (pos != std::string::npos) {
-                                                                ext = file.substr(pos + 1);
-                                                                for (size_t i = 0; i < ext.length(); i++) {
-                                                                        ext[i] = static_cast<char>(tolower(ext[i]));
-                                                                }
                                                                 targetFile = file.substr(0, pos) + ".lua";
                                                         }
                                                         if (!targetPath.empty()) {
@@ -177,3 +174,120 @@ int main(int narg, const char** args) {
         return ret;
 }
 
+#else
+
+extern "C" {
+
+#include "lua.h"
+#include "lauxlib.h"
+
+static const char moonplusCodes[] =
+#include "MoonPlus.h"
+
+static int init_moonplus(lua_State* L) {
+        MoonP::MoonConfig config;
+        std::string s(moonplusCodes, sizeof(moonplusCodes) / sizeof(moonplusCodes[0]) - 1);
+        std::string codes, err;
+        MoonP::GlobalVars globals;
+        std::tie(codes, err, globals) = MoonP::moonCompile(s, config);
+        if (codes.empty()) {
+                luaL_error(L, "fail to compile moonplus init codes.\n%s", err.c_str());
+        }
+        int top = lua_gettop(L);
+        if (luaL_loadbuffer(L, codes.c_str(), codes.size(), "=(moonplus)") != 0) {
+                luaL_error(L, "fail to init moonplus module.");
+        } else {
+                lua_call(L, 0, 0);
+        }
+        lua_settop(L, top);
+        return 0;
+}
+
+static const char stpCodes[] =
+#include "StackTracePlus.h"
+
+static int init_stacktraceplus(lua_State* L) {
+        if (luaL_loadbuffer(L, stpCodes, sizeof(stpCodes) / sizeof(stpCodes[0]) - 1, "=(stacktraceplus)") != 0) {
+                luaL_error(L, "fail to init stacktraceplus module.");
+        } else {
+                lua_call(L, 0, 1);
+        }
+        return 1;
+}
+
+static int moontolua(lua_State* L) {
+        size_t size = 0;
+        const char* input = luaL_checklstring(L, 1, &size);
+        MoonP::MoonConfig config;
+        if (lua_gettop(L) == 2) {
+                lua_pushstring(L, "lint_global");
+                lua_gettable(L, -2);
+                if (!lua_isnil(L, -1)) {
+                        config.lintGlobalVariable = lua_toboolean(L, -1) != 0;
+                }
+                lua_pop(L, 1);
+                lua_pushstring(L, "implicit_return_root");
+                lua_gettable(L, -2);
+                if (!lua_isnil(L, -1)) {
+                        config.implicitReturnRoot = lua_toboolean(L, -1) != 0;
+                }
+                lua_pop(L, 1);
+                lua_pushstring(L, "reserve_line_number");
+                lua_gettable(L, -2);
+                if (!lua_isnil(L, -1)) {
+                        config.reserveLineNumber = lua_toboolean(L, -1) != 0;
+                }
+                lua_pop(L, 1);
+        }
+        std::string s(input, size);
+        std::string codes, err;
+        MoonP::GlobalVars globals;
+        std::tie(codes, err, globals) = MoonP::moonCompile(s, config);
+        if (codes.empty()) {
+                lua_pushnil(L);
+        } else {
+                lua_pushlstring(L, codes.c_str(), codes.size());
+        }
+        if (err.empty()) {
+                lua_pushnil(L);
+        } else {
+                lua_pushlstring(L, err.c_str(), err.size());
+        }
+        if (globals && !globals->empty()) {
+                lua_createtable(L, static_cast<int>(globals->size()), 0);
+                int i = 1;
+                for (const auto& var : *globals) {
+                        lua_createtable(L, 3, 0);
+                        lua_pushlstring(L, var.name.c_str(), var.name.size());
+                        lua_rawseti(L, -2, 1);
+                        lua_pushinteger(L, var.line);
+                        lua_rawseti(L, -2, 2);
+                        lua_pushinteger(L, var.col);
+                        lua_rawseti(L, -2, 3);
+                        lua_rawseti(L, -2, i);
+                        i++;
+                }
+        } else {
+                lua_pushnil(L);
+        }
+        return 3;
+}
+
+int luaopen_moonp(lua_State* L) {
+        lua_getglobal(L, "package");
+        lua_getfield(L, -1, "loaded");
+        lua_createtable(L, 0, 0);
+        lua_pushcfunction(L, moontolua);
+        lua_setfield(L, -2, "to_lua");
+        lua_pushcfunction(L, init_stacktraceplus);
+        lua_setfield(L, -2, "load_stacktraceplus");
+        lua_setfield(L, -2, "moonp");
+        lua_pop(L, 2);
+        init_moonplus(L);
+        return 0;
+}
+
+} // extern "C"
+
+#endif // LIBMOONP
+