Clang Tool: переписать ObjCMessageExpr

Я хочу переписать все сообщения в моем коде, мне нужно заменить только селекторы, но мне нужно иметь возможность заменить вложенные выражения, например:

[super foo:[someInstance someMessage:@""] foo2:[someInstance someMessage2]];

Я пытался сделать это с clang::Rewriter replaceText и просто генерировать новую строку, но есть проблема: не будет работать, если я изменю длину селекторов, потому что я заменяю вложенные сообщения этими старыми позициями.

Итак, я предположил, что мне нужно использовать clang::Rewriter ReplaceStmt(originalStatement, newStatement);

я использую RecursiveASTVisitor чтобы посетить все сообщения, и я хочу скопировать объекты этих сообщений и заменить селекторы:

Как я могу это сделать?

Я пытался использовать ObjCMessageExpr::Create но там так много аргументов, я не знаю как получить ASTContext &Contextand ArrayRef<SourceLocation> SeLocs а также Expr *Receiver параметры из исходного сообщения.

Как правильно заменить селекторы во вложенных сообщениях, используя инструмент clang (интерфейс инструмента clang)?

Обновить:

Должен ли я использовать ReplaceStmtWithStmt обратный вызов и ASTMatchFinder?

Обновить:

Я использую следующую функцию для перезаписи текста в файле:

void ReplaceText(SourceLocation start, unsigned originalLength, StringRef string) { 
    m_rewriter.ReplaceText(start, originalLength, string); 
    m_rewriter.overwriteChangedFiles(); 
}

И я хочу заменить все messageExpr в коде на новый селектор, например:

[object someMessage:[object2 someMessage:obj3 calculate:obj4]];

как это должно быть:

[object newSelector:[object2 newSelector:obj3 newSelector:obj4]];

Я использую ReqoursiveASTVisitor:

bool VisitStmt(Stmt *statement) { 
    if (ObjCMessageExpr *messageExpr = dyn_cast<ObjCMessageExpr>(statement)) { 
         ReplaceMessage(*messageExpr) 
    } 
    return true; 
}

Я создал метод для генерации новой строки expr сообщения:

string StringFromObjCMessageExpr(ObjCMessageExpr& messageExpression) { 
    std::ostringstream stringStream; 
    const string selectorString = messageExpression.getSelector().getAsString(); 
    cout << selectorString << endl; 
    vector<string> methodParts; 
    split(selectorString, ParametersDelimiter, methodParts); 
    stringStream << "[" ; 
    const string receiver = GetStringFromLocations(m_compiler, messageExpression.getReceiverRange().getBegin(), messageExpression.getSelectorStartLoc()); 
    stringStream << receiver; 
    clang::ObjCMessageExpr::arg_iterator argIterator = messageExpression.arg_begin(); 
    for (vector<string>::const_iterator partsIterator = methodParts.begin(); 
         partsIterator != methodParts.end(); 
         ++partsIterator) { 
        stringStream << "newSelector"; 
        if (messageExpression.getNumArgs() != 0) { 
            const clang::Stmt *argument = *argIterator; 
            stringStream << ":" << GetStatementString(*argument) << " "; 
            ++argIterator; 
        } 
    } 
    stringStream << "]"; 
    return stringStream.str(); 
} 

void ReplaceMessage(ObjCMessageExpr& messageExpression) { 
    SourceLocation locStart = messageExpression.getLocStart(); 
    SourceLocation locEnd = messageExpression.getLocEnd(); 
    string newExpr = StringFromObjCMessageExpr(messageExpression); 
    const int exprStringLegth = m_rewriter.getRangeSize(SourceRange(locStart, locEnd)); 
    ReplaceText(locStart, exprStringLegth, newExpr); 
}

Проблема возникает, когда я пытаюсь заменить вложенные сообщения, например так:

[simpleClass doSomeActionWithString:string3 andAnotherString:string4]; 
[simpleClass doSomeActionWithString:str andAnotherString:str2]; 
[simpleClass doSomeActionWithString:@"" andAnotherString:@"asdasdsad"]; 
[simpleClass setSimpleClassZAZAZAZAZAZAZAZA:[simpleClass getSimpleClassZAZAZAZAZAZAZAZA]];

результат:

[simpleClass newSelector:string3 newSelector:string4 ]; 
[simpleClass newSelector:str newSelector:str2 ]; 
[simpleClass newSelector:@"" newSelector:@"asdasdsad" ]; 
[simpleClass newSelector:[simpleClass getSimp[simpleClass newSelector]];

потому что messageExpression имеет "старое" значение getLocStart(); а также getLocEnd(); Как я могу это исправить?

Источник

2 ответа

Вы можете переписать имя селектора, заменив только непрерывные части имени селектора. Например, заменить только подчеркнутые части

[object someMessage:[object2 someMessage:obj3 calculate:obj4]];
        ^~~~~~~~~~~          ^~~~~~~~~~~      ^~~~~~~~~

Для достижения этого вам требуется только

  • количество частей селектора - ObjCMessageExpr::getNumSelectorLocs()
  • их местоположения - ObjCMessageExpr::getSelectorLoc(index)
  • их длина - ObjCMessageExpr::getSelector().getNameForSlot(index).size(),

В целом, вы можете переписать ObjCMessageExpr с помощью следующего RecursiveASTVisitor:

#include "clang/AST/ASTConsumer.h"
#include "clang/AST/ASTContext.h"
#include "clang/AST/RecursiveASTVisitor.h"
#include "clang/Rewrite/Core/Rewriter.h"

namespace clang_tooling
{

using clang::SourceLocation;

class RewritingVisitor : public clang::ASTConsumer,
                         public clang::RecursiveASTVisitor<RewritingVisitor>
{
public:
    // You can obtain SourceManager and LangOptions from CompilerInstance when
    // you are creating visitor (which is also ASTConsumer) in
    // clang::ASTFrontendAction::CreateASTConsumer.
    RewritingVisitor(clang::SourceManager &sourceManager,
                     const clang::LangOptions &langOptions)
        : _sourceManager(sourceManager), _rewriter(sourceManager, langOptions)
    {}

    virtual void HandleTranslationUnit(clang::ASTContext &context)
    {
        TraverseDecl(context.getTranslationUnitDecl());
        _rewriter.overwriteChangedFiles();
    }

    bool VisitObjCMessageExpr(clang::ObjCMessageExpr *messageExpr)
    {
        if (_sourceManager.isInMainFile(messageExpr->getLocStart()))
        {
            clang::Selector selector = messageExpr->getSelector();
            for (unsigned i = 0, end = messageExpr->getNumSelectorLocs();
                 i < end; ++i)
            {
                SourceLocation selectorLoc = messageExpr->getSelectorLoc(i);
                _rewriter.ReplaceText(selectorLoc,
                                      selector.getNameForSlot(i).size(),
                                      "newSelector");
            }
        }
        return Base::VisitObjCMessageExpr(messageExpr);
    }

private:
    typedef clang::RecursiveASTVisitor<RewritingVisitor> Base;

    clang::SourceManager &_sourceManager;
    clang::Rewriter _rewriter;
};

} // end namespace clang_tooling
Источник

Я уверен, что то, что вы хотите, может быть как-то сделано непосредственно с помощью clang, но я не эксперт в этом. Это хорошее начало, если вы можете обнаружить все сообщения. Если бы я был тобой, я бы больше не сражался с лязгом. Я рекомендую вам, если вы не получите правильный ответ на свой первоначальный вопрос, чтобы обойти проблему, используя clang только для анализа, определения точных положений сообщений и выполнения замены самостоятельно. Все, что вам нужно сделать, это сохранить смещение столбца, когда вы делаете замены и корректировать исходные позиции с ним. НТН

Источник
Другие вопросы по тегам