conversations-classic-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

    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 // Default to .file if the extension is not recognized
        }
    }
}