element-ios/Riot/Modules/Pills/PillAttachmentViewProvider.swift

// 
// Copyright 2022 New Vector Ltd
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

import UIKit

/// Provider for mention Pills attachment view.
@available(iOS 15.0, *)
@objc class PillAttachmentViewProvider: NSTextAttachmentViewProvider {
    // MARK: - Properties
    static let pillAttachmentViewSizes = PillAttachmentView.Sizes(verticalMargin: 2.0,
                                                                  horizontalMargin: 6.0,
                                                                  avatarLeading: 2.0,
                                                                  avatarSideLength: 16.0,
                                                                  itemSpacing: 4)
    private weak var pillViewFlusher: PillViewFlusher?

    // MARK: - Override
    override init(textAttachment: NSTextAttachment, parentView: UIView?, textLayoutManager: NSTextLayoutManager?, location: NSTextLocation) {
        super.init(textAttachment: textAttachment, parentView: parentView, textLayoutManager: textLayoutManager, location: location)

        // Try to register a flusher for the pills.
        if let pillViewFlusher = parentView?.superview as? PillViewFlusher {
            self.pillViewFlusher = pillViewFlusher
        } else {
            MXLog.debug("[PillAttachmentViewProvider]: no handler found, pills will not be flushed properly")
        }
    }

    override func loadView() {
        super.loadView()

        guard let textAttachment = self.textAttachment as? PillTextAttachment else {
            MXLog.debug("[PillAttachmentViewProvider]: attachment is missing or not of expected class")
            return
        }

        guard let pillData = textAttachment.data else {
            MXLog.debug("[PillAttachmentViewProvider]: attachment misses pill data")
            return
        }

        let mainSession = AppDelegate.theDelegate().mxSessions.first as? MXSession

        let pillView = PillAttachmentView(frame: CGRect(origin: .zero, size: textAttachment.size(forFont: pillData.font)),
                                          sizes: Self.pillAttachmentViewSizes,
                                          theme: ThemeService.shared().theme,
                                          mediaManager: mainSession?.mediaManager,
                                          andPillData: pillData)
        view = pillView
        pillViewFlusher?.registerPillView(pillView)
    }
}