From a509408b05a4105a163a198ff10a1832964b8090 Mon Sep 17 00:00:00 2001
From: Syping <schiedelrafael@keppe.org>
Date: Thu, 17 Oct 2019 14:25:49 +0200
Subject: [PATCH] improve interpreter support

---
 src/luaengine/main.cpp  | 14 ++++++++-
 src/luaenginec/main.cpp | 69 ++++++++++++++++++++++++++++-------------
 2 files changed, 61 insertions(+), 22 deletions(-)

diff --git a/src/luaengine/main.cpp b/src/luaengine/main.cpp
index cc3bd67..bf29c8c 100644
--- a/src/luaengine/main.cpp
+++ b/src/luaengine/main.cpp
@@ -19,6 +19,7 @@
 #include "LuaEngineIO.h"
 #include "LuaEngineOS.h"
 #include <QApplication>
+#include <QTextStream>
 #include <QFont>
 #include <QFile>
 
@@ -44,11 +45,22 @@ int main(int argc, char *argv[])
     if (arguments.length() >= 2) {
         arguments.removeFirst();
         QFile luaScript(arguments.first().toString());
+        if (!luaScript.open(QIODevice::ReadOnly)) {
+            QTextStream(stderr) << "Error: Failed to open \"" << arguments.first().toString() << "\"." << endl;
+            return 1;
+        }
+
+        if (QString::fromUtf8(luaScript.read(2)) == "#!") {
+            luaScript.readLine();
+        }
+        else {
+            luaScript.reset();
+        }
 
         LuaEngineGui luaEngineGui;
         LuaEngineIO::pushClass(&luaEngineGui);
         LuaEngineOS::pushClass(&luaEngineGui);
-        luaEngineGui.executeLuaScript(&luaScript);
+        luaEngineGui.executeLuaScript(luaScript.readAll());
 
         if (luaEngineGui.executeLuaFunction("main", arguments, true)) {
             QVariant variant = luaEngineGui.returnVariant();
diff --git a/src/luaenginec/main.cpp b/src/luaenginec/main.cpp
index f024ed7..60be8eb 100644
--- a/src/luaenginec/main.cpp
+++ b/src/luaenginec/main.cpp
@@ -22,6 +22,7 @@
 #include <QTextStream>
 #include <QSaveFile>
 #include <QFile>
+#include <QDir>
 
 int main(int argc, char *argv[])
 {
@@ -31,7 +32,7 @@ int main(int argc, char *argv[])
 
     QCommandLineParser parser;
 
-    QCommandLineOption pLuaEngineOption(QStringList() << "le" << "luaengine", "Location of portable Lua Engine executor.", "luaengine");
+    QCommandLineOption pLuaEngineOption(QStringList() << "le" << "luaengine", "Location of portable LuaEngine executor.", "luaengine");
     parser.addOption(pLuaEngineOption);
 
     parser.addPositionalArgument("source", "Source file to build.");
@@ -42,19 +43,45 @@ int main(int argc, char *argv[])
 
     parser.process(a);
 
+    QString inputFile;
+    QString outputFile;
     const QStringList args = parser.positionalArguments();
+    if (args.length() < 1) {
+        QTextStream(stderr) << "Error: No input files." << endl;
+        return 1;
+    }
+    else {
+        inputFile = args.at(0);
+    }
+    if (args.length() < 2) {
+        outputFile = QDir::current().absolutePath() + "/luaenginec.out";
+    }
+    else {
+        outputFile = args.at(1);
+    }
 
-    QFile luaScript(args.at(0));
-
-    LuaEngine luaEngine;
-    if (!luaEngine.loadLuaScript(&luaScript)) {
-        QTextStream(stderr) << "Failed to load \"" << args.at(0) << "\"." << endl;
+    QFile luaScript(inputFile);
+    if (!luaScript.open(QIODevice::ReadOnly)) {
+        QTextStream(stderr) << "Error: Failed to open \"" << inputFile << "\"." << endl;
         return 1;
     }
 
-    QSaveFile outputFile(args.at(1));
-    if (!outputFile.open(QIODevice::WriteOnly)) {
-        QTextStream(stderr) << "Failed to open \"" << args.at(1) << "\"." << endl;
+    if (QString::fromUtf8(luaScript.read(2)) == "#!") {
+        luaScript.readLine();
+    }
+    else {
+        luaScript.reset();
+    }
+
+    LuaEngine luaEngine;
+    if (!luaEngine.loadLuaScript(luaScript.readAll())) {
+        QTextStream(stderr) << "Error: Failed to load \"" << inputFile << "\"." << endl;
+        return 1;
+    }
+
+    QSaveFile outputSaveFile(outputFile);
+    if (!outputSaveFile.open(QIODevice::WriteOnly)) {
+        QTextStream(stderr) << "Error: Failed to open \"" << outputFile << "\"." << endl;
         return 1;
     }
 
@@ -62,24 +89,24 @@ int main(int argc, char *argv[])
     if (!pLuaEngine.isEmpty()) {
         QFile portableLuaEngine(pLuaEngine);
         if (!portableLuaEngine.open(QIODevice::ReadOnly)) {
-            outputFile.cancelWriting();
-            outputFile.commit();
-            QTextStream(stderr) << "Failed to open \"" << pLuaEngine << "\"." << endl;
+            outputSaveFile.cancelWriting();
+            outputSaveFile.commit();
+            QTextStream(stderr) << "Error: Failed to open \"" << pLuaEngine << "\"." << endl;
             return 1;
         }
-        outputFile.write(portableLuaEngine.readAll());
+        outputSaveFile.write(portableLuaEngine.readAll());
     }
 
     const QByteArray luaDump = luaEngine.dumpLuaScript();
     int dumpSize = luaDump.size();
-    outputFile.write(luaDump);
+    outputSaveFile.write(luaDump);
 
     if (!pLuaEngine.isEmpty()) {
         QByteArray lengthArray = QByteArray::number(dumpSize, 16);
         if (lengthArray.size() > 8) {
-            outputFile.cancelWriting();
-            outputFile.commit();
-            QTextStream(stderr) << "Lua Engine script is too large." << endl;
+            outputSaveFile.cancelWriting();
+            outputSaveFile.commit();
+            QTextStream(stderr) << "Error: Lua Engine script is too large." << endl;
             return 1;
         }
         else {
@@ -87,12 +114,12 @@ int main(int argc, char *argv[])
                 lengthArray.insert(0, "0");
             }
         }
-        outputFile.write(QByteArray::fromHex(lengthArray));
-        outputFile.write("\xb4\x00", 2);
+        outputSaveFile.write(QByteArray::fromHex(lengthArray));
+        outputSaveFile.write("\xb4\x00", 2);
     }
 
-    if (!outputFile.commit()) {
-        QTextStream(stderr) << "Failed to write \"" << args.at(1) << "\"." << endl;
+    if (!outputSaveFile.commit()) {
+        QTextStream(stderr) << "Error: Failed to write \"" << args.at(1) << "\"." << endl;
         return 1;
     }