1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
|
// Copyright (C) 2024 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
#include <QCommandLineParser>
#include <QCoreApplication>
#include <QDir>
#include <QFile>
#include <QFileInfo>
#include <QJsonArray>
#include <QJsonDocument>
#include <QJsonObject>
#include <private/qqmljscompilerstats_p.h>
#include <private/qqmljscompilerstatsreporter_p.h>
using namespace Qt::Literals::StringLiterals;
bool saveFormattedStats(const QString &stats, const QString &outputPath)
{
QString directory = QFileInfo(outputPath).dir().path();
if (!QDir().mkpath(directory)) {
qDebug() << "Could not ensure the existence of" << directory;
return false;
}
QFile outputFile(outputPath);
if (!outputFile.open(QIODevice::Text | QIODevice::WriteOnly)) {
qDebug() << "Could not open file" << outputPath;
return false;
}
if (outputFile.write(stats.toLatin1()) == -1) {
qDebug() << "Could not write formatted AOT stats to" << outputPath;
return false;
} else {
qDebug() << "Formatted AOT stats saved to" << outputPath;
}
return true;
}
int main(int argc, char **argv)
{
QCoreApplication app(argc, argv);
QCoreApplication::setApplicationVersion(QLatin1String(QT_VERSION_STR));
QCommandLineParser parser;
parser.addHelpOption();
parser.setApplicationDescription("Internal development tool.");
parser.addPositionalArgument("mode", "Choose whether to aggregate or display aotstats files",
"[aggregate|format]");
parser.addPositionalArgument("input", "Aggregate mode: the aotstatslist file to aggregate. "
"Format mode: the aotstats file to display.");
parser.addPositionalArgument("output", "Aggregate mode: the path where to store the "
"aggregated aotstats. Format mode: the the path where "
"the formatted output will be saved.");
QCommandLineOption emptyModulesOption("empty-modules", QCoreApplication::translate("main", "Format mode: File containing a list of modules with no QML files."), "file");
parser.addOption(emptyModulesOption);
QCommandLineOption onlyBytecodeModulesOption("only-bytecode-modules", QCoreApplication::translate("main", "Format mode: File containing a list of modules for which only the bytecode is generated."), "file");
parser.addOption(onlyBytecodeModulesOption);
parser.process(app);
const auto &positionalArgs = parser.positionalArguments();
if (positionalArgs.size() != 3) {
qDebug().noquote() << parser.helpText();
return EXIT_FAILURE;
}
const auto &mode = positionalArgs.first();
if (mode == u"aggregate"_s) {
const auto aggregated = QQmlJS::AotStats::aggregateAotstatsList(positionalArgs[1]);
if (!aggregated.has_value())
return EXIT_FAILURE;
if (!aggregated->saveToDisk(positionalArgs[2]))
return EXIT_FAILURE;
} else if (mode == u"format"_s) {
const auto aotstats = QQmlJS::AotStats::parseAotstatsFile(positionalArgs[1]);
if (!aotstats.has_value())
return EXIT_FAILURE;
const std::optional<QStringList> emptyModules = parser.isSet(emptyModulesOption)
? QQmlJS::AotStats::readAllLines(parser.value(emptyModulesOption))
: QStringList();
const std::optional<QStringList> onlyBytecodeModules = parser.isSet(onlyBytecodeModulesOption)
? QQmlJS::AotStats::readAllLines(parser.value(onlyBytecodeModulesOption))
: QStringList();
if (!emptyModules || !onlyBytecodeModules)
return EXIT_FAILURE;
const QQmlJS::AotStatsReporter reporter(aotstats.value(), emptyModules.value(),
onlyBytecodeModules.value());
if (!saveFormattedStats(reporter.format(), positionalArgs[2]))
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}
|