Merge pull request #59 from rolandleth/variable-coloring-for-swift

self.property coloring for Swift.
kolinkrewinkel · Aug 10, 2015 · 24db437 · 24db437
2 parents cb8abcb + 570b20b
commit 24db437
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Polychromatic/Categories/DVTTextStorage+PLYHighlightingHook.m b/Polychromatic/Categories/DVTTextStorage+PLYHighlightingHook.m
@@ -53,7 +53,9 @@ - (NSColor *)ply_colorAtCharacterIndex:(unsigned long long)index effectiveRange:
     {
         long long nodeType = [self nodeTypeAtCharacterIndex:newRange.location effectiveRange:effectiveRange context:context];
 
-        if (nodeType == 9)
+        if (nodeType == [DVTSourceNodeTypes registerNodeTypeNamed:@"xcode.syntax.identifier.variable"] ||
+            nodeType == [DVTSourceNodeTypes registerNodeTypeNamed:@"xcode.syntax.identifier.constant"] ||
+            nodeType == [DVTSourceNodeTypes registerNodeTypeNamed:@"xcode.syntax.identifier"])
         {
             PLYMockSwift *fauxSwiftService = (PLYMockSwift *)self.languageService;
             NSRange funcDefinitionRange = [fauxSwiftService methodDefinitionRangeAtIndex:newRange.location];