mirror of
https://bitbucket.org/smil3y/katie.git
synced 2025-02-27 04:13:08 +00:00
273 lines
7.4 KiB
C++
273 lines
7.4 KiB
C++
/****************************************************************************
|
|
**
|
|
** Copyright (C) 2015 The Qt Company Ltd.
|
|
** Contact: http://www.qt.io/licensing/
|
|
**
|
|
** This file is part of the QtDeclarative module of the Qt Toolkit.
|
|
**
|
|
** $QT_BEGIN_LICENSE:LGPL$
|
|
** Commercial License Usage
|
|
** Licensees holding valid commercial Qt licenses may use this file in
|
|
** accordance with the commercial license agreement provided with the
|
|
** Software or, alternatively, in accordance with the terms contained in
|
|
** a written agreement between you and The Qt Company. For licensing terms
|
|
** and conditions see http://www.qt.io/terms-conditions. For further
|
|
** information use the contact form at http://www.qt.io/contact-us.
|
|
**
|
|
** GNU Lesser General Public License Usage
|
|
** Alternatively, this file may be used under the terms of the GNU Lesser
|
|
** General Public License version 2.1 or version 3 as published by the Free
|
|
** Software Foundation and appearing in the file LICENSE.LGPLv21 and
|
|
** LICENSE.LGPLv3 included in the packaging of this file. Please review the
|
|
** following information to ensure the GNU Lesser General Public License
|
|
** requirements will be met: https://www.gnu.org/licenses/lgpl.html and
|
|
** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
|
|
**
|
|
** As a special exception, The Qt Company gives you certain additional
|
|
** rights. These rights are described in The Qt Company LGPL Exception
|
|
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
|
|
**
|
|
** GNU General Public License Usage
|
|
** Alternatively, this file may be used under the terms of the GNU
|
|
** General Public License version 3.0 as published by the Free Software
|
|
** Foundation and appearing in the file LICENSE.GPL included in the
|
|
** packaging of this file. Please review the following information to
|
|
** ensure the GNU General Public License version 3.0 requirements will be
|
|
** met: http://www.gnu.org/copyleft/gpl.html.
|
|
**
|
|
** $QT_END_LICENSE$
|
|
**
|
|
****************************************************************************/
|
|
|
|
#include "qdeclarativerewrite_p.h"
|
|
|
|
#include "qdeclarativeglobal_p.h"
|
|
|
|
#include <QtCore/qdebug.h>
|
|
|
|
QT_BEGIN_NAMESPACE
|
|
|
|
DEFINE_BOOL_CONFIG_OPTION(rewriteDump, QML_REWRITE_DUMP)
|
|
|
|
namespace QDeclarativeRewrite {
|
|
|
|
bool SharedBindingTester::isSharable(const QString &code)
|
|
{
|
|
Engine engine;
|
|
NodePool pool(QString(), &engine);
|
|
Lexer lexer(&engine);
|
|
Parser parser(&engine);
|
|
lexer.setCode(code, 0);
|
|
parser.parseStatement();
|
|
if (!parser.statement())
|
|
return false;
|
|
|
|
return isSharable(parser.statement());
|
|
}
|
|
|
|
bool SharedBindingTester::isSharable(AST::Node *node)
|
|
{
|
|
_sharable = true;
|
|
AST::Node::accept(node, this);
|
|
return _sharable;
|
|
}
|
|
|
|
QString RewriteBinding::operator()(const QString &code, bool *ok, bool *sharable)
|
|
{
|
|
Engine engine;
|
|
NodePool pool(QString(), &engine);
|
|
Lexer lexer(&engine);
|
|
Parser parser(&engine);
|
|
lexer.setCode(code, 0);
|
|
parser.parseStatement();
|
|
if (!parser.statement()) {
|
|
if (ok) *ok = false;
|
|
return QString();
|
|
} else {
|
|
if (ok) *ok = true;
|
|
if (sharable) {
|
|
SharedBindingTester tester;
|
|
*sharable = tester.isSharable(parser.statement());
|
|
}
|
|
}
|
|
return rewrite(code, 0, parser.statement());
|
|
}
|
|
|
|
QString RewriteBinding::operator()(QDeclarativeJS::AST::Node *node, const QString &code, bool *sharable)
|
|
{
|
|
if (!node)
|
|
return code;
|
|
|
|
if (sharable) {
|
|
SharedBindingTester tester;
|
|
*sharable = tester.isSharable(node);
|
|
}
|
|
|
|
QDeclarativeJS::AST::ExpressionNode *expression = node->expressionCast();
|
|
QDeclarativeJS::AST::Statement *statement = node->statementCast();
|
|
if(!expression && !statement)
|
|
return code;
|
|
|
|
TextWriter w;
|
|
_writer = &w;
|
|
_position = expression ? expression->firstSourceLocation().begin() : statement->firstSourceLocation().begin();
|
|
_inLoop = 0;
|
|
|
|
accept(node);
|
|
|
|
unsigned startOfStatement = 0;
|
|
unsigned endOfStatement = (expression ? expression->lastSourceLocation().end() : statement->lastSourceLocation().end()) - _position;
|
|
|
|
QString startString = QLatin1String("(function ") + QString::fromUtf8(_name) + QLatin1String("() { ");
|
|
if (expression)
|
|
startString += QLatin1String("return ");
|
|
_writer->replace(startOfStatement, 0, startString);
|
|
_writer->replace(endOfStatement, 0, QLatin1String(" })"));
|
|
|
|
if (rewriteDump()) {
|
|
qWarning() << "=============================================================";
|
|
qWarning() << "Rewrote:";
|
|
qWarning() << qPrintable(code);
|
|
}
|
|
|
|
QString codeCopy = code;
|
|
w.write(&codeCopy);
|
|
|
|
if (rewriteDump()) {
|
|
qWarning() << "To:";
|
|
qWarning() << qPrintable(code);
|
|
qWarning() << "=============================================================";
|
|
}
|
|
|
|
return codeCopy;
|
|
}
|
|
|
|
void RewriteBinding::accept(AST::Node *node)
|
|
{
|
|
AST::Node::accept(node, this);
|
|
}
|
|
|
|
QString RewriteBinding::rewrite(QString code, unsigned position,
|
|
AST::Statement *node)
|
|
{
|
|
TextWriter w;
|
|
_writer = &w;
|
|
_position = position;
|
|
_inLoop = 0;
|
|
|
|
accept(node);
|
|
|
|
unsigned startOfStatement = node->firstSourceLocation().begin() - _position;
|
|
unsigned endOfStatement = node->lastSourceLocation().end() - _position;
|
|
|
|
_writer->replace(startOfStatement, 0, QLatin1String("(function ") + QString::fromUtf8(_name) + QLatin1String("() { "));
|
|
_writer->replace(endOfStatement, 0, QLatin1String(" })"));
|
|
|
|
if (rewriteDump()) {
|
|
qWarning() << "=============================================================";
|
|
qWarning() << "Rewrote:";
|
|
qWarning() << qPrintable(code);
|
|
}
|
|
|
|
w.write(&code);
|
|
|
|
if (rewriteDump()) {
|
|
qWarning() << "To:";
|
|
qWarning() << qPrintable(code);
|
|
qWarning() << "=============================================================";
|
|
}
|
|
|
|
return code;
|
|
}
|
|
|
|
bool RewriteBinding::visit(AST::Block *ast)
|
|
{
|
|
for (AST::StatementList *it = ast->statements; it; it = it->next) {
|
|
if (! it->next) {
|
|
// we need to rewrite only the last statement of a block.
|
|
accept(it->statement);
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool RewriteBinding::visit(AST::ExpressionStatement *ast)
|
|
{
|
|
if (! _inLoop) {
|
|
unsigned startOfExpressionStatement = ast->firstSourceLocation().begin() - _position;
|
|
_writer->replace(startOfExpressionStatement, 0, QLatin1String("return "));
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool RewriteBinding::visit(AST::DoWhileStatement *)
|
|
{
|
|
++_inLoop;
|
|
return true;
|
|
}
|
|
|
|
void RewriteBinding::endVisit(AST::DoWhileStatement *)
|
|
{
|
|
--_inLoop;
|
|
}
|
|
|
|
bool RewriteBinding::visit(AST::WhileStatement *)
|
|
{
|
|
++_inLoop;
|
|
return true;
|
|
}
|
|
|
|
void RewriteBinding::endVisit(AST::WhileStatement *)
|
|
{
|
|
--_inLoop;
|
|
}
|
|
|
|
bool RewriteBinding::visit(AST::ForStatement *)
|
|
{
|
|
++_inLoop;
|
|
return true;
|
|
}
|
|
|
|
void RewriteBinding::endVisit(AST::ForStatement *)
|
|
{
|
|
--_inLoop;
|
|
}
|
|
|
|
bool RewriteBinding::visit(AST::LocalForStatement *)
|
|
{
|
|
++_inLoop;
|
|
return true;
|
|
}
|
|
|
|
void RewriteBinding::endVisit(AST::LocalForStatement *)
|
|
{
|
|
--_inLoop;
|
|
}
|
|
|
|
bool RewriteBinding::visit(AST::ForEachStatement *)
|
|
{
|
|
++_inLoop;
|
|
return true;
|
|
}
|
|
|
|
void RewriteBinding::endVisit(AST::ForEachStatement *)
|
|
{
|
|
--_inLoop;
|
|
}
|
|
|
|
bool RewriteBinding::visit(AST::LocalForEachStatement *)
|
|
{
|
|
++_inLoop;
|
|
return true;
|
|
}
|
|
|
|
void RewriteBinding::endVisit(AST::LocalForEachStatement *)
|
|
{
|
|
--_inLoop;
|
|
}
|
|
|
|
} // namespace QDeclarativeRewrite
|
|
|
|
QT_END_NAMESPACE
|