1 files changed, 108 insertions, 0 deletions
diff --git a/src/qmlcompiler/qqmljslintercodegen.cpp b/src/qmlcompiler/qqmljslintercodegen.cpp
new file mode 100644
index 0000000000..175f7ee1e0
--- /dev/null
+++ b/src/qmlcompiler/qqmljslintercodegen.cpp
@@ -0,0 +1,108 @@
+// Copyright (C) 2021 The Qt Company Ltd.
+// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
+
+#include "qqmljslintercodegen_p.h"
+
+#include <QtQmlCompiler/private/qqmljsimportvisitor_p.h>
+#include <QtQmlCompiler/private/qqmljsshadowcheck_p.h>
+#include <QtQmlCompiler/private/qqmljsstoragegeneralizer_p.h>
+#include <QtQmlCompiler/private/qqmljstypepropagator_p.h>
+#include <QtQmlCompiler/private/qqmljsfunctioninitializer_p.h>
+
+#include <QFileInfo>
+
+QT_BEGIN_NAMESPACE
+
+using namespace Qt::StringLiterals;
+
+QQmlJSLinterCodegen::QQmlJSLinterCodegen(QQmlJSImporter *importer, const QString &fileName,
+                                         const QStringList &qmldirFiles, QQmlJSLogger *logger)
+    : QQmlJSAotCompiler(importer, fileName, qmldirFiles, logger)
+{
+}
+
+void QQmlJSLinterCodegen::setDocument(const QmlIR::JSCodeGen *codegen,
+                                      const QmlIR::Document *document)
+{
+    Q_UNUSED(codegen);
+    m_document = document;
+    m_unitGenerator = &document->jsGenerator;
+}
+
+std::variant<QQmlJSAotFunction, QQmlJS::DiagnosticMessage>
+QQmlJSLinterCodegen::compileBinding(const QV4::Compiler::Context *context,
+                                    const QmlIR::Binding &irBinding, QQmlJS::AST::Node *astNode)
+{
+    QQmlJSFunctionInitializer initializer(
+                &m_typeResolver, m_currentObject->location, m_currentScope->location);
+
+    QQmlJS::DiagnosticMessage initializationError;
+    const QString name = m_document->stringAt(irBinding.propertyNameIndex);
+    QQmlJSCompilePass::Function function =
+            initializer.run(context, name, astNode, irBinding, &initializationError);
+    if (initializationError.isValid())
+        diagnose(initializationError.message, initializationError.type, initializationError.loc);
+
+    QQmlJS::DiagnosticMessage analyzeError;
+    if (!analyzeFunction(context, &function, &analyzeError)) {
+        // If it's a signal and the function just returns a closure, it's harmless.
+        // Otherwise promote the message to warning level.
+        return diagnose(u"Could not compile binding for %1: %2"_s.arg(name, analyzeError.message),
+                        (function.isSignalHandler && analyzeError.type == QtDebugMsg)
+                            ? QtDebugMsg
+                            : QtWarningMsg,
+                        analyzeError.loc);
+    }
+
+    return QQmlJSAotFunction {};
+}
+
+std::variant<QQmlJSAotFunction, QQmlJS::DiagnosticMessage>
+QQmlJSLinterCodegen::compileFunction(const QV4::Compiler::Context *context,
+                                     const QString &name, QQmlJS::AST::Node *astNode)
+{
+    QQmlJS::DiagnosticMessage initializationError;
+    QQmlJSFunctionInitializer initializer(
+                &m_typeResolver, m_currentObject->location, m_currentScope->location);
+    QQmlJSCompilePass::Function function =
+            initializer.run(context, name, astNode, &initializationError);
+    if (initializationError.isValid())
+        diagnose(initializationError.message, initializationError.type, initializationError.loc);
+
+    QQmlJS::DiagnosticMessage analyzeError;
+    if (!analyzeFunction(context, &function, &analyzeError)) {
+        return diagnose(u"Could not compile function %1: %2"_s.arg(name, analyzeError.message),
+                        QtWarningMsg, analyzeError.loc);
+    }
+
+    return QQmlJSAotFunction {};
+}
+
+bool QQmlJSLinterCodegen::analyzeFunction(const QV4::Compiler::Context *context,
+                                          QQmlJSCompilePass::Function *function,
+                                          QQmlJS::DiagnosticMessage *error)
+{
+    QQmlJSTypePropagator propagator(m_unitGenerator, &m_typeResolver, m_logger,
+                                    {}, {}, m_passManager);
+    auto [basicBlocks, annotations] = propagator.run(function, error);
+    if (!error->isValid()) {
+        QQmlJSShadowCheck shadowCheck(m_unitGenerator, &m_typeResolver, m_logger, basicBlocks,
+                                      annotations);
+        shadowCheck.run(function, error);
+    }
+
+    if (!error->isValid()) {
+        QQmlJSStorageGeneralizer generalizer(m_unitGenerator, &m_typeResolver, m_logger,
+                                             basicBlocks, annotations);
+        generalizer.run(function, error);
+    }
+
+    if (error->isValid()) {
+        error->type = context->returnsClosure ? QtDebugMsg : QtWarningMsg;
+        return false;
+    }
+
+    return true;
+}
+
+QT_END_NAMESPACE