Add maxMessages constant for WalletContract

tonkeeper · Dec 17, 2024 · 6fa4db2 · 6fa4db2
1 parent b4cd884
commit 6fa4db2
Show file tree

Hide file tree

Showing 7 changed files with 24 additions and 10 deletions.
diff --git a/Source/TonSwift/Wallets/WalletContract.swift b/Source/TonSwift/Wallets/WalletContract.swift
@@ -4,6 +4,8 @@ import Foundation
 /// All wallets implement a compatible interface for sending messages
 public protocol WalletContract: Contract {
     func createTransfer(args: WalletTransferData, messageType: MessageType) throws -> WalletTransfer
+
+    var maxMessages: Int { get }
 }
 
 /// Message type (external | internal) to sign. Is using in v5 wallet contract

diff --git a/Source/TonSwift/Wallets/WalletV1.swift b/Source/TonSwift/Wallets/WalletV1.swift
@@ -11,6 +11,8 @@ public final class WalletV1: WalletContract {
     public let stateInit: StateInit
     public let publicKey: Data
     public let revision: WalletContractV1Revision
+
+    public let maxMessages: Int = 4
 
     public init(workchain: Int8, publicKey: Data, revision: WalletContractV1Revision) throws {
         self.workchain = workchain

diff --git a/Source/TonSwift/Wallets/WalletV2.swift b/Source/TonSwift/Wallets/WalletV2.swift
@@ -12,6 +12,8 @@ public final class WalletV2: WalletContract {
     public let publicKey: Data
     public let revision: WalletContractV2Revision
 
+    public let maxMessages: Int = 4
+
     public init(workchain: Int8, publicKey: Data, revision: WalletContractV2Revision) throws {
         self.workchain = workchain
         self.publicKey = publicKey
@@ -34,8 +36,8 @@ public final class WalletV2: WalletContract {
     }
 
     public func createTransfer(args: WalletTransferData, messageType: MessageType = .ext) throws -> WalletTransfer {
-        guard args.messages.count <= 4 else {
-            throw TonError.custom("Maximum number of messages in a single transfer is 4")
+        guard args.messages.count <= maxMessages else {
+            throw TonError.custom("Maximum number of messages in a single transfer is \(maxMessages)")
         }
 
         let signingMessage = try Builder().store(uint: args.seqno, bits: 32)

diff --git a/Source/TonSwift/Wallets/WalletV3.swift b/Source/TonSwift/Wallets/WalletV3.swift
@@ -12,6 +12,8 @@ public final class WalletV3: WalletContract {
     public let publicKey: Data
     public let walletId: UInt64
     public let revision: WalletContractV3Revision
+
+    public var maxMessages: Int = 4
 
     public init(workchain: Int8, publicKey: Data, walletId: UInt64? = nil, revision: WalletContractV3Revision) throws {
         self.workchain = workchain
@@ -43,8 +45,8 @@ public final class WalletV3: WalletContract {
     }
 
     public func createTransfer(args: WalletTransferData, messageType: MessageType = .ext) throws -> WalletTransfer {
-        guard args.messages.count <= 4 else {
-            throw TonError.custom("Maximum number of messages in a single transfer is 4")
+        guard args.messages.count <= maxMessages else {
+            throw TonError.custom("Maximum number of messages in a single transfer is \(maxMessages)")
         }
 
         let signingMessage = try Builder().store(uint: walletId, bits: 32)

diff --git a/Source/TonSwift/Wallets/WalletV4.swift b/Source/TonSwift/Wallets/WalletV4.swift
@@ -34,6 +34,8 @@ public class WalletV4: WalletContract {
     public let walletId: UInt32
     public let plugins: Set<Address>
     public let code: Cell
+
+    public let maxMessages: Int = 4
 
     fileprivate init(code: Cell,
                      seqno: Int64 = 0,
@@ -71,8 +73,8 @@ public class WalletV4: WalletContract {
     }
 
     public func createTransfer(args: WalletTransferData, messageType: MessageType = .ext) throws -> WalletTransfer {
-        guard args.messages.count <= 4 else {
-            throw TonError.custom("Maximum number of messages in a single transfer is 4")
+        guard args.messages.count <= maxMessages else {
+            throw TonError.custom("Maximum number of messages in a single transfer is \(maxMessages)")
         }
 
         let signingMessage = try Builder().store(uint: walletId, bits: 32)

diff --git a/Source/TonSwift/Wallets/WalletV5.swift b/Source/TonSwift/Wallets/WalletV5.swift
@@ -38,6 +38,8 @@ public class WalletV5: WalletContract {
     public let walletId: WalletId
     public let plugins: Set<Address>
     public let code: Cell
+
+    public let maxMessages: Int = 255
 
     fileprivate init(code: Cell,
                      seqno: Int64 = 0,
@@ -113,8 +115,8 @@ public class WalletV5: WalletContract {
     }
 
     public func createTransfer(args: WalletTransferData, messageType: MessageType = .ext) throws -> WalletTransfer {
-        guard args.messages.count <= 255 else {
-            throw TonError.custom("Maximum number of messages in a single transfer is 255")
+        guard args.messages.count <= maxMessages else {
+            throw TonError.custom("Maximum number of messages in a single transfer is \(maxMessages)")
         }
 
         let signingMessage = try Builder()

diff --git a/Source/TonSwift/Wallets/WalletV5Beta.swift b/Source/TonSwift/Wallets/WalletV5Beta.swift
@@ -36,6 +36,8 @@ public class WalletV5BetaContract: WalletContract {
     public let walletId: WalletIdBeta
     public let plugins: Set<Address>
     public let code: Cell
+
+    public let maxMessages: Int = 255
 
     fileprivate init(code: Cell,
                      seqno: Int64 = 0,
@@ -103,8 +105,8 @@ public class WalletV5BetaContract: WalletContract {
     }
 
     public func createTransfer(args: WalletTransferData, messageType: MessageType = .ext) throws -> WalletTransfer {
-        guard args.messages.count <= 255 else {
-            throw TonError.custom("Maximum number of messages in a single transfer is 255")
+        guard args.messages.count <= maxMessages else {
+            throw TonError.custom("Maximum number of messages in a single transfer is \(maxMessages)")
         }
 
         let signingMessage = try Builder()