mirror of
https://github.com/Swiftgram/Telegram-iOS.git
synced 2025-06-16 05:55:20 +00:00
503 lines
21 KiB
Swift
503 lines
21 KiB
Swift
import Foundation
|
|
import Postbox
|
|
import TelegramCore
|
|
import Display
|
|
import UIKit
|
|
import SwiftSignalKit
|
|
import MobileCoreServices
|
|
|
|
private struct MessageContextMenuData {
|
|
let starStatus: Bool?
|
|
let canReply: Bool
|
|
let canPin: Bool
|
|
let canEdit: Bool
|
|
let resourceStatus: MediaResourceStatus?
|
|
let messageActions: ChatAvailableMessageActions
|
|
}
|
|
|
|
private let starIconEmpty = UIImage(bundleImageName: "Chat/Context Menu/StarIconEmpty")?.precomposed()
|
|
private let starIconFilled = UIImage(bundleImageName: "Chat/Context Menu/StarIconFilled")?.precomposed()
|
|
|
|
func canReplyInChat(_ chatPresentationInterfaceState: ChatPresentationInterfaceState) -> Bool {
|
|
guard let peer = chatPresentationInterfaceState.renderedPeer?.peer else {
|
|
return false
|
|
}
|
|
|
|
var canReply = false
|
|
switch chatPresentationInterfaceState.chatLocation {
|
|
case .peer:
|
|
if let channel = peer as? TelegramChannel {
|
|
if case .member = channel.participationStatus {
|
|
switch channel.info {
|
|
case .broadcast:
|
|
canReply = channel.hasAdminRights([.canPostMessages])
|
|
case .group:
|
|
canReply = true
|
|
}
|
|
}
|
|
} else if let group = peer as? TelegramGroup {
|
|
if case .Member = group.membership {
|
|
canReply = true
|
|
}
|
|
} else {
|
|
canReply = true
|
|
}
|
|
case .group:
|
|
break
|
|
}
|
|
return canReply
|
|
}
|
|
|
|
enum ChatMessageContextMenuActionColor {
|
|
case accent
|
|
case destructive
|
|
}
|
|
|
|
struct ChatMessageContextMenuSheetAction {
|
|
let color: ChatMessageContextMenuActionColor
|
|
let title: String
|
|
let action: () -> Void
|
|
}
|
|
|
|
enum ChatMessageContextMenuAction {
|
|
case context(ContextMenuAction)
|
|
case sheet(ChatMessageContextMenuSheetAction)
|
|
}
|
|
|
|
func canEditMessageMedia(message: Message) -> Bool {
|
|
if message.id.peerId.namespace == Namespaces.Peer.SecretChat {
|
|
return false
|
|
}
|
|
if message.groupingKey != nil {
|
|
return false
|
|
}
|
|
for attribute in message.attributes {
|
|
if attribute is AutoremoveTimeoutMessageAttribute {
|
|
return false
|
|
}
|
|
}
|
|
for media in message.media {
|
|
if let _ = media as? TelegramMediaImage {
|
|
return true
|
|
} else if let file = media as? TelegramMediaFile {
|
|
for attribute in file.attributes {
|
|
switch attribute {
|
|
case .Sticker:
|
|
return false
|
|
case .Animated:
|
|
return true
|
|
case let .Video(video):
|
|
if video.flags.contains(.instantRoundVideo) {
|
|
return false
|
|
} else {
|
|
return true
|
|
}
|
|
case let .Audio(audio):
|
|
if audio.isVoice {
|
|
return false
|
|
} else {
|
|
return true
|
|
}
|
|
default:
|
|
break
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func updatedChatEditInterfaceMessagetState(state: ChatPresentationInterfaceState, message: Message) -> ChatPresentationInterfaceState {
|
|
var updated = state
|
|
for media in message.media {
|
|
if let webpage = media as? TelegramMediaWebpage, case let .Loaded(content) = webpage.content {
|
|
updated = updated.updatedEditingUrlPreview((content.url, webpage))
|
|
}
|
|
}
|
|
var isPlaintext = true
|
|
for media in message.media {
|
|
if !(media is TelegramMediaWebpage) {
|
|
isPlaintext = false
|
|
break
|
|
}
|
|
}
|
|
let content: ChatEditInterfaceMessageStateContent
|
|
if isPlaintext {
|
|
content = .plaintext
|
|
} else {
|
|
content = .media(editable: canEditMessageMedia(message: message))
|
|
}
|
|
updated = updated.updatedEditMessageState(ChatEditInterfaceMessageState(content: content, media: nil))
|
|
return updated
|
|
}
|
|
|
|
func contextMenuForChatPresentationIntefaceState(chatPresentationInterfaceState: ChatPresentationInterfaceState, account: Account, messages: [Message], interfaceInteraction: ChatPanelInterfaceInteraction?, debugStreamSingleVideo: @escaping (MessageId) -> Void) -> Signal<[ChatMessageContextMenuAction], NoError> {
|
|
guard let interfaceInteraction = interfaceInteraction else {
|
|
return .single([])
|
|
}
|
|
|
|
let dataSignal: Signal<MessageContextMenuData, NoError>
|
|
|
|
var loadStickerSaveStatus: MediaId?
|
|
var loadCopyMediaResource: MediaResource?
|
|
var isAction = false
|
|
if messages.count == 1 {
|
|
for media in messages[0].media {
|
|
if let file = media as? TelegramMediaFile {
|
|
for attribute in file.attributes {
|
|
if case let .Sticker(_, packInfo, _) = attribute, packInfo != nil {
|
|
loadStickerSaveStatus = file.fileId
|
|
}
|
|
}
|
|
} else if let _ = media as? TelegramMediaAction {
|
|
isAction = true
|
|
} else if let image = media as? TelegramMediaImage {
|
|
if !messages[0].containsSecretMedia {
|
|
loadCopyMediaResource = largestImageRepresentation(image.representations)?.resource
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
var canReply = false
|
|
var canPin = false
|
|
if messages[0].flags.intersection([.Failed, .Unsent]).isEmpty {
|
|
switch chatPresentationInterfaceState.chatLocation {
|
|
case .peer:
|
|
if let channel = messages[0].peers[messages[0].id.peerId] as? TelegramChannel {
|
|
switch channel.info {
|
|
case .broadcast:
|
|
canReply = channel.hasAdminRights([.canPostMessages])
|
|
if !isAction {
|
|
canPin = channel.hasAdminRights([.canEditMessages])
|
|
}
|
|
case .group:
|
|
canReply = true
|
|
if !isAction {
|
|
canPin = channel.hasAdminRights([.canPinMessages])
|
|
}
|
|
}
|
|
} else {
|
|
canReply = true
|
|
}
|
|
case .group:
|
|
break
|
|
}
|
|
}
|
|
|
|
var canEdit = false
|
|
if !isAction {
|
|
let message = messages[0]
|
|
|
|
var hasEditRights = false
|
|
if message.id.peerId.namespace == Namespaces.Peer.SecretChat {
|
|
hasEditRights = false
|
|
} else if let author = message.author, author.id == account.peerId {
|
|
hasEditRights = true
|
|
} else if message.author?.id == message.id.peerId, let peer = message.peers[message.id.peerId] {
|
|
if let peer = peer as? TelegramChannel, case .broadcast = peer.info {
|
|
if peer.hasAdminRights(.canEditMessages) {
|
|
hasEditRights = true
|
|
}
|
|
}
|
|
}
|
|
|
|
if hasEditRights {
|
|
var hasUneditableAttributes = false
|
|
for attribute in message.attributes {
|
|
if let _ = attribute as? InlineBotMessageAttribute {
|
|
hasUneditableAttributes = true
|
|
break
|
|
}
|
|
}
|
|
if message.forwardInfo != nil {
|
|
hasUneditableAttributes = true
|
|
}
|
|
|
|
for media in message.media {
|
|
if let file = media as? TelegramMediaFile {
|
|
if file.isSticker || file.isInstantVideo {
|
|
hasUneditableAttributes = true
|
|
break
|
|
}
|
|
} else if let _ = media as? TelegramMediaContact {
|
|
hasUneditableAttributes = true
|
|
break
|
|
} else if let _ = media as? TelegramMediaExpiredContent {
|
|
hasUneditableAttributes = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !hasUneditableAttributes {
|
|
let timestamp = Int32(CFAbsoluteTimeGetCurrent() + NSTimeIntervalSince1970)
|
|
if message.timestamp >= timestamp - 60 * 60 * 24 * 2 || message.id.peerId == account.peerId {
|
|
canEdit = true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
var loadStickerSaveStatusSignal: Signal<Bool?, NoError> = .single(nil)
|
|
if loadStickerSaveStatus != nil {
|
|
loadStickerSaveStatusSignal = account.postbox.transaction { transaction -> Bool? in
|
|
var starStatus: Bool?
|
|
if let loadStickerSaveStatus = loadStickerSaveStatus {
|
|
if getIsStickerSaved(transaction: transaction, fileId: loadStickerSaveStatus) {
|
|
starStatus = true
|
|
} else {
|
|
starStatus = false
|
|
}
|
|
}
|
|
|
|
return starStatus
|
|
}
|
|
}
|
|
|
|
var loadResourceStatusSignal: Signal<MediaResourceStatus?, NoError> = .single(nil)
|
|
if let loadCopyMediaResource = loadCopyMediaResource {
|
|
loadResourceStatusSignal = account.postbox.mediaBox.resourceStatus(loadCopyMediaResource)
|
|
|> take(1)
|
|
|> map(Optional.init)
|
|
}
|
|
|
|
dataSignal = combineLatest(loadStickerSaveStatusSignal, loadResourceStatusSignal, chatAvailableMessageActions(postbox: account.postbox, accountPeerId: account.peerId, messageIds: Set(messages.map { $0.id })))
|
|
|> map { stickerSaveStatus, resourceStatus, messageActions -> MessageContextMenuData in
|
|
return MessageContextMenuData(starStatus: stickerSaveStatus, canReply: canReply, canPin: canPin, canEdit: canEdit, resourceStatus: resourceStatus, messageActions: messageActions)
|
|
}
|
|
|
|
return dataSignal |> deliverOnMainQueue |> map { data -> [ChatMessageContextMenuAction] in
|
|
var actions: [ChatMessageContextMenuAction] = []
|
|
|
|
if let starStatus = data.starStatus, let image = starStatus ? starIconFilled : starIconEmpty {
|
|
actions.append(.context(ContextMenuAction(content: .icon(image), action: {
|
|
interfaceInteraction.toggleMessageStickerStarred(messages[0].id)
|
|
})))
|
|
}
|
|
|
|
if data.canReply {
|
|
actions.append(.context(ContextMenuAction(content: .text(chatPresentationInterfaceState.strings.Conversation_ContextMenuReply), action: {
|
|
interfaceInteraction.setupReplyMessage(messages[0].id)
|
|
})))
|
|
}
|
|
|
|
if data.canEdit {
|
|
actions.append(.sheet(ChatMessageContextMenuSheetAction(color: .accent, title: chatPresentationInterfaceState.strings.Conversation_Edit, action: {
|
|
interfaceInteraction.setupEditMessage(messages[0].id)
|
|
})))
|
|
}
|
|
|
|
let resourceAvailable: Bool
|
|
if let resourceStatus = data.resourceStatus, case .Local = resourceStatus {
|
|
resourceAvailable = true
|
|
} else {
|
|
resourceAvailable = false
|
|
}
|
|
|
|
if !messages[0].text.isEmpty || resourceAvailable {
|
|
let message = messages[0]
|
|
actions.append(.context(ContextMenuAction(content: .text(chatPresentationInterfaceState.strings.Conversation_ContextMenuCopy), action: {
|
|
if resourceAvailable {
|
|
for media in message.media {
|
|
if let image = media as? TelegramMediaImage, let largest = largestImageRepresentation(image.representations) {
|
|
let _ = (account.postbox.mediaBox.resourceData(largest.resource, option: .incremental(waitUntilFetchStatus: false))
|
|
|> take(1)
|
|
|> deliverOnMainQueue).start(next: { data in
|
|
if data.complete, let imageData = try? Data(contentsOf: URL(fileURLWithPath: data.path)) {
|
|
if let image = UIImage(data: imageData) {
|
|
if !message.text.isEmpty {
|
|
UIPasteboard.general.string = message.text
|
|
/*UIPasteboard.general.items = [
|
|
[kUTTypeUTF8PlainText as String: message.text],
|
|
[kUTTypePNG as String: image]
|
|
]*/
|
|
} else {
|
|
UIPasteboard.general.image = image
|
|
}
|
|
} else {
|
|
UIPasteboard.general.string = message.text
|
|
}
|
|
} else {
|
|
UIPasteboard.general.string = message.text
|
|
}
|
|
})
|
|
}
|
|
}
|
|
} else {
|
|
UIPasteboard.general.string = message.text
|
|
}
|
|
})))
|
|
}
|
|
|
|
if data.canPin {
|
|
if chatPresentationInterfaceState.pinnedMessage?.id != messages[0].id {
|
|
actions.append(.sheet(ChatMessageContextMenuSheetAction(color: .accent, title: chatPresentationInterfaceState.strings.Conversation_Pin, action: {
|
|
interfaceInteraction.pinMessage(messages[0].id)
|
|
})))
|
|
} else {
|
|
actions.append(.sheet(ChatMessageContextMenuSheetAction(color: .accent, title: chatPresentationInterfaceState.strings.Conversation_Unpin, action: {
|
|
interfaceInteraction.unpinMessage()
|
|
})))
|
|
}
|
|
}
|
|
|
|
if messages.count == 1 {
|
|
let message = messages[0]
|
|
|
|
var hasAutoremove = false
|
|
for attribute in message.attributes {
|
|
if let _ = attribute as? AutoremoveTimeoutMessageAttribute {
|
|
hasAutoremove = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !hasAutoremove {
|
|
for media in message.media {
|
|
if let file = media as? TelegramMediaFile {
|
|
if file.isVideo {
|
|
if file.isAnimated {
|
|
actions.append(.sheet(ChatMessageContextMenuSheetAction(color: .accent, title: chatPresentationInterfaceState.strings.Conversation_LinkDialogSave, action: {
|
|
let _ = addSavedGif(postbox: account.postbox, file: file).start()
|
|
})))
|
|
} else if !GlobalExperimentalSettings.isAppStoreBuild {
|
|
actions.append(.sheet(ChatMessageContextMenuSheetAction(color: .accent, title: "Stream", action: {
|
|
debugStreamSingleVideo(message.id)
|
|
})))
|
|
}
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
actions.append(.context(ContextMenuAction(content: .text(chatPresentationInterfaceState.strings.Conversation_ContextMenuMore), action: {
|
|
interfaceInteraction.beginMessageSelection(messages.map { $0.id })
|
|
})))
|
|
|
|
if data.messageActions.options.contains(.forward) {
|
|
actions.append(.sheet(ChatMessageContextMenuSheetAction(color: .accent, title: chatPresentationInterfaceState.strings.Conversation_ContextMenuForward, action: {
|
|
interfaceInteraction.forwardMessages(messages)
|
|
})))
|
|
}
|
|
|
|
if !data.messageActions.options.intersection([.deleteLocally, .deleteGlobally]).isEmpty {
|
|
actions.append(.sheet(ChatMessageContextMenuSheetAction(color: .destructive, title: chatPresentationInterfaceState.strings.Conversation_ContextMenuDelete, action: {
|
|
interfaceInteraction.deleteMessages(messages)
|
|
})))
|
|
}
|
|
|
|
return actions
|
|
}
|
|
}
|
|
|
|
struct ChatAvailableMessageActionOptions: OptionSet {
|
|
var rawValue: Int32
|
|
|
|
init(rawValue: Int32) {
|
|
self.rawValue = rawValue
|
|
}
|
|
|
|
init() {
|
|
self.rawValue = 0
|
|
}
|
|
|
|
static let deleteLocally = ChatAvailableMessageActionOptions(rawValue: 1 << 0)
|
|
static let deleteGlobally = ChatAvailableMessageActionOptions(rawValue: 1 << 1)
|
|
static let forward = ChatAvailableMessageActionOptions(rawValue: 1 << 2)
|
|
static let report = ChatAvailableMessageActionOptions(rawValue: 1 << 3)
|
|
}
|
|
|
|
struct ChatAvailableMessageActions {
|
|
let options: ChatAvailableMessageActionOptions
|
|
let banAuthor: Peer?
|
|
}
|
|
|
|
func chatAvailableMessageActions(postbox: Postbox, accountPeerId: PeerId, messageIds: Set<MessageId>) -> Signal<ChatAvailableMessageActions, NoError> {
|
|
return postbox.transaction { transaction -> ChatAvailableMessageActions in
|
|
var optionsMap: [MessageId: ChatAvailableMessageActionOptions] = [:]
|
|
var banPeer: Peer?
|
|
var hadBanPeerId = false
|
|
for id in messageIds {
|
|
if optionsMap[id] == nil {
|
|
optionsMap[id] = []
|
|
}
|
|
if id.peerId == accountPeerId {
|
|
optionsMap[id]!.insert(.deleteLocally)
|
|
} else if let peer = transaction.getPeer(id.peerId), let message = transaction.getMessage(id) {
|
|
if let channel = peer as? TelegramChannel {
|
|
if message.flags.contains(.Incoming) {
|
|
optionsMap[id]!.insert(.report)
|
|
}
|
|
if channel.hasAdminRights(.canBanUsers), case .group = channel.info {
|
|
if message.flags.contains(.Incoming) {
|
|
if !hadBanPeerId {
|
|
hadBanPeerId = true
|
|
banPeer = message.author
|
|
} else if banPeer?.id != message.author?.id {
|
|
banPeer = nil
|
|
}
|
|
} else {
|
|
hadBanPeerId = true
|
|
banPeer = nil
|
|
}
|
|
}
|
|
if message.id.peerId.namespace != Namespaces.Peer.SecretChat {
|
|
optionsMap[id]!.insert(.forward)
|
|
}
|
|
if !message.flags.contains(.Incoming) {
|
|
optionsMap[id]!.insert(.deleteGlobally)
|
|
} else {
|
|
if channel.hasAdminRights([.canDeleteMessages]) {
|
|
optionsMap[id]!.insert(.deleteGlobally)
|
|
}
|
|
}
|
|
} else if let group = peer as? TelegramGroup {
|
|
if message.id.peerId.namespace != Namespaces.Peer.SecretChat {
|
|
optionsMap[id]!.insert(.forward)
|
|
if message.flags.contains(.Incoming) {
|
|
optionsMap[id]!.insert(.report)
|
|
}
|
|
}
|
|
optionsMap[id]!.insert(.deleteLocally)
|
|
if !message.flags.contains(.Incoming) {
|
|
optionsMap[id]!.insert(.deleteGlobally)
|
|
} else {
|
|
switch group.role {
|
|
case .creator, .admin:
|
|
optionsMap[id]!.insert(.deleteGlobally)
|
|
case .member:
|
|
break
|
|
}
|
|
}
|
|
} else if let _ = peer as? TelegramUser {
|
|
if message.id.peerId.namespace != Namespaces.Peer.SecretChat {
|
|
optionsMap[id]!.insert(.forward)
|
|
}
|
|
optionsMap[id]!.insert(.deleteLocally)
|
|
if !message.flags.contains(.Incoming) {
|
|
optionsMap[id]!.insert(.deleteGlobally)
|
|
}
|
|
} else if let _ = peer as? TelegramSecretChat {
|
|
optionsMap[id]!.insert(.deleteGlobally)
|
|
} else {
|
|
assertionFailure()
|
|
}
|
|
} else {
|
|
optionsMap[id]!.insert(.deleteLocally)
|
|
}
|
|
}
|
|
|
|
if !optionsMap.isEmpty {
|
|
var reducedOptions = optionsMap.values.first!
|
|
for value in optionsMap.values {
|
|
reducedOptions.formIntersection(value)
|
|
}
|
|
return ChatAvailableMessageActions(options: reducedOptions, banAuthor: banPeer)
|
|
} else {
|
|
return ChatAvailableMessageActions(options: [], banAuthor: nil)
|
|
}
|
|
}
|
|
}
|