59 lines
3.6 KiB
Swift
59 lines
3.6 KiB
Swift
|
|
@objc(LKMentionUtilities)
|
|
public final class MentionUtilities : NSObject {
|
|
|
|
override private init() { }
|
|
|
|
@objc public static func highlightMentions(in string: String, threadID: String) -> String {
|
|
return highlightMentions(in: string, isOutgoingMessage: false, threadID: threadID, attributes: [:]).string // isOutgoingMessage and attributes are irrelevant
|
|
}
|
|
|
|
@objc public static func highlightMentions(in string: String, isOutgoingMessage: Bool, threadID: String, attributes: [NSAttributedString.Key:Any]) -> NSAttributedString {
|
|
let userHexEncodedPublicKey = getUserHexEncodedPublicKey()
|
|
var publicChat: LokiPublicChat?
|
|
var userLinkedDeviceHexEncodedPublicKeys: Set<String>!
|
|
OWSPrimaryStorage.shared().dbReadConnection.read { transaction in
|
|
publicChat = LokiDatabaseUtilities.getPublicChat(for: threadID, in: transaction)
|
|
userLinkedDeviceHexEncodedPublicKeys = LokiDatabaseUtilities.getLinkedDeviceHexEncodedPublicKeys(for: userHexEncodedPublicKey, in: transaction)
|
|
}
|
|
var string = string
|
|
let regex = try! NSRegularExpression(pattern: "@[0-9a-fA-F]*", options: [])
|
|
let knownHexEncodedPublicKeys = LokiAPI.userHexEncodedPublicKeyCache[threadID] ?? [] // Should always be populated at this point
|
|
var mentions: [(range: NSRange, hexEncodedPublicKey: String)] = []
|
|
var outerMatch = regex.firstMatch(in: string, options: .withoutAnchoringBounds, range: NSRange(location: 0, length: string.count))
|
|
while let match = outerMatch {
|
|
let hexEncodedPublicKey = String((string as NSString).substring(with: match.range).dropFirst()) // Drop the @
|
|
let matchEnd: Int
|
|
if knownHexEncodedPublicKeys.contains(hexEncodedPublicKey) {
|
|
var displayName: String?
|
|
if hexEncodedPublicKey == userHexEncodedPublicKey {
|
|
displayName = OWSProfileManager.shared().localProfileName()
|
|
} else {
|
|
if let publicChat = publicChat {
|
|
displayName = DisplayNameUtilities.getPublicChatDisplayName(for: hexEncodedPublicKey, in: publicChat.channel, on: publicChat.server)
|
|
} else {
|
|
displayName = DisplayNameUtilities.getPrivateChatDisplayName(for: hexEncodedPublicKey)
|
|
}
|
|
}
|
|
if let displayName = displayName {
|
|
string = (string as NSString).replacingCharacters(in: match.range, with: "@\(displayName)")
|
|
mentions.append((range: NSRange(location: match.range.location, length: displayName.count + 1), hexEncodedPublicKey: hexEncodedPublicKey)) // + 1 to include the @
|
|
matchEnd = match.range.location + displayName.count
|
|
} else {
|
|
matchEnd = match.range.location + match.range.length
|
|
}
|
|
} else {
|
|
matchEnd = match.range.location + match.range.length
|
|
}
|
|
outerMatch = regex.firstMatch(in: string, options: .withoutAnchoringBounds, range: NSRange(location: matchEnd, length: string.count - matchEnd))
|
|
}
|
|
let result = NSMutableAttributedString(string: string, attributes: attributes)
|
|
mentions.forEach { mention in
|
|
guard userLinkedDeviceHexEncodedPublicKeys.contains(mention.hexEncodedPublicKey) else { return }
|
|
result.addAttribute(.foregroundColor, value: Colors.accent, range: mention.range)
|
|
result.addAttribute(.font, value: UIFont.boldSystemFont(ofSize: Values.mediumFontSize), range: mention.range)
|
|
}
|
|
return result
|
|
}
|
|
}
|