feat: introduce Qml to parser

add the Qml tree sitter language support to all places except test files.

Author: itzurabhi

CMakeLists.txt

@@ -135,6 +135,14 @@ add_custom_command(
           ${CMAKE_BINARY_DIR}/compile_commands.json compile_commands.json
   WORKING_DIRECTORY ${CMAKE_CURRENT_LIST_DIR}/test_data/projects/cpp-project)
 
+  add_custom_command(
+    OUTPUT
+      ${CMAKE_CURRENT_LIST_DIR}/test_data/projects/qml-project/compile_commands.json
+    COMMENT "Creating compile_commands.json file for qml-project"
+    COMMAND ${CMAKE_COMMAND} -E create_symlink
+            ${CMAKE_BINARY_DIR}/compile_commands.json compile_commands.json
+    WORKING_DIRECTORY ${CMAKE_CURRENT_LIST_DIR}/test_data/projects/qml-project)
+
 # Post-initialization
 # ##############################################################################
 # Need to be done *after* compiling 3rd parties

src/core/codedocument_p.cpp

@@ -35,8 +35,7 @@ void TreeSitterHelper::clear()
 treesitter::Parser &TreeSitterHelper::parser()
 {
     if (!m_parser) {
-        // TODO: Make language configurable
-        m_parser = treesitter::Parser(tree_sitter_cpp());
+        m_parser = treesitter::Parser(treesitter::Parser::getTSLanguageForDocumentType(m_document->type()));
     }
 
     // Regarding const-ness:

src/core/qmldocument.cpp

@@ -9,12 +9,14 @@
 */
 
 #include "qmldocument.h"
+#include "astnode.h"
 
 namespace Core {
 
 QmlDocument::QmlDocument(QObject *parent)
-    : TextDocument(Type::Qml, parent)
+    : CodeDocument(Type::Qml, parent)
 {
 }
 
+QmlDocument::~QmlDocument() = default;
 }

src/core/qmldocument.h

@@ -10,16 +10,17 @@
 
 #pragma once
 
-#include "textdocument.h"
+#include "codedocument.h"
 
 namespace Core {
 
-class QmlDocument : public TextDocument
+class QmlDocument : public CodeDocument
 {
     Q_OBJECT
 
 public:
     explicit QmlDocument(QObject *parent = nullptr);
+    ~QmlDocument() override;
 };
 
 }

src/gui/mainwindow.cpp

@@ -619,10 +619,11 @@ void MainWindow::updateActions()
     ui->actionExecuteAPI->setEnabled(textDocument != nullptr);
 
     auto *codeDocument = qobject_cast<Core::CodeDocument *>(document);
+    const bool qmlEnabled = codeDocument && qobject_cast<Core::QmlDocument *>(document);
     const bool lspEnabled = codeDocument && codeDocument->hasLspClient();
     ui->actionFollowSymbol->setEnabled(lspEnabled);
     ui->actionSwitchDeclDef->setEnabled(lspEnabled);
-    ui->actionTreeSitterInspector->setEnabled(lspEnabled);
+    ui->actionTreeSitterInspector->setEnabled(lspEnabled || qmlEnabled);
 
     const bool cppEnabled = codeDocument && qobject_cast<Core::CppDocument *>(document);
     ui->actionSwitchHeaderSource->setEnabled(cppEnabled);

src/gui/treesitterinspector.cpp

@@ -80,7 +80,7 @@ void QueryErrorHighlighter::setUtf8Position(int position)
 TreeSitterInspector::TreeSitterInspector(QWidget *parent)
     : QDialog(parent)
     , ui(new Ui::TreeSitterInspector)
-    , m_parser(tree_sitter_cpp())
+    , m_parser(nullptr)
     , m_errorHighlighter(nullptr)
     , m_document(nullptr)
 {
@@ -149,7 +149,8 @@ void TreeSitterInspector::changeQuery()
     }
 
     try {
-        auto query = std::make_shared<treesitter::Query>(tree_sitter_cpp(), ui->query->toPlainText());
+        auto lang = treesitter::Parser::getTSLanguageForDocumentType(m_document->type());
+        auto query = std::make_shared<treesitter::Query>(lang, ui->query->toPlainText());
         m_treemodel.setQuery(query, makePredicates());
         m_errorHighlighter->setUtf8Position(-1);
 
@@ -211,6 +212,7 @@ void TreeSitterInspector::setDocument(Core::CodeDocument *document)
     }
 
     m_document = document;
+    m_parser = treesitter::Parser::getTSLanguageForDocumentType(document->type());
     if (m_document) {
         connect(m_document, &Core::CodeDocument::textChanged, this, &TreeSitterInspector::changeText);
         connect(m_document, &Core::CodeDocument::positionChanged, this, &TreeSitterInspector::changeCursor);
@@ -281,8 +283,9 @@ void TreeSitterInspector::prepareTransformation(
     }
 
     try {
-        auto query = std::make_shared<treesitter::Query>(tree_sitter_cpp(), m_queryText);
-        treesitter::Parser parser(tree_sitter_cpp());
+        auto lang = treesitter::Parser::getTSLanguageForDocumentType(m_document->type());
+        auto query = std::make_shared<treesitter::Query>(lang, m_queryText);
+        treesitter::Parser parser(lang);
 
         treesitter::Transformation transformation(m_document->text(), std::move(parser), query,
                                                   ui->target->toPlainText());

src/treesitter/parser.cpp

@@ -10,6 +10,7 @@
 
 #include "parser.h"
 #include "tree.h"
+#include "treesitter/languages.h"
 
 #include <tree_sitter/api.h>
 #include <utility>
@@ -62,4 +63,18 @@ const TSLanguage *Parser::language() const
     return ts_parser_language(m_parser);
 }
 
+TSLanguage *Parser::getTSLanguageForDocumentType(Core::Document::Type type)
+{
+    switch (type) {
+    case Core::Document::Type::Qml:
+        return tree_sitter_qmljs();
+    // FIXME: make sane default for other documents.
+    case Core::Document::Type::Cpp:
+    default:
+        return tree_sitter_cpp();
+        break;
+    }
+
+    Q_UNREACHABLE();
+}
 }

src/treesitter/parser.h

@@ -10,6 +10,7 @@
 
 #pragma once
 
+#include "core/document.h"
 #include <QString>
 
 struct TSParser;
@@ -38,6 +39,8 @@ class Parser
 
     const TSLanguage *language() const;
 
+    static TSLanguage *getTSLanguageForDocumentType(Core::Document::Type type);
+
 private:
     TSParser *m_parser;
 };

test_data/projects/CMakeLists.txt

@@ -8,3 +8,4 @@
 #
 
 add_subdirectory(cpp-project)
+add_subdirectory(qml-project)