another.im-ios/ConversationsClassic/AppCore/Models/Attachment.swift

import Foundation
import GRDB
import Martin
import SwiftUI

enum AttachmentType: Int, Stateable, DatabaseValueConvertible {
    case movie = 0
    case image = 1
    case audio = 2
    case file = 3
}

struct Attachment: DBStorable {
    static let databaseTableName = "attachments"

    let id: String
    let type: AttachmentType
    let localPath: URL?
    let remotePath: URL?
    let localThumbnailPath: URL?
    let messageId: String
    var downloadFailed: Bool = false

    static let message = belongsTo(Message.self)
    var message: QueryInterfaceRequest<Message> {
        request(for: Attachment.message)
    }
}

extension Attachment: Equatable {}

extension String {
    var attachmentType: AttachmentType {
        let ext = (self as NSString).pathExtension.lowercased()

        switch ext {
        case "mov", "mp4", "avi":
            return .movie

        case "jpg", "png", "gif":
            return .image

        case "mp3", "wav", "m4a":
            return .audio

        case "txt", "doc", "pdf":
            return .file

        default:
            return .file
        }
    }
}