| OLD | NEW |
| (Empty) |
| 1 import 'package:analyzer/src/generated/ast.dart'; | |
| 2 import 'package:analyzer/src/generated/element.dart'; | |
| 3 import 'package:analyzer/src/generated/scanner.dart'; | |
| 4 import 'package:analyzer/src/generated/utilities_dart.dart'; | |
| 5 | |
| 6 int _computeArgIndex(AstNode containingNode, Object entity) { | |
| 7 var argList = containingNode; | |
| 8 if (argList is ArgumentList) { | |
| 9 NodeList<Expression> args = argList.arguments; | |
| 10 for (int index = 0; index < args.length; ++index) { | |
| 11 if (entity == args[index]) { | |
| 12 return index; | |
| 13 } | |
| 14 } | |
| 15 if (args.isEmpty) { | |
| 16 return 0; | |
| 17 } | |
| 18 } | |
| 19 return null; | |
| 20 } | |
| 21 | |
| 22 /** | |
| 23 * A CompletionTarget represents an edge in the parse tree which connects an | |
| 24 * AST node (the [containingNode] of the completion) to one of its children | |
| 25 * (the [entity], which represents the place in the parse tree where the newly | |
| 26 * completed text will be inserted). | |
| 27 * | |
| 28 * To illustrate, consider the following snippet of code, and its associated | |
| 29 * parse tree. (T's represent tokens, N's represent AST nodes. Some trivial | |
| 30 * AST nodes are not shown). | |
| 31 * | |
| 32 * ___N (function declaration) | |
| 33 * / \ | |
| 34 * / __N_______ (function body) | |
| 35 * / / |a \ | |
| 36 * / / N______ \ (statement) | |
| 37 * / / / \ \ | |
| 38 * / / N____ \ \ (assignment expression) | |
| 39 * | / /| \ \ \ | |
| 40 * . | / | _N___ \ | ("as" expression) | |
| 41 * . | | | / | \c | | | |
| 42 * . | N | N |b N | | (simple identifiers) | |
| 43 * | | | | | | | | | |
| 44 * T T T T T T T T | |
| 45 * m() { foo = bar as Baz; } | |
| 46 * | |
| 47 * The Completion target is usually placed as high in the tree as possible so | |
| 48 * that we can produce the most meaningful completions with minimal effort. | |
| 49 * For instance, if the cursor is inside the identifier "foo", the completion | |
| 50 * target will be the edge marked "a", so that we will produce all completions | |
| 51 * that could possibly start a statement, even those which would conflict with | |
| 52 * the current parse (such as the keyword "for", which begins a "for" | |
| 53 * statement). As a consequence of this, the [entity] will usually not be the | |
| 54 * first child of the [containingNode] node. | |
| 55 * | |
| 56 * Note that the [containingNode] is always an AST node, but the [entity] may | |
| 57 * not be. For instance, if the cursor is inside the keyword "as", the | |
| 58 * completion target will be the edge marked "b", so the [entity] is the token | |
| 59 * "as". | |
| 60 * | |
| 61 * If the cursor is between tokens, the completion target is usually associated | |
| 62 * with the token that follows the cursor (since that's the token that will be | |
| 63 * displaced when the new text is inserted). For example, if the cursor is | |
| 64 * after the "{" character, then the completion target will be the edge marked | |
| 65 * "a", just as it is if the cursor is inside the identifier "foo". However | |
| 66 * there is one exception: if the cursor is at the rightmost edge of a keyword | |
| 67 * or identifier, then the completion target is associated with that token, | |
| 68 * since any further letters typed will change the meaning of the identifier or | |
| 69 * keyword, rather than creating a new token. So for instance, if the cursor | |
| 70 * is just after the "s" of "as", the completion target will be the edge marked | |
| 71 * "b", but if the cursor target is after the first space following "as", then | |
| 72 * the completion target will be the edge marked "c". | |
| 73 * | |
| 74 * If the file is empty, or the cursor is after all the text in the file, then | |
| 75 * there may be no edge in the parse tree which is appropriate to act as the | |
| 76 * completion target; in this case, [entity] is set to null and | |
| 77 * [containingNode] is set to the CompilationUnit. | |
| 78 */ | |
| 79 class CompletionTarget { | |
| 80 /** | |
| 81 * The context in which the completion is occurring. This is the AST node | |
| 82 * which is a direct parent of [entity]. | |
| 83 */ | |
| 84 final AstNode containingNode; | |
| 85 | |
| 86 /** | |
| 87 * The entity which the completed text will replace (or which will be | |
| 88 * displaced once the completed text is inserted). This may be an AstNode or | |
| 89 * a Token, or it may be null if the cursor is after all tokens in the file. | |
| 90 * | |
| 91 * Usually, the entity won't be the first child of the [containingNode] (this | |
| 92 * is a consequence of placing the completion target as high in the tree as | |
| 93 * possible). However, there is one exception: when the cursor is inside of | |
| 94 * a multi-character token which is not a keyword or identifier (e.g. a | |
| 95 * comment, or a token like "+=", the entity will be always be the token. | |
| 96 */ | |
| 97 final Object entity; | |
| 98 | |
| 99 /** | |
| 100 * If the target is an argument in an [ArgumentList], then this is the index | |
| 101 * of the argument in the list, otherwise this is `null`. | |
| 102 */ | |
| 103 final int argIndex; | |
| 104 | |
| 105 /** | |
| 106 * Compute the appropriate [CompletionTarget] for the given [offset] within | |
| 107 * the [compilationUnit]. | |
| 108 */ | |
| 109 factory CompletionTarget.forOffset( | |
| 110 CompilationUnit compilationUnit, int offset) { | |
| 111 // The precise algorithm is as follows. We perform a depth-first search of | |
| 112 // all edges in the parse tree (both those that point to AST nodes and | |
| 113 // those that point to tokens), visiting parents before children. The | |
| 114 // first edge which points to an entity satisfying either _isCandidateToken | |
| 115 // or _isCandidateNode is the completion target. If no edge is found that | |
| 116 // satisfies these two predicates, then we set the completion target entity | |
| 117 // to null and the containingNode to the compilationUnit. | |
| 118 // | |
| 119 // Note that if a token is not a candidate target, then none of the tokens | |
| 120 // that precede it are candidate targets either. Therefore any entity | |
| 121 // whose last token is not a candidate target can be skipped. This lets us | |
| 122 // prune the search to the point where no recursion is necessary; at each | |
| 123 // step in the process we know exactly which child node we need to proceed | |
| 124 // to. | |
| 125 AstNode containingNode = compilationUnit; | |
| 126 outerLoop: while (true) { | |
| 127 if (containingNode is Comment) { | |
| 128 // Comments are handled specially: we descend into any CommentReference | |
| 129 // child node that contains the cursor offset. | |
| 130 Comment comment = containingNode; | |
| 131 for (CommentReference commentReference in comment.references) { | |
| 132 if (commentReference.offset <= offset && | |
| 133 offset <= commentReference.end) { | |
| 134 containingNode = commentReference; | |
| 135 continue outerLoop; | |
| 136 } | |
| 137 } | |
| 138 } | |
| 139 for (var entity in containingNode.childEntities) { | |
| 140 if (entity is Token) { | |
| 141 if (_isCandidateToken(entity, offset)) { | |
| 142 // Target found. | |
| 143 return new CompletionTarget._(containingNode, entity); | |
| 144 } else { | |
| 145 // Since entity is a token, we don't need to look inside it; just | |
| 146 // proceed to the next entity. | |
| 147 continue; | |
| 148 } | |
| 149 } else if (entity is AstNode) { | |
| 150 // If the last token in the node isn't a candidate target, then | |
| 151 // neither the node nor any of its descendants can possibly be the | |
| 152 // completion target, so we can skip the node entirely. | |
| 153 if (!_isCandidateToken(entity.endToken, offset)) { | |
| 154 continue; | |
| 155 } | |
| 156 | |
| 157 // If the node is a candidate target, then we are done. | |
| 158 if (_isCandidateNode(entity, offset)) { | |
| 159 // Check to see if the offset is in a preceeding comment | |
| 160 Token commentToken = _getContainingCommentToken(entity, offset); | |
| 161 if (commentToken != null) { | |
| 162 entity = commentToken; | |
| 163 // If the preceeding comment is dartdoc token then update | |
| 164 // the containing node to be the dartdoc comment | |
| 165 Comment docComment = | |
| 166 _getContainingDocComment(containingNode, commentToken); | |
| 167 if (docComment != null) { | |
| 168 containingNode = docComment; | |
| 169 } | |
| 170 } | |
| 171 return new CompletionTarget._(containingNode, entity); | |
| 172 } | |
| 173 | |
| 174 // Otherwise, the completion target is somewhere inside the entity, | |
| 175 // so we need to jump to the start of the outer loop to examine its | |
| 176 // contents. | |
| 177 containingNode = entity; | |
| 178 continue outerLoop; | |
| 179 } else { | |
| 180 // Unexpected entity found (all entities in a parse tree should be | |
| 181 // AST nodes or tokens). | |
| 182 assert(false); | |
| 183 } | |
| 184 } | |
| 185 | |
| 186 // No completion target found. It should only be possible to reach here | |
| 187 // the first time through the outer loop (since we only jump to the start | |
| 188 // of the outer loop after determining that the completion target is | |
| 189 // inside an entity). We can check that assumption by verifying that | |
| 190 // containingNode is still the compilationUnit. | |
| 191 assert(identical(containingNode, compilationUnit)); | |
| 192 | |
| 193 // Since no completion target was found, we set the completion target | |
| 194 // entity to null and use the compilationUnit as the parent. | |
| 195 return new CompletionTarget._(compilationUnit, null); | |
| 196 } | |
| 197 } | |
| 198 | |
| 199 /** | |
| 200 * Create a [CompletionTarget] holding the given [containingNode] and | |
| 201 * [entity]. | |
| 202 */ | |
| 203 CompletionTarget._(AstNode containingNode, Object entity) | |
| 204 : this.containingNode = containingNode, | |
| 205 this.entity = entity, | |
| 206 this.argIndex = _computeArgIndex(containingNode, entity); | |
| 207 | |
| 208 /** | |
| 209 * Return `true` if the target is a functional argument in an argument list. | |
| 210 * The target [AstNode] hierarchy *must* be resolved for this to work. | |
| 211 */ | |
| 212 bool isFunctionalArgument() { | |
| 213 if (argIndex == null) { | |
| 214 return false; | |
| 215 } | |
| 216 AstNode argList = containingNode; | |
| 217 if (argList is! ArgumentList) { | |
| 218 return false; | |
| 219 } | |
| 220 AstNode parent = argList.parent; | |
| 221 if (parent is InstanceCreationExpression) { | |
| 222 DartType instType = parent.bestType; | |
| 223 if (instType != null) { | |
| 224 Element intTypeElem = instType.element; | |
| 225 if (intTypeElem is ClassElement) { | |
| 226 SimpleIdentifier constructorName = parent.constructorName.name; | |
| 227 ConstructorElement constructor = constructorName != null | |
| 228 ? intTypeElem.getNamedConstructor(constructorName.name) | |
| 229 : intTypeElem.unnamedConstructor; | |
| 230 return constructor != null && | |
| 231 _isFunctionalParameter(constructor.parameters, argIndex); | |
| 232 } | |
| 233 } | |
| 234 } else if (parent is MethodInvocation) { | |
| 235 SimpleIdentifier methodName = parent.methodName; | |
| 236 if (methodName != null) { | |
| 237 Element methodElem = methodName.bestElement; | |
| 238 if (methodElem is MethodElement) { | |
| 239 return _isFunctionalParameter(methodElem.parameters, argIndex); | |
| 240 } else if (methodElem is FunctionElement) { | |
| 241 return _isFunctionalParameter(methodElem.parameters, argIndex); | |
| 242 } | |
| 243 } | |
| 244 } | |
| 245 return false; | |
| 246 } | |
| 247 | |
| 248 /** | |
| 249 * Determine if the offset is contained in a preceeding comment token | |
| 250 * and return that token, otherwise return `null`. | |
| 251 */ | |
| 252 static Token _getContainingCommentToken(AstNode node, int offset) { | |
| 253 if (offset >= node.offset) { | |
| 254 return null; | |
| 255 } | |
| 256 Token token = node.beginToken; | |
| 257 if (token == null) { | |
| 258 return null; | |
| 259 } | |
| 260 token = token.precedingComments; | |
| 261 while (token != null) { | |
| 262 if (offset <= token.offset) { | |
| 263 return null; | |
| 264 } | |
| 265 if (offset <= token.end) { | |
| 266 if (token.type == TokenType.SINGLE_LINE_COMMENT || offset < token.end) { | |
| 267 return token; | |
| 268 } | |
| 269 } | |
| 270 token = token.next; | |
| 271 } | |
| 272 return null; | |
| 273 } | |
| 274 | |
| 275 /** | |
| 276 * Determine if the given token is part of the given node's dart doc. | |
| 277 */ | |
| 278 static Comment _getContainingDocComment(AstNode node, Token token) { | |
| 279 if (node is AnnotatedNode) { | |
| 280 Comment docComment = node.documentationComment; | |
| 281 if (docComment != null && docComment.tokens.contains(token)) { | |
| 282 return docComment; | |
| 283 } | |
| 284 } | |
| 285 return null; | |
| 286 } | |
| 287 | |
| 288 /** | |
| 289 * Determine whether [node] could possibly be the [entity] for a | |
| 290 * [CompletionTarget] associated with the given [offset]. | |
| 291 */ | |
| 292 static bool _isCandidateNode(AstNode node, int offset) { | |
| 293 // If the node's first token is a keyword or identifier, then the node is a | |
| 294 // candidate entity if its first token is. | |
| 295 Token beginToken = node.beginToken; | |
| 296 if (beginToken.type == TokenType.KEYWORD || | |
| 297 beginToken.type == TokenType.IDENTIFIER) { | |
| 298 return _isCandidateToken(beginToken, offset); | |
| 299 } | |
| 300 | |
| 301 // Otherwise, the node is a candidate entity only if the offset is before | |
| 302 // the beginning of the node. This ensures that completions within a token | |
| 303 // (e.g. inside a literal string or inside a comment) are evaluated within | |
| 304 // the context of the token itself. | |
| 305 return offset <= node.offset; | |
| 306 } | |
| 307 | |
| 308 /** | |
| 309 * Determine whether [token] could possibly be the [entity] for a | |
| 310 * [CompletionTarget] associated with the given [offset]. | |
| 311 */ | |
| 312 static bool _isCandidateToken(Token token, int offset) { | |
| 313 // A token is considered a candidate entity if the cursor offset is (a) | |
| 314 // before the start of the token, (b) within the token, (c) at the end of | |
| 315 // the token and the token is a keyword or identifier, or (d) at the | |
| 316 // location of the token and the token is zero length. | |
| 317 if (offset < token.end) { | |
| 318 return true; | |
| 319 } else if (offset == token.end) { | |
| 320 return token.type == TokenType.KEYWORD || | |
| 321 token.type == TokenType.IDENTIFIER || | |
| 322 token.length == 0; | |
| 323 } else if (!token.isSynthetic) { | |
| 324 return false; | |
| 325 } | |
| 326 // If the current token is synthetic, then check the previous token | |
| 327 // because it may have been dropped from the parse tree | |
| 328 Token previous = token.previous; | |
| 329 if (offset < previous.end) { | |
| 330 return true; | |
| 331 } else if (offset == previous.end) { | |
| 332 return token.type == TokenType.KEYWORD || | |
| 333 previous.type == TokenType.IDENTIFIER; | |
| 334 } else { | |
| 335 return false; | |
| 336 } | |
| 337 } | |
| 338 | |
| 339 /** | |
| 340 * Return `true` if the parameter is a functional parameter. | |
| 341 */ | |
| 342 static bool _isFunctionalParameter( | |
| 343 List<ParameterElement> parameters, int paramIndex) { | |
| 344 if (paramIndex < parameters.length) { | |
| 345 ParameterElement param = parameters[paramIndex]; | |
| 346 DartType paramType = param.type; | |
| 347 if (param.parameterKind == ParameterKind.NAMED) { | |
| 348 // TODO(danrubel) handle named parameters | |
| 349 return false; | |
| 350 } else { | |
| 351 return paramType is FunctionType || paramType is FunctionTypeAlias; | |
| 352 } | |
| 353 } | |
| 354 return false; | |
| 355 } | |
| 356 } | |
| OLD | NEW |