Add syntax highlighting logic to formatter

Components/Sources/ComponentsObjC/WKSourceEditorFormatterReference.m

@@ -1,5 +1,117 @@
 #import "WKSourceEditorFormatterReference.h"
+#import "WKSourceEditorColors.h"
+
+@interface WKSourceEditorFormatterReference ()
+@property (nonatomic, strong) NSDictionary *refAttributes;
+@property (nonatomic, strong) NSDictionary *refEmptyAttributes;
+
+@property (nonatomic, strong) NSRegularExpression *refOpenAndCloseRegex;
+@property (nonatomic, strong) NSRegularExpression *refOpenRegex;
+@property (nonatomic, strong) NSRegularExpression *refCloseRegex;
+@property (nonatomic, strong) NSRegularExpression *refEmptyRegex;
+@end
 
 @implementation WKSourceEditorFormatterReference
+- (instancetype)initWithColors:(WKSourceEditorColors *)colors fonts:(WKSourceEditorFonts *)fonts {
+    self = [super initWithColors:colors fonts:fonts];
+    if (self) {
+        _refAttributes = @{
+            NSForegroundColorAttributeName: colors.greenForegroundColor,
+            WKSourceEditorCustomKeyColorGreen: [NSNumber numberWithBool:YES]
+        };
+
+        _refEmptyAttributes = @{
+            NSForegroundColorAttributeName: colors.greenForegroundColor,
+            WKSourceEditorCustomKeyColorGreen: [NSNumber numberWithBool:YES]
+        };
+
+        _refOpenAndCloseRegex = [[NSRegularExpression alloc] initWithPattern:@"(<ref(?:[^\\/>]+?)?>)(.*?)(<\\/ref>)" options:0 error:nil];
+        _refOpenRegex = [[NSRegularExpression alloc] initWithPattern:@"<ref(?:[^\\/>]+?)?>" options:0 error:nil];
+        _refCloseRegex = [[NSRegularExpression alloc] initWithPattern:@"<\\/ref>" options:0 error:nil];
+        _refEmptyRegex = [[NSRegularExpression alloc] initWithPattern:@"<ref[^>]+?\\/>" options:0 error:nil];
+    }
+
+    return self;
+}
+
+#pragma mark - Overrides
+
+- (void)addSyntaxHighlightingToAttributedString:(nonnull NSMutableAttributedString *)attributedString inRange:(NSRange)range {
+
+    [self.refOpenAndCloseRegex enumerateMatchesInString:attributedString.string
+                                    options:0
+                                      range:range
+                                 usingBlock:^(NSTextCheckingResult *_Nullable result, NSMatchingFlags flags, BOOL *_Nonnull stop) {
+        NSRange fullMatch = [result rangeAtIndex:0];
+        NSRange openingRange = [result rangeAtIndex:1];
+        NSRange contentRange = [result rangeAtIndex:2];
+        NSRange closingRange = [result rangeAtIndex:3];
+
+        if (openingRange.location != NSNotFound) {
+            [attributedString addAttributes:self.refAttributes range:openingRange];
+        }
+
+        if (closingRange.location != NSNotFound) {
+            [attributedString addAttributes:self.refAttributes range:closingRange];
+        }
+    }];
+
+    [self.refEmptyRegex enumerateMatchesInString:attributedString.string
+                                         options:0
+                                           range:range
+                                      usingBlock:^(NSTextCheckingResult *_Nullable result, NSMatchingFlags flags, BOOL *_Nonnull stop) {
+        NSRange fullMatch = [result rangeAtIndex:0];
+
+        if (fullMatch.location != NSNotFound) {
+            [attributedString addAttributes:self.refAttributes range:fullMatch];
+        }
+    }];
+
+    // refOpenAndClose regex doesn't match everything. This scoops up extra open and closing ref tags that do not have a matching tag on the same line
+
+    [self.refOpenRegex enumerateMatchesInString:attributedString.string
+                                         options:0
+                                           range:range
+                                      usingBlock:^(NSTextCheckingResult *_Nullable result, NSMatchingFlags flags, BOOL *_Nonnull stop) {
+        NSRange fullMatch = [result rangeAtIndex:0];
+
+        if (fullMatch.location != NSNotFound) {
+            [attributedString addAttributes:self.refAttributes range:fullMatch];
+        }
+    }];
+
+    [self.refCloseRegex enumerateMatchesInString:attributedString.string
+                                         options:0
+                                           range:range
+                                      usingBlock:^(NSTextCheckingResult *_Nullable result, NSMatchingFlags flags, BOOL *_Nonnull stop) {
+        NSRange fullMatch = [result rangeAtIndex:0];
+
+        if (fullMatch.location != NSNotFound) {
+            [attributedString addAttributes:self.refAttributes range:fullMatch];
+        }
+    }];
+}
+
+- (void)updateColors:(WKSourceEditorColors *)colors inAttributedString:(NSMutableAttributedString *)attributedString inRange:(NSRange)range {
+
+    NSMutableDictionary *mutAttributes = [[NSMutableDictionary alloc] initWithDictionary:self.refAttributes];
+    [mutAttributes setObject:colors.greenForegroundColor forKey:NSForegroundColorAttributeName];
+    self.refAttributes = [[NSDictionary alloc] initWithDictionary:mutAttributes];
+
+    [attributedString enumerateAttribute:WKSourceEditorCustomKeyColorGreen
+                                 inRange:range
+                                 options:nil
+                              usingBlock:^(id value, NSRange localRange, BOOL *stop) {
+        if ([value isKindOfClass: [NSNumber class]]) {
+            NSNumber *numValue = (NSNumber *)value;
+            if ([numValue boolValue] == YES) {
+                [attributedString addAttributes:self.refAttributes range:localRange];
+            }
+        }
+    }];
+}
 
+- (void)updateFonts:(WKSourceEditorFonts *)fonts inAttributedString:(NSMutableAttributedString *)attributedString inRange:(NSRange)range {
+    // No special font handling needed for references
+}
 @end