Source file liquidity_baking_migration.ml
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
(** This module is used to originate contracts for liquidity baking during
protocol stitching: a CPMM (constant product market making) contract and a
liquidity token FA1.2 contract, with the storage of each containing the
other's address.
The CPMM's storage contains a token address, which corresponds to tzBTC when
originated on mainnet and a reference FA1.2 contract when originated for
testing.
The test FA1.2 contract uses the same script as the liquidity token. Its
manager is initialized to the first bootstrap account. Before originating it,
we make sure we are not on mainnet by both checking for the existence of the
tzBTC contract and that the level is sufficiently low.
The Michelson and Ligo code, as well as Coq proofs, for the CPMM and
liquidity token contracts are available here:
https://gitlab.com/dexter2tz/dexter2tz/-/tree/liquidity_baking
All contracts were generated from Ligo at revision
4d10d07ca05abe0f8a5fb97d15267bf5d339d9f4 and converted to OCaml using
`octez-client convert`.
*)
open Michelson_v1_primitives
open Micheline
let null_address =
Bytes.of_string
"\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000"
let mainnet_tzBTC_address =
Contract_hash.of_b58check_exn "KT1PWx2mnDueood7fEmfbBDKx1D9BAnnXitn"
(** If token_pool, xtz_pool, or lqt_total are ever zero the CPMM will be
permanently broken. Therefore, we initialize it with the null address
registered as a liquidity provider with 1 satoshi tzBTC and 100 mutez
(roughly the current exchange rate). *)
let cpmm_init_storage ~token_address ~lqt_address =
Script_repr.lazy_expr
(Micheline.strip_locations
(Prim
( 0,
D_Pair,
[
Int (1, Z.one);
Int (2, Z.of_int 100);
Int (3, Z.of_int 100);
String (4, token_address);
String (5, lqt_address);
],
[] )))
let lqt_init_storage cpmm_address =
Script_repr.lazy_expr
(Micheline.strip_locations
(Prim
( 0,
D_Pair,
[
Seq
( 1,
[
Prim
( 2,
D_Elt,
[Bytes (3, null_address); Int (4, Z.of_int 100)],
[] );
] );
Seq (5, []);
String (6, cpmm_address);
Int (7, Z.of_int 100);
],
[] )))
let test_fa12_init_storage manager =
Script_repr.lazy_expr
(Micheline.strip_locations
(Prim
( 0,
D_Pair,
[
Seq (1, []);
Seq (2, []);
String (3, manager);
Int (4, Z.of_int 10_000);
],
[] )))
let originate ctxt address_hash ~balance script =
let open Lwt_result_syntax in
let* ctxt =
Contract_storage.raw_originate
ctxt
~prepaid_bootstrap_storage:true
address_hash
~script
in
let address = Contract_repr.Originated address_hash in
let* size = Contract_storage.used_storage_space ctxt address in
let* ctxt, _, origination_updates =
Fees_storage.burn_origination_fees
~origin:Protocol_migration
ctxt
~storage_limit:(Z.of_int64 Int64.max_int)
~payer:`Liquidity_baking_subsidies
in
let* ctxt, _, storage_updates =
Fees_storage.burn_storage_fees
~origin:Protocol_migration
ctxt
~storage_limit:(Z.of_int64 Int64.max_int)
~payer:`Liquidity_baking_subsidies
size
in
let* ctxt, transfer_updates =
Token.transfer
~origin:Protocol_migration
ctxt
`Liquidity_baking_subsidies
(`Contract address)
balance
in
let balance_updates =
origination_updates @ storage_updates @ transfer_updates
in
let result : Migration_repr.origination_result =
{
balance_updates;
originated_contracts = [address_hash];
storage_size = size;
paid_storage_size_diff = size;
}
in
return (ctxt, result)
let originate_test_fa12 ~typecheck ctxt admin =
let open Lwt_result_syntax in
let*? ctxt, fa12_address =
Contract_storage.fresh_contract_from_current_nonce ctxt
in
let script =
Script_repr.
{
code = Script_repr.lazy_expr Liquidity_baking_lqt.script;
storage =
test_fa12_init_storage (Signature.Public_key_hash.to_b58check admin);
}
in
let* script, ctxt = typecheck ctxt script in
let+ ctxt, origination_result =
originate
ctxt
fa12_address
~balance:(Tez_repr.of_mutez_exn 1_000_000L)
script
in
(ctxt, fa12_address, [origination_result])
let first_bootstrap_account =
Signature.Public_key.hash
(Signature.Public_key.of_b58check_exn
"edpkuBknW28nW72KG6RoHtYW7p12T6GKc7nAbwYX5m8Wd9sDVC9yav")
let check_tzBTC ~typecheck current_level ctxt f =
let open Lwt_result_syntax in
let*! exists =
Contract_storage.exists
ctxt
(Contract_repr.Originated mainnet_tzBTC_address)
in
if exists then
f ctxt mainnet_tzBTC_address []
else if
Compare.Int32.(current_level < 1_437_862l)
then
let* ctxt, token_address, token_result =
originate_test_fa12 ~typecheck ctxt first_bootstrap_account
in
f ctxt token_address token_result
else
return (ctxt, [])
let init ctxt ~typecheck =
let open Lwt_result_syntax in
let nonce = Operation_hash.hash_string ["Drip, drip, drip."] in
let ctxt = Raw_context.init_origination_nonce ctxt nonce in
let* ctxt = Storage.Liquidity_baking.Toggle_ema.init ctxt 0l in
let current_level =
Raw_level_repr.to_int32 (Level_storage.current ctxt).level
in
let*? ctxt, cpmm_address =
Contract_storage.fresh_contract_from_current_nonce ctxt
in
let*? ctxt, lqt_address =
Contract_storage.fresh_contract_from_current_nonce ctxt
in
let* ctxt = Storage.Liquidity_baking.Cpmm_address.init ctxt cpmm_address in
check_tzBTC
~typecheck
current_level
ctxt
(fun ctxt token_address token_result ->
let cpmm_script =
Script_repr.
{
code = Script_repr.lazy_expr Liquidity_baking_cpmm.script;
storage =
cpmm_init_storage
~token_address:(Contract_hash.to_b58check token_address)
~lqt_address:(Contract_hash.to_b58check lqt_address);
}
in
let* cpmm_script, ctxt = typecheck ctxt cpmm_script in
let lqt_script =
Script_repr.
{
code = Script_repr.lazy_expr Liquidity_baking_lqt.script;
storage = lqt_init_storage (Contract_hash.to_b58check cpmm_address);
}
in
let* lqt_script, ctxt = typecheck ctxt lqt_script in
let* ctxt, cpmm_result =
originate
ctxt
cpmm_address
~balance:(Tez_repr.of_mutez_exn 100L)
cpmm_script
in
let+ ctxt, lqt_result =
originate ctxt lqt_address ~balance:Tez_repr.zero lqt_script
in
let ctxt = Raw_context.unset_origination_nonce ctxt in
(ctxt, [cpmm_result; lqt_result] @ token_result))