mirror of
https://github.com/github/codeql.git
synced 2025-12-21 11:16:30 +01:00
125aab8107
* visiting now happens in a later stage than fetching labels. While fetching a list of entities to be visited is created, and then acted upon in actual extraction. This partially flattens the recursive nature of `fetchLabel` into a loop inside `SwiftVisitor::extract`. Recursion in `fetchLabel` will only happen on labels fetched while naming an entity (calling into `SwiftMangler`). * The choice whether to name a declaration or type has been moved from the translators to `SwiftMangler`. Acting on this choice is contained in `SwiftDispatcher::createLabel`. * The choice whether to emit a body of a declaration has been moved from `DeclTranslator` to the dispatcher. This choice is also contained in `SwiftDispatcher::createLabel`. * The simple functionality of the `LabelStore` has been moved to the `SwiftDispatcher` as well.
65 lines
2.4 KiB
C++
65 lines
2.4 KiB
C++
#pragma once
|
|
|
|
#include "swift/extractor/infra/SwiftDispatcher.h"
|
|
#include "swift/extractor/translators/DeclTranslator.h"
|
|
#include "swift/extractor/translators/ExprTranslator.h"
|
|
#include "swift/extractor/translators/StmtTranslator.h"
|
|
#include "swift/extractor/translators/TypeTranslator.h"
|
|
#include "swift/extractor/translators/PatternTranslator.h"
|
|
#include "swift/extractor/mangler/SwiftMangler.h"
|
|
|
|
namespace codeql {
|
|
|
|
class SwiftVisitor : private SwiftDispatcher {
|
|
public:
|
|
using SwiftDispatcher::getEncounteredModules;
|
|
using SwiftDispatcher::SwiftDispatcher;
|
|
|
|
template <typename T>
|
|
void extract(const T& entity) {
|
|
fetchLabel(entity);
|
|
visitPending();
|
|
}
|
|
void extract(swift::Token& comment) { emitComment(comment); }
|
|
|
|
private:
|
|
SwiftMangledName name(const swift::Decl* decl) override { return mangler.mangleDecl(*decl); }
|
|
SwiftMangledName name(const swift::TypeBase* type) override { return mangler.mangleType(*type); }
|
|
void visit(const swift::Decl* decl) override { declTranslator.translateAndEmit(*decl); }
|
|
void visit(const swift::Stmt* stmt) override { stmtTranslator.translateAndEmit(*stmt); }
|
|
void visit(const swift::StmtCondition* cond) override { stmtTranslator.translateAndEmit(*cond); }
|
|
void visit(const swift::StmtConditionElement* element) override {
|
|
stmtTranslator.translateAndEmit(*element);
|
|
}
|
|
void visit(const swift::PoundAvailableInfo* availability) override {
|
|
stmtTranslator.translateAndEmit(*availability);
|
|
}
|
|
void visit(const swift::AvailabilitySpec* spec) override {
|
|
stmtTranslator.translateAndEmit(*spec);
|
|
}
|
|
|
|
void visit(const swift::CaseLabelItem* item) override { stmtTranslator.translateAndEmit(*item); }
|
|
void visit(const swift::Expr* expr) override { exprTranslator.translateAndEmit(*expr); }
|
|
void visit(const swift::Pattern* pattern) override {
|
|
patternTranslator.translateAndEmit(*pattern);
|
|
}
|
|
|
|
void visit(const swift::TypeBase* type) override { typeTranslator.translateAndEmit(*type); }
|
|
void visit(const swift::TypeRepr* typeRepr, swift::Type type) override {
|
|
typeTranslator.translateAndEmit(*typeRepr, type);
|
|
}
|
|
|
|
void visit(const swift::CapturedValue* capture) override {
|
|
declTranslator.translateAndEmit(*capture);
|
|
}
|
|
|
|
DeclTranslator declTranslator{*this};
|
|
ExprTranslator exprTranslator{*this};
|
|
StmtTranslator stmtTranslator{*this};
|
|
TypeTranslator typeTranslator{*this};
|
|
PatternTranslator patternTranslator{*this};
|
|
SwiftMangler mangler{*this};
|
|
};
|
|
|
|
} // namespace codeql
|