namespace clang
{
    class ASTConsumer;
    class CompilerInstance;
    class StringRef;
}

namespace Highlighter
{

class HighlighterFrontendAction : public clang::ASTFrontendAction
{
    virtual bool BeginInvocation(clang::CompilerInstance &CI) LLVM_OVERRIDE;
    virtual clang::ASTConsumer *CreateASTConsumer(clang::CompilerInstance &CI,
                                                  llvm::StringRef InFile) LLVM_OVERRIDE;
private:
    clang::Rewriter m_rewriter;
};

}

#endif
