luau-lang/luau
1
0
Fork 0
mirror of https://github.com/luau-lang/luau.git synced 2025-04-11 22:30:53 +01:00
Code Issues Projects Releases Packages Wiki Activity

Update Repl.cpp

Browse source
This commit is contained in:
NotDSF 2021-11-08 07:56:58 +00:00
parent eb40adb727
commit 480ee0df75
1 changed files with 45 additions and 49 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

94
CLI/Repl.cpp
View file

@ -13,6 +13,12 @@
#include <memory> #include <memory>
enum class CompileFormat
{
Default,
Binary
};
static int lua_loadstring(lua_State* L) static int lua_loadstring(lua_State* L)
{ {
size_t l = 0; size_t l = 0;
@ -370,7 +376,7 @@ static void reportError(const char* name, const Luau::CompileError& error)
report(name, error.getLocation(), "CompileError", error.what()); report(name, error.getLocation(), "CompileError", error.what());
} }
static bool compileFile(const char* name) static bool compileFile(const char* name, CompileFormat format)
{ {
std::optional<std::string> source = readFile(name); std::optional<std::string> source = readFile(name);
if (!source) if (!source)
@ -387,40 +393,15 @@ static bool compileFile(const char* name)
Luau::compileOrThrow(bcb, *source); Luau::compileOrThrow(bcb, *source);
printf("%s", bcb.dumpEverything().c_str()); switch (format)
{
return true; case CompileFormat::Default:
} printf("%s", bcb.dumpEverything().c_str());
catch (Luau::ParseErrors& e) break;
{ case CompileFormat::Binary:
for (auto& error : e.getErrors()) printf("%s", bcb.getBytecode().data());
reportError(name, error); break;
return false; }
}
catch (Luau::CompileError& e)
{
reportError(name, e);
return false;
}
}
static bool dumpFile(const char* name) {
std::optional<std::string> source = readFile(name);
if (!source)
{
printf("Error opening %s\n", name);
return false;
}
try
{
Luau::BytecodeBuilder bcb;
Luau::compileOrThrow(bcb, *source);
std::string Bytecode = bcb.getBytecode();
std::ofstream out("luau.out");
out << Bytecode;
out.close();
return true; return true;
} }
@ -445,8 +426,7 @@ static void displayHelp(const char* argv0)
printf("\n"); printf("\n");
printf("Available modes:\n"); printf("Available modes:\n");
printf(" omitted: compile and run input files one by one\n"); printf(" omitted: compile and run input files one by one\n");
printf(" --compile: compile input files and output resulting bytecode\n"); printf(" --compile[=format]: compile input files and output resulting formatted bytecode (binary or text)\n");
printf(" --binary: compile input file and save binary bytecode blob to luau.out\n");
printf("\n"); printf("\n");
printf("Available options:\n"); printf("Available options:\n");
printf(" --profile[=N]: profile the code using N Hz sampling (default 10000) and output results to profile.out\n"); printf(" --profile[=N]: profile the code using N Hz sampling (default 10000) and output results to profile.out\n");
@ -478,16 +458,7 @@ int main(int argc, char** argv)
return 0; return 0;
} }
if (argc >= 2 && strcmp(argv[1], "--binary") == 0) { if (argc >= 2 && strcmp(argv[1], "--compile=text") == 0 || strcmp(argv[1], "--compile") == 0)
if (!argv[2]) {
printf("Please input a file\n");
return 1;
}
return !dumpFile(argv[2]);
}
if (argc >= 2 && strcmp(argv[1], "--compile") == 0)
{ {
int failed = 0; int failed = 0;
@ -500,12 +471,37 @@ int main(int argc, char** argv)
{ {
traverseDirectory(argv[i], [&](const std::string& name) { traverseDirectory(argv[i], [&](const std::string& name) {
if (name.length() > 4 && name.rfind(".lua") == name.length() - 4) if (name.length() > 4 && name.rfind(".lua") == name.length() - 4)
failed += !compileFile(name.c_str()); failed += !compileFile(name.c_str(), CompileFormat::Default);
}); });
} }
else else
{ {
failed += !compileFile(argv[i]); failed += !compileFile(argv[i], CompileFormat::Default);
}
}
return failed;
}
if (argc >= 2 && strcmp(argv[1], "--compile=binary") == 0)
{
int failed = 0;
for (int i = 2; i < argc; ++i)
{
if (argv[i][0] == '-')
continue;
if (isDirectory(argv[i]))
{
traverseDirectory(argv[i], [&](const std::string& name) {
if (name.length() > 4 && name.rfind(".lua") == name.length() - 4)
failed += !compileFile(name.c_str(), CompileFormat::Binary);
});
}
else
{
failed += !compileFile(argv[i], CompileFormat::Binary);
} }
} }