Refactoring

2025-12-22 22:25:57 +00:00 · 2021-03-24 02:56:24 +04:00
parent 50fe822e85
commit 2f9189d220
39 changed files with 16 additions and 468 deletions
--- a/submodules/TelegramCore/Sources/ApiUtils/TelegramChannel.swift
+++ b/submodules/TelegramCore/Sources/ApiUtils/TelegramChannel.swift
@@ -0,0 +1,170 @@
+import Foundation
+import Postbox
+
+import SyncCore
+
+public enum TelegramChannelPermission {
+    case sendMessages
+    case pinMessages
+    case inviteMembers
+    case editAllMessages
+    case deleteAllMessages
+    case banMembers
+    case addAdmins
+    case changeInfo
+    case canBeAnonymous
+    case manageCalls
+}
+
+public extension TelegramChannel {
+    func hasPermission(_ permission: TelegramChannelPermission) -> Bool {
+        if self.flags.contains(.isCreator) {
+            if case .canBeAnonymous = permission {
+                if let adminRights = self.adminRights {
+                    return adminRights.rights.contains(.canBeAnonymous)
+                } else {
+                    return false
+                }
+            }
+            return true
+        }
+        switch permission {
+            case .sendMessages:
+                if case .broadcast = self.info {
+                    if let adminRights = self.adminRights {
+                        return adminRights.rights.contains(.canPostMessages)
+                    } else {
+                        return false
+                    }
+                } else {
+                    if let _ = self.adminRights {
+                        return true
+                    }
+                    if let bannedRights = self.bannedRights, bannedRights.flags.contains(.banSendMessages) {
+                        return false
+                    }
+                    if let defaultBannedRights = self.defaultBannedRights, defaultBannedRights.flags.contains(.banSendMessages) {
+                        return false
+                    }
+                    return true
+                }
+            case .pinMessages:
+                if case .broadcast = self.info {
+                    if let adminRights = self.adminRights {
+                        return adminRights.rights.contains(.canPinMessages) || adminRights.rights.contains(.canEditMessages)
+                    } else {
+                        return false
+                    }
+                } else {
+                    if let adminRights = self.adminRights, adminRights.rights.contains(.canPinMessages) {
+                        return true
+                    }
+                    if let bannedRights = self.bannedRights, bannedRights.flags.contains(.banPinMessages) {
+                        return false
+                    }
+                    if let defaultBannedRights = self.defaultBannedRights, defaultBannedRights.flags.contains(.banPinMessages) {
+                        return false
+                    }
+                    return true
+                }
+            case .inviteMembers:
+                if case .broadcast = self.info {
+                    if let adminRights = self.adminRights {
+                        return adminRights.rights.contains(.canInviteUsers)
+                    } else {
+                        return false
+                    }
+                } else {
+                    if let adminRights = self.adminRights, adminRights.rights.contains(.canInviteUsers) {
+                        return true
+                    }
+                    if let bannedRights = self.bannedRights, bannedRights.flags.contains(.banAddMembers) {
+                        return false
+                    }
+                    if let defaultBannedRights = self.defaultBannedRights, defaultBannedRights.flags.contains(.banAddMembers) {
+                        return false
+                    }
+                    return true
+                }
+            case .editAllMessages:
+                if let adminRights = self.adminRights, adminRights.rights.contains(.canEditMessages) {
+                    return true
+                }
+                return false
+            case .deleteAllMessages:
+                if let adminRights = self.adminRights, adminRights.rights.contains(.canDeleteMessages) {
+                    return true
+                }
+                return false
+            case .banMembers:
+                if let adminRights = self.adminRights, adminRights.rights.contains(.canBanUsers) {
+                    return true
+                }
+                return false
+            case .changeInfo:
+                if case .broadcast = self.info {
+                    if let adminRights = self.adminRights {
+                        return adminRights.rights.contains(.canChangeInfo)
+                    } else {
+                        return false
+                    }
+                } else {
+                    if let adminRights = self.adminRights, adminRights.rights.contains(.canChangeInfo) {
+                        return true
+                    }
+                    if let bannedRights = self.bannedRights, bannedRights.flags.contains(.banChangeInfo) {
+                        return false
+                    }
+                    if let defaultBannedRights = self.defaultBannedRights, defaultBannedRights.flags.contains(.banChangeInfo) {
+                        return false
+                    }
+                    return true
+                }
+            case .addAdmins:
+                if let adminRights = self.adminRights, adminRights.rights.contains(.canAddAdmins) {
+                    return true
+                }
+                return false
+            case .manageCalls:
+                if let adminRights = self.adminRights, adminRights.rights.contains(.canManageCalls) {
+                    return true
+                }
+                return false
+            case .canBeAnonymous:
+                if let adminRights = self.adminRights, adminRights.rights.contains(.canBeAnonymous) {
+                    return true
+                }
+                return false
+        }
+    }
+    
+    func hasBannedPermission(_ rights: TelegramChatBannedRightsFlags) -> (Int32, Bool)? {
+        if self.flags.contains(.isCreator) {
+            return nil
+        }
+        if let _ = self.adminRights {
+            return nil
+        }
+        if let defaultBannedRights = self.defaultBannedRights, defaultBannedRights.flags.contains(rights) {
+            return (Int32.max, false)
+        }
+        if let bannedRights = self.bannedRights, bannedRights.flags.contains(rights) {
+            return (bannedRights.untilDate, true)
+        }
+        return nil
+    }
+    
+    var isRestrictedBySlowmode: Bool {
+        if self.flags.contains(.isCreator) {
+            return false
+        }
+        if let _ = self.adminRights {
+            return false
+        }
+        if case let .group(group) = self.info {
+            return group.flags.contains(.slowModeEnabled)
+        } else {
+            return false
+        }
+    }
+}