mirror of
https://github.com/Swiftgram/Telegram-iOS.git
synced 2025-06-16 05:55:20 +00:00
350 lines
20 KiB
Swift
350 lines
20 KiB
Swift
import Foundation
|
|
import UIKit
|
|
import Postbox
|
|
import TelegramCore
|
|
import Display
|
|
|
|
public func chatInputStateStringWithAppliedEntities(_ text: String, entities: [MessageTextEntity]) -> NSAttributedString {
|
|
var nsString: NSString?
|
|
let string = NSMutableAttributedString(string: text)
|
|
var skipEntity = false
|
|
let stringLength = string.length
|
|
for i in 0 ..< entities.count {
|
|
if skipEntity {
|
|
skipEntity = false
|
|
continue
|
|
}
|
|
let entity = entities[i]
|
|
var range = NSRange(location: entity.range.lowerBound, length: entity.range.upperBound - entity.range.lowerBound)
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
if range.location >= stringLength {
|
|
continue
|
|
}
|
|
if range.location + range.length > stringLength {
|
|
range.length = stringLength - range.location
|
|
}
|
|
switch entity.type {
|
|
case .Url, .Email, .PhoneNumber, .Mention, .Hashtag, .BotCommand:
|
|
break
|
|
case .Bold:
|
|
string.addAttribute(ChatTextInputAttributes.bold, value: true as NSNumber, range: range)
|
|
case .Italic:
|
|
string.addAttribute(ChatTextInputAttributes.italic, value: true as NSNumber, range: range)
|
|
case let .TextMention(peerId):
|
|
string.addAttribute(ChatTextInputAttributes.textMention, value: ChatTextInputTextMentionAttribute(peerId: peerId), range: range)
|
|
case let .TextUrl(url):
|
|
string.addAttribute(ChatTextInputAttributes.textUrl, value: ChatTextInputTextUrlAttribute(url: url), range: range)
|
|
case .Code, .Pre:
|
|
string.addAttribute(ChatTextInputAttributes.monospace, value: true as NSNumber, range: range)
|
|
case .Strikethrough:
|
|
string.addAttribute(ChatTextInputAttributes.strikethrough, value: true as NSNumber, range: range)
|
|
case .Underline:
|
|
string.addAttribute(ChatTextInputAttributes.underline, value: true as NSNumber, range: range)
|
|
case .Spoiler:
|
|
string.addAttribute(ChatTextInputAttributes.spoiler, value: true as NSNumber, range: range)
|
|
case let .CustomEmoji(_, fileId):
|
|
string.addAttribute(ChatTextInputAttributes.customEmoji, value: ChatTextInputTextCustomEmojiAttribute(interactivelySelectedFromPackId: nil, fileId: fileId, file: nil), range: range)
|
|
case .BlockQuote:
|
|
string.addAttribute(ChatTextInputAttributes.quote, value: ChatTextInputTextQuoteAttribute(), range: range)
|
|
default:
|
|
break
|
|
}
|
|
}
|
|
return string
|
|
}
|
|
|
|
public func stringWithAppliedEntities(_ text: String, entities: [MessageTextEntity], baseColor: UIColor, linkColor: UIColor, baseQuoteTintColor: UIColor? = nil, baseFont: UIFont, linkFont: UIFont, boldFont: UIFont, italicFont: UIFont, boldItalicFont: UIFont, fixedFont: UIFont, blockQuoteFont: UIFont, underlineLinks: Bool = true, external: Bool = false, message: Message?, entityFiles: [MediaId: TelegramMediaFile] = [:], adjustQuoteFontSize: Bool = false) -> NSAttributedString {
|
|
let baseQuoteTintColor = baseQuoteTintColor ?? baseColor
|
|
|
|
var nsString: NSString?
|
|
let string = NSMutableAttributedString(string: text, attributes: [NSAttributedString.Key.font: baseFont, NSAttributedString.Key.foregroundColor: baseColor])
|
|
var skipEntity = false
|
|
var underlineAllLinks = false
|
|
if linkColor.argb == baseColor.argb {
|
|
underlineAllLinks = true
|
|
}
|
|
var fontAttributes: [NSRange: ChatTextFontAttributes] = [:]
|
|
|
|
var nextBlockId = 0
|
|
|
|
var rangeOffset: Int = 0
|
|
for i in 0 ..< entities.count {
|
|
if skipEntity {
|
|
skipEntity = false
|
|
continue
|
|
}
|
|
let stringLength = string.length
|
|
let entity = entities[i]
|
|
var range = NSRange(location: entity.range.lowerBound + rangeOffset, length: entity.range.upperBound - entity.range.lowerBound)
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
if range.location > stringLength {
|
|
continue
|
|
} else if range.location + range.length > stringLength {
|
|
range.length = stringLength - range.location
|
|
}
|
|
switch entity.type {
|
|
case .Url:
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: range)
|
|
if underlineLinks {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
}
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.URL), value: nsString!.substring(with: range), range: range)
|
|
case .Email:
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: range)
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
if underlineLinks && underlineAllLinks {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
}
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.URL), value: "mailto:\(nsString!.substring(with: range))", range: range)
|
|
case .PhoneNumber:
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: range)
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
if underlineLinks && underlineAllLinks {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
}
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.URL), value: "tel:\(nsString!.substring(with: range))", range: range)
|
|
case let .TextUrl(url):
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: range)
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
if underlineLinks && underlineAllLinks {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
}
|
|
if external {
|
|
string.addAttribute(NSAttributedString.Key.link, value: url, range: range)
|
|
} else {
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.URL), value: url, range: range)
|
|
}
|
|
case .Bold:
|
|
if let fontAttribute = fontAttributes[range] {
|
|
fontAttributes[range] = fontAttribute.union(.bold)
|
|
} else {
|
|
fontAttributes[range] = .bold
|
|
}
|
|
case .Italic:
|
|
if let fontAttribute = fontAttributes[range] {
|
|
fontAttributes[range] = fontAttribute.union(.italic)
|
|
} else {
|
|
fontAttributes[range] = .italic
|
|
}
|
|
case .Mention:
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: range)
|
|
if underlineLinks && underlineAllLinks {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
}
|
|
if linkFont !== baseFont {
|
|
string.addAttribute(NSAttributedString.Key.font, value: linkFont, range: range)
|
|
}
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.PeerTextMention), value: nsString!.substring(with: range), range: range)
|
|
case .Strikethrough:
|
|
string.addAttribute(NSAttributedString.Key.strikethroughStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
case .Underline:
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
case let .TextMention(peerId):
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: range)
|
|
if underlineLinks && underlineAllLinks {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
}
|
|
if linkFont !== baseFont {
|
|
string.addAttribute(NSAttributedString.Key.font, value: linkFont, range: range)
|
|
}
|
|
let mention = nsString!.substring(with: range)
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.PeerMention), value: TelegramPeerMention(peerId: peerId, mention: mention), range: range)
|
|
case .Hashtag:
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
let hashtag = nsString!.substring(with: range)
|
|
if i + 1 != entities.count {
|
|
if case .Mention = entities[i + 1].type {
|
|
let nextRange = NSRange(location: entities[i + 1].range.lowerBound, length: entities[i + 1].range.upperBound - entities[i + 1].range.lowerBound)
|
|
if nextRange.location == range.location + range.length + 1 && nsString!.character(at: range.location + range.length) == 43 {
|
|
skipEntity = true
|
|
if nextRange.length > 0 {
|
|
let peerName: String = nsString!.substring(with: NSRange(location: nextRange.location + 1, length: nextRange.length - 1))
|
|
|
|
let combinedRange = NSRange(location: range.location, length: nextRange.location + nextRange.length - range.location)
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: combinedRange)
|
|
if linkColor.isEqual(baseColor) {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: combinedRange)
|
|
}
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.Hashtag), value: TelegramHashtag(peerName: peerName, hashtag: hashtag), range: combinedRange)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if !skipEntity {
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: range)
|
|
if underlineLinks && underlineAllLinks {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
}
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.Hashtag), value: TelegramHashtag(peerName: nil, hashtag: hashtag), range: range)
|
|
}
|
|
case .BotCommand:
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: range)
|
|
if underlineLinks && underlineAllLinks {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
}
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.BotCommand), value: nsString!.substring(with: range), range: range)
|
|
case .Pre:
|
|
string.addAttribute(NSAttributedString.Key.font, value: fixedFont, range: range)
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.Pre), value: nsString!.substring(with: range), range: range)
|
|
case .BlockQuote, .Code:
|
|
if let fontAttribute = fontAttributes[range] {
|
|
fontAttributes[range] = fontAttribute.union(.blockQuote)
|
|
} else {
|
|
fontAttributes[range] = .blockQuote
|
|
}
|
|
|
|
let paragraphBreak = "\n"
|
|
|
|
let paragraphRange: NSRange
|
|
if range.lowerBound == 0 {
|
|
paragraphRange = NSRange(location: range.lowerBound, length: range.upperBound - range.lowerBound)
|
|
} else if string.string[string.string.index(string.string.startIndex, offsetBy: range.lowerBound)] == "\n" {
|
|
paragraphRange = NSRange(location: range.lowerBound + 1, length: range.upperBound - range.lowerBound - 1)
|
|
} else if string.string[string.string.index(string.string.startIndex, offsetBy: range.lowerBound - 1)] == "\n" {
|
|
paragraphRange = NSRange(location: range.lowerBound, length: range.upperBound - range.lowerBound)
|
|
} else {
|
|
string.insert(NSAttributedString(string: paragraphBreak), at: range.lowerBound)
|
|
paragraphRange = NSRange(location: range.lowerBound + paragraphBreak.count, length: range.upperBound - range.lowerBound)
|
|
}
|
|
|
|
string.addAttribute(NSAttributedString.Key(rawValue: "Attribute__Blockquote"), value: TextNodeBlockQuoteData(id: nextBlockId, title: nil, color: baseQuoteTintColor), range: paragraphRange)
|
|
nextBlockId += 1
|
|
|
|
if string.string.index(string.string.startIndex, offsetBy: paragraphRange.upperBound) != string.string.endIndex {
|
|
if string.string[string.string.index(string.string.startIndex, offsetBy: paragraphRange.upperBound)] == "\n" {
|
|
string.replaceCharacters(in: NSMakeRange(paragraphRange.upperBound, 1), with: "")
|
|
rangeOffset -= 1
|
|
}
|
|
}
|
|
|
|
rangeOffset += 0
|
|
//rangeOffset += paragraphBreak.count
|
|
case .BankCard:
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: range)
|
|
if underlineLinks && underlineAllLinks {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
}
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.BankCard), value: nsString!.substring(with: range), range: range)
|
|
case .Spoiler:
|
|
if external {
|
|
string.addAttribute(NSAttributedString.Key.backgroundColor, value: UIColor.gray, range: range)
|
|
} else {
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.Spoiler), value: true as NSNumber, range: range)
|
|
}
|
|
case let .Custom(type):
|
|
if type == ApplicationSpecificEntityType.Timecode {
|
|
string.addAttribute(NSAttributedString.Key.foregroundColor, value: linkColor, range: range)
|
|
if underlineLinks && underlineAllLinks {
|
|
string.addAttribute(NSAttributedString.Key.underlineStyle, value: NSUnderlineStyle.single.rawValue as NSNumber, range: range)
|
|
}
|
|
if nsString == nil {
|
|
nsString = text as NSString
|
|
}
|
|
let text = nsString!.substring(with: range)
|
|
if let time = parseTimecodeString(text) {
|
|
string.addAttribute(NSAttributedString.Key(rawValue: TelegramTextAttributes.Timecode), value: TelegramTimecode(time: time, text: text), range: range)
|
|
}
|
|
}
|
|
case let .CustomEmoji(_, fileId):
|
|
let mediaId = MediaId(namespace: Namespaces.Media.CloudFile, id: fileId)
|
|
var emojiFile: TelegramMediaFile?
|
|
if let file = message?.associatedMedia[mediaId] as? TelegramMediaFile {
|
|
emojiFile = file
|
|
} else {
|
|
emojiFile = entityFiles[mediaId]
|
|
}
|
|
string.addAttribute(ChatTextInputAttributes.customEmoji, value: ChatTextInputTextCustomEmojiAttribute(interactivelySelectedFromPackId: nil, fileId: fileId, file: emojiFile), range: range)
|
|
default:
|
|
break
|
|
}
|
|
|
|
var addedAttributes: [(NSRange, ChatTextFontAttributes)] = []
|
|
func addFont(ranges: [NSRange], fontAttributes: ChatTextFontAttributes) {
|
|
for range in ranges {
|
|
var font: UIFont?
|
|
|
|
if fontAttributes == [.bold, .italic] {
|
|
font = boldItalicFont
|
|
} else if fontAttributes == [.bold] {
|
|
font = boldFont
|
|
addedAttributes.append((range, fontAttributes))
|
|
} else if fontAttributes == [.italic] {
|
|
font = italicFont
|
|
addedAttributes.append((range, fontAttributes))
|
|
} else {
|
|
font = baseFont
|
|
}
|
|
|
|
if adjustQuoteFontSize, let fontValue = font, fontAttributes.contains(.blockQuote) {
|
|
font = fontValue.withSize(round(fontValue.pointSize * 0.8235294117647058))
|
|
}
|
|
|
|
if let font = font {
|
|
string.addAttribute(NSAttributedString.Key.font, value: font, range: range)
|
|
}
|
|
}
|
|
}
|
|
|
|
for (range, fontAttributes) in fontAttributes {
|
|
var ranges = [range]
|
|
var fontAttributes = fontAttributes
|
|
if fontAttributes != [.bold, .italic] {
|
|
for (existingRange, existingAttributes) in addedAttributes {
|
|
if let intersection = existingRange.intersection(range) {
|
|
if intersection.length == range.length {
|
|
if existingAttributes == .bold || existingAttributes == .italic {
|
|
fontAttributes.insert(existingAttributes)
|
|
}
|
|
} else {
|
|
var fontAttributes = fontAttributes
|
|
if existingAttributes == .bold || existingAttributes == .italic {
|
|
fontAttributes.insert(existingAttributes)
|
|
}
|
|
addFont(ranges: [intersection], fontAttributes: fontAttributes)
|
|
|
|
ranges = []
|
|
if range.upperBound > existingRange.lowerBound {
|
|
ranges.append(NSRange(location: range.lowerBound, length: existingRange.lowerBound - range.lowerBound))
|
|
}
|
|
if range.upperBound > existingRange.upperBound {
|
|
ranges.append(NSRange(location: existingRange.upperBound, length: range.upperBound - existingRange.upperBound))
|
|
}
|
|
}
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
addFont(ranges: ranges, fontAttributes: fontAttributes)
|
|
}
|
|
}
|
|
return string
|
|
}
|