explicit FunctionCallGraphVisitor(ASTContext *Context) : Context(Context) {}

bool VisitFunctionDecl(FunctionDecl *Func) {

if (!Func->hasBody()) return true;

currentFunc = Func->getNameAsString();

Stmt *Body = Func->getBody();

TraverseStmt(Body);

return true;

}

bool VisitCallExpr(CallExpr *Call) {

if (FunctionDecl *Callee = Call->getDirectCallee()) {

callGraph[currentFunc].insert(Callee->getNameAsString());

}

return true;

}

ASTContext *Context;

std::string currentFunc;

explicit CallGraphConsumer(ASTContext *Context) : Visitor(Context) {}

void HandleTranslationUnit(ASTContext &Context) override {

Visitor.TraverseDecl(Context.getTranslationUnitDecl());

for (auto &[caller, callees] : callGraph) {

for (const auto &callee : callees) {

std::cout << caller << " -> " << callee << std::endl;

}

}

}

FunctionCallGraphVisitor Visitor;

std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &CI, StringRef file) override {

return std::make_unique<CallGraphConsumer>(&CI.getASTContext());

}

auto ExpectedParser = CommonOptionsParser::create(argc, argv, ToolCategory);

if (!ExpectedParser) {

llvm::errs() << ExpectedParser.takeError();

return 1;

}

ClangTool Tool(ExpectedParser->getCompilations(), ExpectedParser->getSourcePathList());

return Tool.run(newFrontendActionFactory<CallGraphAction>().get());