mirror of https://github.com/oxen-io/session-ios
Merge branch 'charlesmchen/linkPreviews2'
Matthew Chen
6 years ago
commit
74ccee26a8
@ -1 +1 @@
|
||||
Subproject commit b46f53105951613fedd8117b83931f43f6ab12f1
|
||||
Subproject commit ea60f60ea01bc51fc2434248890b494e37da98a5
|
||||
@ -0,0 +1,100 @@
|
||||
//
|
||||
// Copyright (c) 2019 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
import Foundation
|
||||
|
||||
@objc(OWSLinkPreview)
|
||||
public class OWSLinkPreview: MTLModel {
|
||||
@objc
|
||||
public var urlString: String?
|
||||
|
||||
@objc
|
||||
public var title: String?
|
||||
|
||||
@objc
|
||||
public var attachmentId: String?
|
||||
|
||||
@objc
|
||||
public init(urlString: String, title: String?, attachmentId: String?) {
|
||||
self.urlString = urlString
|
||||
self.title = title
|
||||
self.attachmentId = attachmentId
|
||||
|
||||
super.init()
|
||||
}
|
||||
|
||||
@objc
|
||||
public required init!(coder: NSCoder) {
|
||||
super.init(coder: coder)
|
||||
}
|
||||
|
||||
@objc
|
||||
public required init(dictionary dictionaryValue: [AnyHashable: Any]!) throws {
|
||||
try super.init(dictionary: dictionaryValue)
|
||||
}
|
||||
|
||||
@objc
|
||||
public class func buildValidatedLinkPreview(dataMessage: SSKProtoDataMessage,
|
||||
body: String?,
|
||||
transaction: YapDatabaseReadWriteTransaction) -> OWSLinkPreview? {
|
||||
guard let previewProto = dataMessage.preview else {
|
||||
return nil
|
||||
}
|
||||
let urlString = previewProto.url
|
||||
|
||||
guard URL(string: urlString) != nil else {
|
||||
owsFailDebug("Could not parse preview URL.")
|
||||
return nil
|
||||
}
|
||||
|
||||
guard let body = body else {
|
||||
owsFailDebug("Preview for message without body.")
|
||||
return nil
|
||||
}
|
||||
let bodyComponents = body.components(separatedBy: .whitespacesAndNewlines)
|
||||
guard bodyComponents.contains(urlString) else {
|
||||
owsFailDebug("URL not present in body.")
|
||||
return nil
|
||||
}
|
||||
|
||||
// TODO: Verify that url host is in whitelist.
|
||||
|
||||
let title: String? = previewProto.title?.trimmingCharacters(in: .whitespacesAndNewlines)
|
||||
|
||||
var imageAttachmentId: String?
|
||||
if let imageProto = previewProto.image {
|
||||
if let imageAttachmentPointer = TSAttachmentPointer(fromProto: imageProto, albumMessage: nil) {
|
||||
imageAttachmentPointer.save(with: transaction)
|
||||
imageAttachmentId = imageAttachmentPointer.uniqueId
|
||||
} else {
|
||||
owsFailDebug("Could not parse image proto.")
|
||||
}
|
||||
}
|
||||
|
||||
var hasTitle = false
|
||||
if let titleValue = title {
|
||||
hasTitle = titleValue.count > 0
|
||||
}
|
||||
let hasImage = imageAttachmentId != nil
|
||||
if !hasTitle && !hasImage {
|
||||
owsFailDebug("Preview has neither title nor image.")
|
||||
return nil
|
||||
}
|
||||
|
||||
return OWSLinkPreview(urlString: urlString, title: title, attachmentId: imageAttachmentId)
|
||||
}
|
||||
|
||||
@objc
|
||||
public func removeAttachment(transaction: YapDatabaseReadWriteTransaction) {
|
||||
guard let attachmentId = attachmentId else {
|
||||
owsFailDebug("No attachment id.")
|
||||
return
|
||||
}
|
||||
guard let attachment = TSAttachment.fetch(uniqueId: attachmentId, transaction: transaction) else {
|
||||
owsFailDebug("Could not load attachment.")
|
||||
return
|
||||
}
|
||||
attachment.remove(with: transaction)
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue