Swiftgram/TelegramUI/MessageContentKind.swift

import Foundation
import Postbox
import TelegramCore

enum MessageContentKind: Equatable {
    case text(String)
    case image
    case video
    case videoMessage
    case audioMessage
    case sticker(String)
    case animation
    case file(String)
    case contact
    case game(String)
    case location
    case liveLocation
    case expiredImage
    case expiredVideo
    
    static func ==(lhs: MessageContentKind, rhs: MessageContentKind) -> Bool {
        switch lhs {
            case let .text(text):
                if case .text(text) = rhs {
                    return true
                } else {
                    return false
                }
            case .image:
                if case .image = rhs {
                    return true
                } else {
                    return false
                }
            case .video:
                if case .video = rhs {
                    return true
                } else {
                    return false
                }
            case .videoMessage:
                if case .videoMessage = rhs {
                    return true
                } else {
                    return false
                }
            case .audioMessage:
                if case .audioMessage = rhs {
                    return true
                } else {
                    return false
                }
            case let .sticker(text):
                if case .sticker(text) = rhs {
                    return true
                } else {
                    return false
                }
            case .animation:
                if case .animation = rhs {
                    return true
                } else {
                    return false
                }
            case let .file(text):
                if case .file(text) = rhs {
                    return true
                } else {
                    return false
                }
            case .contact:
                if case .contact = rhs {
                    return true
                } else {
                    return false
                }
            case let .game(text):
                if case .game(text) = rhs {
                    return true
                } else {
                    return false
                }
            case .location:
                if case .location = rhs {
                    return true
                } else {
                    return false
                }
            case .liveLocation:
                if case .liveLocation = rhs {
                    return true
                } else {
                    return false
                }
            case .expiredImage:
                if case .expiredImage = rhs {
                    return true
                } else {
                    return false
                }
            case .expiredVideo:
                if case .expiredVideo = rhs {
                    return true
                } else {
                    return false
                }
        }
    }
}

func messageContentKind(_ message: Message, strings: PresentationStrings, accountPeerId: PeerId) -> MessageContentKind {
    for media in message.media {
        switch media {
            case let expiredMedia as TelegramMediaExpiredContent:
                switch expiredMedia.data {
                    case .image:
                        return .expiredImage
                    case .file:
                        return .expiredVideo
                }
            case _ as TelegramMediaImage:
                return .image
            case let file as TelegramMediaFile:
                var fileName: String = ""
                for attribute in file.attributes {
                    switch attribute {
                        case let .Sticker(text, _, _):
                            return .sticker(text)
                        case let .FileName(name):
                            fileName = name
                        case let .Audio(isVoice, _, title, performer, _):
                            if isVoice {
                                return .audioMessage
                            } else {
                                if let title = title, let performer = performer, !title.isEmpty, !performer.isEmpty {
                                    return .file(title + " — " + performer)
                                } else if let title = title, !title.isEmpty {
                                    return .file(title)
                                } else if let performer = performer, !performer.isEmpty {
                                    return .file(performer)
                                }
                            }
                        case let .Video(_, _, flags):
                            if file.isAnimated {
                                return .animation
                            } else {
                                if flags.contains(.instantRoundVideo) {
                                    return .videoMessage
                                } else {
                                    return .video
                                }
                            }
                        default:
                            break
                    }
                }
                return .file(fileName)
            case _ as TelegramMediaContact:
                return .contact
            case let game as TelegramMediaGame:
                return .game(game.title)
            case let location as TelegramMediaMap:
                if location.liveBroadcastingTimeout != nil {
                    return .liveLocation
                } else {
                    return .location
                }
            case _ as TelegramMediaAction:
                return .text(plainServiceMessageString(strings: strings, message: message, accountPeerId: accountPeerId) ?? "")
            default:
                break
        }
    }
    return .text(message.text)
}
 
func descriptionStringForMessage(_ message: Message, strings: PresentationStrings, accountPeerId: PeerId) -> (String, Bool) {
    if !message.text.isEmpty {
        return (message.text, false)
    }
    switch messageContentKind(message, strings: strings, accountPeerId: accountPeerId) {
        case let .text(text):
            return (text, false)
        case .image:
            return (strings.Message_Photo, true)
        case .video:
            return (strings.Message_Video, true)
        case .videoMessage:
            return (strings.Message_VideoMessage, true)
        case .audioMessage:
            return (strings.Message_Audio, true)
        case let .sticker(text):
            if text.isEmpty {
                return (strings.Message_Sticker, true)
            } else {
                return ("\(text) \(strings.Message_Sticker)", true)
            }
        case .animation:
            return (strings.Message_Animation, true)
        case let .file(text):
            if text.isEmpty {
                return (strings.Message_File, true)
            } else {
                return (text, true)
            }
        case .contact:
            return (strings.Message_Contact, true)
        case let .game(text):
            return (text, true)
        case .location:
            return (strings.Message_Location, true)
        case .liveLocation:
            return (strings.Message_LiveLocation, true)
        case .expiredImage:
            return (strings.Message_ImageExpired, true)
        case .expiredVideo:
            return (strings.Message_VideoExpired, true)
    }
}