Various improvements

2025-12-22 14:20:20 +00:00 · 2024-03-25 20:43:34 +04:00
parent b7c18980f6
commit 24d7deb3e8
43 changed files with 2076 additions and 1055 deletions
--- a/submodules/TelegramCore/Sources/TelegramEngine/Peers/ChannelAds.swift
+++ b/submodules/TelegramCore/Sources/TelegramEngine/Peers/ChannelAds.swift
@@ -0,0 +1,63 @@
+import Foundation
+import Postbox
+import SwiftSignalKit
+import TelegramApi
+import MtProtoKit
+
+public enum ChannelRestrictAdMessagesValue {
+    case restrict(minCpm: Int32?)
+    case unrestrict
+}
+
+public enum ChannelRestrictAdMessagesError {
+    case generic
+}
+
+func _internal_updateChannelRestrictAdMessages(account: Account, peerId: PeerId, value: ChannelRestrictAdMessagesValue) -> Signal<Never, ChannelRestrictAdMessagesError> {
+    return account.postbox.transaction { transaction -> Peer? in
+        return transaction.getPeer(peerId)
+    }
+    |> castError(ChannelRestrictAdMessagesError.self)
+    |> mapToSignal { peer in
+        guard let peer = peer, let inputChannel = apiInputChannel(peer) else {
+            return .fail(.generic)
+        }
+        
+        var flags: Int32 = 0
+        var minCpm: Int32?
+        var restricted = false
+        if case let .restrict(minCpmValue) = value {
+            if let minCpmValue {
+                flags |= (1 << 0)
+                minCpm = minCpmValue
+            }
+            restricted = true
+        }
+        
+        return account.network.request(Api.functions.channels.restrictSponsoredMessages(flags: flags, channel: inputChannel, restricted: restricted ? .boolTrue : .boolFalse, minCpm: minCpm))
+            |> `catch` { error -> Signal<Api.Updates, ChannelRestrictAdMessagesError> in
+                return .fail(.generic)
+            }
+            |> mapToSignal { updates -> Signal<Never, ChannelRestrictAdMessagesError> in
+                account.stateManager.addUpdates(updates)
+                return account.postbox.transaction { transaction -> Void in
+                    transaction.updatePeerCachedData(peerIds: [peerId], update: { peerId, currentData in
+                        if let currentData = currentData as? CachedChannelData {
+                            var flags = currentData.flags
+                            if restricted {
+                                flags.insert(.adsRestricted)
+                            } else {
+                                flags.remove(.adsRestricted)
+                            }
+                            return currentData.withUpdatedFlags(flags).withUpdatedAdsMinCpm(minCpm)
+                        } else {
+                            return currentData
+                        }
+                    })
+                } 
+                |> castError(ChannelRestrictAdMessagesError.self)
+                |> ignoreValues
+        }
+        
+    }
+}