safe-ios/Multisig/UI/UI Library/String+Highlight.swift

//
//  String+Highlight.swift
//  Multisig
//
//  Created by Moaaz on 11/10/21.
//  Copyright © 2021 Gnosis Ltd. All rights reserved.
//

import Foundation
import UIKit

extension String {
    func highlight(prefix: Int = 6, suffix: Int = 4) -> NSAttributedString {
        let style = GNOTextStyle.bodyTertiary
        let attributedString = NSMutableAttributedString(
            string: self,
            attributes: [
                .font: UIFont.gnoFont(forTextStyle: style),
                .foregroundColor: style.color!,
                .kern: -0.41
            ]
        )

        if self.count < prefix + suffix {
            attributedString.addAttribute(
                .foregroundColor, value: UIColor.labelPrimary, range: NSRange(location: 0, length: count))
        } else {
            attributedString.addAttribute(
                .foregroundColor, value: UIColor.labelPrimary, range: NSRange(location: 0, length: prefix))
            attributedString.addAttribute(
                .foregroundColor, value: UIColor.labelPrimary, range: NSRange(location: count - suffix, length: suffix))
        }
        
        return attributedString
    }

    func highlightRange(originalStyle: GNOTextStyle = .body,
                        highlightStyle: GNOTextStyle = .bodyPrimary,
                        textToHighlight: String? = nil) -> NSMutableAttributedString {
        let attributedString = NSMutableAttributedString(string: self, attributes: originalStyle.attributes)

        if let textToHighlight = textToHighlight {
            let rangeToHighlight = (attributedString.string as NSString).range(of: textToHighlight)
            if rangeToHighlight.location != NSNotFound {
                attributedString.addAttributes(highlightStyle.attributes, range: rangeToHighlight)
            }
        }

        return attributedString
    }
}