[WIP] Edit message preview

2025-12-23 22:55:00 +00:00 · 2024-05-26 18:59:44 +04:00
parent 6ff4cd3681
commit 667aa0349a
16 changed files with 487 additions and 1603 deletions
--- a/submodules/ChatSendMessageActionUI/Sources/ChatSendMessageActionSheetController.swift
+++ b/submodules/ChatSendMessageActionUI/Sources/ChatSendMessageActionSheetController.swift
@@ -11,169 +11,57 @@ import TextFormat
 import ReactionSelectionNode
 import WallpaperBackgroundNode

-private final class ChatSendMessageActionSheetControllerImpl: ViewController, ChatSendMessageActionSheetController {
-    private var controllerNode: ChatSendMessageActionSheetControllerNode {
-        return self.displayNode as! ChatSendMessageActionSheetControllerNode
-    }
-    
-    private let context: AccountContext
-    
-    private let peerId: EnginePeer.Id?
-    private let isScheduledMessages: Bool
-    private let forwardMessageIds: [EngineMessage.Id]?
-    private let hasEntityKeyboard: Bool
-    
-    private let gesture: ContextGesture
-    private let sourceSendButton: ASDisplayNode
-    private let textInputView: UITextView
-    private let attachment: Bool
-    private let canSendWhenOnline: Bool
-    private let completion: () -> Void
-    private let sendMessage: (SendMode, SendParameters?) -> Void
-    private let schedule: (SendParameters?) -> Void
-    private let reactionItems: [ReactionItem]?
-    
-    private var presentationData: PresentationData
-    private var presentationDataDisposable: Disposable?
-    
-    private var didPlayPresentationAnimation = false
-    
-    private var validLayout: ContainerViewLayout?
-    
-    private let hapticFeedback = HapticFeedback()
-    
-    private let emojiViewProvider: ((ChatTextInputTextCustomEmojiAttribute) -> UIView)?
-
-    public init(context: AccountContext, updatedPresentationData: (initial: PresentationData, signal: Signal<PresentationData, NoError>)? = nil, peerId: EnginePeer.Id?, isScheduledMessages: Bool = false, forwardMessageIds: [EngineMessage.Id]?, hasEntityKeyboard: Bool, gesture: ContextGesture, sourceSendButton: ASDisplayNode, textInputView: UITextView, emojiViewProvider: ((ChatTextInputTextCustomEmojiAttribute) -> UIView)?, attachment: Bool = false, canSendWhenOnline: Bool, completion: @escaping () -> Void, sendMessage: @escaping (SendMode, SendParameters?) -> Void, schedule: @escaping (SendParameters?) -> Void, reactionItems: [ReactionItem]? = nil) {
-        self.context = context
-        self.peerId = peerId
-        self.isScheduledMessages = isScheduledMessages
-        self.forwardMessageIds = forwardMessageIds
-        self.hasEntityKeyboard = hasEntityKeyboard
-        self.gesture = gesture
-        self.sourceSendButton = sourceSendButton
-        self.textInputView = textInputView
-        self.emojiViewProvider = emojiViewProvider
-        self.attachment = attachment
-        self.canSendWhenOnline = canSendWhenOnline
-        self.completion = completion
-        self.sendMessage = sendMessage
-        self.schedule = schedule
-        self.reactionItems = reactionItems
+public enum SendMessageActionSheetControllerParams {
+    public final class SendMessage {
+        public let isScheduledMessages: Bool
+        public let mediaPreview: ChatSendMessageContextScreenMediaPreview?
+        public let mediaCaptionIsAbove: (Bool, (Bool) -> Void)?
+        public let attachment: Bool
+        public let canSendWhenOnline: Bool
+        public let forwardMessageIds: [EngineMessage.Id]
        
-        self.presentationData = updatedPresentationData?.initial ?? context.sharedContext.currentPresentationData.with { $0 }
-        
-        super.init(navigationBarPresentationData: nil)
-        
-        self.blocksBackgroundWhenInOverlay = true
-        
-        self.presentationDataDisposable = ((updatedPresentationData?.signal ?? context.sharedContext.presentationData)
-        |> deliverOnMainQueue).startStrict(next: { [weak self] presentationData in
-            if let strongSelf = self {
-                strongSelf.presentationData = presentationData
-                if strongSelf.isNodeLoaded {
-                    strongSelf.controllerNode.updatePresentationData(presentationData)
-                }
-            }
-        }).strict()
-        
-        self.statusBar.statusBarStyle = .Hide
-        self.statusBar.ignoreInCall = true
-    }
-    
-    required init(coder aDecoder: NSCoder) {
-        fatalError("init(coder:) has not been implemented")
-    }
-    
-    deinit {
-        self.presentationDataDisposable?.dispose()
-    }
-    
-    override public func loadDisplayNode() {
-        var forwardedCount: Int?
-        if let forwardMessageIds = self.forwardMessageIds, forwardMessageIds.count > 0 {
-            forwardedCount = forwardMessageIds.count
-        }
-        
-        var reminders = false
-        var isSecret = false
-        var canSchedule = false
-        if let peerId = self.peerId {
-            reminders = peerId == context.account.peerId
-            isSecret = peerId.namespace == Namespaces.Peer.SecretChat
-            canSchedule = !isSecret
-        }
-        if self.isScheduledMessages {
-            canSchedule = false
-        }
-        
-        self.displayNode = ChatSendMessageActionSheetControllerNode(context: self.context, presentationData: self.presentationData, reminders: reminders, gesture: gesture, sourceSendButton: self.sourceSendButton, textInputView: self.textInputView, attachment: self.attachment, canSendWhenOnline: self.canSendWhenOnline, forwardedCount: forwardedCount, hasEntityKeyboard: self.hasEntityKeyboard, emojiViewProvider: self.emojiViewProvider, send: { [weak self] in
-            self?.sendMessage(.generic, nil)
-            self?.dismiss(cancel: false)
-        }, sendSilently: { [weak self] in
-            self?.sendMessage(.silently, nil)
-            self?.dismiss(cancel: false)
-        }, sendWhenOnline: { [weak self] in
-            self?.sendMessage(.whenOnline, nil)
-            self?.dismiss(cancel: false)
-        }, schedule: !canSchedule ? nil : { [weak self] in
-            self?.schedule(nil)
-            self?.dismiss(cancel: false)
-        }, cancel: { [weak self] in
-            self?.dismiss(cancel: true)
-        }, reactionItems: self.reactionItems)
-        self.displayNodeDidLoad()
-    }
-    
-    override public func viewDidAppear(_ animated: Bool) {
-        super.viewDidAppear(animated)
-        
-        if !self.didPlayPresentationAnimation {
-            self.didPlayPresentationAnimation = true
-            
-            self.hapticFeedback.impact()
-            self.controllerNode.animateIn()
+        public init(
+            isScheduledMessages: Bool,
+            mediaPreview: ChatSendMessageContextScreenMediaPreview?,
+            mediaCaptionIsAbove: (Bool, (Bool) -> Void)?,
+            attachment: Bool,
+            canSendWhenOnline: Bool,
+            forwardMessageIds: [EngineMessage.Id]
+        ) {
+            self.isScheduledMessages = isScheduledMessages
+            self.mediaPreview = mediaPreview
+            self.mediaCaptionIsAbove = mediaCaptionIsAbove
+            self.attachment = attachment
+            self.canSendWhenOnline = canSendWhenOnline
+            self.forwardMessageIds = forwardMessageIds
        }
    }
    
-    override public func containerLayoutUpdated(_ layout: ContainerViewLayout, transition: ContainedViewLayoutTransition) {
-        self.validLayout = layout
+    public final class EditMessage {
+        public let messages: [EngineMessage]
+        public let mediaPreview: ChatSendMessageContextScreenMediaPreview?
        
-        super.containerLayoutUpdated(layout, transition: transition)
-        
-        self.controllerNode.containerLayoutUpdated(layout, transition: transition)
+        public init(messages: [EngineMessage], mediaPreview: ChatSendMessageContextScreenMediaPreview?) {
+            self.messages = messages
+            self.mediaPreview = mediaPreview
+        }
    }
    
-    override public func dismiss(completion: (() -> Void)? = nil) {
-        self.dismiss(cancel: true)
-    }
-    
-    private func dismiss(cancel: Bool) {
-        self.statusBar.statusBarStyle = .Ignore
-        self.controllerNode.animateOut(cancel: cancel, completion: { [weak self] in
-            self?.completion()
-            self?.didPlayPresentationAnimation = false
-            self?.presentingViewController?.dismiss(animated: false, completion: nil)
-        })
-    }
+    case sendMessage(SendMessage)
+    case editMessage(EditMessage)
 }

 public func makeChatSendMessageActionSheetController(
    context: AccountContext,
    updatedPresentationData: (initial: PresentationData, signal: Signal<PresentationData, NoError>)? = nil,
    peerId: EnginePeer.Id?,
-    isScheduledMessages: Bool = false,
-    forwardMessageIds: [EngineMessage.Id]?,
+    params: SendMessageActionSheetControllerParams,
    hasEntityKeyboard: Bool,
    gesture: ContextGesture,
    sourceSendButton: ASDisplayNode,
    textInputView: UITextView,
-    mediaPreview: ChatSendMessageContextScreenMediaPreview? = nil,
-    mediaCaptionIsAbove: (Bool, (Bool) -> Void)? = nil,
    emojiViewProvider: ((ChatTextInputTextCustomEmojiAttribute) -> UIView)?,
    wallpaperBackgroundNode: WallpaperBackgroundNode? = nil,
-    attachment: Bool = false,
-    canSendWhenOnline: Bool,
    completion: @escaping () -> Void,
    sendMessage: @escaping (ChatSendMessageActionSheetController.SendMode, ChatSendMessageActionSheetController.SendParameters?) -> Void,
    schedule: @escaping (ChatSendMessageActionSheetController.SendParameters?) -> Void,
@@ -182,43 +70,17 @@ public func makeChatSendMessageActionSheetController(
    availableMessageEffects: AvailableMessageEffects? = nil,
    isPremium: Bool = false
 ) -> ChatSendMessageActionSheetController {
-    if textInputView.text.isEmpty && !"".isEmpty {
-        return ChatSendMessageActionSheetControllerImpl(
-            context: context,
-            updatedPresentationData: updatedPresentationData,
-            peerId: peerId,
-            isScheduledMessages: isScheduledMessages,
-            forwardMessageIds: forwardMessageIds,
-            hasEntityKeyboard: hasEntityKeyboard,
-            gesture: gesture,
-            sourceSendButton: sourceSendButton,
-            textInputView: textInputView,
-            emojiViewProvider: emojiViewProvider,
-            attachment: attachment,
-            canSendWhenOnline: canSendWhenOnline,
-            completion: completion,
-            sendMessage: sendMessage,
-            schedule: schedule,
-            reactionItems: nil
-        )
-    }
-    
    return ChatSendMessageContextScreen(
        context: context,
        updatedPresentationData: updatedPresentationData,
        peerId: peerId,
-        isScheduledMessages: isScheduledMessages,
-        forwardMessageIds: forwardMessageIds,
+        params: params,
        hasEntityKeyboard: hasEntityKeyboard,
        gesture: gesture,
        sourceSendButton: sourceSendButton,
        textInputView: textInputView,
-        mediaPreview: mediaPreview,
-        mediaCaptionIsAbove: mediaCaptionIsAbove,
        emojiViewProvider: emojiViewProvider,
        wallpaperBackgroundNode: wallpaperBackgroundNode,
-        attachment: attachment,
-        canSendWhenOnline: canSendWhenOnline,
        completion: completion,
        sendMessage: sendMessage,
        schedule: schedule,