threema-ios/Threema/ChatFileVideoMessageCell.swift

//  _____ _
// |_   _| |_  _ _ ___ ___ _ __  __ _
//   | | | ' \| '_/ -_) -_) '  \/ _` |_
//   |_| |_||_|_| \___\___|_|_|_\__,_(_)
//
// Threema iOS Client
// Copyright (c) 2019-2020 Threema GmbH
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License, version 3,
// as published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

import Foundation
import CocoaLumberjackSwift

@objc open class ChatFileVideoMessageCell: ChatBlobTextMessageCell {
    override open var message: BaseMessage! {
        didSet {
            setBaseMessage(newMessage: message)
        }
    }
    
    private var _thumbnailView: UIImageView?
    private var _durationBackground: UIImageView?
    private var _downloadBackground: UIImageView?
    private var _playImageView: UIImageView?
    private var _durationLabel: UILabel?
    private var _downloadSizeLabel: UILabel?
    
    private var _observedMessages:[Data] = [Data]()
    
    @objc override public init!(style: UITableViewCell.CellStyle, reuseIdentifier: String!, transparent: Bool) {
        super.init(style: style, reuseIdentifier: reuseIdentifier, transparent: transparent)
        
        _thumbnailView = UIImageView.init()
        _thumbnailView!.clearsContextBeforeDrawing = false
        contentView.addSubview(_thumbnailView!)
        
        _durationBackground = UIImageView.init(image: UIImage.init(named: "VideoDurationBg")?.resizableImage(withCapInsets: UIEdgeInsets.init(top: 0, left: 32, bottom: 0, right: 0)))
        _durationBackground?.isOpaque = false
        _thumbnailView!.addSubview(_durationBackground!)
        
        _downloadBackground = UIImageView.init(image: UIImage.init(named: "VideoDownloadBg")?.resizableImage(withCapInsets: UIEdgeInsets.init(top: 0, left: 32, bottom: 0, right: 0)))
        _downloadBackground!.isOpaque = false
        _thumbnailView!.addSubview(_downloadBackground!)
        
        _durationLabel = UILabel.init()
        _durationLabel!.backgroundColor = .clear
        _durationLabel!.isOpaque = false
        _durationLabel!.font = UIFont.boldSystemFont(ofSize: 12)
        _durationLabel!.textColor = Colors.white()
        _durationLabel!.textAlignment = .right
        _durationBackground?.addSubview(_durationLabel!)
            
        _downloadSizeLabel = UILabel.init()
        _downloadSizeLabel!.backgroundColor = .clear
        _downloadSizeLabel!.isOpaque = false
        _downloadSizeLabel!.font = UIFont.boldSystemFont(ofSize: 12)
        _downloadSizeLabel!.textColor = Colors.white()
        _downloadSizeLabel!.textAlignment = .right
        _downloadSizeLabel!.adjustsFontSizeToFitWidth = true
        _downloadBackground!.addSubview(_downloadSizeLabel!)
            
        if #available(iOS 11.0, *) {
            _thumbnailView?.accessibilityIgnoresInvertColors = true
        }
        
        if #available(iOS 13.0, *) {
            _playImageView = UIImageView.init(image: BundleUtil.imageNamed("Play")?.withTintColor(Colors.white()))
        } else {
            _playImageView = UIImageView.init(image: BundleUtil.imageNamed("Play")?.withTint(Colors.white()))
        }
        _thumbnailView!.addSubview(_playImageView!)
        
        _captionLabel = ChatTextMessageCell.makeAttributedLabel(withFrame: self.bounds)
        _captionLabel?.tapDelegate = self
        _captionLabel?.longPressDelegate = self
        contentView.addSubview(_captionLabel!)

        setupColors()
    }
    
    required public init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
}

extension ChatFileVideoMessageCell {
    // MARK: Override functions
    
    @objc override open class func height(for message: BaseMessage!, forTableWidth tableWidth: CGFloat) -> CGFloat {
        let fileMessage = message as! FileMessage
        
        var cellHeight: CGFloat = 40.0
        let imageInsets = UIEdgeInsets.init(top: 5, left: 5, bottom: 5, right: 5)
        var scaledSize = CGSize.init()
        if fileMessage.thumbnail != nil {
            if fileMessage.thumbnail.data != nil && fileMessage.thumbnail.height.floatValue > 0 {
                let width: CGFloat = CGFloat(fileMessage.thumbnail.width.floatValue)
                let height: CGFloat = CGFloat(fileMessage.thumbnail.height.floatValue)
                let size: CGSize = CGSize.init(width: width, height: height)
                scaledSize = ChatFileVideoMessageCell.scaleImageSize(toCell: size, forTableWidth: tableWidth)
                if scaledSize.height != scaledSize.height || scaledSize.height < 0 {
                    scaledSize.height = 120.0
                }
                cellHeight = scaledSize.height - 17.0
            }
        }
                
        if let caption = fileMessage.getCaption(), caption.count > 0 {
            let x: CGFloat = 30.0
            let maxSize = CGSize.init(width: scaledSize.width - x, height: CGFloat.greatestFiniteMagnitude)
            var textSize: CGSize?
            let captionTextNSString = NSString.init(string: caption)

            if UserSettings.shared().disableBigEmojis && captionTextNSString.isOnlyEmojisMaxCount(3) {
                var dummyLabelEmoji: ZSWTappableLabel? = nil
                if dummyLabelEmoji == nil {
                    dummyLabelEmoji = ChatTextMessageCell.makeAttributedLabel(withFrame: CGRect.init(x: (x/2), y: 0.0, width: maxSize.width, height: maxSize.height))
                }
                dummyLabelEmoji!.font = ChatTextMessageCell.emojiFont()
                dummyLabelEmoji?.attributedText = NSAttributedString.init(string: caption, attributes: [NSAttributedString.Key.font: ChatMessageCell.emojiFont()!])
                textSize = dummyLabelEmoji?.sizeThatFits(maxSize)
                textSize!.height = textSize!.height + 18.0
            } else {
                var dummyLabel: ZSWTappableLabel? = nil
                if dummyLabel == nil {
                    dummyLabel = ChatTextMessageCell.makeAttributedLabel(withFrame: CGRect.init(x: (x/2), y: 0.0, width: maxSize.width, height: maxSize.height))
                }
                dummyLabel!.font = ChatTextMessageCell.textFont()
                let attributed = TextStyleUtils.makeAttributedString(from: caption, with: dummyLabel!.font, textColor: Colors.fontNormal(), isOwn: true, application: UIApplication.shared)
                let formattedAttributeString = NSMutableAttributedString.init(attributedString: (dummyLabel!.applyMarkup(for: attributed))!)
                dummyLabel?.attributedText = TextStyleUtils.makeMentionsAttributedString(for: formattedAttributeString, textFont: dummyLabel!.font!, at: dummyLabel!.textColor.withAlphaComponent(0.4), messageInfo: Int32(message.isOwn!.intValue), application: UIApplication.shared)
                textSize = dummyLabel?.sizeThatFits(maxSize)
                textSize!.height = textSize!.height + 18.0
            }
            cellHeight = cellHeight + textSize!.height
        } else {
            cellHeight += imageInsets.top + imageInsets.bottom
        }
        
        return cellHeight
    }
        
    override public func layoutSubviews() {
        let fileMessage = message as! FileMessage
        let imageInsets = UIEdgeInsets.init(top: 5, left: 5, bottom: 5, right: 5)
        if (fileMessage.thumbnail != nil) {
            var size = CGSize.init(width: CGFloat(fileMessage.thumbnail.width.floatValue), height: CGFloat(fileMessage.thumbnail.height.floatValue))
            var textSize: CGSize = CGSize.init(width: 0.0, height: 0.0)
            let x: CGFloat = 30.0
            
            /* scale to fit maximum cell size */
            size = ChatFileVideoMessageCell.scaleImageSize(toCell: size, forTableWidth: frame.size.width)
            if size.height != size.height {
                size.height = 120.0
            }
            if size.width != size.width {
                size.width = 120.0
            }
            
            if let caption = fileMessage.getCaption(), caption.count > 0 {
                textSize = _captionLabel!.sizeThatFits(CGSize.init(width: size.width - x, height: CGFloat.greatestFiniteMagnitude))
                textSize.height = textSize.height + 12.0
            }
            
            let bubbleSize = CGSize.init(width: size.width + imageInsets.left + imageInsets.right, height: size.height + imageInsets.top + imageInsets.bottom + textSize.height)
            setBubble(bubbleSize)
            
            super.layoutSubviews()

            _thumbnailView!.frame = CGRect.init(x: msgBackground.frame.origin.x + imageInsets.left, y: msgBackground.frame.origin.y + imageInsets.top, width: size.width, height: size.height)
            _captionLabel!.frame  = CGRect.init(x:ceil(msgBackground.frame.origin.x + (x/2)), y: ceil(_thumbnailView!.frame.origin.y + _thumbnailView!.frame.size.height), width: ceil(textSize.width), height: ceil(textSize.height))

            let mask: CALayer = bubbleMaskWithoutArrow(forImageSize: CGSize.init(width: _thumbnailView!.frame.size.width, height: _thumbnailView!.frame.size.height))
            _thumbnailView?.layer.mask = mask
            _thumbnailView?.layer.masksToBounds = true
            
            if fileMessage.isOwn != nil, fileMessage.isOwn.boolValue {
                resendButton.frame = CGRect.init(x: _thumbnailView!.frame.origin.x - kMessageScreenMargin, y: _thumbnailView!.frame.origin.y + (_thumbnailView!.frame.size.height - 32) / 2, width: 114, height: 32)
            }
            
            progressBar.frame = CGRect.init(x: _thumbnailView!.frame.origin.x + 16.0, y: _thumbnailView!.frame.origin.y + _thumbnailView!.frame.size.height - 40.0, width: size.width - 32.0, height: 16.0)
            
            /* duration label */
            _durationBackground!.frame = CGRect.init(x: 0, y: _thumbnailView!.frame.size.height - 22.0, width: _thumbnailView!.frame.size.width + 1, height: 18.0)
            _durationLabel!.frame = CGRect.init(x: _durationBackground!.frame.size.width / 2, y: 0, width: _durationBackground!.frame.size.width / 2 - 12, height: 16.0)
            
            /* download size label */
            _downloadBackground!.frame = CGRect.init(x: 0, y: 1, width: _thumbnailView!.frame.size.width + 1, height: 18.0)
            _downloadSizeLabel!.frame = CGRect.init(x: _downloadBackground!.frame.size.width / 2, y: 1, width: _downloadBackground!.frame.size.width / 2 - 12, height: 16.0)
            
            if bubbleSize.height > 44.0 && bubbleSize.width > 44.0 {
                _playImageView!.frame = CGRect.init(x: (_thumbnailView!.frame.size.width / 2) - 22.0, y: (_thumbnailView!.frame.size.height / 2) - 22.0 - 2.0, width: 44.0, height: 44.0)
            } else {
                var min = Swift.min(bubbleSize.width, bubbleSize.height)
                min = min - 20.0
                _playImageView!.frame = CGRect.init(x: (bubbleSize.width / 2) - (min/2), y: (bubbleSize.height / 2) - (min/2) - 2.0, width: min, height: min)
            }
        } else {
            var textSize: CGSize = CGSize.init(width: 0.0, height: 0.0)
            let size = CGSize.init(width: 80.0, height: 40.0)
            let x: CGFloat = 30.0
            
            if let caption = fileMessage.getCaption(), caption.count > 0 {
                textSize = _captionLabel!.sizeThatFits(CGSize.init(width: size.width - x, height: CGFloat.greatestFiniteMagnitude))
                textSize.height = textSize.height + 12.0
            }
            setBubbleContentSize(CGSize.init(width: size.width, height: size.height + textSize.height))
        }
    }
     
    public override func observeValue(forKeyPath keyPath: String?, of object: Any?, change: [NSKeyValueChangeKey : Any]?, context: UnsafeMutableRawPointer?) {
        super.observeValue(forKeyPath: keyPath, of: object, change: change, context: context)
        
        if let objerveObject = object as? BaseMessage {
            if objerveObject == message && keyPath == "data" {
                updateDownloadSize()
            }
        }
        
    }
    
    override open func accessibilityLabelForContent() -> String! {
        let fileMessage = message as! FileMessage
        
        let preText = "\(BundleUtil.localizedString(forKey: "video")!), \(fileMessage.getDuration()?.intValue ?? 0) \(BundleUtil.localizedString(forKey: "seconds")!)"
        if _captionLabel?.text != nil {
            return "\(preText). \(_captionLabel!.text!))"
        }
        
        return preText
    }
        
    override open func messageTapped(_ sender: Any!) {
        let fileMessage = message as! FileMessage
        chatVc.fileVideoMessageTapped(fileMessage)
    }
    
    override open func canPerformAction(_ action: Selector, withSender sender: Any?) -> Bool {
        let fileMessage = message as! FileMessage
        if action == #selector(resendMessage(_:)) && fileMessage.isOwn.boolValue && fileMessage.sendFailed.boolValue {
            return true
        }
        else if action == #selector(deleteMessage(_:)) && fileMessage.isOwn.boolValue && fileMessage.progress != nil {
            /* don't allow messages in progress to be deleted */
            return false
        }
        else if action == #selector(copyMessage(_:)) && _captionLabel?.text != nil {
            return true
        }
        else if action == #selector(shareMessage(_:)) {
            if #available(iOS 13.0, *) {
                let mdmSetup = MDMSetup.init(setup: false)
                if mdmSetup?.disableShareMedia() == true {
                    return false;
                }
            }
            return fileMessage.data != nil
        }
        else if action == #selector(forwardMessage(_:)) {
            if #available(iOS 13.0, *) {
                return fileMessage.data != nil
            } else {
                return false;
            }
        }
        else if action == #selector(speakMessage(_:)) && _captionLabel?.text != nil {
            return true
        }
        else {
            return super.canPerformAction(action, withSender: sender)
        }
    }
    
    @objc override open func copyMessage(_ menuController: UIMenuController!) {
        let fileMessage = message as! FileMessage
        if let caption = fileMessage.getCaption(), caption.count > 0 {
            UIPasteboard.general.string = fileMessage.getCaption()
        }
    }
            
    open override func performPlayActionForAccessibility() -> Bool {
        messageTapped(self)
        return true
    }
        
    open override func previewViewController() -> UIViewController! {
        return chatVc.headerView.getPhotoBrowser(at: message, forPeeking: true)
    }
    
    open override func previewViewController(for previewingContext: UIViewControllerPreviewing!, viewControllerForLocation location: CGPoint) -> UIViewController! {
        if let controller = super.previewViewController(for: previewingContext, viewControllerForLocation: location) {
            return controller
        }
        return chatVc.headerView.getPhotoBrowser(at: message, forPeeking: true)
    }
    
    open override func setupColors() {
        super.setupColors()
        _captionLabel?.textColor = Colors.fontNormal()
    }
    
    @available(iOS 13.0, *)
    open override func getContextMenu(_ indexPath: IndexPath!, point: CGPoint) -> UIContextMenuConfiguration! {
        if self.isEditing == true {
            return nil
        }
        
        // returns nil if there is no link tapped
        if let menu = contextMenuForLink(indexPath, point: point) {
            return menu
        }

        
        let fileMessage = message as! FileMessage
        if fileMessage.data != nil {
            if fileMessage.data.data != nil {
                let conf = UIContextMenuConfiguration.init(identifier: indexPath as NSIndexPath, previewProvider: { () -> UIViewController? in
                    return self.previewViewController()
                }) { (suggestedActions) -> UIMenu? in
                    var menuItems = super.contextMenuItems()!
                    let saveImage = UIImage.init(systemName: "square.and.arrow.down.fill", compatibleWith: self.traitCollection)
                    let saveAction = UIAction.init(title: BundleUtil.localizedString(forKey: "save"), image: saveImage, identifier: nil, discoverabilityTitle: nil, attributes: [], state: .off) { (action) in
                        let fileName = String.init(format: "%f.%@", Date().timeIntervalSinceReferenceDate, MEDIA_EXTENSION_VIDEO)
                        let tmpurl = URL.init(fileURLWithPath: NSTemporaryDirectory()).appendingPathComponent(fileName)
                        do {
                            try fileMessage.data.data.write(to: tmpurl)
                            AlbumManager.shared.saveMovieToLibrary(movieURL: tmpurl) { (success) in
                                do {
                                    try FileManager.default.removeItem(atPath: tmpurl.path)
                                } catch {
                                    DDLogWarn("Remove moviefile to temporary file failed")
                                }
                            }
                        } catch {
                            DDLogWarn("Writing moviefile to temporary file failed")
                        }
                    }
                    
                    if self.message.isOwn.boolValue == true || self.chatVc.conversation.isGroup() == true {
                        menuItems.insert(saveAction, at: 0)
                    } else {
                        menuItems.insert(saveAction, at: 1)
                    }
                    return UIMenu.init(title: "", image: nil, identifier: nil, options: .displayInline, children: menuItems as! [UIMenuElement])
                }
                return conf
            } else {
                return super.getContextMenu(indexPath, point: point)
            }
        } else {
            return super.getContextMenu(indexPath, point: point)
        }
    }
}

extension ChatFileVideoMessageCell {
    // MARK: Public functions
        
    func setBaseMessage(newMessage: BaseMessage) {
        if message != nil {
            if let index = _observedMessages.firstIndex(of: message.id) {
                message.removeObserver(self, forKeyPath: "data")
                _observedMessages.remove(at: index)
            }
        }
        
        let fileMessage = newMessage as! FileMessage

        super.message = newMessage
        
        if !chatVc.isOpenWithForceTouch {
            _observedMessages.append(message.id)
            message.addObserver(self, forKeyPath: "data", options: [], context: nil)
        }

        if fileMessage.thumbnail != nil, let thumb = fileMessage.thumbnail.uiImage {
            _thumbnailView?.image = thumb
        }
        
        var autoresizingMask: AutoresizingMask = .flexibleRightMargin
        if fileMessage.isOwn.boolValue {
            autoresizingMask = .flexibleLeftMargin
        }
        _thumbnailView?.autoresizingMask = autoresizingMask
        _durationBackground?.autoresizingMask = autoresizingMask
        _durationLabel?.autoresizingMask = autoresizingMask
        _downloadBackground?.autoresizingMask = autoresizingMask
        _downloadSizeLabel?.autoresizingMask = autoresizingMask

        if let seconds = fileMessage.getDuration()?.intValue {
            _durationLabel?.text = DateFormatter.timeFormatted(seconds)
        } else {
            _durationLabel?.text = nil
        }
        
        _downloadSizeLabel!.text = Utils.formatDataLength(CGFloat(fileMessage.fileSize!.floatValue))
        
        updateDownloadSize()
        
        if let captionText = fileMessage.getCaption(), captionText.count > 0, fileMessage.shouldShowCaption() {
            let attributed = TextStyleUtils.makeAttributedString(from: captionText, with: _captionLabel!.font, textColor: Colors.fontNormal(), isOwn: true, application: UIApplication.shared)
            let formattedAttributeString = NSMutableAttributedString.init(attributedString: (_captionLabel!.applyMarkup(for: attributed))!)
            _captionLabel?.attributedText = TextStyleUtils.makeMentionsAttributedString(for: formattedAttributeString, textFont: _captionLabel!.font!, at: _captionLabel!.textColor.withAlphaComponent(0.4), messageInfo: Int32(message.isOwn!.intValue), application: UIApplication.shared)
            _captionLabel?.isHidden = false
        }
        else {
            _captionLabel?.text = nil
            _captionLabel?.isHidden = true
        }
        
        setupColors()

        self.setNeedsLayout()
    }
    
    
    @objc func resendMessage(_ menuController: UIMenuController) {
        let fileMessage = message as! FileMessage
        let sender: FileMessageSender = FileMessageSender.init()
        sender.retryMessage(fileMessage)
    }
    
    @objc func speakMessage(_ menuController: UIMenuController) {
        if _captionLabel?.text != nil {
            let speakText = "\(BundleUtil.localizedString(forKey: "image") ?? "Image"). \(_captionLabel!.text!)"
            let utterance: AVSpeechUtterance = AVSpeechUtterance.init(string: speakText)
            let syn = AVSpeechSynthesizer.init()
            syn.speak(utterance)
        }
    }
}

extension ChatFileVideoMessageCell {
    // MARK: Private functions
    
    private func updateDownloadSize() {
        let fileMessage = message as! FileMessage
        if fileMessage.data != nil {
            self._downloadBackground?.image = UIImage.init(named: "VideoDownloadBgDownloaded")?.resizableImage(withCapInsets: UIEdgeInsets.init(top: 0, left: 32, bottom: 0, right: 0))
        } else {
            self._downloadBackground?.image = UIImage.init(named: "VideoDownloadBg")?.resizableImage(withCapInsets: UIEdgeInsets.init(top: 0, left: 32, bottom: 0, right: 0))
        }
    }
}