comaps/iphone/Maps/UI/Search/TableView/MWMSearchCell.mm

#import "MWMSearchCell.h"
#import "SearchResult.h"

@interface MWMSearchCell ()

@property (weak, nonatomic) IBOutlet UILabel * titleLabel;

@end

@implementation MWMSearchCell

- (void)configureWith:(SearchResult * _Nonnull)result isPartialMatching:(BOOL)isPartialMatching {
  NSString * title = result.titleText;

  if (title.length == 0)
  {
    self.titleLabel.text = @"";
    return;
  }
  
  NSDictionary * selectedTitleAttributes = [self selectedTitleAttributes];
  NSDictionary * unselectedTitleAttributes = [self unselectedTitleAttributes];
  if (!selectedTitleAttributes || !unselectedTitleAttributes)
  {
    self.titleLabel.text = title;
    return;
  }
  
  NSMutableAttributedString * attributedTitle =
      [[NSMutableAttributedString alloc] initWithString:title];
  NSDictionary * titleAttributes = isPartialMatching ? unselectedTitleAttributes : selectedTitleAttributes;

  NSArray<NSValue *> *highlightRanges = result.highlightRanges;
  [attributedTitle addAttributes:titleAttributes range:NSMakeRange(0, title.length)];

  for (NSValue *rangeValue in highlightRanges) {
    NSRange range = [rangeValue rangeValue];
    if (NSMaxRange(range) <= result.titleText.length) {
      [attributedTitle addAttributes:selectedTitleAttributes range:range];
    } else {
      NSLog(@"Incorrect range: %@ for string: %@", NSStringFromRange(range), result.titleText);
    }
  }
  
  // Add branch with thinner font weight if present and not already in title
  if (result.branchText && result.branchText.length > 0 && ![title containsString:result.branchText]) {
    NSDictionary * branchAttributes = isPartialMatching ? unselectedTitleAttributes : @{
      NSForegroundColorAttributeName : [selectedTitleAttributes objectForKey:NSForegroundColorAttributeName],
      NSFontAttributeName : [UIFont regular17]
    };
    NSAttributedString * branchString = [[NSAttributedString alloc] initWithString:[NSString stringWithFormat:@" %@", result.branchText] 
                                                                        attributes:branchAttributes];
    [attributedTitle appendAttributedString:branchString];
  }
  
  self.titleLabel.attributedText = attributedTitle;
  [self.titleLabel sizeToFit];
}

- (NSDictionary *)selectedTitleAttributes
{
  return nil;
}

- (NSDictionary *)unselectedTitleAttributes
{
  return nil;
}

@end