From a293a8e049fea49b8c2b8262fb92a29f73d66989 Mon Sep 17 00:00:00 2001
From: Vitaly Takmazov
Date: Sun, 2 Dec 2018 23:23:31 +0300
Subject: Drop YYModel

---
 Juick/Views/MessageCell.m | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

(limited to 'Juick/Views')

diff --git a/Juick/Views/MessageCell.m b/Juick/Views/MessageCell.m
index faf3492..2686adf 100644
--- a/Juick/Views/MessageCell.m
+++ b/Juick/Views/MessageCell.m
@@ -77,16 +77,16 @@
                                           initWithString:msg.text
                                           attributes:@{NSFontAttributeName:[UIFont preferredFontForTextStyle:UIFontTextStyleBody]}];
         [txt beginEditing];
-        for (NSDictionary *entity in msg.entities) {
-            NSUInteger start = [entity[@"start"] unsignedIntegerValue];
-            NSUInteger end = [entity[@"end"] unsignedIntegerValue];
-            NSString *text = entity[@"text"] ? entity[@"text"] : @"";
+        for (Entity *entity in msg.entities) {
+            NSUInteger start = entity.start ? [entity.start unsignedIntegerValue] : 0;
+            NSUInteger end =  entity.end ? [entity.end unsignedIntegerValue] : 0;
+            NSString *text = entity.text ? entity.text : @"";
             NSRange currentRange = NSMakeRange(start, end - start);
             [txt addAttribute:@"displayText" value:text range:currentRange];
-            if ([entity[@"type"] isEqualToString:@"a"]) {
-                [txt addAttribute:NSLinkAttributeName value:entity[@"url"] range:currentRange];
+            if ([entity.type isEqualToString:@"a"]) {
+                [txt addAttribute:NSLinkAttributeName value:entity.link range:currentRange];
             }
-            if ([entity[@"type"] isEqualToString:@"q"]) {
+            if ([entity.type isEqualToString:@"q"]) {
                 [txt addAttribute:NSForegroundColorAttributeName value:[UIColor darkGrayColor] range:currentRange];
                 NSMutableParagraphStyle *paragraphStyle = [[NSParagraphStyle defaultParagraphStyle] mutableCopy];
                 paragraphStyle.firstLineHeadIndent = 12.0f;
@@ -94,18 +94,18 @@
                 paragraphStyle.paragraphSpacing = 12.0f;
                 [txt addAttribute:NSParagraphStyleAttributeName value:paragraphStyle range:currentRange];
             }
-            if ([entity[@"type"] isEqualToString:@"u"]) {
+            if ([entity.type isEqualToString:@"u"]) {
                 [txt addAttribute:NSUnderlineStyleAttributeName value:@(NSUnderlineStyleSingle) range:currentRange];
             }
             UIFontDescriptor* fontDescriptor = [UIFontDescriptor
                                                 preferredFontDescriptorWithTextStyle:UIFontTextStyleBody];
-            if ([entity[@"type"] isEqualToString:@"b"]) {
+            if ([entity.type isEqualToString:@"b"]) {
                 UIFontDescriptor* boldFontDescriptor = [fontDescriptor
                                                         fontDescriptorWithSymbolicTraits:UIFontDescriptorTraitBold];
                 UIFont* boldFont =  [UIFont fontWithDescriptor:boldFontDescriptor size: 0.0];
                 [txt addAttribute:NSFontAttributeName value:boldFont range:currentRange];
             }
-            if ([entity[@"type"] isEqualToString:@"i"]) {
+            if ([entity.type isEqualToString:@"i"]) {
                 UIFontDescriptor* italicFontDescriptor = [fontDescriptor
                                                         fontDescriptorWithSymbolicTraits:UIFontDescriptorTraitItalic];
                 UIFont* italicFont =  [UIFont fontWithDescriptor:italicFontDescriptor size: 0.0];
-- 
cgit v1.2.3