rooms-ios/NextcloudTalk/InputbarViewController.swift

//
// SPDX-FileCopyrightText: 2023 Nextcloud GmbH and Nextcloud contributors
// SPDX-License-Identifier: GPL-3.0-or-later
//

import Foundation
import NextcloudKit
import PhotosUI
import UIKit

@objcMembers public class InputbarViewController: SLKTextViewController, NCChatTitleViewDelegate {

    // MARK: - Public var
    public var room: NCRoom

    // MARK: - Internal var
    internal var titleView: NCChatTitleView?
    internal var autocompletionUsers: [MentionSuggestion] = []
    internal var mentionsDict: [String: NCMessageParameter] = [:]
    internal var contentView: UIView?

    public init?(for room: NCRoom, tableViewStyle style: UITableView.Style) {
        self.room = room

        super.init(tableViewStyle: style)

        self.commonInit()
    }

    public init?(for room: NCRoom, withView view: UIView) {
        self.room = room
        self.contentView = view

        super.init(tableViewStyle: .plain)

        self.commonInit()

        view.translatesAutoresizingMaskIntoConstraints = false

        self.view.addSubview(view)

        NSLayoutConstraint.activate([
            view.leftAnchor.constraint(equalTo: self.view.safeAreaLayoutGuide.leftAnchor),
            view.rightAnchor.constraint(equalTo: self.view.safeAreaLayoutGuide.rightAnchor),
            view.topAnchor.constraint(equalTo: self.view.safeAreaLayoutGuide.topAnchor),
            view.bottomAnchor.constraint(equalTo: self.textInputbar.topAnchor)
        ])

        // Make sure our contentView does not hide the inputBar and the autocompletionView
        self.view.bringSubviewToFront(self.textInputbar)
        self.view.bringSubviewToFront(self.autoCompletionView)
    }

    private func commonInit() {
        self.registerClass(forTextView: NCMessageTextView.self)
        self.registerClass(forReplyView: ReplyMessageView.self)
        self.registerClass(forTypingIndicatorView: TypingIndicatorView.self)
    }

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

    deinit {
        print("Dealloc InputbarViewController")
    }

    // MARK: - View lifecycle

    public override func viewDidLoad() {
        super.viewDidLoad()

        self.setTitleView()

        self.bounces = false
        self.shakeToClearEnabled = false

        self.textInputbar.autoHideRightButton = false
        self.textInputbar.counterStyle = .limitExceeded
        self.textInputbar.counterPosition = .top
        self.textInputbar.textView.isDynamicTypeEnabled = false
        self.textInputbar.textView.font = .preferredFont(forTextStyle: .body)

        let talkCapabilities = NCDatabaseManager.sharedInstance().roomTalkCapabilities(for: room)

        if let talkCapabilities, talkCapabilities.chatMaxLength > 0 {
            self.textInputbar.maxCharCount = UInt(talkCapabilities.chatMaxLength)
        } else {
            self.textInputbar.maxCharCount = 1000
            self.textInputbar.counterStyle = .countdownReversed
        }

        self.textInputbar.isTranslucent = false
        self.textInputbar.semanticContentAttribute = .forceLeftToRight
        self.textInputbar.contentInset = .init(top: 8, left: 4, bottom: 8, right: 4)
        self.textView.textContainerInset = .init(top: 8, left: 8, bottom: 8, right: 8)

        self.textView.layoutSubviews()
        self.textView.layer.cornerRadius = self.textView.frame.size.height / 2

        // Need a compile-time check here for old xcode version on CI
#if swift(>=5.9)
        if #available(iOS 17.0, *), NCUtils.isiOSAppOnMac() {
            self.textView.inlinePredictionType = .no
        }
#endif

        self.textInputbar.editorTitle.textColor = .darkGray
        self.textInputbar.editorLeftButton.tintColor = .systemBlue
        self.textInputbar.editorRightButton.tintColor = .systemBlue

        self.textInputbar.editorLeftButton.setImage(.init(systemName: "xmark"), for: .normal)
        self.textInputbar.editorRightButton.setImage(.init(systemName: "checkmark"), for: .normal)

        self.textInputbar.editorLeftButton.setTitle("", for: .normal)
        self.textInputbar.editorRightButton.setTitle("", for: .normal)

        self.navigationController?.navigationBar.tintColor = NCAppBranding.themeTextColor()
        self.navigationController?.navigationBar.barTintColor = NCAppBranding.themeColor()
        self.navigationController?.navigationBar.isTranslucent = false
        self.tabBarController?.tabBar.tintColor = NCAppBranding.themeColor()

        let themeColor: UIColor = NCAppBranding.themeColor()
        let themeTextColor: UIColor = NCAppBranding.themeTextColor()

        let appearance = UINavigationBarAppearance()
        appearance.configureWithOpaqueBackground()
        appearance.titleTextAttributes = [.foregroundColor: themeTextColor]
        appearance.backgroundColor = themeColor
        self.navigationItem.standardAppearance = appearance
        self.navigationItem.compactAppearance = appearance
        self.navigationItem.scrollEdgeAppearance = appearance

        self.view.backgroundColor = .systemBackground
        self.textInputbar.backgroundColor = .systemBackground

        self.textInputbar.editorTitle.textColor = .label
        self.textView.layer.borderWidth = 1.0
        self.textView.layer.borderColor = UIColor.systemGray4.cgColor

        // Hide default top border of UIToolbar
        self.textInputbar.setShadowImage(UIImage(), forToolbarPosition: .any)
        self.textView.delegate = self

        self.autoCompletionView.register(AutoCompletionTableViewCell.self, forCellReuseIdentifier: AutoCompletionCellIdentifier)
        self.registerPrefixes(forAutoCompletion: ["@"])

        self.autoCompletionView.backgroundColor = .secondarySystemBackground
        self.autoCompletionView.sectionHeaderTopPadding = 0

        // Align separators to ChatMessageTableViewCell's title label
        self.autoCompletionView.separatorInset = .init(top: 0, left: 50, bottom: 0, right: 0)

        // We can't use UIColor with systemBlueColor directly, because it will switch to indigo. So make sure we actually get a blue tint here
        self.textView.tintColor = UIColor(cgColor: UIColor.systemBlue.cgColor)

        self.restorePendingMessage()

        self.rightButton.setTitle("", for: .normal)
        self.rightButton.setImage(UIImage(systemName: "paperplane"), for: .normal)
        self.rightButton.accessibilityLabel = NSLocalizedString("Send message", comment: "")
        self.rightButton.accessibilityHint = NSLocalizedString("Double tap to send message", comment: "")
    }

    public override func traitCollectionDidChange(_ previousTraitCollection: UITraitCollection?) {
        if self.traitCollection.hasDifferentColorAppearance(comparedTo: previousTraitCollection) {
            // We use a CGColor so we loose the automatic color changing of dynamic colors -> update manually
            self.textView.layer.borderColor = UIColor.systemGray4.cgColor
            self.textView.tintColor = UIColor(cgColor: UIColor.systemBlue.cgColor)
        }
    }

    public override func viewWillTransition(to size: CGSize, with coordinator: UIViewControllerTransitionCoordinator) {
        super.viewWillTransition(to: size, with: coordinator)

        coordinator.animate(alongsideTransition: nil) { _ in
            self.setTitleView()
        }
    }

    // MARK: - Configuration

    func setTitleView() {
        let titleView = NCChatTitleView()

        // Int.max is problematic when running on MacOS, so we use Int32.max here
        titleView.frame = .init(x: 0, y: 0, width: Int(Int32.max), height: 30)
        titleView.delegate = self
        titleView.titleTextView.accessibilityHint = NSLocalizedString("Double tap to go to conversation information", comment: "")

        if self.navigationController?.traitCollection.verticalSizeClass == .compact {
            titleView.showSubtitle = false
        }

        titleView.update(for: self.room)
        self.titleView = titleView
        self.navigationItem.titleView = titleView
    }

    // MARK: - Autocompletion

    public override func didChangeAutoCompletionPrefix(_ prefix: String, andWord word: String) {
        if prefix == "@" {
            self.showSuggestions(for: word)
        }
    }

    public override func heightForAutoCompletionView() -> CGFloat {
        return kAutoCompletionCellHeight * CGFloat(self.autocompletionUsers.count) + (self.autoCompletionView.tableHeaderView?.frame.height ?? 0)
    }

    func showSuggestions(for string: String) {
        self.autocompletionUsers = []

        let activeAccount = NCDatabaseManager.sharedInstance().activeAccount()
        NCAPIController.sharedInstance().getMentionSuggestions(for: activeAccount.accountId, in: self.room.token, with: string) { mentions in
            guard let mentions else { return }

            self.autocompletionUsers = mentions
            let showAutocomplete = !self.autocompletionUsers.isEmpty

            // Check if "@" is still there
            self.textView.look(forPrefixes: self.registeredPrefixes) { prefix, word, _ in
                if prefix?.count ?? 0 > 0 && word?.count ?? 0 > 0 {
                    self.showAutoCompletionView(showAutocomplete)
                } else {
                    self.cancelAutoCompletion()
                }
            }
        }
    }

    internal func replaceMentionsDisplayNamesWithMentionsKeysInMessage(message: String, parameters: String) -> String {
        var resultMessage = message.trimmingCharacters(in: .whitespacesAndNewlines)

        guard let messageParametersDict = NCMessageParameter.messageParametersDict(fromJSONString: parameters) else { return resultMessage }

        for (parameterKey, parameter) in messageParametersDict {
            let parameterKeyString = "{\(parameterKey)}"
            resultMessage = resultMessage.replacingOccurrences(of: parameter.mentionDisplayName, with: parameterKeyString)
        }

        return resultMessage
    }

    // MARK: - UITableViewDataSource methods

    public override func numberOfSections(in tableView: UITableView) -> Int {
        if tableView == self.autoCompletionView {
            return 1
        }

        return 0
    }

    public override func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        if tableView == self.autoCompletionView {
            return self.autocompletionUsers.count
        }

        return 0
    }

    public override func tableView(_ tableView: UITableView, titleForHeaderInSection section: Int) -> String? {
        return nil
    }

    public override func tableView(_ tableView: UITableView, heightForHeaderInSection section: Int) -> CGFloat {
        return 0
    }

    public override func tableView(_ tableView: UITableView, viewForHeaderInSection section: Int) -> UIView? {
        return nil
    }

    public override func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        guard tableView == self.autoCompletionView,
              indexPath.row < self.autocompletionUsers.count,
              let cell = self.autoCompletionView.dequeueReusableCell(withIdentifier: AutoCompletionCellIdentifier) as? AutoCompletionTableViewCell
        else {
            return AutoCompletionTableViewCell(style: .default, reuseIdentifier: AutoCompletionCellIdentifier)
        }

        let suggestion = self.autocompletionUsers[indexPath.row]

        if let details = suggestion.details {
            cell.titleLabel.numberOfLines = 2

            let attributedLabel = (suggestion.label + "\n").withFont(.preferredFont(forTextStyle: .body))
            let attributedDetails = details.withFont(.preferredFont(forTextStyle: .callout)).withTextColor(.secondaryLabel)
            attributedLabel.append(attributedDetails)
            cell.titleLabel.attributedText = attributedLabel
        } else {
            cell.titleLabel.numberOfLines = 1
            cell.titleLabel.text = suggestion.label
        }

        if let suggestionUserStatus = suggestion.userStatus {
            cell.setUserStatus(suggestionUserStatus)
        }

        if suggestion.id == "all" {
            cell.avatarButton.setAvatar(for: self.room)
        } else {
            cell.avatarButton.setActorAvatar(forId: suggestion.id, withType: suggestion.source, withDisplayName: suggestion.label, withRoomToken: self.room.token)
        }

        cell.accessibilityIdentifier = AutoCompletionCellIdentifier
        return cell
    }

    public override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
        guard tableView == self.autoCompletionView,
              indexPath.row < self.autocompletionUsers.count
        else { return }

        let suggestion = self.autocompletionUsers[indexPath.row]

        let mentionKey = "mention-\(self.mentionsDict.count)"
        self.mentionsDict[mentionKey] = suggestion.asMessageParameter()

        let mentionWithWhitespace = suggestion.label + " "
        self.acceptAutoCompletion(with: mentionWithWhitespace, keepPrefix: true)
    }

    public override func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
        return kAutoCompletionCellHeight
    }

    // MARK: - TextView functiosn

    public func setChatMessage(_ chatMessage: String) {
        DispatchQueue.main.async {
            self.textView.text = chatMessage
        }
    }

    public func restorePendingMessage() {
        if let pendingMessage = self.room.pendingMessage {
            self.setChatMessage(pendingMessage)
        }
    }

    public override func textView(_ textView: UITextView, shouldChangeTextIn range: NSRange, replacementText text: String) -> Bool {
        if text.isEmpty, let selectedRange = textView.selectedTextRange, let text = textView.text {
            let cursorOffset = textView.offset(from: textView.beginningOfDocument, to: selectedRange.start)
            let substring = (text as NSString).substring(to: cursorOffset)

            if var lastPossibleMention = substring.components(separatedBy: "@").last {
                lastPossibleMention.insert("@", at: lastPossibleMention.startIndex)

                for (mentionKey, mentionParameter) in self.mentionsDict {
                    if lastPossibleMention != mentionParameter.mentionDisplayName {
                        continue
                    }

                    // Delete mention
                    let range = NSRange(location: cursorOffset - lastPossibleMention.utf16.count, length: lastPossibleMention.utf16.count)
                    textView.text = (text as NSString).replacingCharacters(in: range, with: "")

                    // Only delete it from mentionsDict if there are no more mentions for that user/room
                    // User could have manually added the mention without selecting it from autocompletion
                    // so no mention was added to the mentionsDict
                    if (textView.text as NSString).range(of: lastPossibleMention).location != NSNotFound {
                        self.mentionsDict.removeValue(forKey: mentionKey)
                    }

                    return true
                }
            }
        }

        return super.textView(textView, shouldChangeTextIn: range, replacementText: text)
    }

    // MARK: - TitleView delegate

    public func chatTitleViewTapped(_ titleView: NCChatTitleView!) {
        // Doing nothing here -> override in subclass
    }

}