bundesmessenger-ios/Riot/Categories/NSAttributedString.swift

// 
// Copyright 2022 New Vector Ltd
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

import Foundation

public extension NSAttributedString {

    /// Returns a new attributed string by removing all links from the receiver.
    @objc var vc_byRemovingLinks: NSAttributedString {
        let result = NSMutableAttributedString(attributedString: self)
        result.removeAttribute(.link, range: NSRange(location: 0, length: length))
        return result
    }
    
    /// Enumerate attribute for given key and conveniently ignore any attribute that doesn't match given generic type.
    ///
    /// - Parameters:
    ///   - attrName: The name of the attribute to enumerate.
    ///   - enumerationRange: The range over which the attribute values are enumerated.
    ///   - opts: The options used by the enumeration. For possible values, see NSAttributedStringEnumerationOptions.
    ///   - block: The block to apply to ranges of the specified attribute in the attributed string.
    func vc_enumerateAttribute<T>(_ attrName: NSAttributedString.Key,
                                  in enumerationRange: NSRange,
                                  options opts: NSAttributedString.EnumerationOptions = [],
                                  using block: (T, NSRange, UnsafeMutablePointer<ObjCBool>) -> Void) {
        self.enumerateAttribute(attrName,
                                in: enumerationRange,
                                options: opts) { (attr: Any?, range: NSRange, stop: UnsafeMutablePointer<ObjCBool>) in
            guard let typedAttr = attr as? T else { return }
            
            block(typedAttr, range, stop)
        }
    }
}