Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 1 | //===- CIndexCodeCompletion.cpp - Code Completion API hooks ---------------===// |
| 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 | // This file implements the Clang-C Source Indexing library hooks for |
| 11 | // code completion. |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
| 15 | #include "CIndexer.h" |
Chandler Carruth | f59edb9 | 2012-12-04 09:25:21 +0000 | [diff] [blame] | 16 | #include "CIndexDiagnostic.h" |
Chandler Carruth | b1ba0ef | 2013-01-19 08:09:44 +0000 | [diff] [blame] | 17 | #include "CLog.h" |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 18 | #include "CXCursor.h" |
Ted Kremenek | baf82b0 | 2011-08-17 22:19:53 +0000 | [diff] [blame] | 19 | #include "CXString.h" |
Chandler Carruth | f59edb9 | 2012-12-04 09:25:21 +0000 | [diff] [blame] | 20 | #include "CXTranslationUnit.h" |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 21 | #include "clang/AST/Decl.h" |
| 22 | #include "clang/AST/DeclObjC.h" |
Chandler Carruth | f59edb9 | 2012-12-04 09:25:21 +0000 | [diff] [blame] | 23 | #include "clang/AST/Type.h" |
Benjamin Kramer | b846deb | 2010-04-12 19:45:50 +0000 | [diff] [blame] | 24 | #include "clang/Basic/FileManager.h" |
Chandler Carruth | f59edb9 | 2012-12-04 09:25:21 +0000 | [diff] [blame] | 25 | #include "clang/Basic/SourceManager.h" |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 26 | #include "clang/Frontend/ASTUnit.h" |
Benjamin Kramer | b846deb | 2010-04-12 19:45:50 +0000 | [diff] [blame] | 27 | #include "clang/Frontend/CompilerInstance.h" |
Douglas Gregor | 936ea3b | 2010-01-28 00:56:43 +0000 | [diff] [blame] | 28 | #include "clang/Frontend/FrontendDiagnostic.h" |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 29 | #include "clang/Sema/CodeCompleteConsumer.h" |
David Blaikie | 0b5ca51 | 2013-09-13 18:32:52 +0000 | [diff] [blame] | 30 | #include "clang/Sema/Sema.h" |
Douglas Gregor | df95a13 | 2010-08-09 20:45:32 +0000 | [diff] [blame] | 31 | #include "llvm/ADT/SmallString.h" |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 32 | #include "llvm/ADT/StringExtras.h" |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 33 | #include "llvm/Support/CrashRecoveryContext.h" |
Rafael Espindola | dbb4a35 | 2013-06-26 12:54:23 +0000 | [diff] [blame] | 34 | #include "llvm/Support/FileSystem.h" |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 35 | #include "llvm/Support/MemoryBuffer.h" |
Chandler Carruth | f59edb9 | 2012-12-04 09:25:21 +0000 | [diff] [blame] | 36 | #include "llvm/Support/Program.h" |
Douglas Gregor | df95a13 | 2010-08-09 20:45:32 +0000 | [diff] [blame] | 37 | #include "llvm/Support/Timer.h" |
| 38 | #include "llvm/Support/raw_ostream.h" |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 39 | #include <atomic> |
Douglas Gregor | 3d398aa | 2010-07-26 16:29:14 +0000 | [diff] [blame] | 40 | #include <cstdio> |
Chandler Carruth | f59edb9 | 2012-12-04 09:25:21 +0000 | [diff] [blame] | 41 | #include <cstdlib> |
Dmitri Gribenko | 90b5ac6 | 2013-02-07 19:09:22 +0000 | [diff] [blame] | 42 | #include <string> |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 43 | |
Douglas Gregor | df95a13 | 2010-08-09 20:45:32 +0000 | [diff] [blame] | 44 | |
Ted Kremenek | da7af32 | 2010-04-15 01:02:28 +0000 | [diff] [blame] | 45 | #ifdef UDP_CODE_COMPLETION_LOGGER |
| 46 | #include "clang/Basic/Version.h" |
Ted Kremenek | da7af32 | 2010-04-15 01:02:28 +0000 | [diff] [blame] | 47 | #include <arpa/inet.h> |
| 48 | #include <sys/socket.h> |
| 49 | #include <sys/types.h> |
| 50 | #include <unistd.h> |
| 51 | #endif |
| 52 | |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 53 | using namespace clang; |
Argyrios Kyrtzidis | c6f5c6a | 2013-01-10 18:54:52 +0000 | [diff] [blame] | 54 | using namespace clang::cxindex; |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 55 | |
| 56 | extern "C" { |
| 57 | |
| 58 | enum CXCompletionChunkKind |
| 59 | clang_getCompletionChunkKind(CXCompletionString completion_string, |
| 60 | unsigned chunk_number) { |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 61 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 62 | if (!CCStr || chunk_number >= CCStr->size()) |
| 63 | return CXCompletionChunk_Text; |
| 64 | |
| 65 | switch ((*CCStr)[chunk_number].Kind) { |
| 66 | case CodeCompletionString::CK_TypedText: |
| 67 | return CXCompletionChunk_TypedText; |
| 68 | case CodeCompletionString::CK_Text: |
| 69 | return CXCompletionChunk_Text; |
| 70 | case CodeCompletionString::CK_Optional: |
| 71 | return CXCompletionChunk_Optional; |
| 72 | case CodeCompletionString::CK_Placeholder: |
| 73 | return CXCompletionChunk_Placeholder; |
| 74 | case CodeCompletionString::CK_Informative: |
| 75 | return CXCompletionChunk_Informative; |
| 76 | case CodeCompletionString::CK_ResultType: |
| 77 | return CXCompletionChunk_ResultType; |
| 78 | case CodeCompletionString::CK_CurrentParameter: |
| 79 | return CXCompletionChunk_CurrentParameter; |
| 80 | case CodeCompletionString::CK_LeftParen: |
| 81 | return CXCompletionChunk_LeftParen; |
| 82 | case CodeCompletionString::CK_RightParen: |
| 83 | return CXCompletionChunk_RightParen; |
| 84 | case CodeCompletionString::CK_LeftBracket: |
| 85 | return CXCompletionChunk_LeftBracket; |
| 86 | case CodeCompletionString::CK_RightBracket: |
| 87 | return CXCompletionChunk_RightBracket; |
| 88 | case CodeCompletionString::CK_LeftBrace: |
| 89 | return CXCompletionChunk_LeftBrace; |
| 90 | case CodeCompletionString::CK_RightBrace: |
| 91 | return CXCompletionChunk_RightBrace; |
| 92 | case CodeCompletionString::CK_LeftAngle: |
| 93 | return CXCompletionChunk_LeftAngle; |
| 94 | case CodeCompletionString::CK_RightAngle: |
| 95 | return CXCompletionChunk_RightAngle; |
| 96 | case CodeCompletionString::CK_Comma: |
| 97 | return CXCompletionChunk_Comma; |
Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 98 | case CodeCompletionString::CK_Colon: |
| 99 | return CXCompletionChunk_Colon; |
| 100 | case CodeCompletionString::CK_SemiColon: |
| 101 | return CXCompletionChunk_SemiColon; |
| 102 | case CodeCompletionString::CK_Equal: |
| 103 | return CXCompletionChunk_Equal; |
| 104 | case CodeCompletionString::CK_HorizontalSpace: |
| 105 | return CXCompletionChunk_HorizontalSpace; |
| 106 | case CodeCompletionString::CK_VerticalSpace: |
| 107 | return CXCompletionChunk_VerticalSpace; |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 108 | } |
| 109 | |
David Blaikie | 7530c03 | 2012-01-17 06:56:22 +0000 | [diff] [blame] | 110 | llvm_unreachable("Invalid CompletionKind!"); |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 111 | } |
| 112 | |
Ted Kremenek | 2ef6f8f | 2010-02-17 01:42:24 +0000 | [diff] [blame] | 113 | CXString clang_getCompletionChunkText(CXCompletionString completion_string, |
| 114 | unsigned chunk_number) { |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 115 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 116 | if (!CCStr || chunk_number >= CCStr->size()) |
Dmitri Gribenko | dad4c1a | 2013-02-01 14:13:32 +0000 | [diff] [blame] | 117 | return cxstring::createNull(); |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 118 | |
| 119 | switch ((*CCStr)[chunk_number].Kind) { |
| 120 | case CodeCompletionString::CK_TypedText: |
| 121 | case CodeCompletionString::CK_Text: |
| 122 | case CodeCompletionString::CK_Placeholder: |
| 123 | case CodeCompletionString::CK_CurrentParameter: |
| 124 | case CodeCompletionString::CK_Informative: |
| 125 | case CodeCompletionString::CK_LeftParen: |
| 126 | case CodeCompletionString::CK_RightParen: |
| 127 | case CodeCompletionString::CK_LeftBracket: |
| 128 | case CodeCompletionString::CK_RightBracket: |
| 129 | case CodeCompletionString::CK_LeftBrace: |
| 130 | case CodeCompletionString::CK_RightBrace: |
| 131 | case CodeCompletionString::CK_LeftAngle: |
| 132 | case CodeCompletionString::CK_RightAngle: |
| 133 | case CodeCompletionString::CK_Comma: |
| 134 | case CodeCompletionString::CK_ResultType: |
Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 135 | case CodeCompletionString::CK_Colon: |
| 136 | case CodeCompletionString::CK_SemiColon: |
| 137 | case CodeCompletionString::CK_Equal: |
| 138 | case CodeCompletionString::CK_HorizontalSpace: |
Douglas Gregor | 21c241f | 2010-05-25 06:14:46 +0000 | [diff] [blame] | 139 | case CodeCompletionString::CK_VerticalSpace: |
Dmitri Gribenko | 0c4394c | 2013-02-02 00:02:12 +0000 | [diff] [blame] | 140 | return cxstring::createRef((*CCStr)[chunk_number].Text); |
Douglas Gregor | 21c241f | 2010-05-25 06:14:46 +0000 | [diff] [blame] | 141 | |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 142 | case CodeCompletionString::CK_Optional: |
| 143 | // Note: treated as an empty text block. |
Dmitri Gribenko | dc66adb | 2013-02-01 14:21:22 +0000 | [diff] [blame] | 144 | return cxstring::createEmpty(); |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 145 | } |
| 146 | |
David Blaikie | 3026348 | 2012-01-20 21:50:17 +0000 | [diff] [blame] | 147 | llvm_unreachable("Invalid CodeCompletionString Kind!"); |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 148 | } |
| 149 | |
Ted Kremenek | 2ef6f8f | 2010-02-17 01:42:24 +0000 | [diff] [blame] | 150 | |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 151 | CXCompletionString |
| 152 | clang_getCompletionChunkCompletionString(CXCompletionString completion_string, |
| 153 | unsigned chunk_number) { |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 154 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 155 | if (!CCStr || chunk_number >= CCStr->size()) |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 156 | return nullptr; |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 157 | |
| 158 | switch ((*CCStr)[chunk_number].Kind) { |
| 159 | case CodeCompletionString::CK_TypedText: |
| 160 | case CodeCompletionString::CK_Text: |
| 161 | case CodeCompletionString::CK_Placeholder: |
| 162 | case CodeCompletionString::CK_CurrentParameter: |
| 163 | case CodeCompletionString::CK_Informative: |
| 164 | case CodeCompletionString::CK_LeftParen: |
| 165 | case CodeCompletionString::CK_RightParen: |
| 166 | case CodeCompletionString::CK_LeftBracket: |
| 167 | case CodeCompletionString::CK_RightBracket: |
| 168 | case CodeCompletionString::CK_LeftBrace: |
| 169 | case CodeCompletionString::CK_RightBrace: |
| 170 | case CodeCompletionString::CK_LeftAngle: |
| 171 | case CodeCompletionString::CK_RightAngle: |
| 172 | case CodeCompletionString::CK_Comma: |
| 173 | case CodeCompletionString::CK_ResultType: |
Douglas Gregor | 01dfea0 | 2010-01-10 23:08:15 +0000 | [diff] [blame] | 174 | case CodeCompletionString::CK_Colon: |
| 175 | case CodeCompletionString::CK_SemiColon: |
| 176 | case CodeCompletionString::CK_Equal: |
| 177 | case CodeCompletionString::CK_HorizontalSpace: |
| 178 | case CodeCompletionString::CK_VerticalSpace: |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 179 | return nullptr; |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 180 | |
| 181 | case CodeCompletionString::CK_Optional: |
| 182 | // Note: treated as an empty text block. |
| 183 | return (*CCStr)[chunk_number].Optional; |
| 184 | } |
| 185 | |
David Blaikie | 7530c03 | 2012-01-17 06:56:22 +0000 | [diff] [blame] | 186 | llvm_unreachable("Invalid CompletionKind!"); |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 187 | } |
| 188 | |
| 189 | unsigned clang_getNumCompletionChunks(CXCompletionString completion_string) { |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 190 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 191 | return CCStr? CCStr->size() : 0; |
| 192 | } |
| 193 | |
Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 194 | unsigned clang_getCompletionPriority(CXCompletionString completion_string) { |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 195 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
Bill Wendling | a2ace58 | 2010-05-27 18:35:05 +0000 | [diff] [blame] | 196 | return CCStr? CCStr->getPriority() : unsigned(CCP_Unlikely); |
Douglas Gregor | 12e1313 | 2010-05-26 22:00:08 +0000 | [diff] [blame] | 197 | } |
| 198 | |
Douglas Gregor | 58ddb60 | 2010-08-23 23:00:57 +0000 | [diff] [blame] | 199 | enum CXAvailabilityKind |
| 200 | clang_getCompletionAvailability(CXCompletionString completion_string) { |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 201 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
| 202 | return CCStr? static_cast<CXAvailabilityKind>(CCStr->getAvailability()) |
| 203 | : CXAvailability_Available; |
Douglas Gregor | 58ddb60 | 2010-08-23 23:00:57 +0000 | [diff] [blame] | 204 | } |
| 205 | |
Erik Verbruggen | 6164ea1 | 2011-10-14 15:31:08 +0000 | [diff] [blame] | 206 | unsigned clang_getCompletionNumAnnotations(CXCompletionString completion_string) |
| 207 | { |
| 208 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
| 209 | return CCStr ? CCStr->getAnnotationCount() : 0; |
| 210 | } |
| 211 | |
| 212 | CXString clang_getCompletionAnnotation(CXCompletionString completion_string, |
| 213 | unsigned annotation_number) { |
| 214 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
Dmitri Gribenko | 0c4394c | 2013-02-02 00:02:12 +0000 | [diff] [blame] | 215 | return CCStr ? cxstring::createRef(CCStr->getAnnotation(annotation_number)) |
Dmitri Gribenko | dad4c1a | 2013-02-01 14:13:32 +0000 | [diff] [blame] | 216 | : cxstring::createNull(); |
Erik Verbruggen | 6164ea1 | 2011-10-14 15:31:08 +0000 | [diff] [blame] | 217 | } |
| 218 | |
Douglas Gregor | ba10306 | 2012-03-27 23:34:16 +0000 | [diff] [blame] | 219 | CXString |
| 220 | clang_getCompletionParent(CXCompletionString completion_string, |
| 221 | CXCursorKind *kind) { |
| 222 | if (kind) |
| 223 | *kind = CXCursor_NotImplemented; |
| 224 | |
| 225 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
| 226 | if (!CCStr) |
Dmitri Gribenko | dad4c1a | 2013-02-01 14:13:32 +0000 | [diff] [blame] | 227 | return cxstring::createNull(); |
Douglas Gregor | ba10306 | 2012-03-27 23:34:16 +0000 | [diff] [blame] | 228 | |
Dmitri Gribenko | 5595ded | 2013-02-02 02:19:29 +0000 | [diff] [blame] | 229 | return cxstring::createRef(CCStr->getParentContextName()); |
Douglas Gregor | ba10306 | 2012-03-27 23:34:16 +0000 | [diff] [blame] | 230 | } |
Dmitri Gribenko | d99ef53 | 2012-07-02 17:35:10 +0000 | [diff] [blame] | 231 | |
| 232 | CXString |
| 233 | clang_getCompletionBriefComment(CXCompletionString completion_string) { |
| 234 | CodeCompletionString *CCStr = (CodeCompletionString *)completion_string; |
| 235 | |
| 236 | if (!CCStr) |
Dmitri Gribenko | dad4c1a | 2013-02-01 14:13:32 +0000 | [diff] [blame] | 237 | return cxstring::createNull(); |
Dmitri Gribenko | d99ef53 | 2012-07-02 17:35:10 +0000 | [diff] [blame] | 238 | |
Dmitri Gribenko | 0c4394c | 2013-02-02 00:02:12 +0000 | [diff] [blame] | 239 | return cxstring::createRef(CCStr->getBriefComment()); |
Dmitri Gribenko | d99ef53 | 2012-07-02 17:35:10 +0000 | [diff] [blame] | 240 | } |
| 241 | |
Benjamin Kramer | 4d9f4e5 | 2012-11-22 15:02:44 +0000 | [diff] [blame] | 242 | namespace { |
| 243 | |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 244 | /// \brief The CXCodeCompleteResults structure we allocate internally; |
| 245 | /// the client only sees the initial CXCodeCompleteResults structure. |
Dmitri Gribenko | 90b5ac6 | 2013-02-07 19:09:22 +0000 | [diff] [blame] | 246 | /// |
| 247 | /// Normally, clients of CXString shouldn't care whether or not a CXString is |
| 248 | /// managed by a pool or by explicitly malloc'ed memory. But |
| 249 | /// AllocatedCXCodeCompleteResults outlives the CXTranslationUnit, so we can |
| 250 | /// not rely on the StringPool in the TU. |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 251 | struct AllocatedCXCodeCompleteResults : public CXCodeCompleteResults { |
Stephen Hines | 6bcf27b | 2014-05-29 04:14:42 -0700 | [diff] [blame] | 252 | AllocatedCXCodeCompleteResults(IntrusiveRefCntPtr<FileManager> FileMgr); |
Douglas Gregor | 0a812cf | 2010-02-18 23:07:20 +0000 | [diff] [blame] | 253 | ~AllocatedCXCodeCompleteResults(); |
| 254 | |
Douglas Gregor | a88084b | 2010-02-18 18:08:43 +0000 | [diff] [blame] | 255 | /// \brief Diagnostics produced while performing code completion. |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 256 | SmallVector<StoredDiagnostic, 8> Diagnostics; |
Douglas Gregor | a88084b | 2010-02-18 18:08:43 +0000 | [diff] [blame] | 257 | |
Stephen Hines | 6bcf27b | 2014-05-29 04:14:42 -0700 | [diff] [blame] | 258 | /// \brief Allocated API-exposed wrappters for Diagnostics. |
| 259 | SmallVector<CXStoredDiagnostic *, 8> DiagnosticsWrappers; |
| 260 | |
Douglas Gregor | 02c23eb | 2012-10-23 22:26:28 +0000 | [diff] [blame] | 261 | IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts; |
| 262 | |
Douglas Gregor | f715ca1 | 2010-03-16 00:06:06 +0000 | [diff] [blame] | 263 | /// \brief Diag object |
Dylan Noblesmith | c93dc78 | 2012-02-20 14:00:23 +0000 | [diff] [blame] | 264 | IntrusiveRefCntPtr<DiagnosticsEngine> Diag; |
Douglas Gregor | f715ca1 | 2010-03-16 00:06:06 +0000 | [diff] [blame] | 265 | |
Douglas Gregor | a88084b | 2010-02-18 18:08:43 +0000 | [diff] [blame] | 266 | /// \brief Language options used to adjust source locations. |
Daniel Dunbar | 35b8440 | 2010-01-30 23:31:40 +0000 | [diff] [blame] | 267 | LangOptions LangOpts; |
Argyrios Kyrtzidis | 389db16 | 2010-11-03 22:45:23 +0000 | [diff] [blame] | 268 | |
Chris Lattner | 39b49bc | 2010-11-23 08:35:12 +0000 | [diff] [blame] | 269 | /// \brief File manager, used for diagnostics. |
Dylan Noblesmith | c93dc78 | 2012-02-20 14:00:23 +0000 | [diff] [blame] | 270 | IntrusiveRefCntPtr<FileManager> FileMgr; |
Chris Lattner | 39b49bc | 2010-11-23 08:35:12 +0000 | [diff] [blame] | 271 | |
Argyrios Kyrtzidis | 389db16 | 2010-11-03 22:45:23 +0000 | [diff] [blame] | 272 | /// \brief Source manager, used for diagnostics. |
Dylan Noblesmith | c93dc78 | 2012-02-20 14:00:23 +0000 | [diff] [blame] | 273 | IntrusiveRefCntPtr<SourceManager> SourceMgr; |
Douglas Gregor | 313e26c | 2010-02-18 23:35:40 +0000 | [diff] [blame] | 274 | |
| 275 | /// \brief Temporary files that should be removed once we have finished |
| 276 | /// with the code-completion results. |
Rafael Espindola | dbb4a35 | 2013-06-26 12:54:23 +0000 | [diff] [blame] | 277 | std::vector<std::string> TemporaryFiles; |
Douglas Gregor | b75d3df | 2010-08-04 17:07:00 +0000 | [diff] [blame] | 278 | |
Chris Lattner | 7ad97ff | 2010-11-23 07:51:02 +0000 | [diff] [blame] | 279 | /// \brief Temporary buffers that will be deleted once we have finished with |
| 280 | /// the code-completion results. |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 281 | SmallVector<const llvm::MemoryBuffer *, 1> TemporaryBuffers; |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 282 | |
Douglas Gregor | 48601b3 | 2011-02-16 19:08:06 +0000 | [diff] [blame] | 283 | /// \brief Allocator used to store globally cached code-completion results. |
Dylan Noblesmith | c93dc78 | 2012-02-20 14:00:23 +0000 | [diff] [blame] | 284 | IntrusiveRefCntPtr<clang::GlobalCodeCompletionAllocator> |
Douglas Gregor | 48601b3 | 2011-02-16 19:08:06 +0000 | [diff] [blame] | 285 | CachedCompletionAllocator; |
| 286 | |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 287 | /// \brief Allocator used to store code completion results. |
Argyrios Kyrtzidis | 28a83f5 | 2012-04-10 17:23:48 +0000 | [diff] [blame] | 288 | IntrusiveRefCntPtr<clang::GlobalCodeCompletionAllocator> |
| 289 | CodeCompletionAllocator; |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 290 | |
| 291 | /// \brief Context under which completion occurred. |
| 292 | enum clang::CodeCompletionContext::Kind ContextKind; |
| 293 | |
| 294 | /// \brief A bitfield representing the acceptable completions for the |
| 295 | /// current context. |
| 296 | unsigned long long Contexts; |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 297 | |
Douglas Gregor | 0a47d69 | 2011-07-26 15:24:30 +0000 | [diff] [blame] | 298 | /// \brief The kind of the container for the current context for completions. |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 299 | enum CXCursorKind ContainerKind; |
Dmitri Gribenko | 90b5ac6 | 2013-02-07 19:09:22 +0000 | [diff] [blame] | 300 | |
Douglas Gregor | 0a47d69 | 2011-07-26 15:24:30 +0000 | [diff] [blame] | 301 | /// \brief The USR of the container for the current context for completions. |
Dmitri Gribenko | 90b5ac6 | 2013-02-07 19:09:22 +0000 | [diff] [blame] | 302 | std::string ContainerUSR; |
| 303 | |
Douglas Gregor | 0a47d69 | 2011-07-26 15:24:30 +0000 | [diff] [blame] | 304 | /// \brief a boolean value indicating whether there is complete information |
| 305 | /// about the container |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 306 | unsigned ContainerIsIncomplete; |
Douglas Gregor | 0a47d69 | 2011-07-26 15:24:30 +0000 | [diff] [blame] | 307 | |
| 308 | /// \brief A string containing the Objective-C selector entered thus far for a |
| 309 | /// message send. |
| 310 | std::string Selector; |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 311 | }; |
| 312 | |
Benjamin Kramer | 4d9f4e5 | 2012-11-22 15:02:44 +0000 | [diff] [blame] | 313 | } // end anonymous namespace |
| 314 | |
Douglas Gregor | e3c60a7 | 2010-11-17 00:13:31 +0000 | [diff] [blame] | 315 | /// \brief Tracks the number of code-completion result objects that are |
| 316 | /// currently active. |
| 317 | /// |
| 318 | /// Used for debugging purposes only. |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 319 | static std::atomic<unsigned> CodeCompletionResultObjects; |
Douglas Gregor | e3c60a7 | 2010-11-17 00:13:31 +0000 | [diff] [blame] | 320 | |
Anders Carlsson | 0d8d7e6 | 2011-03-18 18:22:40 +0000 | [diff] [blame] | 321 | AllocatedCXCodeCompleteResults::AllocatedCXCodeCompleteResults( |
Stephen Hines | 6bcf27b | 2014-05-29 04:14:42 -0700 | [diff] [blame] | 322 | IntrusiveRefCntPtr<FileManager> FileMgr) |
| 323 | : CXCodeCompleteResults(), |
| 324 | DiagOpts(new DiagnosticOptions), |
| 325 | Diag(new DiagnosticsEngine( |
| 326 | IntrusiveRefCntPtr<DiagnosticIDs>(new DiagnosticIDs), &*DiagOpts)), |
| 327 | FileMgr(FileMgr), SourceMgr(new SourceManager(*Diag, *FileMgr)), |
| 328 | CodeCompletionAllocator(new clang::GlobalCodeCompletionAllocator), |
| 329 | Contexts(CXCompletionContext_Unknown), |
| 330 | ContainerKind(CXCursor_InvalidCode), ContainerIsIncomplete(1) { |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 331 | if (getenv("LIBCLANG_OBJTRACKING")) |
| 332 | fprintf(stderr, "+++ %u completion results\n", |
| 333 | ++CodeCompletionResultObjects); |
Douglas Gregor | e3c60a7 | 2010-11-17 00:13:31 +0000 | [diff] [blame] | 334 | } |
Douglas Gregor | 0a812cf | 2010-02-18 23:07:20 +0000 | [diff] [blame] | 335 | |
| 336 | AllocatedCXCodeCompleteResults::~AllocatedCXCodeCompleteResults() { |
Stephen Hines | 6bcf27b | 2014-05-29 04:14:42 -0700 | [diff] [blame] | 337 | llvm::DeleteContainerPointers(DiagnosticsWrappers); |
Douglas Gregor | 0a812cf | 2010-02-18 23:07:20 +0000 | [diff] [blame] | 338 | delete [] Results; |
Dmitri Gribenko | 90b5ac6 | 2013-02-07 19:09:22 +0000 | [diff] [blame] | 339 | |
Douglas Gregor | 313e26c | 2010-02-18 23:35:40 +0000 | [diff] [blame] | 340 | for (unsigned I = 0, N = TemporaryFiles.size(); I != N; ++I) |
Rafael Espindola | dbb4a35 | 2013-06-26 12:54:23 +0000 | [diff] [blame] | 341 | llvm::sys::fs::remove(TemporaryFiles[I]); |
Douglas Gregor | b75d3df | 2010-08-04 17:07:00 +0000 | [diff] [blame] | 342 | for (unsigned I = 0, N = TemporaryBuffers.size(); I != N; ++I) |
| 343 | delete TemporaryBuffers[I]; |
Douglas Gregor | e3c60a7 | 2010-11-17 00:13:31 +0000 | [diff] [blame] | 344 | |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 345 | if (getenv("LIBCLANG_OBJTRACKING")) |
| 346 | fprintf(stderr, "--- %u completion results\n", |
| 347 | --CodeCompletionResultObjects); |
Douglas Gregor | 0a812cf | 2010-02-18 23:07:20 +0000 | [diff] [blame] | 348 | } |
| 349 | |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 350 | } // end extern "C" |
| 351 | |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 352 | static unsigned long long getContextsForContextKind( |
| 353 | enum CodeCompletionContext::Kind kind, |
| 354 | Sema &S) { |
| 355 | unsigned long long contexts = 0; |
| 356 | switch (kind) { |
| 357 | case CodeCompletionContext::CCC_OtherWithMacros: { |
| 358 | //We can allow macros here, but we don't know what else is permissible |
| 359 | //So we'll say the only thing permissible are macros |
| 360 | contexts = CXCompletionContext_MacroName; |
| 361 | break; |
| 362 | } |
| 363 | case CodeCompletionContext::CCC_TopLevel: |
| 364 | case CodeCompletionContext::CCC_ObjCIvarList: |
| 365 | case CodeCompletionContext::CCC_ClassStructUnion: |
| 366 | case CodeCompletionContext::CCC_Type: { |
| 367 | contexts = CXCompletionContext_AnyType | |
| 368 | CXCompletionContext_ObjCInterface; |
David Blaikie | 4e4d084 | 2012-03-11 07:00:24 +0000 | [diff] [blame] | 369 | if (S.getLangOpts().CPlusPlus) { |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 370 | contexts |= CXCompletionContext_EnumTag | |
| 371 | CXCompletionContext_UnionTag | |
| 372 | CXCompletionContext_StructTag | |
| 373 | CXCompletionContext_ClassTag | |
| 374 | CXCompletionContext_NestedNameSpecifier; |
| 375 | } |
| 376 | break; |
| 377 | } |
| 378 | case CodeCompletionContext::CCC_Statement: { |
| 379 | contexts = CXCompletionContext_AnyType | |
| 380 | CXCompletionContext_ObjCInterface | |
| 381 | CXCompletionContext_AnyValue; |
David Blaikie | 4e4d084 | 2012-03-11 07:00:24 +0000 | [diff] [blame] | 382 | if (S.getLangOpts().CPlusPlus) { |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 383 | contexts |= CXCompletionContext_EnumTag | |
| 384 | CXCompletionContext_UnionTag | |
| 385 | CXCompletionContext_StructTag | |
| 386 | CXCompletionContext_ClassTag | |
| 387 | CXCompletionContext_NestedNameSpecifier; |
| 388 | } |
| 389 | break; |
| 390 | } |
| 391 | case CodeCompletionContext::CCC_Expression: { |
| 392 | contexts = CXCompletionContext_AnyValue; |
David Blaikie | 4e4d084 | 2012-03-11 07:00:24 +0000 | [diff] [blame] | 393 | if (S.getLangOpts().CPlusPlus) { |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 394 | contexts |= CXCompletionContext_AnyType | |
| 395 | CXCompletionContext_ObjCInterface | |
| 396 | CXCompletionContext_EnumTag | |
| 397 | CXCompletionContext_UnionTag | |
| 398 | CXCompletionContext_StructTag | |
| 399 | CXCompletionContext_ClassTag | |
| 400 | CXCompletionContext_NestedNameSpecifier; |
| 401 | } |
| 402 | break; |
| 403 | } |
| 404 | case CodeCompletionContext::CCC_ObjCMessageReceiver: { |
| 405 | contexts = CXCompletionContext_ObjCObjectValue | |
| 406 | CXCompletionContext_ObjCSelectorValue | |
| 407 | CXCompletionContext_ObjCInterface; |
David Blaikie | 4e4d084 | 2012-03-11 07:00:24 +0000 | [diff] [blame] | 408 | if (S.getLangOpts().CPlusPlus) { |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 409 | contexts |= CXCompletionContext_CXXClassTypeValue | |
| 410 | CXCompletionContext_AnyType | |
| 411 | CXCompletionContext_EnumTag | |
| 412 | CXCompletionContext_UnionTag | |
| 413 | CXCompletionContext_StructTag | |
| 414 | CXCompletionContext_ClassTag | |
| 415 | CXCompletionContext_NestedNameSpecifier; |
| 416 | } |
| 417 | break; |
| 418 | } |
| 419 | case CodeCompletionContext::CCC_DotMemberAccess: { |
| 420 | contexts = CXCompletionContext_DotMemberAccess; |
| 421 | break; |
| 422 | } |
| 423 | case CodeCompletionContext::CCC_ArrowMemberAccess: { |
| 424 | contexts = CXCompletionContext_ArrowMemberAccess; |
| 425 | break; |
| 426 | } |
| 427 | case CodeCompletionContext::CCC_ObjCPropertyAccess: { |
| 428 | contexts = CXCompletionContext_ObjCPropertyAccess; |
| 429 | break; |
| 430 | } |
| 431 | case CodeCompletionContext::CCC_EnumTag: { |
| 432 | contexts = CXCompletionContext_EnumTag | |
| 433 | CXCompletionContext_NestedNameSpecifier; |
| 434 | break; |
| 435 | } |
| 436 | case CodeCompletionContext::CCC_UnionTag: { |
| 437 | contexts = CXCompletionContext_UnionTag | |
| 438 | CXCompletionContext_NestedNameSpecifier; |
| 439 | break; |
| 440 | } |
| 441 | case CodeCompletionContext::CCC_ClassOrStructTag: { |
| 442 | contexts = CXCompletionContext_StructTag | |
| 443 | CXCompletionContext_ClassTag | |
| 444 | CXCompletionContext_NestedNameSpecifier; |
| 445 | break; |
| 446 | } |
| 447 | case CodeCompletionContext::CCC_ObjCProtocolName: { |
| 448 | contexts = CXCompletionContext_ObjCProtocol; |
| 449 | break; |
| 450 | } |
| 451 | case CodeCompletionContext::CCC_Namespace: { |
| 452 | contexts = CXCompletionContext_Namespace; |
| 453 | break; |
| 454 | } |
| 455 | case CodeCompletionContext::CCC_PotentiallyQualifiedName: { |
| 456 | contexts = CXCompletionContext_NestedNameSpecifier; |
| 457 | break; |
| 458 | } |
| 459 | case CodeCompletionContext::CCC_MacroNameUse: { |
| 460 | contexts = CXCompletionContext_MacroName; |
| 461 | break; |
| 462 | } |
| 463 | case CodeCompletionContext::CCC_NaturalLanguage: { |
| 464 | contexts = CXCompletionContext_NaturalLanguage; |
| 465 | break; |
| 466 | } |
| 467 | case CodeCompletionContext::CCC_SelectorName: { |
| 468 | contexts = CXCompletionContext_ObjCSelectorName; |
| 469 | break; |
| 470 | } |
| 471 | case CodeCompletionContext::CCC_ParenthesizedExpression: { |
| 472 | contexts = CXCompletionContext_AnyType | |
| 473 | CXCompletionContext_ObjCInterface | |
| 474 | CXCompletionContext_AnyValue; |
David Blaikie | 4e4d084 | 2012-03-11 07:00:24 +0000 | [diff] [blame] | 475 | if (S.getLangOpts().CPlusPlus) { |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 476 | contexts |= CXCompletionContext_EnumTag | |
| 477 | CXCompletionContext_UnionTag | |
| 478 | CXCompletionContext_StructTag | |
| 479 | CXCompletionContext_ClassTag | |
| 480 | CXCompletionContext_NestedNameSpecifier; |
| 481 | } |
| 482 | break; |
| 483 | } |
| 484 | case CodeCompletionContext::CCC_ObjCInstanceMessage: { |
| 485 | contexts = CXCompletionContext_ObjCInstanceMessage; |
| 486 | break; |
| 487 | } |
| 488 | case CodeCompletionContext::CCC_ObjCClassMessage: { |
| 489 | contexts = CXCompletionContext_ObjCClassMessage; |
| 490 | break; |
| 491 | } |
Douglas Gregor | 0f91c8c | 2011-07-30 06:55:39 +0000 | [diff] [blame] | 492 | case CodeCompletionContext::CCC_ObjCInterfaceName: { |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 493 | contexts = CXCompletionContext_ObjCInterface; |
| 494 | break; |
| 495 | } |
| 496 | case CodeCompletionContext::CCC_ObjCCategoryName: { |
| 497 | contexts = CXCompletionContext_ObjCCategory; |
| 498 | break; |
| 499 | } |
| 500 | case CodeCompletionContext::CCC_Other: |
| 501 | case CodeCompletionContext::CCC_ObjCInterface: |
| 502 | case CodeCompletionContext::CCC_ObjCImplementation: |
| 503 | case CodeCompletionContext::CCC_Name: |
| 504 | case CodeCompletionContext::CCC_MacroName: |
| 505 | case CodeCompletionContext::CCC_PreprocessorExpression: |
| 506 | case CodeCompletionContext::CCC_PreprocessorDirective: |
| 507 | case CodeCompletionContext::CCC_TypeQualifiers: { |
| 508 | //Only Clang results should be accepted, so we'll set all of the other |
| 509 | //context bits to 0 (i.e. the empty set) |
| 510 | contexts = CXCompletionContext_Unexposed; |
| 511 | break; |
| 512 | } |
| 513 | case CodeCompletionContext::CCC_Recovery: { |
| 514 | //We don't know what the current context is, so we'll return unknown |
| 515 | //This is the equivalent of setting all of the other context bits |
| 516 | contexts = CXCompletionContext_Unknown; |
| 517 | break; |
| 518 | } |
| 519 | } |
| 520 | return contexts; |
| 521 | } |
| 522 | |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 523 | namespace { |
| 524 | class CaptureCompletionResults : public CodeCompleteConsumer { |
| 525 | AllocatedCXCodeCompleteResults &AllocatedResults; |
Argyrios Kyrtzidis | 28a83f5 | 2012-04-10 17:23:48 +0000 | [diff] [blame] | 526 | CodeCompletionTUInfo CCTUInfo; |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 527 | SmallVector<CXCompletionResult, 16> StoredResults; |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 528 | CXTranslationUnit *TU; |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 529 | public: |
Dmitri Gribenko | d99ef53 | 2012-07-02 17:35:10 +0000 | [diff] [blame] | 530 | CaptureCompletionResults(const CodeCompleteOptions &Opts, |
| 531 | AllocatedCXCodeCompleteResults &Results, |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 532 | CXTranslationUnit *TranslationUnit) |
Dmitri Gribenko | d99ef53 | 2012-07-02 17:35:10 +0000 | [diff] [blame] | 533 | : CodeCompleteConsumer(Opts, false), |
Argyrios Kyrtzidis | 28a83f5 | 2012-04-10 17:23:48 +0000 | [diff] [blame] | 534 | AllocatedResults(Results), CCTUInfo(Results.CodeCompletionAllocator), |
| 535 | TU(TranslationUnit) { } |
Pirama Arumuga Nainar | 33337ca | 2015-05-06 11:48:57 -0700 | [diff] [blame] | 536 | ~CaptureCompletionResults() override { Finish(); } |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 537 | |
| 538 | void ProcessCodeCompleteResults(Sema &S, |
| 539 | CodeCompletionContext Context, |
| 540 | CodeCompletionResult *Results, |
| 541 | unsigned NumResults) override { |
Douglas Gregor | 32be4a5 | 2010-10-11 21:37:58 +0000 | [diff] [blame] | 542 | StoredResults.reserve(StoredResults.size() + NumResults); |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 543 | for (unsigned I = 0; I != NumResults; ++I) { |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 544 | CodeCompletionString *StoredCompletion |
Argyrios Kyrtzidis | 28a83f5 | 2012-04-10 17:23:48 +0000 | [diff] [blame] | 545 | = Results[I].CreateCodeCompletionString(S, getAllocator(), |
Dmitri Gribenko | d99ef53 | 2012-07-02 17:35:10 +0000 | [diff] [blame] | 546 | getCodeCompletionTUInfo(), |
| 547 | includeBriefComments()); |
Douglas Gregor | 32be4a5 | 2010-10-11 21:37:58 +0000 | [diff] [blame] | 548 | |
| 549 | CXCompletionResult R; |
| 550 | R.CursorKind = Results[I].CursorKind; |
| 551 | R.CompletionString = StoredCompletion; |
| 552 | StoredResults.push_back(R); |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 553 | } |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 554 | |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 555 | enum CodeCompletionContext::Kind contextKind = Context.getKind(); |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 556 | |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 557 | AllocatedResults.ContextKind = contextKind; |
| 558 | AllocatedResults.Contexts = getContextsForContextKind(contextKind, S); |
| 559 | |
Douglas Gregor | 0a47d69 | 2011-07-26 15:24:30 +0000 | [diff] [blame] | 560 | AllocatedResults.Selector = ""; |
Dmitri Gribenko | 5c1a9f3 | 2013-06-18 04:41:50 +0000 | [diff] [blame] | 561 | ArrayRef<IdentifierInfo *> SelIdents = Context.getSelIdents(); |
| 562 | for (ArrayRef<IdentifierInfo *>::iterator I = SelIdents.begin(), |
| 563 | E = SelIdents.end(); |
| 564 | I != E; ++I) { |
| 565 | if (IdentifierInfo *selIdent = *I) |
| 566 | AllocatedResults.Selector += selIdent->getName(); |
Dmitri Gribenko | 68effa0 | 2013-06-18 04:02:26 +0000 | [diff] [blame] | 567 | AllocatedResults.Selector += ":"; |
Douglas Gregor | 0a47d69 | 2011-07-26 15:24:30 +0000 | [diff] [blame] | 568 | } |
| 569 | |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 570 | QualType baseType = Context.getBaseType(); |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 571 | NamedDecl *D = nullptr; |
| 572 | |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 573 | if (!baseType.isNull()) { |
| 574 | // Get the declaration for a class/struct/union/enum type |
| 575 | if (const TagType *Tag = baseType->getAs<TagType>()) |
| 576 | D = Tag->getDecl(); |
| 577 | // Get the @interface declaration for a (possibly-qualified) Objective-C |
| 578 | // object pointer type, e.g., NSString* |
| 579 | else if (const ObjCObjectPointerType *ObjPtr = |
| 580 | baseType->getAs<ObjCObjectPointerType>()) |
| 581 | D = ObjPtr->getInterfaceDecl(); |
| 582 | // Get the @interface declaration for an Objective-C object type |
| 583 | else if (const ObjCObjectType *Obj = baseType->getAs<ObjCObjectType>()) |
| 584 | D = Obj->getInterface(); |
| 585 | // Get the class for a C++ injected-class-name |
| 586 | else if (const InjectedClassNameType *Injected = |
| 587 | baseType->getAs<InjectedClassNameType>()) |
| 588 | D = Injected->getDecl(); |
| 589 | } |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 590 | |
| 591 | if (D != nullptr) { |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 592 | CXCursor cursor = cxcursor::MakeCXCursor(D, *TU); |
Dmitri Gribenko | 90b5ac6 | 2013-02-07 19:09:22 +0000 | [diff] [blame] | 593 | |
| 594 | AllocatedResults.ContainerKind = clang_getCursorKind(cursor); |
| 595 | |
| 596 | CXString CursorUSR = clang_getCursorUSR(cursor); |
| 597 | AllocatedResults.ContainerUSR = clang_getCString(CursorUSR); |
| 598 | clang_disposeString(CursorUSR); |
| 599 | |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 600 | const Type *type = baseType.getTypePtrOrNull(); |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 601 | if (type) { |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 602 | AllocatedResults.ContainerIsIncomplete = type->isIncompleteType(); |
| 603 | } |
| 604 | else { |
| 605 | AllocatedResults.ContainerIsIncomplete = 1; |
| 606 | } |
| 607 | } |
| 608 | else { |
| 609 | AllocatedResults.ContainerKind = CXCursor_InvalidCode; |
Dmitri Gribenko | 90b5ac6 | 2013-02-07 19:09:22 +0000 | [diff] [blame] | 610 | AllocatedResults.ContainerUSR.clear(); |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 611 | AllocatedResults.ContainerIsIncomplete = 1; |
| 612 | } |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 613 | } |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 614 | |
| 615 | void ProcessOverloadCandidates(Sema &S, unsigned CurrentArg, |
| 616 | OverloadCandidate *Candidates, |
| 617 | unsigned NumCandidates) override { |
Douglas Gregor | 32be4a5 | 2010-10-11 21:37:58 +0000 | [diff] [blame] | 618 | StoredResults.reserve(StoredResults.size() + NumCandidates); |
| 619 | for (unsigned I = 0; I != NumCandidates; ++I) { |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 620 | CodeCompletionString *StoredCompletion |
Argyrios Kyrtzidis | 28a83f5 | 2012-04-10 17:23:48 +0000 | [diff] [blame] | 621 | = Candidates[I].CreateSignatureString(CurrentArg, S, getAllocator(), |
Stephen Hines | 0e2c34f | 2015-03-23 12:09:02 -0700 | [diff] [blame] | 622 | getCodeCompletionTUInfo(), |
| 623 | includeBriefComments()); |
Douglas Gregor | 32be4a5 | 2010-10-11 21:37:58 +0000 | [diff] [blame] | 624 | |
| 625 | CXCompletionResult R; |
Stephen Hines | 0e2c34f | 2015-03-23 12:09:02 -0700 | [diff] [blame] | 626 | R.CursorKind = CXCursor_OverloadCandidate; |
Douglas Gregor | 32be4a5 | 2010-10-11 21:37:58 +0000 | [diff] [blame] | 627 | R.CompletionString = StoredCompletion; |
| 628 | StoredResults.push_back(R); |
| 629 | } |
| 630 | } |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 631 | |
| 632 | CodeCompletionAllocator &getAllocator() override { |
Argyrios Kyrtzidis | 28a83f5 | 2012-04-10 17:23:48 +0000 | [diff] [blame] | 633 | return *AllocatedResults.CodeCompletionAllocator; |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 634 | } |
Argyrios Kyrtzidis | 28a83f5 | 2012-04-10 17:23:48 +0000 | [diff] [blame] | 635 | |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 636 | CodeCompletionTUInfo &getCodeCompletionTUInfo() override { return CCTUInfo;} |
| 637 | |
Douglas Gregor | 32be4a5 | 2010-10-11 21:37:58 +0000 | [diff] [blame] | 638 | private: |
| 639 | void Finish() { |
| 640 | AllocatedResults.Results = new CXCompletionResult [StoredResults.size()]; |
| 641 | AllocatedResults.NumResults = StoredResults.size(); |
| 642 | std::memcpy(AllocatedResults.Results, StoredResults.data(), |
| 643 | StoredResults.size() * sizeof(CXCompletionResult)); |
| 644 | StoredResults.clear(); |
| 645 | } |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 646 | }; |
| 647 | } |
| 648 | |
| 649 | extern "C" { |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 650 | struct CodeCompleteAtInfo { |
| 651 | CXTranslationUnit TU; |
| 652 | const char *complete_filename; |
| 653 | unsigned complete_line; |
| 654 | unsigned complete_column; |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 655 | ArrayRef<CXUnsavedFile> unsaved_files; |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 656 | unsigned options; |
| 657 | CXCodeCompleteResults *result; |
| 658 | }; |
Pirama Arumuga Nainar | 3ea9e33 | 2015-04-08 08:57:32 -0700 | [diff] [blame] | 659 | static void clang_codeCompleteAt_Impl(void *UserData) { |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 660 | CodeCompleteAtInfo *CCAI = static_cast<CodeCompleteAtInfo*>(UserData); |
| 661 | CXTranslationUnit TU = CCAI->TU; |
| 662 | const char *complete_filename = CCAI->complete_filename; |
| 663 | unsigned complete_line = CCAI->complete_line; |
| 664 | unsigned complete_column = CCAI->complete_column; |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 665 | unsigned options = CCAI->options; |
Dmitri Gribenko | d99ef53 | 2012-07-02 17:35:10 +0000 | [diff] [blame] | 666 | bool IncludeBriefComments = options & CXCodeComplete_IncludeBriefComments; |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 667 | CCAI->result = nullptr; |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 668 | |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 669 | #ifdef UDP_CODE_COMPLETION_LOGGER |
| 670 | #ifdef UDP_CODE_COMPLETION_LOGGER_PORT |
| 671 | const llvm::TimeRecord &StartTime = llvm::TimeRecord::getCurrentTime(); |
| 672 | #endif |
| 673 | #endif |
| 674 | |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 675 | bool EnableLogging = getenv("LIBCLANG_CODE_COMPLETION_LOGGING") != nullptr; |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 676 | |
| 677 | if (cxtu::isNotUsableTU(TU)) { |
| 678 | LOG_BAD_TU(TU); |
| 679 | return; |
| 680 | } |
| 681 | |
Dmitri Gribenko | 5694feb | 2013-01-26 18:53:38 +0000 | [diff] [blame] | 682 | ASTUnit *AST = cxtu::getASTUnit(TU); |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 683 | if (!AST) |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 684 | return; |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 685 | |
Dmitri Gribenko | 8c718e7 | 2013-01-26 21:49:50 +0000 | [diff] [blame] | 686 | CIndexer *CXXIdx = TU->CIdx; |
Argyrios Kyrtzidis | fdc1795 | 2012-03-28 02:18:05 +0000 | [diff] [blame] | 687 | if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing)) |
Argyrios Kyrtzidis | 81b5ac3 | 2012-03-28 02:49:54 +0000 | [diff] [blame] | 688 | setThreadBackgroundPriority(); |
Argyrios Kyrtzidis | fdc1795 | 2012-03-28 02:18:05 +0000 | [diff] [blame] | 689 | |
Douglas Gregor | 593b0c1 | 2010-09-23 18:47:53 +0000 | [diff] [blame] | 690 | ASTUnit::ConcurrencyCheck Check(*AST); |
| 691 | |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 692 | // Perform the remapping of source files. |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 693 | SmallVector<ASTUnit::RemappedFile, 4> RemappedFiles; |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 694 | |
| 695 | for (auto &UF : CCAI->unsaved_files) { |
Stephen Hines | 176edba | 2014-12-01 14:53:08 -0800 | [diff] [blame] | 696 | std::unique_ptr<llvm::MemoryBuffer> MB = |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 697 | llvm::MemoryBuffer::getMemBufferCopy(getContents(UF), UF.Filename); |
Stephen Hines | 176edba | 2014-12-01 14:53:08 -0800 | [diff] [blame] | 698 | RemappedFiles.push_back(std::make_pair(UF.Filename, MB.release())); |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 699 | } |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 700 | |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 701 | if (EnableLogging) { |
| 702 | // FIXME: Add logging. |
| 703 | } |
| 704 | |
| 705 | // Parse the resulting source file to find code-completion results. |
Stephen Hines | 6bcf27b | 2014-05-29 04:14:42 -0700 | [diff] [blame] | 706 | AllocatedCXCodeCompleteResults *Results = new AllocatedCXCodeCompleteResults( |
| 707 | &AST->getFileManager()); |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 708 | Results->Results = nullptr; |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 709 | Results->NumResults = 0; |
Douglas Gregor | 48601b3 | 2011-02-16 19:08:06 +0000 | [diff] [blame] | 710 | |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 711 | // Create a code-completion consumer to capture the results. |
Dmitri Gribenko | d99ef53 | 2012-07-02 17:35:10 +0000 | [diff] [blame] | 712 | CodeCompleteOptions Opts; |
| 713 | Opts.IncludeBriefComments = IncludeBriefComments; |
| 714 | CaptureCompletionResults Capture(Opts, *Results, &TU); |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 715 | |
| 716 | // Perform completion. |
| 717 | AST->CodeComplete(complete_filename, complete_line, complete_column, |
Stephen Hines | 651f13c | 2014-04-23 16:59:28 -0700 | [diff] [blame] | 718 | RemappedFiles, |
Douglas Gregor | cee235c | 2010-08-05 09:09:23 +0000 | [diff] [blame] | 719 | (options & CXCodeComplete_IncludeMacros), |
| 720 | (options & CXCodeComplete_IncludeCodePatterns), |
Dmitri Gribenko | d99ef53 | 2012-07-02 17:35:10 +0000 | [diff] [blame] | 721 | IncludeBriefComments, |
Douglas Gregor | cee235c | 2010-08-05 09:09:23 +0000 | [diff] [blame] | 722 | Capture, |
Ted Kremenek | 4f32786 | 2011-03-21 18:40:17 +0000 | [diff] [blame] | 723 | *Results->Diag, Results->LangOpts, *Results->SourceMgr, |
| 724 | *Results->FileMgr, Results->Diagnostics, |
Douglas Gregor | 2283d79 | 2010-08-20 00:59:43 +0000 | [diff] [blame] | 725 | Results->TemporaryBuffers); |
Stephen Hines | 6bcf27b | 2014-05-29 04:14:42 -0700 | [diff] [blame] | 726 | |
| 727 | Results->DiagnosticsWrappers.resize(Results->Diagnostics.size()); |
| 728 | |
Douglas Gregor | 48601b3 | 2011-02-16 19:08:06 +0000 | [diff] [blame] | 729 | // Keep a reference to the allocator used for cached global completions, so |
| 730 | // that we can be sure that the memory used by our code completion strings |
| 731 | // doesn't get freed due to subsequent reparses (while the code completion |
| 732 | // results are still active). |
| 733 | Results->CachedCompletionAllocator = AST->getCachedCompletionAllocator(); |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 734 | |
| 735 | |
| 736 | |
| 737 | #ifdef UDP_CODE_COMPLETION_LOGGER |
| 738 | #ifdef UDP_CODE_COMPLETION_LOGGER_PORT |
| 739 | const llvm::TimeRecord &EndTime = llvm::TimeRecord::getCurrentTime(); |
Dylan Noblesmith | 36d5927 | 2012-02-13 12:32:26 +0000 | [diff] [blame] | 740 | SmallString<256> LogResult; |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 741 | llvm::raw_svector_ostream os(LogResult); |
| 742 | |
| 743 | // Figure out the language and whether or not it uses PCH. |
| 744 | const char *lang = 0; |
| 745 | bool usesPCH = false; |
| 746 | |
| 747 | for (std::vector<const char*>::iterator I = argv.begin(), E = argv.end(); |
| 748 | I != E; ++I) { |
| 749 | if (*I == 0) |
| 750 | continue; |
| 751 | if (strcmp(*I, "-x") == 0) { |
| 752 | if (I + 1 != E) { |
| 753 | lang = *(++I); |
| 754 | continue; |
| 755 | } |
| 756 | } |
| 757 | else if (strcmp(*I, "-include") == 0) { |
| 758 | if (I+1 != E) { |
| 759 | const char *arg = *(++I); |
Dylan Noblesmith | 36d5927 | 2012-02-13 12:32:26 +0000 | [diff] [blame] | 760 | SmallString<512> pchName; |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 761 | { |
| 762 | llvm::raw_svector_ostream os(pchName); |
| 763 | os << arg << ".pth"; |
| 764 | } |
| 765 | pchName.push_back('\0'); |
| 766 | struct stat stat_results; |
Douglas Gregor | 0a47d69 | 2011-07-26 15:24:30 +0000 | [diff] [blame] | 767 | if (stat(pchName.str().c_str(), &stat_results) == 0) |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 768 | usesPCH = true; |
| 769 | continue; |
| 770 | } |
| 771 | } |
| 772 | } |
| 773 | |
| 774 | os << "{ "; |
| 775 | os << "\"wall\": " << (EndTime.getWallTime() - StartTime.getWallTime()); |
| 776 | os << ", \"numRes\": " << Results->NumResults; |
| 777 | os << ", \"diags\": " << Results->Diagnostics.size(); |
| 778 | os << ", \"pch\": " << (usesPCH ? "true" : "false"); |
| 779 | os << ", \"lang\": \"" << (lang ? lang : "<unknown>") << '"'; |
| 780 | const char *name = getlogin(); |
| 781 | os << ", \"user\": \"" << (name ? name : "unknown") << '"'; |
| 782 | os << ", \"clangVer\": \"" << getClangFullVersion() << '"'; |
| 783 | os << " }"; |
| 784 | |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 785 | StringRef res = os.str(); |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 786 | if (res.size() > 0) { |
| 787 | do { |
| 788 | // Setup the UDP socket. |
| 789 | struct sockaddr_in servaddr; |
| 790 | bzero(&servaddr, sizeof(servaddr)); |
| 791 | servaddr.sin_family = AF_INET; |
| 792 | servaddr.sin_port = htons(UDP_CODE_COMPLETION_LOGGER_PORT); |
| 793 | if (inet_pton(AF_INET, UDP_CODE_COMPLETION_LOGGER, |
| 794 | &servaddr.sin_addr) <= 0) |
| 795 | break; |
| 796 | |
| 797 | int sockfd = socket(AF_INET, SOCK_DGRAM, 0); |
| 798 | if (sockfd < 0) |
| 799 | break; |
| 800 | |
| 801 | sendto(sockfd, res.data(), res.size(), 0, |
| 802 | (struct sockaddr *)&servaddr, sizeof(servaddr)); |
| 803 | close(sockfd); |
| 804 | } |
| 805 | while (false); |
| 806 | } |
| 807 | #endif |
| 808 | #endif |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 809 | CCAI->result = Results; |
| 810 | } |
| 811 | CXCodeCompleteResults *clang_codeCompleteAt(CXTranslationUnit TU, |
| 812 | const char *complete_filename, |
| 813 | unsigned complete_line, |
| 814 | unsigned complete_column, |
| 815 | struct CXUnsavedFile *unsaved_files, |
| 816 | unsigned num_unsaved_files, |
| 817 | unsigned options) { |
Argyrios Kyrtzidis | c6f5c6a | 2013-01-10 18:54:52 +0000 | [diff] [blame] | 818 | LOG_FUNC_SECTION { |
| 819 | *Log << TU << ' ' |
| 820 | << complete_filename << ':' << complete_line << ':' << complete_column; |
| 821 | } |
| 822 | |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 823 | if (num_unsaved_files && !unsaved_files) |
| 824 | return nullptr; |
| 825 | |
| 826 | CodeCompleteAtInfo CCAI = {TU, complete_filename, complete_line, |
| 827 | complete_column, llvm::makeArrayRef(unsaved_files, num_unsaved_files), |
| 828 | options, nullptr}; |
Argyrios Kyrtzidis | 772e713 | 2012-12-22 04:48:07 +0000 | [diff] [blame] | 829 | |
| 830 | if (getenv("LIBCLANG_NOTHREADS")) { |
| 831 | clang_codeCompleteAt_Impl(&CCAI); |
| 832 | return CCAI.result; |
| 833 | } |
| 834 | |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 835 | llvm::CrashRecoveryContext CRC; |
| 836 | |
Daniel Dunbar | bf44c3b | 2010-11-05 07:19:31 +0000 | [diff] [blame] | 837 | if (!RunSafely(CRC, clang_codeCompleteAt_Impl, &CCAI)) { |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 838 | fprintf(stderr, "libclang: crash detected in code completion\n"); |
Dmitri Gribenko | 5694feb | 2013-01-26 18:53:38 +0000 | [diff] [blame] | 839 | cxtu::getASTUnit(TU)->setUnsafeToFree(true); |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 840 | return nullptr; |
Douglas Gregor | 6df7873 | 2011-05-05 20:27:22 +0000 | [diff] [blame] | 841 | } else if (getenv("LIBCLANG_RESOURCE_USAGE")) |
| 842 | PrintLibclangResourceUsage(TU); |
Daniel Dunbar | b1fd345 | 2010-08-19 23:44:10 +0000 | [diff] [blame] | 843 | |
| 844 | return CCAI.result; |
Douglas Gregor | 1abc6bc | 2010-08-04 16:47:14 +0000 | [diff] [blame] | 845 | } |
| 846 | |
Douglas Gregor | cee235c | 2010-08-05 09:09:23 +0000 | [diff] [blame] | 847 | unsigned clang_defaultCodeCompleteOptions(void) { |
| 848 | return CXCodeComplete_IncludeMacros; |
| 849 | } |
| 850 | |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 851 | void clang_disposeCodeCompleteResults(CXCodeCompleteResults *ResultsIn) { |
| 852 | if (!ResultsIn) |
| 853 | return; |
| 854 | |
| 855 | AllocatedCXCodeCompleteResults *Results |
| 856 | = static_cast<AllocatedCXCodeCompleteResults*>(ResultsIn); |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 857 | delete Results; |
| 858 | } |
Douglas Gregor | 58ddb60 | 2010-08-23 23:00:57 +0000 | [diff] [blame] | 859 | |
Douglas Gregor | a88084b | 2010-02-18 18:08:43 +0000 | [diff] [blame] | 860 | unsigned |
| 861 | clang_codeCompleteGetNumDiagnostics(CXCodeCompleteResults *ResultsIn) { |
| 862 | AllocatedCXCodeCompleteResults *Results |
| 863 | = static_cast<AllocatedCXCodeCompleteResults*>(ResultsIn); |
| 864 | if (!Results) |
| 865 | return 0; |
| 866 | |
| 867 | return Results->Diagnostics.size(); |
| 868 | } |
| 869 | |
| 870 | CXDiagnostic |
| 871 | clang_codeCompleteGetDiagnostic(CXCodeCompleteResults *ResultsIn, |
| 872 | unsigned Index) { |
| 873 | AllocatedCXCodeCompleteResults *Results |
| 874 | = static_cast<AllocatedCXCodeCompleteResults*>(ResultsIn); |
| 875 | if (!Results || Index >= Results->Diagnostics.size()) |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 876 | return nullptr; |
Douglas Gregor | a88084b | 2010-02-18 18:08:43 +0000 | [diff] [blame] | 877 | |
Stephen Hines | 6bcf27b | 2014-05-29 04:14:42 -0700 | [diff] [blame] | 878 | CXStoredDiagnostic *Diag = Results->DiagnosticsWrappers[Index]; |
| 879 | if (!Diag) |
| 880 | Results->DiagnosticsWrappers[Index] = Diag = |
| 881 | new CXStoredDiagnostic(Results->Diagnostics[Index], Results->LangOpts); |
| 882 | return Diag; |
Douglas Gregor | a88084b | 2010-02-18 18:08:43 +0000 | [diff] [blame] | 883 | } |
| 884 | |
Douglas Gregor | 3da626b | 2011-07-07 16:03:39 +0000 | [diff] [blame] | 885 | unsigned long long |
| 886 | clang_codeCompleteGetContexts(CXCodeCompleteResults *ResultsIn) { |
| 887 | AllocatedCXCodeCompleteResults *Results |
| 888 | = static_cast<AllocatedCXCodeCompleteResults*>(ResultsIn); |
| 889 | if (!Results) |
| 890 | return 0; |
| 891 | |
| 892 | return Results->Contexts; |
| 893 | } |
Douglas Gregor | a88084b | 2010-02-18 18:08:43 +0000 | [diff] [blame] | 894 | |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 895 | enum CXCursorKind clang_codeCompleteGetContainerKind( |
| 896 | CXCodeCompleteResults *ResultsIn, |
| 897 | unsigned *IsIncomplete) { |
| 898 | AllocatedCXCodeCompleteResults *Results = |
| 899 | static_cast<AllocatedCXCodeCompleteResults *>(ResultsIn); |
| 900 | if (!Results) |
| 901 | return CXCursor_InvalidCode; |
Stephen Hines | c568f1e | 2014-07-21 00:47:37 -0700 | [diff] [blame] | 902 | |
| 903 | if (IsIncomplete != nullptr) { |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 904 | *IsIncomplete = Results->ContainerIsIncomplete; |
| 905 | } |
| 906 | |
| 907 | return Results->ContainerKind; |
| 908 | } |
| 909 | |
| 910 | CXString clang_codeCompleteGetContainerUSR(CXCodeCompleteResults *ResultsIn) { |
| 911 | AllocatedCXCodeCompleteResults *Results = |
| 912 | static_cast<AllocatedCXCodeCompleteResults *>(ResultsIn); |
| 913 | if (!Results) |
Dmitri Gribenko | dc66adb | 2013-02-01 14:21:22 +0000 | [diff] [blame] | 914 | return cxstring::createEmpty(); |
Dmitri Gribenko | 90b5ac6 | 2013-02-07 19:09:22 +0000 | [diff] [blame] | 915 | |
| 916 | return cxstring::createRef(Results->ContainerUSR.c_str()); |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 917 | } |
Douglas Gregor | 0a47d69 | 2011-07-26 15:24:30 +0000 | [diff] [blame] | 918 | |
| 919 | |
| 920 | CXString clang_codeCompleteGetObjCSelector(CXCodeCompleteResults *ResultsIn) { |
| 921 | AllocatedCXCodeCompleteResults *Results = |
| 922 | static_cast<AllocatedCXCodeCompleteResults *>(ResultsIn); |
| 923 | if (!Results) |
Dmitri Gribenko | dc66adb | 2013-02-01 14:21:22 +0000 | [diff] [blame] | 924 | return cxstring::createEmpty(); |
Douglas Gregor | 0a47d69 | 2011-07-26 15:24:30 +0000 | [diff] [blame] | 925 | |
Dmitri Gribenko | 5595ded | 2013-02-02 02:19:29 +0000 | [diff] [blame] | 926 | return cxstring::createDup(Results->Selector); |
Douglas Gregor | 0a47d69 | 2011-07-26 15:24:30 +0000 | [diff] [blame] | 927 | } |
Douglas Gregor | e081a61 | 2011-07-21 01:05:26 +0000 | [diff] [blame] | 928 | |
Ted Kremenek | ab18893 | 2010-01-05 19:32:54 +0000 | [diff] [blame] | 929 | } // end extern "C" |
Douglas Gregor | 1e5e668 | 2010-08-26 13:48:20 +0000 | [diff] [blame] | 930 | |
Douglas Gregor | c7b7b7a | 2010-10-18 21:05:04 +0000 | [diff] [blame] | 931 | /// \brief Simple utility function that appends a \p New string to the given |
| 932 | /// \p Old string, using the \p Buffer for storage. |
| 933 | /// |
| 934 | /// \param Old The string to which we are appending. This parameter will be |
| 935 | /// updated to reflect the complete string. |
| 936 | /// |
| 937 | /// |
| 938 | /// \param New The string to append to \p Old. |
| 939 | /// |
| 940 | /// \param Buffer A buffer that stores the actual, concatenated string. It will |
| 941 | /// be used if the old string is already-non-empty. |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 942 | static void AppendToString(StringRef &Old, StringRef New, |
Dylan Noblesmith | 36d5927 | 2012-02-13 12:32:26 +0000 | [diff] [blame] | 943 | SmallString<256> &Buffer) { |
Douglas Gregor | c7b7b7a | 2010-10-18 21:05:04 +0000 | [diff] [blame] | 944 | if (Old.empty()) { |
| 945 | Old = New; |
| 946 | return; |
| 947 | } |
| 948 | |
| 949 | if (Buffer.empty()) |
| 950 | Buffer.append(Old.begin(), Old.end()); |
| 951 | Buffer.append(New.begin(), New.end()); |
| 952 | Old = Buffer.str(); |
| 953 | } |
| 954 | |
| 955 | /// \brief Get the typed-text blocks from the given code-completion string |
| 956 | /// and return them as a single string. |
| 957 | /// |
| 958 | /// \param String The code-completion string whose typed-text blocks will be |
| 959 | /// concatenated. |
| 960 | /// |
| 961 | /// \param Buffer A buffer used for storage of the completed name. |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 962 | static StringRef GetTypedName(CodeCompletionString *String, |
Dylan Noblesmith | 36d5927 | 2012-02-13 12:32:26 +0000 | [diff] [blame] | 963 | SmallString<256> &Buffer) { |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 964 | StringRef Result; |
Douglas Gregor | c7b7b7a | 2010-10-18 21:05:04 +0000 | [diff] [blame] | 965 | for (CodeCompletionString::iterator C = String->begin(), CEnd = String->end(); |
| 966 | C != CEnd; ++C) { |
| 967 | if (C->Kind == CodeCompletionString::CK_TypedText) |
| 968 | AppendToString(Result, C->Text, Buffer); |
| 969 | } |
| 970 | |
| 971 | return Result; |
| 972 | } |
| 973 | |
Douglas Gregor | 1e5e668 | 2010-08-26 13:48:20 +0000 | [diff] [blame] | 974 | namespace { |
| 975 | struct OrderCompletionResults { |
| 976 | bool operator()(const CXCompletionResult &XR, |
| 977 | const CXCompletionResult &YR) const { |
Douglas Gregor | 218937c | 2011-02-01 19:23:04 +0000 | [diff] [blame] | 978 | CodeCompletionString *X |
| 979 | = (CodeCompletionString *)XR.CompletionString; |
| 980 | CodeCompletionString *Y |
| 981 | = (CodeCompletionString *)YR.CompletionString; |
Douglas Gregor | 1e5e668 | 2010-08-26 13:48:20 +0000 | [diff] [blame] | 982 | |
Dylan Noblesmith | 36d5927 | 2012-02-13 12:32:26 +0000 | [diff] [blame] | 983 | SmallString<256> XBuffer; |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 984 | StringRef XText = GetTypedName(X, XBuffer); |
Dylan Noblesmith | 36d5927 | 2012-02-13 12:32:26 +0000 | [diff] [blame] | 985 | SmallString<256> YBuffer; |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 986 | StringRef YText = GetTypedName(Y, YBuffer); |
Douglas Gregor | 1e5e668 | 2010-08-26 13:48:20 +0000 | [diff] [blame] | 987 | |
Douglas Gregor | c7b7b7a | 2010-10-18 21:05:04 +0000 | [diff] [blame] | 988 | if (XText.empty() || YText.empty()) |
| 989 | return !XText.empty(); |
| 990 | |
| 991 | int result = XText.compare_lower(YText); |
Douglas Gregor | 1e5e668 | 2010-08-26 13:48:20 +0000 | [diff] [blame] | 992 | if (result < 0) |
| 993 | return true; |
| 994 | if (result > 0) |
| 995 | return false; |
| 996 | |
Douglas Gregor | c7b7b7a | 2010-10-18 21:05:04 +0000 | [diff] [blame] | 997 | result = XText.compare(YText); |
Douglas Gregor | 1aad340 | 2010-09-10 23:05:54 +0000 | [diff] [blame] | 998 | return result < 0; |
Douglas Gregor | 1e5e668 | 2010-08-26 13:48:20 +0000 | [diff] [blame] | 999 | } |
| 1000 | }; |
| 1001 | } |
| 1002 | |
| 1003 | extern "C" { |
| 1004 | void clang_sortCodeCompletionResults(CXCompletionResult *Results, |
| 1005 | unsigned NumResults) { |
| 1006 | std::stable_sort(Results, Results + NumResults, OrderCompletionResults()); |
| 1007 | } |
Douglas Gregor | 458433d | 2010-08-26 15:07:07 +0000 | [diff] [blame] | 1008 | } |