Files
qt-creator/tests/tools/qml-ast2dot/main.cpp
Christian Stenger f720835680 Tests: Fix build
Change-Id: I6d9475bf2d95a1075d7a761f49d8edff89a30f78
Reviewed-by: <github-actions-qt-creator@cristianadam.eu>
Reviewed-by: Eike Ziller <eike.ziller@qt.io>
2024-04-18 09:52:28 +00:00

340 lines
16 KiB
C++

// Copyright (C) 2016 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
#include <qmljs/parser/qmljsast_p.h>
#include <qmljs/parser/qmljsastvisitor_p.h>
#include <qmljs/parser/qmljsengine_p.h>
#include <qmljs/qmljsdocument.h>
#include <qmljs/qmljsmodelmanagerinterface.h>
#include <utils/filepath.h>
#include <QFile>
#include <QList>
#include <QCoreApplication>
#include <QStringList>
#include <QFileInfo>
#include <QTime>
#include <QDebug>
#include <cstdio>
#include <cstdlib>
#include <fstream>
#include <iostream>
#include <string>
#ifdef __GNUC__
# include <cxxabi.h>
#endif
using namespace QmlJS;
using namespace QmlJS::AST;
using namespace std;
class ASTDump: protected Visitor
{
public:
void operator()(const QString &fileName, const QByteArray &src, Node *ast) {
_src = src;
QString basename = fileName;
int dotIdx = basename.lastIndexOf(QLatin1Char('.'));
if (dotIdx != -1)
basename.truncate(dotIdx);
basename.append(QLatin1String(".ast.dot"));
out.open(basename.toUtf8().constData());
out << "digraph AST { ordering=out;" << endl;
// cout << "rankdir = \"LR\";" << endl;
Node::accept(ast, this);
typedef QPair<QByteArray, QByteArray> Pair;
for (const Pair &conn : std::as_const(_connections))
out << conn.first.constData() << " -> " << conn.second.constData() << endl;
alignTerminals();
out << "}" << endl;
out.close();
cout << qPrintable(basename) << endl;
}
protected:
void alignTerminals() {
out<<"{ rank=same;" << endl;
for (const QByteArray &terminalShape : std::as_const(_terminalShapes)) {
out << " " << string(terminalShape.constData(), terminalShape.size()).c_str() << ";" << endl;
}
out<<"}"<<endl;
}
static QByteArray name(Node *ast) {
#ifdef __GNUC__
QByteArray name = abi::__cxa_demangle(typeid(*ast).name(), 0, 0, 0) + 12;
#else
QByteArray name = typeid(*ast).name();
#endif
return name;
}
QString spell(const SourceLocation &token) {
return QString::fromLatin1(_src.mid(token.offset, token.length).replace('\'', "\\\\").replace('"', "\\\""));
}
void terminal(const SourceLocation &token) {
if (!token.isValid())
return;
static int count = 1;
QByteArray id = 't' + QByteArray::number(count++);
Node *node = _stack.last();
_connections.append(qMakePair(_id[node], id));
QByteArray t;
t.append(id);
t.append(" [label = \"");
t.append(spell(token).toUtf8());
t.append("\" shape=rect]");
_terminalShapes.append(t);
}
virtual void nonterminal(Node *ast) {
Node::accept(ast, this);
}
virtual void node(Node *ast) {
out << _id[ast].constData() << " [label=\"" << name(ast).constData() << "\"];" << endl;
}
virtual bool preVisit(Node *ast) {
static int count = 1;
const QByteArray id = 'n' + QByteArray::number(count++);
_id[ast] = id;
if (! _stack.isEmpty())
_connections.append(qMakePair(_id[_stack.last()], id));
_stack.append(ast);
node(ast);
return true;
}
virtual void postVisit(Node *) {
_stack.removeLast();
}
protected: // visiting functions:
virtual bool visit(UiImport *ast) {
terminal(ast->importToken);
if (ast->importUri)
nonterminal(ast->importUri);
else
terminal(ast->fileNameToken);
terminal(ast->version->firstSourceLocation());
terminal(ast->asToken);
terminal(ast->importIdToken);
terminal(ast->semicolonToken);
return false;
}
virtual bool visit(UiObjectBinding *ast) {
if (ast->hasOnToken) {
nonterminal(ast->qualifiedTypeNameId);
terminal(ast->colonToken);
nonterminal(ast->qualifiedId);
} else {
nonterminal(ast->qualifiedId);
terminal(ast->colonToken);
nonterminal(ast->qualifiedTypeNameId);
}
nonterminal(ast->initializer);
return false;
}
virtual bool visit(UiObjectDefinition *ast) {
nonterminal(ast->qualifiedTypeNameId);
nonterminal(ast->initializer);
return false;
}
virtual bool visit(UiObjectInitializer *ast) {
terminal(ast->lbraceToken);
nonterminal(ast->members);
terminal(ast->rbraceToken);
return false;
}
virtual bool visit(UiScriptBinding *ast) {
nonterminal(ast->qualifiedId);
terminal(ast->colonToken);
nonterminal(ast->statement);
return false;
}
virtual bool visit(UiArrayBinding *ast) {
nonterminal(ast->qualifiedId);
terminal(ast->colonToken);
terminal(ast->lbracketToken);
nonterminal(ast->members);
terminal(ast->rbracketToken);
return false;
}
virtual bool visit(UiArrayMemberList *ast) {
terminal(ast->commaToken);
nonterminal(ast->member);
nonterminal(ast->next);
return false;
}
virtual bool visit(UiQualifiedId *ast) {
terminal(ast->identifierToken);
nonterminal(ast->next);
return false;
}
virtual bool visit(UiPublicMember *ast) {
// TODO: place the parameters...
// UiParameterList *parameters;
terminal(ast->defaultToken());
terminal(ast->readonlyToken());
terminal(ast->propertyToken());
terminal(ast->typeModifierToken);
terminal(ast->typeToken);
terminal(ast->identifierToken);
terminal(ast->colonToken);
nonterminal(ast->statement);
nonterminal(ast->binding);
terminal(ast->semicolonToken);
return false;
}
virtual bool visit(StringLiteral *ast) { terminal(ast->literalToken); return false; }
virtual bool visit(NumericLiteral *ast) { terminal(ast->literalToken); return false; }
virtual bool visit(TrueLiteral *ast) { terminal(ast->trueToken); return false; }
virtual bool visit(FalseLiteral *ast) { terminal(ast->falseToken); return false; }
virtual bool visit(IdentifierExpression *ast) { terminal(ast->identifierToken); return false; }
virtual bool visit(FieldMemberExpression *ast) { nonterminal(ast->base); terminal(ast->dotToken); terminal(ast->identifierToken); return false; }
virtual bool visit(BinaryExpression *ast) { nonterminal(ast->left); terminal(ast->operatorToken); nonterminal(ast->right); return false; }
virtual bool visit(UnaryPlusExpression *ast) { terminal(ast->plusToken); nonterminal(ast->expression); return false; }
virtual bool visit(UnaryMinusExpression *ast) { terminal(ast->minusToken); nonterminal(ast->expression); return false; }
virtual bool visit(NestedExpression *ast) { terminal(ast->lparenToken); nonterminal(ast->expression); terminal(ast->rparenToken); return false; }
virtual bool visit(ThisExpression *ast) { terminal(ast->thisToken); return false; }
virtual bool visit(NullExpression *ast) { terminal(ast->nullToken); return false; }
virtual bool visit(RegExpLiteral *ast) { terminal(ast->literalToken); return false; }
virtual bool visit(ArrayPattern *ast) { terminal(ast->lbracketToken); nonterminal(ast->elements); terminal(ast->commaToken); terminal(ast->rbracketToken); return false; }
virtual bool visit(ObjectPattern *ast) { terminal(ast->lbraceToken); nonterminal(ast->properties); terminal(ast->rbraceToken); return false; }
virtual bool visit(PatternElementList *ast) { nonterminal(ast->next); nonterminal(ast->elision); nonterminal(ast->element->initializer); return false; }
virtual bool visit(Elision *ast) { nonterminal(ast->next); terminal(ast->commaToken); return false; }
virtual bool visit(PatternPropertyList *ast) { nonterminal(AST::cast<AST::PatternProperty *>(ast->property)); nonterminal(ast->next); return false; }
virtual bool visit(IdentifierPropertyName *ast) { terminal(ast->propertyNameToken); return false; }
virtual bool visit(StringLiteralPropertyName *ast) { terminal(ast->propertyNameToken); return false; }
virtual bool visit(NumericLiteralPropertyName *ast) { terminal(ast->propertyNameToken); return false; }
virtual bool visit(ArrayMemberExpression *ast) { nonterminal(ast->base); terminal(ast->lbracketToken); nonterminal(ast->expression); terminal(ast->rbracketToken); return false; }
virtual bool visit(NewMemberExpression *ast) { terminal(ast->newToken); nonterminal(ast->base); terminal(ast->lparenToken); nonterminal(ast->arguments); terminal(ast->rparenToken); return false; }
virtual bool visit(NewExpression *ast) { terminal(ast->newToken); nonterminal(ast->expression); return false; }
virtual bool visit(CallExpression *ast) { nonterminal(ast->base); terminal(ast->lparenToken); nonterminal(ast->arguments); terminal(ast->rparenToken); return false; }
virtual bool visit(ArgumentList *ast) { nonterminal(ast->expression); terminal(ast->commaToken); nonterminal(ast->next); return false; }
virtual bool visit(PostIncrementExpression *ast) { nonterminal(ast->base); terminal(ast->incrementToken); return false; }
virtual bool visit(PostDecrementExpression *ast) { nonterminal(ast->base); terminal(ast->decrementToken); return false; }
virtual bool visit(DeleteExpression *ast) { terminal(ast->deleteToken); nonterminal(ast->expression); return false; }
virtual bool visit(VoidExpression *ast) { terminal(ast->voidToken); nonterminal(ast->expression); return false; }
virtual bool visit(TypeOfExpression *ast) { terminal(ast->typeofToken); nonterminal(ast->expression); return false; }
virtual bool visit(PreIncrementExpression *ast) { terminal(ast->incrementToken); nonterminal(ast->expression); return false; }
virtual bool visit(PreDecrementExpression *ast) { terminal(ast->decrementToken); nonterminal(ast->expression); return false; }
virtual bool visit(TildeExpression *ast) { terminal(ast->tildeToken); nonterminal(ast->expression); return false; }
virtual bool visit(NotExpression *ast) { terminal(ast->notToken); nonterminal(ast->expression); return false; }
virtual bool visit(ConditionalExpression *ast) { nonterminal(ast->expression); terminal(ast->questionToken); nonterminal(ast->ok); terminal(ast->colonToken); nonterminal(ast->ko); return false; }
virtual bool visit(Expression *ast) { nonterminal(ast->left); terminal(ast->commaToken); nonterminal(ast->right); return false; }
virtual bool visit(Block *ast) { terminal(ast->lbraceToken); nonterminal(ast->statements); terminal(ast->rbraceToken); return false; }
virtual bool visit(VariableStatement *ast) { terminal(ast->declarationKindToken); nonterminal(ast->declarations); return false; }
//virtual bool visit(VariableDeclaration *ast) { terminal(ast->identifierToken); nonterminal(ast->expression); return false; }
virtual bool visit(VariableDeclarationList *ast) { nonterminal(ast->declaration); terminal(ast->commaToken); nonterminal(ast->next); return false; }
virtual bool visit(EmptyStatement* ast) { terminal(ast->semicolonToken); return false; }
virtual bool visit(ExpressionStatement *ast) { nonterminal(ast->expression); terminal(ast->semicolonToken); return false; }
virtual bool visit(IfStatement *ast) { terminal(ast->ifToken); terminal(ast->lparenToken); nonterminal(ast->expression); terminal(ast->rparenToken); nonterminal(ast->ok); terminal(ast->elseToken); nonterminal(ast->ko); return false; }
virtual bool visit(DoWhileStatement *ast) { terminal(ast->doToken); nonterminal(ast->statement); terminal(ast->whileToken); terminal(ast->lparenToken); nonterminal(ast->expression); terminal(ast->rparenToken); terminal(ast->semicolonToken); return false; }
virtual bool visit(WhileStatement *ast) { terminal(ast->whileToken); terminal(ast->lparenToken); nonterminal(ast->expression); terminal(ast->rparenToken); nonterminal(ast->statement); return false; }
virtual bool visit(ForStatement *ast) { terminal(ast->forToken); terminal(ast->lparenToken); nonterminal(ast->initialiser); terminal(ast->firstSemicolonToken); nonterminal(ast->condition); terminal(ast->secondSemicolonToken); nonterminal(ast->expression); terminal(ast->rparenToken); nonterminal(ast->statement); return false; }
virtual bool visit(ForEachStatement *ast) { terminal(ast->forToken); terminal(ast->lparenToken); nonterminal(ast->expression); terminal(ast->rparenToken); nonterminal(ast->statement); return false; }
virtual bool visit(ContinueStatement *ast) { terminal(ast->continueToken); return false; }
virtual bool visit(BreakStatement *ast) { terminal(ast->breakToken); return false; }
virtual bool visit(ReturnStatement *ast) { terminal(ast->returnToken); nonterminal(ast->expression); return false; }
virtual bool visit(WithStatement *ast) { terminal(ast->withToken); terminal(ast->lparenToken); nonterminal(ast->expression); terminal(ast->rparenToken); nonterminal(ast->statement); return false; }
virtual bool visit(CaseBlock *ast) { terminal(ast->lbraceToken); nonterminal(ast->clauses); nonterminal(ast->defaultClause); nonterminal(ast->moreClauses); terminal(ast->rbraceToken); return false; }
virtual bool visit(SwitchStatement *ast) { terminal(ast->switchToken); terminal(ast->lparenToken); nonterminal(ast->expression); terminal(ast->rparenToken); nonterminal(ast->block); return false; }
virtual bool visit(CaseClause *ast) { terminal(ast->caseToken); nonterminal(ast->expression); terminal(ast->colonToken); nonterminal(ast->statements); return false; }
virtual bool visit(DefaultClause *ast) { terminal(ast->defaultToken); terminal(ast->colonToken); nonterminal(ast->statements); return false; }
virtual bool visit(LabelledStatement *ast) { terminal(ast->identifierToken); terminal(ast->colonToken); nonterminal(ast->statement); return false; }
virtual bool visit(ThrowStatement *ast) { terminal(ast->throwToken); nonterminal(ast->expression); return false; }
virtual bool visit(Catch *ast) { terminal(ast->catchToken); terminal(ast->lparenToken); terminal(ast->identifierToken); terminal(ast->rparenToken); nonterminal(ast->statement); return false; }
virtual bool visit(Finally *ast) { terminal(ast->finallyToken); nonterminal(ast->statement); return false; }
virtual bool visit(FunctionExpression *ast) { terminal(ast->functionToken); terminal(ast->identifierToken); terminal(ast->lparenToken); nonterminal(ast->formals); terminal(ast->rparenToken); terminal(ast->lbraceToken); nonterminal(ast->body); terminal(ast->rbraceToken); return false; }
virtual bool visit(FunctionDeclaration *ast) { return visit(static_cast<FunctionExpression *>(ast)); }
virtual bool visit(DebuggerStatement *ast) { terminal(ast->debuggerToken); terminal(ast->semicolonToken); return false; }
virtual bool visit(UiParameterList *ast) { terminal(ast->commaToken); terminal(ast->identifierToken); nonterminal(ast->next); return false; }
void throwRecursionDepthError() override {
qWarning() << "Reached maximum recursion depth.";
}
private:
QHash<Node *, QByteArray> _id;
QList<QPair<QByteArray, QByteArray> > _connections;
QList<Node *> _stack;
QList<QByteArray> _terminalShapes;
ofstream out;
QByteArray _src;
};
int main(int argc, char *argv[])
{
QCoreApplication app(argc, argv);
QStringList files = app.arguments();
files.removeFirst();
for (const QString &fileName : std::as_const(files)) {
QFile file(fileName);
if (! file.open(QFile::ReadOnly)) {
cerr << "Cannot open \"" << qPrintable(fileName)
<< "\", skipping it." << endl;
continue;
}
const QByteArray source = file.readAll();
file.close();
const Utils::FilePath filePath = Utils::FilePath::fromUserInput(fileName);
Document::MutablePtr doc = Document::create(filePath, ModelManagerInterface::guessLanguageOfFile(filePath));
doc->setSource(QString::fromUtf8(source));
doc->parse();
const QList<DiagnosticMessage> messages = doc->diagnosticMessages();
for (const DiagnosticMessage &m : messages) {
ostream *os;
if (m.isError()) {
os = &cerr;
*os << "Error:";
} else {
os = &cout;
*os << "Warning:";
}
if (m.loc.isValid())
*os << m.loc.startLine << ':' << m.loc.startColumn << ':';
*os << ' ';
*os << qPrintable(m.message) << endl;
}
ASTDump dump;
dump(fileName, source, doc->ast());
}
return EXIT_SUCCESS;
}