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.")
        }
    }
}