|
| 1 | +use starknet::ContractAddress; |
1 | 2 | #[starknet::interface]
|
2 | 3 | trait ITokenWrapper<TContractState> {
|
3 | 4 | fn transfer_token(
|
4 | 5 | ref self: TContractState,
|
5 |
| - address: starknet::ContractAddress, |
6 |
| - selector: felt252, |
7 |
| - calldata: Array<felt252> |
| 6 | + address: ContractAddress, |
| 7 | + sender: ContractAddress, |
| 8 | + recipient: ContractAddress, |
| 9 | + amount: u256 |
8 | 10 | ) -> bool;
|
9 | 11 | }
|
10 | 12 |
|
11 | 13 | #[starknet::contract]
|
12 | 14 | mod TokenWrapper {
|
13 | 15 | use super::ITokenWrapper;
|
| 16 | + use serde::Serde; |
14 | 17 | use starknet::SyscallResultTrait;
|
| 18 | + use starknet::ContractAddress; |
15 | 19 |
|
16 | 20 | #[storage]
|
17 | 21 | struct Storage {}
|
18 | 22 |
|
19 | 23 | impl TokenWrapper of ITokenWrapper<ContractState> {
|
20 | 24 | fn transfer_token(
|
21 | 25 | ref self: ContractState,
|
22 |
| - address: starknet::ContractAddress, |
23 |
| - selector: felt252, |
24 |
| - calldata: Array<felt252> |
| 26 | + address: ContractAddress, |
| 27 | + sender: ContractAddress, |
| 28 | + recipient: ContractAddress, |
| 29 | + amount: u256 |
25 | 30 | ) -> bool {
|
26 |
| - let mut res = starknet::call_contract_syscall(address, selector, calldata.span()) |
| 31 | + let mut call_data: Array<felt252> = ArrayTrait::new(); |
| 32 | + Serde::serialize(@sender, ref call_data); |
| 33 | + Serde::serialize(@recipient, ref call_data); |
| 34 | + Serde::serialize(@amount, ref call_data); |
| 35 | + let mut res = starknet::call_contract_syscall( |
| 36 | + address, selector!("transferFrom"), call_data.span() |
| 37 | + ) |
27 | 38 | .unwrap_syscall();
|
28 | 39 | Serde::<bool>::deserialize(ref res).unwrap()
|
29 | 40 | }
|
30 | 41 | }
|
31 | 42 | }
|
| 43 | + |
0 commit comments