OLD | NEW |
| (Empty) |
1 # Copyright 2015 The Chromium Authors. All rights reserved. | |
2 # Use of this source code is governed by a BSD-style license that can be | |
3 # found in the LICENSE file. | |
4 | |
5 import itertools | |
6 import symbol | |
7 import token | |
8 | |
9 from catapult_base.refactor.annotated_symbol import base_symbol | |
10 from catapult_base.refactor import snippet | |
11 | |
12 | |
13 __all__ = [ | |
14 'Reference', | |
15 ] | |
16 | |
17 | |
18 class Reference(base_symbol.AnnotatedSymbol): | |
19 @classmethod | |
20 def Annotate(cls, nodes): | |
21 if not nodes: | |
22 return None | |
23 if nodes[0].type != symbol.atom: | |
24 return None | |
25 if not nodes[0].children or nodes[0].children[0].type != token.NAME: | |
26 return None | |
27 | |
28 for i in xrange(1, len(nodes)): | |
29 if not nodes: | |
30 break | |
31 if nodes[i].type != symbol.trailer: | |
32 break | |
33 if len(nodes[i].children) != 2: | |
34 break | |
35 if (nodes[i].children[0].type != token.DOT or | |
36 nodes[i].children[1].type != token.NAME): | |
37 break | |
38 else: | |
39 i = len(nodes) | |
40 | |
41 return [cls(nodes[:i])] + nodes[i:] | |
42 | |
43 def __init__(self, children): | |
44 super(Reference, self).__init__(-1, children) | |
45 | |
46 @property | |
47 def type_name(self): | |
48 return 'attribute_reference' | |
49 | |
50 @property | |
51 def value(self): | |
52 return ''.join(token_snippet.value | |
53 for child in self.children | |
54 for token_snippet in child.children) | |
55 | |
56 @value.setter | |
57 def value(self, value): | |
58 value_parts = value.split('.') | |
59 | |
60 # If we have too many children, cut the list down to size. | |
61 self._children = self._children[:len(value_parts)] | |
62 | |
63 # Update child nodes. | |
64 for child, value_part in itertools.izip_longest( | |
65 self._children, value_parts): | |
66 if child: | |
67 # Modify existing children. This helps preserve comments and spaces. | |
68 child.children[-1].value = value_part | |
69 else: | |
70 # Add children as needed. | |
71 token_snippets = [ | |
72 snippet.TokenSnippet.Create(token.DOT, '.'), | |
73 snippet.TokenSnippet.Create(token.NAME, value_part), | |
74 ] | |
75 self._children.append(snippet.Symbol(symbol.trailer, token_snippets)) | |
OLD | NEW |