-
Notifications
You must be signed in to change notification settings - Fork 1
/
prepare-sapling.ts
120 lines (110 loc) · 3.42 KB
/
prepare-sapling.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
import {
TezosSaplingProtocol,
isHex,
NetworkType,
TezosContractCall,
TezosAddress,
TezosProtocolNetwork,
MichelsonList,
MichelsonOption,
MichelsonPair,
MichelsonBytes,
MichelsonString,
TezosTransactionParameters,
TezosSaplingWrappedTransaction,
isMichelinePrimitive,
} from "airgap-coin-lib"
import BigNumber from "bignumber.js"
import {
TezosSaplingProtocolOptions,
TezosShieldedTezProtocolConfig,
} from "./TezosSaplingProtocolOptions"
// An example sapling contract, to be removed
export class TezosShieldedTezProtocol extends TezosSaplingProtocol {
constructor(
options: TezosSaplingProtocolOptions = new TezosSaplingProtocolOptions(
new TezosProtocolNetwork(
"Florencenet",
NetworkType.TESTNET,
"https://tezos-florencenet-node.prod.gke.papers.tech"
),
new TezosShieldedTezProtocolConfig()
)
) {
super(options)
}
public async prepareContractCalls(
transactions: TezosSaplingWrappedTransaction[]
): Promise<TezosContractCall[]> {
const balances: BigNumber[] = transactions.map(
(transaction: TezosSaplingWrappedTransaction) =>
isHex(transaction.signed)
? this.encoder.decodeBalanceFromTransaction(
Buffer.from(transaction.signed, "hex")
)
: new BigNumber(0)
)
const callAmount: BigNumber = balances.reduce(
(sum: BigNumber, next: BigNumber) =>
next.isNegative() ? sum.plus(next.negated()) : sum,
new BigNumber(0)
)
const contractCall: TezosContractCall =
await this.contract.createContractCall(
"default",
transactions.map((transaction: TezosSaplingWrappedTransaction) => [
transaction.signed,
transaction.unshieldTarget?.getValue(),
]),
callAmount
)
return [contractCall]
}
public async parseParameters(
parameters: TezosTransactionParameters
): Promise<TezosSaplingWrappedTransaction[]> {
if (parameters.entrypoint === "default") {
try {
const callArgumentsList = MichelsonList.from(
parameters.value,
(pairJSON) =>
MichelsonPair.from(
pairJSON,
undefined,
(bytesJSON) => MichelsonBytes.from(bytesJSON, "tx"),
(optionJSON) =>
MichelsonOption.from(
optionJSON,
(valueJSON) => {
if (isMichelinePrimitive("bytes", valueJSON)) {
return MichelsonBytes.from(valueJSON)
} else if (isMichelinePrimitive("string", valueJSON)) {
return MichelsonString.from(valueJSON)
} else {
return undefined
}
},
"unshieldTarget"
)
)
).asRawValue()
return Array.isArray(callArgumentsList)
? Promise.all(
callArgumentsList.map(async (args) => ({
signed: args.tx,
unshieldTarget: args.unshieldTarget
? TezosAddress.isTzAddress(args.unshieldTarget)
? await TezosAddress.fromValue(args.unshieldTarget)
: await TezosAddress.fromRawTz(args.unshieldTarget)
: undefined,
}))
)
: []
} catch (error) {
console.error(error)
return []
}
}
return []
}
}