contrib/llvm/tools/clang/lib/Index/SimpleFormatContext.h

   1 //===--- SimpleFormatContext.h ----------------------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 /// \file
  11 ///
  12 /// Defines a utility class for use of clang-format in libclang
  13 //
  14 //===----------------------------------------------------------------------===//
  15
  16 #ifndef LLVM_CLANG_LIB_INDEX_SIMPLEFORMATCONTEXT_H
  17 #define LLVM_CLANG_LIB_INDEX_SIMPLEFORMATCONTEXT_H
  18
  19 #include "clang/Basic/Diagnostic.h"
  20 #include "clang/Basic/DiagnosticOptions.h"
  21 #include "clang/Basic/FileManager.h"
  22 #include "clang/Basic/LangOptions.h"
  23 #include "clang/Basic/SourceManager.h"
  24 #include "clang/Rewrite/Core/Rewriter.h"
  25 #include "llvm/Support/FileSystem.h"
  26 #include "llvm/Support/Path.h"
  27 #include "llvm/Support/raw_ostream.h"
  28
  29 namespace clang {
  30 namespace index {
  31
  32 /// A small class to be used by libclang clients to format
  33 /// a declaration string in memory. This object is instantiated once
  34 /// and used each time a formatting is needed.
  35 class SimpleFormatContext {
  36 public:
  37   SimpleFormatContext(LangOptions Options)
  38       : DiagOpts(new DiagnosticOptions()),
  39         Diagnostics(new DiagnosticsEngine(new DiagnosticIDs,
  40                                           DiagOpts.get())),
  41         InMemoryFileSystem(new vfs::InMemoryFileSystem),
  42         Files(FileSystemOptions(), InMemoryFileSystem),
  43         Sources(*Diagnostics, Files),
  44         Rewrite(Sources, Options) {
  45     Diagnostics->setClient(new IgnoringDiagConsumer, true);
  46   }
  47
  48   FileID createInMemoryFile(StringRef Name, StringRef Content) {
  49     InMemoryFileSystem->addFile(Name, 0,
  50                                 llvm::MemoryBuffer::getMemBuffer(Content));
  51     const FileEntry *Entry = Files.getFile(Name);
  52     assert(Entry != nullptr);
  53     return Sources.createFileID(Entry, SourceLocation(), SrcMgr::C_User);
  54   }
  55
  56   std::string getRewrittenText(FileID ID) {
  57     std::string Result;
  58     llvm::raw_string_ostream OS(Result);
  59     Rewrite.getEditBuffer(ID).write(OS);
  60     OS.flush();
  61     return Result;
  62   }
  63
  64   IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts;
  65   IntrusiveRefCntPtr<DiagnosticsEngine> Diagnostics;
  66   IntrusiveRefCntPtr<vfs::InMemoryFileSystem> InMemoryFileSystem;
  67   FileManager Files;
  68   SourceManager Sources;
  69   Rewriter Rewrite;
  70 };
  71
  72 } // end namespace index
  73 } // end namespace clang
  74
  75 #endif