mirror of https://github.com/oxen-io/session-ios
Parse attachment pointers
nielsandriesse
4 years ago
parent
7d207ddfb7
commit
961878f74c
@ -0,0 +1,60 @@
|
||||
import CoreGraphics
|
||||
import SessionUtilitiesKit
|
||||
|
||||
public extension VisibleMessage {
|
||||
|
||||
@objc(SNAttachment)
|
||||
class Attachment : NSObject, NSCoding {
|
||||
public var fileName: String?
|
||||
public var contentType: String?
|
||||
public var key: Data?
|
||||
public var digest: Data?
|
||||
public var kind: Kind?
|
||||
public var caption: String?
|
||||
public var size: CGSize?
|
||||
public var sizeInBytes: UInt?
|
||||
public var url: String?
|
||||
|
||||
public var isValid: Bool {
|
||||
fileName != nil && contentType != nil && key != nil && digest != nil && kind != nil && size != nil && sizeInBytes != nil && url != nil
|
||||
}
|
||||
|
||||
public enum Kind : String {
|
||||
case voiceMessage, generic
|
||||
}
|
||||
|
||||
public override init() { super.init() }
|
||||
|
||||
public required init?(coder: NSCoder) {
|
||||
if let fileName = coder.decodeObject(forKey: "fileName") as! String? { self.fileName = fileName }
|
||||
if let contentType = coder.decodeObject(forKey: "contentType") as! String? { self.contentType = contentType }
|
||||
if let key = coder.decodeObject(forKey: "key") as! Data? { self.key = key }
|
||||
if let digest = coder.decodeObject(forKey: "digest") as! Data? { self.digest = digest }
|
||||
if let rawKind = coder.decodeObject(forKey: "kind") as! String? { self.kind = Kind(rawValue: rawKind) }
|
||||
if let caption = coder.decodeObject(forKey: "caption") as! String? { self.caption = caption }
|
||||
if let size = coder.decodeObject(forKey: "size") as! CGSize? { self.size = size }
|
||||
if let sizeInBytes = coder.decodeObject(forKey: "sizeInBytes") as! UInt? { self.sizeInBytes = sizeInBytes }
|
||||
if let url = coder.decodeObject(forKey: "url") as! String? { self.url = url }
|
||||
}
|
||||
|
||||
public func encode(with coder: NSCoder) {
|
||||
coder.encode(fileName, forKey: "fileName")
|
||||
coder.encode(contentType, forKey: "contentType")
|
||||
coder.encode(key, forKey: "key")
|
||||
coder.encode(digest, forKey: "digest")
|
||||
coder.encode(kind?.rawValue, forKey: "kind")
|
||||
coder.encode(caption, forKey: "caption")
|
||||
coder.encode(size, forKey: "size")
|
||||
coder.encode(sizeInBytes, forKey: "sizeInBytes")
|
||||
coder.encode(url, forKey: "url")
|
||||
}
|
||||
|
||||
public static func fromProto(_ proto: SNProtoAttachmentPointer) -> Attachment? {
|
||||
preconditionFailure("Use MessageReceiverDelegate.parseAttachments(from:) instead.")
|
||||
}
|
||||
|
||||
public func toProto() -> SNProtoDataMessageQuote? {
|
||||
fatalError("Not implemented.")
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,22 @@
|
||||
|
||||
extension TSAttachmentPointer {
|
||||
|
||||
public static func from(_ attachment: VisibleMessage.Attachment) -> TSAttachmentPointer {
|
||||
let kind: TSAttachmentType
|
||||
switch attachment.kind! {
|
||||
case .generic: kind = .default
|
||||
case .voiceMessage: kind = .voiceMessage
|
||||
}
|
||||
return TSAttachmentPointer(
|
||||
serverId: 0,
|
||||
key: attachment.key,
|
||||
digest: attachment.digest,
|
||||
byteCount: UInt32(attachment.sizeInBytes!),
|
||||
contentType: attachment.contentType!,
|
||||
sourceFilename: attachment.fileName,
|
||||
caption: attachment.caption,
|
||||
albumMessageId: nil,
|
||||
attachmentType: kind,
|
||||
mediaSize: attachment.size!)
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue