-
Notifications
You must be signed in to change notification settings - Fork 9
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #13 from tonkeeper/feature/v5r1-wallet
Add v5r1 contract
- Loading branch information
Showing
3 changed files
with
196 additions
and
20 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,135 @@ | ||
| import Foundation | ||
| import BigInt | ||
| import TweetNacl | ||
|
|
||
| public struct WalletIdBeta { | ||
| public let walletVersion: Int8 = 0 | ||
| public let subwalletNumber: Int32 = 0 | ||
| public let networkGlobalId: Int32 | ||
| public let workchain: Int8 | ||
|
|
||
| public init(networkGlobalId: Int32, workchain: Int8) { | ||
| self.networkGlobalId = networkGlobalId | ||
| self.workchain = workchain | ||
| } | ||
| } | ||
|
|
||
| /// WARNING: WalletW5 contract is still in beta. use at your own risk | ||
| public class WalletV5Beta: WalletV5BetaContract { | ||
| public init(seqno: Int64 = 0, | ||
| workchain: Int8 = 0, | ||
| publicKey: Data, | ||
| walletId: WalletIdBeta, | ||
| plugins: Set<Address> = [] | ||
| ) { | ||
| let code = try! Cell.fromBase64(src: "te6cckEBAQEAIwAIQgLkzzsvTG1qYeoPK1RH0mZ4WyavNjfbLe7mvNGqgm80Eg3NjhE=" | ||
| ) | ||
| super.init(code:code, seqno: seqno, workchain: workchain, publicKey: publicKey, walletId: walletId, plugins: plugins) | ||
| } | ||
| } | ||
|
|
||
| /// Internal WalletV5 implementation. Use specific revision `WalletV5R1` instead. | ||
| public class WalletV5BetaContract: WalletContract { | ||
| public let seqno: Int64 | ||
| public let workchain: Int8 | ||
| public let publicKey: Data | ||
| public let walletId: WalletIdBeta | ||
| public let plugins: Set<Address> | ||
| public let code: Cell | ||
|
|
||
| fileprivate init(code: Cell, | ||
| seqno: Int64 = 0, | ||
| workchain: Int8 = 0, | ||
| publicKey: Data, | ||
| walletId: WalletIdBeta, | ||
| plugins: Set<Address> = [] | ||
| ) { | ||
| self.code = code | ||
| self.seqno = seqno | ||
| self.workchain = workchain | ||
| self.publicKey = publicKey | ||
|
|
||
| self.walletId = walletId | ||
|
|
||
| self.plugins = plugins | ||
| } | ||
|
|
||
| func storeWalletId() -> Builder { | ||
| return try! Builder() | ||
| .store(int: self.walletId.networkGlobalId, bits: 32) | ||
| .store(int: self.walletId.workchain, bits: 8) | ||
| .store(uint: self.walletId.walletVersion, bits: 8) | ||
| .store(uint: self.walletId.subwalletNumber, bits: 32) | ||
| } | ||
|
|
||
| public var stateInit: StateInit { | ||
| let data = try! Builder() | ||
| .store(uint: 0, bits: 33) // initial seqno = 0 | ||
| .store(self.storeWalletId()) | ||
| .store(data: publicKey) | ||
| .store(bit: 0) | ||
| .endCell() | ||
|
|
||
| return StateInit(code: self.code, data: data) | ||
| } | ||
|
|
||
| func pluginsCompact() -> Set<CompactAddress> { | ||
| Set(self.plugins.map{ a in CompactAddress(a) }) | ||
| } | ||
|
|
||
| /* | ||
| out_list_empty$_ = OutList 0; | ||
| out_list$_ {n:#} prev:^(OutList n) action:OutAction | ||
| = OutList (n + 1); | ||
| */ | ||
| private func storeOutList(messages: [MessageRelaxed], sendMode: UInt64) throws -> Builder { | ||
|
|
||
| var latestCell = Builder() | ||
| for message in messages { | ||
| latestCell = try Builder() | ||
| .store(uint: OpCodes.OUT_ACTION_SEND_MSG_TAG, bits: 32) | ||
| .store(uint: sendMode, bits: 8) | ||
| .store(ref: latestCell) | ||
| .store(ref: try Builder().store(message)) | ||
| } | ||
|
|
||
| return latestCell | ||
| } | ||
|
|
||
| private func storeOutListExtended(messages: [MessageRelaxed], sendMode: UInt64) throws -> Builder { | ||
| try Builder() | ||
| .store(uint: 0, bits: 1) | ||
| .store(ref: self.storeOutList(messages: messages, sendMode: sendMode)) | ||
| } | ||
|
|
||
| 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") | ||
| } | ||
|
|
||
| let signingMessage = try Builder() | ||
| .store(uint: messageType.opCode, bits: 32) | ||
| .store(self.storeWalletId()) | ||
|
|
||
| if (args.seqno == 0) { | ||
| // 32 bits with 1 | ||
| for _ in 0..<4 { | ||
| try signingMessage.store(uint: 0xFFFFFFFF, bits: 8) | ||
| } | ||
| } else { | ||
| let defaultTimeout = UInt64(Date().timeIntervalSince1970) + 60 // Default timeout: 60 seconds | ||
| try signingMessage.store(uint: args.timeout ?? defaultTimeout, bits: 32) | ||
| } | ||
|
|
||
| try signingMessage | ||
| .store(uint: args.seqno, bits: 32) | ||
| .store( | ||
| self.storeOutListExtended( | ||
| messages: args.messages, | ||
| sendMode: UInt64(args.sendMode.rawValue) | ||
| ) | ||
| ) | ||
|
|
||
| return WalletTransfer(signingMessage: signingMessage, signaturePosition: .tail) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters