Swiftgram/TelegramUI/Pasteboard.swift

123 lines
5.7 KiB
Swift

import Foundation
import UIKit
import Display
import TelegramCore
import MobileCoreServices
private func rtfStringWithAppliedEntities(_ text: String, entities: [MessageTextEntity]) -> String {
var string: String = #"""
{\rtf1\ansi\ansicpg1252{\fonttbl\f0\fnil\fcharset0 .SFUIText;\f1\fnil\fcharset0 .SFUIText-Semibold;\f2\fnil\fcharset0 .SFUIText-Italic;\f3\fnil\fcharset0 Menlo-Regular;}
"""#
string.append("\\f0 ")
let nsString = text as NSString
var remainingRange = NSMakeRange(0, text.count)
for i in 0 ..< entities.count {
let entity = entities[i]
var range = NSRange(location: entity.range.lowerBound, length: entity.range.upperBound - entity.range.lowerBound)
if range.location + range.length > text.count {
range.location = max(0, text.count - range.length)
range.length = text.count - range.location
}
if range.location != remainingRange.location {
string.append(nsString.substring(with: NSMakeRange(remainingRange.location, range.location - remainingRange.location)))
remainingRange = NSMakeRange(range.location, remainingRange.location + remainingRange.length - range.location)
}
switch entity.type {
// case .Url:
// string.addAttribute(NSAttributedStringKey.foregroundColor, value: linkColor, range: range)
// if underlineLinks {
// string.addAttribute(NSAttributedStringKey.underlineStyle, value: NSUnderlineStyle.styleSingle.rawValue as NSNumber, range: range)
// }
// if nsString == nil {
// nsString = text as NSString
// }
// string.addAttribute(NSAttributedStringKey(rawValue: TelegramTextAttributes.URL), value: nsString!.substring(with: range), range: range)
// case .Email:
// string.addAttribute(NSAttributedStringKey.foregroundColor, value: linkColor, range: range)
// if nsString == nil {
// nsString = text as NSString
// }
// if underlineLinks && underlineAllLinks {
// string.addAttribute(NSAttributedStringKey.underlineStyle, value: NSUnderlineStyle.styleSingle.rawValue as NSNumber, range: range)
// }
// string.addAttribute(NSAttributedStringKey(rawValue: TelegramTextAttributes.URL), value: "mailto:\(nsString!.substring(with: range))", range: range)
// case let .TextUrl(url):
// string.addAttribute(NSAttributedStringKey.foregroundColor, value: linkColor, range: range)
// if nsString == nil {
// nsString = text as NSString
// }
// if underlineLinks && underlineAllLinks {
// string.addAttribute(NSAttributedStringKey.underlineStyle, value: NSUnderlineStyle.styleSingle.rawValue as NSNumber, range: range)
// }
// string.addAttribute(NSAttributedStringKey(rawValue: TelegramTextAttributes.URL), value: url, range: range)
case .Bold:
string.append("\\f1\\b ")
string.append(nsString.substring(with: range))
string.append("\\b0\\f0 ")
case .Italic:
string.append("\\f2\\i ")
string.append(nsString.substring(with: range))
string.append("\\i0\\f0 ")
case .Code, .Pre:
string.append("\\f3 ")
string.append(nsString.substring(with: range))
string.append("\\f0 ")
default:
string.append(nsString.substring(with: range))
break
}
remainingRange = NSMakeRange(range.location + range.length, remainingRange.location + remainingRange.length - (range.location + range.length))
}
if remainingRange.length > 0 {
string.append(nsString.substring(with: NSMakeRange(remainingRange.location, remainingRange.length)))
}
string.append("}")
string = string.replacingOccurrences(of: "\n", with: "\\line")
return string
}
func chatInputStateStringFromRTF(_ data: Data, type: NSAttributedString.DocumentType) -> NSAttributedString? {
if let attributedString = try? NSAttributedString(data: data, options: [NSAttributedString.DocumentReadingOptionKey.documentType: type], documentAttributes: nil) {
let string = NSMutableAttributedString(string: attributedString.string)
attributedString.enumerateAttribute(.font, in: NSRange(location: 0, length: attributedString.length), options: [], using: { value, range, _ in
if let font = value as? UIFont {
let fontName = font.fontName.lowercased()
if fontName.contains("bold") {
string.addAttribute(ChatTextInputAttributes.bold, value: true as NSNumber, range: range)
} else if fontName.contains("italic") {
string.addAttribute(ChatTextInputAttributes.italic, value: true as NSNumber, range: range)
} else if fontName.contains("menlo") {
string.addAttribute(ChatTextInputAttributes.monospace, value: true as NSNumber, range: range)
}
}
})
return string
}
return nil
}
func storeMessageTextInPasteboard(_ text: String, entities: [MessageTextEntity]?) {
var items: [String: Any] = [:]
items[kUTTypeUTF8PlainText as String] = text
if let entities = entities {
items[kUTTypeRTF as String] = rtfStringWithAppliedEntities(text, entities: entities)
}
UIPasteboard.general.items = [items]
}
func storeInputTextInPasteboard(_ text: NSAttributedString) {
let entities = generateChatInputTextEntities(text)
storeMessageTextInPasteboard(text.string, entities: entities)
}