mirror of
https://github.com/vector-im/element-ios.git
synced 2024-09-29 07:42:40 +00:00
178 lines
7.3 KiB
Swift
178 lines
7.3 KiB
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
|
|
import UIKit
|
|
|
|
/// Provides utilities funcs to handle Pills inside attributed strings.
|
|
@available (iOS 15.0, *)
|
|
@objcMembers
|
|
class PillsFormatter: NSObject {
|
|
// MARK: - Internal Properties
|
|
/// UTType identifier for pills. Should be declared as Document type & Exported type identifier inside Info.plist
|
|
static let pillUTType: String = "im.vector.app.pills"
|
|
|
|
// MARK: - Internal Enums
|
|
/// Defines a replacement mode for converting Pills to plain text.
|
|
@objc enum PillsReplacementTextMode: Int {
|
|
case displayname
|
|
case identifier
|
|
case markdown
|
|
}
|
|
|
|
// MARK: - Internal Methods
|
|
/// Insert text attachments for pills inside given message attributed string.
|
|
///
|
|
/// - Parameters:
|
|
/// - attributedString: message string to update
|
|
/// - session: current session
|
|
/// - eventFormatter: the event formatter
|
|
/// - event: the event
|
|
/// - roomState: room state for message
|
|
/// - latestRoomState: latest room state of the room containing this message
|
|
/// - isEditMode: whether this string will be used in the composer
|
|
/// - Returns: new attributed string with pills
|
|
static func insertPills(in attributedString: NSAttributedString,
|
|
withSession session: MXSession,
|
|
eventFormatter: MXKEventFormatter,
|
|
event: MXEvent,
|
|
roomState: MXRoomState,
|
|
andLatestRoomState latestRoomState: MXRoomState?,
|
|
isEditMode: Bool = false) -> NSAttributedString {
|
|
|
|
let newAttr = NSMutableAttributedString(attributedString: attributedString)
|
|
newAttr.vc_enumerateAttribute(.link) { (url: URL, range: NSRange, _) in
|
|
|
|
let provider = PillProvider(withSession: session,
|
|
eventFormatter: eventFormatter,
|
|
event: event,
|
|
roomState: roomState,
|
|
andLatestRoomState: latestRoomState,
|
|
isEditMode: isEditMode)
|
|
|
|
// try to get a mention pill from the url
|
|
let label = Range(range, in: newAttr.string).flatMap { String(newAttr.string[$0]) }
|
|
if let attachmentString: NSAttributedString = provider.pillTextAttachmentString(forUrl: url, withLabel: label ?? "", event: event) {
|
|
// replace the url with the pill
|
|
newAttr.replaceCharacters(in: range, with: attachmentString)
|
|
}
|
|
}
|
|
|
|
return newAttr
|
|
}
|
|
|
|
/// Creates a string with all pills of given attributed string replaced by display names.
|
|
///
|
|
/// - Parameters:
|
|
/// - attributedString: attributed string with pills
|
|
/// - mode: replacement mode for pills (default: displayname)
|
|
/// - Returns: string with display names
|
|
static func stringByReplacingPills(in attributedString: NSAttributedString,
|
|
mode: PillsReplacementTextMode = .displayname) -> String {
|
|
let newAttr = NSMutableAttributedString(attributedString: attributedString)
|
|
newAttr.vc_enumerateAttribute(.attachment) { (attachment: PillTextAttachment, range: NSRange, _) in
|
|
guard let data = attachment.data else {
|
|
return
|
|
}
|
|
|
|
var pillString: String
|
|
switch mode {
|
|
case .displayname:
|
|
pillString = data.displayText
|
|
case .identifier:
|
|
pillString = data.pillIdentifier
|
|
case .markdown:
|
|
pillString = data.markdown
|
|
}
|
|
|
|
newAttr.replaceCharacters(in: range, with: pillString)
|
|
}
|
|
|
|
return newAttr.string
|
|
}
|
|
|
|
|
|
/// Creates an attributed string containing a pill for given room member.
|
|
///
|
|
/// - Parameters:
|
|
/// - roomMember: the room member
|
|
/// - url: URL to room member profile. Should be provided to make pill act as a link.
|
|
/// - isHighlighted: true to indicate that the pill should be highlighted
|
|
/// - font: the text font
|
|
/// - Returns: attributed string with a pill attachment and an optional link
|
|
static func mentionPill(withRoomMember roomMember: MXRoomMember,
|
|
andUrl url: URL? = nil,
|
|
isHighlighted: Bool,
|
|
font: UIFont) -> NSAttributedString {
|
|
|
|
guard let attachment = PillTextAttachment(withRoomMember: roomMember, isHighlighted: isHighlighted, font: font) else {
|
|
return NSAttributedString(string: roomMember.displayname)
|
|
}
|
|
return attributedStringWithAttachment(attachment, link: url, font: font)
|
|
}
|
|
|
|
/// Update alpha of all `PillTextAttachment` contained in given attributed string.
|
|
///
|
|
/// - Parameters:
|
|
/// - alpha: Alpha value to apply
|
|
/// - attributedString: Attributed string containing the pills
|
|
static func setPillAlpha(_ alpha: CGFloat, inAttributedString attributedString: NSAttributedString) {
|
|
attributedString.vc_enumerateAttribute(.attachment) { (pill: PillTextAttachment, range: NSRange, _) in
|
|
pill.data?.alpha = alpha
|
|
}
|
|
}
|
|
|
|
/// Refresh pills inside given attributed string.
|
|
///
|
|
/// - Parameters:
|
|
/// - attributedString: attributed string to update
|
|
/// - roomState: room state for refresh, should be the latest available
|
|
static func refreshPills(in attributedString: NSAttributedString, with roomState: MXRoomState) {
|
|
attributedString.vc_enumerateAttribute(.attachment) { (pill: PillTextAttachment, range: NSRange, _) in
|
|
|
|
switch pill.data?.pillType {
|
|
case .user(let userId):
|
|
guard let roomMember = roomState.members.member(withUserId: userId) else {
|
|
return
|
|
}
|
|
|
|
pill.data?.items = [
|
|
.avatar(url: roomMember.avatarUrl,
|
|
string: roomMember.displayname,
|
|
matrixId: roomMember.userId),
|
|
.text(roomMember.displayname)
|
|
]
|
|
default:
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// MARK: - Private Methods
|
|
@available (iOS 15.0, *)
|
|
extension PillsFormatter {
|
|
|
|
static func attributedStringWithAttachment(_ attachment: PillTextAttachment, link: URL?, font: UIFont) -> NSAttributedString {
|
|
let string = NSMutableAttributedString(attachment: attachment)
|
|
string.addAttribute(.font, value: font, range: .init(location: 0, length: string.length))
|
|
if let url = link {
|
|
string.addAttribute(.link, value: url, range: .init(location: 0, length: string.length))
|
|
}
|
|
return string
|
|
}
|
|
}
|