From d2386fac9b9f1432bfe70e5a0dc295fd11febb12 Mon Sep 17 00:00:00 2001
From: BigBahss <tdennis4496@gmail.com>
Date: Fri, 29 Jan 2021 22:05:08 -0500
Subject: [PATCH] Add functions to parse all global event scripts labels in the
 project

---
 include/core/parseutil.h |  9 ++++++++
 src/core/parseutil.cpp   | 50 +++++++++++++++++++++++++++++++++++++---
 2 files changed, 56 insertions(+), 3 deletions(-)

diff --git a/include/core/parseutil.h b/include/core/parseutil.h
index 82fad4e2..17a1cf4c 100644
--- a/include/core/parseutil.h
+++ b/include/core/parseutil.h
@@ -60,6 +60,9 @@ public:
     static int getScriptLineNumber(const QString &filePath, const QString &scriptLabel);
     static int getRawScriptLineNumber(QString text, const QString &scriptLabel);
     static int getPoryScriptLineNumber(QString text, const QString &scriptLabel);
+    static QStringList getGlobalScriptLabels(const QString &filePath);
+    static QStringList getGlobalRawScriptLabels(QString text);
+    static QStringList getGlobalPoryScriptLabels(QString text);
     static QString &removeStringLiterals(QString &text);
     static QString &removeLineComments(QString &text, const QString &commentSymbol);
     static QString &removeLineComments(QString &text, const QStringList &commentSymbols);
@@ -74,6 +77,12 @@ private:
     QList<Token> generatePostfix(QList<Token> tokens);
     int evaluatePostfix(QList<Token> postfix);
     void error(QString message, QString expression);
+
+    static const QRegularExpression re_incScriptLabel;
+    static const QRegularExpression re_globalIncScriptLabel;
+    static const QRegularExpression re_poryScriptLabel;
+    static const QRegularExpression re_globalPoryScriptLabel;
+    static const QRegularExpression re_poryRawSection;
 };
 
 #endif // PARSEUTIL_H
diff --git a/src/core/parseutil.cpp b/src/core/parseutil.cpp
index 74330738..e37772e8 100644
--- a/src/core/parseutil.cpp
+++ b/src/core/parseutil.cpp
@@ -6,6 +6,12 @@
 #include <QJsonObject>
 #include <QStack>
 
+const QRegularExpression ParseUtil::re_incScriptLabel("\\b(?<label>[\\w_][\\w\\d_]*):{1,2}");
+const QRegularExpression ParseUtil::re_globalIncScriptLabel("\\b(?<label>[\\w_][\\w\\d_]*)::");
+const QRegularExpression ParseUtil::re_poryScriptLabel("\\b(script)(\\((global|local)\\))?\\s*\\b(?<label>[\\w_][\\w\\d_]*)");
+const QRegularExpression ParseUtil::re_globalPoryScriptLabel("\\b(script)(\\((global)\\))?\\s*\\b(?<label>[\\w_][\\w\\d_]*)");
+const QRegularExpression ParseUtil::re_poryRawSection("\\b(raw)\\s*`(?<raw_script>[^`]*)");
+
 ParseUtil::ParseUtil()
 {
 }
@@ -437,7 +443,6 @@ int ParseUtil::getRawScriptLineNumber(QString text, const QString &scriptLabel)
     removeStringLiterals(text);
     removeLineComments(text, "@");
 
-    static const QRegularExpression re_incScriptLabel("\\b(?<label>[\\w_][\\w\\d_]*):{1,2}");
     QRegularExpressionMatchIterator it = re_incScriptLabel.globalMatch(text);
     while (it.hasNext()) {
         const QRegularExpressionMatch match = it.next();
@@ -452,7 +457,6 @@ int ParseUtil::getPoryScriptLineNumber(QString text, const QString &scriptLabel)
     removeStringLiterals(text);
     removeLineComments(text, {"//", "#"});
 
-    static const QRegularExpression re_poryScriptLabel("\\b(script)(\\((global|local)\\))?\\s*\\b(?<label>[\\w_][\\w\\d_]*)");
     QRegularExpressionMatchIterator it = re_poryScriptLabel.globalMatch(text);
     while (it.hasNext()) {
         const QRegularExpressionMatch match = it.next();
@@ -460,7 +464,6 @@ int ParseUtil::getPoryScriptLineNumber(QString text, const QString &scriptLabel)
             return text.left(match.capturedStart("label")).count('\n') + 1;
     }
 
-    static const QRegularExpression re_poryRawSection("\\b(raw)\\s*`(?<raw_script>[^`]*)");
     QRegularExpressionMatchIterator raw_it = re_poryRawSection.globalMatch(text);
     while (raw_it.hasNext()) {
         const QRegularExpressionMatch match = raw_it.next();
@@ -472,6 +475,47 @@ int ParseUtil::getPoryScriptLineNumber(QString text, const QString &scriptLabel)
     return 0;
 }
 
+QStringList ParseUtil::getGlobalScriptLabels(const QString &filePath) {
+    if (filePath.endsWith(".inc") || filePath.endsWith(".s"))
+        return getGlobalRawScriptLabels(readTextFile(filePath));
+    else if (filePath.endsWith(".pory"))
+        return getGlobalPoryScriptLabels(readTextFile(filePath));
+
+    return { };
+}
+
+QStringList ParseUtil::getGlobalRawScriptLabels(QString text) {
+    removeStringLiterals(text);
+    removeLineComments(text, "@");
+
+    auto rawScriptLabels = QStringList();
+
+    QRegularExpressionMatchIterator it = re_globalIncScriptLabel.globalMatch(text);
+    while (it.hasNext()) {
+        const QRegularExpressionMatch match = it.next();
+        rawScriptLabels << match.captured("label");
+    }
+
+    return rawScriptLabels;
+}
+
+QStringList ParseUtil::getGlobalPoryScriptLabels(QString text) {
+    removeStringLiterals(text);
+    removeLineComments(text, {"//", "#"});
+
+    auto poryScriptLabels = QStringList();
+
+    QRegularExpressionMatchIterator it = re_globalPoryScriptLabel.globalMatch(text);
+    while (it.hasNext())
+        poryScriptLabels << it.next().captured("label");
+
+    QRegularExpressionMatchIterator raw_it = re_poryRawSection.globalMatch(text);
+    while (raw_it.hasNext())
+        poryScriptLabels << getGlobalRawScriptLabels(raw_it.next().captured("raw_script"));
+
+    return poryScriptLabels;
+}
+
 QString &ParseUtil::removeStringLiterals(QString &text) {
     static const QRegularExpression re_string("\".*\"");
     return text.remove(re_string);