Swiftgram/submodules/TelegramCore/Sources/ApiUtils/TextEntitiesMessageAttribute.swift

import Foundation
import Postbox
import TelegramApi


func apiEntitiesFromMessageTextEntities(_ entities: [MessageTextEntity], associatedPeers: SimpleDictionary<PeerId, Peer>) -> [Api.MessageEntity] {
    var apiEntities: [Api.MessageEntity] = []
    
    for entity in entities {
        let offset: Int32 = Int32(entity.range.lowerBound)
        let length: Int32 = Int32(entity.range.upperBound - entity.range.lowerBound)
        switch entity.type {
        case .Unknown:
            break
        case .Mention:
            apiEntities.append(.messageEntityMention(.init(offset: offset, length: length)))
        case .Hashtag:
            apiEntities.append(.messageEntityHashtag(.init(offset: offset, length: length)))
        case .BotCommand:
            apiEntities.append(.messageEntityBotCommand(.init(offset: offset, length: length)))
        case .Url:
            apiEntities.append(.messageEntityUrl(.init(offset: offset, length: length)))
        case .Email:
            apiEntities.append(.messageEntityEmail(.init(offset: offset, length: length)))
        case .Bold:
            apiEntities.append(.messageEntityBold(.init(offset: offset, length: length)))
        case .Italic:
            apiEntities.append(.messageEntityItalic(.init(offset: offset, length: length)))
        case .Code:
            apiEntities.append(.messageEntityCode(.init(offset: offset, length: length)))
        case let .Pre(language):
            apiEntities.append(.messageEntityPre(.init(offset: offset, length: length, language: language ?? "")))
        case let .TextUrl(url):
            apiEntities.append(.messageEntityTextUrl(.init(offset: offset, length: length, url: url)))
        case let .TextMention(peerId):
            if let peer = associatedPeers[peerId], let inputUser = apiInputUser(peer) {
                apiEntities.append(.inputMessageEntityMentionName(.init(offset: offset, length: length, userId: inputUser)))
            }
        case .PhoneNumber:
            break
        case .Strikethrough:
            apiEntities.append(.messageEntityStrike(.init(offset: offset, length: length)))
        case let .BlockQuote(isCollapsed):
            var flags: Int32 = 0
            if isCollapsed {
                flags |= 1 << 0
            }
            apiEntities.append(.messageEntityBlockquote(.init(flags: flags, offset: offset, length: length)))
        case .Underline:
            apiEntities.append(.messageEntityUnderline(.init(offset: offset, length: length)))
        case .BankCard:
            apiEntities.append(.messageEntityBankCard(.init(offset: offset, length: length)))
        case .Spoiler:
            apiEntities.append(.messageEntitySpoiler(.init(offset: offset, length: length)))
        case let .CustomEmoji(_, fileId):
            apiEntities.append(.messageEntityCustomEmoji(.init(offset: offset, length: length, documentId: fileId)))
        case let .FormattedDate(format, date):
            var flags: Int32 = 0
            switch format {
            case .relative:
                flags |= 1 << 0
            case let .full(timeFormat, dateFormat, dayOfWeek):
                switch timeFormat {
                case .short:
                    flags |= 1 << 1
                case .long:
                    flags |= 1 << 2
                default:
                    break
                }
                switch dateFormat {
                case .short:
                    flags |= 1 << 3
                case .long:
                    flags |= 1 << 4
                default:
                    break
                }
                if dayOfWeek {
                    flags |= 1 << 5
                }
            default:
                break
            }
            apiEntities.append(.messageEntityFormattedDate(.init(flags: flags, offset: offset, length: length, date: date)))
        case .Custom:
            break
        }
    }
    
    return apiEntities
}

func apiTextAttributeEntities(_ attribute: TextEntitiesMessageAttribute, associatedPeers: SimpleDictionary<PeerId, Peer>) -> [Api.MessageEntity] {
    return apiEntitiesFromMessageTextEntities(attribute.entities, associatedPeers: associatedPeers)
}