Merge commit '7621e2f8dec938cf48181c8b10afc9b01f444e68' into beta

third-party/libprisma/Sources/LanguageTree.h

@@ -0,0 +1,60 @@
+#pragma once
+
+#include <sstream>
+#include <fstream>
+#include <optional>
+
+#include "Highlight.h"
+
+struct Buffer {
+    const std::string &content;
+    int64_t offset = 0;
+};
+
+class LanguageTree : public std::enable_shared_from_this<LanguageTree>
+{
+public:
+    LanguageTree() = default;
+
+    void load(const std::string& content);
+
+    const Pattern* resolvePattern(size_t path);
+    const Grammar* resolveGrammar(size_t path);
+
+    std::map<std::string, std::string> keys() const
+    {
+        std::map<std::string, std::string> keys;
+
+        for (const auto& kv : m_languages)
+        {
+            if (kv.second.first.empty())
+            {
+                continue;
+            }
+
+            keys.emplace(kv.first, kv.second.first);
+        }
+
+        return keys;
+    }
+
+    const Grammar* find(const std::string& key) const
+    {
+        const auto& value = m_languages.find(key);
+        if (value != m_languages.end())
+        {
+            return m_grammars[value->second.second].get();
+        }
+
+        return nullptr;
+    }
+
+private:
+    void parseLanguages(Buffer &buffer);
+    void parseGrammars(Buffer &buffer);
+    void parsePatterns(Buffer &buffer);
+
+    std::map<std::string, std::pair<std::string, size_t>> m_languages;
+    std::vector<std::shared_ptr<Grammar>> m_grammars;
+    std::vector<std::shared_ptr<Pattern>> m_patterns;
+};