cppquickfixes.cpp 72 KB
Newer Older
hjk's avatar
hjk committed
1
/****************************************************************************
2
**
3
** Copyright (C) 2013 Digia Plc and/or its subsidiary(-ies).
hjk's avatar
hjk committed
4
** Contact: http://www.qt-project.org/legal
5
**
hjk's avatar
hjk committed
6
** This file is part of Qt Creator.
7
**
hjk's avatar
hjk committed
8 9 10 11 12 13 14
** 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 Digia.  For licensing terms and
** conditions see http://qt.digia.com/licensing.  For further information
** use the contact form at http://qt.digia.com/contact-us.
15 16
**
** GNU Lesser General Public License Usage
hjk's avatar
hjk committed
17 18 19 20 21 22 23 24 25
** 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, Digia gives you certain additional
** rights.  These rights are described in the Digia Qt LGPL Exception
con's avatar
con committed
26 27
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
**
hjk's avatar
hjk committed
28
****************************************************************************/
29

Nikolai Kosjar's avatar
Nikolai Kosjar committed
30 31
#include "cppquickfixes.h"

32
#include "cppcompleteswitch.h"
33
#include "cppeditor.h"
34
#include "cppinsertdecldef.h"
35
#include "cppinsertqtpropertymembers.h"
Leandro Melo's avatar
Leandro Melo committed
36
#include "cppquickfixassistant.h"
37 38 39 40 41 42 43 44 45

#include <Literals.h>
#include <Name.h>
#include <Names.h>
#include <Symbol.h>
#include <Symbols.h>
#include <Token.h>
#include <TranslationUnit.h>

46
#include <cplusplus/CppRewriter.h>
47 48 49
#include <cplusplus/DependencyTable.h>
#include <cplusplus/Overview.h>
#include <cplusplus/TypeOfExpression.h>
50
#include <cpptools/cppclassesfilter.h>
51
#include <cpptools/cppcodestylesettings.h>
52 53 54
#include <cpptools/cpppointerdeclarationformatter.h>
#include <cpptools/cpptoolsconstants.h>
#include <cpptools/ModelManagerInterface.h>
55
#include <cpptools/symbolfinder.h>
56
#include <extensionsystem/pluginmanager.h>
Nikolai Kosjar's avatar
Nikolai Kosjar committed
57
#include <utils/changeset.h>
58 59
#include <utils/qtcassert.h>

60
#include <QApplication>
61
#include <QFileInfo>
Nikolai Kosjar's avatar
Nikolai Kosjar committed
62
#include <QSharedPointer>
63 64
#include <QTextBlock>
#include <QTextCursor>
65

66 67
using namespace CppEditor;
using namespace CppEditor::Internal;
Nikolai Kosjar's avatar
Nikolai Kosjar committed
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 99 100 101 102 103 104

void CppEditor::Internal::registerQuickFixes(ExtensionSystem::IPlugin *plugIn)
{
    plugIn->addAutoReleasedObject(new AddIncludeForUndefinedIdentifier);
    plugIn->addAutoReleasedObject(new AddIncludeForForwardDeclaration);

    plugIn->addAutoReleasedObject(new FlipLogicalOperands);
    plugIn->addAutoReleasedObject(new InverseLogicalComparison);
    plugIn->addAutoReleasedObject(new RewriteLogicalAnd);

    plugIn->addAutoReleasedObject(new ConvertToCamelCase);

    plugIn->addAutoReleasedObject(new ConvertCStringToNSString);
    plugIn->addAutoReleasedObject(new ConvertNumericLiteral);
    plugIn->addAutoReleasedObject(new TranslateStringLiteral);
    plugIn->addAutoReleasedObject(new WrapStringLiteral);

    plugIn->addAutoReleasedObject(new MoveDeclarationOutOfIf);
    plugIn->addAutoReleasedObject(new MoveDeclarationOutOfWhile);

    plugIn->addAutoReleasedObject(new SplitIfStatement);
    plugIn->addAutoReleasedObject(new SplitSimpleDeclaration);

    plugIn->addAutoReleasedObject(new AddLocalDeclaration);
    plugIn->addAutoReleasedObject(new AddBracesToIf);
    plugIn->addAutoReleasedObject(new RearrangeParamDeclarationList);
    plugIn->addAutoReleasedObject(new ReformatPointerDeclaration);

    plugIn->addAutoReleasedObject(new CompleteSwitchCaseStatement);
    plugIn->addAutoReleasedObject(new InsertQtPropertyMembers);

    plugIn->addAutoReleasedObject(new ApplyDeclDefLinkChanges);
    plugIn->addAutoReleasedObject(new ExtractFunction);
    plugIn->addAutoReleasedObject(new GenerateGetterSetter);
    plugIn->addAutoReleasedObject(new InsertDeclFromDef);
    plugIn->addAutoReleasedObject(new InsertDefFromDecl);
}
105

106 107 108 109 110
static inline bool isQtStringLiteral(const QByteArray &id)
{
    return id == "QLatin1String" || id == "QLatin1Literal" || id == "QStringLiteral";
}

111 112 113 114 115
static inline bool isQtStringTranslation(const QByteArray &id)
{
    return id == "tr" || id == "trUtf8" || id == "translate" || id == "QT_TRANSLATE_NOOP";
}

Nikolai Kosjar's avatar
Nikolai Kosjar committed
116
class InverseLogicalComparisonOp: public CppQuickFixOperation
117 118
{
public:
Nikolai Kosjar's avatar
Nikolai Kosjar committed
119 120 121 122
    InverseLogicalComparisonOp(const CppQuickFixInterface &interface, int priority,
                               BinaryExpressionAST *binary, Kind invertToken)
        : CppQuickFixOperation(interface, priority)
        , binary(binary), nested(0), negation(0)
123
    {
Nikolai Kosjar's avatar
Nikolai Kosjar committed
124 125 126
        Token tok;
        tok.f.kind = invertToken;
        replacement = QLatin1String(tok.spell());
127

Nikolai Kosjar's avatar
Nikolai Kosjar committed
128 129 130
        // check for enclosing nested expression
        if (priority - 1 >= 0)
            nested = interface->path()[priority - 1]->asNestedExpression();
131

Nikolai Kosjar's avatar
Nikolai Kosjar committed
132 133 134 135 136
        // check for ! before parentheses
        if (nested && priority - 2 >= 0) {
            negation = interface->path()[priority - 2]->asUnaryExpression();
            if (negation && ! interface->currentFile()->tokenAt(negation->unary_op_token).is(T_EXCLAIM))
                negation = 0;
137 138 139
        }
    }

Nikolai Kosjar's avatar
Nikolai Kosjar committed
140
    QString description() const
141
    {
Nikolai Kosjar's avatar
Nikolai Kosjar committed
142 143
        return QApplication::translate("CppTools::QuickFix", "Rewrite Using %1").arg(replacement);
    }
144

Nikolai Kosjar's avatar
Nikolai Kosjar committed
145 146 147 148 149 150 151 152 153 154 155 156 157 158
    void perform()
    {
        CppRefactoringChanges refactoring(snapshot());
        CppRefactoringFilePtr currentFile = refactoring.file(fileName());

        ChangeSet changes;
        if (negation) {
            // can't remove parentheses since that might break precedence
            changes.remove(currentFile->range(negation->unary_op_token));
        } else if (nested) {
            changes.insert(currentFile->startOf(nested), QLatin1String("!"));
        } else {
            changes.insert(currentFile->startOf(binary), QLatin1String("!("));
            changes.insert(currentFile->endOf(binary), QLatin1String(")"));
159
        }
Nikolai Kosjar's avatar
Nikolai Kosjar committed
160 161 162 163
        changes.replace(currentFile->range(binary->binary_op_token), replacement);
        currentFile->setChangeSet(changes);
        currentFile->apply();
    }
164

Nikolai Kosjar's avatar
Nikolai Kosjar committed
165 166 167 168
private:
    BinaryExpressionAST *binary;
    NestedExpressionAST *nested;
    UnaryExpressionAST *negation;
169

Nikolai Kosjar's avatar
Nikolai Kosjar committed
170
    QString replacement;
171 172
};

Nikolai Kosjar's avatar
Nikolai Kosjar committed
173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208
void InverseLogicalComparison::match(const CppQuickFixInterface &interface,
                                     QuickFixOperations &result)
{
    CppRefactoringFilePtr file = interface->currentFile();

    const QList<AST *> &path = interface->path();
    int index = path.size() - 1;
    BinaryExpressionAST *binary = path.at(index)->asBinaryExpression();
    if (! binary)
        return;
    if (! interface->isCursorOn(binary->binary_op_token))
        return;

    Kind invertToken;
    switch (file->tokenAt(binary->binary_op_token).kind()) {
    case T_LESS_EQUAL:
        invertToken = T_GREATER;
        break;
    case T_LESS:
        invertToken = T_GREATER_EQUAL;
        break;
    case T_GREATER:
        invertToken = T_LESS_EQUAL;
        break;
    case T_GREATER_EQUAL:
        invertToken = T_LESS;
        break;
    case T_EQUAL_EQUAL:
        invertToken = T_EXCLAIM_EQUAL;
        break;
    case T_EXCLAIM_EQUAL:
        invertToken = T_EQUAL_EQUAL;
        break;
    default:
        return;
    }
209

Nikolai Kosjar's avatar
Nikolai Kosjar committed
210 211 212
    result.append(CppQuickFixOperation::Ptr(
        new InverseLogicalComparisonOp(interface, index, binary, invertToken)));
}
213

Nikolai Kosjar's avatar
Nikolai Kosjar committed
214
class FlipLogicalOperandsOp: public CppQuickFixOperation
215 216
{
public:
Nikolai Kosjar's avatar
Nikolai Kosjar committed
217 218 219 220 221
    FlipLogicalOperandsOp(const CppQuickFixInterface &interface, int priority,
                          BinaryExpressionAST *binary, QString replacement)
        : CppQuickFixOperation(interface)
        , binary(binary)
        , replacement(replacement)
222
    {
Nikolai Kosjar's avatar
Nikolai Kosjar committed
223 224
        setPriority(priority);
    }
225

Nikolai Kosjar's avatar
Nikolai Kosjar committed
226 227 228 229 230 231 232
    QString description() const
    {
        if (replacement.isEmpty())
            return QApplication::translate("CppTools::QuickFix", "Swap Operands");
        else
            return QApplication::translate("CppTools::QuickFix", "Rewrite Using %1").arg(replacement);
    }
233

Nikolai Kosjar's avatar
Nikolai Kosjar committed
234 235 236 237
    void perform()
    {
        CppRefactoringChanges refactoring(snapshot());
        CppRefactoringFilePtr currentFile = refactoring.file(fileName());
238

Nikolai Kosjar's avatar
Nikolai Kosjar committed
239 240 241 242
        ChangeSet changes;
        changes.flip(currentFile->range(binary->left_expression), currentFile->range(binary->right_expression));
        if (! replacement.isEmpty())
            changes.replace(currentFile->range(binary->binary_op_token), replacement);
243

Nikolai Kosjar's avatar
Nikolai Kosjar committed
244 245
        currentFile->setChangeSet(changes);
        currentFile->apply();
246 247 248
    }

private:
Nikolai Kosjar's avatar
Nikolai Kosjar committed
249 250 251
    BinaryExpressionAST *binary;
    QString replacement;
};
252

Nikolai Kosjar's avatar
Nikolai Kosjar committed
253 254 255 256
void FlipLogicalOperands::match(const CppQuickFixInterface &interface, QuickFixOperations &result)
{
    const QList<AST *> &path = interface->path();
    CppRefactoringFilePtr file = interface->currentFile();
257

Nikolai Kosjar's avatar
Nikolai Kosjar committed
258 259 260 261 262 263
    int index = path.size() - 1;
    BinaryExpressionAST *binary = path.at(index)->asBinaryExpression();
    if (! binary)
        return;
    if (! interface->isCursorOn(binary->binary_op_token))
        return;
264

Nikolai Kosjar's avatar
Nikolai Kosjar committed
265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287
    Kind flipToken;
    switch (file->tokenAt(binary->binary_op_token).kind()) {
    case T_LESS_EQUAL:
        flipToken = T_GREATER_EQUAL;
        break;
    case T_LESS:
        flipToken = T_GREATER;
        break;
    case T_GREATER:
        flipToken = T_LESS;
        break;
    case T_GREATER_EQUAL:
        flipToken = T_LESS_EQUAL;
        break;
    case T_EQUAL_EQUAL:
    case T_EXCLAIM_EQUAL:
    case T_AMPER_AMPER:
    case T_PIPE_PIPE:
        flipToken = T_EOF_SYMBOL;
        break;
    default:
        return;
    }
288

Nikolai Kosjar's avatar
Nikolai Kosjar committed
289 290 291 292 293 294
    QString replacement;
    if (flipToken != T_EOF_SYMBOL) {
        Token tok;
        tok.f.kind = flipToken;
        replacement = QLatin1String(tok.spell());
    }
295

Nikolai Kosjar's avatar
Nikolai Kosjar committed
296 297 298
    result.append(QuickFixOperation::Ptr(
        new FlipLogicalOperandsOp(interface, index, binary, replacement)));
}
299

Nikolai Kosjar's avatar
Nikolai Kosjar committed
300
class RewriteLogicalAndOp: public CppQuickFixOperation
301 302
{
public:
Nikolai Kosjar's avatar
Nikolai Kosjar committed
303 304 305 306 307 308 309 310
    QSharedPointer<ASTPatternBuilder> mk;
    UnaryExpressionAST *left;
    UnaryExpressionAST *right;
    BinaryExpressionAST *pattern;

    RewriteLogicalAndOp(const CppQuickFixInterface &interface)
        : CppQuickFixOperation(interface)
        , mk(new ASTPatternBuilder)
311
    {
Nikolai Kosjar's avatar
Nikolai Kosjar committed
312 313 314 315
        left = mk->UnaryExpression();
        right = mk->UnaryExpression();
        pattern = mk->BinaryExpression(left, right);
    }
316

Nikolai Kosjar's avatar
Nikolai Kosjar committed
317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335
    void perform()
    {
        CppRefactoringChanges refactoring(snapshot());
        CppRefactoringFilePtr currentFile = refactoring.file(fileName());

        ChangeSet changes;
        changes.replace(currentFile->range(pattern->binary_op_token), QLatin1String("||"));
        changes.remove(currentFile->range(left->unary_op_token));
        changes.remove(currentFile->range(right->unary_op_token));
        const int start = currentFile->startOf(pattern);
        const int end = currentFile->endOf(pattern);
        changes.insert(start, QLatin1String("!("));
        changes.insert(end, QLatin1String(")"));

        currentFile->setChangeSet(changes);
        currentFile->appendIndentRange(currentFile->range(pattern));
        currentFile->apply();
    }
};
336

Nikolai Kosjar's avatar
Nikolai Kosjar committed
337 338 339 340 341
void RewriteLogicalAnd::match(const CppQuickFixInterface &interface, QuickFixOperations &result)
{
    BinaryExpressionAST *expression = 0;
    const QList<AST *> &path = interface->path();
    CppRefactoringFilePtr file = interface->currentFile();
342

Nikolai Kosjar's avatar
Nikolai Kosjar committed
343 344 345 346 347
    int index = path.size() - 1;
    for (; index != -1; --index) {
        expression = path.at(index)->asBinaryExpression();
        if (expression)
            break;
348 349
    }

Nikolai Kosjar's avatar
Nikolai Kosjar committed
350 351
    if (! expression)
        return;
352

Nikolai Kosjar's avatar
Nikolai Kosjar committed
353 354
    if (! interface->isCursorOn(expression->binary_op_token))
        return;
355

Nikolai Kosjar's avatar
Nikolai Kosjar committed
356
    QSharedPointer<RewriteLogicalAndOp> op(new RewriteLogicalAndOp(interface));
357

Nikolai Kosjar's avatar
Nikolai Kosjar committed
358 359 360 361 362 363 364 365 366
    if (expression->match(op->pattern, &matcher) &&
            file->tokenAt(op->pattern->binary_op_token).is(T_AMPER_AMPER) &&
            file->tokenAt(op->left->unary_op_token).is(T_EXCLAIM) &&
            file->tokenAt(op->right->unary_op_token).is(T_EXCLAIM)) {
        op->setDescription(QApplication::translate("CppTools::QuickFix", "Rewrite Condition Using ||"));
        op->setPriority(index);
        result.append(op);
    }
}
367

Nikolai Kosjar's avatar
Nikolai Kosjar committed
368 369 370 371
bool SplitSimpleDeclaration::checkDeclaration(SimpleDeclarationAST *declaration)
{
    if (! declaration->semicolon_token)
        return false;
372

Nikolai Kosjar's avatar
Nikolai Kosjar committed
373 374
    if (! declaration->decl_specifier_list)
        return false;
375

Nikolai Kosjar's avatar
Nikolai Kosjar committed
376 377
    for (SpecifierListAST *it = declaration->decl_specifier_list; it; it = it->next) {
        SpecifierAST *specifier = it->value;
378

Nikolai Kosjar's avatar
Nikolai Kosjar committed
379
        if (specifier->asEnumSpecifier() != 0)
380 381
            return false;

Nikolai Kosjar's avatar
Nikolai Kosjar committed
382
        else if (specifier->asClassSpecifier() != 0)
383
            return false;
Nikolai Kosjar's avatar
Nikolai Kosjar committed
384
    }
385

Nikolai Kosjar's avatar
Nikolai Kosjar committed
386 387
    if (! declaration->declarator_list)
        return false;
388

Nikolai Kosjar's avatar
Nikolai Kosjar committed
389 390
    else if (! declaration->declarator_list->next)
        return false;
391

Nikolai Kosjar's avatar
Nikolai Kosjar committed
392 393
    return true;
}
394

Nikolai Kosjar's avatar
Nikolai Kosjar committed
395 396 397 398 399 400 401 402 403 404
class SplitSimpleDeclarationOp: public CppQuickFixOperation
{
public:
    SplitSimpleDeclarationOp(const CppQuickFixInterface &interface, int priority,
                             SimpleDeclarationAST *decl)
        : CppQuickFixOperation(interface, priority)
        , declaration(decl)
    {
        setDescription(QApplication::translate("CppTools::QuickFix",
                                               "Split Declaration"));
405 406
    }

Nikolai Kosjar's avatar
Nikolai Kosjar committed
407
    void perform()
408
    {
Nikolai Kosjar's avatar
Nikolai Kosjar committed
409 410
        CppRefactoringChanges refactoring(snapshot());
        CppRefactoringFilePtr currentFile = refactoring.file(fileName());
411

Nikolai Kosjar's avatar
Nikolai Kosjar committed
412
        ChangeSet changes;
413

Nikolai Kosjar's avatar
Nikolai Kosjar committed
414 415 416 417
        SpecifierListAST *specifiers = declaration->decl_specifier_list;
        int declSpecifiersStart = currentFile->startOf(specifiers->firstToken());
        int declSpecifiersEnd = currentFile->endOf(specifiers->lastToken() - 1);
        int insertPos = currentFile->endOf(declaration->semicolon_token);
418

Nikolai Kosjar's avatar
Nikolai Kosjar committed
419
        DeclaratorAST *prevDeclarator = declaration->declarator_list->value;
420

Nikolai Kosjar's avatar
Nikolai Kosjar committed
421 422
        for (DeclaratorListAST *it = declaration->declarator_list->next; it; it = it->next) {
            DeclaratorAST *declarator = it->value;
423

Nikolai Kosjar's avatar
Nikolai Kosjar committed
424 425 426 427 428
            changes.insert(insertPos, QLatin1String("\n"));
            changes.copy(declSpecifiersStart, declSpecifiersEnd, insertPos);
            changes.insert(insertPos, QLatin1String(" "));
            changes.move(currentFile->range(declarator), insertPos);
            changes.insert(insertPos, QLatin1String(";"));
429

Nikolai Kosjar's avatar
Nikolai Kosjar committed
430 431
            const int prevDeclEnd = currentFile->endOf(prevDeclarator);
            changes.remove(prevDeclEnd, currentFile->startOf(declarator));
432

Nikolai Kosjar's avatar
Nikolai Kosjar committed
433
            prevDeclarator = declarator;
434
        }
Nikolai Kosjar's avatar
Nikolai Kosjar committed
435 436 437 438

        currentFile->setChangeSet(changes);
        currentFile->appendIndentRange(currentFile->range(declaration));
        currentFile->apply();
439 440 441
    }

private:
Nikolai Kosjar's avatar
Nikolai Kosjar committed
442 443
    SimpleDeclarationAST *declaration;
};
444

Nikolai Kosjar's avatar
Nikolai Kosjar committed
445 446 447 448 449 450 451
void SplitSimpleDeclaration::match(const CppQuickFixInterface &interface,
                                   QuickFixOperations &result)
{
    CoreDeclaratorAST *core_declarator = 0;
    const QList<AST *> &path = interface->path();
    CppRefactoringFilePtr file = interface->currentFile();
    const int cursorPosition = file->cursor().selectionStart();
452

Nikolai Kosjar's avatar
Nikolai Kosjar committed
453 454
    for (int index = path.size() - 1; index != -1; --index) {
        AST *node = path.at(index);
455

Nikolai Kosjar's avatar
Nikolai Kosjar committed
456 457
        if (CoreDeclaratorAST *coreDecl = node->asCoreDeclarator())
            core_declarator = coreDecl;
458

Nikolai Kosjar's avatar
Nikolai Kosjar committed
459 460 461
        else if (SimpleDeclarationAST *simpleDecl = node->asSimpleDeclaration()) {
            if (checkDeclaration(simpleDecl)) {
                SimpleDeclarationAST *declaration = simpleDecl;
462

Nikolai Kosjar's avatar
Nikolai Kosjar committed
463 464
                const int startOfDeclSpecifier = file->startOf(declaration->decl_specifier_list->firstToken());
                const int endOfDeclSpecifier = file->endOf(declaration->decl_specifier_list->lastToken() - 1);
465

Nikolai Kosjar's avatar
Nikolai Kosjar committed
466 467 468 469 470 471
                if (cursorPosition >= startOfDeclSpecifier && cursorPosition <= endOfDeclSpecifier) {
                    // the AST node under cursor is a specifier.
                    result.append(QuickFixOperation::Ptr(
                        new SplitSimpleDeclarationOp(interface, index, declaration)));
                    return;
                }
472

Nikolai Kosjar's avatar
Nikolai Kosjar committed
473 474 475 476 477 478
                if (core_declarator && interface->isCursorOn(core_declarator)) {
                    // got a core-declarator under the text cursor.
                    result.append(QuickFixOperation::Ptr(
                        new SplitSimpleDeclarationOp(interface, index, declaration)));
                    return;
                }
479 480
            }

Nikolai Kosjar's avatar
Nikolai Kosjar committed
481
            return;
482
        }
Nikolai Kosjar's avatar
Nikolai Kosjar committed
483 484
    }
}
485

Nikolai Kosjar's avatar
Nikolai Kosjar committed
486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502
class AddBracesToIfOp: public CppQuickFixOperation
{
public:
    AddBracesToIfOp(const CppQuickFixInterface &interface, int priority, StatementAST *statement)
        : CppQuickFixOperation(interface, priority)
        , _statement(statement)
    {
        setDescription(QApplication::translate("CppTools::QuickFix",
                                               "Add Curly Braces"));
    }

    void perform()
    {
        CppRefactoringChanges refactoring(snapshot());
        CppRefactoringFilePtr currentFile = refactoring.file(fileName());

        ChangeSet changes;
503

Nikolai Kosjar's avatar
Nikolai Kosjar committed
504 505
        const int start = currentFile->endOf(_statement->firstToken() - 1);
        changes.insert(start, QLatin1String(" {"));
506

Nikolai Kosjar's avatar
Nikolai Kosjar committed
507 508
        const int end = currentFile->endOf(_statement->lastToken() - 1);
        changes.insert(end, QLatin1String("\n}"));
509

Nikolai Kosjar's avatar
Nikolai Kosjar committed
510 511 512 513 514 515 516 517 518 519
        currentFile->setChangeSet(changes);
        currentFile->appendIndentRange(Utils::ChangeSet::Range(start, end));
        currentFile->apply();
    }

private:
    StatementAST *_statement;
};

void AddBracesToIf::match(const CppQuickFixInterface &interface, QuickFixOperations &result)
520
{
Nikolai Kosjar's avatar
Nikolai Kosjar committed
521
    const QList<AST *> &path = interface->path();
522

Nikolai Kosjar's avatar
Nikolai Kosjar committed
523 524 525 526 527 528 529 530 531 532 533 534 535
    // show when we're on the 'if' of an if statement
    int index = path.size() - 1;
    IfStatementAST *ifStatement = path.at(index)->asIfStatement();
    if (ifStatement && interface->isCursorOn(ifStatement->if_token) && ifStatement->statement
        && ! ifStatement->statement->asCompoundStatement()) {
        result.append(QuickFixOperation::Ptr(
            new AddBracesToIfOp(interface, index, ifStatement->statement)));
        return;
    }

    // or if we're on the statement contained in the if
    // ### This may not be such a good idea, consider nested ifs...
    for (; index != -1; --index) {
536
        IfStatementAST *ifStatement = path.at(index)->asIfStatement();
Nikolai Kosjar's avatar
Nikolai Kosjar committed
537 538
        if (ifStatement && ifStatement->statement
            && interface->isCursorOn(ifStatement->statement)
539
            && ! ifStatement->statement->asCompoundStatement()) {
Nikolai Kosjar's avatar
Nikolai Kosjar committed
540 541
            result.append(QuickFixOperation::Ptr(
                new AddBracesToIfOp(interface, index, ifStatement->statement)));
542
            return;
543
        }
Nikolai Kosjar's avatar
Nikolai Kosjar committed
544
    }
545

Nikolai Kosjar's avatar
Nikolai Kosjar committed
546 547 548
    // ### This could very well be extended to the else branch
    // and other nodes entirely.
}
549

Nikolai Kosjar's avatar
Nikolai Kosjar committed
550 551 552 553 554 555 556 557 558 559 560
class MoveDeclarationOutOfIfOp: public CppQuickFixOperation
{
public:
    MoveDeclarationOutOfIfOp(const CppQuickFixInterface &interface)
        : CppQuickFixOperation(interface)
    {
        setDescription(QApplication::translate("CppTools::QuickFix",
                                               "Move Declaration out of Condition"));

        condition = mk.Condition();
        pattern = mk.IfStatement(condition);
561 562
    }

Nikolai Kosjar's avatar
Nikolai Kosjar committed
563
    void perform()
564
    {
Nikolai Kosjar's avatar
Nikolai Kosjar committed
565 566
        CppRefactoringChanges refactoring(snapshot());
        CppRefactoringFilePtr currentFile = refactoring.file(fileName());
567

Nikolai Kosjar's avatar
Nikolai Kosjar committed
568
        ChangeSet changes;
569

Nikolai Kosjar's avatar
Nikolai Kosjar committed
570
        changes.copy(currentFile->range(core), currentFile->startOf(condition));
571

Nikolai Kosjar's avatar
Nikolai Kosjar committed
572 573 574
        int insertPos = currentFile->startOf(pattern);
        changes.move(currentFile->range(condition), insertPos);
        changes.insert(insertPos, QLatin1String(";\n"));
575

Nikolai Kosjar's avatar
Nikolai Kosjar committed
576 577 578 579
        currentFile->setChangeSet(changes);
        currentFile->appendIndentRange(currentFile->range(pattern));
        currentFile->apply();
    }
580

Nikolai Kosjar's avatar
Nikolai Kosjar committed
581 582 583 584 585 586
    ASTMatcher matcher;
    ASTPatternBuilder mk;
    CPPEditorWidget *editor;
    ConditionAST *condition;
    IfStatementAST *pattern;
    CoreDeclaratorAST *core;
587 588
};

Nikolai Kosjar's avatar
Nikolai Kosjar committed
589 590
void MoveDeclarationOutOfIf::match(const CppQuickFixInterface &interface,
                                   QuickFixOperations &result)
591
{
Nikolai Kosjar's avatar
Nikolai Kosjar committed
592 593 594 595 596 597 598 599 600 601 602 603
    const QList<AST *> &path = interface->path();
    typedef QSharedPointer<MoveDeclarationOutOfIfOp> Ptr;
    Ptr op(new MoveDeclarationOutOfIfOp(interface));

    int index = path.size() - 1;
    for (; index != -1; --index) {
        if (IfStatementAST *statement = path.at(index)->asIfStatement()) {
            if (statement->match(op->pattern, &op->matcher) && op->condition->declarator) {
                DeclaratorAST *declarator = op->condition->declarator;
                op->core = declarator->core_declarator;
                if (! op->core)
                    return;
604

Nikolai Kosjar's avatar
Nikolai Kosjar committed
605 606 607 608
                if (interface->isCursorOn(op->core)) {
                    op->setPriority(index);
                    result.append(op);
                    return;
609 610 611 612
                }
            }
        }
    }
Nikolai Kosjar's avatar
Nikolai Kosjar committed
613
}
614

Nikolai Kosjar's avatar
Nikolai Kosjar committed
615 616 617 618 619
class MoveDeclarationOutOfWhileOp: public CppQuickFixOperation
{
public:
    MoveDeclarationOutOfWhileOp(const CppQuickFixInterface &interface)
        : CppQuickFixOperation(interface)
620
    {
Nikolai Kosjar's avatar
Nikolai Kosjar committed
621 622
        setDescription(QApplication::translate("CppTools::QuickFix",
                                               "Move Declaration out of Condition"));
623

Nikolai Kosjar's avatar
Nikolai Kosjar committed
624 625 626
        condition = mk.Condition();
        pattern = mk.WhileStatement(condition);
    }
627

Nikolai Kosjar's avatar
Nikolai Kosjar committed
628 629 630 631
    void perform()
    {
        CppRefactoringChanges refactoring(snapshot());
        CppRefactoringFilePtr currentFile = refactoring.file(fileName());
632

Nikolai Kosjar's avatar
Nikolai Kosjar committed
633
        ChangeSet changes;
634

Nikolai Kosjar's avatar
Nikolai Kosjar committed
635 636
        changes.insert(currentFile->startOf(condition), QLatin1String("("));
        changes.insert(currentFile->endOf(condition), QLatin1String(") != 0"));
637

Nikolai Kosjar's avatar
Nikolai Kosjar committed
638 639 640 641 642
        int insertPos = currentFile->startOf(pattern);
        const int conditionStart = currentFile->startOf(condition);
        changes.move(conditionStart, currentFile->startOf(core), insertPos);
        changes.copy(currentFile->range(core), insertPos);
        changes.insert(insertPos, QLatin1String(";\n"));
643

Nikolai Kosjar's avatar
Nikolai Kosjar committed
644 645 646 647
        currentFile->setChangeSet(changes);
        currentFile->appendIndentRange(currentFile->range(pattern));
        currentFile->apply();
    }
648

Nikolai Kosjar's avatar
Nikolai Kosjar committed
649 650 651 652 653 654 655
    ASTMatcher matcher;
    ASTPatternBuilder mk;
    CPPEditorWidget *editor;
    ConditionAST *condition;
    WhileStatementAST *pattern;
    CoreDeclaratorAST *core;
};
656

Nikolai Kosjar's avatar
Nikolai Kosjar committed
657 658
void MoveDeclarationOutOfWhile::match(const CppQuickFixInterface &interface,
                                      QuickFixOperations &result)
659
{
Nikolai Kosjar's avatar
Nikolai Kosjar committed
660 661
    const QList<AST *> &path = interface->path();
    QSharedPointer<MoveDeclarationOutOfWhileOp> op(new MoveDeclarationOutOfWhileOp(interface));
662

Nikolai Kosjar's avatar
Nikolai Kosjar committed
663 664 665 666 667 668
    int index = path.size() - 1;
    for (; index != -1; --index) {
        if (WhileStatementAST *statement = path.at(index)->asWhileStatement()) {
            if (statement->match(op->pattern, &op->matcher) && op->condition->declarator) {
                DeclaratorAST *declarator = op->condition->declarator;
                op->core = declarator->core_declarator;
669

Nikolai Kosjar's avatar
Nikolai Kosjar committed
670 671
                if (! op->core)
                    return;
672

Nikolai Kosjar's avatar
Nikolai Kosjar committed
673 674
                if (! declarator->equal_token)
                    return;
675

Nikolai Kosjar's avatar
Nikolai Kosjar committed
676 677
                if (! declarator->initializer)
                    return;
678

Nikolai Kosjar's avatar
Nikolai Kosjar committed
679 680 681 682
                if (interface->isCursorOn(op->core)) {
                    op->setPriority(index);
                    result.append(op);
                    return;
683 684 685 686
                }
            }
        }
    }
Nikolai Kosjar's avatar
Nikolai Kosjar committed
687
}
688

Nikolai Kosjar's avatar
Nikolai Kosjar committed
689 690 691 692 693 694 695 696
class SplitIfStatementOp: public CppQuickFixOperation
{
public:
    SplitIfStatementOp(const CppQuickFixInterface &interface, int priority,
                       IfStatementAST *pattern, BinaryExpressionAST *condition)
        : CppQuickFixOperation(interface, priority)
        , pattern(pattern)
        , condition(condition)
697
    {
Nikolai Kosjar's avatar
Nikolai Kosjar committed
698 699 700 701 702 703 704 705
        setDescription(QApplication::translate("CppTools::QuickFix",
                                               "Split if Statement"));
    }

    void perform()
    {
        CppRefactoringChanges refactoring(snapshot());
        CppRefactoringFilePtr currentFile = refactoring.file(fileName());
706

Nikolai Kosjar's avatar
Nikolai Kosjar committed
707
        const Token binaryToken = currentFile->tokenAt(condition->binary_op_token);
708

Nikolai Kosjar's avatar
Nikolai Kosjar committed
709 710 711 712 713
        if (binaryToken.is(T_AMPER_AMPER))
            splitAndCondition(currentFile);
        else
            splitOrCondition(currentFile);
    }
714

Nikolai Kosjar's avatar
Nikolai Kosjar committed
715 716 717
    void splitAndCondition(CppRefactoringFilePtr currentFile) const
    {
        ChangeSet changes;
718

Nikolai Kosjar's avatar
Nikolai Kosjar committed
719 720 721 722
        int startPos = currentFile->startOf(pattern);
        changes.insert(startPos, QLatin1String("if ("));
        changes.move(currentFile->range(condition->left_expression), startPos);
        changes.insert(startPos, QLatin1String(") {\n"));
723

Nikolai Kosjar's avatar
Nikolai Kosjar committed
724 725 726
        const int lExprEnd = currentFile->endOf(condition->left_expression);
        changes.remove(lExprEnd, currentFile->startOf(condition->right_expression));
        changes.insert(currentFile->endOf(pattern), QLatin1String("\n}"));
727

Nikolai Kosjar's avatar
Nikolai Kosjar committed
728 729 730 731
        currentFile->setChangeSet(changes);
        currentFile->appendIndentRange(currentFile->range(pattern));
        currentFile->apply();
    }
732

Nikolai Kosjar's avatar
Nikolai Kosjar committed
733
    void splitOrCondition(CppRefactoringFilePtr currentFile) const
734
    {
Nikolai Kosjar's avatar
Nikolai Kosjar committed
735
        ChangeSet changes;
736

Nikolai Kosjar's avatar
Nikolai Kosjar committed
737 738
        StatementAST *ifTrueStatement = pattern->statement;
        CompoundStatementAST *compoundStatement = ifTrueStatement->asCompoundStatement();
739

Nikolai Kosjar's avatar
Nikolai Kosjar committed
740 741 742 743 744 745
        int insertPos = currentFile->endOf(ifTrueStatement);
        if (compoundStatement)
            changes.insert(insertPos, QLatin1String(" "));
        else
            changes.insert(insertPos, QLatin1String("\n"));
        changes.insert(insertPos, QLatin1String("else if ("));
746

Nikolai Kosjar's avatar
Nikolai Kosjar committed
747 748 749
        const int rExprStart = currentFile->startOf(condition->right_expression);
        changes.move(rExprStart, currentFile->startOf(pattern->rparen_token), insertPos);
        changes.insert(insertPos, QLatin1String(")"));
750

Nikolai Kosjar's avatar
Nikolai Kosjar committed
751 752
        const int rParenEnd = currentFile->endOf(pattern->rparen_token);
        changes.copy(rParenEnd, currentFile->endOf(pattern->statement), insertPos);
753

Nikolai Kosjar's avatar
Nikolai Kosjar committed
754 755 756 757 758 759
        const int lExprEnd = currentFile->endOf(condition->left_expression);
        changes.remove(lExprEnd, currentFile->startOf(condition->right_expression));

        currentFile->setChangeSet(changes);
        currentFile->appendIndentRange(currentFile->range(pattern));
        currentFile->apply();
760 761 762
    }

private:
Nikolai Kosjar's avatar
Nikolai Kosjar committed
763 764 765
    IfStatementAST *pattern;
    BinaryExpressionAST *condition;
};
766

Nikolai Kosjar's avatar
Nikolai Kosjar committed
767 768 769 770
void SplitIfStatement::match(const CppQuickFixInterface &interface, QuickFixOperations &result)
{
    IfStatementAST *pattern = 0;
    const QList<AST *> &path = interface->path();
771

Nikolai Kosjar's avatar
Nikolai Kosjar committed
772 773 774 775 776 777
    int index = path.size() - 1;
    for (; index != -1; --index) {
        AST *node = path.at(index);
        if (IfStatementAST *stmt = node->asIfStatement()) {
            pattern = stmt;
            break;
778
        }
Nikolai Kosjar's avatar
Nikolai Kosjar committed
779
    }
780

Nikolai Kosjar's avatar
Nikolai Kosjar committed
781 782
    if (! pattern || ! pattern->statement)
        return;
783

Nikolai Kosjar's avatar
Nikolai Kosjar committed
784 785 786 787 788 789
    unsigned splitKind = 0;
    for (++index; index < path.size(); ++index) {
        AST *node = path.at(index);
        BinaryExpressionAST *condition = node->asBinaryExpression();
        if (! condition)
            return;
790

Nikolai Kosjar's avatar
Nikolai Kosjar committed
791
        Token binaryToken = interface->currentFile()->tokenAt(condition->binary_op_token);
792

Nikolai Kosjar's avatar
Nikolai Kosjar committed
793 794 795 796 797 798 799 800 801 802
        // only accept a chain of ||s or &&s - no mixing
        if (! splitKind) {
            splitKind = binaryToken.kind();
            if (splitKind != T_AMPER_AMPER && splitKind != T_PIPE_PIPE)
                return;
            // we can't reliably split &&s in ifs with an else branch
            if (splitKind == T_AMPER_AMPER && pattern->else_statement)
                return;
        } else if (splitKind != binaryToken.kind()) {
            return;
803 804
        }

Nikolai Kosjar's avatar
Nikolai Kosjar committed
805 806 807 808
        if (interface->isCursorOn(condition->binary_op_token)) {
            result.append(QuickFixOperation::Ptr(
                new SplitIfStatementOp(interface, index, pattern, condition)));
            return;
809
        }
Nikolai Kosjar's avatar
Nikolai Kosjar committed
810 811
    }
}
812 813 814 815 816 817 818

static inline QString msgQtStringLiteralDescription(const QString &replacement, int qtVersion)
{
    return QApplication::translate("CppTools::QuickFix", "Enclose in %1(...) (Qt %2)")
           .arg(replacement).arg(qtVersion);
}

819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851
static inline QString msgQtStringLiteralDescription(const QString &replacement)
{
    return QApplication::translate("CppTools::QuickFix", "Enclose in %1(...)").arg(replacement);
}

/* Analze a string/character literal like "x", QLatin1String("x") and return the literal
 * (StringLiteral or NumericLiteral for characters) and its type
 * and the enclosing function (QLatin1String, tr...) */
ExpressionAST *WrapStringLiteral::analyze(const QList<AST *> &path,
                                          const CppRefactoringFilePtr &file,
                                          Type *type,
                                          QByteArray *enclosingFunction /* = 0 */,
                                          CallAST **enclosingFunctionCall /* = 0 */)
{
    *type = TypeNone;
    if (enclosingFunction)
        enclosingFunction->clear();
    if (enclosingFunctionCall)
        *enclosingFunctionCall = 0;

    if (path.isEmpty())
        return 0;

    ExpressionAST *literal = path.last()->asExpression();
    if (literal) {
        if (literal->asStringLiteral()) {
            // Check for Objective C string (@"bla")
            const QChar firstChar = file->charAt(file->startOf(literal));
            *type = firstChar == QLatin1Char('@') ? TypeObjCString : TypeString;
        } else if (NumericLiteralAST *numericLiteral = literal->asNumericLiteral()) {
            // character ('c') constants are numeric.
            if (file->tokenAt(numericLiteral->literal_token).is(T_CHAR_LITERAL))
                *type = TypeChar;
852
        }
853
    }
854

855 856 857 858 859 860 861 862
    if (*type != TypeNone && enclosingFunction && path.size() > 1) {
        if (CallAST *call = path.at(path.size() - 2)->asCall()) {
            if (call->base_expression) {
                if (IdExpressionAST *idExpr = call->base_expression->asIdExpression()) {
                    if (SimpleNameAST *functionName = idExpr->name->asSimpleName()) {
                        *enclosingFunction = file->tokenAt(functionName->identifier_token).identifier->chars();
                        if (enclosingFunctionCall)
                            *enclosingFunctionCall = call;
863 864 865 866
                    }
                }
            }
        }
867 868 869
    }
    return literal;
}
870

Nikolai Kosjar's avatar
Nikolai Kosjar committed
871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951
/// Operation performs the operations of type ActionFlags passed in as actions.
class WrapStringLiteralOp : public CppQuickFixOperation
{
public:
    typedef WrapStringLiteral Factory;

    WrapStringLiteralOp(const CppQuickFixInterface &interface, int priority,
                        unsigned actions, const QString &description, ExpressionAST *literal,
                        const QString &translationContext = QString())
        : CppQuickFixOperation(interface, priority), m_actions(actions), m_literal(literal),
          m_translationContext(translationContext)
    {
        setDescription(description);
    }

    void perform()
    {
        CppRefactoringChanges refactoring(snapshot());
        CppRefactoringFilePtr currentFile = refactoring.file(fileName());

        ChangeSet changes;

        const int startPos = currentFile->startOf(m_literal);
        const int endPos = currentFile->endOf(m_literal);

        // kill leading '@'. No need to adapt endPos, that is done by ChangeSet
        if (m_actions & Factory::RemoveObjectiveCAction)
            changes.remove(startPos, startPos + 1);

        // Fix quotes
        if (m_actions & (Factory::SingleQuoteAction | Factory::DoubleQuoteAction)) {
            const QString newQuote((m_actions & Factory::SingleQuoteAction) ? QLatin1Char('\'') : QLatin1Char('"'));
            changes.replace(startPos, startPos + 1, newQuote);
            changes.replace(endPos - 1, endPos, newQuote);
        }

        // Convert single character strings into character constants
        if (m_actions & Factory::ConvertEscapeSequencesToCharAction) {
            StringLiteralAST *stringLiteral = m_literal->asStringLiteral();
            QTC_ASSERT(stringLiteral, return ;);
            const QByteArray oldContents(currentFile->tokenAt(stringLiteral->literal_token).identifier->chars());
            const QByteArray newContents = Factory::stringToCharEscapeSequences(oldContents);
            QTC_ASSERT(!newContents.isEmpty(), return ;);
            if (oldContents != newContents)
                changes.replace(startPos + 1, endPos -1, QString::fromLatin1(newContents));
        }

        // Convert character constants into strings constants
        if (m_actions & Factory::ConvertEscapeSequencesToStringAction) {
            NumericLiteralAST *charLiteral = m_literal->asNumericLiteral(); // char 'c' constants are numerical.
            QTC_ASSERT(charLiteral, return ;);
            const QByteArray oldContents(currentFile->tokenAt(charLiteral->literal_token).identifier->chars());
            const QByteArray newContents = Factory::charToStringEscapeSequences(oldContents);
            QTC_ASSERT(!newContents.isEmpty(), return ;);
            if (oldContents != newContents)
                changes.replace(startPos + 1, endPos -1, QString::fromLatin1(newContents));
        }

        // Enclose in literal or translation function, macro.
        if (m_actions & (Factory::EncloseActionMask | Factory::TranslationMask)) {
            changes.insert(endPos, QString(QLatin1Char(')')));
            QString leading = Factory::replacement(m_actions);
            leading += QLatin1Char('(');
            if (m_actions & (Factory::TranslateQCoreApplicationAction | Factory::TranslateNoopAction)) {
                leading += QLatin1Char('"');
                leading += m_translationContext;
                leading += QLatin1String("\", ");
            }
            changes.insert(startPos, leading);
        }

        currentFile->setChangeSet(changes);
        currentFile->apply();
    }

private:
    const unsigned m_actions;
    ExpressionAST *m_literal;
    const QString m_translationContext;
};

952
void WrapStringLiteral::match(const CppQuickFixInterface &interface, QuickFixOperations &result)
953 954 955 956 957 958 959
{
    typedef CppQuickFixOperation::Ptr OperationPtr;

    Type type = TypeNone;
    QByteArray enclosingFunction;
    const QList<AST *> &path = interface->path();
    CppRefactoringFilePtr file = interface->currentFile();
Nikolai Kosjar's avatar
Nikolai Kosjar committed
960
    ExpressionAST *literal = analyze(path, file, &type, &enclosingFunction);
961
    if (!literal || type == TypeNone)
962
        return;
963 964 965
    if ((type == TypeChar && enclosingFunction == "QLatin1Char")
        || isQtStringLiteral(enclosingFunction)
        || isQtStringTranslation(enclosingFunction))
966
        return;
967

968 969 970
    const int priority = path.size() - 1; // very high priority
    if (type == TypeChar) {
        unsigned actions = EncloseInQLatin1CharAction;
Nikolai Kosjar's avatar
Nikolai Kosjar committed
971 972 973
        QString description = msgQtStringLiteralDescription(replacement(actions));
        result << OperationPtr(new WrapStringLiteralOp(interface, priority, actions,
                                                             description, literal));
974 975 976 977 978 979
        if (NumericLiteralAST *charLiteral = literal->asNumericLiteral()) {
            const QByteArray contents(file->tokenAt(charLiteral->literal_token).identifier->chars());
            if (!charToStringEscapeSequences(contents).isEmpty()) {
                actions = DoubleQuoteAction | ConvertEscapeSequencesToStringAction;
                description = QApplication::translate("CppTools::QuickFix",
                              "Convert to String Literal");
Nikolai Kosjar's avatar
Nikolai Kosjar committed
980 981
                result << OperationPtr(new WrapStringLiteralOp(interface, priority, actions,
                                                                     description, literal));
982
            }
983
        }
984 985 986 987 988 989 990 991 992 993 994
    } else {
        const unsigned objectiveCActions = type == TypeObjCString ?
                                           unsigned(RemoveObjectiveCAction) : 0u;
        unsigned actions = 0;
        if (StringLiteralAST *stringLiteral = literal->asStringLiteral()) {
            const QByteArray contents(file->tokenAt(stringLiteral->literal_token).identifier->chars());
            if (!stringToCharEscapeSequences(contents).isEmpty()) {
                actions = EncloseInQLatin1CharAction | SingleQuoteAction
                          | ConvertEscapeSequencesToCharAction | objectiveCActions;
                QString description = QApplication::translate("CppTools::QuickFix",
                                      "Convert to Character Literal and Enclose in QLatin1Char(...)");
Nikolai Kosjar's avatar
Nikolai Kosjar committed
995 996
                result << OperationPtr(new WrapStringLiteralOp(interface, priority,
                                                                     actions, description, literal));
997 998 999
                actions &= ~EncloseInQLatin1CharAction;
                description = QApplication::translate("CppTools::QuickFix",
                              "Convert to Character Literal");
Nikolai Kosjar's avatar
Nikolai Kosjar committed
1000 1001
                result << OperationPtr(new WrapStringLiteralOp(interface, priority,
                                                                     actions, description, literal));
1002 1003 1004
            }
        }
        actions = EncloseInQLatin1StringAction | objectiveCActions;
Nikolai Kosjar's avatar
Nikolai Kosjar committed
1005 1006 1007 1008
        result << OperationPtr(
            new WrapStringLiteralOp(interface, priority, actions,
                msgQtStringLiteralDescription(replacement(actions), 4),
                    literal));
1009
        actions = EncloseInQStringLiteralAction | objectiveCActions;
Nikolai Kosjar's avatar
Nikolai Kosjar committed
1010 1011 1012
        result << OperationPtr(
            new WrapStringLiteralOp(interface, priority, actions,
                msgQtStringLiteralDescription(replacement(actions), 5), literal));
1013
    }
1014
}
1015

1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031
QString WrapStringLiteral::replacement(unsigned actions)
{
    if (actions & EncloseInQLatin1CharAction)
        return QLatin1String("QLatin1Char");
    if (actions & EncloseInQLatin1StringAction)
        return QLatin1String("QLatin1String");
    if (actions & EncloseInQStringLiteralAction)
        return QLatin1String("QStringLiteral");
    if (actions & TranslateTrAction)
        return QLatin1String("tr");
    if (actions & TranslateQCoreApplicationAction)
        return QLatin1String("QCoreApplication::translate");
    if (actions & TranslateNoopAction)
        return QLatin1String("QT_TRANSLATE_NOOP");
    return QString();
}
1032

1033 1034 1035 1036 1037 1038 1039 1040 1041 1042
/* Convert single-character string literals into character literals with some
 * special cases "a" --> 'a', "'" --> '\'', "\n" --> '\n', "\"" --> '"'. */
QByteArray WrapStringLiteral::stringToCharEscapeSequences(const QByteArray &content)
{
    if (content.size() == 1)
        return content.at(0) == '\'' ? QByteArray("\\'") : content;
    if (content.size() == 2 && content.at(0) == '\\')
        return content == "\\\"" ? QByteArray(1, '"') : content;
    return QByteArray();
}
1043

1044 1045 1046 1047 1048 1049 1050 1051 1052 1053
/* Convert character literal into a string literal with some special cases
 * 'a' -> "a", '\n' -> "\n", '\'' --> "'", '"' --> "\"". */
QByteArray WrapStringLiteral::charToStringEscapeSequences(const QByteArray &content)
{
    if (content.size() == 1)
        return content.at(0) == '"' ? QByteArray("\\\"") : content;
    if (content.size() == 2)
        return content == "\\'" ? QByteArray("'") : content;
    return QByteArray();
}
1054

Nikolai Kosjar's avatar
Nikolai Kosjar committed
1055 1056
void TranslateStringLiteral::match(const CppQuickFixInterface &interface,
                                   QuickFixOperations &result)
1057
{
Nikolai Kosjar's avatar
Nikolai Kosjar committed
1058 1059 1060 1061 1062 1063 1064 1065 1066
    // Initialize
    WrapStringLiteral::Type type = WrapStringLiteral::TypeNone;
    QByteArray enclosingFunction;
    const QList<AST *> &path = interface->path();
    CppRefactoringFilePtr file = interface->currentFile();
    ExpressionAST *literal = WrapStringLiteral::analyze(path, file, &type, &enclosingFunction);
    if (!literal || type != WrapStringLiteral::TypeString
       || isQtStringLiteral(enclosingFunction) || isQtStringTranslation(enclosingFunction))
        return;
1067

Nikolai Kosjar's avatar
Nikolai Kosjar committed
1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108
    QString trContext;

    QSharedPointer<Control> control = interface->context().control();
    const Name *trName = control->identifier("tr");

    // Check whether we are in a method:
    const QString description = QApplication::translate("CppTools::QuickFix", "Mark as Translatable");
    for (int i = path.size() - 1; i >= 0; --i) {
        if (FunctionDefinitionAST *definition = path.at(i)->asFunctionDefinition())