1 files changed, 157 insertions, 0 deletions
diff --git a/src/declarative/qml/qmlrewrite.cpp b/src/declarative/qml/qmlrewrite.cpp
new file mode 100644
index 0000000..02bf8fa
--- /dev/null
+++ b/src/declarative/qml/qmlrewrite.cpp
@@ -0,0 +1,157 @@
+/****************************************************************************
+**
+** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
+** Contact: Qt Software Information (qt-info@nokia.com)
+**
+** This file is part of the QtDeclarative module of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:LGPL$
+** No Commercial Usage
+** This file contains pre-release code and may not be distributed.
+** You may use this file in accordance with the terms and conditions
+** contained in the either Technology Preview License Agreement or the
+** Beta Release License Agreement.
+**
+** 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 as published by the Free Software
+** Foundation and appearing in the file LICENSE.LGPL included in the
+** packaging of this file.  Please review the following information to
+** ensure the GNU Lesser General Public License version 2.1 requirements
+** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
+**
+** In addition, as a special exception, Nokia gives you certain
+** additional rights. These rights are described in the Nokia Qt LGPL
+** Exception version 1.0, 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.
+**
+** If you are unsure which license is appropriate for your use, please
+** contact the sales department at qt-sales@nokia.com.
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#include "qmlrewrite_p.h"
+
+QT_BEGIN_NAMESPACE
+
+namespace QmlRewrite {
+
+QString RewriteBinding::operator()(const QString &code)
+{
+    Engine engine;
+    NodePool pool(QString(), &engine);
+    Lexer lexer(&engine);
+    Parser parser(&engine);
+    lexer.setCode(code, 0);
+    parser.parseStatement();
+    return rewrite(code, 0, parser.statement());
+}
+
+void RewriteBinding::accept(AST::Node *node)
+{
+    AST::Node::acceptChild(node, this);
+}
+
+QString RewriteBinding::rewrite(QString code, unsigned position, 
+                                AST::Statement *node)
+{
+    TextWriter w;
+    _writer = &w;
+    _position = position;
+
+    accept(node);
+
+    unsigned startOfStatement = node->firstSourceLocation().begin() - _position;
+    unsigned endOfStatement = node->lastSourceLocation().end() - _position;
+
+    _writer->replace(startOfStatement, 0, QLatin1String("function() {\n"));
+    _writer->replace(endOfStatement, 0, QLatin1String("\n}"));
+
+    w.write(&code);
+
+    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)
+{
+    unsigned startOfExpressionStatement = ast->firstSourceLocation().begin() - _position;
+    _writer->replace(startOfExpressionStatement, 0, QLatin1String("return "));
+
+    return false;
+}
+
+bool RewriteBinding::visit(AST::NumericLiteral *node)
+{
+    if (node->suffix != AST::NumericLiteral::noSuffix) {
+        const int suffixLength = AST::NumericLiteral::suffixLength[node->suffix];
+        const char *suffixSpell = AST::NumericLiteral::suffixSpell[node->suffix];
+        QString pre;
+        pre += QLatin1String("qmlNumberFrom");
+        pre += QChar(QLatin1Char(suffixSpell[0])).toUpper();
+        pre += QLatin1String(&suffixSpell[1]);
+        pre += QLatin1Char('(');
+        _writer->replace(node->literalToken.begin() - _position, 0, pre);
+        _writer->replace(node->literalToken.end() - _position - suffixLength,
+                         suffixLength,
+                         QLatin1String(")"));
+    }
+
+    return false;
+}
+
+QString RewriteNumericLiterals::operator()(QString code, unsigned position, AST::Node *node)
+{
+    TextWriter w;
+    _writer = &w;
+    _position = position;
+
+    AST::Node::acceptChild(node, this);
+
+    w.write(&code);
+
+    return code;
+}
+
+bool RewriteNumericLiterals::visit(AST::NumericLiteral *node)
+{
+    if (node->suffix != AST::NumericLiteral::noSuffix) {
+        const int suffixLength = AST::NumericLiteral::suffixLength[node->suffix];
+        const char *suffixSpell = AST::NumericLiteral::suffixSpell[node->suffix];
+        QString pre;
+        pre += QLatin1String("qmlNumberFrom");
+        pre += QChar(QLatin1Char(suffixSpell[0])).toUpper();
+        pre += QLatin1String(&suffixSpell[1]);
+        pre += QLatin1Char('(');
+        _writer->replace(node->literalToken.begin() - _position, 0, pre);
+        _writer->replace(node->literalToken.end() - _position - suffixLength,
+                         suffixLength,
+                         QLatin1String(")"));
+    }
+
+    return false;
+}
+
+} // namespace QmlRewrite
+
+QT_END_NAMESPACE