From af0efc325bdd6f18d08fc9081038df8de7095e6a Mon Sep 17 00:00:00 2001
From: nielsandriesse <andriesseniels@gmail.com>
Date: Wed, 17 Feb 2021 10:06:17 +1100
Subject: [PATCH] Glue things together

---
 .../ConversationVC+Interaction.swift          | 28 +++++++++++++++++--
 1 file changed, 25 insertions(+), 3 deletions(-)

diff --git a/Session/Conversations V2/ConversationVC+Interaction.swift b/Session/Conversations V2/ConversationVC+Interaction.swift
index b6231985c..b2b24c1b0 100644
--- a/Session/Conversations V2/ConversationVC+Interaction.swift	
+++ b/Session/Conversations V2/ConversationVC+Interaction.swift	
@@ -1,7 +1,8 @@
 import CoreServices
 import Photos
 
-extension ConversationVC : InputViewDelegate, MessageCellDelegate, ContextMenuActionDelegate, ScrollToBottomButtonDelegate, SendMediaNavDelegate, UIDocumentPickerDelegate {
+extension ConversationVC : InputViewDelegate, MessageCellDelegate, ContextMenuActionDelegate, ScrollToBottomButtonDelegate,
+    SendMediaNavDelegate, UIDocumentPickerDelegate, AttachmentApprovalViewControllerDelegate {
 
     @objc func openSettings() {
         let settingsVC = OWSConversationSettingsViewController()
@@ -43,6 +44,8 @@ extension ConversationVC : InputViewDelegate, MessageCellDelegate, ContextMenuAc
     func sendMediaNav(_ sendMediaNavigationController: SendMediaNavigationController, didApproveAttachments attachments: [SignalAttachment], messageText: String?) {
         sendAttachments(attachments, with: messageText ?? "")
         scrollToBottom(isAnimated: false)
+        // TODO: Reset mentions
+        self.snInputView.text = ""
         dismiss(animated: true) { }
     }
 
@@ -54,6 +57,22 @@ extension ConversationVC : InputViewDelegate, MessageCellDelegate, ContextMenuAc
         snInputView.text = newMessageText ?? ""
     }
 
+    func attachmentApproval(_ attachmentApproval: AttachmentApprovalViewController, didApproveAttachments attachments: [SignalAttachment], messageText: String?) {
+        sendAttachments(attachments, with: messageText ?? "")
+        scrollToBottom(isAnimated: false)
+        // TODO: Reset mentions
+        self.snInputView.text = ""
+        dismiss(animated: true) { }
+    }
+
+    func attachmentApprovalDidCancel(_ attachmentApproval: AttachmentApprovalViewController) {
+        dismiss(animated: true, completion: nil)
+    }
+
+    func attachmentApproval(_ attachmentApproval: AttachmentApprovalViewController, didChangeMessageText newMessageText: String?) {
+        snInputView.text = newMessageText ?? ""
+    }
+
     func handleCameraButtonTapped() {
         guard requestCameraPermissionIfNeeded() else { return }
         requestMicrophonePermissionIfNeeded { }
@@ -82,14 +101,17 @@ extension ConversationVC : InputViewDelegate, MessageCellDelegate, ContextMenuAc
         // It uses more memory than "open" but lets us avoid working with security scoped URLs.
         let documentPickerVC = UIDocumentPickerViewController(documentTypes: [ kUTTypeItem as String ], in: UIDocumentPickerMode.import)
         documentPickerVC.delegate = self
+        documentPickerVC.modalPresentationStyle = .fullScreen
+        AppearanceUtilities.switchToImagePickerAppearance()
         present(documentPickerVC, animated: true, completion: nil)
     }
 
     func documentPickerWasCancelled(_ controller: UIDocumentPickerViewController) {
-        // Do nothing
+        AppearanceUtilities.switchToSessionAppearance()
     }
 
     func documentPicker(_ controller: UIDocumentPickerViewController, didPickDocumentsAt urls: [URL]) {
+        AppearanceUtilities.switchToSessionAppearance()
         guard let url = urls.first else { return } // TODO: Handle multiple?
         let urlResourceValues: URLResourceValues
         do {
@@ -141,7 +163,7 @@ extension ConversationVC : InputViewDelegate, MessageCellDelegate, ContextMenuAc
                 guard !modalActivityIndicator.wasCancelled, let attachment = attachment as? SignalAttachment else { return }
                 modalActivityIndicator.dismiss {
                     if !attachment.hasError {
-                        self?.showApprovalDialog(for: [ attachment ])
+                        self?.showAttachmentApprovalDialog(for: [ attachment ])
                     } else {
                         self?.showErrorAlert(for: attachment)
                     }